mongodb-mcp-server 2.1.0 → 3.0.0-prerelease.2

package/dist/cjs/tools/mongodb/read/aggregate.js

@@ -1,371 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.AggregateTool = exports.AggregateArgs = exports.pipelineDescriptionWithVectorSearch = void 0;
-const zod_1 = require("zod");
-const mongodbTool_js_1 = require("../mongodbTool.js");
-const tool_js_1 = require("../../tool.js");
-const indexCheck_js_1 = require("../../../helpers/indexCheck.js");
-const errors_js_1 = require("../../../common/errors.js");
-const collectCursorUntilMaxBytes_js_1 = require("../../../helpers/collectCursorUntilMaxBytes.js");
-const operationWithFallback_js_1 = require("../../../helpers/operationWithFallback.js");
-const constants_js_1 = require("../../../helpers/constants.js");
-const index_js_1 = require("../../../common/logging/index.js");
-const mongodbSchemas_js_1 = require("../mongodbSchemas.js");
-const assertVectorSearchFilterFieldsAreIndexed_js_1 = require("../../../helpers/assertVectorSearchFilterFieldsAreIndexed.js");
-const mqlGuards_js_1 = require("../../../helpers/mqlGuards.js");
-const bsonToJson_js_1 = require("../../../helpers/bsonToJson.js");
-exports.pipelineDescriptionWithVectorSearch = `\
-An array of aggregation stages to execute.
-
-If the user has asked for a vector search, \`$vectorSearch\` **MUST** be the first stage
-of the pipeline (or the first stage of a \`$unionWith\` sub-pipeline only when explicitly
-combining unrelated result sets — for hybrid full-text + vector search, use \`$rankFusion\`
-or \`$scoreFusion\` instead, see below).
-
-If the user has asked for lexical/Atlas search, use \`$search\` instead of \`$text\`.
-### Usage Rules for \`$vectorSearch\`
-- **Index Type Detection:**
-  Use the collection-indexes tool to determine if the target field has a classic vector index (type: 'vector') or an auto-embed index (type: 'autoEmbed').
-- **Classic Vector Search (type: 'vector'):**
-  Use 'queryVector' with embeddings as an array of numbers.
-- **Auto-Embed Vector Search (type: 'autoEmbed'):**
-  Use 'query' - MongoDB automatically generates embeddings at query time. Do NOT use 'queryVector' or 'embeddingParameters' for auto-embed indexes.
-- **Unset embeddings:**
-  Unless the user explicitly requests the embeddings, add an \`$unset\` stage **at the end of the pipeline** to remove the embedding field and avoid context limits. **The $unset stage in this situation is mandatory**.
-- **Pre-filtering:**
-  If the user requests additional filtering, include filters in \`$vectorSearch.filter\` only for pre-filter fields in the vector index.
-  NEVER include fields in $vectorSearch.filter that are not part of the vector index.
-- **Post-filtering:**
-  For all remaining filters, add a $match stage after $vectorSearch.
-- If unsure which fields are filterable, use the collection-indexes tool to determine valid prefilter fields.
-- If no requested filters are valid prefilters, omit the filter key from $vectorSearch.
-
-### Usage Rules for \`$search\`
-- Include the index name, unless you know for a fact there's a default index. If unsure, use the collection-indexes tool to determine the index name.
-- The \`$search\` stage supports multiple operators, such as 'autocomplete', 'text', 'geoWithin', and others. Choose the appropriate operator based on the user's query. If unsure of the exact syntax, consult the MongoDB Atlas Search documentation, which can be found here: https://www.mongodb.com/docs/atlas/atlas-search/operators-and-collectors/
-
-### Usage Rules for \`$rankFusion\` and \`$scoreFusion\` (Hybrid Search)
-Use these stages when the user wants to combine full-text (\`$search\`) and vector
-(\`$vectorSearch\`) retrieval into a single fused result set. **Prefer native
-fusion over a \`$unionWith\` + \`$group\` workaround** — the workaround averages
-incompatible score scales and produces wrong rankings.
-
-**Which stage to use:**
-- \`$rankFusion\` (MongoDB 8.0+) — Reciprocal Rank Fusion. The recommended default.
-  Normalizes scores across incompatible scales automatically. No score tuning needed.
-- \`$scoreFusion\` (MongoDB 8.2+) — Score-based fusion. Use when the user needs explicit
-  per-pipeline weights, score normalisation (sigmoid / minMaxScaler), or a custom
-  combination expression.
-
-**Construction rules:**
-- \`$rankFusion\` / \`$scoreFusion\` MUST be the first stage of the top-level pipeline.
-- Sub-pipelines go inside \`input.pipelines\` as a named map (not an array). Each name
-  must be non-empty, must not start with \`$\`, and must not contain \`.\` or null bytes.
-- Allowed stages inside sub-pipelines: \`$search\`, \`$vectorSearch\`, \`$match\`, \`$sort\`,
-  \`$geoNear\`, \`$skip\`, \`$limit\`. \`$project\` and \`$unset\` are NOT allowed inside sub-pipelines.
-- Do field shaping (\`$project\` / \`$unset\`) only AFTER the fusion stage, at the root.
-- Both a vectorSearch (or autoEmbed) index AND a search (lexical) index must exist on
-  the collection. Use the collection-indexes tool to confirm both before running a hybrid query.
-- Add a \`$limit\` stage after the fusion stage to cap the final result set.
-- Add \`$unset\` at the end to remove embedding fields and avoid context bloat.
-
-### Usage Rules for \`$rerank\` (Native Reranking)
-Use this stage when the user wants to reorder a set of candidate documents using a cross-encoder reranker model.
-
-**Construction rules:**
-- \`$rerank\` can be any stage in the pipeline on an Atlas cluster running MongoDB 8.3 or higher.
-- It is recommended to use \`$rerank\` after a sorted pipeline, e.g. \`$search\`, \`$vectorSearch\`, \`$rankFusion\`, \`$scoreFusion\`, or [\`$match\`, \`$sort\`].
-- $rerank must be enabled via the Native Reranking Project Setting
-- Set \`numDocsToRerank\` as the number of documents passed into \`$rerank\`. This will also limit the number of documents returned by \`$rerank\`
-- Set \`path\` as a field name or an array of field names that exist in all documents. Use \`$match\` or \`$set\` before \`$rerank\` to validate no fields are missing.
-- Add \`$addFields\` after \`$rerank\` to retrieve the reranker score.
-
-**\`$rerank\` example (recommended default):**
-\`\`\`javascript
-[
-  {
-    $match: {
-      description: { $exists: true },
-      name: { $exists: true }
-    }
-  },
-  {
-    $sort: {
-      lastUpdated: -1
-    }
-  },
-  {
-    $rerank: {
-      query: {
-        text: "query text including instructions"
-      },
-      model: "rerank-2.5",
-      numDocsToRerank: 100,
-      path: ["description", "name"]
-    }
-  },
-  {
-    $addFields: {
-      rerankScore: { $meta: "score" }
-    }
-  }
-]
-\`\`\`
-`;
-const AggregateOutputSchema = {
-    documents: zod_1.z.array(zod_1.z.unknown()).describe("The documents returned by the aggregation pipeline"),
-    count: zod_1.z
-        .number()
-        .or(zod_1.z.literal("indeterminate"))
-        .describe("The total number of documents returned by the aggregation pipeline"),
-    appliedLimits: zod_1.z.array(constants_js_1.CURSOR_LIMIT_KEYS).describe("The limits applied to the aggregation pipeline"),
-};
-exports.AggregateArgs = {
-    pipeline: zod_1.z.array(zod_1.z.union([mongodbSchemas_js_1.VectorSearchStage, mongodbSchemas_js_1.AnyAggregateStage])).describe(exports.pipelineDescriptionWithVectorSearch),
-};
-class AggregateTool extends mongodbTool_js_1.MongoDBToolBase {
-    constructor() {
-        super(...arguments);
-        this.description = "Run an aggregation against a MongoDB collection";
-        this.argsShape = {
-            ...mongodbTool_js_1.ConnectionIdArgs,
-            ...mongodbTool_js_1.CollOperationArgs,
-            ...exports.AggregateArgs,
-            responseBytesLimit: zod_1.z
-                .number()
-                .optional()
-                .default(constants_js_1.ONE_MB)
-                .describe("The maximum number of bytes to return in the response. This value is capped by the server's configured maximum and cannot be exceeded."),
-        };
-        this.outputSchema = AggregateOutputSchema;
-    }
-    async execute({ connectionId, database, collection, pipeline, responseBytesLimit }, context) {
-        const { signal } = context;
-        let aggregationCursor = undefined;
-        try {
-            const provider = await this.resolveConnection(connectionId);
-            const isSearchSupported = await this.isSearchSupported(connectionId);
-            this.assertOnlyUsesPermittedStages({ isSearchSupported }, pipeline);
-            if (isSearchSupported) {
-                let searchIndexes;
-                try {
-                    searchIndexes = (await provider.getSearchIndexes(database, collection));
-                }
-                catch (error) {
-                    this.session.logger.debug({
-                        id: index_js_1.LogId.mongodbGetSearchIndexesFailure,
-                        context: "aggregate tool",
-                        message: `Failed to fetch search indexes for pre-filter validation, skipping check: ${error instanceof Error ? error.message : String(error)}`,
-                    });
-                }
-                if (searchIndexes !== undefined) {
-                    (0, assertVectorSearchFilterFieldsAreIndexed_js_1.assertVectorSearchFilterFieldsAreIndexed)({
-                        searchIndexes,
-                        pipeline,
-                        logger: this.session.logger,
-                    });
-                }
-            }
-            // Check if aggregate operation uses an index if enabled
-            if (this.config.indexCheck) {
-                const [usesVectorSearchIndex, indexName] = await this.isVectorSearchIndexUsed({ isSearchSupported, provider }, {
-                    database,
-                    collection,
-                    pipeline,
-                });
-                switch (usesVectorSearchIndex) {
-                    case "not-vector-search-query":
-                        await (0, indexCheck_js_1.checkIndexUsage)({
-                            database,
-                            collection,
-                            operation: "aggregate",
-                            explainCallback: async () => {
-                                return provider
-                                    .aggregate(database, collection, pipeline, {
-                                    ...this.getOperationOptions(signal),
-                                }, { writeConcern: undefined })
-                                    .explain("queryPlanner");
-                            },
-                            logger: this.session.logger,
-                        });
-                        break;
-                    case "non-existent-index":
-                        throw new errors_js_1.MongoDBError(errors_js_1.ErrorCodes.AtlasVectorSearchIndexNotFound, `Could not find an index with name "${indexName}" in namespace "${database}.${collection}".`);
-                    case "valid-index":
-                        // nothing to do, everything is correct so ready to run the query
-                        break;
-                }
-            }
-            let successMessage;
-            let documents;
-            let count;
-            let appliedLimits = [];
-            const writeStageTargets = (0, mqlGuards_js_1.getWriteStageTargets)(pipeline, database);
-            if (writeStageTargets.length > 0) {
-                await this.confirmWriteStages(writeStageTargets, context);
-                // This is a write pipeline, so special-case it and don't attempt to apply limits or caps
-                aggregationCursor = provider.aggregate(database, collection, pipeline, {
-                    signal,
-                });
-                documents = await aggregationCursor.toArray();
-                successMessage = "The aggregation pipeline executed successfully.";
-            }
-            else {
-                const cappedResultsPipeline = [...pipeline];
-                if (this.config.maxDocumentsPerQuery > 0) {
-                    cappedResultsPipeline.push({ $limit: this.config.maxDocumentsPerQuery });
-                }
-                aggregationCursor = provider.aggregate(database, collection, cappedResultsPipeline, {
-                    ...this.getOperationOptions(signal),
-                });
-                const [totalDocuments, cursorResults] = await Promise.all([
-                    this.countAggregationResultDocuments({
-                        provider,
-                        database,
-                        collection,
-                        pipeline,
-                        abortSignal: signal,
-                    }),
-                    (0, collectCursorUntilMaxBytes_js_1.collectCursorUntilMaxBytesLimit)({
-                        cursor: aggregationCursor,
-                        configuredMaxBytesPerQuery: this.config.maxBytesPerQuery,
-                        toolResponseBytesLimit: responseBytesLimit,
-                        abortSignal: signal,
-                    }),
-                ]);
-                // If the total number of documents that the aggregation would've
-                // resulted in would be greater than the configured
-                // maxDocumentsPerQuery then we know for sure that the results were
-                // capped.
-                const aggregationResultsCappedByMaxDocumentsLimit = this.config.maxDocumentsPerQuery > 0 &&
-                    !!totalDocuments &&
-                    totalDocuments > this.config.maxDocumentsPerQuery;
-                documents = cursorResults.documents;
-                count = totalDocuments;
-                appliedLimits = [
-                    aggregationResultsCappedByMaxDocumentsLimit ? "config.maxDocumentsPerQuery" : undefined,
-                    cursorResults.cappedBy,
-                ].filter((limit) => !!limit);
-                successMessage = this.generateMessage({
-                    count,
-                    documents,
-                    appliedLimits,
-                });
-            }
-            documents = (0, bsonToJson_js_1.bsonToJson)(documents);
-            return {
-                content: (0, tool_js_1.formatUntrustedData)(successMessage, ...(documents.length > 0 ? [JSON.stringify(documents)] : [])),
-                structuredContent: {
-                    documents,
-                    count: count ?? "indeterminate",
-                    appliedLimits,
-                },
-            };
-        }
-        finally {
-            if (aggregationCursor) {
-                void this.safeCloseCursor(aggregationCursor);
-            }
-        }
-    }
-    async safeCloseCursor(cursor) {
-        try {
-            await cursor.close();
-        }
-        catch (error) {
-            this.session.logger.warning({
-                id: index_js_1.LogId.mongodbCursorCloseError,
-                context: "aggregate tool",
-                message: `Error when closing the cursor - ${error instanceof Error ? error.message : String(error)}`,
-            });
-        }
-    }
-    assertOnlyUsesPermittedStages({ isSearchSupported }, pipeline) {
-        this.assertMqlIsAllowed(pipeline);
-        for (const stage of pipeline) {
-            // This ensure that you can't use $search if the cluster does not support MongoDB Search
-            // either in Atlas or in a local cluster.
-            if (this.isSearchStage(stage) && !isSearchSupported) {
-                throw new errors_js_1.MongoDBError(errors_js_1.ErrorCodes.AtlasSearchNotSupported, "Atlas Search is not supported in this cluster.");
-            }
-        }
-    }
-    async countAggregationResultDocuments({ provider, database, collection, pipeline, abortSignal, }) {
-        const resultsCountAggregation = [...pipeline, { $count: "totalDocuments" }];
-        return await (0, operationWithFallback_js_1.operationWithFallback)(async () => {
-            const aggregationResults = await provider
-                .aggregate(database, collection, resultsCountAggregation, {
-                signal: abortSignal,
-            })
-                .maxTimeMS(this.config.maxTimeMS !== undefined
-                ? Math.min(this.config.maxTimeMS, constants_js_1.AGG_COUNT_MAX_TIME_MS_CAP)
-                : constants_js_1.AGG_COUNT_MAX_TIME_MS_CAP)
-                .toArray();
-            const documentWithCount = aggregationResults.length === 1 ? aggregationResults[0] : undefined;
-            const totalDocuments = documentWithCount &&
-                typeof documentWithCount === "object" &&
-                "totalDocuments" in documentWithCount &&
-                typeof documentWithCount.totalDocuments === "number"
-                ? documentWithCount.totalDocuments
-                : 0;
-            return totalDocuments;
-        }, undefined);
-    }
-    async isVectorSearchIndexUsed({ isSearchSupported, provider }, { database, collection, pipeline, }) {
-        // check if the pipeline contains a $vectorSearch stage
-        let usesVectorSearch = false;
-        let indexName = "default";
-        for (const stage of pipeline) {
-            if ("$vectorSearch" in stage) {
-                const { $vectorSearch: vectorSearchStage } = stage;
-                usesVectorSearch = true;
-                indexName = vectorSearchStage.index;
-                break;
-            }
-        }
-        if (!usesVectorSearch) {
-            return ["not-vector-search-query"];
-        }
-        let indexExists = false;
-        if (isSearchSupported) {
-            try {
-                const indexes = await provider.getSearchIndexes(database, collection, indexName);
-                indexExists = indexes.length >= 1;
-            }
-            catch (error) {
-                this.session.logger.debug({
-                    id: index_js_1.LogId.mongodbGetSearchIndexesFailure,
-                    context: "aggregate tool",
-                    message: `Failed to fetch search indexes for vector search index check, skipping check: ${error instanceof Error ? error.message : String(error)}`,
-                });
-                return ["valid-index", indexName];
-            }
-        }
-        return [indexExists ? "valid-index" : "non-existent-index", indexName];
-    }
-    generateMessage({ count, documents, appliedLimits, }) {
-        let message = `The aggregation resulted in ${count === undefined ? "indeterminable number of" : count} documents.`;
-        // If we applied a limit or the count is different from the aggregation result count,
-        // communicate what is the actual number of returned documents
-        if (documents.length !== count || appliedLimits.length) {
-            message += ` Returning ${documents.length} documents`;
-            if (appliedLimits.length) {
-                message += ` while respecting the applied limits of ${appliedLimits
-                    .map((limit) => constants_js_1.CURSOR_LIMITS_TO_LLM_TEXT[limit])
-                    .join(", ")}`;
-                if (this.isExportToolAvailable) {
-                    message += `. If the entire aggregation result is required, use the "export" tool to retrieve the full result set`;
-                }
-            }
-            message += ".";
-        }
-        return message;
-    }
-    isSearchStage(stage) {
-        return "$vectorSearch" in stage || "$search" in stage || "$searchMeta" in stage;
-    }
-}
-exports.AggregateTool = AggregateTool;
-AggregateTool.toolName = "aggregate";
-AggregateTool.operationType = "read";
-//# sourceMappingURL=aggregate.js.map

