mongodb-mcp-server 1.1.0-prerelease.1 → 1.2.0

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

@@ -3,14 +3,226 @@ import type { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
 import { MongoDBToolBase } from "../mongodbTool.js";
 import type { ToolArgs, OperationType, ToolExecutionContext } from "../../tool.js";
 export declare const AggregateArgs: {
-    pipeline: z.ZodArray<z.AnyZodObject, "many">;
+    pipeline: z.ZodArray<z.ZodUnion<[z.AnyZodObject, z.ZodObject<{
+        $vectorSearch: z.ZodObject<{
+            exact: z.ZodDefault<z.ZodOptional<z.ZodBoolean>>;
+            index: z.ZodString;
+            path: z.ZodString;
+            queryVector: z.ZodUnion<[z.ZodString, z.ZodArray<z.ZodNumber, "many">]>;
+            numCandidates: z.ZodOptional<z.ZodNumber>;
+            limit: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
+            filter: z.ZodOptional<z.AnyZodObject>;
+            embeddingParameters: z.ZodOptional<z.ZodObject<{
+                outputDimension: z.ZodDefault<z.ZodOptional<z.ZodUnion<[z.ZodLiteral<256>, z.ZodLiteral<512>, z.ZodLiteral<1024>, z.ZodLiteral<2048>, z.ZodLiteral<4096>]>>>;
+                outputDType: z.ZodDefault<z.ZodOptional<z.ZodEnum<["float", "int8", "uint8", "binary", "ubinary"]>>>;
+            } & {
+                model: z.ZodDefault<z.ZodEnum<["voyage-3-large", "voyage-3.5", "voyage-3.5-lite", "voyage-code-3"]>>;
+            }, "strip", z.ZodTypeAny, {
+                outputDimension: 256 | 512 | 1024 | 2048 | 4096;
+                outputDType: "binary" | "float" | "int8" | "uint8" | "ubinary";
+                model: "voyage-3-large" | "voyage-3.5" | "voyage-3.5-lite" | "voyage-code-3";
+            }, {
+                outputDimension?: 256 | 512 | 1024 | 2048 | 4096 | undefined;
+                outputDType?: "binary" | "float" | "int8" | "uint8" | "ubinary" | undefined;
+                model?: "voyage-3-large" | "voyage-3.5" | "voyage-3.5-lite" | "voyage-code-3" | undefined;
+            }>>;
+        }, "passthrough", z.ZodTypeAny, z.objectOutputType<{
+            exact: z.ZodDefault<z.ZodOptional<z.ZodBoolean>>;
+            index: z.ZodString;
+            path: z.ZodString;
+            queryVector: z.ZodUnion<[z.ZodString, z.ZodArray<z.ZodNumber, "many">]>;
+            numCandidates: z.ZodOptional<z.ZodNumber>;
+            limit: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
+            filter: z.ZodOptional<z.AnyZodObject>;
+            embeddingParameters: z.ZodOptional<z.ZodObject<{
+                outputDimension: z.ZodDefault<z.ZodOptional<z.ZodUnion<[z.ZodLiteral<256>, z.ZodLiteral<512>, z.ZodLiteral<1024>, z.ZodLiteral<2048>, z.ZodLiteral<4096>]>>>;
+                outputDType: z.ZodDefault<z.ZodOptional<z.ZodEnum<["float", "int8", "uint8", "binary", "ubinary"]>>>;
+            } & {
+                model: z.ZodDefault<z.ZodEnum<["voyage-3-large", "voyage-3.5", "voyage-3.5-lite", "voyage-code-3"]>>;
+            }, "strip", z.ZodTypeAny, {
+                outputDimension: 256 | 512 | 1024 | 2048 | 4096;
+                outputDType: "binary" | "float" | "int8" | "uint8" | "ubinary";
+                model: "voyage-3-large" | "voyage-3.5" | "voyage-3.5-lite" | "voyage-code-3";
+            }, {
+                outputDimension?: 256 | 512 | 1024 | 2048 | 4096 | undefined;
+                outputDType?: "binary" | "float" | "int8" | "uint8" | "ubinary" | undefined;
+                model?: "voyage-3-large" | "voyage-3.5" | "voyage-3.5-lite" | "voyage-code-3" | undefined;
+            }>>;
+        }, z.ZodTypeAny, "passthrough">, z.objectInputType<{
+            exact: z.ZodDefault<z.ZodOptional<z.ZodBoolean>>;
+            index: z.ZodString;
+            path: z.ZodString;
+            queryVector: z.ZodUnion<[z.ZodString, z.ZodArray<z.ZodNumber, "many">]>;
+            numCandidates: z.ZodOptional<z.ZodNumber>;
+            limit: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
+            filter: z.ZodOptional<z.AnyZodObject>;
+            embeddingParameters: z.ZodOptional<z.ZodObject<{
+                outputDimension: z.ZodDefault<z.ZodOptional<z.ZodUnion<[z.ZodLiteral<256>, z.ZodLiteral<512>, z.ZodLiteral<1024>, z.ZodLiteral<2048>, z.ZodLiteral<4096>]>>>;
+                outputDType: z.ZodDefault<z.ZodOptional<z.ZodEnum<["float", "int8", "uint8", "binary", "ubinary"]>>>;
+            } & {
+                model: z.ZodDefault<z.ZodEnum<["voyage-3-large", "voyage-3.5", "voyage-3.5-lite", "voyage-code-3"]>>;
+            }, "strip", z.ZodTypeAny, {
+                outputDimension: 256 | 512 | 1024 | 2048 | 4096;
+                outputDType: "binary" | "float" | "int8" | "uint8" | "ubinary";
+                model: "voyage-3-large" | "voyage-3.5" | "voyage-3.5-lite" | "voyage-code-3";
+            }, {
+                outputDimension?: 256 | 512 | 1024 | 2048 | 4096 | undefined;
+                outputDType?: "binary" | "float" | "int8" | "uint8" | "ubinary" | undefined;
+                model?: "voyage-3-large" | "voyage-3.5" | "voyage-3.5-lite" | "voyage-code-3" | undefined;
+            }>>;
+        }, z.ZodTypeAny, "passthrough">>;
+    }, "strip", z.ZodTypeAny, {
+        $vectorSearch: {
+            path: string;
+            exact: boolean;
+            index: string;
+            limit: number;
+            queryVector: string | number[];
+            filter?: {
+                [x: string]: any;
+            } | undefined;
+            embeddingParameters?: {
+                outputDimension: 256 | 512 | 1024 | 2048 | 4096;
+                outputDType: "binary" | "float" | "int8" | "uint8" | "ubinary";
+                model: "voyage-3-large" | "voyage-3.5" | "voyage-3.5-lite" | "voyage-code-3";
+            } | undefined;
+            numCandidates?: number | undefined;
+        } & {
+            [k: string]: unknown;
+        };
+    }, {
+        $vectorSearch: {
+            path: string;
+            index: string;
+            queryVector: string | number[];
+            exact?: boolean | undefined;
+            filter?: {
+                [x: string]: any;
+            } | undefined;
+            embeddingParameters?: {
+                outputDimension?: 256 | 512 | 1024 | 2048 | 4096 | undefined;
+                outputDType?: "binary" | "float" | "int8" | "uint8" | "ubinary" | undefined;
+                model?: "voyage-3-large" | "voyage-3.5" | "voyage-3.5-lite" | "voyage-code-3" | undefined;
+            } | undefined;
+            limit?: number | undefined;
+            numCandidates?: number | undefined;
+        } & {
+            [k: string]: unknown;
+        };
+    }>]>, "many">;
     responseBytesLimit: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
 };
 export declare class AggregateTool extends MongoDBToolBase {
     name: string;
     protected description: string;
     protected argsShape: {
-        pipeline: z.ZodArray<z.AnyZodObject, "many">;
+        pipeline: z.ZodArray<z.ZodUnion<[z.AnyZodObject, z.ZodObject<{
+            $vectorSearch: z.ZodObject<{
+                exact: z.ZodDefault<z.ZodOptional<z.ZodBoolean>>;
+                index: z.ZodString;
+                path: z.ZodString;
+                queryVector: z.ZodUnion<[z.ZodString, z.ZodArray<z.ZodNumber, "many">]>;
+                numCandidates: z.ZodOptional<z.ZodNumber>;
+                limit: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
+                filter: z.ZodOptional<z.AnyZodObject>;
+                embeddingParameters: z.ZodOptional<z.ZodObject<{
+                    outputDimension: z.ZodDefault<z.ZodOptional<z.ZodUnion<[z.ZodLiteral<256>, z.ZodLiteral<512>, z.ZodLiteral<1024>, z.ZodLiteral<2048>, z.ZodLiteral<4096>]>>>;
+                    outputDType: z.ZodDefault<z.ZodOptional<z.ZodEnum<["float", "int8", "uint8", "binary", "ubinary"]>>>;
+                } & {
+                    model: z.ZodDefault<z.ZodEnum<["voyage-3-large", "voyage-3.5", "voyage-3.5-lite", "voyage-code-3"]>>;
+                }, "strip", z.ZodTypeAny, {
+                    outputDimension: 256 | 512 | 1024 | 2048 | 4096;
+                    outputDType: "binary" | "float" | "int8" | "uint8" | "ubinary";
+                    model: "voyage-3-large" | "voyage-3.5" | "voyage-3.5-lite" | "voyage-code-3";
+                }, {
+                    outputDimension?: 256 | 512 | 1024 | 2048 | 4096 | undefined;
+                    outputDType?: "binary" | "float" | "int8" | "uint8" | "ubinary" | undefined;
+                    model?: "voyage-3-large" | "voyage-3.5" | "voyage-3.5-lite" | "voyage-code-3" | undefined;
+                }>>;
+            }, "passthrough", z.ZodTypeAny, z.objectOutputType<{
+                exact: z.ZodDefault<z.ZodOptional<z.ZodBoolean>>;
+                index: z.ZodString;
+                path: z.ZodString;
+                queryVector: z.ZodUnion<[z.ZodString, z.ZodArray<z.ZodNumber, "many">]>;
+                numCandidates: z.ZodOptional<z.ZodNumber>;
+                limit: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
+                filter: z.ZodOptional<z.AnyZodObject>;
+                embeddingParameters: z.ZodOptional<z.ZodObject<{
+                    outputDimension: z.ZodDefault<z.ZodOptional<z.ZodUnion<[z.ZodLiteral<256>, z.ZodLiteral<512>, z.ZodLiteral<1024>, z.ZodLiteral<2048>, z.ZodLiteral<4096>]>>>;
+                    outputDType: z.ZodDefault<z.ZodOptional<z.ZodEnum<["float", "int8", "uint8", "binary", "ubinary"]>>>;
+                } & {
+                    model: z.ZodDefault<z.ZodEnum<["voyage-3-large", "voyage-3.5", "voyage-3.5-lite", "voyage-code-3"]>>;
+                }, "strip", z.ZodTypeAny, {
+                    outputDimension: 256 | 512 | 1024 | 2048 | 4096;
+                    outputDType: "binary" | "float" | "int8" | "uint8" | "ubinary";
+                    model: "voyage-3-large" | "voyage-3.5" | "voyage-3.5-lite" | "voyage-code-3";
+                }, {
+                    outputDimension?: 256 | 512 | 1024 | 2048 | 4096 | undefined;
+                    outputDType?: "binary" | "float" | "int8" | "uint8" | "ubinary" | undefined;
+                    model?: "voyage-3-large" | "voyage-3.5" | "voyage-3.5-lite" | "voyage-code-3" | undefined;
+                }>>;
+            }, z.ZodTypeAny, "passthrough">, z.objectInputType<{
+                exact: z.ZodDefault<z.ZodOptional<z.ZodBoolean>>;
+                index: z.ZodString;
+                path: z.ZodString;
+                queryVector: z.ZodUnion<[z.ZodString, z.ZodArray<z.ZodNumber, "many">]>;
+                numCandidates: z.ZodOptional<z.ZodNumber>;
+                limit: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
+                filter: z.ZodOptional<z.AnyZodObject>;
+                embeddingParameters: z.ZodOptional<z.ZodObject<{
+                    outputDimension: z.ZodDefault<z.ZodOptional<z.ZodUnion<[z.ZodLiteral<256>, z.ZodLiteral<512>, z.ZodLiteral<1024>, z.ZodLiteral<2048>, z.ZodLiteral<4096>]>>>;
+                    outputDType: z.ZodDefault<z.ZodOptional<z.ZodEnum<["float", "int8", "uint8", "binary", "ubinary"]>>>;
+                } & {
+                    model: z.ZodDefault<z.ZodEnum<["voyage-3-large", "voyage-3.5", "voyage-3.5-lite", "voyage-code-3"]>>;
+                }, "strip", z.ZodTypeAny, {
+                    outputDimension: 256 | 512 | 1024 | 2048 | 4096;
+                    outputDType: "binary" | "float" | "int8" | "uint8" | "ubinary";
+                    model: "voyage-3-large" | "voyage-3.5" | "voyage-3.5-lite" | "voyage-code-3";
+                }, {
+                    outputDimension?: 256 | 512 | 1024 | 2048 | 4096 | undefined;
+                    outputDType?: "binary" | "float" | "int8" | "uint8" | "ubinary" | undefined;
+                    model?: "voyage-3-large" | "voyage-3.5" | "voyage-3.5-lite" | "voyage-code-3" | undefined;
+                }>>;
+            }, z.ZodTypeAny, "passthrough">>;
+        }, "strip", z.ZodTypeAny, {
+            $vectorSearch: {
+                path: string;
+                exact: boolean;
+                index: string;
+                limit: number;
+                queryVector: string | number[];
+                filter?: {
+                    [x: string]: any;
+                } | undefined;
+                embeddingParameters?: {
+                    outputDimension: 256 | 512 | 1024 | 2048 | 4096;
+                    outputDType: "binary" | "float" | "int8" | "uint8" | "ubinary";
+                    model: "voyage-3-large" | "voyage-3.5" | "voyage-3.5-lite" | "voyage-code-3";
+                } | undefined;
+                numCandidates?: number | undefined;
+            } & {
+                [k: string]: unknown;
+            };
+        }, {
+            $vectorSearch: {
+                path: string;
+                index: string;
+                queryVector: string | number[];
+                exact?: boolean | undefined;
+                filter?: {
+                    [x: string]: any;
+                } | undefined;
+                embeddingParameters?: {
+                    outputDimension?: 256 | 512 | 1024 | 2048 | 4096 | undefined;
+                    outputDType?: "binary" | "float" | "int8" | "uint8" | "ubinary" | undefined;
+                    model?: "voyage-3-large" | "voyage-3.5" | "voyage-3.5-lite" | "voyage-code-3" | undefined;
+                } | undefined;
+                limit?: number | undefined;
+                numCandidates?: number | undefined;
+            } & {
+                [k: string]: unknown;
+            };
+        }>]>, "many">;
         responseBytesLimit: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
         database: z.ZodString;
         collection: z.ZodString;
@@ -20,6 +232,7 @@ export declare class AggregateTool extends MongoDBToolBase {
     private safeCloseCursor;
     private assertOnlyUsesPermittedStages;
     private countAggregationResultDocuments;
+    private replaceRawValuesWithEmbeddingsIfNecessary;
     private generateMessage;
 }
 //# sourceMappingURL=aggregate.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"aggregate.d.ts","sourceRoot":"","sources":["../../../../../src/tools/mongodb/read/aggregate.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,oCAAoC,CAAC;AAEzE,OAAO,EAAmB,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACrE,OAAO,KAAK,EAAE,QAAQ,EAAE,aAAa,EAAE,oBAAoB,EAAE,MAAM,eAAe,CAAC;AAWnF,eAAO,MAAM,aAAa;;;CAMzB,CAAC;AAEF,qBAAa,aAAc,SAAQ,eAAe;IACvC,IAAI,SAAe;IAC1B,SAAS,CAAC,WAAW,SAAqD;IAC1E,SAAS,CAAC,SAAS;;;;;MAGjB;IACK,aAAa,EAAE,aAAa,CAAU;cAE7B,OAAO,CACnB,EAAE,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE,kBAAkB,EAAE,EAAE,QAAQ,CAAC,OAAO,IAAI,CAAC,SAAS,CAAC,EACvF,EAAE,MAAM,EAAE,EAAE,oBAAoB,GACjC,OAAO,CAAC,cAAc,CAAC;YA6DZ,eAAe;IAY7B,OAAO,CAAC,6BAA6B;YAsBvB,+BAA+B;IA+B7C,OAAO,CAAC,eAAe;CAqB1B"}
+{"version":3,"file":"aggregate.d.ts","sourceRoot":"","sources":["../../../../../src/tools/mongodb/read/aggregate.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,oCAAoC,CAAC;AAEzE,OAAO,EAAmB,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACrE,OAAO,KAAK,EAAE,QAAQ,EAAE,aAAa,EAAE,oBAAoB,EAAE,MAAM,eAAe,CAAC;AAuDnF,eAAO,MAAM,aAAa;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAUzB,CAAC;AAEF,qBAAa,aAAc,SAAQ,eAAe;IACvC,IAAI,SAAe;IAC1B,SAAS,CAAC,WAAW,SAAqD;IAC1E,SAAS,CAAC,SAAS;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;MAGjB;IACK,aAAa,EAAE,aAAa,CAAU;cAE7B,OAAO,CACnB,EAAE,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE,kBAAkB,EAAE,EAAE,QAAQ,CAAC,OAAO,IAAI,CAAC,SAAS,CAAC,EACvF,EAAE,MAAM,EAAE,EAAE,oBAAoB,GACjC,OAAO,CAAC,cAAc,CAAC;YAkEZ,eAAe;YAYf,6BAA6B;YAgC7B,+BAA+B;YA+B/B,yCAAyC;IA8CvD,OAAO,CAAC,eAAe;CAqB1B"}

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

@@ -12,8 +12,43 @@ const operationWithFallback_js_1 = require("../../../helpers/operationWithFallba
 const constants_js_1 = require("../../../helpers/constants.js");
 const args_js_1 = require("../../args.js");
 const logger_js_1 = require("../../../common/logger.js");
+const embeddingsProvider_js_1 = require("../../../common/search/embeddingsProvider.js");
+const AnyStage = (0, args_js_1.zEJSON)();
+const VectorSearchStage = zod_1.z.object({
+    $vectorSearch: zod_1.z
+        .object({
+        exact: zod_1.z
+            .boolean()
+            .optional()
+            .default(false)
+            .describe("When true, uses an ENN algorithm, otherwise uses ANN. Using ENN is not compatible with numCandidates, in that case, numCandidates must be left empty."),
+        index: zod_1.z.string().describe("Name of the index, as retrieved from the `collection-indexes` tool."),
+        path: zod_1.z
+            .string()
+            .describe("Field, in dot notation, where to search. There must be a vector search index for that field. Note to LLM: When unsure, use the 'collection-indexes' tool to validate that the field is indexed with a vector search index."),
+        queryVector: zod_1.z
+            .union([zod_1.z.string(), zod_1.z.array(zod_1.z.number())])
+            .describe("The content to search for. The embeddingParameters field is mandatory if the queryVector is a string, in that case, the tool generates the embedding automatically using the provided configuration."),
+        numCandidates: zod_1.z
+            .number()
+            .int()
+            .positive()
+            .optional()
+            .describe("Number of candidates for the ANN algorithm. Mandatory when exact is false."),
+        limit: zod_1.z.number().int().positive().optional().default(10),
+        filter: (0, args_js_1.zEJSON)()
+            .optional()
+            .describe("MQL filter that can only use pre-filter fields from the index definition. Note to LLM: If unsure, use the `collection-indexes` tool to learn which fields can be used for pre-filtering."),
+        embeddingParameters: embeddingsProvider_js_1.zSupportedEmbeddingParameters
+            .optional()
+            .describe("The embedding model and its parameters to use to generate embeddings before searching. It is mandatory if queryVector is a string value. Note to LLM: If unsure, ask the user before providing one."),
+    })
+        .passthrough(),
+});
 exports.AggregateArgs = {
-    pipeline: zod_1.z.array((0, args_js_1.zEJSON)()).describe("An array of aggregation stages to execute"),
+    pipeline: zod_1.z
+        .array(zod_1.z.union([AnyStage, VectorSearchStage]))
+        .describe("An array of aggregation stages to execute. $vectorSearch can only appear as the first stage of the aggregation pipeline or as the first stage of a $unionWith subpipeline. When using $vectorSearch, unless the user explicitly asks for the embeddings, $unset any embedding field to avoid reaching context limits."),
     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 maxBytesPerQuery and cannot be exceeded. \
 Note to LLM: If the entire aggregation result is required, use the "export" tool instead of increasing this limit.\
@@ -34,7 +69,7 @@ class AggregateTool extends mongodbTool_js_1.MongoDBToolBase {
         let aggregationCursor = undefined;
         try {
             const provider = await this.ensureConnected();
-            this.assertOnlyUsesPermittedStages(pipeline);
+            await this.assertOnlyUsesPermittedStages(pipeline);
             // Check if aggregate operation uses an index if enabled
             if (this.config.indexCheck) {
                 await (0, indexCheck_js_1.checkIndexUsage)(provider, database, collection, "aggregate", async () => {
@@ -43,6 +78,11 @@ class AggregateTool extends mongodbTool_js_1.MongoDBToolBase {
                         .explain("queryPlanner");
                 });
             }
+            pipeline = await this.replaceRawValuesWithEmbeddingsIfNecessary({
+                database,
+                collection,
+                pipeline,
+            });
             const cappedResultsPipeline = [...pipeline];
             if (this.config.maxDocumentsPerQuery > 0) {
                 cappedResultsPipeline.push({ $limit: this.config.maxDocumentsPerQuery });
@@ -72,7 +112,7 @@ class AggregateTool extends mongodbTool_js_1.MongoDBToolBase {
                         aggregationResultsCappedByMaxDocumentsLimit ? "config.maxDocumentsPerQuery" : undefined,
                         cursorResults.cappedBy,
                     ].filter((limit) => !!limit),
-                }), cursorResults.documents.length > 0 ? bson_1.EJSON.stringify(cursorResults.documents) : undefined),
+                }), ...(cursorResults.documents.length > 0 ? [bson_1.EJSON.stringify(cursorResults.documents)] : [])),
             };
         }
         finally {
@@ -93,8 +133,9 @@ class AggregateTool extends mongodbTool_js_1.MongoDBToolBase {
             });
         }
     }
