Setting the size to k.

o19s · Dec 3, 2024 · 6720e5c · 6720e5c
1 parent 0e240b8
commit 6720e5c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/...evaluation-plugin/src/main/java/org/opensearch/eval/runners/OpenSearchQuerySetRunner.java b/...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[] {};