package/dist/cjs/tools/mongodb/read/aggregate.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"aggregate.js","sourceRoot":"","sources":["../../../../../src/tools/mongodb/read/aggregate.ts"],"names":[],"mappings":";;;AAAA,6BAAwB;AAGxB,sDAAyF;AAEzF,2CAAoD;AACpD,kEAAiE;AAEjE,yDAAqE;AACrE,kGAAiG;AACjG,wFAAkF;AAClF,gEAMuC;AACvC,+DAAyD;AACzD,4DAA4E;AAC5E,8HAGsE;AACtE,gEAAqE;AACrE,kEAA4D;AAE/C,QAAA,mCAAmC,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAiGlD,CAAC;AAEF,MAAM,qBAAqB,GAAG;IAC1B,SAAS,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,OAAO,EAAE,CAAC,CAAC,QAAQ,CAAC,oDAAoD,CAAC;IAC9F,KAAK,EAAE,OAAC;SACH,MAAM,EAAE;SACR,EAAE,CAAC,OAAC,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;SAC9B,QAAQ,CAAC,oEAAoE,CAAC;IACnF,aAAa,EAAE,OAAC,CAAC,KAAK,CAAC,gCAAiB,CAAC,CAAC,QAAQ,CAAC,gDAAgD,CAAC;CACvG,CAAC;AAEW,QAAA,aAAa,GAAG;IACzB,QAAQ,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,KAAK,CAAC,CAAC,qCAAiB,EAAE,qCAAiB,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,2CAAmC,CAAC;CACnH,CAAC;AAEF,MAAa,aAAc,SAAQ,gCAAe;IAAlD;;QAEW,gBAAW,GAAG,iDAAiD,CAAC;QAChE,cAAS,GAAG;YACf,GAAG,iCAAgB;YACnB,GAAG,kCAAiB;YACpB,GAAG,qBAAa;YAChB,kBAAkB,EAAE,OAAC;iBAChB,MAAM,EAAE;iBACR,QAAQ,EAAE;iBACV,OAAO,CAAC,qBAAM,CAAC;iBACf,QAAQ,CACL,wIAAwI,CAC3I;SACR,CAAC;QAGc,iBAAY,GAAG,qBAAqB,CAAC;IAmTzD,CAAC;IAjTa,KAAK,CAAC,OAAO,CACnB,EAAE,YAAY,EAAE,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE,kBAAkB,EAAmC,EACrG,OAA6B;QAE7B,MAAM,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;QAC3B,IAAI,iBAAiB,GAAkC,SAAS,CAAC;QACjE,IAAI,CAAC;YACD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;YAC5D,MAAM,iBAAiB,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;YACrE,IAAI,CAAC,6BAA6B,CAAC,EAAE,iBAAiB,EAAE,EAAE,QAAQ,CAAC,CAAC;YACpE,IAAI,iBAAiB,EAAE,CAAC;gBACpB,IAAI,aAAwC,CAAC;gBAC7C,IAAI,CAAC;oBACD,aAAa,GAAG,CAAC,MAAM,QAAQ,CAAC,gBAAgB,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAkB,CAAC;gBAC7F,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACb,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC;wBACtB,EAAE,EAAE,gBAAK,CAAC,8BAA8B;wBACxC,OAAO,EAAE,gBAAgB;wBACzB,OAAO,EAAE,6EAA6E,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE;qBACjJ,CAAC,CAAC;gBACP,CAAC;gBACD,IAAI,aAAa,KAAK,SAAS,EAAE,CAAC;oBAC9B,IAAA,sFAAwC,EAAC;wBACrC,aAAa;wBACb,QAAQ;wBACR,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM;qBAC9B,CAAC,CAAC;gBACP,CAAC;YACL,CAAC;YAED,wDAAwD;YACxD,IAAI,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;gBACzB,MAAM,CAAC,qBAAqB,EAAE,SAAS,CAAC,GAAG,MAAM,IAAI,CAAC,uBAAuB,CACzE,EAAE,iBAAiB,EAAE,QAAQ,EAAE,EAC/B;oBACI,QAAQ;oBACR,UAAU;oBACV,QAAQ;iBACX,CACJ,CAAC;gBACF,QAAQ,qBAAqB,EAAE,CAAC;oBAC5B,KAAK,yBAAyB;wBAC1B,MAAM,IAAA,+BAAe,EAAC;4BAClB,QAAQ;4BACR,UAAU;4BACV,SAAS,EAAE,WAAW;4BACtB,eAAe,EAAE,KAAK,IAAI,EAAE;gCACxB,OAAO,QAAQ;qCACV,SAAS,CACN,QAAQ,EACR,UAAU,EACV,QAAQ,EACR;oCACI,GAAG,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC;iCACtC,EACD,EAAE,YAAY,EAAE,SAAS,EAAE,CAC9B;qCACA,OAAO,CAAC,cAAc,CAAC,CAAC;4BACjC,CAAC;4BACD,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM;yBAC9B,CAAC,CAAC;wBACH,MAAM;oBACV,KAAK,oBAAoB;wBACrB,MAAM,IAAI,wBAAY,CAClB,sBAAU,CAAC,8BAA8B,EACzC,sCAAsC,SAAS,mBAAmB,QAAQ,IAAI,UAAU,IAAI,CAC/F,CAAC;oBACN,KAAK,aAAa;wBACd,iEAAiE;wBACjE,MAAM;gBACd,CAAC;YACL,CAAC;YAED,IAAI,cAAsB,CAAC;YAC3B,IAAI,SAAoB,CAAC;YACzB,IAAI,KAAyB,CAAC;YAC9B,IAAI,aAAa,GAAqB,EAAE,CAAC;YAEzC,MAAM,iBAAiB,GAAG,IAAA,mCAAoB,EAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;YACnE,IAAI,iBAAiB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC/B,MAAM,IAAI,CAAC,kBAAkB,CAAC,iBAAiB,EAAE,OAAO,CAAC,CAAC;gBAE1D,yFAAyF;gBACzF,iBAAiB,GAAG,QAAQ,CAAC,SAAS,CAAC,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE;oBACnE,MAAM;iBACT,CAAC,CAAC;gBAEH,SAAS,GAAG,MAAM,iBAAiB,CAAC,OAAO,EAAE,CAAC;gBAC9C,cAAc,GAAG,iDAAiD,CAAC;YACvE,CAAC;iBAAM,CAAC;gBACJ,MAAM,qBAAqB,GAAe,CAAC,GAAG,QAAQ,CAAC,CAAC;gBACxD,IAAI,IAAI,CAAC,MAAM,CAAC,oBAAoB,GAAG,CAAC,EAAE,CAAC;oBACvC,qBAAqB,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC,CAAC;gBAC7E,CAAC;gBACD,iBAAiB,GAAG,QAAQ,CAAC,SAAS,CAAC,QAAQ,EAAE,UAAU,EAAE,qBAAqB,EAAE;oBAChF,GAAG,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC;iBACtC,CAAC,CAAC;gBAEH,MAAM,CAAC,cAAc,EAAE,aAAa,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;oBACtD,IAAI,CAAC,+BAA+B,CAAC;wBACjC,QAAQ;wBACR,QAAQ;wBACR,UAAU;wBACV,QAAQ;wBACR,WAAW,EAAE,MAAM;qBACtB,CAAC;oBACF,IAAA,+DAA+B,EAAC;wBAC5B,MAAM,EAAE,iBAAiB;wBACzB,0BAA0B,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB;wBACxD,sBAAsB,EAAE,kBAAkB;wBAC1C,WAAW,EAAE,MAAM;qBACtB,CAAC;iBACL,CAAC,CAAC;gBAEH,iEAAiE;gBACjE,mDAAmD;gBACnD,mEAAmE;gBACnE,UAAU;gBACV,MAAM,2CAA2C,GAC7C,IAAI,CAAC,MAAM,CAAC,oBAAoB,GAAG,CAAC;oBACpC,CAAC,CAAC,cAAc;oBAChB,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC,oBAAoB,CAAC;gBAEtD,SAAS,GAAG,aAAa,CAAC,SAAS,CAAC;gBACpC,KAAK,GAAG,cAAc,CAAC;gBACvB,aAAa,GAAG;oBACZ,2CAA2C,CAAC,CAAC,CAAC,6BAA6B,CAAC,CAAC,CAAC,SAAS;oBACvF,aAAa,CAAC,QAAQ;iBACzB,CAAC,MAAM,CAAC,CAAC,KAAK,EAA2B,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;gBACtD,cAAc,GAAG,IAAI,CAAC,eAAe,CAAC;oBAClC,KAAK;oBACL,SAAS;oBACT,aAAa;iBAChB,CAAC,CAAC;YACP,CAAC;YAED,SAAS,GAAG,IAAA,0BAAU,EAAC,SAAS,CAAC,CAAC;YAElC,OAAO;gBACH,OAAO,EAAE,IAAA,6BAAmB,EACxB,cAAc,EACd,GAAG,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAC/D;gBACD,iBAAiB,EAAE;oBACf,SAAS;oBACT,KAAK,EAAE,KAAK,IAAI,eAAe;oBAC/B,aAAa;iBAChB;aACJ,CAAC;QACN,CAAC;gBAAS,CAAC;YACP,IAAI,iBAAiB,EAAE,CAAC;gBACpB,KAAK,IAAI,CAAC,eAAe,CAAC,iBAAiB,CAAC,CAAC;YACjD,CAAC;QACL,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,eAAe,CAAC,MAAkC;QAC5D,IAAI,CAAC;YACD,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;QACzB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC;gBACxB,EAAE,EAAE,gBAAK,CAAC,uBAAuB;gBACjC,OAAO,EAAE,gBAAgB;gBACzB,OAAO,EAAE,mCAAmC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE;aACvG,CAAC,CAAC;QACP,CAAC;IACL,CAAC;IAEO,6BAA6B,CACjC,EAAE,iBAAiB,EAAkC,EACrD,QAAmC;QAEnC,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QAElC,KAAK,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;YAC3B,wFAAwF;YACxF,yCAAyC;YACzC,IAAI,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC;gBAClD,MAAM,IAAI,wBAAY,CAClB,sBAAU,CAAC,uBAAuB,EAClC,gDAAgD,CACnD,CAAC;YACN,CAAC;QACL,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,+BAA+B,CAAC,EAC1C,QAAQ,EACR,QAAQ,EACR,UAAU,EACV,QAAQ,EACR,WAAW,GAOd;QACG,MAAM,uBAAuB,GAAG,CAAC,GAAG,QAAQ,EAAE,EAAE,MAAM,EAAE,gBAAgB,EAAE,CAAC,CAAC;QAC5E,OAAO,MAAM,IAAA,gDAAqB,EAAC,KAAK,IAAiC,EAAE;YACvE,MAAM,kBAAkB,GAAG,MAAM,QAAQ;iBACpC,SAAS,CAAC,QAAQ,EAAE,UAAU,EAAE,uBAAuB,EAAE;gBACtD,MAAM,EAAE,WAAW;aACtB,CAAC;iBACD,SAAS,CACN,IAAI,CAAC,MAAM,CAAC,SAAS,KAAK,SAAS;gBAC/B,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,wCAAyB,CAAC;gBAC5D,CAAC,CAAC,wCAAyB,CAClC;iBACA,OAAO,EAAE,CAAC;YAEf,MAAM,iBAAiB,GAAY,kBAAkB,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YACvG,MAAM,cAAc,GAChB,iBAAiB;gBACjB,OAAO,iBAAiB,KAAK,QAAQ;gBACrC,gBAAgB,IAAI,iBAAiB;gBACrC,OAAO,iBAAiB,CAAC,cAAc,KAAK,QAAQ;gBAChD,CAAC,CAAC,iBAAiB,CAAC,cAAc;gBAClC,CAAC,CAAC,CAAC,CAAC;YAEZ,OAAO,cAAc,CAAC;QAC1B,CAAC,EAAE,SAAS,CAAC,CAAC;IAClB,CAAC;IAEO,KAAK,CAAC,uBAAuB,CACjC,EAAE,iBAAiB,EAAE,QAAQ,EAAuE,EACpG,EACI,QAAQ,EACR,UAAU,EACV,QAAQ,GAKX;QAED,uDAAuD;QACvD,IAAI,gBAAgB,GAAG,KAAK,CAAC;QAC7B,IAAI,SAAS,GAAW,SAAS,CAAC;QAElC,KAAK,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;YAC3B,IAAI,eAAe,IAAI,KAAK,EAAE,CAAC;gBAC3B,MAAM,EAAE,aAAa,EAAE,iBAAiB,EAAE,GAAG,KAA0C,CAAC;gBACxF,gBAAgB,GAAG,IAAI,CAAC;gBACxB,SAAS,GAAG,iBAAiB,CAAC,KAAK,CAAC;gBACpC,MAAM;YACV,CAAC;QACL,CAAC;QAED,IAAI,CAAC,gBAAgB,EAAE,CAAC;YACpB,OAAO,CAAC,yBAAyB,CAAC,CAAC;QACvC,CAAC;QAED,IAAI,WAAW,GAAG,KAAK,CAAC;QACxB,IAAI,iBAAiB,EAAE,CAAC;YACpB,IAAI,CAAC;gBACD,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,gBAAgB,CAAC,QAAQ,EAAE,UAAU,EAAE,SAAS,CAAC,CAAC;gBACjF,WAAW,GAAG,OAAO,CAAC,MAAM,IAAI,CAAC,CAAC;YACtC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACb,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC;oBACtB,EAAE,EAAE,gBAAK,CAAC,8BAA8B;oBACxC,OAAO,EAAE,gBAAgB;oBACzB,OAAO,EAAE,iFAAiF,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE;iBACrJ,CAAC,CAAC;gBACH,OAAO,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC;YACtC,CAAC;QACL,CAAC;QAED,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,oBAAoB,EAAE,SAAS,CAAC,CAAC;IAC3E,CAAC;IAEO,eAAe,CAAC,EACpB,KAAK,EACL,SAAS,EACT,aAAa,GAKhB;QACG,IAAI,OAAO,GAAG,+BAA+B,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,0BAA0B,CAAC,CAAC,CAAC,KAAK,aAAa,CAAC;QAEnH,qFAAqF;QACrF,8DAA8D;QAC9D,IAAI,SAAS,CAAC,MAAM,KAAK,KAAK,IAAI,aAAa,CAAC,MAAM,EAAE,CAAC;YACrD,OAAO,IAAI,cAAc,SAAS,CAAC,MAAM,YAAY,CAAC;YACtD,IAAI,aAAa,CAAC,MAAM,EAAE,CAAC;gBACvB,OAAO,IAAI,2CAA2C,aAAa;qBAC9D,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,wCAAyB,CAAC,KAAK,CAAC,CAAC;qBAChD,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;gBAClB,IAAI,IAAI,CAAC,qBAAqB,EAAE,CAAC;oBAC7B,OAAO,IAAI,uGAAuG,CAAC;gBACvH,CAAC;YACL,CAAC;YAED,OAAO,IAAI,GAAG,CAAC;QACnB,CAAC;QAED,OAAO,OAAO,CAAC;IACnB,CAAC;IAEO,aAAa,CAAC,KAA8B;QAChD,OAAO,eAAe,IAAI,KAAK,IAAI,SAAS,IAAI,KAAK,IAAI,aAAa,IAAI,KAAK,CAAC;IACpF,CAAC;;AAnUL,sCAoUC;AAnUU,sBAAQ,GAAG,WAAW,AAAd,CAAe;AAcvB,2BAAa,GAAkB,MAAM,AAAxB,CAAyB"}

