From 4b02c472a1ab4b99aedefa4a1349808f7766e391 Mon Sep 17 00:00:00 2001
From: sunxiqing <2240398334@qq.com>
Date: Mon, 17 Mar 2025 10:37:36 +0800
Subject: [PATCH] =?UTF-8?q?fix(llm):=20=E7=A7=BB=E9=99=A4=E6=95=B0?=
 =?UTF-8?q?=E6=8D=AE=E5=9B=9E=E6=B5=81=E4=B8=AD=E7=9A=84=E6=80=9D=E8=80=83?=
 =?UTF-8?q?=E5=86=85=E5=AE=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 在保存数据回流时，从模型的回答中移除了包含 <think> 标签的内容
- 这样可以避免将不应持久化或索引的思考过程数据保存到数据库中
---
 .../llm/service/conversation/ConversationServiceImpl.java       | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/conversation/ConversationServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/conversation/ConversationServiceImpl.java
index bedaf926a..0b0242a31 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/conversation/ConversationServiceImpl.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/conversation/ConversationServiceImpl.java
@@ -532,7 +532,7 @@ public class ConversationServiceImpl implements ConversationService {
         dataRefluxDataSaveReqVO.setModelType(chatReqVO.getModelType());
         dataRefluxDataSaveReqVO.setPrompt(chatReqVO.getPrompt());
         dataRefluxDataSaveReqVO.setSystem(modelCompletionsRespVO.getSystem());
-        dataRefluxDataSaveReqVO.setResponse(modelCompletionsRespVO.getAnswer());
+        dataRefluxDataSaveReqVO.setResponse(modelCompletionsRespVO.getAnswer().replaceAll("<think>.*?</think>", ""));
         dataRefluxDataSaveReqVO.setMaxTokens(chatReqVO.getMaxTokens());
         dataRefluxDataSaveReqVO.setTemperature(chatReqVO.getTemperature());
         dataRefluxDataService.saveDataRefluxData(dataRefluxDataSaveReqVO);