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

This commit is contained in:
YunaiV 2024-07-06 22:47:33 +08:00
parent c5ed07a091
commit 62d5ed8395
1 changed files with 62 additions and 0 deletions

View File

@ -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);
}
}