add wenxinyiyan support

robinji0 · robinji0 · commit f461d9da80e4 · 2023-11-06T17:36:24.000+08:00
diff --git a/chat2db-client/src/typings/ai.ts b/chat2db-client/src/typings/ai.ts
@@ -2,7 +2,7 @@ export enum AIType {
   CHAT2DBAI = 'CHAT2DBAI',
   ZHIPUAI = 'ZHIPUAI',
   BAICHUANAI='BAICHUANAI',
-  // WENXINAI='WENXINAI',
+  WENXINAI='WENXINAI',
   // TONGYIQIANWENAI='TONGYIQIANWENAI',
   OPENAI = 'OPENAI',
   AZUREAI = 'AZUREAI',
diff --git a/chat2db-server/chat2db-server-start/src/main/resources/application.yml b/chat2db-server/chat2db-server-start/src/main/resources/application.yml
@@ -73,7 +73,7 @@ chatgpt:
   # 可以选择GPT3或GPT35
   version: GPT35
   context:
-    length: 4
+    length: 1
 
 # Print the HTTP log
 logbook:
diff --git a/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/ai/ChatController.java b/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/ai/ChatController.java
@@ -480,6 +480,9 @@ private List<FastChatMessage> getFastChatMessage(String uid, String prompt) {
     private SseEmitter chatWithWenxinAi(ChatQueryRequest queryRequest, SseEmitter sseEmitter, String uid) throws IOException {
         String prompt = buildPrompt(queryRequest);
         List<FastChatMessage> messages = getFastChatMessage(uid, prompt);
+        if (messages.size() >= 2 && messages.size() % 2 == 0) {
+            messages.remove(messages.size() - 1);
+        }
 
         buildSseEmitter(sseEmitter, uid);
 
