Просмотр исходного кода

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

zrd 3 недель назад
Родитель
Сommit
5953ad0f64

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

@@ -328,7 +328,12 @@ public class AiChatMessageServiceImpl implements AiChatMessageService {
             return success(result);
         }).doOnComplete(() -> {
             // 忽略租户,因为 Flux 异步无法透传租户
-            assistantMessage.setContent(contentBuffer.toString());
+            if (!StrUtil.contains(contentBuffer.toString(), "><")) {
+                assistantMessage.setContent(contentBuffer.toString());
+            } else {
+                assistantMessage.setContent("选中");
+            }
+            
             // 1.2 保存消息
             //更新 创建时间 方便排序
             assistantMessage.setCreateTime(LocalDateTime.now());
@@ -337,7 +342,7 @@ public class AiChatMessageServiceImpl implements AiChatMessageService {
             AiChatConversationUpdateMyReqVO updateReqVO = new AiChatConversationUpdateMyReqVO();
             updateReqVO.setId(conversation.getId());
             updateReqVO.setDifyConversationId(difyId.toString());
-            if (!StrUtil.contains(assistantMessage.getContent(), "ecologicalValueCalculate")) {
+            if (!StrUtil.contains(assistantMessage.getContent(), "><")) {
                 updateReqVO.setTitle(StrUtil.subPre(sendReqVO.getContent(), 20));
             }
             TenantUtils.executeIgnore(() ->