mongodb-mcp-server 1.2.0 → 1.3.0-pre.2

package/dist/cjs/common/search/vectorSearchEmbeddingsManager.js

@@ -3,12 +3,12 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.VectorSearchEmbeddingsManager = exports.quantizationEnum = exports.similarityEnum = void 0;
+exports.VectorSearchEmbeddingsManager = exports.quantizationEnum = void 0;
 const bson_1 = require("bson");
 const zod_1 = __importDefault(require("zod"));
 const errors_js_1 = require("../errors.js");
 const embeddingsProvider_js_1 = require("./embeddingsProvider.js");
-exports.similarityEnum = zod_1.default.enum(["cosine", "euclidean", "dotProduct"]);
+const tool_js_1 = require("../../tools/tool.js");
 exports.quantizationEnum = zod_1.default.enum(["none", "scalar", "binary"]);
 class VectorSearchEmbeddingsManager {
     constructor(config, connectionManager, embeddings = new Map(), embeddingsProvider = embeddingsProvider_js_1.getEmbeddingsProvider) {
@@ -24,6 +24,14 @@ class VectorSearchEmbeddingsManager {
         const embeddingDefKey = `${database}.${collection}`;
         this.embeddings.delete(embeddingDefKey);
     }
+    async indexExists({ database, collection, indexName, }) {
+        const provider = await this.atlasSearchEnabledProvider();
+        if (!provider) {
+            return false;
+        }
+        const searchIndexesWithName = await provider.getSearchIndexes(database, collection, indexName);
+        return searchIndexesWithName.length >= 1;
+    }
     async embeddingsForNamespace({ database, collection, }) {
         const provider = await this.atlasSearchEnabledProvider();
         if (!provider) {
@@ -31,7 +39,7 @@ class VectorSearchEmbeddingsManager {
         }
         // We only need the embeddings for validation now, so don't query them if
         // validation is disabled.
-        if (this.config.disableEmbeddingsValidation) {
+        if (!this.config.embeddingsValidation) {
             return [];
         }
         const embeddingDefKey = `${database}.${collection}`;
@@ -48,6 +56,17 @@ class VectorSearchEmbeddingsManager {
         }
         return definition;
     }
+    async assertFieldsHaveCorrectEmbeddings({ database, collection }, documents) {
+        const embeddingValidationResults = (await Promise.all(documents.map((document) => this.findFieldsWithWrongEmbeddings({ database, collection }, document)))).flat();
+        if (embeddingValidationResults.length > 0) {
+            const embeddingValidationMessages = embeddingValidationResults.map((validation) => `- Field ${validation.path} is an embedding with ${validation.expectedNumDimensions} dimensions,` +
+                ` and the provided value is not compatible. Actual dimensions: ${validation.actualNumDimensions},` +
+                ` Error: ${validation.error}`);
+            throw new errors_js_1.MongoDBError(errors_js_1.ErrorCodes.AtlasVectorSearchInvalidQuery, (0, tool_js_1.formatUntrustedData)("", ...embeddingValidationMessages)
+                .map(({ text }) => text)
+                .join("\n"));
+        }
+    }
     async findFieldsWithWrongEmbeddings({ database, collection, }, document) {
         const provider = await this.atlasSearchEnabledProvider();
         if (!provider) {
@@ -56,7 +75,7 @@ class VectorSearchEmbeddingsManager {
         // While we can do our best effort to ensure that the embedding validation is correct
         // based on https://www.mongodb.com/docs/atlas/atlas-vector-search/vector-quantization/
         // it's a complex process so we will also give the user the ability to disable this validation
-        if (this.config.disableEmbeddingsValidation) {
+        if (!this.config.embeddingsValidation) {
             return [];
         }
         const embeddings = await this.embeddingsForNamespace({ database, collection });
@@ -80,11 +99,29 @@ class VectorSearchEmbeddingsManager {
         const constructError = (details) => ({
             path: definition.path,
             expectedNumDimensions: definition.numDimensions,
-            expectedQuantization: definition.quantization,
             actualNumDimensions: details.actualNumDimensions ?? "unknown",
-            actualQuantization: details.actualQuantization ?? "unknown",
             error: details.error ?? "not-a-vector",
         });
+        const extractUnderlyingVector = (fieldRef) => {
+            if (fieldRef instanceof bson_1.BSON.Binary) {
+                try {
+                    return fieldRef.toFloat32Array();
+                }
+                catch {
+                    // nothing to do here
+                }
+                try {
+                    return fieldRef.toBits();
+                }
+                catch {
+                    // nothing to do here
+                }
+            }
+            if (Array.isArray(fieldRef)) {
+                return fieldRef;
+            }
+            return undefined;
+        };
         for (const field of fieldPath) {
             if (fieldRef && typeof fieldRef === "object" && field in fieldRef) {
                 fieldRef = fieldRef[field];
@@ -93,72 +130,34 @@ class VectorSearchEmbeddingsManager {
                 return undefined;
             }
         }
-        switch (definition.quantization) {
-            // Because quantization is not defined by the user
-            // we have to trust them in the format they use.
-            case "none":
-                return undefined;
-            case "scalar":
-            case "binary":
-                if (fieldRef instanceof bson_1.BSON.Binary) {
-                    try {
-                        const elements = fieldRef.toFloat32Array();
-                        if (elements.length !== definition.numDimensions) {
-                            return constructError({
-                                actualNumDimensions: elements.length,
-                                actualQuantization: "binary",
-                                error: "dimension-mismatch",
-                            });
-                        }
-                        return undefined;
-                    }
-                    catch {
-                        // bits are also supported
-                        try {
-                            const bits = fieldRef.toBits();
-                            if (bits.length !== definition.numDimensions) {
-                                return constructError({
-                                    actualNumDimensions: bits.length,
-                                    actualQuantization: "binary",
-                                    error: "dimension-mismatch",
-                                });
-                            }
-                            return undefined;
-                        }
-                        catch {
-                            return constructError({
-                                actualQuantization: "binary",
-                                error: "not-a-vector",
-                            });
-                        }
-                    }
-                }
-                else {
-                    if (!Array.isArray(fieldRef)) {
-                        return constructError({
-                            error: "not-a-vector",
-                        });
-                    }
-                    if (fieldRef.length !== definition.numDimensions) {
-                        return constructError({
-                            actualNumDimensions: fieldRef.length,
-                            actualQuantization: "scalar",
-                            error: "dimension-mismatch",
-                        });
-                    }
-                    if (!fieldRef.every((e) => this.isANumber(e))) {
-                        return constructError({
-                            actualNumDimensions: fieldRef.length,
-                            actualQuantization: "scalar",
-                            error: "not-numeric",
-                        });
-                    }
-                }
-                break;
+        const maybeVector = extractUnderlyingVector(fieldRef);
+        if (!maybeVector) {
+            return constructError({
+                error: "not-a-vector",
+            });
+        }
+        if (maybeVector.length !== definition.numDimensions) {
+            return constructError({
+                actualNumDimensions: maybeVector.length,
+                error: "dimension-mismatch",
+            });
+        }
+        if (Array.isArray(maybeVector) && maybeVector.some((e) => !this.isANumber(e))) {
+            return constructError({
+                actualNumDimensions: maybeVector.length,
+                error: "not-numeric",
+            });
         }
         return undefined;
     }
-    async generateEmbeddings({ database, collection, path, rawValues, embeddingParameters, inputType, }) {
+    async assertVectorSearchIndexExists({ database, collection, path, }) {
+        const embeddingInfoForCollection = await this.embeddingsForNamespace({ database, collection });
+        const embeddingInfoForPath = embeddingInfoForCollection.find((definition) => definition.path === path);
+        if (!embeddingInfoForPath) {
+            throw new errors_js_1.MongoDBError(errors_js_1.ErrorCodes.AtlasVectorSearchIndexNotFound, `No Vector Search index found for path "${path}" in namespace "${database}.${collection}"`);
+        }
+    }
+    async generateEmbeddings({ rawValues, embeddingParameters, inputType, }) {
         const provider = await this.atlasSearchEnabledProvider();
         if (!provider) {
             throw new errors_js_1.MongoDBError(errors_js_1.ErrorCodes.AtlasSearchNotSupported, "Atlas Search is not supported in this cluster.");
@@ -167,17 +166,6 @@ class VectorSearchEmbeddingsManager {
         if (!embeddingsProvider) {
             throw new errors_js_1.MongoDBError(errors_js_1.ErrorCodes.NoEmbeddingsProviderConfigured, "No embeddings provider configured.");
         }
-        if (this.config.disableEmbeddingsValidation) {
-            return await embeddingsProvider.embed(embeddingParameters.model, rawValues, {
-                inputType,
-                ...embeddingParameters,
-            });
-        }
-        const embeddingInfoForCollection = await this.embeddingsForNamespace({ database, collection });
-        const embeddingInfoForPath = embeddingInfoForCollection.find((definition) => definition.path === path);
-        if (!embeddingInfoForPath) {
-            throw new errors_js_1.MongoDBError(errors_js_1.ErrorCodes.AtlasVectorSearchIndexNotFound, `No Vector Search index found for path "${path}" in namespace "${database}.${collection}"`);
-        }
         return await embeddingsProvider.embed(embeddingParameters.model, rawValues, {
             inputType,
             ...embeddingParameters,

package/dist/cjs/common/search/vectorSearchEmbeddingsManager.js.map

@@ -1 +1 @@
-{"version":3,"file":"vectorSearchEmbeddingsManager.js","sourceRoot":"","sources":["../../../../src/common/search/vectorSearchEmbeddingsManager.ts"],"names":[],"mappings":";;;;;;AACA,+BAA2C;AAG3C,8CAAoB;AACpB,4CAAwD;AACxD,mEAAgE;AAGnD,QAAA,cAAc,GAAG,aAAC,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC,CAAC;AAG/D,QAAA,gBAAgB,GAAG,aAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC;AAqBrE,MAAa,6BAA6B;IACtC,YACqB,MAAkB,EAClB,iBAAoC,EACpC,aAAoE,IAAI,GAAG,EAAE,EAC7E,qBAAmD,6CAAqB;QAHxE,WAAM,GAAN,MAAM,CAAY;QAClB,sBAAiB,GAAjB,iBAAiB,CAAmB;QACpC,eAAU,GAAV,UAAU,CAAmE;QAC7E,uBAAkB,GAAlB,kBAAkB,CAAsD;QAEzF,iBAAiB,CAAC,MAAM,CAAC,EAAE,CAAC,kBAAkB,EAAE,GAAG,EAAE;YACjD,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;QAC5B,CAAC,CAAC,CAAC;IACP,CAAC;IAED,6BAA6B,CAAC,EAAE,QAAQ,EAAE,UAAU,EAA4C;QAC5F,MAAM,eAAe,GAAuB,GAAG,QAAQ,IAAI,UAAU,EAAE,CAAC;QACxE,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,sBAAsB,CAAC,EACzB,QAAQ,EACR,UAAU,GAIb;QACG,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACzD,IAAI,CAAC,QAAQ,EAAE,CAAC;YACZ,OAAO,EAAE,CAAC;QACd,CAAC;QAED,yEAAyE;QACzE,0BAA0B;QAC1B,IAAI,IAAI,CAAC,MAAM,CAAC,2BAA2B,EAAE,CAAC;YAC1C,OAAO,EAAE,CAAC;QACd,CAAC;QAED,MAAM,eAAe,GAAuB,GAAG,QAAQ,IAAI,UAAU,EAAE,CAAC;QACxE,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;QAExD,IAAI,CAAC,UAAU,EAAE,CAAC;YACd,MAAM,gBAAgB,GAAG,MAAM,QAAQ,CAAC,gBAAgB,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;YAC/E,MAAM,mBAAmB,GAAG,gBAAgB,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,cAAc,CAAC,CAAC;YAC9F,MAAM,YAAY,GAAG,mBAAmB;gBACpC,sEAAsE;iBACrE,OAAO,CAAW,CAAC,KAAK,EAAE,EAAE,CAAE,KAAK,CAAC,gBAAgB,EAAE,MAAqB,IAAI,EAAE,CAAC;iBAClF,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,4BAA4B,CAAC,KAAK,CAAC,CAAC,CAAC;YAEjE,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,eAAe,EAAE,YAAY,CAAC,CAAC;YACnD,OAAO,YAAY,CAAC;QACxB,CAAC;QAED,OAAO,UAAU,CAAC;IACtB,CAAC;IAED,KAAK,CAAC,6BAA6B,CAC/B,EACI,QAAQ,EACR,UAAU,GAIb,EACD,QAAkB;QAElB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACzD,IAAI,CAAC,QAAQ,EAAE,CAAC;YACZ,OAAO,EAAE,CAAC;QACd,CAAC;QAED,qFAAqF;QACrF,uFAAuF;QACvF,8FAA8F;QAC9F,IAAI,IAAI,CAAC,MAAM,CAAC,2BAA2B,EAAE,CAAC;YAC1C,OAAO,EAAE,CAAC;QACd,CAAC;QAED,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,sBAAsB,CAAC,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC,CAAC;QAC/E,OAAO,UAAU;aACZ,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,6BAA6B,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;aAC/D,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC;IACxC,CAAC;IAEO,KAAK,CAAC,0BAA0B;QACpC,MAAM,eAAe,GAAG,IAAI,CAAC,iBAAiB,CAAC,sBAAsB,CAAC;QACtE,IAAI,eAAe,CAAC,GAAG,KAAK,WAAW,IAAI,CAAC,MAAM,eAAe,CAAC,iBAAiB,EAAE,CAAC,EAAE,CAAC;YACrF,OAAO,eAAe,CAAC,eAAe,CAAC;QAC3C,CAAC;QAED,OAAO,IAAI,CAAC;IAChB,CAAC;IAEO,4BAA4B,CAAC,GAAa;QAC9C,OAAO,GAAG,CAAC,MAAM,CAAC,KAAK,QAAQ,CAAC;IACpC,CAAC;IAEO,6BAA6B,CACjC,UAAsC,EACtC,QAAkB;QAElB,MAAM,SAAS,GAAG,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC7C,IAAI,QAAQ,GAAY,QAAQ,CAAC;QAEjC,MAAM,cAAc,GAAG,CACnB,OAA0G,EAChF,EAAE,CAAC,CAAC;YAC9B,IAAI,EAAE,UAAU,CAAC,IAAI;YACrB,qBAAqB,EAAE,UAAU,CAAC,aAAa;YAC/C,oBAAoB,EAAE,UAAU,CAAC,YAAY;YAC7C,mBAAmB,EAAE,OAAO,CAAC,mBAAmB,IAAI,SAAS;YAC7D,kBAAkB,EAAE,OAAO,CAAC,kBAAkB,IAAI,SAAS;YAC3D,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,cAAc;SACzC,CAAC,CAAC;QAEH,KAAK,MAAM,KAAK,IAAI,SAAS,EAAE,CAAC;YAC5B,IAAI,QAAQ,IAAI,OAAO,QAAQ,KAAK,QAAQ,IAAI,KAAK,IAAI,QAAQ,EAAE,CAAC;gBAChE,QAAQ,GAAI,QAAoC,CAAC,KAAK,CAAC,CAAC;YAC5D,CAAC;iBAAM,CAAC;gBACJ,OAAO,SAAS,CAAC;YACrB,CAAC;QACL,CAAC;QAED,QAAQ,UAAU,CAAC,YAAY,EAAE,CAAC;YAC9B,kDAAkD;YAClD,gDAAgD;YAChD,KAAK,MAAM;gBACP,OAAO,SAAS,CAAC;YACrB,KAAK,QAAQ,CAAC;YACd,KAAK,QAAQ;gBACT,IAAI,QAAQ,YAAY,WAAI,CAAC,MAAM,EAAE,CAAC;oBAClC,IAAI,CAAC;wBACD,MAAM,QAAQ,GAAG,QAAQ,CAAC,cAAc,EAAE,CAAC;wBAC3C,IAAI,QAAQ,CAAC,MAAM,KAAK,UAAU,CAAC,aAAa,EAAE,CAAC;4BAC/C,OAAO,cAAc,CAAC;gCAClB,mBAAmB,EAAE,QAAQ,CAAC,MAAM;gCACpC,kBAAkB,EAAE,QAAQ;gCAC5B,KAAK,EAAE,oBAAoB;6BAC9B,CAAC,CAAC;wBACP,CAAC;wBAED,OAAO,SAAS,CAAC;oBACrB,CAAC;oBAAC,MAAM,CAAC;wBACL,0BAA0B;wBAC1B,IAAI,CAAC;4BACD,MAAM,IAAI,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC;4BAC/B,IAAI,IAAI,CAAC,MAAM,KAAK,UAAU,CAAC,aAAa,EAAE,CAAC;gCAC3C,OAAO,cAAc,CAAC;oCAClB,mBAAmB,EAAE,IAAI,CAAC,MAAM;oCAChC,kBAAkB,EAAE,QAAQ;oCAC5B,KAAK,EAAE,oBAAoB;iCAC9B,CAAC,CAAC;4BACP,CAAC;4BAED,OAAO,SAAS,CAAC;wBACrB,CAAC;wBAAC,MAAM,CAAC;4BACL,OAAO,cAAc,CAAC;gCAClB,kBAAkB,EAAE,QAAQ;gCAC5B,KAAK,EAAE,cAAc;6BACxB,CAAC,CAAC;wBACP,CAAC;oBACL,CAAC;gBACL,CAAC;qBAAM,CAAC;oBACJ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;wBAC3B,OAAO,cAAc,CAAC;4BAClB,KAAK,EAAE,cAAc;yBACxB,CAAC,CAAC;oBACP,CAAC;oBAED,IAAI,QAAQ,CAAC,MAAM,KAAK,UAAU,CAAC,aAAa,EAAE,CAAC;wBAC/C,OAAO,cAAc,CAAC;4BAClB,mBAAmB,EAAE,QAAQ,CAAC,MAAM;4BACpC,kBAAkB,EAAE,QAAQ;4BAC5B,KAAK,EAAE,oBAAoB;yBAC9B,CAAC,CAAC;oBACP,CAAC;oBAED,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;wBAC5C,OAAO,cAAc,CAAC;4BAClB,mBAAmB,EAAE,QAAQ,CAAC,MAAM;4BACpC,kBAAkB,EAAE,QAAQ;4BAC5B,KAAK,EAAE,aAAa;yBACvB,CAAC,CAAC;oBACP,CAAC;gBACL,CAAC;gBAED,MAAM;QACd,CAAC;QAED,OAAO,SAAS,CAAC;IACrB,CAAC;IAEM,KAAK,CAAC,kBAAkB,CAAC,EAC5B,QAAQ,EACR,UAAU,EACV,IAAI,EACJ,SAAS,EACT,mBAAmB,EACnB,SAAS,GAQZ;QACG,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACzD,IAAI,CAAC,QAAQ,EAAE,CAAC;YACZ,MAAM,IAAI,wBAAY,CAClB,sBAAU,CAAC,uBAAuB,EAClC,gDAAgD,CACnD,CAAC;QACN,CAAC;QAED,MAAM,kBAAkB,GAAG,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAEhE,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACtB,MAAM,IAAI,wBAAY,CAAC,sBAAU,CAAC,8BAA8B,EAAE,oCAAoC,CAAC,CAAC;QAC5G,CAAC;QAED,IAAI,IAAI,CAAC,MAAM,CAAC,2BAA2B,EAAE,CAAC;YAC1C,OAAO,MAAM,kBAAkB,CAAC,KAAK,CAAC,mBAAmB,CAAC,KAAK,EAAE,SAAS,EAAE;gBACxE,SAAS;gBACT,GAAG,mBAAmB;aACzB,CAAC,CAAC;QACP,CAAC;QAED,MAAM,0BAA0B,GAAG,MAAM,IAAI,CAAC,sBAAsB,CAAC,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC,CAAC;QAC/F,MAAM,oBAAoB,GAAG,0BAA0B,CAAC,IAAI,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;QACvG,IAAI,CAAC,oBAAoB,EAAE,CAAC;YACxB,MAAM,IAAI,wBAAY,CAClB,sBAAU,CAAC,8BAA8B,EACzC,0CAA0C,IAAI,mBAAmB,QAAQ,IAAI,UAAU,GAAG,CAC7F,CAAC;QACN,CAAC;QAED,OAAO,MAAM,kBAAkB,CAAC,KAAK,CAAC,mBAAmB,CAAC,KAAK,EAAE,SAAS,EAAE;YACxE,SAAS;YACT,GAAG,mBAAmB;SACzB,CAAC,CAAC;IACP,CAAC;IAEO,SAAS,CAAC,KAAc;QAC5B,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC5B,OAAO,IAAI,CAAC;QAChB,CAAC;QAED,IACI,KAAK,YAAY,WAAI,CAAC,KAAK;YAC3B,KAAK,YAAY,WAAI,CAAC,UAAU;YAChC,KAAK,YAAY,WAAI,CAAC,MAAM;YAC5B,KAAK,YAAY,WAAI,CAAC,IAAI,EAC5B,CAAC;YACC,OAAO,IAAI,CAAC;QAChB,CAAC;QAED,OAAO,KAAK,CAAC;IACjB,CAAC;CACJ;AAhQD,sEAgQC"}
+{"version":3,"file":"vectorSearchEmbeddingsManager.js","sourceRoot":"","sources":["../../../../src/common/search/vectorSearchEmbeddingsManager.ts"],"names":[],"mappings":";;;;;;AACA,+BAA2C;AAG3C,8CAAoB;AACpB,4CAAwD;AACxD,mEAAgE;AAEhE,iDAA0D;AAI7C,QAAA,gBAAgB,GAAG,aAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC;AAmBrE,MAAa,6BAA6B;IACtC,YACqB,MAAkB,EAClB,iBAAoC,EACpC,aAAoE,IAAI,GAAG,EAAE,EAC7E,qBAAmD,6CAAqB;QAHxE,WAAM,GAAN,MAAM,CAAY;QAClB,sBAAiB,GAAjB,iBAAiB,CAAmB;QACpC,eAAU,GAAV,UAAU,CAAmE;QAC7E,uBAAkB,GAAlB,kBAAkB,CAAsD;QAEzF,iBAAiB,CAAC,MAAM,CAAC,EAAE,CAAC,kBAAkB,EAAE,GAAG,EAAE;YACjD,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;QAC5B,CAAC,CAAC,CAAC;IACP,CAAC;IAED,6BAA6B,CAAC,EAAE,QAAQ,EAAE,UAAU,EAA4C;QAC5F,MAAM,eAAe,GAAuB,GAAG,QAAQ,IAAI,UAAU,EAAE,CAAC;QACxE,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,EACd,QAAQ,EACR,UAAU,EACV,SAAS,GAKZ;QACG,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACzD,IAAI,CAAC,QAAQ,EAAE,CAAC;YACZ,OAAO,KAAK,CAAC;QACjB,CAAC;QAED,MAAM,qBAAqB,GAAG,MAAM,QAAQ,CAAC,gBAAgB,CAAC,QAAQ,EAAE,UAAU,EAAE,SAAS,CAAC,CAAC;QAE/F,OAAO,qBAAqB,CAAC,MAAM,IAAI,CAAC,CAAC;IAC7C,CAAC;IAED,KAAK,CAAC,sBAAsB,CAAC,EACzB,QAAQ,EACR,UAAU,GAIb;QACG,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACzD,IAAI,CAAC,QAAQ,EAAE,CAAC;YACZ,OAAO,EAAE,CAAC;QACd,CAAC;QAED,yEAAyE;QACzE,0BAA0B;QAC1B,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC;YACpC,OAAO,EAAE,CAAC;QACd,CAAC;QAED,MAAM,eAAe,GAAuB,GAAG,QAAQ,IAAI,UAAU,EAAE,CAAC;QACxE,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;QAExD,IAAI,CAAC,UAAU,EAAE,CAAC;YACd,MAAM,gBAAgB,GAAG,MAAM,QAAQ,CAAC,gBAAgB,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;YAC/E,MAAM,mBAAmB,GAAG,gBAAgB,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,cAAc,CAAC,CAAC;YAC9F,MAAM,YAAY,GAAG,mBAAmB;gBACpC,sEAAsE;iBACrE,OAAO,CAAW,CAAC,KAAK,EAAE,EAAE,CAAE,KAAK,CAAC,gBAAgB,EAAE,MAAqB,IAAI,EAAE,CAAC;iBAClF,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,4BAA4B,CAAC,KAAK,CAAC,CAAC,CAAC;YAEjE,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,eAAe,EAAE,YAAY,CAAC,CAAC;YACnD,OAAO,YAAY,CAAC;QACxB,CAAC;QAED,OAAO,UAAU,CAAC;IACtB,CAAC;IAED,KAAK,CAAC,iCAAiC,CACnC,EAAE,QAAQ,EAAE,UAAU,EAA4C,EAClE,SAAqB;QAErB,MAAM,0BAA0B,GAAG,CAC/B,MAAM,OAAO,CAAC,GAAG,CACb,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,IAAI,CAAC,6BAA6B,CAAC,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,QAAQ,CAAC,CAAC,CACtG,CACJ,CAAC,IAAI,EAAE,CAAC;QAET,IAAI,0BAA0B,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxC,MAAM,2BAA2B,GAAG,0BAA0B,CAAC,GAAG,CAC9D,CAAC,UAAU,EAAE,EAAE,CACX,WAAW,UAAU,CAAC,IAAI,yBAAyB,UAAU,CAAC,qBAAqB,cAAc;gBACjG,iEAAiE,UAAU,CAAC,mBAAmB,GAAG;gBAClG,WAAW,UAAU,CAAC,KAAK,EAAE,CACpC,CAAC;YAEF,MAAM,IAAI,wBAAY,CAClB,sBAAU,CAAC,6BAA6B,EACxC,IAAA,6BAAmB,EAAC,EAAE,EAAE,GAAG,2BAA2B,CAAC;iBAClD,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC;iBACvB,IAAI,CAAC,IAAI,CAAC,CAClB,CAAC;QACN,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,6BAA6B,CACtC,EACI,QAAQ,EACR,UAAU,GAIb,EACD,QAAkB;QAElB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACzD,IAAI,CAAC,QAAQ,EAAE,CAAC;YACZ,OAAO,EAAE,CAAC;QACd,CAAC;QAED,qFAAqF;QACrF,uFAAuF;QACvF,8FAA8F;QAC9F,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC;YACpC,OAAO,EAAE,CAAC;QACd,CAAC;QAED,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,sBAAsB,CAAC,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC,CAAC;QAC/E,OAAO,UAAU;aACZ,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,6BAA6B,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;aAC/D,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC;IACxC,CAAC;IAEO,KAAK,CAAC,0BAA0B;QACpC,MAAM,eAAe,GAAG,IAAI,CAAC,iBAAiB,CAAC,sBAAsB,CAAC;QACtE,IAAI,eAAe,CAAC,GAAG,KAAK,WAAW,IAAI,CAAC,MAAM,eAAe,CAAC,iBAAiB,EAAE,CAAC,EAAE,CAAC;YACrF,OAAO,eAAe,CAAC,eAAe,CAAC;QAC3C,CAAC;QAED,OAAO,IAAI,CAAC;IAChB,CAAC;IAEO,4BAA4B,CAAC,GAAa;QAC9C,OAAO,GAAG,CAAC,MAAM,CAAC,KAAK,QAAQ,CAAC;IACpC,CAAC;IAEO,6BAA6B,CACjC,UAAsC,EACtC,QAAkB;QAElB,MAAM,SAAS,GAAG,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC7C,IAAI,QAAQ,GAAY,QAAQ,CAAC;QAEjC,MAAM,cAAc,GAAG,CACnB,OAAmF,EACzD,EAAE,CAAC,CAAC;YAC9B,IAAI,EAAE,UAAU,CAAC,IAAI;YACrB,qBAAqB,EAAE,UAAU,CAAC,aAAa;YAC/C,mBAAmB,EAAE,OAAO,CAAC,mBAAmB,IAAI,SAAS;YAC7D,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,cAAc;SACzC,CAAC,CAAC;QAEH,MAAM,uBAAuB,GAAG,CAAC,QAAiB,EAAkC,EAAE;YAClF,IAAI,QAAQ,YAAY,WAAI,CAAC,MAAM,EAAE,CAAC;gBAClC,IAAI,CAAC;oBACD,OAAO,QAAQ,CAAC,cAAc,EAAE,CAAC;gBACrC,CAAC;gBAAC,MAAM,CAAC;oBACL,qBAAqB;gBACzB,CAAC;gBAED,IAAI,CAAC;oBACD,OAAO,QAAQ,CAAC,MAAM,EAAE,CAAC;gBAC7B,CAAC;gBAAC,MAAM,CAAC;oBACL,qBAAqB;gBACzB,CAAC;YACL,CAAC;YAED,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC1B,OAAO,QAA0B,CAAC;YACtC,CAAC;YAED,OAAO,SAAS,CAAC;QACrB,CAAC,CAAC;QAEF,KAAK,MAAM,KAAK,IAAI,SAAS,EAAE,CAAC;YAC5B,IAAI,QAAQ,IAAI,OAAO,QAAQ,KAAK,QAAQ,IAAI,KAAK,IAAI,QAAQ,EAAE,CAAC;gBAChE,QAAQ,GAAI,QAAoC,CAAC,KAAK,CAAC,CAAC;YAC5D,CAAC;iBAAM,CAAC;gBACJ,OAAO,SAAS,CAAC;YACrB,CAAC;QACL,CAAC;QAED,MAAM,WAAW,GAAG,uBAAuB,CAAC,QAAQ,CAAC,CAAC;QACtD,IAAI,CAAC,WAAW,EAAE,CAAC;YACf,OAAO,cAAc,CAAC;gBAClB,KAAK,EAAE,cAAc;aACxB,CAAC,CAAC;QACP,CAAC;QAED,IAAI,WAAW,CAAC,MAAM,KAAK,UAAU,CAAC,aAAa,EAAE,CAAC;YAClD,OAAO,cAAc,CAAC;gBAClB,mBAAmB,EAAE,WAAW,CAAC,MAAM;gBACvC,KAAK,EAAE,oBAAoB;aAC9B,CAAC,CAAC;QACP,CAAC;QAED,IAAI,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,IAAI,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YAC5E,OAAO,cAAc,CAAC;gBAClB,mBAAmB,EAAE,WAAW,CAAC,MAAM;gBACvC,KAAK,EAAE,aAAa;aACvB,CAAC,CAAC;QACP,CAAC;QAED,OAAO,SAAS,CAAC;IACrB,CAAC;IAEM,KAAK,CAAC,6BAA6B,CAAC,EACvC,QAAQ,EACR,UAAU,EACV,IAAI,GAKP;QACG,MAAM,0BAA0B,GAAG,MAAM,IAAI,CAAC,sBAAsB,CAAC,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC,CAAC;QAC/F,MAAM,oBAAoB,GAAG,0BAA0B,CAAC,IAAI,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;QACvG,IAAI,CAAC,oBAAoB,EAAE,CAAC;YACxB,MAAM,IAAI,wBAAY,CAClB,sBAAU,CAAC,8BAA8B,EACzC,0CAA0C,IAAI,mBAAmB,QAAQ,IAAI,UAAU,GAAG,CAC7F,CAAC;QACN,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,kBAAkB,CAAC,EAC5B,SAAS,EACT,mBAAmB,EACnB,SAAS,GAKZ;QACG,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACzD,IAAI,CAAC,QAAQ,EAAE,CAAC;YACZ,MAAM,IAAI,wBAAY,CAClB,sBAAU,CAAC,uBAAuB,EAClC,gDAAgD,CACnD,CAAC;QACN,CAAC;QAED,MAAM,kBAAkB,GAAG,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAEhE,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACtB,MAAM,IAAI,wBAAY,CAAC,sBAAU,CAAC,8BAA8B,EAAE,oCAAoC,CAAC,CAAC;QAC5G,CAAC;QAED,OAAO,MAAM,kBAAkB,CAAC,KAAK,CAAC,mBAAmB,CAAC,KAAK,EAAE,SAAS,EAAE;YACxE,SAAS;YACT,GAAG,mBAAmB;SACzB,CAAC,CAAC;IACP,CAAC;IAEO,SAAS,CAAC,KAAc;QAC5B,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC5B,OAAO,IAAI,CAAC;QAChB,CAAC;QAED,IACI,KAAK,YAAY,WAAI,CAAC,KAAK;YAC3B,KAAK,YAAY,WAAI,CAAC,UAAU;YAChC,KAAK,YAAY,WAAI,CAAC,MAAM;YAC5B,KAAK,YAAY,WAAI,CAAC,IAAI,EAC5B,CAAC;YACC,OAAO,IAAI,CAAC;QAChB,CAAC;QAED,OAAO,KAAK,CAAC;IACjB,CAAC;CACJ;AAlRD,sEAkRC"}

package/dist/cjs/common/session.d.ts

@@ -2,16 +2,15 @@ import { ApiClient } from "./atlas/apiClient.js";
 import type { Implementation } from "@modelcontextprotocol/sdk/types.js";
 import type { CompositeLogger } from "./logger.js";
 import EventEmitter from "events";
-import type { AtlasClusterConnectionInfo, ConnectionManager, ConnectionSettings, ConnectionStateErrored } from "./connectionManager.js";
+import type { AtlasClusterConnectionInfo, ConnectionManager, ConnectionSettings, ConnectionStateErrored, ConnectionStringAuthType } from "./connectionManager.js";
 import type { NodeDriverServiceProvider } from "@mongosh/service-provider-node-driver";
 import type { ExportsManager } from "./exportsManager.js";
 import type { Client } from "@mongodb-js/atlas-local";
 import type { Keychain } from "./keychain.js";
 import type { VectorSearchEmbeddingsManager } from "./search/vectorSearchEmbeddingsManager.js";
+import { type UserConfig } from "../common/config/userConfig.js";
 export interface SessionOptions {
-    apiBaseUrl: string;
-    apiClientId?: string;
-    apiClientSecret?: string;
+    userConfig: UserConfig;
     logger: CompositeLogger;
     exportsManager: ExportsManager;
     connectionManager: ConnectionManager;
@@ -26,6 +25,7 @@ export type SessionEvents = {
     "connection-error": [ConnectionStateErrored];
 };
 export declare class Session extends EventEmitter<SessionEvents> {
+    private readonly userConfig;
     readonly sessionId: string;
     readonly exportsManager: ExportsManager;
     readonly connectionManager: ConnectionManager;
@@ -39,15 +39,17 @@ export declare class Session extends EventEmitter<SessionEvents> {
         title?: string;
     };
     logger: CompositeLogger;
-    constructor({ apiBaseUrl, apiClientId, apiClientSecret, logger, connectionManager, exportsManager, keychain, atlasLocalClient, vectorSearchEmbeddingsManager, }: SessionOptions);
+    constructor({ userConfig, logger, connectionManager, exportsManager, keychain, atlasLocalClient, vectorSearchEmbeddingsManager, }: SessionOptions);
     setMcpClient(mcpClient: Implementation | undefined): void;
     disconnect(): Promise<void>;
     close(): Promise<void>;
+    connectToConfiguredConnection(): Promise<void>;
     connectToMongoDB(settings: ConnectionSettings): Promise<void>;
     get isConnectedToMongoDB(): boolean;
     isSearchSupported(): Promise<boolean>;
     assertSearchSupported(): Promise<void>;
     get serviceProvider(): NodeDriverServiceProvider;
     get connectedAtlasCluster(): AtlasClusterConnectionInfo | undefined;
+    get connectionStringAuthType(): ConnectionStringAuthType | undefined;
 }
 //# sourceMappingURL=session.d.ts.map

package/dist/cjs/common/session.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"session.d.ts","sourceRoot":"","sources":["../../../src/common/session.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACjD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,oCAAoC,CAAC;AACzE,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAEnD,OAAO,YAAY,MAAM,QAAQ,CAAC;AAClC,OAAO,KAAK,EACR,0BAA0B,EAC1B,iBAAiB,EACjB,kBAAkB,EAElB,sBAAsB,EACzB,MAAM,wBAAwB,CAAC;AAChC,OAAO,KAAK,EAAE,yBAAyB,EAAE,MAAM,uCAAuC,CAAC;AAEvF,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAC1D,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,yBAAyB,CAAC;AACtD,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AAC9C,OAAO,KAAK,EAAE,6BAA6B,EAAE,MAAM,2CAA2C,CAAC;AAE/F,MAAM,WAAW,cAAc;IAC3B,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,MAAM,EAAE,eAAe,CAAC;IACxB,cAAc,EAAE,cAAc,CAAC;IAC/B,iBAAiB,EAAE,iBAAiB,CAAC;IACrC,QAAQ,EAAE,QAAQ,CAAC;IACnB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,6BAA6B,EAAE,6BAA6B,CAAC;CAChE;AAED,MAAM,MAAM,aAAa,GAAG;IACxB,OAAO,EAAE,EAAE,CAAC;IACZ,KAAK,EAAE,EAAE,CAAC;IACV,UAAU,EAAE,EAAE,CAAC;IACf,kBAAkB,EAAE,CAAC,sBAAsB,CAAC,CAAC;CAChD,CAAC;AAEF,qBAAa,OAAQ,SAAQ,YAAY,CAAC,aAAa,CAAC;IACpD,QAAQ,CAAC,SAAS,EAAE,MAAM,CAA6B;IACvD,QAAQ,CAAC,cAAc,EAAE,cAAc,CAAC;IACxC,QAAQ,CAAC,iBAAiB,EAAE,iBAAiB,CAAC;IAC9C,QAAQ,CAAC,SAAS,EAAE,SAAS,CAAC;IAC9B,QAAQ,CAAC,gBAAgB,CAAC,EAAE,MAAM,CAAC;IACnC,QAAQ,CAAC,QAAQ,EAAE,QAAQ,CAAC;IAC5B,QAAQ,CAAC,6BAA6B,EAAE,6BAA6B,CAAC;IAEtE,SAAS,CAAC,EAAE;QACR,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,KAAK,CAAC,EAAE,MAAM,CAAC;KAClB,CAAC;IAEK,MAAM,EAAE,eAAe,CAAC;gBAEnB,EACR,UAAU,EACV,WAAW,EACX,eAAe,EACf,MAAM,EACN,iBAAiB,EACjB,cAAc,EACd,QAAQ,EACR,gBAAgB,EAChB,6BAA6B,GAChC,EAAE,cAAc;IAwBjB,YAAY,CAAC,SAAS,EAAE,cAAc,GAAG,SAAS,GAAG,IAAI;IAoBnD,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IA2B3B,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAOtB,gBAAgB,CAAC,QAAQ,EAAE,kBAAkB,GAAG,OAAO,CAAC,IAAI,CAAC;IAInE,IAAI,oBAAoB,IAAI,OAAO,CAElC;IAEK,iBAAiB,IAAI,OAAO,CAAC,OAAO,CAAC;IASrC,qBAAqB,IAAI,OAAO,CAAC,IAAI,CAAC;IAU5C,IAAI,eAAe,IAAI,yBAAyB,CAO/C;IAED,IAAI,qBAAqB,IAAI,0BAA0B,GAAG,SAAS,CAElE;CACJ"}
+{"version":3,"file":"session.d.ts","sourceRoot":"","sources":["../../../src/common/session.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACjD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,oCAAoC,CAAC;AACzE,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAEnD,OAAO,YAAY,MAAM,QAAQ,CAAC;AAClC,OAAO,KAAK,EACR,0BAA0B,EAC1B,iBAAiB,EACjB,kBAAkB,EAElB,sBAAsB,EACtB,wBAAwB,EAC3B,MAAM,wBAAwB,CAAC;AAChC,OAAO,KAAK,EAAE,yBAAyB,EAAE,MAAM,uCAAuC,CAAC;AAEvF,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAC1D,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,yBAAyB,CAAC;AACtD,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AAC9C,OAAO,KAAK,EAAE,6BAA6B,EAAE,MAAM,2CAA2C,CAAC;AAE/F,OAAO,EAAE,KAAK,UAAU,EAAE,MAAM,gCAAgC,CAAC;AAEjE,MAAM,WAAW,cAAc;IAC3B,UAAU,EAAE,UAAU,CAAC;IACvB,MAAM,EAAE,eAAe,CAAC;IACxB,cAAc,EAAE,cAAc,CAAC;IAC/B,iBAAiB,EAAE,iBAAiB,CAAC;IACrC,QAAQ,EAAE,QAAQ,CAAC;IACnB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,6BAA6B,EAAE,6BAA6B,CAAC;CAChE;AAED,MAAM,MAAM,aAAa,GAAG;IACxB,OAAO,EAAE,EAAE,CAAC;IACZ,KAAK,EAAE,EAAE,CAAC;IACV,UAAU,EAAE,EAAE,CAAC;IACf,kBAAkB,EAAE,CAAC,sBAAsB,CAAC,CAAC;CAChD,CAAC;AAEF,qBAAa,OAAQ,SAAQ,YAAY,CAAC,aAAa,CAAC;IACpD,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAa;IACxC,QAAQ,CAAC,SAAS,EAAE,MAAM,CAA6B;IACvD,QAAQ,CAAC,cAAc,EAAE,cAAc,CAAC;IACxC,QAAQ,CAAC,iBAAiB,EAAE,iBAAiB,CAAC;IAC9C,QAAQ,CAAC,SAAS,EAAE,SAAS,CAAC;IAC9B,QAAQ,CAAC,gBAAgB,CAAC,EAAE,MAAM,CAAC;IACnC,QAAQ,CAAC,QAAQ,EAAE,QAAQ,CAAC;IAC5B,QAAQ,CAAC,6BAA6B,EAAE,6BAA6B,CAAC;IAEtE,SAAS,CAAC,EAAE;QACR,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,KAAK,CAAC,EAAE,MAAM,CAAC;KAClB,CAAC;IAEK,MAAM,EAAE,eAAe,CAAC;gBAEnB,EACR,UAAU,EACV,MAAM,EACN,iBAAiB,EACjB,cAAc,EACd,QAAQ,EACR,gBAAgB,EAChB,6BAA6B,GAChC,EAAE,cAAc;IAyBjB,YAAY,CAAC,SAAS,EAAE,cAAc,GAAG,SAAS,GAAG,IAAI;IAoBnD,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IA2B3B,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAOtB,6BAA6B,IAAI,OAAO,CAAC,IAAI,CAAC;IAQ9C,gBAAgB,CAAC,QAAQ,EAAE,kBAAkB,GAAG,OAAO,CAAC,IAAI,CAAC;IAInE,IAAI,oBAAoB,IAAI,OAAO,CAElC;IAEK,iBAAiB,IAAI,OAAO,CAAC,OAAO,CAAC;IASrC,qBAAqB,IAAI,OAAO,CAAC,IAAI,CAAC;IAU5C,IAAI,eAAe,IAAI,yBAAyB,CAO/C;IAED,IAAI,qBAAqB,IAAI,0BAA0B,GAAG,SAAS,CAElE;IAED,IAAI,wBAAwB,IAAI,wBAAwB,GAAG,SAAS,CAEnE;CACJ"}

package/dist/cjs/common/session.js

@@ -9,19 +9,21 @@ const apiClient_js_1 = require("./atlas/apiClient.js");
 const logger_js_1 = require("./logger.js");
 const events_1 = __importDefault(require("events"));
 const errors_js_1 = require("./errors.js");
+const arg_parser_1 = require("@mongosh/arg-parser");
 class Session extends events_1.default {
-    constructor({ apiBaseUrl, apiClientId, apiClientSecret, logger, connectionManager, exportsManager, keychain, atlasLocalClient, vectorSearchEmbeddingsManager, }) {
+    constructor({ userConfig, logger, connectionManager, exportsManager, keychain, atlasLocalClient, vectorSearchEmbeddingsManager, }) {
         super();
         this.sessionId = new bson_1.ObjectId().toString();
+        this.userConfig = userConfig;
         this.keychain = keychain;
         this.logger = logger;
-        const credentials = apiClientId && apiClientSecret
+        const credentials = userConfig.apiClientId && userConfig.apiClientSecret
             ? {
-                clientId: apiClientId,
-                clientSecret: apiClientSecret,
+                clientId: userConfig.apiClientId,
+                clientSecret: userConfig.apiClientSecret,
             }
             : undefined;
-        this.apiClient = new apiClient_js_1.ApiClient({ baseUrl: apiBaseUrl, credentials }, logger);
+        this.apiClient = new apiClient_js_1.ApiClient({ baseUrl: userConfig.apiBaseUrl, credentials }, logger);
         this.atlasLocalClient = atlasLocalClient;
         this.exportsManager = exportsManager;
         this.connectionManager = connectionManager;
@@ -78,6 +80,13 @@ class Session extends events_1.default {
         await this.exportsManager.close();
         this.emit("close");
     }
+    async connectToConfiguredConnection() {
+        const connectionInfo = (0, arg_parser_1.generateConnectionInfoFromCliArgs)({
+            ...this.userConfig,
+            connectionSpecifier: this.userConfig.connectionString,
+        });
+        await this.connectToMongoDB(connectionInfo);
+    }
     async connectToMongoDB(settings) {
         await this.connectionManager.connect({ ...settings });
     }
@@ -107,6 +116,9 @@ class Session extends events_1.default {
     get connectedAtlasCluster() {
         return this.connectionManager.currentConnectionState.connectedAtlasCluster;
     }
+    get connectionStringAuthType() {
+        return this.connectionManager.currentConnectionState.connectionStringAuthType;
+    }
 }
 exports.Session = Session;
 //# sourceMappingURL=session.js.map

package/dist/cjs/common/session.js.map

@@ -1 +1 @@
-{"version":3,"file":"session.js","sourceRoot":"","sources":["../../../src/common/session.ts"],"names":[],"mappings":";;;;;;AAAA,+BAAgC;AAEhC,uDAAiD;AAGjD,2CAAoC;AACpC,oDAAkC;AASlC,2CAAuD;AAyBvD,MAAa,OAAQ,SAAQ,gBAA2B;IAiBpD,YAAY,EACR,UAAU,EACV,WAAW,EACX,eAAe,EACf,MAAM,EACN,iBAAiB,EACjB,cAAc,EACd,QAAQ,EACR,gBAAgB,EAChB,6BAA6B,GAChB;QACb,KAAK,EAAE,CAAC;QA3BH,cAAS,GAAW,IAAI,eAAQ,EAAE,CAAC,QAAQ,EAAE,CAAC;QA6BnD,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,MAAM,WAAW,GACb,WAAW,IAAI,eAAe;YAC1B,CAAC,CAAC;gBACI,QAAQ,EAAE,WAAW;gBACrB,YAAY,EAAE,eAAe;aAChC;YACH,CAAC,CAAC,SAAS,CAAC;QAEpB,IAAI,CAAC,SAAS,GAAG,IAAI,wBAAS,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,WAAW,EAAE,EAAE,MAAM,CAAC,CAAC;QAC7E,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QACrC,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;QAC3C,IAAI,CAAC,6BAA6B,GAAG,6BAA6B,CAAC;QACnE,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,EAAE,CAAC,oBAAoB,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC;QACnF,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,EAAE,CAAC,qBAAqB,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,KAAK,CAAC,CAAC,CAAC;QACzG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,EAAE,CAAC,kBAAkB,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC;QACpF,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,EAAE,CAAC,kBAAkB,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,KAAK,CAAC,CAAC,CAAC;IAC1G,CAAC;IAED,YAAY,CAAC,SAAqC;QAC9C,IAAI,CAAC,SAAS,EAAE,CAAC;YACb,IAAI,CAAC,iBAAiB,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;YAChD,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;gBACd,EAAE,EAAE,iBAAK,CAAC,kBAAkB;gBAC5B,OAAO,EAAE,SAAS;gBAClB,OAAO,EAAE,2BAA2B;aACvC,CAAC,CAAC;QACP,CAAC;QAED,IAAI,CAAC,SAAS,GAAG;YACb,IAAI,EAAE,SAAS,EAAE,IAAI,IAAI,SAAS;YAClC,OAAO,EAAE,SAAS,EAAE,OAAO,IAAI,SAAS;YACxC,KAAK,EAAE,SAAS,EAAE,KAAK,IAAI,SAAS;SACvC,CAAC;QAEF,uEAAuE;QACvE,IAAI,CAAC,iBAAiB,CAAC,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,IAAI,SAAS,CAAC,CAAC;IAC3E,CAAC;IAED,KAAK,CAAC,UAAU;QACZ,MAAM,YAAY,GAAG,IAAI,CAAC,qBAAqB,CAAC;QAEhD,MAAM,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,CAAC;QAErC,IAAI,YAAY,EAAE,QAAQ,IAAI,YAAY,EAAE,SAAS,EAAE,CAAC;YACpD,KAAK,IAAI,CAAC,SAAS;iBACd,kBAAkB,CAAC;gBAChB,MAAM,EAAE;oBACJ,IAAI,EAAE;wBACF,OAAO,EAAE,YAAY,CAAC,SAAS;wBAC/B,QAAQ,EAAE,YAAY,CAAC,QAAQ;wBAC/B,YAAY,EAAE,OAAO;qBACxB;iBACJ;aACJ,CAAC;iBACD,KAAK,CAAC,CAAC,GAAY,EAAE,EAAE;gBACpB,MAAM,KAAK,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;gBAClE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;oBACd,EAAE,EAAE,iBAAK,CAAC,8BAA8B;oBACxC,OAAO,EAAE,SAAS;oBAClB,OAAO,EAAE,0CAA0C,KAAK,CAAC,OAAO,EAAE;iBACrE,CAAC,CAAC;YACP,CAAC,CAAC,CAAC;QACX,CAAC;IACL,CAAC;IAED,KAAK,CAAC,KAAK;QACP,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,MAAM,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;QAC7B,MAAM,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,CAAC;QAClC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,QAA4B;QAC/C,MAAM,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,EAAE,GAAG,QAAQ,EAAE,CAAC,CAAC;IAC1D,CAAC;IAED,IAAI,oBAAoB;QACpB,OAAO,IAAI,CAAC,iBAAiB,CAAC,sBAAsB,CAAC,GAAG,KAAK,WAAW,CAAC;IAC7E,CAAC;IAED,KAAK,CAAC,iBAAiB;QACnB,MAAM,KAAK,GAAG,IAAI,CAAC,iBAAiB,CAAC,sBAAsB,CAAC;QAC5D,IAAI,KAAK,CAAC,GAAG,KAAK,WAAW,EAAE,CAAC;YAC5B,OAAO,MAAM,KAAK,CAAC,iBAAiB,EAAE,CAAC;QAC3C,CAAC;QAED,OAAO,KAAK,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,qBAAqB;QACvB,MAAM,iBAAiB,GAAG,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACzD,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACrB,MAAM,IAAI,wBAAY,CAClB,sBAAU,CAAC,uBAAuB,EAClC,uDAAuD,CAC1D,CAAC;QACN,CAAC;IACL,CAAC;IAED,IAAI,eAAe;QACf,IAAI,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC5B,MAAM,KAAK,GAAG,IAAI,CAAC,iBAAiB,CAAC,sBAAkD,CAAC;YACxF,OAAO,KAAK,CAAC,eAAe,CAAC;QACjC,CAAC;QAED,MAAM,IAAI,wBAAY,CAAC,sBAAU,CAAC,qBAAqB,EAAE,0BAA0B,CAAC,CAAC;IACzF,CAAC;IAED,IAAI,qBAAqB;QACrB,OAAO,IAAI,CAAC,iBAAiB,CAAC,sBAAsB,CAAC,qBAAqB,CAAC;IAC/E,CAAC;CACJ;AAhJD,0BAgJC"}
+{"version":3,"file":"session.js","sourceRoot":"","sources":["../../../src/common/session.ts"],"names":[],"mappings":";;;;;;AAAA,+BAAgC;AAEhC,uDAAiD;AAGjD,2CAAoC;AACpC,oDAAkC;AAUlC,2CAAuD;AAKvD,oDAAwE;AAoBxE,MAAa,OAAQ,SAAQ,gBAA2B;IAkBpD,YAAY,EACR,UAAU,EACV,MAAM,EACN,iBAAiB,EACjB,cAAc,EACd,QAAQ,EACR,gBAAgB,EAChB,6BAA6B,GAChB;QACb,KAAK,EAAE,CAAC;QAzBH,cAAS,GAAW,IAAI,eAAQ,EAAE,CAAC,QAAQ,EAAE,CAAC;QA2BnD,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,MAAM,WAAW,GACb,UAAU,CAAC,WAAW,IAAI,UAAU,CAAC,eAAe;YAChD,CAAC,CAAC;gBACI,QAAQ,EAAE,UAAU,CAAC,WAAW;gBAChC,YAAY,EAAE,UAAU,CAAC,eAAe;aAC3C;YACH,CAAC,CAAC,SAAS,CAAC;QAEpB,IAAI,CAAC,SAAS,GAAG,IAAI,wBAAS,CAAC,EAAE,OAAO,EAAE,UAAU,CAAC,UAAU,EAAE,WAAW,EAAE,EAAE,MAAM,CAAC,CAAC;QACxF,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QACrC,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;QAC3C,IAAI,CAAC,6BAA6B,GAAG,6BAA6B,CAAC;QACnE,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,EAAE,CAAC,oBAAoB,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC;QACnF,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,EAAE,CAAC,qBAAqB,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,KAAK,CAAC,CAAC,CAAC;QACzG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,EAAE,CAAC,kBAAkB,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC;QACpF,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,EAAE,CAAC,kBAAkB,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,KAAK,CAAC,CAAC,CAAC;IAC1G,CAAC;IAED,YAAY,CAAC,SAAqC;QAC9C,IAAI,CAAC,SAAS,EAAE,CAAC;YACb,IAAI,CAAC,iBAAiB,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;YAChD,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;gBACd,EAAE,EAAE,iBAAK,CAAC,kBAAkB;gBAC5B,OAAO,EAAE,SAAS;gBAClB,OAAO,EAAE,2BAA2B;aACvC,CAAC,CAAC;QACP,CAAC;QAED,IAAI,CAAC,SAAS,GAAG;YACb,IAAI,EAAE,SAAS,EAAE,IAAI,IAAI,SAAS;YAClC,OAAO,EAAE,SAAS,EAAE,OAAO,IAAI,SAAS;YACxC,KAAK,EAAE,SAAS,EAAE,KAAK,IAAI,SAAS;SACvC,CAAC;QAEF,uEAAuE;QACvE,IAAI,CAAC,iBAAiB,CAAC,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,IAAI,SAAS,CAAC,CAAC;IAC3E,CAAC;IAED,KAAK,CAAC,UAAU;QACZ,MAAM,YAAY,GAAG,IAAI,CAAC,qBAAqB,CAAC;QAEhD,MAAM,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,CAAC;QAErC,IAAI,YAAY,EAAE,QAAQ,IAAI,YAAY,EAAE,SAAS,EAAE,CAAC;YACpD,KAAK,IAAI,CAAC,SAAS;iBACd,kBAAkB,CAAC;gBAChB,MAAM,EAAE;oBACJ,IAAI,EAAE;wBACF,OAAO,EAAE,YAAY,CAAC,SAAS;wBAC/B,QAAQ,EAAE,YAAY,CAAC,QAAQ;wBAC/B,YAAY,EAAE,OAAO;qBACxB;iBACJ;aACJ,CAAC;iBACD,KAAK,CAAC,CAAC,GAAY,EAAE,EAAE;gBACpB,MAAM,KAAK,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;gBAClE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;oBACd,EAAE,EAAE,iBAAK,CAAC,8BAA8B;oBACxC,OAAO,EAAE,SAAS;oBAClB,OAAO,EAAE,0CAA0C,KAAK,CAAC,OAAO,EAAE;iBACrE,CAAC,CAAC;YACP,CAAC,CAAC,CAAC;QACX,CAAC;IACL,CAAC;IAED,KAAK,CAAC,KAAK;QACP,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,MAAM,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;QAC7B,MAAM,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,CAAC;QAClC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,6BAA6B;QAC/B,MAAM,cAAc,GAAG,IAAA,8CAAiC,EAAC;YACrD,GAAG,IAAI,CAAC,UAAU;YAClB,mBAAmB,EAAE,IAAI,CAAC,UAAU,CAAC,gBAAgB;SACxD,CAAC,CAAC;QACH,MAAM,IAAI,CAAC,gBAAgB,CAAC,cAAc,CAAC,CAAC;IAChD,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,QAA4B;QAC/C,MAAM,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,EAAE,GAAG,QAAQ,EAAE,CAAC,CAAC;IAC1D,CAAC;IAED,IAAI,oBAAoB;QACpB,OAAO,IAAI,CAAC,iBAAiB,CAAC,sBAAsB,CAAC,GAAG,KAAK,WAAW,CAAC;IAC7E,CAAC;IAED,KAAK,CAAC,iBAAiB;QACnB,MAAM,KAAK,GAAG,IAAI,CAAC,iBAAiB,CAAC,sBAAsB,CAAC;QAC5D,IAAI,KAAK,CAAC,GAAG,KAAK,WAAW,EAAE,CAAC;YAC5B,OAAO,MAAM,KAAK,CAAC,iBAAiB,EAAE,CAAC;QAC3C,CAAC;QAED,OAAO,KAAK,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,qBAAqB;QACvB,MAAM,iBAAiB,GAAG,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACzD,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACrB,MAAM,IAAI,wBAAY,CAClB,sBAAU,CAAC,uBAAuB,EAClC,uDAAuD,CAC1D,CAAC;QACN,CAAC;IACL,CAAC;IAED,IAAI,eAAe;QACf,IAAI,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC5B,MAAM,KAAK,GAAG,IAAI,CAAC,iBAAiB,CAAC,sBAAkD,CAAC;YACxF,OAAO,KAAK,CAAC,eAAe,CAAC;QACjC,CAAC;QAED,MAAM,IAAI,wBAAY,CAAC,sBAAU,CAAC,qBAAqB,EAAE,0BAA0B,CAAC,CAAC;IACzF,CAAC;IAED,IAAI,qBAAqB;QACrB,OAAO,IAAI,CAAC,iBAAiB,CAAC,sBAAsB,CAAC,qBAAqB,CAAC;IAC/E,CAAC;IAED,IAAI,wBAAwB;QACxB,OAAO,IAAI,CAAC,iBAAiB,CAAC,sBAAsB,CAAC,wBAAwB,CAAC;IAClF,CAAC;CACJ;AA5JD,0BA4JC"}

package/dist/cjs/helpers/assertVectorSearchFilterFieldsAreIndexed.d.ts

@@ -0,0 +1,27 @@
+import { type CompositeLogger } from "../common/logger.js";
+export type SearchIndex = VectorSearchIndex | AtlasSearchIndex;
+type VectorSearchIndex = {
+    name: string;
+    latestDefinition: {
+        fields: Array<{
+            type: "vector";
+        } | {
+            type: "filter";
+            path: string;
+        }>;
+    };
+    type: "vectorSearch";
+};
+type AtlasSearchIndex = {
+    name: string;
+    latestDefinition: unknown;
+    type: "search";
+};
+export declare function assertVectorSearchFilterFieldsAreIndexed({ searchIndexes, pipeline, logger, }: {
+    searchIndexes: SearchIndex[];
+    pipeline: Record<string, unknown>[];
+    logger: CompositeLogger;
+}): void;
+export declare function collectFieldsFromVectorSearchFilter(filter: unknown): string[];
+export {};
+//# sourceMappingURL=assertVectorSearchFilterFieldsAreIndexed.d.ts.map

package/dist/cjs/helpers/assertVectorSearchFilterFieldsAreIndexed.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"assertVectorSearchFilterFieldsAreIndexed.d.ts","sourceRoot":"","sources":["../../../src/helpers/assertVectorSearchFilterFieldsAreIndexed.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,KAAK,eAAe,EAAS,MAAM,qBAAqB,CAAC;AAKlE,MAAM,MAAM,WAAW,GAAG,iBAAiB,GAAG,gBAAgB,CAAC;AAE/D,KAAK,iBAAiB,GAAG;IACrB,IAAI,EAAE,MAAM,CAAC;IACb,gBAAgB,EAAE;QACd,MAAM,EAAE,KAAK,CACP;YACI,IAAI,EAAE,QAAQ,CAAC;SAClB,GACD;YACI,IAAI,EAAE,QAAQ,CAAC;YACf,IAAI,EAAE,MAAM,CAAC;SAChB,CACN,CAAC;KACL,CAAC;IACF,IAAI,EAAE,cAAc,CAAC;CACxB,CAAC;AAEF,KAAK,gBAAgB,GAAG;IACpB,IAAI,EAAE,MAAM,CAAC;IACb,gBAAgB,EAAE,OAAO,CAAC;IAC1B,IAAI,EAAE,QAAQ,CAAC;CAClB,CAAC;AAEF,wBAAgB,wCAAwC,CAAC,EACrD,aAAa,EACb,QAAQ,EACR,MAAM,GACT,EAAE;IACC,aAAa,EAAE,WAAW,EAAE,CAAC;IAC7B,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC;IACpC,MAAM,EAAE,eAAe,CAAC;CAC3B,GAAG,IAAI,CAqCP;AAED,wBAAgB,mCAAmC,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,EAAE,CAiB7E"}

package/dist/cjs/helpers/assertVectorSearchFilterFieldsAreIndexed.js

@@ -0,0 +1,58 @@
+"use strict";
+// Based on -
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.assertVectorSearchFilterFieldsAreIndexed = assertVectorSearchFilterFieldsAreIndexed;
+exports.collectFieldsFromVectorSearchFilter = collectFieldsFromVectorSearchFilter;
+const errors_js_1 = require("../common/errors.js");
+const logger_js_1 = require("../common/logger.js");
+// https://www.mongodb.com/docs/atlas/atlas-vector-search/vector-search-stage/#mongodb-vector-search-pre-filter
+const ALLOWED_LOGICAL_OPERATORS = ["$not", "$nor", "$and", "$or"];
+function assertVectorSearchFilterFieldsAreIndexed({ searchIndexes, pipeline, logger, }) {
+    const searchIndexesWithFilterFields = searchIndexes
+        // Ensure we only process vector search indexes and not lexical search ones
+        .filter((index) => index.type === "vectorSearch")
+        .reduce((indexFieldMap, searchIndex) => {
+        const filterFields = searchIndex.latestDefinition.fields
+            .map((field) => {
+            return field.type === "filter" ? field.path : undefined;
+        })
+            .filter((filterField) => filterField !== undefined);
+        indexFieldMap[searchIndex.name] = filterFields;
+        return indexFieldMap;
+    }, {});
+    for (const stage of pipeline) {
+        if ("$vectorSearch" in stage) {
+            const { $vectorSearch: vectorSearchStage } = stage;
+            const allowedFilterFields = searchIndexesWithFilterFields[vectorSearchStage.index];
+            if (!allowedFilterFields) {
+                logger.warning({
+                    id: logger_js_1.LogId.toolValidationError,
+                    context: "aggregate tool",
+                    message: `Could not assert if filter fields are indexed - No filter fields found for index ${vectorSearchStage.index}`,
+                });
+                return;
+            }
+            const filterFieldsInStage = collectFieldsFromVectorSearchFilter(vectorSearchStage.filter);
+            const filterFieldsNotIndexed = filterFieldsInStage.filter((field) => !allowedFilterFields.includes(field));
+            if (filterFieldsNotIndexed.length) {
+                throw new errors_js_1.MongoDBError(errors_js_1.ErrorCodes.AtlasVectorSearchInvalidQuery, `Vector search stage contains filter on fields that are not indexed by index ${vectorSearchStage.index} - ${filterFieldsNotIndexed.join(", ")}`);
+            }
+        }
+    }
+}
+function collectFieldsFromVectorSearchFilter(filter) {
+    if (!filter || typeof filter !== "object" || !Object.keys(filter).length) {
+        return [];
+    }
+    const collectedFields = Object.entries(filter).reduce((collectedFields, [maybeField, fieldMQL]) => {
+        if (ALLOWED_LOGICAL_OPERATORS.includes(maybeField) && Array.isArray(fieldMQL)) {
+            return fieldMQL.flatMap((mql) => collectFieldsFromVectorSearchFilter(mql));
+        }
+        if (!ALLOWED_LOGICAL_OPERATORS.includes(maybeField)) {
+            collectedFields.push(maybeField);
+        }
+        return collectedFields;
+    }, []);
+    return Array.from(new Set(collectedFields));
+}
+//# sourceMappingURL=assertVectorSearchFilterFieldsAreIndexed.js.map

package/dist/cjs/helpers/assertVectorSearchFilterFieldsAreIndexed.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"assertVectorSearchFilterFieldsAreIndexed.js","sourceRoot":"","sources":["../../../src/helpers/assertVectorSearchFilterFieldsAreIndexed.ts"],"names":[],"mappings":";AAAA,aAAa;;AAkCb,4FA6CC;AAED,kFAiBC;AA/FD,mDAA+D;AAE/D,mDAAkE;AAElE,+GAA+G;AAC/G,MAAM,yBAAyB,GAAG,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;AA0BlE,SAAgB,wCAAwC,CAAC,EACrD,aAAa,EACb,QAAQ,EACR,MAAM,GAKT;IACG,MAAM,6BAA6B,GAAG,aAAa;QAC/C,2EAA2E;SAC1E,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,cAAc,CAAC;SAChD,MAAM,CAA2B,CAAC,aAAa,EAAE,WAAW,EAAE,EAAE;QAC7D,MAAM,YAAY,GAAG,WAAW,CAAC,gBAAgB,CAAC,MAAM;aACnD,GAAG,CAAqB,CAAC,KAAK,EAAE,EAAE;YAC/B,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;QAC5D,CAAC,CAAC;aACD,MAAM,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC;QAExD,aAAa,CAAC,WAAW,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC;QAC/C,OAAO,aAAa,CAAC;IACzB,CAAC,EAAE,EAAE,CAAC,CAAC;IACX,KAAK,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;QAC3B,IAAI,eAAe,IAAI,KAAK,EAAE,CAAC;YAC3B,MAAM,EAAE,aAAa,EAAE,iBAAiB,EAAE,GAAG,KAA0C,CAAC;YACxF,MAAM,mBAAmB,GAAG,6BAA6B,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;YACnF,IAAI,CAAC,mBAAmB,EAAE,CAAC;gBACvB,MAAM,CAAC,OAAO,CAAC;oBACX,EAAE,EAAE,iBAAK,CAAC,mBAAmB;oBAC7B,OAAO,EAAE,gBAAgB;oBACzB,OAAO,EAAE,oFAAoF,iBAAiB,CAAC,KAAK,EAAE;iBACzH,CAAC,CAAC;gBACH,OAAO;YACX,CAAC;YAED,MAAM,mBAAmB,GAAG,mCAAmC,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YAC1F,MAAM,sBAAsB,GAAG,mBAAmB,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,mBAAmB,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC;YAC3G,IAAI,sBAAsB,CAAC,MAAM,EAAE,CAAC;gBAChC,MAAM,IAAI,wBAAY,CAClB,sBAAU,CAAC,6BAA6B,EACxC,+EAA+E,iBAAiB,CAAC,KAAK,MAAM,sBAAsB,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAClJ,CAAC;YACN,CAAC;QACL,CAAC;IACL,CAAC;AACL,CAAC;AAED,SAAgB,mCAAmC,CAAC,MAAe;IAC/D,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,CAAC;QACvE,OAAO,EAAE,CAAC;IACd,CAAC;IAED,MAAM,eAAe,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,MAAM,CAAW,CAAC,eAAe,EAAE,CAAC,UAAU,EAAE,QAAQ,CAAC,EAAE,EAAE;QACxG,IAAI,yBAAyB,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC5E,OAAO,QAAQ,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,mCAAmC,CAAC,GAAG,CAAC,CAAC,CAAC;QAC/E,CAAC;QAED,IAAI,CAAC,yBAAyB,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;YAClD,eAAe,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACrC,CAAC;QACD,OAAO,eAAe,CAAC;IAC3B,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,eAAe,CAAC,CAAC,CAAC;AAChD,CAAC"}

package/dist/cjs/index.d.ts

@@ -1,3 +1,4 @@
 #!/usr/bin/env node
-export {};
+import { type UserConfig } from "./common/config/userConfig.js";
+export declare function handleDryRunRequest(config: UserConfig): Promise<never>;
 //# sourceMappingURL=index.d.ts.map

package/dist/cjs/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":""}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";AAwCA,OAAO,EAAE,KAAK,UAAU,EAAE,MAAM,+BAA+B,CAAC;AAoHhE,wBAAsB,mBAAmB,CAAC,MAAM,EAAE,UAAU,GAAG,OAAO,CAAC,KAAK,CAAC,CAoB5E"}

package/dist/cjs/index.js

@@ -4,6 +4,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.handleDryRunRequest = handleDryRunRequest;
 function enableFipsIfRequested() {
     let fipsError;
     const tlsFIPSMode = process.argv.includes("--tlsFIPSMode");
@@ -34,24 +35,33 @@ function enableFipsIfRequested() {
     }
 }
 enableFipsIfRequested();
-const logger_js_1 = require("./common/logger.js");
-const config_js_1 = require("./common/config.js");
 const crypto_1 = __importDefault(require("crypto"));
+const logger_js_1 = require("./common/logger.js");
+const createUserConfig_js_1 = require("./common/config/createUserConfig.js");
 const packageInfo_js_1 = require("./common/packageInfo.js");
 const stdio_js_1 = require("./transports/stdio.js");
 const streamableHttp_js_1 = require("./transports/streamableHttp.js");
 const devtools_proxy_support_1 = require("@mongodb-js/devtools-proxy-support");
 const keychain_js_1 = require("./common/keychain.js");
+const dryModeRunner_js_1 = require("./transports/dryModeRunner.js");
 async function main() {
     (0, devtools_proxy_support_1.systemCA)().catch(() => undefined); // load system CA asynchronously as in mongosh
-    assertHelpMode();
-    assertVersionMode();
-    const transportRunner = config_js_1.config.transport === "stdio"
+    const config = (0, createUserConfig_js_1.createUserConfig)();
+    if (config.help) {
+        handleHelpRequest();
+    }
+    if (config.version) {
+        handleVersionRequest();
+    }
+    if (config.dryRun) {
+        await handleDryRunRequest(config);
+    }
+    const transportRunner = config.transport === "stdio"
         ? new stdio_js_1.StdioRunner({
-            userConfig: config_js_1.config,
+            userConfig: config,
         })
         : new streamableHttp_js_1.StreamableHttpRunner({
-            userConfig: config_js_1.config,
+            userConfig: config,
         });
     const shutdown = () => {
         transportRunner.logger.info({
@@ -122,17 +132,35 @@ main().catch((error) => {
     });
     process.exit(1);
 });
-function assertHelpMode() {
-    if (config_js_1.config.help) {
-        console.log("For usage information refer to the README.md:");
-        console.log("https://github.com/mongodb-js/mongodb-mcp-server?tab=readme-ov-file#quick-start");
-        process.exit(0);
-    }
+function handleHelpRequest() {
+    console.log("For usage information refer to the README.md:");
+    console.log("https://github.com/mongodb-js/mongodb-mcp-server?tab=readme-ov-file#quick-start");
+    process.exit(0);
 }
-function assertVersionMode() {
-    if (config_js_1.config.version) {
-        console.log(packageInfo_js_1.packageInfo.version);
+function handleVersionRequest() {
+    console.log(packageInfo_js_1.packageInfo.version);
+    process.exit(0);
+}
+async function handleDryRunRequest(config) {
+    try {
+        const runner = new dryModeRunner_js_1.DryRunModeRunner({
+            userConfig: config,
+            logger: {
+                log(message) {
+                    console.log(message);
+                },
+                error(message) {
+                    console.error(message);
+                },
+            },
+        });
+        await runner.start();
+        await runner.close();
         process.exit(0);
     }
+    catch (error) {
+        console.error(`Fatal error running server in dry run mode: ${error}`);
+        process.exit(1);
+    }
 }
 //# sourceMappingURL=index.js.map

package/dist/cjs/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";;;;;;AAEA,SAAS,qBAAqB;IAC1B,IAAI,SAA4B,CAAC;IACjC,MAAM,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC;IAE3D,IAAI,WAAW,EAAE,CAAC;QACd,IAAI,CAAC;YACD,2BAA2B;YAC3B,OAAO,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QACjC,CAAC;QAAC,OAAO,GAAY,EAAE,CAAC;YACpB,SAAS,KAAT,SAAS,GAAK,GAAY,EAAC;QAC/B,CAAC;IACL,CAAC;IAED,IAAI,WAAW,EAAE,CAAC;QACd,IAAI,CAAC,SAAS,IAAI,CAAC,gBAAM,CAAC,OAAO,EAAE,EAAE,CAAC;YAClC,SAAS,GAAG,IAAI,KAAK,CAAC,+DAA+D,CAAC,CAAC;QAC3F,CAAC;IACL,CAAC;IAED,IAAI,SAAS,EAAE,CAAC;QACZ,IAAI,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,mBAAmB,EAAE,CAAC;YAC/C,OAAO,CAAC,KAAK,CACT,gGAAgG,CACnG,CAAC;QACN,CAAC;aAAM,CAAC;YACJ,OAAO,CAAC,KAAK,CAAC,gFAAgF,CAAC,CAAC;QACpG,CAAC;QACD,OAAO,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;QAChC,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QACzB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACpB,CAAC;AACL,CAAC;AAED,qBAAqB,EAAE,CAAC;AAExB,kDAA0D;AAC1D,kDAA4C;AAC5C,oDAA4B;AAC5B,4DAAsD;AACtD,oDAAoD;AACpD,sEAAsE;AACtE,+EAA8D;AAC9D,sDAAgD;AAEhD,KAAK,UAAU,IAAI;IACf,IAAA,iCAAQ,GAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC,8CAA8C;IAEjF,cAAc,EAAE,CAAC;IACjB,iBAAiB,EAAE,CAAC;IAEpB,MAAM,eAAe,GACjB,kBAAM,CAAC,SAAS,KAAK,OAAO;QACxB,CAAC,CAAC,IAAI,sBAAW,CAAC;YACZ,UAAU,EAAE,kBAAM;SACrB,CAAC;QACJ,CAAC,CAAC,IAAI,wCAAoB,CAAC;YACrB,UAAU,EAAE,kBAAM;SACrB,CAAC,CAAC;IACb,MAAM,QAAQ,GAAG,GAAS,EAAE;QACxB,eAAe,CAAC,MAAM,CAAC,IAAI,CAAC;YACxB,EAAE,EAAE,iBAAK,CAAC,oBAAoB;YAC9B,OAAO,EAAE,QAAQ;YACjB,OAAO,EAAE,wBAAwB;SACpC,CAAC,CAAC;QAEH,eAAe;aACV,KAAK,EAAE;aACP,IAAI,CAAC,GAAG,EAAE;YACP,eAAe,CAAC,MAAM,CAAC,IAAI,CAAC;gBACxB,EAAE,EAAE,iBAAK,CAAC,YAAY;gBACtB,OAAO,EAAE,QAAQ;gBACjB,OAAO,EAAE,eAAe;aAC3B,CAAC,CAAC;YACH,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACpB,CAAC,CAAC;aACD,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;YACtB,eAAe,CAAC,MAAM,CAAC,KAAK,CAAC;gBACzB,EAAE,EAAE,iBAAK,CAAC,kBAAkB;gBAC5B,OAAO,EAAE,QAAQ;gBACjB,OAAO,EAAE,yBAAyB,KAAe,EAAE;aACtD,CAAC,CAAC;YACH,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACpB,CAAC,CAAC,CAAC;IACX,CAAC,CAAC;IAEF,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;IAC/B,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IAChC,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IAChC,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IAEhC,IAAI,CAAC;QACD,MAAM,eAAe,CAAC,KAAK,EAAE,CAAC;IAClC,CAAC;IAAC,OAAO,KAAc,EAAE,CAAC;QACtB,eAAe,CAAC,MAAM,CAAC,IAAI,CAAC;YACxB,EAAE,EAAE,iBAAK,CAAC,oBAAoB;YAC9B,OAAO,EAAE,QAAQ;YACjB,OAAO,EAAE,gCAAgC,KAAe,EAAE;YAC1D,WAAW,EAAE,IAAI;SACpB,CAAC,CAAC;QAEH,IAAI,CAAC;YACD,MAAM,eAAe,CAAC,KAAK,EAAE,CAAC;YAC9B,eAAe,CAAC,MAAM,CAAC,IAAI,CAAC;gBACxB,EAAE,EAAE,iBAAK,CAAC,YAAY;gBACtB,OAAO,EAAE,QAAQ;gBACjB,OAAO,EAAE,eAAe;aAC3B,CAAC,CAAC;QACP,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACtB,eAAe,CAAC,MAAM,CAAC,KAAK,CAAC;gBACzB,EAAE,EAAE,iBAAK,CAAC,kBAAkB;gBAC5B,OAAO,EAAE,QAAQ;gBACjB,OAAO,EAAE,yBAAyB,KAAe,EAAE;aACtD,CAAC,CAAC;QACP,CAAC;QACD,MAAM,KAAK,CAAC;IAChB,CAAC;AACL,CAAC;AAED,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;IAC5B,kFAAkF;IAClF,gFAAgF;IAChF,kBAAkB;IAClB,MAAM,MAAM,GAAG,IAAI,yBAAa,CAAC,sBAAQ,CAAC,IAAI,CAAC,CAAC;IAChD,MAAM,CAAC,SAAS,CAAC;QACb,EAAE,EAAE,iBAAK,CAAC,kBAAkB;QAC5B,OAAO,EAAE,QAAQ;QACjB,OAAO,EAAE,+BAA+B,KAAe,EAAE;KAC5D,CAAC,CAAC;IACH,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AACpB,CAAC,CAAC,CAAC;AAEH,SAAS,cAAc;IACnB,IAAI,kBAAM,CAAC,IAAI,EAAE,CAAC;QACd,OAAO,CAAC,GAAG,CAAC,+CAA+C,CAAC,CAAC;QAC7D,OAAO,CAAC,GAAG,CAAC,iFAAiF,CAAC,CAAC;QAC/F,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACpB,CAAC;AACL,CAAC;AAED,SAAS,iBAAiB;IACtB,IAAI,kBAAM,CAAC,OAAO,EAAE,CAAC;QACjB,OAAO,CAAC,GAAG,CAAC,4BAAW,CAAC,OAAO,CAAC,CAAC;QACjC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACpB,CAAC;AACL,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";;;;;;AA4JA,kDAoBC;AA9KD,SAAS,qBAAqB;IAC1B,IAAI,SAA4B,CAAC;IACjC,MAAM,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC;IAE3D,IAAI,WAAW,EAAE,CAAC;QACd,IAAI,CAAC;YACD,2BAA2B;YAC3B,OAAO,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QACjC,CAAC;QAAC,OAAO,GAAY,EAAE,CAAC;YACpB,SAAS,KAAT,SAAS,GAAK,GAAY,EAAC;QAC/B,CAAC;IACL,CAAC;IAED,IAAI,WAAW,EAAE,CAAC;QACd,IAAI,CAAC,SAAS,IAAI,CAAC,gBAAM,CAAC,OAAO,EAAE,EAAE,CAAC;YAClC,SAAS,GAAG,IAAI,KAAK,CAAC,+DAA+D,CAAC,CAAC;QAC3F,CAAC;IACL,CAAC;IAED,IAAI,SAAS,EAAE,CAAC;QACZ,IAAI,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,mBAAmB,EAAE,CAAC;YAC/C,OAAO,CAAC,KAAK,CACT,gGAAgG,CACnG,CAAC;QACN,CAAC;aAAM,CAAC;YACJ,OAAO,CAAC,KAAK,CAAC,gFAAgF,CAAC,CAAC;QACpG,CAAC;QACD,OAAO,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;QAChC,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QACzB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACpB,CAAC;AACL,CAAC;AAED,qBAAqB,EAAE,CAAC;AAExB,oDAA4B;AAC5B,kDAA0D;AAC1D,6EAAuE;AAEvE,4DAAsD;AACtD,oDAAoD;AACpD,sEAAsE;AACtE,+EAA8D;AAC9D,sDAAgD;AAChD,oEAAiE;AAEjE,KAAK,UAAU,IAAI;IACf,IAAA,iCAAQ,GAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC,8CAA8C;IAEjF,MAAM,MAAM,GAAG,IAAA,sCAAgB,GAAE,CAAC;IAClC,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;QACd,iBAAiB,EAAE,CAAC;IACxB,CAAC;IAED,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;QACjB,oBAAoB,EAAE,CAAC;IAC3B,CAAC;IAED,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;QAChB,MAAM,mBAAmB,CAAC,MAAM,CAAC,CAAC;IACtC,CAAC;IAED,MAAM,eAAe,GACjB,MAAM,CAAC,SAAS,KAAK,OAAO;QACxB,CAAC,CAAC,IAAI,sBAAW,CAAC;YACZ,UAAU,EAAE,MAAM;SACrB,CAAC;QACJ,CAAC,CAAC,IAAI,wCAAoB,CAAC;YACrB,UAAU,EAAE,MAAM;SACrB,CAAC,CAAC;IACb,MAAM,QAAQ,GAAG,GAAS,EAAE;QACxB,eAAe,CAAC,MAAM,CAAC,IAAI,CAAC;YACxB,EAAE,EAAE,iBAAK,CAAC,oBAAoB;YAC9B,OAAO,EAAE,QAAQ;YACjB,OAAO,EAAE,wBAAwB;SACpC,CAAC,CAAC;QAEH,eAAe;aACV,KAAK,EAAE;aACP,IAAI,CAAC,GAAG,EAAE;YACP,eAAe,CAAC,MAAM,CAAC,IAAI,CAAC;gBACxB,EAAE,EAAE,iBAAK,CAAC,YAAY;gBACtB,OAAO,EAAE,QAAQ;gBACjB,OAAO,EAAE,eAAe;aAC3B,CAAC,CAAC;YACH,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACpB,CAAC,CAAC;aACD,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;YACtB,eAAe,CAAC,MAAM,CAAC,KAAK,CAAC;gBACzB,EAAE,EAAE,iBAAK,CAAC,kBAAkB;gBAC5B,OAAO,EAAE,QAAQ;gBACjB,OAAO,EAAE,yBAAyB,KAAe,EAAE;aACtD,CAAC,CAAC;YACH,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACpB,CAAC,CAAC,CAAC;IACX,CAAC,CAAC;IAEF,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;IAC/B,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IAChC,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IAChC,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IAEhC,IAAI,CAAC;QACD,MAAM,eAAe,CAAC,KAAK,EAAE,CAAC;IAClC,CAAC;IAAC,OAAO,KAAc,EAAE,CAAC;QACtB,eAAe,CAAC,MAAM,CAAC,IAAI,CAAC;YACxB,EAAE,EAAE,iBAAK,CAAC,oBAAoB;YAC9B,OAAO,EAAE,QAAQ;YACjB,OAAO,EAAE,gCAAgC,KAAe,EAAE;YAC1D,WAAW,EAAE,IAAI;SACpB,CAAC,CAAC;QAEH,IAAI,CAAC;YACD,MAAM,eAAe,CAAC,KAAK,EAAE,CAAC;YAC9B,eAAe,CAAC,MAAM,CAAC,IAAI,CAAC;gBACxB,EAAE,EAAE,iBAAK,CAAC,YAAY;gBACtB,OAAO,EAAE,QAAQ;gBACjB,OAAO,EAAE,eAAe;aAC3B,CAAC,CAAC;QACP,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACtB,eAAe,CAAC,MAAM,CAAC,KAAK,CAAC;gBACzB,EAAE,EAAE,iBAAK,CAAC,kBAAkB;gBAC5B,OAAO,EAAE,QAAQ;gBACjB,OAAO,EAAE,yBAAyB,KAAe,EAAE;aACtD,CAAC,CAAC;QACP,CAAC;QACD,MAAM,KAAK,CAAC;IAChB,CAAC;AACL,CAAC;AAED,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;IAC5B,kFAAkF;IAClF,gFAAgF;IAChF,kBAAkB;IAClB,MAAM,MAAM,GAAG,IAAI,yBAAa,CAAC,sBAAQ,CAAC,IAAI,CAAC,CAAC;IAChD,MAAM,CAAC,SAAS,CAAC;QACb,EAAE,EAAE,iBAAK,CAAC,kBAAkB;QAC5B,OAAO,EAAE,QAAQ;QACjB,OAAO,EAAE,+BAA+B,KAAe,EAAE;KAC5D,CAAC,CAAC;IACH,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AACpB,CAAC,CAAC,CAAC;AAEH,SAAS,iBAAiB;IACtB,OAAO,CAAC,GAAG,CAAC,+CAA+C,CAAC,CAAC;IAC7D,OAAO,CAAC,GAAG,CAAC,iFAAiF,CAAC,CAAC;IAC/F,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AACpB,CAAC;AAED,SAAS,oBAAoB;IACzB,OAAO,CAAC,GAAG,CAAC,4BAAW,CAAC,OAAO,CAAC,CAAC;IACjC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AACpB,CAAC;AAEM,KAAK,UAAU,mBAAmB,CAAC,MAAkB;IACxD,IAAI,CAAC;QACD,MAAM,MAAM,GAAG,IAAI,mCAAgB,CAAC;YAChC,UAAU,EAAE,MAAM;YAClB,MAAM,EAAE;gBACJ,GAAG,CAAC,OAAO;oBACP,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBACzB,CAAC;gBACD,KAAK,CAAC,OAAO;oBACT,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;gBAC3B,CAAC;aACJ;SACJ,CAAC,CAAC;QACH,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;QACrB,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;QACrB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACpB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACb,OAAO,CAAC,KAAK,CAAC,+CAA+C,KAAe,EAAE,CAAC,CAAC;QAChF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACpB,CAAC;AACL,CAAC"}