diff --git a/spring-ai-core/src/main/java/org/springframework/ai/evaluation/EvaluationRequest.java b/spring-ai-core/src/main/java/org/springframework/ai/evaluation/EvaluationRequest.java index d5629cc0f..0a17e7011 100644 --- a/spring-ai-core/src/main/java/org/springframework/ai/evaluation/EvaluationRequest.java +++ b/spring-ai-core/src/main/java/org/springframework/ai/evaluation/EvaluationRequest.java @@ -20,6 +20,7 @@ import java.util.Collections; import java.util.List; import java.util.Objects; +import org.springframework.ai.document.Document; import org.springframework.ai.model.Content; /** @@ -35,7 +36,7 @@ public class EvaluationRequest { private final String userText; - private final List dataList; + private final List dataList; private final String responseContent; @@ -43,11 +44,11 @@ public class EvaluationRequest { this(userText, Collections.emptyList(), responseContent); } - public EvaluationRequest(List dataList, String responseContent) { + public EvaluationRequest(List dataList, String responseContent) { this("", dataList, responseContent); } - public EvaluationRequest(String userText, List dataList, String responseContent) { + public EvaluationRequest(String userText, List dataList, String responseContent) { this.userText = userText; this.dataList = dataList; this.responseContent = responseContent; @@ -57,7 +58,7 @@ public class EvaluationRequest { return this.userText; } - public List getDataList() { + public List getDataList() { return this.dataList; } diff --git a/spring-ai-core/src/main/java/org/springframework/ai/evaluation/Evaluator.java b/spring-ai-core/src/main/java/org/springframework/ai/evaluation/Evaluator.java index 927b97acb..902a91fdf 100644 --- a/spring-ai-core/src/main/java/org/springframework/ai/evaluation/Evaluator.java +++ b/spring-ai-core/src/main/java/org/springframework/ai/evaluation/Evaluator.java @@ -19,6 +19,7 @@ package org.springframework.ai.evaluation; import java.util.List; import java.util.stream.Collectors; +import org.springframework.ai.document.Document; import org.springframework.ai.model.Content; import org.springframework.util.StringUtils; @@ -28,9 +29,9 @@ public interface Evaluator { EvaluationResponse evaluate(EvaluationRequest evaluationRequest); default String doGetSupportingData(EvaluationRequest evaluationRequest) { - List data = evaluationRequest.getDataList(); + List data = evaluationRequest.getDataList(); return data.stream() - .map(Content::getText) + .map(Document::getText) .filter(StringUtils::hasText) .collect(Collectors.joining(System.lineSeparator())); }