【代码优化】AI：无法适配 Spring AI 1.0.6 对百度 QianFan 的逻辑：无语！！！

YunaiV
1 parent d05a7bd5
Showing 2 changed files with 10 additions and 8 deletions
yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/chat/YiYanChatModelTests.java
yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/image/QianFanImageTests.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
@@ -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