diff --git a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java index 8c25c7fc..337ef4b6 100644 --- a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java +++ b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java @@ -1221,7 +1221,7 @@ public class GenerateServiceImpl extends ServiceImpl i String style = generateDTO.getText().substring(0, firstCommaIndex).trim(); String prompt = generateDTO.getText().substring(firstCommaIndex + 1).trim(); - prompt = getPrintboardPrompt(style, prompt,modelName); + prompt = getPrintboardPrompt(style, prompt,modelName,isUseImage); modelAndPromptMap.put(ModelConstants.PROMPT, prompt); @@ -1570,7 +1570,7 @@ public class GenerateServiceImpl extends ServiceImpl i } - private String getPrintboardPrompt(String style, String userInput, String modelName) { + private String getPrintboardPrompt(String style, String userInput, String modelName, boolean isUseImage) { String systemPrompt = null; String prompt; @@ -1601,7 +1601,11 @@ public class GenerateServiceImpl extends ServiceImpl i } if (userInput == null || userInput.trim().isEmpty()) { - throw new BusinessException("prompt null"); + if (isUseImage){ + prompt = "Theme: Image content" + "\nRequirement: " + systemPrompt; + }else { + throw new BusinessException("prompt null"); + } } else { prompt = "Theme: " + userInput.trim() + "\nRequirement: " + systemPrompt; }