Fix RAG tests after Document refactoring

Author: markpollack

spring-ai-core/src/main/java/org/springframework/ai/evaluation/EvaluationRequest.java

@@ -20,6 +20,7 @@
 import java.util.List;
 import java.util.Objects;
 
+import org.springframework.ai.document.Document;
 import org.springframework.ai.model.Content;
 
 /**
@@ -35,19 +36,19 @@ public class EvaluationRequest {
 
 	private final String userText;
 
-	private final List<Content> dataList;
+	private final List<Document> dataList;
 
 	private final String responseContent;
 
 	public EvaluationRequest(String userText, String responseContent) {
 		this(userText, Collections.emptyList(), responseContent);
 	}
 
-	public EvaluationRequest(List<Content> dataList, String responseContent) {
+	public EvaluationRequest(List<Document> dataList, String responseContent) {
 		this("", dataList, responseContent);
 	}
 
-	public EvaluationRequest(String userText, List<Content> dataList, String responseContent) {
+	public EvaluationRequest(String userText, List<Document> dataList, String responseContent) {
 		this.userText = userText;
 		this.dataList = dataList;
 		this.responseContent = responseContent;
@@ -57,7 +58,7 @@ public String getUserText() {
 		return this.userText;
 	}
 
-	public List<Content> getDataList() {
+	public List<Document> getDataList() {
 		return this.dataList;
 	}
 

spring-ai-core/src/main/java/org/springframework/ai/evaluation/Evaluator.java

@@ -19,6 +19,7 @@
 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<Content> data = evaluationRequest.getDataList();
+		List<Document> data = evaluationRequest.getDataList();
 		return data.stream()
-			.map(Content::getText)
+			.map(Document::getText)
 			.filter(StringUtils::hasText)
 			.collect(Collectors.joining(System.lineSeparator()));
 	}