huangge1199/vue-pro
1
0
Fork 0
mirror of https://gitee.com/huangge1199_admin/vue-pro.git synced 2024-11-26 17:21:53 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

增加 mj Imagine 测试类

Browse Source
This commit is contained in:
cherishsince 2024-04-04 19:38:03 +08:00
parent f1f003b76d
commit b77e8f16e9
1 changed files with 34 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/image/MjTests.java Normal file
View File

@ -0,0 +1,34 @@
package cn.iocoder.yudao.framework.ai.image;
import cn.hutool.core.io.FileUtil;
import cn.iocoder.yudao.framework.ai.midjourney.MidjourneyConfig;
import cn.iocoder.yudao.framework.ai.midjourney.interactions.MjImagineInteractions;
import org.junit.Test;
import java.io.File;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
/**
* mj
*
* author: fansili
* time: 2024/4/4 18:59
*/
public class MjTests {
@Test
public void mjImage() {
String token = "NTY5MDY4NDAxNzEyOTU1Mzky.G4-Fu0.MzD-7ll-ElbXTTgDPHF-WS_UyhMAfbKN3WyyBc";
Map<String, String> requestTemplates = new HashMap<>();
List<File> files = FileUtil.loopFiles("/Users/fansili/projects/github/ruoyi-vue-pro/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/resources/http-body");
for (File file : files) {
requestTemplates.put(file.getName().replace(".json", ""), FileUtil.readUtf8String(file));
}
MidjourneyConfig midjourneyConfig = new MidjourneyConfig(
token, "1221445697157468200", "1221445862962630706", requestTemplates);
MjImagineInteractions mjImagineInteractions = new MjImagineInteractions(midjourneyConfig);
mjImagineInteractions.execute("童话里应该是什么样子？");
}
}