【代码优化】AI:完善 MidjourneyApiTests 单测,方便大家快速体验
This commit is contained in:
parent
c5ed07a091
commit
62d5ed8395
|
@ -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);
|
||||
}
|
||||
|
||||
}
|
Loading…
Reference in New Issue