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 | +} | ... | ... |