From c96d966a41622bd41d3dbbfad6062e66689d64c4 Mon Sep 17 00:00:00 2001
From: YunaiV <zhijiantianya@gmail.com>
Date: Sun, 23 Feb 2025 17:57:59 +0800
Subject: [PATCH] =?UTF-8?q?=E3=80=90=E4=BB=A3=E7=A0=81=E4=BC=98=E5=8C=96?=
 =?UTF-8?q?=E3=80=91AI=EF=BC=9A=E6=97=A0=E6=B3=95=E9=80=82=E9=85=8D=20Spri?=
 =?UTF-8?q?ng=20AI=201.0.6=20=E5=AF=B9=E7=99=BE=E5=BA=A6=20QianFan=20?=
 =?UTF-8?q?=E7=9A=84=E9=80=BB=E8=BE=91=EF=BC=9A=E6=97=A0=E8=AF=AD=EF=BC=81?=
 =?UTF-8?q?=EF=BC=81=EF=BC=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../yudao/framework/ai/chat/YiYanChatModelTests.java  | 11 ++++++-----
 .../yudao/framework/ai/image/QianFanImageTests.java   |  7 ++++---
 2 files changed, 10 insertions(+), 8 deletions(-)

diff --git a/yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/chat/YiYanChatModelTests.java b/yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/chat/YiYanChatModelTests.java
index e271decbc5..baa36d86e8 100644
--- a/yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/chat/YiYanChatModelTests.java
+++ b/yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/chat/YiYanChatModelTests.java
@@ -14,6 +14,7 @@ import reactor.core.publisher.Flux;
 import java.util.ArrayList;
 import java.util.List;
 
+// TODO @芋艿：百度千帆 API 提供了 V2 版本，目前 Spring AI 不兼容，可关键 <https://github.com/spring-projects/spring-ai/issues/2179> 进展
 /**
  * {@link QianFanChatModel} 的集成测试
  *
@@ -21,11 +22,11 @@ import java.util.List;
  */
 public class YiYanChatModelTests {
 
-    private final QianFanApi qianFanApi = new QianFanApi(
-            "qS8k8dYr2nXunagK4SSU8Xjj",
-            "pHGbx51ql2f0hOyabQvSZezahVC3hh3e");
-    private final QianFanChatModel chatModel = new QianFanChatModel(qianFanApi,
-            QianFanChatOptions.builder().model(QianFanApi.ChatModel.ERNIE_Tiny_8K.getValue()).build()
+    private final QianFanChatModel chatModel = new QianFanChatModel(
+            new QianFanApi("qS8k8dYr2nXunagK4SSU8Xjj", "pHGbx51ql2f0hOyabQvSZezahVC3hh3e"), // 密钥
+            QianFanChatOptions.builder()
+                    .model(QianFanApi.ChatModel.ERNIE_4_0_8K_Preview.getValue())
+                    .build()
     );
 
     @Test
diff --git a/yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/image/QianFanImageTests.java b/yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/image/QianFanImageTests.java
index 4c0a9eb36d..c284d8c76d 100644
--- a/yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/image/QianFanImageTests.java
+++ b/yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/image/QianFanImageTests.java
@@ -10,14 +10,15 @@ import org.springframework.ai.qianfan.api.QianFanImageApi;
 
 import static cn.iocoder.yudao.framework.ai.image.StabilityAiImageModelTests.viewImage;
 
+// TODO @芋艿：百度千帆 API 提供了 V2 版本，目前 Spring AI 不兼容，可关键 <https://github.com/spring-projects/spring-ai/issues/2179> 进展
+
 /**
  * {@link QianFanImageModel} 集成测试类
  */
 public class QianFanImageTests {
 
-    private final QianFanImageApi imageApi = new QianFanImageApi(
-            "qS8k8dYr2nXunagK4SSU8Xjj", "pHGbx51ql2f0hOyabQvSZezahVC3hh3e");
-    private final QianFanImageModel imageModel = new QianFanImageModel(imageApi);
+    private final QianFanImageModel imageModel = new QianFanImageModel(
+            new QianFanImageApi("qS8k8dYr2nXunagK4SSU8Xjj", "pHGbx51ql2f0hOyabQvSZezahVC3hh3e")); // 密钥
 
     @Test
     @Disabled