Pagination in Hybrid query

src/main/java/org/opensearch/neuralsearch/query/HybridQueryBuilder.java

@@ -337,7 +337,7 @@ private static void validatePaginationDepth(final int paginationDepth, final Que
         int maxResultWindowIndexSetting = queryShardContext.getIndexSettings().getMaxResultWindow();
         if (paginationDepth > maxResultWindowIndexSetting) {
             throw new IllegalArgumentException(
-                String.format(Locale.ROOT, "pagination_depth should be less than index.max_result_window setting")
+                String.format(Locale.ROOT, "pagination_depth should be less than or equal to index.max_result_window setting")
             );
         }
     }

src/main/java/org/opensearch/neuralsearch/util/HybridQueryUtil.java

@@ -20,6 +20,9 @@
 @NoArgsConstructor(access = AccessLevel.PRIVATE)
 public class HybridQueryUtil {
 
+    /**
+     * This method validates whether the query object is an instance of hybrid query
+     */
     public static boolean isHybridQuery(final Query query, final SearchContext searchContext) {
         if (query instanceof HybridQuery) {
             return true;
@@ -52,19 +55,22 @@ public static boolean isHybridQuery(final Query query, final SearchContext searc
         return false;
     }
 
-    public static boolean hasNestedFieldOrNestedDocs(final Query query, final SearchContext searchContext) {
+    private static boolean hasNestedFieldOrNestedDocs(final Query query, final SearchContext searchContext) {
         return searchContext.mapperService().hasNested() && new NestedHelper(searchContext.mapperService()).mightMatchNestedDocs(query);
     }
 
-    public static boolean isWrappedHybridQuery(final Query query) {
+    private static boolean isWrappedHybridQuery(final Query query) {
         return query instanceof BooleanQuery
             && ((BooleanQuery) query).clauses().stream().anyMatch(clauseQuery -> clauseQuery.getQuery() instanceof HybridQuery);
     }
 
-    public static boolean hasAliasFilter(final Query query, final SearchContext searchContext) {
+    private static boolean hasAliasFilter(final Query query, final SearchContext searchContext) {
         return Objects.nonNull(searchContext.aliasFilter());
     }
 
+    /**
+     * This method checks whether hybrid query is wrapped under boolean query object
+     */
     public static boolean isHybridQueryWrappedInBooleanQuery(final SearchContext searchContext, final Query query) {
         return ((hasAliasFilter(query, searchContext) || hasNestedFieldOrNestedDocs(query, searchContext))
             && isWrappedHybridQuery(query)

src/test/java/org/opensearch/neuralsearch/query/HybridQueryBuilderTests.java

@@ -249,7 +249,10 @@ public void testDoToQuery_whenPaginationDepthIsGreaterThan10000_thenBuildSuccess
             IllegalArgumentException.class,
             () -> queryBuilder.doToQuery(mockQueryShardContext)
         );
-        assertThat(exception.getMessage(), containsString("pagination_depth should be less than index.max_result_window setting"));
+        assertThat(
+            exception.getMessage(),
+            containsString("pagination_depth should be less than or equal to index.max_result_window setting")
+        );
     }
 
     @SneakyThrows

src/test/java/org/opensearch/neuralsearch/query/HybridQueryIT.java

@@ -968,7 +968,7 @@ public void testHybridQuery_whenPaginationDepthIsOutOfRange_thenFail() {
 
             org.hamcrest.MatcherAssert.assertThat(
                 responseException.getMessage(),
-                allOf(containsString("pagination_depth should be less than index.max_result_window setting"))
+                allOf(containsString("pagination_depth should be less than or equal to index.max_result_window setting"))
             );
         } finally {
             wipeOfTestResources(TEST_MULTI_DOC_INDEX_NAME_ONE_SHARD, null, null, SEARCH_PIPELINE);