【代码优化】AI：完善 MidjourneyApiTests 单测，方便大家快速体验

2024-11-26 01:01:52 +08:00 · 2024-07-06 22:47:33 +08:00 · 2024-07-06 22:47:33 +08:00 · 62d5ed8395
commit 62d5ed8395
parent c5ed07a091
1 changed files with 62 additions and 0 deletions
--- a/yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/image/MidjourneyApiTests.java
+++ b/yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/image/MidjourneyApiTests.java
@ -0,0 +1,62 @@
+package cn.iocoder.yudao.framework.ai.image;
+
+import cn.iocoder.yudao.framework.ai.core.model.midjourney.api.MidjourneyApi;
+import org.junit.jupiter.api.Disabled;
+import org.junit.jupiter.api.Test;
+
+import java.util.Collections;
+import java.util.List;
+
+/**
+ * {@link MidjourneyApi} 集成测试
+ *
+ * @author 芋道源码
+ */
+public class MidjourneyApiTests {
+
+    private final MidjourneyApi midjourneyApi = new MidjourneyApi(
+            "https://api.holdai.top/mj",
+            "sk-dZEPiVaNcT3FHhef51996bAa0bC74806BeAb620dA5Da10Bf",
+            null);
+
+    @Test
+    @Disabled
+    public void testImagine() {
+        // 准备参数
+        MidjourneyApi.ImagineRequest request = new MidjourneyApi.ImagineRequest(null,
+                "生成一个小猫，可爱的", null,
+                MidjourneyApi.ImagineRequest.buildState(512, 512, "6.0", MidjourneyApi.ModelEnum.MIDJOURNEY.getModel()));
+
+        // 方法调用
+        MidjourneyApi.SubmitResponse response = midjourneyApi.imagine(request);
+        // 打印结果
+        System.out.println(response);
+    }
+
+    @Test
+    @Disabled
+    public void testAction() {
+        // 准备参数
+        MidjourneyApi.ActionRequest request = new MidjourneyApi.ActionRequest("1720277033455953",
+                "MJ::JOB::upsample::1::ee267661-ee52-4ced-a530-0343ba95af3b", null);
+
+        // 方法调用
+        MidjourneyApi.SubmitResponse response = midjourneyApi.action(request);
+        // 打印结果
+        System.out.println(response);
+    }
+
+    @Test
+    @Disabled
+    public void testGetTaskList() {
+        // 准备参数。该参数可以通过 MidjourneyApi.SubmitResponse 的 result 获取
+//        String taskId = "1720277033455953";
+        String taskId = "1720277214045971";
+
+        // 方法调用
+        List<MidjourneyApi.Notify> taskList = midjourneyApi.getTaskList(Collections.singletonList(taskId));
+        // 打印结果
+        System.out.println(taskList);
+    }
+
+}