From 6720e5cba6abbcf0d64467fa4e0fc6adc5007db7 Mon Sep 17 00:00:00 2001
From: jzonthemtn <jeff.zemerick@mtnfog.com>
Date: Tue, 3 Dec 2024 10:36:16 -0500
Subject: [PATCH] Setting the size to k.

---
 .../org/opensearch/eval/runners/OpenSearchQuerySetRunner.java  | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/opensearch-search-quality-evaluation-plugin/src/main/java/org/opensearch/eval/runners/OpenSearchQuerySetRunner.java b/opensearch-search-quality-evaluation-plugin/src/main/java/org/opensearch/eval/runners/OpenSearchQuerySetRunner.java
index 7679ab1..89ddc3b 100644
--- a/opensearch-search-quality-evaluation-plugin/src/main/java/org/opensearch/eval/runners/OpenSearchQuerySetRunner.java
+++ b/opensearch-search-quality-evaluation-plugin/src/main/java/org/opensearch/eval/runners/OpenSearchQuerySetRunner.java
@@ -83,7 +83,8 @@ public QuerySetRunResult run(final String querySetId, final String judgmentsId,
                     final SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder();
                     searchSourceBuilder.query(QueryBuilders.wrapperQuery(q));
                     searchSourceBuilder.from(0);
-                    searchSourceBuilder.size(10);
+                    // TODO: If k is > 10, we'll need to page through these.
+                    searchSourceBuilder.size(k);
 
                     String[] includeFields = new String[] {idField};
                     String[] excludeFields = new String[] {};