huangge1199/vue-pro
1
0
Fork 0
mirror of https://gitee.com/huangge1199_admin/vue-pro.git synced 2025-01-31 09:30:05 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

增加contentparse 工具类

Browse Source
This commit is contained in:
cherishsince 2024-04-06 20:32:08 +08:00
parent 1b02733c10
commit b38c94847e
1 changed files with 68 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

68
yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/midjourney/util/MjUtil.java Normal file
View File

@ -0,0 +1,68 @@
package cn.iocoder.yudao.framework.ai.midjourney.util;
import cn.hutool.core.text.CharSequenceUtil;
import cn.iocoder.yudao.framework.ai.midjourney.MjMessage;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
/**
* author: fansili
* time: 2024/4/6 19:00
*/
public class MjUtil {
/**
* content正则匹配prompt和进度.
*/
public static final String CONTENT_REGEX = ".*?\\*\\*(.*?)\\*\\*.+<@\\d+> \\((.*?)\\)";
public static final String CONTENT_PROGRESS_REGEX = "\\(([^)]*)\\)";
public static MjMessage.Content parseContent(String content) {
// "**南极应该是什么样子？ --v 6.0 --style raw** - <@972721304891453450> (32%) (fast, stealth)",
// "**南极应该是什么样子？ --v 6.0 --style raw** - <@972721304891453450> (fast, stealth)"
MjMessage.Content mjContent = new MjMessage.Content();
if (CharSequenceUtil.isBlank(content)) {
return null;
}
if (!content.contains("raw**")) {
return mjContent.setPrompt(content);
}
int rawIndex = content.indexOf("raw**") + 5;
String prompt = content.substring(0, rawIndex).trim();
String contentTail = content.substring(rawIndex).trim();
// 检查是否存在进度条
Pattern pattern = Pattern.compile(CONTENT_PROGRESS_REGEX);
Matcher matcher = pattern.matcher(contentTail);
if (contentTail.contains("%")) {
if (matcher.find()) {
// 获取第一个（也是此处唯一的）捕获组的内容
String progress = matcher.group(1);
mjContent.setProgress(progress);
}
if (matcher.find()) {
String status = matcher.group(1);
mjContent.setStatus(status);
}
} else {
if (matcher.find()) {
// 获取第一个（也是此处唯一的）捕获组的内容
String status = matcher.group(1);
mjContent.setStatus(status);
}
}
mjContent.setPrompt(prompt);
// tip：contentArray
return mjContent;
}
public static void main(String[] args) {
String content1 = "**南极应该是什么样子？ --v 6.0 --style raw** - <@972721304891453450> (32%) (fast, stealth)";
String content2 = "**南极应该是什么样子？ --v 6.0 --style raw** - <@972721304891453450> (fast, stealth)";
System.err.println(parseContent(content1));;
System.err.println(parseContent(content2));;
}
}