Fix tools test configuration

spring-projects · Jan 26, 2025 · 39be9ae · 39be9ae
1 parent e7d6413
commit 39be9ae
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 5 deletions.
diff --git a/...-openai/src/test/java/org/springframework/ai/openai/chat/OpenAiCompatibleChatModelIT.java b/...-openai/src/test/java/org/springframework/ai/openai/chat/OpenAiCompatibleChatModelIT.java
@@ -57,10 +57,13 @@ static Stream<ChatModel> openAiCompatibleApis() {
 
 		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/...st/java/org/springframework/ai/autoconfigure/openai/tool/FunctionCallbackInPrompt2IT.java b/...st/java/org/springframework/ai/autoconfigure/openai/tool/FunctionCallbackInPrompt2IT.java
@@ -138,7 +138,11 @@ void streamingFunctionCallTest() {
 			// @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