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:56:42 +08:00
parent 16e4f7c5b5
commit 4dfee09abb
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -155,8 +155,10 @@ public class LLMServiceImpl implements LLMService {
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;
receiveCollectionElement.setMinioUrl(minioUtil.getPreSignedUrl(receiveCollectionElement.getUrl(), 24 * 60)); receiveCollectionElement.setMinioUrl(minioUtil.getPreSignedUrl(receiveCollectionElement.getUrl(), 24 * 60));
} }
} }
@@ -165,7 +167,11 @@ public class LLMServiceImpl implements LLMService {
ChatMessage systemImage = new ChatMessage(); ChatMessage systemImage = new ChatMessage();
systemImage.setRole("system"); systemImage.setRole("system");
systemImage.setIsImage(1); if (color) {
systemImage.setIsImage(1);
}else {
systemImage.setIsImage(2);
}
systemImage.setProjectId(projectId); systemImage.setProjectId(projectId);
systemImage.setSeq(getNextSeq(projectId)); systemImage.setSeq(getNextSeq(projectId));
systemImage.setCreateTime(LocalDateTime.now()); systemImage.setCreateTime(LocalDateTime.now());