package/dist/cjs/tools/mongodb/read/aggregateDB.d.ts

@@ -1,32 +0,0 @@
-import { z } from "zod";
-import { MongoDBToolBase } from "../mongodbTool.js";
-import type { ToolArgs, OperationType, ToolExecutionContext, ToolResult } from "../../tool.js";
-export declare const AggregateArgs: {
-    pipeline: z.ZodArray<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
-};
-export declare class AggregateDBTool extends MongoDBToolBase {
-    static toolName: string;
-    description: string;
-    argsShape: {
-        responseBytesLimit: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
-        pipeline: z.ZodArray<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
-        database: z.ZodString;
-        connectionId: z.ZodString;
-    };
-    static operationType: OperationType;
-    outputSchema: {
-        documents: z.ZodArray<z.ZodUnknown>;
-        aggResultsCount: z.ZodOptional<z.ZodNumber>;
-        appliedLimits: z.ZodArray<z.ZodEnum<{
-            "config.maxDocumentsPerQuery": "config.maxDocumentsPerQuery";
-            "config.maxBytesPerQuery": "config.maxBytesPerQuery";
-            "tool.responseBytesLimit": "tool.responseBytesLimit";
-        }>>;
-    };
-    protected execute({ connectionId, database, pipeline, responseBytesLimit }: ToolArgs<typeof this.argsShape>, context: ToolExecutionContext): Promise<ToolResult<typeof this.outputSchema>>;
-    private safeCloseCursor;
-    private assertOnlyUsesPermittedStages;
-    private countAggregationResultDocuments;
-    private generateMessage;
-}
-//# sourceMappingURL=aggregateDB.d.ts.map

