@azure/cosmos 4.9.1 → 4.9.2

package/dist/commonjs/queryExecutionContext/pipelinedQueryExecutionContext.js

@@ -1,244 +1,294 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.PipelinedQueryExecutionContext = void 0;
-const ErrorResponse_js_1 = require("../request/ErrorResponse.js");
-const OffsetLimitEndpointComponent_js_1 = require("./EndpointComponent/OffsetLimitEndpointComponent.js");
-const OrderByEndpointComponent_js_1 = require("./EndpointComponent/OrderByEndpointComponent.js");
-const OrderedDistinctEndpointComponent_js_1 = require("./EndpointComponent/OrderedDistinctEndpointComponent.js");
-const UnorderedDistinctEndpointComponent_js_1 = require("./EndpointComponent/UnorderedDistinctEndpointComponent.js");
-const GroupByEndpointComponent_js_1 = require("./EndpointComponent/GroupByEndpointComponent.js");
-const orderByQueryExecutionContext_js_1 = require("./orderByQueryExecutionContext.js");
-const parallelQueryExecutionContext_js_1 = require("./parallelQueryExecutionContext.js");
-const GroupByValueEndpointComponent_js_1 = require("./EndpointComponent/GroupByValueEndpointComponent.js");
-const NonStreamingOrderByDistinctEndpointComponent_js_1 = require("./EndpointComponent/NonStreamingOrderByDistinctEndpointComponent.js");
-const NonStreamingOrderByEndpointComponent_js_1 = require("./EndpointComponent/NonStreamingOrderByEndpointComponent.js");
-const QueryValidationHelper_js_1 = require("./QueryValidationHelper.js");
-const ContinuationTokenParser_js_1 = require("./ContinuationTokenParser.js");
-const LegacyFetchImplementation_js_1 = require("./LegacyFetchImplementation.js");
-const QueryControlFetchImplementation_js_1 = require("./QueryControlFetchImplementation.js");
-const constants_js_1 = require("../common/constants.js");
-/** @hidden */
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var pipelinedQueryExecutionContext_exports = {};
+__export(pipelinedQueryExecutionContext_exports, {
+  PipelinedQueryExecutionContext: () => PipelinedQueryExecutionContext
+});
+module.exports = __toCommonJS(pipelinedQueryExecutionContext_exports);
+var import_ErrorResponse = require("../request/ErrorResponse.js");
+var import_OffsetLimitEndpointComponent = require("./EndpointComponent/OffsetLimitEndpointComponent.js");
+var import_OrderByEndpointComponent = require("./EndpointComponent/OrderByEndpointComponent.js");
+var import_OrderedDistinctEndpointComponent = require("./EndpointComponent/OrderedDistinctEndpointComponent.js");
+var import_UnorderedDistinctEndpointComponent = require("./EndpointComponent/UnorderedDistinctEndpointComponent.js");
+var import_GroupByEndpointComponent = require("./EndpointComponent/GroupByEndpointComponent.js");
+var import_orderByQueryExecutionContext = require("./orderByQueryExecutionContext.js");
+var import_parallelQueryExecutionContext = require("./parallelQueryExecutionContext.js");
+var import_GroupByValueEndpointComponent = require("./EndpointComponent/GroupByValueEndpointComponent.js");
+var import_NonStreamingOrderByDistinctEndpointComponent = require("./EndpointComponent/NonStreamingOrderByDistinctEndpointComponent.js");
+var import_NonStreamingOrderByEndpointComponent = require("./EndpointComponent/NonStreamingOrderByEndpointComponent.js");
+var import_QueryValidationHelper = require("./QueryValidationHelper.js");
+var import_ContinuationTokenParser = require("./ContinuationTokenParser.js");
+var import_LegacyFetchImplementation = require("./LegacyFetchImplementation.js");
+var import_QueryControlFetchImplementation = require("./QueryControlFetchImplementation.js");
+var import_constants = require("../common/constants.js");
 class PipelinedQueryExecutionContext {
-    clientContext;
-    collectionLink;
-    query;
-    options;
-    partitionedQueryExecutionInfo;
-    emitRawOrderByPayload;
-    supportsContinuationTokens;
-    fetchBuffer;
-    endpoint;
-    fetchImplementation;
-    constructor(clientContext, collectionLink, query, options, partitionedQueryExecutionInfo, correlatedActivityId, emitRawOrderByPayload = false, supportsContinuationTokens = true) {
-        this.clientContext = clientContext;
-        this.collectionLink = collectionLink;
-        this.query = query;
-        this.options = options;
-        this.partitionedQueryExecutionInfo = partitionedQueryExecutionInfo;
-        this.emitRawOrderByPayload = emitRawOrderByPayload;
-        this.supportsContinuationTokens = supportsContinuationTokens;
-        // Validate that queryInfo is present in partitioned query execution info
-        if (!partitionedQueryExecutionInfo.queryInfo) {
-            throw new ErrorResponse_js_1.ErrorResponse("Query execution requires valid query plan information. " +
-                "The partitioned query execution info is missing queryInfo. " +
-                "This may indicate an invalid query or a problem with query planning.");
-        }
-        if (!this.options.maxItemCount) {
-            this.options.maxItemCount = constants_js_1.QueryExecution.DEFAULT_PAGE_SIZE;
-        }
-        const pageSize = this.options.maxItemCount;
-        // Extract query information and characteristics
-        const analyzedQueryInfo = this.analyzeQueryInfo(partitionedQueryExecutionInfo.queryInfo);
-        const { sortOrders, nonStreamingOrderBy, isOrderByQuery, isGroupByQuery, isUnorderedDistinctQuery, querySupportsTokens, } = analyzedQueryInfo;
-        // Reject continuation token usage for unsupported query types
-        if (!querySupportsTokens) {
-            (0, QueryValidationHelper_js_1.rejectContinuationTokenForUnsupportedQueries)(this.options.continuationToken, [
-                QueryValidationHelper_js_1.QueryTypes.nonStreamingOrderBy(nonStreamingOrderBy),
-                QueryValidationHelper_js_1.QueryTypes.groupBy(isGroupByQuery),
-                QueryValidationHelper_js_1.QueryTypes.unorderedDistinct(isUnorderedDistinctQuery),
-            ]);
-        }
-        // Parse continuation token fields once for reuse in pipeline construction
-        const queryContinuationFields = this.options.continuationToken
-            ? (0, ContinuationTokenParser_js_1.parseContinuationTokenFields)(this.options.continuationToken)
-            : undefined;
-        // Pick between non-streaming vs streaming execution context
-        this.endpoint = nonStreamingOrderBy
-            ? this.createNonStreamingEndpoint(partitionedQueryExecutionInfo, sortOrders, correlatedActivityId, options)
-            : this.createStreamingEndpoint(partitionedQueryExecutionInfo, sortOrders, correlatedActivityId, isGroupByQuery, queryContinuationFields);
-        this.fetchBuffer = [];
-        // Initialize the appropriate fetch implementation based on enableQueryControl
-        if (this.options.enableQueryControl) {
-            const querySupportsContinuationTokens = this.supportsContinuationTokens && querySupportsTokens;
-            this.fetchImplementation = new QueryControlFetchImplementation_js_1.QueryControlFetchImplementation(this.endpoint, pageSize, this.collectionLink, this.options.continuationToken, isOrderByQuery, querySupportsContinuationTokens);
-        }
-        else {
-            this.fetchImplementation = new LegacyFetchImplementation_js_1.LegacyFetchImplementation(this.endpoint, pageSize);
-        }
+  constructor(clientContext, collectionLink, query, options, partitionedQueryExecutionInfo, correlatedActivityId, emitRawOrderByPayload = false, supportsContinuationTokens = true) {
+    this.clientContext = clientContext;
+    this.collectionLink = collectionLink;
+    this.query = query;
+    this.options = options;
+    this.partitionedQueryExecutionInfo = partitionedQueryExecutionInfo;
+    this.emitRawOrderByPayload = emitRawOrderByPayload;
+    this.supportsContinuationTokens = supportsContinuationTokens;
+    if (!partitionedQueryExecutionInfo.queryInfo) {
+      throw new import_ErrorResponse.ErrorResponse(
+        "Query execution requires valid query plan information. The partitioned query execution info is missing queryInfo. This may indicate an invalid query or a problem with query planning."
+      );
     }
-    hasMoreResults() {
-        return this.fetchBuffer.length !== 0 || this.endpoint.hasMoreResults();
+    if (!this.options.maxItemCount) {
+      this.options.maxItemCount = import_constants.QueryExecution.DEFAULT_PAGE_SIZE;
     }
-    async fetchMore(diagnosticNode) {
-        return this.fetchImplementation.fetchMore(diagnosticNode, this.fetchBuffer);
+    const pageSize = this.options.maxItemCount;
+    const analyzedQueryInfo = this.analyzeQueryInfo(partitionedQueryExecutionInfo.queryInfo);
+    const {
+      sortOrders,
+      nonStreamingOrderBy,
+      isOrderByQuery,
+      isGroupByQuery,
+      isUnorderedDistinctQuery,
+      querySupportsTokens
+    } = analyzedQueryInfo;
+    if (!querySupportsTokens) {
+      (0, import_QueryValidationHelper.rejectContinuationTokenForUnsupportedQueries)(this.options.continuationToken, [
+        import_QueryValidationHelper.QueryTypes.nonStreamingOrderBy(nonStreamingOrderBy),
+        import_QueryValidationHelper.QueryTypes.groupBy(isGroupByQuery),
+        import_QueryValidationHelper.QueryTypes.unorderedDistinct(isUnorderedDistinctQuery)
+      ]);
     }
-    /**
-     * Creates a non-streaming endpoint for vector search and similar queries that require buffering
-     */
-    createNonStreamingEndpoint(partitionedQueryExecutionInfo, sortOrders, correlatedActivityId, options) {
-        const queryInfo = partitionedQueryExecutionInfo.queryInfo; // Safe to use ! after validation in constructor
-        if (!options.allowUnboundedNonStreamingQueries) {
-            this.checkQueryConstraints(queryInfo);
-        }
-        const vectorSearchBufferSize = this.calculateVectorSearchBufferSize(queryInfo, options);
-        this.validateVectorSearchBufferSize(vectorSearchBufferSize, options);
-        const baseContext = new parallelQueryExecutionContext_js_1.ParallelQueryExecutionContext(this.clientContext, this.collectionLink, this.query, this.options, this.partitionedQueryExecutionInfo, correlatedActivityId);
-        return this.wrapWithNonStreamingComponent(baseContext, queryInfo, sortOrders, vectorSearchBufferSize);
+    const queryContinuationFields = this.options.continuationToken ? (0, import_ContinuationTokenParser.parseContinuationTokenFields)(this.options.continuationToken) : void 0;
+    this.endpoint = nonStreamingOrderBy ? this.createNonStreamingEndpoint(
+      partitionedQueryExecutionInfo,
+      sortOrders,
+      correlatedActivityId,
+      options
+    ) : this.createStreamingEndpoint(
+      partitionedQueryExecutionInfo,
+      sortOrders,
+      correlatedActivityId,
+      isGroupByQuery,
+      queryContinuationFields
+    );
+    this.fetchBuffer = [];
+    if (this.options.enableQueryControl) {
+      const querySupportsContinuationTokens = this.supportsContinuationTokens && querySupportsTokens;
+      this.fetchImplementation = new import_QueryControlFetchImplementation.QueryControlFetchImplementation(
+        this.endpoint,
+        pageSize,
+        this.collectionLink,
+        this.options.continuationToken,
+        isOrderByQuery,
+        querySupportsContinuationTokens
+      );
+    } else {
+      this.fetchImplementation = new import_LegacyFetchImplementation.LegacyFetchImplementation(this.endpoint, pageSize);
     }
-    /**
-     * Creates a streaming endpoint with proper pipeline components
-     */
-    createStreamingEndpoint(partitionedQueryExecutionInfo, sortOrders, correlatedActivityId, isGroupByQuery, queryContinuationFields) {
-        const queryInfo = partitionedQueryExecutionInfo.queryInfo; // Safe to use ! after validation in constructor
-        // Create base execution context
-        let endpoint = this.createBaseExecutionContext(partitionedQueryExecutionInfo, sortOrders, correlatedActivityId);
-        // Apply pipeline transformations
-        endpoint = this.applyGroupByComponents(endpoint, queryInfo, isGroupByQuery);
-        endpoint = this.applyDistinctComponents(endpoint, queryInfo, queryContinuationFields);
-        endpoint = this.applyLimitComponents(endpoint, queryInfo, queryContinuationFields);
-        return endpoint;
+  }
+  fetchBuffer;
+  endpoint;
+  fetchImplementation;
+  hasMoreResults() {
+    return this.fetchBuffer.length !== 0 || this.endpoint.hasMoreResults();
+  }
+  async fetchMore(diagnosticNode) {
+    return this.fetchImplementation.fetchMore(diagnosticNode, this.fetchBuffer);
+  }
+  /**
+   * Creates a non-streaming endpoint for vector search and similar queries that require buffering
+   */
+  createNonStreamingEndpoint(partitionedQueryExecutionInfo, sortOrders, correlatedActivityId, options) {
+    const queryInfo = partitionedQueryExecutionInfo.queryInfo;
+    if (!options.allowUnboundedNonStreamingQueries) {
+      this.checkQueryConstraints(queryInfo);
     }
-    /**
-     * Creates the base execution context (OrderBy or Parallel)
-     */
-    createBaseExecutionContext(partitionedQueryExecutionInfo, sortOrders, correlatedActivityId) {
-        if (Array.isArray(sortOrders) && sortOrders.length > 0) {
-            // OrderBy queries need special wrapping for payload structure
-            return new OrderByEndpointComponent_js_1.OrderByEndpointComponent(new orderByQueryExecutionContext_js_1.OrderByQueryExecutionContext(this.clientContext, this.collectionLink, this.query, this.options, partitionedQueryExecutionInfo, correlatedActivityId), this.emitRawOrderByPayload);
-        }
-        // Parallel queries
-        return new parallelQueryExecutionContext_js_1.ParallelQueryExecutionContext(this.clientContext, this.collectionLink, this.query, this.options, partitionedQueryExecutionInfo, correlatedActivityId);
+    const vectorSearchBufferSize = this.calculateVectorSearchBufferSize(queryInfo, options);
+    this.validateVectorSearchBufferSize(vectorSearchBufferSize, options);
+    const baseContext = new import_parallelQueryExecutionContext.ParallelQueryExecutionContext(
+      this.clientContext,
+      this.collectionLink,
+      this.query,
+      this.options,
+      this.partitionedQueryExecutionInfo,
+      correlatedActivityId
+    );
+    return this.wrapWithNonStreamingComponent(
+      baseContext,
+      queryInfo,
+      sortOrders,
+      vectorSearchBufferSize
+    );
+  }
+  /**
+   * Creates a streaming endpoint with proper pipeline components
+   */
+  createStreamingEndpoint(partitionedQueryExecutionInfo, sortOrders, correlatedActivityId, isGroupByQuery, queryContinuationFields) {
+    const queryInfo = partitionedQueryExecutionInfo.queryInfo;
+    let endpoint = this.createBaseExecutionContext(
+      partitionedQueryExecutionInfo,
+      sortOrders,
+      correlatedActivityId
+    );
+    endpoint = this.applyGroupByComponents(endpoint, queryInfo, isGroupByQuery);
+    endpoint = this.applyDistinctComponents(endpoint, queryInfo, queryContinuationFields);
+    endpoint = this.applyLimitComponents(endpoint, queryInfo, queryContinuationFields);
+    return endpoint;
+  }
+  /**
+   * Creates the base execution context (OrderBy or Parallel)
+   */
+  createBaseExecutionContext(partitionedQueryExecutionInfo, sortOrders, correlatedActivityId) {
+    if (Array.isArray(sortOrders) && sortOrders.length > 0) {
+      return new import_OrderByEndpointComponent.OrderByEndpointComponent(
+        new import_orderByQueryExecutionContext.OrderByQueryExecutionContext(
+          this.clientContext,
+          this.collectionLink,
+          this.query,
+          this.options,
+          partitionedQueryExecutionInfo,
+          correlatedActivityId
+        ),
+        this.emitRawOrderByPayload
+      );
     }
-    /**
-     * Wraps base context with appropriate non-streaming component
-     */
-    wrapWithNonStreamingComponent(baseContext, queryInfo, sortOrders, vectorSearchBufferSize) {
-        const distinctType = queryInfo.distinctType;
-        if (distinctType === "None") {
-            return new NonStreamingOrderByEndpointComponent_js_1.NonStreamingOrderByEndpointComponent(baseContext, sortOrders, vectorSearchBufferSize, queryInfo.offset, this.emitRawOrderByPayload);
-        }
-        return new NonStreamingOrderByDistinctEndpointComponent_js_1.NonStreamingOrderByDistinctEndpointComponent(baseContext, queryInfo, vectorSearchBufferSize, this.emitRawOrderByPayload);
+    return new import_parallelQueryExecutionContext.ParallelQueryExecutionContext(
+      this.clientContext,
+      this.collectionLink,
+      this.query,
+      this.options,
+      partitionedQueryExecutionInfo,
+      correlatedActivityId
+    );
+  }
+  /**
+   * Wraps base context with appropriate non-streaming component
+   */
+  wrapWithNonStreamingComponent(baseContext, queryInfo, sortOrders, vectorSearchBufferSize) {
+    const distinctType = queryInfo.distinctType;
+    if (distinctType === "None") {
+      return new import_NonStreamingOrderByEndpointComponent.NonStreamingOrderByEndpointComponent(
+        baseContext,
+        sortOrders,
+        vectorSearchBufferSize,
+        queryInfo.offset,
+        this.emitRawOrderByPayload
+      );
     }
-    /**
-     * Applies GROUP BY components to the pipeline if needed
-     */
-    applyGroupByComponents(endpoint, queryInfo, isGroupByQuery) {
-        if (!isGroupByQuery) {
-            return endpoint;
-        }
-        return queryInfo.hasSelectValue
-            ? new GroupByValueEndpointComponent_js_1.GroupByValueEndpointComponent(endpoint, queryInfo)
-            : new GroupByEndpointComponent_js_1.GroupByEndpointComponent(endpoint, queryInfo);
+    return new import_NonStreamingOrderByDistinctEndpointComponent.NonStreamingOrderByDistinctEndpointComponent(
+      baseContext,
+      queryInfo,
+      vectorSearchBufferSize,
+      this.emitRawOrderByPayload
+    );
+  }
+  /**
+   * Applies GROUP BY components to the pipeline if needed
+   */
+  applyGroupByComponents(endpoint, queryInfo, isGroupByQuery) {
+    if (!isGroupByQuery) {
+      return endpoint;
     }
-    /**
-     * Applies DISTINCT components to the pipeline if needed
-     */
-    applyDistinctComponents(endpoint, queryInfo, queryContinuationFields) {
-        const distinctType = queryInfo.distinctType;
-        if (distinctType === "Ordered") {
-            const lastHash = queryContinuationFields?.hashedLastResult;
-            return new OrderedDistinctEndpointComponent_js_1.OrderedDistinctEndpointComponent(endpoint, lastHash);
-        }
-        if (distinctType === "Unordered") {
-            return new UnorderedDistinctEndpointComponent_js_1.UnorderedDistinctEndpointComponent(endpoint);
-        }
-        return endpoint;
+    return queryInfo.hasSelectValue ? new import_GroupByValueEndpointComponent.GroupByValueEndpointComponent(endpoint, queryInfo) : new import_GroupByEndpointComponent.GroupByEndpointComponent(endpoint, queryInfo);
+  }
+  /**
+   * Applies DISTINCT components to the pipeline if needed
+   */
+  applyDistinctComponents(endpoint, queryInfo, queryContinuationFields) {
+    const distinctType = queryInfo.distinctType;
+    if (distinctType === "Ordered") {
+      const lastHash = queryContinuationFields?.hashedLastResult;
+      return new import_OrderedDistinctEndpointComponent.OrderedDistinctEndpointComponent(endpoint, lastHash);
     }
-    /**
-     * Applies TOP and OFFSET+LIMIT components to the pipeline if needed
-     */
-    applyLimitComponents(endpoint, queryInfo, queryContinuationFields) {
-        // Apply TOP component (TOP N is effectively OFFSET 0 LIMIT N)
-        let top = queryInfo.top;
-        if (typeof top === "number") {
-            if (queryContinuationFields?.limit !== undefined) {
-                top = queryContinuationFields.limit;
-            }
-            endpoint = new OffsetLimitEndpointComponent_js_1.OffsetLimitEndpointComponent(endpoint, 0, top);
-        }
-        // Apply OFFSET+LIMIT component
-        let limit = queryInfo.limit;
-        let offset = queryInfo.offset;
-        if (queryContinuationFields) {
-            if (queryContinuationFields.limit !== undefined) {
-                limit = queryContinuationFields.limit;
-            }
-            if (queryContinuationFields.offset !== undefined) {
-                offset = queryContinuationFields.offset;
-            }
-        }
-        if (typeof limit === "number" && typeof offset === "number") {
-            endpoint = new OffsetLimitEndpointComponent_js_1.OffsetLimitEndpointComponent(endpoint, offset, limit);
-        }
-        return endpoint;
+    if (distinctType === "Unordered") {
+      return new import_UnorderedDistinctEndpointComponent.UnorderedDistinctEndpointComponent(endpoint);
     }
-    /**
-     * Validates vector search buffer size constraints
-     */
-    validateVectorSearchBufferSize(vectorSearchBufferSize, options) {
-        const maxBufferSize = options["vectorSearchBufferSize"]
-            ? options["vectorSearchBufferSize"]
-            : constants_js_1.QueryExecution.DEFAULT_MAX_VECTOR_SEARCH_BUFFER_SIZE;
-        if (vectorSearchBufferSize > maxBufferSize) {
-            throw new ErrorResponse_js_1.ErrorResponse(`Executing a vector search query with TOP or OFFSET + LIMIT value ${vectorSearchBufferSize} larger than the vectorSearchBufferSize ${maxBufferSize} ` +
-                `is not allowed`);
-        }
+    return endpoint;
+  }
+  /**
+   * Applies TOP and OFFSET+LIMIT components to the pipeline if needed
+   */
+  applyLimitComponents(endpoint, queryInfo, queryContinuationFields) {
+    let top = queryInfo.top;
+    if (typeof top === "number") {
+      if (queryContinuationFields?.limit !== void 0) {
+        top = queryContinuationFields.limit;
+      }
+      endpoint = new import_OffsetLimitEndpointComponent.OffsetLimitEndpointComponent(endpoint, 0, top);
     }
-    calculateVectorSearchBufferSize(queryInfo, options) {
-        if (queryInfo.top === 0 || queryInfo.limit === 0)
-            return 0;
-        return queryInfo.top
-            ? queryInfo.top
-            : queryInfo.limit
-                ? queryInfo.offset + queryInfo.limit
-                : options["vectorSearchBufferSize"] && options["vectorSearchBufferSize"] > 0
-                    ? options["vectorSearchBufferSize"]
-                    : constants_js_1.QueryExecution.DEFAULT_MAX_VECTOR_SEARCH_BUFFER_SIZE;
+    let limit = queryInfo.limit;
+    let offset = queryInfo.offset;
+    if (queryContinuationFields) {
+      if (queryContinuationFields.limit !== void 0) {
+        limit = queryContinuationFields.limit;
+      }
+      if (queryContinuationFields.offset !== void 0) {
+        offset = queryContinuationFields.offset;
+      }
     }
-    checkQueryConstraints(queryInfo) {
-        const hasTop = queryInfo.top || queryInfo.top === 0;
-        const hasLimit = queryInfo.limit || queryInfo.limit === 0;
-        if (!hasTop && !hasLimit) {
-            throw new ErrorResponse_js_1.ErrorResponse("Executing a non-streaming search query without TOP or LIMIT can consume a large number of RUs " +
-                "very fast and have long runtimes. Please ensure you are using one of the above two filters " +
-                "with your vector search query.");
-        }
-        return;
+    if (typeof limit === "number" && typeof offset === "number") {
+      endpoint = new import_OffsetLimitEndpointComponent.OffsetLimitEndpointComponent(endpoint, offset, limit);
     }
-    /**
-     * Analyzes query information and extracts key characteristics for query execution planning
-     */
-    analyzeQueryInfo(queryInfo) {
-        const sortOrders = queryInfo.orderBy;
-        const nonStreamingOrderBy = queryInfo.hasNonStreamingOrderBy;
-        const isOrderByQuery = Array.isArray(sortOrders) && sortOrders.length > 0;
-        // Check if this is a GROUP BY query
-        const isGroupByQuery = Object.keys(queryInfo.groupByAliasToAggregateType || {}).length > 0 ||
-            (queryInfo.aggregates?.length || 0) > 0 ||
-            (queryInfo.groupByExpressions?.length || 0) > 0;
-        // Check if this is an unordered DISTINCT query
-        const isUnorderedDistinctQuery = queryInfo.distinctType === "Unordered";
-        // Determine if this query type supports continuation tokens
-        const querySupportsTokens = !isUnorderedDistinctQuery && !isGroupByQuery && !nonStreamingOrderBy;
-        return {
-            sortOrders,
-            nonStreamingOrderBy,
-            isOrderByQuery,
-            isGroupByQuery,
-            isUnorderedDistinctQuery,
-            querySupportsTokens,
-        };
+    return endpoint;
+  }
+  /**
+   * Validates vector search buffer size constraints
+   */
+  validateVectorSearchBufferSize(vectorSearchBufferSize, options) {
+    const maxBufferSize = options["vectorSearchBufferSize"] ? options["vectorSearchBufferSize"] : import_constants.QueryExecution.DEFAULT_MAX_VECTOR_SEARCH_BUFFER_SIZE;
+    if (vectorSearchBufferSize > maxBufferSize) {
+      throw new import_ErrorResponse.ErrorResponse(
+        `Executing a vector search query with TOP or OFFSET + LIMIT value ${vectorSearchBufferSize} larger than the vectorSearchBufferSize ${maxBufferSize} is not allowed`
+      );
     }
+  }
+  calculateVectorSearchBufferSize(queryInfo, options) {
+    if (queryInfo.top === 0 || queryInfo.limit === 0) return 0;
+    return queryInfo.top ? queryInfo.top : queryInfo.limit ? queryInfo.offset + queryInfo.limit : options["vectorSearchBufferSize"] && options["vectorSearchBufferSize"] > 0 ? options["vectorSearchBufferSize"] : import_constants.QueryExecution.DEFAULT_MAX_VECTOR_SEARCH_BUFFER_SIZE;
+  }
+  checkQueryConstraints(queryInfo) {
+    const hasTop = queryInfo.top || queryInfo.top === 0;
+    const hasLimit = queryInfo.limit || queryInfo.limit === 0;
+    if (!hasTop && !hasLimit) {
+      throw new import_ErrorResponse.ErrorResponse(
+        "Executing a non-streaming search query without TOP or LIMIT can consume a large number of RUs very fast and have long runtimes. Please ensure you are using one of the above two filters with your vector search query."
+      );
+    }
+    return;
+  }
+  /**
+   * Analyzes query information and extracts key characteristics for query execution planning
+   */
+  analyzeQueryInfo(queryInfo) {
+    const sortOrders = queryInfo.orderBy;
+    const nonStreamingOrderBy = queryInfo.hasNonStreamingOrderBy;
+    const isOrderByQuery = Array.isArray(sortOrders) && sortOrders.length > 0;
+    const isGroupByQuery = Object.keys(queryInfo.groupByAliasToAggregateType || {}).length > 0 || (queryInfo.aggregates?.length || 0) > 0 || (queryInfo.groupByExpressions?.length || 0) > 0;
+    const isUnorderedDistinctQuery = queryInfo.distinctType === "Unordered";
+    const querySupportsTokens = !isUnorderedDistinctQuery && !isGroupByQuery && !nonStreamingOrderBy;
+    return {
+      sortOrders,
+      nonStreamingOrderBy,
+      isOrderByQuery,
+      isGroupByQuery,
+      isUnorderedDistinctQuery,
+      querySupportsTokens
+    };
+  }
 }
-exports.PipelinedQueryExecutionContext = PipelinedQueryExecutionContext;
-//# sourceMappingURL=pipelinedQueryExecutionContext.js.map
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  PipelinedQueryExecutionContext
+});

package/dist/commonjs/queryExecutionContext/pipelinedQueryExecutionContext.js.map

@@ -1 +1,7 @@
-{"version":3,"file":"pipelinedQueryExecutionContext.js","sourceRoot":"","sources":["../../../src/queryExecutionContext/pipelinedQueryExecutionContext.ts"],"names":[],"mappings":";;;AAKA,kEAA4D;AAC5D,yGAAmG;AACnG,iGAA2F;AAC3F,iHAA2G;AAC3G,qHAA+G;AAC/G,iGAA2F;AAE3F,uFAAiF;AACjF,yFAAmF;AACnF,2GAAqG;AAGrG,yIAAmI;AACnI,yHAAmH;AACnH,yEAGoC;AACpC,6EAA4E;AAC5E,iFAA2E;AAC3E,6FAAuF;AACvF,yDAAwD;AAExD,cAAc;AACd,MAAa,8BAA8B;IAM/B;IACA;IACA;IACA;IACA;IAEA;IACA;IAZF,WAAW,CAAQ;IACnB,QAAQ,CAAmB;IAClB,mBAAmB,CAA8D;IAElG,YACU,aAA4B,EAC5B,cAAsB,EACtB,KAA4B,EAC5B,OAAoB,EACpB,6BAA4D,EACpE,oBAA4B,EACpB,wBAAiC,KAAK,EACtC,6BAAsC,IAAI;QAP1C,kBAAa,GAAb,aAAa,CAAe;QAC5B,mBAAc,GAAd,cAAc,CAAQ;QACtB,UAAK,GAAL,KAAK,CAAuB;QAC5B,YAAO,GAAP,OAAO,CAAa;QACpB,kCAA6B,GAA7B,6BAA6B,CAA+B;QAE5D,0BAAqB,GAArB,qBAAqB,CAAiB;QACtC,+BAA0B,GAA1B,0BAA0B,CAAgB;QAElD,yEAAyE;QACzE,IAAI,CAAC,6BAA6B,CAAC,SAAS,EAAE,CAAC;YAC7C,MAAM,IAAI,gCAAa,CACrB,yDAAyD;gBACvD,6DAA6D;gBAC7D,sEAAsE,CACzE,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC;YAC/B,IAAI,CAAC,OAAO,CAAC,YAAY,GAAG,6BAAc,CAAC,iBAAiB,CAAC;QAC/D,CAAC;QACD,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC;QAE3C,gDAAgD;QAChD,MAAM,iBAAiB,GAAG,IAAI,CAAC,gBAAgB,CAAC,6BAA6B,CAAC,SAAS,CAAC,CAAC;QACzF,MAAM,EACJ,UAAU,EACV,mBAAmB,EACnB,cAAc,EACd,cAAc,EACd,wBAAwB,EACxB,mBAAmB,GACpB,GAAG,iBAAiB,CAAC;QAEtB,8DAA8D;QAC9D,IAAI,CAAC,mBAAmB,EAAE,CAAC;YACzB,IAAA,uEAA4C,EAAC,IAAI,CAAC,OAAO,CAAC,iBAAiB,EAAE;gBAC3E,qCAAU,CAAC,mBAAmB,CAAC,mBAAmB,CAAC;gBACnD,qCAAU,CAAC,OAAO,CAAC,cAAc,CAAC;gBAClC,qCAAU,CAAC,iBAAiB,CAAC,wBAAwB,CAAC;aACvD,CAAC,CAAC;QACL,CAAC;QAED,0EAA0E;QAC1E,MAAM,uBAAuB,GAAG,IAAI,CAAC,OAAO,CAAC,iBAAiB;YAC5D,CAAC,CAAC,IAAA,yDAA4B,EAAC,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC;YAC9D,CAAC,CAAC,SAAS,CAAC;QAEd,4DAA4D;QAC5D,IAAI,CAAC,QAAQ,GAAG,mBAAmB;YACjC,CAAC,CAAC,IAAI,CAAC,0BAA0B,CAC7B,6BAA6B,EAC7B,UAAU,EACV,oBAAoB,EACpB,OAAO,CACR;YACH,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAC1B,6BAA6B,EAC7B,UAAU,EACV,oBAAoB,EACpB,cAAc,EACd,uBAAuB,CACxB,CAAC;QACN,IAAI,CAAC,WAAW,GAAG,EAAE,CAAC;QACtB,8EAA8E;QAC9E,IAAI,IAAI,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC;YACpC,MAAM,+BAA+B,GACnC,IAAI,CAAC,0BAA0B,IAAI,mBAAmB,CAAC;YACzD,IAAI,CAAC,mBAAmB,GAAG,IAAI,oEAA+B,CAC5D,IAAI,CAAC,QAAQ,EACb,QAAQ,EACR,IAAI,CAAC,cAAc,EACnB,IAAI,CAAC,OAAO,CAAC,iBAAiB,EAC9B,cAAc,EACd,+BAA+B,CAChC,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,mBAAmB,GAAG,IAAI,wDAAyB,CAAC,IAAI,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QACpF,CAAC;IACH,CAAC;IAEM,cAAc;QACnB,OAAO,IAAI,CAAC,WAAW,CAAC,MAAM,KAAK,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,cAAc,EAAE,CAAC;IACzE,CAAC;IAEM,KAAK,CAAC,SAAS,CAAC,cAAsC;QAC3D,OAAO,IAAI,CAAC,mBAAmB,CAAC,SAAS,CAAC,cAAc,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;IAC9E,CAAC;IAED;;OAEG;IACK,0BAA0B,CAChC,6BAA4D,EAC5D,UAAiB,EACjB,oBAA4B,EAC5B,OAAoB;QAEpB,MAAM,SAAS,GAAG,6BAA6B,CAAC,SAAU,CAAC,CAAC,gDAAgD;QAE5G,IAAI,CAAC,OAAO,CAAC,iCAAiC,EAAE,CAAC;YAC/C,IAAI,CAAC,qBAAqB,CAAC,SAAS,CAAC,CAAC;QACxC,CAAC;QAED,MAAM,sBAAsB,GAAG,IAAI,CAAC,+BAA+B,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QAExF,IAAI,CAAC,8BAA8B,CAAC,sBAAsB,EAAE,OAAO,CAAC,CAAC;QAErE,MAAM,WAAW,GAAG,IAAI,gEAA6B,CACnD,IAAI,CAAC,aAAa,EAClB,IAAI,CAAC,cAAc,EACnB,IAAI,CAAC,KAAK,EACV,IAAI,CAAC,OAAO,EACZ,IAAI,CAAC,6BAA6B,EAClC,oBAAoB,CACrB,CAAC;QAEF,OAAO,IAAI,CAAC,6BAA6B,CACvC,WAAW,EACX,SAAS,EACT,UAAU,EACV,sBAAsB,CACvB,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,uBAAuB,CAC7B,6BAA4D,EAC5D,UAAiB,EACjB,oBAA4B,EAC5B,cAAuB,EACvB,uBAA4B;QAE5B,MAAM,SAAS,GAAG,6BAA6B,CAAC,SAAU,CAAC,CAAC,gDAAgD;QAE5G,gCAAgC;QAChC,IAAI,QAAQ,GAAG,IAAI,CAAC,0BAA0B,CAC5C,6BAA6B,EAC7B,UAAU,EACV,oBAAoB,CACrB,CAAC;QAEF,iCAAiC;QACjC,QAAQ,GAAG,IAAI,CAAC,sBAAsB,CAAC,QAAQ,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC;QAC5E,QAAQ,GAAG,IAAI,CAAC,uBAAuB,CAAC,QAAQ,EAAE,SAAS,EAAE,uBAAuB,CAAC,CAAC;QACtF,QAAQ,GAAG,IAAI,CAAC,oBAAoB,CAAC,QAAQ,EAAE,SAAS,EAAE,uBAAuB,CAAC,CAAC;QAEnF,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;OAEG;IACK,0BAA0B,CAChC,6BAA4D,EAC5D,UAAiB,EACjB,oBAA4B;QAE5B,IAAI,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvD,8DAA8D;YAC9D,OAAO,IAAI,sDAAwB,CACjC,IAAI,8DAA4B,CAC9B,IAAI,CAAC,aAAa,EAClB,IAAI,CAAC,cAAc,EACnB,IAAI,CAAC,KAAK,EACV,IAAI,CAAC,OAAO,EACZ,6BAA6B,EAC7B,oBAAoB,CACrB,EACD,IAAI,CAAC,qBAAqB,CAC3B,CAAC;QACJ,CAAC;QAED,mBAAmB;QACnB,OAAO,IAAI,gEAA6B,CACtC,IAAI,CAAC,aAAa,EAClB,IAAI,CAAC,cAAc,EACnB,IAAI,CAAC,KAAK,EACV,IAAI,CAAC,OAAO,EACZ,6BAA6B,EAC7B,oBAAoB,CACrB,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,6BAA6B,CACnC,WAA6B,EAC7B,SAAoB,EACpB,UAAiB,EACjB,sBAA8B;QAE9B,MAAM,YAAY,GAAG,SAAS,CAAC,YAAY,CAAC;QAE5C,IAAI,YAAY,KAAK,MAAM,EAAE,CAAC;YAC5B,OAAO,IAAI,8EAAoC,CAC7C,WAAW,EACX,UAAU,EACV,sBAAsB,EACtB,SAAS,CAAC,MAAM,EAChB,IAAI,CAAC,qBAAqB,CAC3B,CAAC;QACJ,CAAC;QAED,OAAO,IAAI,8FAA4C,CACrD,WAAW,EACX,SAAS,EACT,sBAAsB,EACtB,IAAI,CAAC,qBAAqB,CAC3B,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,sBAAsB,CAC5B,QAA0B,EAC1B,SAAoB,EACpB,cAAuB;QAEvB,IAAI,CAAC,cAAc,EAAE,CAAC;YACpB,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,OAAO,SAAS,CAAC,cAAc;YAC7B,CAAC,CAAC,IAAI,gEAA6B,CAAC,QAAQ,EAAE,SAAS,CAAC;YACxD,CAAC,CAAC,IAAI,sDAAwB,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;IACxD,CAAC;IAED;;OAEG;IACK,uBAAuB,CAC7B,QAA0B,EAC1B,SAAoB,EACpB,uBAA4B;QAE5B,MAAM,YAAY,GAAG,SAAS,CAAC,YAAY,CAAC;QAE5C,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;YAC/B,MAAM,QAAQ,GAAG,uBAAuB,EAAE,gBAAgB,CAAC;YAC3D,OAAO,IAAI,sEAAgC,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QAClE,CAAC;QAED,IAAI,YAAY,KAAK,WAAW,EAAE,CAAC;YACjC,OAAO,IAAI,0EAAkC,CAAC,QAAQ,CAAC,CAAC;QAC1D,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;OAEG;IACK,oBAAoB,CAC1B,QAA0B,EAC1B,SAAoB,EACpB,uBAA4B;QAE5B,8DAA8D;QAC9D,IAAI,GAAG,GAAG,SAAS,CAAC,GAAG,CAAC;QACxB,IAAI,OAAO,GAAG,KAAK,QAAQ,EAAE,CAAC;YAC5B,IAAI,uBAAuB,EAAE,KAAK,KAAK,SAAS,EAAE,CAAC;gBACjD,GAAG,GAAG,uBAAuB,CAAC,KAAK,CAAC;YACtC,CAAC;YACD,QAAQ,GAAG,IAAI,8DAA4B,CAAC,QAAQ,EAAE,CAAC,EAAE,GAAG,CAAC,CAAC;QAChE,CAAC;QAED,+BAA+B;QAC/B,IAAI,KAAK,GAAG,SAAS,CAAC,KAAK,CAAC;QAC5B,IAAI,MAAM,GAAG,SAAS,CAAC,MAAM,CAAC;QAE9B,IAAI,uBAAuB,EAAE,CAAC;YAC5B,IAAI,uBAAuB,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;gBAChD,KAAK,GAAG,uBAAuB,CAAC,KAAK,CAAC;YACxC,CAAC;YACD,IAAI,uBAAuB,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;gBACjD,MAAM,GAAG,uBAAuB,CAAC,MAAM,CAAC;YAC1C,CAAC;QACH,CAAC;QAED,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC5D,QAAQ,GAAG,IAAI,8DAA4B,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;QACvE,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;OAEG;IACK,8BAA8B,CACpC,sBAA8B,EAC9B,OAAoB;QAEpB,MAAM,aAAa,GAAG,OAAO,CAAC,wBAAwB,CAAC;YACrD,CAAC,CAAC,OAAO,CAAC,wBAAwB,CAAC;YACnC,CAAC,CAAC,6BAAc,CAAC,qCAAqC,CAAC;QAEzD,IAAI,sBAAsB,GAAG,aAAa,EAAE,CAAC;YAC3C,MAAM,IAAI,gCAAa,CACrB,oEAAoE,sBAAsB,2CAA2C,aAAa,GAAG;gBACnJ,gBAAgB,CACnB,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,+BAA+B,CAAC,SAAoB,EAAE,OAAoB;QAChF,IAAI,SAAS,CAAC,GAAG,KAAK,CAAC,IAAI,SAAS,CAAC,KAAK,KAAK,CAAC;YAAE,OAAO,CAAC,CAAC;QAC3D,OAAO,SAAS,CAAC,GAAG;YAClB,CAAC,CAAC,SAAS,CAAC,GAAG;YACf,CAAC,CAAC,SAAS,CAAC,KAAK;gBACf,CAAC,CAAC,SAAS,CAAC,MAAM,GAAG,SAAS,CAAC,KAAK;gBACpC,CAAC,CAAC,OAAO,CAAC,wBAAwB,CAAC,IAAI,OAAO,CAAC,wBAAwB,CAAC,GAAG,CAAC;oBAC1E,CAAC,CAAC,OAAO,CAAC,wBAAwB,CAAC;oBACnC,CAAC,CAAC,6BAAc,CAAC,qCAAqC,CAAC;IAC/D,CAAC;IAEO,qBAAqB,CAAC,SAAoB;QAChD,MAAM,MAAM,GAAG,SAAS,CAAC,GAAG,IAAI,SAAS,CAAC,GAAG,KAAK,CAAC,CAAC;QACpD,MAAM,QAAQ,GAAG,SAAS,CAAC,KAAK,IAAI,SAAS,CAAC,KAAK,KAAK,CAAC,CAAC;QAC1D,IAAI,CAAC,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YACzB,MAAM,IAAI,gCAAa,CACrB,gGAAgG;gBAC9F,6FAA6F;gBAC7F,gCAAgC,CACnC,CAAC;QACJ,CAAC;QACD,OAAO;IACT,CAAC;IAED;;OAEG;IACK,gBAAgB,CAAC,SAAoB;QAC3C,MAAM,UAAU,GAAG,SAAS,CAAC,OAAO,CAAC;QACrC,MAAM,mBAAmB,GAAG,SAAS,CAAC,sBAAsB,CAAC;QAC7D,MAAM,cAAc,GAAG,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC;QAE1E,oCAAoC;QACpC,MAAM,cAAc,GAClB,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,2BAA2B,IAAI,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC;YACnE,CAAC,SAAS,CAAC,UAAU,EAAE,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC;YACvC,CAAC,SAAS,CAAC,kBAAkB,EAAE,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;QAElD,+CAA+C;QAC/C,MAAM,wBAAwB,GAAG,SAAS,CAAC,YAAY,KAAK,WAAW,CAAC;QAExE,4DAA4D;QAC5D,MAAM,mBAAmB,GACvB,CAAC,wBAAwB,IAAI,CAAC,cAAc,IAAI,CAAC,mBAAmB,CAAC;QAEvE,OAAO;YACL,UAAU;YACV,mBAAmB;YACnB,cAAc;YACd,cAAc;YACd,wBAAwB;YACxB,mBAAmB;SACpB,CAAC;IACJ,CAAC;CACF;AAlXD,wEAkXC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\nimport type { ClientContext } from \"../ClientContext.js\";\nimport type { Response, FeedOptions } from \"../request/index.js\";\nimport type { PartitionedQueryExecutionInfo, QueryInfo } from \"../request/ErrorResponse.js\";\nimport { ErrorResponse } from \"../request/ErrorResponse.js\";\nimport { OffsetLimitEndpointComponent } from \"./EndpointComponent/OffsetLimitEndpointComponent.js\";\nimport { OrderByEndpointComponent } from \"./EndpointComponent/OrderByEndpointComponent.js\";\nimport { OrderedDistinctEndpointComponent } from \"./EndpointComponent/OrderedDistinctEndpointComponent.js\";\nimport { UnorderedDistinctEndpointComponent } from \"./EndpointComponent/UnorderedDistinctEndpointComponent.js\";\nimport { GroupByEndpointComponent } from \"./EndpointComponent/GroupByEndpointComponent.js\";\nimport type { ExecutionContext } from \"./ExecutionContext.js\";\nimport { OrderByQueryExecutionContext } from \"./orderByQueryExecutionContext.js\";\nimport { ParallelQueryExecutionContext } from \"./parallelQueryExecutionContext.js\";\nimport { GroupByValueEndpointComponent } from \"./EndpointComponent/GroupByValueEndpointComponent.js\";\nimport type { SqlQuerySpec } from \"./SqlQuerySpec.js\";\nimport type { DiagnosticNodeInternal } from \"../diagnostics/DiagnosticNodeInternal.js\";\nimport { NonStreamingOrderByDistinctEndpointComponent } from \"./EndpointComponent/NonStreamingOrderByDistinctEndpointComponent.js\";\nimport { NonStreamingOrderByEndpointComponent } from \"./EndpointComponent/NonStreamingOrderByEndpointComponent.js\";\nimport {\n  rejectContinuationTokenForUnsupportedQueries,\n  QueryTypes,\n} from \"./QueryValidationHelper.js\";\nimport { parseContinuationTokenFields } from \"./ContinuationTokenParser.js\";\nimport { LegacyFetchImplementation } from \"./LegacyFetchImplementation.js\";\nimport { QueryControlFetchImplementation } from \"./QueryControlFetchImplementation.js\";\nimport { QueryExecution } from \"../common/constants.js\";\n\n/** @hidden */\nexport class PipelinedQueryExecutionContext implements ExecutionContext {\n  private fetchBuffer: any[];\n  private endpoint: ExecutionContext;\n  private readonly fetchImplementation: LegacyFetchImplementation | QueryControlFetchImplementation;\n\n  constructor(\n    private clientContext: ClientContext,\n    private collectionLink: string,\n    private query: string | SqlQuerySpec,\n    private options: FeedOptions,\n    private partitionedQueryExecutionInfo: PartitionedQueryExecutionInfo,\n    correlatedActivityId: string,\n    private emitRawOrderByPayload: boolean = false,\n    private supportsContinuationTokens: boolean = true,\n  ) {\n    // Validate that queryInfo is present in partitioned query execution info\n    if (!partitionedQueryExecutionInfo.queryInfo) {\n      throw new ErrorResponse(\n        \"Query execution requires valid query plan information. \" +\n          \"The partitioned query execution info is missing queryInfo. \" +\n          \"This may indicate an invalid query or a problem with query planning.\",\n      );\n    }\n\n    if (!this.options.maxItemCount) {\n      this.options.maxItemCount = QueryExecution.DEFAULT_PAGE_SIZE;\n    }\n    const pageSize = this.options.maxItemCount;\n\n    // Extract query information and characteristics\n    const analyzedQueryInfo = this.analyzeQueryInfo(partitionedQueryExecutionInfo.queryInfo);\n    const {\n      sortOrders,\n      nonStreamingOrderBy,\n      isOrderByQuery,\n      isGroupByQuery,\n      isUnorderedDistinctQuery,\n      querySupportsTokens,\n    } = analyzedQueryInfo;\n\n    // Reject continuation token usage for unsupported query types\n    if (!querySupportsTokens) {\n      rejectContinuationTokenForUnsupportedQueries(this.options.continuationToken, [\n        QueryTypes.nonStreamingOrderBy(nonStreamingOrderBy),\n        QueryTypes.groupBy(isGroupByQuery),\n        QueryTypes.unorderedDistinct(isUnorderedDistinctQuery),\n      ]);\n    }\n\n    // Parse continuation token fields once for reuse in pipeline construction\n    const queryContinuationFields = this.options.continuationToken\n      ? parseContinuationTokenFields(this.options.continuationToken)\n      : undefined;\n\n    // Pick between non-streaming vs streaming execution context\n    this.endpoint = nonStreamingOrderBy\n      ? this.createNonStreamingEndpoint(\n          partitionedQueryExecutionInfo,\n          sortOrders,\n          correlatedActivityId,\n          options,\n        )\n      : this.createStreamingEndpoint(\n          partitionedQueryExecutionInfo,\n          sortOrders,\n          correlatedActivityId,\n          isGroupByQuery,\n          queryContinuationFields,\n        );\n    this.fetchBuffer = [];\n    // Initialize the appropriate fetch implementation based on enableQueryControl\n    if (this.options.enableQueryControl) {\n      const querySupportsContinuationTokens =\n        this.supportsContinuationTokens && querySupportsTokens;\n      this.fetchImplementation = new QueryControlFetchImplementation(\n        this.endpoint,\n        pageSize,\n        this.collectionLink,\n        this.options.continuationToken,\n        isOrderByQuery,\n        querySupportsContinuationTokens,\n      );\n    } else {\n      this.fetchImplementation = new LegacyFetchImplementation(this.endpoint, pageSize);\n    }\n  }\n\n  public hasMoreResults(): boolean {\n    return this.fetchBuffer.length !== 0 || this.endpoint.hasMoreResults();\n  }\n\n  public async fetchMore(diagnosticNode: DiagnosticNodeInternal): Promise<Response<any>> {\n    return this.fetchImplementation.fetchMore(diagnosticNode, this.fetchBuffer);\n  }\n\n  /**\n   * Creates a non-streaming endpoint for vector search and similar queries that require buffering\n   */\n  private createNonStreamingEndpoint(\n    partitionedQueryExecutionInfo: PartitionedQueryExecutionInfo,\n    sortOrders: any[],\n    correlatedActivityId: string,\n    options: FeedOptions,\n  ): ExecutionContext {\n    const queryInfo = partitionedQueryExecutionInfo.queryInfo!; // Safe to use ! after validation in constructor\n\n    if (!options.allowUnboundedNonStreamingQueries) {\n      this.checkQueryConstraints(queryInfo);\n    }\n\n    const vectorSearchBufferSize = this.calculateVectorSearchBufferSize(queryInfo, options);\n\n    this.validateVectorSearchBufferSize(vectorSearchBufferSize, options);\n\n    const baseContext = new ParallelQueryExecutionContext(\n      this.clientContext,\n      this.collectionLink,\n      this.query,\n      this.options,\n      this.partitionedQueryExecutionInfo,\n      correlatedActivityId,\n    );\n\n    return this.wrapWithNonStreamingComponent(\n      baseContext,\n      queryInfo,\n      sortOrders,\n      vectorSearchBufferSize,\n    );\n  }\n\n  /**\n   * Creates a streaming endpoint with proper pipeline components\n   */\n  private createStreamingEndpoint(\n    partitionedQueryExecutionInfo: PartitionedQueryExecutionInfo,\n    sortOrders: any[],\n    correlatedActivityId: string,\n    isGroupByQuery: boolean,\n    queryContinuationFields: any,\n  ): ExecutionContext {\n    const queryInfo = partitionedQueryExecutionInfo.queryInfo!; // Safe to use ! after validation in constructor\n\n    // Create base execution context\n    let endpoint = this.createBaseExecutionContext(\n      partitionedQueryExecutionInfo,\n      sortOrders,\n      correlatedActivityId,\n    );\n\n    // Apply pipeline transformations\n    endpoint = this.applyGroupByComponents(endpoint, queryInfo, isGroupByQuery);\n    endpoint = this.applyDistinctComponents(endpoint, queryInfo, queryContinuationFields);\n    endpoint = this.applyLimitComponents(endpoint, queryInfo, queryContinuationFields);\n\n    return endpoint;\n  }\n\n  /**\n   * Creates the base execution context (OrderBy or Parallel)\n   */\n  private createBaseExecutionContext(\n    partitionedQueryExecutionInfo: PartitionedQueryExecutionInfo,\n    sortOrders: any[],\n    correlatedActivityId: string,\n  ): ExecutionContext {\n    if (Array.isArray(sortOrders) && sortOrders.length > 0) {\n      // OrderBy queries need special wrapping for payload structure\n      return new OrderByEndpointComponent(\n        new OrderByQueryExecutionContext(\n          this.clientContext,\n          this.collectionLink,\n          this.query,\n          this.options,\n          partitionedQueryExecutionInfo,\n          correlatedActivityId,\n        ),\n        this.emitRawOrderByPayload,\n      );\n    }\n\n    // Parallel queries\n    return new ParallelQueryExecutionContext(\n      this.clientContext,\n      this.collectionLink,\n      this.query,\n      this.options,\n      partitionedQueryExecutionInfo,\n      correlatedActivityId,\n    );\n  }\n\n  /**\n   * Wraps base context with appropriate non-streaming component\n   */\n  private wrapWithNonStreamingComponent(\n    baseContext: ExecutionContext,\n    queryInfo: QueryInfo,\n    sortOrders: any[],\n    vectorSearchBufferSize: number,\n  ): ExecutionContext {\n    const distinctType = queryInfo.distinctType;\n\n    if (distinctType === \"None\") {\n      return new NonStreamingOrderByEndpointComponent(\n        baseContext,\n        sortOrders,\n        vectorSearchBufferSize,\n        queryInfo.offset,\n        this.emitRawOrderByPayload,\n      );\n    }\n\n    return new NonStreamingOrderByDistinctEndpointComponent(\n      baseContext,\n      queryInfo,\n      vectorSearchBufferSize,\n      this.emitRawOrderByPayload,\n    );\n  }\n\n  /**\n   * Applies GROUP BY components to the pipeline if needed\n   */\n  private applyGroupByComponents(\n    endpoint: ExecutionContext,\n    queryInfo: QueryInfo,\n    isGroupByQuery: boolean,\n  ): ExecutionContext {\n    if (!isGroupByQuery) {\n      return endpoint;\n    }\n\n    return queryInfo.hasSelectValue\n      ? new GroupByValueEndpointComponent(endpoint, queryInfo)\n      : new GroupByEndpointComponent(endpoint, queryInfo);\n  }\n\n  /**\n   * Applies DISTINCT components to the pipeline if needed\n   */\n  private applyDistinctComponents(\n    endpoint: ExecutionContext,\n    queryInfo: QueryInfo,\n    queryContinuationFields: any,\n  ): ExecutionContext {\n    const distinctType = queryInfo.distinctType;\n\n    if (distinctType === \"Ordered\") {\n      const lastHash = queryContinuationFields?.hashedLastResult;\n      return new OrderedDistinctEndpointComponent(endpoint, lastHash);\n    }\n\n    if (distinctType === \"Unordered\") {\n      return new UnorderedDistinctEndpointComponent(endpoint);\n    }\n\n    return endpoint;\n  }\n\n  /**\n   * Applies TOP and OFFSET+LIMIT components to the pipeline if needed\n   */\n  private applyLimitComponents(\n    endpoint: ExecutionContext,\n    queryInfo: QueryInfo,\n    queryContinuationFields: any,\n  ): ExecutionContext {\n    // Apply TOP component (TOP N is effectively OFFSET 0 LIMIT N)\n    let top = queryInfo.top;\n    if (typeof top === \"number\") {\n      if (queryContinuationFields?.limit !== undefined) {\n        top = queryContinuationFields.limit;\n      }\n      endpoint = new OffsetLimitEndpointComponent(endpoint, 0, top);\n    }\n\n    // Apply OFFSET+LIMIT component\n    let limit = queryInfo.limit;\n    let offset = queryInfo.offset;\n\n    if (queryContinuationFields) {\n      if (queryContinuationFields.limit !== undefined) {\n        limit = queryContinuationFields.limit;\n      }\n      if (queryContinuationFields.offset !== undefined) {\n        offset = queryContinuationFields.offset;\n      }\n    }\n\n    if (typeof limit === \"number\" && typeof offset === \"number\") {\n      endpoint = new OffsetLimitEndpointComponent(endpoint, offset, limit);\n    }\n\n    return endpoint;\n  }\n\n  /**\n   * Validates vector search buffer size constraints\n   */\n  private validateVectorSearchBufferSize(\n    vectorSearchBufferSize: number,\n    options: FeedOptions,\n  ): void {\n    const maxBufferSize = options[\"vectorSearchBufferSize\"]\n      ? options[\"vectorSearchBufferSize\"]\n      : QueryExecution.DEFAULT_MAX_VECTOR_SEARCH_BUFFER_SIZE;\n\n    if (vectorSearchBufferSize > maxBufferSize) {\n      throw new ErrorResponse(\n        `Executing a vector search query with TOP or OFFSET + LIMIT value ${vectorSearchBufferSize} larger than the vectorSearchBufferSize ${maxBufferSize} ` +\n          `is not allowed`,\n      );\n    }\n  }\n\n  private calculateVectorSearchBufferSize(queryInfo: QueryInfo, options: FeedOptions): number {\n    if (queryInfo.top === 0 || queryInfo.limit === 0) return 0;\n    return queryInfo.top\n      ? queryInfo.top\n      : queryInfo.limit\n        ? queryInfo.offset + queryInfo.limit\n        : options[\"vectorSearchBufferSize\"] && options[\"vectorSearchBufferSize\"] > 0\n          ? options[\"vectorSearchBufferSize\"]\n          : QueryExecution.DEFAULT_MAX_VECTOR_SEARCH_BUFFER_SIZE;\n  }\n\n  private checkQueryConstraints(queryInfo: QueryInfo): void {\n    const hasTop = queryInfo.top || queryInfo.top === 0;\n    const hasLimit = queryInfo.limit || queryInfo.limit === 0;\n    if (!hasTop && !hasLimit) {\n      throw new ErrorResponse(\n        \"Executing a non-streaming search query without TOP or LIMIT can consume a large number of RUs \" +\n          \"very fast and have long runtimes. Please ensure you are using one of the above two filters \" +\n          \"with your vector search query.\",\n      );\n    }\n    return;\n  }\n\n  /**\n   * Analyzes query information and extracts key characteristics for query execution planning\n   */\n  private analyzeQueryInfo(queryInfo: QueryInfo) {\n    const sortOrders = queryInfo.orderBy;\n    const nonStreamingOrderBy = queryInfo.hasNonStreamingOrderBy;\n    const isOrderByQuery = Array.isArray(sortOrders) && sortOrders.length > 0;\n\n    // Check if this is a GROUP BY query\n    const isGroupByQuery =\n      Object.keys(queryInfo.groupByAliasToAggregateType || {}).length > 0 ||\n      (queryInfo.aggregates?.length || 0) > 0 ||\n      (queryInfo.groupByExpressions?.length || 0) > 0;\n\n    // Check if this is an unordered DISTINCT query\n    const isUnorderedDistinctQuery = queryInfo.distinctType === \"Unordered\";\n\n    // Determine if this query type supports continuation tokens\n    const querySupportsTokens =\n      !isUnorderedDistinctQuery && !isGroupByQuery && !nonStreamingOrderBy;\n\n    return {\n      sortOrders,\n      nonStreamingOrderBy,\n      isOrderByQuery,\n      isGroupByQuery,\n      isUnorderedDistinctQuery,\n      querySupportsTokens,\n    };\n  }\n}\n"]}
+{
+  "version": 3,
+  "sources": ["/mnt/vss/_work/1/s/sdk/cosmosdb/cosmos/src/queryExecutionContext/pipelinedQueryExecutionContext.ts"],
+  "sourcesContent": ["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\nimport type { ClientContext } from \"../ClientContext.js\";\nimport type { Response, FeedOptions } from \"../request/index.js\";\nimport type { PartitionedQueryExecutionInfo, QueryInfo } from \"../request/ErrorResponse.js\";\nimport { ErrorResponse } from \"../request/ErrorResponse.js\";\nimport { OffsetLimitEndpointComponent } from \"./EndpointComponent/OffsetLimitEndpointComponent.js\";\nimport { OrderByEndpointComponent } from \"./EndpointComponent/OrderByEndpointComponent.js\";\nimport { OrderedDistinctEndpointComponent } from \"./EndpointComponent/OrderedDistinctEndpointComponent.js\";\nimport { UnorderedDistinctEndpointComponent } from \"./EndpointComponent/UnorderedDistinctEndpointComponent.js\";\nimport { GroupByEndpointComponent } from \"./EndpointComponent/GroupByEndpointComponent.js\";\nimport type { ExecutionContext } from \"./ExecutionContext.js\";\nimport { OrderByQueryExecutionContext } from \"./orderByQueryExecutionContext.js\";\nimport { ParallelQueryExecutionContext } from \"./parallelQueryExecutionContext.js\";\nimport { GroupByValueEndpointComponent } from \"./EndpointComponent/GroupByValueEndpointComponent.js\";\nimport type { SqlQuerySpec } from \"./SqlQuerySpec.js\";\nimport type { DiagnosticNodeInternal } from \"../diagnostics/DiagnosticNodeInternal.js\";\nimport { NonStreamingOrderByDistinctEndpointComponent } from \"./EndpointComponent/NonStreamingOrderByDistinctEndpointComponent.js\";\nimport { NonStreamingOrderByEndpointComponent } from \"./EndpointComponent/NonStreamingOrderByEndpointComponent.js\";\nimport {\n  rejectContinuationTokenForUnsupportedQueries,\n  QueryTypes,\n} from \"./QueryValidationHelper.js\";\nimport { parseContinuationTokenFields } from \"./ContinuationTokenParser.js\";\nimport { LegacyFetchImplementation } from \"./LegacyFetchImplementation.js\";\nimport { QueryControlFetchImplementation } from \"./QueryControlFetchImplementation.js\";\nimport { QueryExecution } from \"../common/constants.js\";\n\n/** @hidden */\nexport class PipelinedQueryExecutionContext implements ExecutionContext {\n  private fetchBuffer: any[];\n  private endpoint: ExecutionContext;\n  private readonly fetchImplementation: LegacyFetchImplementation | QueryControlFetchImplementation;\n\n  constructor(\n    private clientContext: ClientContext,\n    private collectionLink: string,\n    private query: string | SqlQuerySpec,\n    private options: FeedOptions,\n    private partitionedQueryExecutionInfo: PartitionedQueryExecutionInfo,\n    correlatedActivityId: string,\n    private emitRawOrderByPayload: boolean = false,\n    private supportsContinuationTokens: boolean = true,\n  ) {\n    // Validate that queryInfo is present in partitioned query execution info\n    if (!partitionedQueryExecutionInfo.queryInfo) {\n      throw new ErrorResponse(\n        \"Query execution requires valid query plan information. \" +\n          \"The partitioned query execution info is missing queryInfo. \" +\n          \"This may indicate an invalid query or a problem with query planning.\",\n      );\n    }\n\n    if (!this.options.maxItemCount) {\n      this.options.maxItemCount = QueryExecution.DEFAULT_PAGE_SIZE;\n    }\n    const pageSize = this.options.maxItemCount;\n\n    // Extract query information and characteristics\n    const analyzedQueryInfo = this.analyzeQueryInfo(partitionedQueryExecutionInfo.queryInfo);\n    const {\n      sortOrders,\n      nonStreamingOrderBy,\n      isOrderByQuery,\n      isGroupByQuery,\n      isUnorderedDistinctQuery,\n      querySupportsTokens,\n    } = analyzedQueryInfo;\n\n    // Reject continuation token usage for unsupported query types\n    if (!querySupportsTokens) {\n      rejectContinuationTokenForUnsupportedQueries(this.options.continuationToken, [\n        QueryTypes.nonStreamingOrderBy(nonStreamingOrderBy),\n        QueryTypes.groupBy(isGroupByQuery),\n        QueryTypes.unorderedDistinct(isUnorderedDistinctQuery),\n      ]);\n    }\n\n    // Parse continuation token fields once for reuse in pipeline construction\n    const queryContinuationFields = this.options.continuationToken\n      ? parseContinuationTokenFields(this.options.continuationToken)\n      : undefined;\n\n    // Pick between non-streaming vs streaming execution context\n    this.endpoint = nonStreamingOrderBy\n      ? this.createNonStreamingEndpoint(\n          partitionedQueryExecutionInfo,\n          sortOrders,\n          correlatedActivityId,\n          options,\n        )\n      : this.createStreamingEndpoint(\n          partitionedQueryExecutionInfo,\n          sortOrders,\n          correlatedActivityId,\n          isGroupByQuery,\n          queryContinuationFields,\n        );\n    this.fetchBuffer = [];\n    // Initialize the appropriate fetch implementation based on enableQueryControl\n    if (this.options.enableQueryControl) {\n      const querySupportsContinuationTokens =\n        this.supportsContinuationTokens && querySupportsTokens;\n      this.fetchImplementation = new QueryControlFetchImplementation(\n        this.endpoint,\n        pageSize,\n        this.collectionLink,\n        this.options.continuationToken,\n        isOrderByQuery,\n        querySupportsContinuationTokens,\n      );\n    } else {\n      this.fetchImplementation = new LegacyFetchImplementation(this.endpoint, pageSize);\n    }\n  }\n\n  public hasMoreResults(): boolean {\n    return this.fetchBuffer.length !== 0 || this.endpoint.hasMoreResults();\n  }\n\n  public async fetchMore(diagnosticNode: DiagnosticNodeInternal): Promise<Response<any>> {\n    return this.fetchImplementation.fetchMore(diagnosticNode, this.fetchBuffer);\n  }\n\n  /**\n   * Creates a non-streaming endpoint for vector search and similar queries that require buffering\n   */\n  private createNonStreamingEndpoint(\n    partitionedQueryExecutionInfo: PartitionedQueryExecutionInfo,\n    sortOrders: any[],\n    correlatedActivityId: string,\n    options: FeedOptions,\n  ): ExecutionContext {\n    const queryInfo = partitionedQueryExecutionInfo.queryInfo!; // Safe to use ! after validation in constructor\n\n    if (!options.allowUnboundedNonStreamingQueries) {\n      this.checkQueryConstraints(queryInfo);\n    }\n\n    const vectorSearchBufferSize = this.calculateVectorSearchBufferSize(queryInfo, options);\n\n    this.validateVectorSearchBufferSize(vectorSearchBufferSize, options);\n\n    const baseContext = new ParallelQueryExecutionContext(\n      this.clientContext,\n      this.collectionLink,\n      this.query,\n      this.options,\n      this.partitionedQueryExecutionInfo,\n      correlatedActivityId,\n    );\n\n    return this.wrapWithNonStreamingComponent(\n      baseContext,\n      queryInfo,\n      sortOrders,\n      vectorSearchBufferSize,\n    );\n  }\n\n  /**\n   * Creates a streaming endpoint with proper pipeline components\n   */\n  private createStreamingEndpoint(\n    partitionedQueryExecutionInfo: PartitionedQueryExecutionInfo,\n    sortOrders: any[],\n    correlatedActivityId: string,\n    isGroupByQuery: boolean,\n    queryContinuationFields: any,\n  ): ExecutionContext {\n    const queryInfo = partitionedQueryExecutionInfo.queryInfo!; // Safe to use ! after validation in constructor\n\n    // Create base execution context\n    let endpoint = this.createBaseExecutionContext(\n      partitionedQueryExecutionInfo,\n      sortOrders,\n      correlatedActivityId,\n    );\n\n    // Apply pipeline transformations\n    endpoint = this.applyGroupByComponents(endpoint, queryInfo, isGroupByQuery);\n    endpoint = this.applyDistinctComponents(endpoint, queryInfo, queryContinuationFields);\n    endpoint = this.applyLimitComponents(endpoint, queryInfo, queryContinuationFields);\n\n    return endpoint;\n  }\n\n  /**\n   * Creates the base execution context (OrderBy or Parallel)\n   */\n  private createBaseExecutionContext(\n    partitionedQueryExecutionInfo: PartitionedQueryExecutionInfo,\n    sortOrders: any[],\n    correlatedActivityId: string,\n  ): ExecutionContext {\n    if (Array.isArray(sortOrders) && sortOrders.length > 0) {\n      // OrderBy queries need special wrapping for payload structure\n      return new OrderByEndpointComponent(\n        new OrderByQueryExecutionContext(\n          this.clientContext,\n          this.collectionLink,\n          this.query,\n          this.options,\n          partitionedQueryExecutionInfo,\n          correlatedActivityId,\n        ),\n        this.emitRawOrderByPayload,\n      );\n    }\n\n    // Parallel queries\n    return new ParallelQueryExecutionContext(\n      this.clientContext,\n      this.collectionLink,\n      this.query,\n      this.options,\n      partitionedQueryExecutionInfo,\n      correlatedActivityId,\n    );\n  }\n\n  /**\n   * Wraps base context with appropriate non-streaming component\n   */\n  private wrapWithNonStreamingComponent(\n    baseContext: ExecutionContext,\n    queryInfo: QueryInfo,\n    sortOrders: any[],\n    vectorSearchBufferSize: number,\n  ): ExecutionContext {\n    const distinctType = queryInfo.distinctType;\n\n    if (distinctType === \"None\") {\n      return new NonStreamingOrderByEndpointComponent(\n        baseContext,\n        sortOrders,\n        vectorSearchBufferSize,\n        queryInfo.offset,\n        this.emitRawOrderByPayload,\n      );\n    }\n\n    return new NonStreamingOrderByDistinctEndpointComponent(\n      baseContext,\n      queryInfo,\n      vectorSearchBufferSize,\n      this.emitRawOrderByPayload,\n    );\n  }\n\n  /**\n   * Applies GROUP BY components to the pipeline if needed\n   */\n  private applyGroupByComponents(\n    endpoint: ExecutionContext,\n    queryInfo: QueryInfo,\n    isGroupByQuery: boolean,\n  ): ExecutionContext {\n    if (!isGroupByQuery) {\n      return endpoint;\n    }\n\n    return queryInfo.hasSelectValue\n      ? new GroupByValueEndpointComponent(endpoint, queryInfo)\n      : new GroupByEndpointComponent(endpoint, queryInfo);\n  }\n\n  /**\n   * Applies DISTINCT components to the pipeline if needed\n   */\n  private applyDistinctComponents(\n    endpoint: ExecutionContext,\n    queryInfo: QueryInfo,\n    queryContinuationFields: any,\n  ): ExecutionContext {\n    const distinctType = queryInfo.distinctType;\n\n    if (distinctType === \"Ordered\") {\n      const lastHash = queryContinuationFields?.hashedLastResult;\n      return new OrderedDistinctEndpointComponent(endpoint, lastHash);\n    }\n\n    if (distinctType === \"Unordered\") {\n      return new UnorderedDistinctEndpointComponent(endpoint);\n    }\n\n    return endpoint;\n  }\n\n  /**\n   * Applies TOP and OFFSET+LIMIT components to the pipeline if needed\n   */\n  private applyLimitComponents(\n    endpoint: ExecutionContext,\n    queryInfo: QueryInfo,\n    queryContinuationFields: any,\n  ): ExecutionContext {\n    // Apply TOP component (TOP N is effectively OFFSET 0 LIMIT N)\n    let top = queryInfo.top;\n    if (typeof top === \"number\") {\n      if (queryContinuationFields?.limit !== undefined) {\n        top = queryContinuationFields.limit;\n      }\n      endpoint = new OffsetLimitEndpointComponent(endpoint, 0, top);\n    }\n\n    // Apply OFFSET+LIMIT component\n    let limit = queryInfo.limit;\n    let offset = queryInfo.offset;\n\n    if (queryContinuationFields) {\n      if (queryContinuationFields.limit !== undefined) {\n        limit = queryContinuationFields.limit;\n      }\n      if (queryContinuationFields.offset !== undefined) {\n        offset = queryContinuationFields.offset;\n      }\n    }\n\n    if (typeof limit === \"number\" && typeof offset === \"number\") {\n      endpoint = new OffsetLimitEndpointComponent(endpoint, offset, limit);\n    }\n\n    return endpoint;\n  }\n\n  /**\n   * Validates vector search buffer size constraints\n   */\n  private validateVectorSearchBufferSize(\n    vectorSearchBufferSize: number,\n    options: FeedOptions,\n  ): void {\n    const maxBufferSize = options[\"vectorSearchBufferSize\"]\n      ? options[\"vectorSearchBufferSize\"]\n      : QueryExecution.DEFAULT_MAX_VECTOR_SEARCH_BUFFER_SIZE;\n\n    if (vectorSearchBufferSize > maxBufferSize) {\n      throw new ErrorResponse(\n        `Executing a vector search query with TOP or OFFSET + LIMIT value ${vectorSearchBufferSize} larger than the vectorSearchBufferSize ${maxBufferSize} ` +\n          `is not allowed`,\n      );\n    }\n  }\n\n  private calculateVectorSearchBufferSize(queryInfo: QueryInfo, options: FeedOptions): number {\n    if (queryInfo.top === 0 || queryInfo.limit === 0) return 0;\n    return queryInfo.top\n      ? queryInfo.top\n      : queryInfo.limit\n        ? queryInfo.offset + queryInfo.limit\n        : options[\"vectorSearchBufferSize\"] && options[\"vectorSearchBufferSize\"] > 0\n          ? options[\"vectorSearchBufferSize\"]\n          : QueryExecution.DEFAULT_MAX_VECTOR_SEARCH_BUFFER_SIZE;\n  }\n\n  private checkQueryConstraints(queryInfo: QueryInfo): void {\n    const hasTop = queryInfo.top || queryInfo.top === 0;\n    const hasLimit = queryInfo.limit || queryInfo.limit === 0;\n    if (!hasTop && !hasLimit) {\n      throw new ErrorResponse(\n        \"Executing a non-streaming search query without TOP or LIMIT can consume a large number of RUs \" +\n          \"very fast and have long runtimes. Please ensure you are using one of the above two filters \" +\n          \"with your vector search query.\",\n      );\n    }\n    return;\n  }\n\n  /**\n   * Analyzes query information and extracts key characteristics for query execution planning\n   */\n  private analyzeQueryInfo(queryInfo: QueryInfo) {\n    const sortOrders = queryInfo.orderBy;\n    const nonStreamingOrderBy = queryInfo.hasNonStreamingOrderBy;\n    const isOrderByQuery = Array.isArray(sortOrders) && sortOrders.length > 0;\n\n    // Check if this is a GROUP BY query\n    const isGroupByQuery =\n      Object.keys(queryInfo.groupByAliasToAggregateType || {}).length > 0 ||\n      (queryInfo.aggregates?.length || 0) > 0 ||\n      (queryInfo.groupByExpressions?.length || 0) > 0;\n\n    // Check if this is an unordered DISTINCT query\n    const isUnorderedDistinctQuery = queryInfo.distinctType === \"Unordered\";\n\n    // Determine if this query type supports continuation tokens\n    const querySupportsTokens =\n      !isUnorderedDistinctQuery && !isGroupByQuery && !nonStreamingOrderBy;\n\n    return {\n      sortOrders,\n      nonStreamingOrderBy,\n      isOrderByQuery,\n      isGroupByQuery,\n      isUnorderedDistinctQuery,\n      querySupportsTokens,\n    };\n  }\n}\n"],
+  "mappings": ";;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAKA,2BAA8B;AAC9B,0CAA6C;AAC7C,sCAAyC;AACzC,8CAAiD;AACjD,gDAAmD;AACnD,sCAAyC;AAEzC,0CAA6C;AAC7C,2CAA8C;AAC9C,2CAA8C;AAG9C,0DAA6D;AAC7D,kDAAqD;AACrD,mCAGO;AACP,qCAA6C;AAC7C,uCAA0C;AAC1C,6CAAgD;AAChD,uBAA+B;AAGxB,MAAM,+BAA2D;AAAA,EAKtE,YACU,eACA,gBACA,OACA,SACA,+BACR,sBACQ,wBAAiC,OACjC,6BAAsC,MAC9C;AARQ;AACA;AACA;AACA;AACA;AAEA;AACA;AAGR,QAAI,CAAC,8BAA8B,WAAW;AAC5C,YAAM,IAAI;AAAA,QACR;AAAA,MAGF;AAAA,IACF;AAEA,QAAI,CAAC,KAAK,QAAQ,cAAc;AAC9B,WAAK,QAAQ,eAAe,gCAAe;AAAA,IAC7C;AACA,UAAM,WAAW,KAAK,QAAQ;AAG9B,UAAM,oBAAoB,KAAK,iBAAiB,8BAA8B,SAAS;AACvF,UAAM;AAAA,MACJ;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF,IAAI;AAGJ,QAAI,CAAC,qBAAqB;AACxB,qFAA6C,KAAK,QAAQ,mBAAmB;AAAA,QAC3E,wCAAW,oBAAoB,mBAAmB;AAAA,QAClD,wCAAW,QAAQ,cAAc;AAAA,QACjC,wCAAW,kBAAkB,wBAAwB;AAAA,MACvD,CAAC;AAAA,IACH;AAGA,UAAM,0BAA0B,KAAK,QAAQ,wBACzC,6DAA6B,KAAK,QAAQ,iBAAiB,IAC3D;AAGJ,SAAK,WAAW,sBACZ,KAAK;AAAA,MACH;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF,IACA,KAAK;AAAA,MACH;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF;AACJ,SAAK,cAAc,CAAC;AAEpB,QAAI,KAAK,QAAQ,oBAAoB;AACnC,YAAM,kCACJ,KAAK,8BAA8B;AACrC,WAAK,sBAAsB,IAAI;AAAA,QAC7B,KAAK;AAAA,QACL;AAAA,QACA,KAAK;AAAA,QACL,KAAK,QAAQ;AAAA,QACb;AAAA,QACA;AAAA,MACF;AAAA,IACF,OAAO;AACL,WAAK,sBAAsB,IAAI,2DAA0B,KAAK,UAAU,QAAQ;AAAA,IAClF;AAAA,EACF;AAAA,EApFQ;AAAA,EACA;AAAA,EACS;AAAA,EAoFV,iBAA0B;AAC/B,WAAO,KAAK,YAAY,WAAW,KAAK,KAAK,SAAS,eAAe;AAAA,EACvE;AAAA,EAEA,MAAa,UAAU,gBAAgE;AACrF,WAAO,KAAK,oBAAoB,UAAU,gBAAgB,KAAK,WAAW;AAAA,EAC5E;AAAA;AAAA;AAAA;AAAA,EAKQ,2BACN,+BACA,YACA,sBACA,SACkB;AAClB,UAAM,YAAY,8BAA8B;AAEhD,QAAI,CAAC,QAAQ,mCAAmC;AAC9C,WAAK,sBAAsB,SAAS;AAAA,IACtC;AAEA,UAAM,yBAAyB,KAAK,gCAAgC,WAAW,OAAO;AAEtF,SAAK,+BAA+B,wBAAwB,OAAO;AAEnE,UAAM,cAAc,IAAI;AAAA,MACtB,KAAK;AAAA,MACL,KAAK;AAAA,MACL,KAAK;AAAA,MACL,KAAK;AAAA,MACL,KAAK;AAAA,MACL;AAAA,IACF;AAEA,WAAO,KAAK;AAAA,MACV;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKQ,wBACN,+BACA,YACA,sBACA,gBACA,yBACkB;AAClB,UAAM,YAAY,8BAA8B;AAGhD,QAAI,WAAW,KAAK;AAAA,MAClB;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAGA,eAAW,KAAK,uBAAuB,UAAU,WAAW,cAAc;AAC1E,eAAW,KAAK,wBAAwB,UAAU,WAAW,uBAAuB;AACpF,eAAW,KAAK,qBAAqB,UAAU,WAAW,uBAAuB;AAEjF,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKQ,2BACN,+BACA,YACA,sBACkB;AAClB,QAAI,MAAM,QAAQ,UAAU,KAAK,WAAW,SAAS,GAAG;AAEtD,aAAO,IAAI;AAAA,QACT,IAAI;AAAA,UACF,KAAK;AAAA,UACL,KAAK;AAAA,UACL,KAAK;AAAA,UACL,KAAK;AAAA,UACL;AAAA,UACA;AAAA,QACF;AAAA,QACA,KAAK;AAAA,MACP;AAAA,IACF;AAGA,WAAO,IAAI;AAAA,MACT,KAAK;AAAA,MACL,KAAK;AAAA,MACL,KAAK;AAAA,MACL,KAAK;AAAA,MACL;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKQ,8BACN,aACA,WACA,YACA,wBACkB;AAClB,UAAM,eAAe,UAAU;AAE/B,QAAI,iBAAiB,QAAQ;AAC3B,aAAO,IAAI;AAAA,QACT;AAAA,QACA;AAAA,QACA;AAAA,QACA,UAAU;AAAA,QACV,KAAK;AAAA,MACP;AAAA,IACF;AAEA,WAAO,IAAI;AAAA,MACT;AAAA,MACA;AAAA,MACA;AAAA,MACA,KAAK;AAAA,IACP;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKQ,uBACN,UACA,WACA,gBACkB;AAClB,QAAI,CAAC,gBAAgB;AACnB,aAAO;AAAA,IACT;AAEA,WAAO,UAAU,iBACb,IAAI,mEAA8B,UAAU,SAAS,IACrD,IAAI,yDAAyB,UAAU,SAAS;AAAA,EACtD;AAAA;AAAA;AAAA;AAAA,EAKQ,wBACN,UACA,WACA,yBACkB;AAClB,UAAM,eAAe,UAAU;AAE/B,QAAI,iBAAiB,WAAW;AAC9B,YAAM,WAAW,yBAAyB;AAC1C,aAAO,IAAI,yEAAiC,UAAU,QAAQ;AAAA,IAChE;AAEA,QAAI,iBAAiB,aAAa;AAChC,aAAO,IAAI,6EAAmC,QAAQ;AAAA,IACxD;AAEA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKQ,qBACN,UACA,WACA,yBACkB;AAElB,QAAI,MAAM,UAAU;AACpB,QAAI,OAAO,QAAQ,UAAU;AAC3B,UAAI,yBAAyB,UAAU,QAAW;AAChD,cAAM,wBAAwB;AAAA,MAChC;AACA,iBAAW,IAAI,iEAA6B,UAAU,GAAG,GAAG;AAAA,IAC9D;AAGA,QAAI,QAAQ,UAAU;AACtB,QAAI,SAAS,UAAU;AAEvB,QAAI,yBAAyB;AAC3B,UAAI,wBAAwB,UAAU,QAAW;AAC/C,gBAAQ,wBAAwB;AAAA,MAClC;AACA,UAAI,wBAAwB,WAAW,QAAW;AAChD,iBAAS,wBAAwB;AAAA,MACnC;AAAA,IACF;AAEA,QAAI,OAAO,UAAU,YAAY,OAAO,WAAW,UAAU;AAC3D,iBAAW,IAAI,iEAA6B,UAAU,QAAQ,KAAK;AAAA,IACrE;AAEA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKQ,+BACN,wBACA,SACM;AACN,UAAM,gBAAgB,QAAQ,wBAAwB,IAClD,QAAQ,wBAAwB,IAChC,gCAAe;AAEnB,QAAI,yBAAyB,eAAe;AAC1C,YAAM,IAAI;AAAA,QACR,oEAAoE,sBAAsB,2CAA2C,aAAa;AAAA,MAEpJ;AAAA,IACF;AAAA,EACF;AAAA,EAEQ,gCAAgC,WAAsB,SAA8B;AAC1F,QAAI,UAAU,QAAQ,KAAK,UAAU,UAAU,EAAG,QAAO;AACzD,WAAO,UAAU,MACb,UAAU,MACV,UAAU,QACR,UAAU,SAAS,UAAU,QAC7B,QAAQ,wBAAwB,KAAK,QAAQ,wBAAwB,IAAI,IACvE,QAAQ,wBAAwB,IAChC,gCAAe;AAAA,EACzB;AAAA,EAEQ,sBAAsB,WAA4B;AACxD,UAAM,SAAS,UAAU,OAAO,UAAU,QAAQ;AAClD,UAAM,WAAW,UAAU,SAAS,UAAU,UAAU;AACxD,QAAI,CAAC,UAAU,CAAC,UAAU;AACxB,YAAM,IAAI;AAAA,QACR;AAAA,MAGF;AAAA,IACF;AACA;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKQ,iBAAiB,WAAsB;AAC7C,UAAM,aAAa,UAAU;AAC7B,UAAM,sBAAsB,UAAU;AACtC,UAAM,iBAAiB,MAAM,QAAQ,UAAU,KAAK,WAAW,SAAS;AAGxE,UAAM,iBACJ,OAAO,KAAK,UAAU,+BAA+B,CAAC,CAAC,EAAE,SAAS,MACjE,UAAU,YAAY,UAAU,KAAK,MACrC,UAAU,oBAAoB,UAAU,KAAK;AAGhD,UAAM,2BAA2B,UAAU,iBAAiB;AAG5D,UAAM,sBACJ,CAAC,4BAA4B,CAAC,kBAAkB,CAAC;AAEnD,WAAO;AAAA,MACL;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AACF;",
+  "names": []
+}