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

fix(integration-tests) fix test fixture

Browse Source 
Signed-off-by: Christian Tzolov <christian.tzolov@broadcom.com>
This commit is contained in:
Christian Tzolov
2025-02-14 08:49:21 +01:00
parent e4d8a9c3ef
commit 12729b0900
2 changed files with 8 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
spring-ai-integration-tests/src/test/java/org/springframework/ai/integration/tests/tool/MethodToolCallbackTests.java
View File

@@ -16,10 +16,13 @@
package org.springframework.ai.integration.tests.tool;
import java.util.List;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.condition.EnabledIfEnvironmentVariable;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.ai.chat.client.ChatClient;
import org.springframework.ai.integration.tests.TestApplication;
import org.springframework.ai.integration.tests.tool.domain.Author;
@@ -32,10 +35,6 @@ import org.springframework.ai.tool.method.MethodToolCallback;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import static org.assertj.core.api.Assertions.assertThat;
/**
@@ -96,7 +95,8 @@ public class MethodToolCallbackTests {
var content = ChatClient.builder(this.openAiChatModel)
.build()
.prompt()
.user("What authors wrote the books %s and %s available in the library?".formatted("The Hobbit", "Narnia"))
.user("What authors wrote the books %s and %s available in the library?".formatted("The Hobbit",
"The Lion, the Witch and the Wardrobe"))
.tools(tools)
.call()
.content();
@@ -108,7 +108,8 @@ public class MethodToolCallbackTests {
var content = ChatClient.builder(this.openAiChatModel)
.build()
.prompt()
.user("What authors wrote the books %s and %s available in the library?".formatted("The Hobbit", "Narnia"))
.user("What authors wrote the books %s and %s available in the library?".formatted("The Hobbit",
"The Lion, the Witch and the Wardrobe"))
.tools(ToolCallbacks.from(tools))
.call()
.content();

2
spring-ai-integration-tests/src/test/java/org/springframework/ai/integration/tests/tool/ToolCallingManagerTests.java
View File

@@ -139,7 +139,7 @@ public class ToolCallingManagerTests {
assertThat(chatResponse).isNotNull();
assertThat(chatResponse.getResult().getOutput().getText()).isNotEmpty()
.contains("His Dark Materials")
.contains("Narnia")
.contains("The Lion, the Witch and the Wardrob")
.contains("The Hobbit")
.contains("The Lord of The Rings")
.contains("The Silmarillion");