增加会话id

xigua0824 · Dec 19, 2023 · 18ee3b9 · 18ee3b9
1 parent 0b72bc3
commit 18ee3b9
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/src/main/java/com/beauty/aide/adaptor/TongYiAdaptor.java b/src/main/java/com/beauty/aide/adaptor/TongYiAdaptor.java
@@ -30,8 +30,7 @@ public void afterPropertiesSet() {
         AccessTokenClient tokenClient = new AccessTokenClient(adminAk, adminSk, agentKey);
         String token = tokenClient.getToken();
         BaiLianConfig config = new BaiLianConfig().setApiKey(token);
-        ApplicationClient client = new ApplicationClient(config);
-        client = new ApplicationClient(config);
+        this.client = new ApplicationClient(config);
     }
 
 
@@ -41,13 +40,14 @@ public void afterPropertiesSet() {
      * @param prompt
      * @return
      */
-    public String sendChatRequest(String prompt) {
+    public String sendChatRequest(String prompt, String uuid) {
         CompletionsRequest request = new CompletionsRequest()
                 .setAppId(appId)
+                // 上下文会话id
+                .setSessionId(uuid)
                 .setPrompt(prompt);
         CompletionsResponse response = client.completions(request);
         String content = response.getData().getText();
-        content = content.substring(0, content.lastIndexOf("<sup>"));
         return content;
     }
 

diff --git a/src/main/java/com/beauty/aide/manager/AccountManager.java b/src/main/java/com/beauty/aide/manager/AccountManager.java
@@ -88,9 +88,13 @@ public AccountDO vo2Do(AccountVO accountVO) {
      * @return
      */
     public AccountVO getLoginUser(HttpServletRequest request) {
+        if (request == null) {
+            return null;
+        }
         AccountVO accountVO = (AccountVO) request.getSession().getAttribute(UserConstant.USER_LOGIN_STATE);
         if (accountVO == null) {
-            throw new BusinessException(UserErrorCode.USER_NOT_LOGIN.getMessage());
+//            throw new BusinessException(UserErrorCode.USER_NOT_LOGIN.getMessage());
+            return null;
         }
         return accountVO;
     }