fix：【ai】单测在 spring-ai 1.1.0 版本下的报错

YunaiV
1 parent 30c4be14
Showing 5 changed files with 10 additions and 9 deletions
yudao-module-ai/src/test/java/cn/iocoder/yudao/module/ai/framework/ai/core/model/chat/AnthropicChatModelTest.java
yudao-module-ai/src/test/java/cn/iocoder/yudao/module/ai/framework/ai/core/model/chat/LlamaChatModelTests.java
yudao-module-ai/src/test/java/cn/iocoder/yudao/module/ai/framework/ai/core/model/chat/OllamaChatModelTests.java
yudao-module-ai/src/test/java/cn/iocoder/yudao/module/ai/framework/ai/core/model/chat/ZhiPuAiChatModelTests.java
yudao-module-ai/src/test/java/cn/iocoder/yudao/module/ai/framework/ai/core/websearch/AiBoChaWebSearchClientTest.java
@@ -28,7 +28,7 @@ public class AnthropicChatModelTest {
                     .baseUrl("https://aihubmix.com")
                     .build())
             .defaultOptions(AnthropicChatOptions.builder()
-                    .model(AnthropicApi.ChatModel.CLAUDE_SONNET_4)
+                    .model(AnthropicApi.ChatModel.CLAUDE_SONNET_4_5)
                     .temperature(0.7)
                     .maxTokens(4096)
                     .build())
@@ -70,7 +70,7 @@ public class AnthropicChatModelTest {
         List<Message> messages = new ArrayList<>();
         messages.add(new UserMessage("thkinking 下，1+1 为什么等于 2 "));
         AnthropicChatOptions options = AnthropicChatOptions.builder()
-                .model(AnthropicApi.ChatModel.CLAUDE_SONNET_4)
+                .model(AnthropicApi.ChatModel.CLAUDE_SONNET_4_5)
                 .thinking(AnthropicApi.ThinkingType.ENABLED, 3096)
                 .temperature(1D)
                 .build();
@@ -9,8 +9,8 @@ import org.springframework.ai.chat.model.ChatResponse;
 import org.springframework.ai.chat.prompt.Prompt;
 import org.springframework.ai.ollama.OllamaChatModel;
 import org.springframework.ai.ollama.api.OllamaApi;
+import org.springframework.ai.ollama.api.OllamaChatOptions;
 import org.springframework.ai.ollama.api.OllamaModel;
-import org.springframework.ai.ollama.api.OllamaOptions;
 import reactor.core.publisher.Flux;
 import java.util.ArrayList;
@@ -27,7 +27,7 @@ public class LlamaChatModelTests {
             .ollamaApi(OllamaApi.builder()
                     .baseUrl("http://127.0.0.1:11434")  // Ollama 服务地址
                     .build())
-            .defaultOptions(OllamaOptions.builder()
+            .defaultOptions(OllamaChatOptions.builder()
                     .model(OllamaModel.LLAMA3.getName()) // 模型
                     .build())
             .build();
@@ -70,7 +70,7 @@ public class LlamaChatModelTests {
         // 准备参数
         List<Message> messages = new ArrayList<>();
         messages.add(new UserMessage("详细分析下，如何设计一个电商系统？"));
-        OllamaOptions options = OllamaOptions.builder()
+        OllamaChatOptions options = OllamaChatOptions.builder()
                 .model("qwen3")
                 .build();
@@ -83,5 +83,4 @@ public class LlamaChatModelTests {
         }).then().block();
     }
-
 }
@@ -9,7 +9,7 @@ import org.springframework.ai.chat.model.ChatResponse;
 import org.springframework.ai.chat.prompt.Prompt;
 import org.springframework.ai.ollama.OllamaChatModel;
 import org.springframework.ai.ollama.api.OllamaApi;
-import org.springframework.ai.ollama.api.OllamaOptions;
+import org.springframework.ai.ollama.api.OllamaChatOptions;
 import reactor.core.publisher.Flux;
 import java.util.ArrayList;
@@ -26,7 +26,7 @@ public class OllamaChatModelTests {
             .ollamaApi(OllamaApi.builder()
                     .baseUrl("http://127.0.0.1:11434") // Ollama 服务地址
                     .build())
-            .defaultOptions(OllamaOptions.builder()
+            .defaultOptions(OllamaChatOptions.builder()
 //                    .model("qwen") // 模型（https://ollama.com/library/qwen）
                     .model("deepseek-r1") // 模型（https://ollama.com/library/deepseek-r1）
                     .build())
@@ -23,7 +23,7 @@ import java.util.List;
 public class ZhiPuAiChatModelTests {
     private final ZhiPuAiChatModel chatModel = new ZhiPuAiChatModel(
-            new ZhiPuAiApi("2f35fb6ca4ea41fab898729b7fac086c.6ESSfPcCkxaKEUlR"), // 密钥
+            ZhiPuAiApi.builder().apiKey("2f35fb6ca4ea41fab898729b7fac086c.6ESSfPcCkxaKEUlR").build(), // 密钥
             ZhiPuAiChatOptions.builder()
                     .model(ZhiPuAiApi.ChatModel.GLM_4.getName()) // 模型
                     .build()
@@ -4,6 +4,7 @@ import cn.iocoder.yudao.framework.common.util.json.JsonUtils;
 import cn.iocoder.yudao.module.ai.framework.ai.core.webserch.AiWebSearchRequest;
 import cn.iocoder.yudao.module.ai.framework.ai.core.webserch.AiWebSearchResponse;
 import cn.iocoder.yudao.module.ai.framework.ai.core.webserch.bocha.AiBoChaWebSearchClient;
+import org.junit.jupiter.api.Disabled;
 import org.junit.jupiter.api.Test;
 /**
@@ -17,6 +18,7 @@ public class AiBoChaWebSearchClientTest {
             "sk-40500e52840f4d24b956d0b1d80d9abe");
     @Test
+    @Disabled
     public void testSearch() {
         AiWebSearchRequest request = new AiWebSearchRequest()
                 .setQuery("阿里巴巴")