package/dist/cjs/tools/mongodb/read/aggregateDB.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"aggregateDB.d.ts","sourceRoot":"","sources":["../../../../../src/tools/mongodb/read/aggregateDB.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAGxB,OAAO,EAAqC,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACvF,OAAO,KAAK,EAAE,QAAQ,EAAE,aAAa,EAAE,oBAAoB,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AA2B/F,eAAO,MAAM,aAAa;;CAOzB,CAAC;AAEF,qBAAa,eAAgB,SAAQ,eAAe;IAChD,MAAM,CAAC,QAAQ,SAAkB;IAC1B,WAAW,SAAmD;IAC9D,SAAS;;;;;MAWd;IACF,MAAM,CAAC,aAAa,EAAE,aAAa,CAAU;IAE7B,YAAY;;;;;;;;MAA2B;cAEvC,OAAO,CACnB,EAAE,YAAY,EAAE,QAAQ,EAAE,QAAQ,EAAE,kBAAkB,EAAE,EAAE,QAAQ,CAAC,OAAO,IAAI,CAAC,SAAS,CAAC,EACzF,OAAO,EAAE,oBAAoB,GAC9B,OAAO,CAAC,UAAU,CAAC,OAAO,IAAI,CAAC,YAAY,CAAC,CAAC;YAuFlC,eAAe;IAY7B,OAAO,CAAC,6BAA6B;YAYvB,+BAA+B;IAqC7C,OAAO,CAAC,eAAe;CA0B1B"}

