From 39be9aeba29be4bfab8f8a35857381d845b8c807 Mon Sep 17 00:00:00 2001 From: Christian Tzolov Date: Sun, 26 Jan 2025 07:28:51 +0100 Subject: [PATCH] Fix tools test configuration --- .../ai/openai/chat/OpenAiCompatibleChatModelIT.java | 11 +++++++---- .../openai/tool/FunctionCallbackInPrompt2IT.java | 6 +++++- 2 files changed, 12 insertions(+), 5 deletions(-) diff --git a/models/spring-ai-openai/src/test/java/org/springframework/ai/openai/chat/OpenAiCompatibleChatModelIT.java b/models/spring-ai-openai/src/test/java/org/springframework/ai/openai/chat/OpenAiCompatibleChatModelIT.java index a373eb581..56cd6fa21 100644 --- a/models/spring-ai-openai/src/test/java/org/springframework/ai/openai/chat/OpenAiCompatibleChatModelIT.java +++ b/models/spring-ai-openai/src/test/java/org/springframework/ai/openai/chat/OpenAiCompatibleChatModelIT.java @@ -57,10 +57,13 @@ public class OpenAiCompatibleChatModelIT { builder.add(new OpenAiChatModel(new OpenAiApi(System.getenv("OPENAI_API_KEY")), forModelName("gpt-3.5-turbo"))); - if (System.getenv("GROQ_API_KEY") != null) { - builder.add(new OpenAiChatModel(new OpenAiApi("https://api.groq.com/openai", System.getenv("GROQ_API_KEY")), - forModelName("llama3-8b-8192"))); - } + // (26.01.2025) Disable because the Groq API is down. TODO: Re-enable when the API + // is back up. + // if (System.getenv("GROQ_API_KEY") != null) { + // builder.add(new OpenAiChatModel(new OpenAiApi("https://api.groq.com/openai", + // System.getenv("GROQ_API_KEY")), + // forModelName("llama3-8b-8192"))); + // } if (System.getenv("OPEN_ROUTER_API_KEY") != null) { builder.add(new OpenAiChatModel( diff --git a/spring-ai-spring-boot-autoconfigure/src/test/java/org/springframework/ai/autoconfigure/openai/tool/FunctionCallbackInPrompt2IT.java b/spring-ai-spring-boot-autoconfigure/src/test/java/org/springframework/ai/autoconfigure/openai/tool/FunctionCallbackInPrompt2IT.java index f0988a39d..0eaaf4762 100644 --- a/spring-ai-spring-boot-autoconfigure/src/test/java/org/springframework/ai/autoconfigure/openai/tool/FunctionCallbackInPrompt2IT.java +++ b/spring-ai-spring-boot-autoconfigure/src/test/java/org/springframework/ai/autoconfigure/openai/tool/FunctionCallbackInPrompt2IT.java @@ -138,7 +138,11 @@ public class FunctionCallbackInPrompt2IT { // @formatter:off String content = ChatClient.builder(chatModel).build().prompt() .user("What's the weather like in San Francisco, Tokyo, and Paris?") - .functions(FunctionCallback.builder().function("CurrentWeatherService", new MockWeatherService()).description("Get the weather in location").build()) + .functions(FunctionCallback.builder() + .function("CurrentWeatherService", new MockWeatherService()) + .description("Get the weather in location") + .inputType(MockWeatherService.Request.class) + .build()) .stream().content() .collectList().block().stream().collect(Collectors.joining()); // @formatter:on