From 9956367fc842334dffdf2c3cfa078c9255895b97 Mon Sep 17 00:00:00 2001 From: Craig Walls Date: Mon, 24 Jun 2024 16:06:21 -0600 Subject: [PATCH] Accept String answer in BasicEvaluationTest instead of ChatResponse --- .../springframework/ai/evaluation/BasicEvaluationTest.java | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/spring-ai-test/src/main/java/org/springframework/ai/evaluation/BasicEvaluationTest.java b/spring-ai-test/src/main/java/org/springframework/ai/evaluation/BasicEvaluationTest.java index f883bc38c..3224094b8 100644 --- a/spring-ai-test/src/main/java/org/springframework/ai/evaluation/BasicEvaluationTest.java +++ b/spring-ai-test/src/main/java/org/springframework/ai/evaluation/BasicEvaluationTest.java @@ -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,