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

Remove assertion that depends on unreliable response from AI Model

Browse Source
This commit is contained in:
Mark Pollack
2024-05-08 15:46:17 -04:00
parent ad527303ed
commit b774a9549d
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
spring-ai-test/src/main/java/org/springframework/ai/evaluation/BaseMemoryTest.java
View File

@@ -59,7 +59,8 @@ public class BaseMemoryTest {
var chatBotResponse1 = this.chatBot.call(promptContext);
logger.info("Response1: " + chatBotResponse1.getChatResponse().getResult().getOutput().getContent());
assertThat(chatBotResponse1.getChatResponse().getResult().getOutput().getContent()).contains("John");
// response varies too much.
// assertThat(chatBotResponse1.getChatResponse().getResult().getOutput().getContent()).contains("John");
var chatBotResponse2 = this.chatBot.call(new PromptContext(new Prompt(new String("What is my name?"))));
logger.info("Response2: " + chatBotResponse2.getChatResponse().getResult().getOutput().getContent());
@@ -88,7 +89,7 @@ public class BaseMemoryTest {
.collect(Collectors.joining());
logger.info("Response1: " + chatBotResponse1);
assertThat(chatBotResponse1).contains("John");
// response varies too much assertThat(chatBotResponse1).contains("John");
var fluxChatBotResponse2 = this.streamingChatBot
.stream(new PromptContext(new Prompt(new String("What is my name?"))));