Browse Source

【增加】百度千帆 image tests

cherishsince 8 months ago
parent
commit
6b7238a79d

+ 47 - 0
yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/image/QianFanImageTests.java

@@ -0,0 +1,47 @@
+package cn.iocoder.yudao.framework.ai.image;
+
+import cn.iocoder.yudao.framework.common.util.json.JsonUtils;
+import org.junit.jupiter.api.Test;
+import org.springframework.ai.image.ImageOptionsBuilder;
+import org.springframework.ai.image.ImagePrompt;
+import org.springframework.ai.image.ImageResponse;
+import org.springframework.ai.qianfan.QianFanImageModel;
+import org.springframework.ai.qianfan.QianFanImageOptions;
+import org.springframework.ai.qianfan.api.QianFanApi;
+import org.springframework.ai.qianfan.api.QianFanImageApi;
+
+/**
+ * 百度千帆 image
+ */
+public class QianFanImageTests {
+
+    @Test
+    public void callTest() {
+        QianFanImageApi qianFanImageApi = new QianFanImageApi(
+                "ghbbvbW2t7HK7WtYmEITAupm", "njJEr5AsQ5fkB3ucYYDjiQqsOZK20SGb");
+        QianFanImageModel qianFanImageModel = new QianFanImageModel(qianFanImageApi);
+
+        QianFanImageOptions imageOptions = QianFanImageOptions.builder()
+                .withWidth(512)
+                .withHeight(512)
+                .build();
+        ImagePrompt imagePrompt = new ImagePrompt("薄涂炫酷少女头像,田野花朵盛开", imageOptions);
+        ImageResponse call = qianFanImageModel.call(imagePrompt);
+        System.err.println(JsonUtils.toJsonString(call));
+    }
+
+    @Test
+    public void call2Test() {
+        // 官方测试 test https://github.com/spring-projects/spring-ai/blob/main/models/spring-ai-qianfan/src/test/java/org/springframework/ai/qianfan/image/QianFanImageModelIT.java
+        var options = ImageOptionsBuilder.builder().withHeight(1024).withWidth(1024).build();
+        var instructions = "薄涂炫酷少女头像,田野花朵盛开";
+
+        ImagePrompt imagePrompt = new ImagePrompt(instructions, options);
+
+        QianFanImageApi qianFanImageApi = new QianFanImageApi(
+                "ghbbvbW2t7HK7WtYmEITAupm", "njJEr5AsQ5fkB3ucYYDjiQqsOZK20SGb");
+        QianFanImageModel imageModel = new QianFanImageModel(qianFanImageApi);
+        ImageResponse imageResponse = imageModel.call(imagePrompt);
+    }
+
+}