-    assertOnlyUsesPermittedStages(pipeline) {
+    async assertOnlyUsesPermittedStages(pipeline) {
         const writeOperations = ["update", "create", "delete"];
+        const isSearchSupported = await this.session.isSearchSupported();
         let writeStageForbiddenError = "";
         if (this.config.readOnly) {
             writeStageForbiddenError = "In readOnly mode you can not run pipelines with $out or $merge stages.";
@@ -103,13 +144,18 @@ class AggregateTool extends mongodbTool_js_1.MongoDBToolBase {
             writeStageForbiddenError =
                 "When 'create', 'update', or 'delete' operations are disabled, you can not run pipelines with $out or $merge stages.";
         }
-        if (!writeStageForbiddenError) {
-            return;
-        }
         for (const stage of pipeline) {
-            if (stage.$out || stage.$merge) {
+            // This validates that in readOnly mode or "write" operations are disabled, we can't use $out or $merge.
+            // This is really important because aggregates are the only "multi-faceted" tool in the MQL, where you
+            // can both read and write.
+            if ((stage.$out || stage.$merge) && writeStageForbiddenError) {
                 throw new errors_js_1.MongoDBError(errors_js_1.ErrorCodes.ForbiddenWriteOperation, writeStageForbiddenError);
             }
+            // This ensure that you can't use $vectorSearch if the cluster does not support MongoDB Search
+            // either in Atlas or in a local cluster.
+            if (stage.$vectorSearch && !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, }) {
@@ -129,6 +175,34 @@ class AggregateTool extends mongodbTool_js_1.MongoDBToolBase {
             return totalDocuments;
         }, undefined);
     }
+    async replaceRawValuesWithEmbeddingsIfNecessary({ database, collection, pipeline, }) {
+        for (const stage of pipeline) {
+            if ("$vectorSearch" in stage) {
+                const { $vectorSearch: vectorSearchStage } = stage;
+                if (Array.isArray(vectorSearchStage.queryVector)) {
+                    continue;
+                }
+                if (!vectorSearchStage.embeddingParameters) {
+                    throw new errors_js_1.MongoDBError(errors_js_1.ErrorCodes.AtlasVectorSearchInvalidQuery, "embeddingModel is mandatory if queryVector is a raw string.");
+                }
+                const embeddingParameters = vectorSearchStage.embeddingParameters;
+                delete vectorSearchStage.embeddingParameters;
+                const [embeddings] = await this.session.vectorSearchEmbeddingsManager.generateEmbeddings({
+                    database,
+                    collection,
+                    path: vectorSearchStage.path,
+                    rawValues: [vectorSearchStage.queryVector],
+                    embeddingParameters,
+                    inputType: "query",
+                });
+                // $vectorSearch.queryVector can be a BSON.Binary: that it's not either number or an array.
+                // It's not exactly valid from the LLM perspective (they can't provide binaries).
+                // That's why we overwrite the stage in an untyped way, as what we expose and what LLMs can use is different.
+                vectorSearchStage.queryVector = embeddings;
+            }
+        }
+        return pipeline;
+    }
     generateMessage({ aggResultsCount, documents, appliedLimits, }) {
         const appliedLimitText = appliedLimits.length
             ? `\

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

@@ -1 +1 @@
-{"version":3,"file":"aggregate.js","sourceRoot":"","sources":["../../../../../src/tools/mongodb/read/aggregate.ts"],"names":[],"mappings":";;;AAAA,6BAAwB;AAIxB,sDAAqE;AAErE,2CAAoD;AACpD,kEAAiE;AACjE,+BAA4C;AAC5C,yDAAqE;AACrE,kGAAiG;AACjG,wFAAkF;AAClF,gEAA6G;AAC7G,2CAAuC;AACvC,yDAAkD;AAErC,QAAA,aAAa,GAAG;IACzB,QAAQ,EAAE,OAAC,CAAC,KAAK,CAAC,IAAA,gBAAM,GAAE,CAAC,CAAC,QAAQ,CAAC,2CAA2C,CAAC;IACjF,kBAAkB,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,qBAAM,CAAC,CAAC,QAAQ,CAAC;;;CAGtE,CAAC;CACD,CAAC;AAEF,MAAa,aAAc,SAAQ,gCAAe;IAAlD;;QACW,SAAI,GAAG,WAAW,CAAC;QAChB,gBAAW,GAAG,iDAAiD,CAAC;QAChE,cAAS,GAAG;YAClB,GAAG,gCAAe;YAClB,GAAG,qBAAa;SACnB,CAAC;QACK,kBAAa,GAAkB,MAAM,CAAC;IAwJjD,CAAC;IAtJa,KAAK,CAAC,OAAO,CACnB,EAAE,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE,kBAAkB,EAAmC,EACvF,EAAE,MAAM,EAAwB;QAEhC,IAAI,iBAAiB,GAAkC,SAAS,CAAC;QACjE,IAAI,CAAC;YACD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;YAE9C,IAAI,CAAC,6BAA6B,CAAC,QAAQ,CAAC,CAAC;YAE7C,wDAAwD;YACxD,IAAI,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;gBACzB,MAAM,IAAA,+BAAe,EAAC,QAAQ,EAAE,QAAQ,EAAE,UAAU,EAAE,WAAW,EAAE,KAAK,IAAI,EAAE;oBAC1E,OAAO,QAAQ;yBACV,SAAS,CAAC,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,EAAE,EAAE,YAAY,EAAE,SAAS,EAAE,CAAC;yBAC1E,OAAO,CAAC,cAAc,CAAC,CAAC;gBACjC,CAAC,CAAC,CAAC;YACP,CAAC;YAED,MAAM,qBAAqB,GAAG,CAAC,GAAG,QAAQ,CAAC,CAAC;YAC5C,IAAI,IAAI,CAAC,MAAM,CAAC,oBAAoB,GAAG,CAAC,EAAE,CAAC;gBACvC,qBAAqB,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC,CAAC;YAC7E,CAAC;YACD,iBAAiB,GAAG,QAAQ,CAAC,SAAS,CAAC,QAAQ,EAAE,UAAU,EAAE,qBAAqB,CAAC,CAAC;YAEpF,MAAM,CAAC,cAAc,EAAE,aAAa,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBACtD,IAAI,CAAC,+BAA+B,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC;gBAClF,IAAA,+DAA+B,EAAC;oBAC5B,MAAM,EAAE,iBAAiB;oBACzB,0BAA0B,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB;oBACxD,sBAAsB,EAAE,kBAAkB;oBAC1C,WAAW,EAAE,MAAM;iBACtB,CAAC;aACL,CAAC,CAAC;YAEH,iEAAiE;YACjE,mDAAmD;YACnD,mEAAmE;YACnE,UAAU;YACV,MAAM,2CAA2C,GAC7C,IAAI,CAAC,MAAM,CAAC,oBAAoB,GAAG,CAAC;gBACpC,CAAC,CAAC,cAAc;gBAChB,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC,oBAAoB,CAAC;YAEtD,OAAO;gBACH,OAAO,EAAE,IAAA,6BAAmB,EACxB,IAAI,CAAC,eAAe,CAAC;oBACjB,eAAe,EAAE,cAAc;oBAC/B,SAAS,EAAE,aAAa,CAAC,SAAS;oBAClC,aAAa,EAAE;wBACX,2CAA2C,CAAC,CAAC,CAAC,6BAA6B,CAAC,CAAC,CAAC,SAAS;wBACvF,aAAa,CAAC,QAAQ;qBACzB,CAAC,MAAM,CAAC,CAAC,KAAK,EAAmD,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC;iBAChF,CAAC,EACF,aAAa,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,YAAK,CAAC,SAAS,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAC5F;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,iBAAK,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,CAAC,QAAmC;QACrE,MAAM,eAAe,GAAoB,CAAC,QAAQ,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC;QACxE,IAAI,wBAAwB,GAAG,EAAE,CAAC;QAElC,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;YACvB,wBAAwB,GAAG,wEAAwE,CAAC;QACxG,CAAC;aAAM,IAAI,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAkB,CAAC,CAAC,EAAE,CAAC;YAC7F,wBAAwB;gBACpB,qHAAqH,CAAC;QAC9H,CAAC;QAED,IAAI,CAAC,wBAAwB,EAAE,CAAC;YAC5B,OAAO;QACX,CAAC;QAED,KAAK,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;YAC3B,IAAI,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;gBAC7B,MAAM,IAAI,wBAAY,CAAC,sBAAU,CAAC,uBAAuB,EAAE,wBAAwB,CAAC,CAAC;YACzF,CAAC;QACL,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,+BAA+B,CAAC,EAC1C,QAAQ,EACR,QAAQ,EACR,UAAU,EACV,QAAQ,GAMX;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,CAAC;iBACxD,SAAS,CAAC,wCAAyB,CAAC;iBACpC,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,MAAM,gBAAgB,GAAG,aAAa,CAAC,MAAM;YACzC,CAAC,CAAC;yCAC2B,aAAa,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,wCAAyB,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;;CAEjH;YACW,CAAC,CAAC,EAAE,CAAC;QAET,OAAO;8BACe,eAAe,KAAK,SAAS,CAAC,CAAC,CAAC,0BAA0B,CAAC,CAAC,CAAC,eAAe;YAC9F,SAAS,CAAC,MAAM,aAAa,gBAAgB,CAAC,CAAC,CAAC,IAAI,gBAAgB,EAAE,CAAC,CAAC,CAAC,GAAG;CACvF,CAAC;IACE,CAAC;CACJ;AA/JD,sCA+JC"}
+{"version":3,"file":"aggregate.js","sourceRoot":"","sources":["../../../../../src/tools/mongodb/read/aggregate.ts"],"names":[],"mappings":";;;AAAA,6BAAwB;AAIxB,sDAAqE;AAErE,2CAAoD;AACpD,kEAAiE;AACjE,+BAA4C;AAC5C,yDAAqE;AACrE,kGAAiG;AACjG,wFAAkF;AAClF,gEAA6G;AAC7G,2CAAuC;AACvC,yDAAkD;AAClD,wFAA6F;AAE7F,MAAM,QAAQ,GAAG,IAAA,gBAAM,GAAE,CAAC;AAC1B,MAAM,iBAAiB,GAAG,OAAC,CAAC,MAAM,CAAC;IAC/B,aAAa,EAAE,OAAC;SACX,MAAM,CAAC;QACJ,KAAK,EAAE,OAAC;aACH,OAAO,EAAE;aACT,QAAQ,EAAE;aACV,OAAO,CAAC,KAAK,CAAC;aACd,QAAQ,CACL,uJAAuJ,CAC1J;QACL,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,qEAAqE,CAAC;QACjG,IAAI,EAAE,OAAC;aACF,MAAM,EAAE;aACR,QAAQ,CACL,4NAA4N,CAC/N;QACL,WAAW,EAAE,OAAC;aACT,KAAK,CAAC,CAAC,OAAC,CAAC,MAAM,EAAE,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;aACxC,QAAQ,CACL,sMAAsM,CACzM;QACL,aAAa,EAAE,OAAC;aACX,MAAM,EAAE;aACR,GAAG,EAAE;aACL,QAAQ,EAAE;aACV,QAAQ,EAAE;aACV,QAAQ,CAAC,4EAA4E,CAAC;QAC3F,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;QACzD,MAAM,EAAE,IAAA,gBAAM,GAAE;aACX,QAAQ,EAAE;aACV,QAAQ,CACL,0LAA0L,CAC7L;QACL,mBAAmB,EAAE,qDAA6B;aAC7C,QAAQ,EAAE;aACV,QAAQ,CACL,qMAAqM,CACxM;KACR,CAAC;SACD,WAAW,EAAE;CACrB,CAAC,CAAC;AAEU,QAAA,aAAa,GAAG;IACzB,QAAQ,EAAE,OAAC;SACN,KAAK,CAAC,OAAC,CAAC,KAAK,CAAC,CAAC,QAAQ,EAAE,iBAAiB,CAAC,CAAC,CAAC;SAC7C,QAAQ,CACL,uTAAuT,CAC1T;IACL,kBAAkB,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,qBAAM,CAAC,CAAC,QAAQ,CAAC;;;CAGtE,CAAC;CACD,CAAC;AAEF,MAAa,aAAc,SAAQ,gCAAe;IAAlD;;QACW,SAAI,GAAG,WAAW,CAAC;QAChB,gBAAW,GAAG,iDAAiD,CAAC;QAChE,cAAS,GAAG;YAClB,GAAG,gCAAe;YAClB,GAAG,qBAAa;SACnB,CAAC;QACK,kBAAa,GAAkB,MAAM,CAAC;IAqNjD,CAAC;IAnNa,KAAK,CAAC,OAAO,CACnB,EAAE,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE,kBAAkB,EAAmC,EACvF,EAAE,MAAM,EAAwB;QAEhC,IAAI,iBAAiB,GAAkC,SAAS,CAAC;QACjE,IAAI,CAAC;YACD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;YAC9C,MAAM,IAAI,CAAC,6BAA6B,CAAC,QAAQ,CAAC,CAAC;YAEnD,wDAAwD;YACxD,IAAI,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;gBACzB,MAAM,IAAA,+BAAe,EAAC,QAAQ,EAAE,QAAQ,EAAE,UAAU,EAAE,WAAW,EAAE,KAAK,IAAI,EAAE;oBAC1E,OAAO,QAAQ;yBACV,SAAS,CAAC,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,EAAE,EAAE,YAAY,EAAE,SAAS,EAAE,CAAC;yBAC1E,OAAO,CAAC,cAAc,CAAC,CAAC;gBACjC,CAAC,CAAC,CAAC;YACP,CAAC;YAED,QAAQ,GAAG,MAAM,IAAI,CAAC,yCAAyC,CAAC;gBAC5D,QAAQ;gBACR,UAAU;gBACV,QAAQ;aACX,CAAC,CAAC;YAEH,MAAM,qBAAqB,GAAG,CAAC,GAAG,QAAQ,CAAC,CAAC;YAC5C,IAAI,IAAI,CAAC,MAAM,CAAC,oBAAoB,GAAG,CAAC,EAAE,CAAC;gBACvC,qBAAqB,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC,CAAC;YAC7E,CAAC;YACD,iBAAiB,GAAG,QAAQ,CAAC,SAAS,CAAC,QAAQ,EAAE,UAAU,EAAE,qBAAqB,CAAC,CAAC;YAEpF,MAAM,CAAC,cAAc,EAAE,aAAa,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBACtD,IAAI,CAAC,+BAA+B,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC;gBAClF,IAAA,+DAA+B,EAAC;oBAC5B,MAAM,EAAE,iBAAiB;oBACzB,0BAA0B,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB;oBACxD,sBAAsB,EAAE,kBAAkB;oBAC1C,WAAW,EAAE,MAAM;iBACtB,CAAC;aACL,CAAC,CAAC;YAEH,iEAAiE;YACjE,mDAAmD;YACnD,mEAAmE;YACnE,UAAU;YACV,MAAM,2CAA2C,GAC7C,IAAI,CAAC,MAAM,CAAC,oBAAoB,GAAG,CAAC;gBACpC,CAAC,CAAC,cAAc;gBAChB,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC,oBAAoB,CAAC;YAEtD,OAAO;gBACH,OAAO,EAAE,IAAA,6BAAmB,EACxB,IAAI,CAAC,eAAe,CAAC;oBACjB,eAAe,EAAE,cAAc;oBAC/B,SAAS,EAAE,aAAa,CAAC,SAAS;oBAClC,aAAa,EAAE;wBACX,2CAA2C,CAAC,CAAC,CAAC,6BAA6B,CAAC,CAAC,CAAC,SAAS;wBACvF,aAAa,CAAC,QAAQ;qBACzB,CAAC,MAAM,CAAC,CAAC,KAAK,EAAmD,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC;iBAChF,CAAC,EACF,GAAG,CAAC,aAAa,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,YAAK,CAAC,SAAS,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAC5F;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,iBAAK,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,KAAK,CAAC,6BAA6B,CAAC,QAAmC;QAC3E,MAAM,eAAe,GAAoB,CAAC,QAAQ,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC;QACxE,MAAM,iBAAiB,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,iBAAiB,EAAE,CAAC;QAEjE,IAAI,wBAAwB,GAAG,EAAE,CAAC;QAElC,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;YACvB,wBAAwB,GAAG,wEAAwE,CAAC;QACxG,CAAC;aAAM,IAAI,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAkB,CAAC,CAAC,EAAE,CAAC;YAC7F,wBAAwB;gBACpB,qHAAqH,CAAC;QAC9H,CAAC;QAED,KAAK,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;YAC3B,wGAAwG;YACxG,sGAAsG;YACtG,2BAA2B;YAC3B,IAAI,CAAC,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,MAAM,CAAC,IAAI,wBAAwB,EAAE,CAAC;gBAC3D,MAAM,IAAI,wBAAY,CAAC,sBAAU,CAAC,uBAAuB,EAAE,wBAAwB,CAAC,CAAC;YACzF,CAAC;YAED,8FAA8F;YAC9F,yCAAyC;YACzC,IAAI,KAAK,CAAC,aAAa,IAAI,CAAC,iBAAiB,EAAE,CAAC;gBAC5C,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,GAMX;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,CAAC;iBACxD,SAAS,CAAC,wCAAyB,CAAC;iBACpC,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,yCAAyC,CAAC,EACpD,QAAQ,EACR,UAAU,EACV,QAAQ,GAKX;QACG,KAAK,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;YAC3B,IAAI,eAAe,IAAI,KAAK,EAAE,CAAC;gBAC3B,MAAM,EAAE,aAAa,EAAE,iBAAiB,EAAE,GAAG,KAA0C,CAAC;gBAExF,IAAI,KAAK,CAAC,OAAO,CAAC,iBAAiB,CAAC,WAAW,CAAC,EAAE,CAAC;oBAC/C,SAAS;gBACb,CAAC;gBAED,IAAI,CAAC,iBAAiB,CAAC,mBAAmB,EAAE,CAAC;oBACzC,MAAM,IAAI,wBAAY,CAClB,sBAAU,CAAC,6BAA6B,EACxC,6DAA6D,CAChE,CAAC;gBACN,CAAC;gBAED,MAAM,mBAAmB,GAAG,iBAAiB,CAAC,mBAAmB,CAAC;gBAClE,OAAO,iBAAiB,CAAC,mBAAmB,CAAC;gBAE7C,MAAM,CAAC,UAAU,CAAC,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,6BAA6B,CAAC,kBAAkB,CAAC;oBACrF,QAAQ;oBACR,UAAU;oBACV,IAAI,EAAE,iBAAiB,CAAC,IAAI;oBAC5B,SAAS,EAAE,CAAC,iBAAiB,CAAC,WAAW,CAAC;oBAC1C,mBAAmB;oBACnB,SAAS,EAAE,OAAO;iBACrB,CAAC,CAAC;gBAEH,2FAA2F;gBAC3F,iFAAiF;gBACjF,6GAA6G;gBAC7G,iBAAiB,CAAC,WAAW,GAAG,UAAsB,CAAC;YAC3D,CAAC;QACL,CAAC;QAED,OAAO,QAAQ,CAAC;IACpB,CAAC;IAEO,eAAe,CAAC,EACpB,eAAe,EACf,SAAS,EACT,aAAa,GAKhB;QACG,MAAM,gBAAgB,GAAG,aAAa,CAAC,MAAM;YACzC,CAAC,CAAC;yCAC2B,aAAa,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,wCAAyB,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;;CAEjH;YACW,CAAC,CAAC,EAAE,CAAC;QAET,OAAO;8BACe,eAAe,KAAK,SAAS,CAAC,CAAC,CAAC,0BAA0B,CAAC,CAAC,CAAC,eAAe;YAC9F,SAAS,CAAC,MAAM,aAAa,gBAAgB,CAAC,CAAC,CAAC,IAAI,gBAAgB,EAAE,CAAC,CAAC,CAAC,GAAG;CACvF,CAAC;IACE,CAAC;CACJ;AA5ND,sCA4NC"}

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

@@ -57,33 +57,215 @@ export declare class ExportTool extends MongoDBToolBase {
         }>, z.ZodObject<{
             name: z.ZodLiteral<"aggregate">;
             arguments: z.ZodObject<{
-                pipeline: z.ZodArray<z.AnyZodObject, "many">;
+                pipeline: z.ZodArray<z.ZodUnion<[z.AnyZodObject, z.ZodObject<{
+                    $vectorSearch: z.ZodObject<{
+                        exact: z.ZodDefault<z.ZodOptional<z.ZodBoolean>>;
+                        index: z.ZodString;
+                        path: z.ZodString;
+                        queryVector: z.ZodUnion<[z.ZodString, z.ZodArray<z.ZodNumber, "many">]>;
+                        numCandidates: z.ZodOptional<z.ZodNumber>;
+                        limit: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
+                        filter: z.ZodOptional<z.AnyZodObject>;
+                        embeddingParameters: z.ZodOptional<z.ZodObject<{
+                            outputDimension: z.ZodDefault<z.ZodOptional<z.ZodUnion<[z.ZodLiteral<256>, z.ZodLiteral<512>, z.ZodLiteral<1024>, z.ZodLiteral<2048>, z.ZodLiteral<4096>]>>>;
+                            outputDType: z.ZodDefault<z.ZodOptional<z.ZodEnum<["float", "int8", "uint8", "binary", "ubinary"]>>>;
+                        } & {
+                            model: z.ZodDefault<z.ZodEnum<["voyage-3-large", "voyage-3.5", "voyage-3.5-lite", "voyage-code-3"]>>;
+                        }, "strip", z.ZodTypeAny, {
+                            outputDimension: 256 | 512 | 1024 | 2048 | 4096;
+                            outputDType: "binary" | "float" | "int8" | "uint8" | "ubinary";
+                            model: "voyage-3-large" | "voyage-3.5" | "voyage-3.5-lite" | "voyage-code-3";
+                        }, {
+                            outputDimension?: 256 | 512 | 1024 | 2048 | 4096 | undefined;
+                            outputDType?: "binary" | "float" | "int8" | "uint8" | "ubinary" | undefined;
+                            model?: "voyage-3-large" | "voyage-3.5" | "voyage-3.5-lite" | "voyage-code-3" | undefined;
+                        }>>;
+                    }, "passthrough", z.ZodTypeAny, z.objectOutputType<{
+                        exact: z.ZodDefault<z.ZodOptional<z.ZodBoolean>>;
+                        index: z.ZodString;
+                        path: z.ZodString;
+                        queryVector: z.ZodUnion<[z.ZodString, z.ZodArray<z.ZodNumber, "many">]>;
+                        numCandidates: z.ZodOptional<z.ZodNumber>;
+                        limit: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
+                        filter: z.ZodOptional<z.AnyZodObject>;
+                        embeddingParameters: z.ZodOptional<z.ZodObject<{
+                            outputDimension: z.ZodDefault<z.ZodOptional<z.ZodUnion<[z.ZodLiteral<256>, z.ZodLiteral<512>, z.ZodLiteral<1024>, z.ZodLiteral<2048>, z.ZodLiteral<4096>]>>>;
+                            outputDType: z.ZodDefault<z.ZodOptional<z.ZodEnum<["float", "int8", "uint8", "binary", "ubinary"]>>>;
+                        } & {
+                            model: z.ZodDefault<z.ZodEnum<["voyage-3-large", "voyage-3.5", "voyage-3.5-lite", "voyage-code-3"]>>;
+                        }, "strip", z.ZodTypeAny, {
+                            outputDimension: 256 | 512 | 1024 | 2048 | 4096;
+                            outputDType: "binary" | "float" | "int8" | "uint8" | "ubinary";
+                            model: "voyage-3-large" | "voyage-3.5" | "voyage-3.5-lite" | "voyage-code-3";
+                        }, {
+                            outputDimension?: 256 | 512 | 1024 | 2048 | 4096 | undefined;
+                            outputDType?: "binary" | "float" | "int8" | "uint8" | "ubinary" | undefined;
+                            model?: "voyage-3-large" | "voyage-3.5" | "voyage-3.5-lite" | "voyage-code-3" | undefined;
+                        }>>;
+                    }, z.ZodTypeAny, "passthrough">, z.objectInputType<{
+                        exact: z.ZodDefault<z.ZodOptional<z.ZodBoolean>>;
+                        index: z.ZodString;
+                        path: z.ZodString;
+                        queryVector: z.ZodUnion<[z.ZodString, z.ZodArray<z.ZodNumber, "many">]>;
+                        numCandidates: z.ZodOptional<z.ZodNumber>;
+                        limit: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
+                        filter: z.ZodOptional<z.AnyZodObject>;
+                        embeddingParameters: z.ZodOptional<z.ZodObject<{
+                            outputDimension: z.ZodDefault<z.ZodOptional<z.ZodUnion<[z.ZodLiteral<256>, z.ZodLiteral<512>, z.ZodLiteral<1024>, z.ZodLiteral<2048>, z.ZodLiteral<4096>]>>>;
+                            outputDType: z.ZodDefault<z.ZodOptional<z.ZodEnum<["float", "int8", "uint8", "binary", "ubinary"]>>>;
+                        } & {
+                            model: z.ZodDefault<z.ZodEnum<["voyage-3-large", "voyage-3.5", "voyage-3.5-lite", "voyage-code-3"]>>;
+                        }, "strip", z.ZodTypeAny, {
+                            outputDimension: 256 | 512 | 1024 | 2048 | 4096;
+                            outputDType: "binary" | "float" | "int8" | "uint8" | "ubinary";
+                            model: "voyage-3-large" | "voyage-3.5" | "voyage-3.5-lite" | "voyage-code-3";
+                        }, {
+                            outputDimension?: 256 | 512 | 1024 | 2048 | 4096 | undefined;
+                            outputDType?: "binary" | "float" | "int8" | "uint8" | "ubinary" | undefined;
+                            model?: "voyage-3-large" | "voyage-3.5" | "voyage-3.5-lite" | "voyage-code-3" | undefined;
+                        }>>;
+                    }, z.ZodTypeAny, "passthrough">>;
+                }, "strip", z.ZodTypeAny, {
+                    $vectorSearch: {
+                        path: string;
+                        exact: boolean;
+                        index: string;
+                        limit: number;
+                        queryVector: string | number[];
+                        filter?: {
+                            [x: string]: any;
+                        } | undefined;
+                        embeddingParameters?: {
+                            outputDimension: 256 | 512 | 1024 | 2048 | 4096;
+                            outputDType: "binary" | "float" | "int8" | "uint8" | "ubinary";
+                            model: "voyage-3-large" | "voyage-3.5" | "voyage-3.5-lite" | "voyage-code-3";
+                        } | undefined;
+                        numCandidates?: number | undefined;
+                    } & {
+                        [k: string]: unknown;
+                    };
+                }, {
+                    $vectorSearch: {
+                        path: string;
+                        index: string;
+                        queryVector: string | number[];
+                        exact?: boolean | undefined;
+                        filter?: {
+                            [x: string]: any;
+                        } | undefined;
+                        embeddingParameters?: {
+                            outputDimension?: 256 | 512 | 1024 | 2048 | 4096 | undefined;
+                            outputDType?: "binary" | "float" | "int8" | "uint8" | "ubinary" | undefined;
+                            model?: "voyage-3-large" | "voyage-3.5" | "voyage-3.5-lite" | "voyage-code-3" | undefined;
+                        } | undefined;
+                        limit?: number | undefined;
+                        numCandidates?: number | undefined;
+                    } & {
+                        [k: string]: unknown;
+                    };
+                }>]>, "many">;
                 responseBytesLimit: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
             }, "strip", z.ZodTypeAny, {
                 responseBytesLimit: number;
-                pipeline: {
+                pipeline: ({
                     [x: string]: any;
-                }[];
+                } | {
+                    $vectorSearch: {
+                        path: string;
+                        exact: boolean;
+                        index: string;
+                        limit: number;
+                        queryVector: string | number[];
+                        filter?: {
+                            [x: string]: any;
+                        } | undefined;
+                        embeddingParameters?: {
+                            outputDimension: 256 | 512 | 1024 | 2048 | 4096;
+                            outputDType: "binary" | "float" | "int8" | "uint8" | "ubinary";
+                            model: "voyage-3-large" | "voyage-3.5" | "voyage-3.5-lite" | "voyage-code-3";
+                        } | undefined;
+                        numCandidates?: number | undefined;
+                    } & {
+                        [k: string]: unknown;
+                    };
+                })[];
             }, {
-                pipeline: {
+                pipeline: ({
                     [x: string]: any;
-                }[];
+                } | {
+                    $vectorSearch: {
+                        path: string;
+                        index: string;
+                        queryVector: string | number[];
+                        exact?: boolean | undefined;
+                        filter?: {
+                            [x: string]: any;
+                        } | undefined;
+                        embeddingParameters?: {
+                            outputDimension?: 256 | 512 | 1024 | 2048 | 4096 | undefined;
+                            outputDType?: "binary" | "float" | "int8" | "uint8" | "ubinary" | undefined;
+                            model?: "voyage-3-large" | "voyage-3.5" | "voyage-3.5-lite" | "voyage-code-3" | undefined;
+                        } | undefined;
+                        limit?: number | undefined;
+                        numCandidates?: number | undefined;
+                    } & {
+                        [k: string]: unknown;
+                    };
+                })[];
                 responseBytesLimit?: number | undefined;
             }>;
         }, "strip", z.ZodTypeAny, {
             name: "aggregate";
             arguments: {
                 responseBytesLimit: number;
-                pipeline: {
+                pipeline: ({
                     [x: string]: any;
-                }[];
+                } | {
+                    $vectorSearch: {
+                        path: string;
+                        exact: boolean;
+                        index: string;
+                        limit: number;
+                        queryVector: string | number[];
+                        filter?: {
+                            [x: string]: any;
+                        } | undefined;
+                        embeddingParameters?: {
+                            outputDimension: 256 | 512 | 1024 | 2048 | 4096;
+                            outputDType: "binary" | "float" | "int8" | "uint8" | "ubinary";
+                            model: "voyage-3-large" | "voyage-3.5" | "voyage-3.5-lite" | "voyage-code-3";
+                        } | undefined;
+                        numCandidates?: number | undefined;
+                    } & {
+                        [k: string]: unknown;
+                    };
+                })[];
             };
         }, {
             name: "aggregate";
             arguments: {
-                pipeline: {
+                pipeline: ({
                     [x: string]: any;
-                }[];
+                } | {
+                    $vectorSearch: {
+                        path: string;
+                        index: string;
+                        queryVector: string | number[];
+                        exact?: boolean | undefined;
+                        filter?: {
+                            [x: string]: any;
+                        } | undefined;
+                        embeddingParameters?: {
+                            outputDimension?: 256 | 512 | 1024 | 2048 | 4096 | undefined;
+                            outputDType?: "binary" | "float" | "int8" | "uint8" | "ubinary" | undefined;
+                            model?: "voyage-3-large" | "voyage-3.5" | "voyage-3.5-lite" | "voyage-code-3" | undefined;
+                        } | undefined;
+                        limit?: number | undefined;
+                        numCandidates?: number | undefined;
+                    } & {
+                        [k: string]: unknown;
+                    };
+                })[];
                 responseBytesLimit?: number | undefined;
             };
         }>]>, "many">;