hhyykk/ipms-sjy
1
1
Fork 0
mirror of https://gitee.com/hhyykk/ipms-sjy.git synced 2025-10-30 01:38:43 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

修改test，增加before初始化

Browse Source
This commit is contained in:
cherishsince
2024-04-05 10:16:46 +08:00
parent 44e44dc4bb
commit dfbd75b53c
2 changed files with 14 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -3,6 +3,7 @@ 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.Before;
import org.junit.Test;
import java.io.File;
@@ -18,16 +19,20 @@ import java.util.Map;
*/
public class MjTests {
@Test
public void mjImage() {
String token = "NTY5MDY4NDAxNzEyOTU1Mzky.G4-Fu0.MzD-7ll-ElbXTTgDPHF-WS_UyhMAfbKN3WyyBc";
private MidjourneyConfig midjourneyConfig;
@Before
public void setup() {
String token = "OTcwNDc3NzQxMjUyMTY5NzI4.GJcVxa.VrzMii8dsHOJAPZn4Mw8GuEo7_nIUJij9JIHD4";
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);
this.midjourneyConfig = new MidjourneyConfig(token, "1225414986084388926", "1225414986587832385", requestTemplates);
}
@Test
public void mjImage() {
MjImagineInteractions mjImagineInteractions = new MjImagineInteractions(midjourneyConfig);
mjImagineInteractions.execute("童话里应该是什么样子？");
}

8
yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/image/MjWebSocketTests.java
View File

@@ -2,8 +2,8 @@ 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.demo.wss.user.MjMessageListener;
import cn.iocoder.yudao.framework.ai.midjourney.demo.wss.user.MjWebSocketStarter;
import cn.iocoder.yudao.framework.ai.midjourney.webSocket.listener.MjMessageListener;
import cn.iocoder.yudao.framework.ai.midjourney.webSocket.MjWebSocketStarter;
import org.junit.Before;
import org.junit.Test;
@@ -23,13 +23,13 @@ public class MjWebSocketTests {
@Before
public void setup() {
String token = "OTcwNDc3NzQxMjUyMTY5NzI4.GJcVxa.VrzMii8dsHOJAPZn4Mw8GuEo7_nIUJij9JIHD4";
String token = "OTcyNzIxMzA0ODkxNDUzNDUw.G_vMOz.BO_Q0sXAD80u5ZKIHPNYDTRX_FgeKL3cKFc53I";
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));
}
this.midjourneyConfig = new MidjourneyConfig(token, "1225414986084388926", "1225414986587832385", requestTemplates);
this.midjourneyConfig = new MidjourneyConfig(token, "1225608134878302329", "1225608134878302332", requestTemplates);
}
@Test