package/dist/cjs/tools/mongodb/read/aggregateDB.js

@@ -1,180 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.AggregateDBTool = exports.AggregateArgs = void 0;
-const zod_1 = require("zod");
-const mongodbTool_js_1 = require("../mongodbTool.js");
-const tool_js_1 = require("../../tool.js");
-const errors_js_1 = require("../../../common/errors.js");
-const collectCursorUntilMaxBytes_js_1 = require("../../../helpers/collectCursorUntilMaxBytes.js");
-const operationWithFallback_js_1 = require("../../../helpers/operationWithFallback.js");
-const mqlGuards_js_1 = require("../../../helpers/mqlGuards.js");
-const constants_js_1 = require("../../../helpers/constants.js");
-const index_js_1 = require("../../../common/logging/index.js");
-const mongodbSchemas_js_1 = require("../mongodbSchemas.js");
-const bsonToJson_js_1 = require("../../../helpers/bsonToJson.js");
-const AggregateDBOutputSchema = {
-    documents: zod_1.z.array(zod_1.z.unknown()).describe("The documents returned by the aggregation pipeline"),
-    aggResultsCount: zod_1.z
-        .number()
-        .optional()
-        .describe("The total number of documents returned by the aggregation pipeline"),
-    appliedLimits: zod_1.z.array(constants_js_1.CURSOR_LIMIT_KEYS).describe("The limits applied to the aggregation pipeline"),
-};
-exports.AggregateArgs = {
-    pipeline: zod_1.z
-        .array(mongodbSchemas_js_1.AnyAggregateStage)
-        .min(1)
-        .describe(`An array of aggregation stages to execute. The first stage must be a database-level aggregation stage (one of ${mongodbSchemas_js_1.DB_AGGREGATE_STAGE_OPERATORS.map((op) => `\`${op}\``).join(", ")}). https://www.mongodb.com/docs/manual/reference/mql/aggregation-stages/#db.aggregate---stages`),
-};
-class AggregateDBTool extends mongodbTool_js_1.MongoDBToolBase {
-    constructor() {
-        super(...arguments);
-        this.description = "Run an aggregation against a MongoDB database";
-        this.argsShape = {
-            ...mongodbTool_js_1.ConnectionIdArgs,
-            ...mongodbTool_js_1.DBOperationArgs,
-            ...exports.AggregateArgs,
-            responseBytesLimit: zod_1.z
-                .number()
-                .optional()
-                .default(constants_js_1.ONE_MB)
-                .describe("The maximum number of bytes to return in the response. This value is capped by the server's configured maximum and cannot be exceeded."),
-        };
-        this.outputSchema = AggregateDBOutputSchema;
-    }
-    async execute({ connectionId, database, pipeline, responseBytesLimit }, context) {
-        const { signal } = context;
-        let aggregationCursor = undefined;
-        try {
-            const provider = await this.resolveConnection(connectionId);
-            this.assertOnlyUsesPermittedStages(pipeline);
-            let successMessage;
-            let documents;
-            let aggResultsCount;
-            let appliedLimits = [];
-            const writeStageTargets = (0, mqlGuards_js_1.getWriteStageTargets)(pipeline, database);
-            if (writeStageTargets.length > 0) {
-                await this.confirmWriteStages(writeStageTargets, context);
-                // This is a write pipeline, so special-case it and don't attempt to apply limits or caps
-                aggregationCursor = provider.aggregateDb(database, pipeline, {
-                    ...this.getOperationOptions(signal),
-                });
-                documents = await aggregationCursor.toArray();
-                successMessage = "The aggregation pipeline executed successfully.";
-            }
-            else {
-                const cappedResultsPipeline = [...pipeline];
-                if (this.config.maxDocumentsPerQuery > 0) {
-                    cappedResultsPipeline.push({ $limit: this.config.maxDocumentsPerQuery });
-                }
-                aggregationCursor = provider.aggregateDb(database, cappedResultsPipeline, {
-                    ...this.getOperationOptions(signal),
-                });
-                const [totalDocuments, cursorResults] = await Promise.all([
-                    this.countAggregationResultDocuments({
-                        provider,
-                        database,
-                        pipeline,
-                        abortSignal: signal,
-                    }),
-                    (0, collectCursorUntilMaxBytes_js_1.collectCursorUntilMaxBytesLimit)({
-                        cursor: aggregationCursor,
-                        configuredMaxBytesPerQuery: this.config.maxBytesPerQuery,
-                        toolResponseBytesLimit: responseBytesLimit,
-                        abortSignal: signal,
-                    }),
-                ]);
-                // If the total number of documents that the aggregation would've
-                // resulted in would be greater than the configured
-                // maxDocumentsPerQuery then we know for sure that the results were
-                // capped.
-                const aggregationResultsCappedByMaxDocumentsLimit = this.config.maxDocumentsPerQuery > 0 &&
-                    !!totalDocuments &&
-                    totalDocuments > this.config.maxDocumentsPerQuery;
-                documents = (0, bsonToJson_js_1.bsonToJson)(cursorResults.documents);
-                aggResultsCount = totalDocuments;
-                appliedLimits = [
-                    aggregationResultsCappedByMaxDocumentsLimit ? "config.maxDocumentsPerQuery" : undefined,
-                    cursorResults.cappedBy,
-                ].filter((limit) => !!limit);
-                successMessage = this.generateMessage({
-                    aggResultsCount,
-                    documents,
-                    appliedLimits,
-                });
-            }
-            return {
-                content: (0, tool_js_1.formatUntrustedData)(successMessage, ...(documents.length > 0 ? [JSON.stringify(documents)] : [])),
-                structuredContent: {
-                    documents,
-                    ...(aggResultsCount !== undefined ? { aggResultsCount } : {}),
-                    appliedLimits,
-                },
-            };
-        }
-        finally {
-            if (aggregationCursor) {
-                void this.safeCloseCursor(aggregationCursor);
-            }
-        }
-    }
-    async safeCloseCursor(cursor) {
-        try {
-            await cursor.close();
-        }
-        catch (error) {
-            this.session.logger.warning({
-                id: index_js_1.LogId.mongodbCursorCloseError,
-                context: "aggregate-db tool",
-                message: `Error when closing the cursor - ${error instanceof Error ? error.message : String(error)}`,
-            });
-        }
-    }
-    assertOnlyUsesPermittedStages(pipeline) {
-        const firstStage = pipeline[0];
-        if (!firstStage || !mongodbSchemas_js_1.DB_AGGREGATE_STAGE_OPERATORS.some((op) => op in firstStage)) {
-            throw new errors_js_1.MongoDBError(errors_js_1.ErrorCodes.InvalidPipeline, `The first stage of the pipeline must be a database-level aggregation stage (one of ${mongodbSchemas_js_1.DB_AGGREGATE_STAGE_OPERATORS.join(", ")})`);
-        }
-        this.assertMqlIsAllowed(pipeline);
-    }
-    async countAggregationResultDocuments({ provider, database, pipeline, abortSignal, }) {
-        const resultsCountAggregation = [...pipeline, { $count: "totalDocuments" }];
-        return await (0, operationWithFallback_js_1.operationWithFallback)(async () => {
-            const aggregationResults = await provider
-                .aggregateDb(database, resultsCountAggregation, {
-                signal: abortSignal,
-            })
-                .maxTimeMS(this.config.maxTimeMS !== undefined
-                ? Math.min(this.config.maxTimeMS, constants_js_1.AGG_COUNT_MAX_TIME_MS_CAP)
-                : constants_js_1.AGG_COUNT_MAX_TIME_MS_CAP)
-                .toArray();
-            const documentWithCount = aggregationResults.length === 1 ? aggregationResults[0] : undefined;
-            const totalDocuments = documentWithCount &&
-                typeof documentWithCount === "object" &&
-                "totalDocuments" in documentWithCount &&
-                typeof documentWithCount.totalDocuments === "number"
-                ? documentWithCount.totalDocuments
-                : 0;
-            return totalDocuments;
-        }, undefined);
-    }
-    generateMessage({ aggResultsCount, documents, appliedLimits, }) {
-        let message = `The aggregation resulted in ${aggResultsCount === undefined ? "indeterminable number of" : aggResultsCount} documents.`;
-        // If we applied a limit or the count is different from the aggregation result count,
-        // communicate what is the actual number of returned documents
-        if (documents.length !== aggResultsCount || appliedLimits.length) {
-            message += ` Returning ${documents.length} documents`;
-            if (appliedLimits.length) {
-                message += ` while respecting the applied limits of ${appliedLimits
-                    .map((limit) => constants_js_1.CURSOR_LIMITS_TO_LLM_TEXT[limit])
-                    .join(", ")}`;
-            }
-            message += ".";
-        }
-        return message;
-    }
-}
-exports.AggregateDBTool = AggregateDBTool;
-AggregateDBTool.toolName = "aggregate-db";
-AggregateDBTool.operationType = "read";
-//# sourceMappingURL=aggregateDB.js.map

