FTS: Hybrid search query pipeline

sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/implementation/Configs.java

@@ -178,6 +178,9 @@ public class Configs {
     private static final String MAX_ITEM_COUNT_FOR_VECTOR_SEARCH = "COSMOS.MAX_ITEM_SIZE_FOR_VECTOR_SEARCH";
     public static final int DEFAULT_MAX_ITEM_COUNT_FOR_VECTOR_SEARCH = 50000;
 
+    private static final String MAX_ITEM_COUNT_FOR_HYBRID_SEARCH = "COSMOS.MAX_ITEM_SIZE_FOR_HYBRID_SEARCH";
+    public static final int DEFAULT_MAX_ITEM_COUNT_FOR_HYBRID_SEARCH = 1000;
+
     private static final String AZURE_COSMOS_DISABLE_NON_STREAMING_ORDER_BY = "COSMOS.AZURE_COSMOS_DISABLE_NON_STREAMING_ORDER_BY";
 
     private static final boolean DEFAULT_AZURE_COSMOS_DISABLE_NON_STREAMING_ORDER_BY = false;
@@ -574,6 +577,13 @@ public static int getMaxItemCountForVectorSearch() {
                 String.valueOf(DEFAULT_MAX_ITEM_COUNT_FOR_VECTOR_SEARCH))));
     }
 
+    public static int getMaxItemCountForHybridSearchSearch() {
+        return Integer.parseInt(System.getProperty(MAX_ITEM_COUNT_FOR_HYBRID_SEARCH,
+            firstNonNull(
+                emptyToNull(System.getenv().get(MAX_ITEM_COUNT_FOR_HYBRID_SEARCH)),
+                String.valueOf(DEFAULT_MAX_ITEM_COUNT_FOR_HYBRID_SEARCH))));
+    }
+
     public static boolean getAzureCosmosNonStreamingOrderByDisabled() {
         if(logger.isTraceEnabled()) {
             logger.trace(

sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/implementation/Constants.java

@@ -129,6 +129,17 @@ public static final class Properties {
         public static final String FULL_TEXT_POLICY = "fullTextPolicy";
         public static final String FULL_TEXT_PATHS = "fullTextPaths";
 
+        // Hybrid Search Query
+        public static final String GLOBAL_STATISTICS_QUERY = "globalStatisticsQuery";
+        public static final String COMPONENT_QUERY_INFOS = "componentQueryInfos";
+        public static final String PROJECTION_QUERY_INFO = "projectionQueryInfo";
+        public static final String SKIP = "skip";
+        public static final String TAKE = "take";
+        public static final String REQUIRES_GLOBAL_STATISTICS = "requiresGlobalStatistics";
+        public static final String TOTAL_WORD_COUNT = "totalWordCount";
+        public static final String HIT_COUNTS = "hitCounts";
+        public static final String DOCUMENT_COUNT = "documentCount";
+        public static final String FULL_TEXT_QUERY_STATISTICS = "fullTextStatistics";
 
         // Vector Embedding Policy
         public static final String VECTOR_EMBEDDING_POLICY = "vectorEmbeddingPolicy";

sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/implementation/CosmosQueryRequestOptionsImpl.java

@@ -29,6 +29,7 @@ public final class CosmosQueryRequestOptionsImpl extends CosmosQueryRequestOptio
     private boolean emptyPageDiagnosticsEnabled;
     private String queryName;
     private Integer maxItemCountForVectorSearch;
+    private Integer maxItemCountForHybridSearch;
     private List<CosmosDiagnostics> cancelledRequestDiagnosticsTracker = new ArrayList<>();
     private String collectionRid;
 
@@ -69,6 +70,7 @@ public CosmosQueryRequestOptionsImpl(CosmosQueryRequestOptionsImpl options) {
         this.feedRange = options.feedRange;
         this.cancelledRequestDiagnosticsTracker = options.cancelledRequestDiagnosticsTracker;
         this.maxItemCountForVectorSearch = options.maxItemCountForVectorSearch;
+        this.maxItemCountForHybridSearch = options.maxItemCountForHybridSearch;
         this.collectionRid = options.collectionRid;
     }
 
@@ -235,6 +237,18 @@ public Integer getMaxItemCountForVectorSearch() {
         return this.maxItemCountForVectorSearch;
     }
 
+    /**
+     * Gets the maximum item size to fetch during hybrid search queries.
+     *
+     * @return the max number of items for hybrid search.
+     */
+    public Integer getMaxItemCountForHybridSearch() {
+        if (this.maxItemCountForHybridSearch == null) {
+            this.maxItemCountForHybridSearch = Configs.DEFAULT_MAX_ITEM_COUNT_FOR_HYBRID_SEARCH;
+        }
+        return this.maxItemCountForHybridSearch;
+    }
+
     /**
      * Sets the maximum item size to fetch during non-streaming order by queries.
      *

sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/implementation/ImplementationBridgeHelpers.java

@@ -297,6 +297,8 @@ void setCancelledRequestDiagnosticsTracker(
 
             Integer getMaxItemCountForVectorSearch(CosmosQueryRequestOptions options);
 
+            Integer getMaxItemCountForHybridSearch(CosmosQueryRequestOptions options);
+
             void setPartitionKeyDefinition(CosmosQueryRequestOptions options, PartitionKeyDefinition partitionKeyDefinition);
 
             PartitionKeyDefinition getPartitionKeyDefinition(CosmosQueryRequestOptions options);

sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/implementation/JsonSerializable.java

@@ -384,7 +384,7 @@ public Integer getInt(String propertyName) {
      * @param propertyName the property to get.
      * @return the long value
      */
-    protected Long getLong(String propertyName) {
+    public Long getLong(String propertyName) {
         return getWithMapping(propertyName, JsonNode::asLong);
     }