wangyf10 2 روز پیش
والد
کامیت
30b7fc28d8

+ 19 - 0
yusys-ai-common/pom.xml

@@ -0,0 +1,19 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project xmlns="http://maven.apache.org/POM/4.0.0"
+         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd">
+    <modelVersion>4.0.0</modelVersion>
+
+    <parent>
+        <groupId>cn.com.yusys.ai</groupId>
+        <artifactId>yusys-ai-platform</artifactId>
+        <version>1.0</version>
+        <relativePath>../pom.xml</relativePath>
+    </parent>
+    <artifactId>yusys-ai-common</artifactId>
+    <name>yusys-ai-common</name>
+    <description>yusys-ai-common</description>
+    <dependencies>
+
+    </dependencies>
+</project>

+ 5 - 0
yusys-ai-common/src/main/java/cn/com/yusys/ai/common/Cons.java

@@ -0,0 +1,5 @@
+package cn.com.yusys.ai.common;
+
+public class Cons {
+    public final static String HEADER_USER_ID = "x-user-id";
+}

+ 1 - 1
yusys-ai-gateway-agent/src/main/resources/application.yml

@@ -55,7 +55,7 @@ spring:
   reactor:
     context-propagation: auto
 agent:
-  session-key-in-header: sessionid
+  session-key-in-header: x-user-id
 logging:
   pattern:
     correlation: "[${spring.application.name:},%X{traceId:-},%X{spanId:-}] "

+ 5 - 0
yusys-ai-plus/yusys-ai-harness/yusys-ai-harness-core/pom.xml

@@ -17,6 +17,11 @@
 
     <dependencies>
         <dependency>
+            <groupId>cn.com.yusys.ai</groupId>
+            <artifactId>yusys-ai-common</artifactId>
+            <version>${project.version}</version>
+        </dependency>
+        <dependency>
             <groupId>org.springframework.boot</groupId>
             <artifactId>spring-boot-starter-web</artifactId>
         </dependency>

+ 2 - 1
yusys-ai-plus/yusys-ai-harness/yusys-ai-harness-core/src/main/java/cn/com/yusys/ai/plus/harness/controller/HarnessAgentRest.java

@@ -1,5 +1,6 @@
 package cn.com.yusys.ai.plus.harness.controller;
 
+import cn.com.yusys.ai.common.Cons;
 import com.fasterxml.jackson.core.JsonProcessingException;
 import com.fasterxml.jackson.databind.ObjectMapper;
 import io.agentscope.core.agent.Agent;
@@ -39,7 +40,7 @@ public class HarnessAgentRest {
     @PostMapping(value = "/harness", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
     public Flux<ServerSentEvent<String>> run(
             @RequestBody String body,
-            @RequestHeader(value = "x-user-id", required = true) String userId) throws JsonProcessingException {
+            @RequestHeader(value = Cons.HEADER_USER_ID, required = true) String userId) throws JsonProcessingException {
         RunAgentInput input = jackson2.readValue(body, RunAgentInput.class);
         RuntimeContext ctx = RuntimeContext.builder()
                 .userId(userId)          // null 也没关系 = 匿名