Merge branch 'deepseek' of wangfan/adweb3-server into master

jeecg-module-system/jeecg-system-biz/src/main/java/org/jeecg/modules/adweb/gpt/service/impl/ChatServiceImpl.java

@@ -1,11 +1,15 @@
 package org.jeecg.modules.adweb.gpt.service.impl;
 
+import com.google.common.collect.Iterables;
 import com.google.common.collect.Lists;
 import com.unfbx.chatgpt.OpenAiStreamClient;
+import com.unfbx.chatgpt.entity.chat.BaseChatCompletion;
 import com.unfbx.chatgpt.entity.chat.ChatCompletion;
 import com.unfbx.chatgpt.entity.chat.Message;
 import com.unfbx.chatgpt.exception.BaseException;
 
+import jakarta.annotation.PostConstruct;
+
 import lombok.extern.slf4j.Slf4j;
 
 import org.apache.commons.lang3.StringUtils;
@@ -13,7 +17,6 @@ import org.apache.shiro.SecurityUtils;
 import org.jeecg.chatgpt.prop.AiChatProperties;
 import org.jeecg.common.exception.JeecgBootException;
 import org.jeecg.common.system.vo.LoginUser;
-import org.jeecg.common.util.SpringContextUtils;
 import org.jeecg.common.util.UUIDGenerator;
 import org.jeecg.modules.adweb.common.util.AdwebRedisUtil;
 import org.jeecg.modules.adweb.gpt.cache.SseEmitterCache;
@@ -53,13 +56,24 @@ public class ChatServiceImpl implements ChatService {
 
     @Autowired private IChatHistoryService chatHistoryService;
 
-    /** 防止client不能成功注入 */
-    private OpenAiStreamClient ensureClient() {
-        if (Objects.isNull(this.openAiStreamClient)) {
-            this.openAiStreamClient = SpringContextUtils.getBean(OpenAiStreamClient.class);
-            this.aiChatProperties = SpringContextUtils.getBean(AiChatProperties.class);
-        }
-        return this.openAiStreamClient;
+    // 是否使用DeepSeek模型
+    private boolean isDeepSeek = false;
+
+    //    /** 防止client不能成功注入 */
+    //    private OpenAiStreamClient ensureClient() {
+    //        if (Objects.isNull(this.openAiStreamClient)) {
+    //            this.openAiStreamClient = SpringContextUtils.getBean(OpenAiStreamClient.class);
+    //            this.aiChatProperties = SpringContextUtils.getBean(AiChatProperties.class);
+    //        }
+    //        return this.openAiStreamClient;
+    //    }
+
+    @PostConstruct
+    private void init() {
+        this.isDeepSeek =
+                Arrays.stream(BaseChatCompletion.Model.values())
+                        .map(BaseChatCompletion.Model::getName)
+                        .noneMatch(aiChatProperties.getModel()::equals);
     }
 
     @Override
@@ -148,10 +162,14 @@ public class ChatServiceImpl implements ChatService {
                 new OpenAISSEEventSourceListener(topicId, sseEmitter);
         ChatCompletion completion =
                 ChatCompletion.builder()
-                        .messages(topicContext)
+                        // 对DeepSeek模型不发送上下文
+                        .messages(
+                                this.isDeepSeek
+                                        ? Collections.singletonList(Iterables.getLast(topicContext))
+                                        : topicContext)
                         .model(aiChatProperties.getModel())
                         .build();
-        ensureClient().streamChatCompletion(completion, openAIEventSourceListener);
+        openAiStreamClient.streamChatCompletion(completion, openAIEventSourceListener);
 
         // 3. 将当前话题的context保存到Redis
         adwebRedisUtil.del(contextKey);