aidlab/aida_back
7
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

TASK:LLM；

Browse Source
This commit is contained in:
shahaibo
2025-05-20 13:37:46 +08:00
parent e9e40e53bc
commit 16e4f7c5b5
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/main/java/com/ai/da/service/impl/LLMServiceImpl.java
View File

@@ -112,12 +112,10 @@ public class LLMServiceImpl implements LLMService {
// 2. 流式接收并累积内容 // 2. 流式接收并累积内容
// 3. 存储系统回复 // 3. 存储系统回复
int systemSeq = getNextSeq(projectId);
ChatMessage systemMessage = new ChatMessage(); ChatMessage systemMessage = new ChatMessage();
systemMessage.setRole("system"); systemMessage.setRole("system");
systemMessage.setIsImage(0); systemMessage.setIsImage(0);
systemMessage.setProjectId(projectId); systemMessage.setProjectId(projectId);
systemMessage.setSeq(systemSeq);
// systemMessage.setCreateTime(LocalDateTime.now()); // systemMessage.setCreateTime(LocalDateTime.now());
// systemMessage.setContent(responseBuilder.toString()); // systemMessage.setContent(responseBuilder.toString());
systemMessage.setAccountId(accountId); systemMessage.setAccountId(accountId);
@@ -146,11 +144,11 @@ public class LLMServiceImpl implements LLMService {
} }
}else { }else {
if (responseContentBuilder.length() != 0) { if (responseContentBuilder.length() != 0) {
systemMessage.setSeq(getNextSeq(projectId));
systemMessage.setCreateTime(LocalDateTime.now()); systemMessage.setCreateTime(LocalDateTime.now());
systemMessage.setContent(responseContentBuilder.toString()); systemMessage.setContent(responseContentBuilder.toString());
chatMessageMapper.insert(systemMessage); chatMessageMapper.insert(systemMessage);
systemMessage.setId(null); systemMessage.setId(null);
systemMessage.setSeq(getNextSeq(projectId));
responseContentBuilder = new StringBuilder(); responseContentBuilder = new StringBuilder();
} }
JSONObject toolsData = json.getJSONObject("tools_data"); JSONObject toolsData = json.getJSONObject("tools_data");
@@ -181,6 +179,7 @@ public class LLMServiceImpl implements LLMService {
} }
} }
if (responseContentBuilder.length() != 0) { if (responseContentBuilder.length() != 0) {
systemMessage.setSeq(getNextSeq(projectId));
systemMessage.setCreateTime(LocalDateTime.now()); systemMessage.setCreateTime(LocalDateTime.now());
systemMessage.setContent(responseContentBuilder.toString()); systemMessage.setContent(responseContentBuilder.toString());
chatMessageMapper.insert(systemMessage); chatMessageMapper.insert(systemMessage);