package/dist/cjs/tools/mongodb/read/aggregateDB.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"aggregateDB.js","sourceRoot":"","sources":["../../../../../src/tools/mongodb/read/aggregateDB.ts"],"names":[],"mappings":";;;AAAA,6BAAwB;AAGxB,sDAAuF;AAEvF,2CAAoD;AAEpD,yDAAqE;AACrE,kGAAiG;AACjG,wFAAkF;AAClF,gEAAqE;AACrE,gEAMuC;AACvC,+DAAyD;AACzD,4DAAuF;AACvF,kEAA4D;AAE5D,MAAM,uBAAuB,GAAG;IAC5B,SAAS,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,OAAO,EAAE,CAAC,CAAC,QAAQ,CAAC,oDAAoD,CAAC;IAC9F,eAAe,EAAE,OAAC;SACb,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CAAC,oEAAoE,CAAC;IACnF,aAAa,EAAE,OAAC,CAAC,KAAK,CAAC,gCAAiB,CAAC,CAAC,QAAQ,CAAC,gDAAgD,CAAC;CACvG,CAAC;AAEW,QAAA,aAAa,GAAG;IACzB,QAAQ,EAAE,OAAC;SACN,KAAK,CAAC,qCAAiB,CAAC;SACxB,GAAG,CAAC,CAAC,CAAC;SACN,QAAQ,CACL,iHAAiH,gDAA4B,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,gGAAgG,CACpR;CACR,CAAC;AAEF,MAAa,eAAgB,SAAQ,gCAAe;IAApD;;QAEW,gBAAW,GAAG,+CAA+C,CAAC;QAC9D,cAAS,GAAG;YACf,GAAG,iCAAgB;YACnB,GAAG,gCAAe;YAClB,GAAG,qBAAa;YAChB,kBAAkB,EAAE,OAAC;iBAChB,MAAM,EAAE;iBACR,QAAQ,EAAE;iBACV,OAAO,CAAC,qBAAM,CAAC;iBACf,QAAQ,CACL,wIAAwI,CAC3I;SACR,CAAC;QAGc,iBAAY,GAAG,uBAAuB,CAAC;IAmL3D,CAAC;IAjLa,KAAK,CAAC,OAAO,CACnB,EAAE,YAAY,EAAE,QAAQ,EAAE,QAAQ,EAAE,kBAAkB,EAAmC,EACzF,OAA6B;QAE7B,MAAM,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;QAC3B,IAAI,iBAAiB,GAAkC,SAAS,CAAC;QACjE,IAAI,CAAC;YACD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;YAC5D,IAAI,CAAC,6BAA6B,CAAC,QAAQ,CAAC,CAAC;YAE7C,IAAI,cAAsB,CAAC;YAC3B,IAAI,SAAoB,CAAC;YACzB,IAAI,eAAmC,CAAC;YACxC,IAAI,aAAa,GAAqB,EAAE,CAAC;YAEzC,MAAM,iBAAiB,GAAG,IAAA,mCAAoB,EAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;YACnE,IAAI,iBAAiB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC/B,MAAM,IAAI,CAAC,kBAAkB,CAAC,iBAAiB,EAAE,OAAO,CAAC,CAAC;gBAE1D,yFAAyF;gBACzF,iBAAiB,GAAG,QAAQ,CAAC,WAAW,CAAC,QAAQ,EAAE,QAAQ,EAAE;oBACzD,GAAG,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC;iBACtC,CAAC,CAAC;gBAEH,SAAS,GAAG,MAAM,iBAAiB,CAAC,OAAO,EAAE,CAAC;gBAC9C,cAAc,GAAG,iDAAiD,CAAC;YACvE,CAAC;iBAAM,CAAC;gBACJ,MAAM,qBAAqB,GAAG,CAAC,GAAG,QAAQ,CAAC,CAAC;gBAC5C,IAAI,IAAI,CAAC,MAAM,CAAC,oBAAoB,GAAG,CAAC,EAAE,CAAC;oBACvC,qBAAqB,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC,CAAC;gBAC7E,CAAC;gBACD,iBAAiB,GAAG,QAAQ,CAAC,WAAW,CAAC,QAAQ,EAAE,qBAAqB,EAAE;oBACtE,GAAG,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC;iBACtC,CAAC,CAAC;gBAEH,MAAM,CAAC,cAAc,EAAE,aAAa,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;oBACtD,IAAI,CAAC,+BAA+B,CAAC;wBACjC,QAAQ;wBACR,QAAQ;wBACR,QAAQ;wBACR,WAAW,EAAE,MAAM;qBACtB,CAAC;oBACF,IAAA,+DAA+B,EAAC;wBAC5B,MAAM,EAAE,iBAAiB;wBACzB,0BAA0B,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB;wBACxD,sBAAsB,EAAE,kBAAkB;wBAC1C,WAAW,EAAE,MAAM;qBACtB,CAAC;iBACL,CAAC,CAAC;gBAEH,iEAAiE;gBACjE,mDAAmD;gBACnD,mEAAmE;gBACnE,UAAU;gBACV,MAAM,2CAA2C,GAC7C,IAAI,CAAC,MAAM,CAAC,oBAAoB,GAAG,CAAC;oBACpC,CAAC,CAAC,cAAc;oBAChB,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC,oBAAoB,CAAC;gBAEtD,SAAS,GAAG,IAAA,0BAAU,EAAC,aAAa,CAAC,SAAS,CAAC,CAAC;gBAChD,eAAe,GAAG,cAAc,CAAC;gBACjC,aAAa,GAAG;oBACZ,2CAA2C,CAAC,CAAC,CAAC,6BAA6B,CAAC,CAAC,CAAC,SAAS;oBACvF,aAAa,CAAC,QAAQ;iBACzB,CAAC,MAAM,CAAC,CAAC,KAAK,EAA2B,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;gBACtD,cAAc,GAAG,IAAI,CAAC,eAAe,CAAC;oBAClC,eAAe;oBACf,SAAS;oBACT,aAAa;iBAChB,CAAC,CAAC;YACP,CAAC;YAED,OAAO;gBACH,OAAO,EAAE,IAAA,6BAAmB,EACxB,cAAc,EACd,GAAG,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAC/D;gBACD,iBAAiB,EAAE;oBACf,SAAS;oBACT,GAAG,CAAC,eAAe,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC7D,aAAa;iBAChB;aACJ,CAAC;QACN,CAAC;gBAAS,CAAC;YACP,IAAI,iBAAiB,EAAE,CAAC;gBACpB,KAAK,IAAI,CAAC,eAAe,CAAC,iBAAiB,CAAC,CAAC;YACjD,CAAC;QACL,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,eAAe,CAAC,MAAkC;QAC5D,IAAI,CAAC;YACD,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;QACzB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC;gBACxB,EAAE,EAAE,gBAAK,CAAC,uBAAuB;gBACjC,OAAO,EAAE,mBAAmB;gBAC5B,OAAO,EAAE,mCAAmC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE;aACvG,CAAC,CAAC;QACP,CAAC;IACL,CAAC;IAEO,6BAA6B,CAAC,QAAmC;QACrE,MAAM,UAAU,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;QAC/B,IAAI,CAAC,UAAU,IAAI,CAAC,gDAA4B,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,IAAI,UAAU,CAAC,EAAE,CAAC;YAC9E,MAAM,IAAI,wBAAY,CAClB,sBAAU,CAAC,eAAe,EAC1B,sFAAsF,gDAA4B,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CACnI,CAAC;QACN,CAAC;QAED,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;IACtC,CAAC;IAEO,KAAK,CAAC,+BAA+B,CAAC,EAC1C,QAAQ,EACR,QAAQ,EACR,QAAQ,EACR,WAAW,GAMd;QACG,MAAM,uBAAuB,GAAG,CAAC,GAAG,QAAQ,EAAE,EAAE,MAAM,EAAE,gBAAgB,EAAE,CAAC,CAAC;QAC5E,OAAO,MAAM,IAAA,gDAAqB,EAAC,KAAK,IAAiC,EAAE;YACvE,MAAM,kBAAkB,GAAG,MAAM,QAAQ;iBACpC,WAAW,CAAC,QAAQ,EAAE,uBAAuB,EAAE;gBAC5C,MAAM,EAAE,WAAW;aACtB,CAAC;iBACD,SAAS,CACN,IAAI,CAAC,MAAM,CAAC,SAAS,KAAK,SAAS;gBAC/B,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,wCAAyB,CAAC;gBAC5D,CAAC,CAAC,wCAAyB,CAClC;iBACA,OAAO,EAAE,CAAC;YAEf,MAAM,iBAAiB,GAAY,kBAAkB,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YACvG,MAAM,cAAc,GAChB,iBAAiB;gBACjB,OAAO,iBAAiB,KAAK,QAAQ;gBACrC,gBAAgB,IAAI,iBAAiB;gBACrC,OAAO,iBAAiB,CAAC,cAAc,KAAK,QAAQ;gBAChD,CAAC,CAAC,iBAAiB,CAAC,cAAc;gBAClC,CAAC,CAAC,CAAC,CAAC;YAEZ,OAAO,cAAc,CAAC;QAC1B,CAAC,EAAE,SAAS,CAAC,CAAC;IAClB,CAAC;IAEO,eAAe,CAAC,EACpB,eAAe,EACf,SAAS,EACT,aAAa,GAKhB;QACG,IAAI,OAAO,GAAG,+BAA+B,eAAe,KAAK,SAAS,CAAC,CAAC,CAAC,0BAA0B,CAAC,CAAC,CAAC,eAAe,aAAa,CAAC;QAEvI,qFAAqF;QACrF,8DAA8D;QAC9D,IAAI,SAAS,CAAC,MAAM,KAAK,eAAe,IAAI,aAAa,CAAC,MAAM,EAAE,CAAC;YAC/D,OAAO,IAAI,cAAc,SAAS,CAAC,MAAM,YAAY,CAAC;YACtD,IAAI,aAAa,CAAC,MAAM,EAAE,CAAC;gBACvB,OAAO,IAAI,2CAA2C,aAAa;qBAC9D,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,wCAAyB,CAAC,KAAK,CAAC,CAAC;qBAChD,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YACtB,CAAC;YAED,OAAO,IAAI,GAAG,CAAC;QACnB,CAAC;QAED,OAAO,OAAO,CAAC;IACnB,CAAC;;AAnML,0CAoMC;AAnMU,wBAAQ,GAAG,cAAc,AAAjB,CAAkB;AAc1B,6BAAa,GAAkB,MAAM,AAAxB,CAAyB"}

