From b09fc5579c86a8a7c53be889e7bac55b3a989d62 Mon Sep 17 00:00:00 2001 From: cherishsince Date: Sat, 6 Apr 2024 21:04:53 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A4=84=E7=90=86=E8=A1=A8=E6=83=85=E7=AC=A6?= =?UTF-8?q?=E5=8F=B7=EF=BC=8Ccontent=E8=A7=A3=E6=9E=90=E9=97=AE=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../cn/iocoder/yudao/framework/ai/midjourney/util/MjUtil.java | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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(); // 检查是否存在进度条