Adding log of pagination_depth and correcting is null check

Signed-off-by: Varun Jain <[email protected]>
opensearch-project · Jan 14, 2025 · 0eb2641 · 0eb2641
1 parent 2c029ab
commit 0eb2641
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/main/java/org/opensearch/neuralsearch/query/HybridQuery.java b/src/main/java/org/opensearch/neuralsearch/query/HybridQuery.java
@@ -47,7 +47,7 @@ public HybridQuery(final Collection<Query> subQueries, final List<Query> filterQ
             throw new IllegalArgumentException("collection of queries must not be empty");
         }
         Integer paginationDepth = hybridQueryContext.getPaginationDepth();
-        if (paginationDepth != null && paginationDepth == 0) {
+        if (Objects.nonNull(paginationDepth) && paginationDepth == 0) {
             throw new IllegalArgumentException("pagination_depth must not be zero");
         }
         if (Objects.isNull(filterQueries) || filterQueries.isEmpty()) {

diff --git a/src/main/java/org/opensearch/neuralsearch/search/query/HybridCollectorManager.java b/src/main/java/org/opensearch/neuralsearch/search/query/HybridCollectorManager.java
@@ -491,6 +491,7 @@ private static int getSubqueryResultsRetrievalSize(final SearchContext searchCon
         if (searchContext.from() == 0) {
             return searchContext.size();
         }
+        log.info("pagination_depth is {}", paginationDepth);
         return paginationDepth;
     }