package/dist/cjs/tools/mongodb/read/count.d.ts

@@ -1,22 +0,0 @@
-import { MongoDBToolBase } from "../mongodbTool.js";
-import type { ToolArgs, OperationType, ToolExecutionContext, ToolResult } from "../../tool.js";
-import { z } from "zod";
-export declare const CountArgs: {
-    query: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
-};
-export declare class CountTool extends MongoDBToolBase {
-    static toolName: string;
-    description: string;
-    argsShape: {
-        query: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
-        collection: z.ZodString;
-        database: z.ZodString;
-        connectionId: z.ZodString;
-    };
-    static operationType: OperationType;
-    outputSchema: {
-        count: z.ZodNumber;
-    };
-    protected execute({ connectionId, database, collection, query }: ToolArgs<typeof this.argsShape>, { signal }: ToolExecutionContext): Promise<ToolResult<typeof this.outputSchema>>;
-}
-//# sourceMappingURL=count.d.ts.map

package/dist/cjs/tools/mongodb/read/count.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"count.d.ts","sourceRoot":"","sources":["../../../../../src/tools/mongodb/read/count.ts"],"names":[],"mappings":"AAAA,OAAO,EAAuC,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACzF,OAAO,KAAK,EAAE,QAAQ,EAAE,aAAa,EAAE,oBAAoB,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AAG/F,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,eAAO,MAAM,SAAS;;CAMrB,CAAC;AAMF,qBAAa,SAAU,SAAQ,eAAe;IAC1C,MAAM,CAAC,QAAQ,SAAW;IACnB,WAAW,SAC+G;IAC1H,SAAS;;;;;MAId;IAEF,MAAM,CAAC,aAAa,EAAE,aAAa,CAAU;IAE7B,YAAY;;MAAqB;cAEjC,OAAO,CACnB,EAAE,YAAY,EAAE,QAAQ,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,QAAQ,CAAC,OAAO,IAAI,CAAC,SAAS,CAAC,EAC9E,EAAE,MAAM,EAAE,EAAE,oBAAoB,GACjC,OAAO,CAAC,UAAU,CAAC,OAAO,IAAI,CAAC,YAAY,CAAC,CAAC;CA+CnD"}