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

Accept String answer in BasicEvaluationTest instead of ChatResponse

Browse Source
This commit is contained in:
Craig Walls
2024-06-24 16:06:21 -06:00
committed by Mark Pollack
parent 0f07b7bb8a
commit 9956367fc8
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
spring-ai-test/src/main/java/org/springframework/ai/evaluation/BasicEvaluationTest.java
View File

@@ -52,9 +52,9 @@ public class BasicEvaluationTest {
@Value("classpath:/prompts/spring/test/evaluation/user-evaluator-message.st")
protected Resource userEvaluatorResource;
protected void evaluateQuestionAndAnswer(String question, ChatResponse response, boolean factBased) {
assertThat(response).isNotNull();
String answer = response.getResult().getOutput().getContent();
protected void evaluateQuestionAndAnswer(String question, String answer, boolean factBased) {
assertThat(question).isNotNull();
assertThat(answer).isNotNull();
logger.info("Question: " + question);
logger.info("Answer:" + answer);
PromptTemplate userPromptTemplate = new PromptTemplate(userEvaluatorResource,