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 14:31:05 +08:00
parent f1adbb4da7
commit b030e9c09b
1 changed files with 25 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -152,10 +152,11 @@ public class LLMServiceImpl implements LLMService {
responseContentBuilder = new StringBuilder(); responseContentBuilder = new StringBuilder();
} }
JSONObject toolsData = json.getJSONObject("tools_data"); JSONObject toolsData = json.getJSONObject("tools_data");
if (Objects.nonNull(toolsData)) {
boolean color = true;
ReceiveDesignParam receiveDesignParam = JSONObject.parseObject(JSONObject.toJSONString(toolsData), ReceiveDesignParam.class); ReceiveDesignParam receiveDesignParam = JSONObject.parseObject(JSONObject.toJSONString(toolsData), ReceiveDesignParam.class);
receiveDesignParam.setProjectId(projectId); receiveDesignParam.setProjectId(projectId);
designService.receiveDesignParams(receiveDesignParam); designService.receiveDesignParams(receiveDesignParam);
boolean color = true;
for (ReceiveCollectionElement receiveCollectionElement : receiveDesignParam.getReceiveCollectionElementList()) { for (ReceiveCollectionElement receiveCollectionElement : receiveDesignParam.getReceiveCollectionElementList()) {
if (!StringUtils.isEmpty(receiveCollectionElement.getUrl())) { if (!StringUtils.isEmpty(receiveCollectionElement.getUrl())) {
color = false; color = false;
@@ -164,7 +165,6 @@ public class LLMServiceImpl implements LLMService {
} }
String jsonString = JSONObject.toJSONString(receiveDesignParam); String jsonString = JSONObject.toJSONString(receiveDesignParam);
json.put("tools_data", jsonString); json.put("tools_data", jsonString);
ChatMessage systemImage = new ChatMessage(); ChatMessage systemImage = new ChatMessage();
systemImage.setRole("system"); systemImage.setRole("system");
if (color) { if (color) {
@@ -178,7 +178,7 @@ public class LLMServiceImpl implements LLMService {
systemImage.setContent(JSONObject.toJSONString(receiveDesignParam.getReceiveCollectionElementList())); systemImage.setContent(JSONObject.toJSONString(receiveDesignParam.getReceiveCollectionElementList()));
systemImage.setAccountId(accountId); systemImage.setAccountId(accountId);
chatMessageMapper.insert(systemImage); chatMessageMapper.insert(systemImage);
}
emitter.send(json.toJSONString()); emitter.send(json.toJSONString());
} }
} }