Draft solution

Signed-off-by: Martin Gaievski <[email protected]>
martin-gaievski · Nov 16, 2024 · 898b34c · 898b34c
1 parent 0119907
commit 898b34c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/main/java/org/opensearch/neuralsearch/search/query/HybridQueryPhaseSearcher.java b/src/main/java/org/opensearch/neuralsearch/search/query/HybridQueryPhaseSearcher.java
@@ -15,6 +15,7 @@
 import org.apache.lucene.search.BooleanClause;
 import org.apache.lucene.search.BooleanQuery;
 import org.apache.lucene.search.Query;
+import org.apache.lucene.search.ScoreMode;
 import org.opensearch.common.settings.Settings;
 import org.opensearch.index.mapper.MapperService;
 import org.opensearch.neuralsearch.query.HybridQuery;
@@ -179,7 +180,7 @@ protected boolean searchWithCollector(
                 searcher,
                 query,
                 collectors,
-                QueryCollectorContext.EMPTY_CONTEXT,
+                QueryCollectorContext.getContextForScoreMode(ScoreMode.TOP_SCORES),
                 hasFilterCollector,
                 hasTimeout
             );
@@ -207,7 +208,7 @@ protected boolean searchWithCollector(
                 searcher,
                 query,
                 collectors,
-                QueryCollectorContext.EMPTY_CONTEXT,
+                QueryCollectorContext.getContextForScoreMode(ScoreMode.TOP_SCORES),
                 hasFilterCollector,
                 hasTimeout
             );