diff --git a/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/midjourney/util/MjUtil.java b/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/midjourney/util/MjUtil.java index 19ad63184..bc7ee5504 100644 --- a/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/midjourney/util/MjUtil.java +++ b/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/midjourney/util/MjUtil.java @@ -25,10 +25,10 @@ public class MjUtil { if (CharSequenceUtil.isBlank(content)) { return null; } - if (!content.contains("raw**")) { + if (!content.contains("<@")) { return mjContent.setPrompt(content); } - int rawIndex = content.indexOf("raw**") + 5; + int rawIndex = content.indexOf("<@") - 3; String prompt = content.substring(0, rawIndex).trim(); String contentTail = content.substring(rawIndex).trim(); // 检查是否存在进度条