refactor(yudao-module-llm): 优化数据回流 response 处理

- 使用 "(?s)" 标志使 "."匹配换行符,确保<think> 标签内的内容被正确替换
- 添加 trim() 方法去除回答两端的空白字符,提高数据准确性
This commit is contained in:
sunxiqing 2025-03-17 10:54:02 +08:00
parent 4b02c472a1
commit fe6bbbc633

View File

@ -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().replaceAll("<think>.*?</think>", ""));
dataRefluxDataSaveReqVO.setResponse(modelCompletionsRespVO.getAnswer().replaceAll("(?s)<think>.*?</think>", "").trim());
dataRefluxDataSaveReqVO.setMaxTokens(chatReqVO.getMaxTokens());
dataRefluxDataSaveReqVO.setTemperature(chatReqVO.getTemperature());
dataRefluxDataService.saveDataRefluxData(dataRefluxDataSaveReqVO);