Commit 62d5ed8395cdb7e17d09dbe0a0a78f53a7299263
1 parent
c5ed07a0
【代码优化】AI：完善 MidjourneyApiTests 单测，方便大家快速体验
Showing
1 changed file
with
62 additions
and
0 deletions
yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/image/MidjourneyApiTests.java
0 → 100644
| 1 | +package cn.iocoder.yudao.framework.ai.image; | ||
| 2 | + | ||
| 3 | +import cn.iocoder.yudao.framework.ai.core.model.midjourney.api.MidjourneyApi; | ||
| 4 | +import org.junit.jupiter.api.Disabled; | ||
| 5 | +import org.junit.jupiter.api.Test; | ||
| 6 | + | ||
| 7 | +import java.util.Collections; | ||
| 8 | +import java.util.List; | ||
| 9 | + | ||
| 10 | +/** | ||
| 11 | + * {@link MidjourneyApi} 集成测试 | ||
| 12 | + * | ||
| 13 | + * @author 芋道源码 | ||
| 14 | + */ | ||
| 15 | +public class MidjourneyApiTests { | ||
| 16 | + | ||
| 17 | + private final MidjourneyApi midjourneyApi = new MidjourneyApi( | ||
| 18 | + "https://api.holdai.top/mj", | ||
| 19 | + "sk-dZEPiVaNcT3FHhef51996bAa0bC74806BeAb620dA5Da10Bf", | ||
| 20 | + null); | ||
| 21 | + | ||
| 22 | + @Test | ||
| 23 | + @Disabled | ||
| 24 | + public void testImagine() { | ||
| 25 | + // 准备参数 | ||
| 26 | + MidjourneyApi.ImagineRequest request = new MidjourneyApi.ImagineRequest(null, | ||
| 27 | + "生成一个小猫，可爱的", null, | ||
| 28 | + MidjourneyApi.ImagineRequest.buildState(512, 512, "6.0", MidjourneyApi.ModelEnum.MIDJOURNEY.getModel())); | ||
| 29 | + | ||
| 30 | + // 方法调用 | ||
| 31 | + MidjourneyApi.SubmitResponse response = midjourneyApi.imagine(request); | ||
| 32 | + // 打印结果 | ||
| 33 | + System.out.println(response); | ||
| 34 | + } | ||
| 35 | + | ||
| 36 | + @Test | ||
| 37 | + @Disabled | ||
| 38 | + public void testAction() { | ||
| 39 | + // 准备参数 | ||
| 40 | + MidjourneyApi.ActionRequest request = new MidjourneyApi.ActionRequest("1720277033455953", | ||
| 41 | + "MJ::JOB::upsample::1::ee267661-ee52-4ced-a530-0343ba95af3b", null); | ||
| 42 | + | ||
| 43 | + // 方法调用 | ||
| 44 | + MidjourneyApi.SubmitResponse response = midjourneyApi.action(request); | ||
| 45 | + // 打印结果 | ||
| 46 | + System.out.println(response); | ||
| 47 | + } | ||
| 48 | + | ||
| 49 | + @Test | ||
| 50 | + @Disabled | ||
| 51 | + public void testGetTaskList() { | ||
| 52 | + // 准备参数。该参数可以通过 MidjourneyApi.SubmitResponse 的 result 获取 | ||
| 53 | +// String taskId = "1720277033455953"; | ||
| 54 | + String taskId = "1720277214045971"; | ||
| 55 | + | ||
| 56 | + // 方法调用 | ||
| 57 | + List<MidjourneyApi.Notify> taskList = midjourneyApi.getTaskList(Collections.singletonList(taskId)); | ||
| 58 | + // 打印结果 | ||
| 59 | + System.out.println(taskList); | ||
| 60 | + } | ||
| 61 | + | ||
| 62 | +} |