diff --git a/src/main/java/com/ai/da/service/impl/LLMServiceImpl.java b/src/main/java/com/ai/da/service/impl/LLMServiceImpl.java index 458742c5..9dc5c9a6 100644 --- a/src/main/java/com/ai/da/service/impl/LLMServiceImpl.java +++ b/src/main/java/com/ai/da/service/impl/LLMServiceImpl.java @@ -155,8 +155,10 @@ public class LLMServiceImpl implements LLMService { ReceiveDesignParam receiveDesignParam = JSONObject.parseObject(JSONObject.toJSONString(toolsData), ReceiveDesignParam.class); receiveDesignParam.setProjectId(projectId); designService.receiveDesignParams(receiveDesignParam); + boolean color = true; for (ReceiveCollectionElement receiveCollectionElement : receiveDesignParam.getReceiveCollectionElementList()) { if (!StringUtils.isEmpty(receiveCollectionElement.getUrl())) { + color = false; receiveCollectionElement.setMinioUrl(minioUtil.getPreSignedUrl(receiveCollectionElement.getUrl(), 24 * 60)); } } @@ -165,7 +167,11 @@ public class LLMServiceImpl implements LLMService { ChatMessage systemImage = new ChatMessage(); systemImage.setRole("system"); - systemImage.setIsImage(1); + if (color) { + systemImage.setIsImage(1); + }else { + systemImage.setIsImage(2); + } systemImage.setProjectId(projectId); systemImage.setSeq(getNextSeq(projectId)); systemImage.setCreateTime(LocalDateTime.now());