Sfoglia il codice sorgente

refactor(ai): 优化 AI聊天消息响应

zrd 3 settimane fa
parent
commit
dc22406f42

+ 3 - 3
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/chat/AiChatMessageServiceImpl.java

@@ -239,8 +239,8 @@ public class AiChatMessageServiceImpl implements AiChatMessageService {
         String apiKey = DictFrameworkUtils.parseDictDataValue("ai_key", "多轮对话");
         JSONObject requestBody = new JSONObject();
         String kcb = "";
-        if (StrUtil.isNotBlank(conversation.getDifyConversationId()) && !StrUtil.equals("0",
-                conversation.getDifyConversationId())) {
+        if (StrUtil.isNotBlank(sendReqVO.getDifyConversationId()) && !StrUtil.equals("0",
+                sendReqVO.getDifyConversationId())) {
             requestBody.set("conversation_id", conversation.getDifyConversationId());
         } else {
             kcb = getKcb(apiKey);
@@ -303,7 +303,7 @@ public class AiChatMessageServiceImpl implements AiChatMessageService {
                 userId, conversation.getRoleId(), MessageType.ASSISTANT, kcb, sendReqVO.getUseContext(),
                 knowledgeSegments, sendReqVO.getContentType(), sendReqVO.getStateId());
         
-        if (StrUtil.isBlank(conversation.getDifyConversationId())) {
+        if (StrUtil.isBlank(sendReqVO.getDifyConversationId())) {
             AiChatMessageSendRespVO.Message rs = BeanUtils.toBean(assistantMessage,
                     AiChatMessageSendRespVO.Message.class);
             rs.setDifyConversationId("0");