Spring/spring-ai
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix tools test configuration

Browse Source
This commit is contained in:
Christian Tzolov
2025-01-26 07:28:51 +01:00
parent e7d6413411
commit 39be9aeba2
2 changed files with 12 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
models/spring-ai-openai/src/test/java/org/springframework/ai/openai/chat/OpenAiCompatibleChatModelIT.java
View File

@@ -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(

6
spring-ai-spring-boot-autoconfigure/src/test/java/org/springframework/ai/autoconfigure/openai/tool/FunctionCallbackInPrompt2IT.java
View File

@@ -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