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

package/dist/esm/common/search/embeddingsProvider.d.ts

@@ -0,0 +1,41 @@
+import type { UserConfig } from "../config.js";
+import { z } from "zod";
+type EmbeddingsInput = string;
+type Embeddings = number[];
+export type EmbeddingParameters = {
+    inputType: "query" | "document";
+};
+export interface EmbeddingsProvider<SupportedModels extends string, SupportedEmbeddingParameters extends EmbeddingParameters> {
+    embed(modelId: SupportedModels, content: EmbeddingsInput[], parameters: SupportedEmbeddingParameters): Promise<Embeddings[]>;
+}
+export declare const zVoyageModels: z.ZodDefault<z.ZodEnum<["voyage-3-large", "voyage-3.5", "voyage-3.5-lite", "voyage-code-3"]>>;
+export declare const zVoyageEmbeddingParameters: 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"]>>>;
+}, "strip", z.ZodTypeAny, {
+    outputDimension: 256 | 512 | 1024 | 2048 | 4096;
+    outputDType: "binary" | "float" | "int8" | "uint8" | "ubinary";
+}, {
+    outputDimension?: 256 | 512 | 1024 | 2048 | 4096 | undefined;
+    outputDType?: "binary" | "float" | "int8" | "uint8" | "ubinary" | undefined;
+}>;
+type VoyageModels = z.infer<typeof zVoyageModels>;
+type VoyageEmbeddingParameters = z.infer<typeof zVoyageEmbeddingParameters> & EmbeddingParameters;
+export declare function getEmbeddingsProvider(userConfig: UserConfig): EmbeddingsProvider<VoyageModels, VoyageEmbeddingParameters> | undefined;
+export declare const zSupportedEmbeddingParameters: 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;
+}>;
+export type SupportedEmbeddingParameters = z.infer<typeof zSupportedEmbeddingParameters>;
+export {};
+//# sourceMappingURL=embeddingsProvider.d.ts.map

package/dist/esm/common/search/embeddingsProvider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"embeddingsProvider.d.ts","sourceRoot":"","sources":["../../../../src/common/search/embeddingsProvider.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAG/C,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,KAAK,eAAe,GAAG,MAAM,CAAC;AAC9B,KAAK,UAAU,GAAG,MAAM,EAAE,CAAC;AAC3B,MAAM,MAAM,mBAAmB,GAAG;IAC9B,SAAS,EAAE,OAAO,GAAG,UAAU,CAAC;CACnC,CAAC;AAEF,MAAM,WAAW,kBAAkB,CAC/B,eAAe,SAAS,MAAM,EAC9B,4BAA4B,SAAS,mBAAmB;IAExD,KAAK,CACD,OAAO,EAAE,eAAe,EACxB,OAAO,EAAE,eAAe,EAAE,EAC1B,UAAU,EAAE,4BAA4B,GACzC,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC;CAC5B;AAED,eAAO,MAAM,aAAa,+FAEI,CAAC;AAE/B,eAAO,MAAM,0BAA0B;;;;;;;;;EAMrC,CAAC;AAEH,KAAK,YAAY,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,aAAa,CAAC,CAAC;AAClD,KAAK,yBAAyB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,0BAA0B,CAAC,GAAG,mBAAmB,CAAC;AAqClG,wBAAgB,qBAAqB,CACjC,UAAU,EAAE,UAAU,GACvB,kBAAkB,CAAC,YAAY,EAAE,yBAAyB,CAAC,GAAG,SAAS,CAMzE;AAED,eAAO,MAAM,6BAA6B;;;;;;;;;;;;;EAA8D,CAAC;AACzG,MAAM,MAAM,4BAA4B,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,6BAA6B,CAAC,CAAC"}

package/dist/esm/common/search/embeddingsProvider.js

@@ -0,0 +1,46 @@
+import { createVoyage } from "voyage-ai-provider";
+import { embedMany } from "ai";
+import assert from "assert";
+import { createFetch } from "@mongodb-js/devtools-proxy-support";
+import { z } from "zod";
+export const zVoyageModels = z
+    .enum(["voyage-3-large", "voyage-3.5", "voyage-3.5-lite", "voyage-code-3"])
+    .default("voyage-3-large");
+export const zVoyageEmbeddingParameters = z.object({
+    outputDimension: z
+        .union([z.literal(256), z.literal(512), z.literal(1024), z.literal(2048), z.literal(4096)])
+        .optional()
+        .default(1024),
+    outputDType: z.enum(["float", "int8", "uint8", "binary", "ubinary"]).optional().default("float"),
+});
+class VoyageEmbeddingsProvider {
+    constructor({ voyageApiKey }, providedFetch) {
+        assert(voyageApiKey, "The VoyageAI API Key does not exist. This is likely a bug.");
+        // We should always use, by default, any enterprise proxy that the user has configured.
+        // Direct requests to VoyageAI might get blocked by the network if they don't go through
+        // the provided proxy.
+        const customFetch = (providedFetch ??
+            createFetch({ useEnvironmentVariableProxies: true }));
+        this.voyage = createVoyage({ apiKey: voyageApiKey, fetch: customFetch });
+    }
+    static isConfiguredIn({ voyageApiKey }) {
+        return !!voyageApiKey;
+    }
+    async embed(modelId, content, parameters) {
+        const model = this.voyage.textEmbeddingModel(modelId);
+        const { embeddings } = await embedMany({
+            model,
+            values: content,
+            providerOptions: { voyage: parameters },
+        });
+        return embeddings;
+    }
+}
+export function getEmbeddingsProvider(userConfig) {
+    if (VoyageEmbeddingsProvider.isConfiguredIn(userConfig)) {
+        return new VoyageEmbeddingsProvider(userConfig);
+    }
+    return undefined;
+}
+export const zSupportedEmbeddingParameters = zVoyageEmbeddingParameters.extend({ model: zVoyageModels });
+//# sourceMappingURL=embeddingsProvider.js.map

package/dist/esm/common/search/embeddingsProvider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"embeddingsProvider.js","sourceRoot":"","sources":["../../../../src/common/search/embeddingsProvider.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAElD,OAAO,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAE/B,OAAO,MAAM,MAAM,QAAQ,CAAC;AAC5B,OAAO,EAAE,WAAW,EAAE,MAAM,oCAAoC,CAAC;AACjE,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAmBxB,MAAM,CAAC,MAAM,aAAa,GAAG,CAAC;KACzB,IAAI,CAAC,CAAC,gBAAgB,EAAE,YAAY,EAAE,iBAAiB,EAAE,eAAe,CAAC,CAAC;KAC1E,OAAO,CAAC,gBAAgB,CAAC,CAAC;AAE/B,MAAM,CAAC,MAAM,0BAA0B,GAAG,CAAC,CAAC,MAAM,CAAC;IAC/C,eAAe,EAAE,CAAC;SACb,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;SAC1F,QAAQ,EAAE;SACV,OAAO,CAAC,IAAI,CAAC;IAClB,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,SAAS,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC;CACnG,CAAC,CAAC;AAKH,MAAM,wBAAwB;IAG1B,YAAY,EAAE,YAAY,EAAc,EAAE,aAA4B;QAClE,MAAM,CAAC,YAAY,EAAE,4DAA4D,CAAC,CAAC;QAEnF,uFAAuF;QACvF,wFAAwF;QACxF,sBAAsB;QACtB,MAAM,WAAW,GAAiB,CAAC,aAAa;YAC5C,WAAW,CAAC,EAAE,6BAA6B,EAAE,IAAI,EAAE,CAAC,CAA4B,CAAC;QAErF,IAAI,CAAC,MAAM,GAAG,YAAY,CAAC,EAAE,MAAM,EAAE,YAAY,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;IAC7E,CAAC;IAED,MAAM,CAAC,cAAc,CAAC,EAAE,YAAY,EAAc;QAC9C,OAAO,CAAC,CAAC,YAAY,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,KAAK,CACP,OAAc,EACd,OAA0B,EAC1B,UAAqC;QAErC,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC;QACtD,MAAM,EAAE,UAAU,EAAE,GAAG,MAAM,SAAS,CAAC;YACnC,KAAK;YACL,MAAM,EAAE,OAAO;YACf,eAAe,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE;SAC1C,CAAC,CAAC;QAEH,OAAO,UAAU,CAAC;IACtB,CAAC;CACJ;AAED,MAAM,UAAU,qBAAqB,CACjC,UAAsB;IAEtB,IAAI,wBAAwB,CAAC,cAAc,CAAC,UAAU,CAAC,EAAE,CAAC;QACtD,OAAO,IAAI,wBAAwB,CAAC,UAAU,CAAC,CAAC;IACpD,CAAC;IAED,OAAO,SAAS,CAAC;AACrB,CAAC;AAED,MAAM,CAAC,MAAM,6BAA6B,GAAG,0BAA0B,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC,CAAC"}

package/dist/esm/common/search/vectorSearchEmbeddingsManager.d.ts

@@ -0,0 +1,58 @@
+import { type Document } from "bson";
+import type { UserConfig } from "../config.js";
+import type { ConnectionManager } from "../connectionManager.js";
+import z from "zod";
+import { getEmbeddingsProvider } from "./embeddingsProvider.js";
+import type { EmbeddingParameters, SupportedEmbeddingParameters } from "./embeddingsProvider.js";
+export declare const similarityEnum: z.ZodEnum<["cosine", "euclidean", "dotProduct"]>;
+export type Similarity = z.infer<typeof similarityEnum>;
+export declare const quantizationEnum: z.ZodEnum<["none", "scalar", "binary"]>;
+export type Quantization = z.infer<typeof quantizationEnum>;
+export type VectorFieldIndexDefinition = {
+    type: "vector";
+    path: string;
+    numDimensions: number;
+    quantization: Quantization;
+    similarity: Similarity;
+};
+export type VectorFieldValidationError = {
+    path: string;
+    expectedNumDimensions: number;
+    expectedQuantization: Quantization;
+    actualNumDimensions: number | "unknown";
+    actualQuantization: Quantization | "unknown";
+    error: "dimension-mismatch" | "quantization-mismatch" | "not-a-vector" | "not-numeric";
+};
+export type EmbeddingNamespace = `${string}.${string}`;
+export declare class VectorSearchEmbeddingsManager {
+    private readonly config;
+    private readonly connectionManager;
+    private readonly embeddings;
+    private readonly embeddingsProvider;
+    constructor(config: UserConfig, connectionManager: ConnectionManager, embeddings?: Map<EmbeddingNamespace, VectorFieldIndexDefinition[]>, embeddingsProvider?: typeof getEmbeddingsProvider);
+    cleanupEmbeddingsForNamespace({ database, collection }: {
+        database: string;
+        collection: string;
+    }): void;
+    embeddingsForNamespace({ database, collection, }: {
+        database: string;
+        collection: string;
+    }): Promise<VectorFieldIndexDefinition[]>;
+    findFieldsWithWrongEmbeddings({ database, collection, }: {
+        database: string;
+        collection: string;
+    }, document: Document): Promise<VectorFieldValidationError[]>;
+    private atlasSearchEnabledProvider;
+    private isVectorFieldIndexDefinition;
+    private getValidationErrorForDocument;
+    generateEmbeddings({ database, collection, path, rawValues, embeddingParameters, inputType, }: {
+        database: string;
+        collection: string;
+        path: string;
+        rawValues: string[];
+        embeddingParameters: SupportedEmbeddingParameters;
+        inputType: EmbeddingParameters["inputType"];
+    }): Promise<unknown[]>;
+    private isANumber;
+}
+//# sourceMappingURL=vectorSearchEmbeddingsManager.d.ts.map

package/dist/esm/common/search/vectorSearchEmbeddingsManager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"vectorSearchEmbeddingsManager.d.ts","sourceRoot":"","sources":["../../../../src/common/search/vectorSearchEmbeddingsManager.ts"],"names":[],"mappings":"AACA,OAAO,EAAQ,KAAK,QAAQ,EAAE,MAAM,MAAM,CAAC;AAC3C,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAC/C,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AACjE,OAAO,CAAC,MAAM,KAAK,CAAC;AAEpB,OAAO,EAAE,qBAAqB,EAAE,MAAM,yBAAyB,CAAC;AAChE,OAAO,KAAK,EAAE,mBAAmB,EAAE,4BAA4B,EAAE,MAAM,yBAAyB,CAAC;AAEjG,eAAO,MAAM,cAAc,kDAAgD,CAAC;AAC5E,MAAM,MAAM,UAAU,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,cAAc,CAAC,CAAC;AAExD,eAAO,MAAM,gBAAgB,yCAAuC,CAAC;AACrE,MAAM,MAAM,YAAY,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,gBAAgB,CAAC,CAAC;AAE5D,MAAM,MAAM,0BAA0B,GAAG;IACrC,IAAI,EAAE,QAAQ,CAAC;IACf,IAAI,EAAE,MAAM,CAAC;IACb,aAAa,EAAE,MAAM,CAAC;IACtB,YAAY,EAAE,YAAY,CAAC;IAC3B,UAAU,EAAE,UAAU,CAAC;CAC1B,CAAC;AAEF,MAAM,MAAM,0BAA0B,GAAG;IACrC,IAAI,EAAE,MAAM,CAAC;IACb,qBAAqB,EAAE,MAAM,CAAC;IAC9B,oBAAoB,EAAE,YAAY,CAAC;IACnC,mBAAmB,EAAE,MAAM,GAAG,SAAS,CAAC;IACxC,kBAAkB,EAAE,YAAY,GAAG,SAAS,CAAC;IAC7C,KAAK,EAAE,oBAAoB,GAAG,uBAAuB,GAAG,cAAc,GAAG,aAAa,CAAC;CAC1F,CAAC;AAEF,MAAM,MAAM,kBAAkB,GAAG,GAAG,MAAM,IAAI,MAAM,EAAE,CAAC;AACvD,qBAAa,6BAA6B;IAElC,OAAO,CAAC,QAAQ,CAAC,MAAM;IACvB,OAAO,CAAC,QAAQ,CAAC,iBAAiB;IAClC,OAAO,CAAC,QAAQ,CAAC,UAAU;IAC3B,OAAO,CAAC,QAAQ,CAAC,kBAAkB;gBAHlB,MAAM,EAAE,UAAU,EAClB,iBAAiB,EAAE,iBAAiB,EACpC,UAAU,GAAE,GAAG,CAAC,kBAAkB,EAAE,0BAA0B,EAAE,CAAa,EAC7E,kBAAkB,GAAE,OAAO,qBAA6C;IAO7F,6BAA6B,CAAC,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,UAAU,EAAE,MAAM,CAAA;KAAE,GAAG,IAAI;IAKjG,sBAAsB,CAAC,EACzB,QAAQ,EACR,UAAU,GACb,EAAE;QACC,QAAQ,EAAE,MAAM,CAAC;QACjB,UAAU,EAAE,MAAM,CAAC;KACtB,GAAG,OAAO,CAAC,0BAA0B,EAAE,CAAC;IA8BnC,6BAA6B,CAC/B,EACI,QAAQ,EACR,UAAU,GACb,EAAE;QACC,QAAQ,EAAE,MAAM,CAAC;QACjB,UAAU,EAAE,MAAM,CAAC;KACtB,EACD,QAAQ,EAAE,QAAQ,GACnB,OAAO,CAAC,0BAA0B,EAAE,CAAC;YAmB1B,0BAA0B;IASxC,OAAO,CAAC,4BAA4B;IAIpC,OAAO,CAAC,6BAA6B;IA+FxB,kBAAkB,CAAC,EAC5B,QAAQ,EACR,UAAU,EACV,IAAI,EACJ,SAAS,EACT,mBAAmB,EACnB,SAAS,GACZ,EAAE;QACC,QAAQ,EAAE,MAAM,CAAC;QACjB,UAAU,EAAE,MAAM,CAAC;QACnB,IAAI,EAAE,MAAM,CAAC;QACb,SAAS,EAAE,MAAM,EAAE,CAAC;QACpB,mBAAmB,EAAE,4BAA4B,CAAC;QAClD,SAAS,EAAE,mBAAmB,CAAC,WAAW,CAAC,CAAC;KAC/C,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;IAqCtB,OAAO,CAAC,SAAS;CAgBpB"}

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

@@ -0,0 +1,193 @@
+import { BSON } from "bson";
+import z from "zod";
+import { ErrorCodes, MongoDBError } from "../errors.js";
+import { getEmbeddingsProvider } from "./embeddingsProvider.js";
+export const similarityEnum = z.enum(["cosine", "euclidean", "dotProduct"]);
+export const quantizationEnum = z.enum(["none", "scalar", "binary"]);
+export class VectorSearchEmbeddingsManager {
+    constructor(config, connectionManager, embeddings = new Map(), embeddingsProvider = getEmbeddingsProvider) {
+        this.config = config;
+        this.connectionManager = connectionManager;
+        this.embeddings = embeddings;
+        this.embeddingsProvider = embeddingsProvider;
+        connectionManager.events.on("connection-close", () => {
+            this.embeddings.clear();
+        });
+    }
+    cleanupEmbeddingsForNamespace({ database, collection }) {
+        const embeddingDefKey = `${database}.${collection}`;
+        this.embeddings.delete(embeddingDefKey);
+    }
+    async embeddingsForNamespace({ database, collection, }) {
+        const provider = await this.atlasSearchEnabledProvider();
+        if (!provider) {
+            return [];
+        }
+        // We only need the embeddings for validation now, so don't query them if
+        // validation is disabled.
+        if (this.config.disableEmbeddingsValidation) {
+            return [];
+        }
+        const embeddingDefKey = `${database}.${collection}`;
+        const definition = this.embeddings.get(embeddingDefKey);
+        if (!definition) {
+            const allSearchIndexes = await provider.getSearchIndexes(database, collection);
+            const vectorSearchIndexes = allSearchIndexes.filter((index) => index.type === "vectorSearch");
+            const vectorFields = vectorSearchIndexes
+                // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
+                .flatMap((index) => index.latestDefinition?.fields ?? [])
+                .filter((field) => this.isVectorFieldIndexDefinition(field));
+            this.embeddings.set(embeddingDefKey, vectorFields);
+            return vectorFields;
+        }
+        return definition;
+    }
+    async findFieldsWithWrongEmbeddings({ database, collection, }, document) {
+        const provider = await this.atlasSearchEnabledProvider();
+        if (!provider) {
+            return [];
+        }
+        // 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) {
+            return [];
+        }
+        const embeddings = await this.embeddingsForNamespace({ database, collection });
+        return embeddings
+            .map((emb) => this.getValidationErrorForDocument(emb, document))
+            .filter((e) => e !== undefined);
+    }
+    async atlasSearchEnabledProvider() {
+        const connectionState = this.connectionManager.currentConnectionState;
+        if (connectionState.tag === "connected" && (await connectionState.isSearchSupported())) {
+            return connectionState.serviceProvider;
+        }
+        return null;
+    }
+    isVectorFieldIndexDefinition(doc) {
+        return doc["type"] === "vector";
+    }
+    getValidationErrorForDocument(definition, document) {
+        const fieldPath = definition.path.split(".");
+        let fieldRef = document;
+        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",
+        });
+        for (const field of fieldPath) {
+            if (fieldRef && typeof fieldRef === "object" && field in fieldRef) {
+                fieldRef = fieldRef[field];
+            }
+            else {
+                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.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;
+        }
+        return undefined;
+    }
+    async generateEmbeddings({ database, collection, path, rawValues, embeddingParameters, inputType, }) {
+        const provider = await this.atlasSearchEnabledProvider();
+        if (!provider) {
+            throw new MongoDBError(ErrorCodes.AtlasSearchNotSupported, "Atlas Search is not supported in this cluster.");
+        }
+        const embeddingsProvider = this.embeddingsProvider(this.config);
+        if (!embeddingsProvider) {
+            throw new MongoDBError(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 MongoDBError(ErrorCodes.AtlasVectorSearchIndexNotFound, `No Vector Search index found for path "${path}" in namespace "${database}.${collection}"`);
+        }
+        return await embeddingsProvider.embed(embeddingParameters.model, rawValues, {
+            inputType,
+            ...embeddingParameters,
+        });
+    }
+    isANumber(value) {
+        if (typeof value === "number") {
+            return true;
+        }
+        if (value instanceof BSON.Int32 ||
+            value instanceof BSON.Decimal128 ||
+            value instanceof BSON.Double ||
+            value instanceof BSON.Long) {
+            return true;
+        }
+        return false;
+    }
+}
+//# sourceMappingURL=vectorSearchEmbeddingsManager.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"vectorSearchEmbeddingsManager.js","sourceRoot":"","sources":["../../../../src/common/search/vectorSearchEmbeddingsManager.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,IAAI,EAAiB,MAAM,MAAM,CAAC;AAG3C,OAAO,CAAC,MAAM,KAAK,CAAC;AACpB,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AACxD,OAAO,EAAE,qBAAqB,EAAE,MAAM,yBAAyB,CAAC;AAGhE,MAAM,CAAC,MAAM,cAAc,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC,CAAC;AAG5E,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC;AAqBrE,MAAM,OAAO,6BAA6B;IACtC,YACqB,MAAkB,EAClB,iBAAoC,EACpC,aAAoE,IAAI,GAAG,EAAE,EAC7E,qBAAmD,qBAAqB;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,IAAI,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,YAAY,CAClB,UAAU,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,YAAY,CAAC,UAAU,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,YAAY,CAClB,UAAU,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,IAAI,CAAC,KAAK;YAC3B,KAAK,YAAY,IAAI,CAAC,UAAU;YAChC,KAAK,YAAY,IAAI,CAAC,MAAM;YAC5B,KAAK,YAAY,IAAI,CAAC,IAAI,EAC5B,CAAC;YACC,OAAO,IAAI,CAAC;QAChB,CAAC;QAED,OAAO,KAAK,CAAC;IACjB,CAAC;CACJ"}

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

@@ -5,7 +5,9 @@ import EventEmitter from "events";
 import type { AtlasClusterConnectionInfo, ConnectionManager, ConnectionSettings, ConnectionStateErrored } 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";
 export interface SessionOptions {
     apiBaseUrl: string;
     apiClientId?: string;
@@ -14,6 +16,8 @@ export interface SessionOptions {
     exportsManager: ExportsManager;
     connectionManager: ConnectionManager;
     keychain: Keychain;
+    atlasLocalClient?: Client;
+    vectorSearchEmbeddingsManager: VectorSearchEmbeddingsManager;
 }
 export type SessionEvents = {
     connect: [];
@@ -26,19 +30,23 @@ export declare class Session extends EventEmitter<SessionEvents> {
     readonly exportsManager: ExportsManager;
     readonly connectionManager: ConnectionManager;
     readonly apiClient: ApiClient;
+    readonly atlasLocalClient?: Client;
     readonly keychain: Keychain;
+    readonly vectorSearchEmbeddingsManager: VectorSearchEmbeddingsManager;
     mcpClient?: {
         name?: string;
         version?: string;
         title?: string;
     };
     logger: CompositeLogger;
-    constructor({ apiBaseUrl, apiClientId, apiClientSecret, logger, connectionManager, exportsManager, keychain, }: SessionOptions);
+    constructor({ apiBaseUrl, apiClientId, apiClientSecret, logger, connectionManager, exportsManager, keychain, atlasLocalClient, vectorSearchEmbeddingsManager, }: SessionOptions);
     setMcpClient(mcpClient: Implementation | undefined): void;
     disconnect(): Promise<void>;
     close(): Promise<void>;
     connectToMongoDB(settings: ConnectionSettings): Promise<void>;
     get isConnectedToMongoDB(): boolean;
+    isSearchSupported(): Promise<boolean>;
+    assertSearchSupported(): Promise<void>;
     get serviceProvider(): NodeDriverServiceProvider;
     get connectedAtlasCluster(): AtlasClusterConnectionInfo | undefined;
 }

package/dist/esm/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,QAAQ,EAAE,MAAM,eAAe,CAAC;AAE9C,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;CACtB;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,QAAQ,EAAE,QAAQ,CAAC;IAE5B,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,GACX,EAAE,cAAc;IAsBjB,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;IAED,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,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"}

package/dist/esm/common/session.js

@@ -4,7 +4,7 @@ import { LogId } from "./logger.js";
 import EventEmitter from "events";
 import { ErrorCodes, MongoDBError } from "./errors.js";
 export class Session extends EventEmitter {
-    constructor({ apiBaseUrl, apiClientId, apiClientSecret, logger, connectionManager, exportsManager, keychain, }) {
+    constructor({ apiBaseUrl, apiClientId, apiClientSecret, logger, connectionManager, exportsManager, keychain, atlasLocalClient, vectorSearchEmbeddingsManager, }) {
         super();
         this.sessionId = new ObjectId().toString();
         this.keychain = keychain;
@@ -16,8 +16,10 @@ export class Session extends EventEmitter {
             }
             : undefined;
         this.apiClient = new ApiClient({ baseUrl: apiBaseUrl, credentials }, logger);
+        this.atlasLocalClient = atlasLocalClient;
         this.exportsManager = exportsManager;
         this.connectionManager = connectionManager;
+        this.vectorSearchEmbeddingsManager = vectorSearchEmbeddingsManager;
         this.connectionManager.events.on("connection-success", () => this.emit("connect"));
         this.connectionManager.events.on("connection-time-out", (error) => this.emit("connection-error", error));
         this.connectionManager.events.on("connection-close", () => this.emit("disconnect"));
@@ -76,6 +78,19 @@ export class Session extends EventEmitter {
     get isConnectedToMongoDB() {
         return this.connectionManager.currentConnectionState.tag === "connected";
     }
+    async isSearchSupported() {
+        const state = this.connectionManager.currentConnectionState;
+        if (state.tag === "connected") {
+            return await state.isSearchSupported();
+        }
+        return false;
+    }
+    async assertSearchSupported() {
+        const isSearchSupported = await this.isSearchSupported();
+        if (!isSearchSupported) {
+            throw new MongoDBError(ErrorCodes.AtlasSearchNotSupported, "Atlas Search is not supported in the current cluster.");
+        }
+    }
     get serviceProvider() {
         if (this.isConnectedToMongoDB) {
             const state = this.connectionManager.currentConnectionState;

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

@@ -1 +1 @@
-{"version":3,"file":"session.js","sourceRoot":"","sources":["../../../src/common/session.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,MAAM,CAAC;AAEhC,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AAGjD,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AACpC,OAAO,YAAY,MAAM,QAAQ,CAAC;AASlC,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAqBvD,MAAM,OAAO,OAAQ,SAAQ,YAA2B;IAepD,YAAY,EACR,UAAU,EACV,WAAW,EACX,eAAe,EACf,MAAM,EACN,iBAAiB,EACjB,cAAc,EACd,QAAQ,GACK;QACb,KAAK,EAAE,CAAC;QAvBH,cAAS,GAAW,IAAI,QAAQ,EAAE,CAAC,QAAQ,EAAE,CAAC;QAyBnD,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,SAAS,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,WAAW,EAAE,EAAE,MAAM,CAAC,CAAC;QAC7E,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QACrC,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;QAC3C,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,KAAK,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,KAAK,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,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,YAAY,CAAC,UAAU,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"}
+{"version":3,"file":"session.js","sourceRoot":"","sources":["../../../src/common/session.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,MAAM,CAAC;AAEhC,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AAGjD,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AACpC,OAAO,YAAY,MAAM,QAAQ,CAAC;AASlC,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAyBvD,MAAM,OAAO,OAAQ,SAAQ,YAA2B;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,QAAQ,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,SAAS,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,KAAK,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,KAAK,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,YAAY,CAClB,UAAU,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,YAAY,CAAC,UAAU,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"}

package/dist/esm/resources/common/debug.d.ts

@@ -11,7 +11,7 @@ type ConnectionStateDebuggingInformation = {
 export declare class DebugResource extends ReactiveResource<ConnectionStateDebuggingInformation, readonly ["connect", "disconnect", "close", "connection-error"]> {
     constructor(session: Session, config: UserConfig, telemetry: Telemetry);
     reduce(eventName: "connect" | "disconnect" | "close" | "connection-error", event: ConnectionStateErrored | undefined): ConnectionStateDebuggingInformation;
-    toOutput(): string;
+    toOutput(): Promise<string>;
 }
 export {};
 //# sourceMappingURL=debug.d.ts.map

package/dist/esm/resources/common/debug.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"debug.d.ts","sourceRoot":"","sources":["../../../../src/resources/common/debug.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAClD,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,8BAA8B,CAAC;AAC9D,OAAO,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AACxD,OAAO,KAAK,EAAE,0BAA0B,EAAE,sBAAsB,EAAE,MAAM,mCAAmC,CAAC;AAE5G,KAAK,mCAAmC,GAAG;IACvC,QAAQ,CAAC,GAAG,EAAE,WAAW,GAAG,YAAY,GAAG,cAAc,GAAG,SAAS,CAAC;IACtE,QAAQ,CAAC,wBAAwB,CAAC,EAAE,OAAO,GAAG,MAAM,GAAG,UAAU,GAAG,gBAAgB,GAAG,kBAAkB,GAAG,OAAO,CAAC;IACpH,QAAQ,CAAC,WAAW,CAAC,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,qBAAqB,CAAC,EAAE,0BAA0B,CAAC;CAC/D,CAAC;AAEF,qBAAa,aAAc,SAAQ,gBAAgB,CAC/C,mCAAmC,EACnC,SAAS,CAAC,SAAS,EAAE,YAAY,EAAE,OAAO,EAAE,kBAAkB,CAAC,CAClE;gBACe,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,SAAS;IAmBtE,MAAM,CACF,SAAS,EAAE,SAAS,GAAG,YAAY,GAAG,OAAO,GAAG,kBAAkB,EAClE,KAAK,EAAE,sBAAsB,GAAG,SAAS,GAC1C,mCAAmC;IAoBtC,QAAQ,IAAI,MAAM;CA0BrB"}
+{"version":3,"file":"debug.d.ts","sourceRoot":"","sources":["../../../../src/resources/common/debug.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAClD,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,8BAA8B,CAAC;AAC9D,OAAO,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AACxD,OAAO,KAAK,EAAE,0BAA0B,EAAE,sBAAsB,EAAE,MAAM,mCAAmC,CAAC;AAE5G,KAAK,mCAAmC,GAAG;IACvC,QAAQ,CAAC,GAAG,EAAE,WAAW,GAAG,YAAY,GAAG,cAAc,GAAG,SAAS,CAAC;IACtE,QAAQ,CAAC,wBAAwB,CAAC,EAAE,OAAO,GAAG,MAAM,GAAG,UAAU,GAAG,gBAAgB,GAAG,kBAAkB,GAAG,OAAO,CAAC;IACpH,QAAQ,CAAC,WAAW,CAAC,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,qBAAqB,CAAC,EAAE,0BAA0B,CAAC;CAC/D,CAAC;AAEF,qBAAa,aAAc,SAAQ,gBAAgB,CAC/C,mCAAmC,EACnC,SAAS,CAAC,SAAS,EAAE,YAAY,EAAE,OAAO,EAAE,kBAAkB,CAAC,CAClE;gBACe,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,SAAS;IAmBtE,MAAM,CACF,SAAS,EAAE,SAAS,GAAG,YAAY,GAAG,OAAO,GAAG,kBAAkB,EAClE,KAAK,EAAE,sBAAsB,GAAG,SAAS,GAC1C,mCAAmC;IAoBhC,QAAQ,IAAI,OAAO,CAAC,MAAM,CAAC;CA4BpC"}

package/dist/esm/resources/common/debug.js

@@ -36,12 +36,14 @@ export class DebugResource extends ReactiveResource {
                 return { tag: "disconnected" };
         }
     }
-    toOutput() {
+    async toOutput() {
         let result = "";
         switch (this.current.tag) {
-            case "connected":
-                result += "The user is connected to the MongoDB cluster.";
+            case "connected": {
+                const searchIndexesSupported = await this.session.isSearchSupported();
+                result += `The user is connected to the MongoDB cluster${searchIndexesSupported ? " with support for search indexes" : " without any support for search indexes"}.`;
                 break;
+            }
             case "errored":
                 result += `The user is not connected to a MongoDB cluster because of an error.\n`;
                 if (this.current.connectedAtlasCluster) {

package/dist/esm/resources/common/debug.js.map

@@ -1 +1 @@
-{"version":3,"file":"debug.js","sourceRoot":"","sources":["../../../../src/resources/common/debug.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAYlD,MAAM,OAAO,aAAc,SAAQ,gBAGlC;IACG,YAAY,OAAgB,EAAE,MAAkB,EAAE,SAAoB;QAClE,KAAK,CAAC;YACF,qBAAqB,EAAE;gBACnB,IAAI,EAAE,eAAe;gBACrB,GAAG,EAAE,iBAAiB;gBACtB,MAAM,EAAE;oBACJ,WAAW,EACP,oHAAoH;iBAC3H;aACJ;YACD,OAAO,EAAE;gBACL,OAAO,EAAE,EAAE,GAAG,EAAE,cAAc,EAAE;gBAChC,MAAM,EAAE,CAAC,SAAS,EAAE,YAAY,EAAE,OAAO,EAAE,kBAAkB,CAAC;aACjE;YACD,OAAO;YACP,MAAM;YACN,SAAS;SACZ,CAAC,CAAC;IACP,CAAC;IACD,MAAM,CACF,SAAkE,EAClE,KAAyC;QAEzC,QAAQ,SAAS,EAAE,CAAC;YAChB,KAAK,SAAS;gBACV,OAAO,EAAE,GAAG,EAAE,WAAW,EAAE,CAAC;YAChC,KAAK,kBAAkB,CAAC,CAAC,CAAC;gBACtB,OAAO;oBACH,GAAG,EAAE,SAAS;oBACd,wBAAwB,EAAE,KAAK,EAAE,wBAAwB;oBACzD,qBAAqB,EAAE,KAAK,EAAE,qBAAqB;oBACnD,WAAW,EACP,KAAK,EAAE,WAAW;wBAClB,2IAA2I;iBAClJ,CAAC;YACN,CAAC;YACD,KAAK,YAAY,CAAC;YAClB,KAAK,OAAO;gBACR,OAAO,EAAE,GAAG,EAAE,cAAc,EAAE,CAAC;QACvC,CAAC;IACL,CAAC;IAED,QAAQ;QACJ,IAAI,MAAM,GAAG,EAAE,CAAC;QAEhB,QAAQ,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC;YACvB,KAAK,WAAW;gBACZ,MAAM,IAAI,+CAA+C,CAAC;gBAC1D,MAAM;YACV,KAAK,SAAS;gBACV,MAAM,IAAI,uEAAuE,CAAC;gBAClF,IAAI,IAAI,CAAC,OAAO,CAAC,qBAAqB,EAAE,CAAC;oBACrC,MAAM,IAAI,0CAA0C,IAAI,CAAC,OAAO,CAAC,qBAAqB,CAAC,WAAW,yBAAyB,IAAI,CAAC,OAAO,CAAC,qBAAqB,CAAC,SAAS,MAAM,CAAC;gBAClL,CAAC;gBAED,IAAI,IAAI,CAAC,OAAO,CAAC,wBAAwB,KAAK,SAAS,EAAE,CAAC;oBACtD,MAAM,IAAI,6CAA6C,IAAI,CAAC,OAAO,CAAC,wBAAwB,MAAM,CAAC;gBACvG,CAAC;gBACD,MAAM,IAAI,UAAU,IAAI,CAAC,OAAO,CAAC,WAAW,UAAU,CAAC;gBACvD,MAAM;YACV,KAAK,YAAY,CAAC;YAClB,KAAK,cAAc;gBACf,MAAM,IAAI,iDAAiD,CAAC;gBAC5D,MAAM;QACd,CAAC;QAED,OAAO,MAAM,CAAC;IAClB,CAAC;CACJ"}
+{"version":3,"file":"debug.js","sourceRoot":"","sources":["../../../../src/resources/common/debug.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAYlD,MAAM,OAAO,aAAc,SAAQ,gBAGlC;IACG,YAAY,OAAgB,EAAE,MAAkB,EAAE,SAAoB;QAClE,KAAK,CAAC;YACF,qBAAqB,EAAE;gBACnB,IAAI,EAAE,eAAe;gBACrB,GAAG,EAAE,iBAAiB;gBACtB,MAAM,EAAE;oBACJ,WAAW,EACP,oHAAoH;iBAC3H;aACJ;YACD,OAAO,EAAE;gBACL,OAAO,EAAE,EAAE,GAAG,EAAE,cAAc,EAAE;gBAChC,MAAM,EAAE,CAAC,SAAS,EAAE,YAAY,EAAE,OAAO,EAAE,kBAAkB,CAAC;aACjE;YACD,OAAO;YACP,MAAM;YACN,SAAS;SACZ,CAAC,CAAC;IACP,CAAC;IACD,MAAM,CACF,SAAkE,EAClE,KAAyC;QAEzC,QAAQ,SAAS,EAAE,CAAC;YAChB,KAAK,SAAS;gBACV,OAAO,EAAE,GAAG,EAAE,WAAW,EAAE,CAAC;YAChC,KAAK,kBAAkB,CAAC,CAAC,CAAC;gBACtB,OAAO;oBACH,GAAG,EAAE,SAAS;oBACd,wBAAwB,EAAE,KAAK,EAAE,wBAAwB;oBACzD,qBAAqB,EAAE,KAAK,EAAE,qBAAqB;oBACnD,WAAW,EACP,KAAK,EAAE,WAAW;wBAClB,2IAA2I;iBAClJ,CAAC;YACN,CAAC;YACD,KAAK,YAAY,CAAC;YAClB,KAAK,OAAO;gBACR,OAAO,EAAE,GAAG,EAAE,cAAc,EAAE,CAAC;QACvC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,QAAQ;QACV,IAAI,MAAM,GAAG,EAAE,CAAC;QAEhB,QAAQ,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC;YACvB,KAAK,WAAW,CAAC,CAAC,CAAC;gBACf,MAAM,sBAAsB,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,iBAAiB,EAAE,CAAC;gBACtE,MAAM,IAAI,+CAA+C,sBAAsB,CAAC,CAAC,CAAC,kCAAkC,CAAC,CAAC,CAAC,yCAAyC,GAAG,CAAC;gBACpK,MAAM;YACV,CAAC;YACD,KAAK,SAAS;gBACV,MAAM,IAAI,uEAAuE,CAAC;gBAClF,IAAI,IAAI,CAAC,OAAO,CAAC,qBAAqB,EAAE,CAAC;oBACrC,MAAM,IAAI,0CAA0C,IAAI,CAAC,OAAO,CAAC,qBAAqB,CAAC,WAAW,yBAAyB,IAAI,CAAC,OAAO,CAAC,qBAAqB,CAAC,SAAS,MAAM,CAAC;gBAClL,CAAC;gBAED,IAAI,IAAI,CAAC,OAAO,CAAC,wBAAwB,KAAK,SAAS,EAAE,CAAC;oBACtD,MAAM,IAAI,6CAA6C,IAAI,CAAC,OAAO,CAAC,wBAAwB,MAAM,CAAC;gBACvG,CAAC;gBACD,MAAM,IAAI,UAAU,IAAI,CAAC,OAAO,CAAC,WAAW,UAAU,CAAC;gBACvD,MAAM;YACV,KAAK,YAAY,CAAC;YAClB,KAAK,cAAc;gBACf,MAAM,IAAI,iDAAiD,CAAC;gBAC5D,MAAM;QACd,CAAC;QAED,OAAO,MAAM,CAAC;IAClB,CAAC;CACJ"}

package/dist/esm/resources/resource.d.ts

@@ -38,7 +38,7 @@ export declare abstract class ReactiveResource<Value, RelevantEvents extends rea
     private triggerUpdate;
     reduceApply(eventName: RelevantEvents[number], ...event: PayloadOf<RelevantEvents[number]>[]): void;
     protected abstract reduce(eventName: RelevantEvents[number], ...event: PayloadOf<RelevantEvents[number]>[]): Value;
-    abstract toOutput(): string;
+    abstract toOutput(): string | Promise<string>;
 }
 export {};
 //# sourceMappingURL=resource.d.ts.map

package/dist/esm/resources/resource.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"resource.d.ts","sourceRoot":"","sources":["../../../src/resources/resource.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAC3C,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,sBAAsB,CAAC;AACpD,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AACtD,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,2BAA2B,CAAC;AAC3D,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,KAAK,EAAwB,gBAAgB,EAAE,MAAM,yCAAyC,CAAC;AAGtG,KAAK,SAAS,CAAC,CAAC,SAAS,MAAM,aAAa,IAAI,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AAEpE,MAAM,MAAM,qBAAqB,GAAG;IAChC,IAAI,EAAE,MAAM,CAAC;IACb,GAAG,EAAE,MAAM,CAAC;IACZ,MAAM,EAAE,gBAAgB,CAAC;CAC5B,CAAC;AAEF,MAAM,MAAM,uBAAuB,CAAC,KAAK,EAAE,cAAc,SAAS,SAAS,CAAC,MAAM,aAAa,CAAC,EAAE,IAAI;IAClG,OAAO,EAAE,KAAK,CAAC;IACf,MAAM,EAAE,cAAc,CAAC;CAC1B,CAAC;AAEF,8BAAsB,gBAAgB,CAAC,KAAK,EAAE,cAAc,SAAS,SAAS,CAAC,MAAM,aAAa,CAAC,EAAE;IACjG,SAAS,CAAC,MAAM,CAAC,EAAE,MAAM,CAAC;IAC1B,SAAS,CAAC,OAAO,EAAE,OAAO,CAAC;IAC3B,SAAS,CAAC,MAAM,EAAE,UAAU,CAAC;IAC7B,SAAS,CAAC,SAAS,EAAE,SAAS,CAAC;IAE/B,SAAS,CAAC,OAAO,EAAE,KAAK,CAAC;IACzB,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IAChC,SAAS,CAAC,QAAQ,CAAC,GAAG,EAAE,MAAM,CAAC;IAC/B,SAAS,CAAC,QAAQ,CAAC,cAAc,EAAE,gBAAgB,CAAC;IACpD,SAAS,CAAC,QAAQ,CAAC,MAAM,EAAE,cAAc,CAAC;gBAE9B,EACR,qBAAqB,EACrB,OAAO,EACP,OAAO,EACP,MAAM,EACN,SAAS,EACT,OAAO,GACV,EAAE;QACC,qBAAqB,EAAE,qBAAqB,CAAC;QAC7C,OAAO,EAAE,uBAAuB,CAAC,KAAK,EAAE,cAAc,CAAC,CAAC;QACxD,OAAO,EAAE,OAAO,CAAC;QACjB,MAAM,EAAE,UAAU,CAAC;QACnB,SAAS,EAAE,SAAS,CAAC;QACrB,OAAO,CAAC,EAAE,KAAK,CAAC;KACnB;IAcD,OAAO,CAAC,mBAAmB;IASpB,QAAQ,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAKrC,OAAO,CAAC,gBAAgB,CAQrB;IAEH,OAAO,CAAC,aAAa;IAad,WAAW,CAAC,SAAS,EAAE,cAAc,CAAC,MAAM,CAAC,EAAE,GAAG,KAAK,EAAE,SAAS,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,EAAE,GAAG,IAAI;IAI1G,SAAS,CAAC,QAAQ,CAAC,MAAM,CAAC,SAAS,EAAE,cAAc,CAAC,MAAM,CAAC,EAAE,GAAG,KAAK,EAAE,SAAS,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,EAAE,GAAG,KAAK;aAClG,QAAQ,IAAI,MAAM;CACrC"}
+{"version":3,"file":"resource.d.ts","sourceRoot":"","sources":["../../../src/resources/resource.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAC3C,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,sBAAsB,CAAC;AACpD,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AACtD,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,2BAA2B,CAAC;AAC3D,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,KAAK,EAAwB,gBAAgB,EAAE,MAAM,yCAAyC,CAAC;AAGtG,KAAK,SAAS,CAAC,CAAC,SAAS,MAAM,aAAa,IAAI,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AAEpE,MAAM,MAAM,qBAAqB,GAAG;IAChC,IAAI,EAAE,MAAM,CAAC;IACb,GAAG,EAAE,MAAM,CAAC;IACZ,MAAM,EAAE,gBAAgB,CAAC;CAC5B,CAAC;AAEF,MAAM,MAAM,uBAAuB,CAAC,KAAK,EAAE,cAAc,SAAS,SAAS,CAAC,MAAM,aAAa,CAAC,EAAE,IAAI;IAClG,OAAO,EAAE,KAAK,CAAC;IACf,MAAM,EAAE,cAAc,CAAC;CAC1B,CAAC;AAEF,8BAAsB,gBAAgB,CAAC,KAAK,EAAE,cAAc,SAAS,SAAS,CAAC,MAAM,aAAa,CAAC,EAAE;IACjG,SAAS,CAAC,MAAM,CAAC,EAAE,MAAM,CAAC;IAC1B,SAAS,CAAC,OAAO,EAAE,OAAO,CAAC;IAC3B,SAAS,CAAC,MAAM,EAAE,UAAU,CAAC;IAC7B,SAAS,CAAC,SAAS,EAAE,SAAS,CAAC;IAE/B,SAAS,CAAC,OAAO,EAAE,KAAK,CAAC;IACzB,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IAChC,SAAS,CAAC,QAAQ,CAAC,GAAG,EAAE,MAAM,CAAC;IAC/B,SAAS,CAAC,QAAQ,CAAC,cAAc,EAAE,gBAAgB,CAAC;IACpD,SAAS,CAAC,QAAQ,CAAC,MAAM,EAAE,cAAc,CAAC;gBAE9B,EACR,qBAAqB,EACrB,OAAO,EACP,OAAO,EACP,MAAM,EACN,SAAS,EACT,OAAO,GACV,EAAE;QACC,qBAAqB,EAAE,qBAAqB,CAAC;QAC7C,OAAO,EAAE,uBAAuB,CAAC,KAAK,EAAE,cAAc,CAAC,CAAC;QACxD,OAAO,EAAE,OAAO,CAAC;QACjB,MAAM,EAAE,UAAU,CAAC;QACnB,SAAS,EAAE,SAAS,CAAC;QACrB,OAAO,CAAC,EAAE,KAAK,CAAC;KACnB;IAcD,OAAO,CAAC,mBAAmB;IASpB,QAAQ,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAKrC,OAAO,CAAC,gBAAgB,CAQrB;IAEH,OAAO,CAAC,aAAa;IAad,WAAW,CAAC,SAAS,EAAE,cAAc,CAAC,MAAM,CAAC,EAAE,GAAG,KAAK,EAAE,SAAS,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,EAAE,GAAG,IAAI;IAI1G,SAAS,CAAC,QAAQ,CAAC,MAAM,CAAC,SAAS,EAAE,cAAc,CAAC,MAAM,CAAC,EAAE,GAAG,KAAK,EAAE,SAAS,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,EAAE,GAAG,KAAK;aAClG,QAAQ,IAAI,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;CACvD"}

package/dist/esm/resources/resource.js

@@ -1,10 +1,10 @@
 import { LogId } from "../common/logger.js";
 export class ReactiveResource {
     constructor({ resourceConfiguration, options, session, config, telemetry, current, }) {
-        this.resourceCallback = (uri) => ({
+        this.resourceCallback = async (uri) => ({
             contents: [
                 {
-                    text: this.toOutput(),
+                    text: await this.toOutput(),
                     mimeType: "application/json",
                     uri: uri.href,
                 },

package/dist/esm/resources/resource.js.map

@@ -1 +1 @@
-{"version":3,"file":"resource.js","sourceRoot":"","sources":["../../../src/resources/resource.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,KAAK,EAAE,MAAM,qBAAqB,CAAC;AAe5C,MAAM,OAAgB,gBAAgB;IAYlC,YAAY,EACR,qBAAqB,EACrB,OAAO,EACP,OAAO,EACP,MAAM,EACN,SAAS,EACT,OAAO,GAQV;QA4BO,qBAAgB,GAAyB,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YACvD,QAAQ,EAAE;gBACN;oBACI,IAAI,EAAE,IAAI,CAAC,QAAQ,EAAE;oBACrB,QAAQ,EAAE,kBAAkB;oBAC5B,GAAG,EAAE,GAAG,CAAC,IAAI;iBAChB;aACJ;SACJ,CAAC,CAAC;QAnCC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAE3B,IAAI,CAAC,IAAI,GAAG,qBAAqB,CAAC,IAAI,CAAC;QACvC,IAAI,CAAC,GAAG,GAAG,qBAAqB,CAAC,GAAG,CAAC;QACrC,IAAI,CAAC,cAAc,GAAG,qBAAqB,CAAC,MAAM,CAAC;QACnD,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,IAAI,CAAC,OAAO,GAAG,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC;QAE1C,IAAI,CAAC,mBAAmB,EAAE,CAAC;IAC/B,CAAC;IAEO,mBAAmB;QACvB,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAC9B,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,GAAG,IAAiC,EAAE,EAAE;gBAC5D,IAAI,CAAC,WAAW,CAAC,KAAK,EAAG,IAAkB,CAAC,CAAC,CAA4B,CAAC,CAAC;gBAC3E,KAAK,IAAI,CAAC,aAAa,EAAE,CAAC;YAC9B,CAAC,CAAC,CAAC;QACP,CAAC;IACL,CAAC;IAEM,QAAQ,CAAC,MAAc;QAC1B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,gBAAgB,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC;IAC5G,CAAC;IAYO,aAAa;QACjB,IAAI,CAAC;YACD,IAAI,CAAC,MAAM,EAAE,uBAAuB,EAAE,CAAC;YACvC,IAAI,CAAC,MAAM,EAAE,mBAAmB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC/C,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACtB,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC;gBACxB,EAAE,EAAE,KAAK,CAAC,qBAAqB;gBAC/B,OAAO,EAAE,UAAU;gBACnB,OAAO,EAAE,sDAAsD,KAAe,EAAE;aACnF,CAAC,CAAC;QACP,CAAC;IACL,CAAC;IAEM,WAAW,CAAC,SAAiC,EAAE,GAAG,KAA0C;QAC/F,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,GAAG,KAAK,CAAC,CAAC;IACpD,CAAC;CAIJ"}
+{"version":3,"file":"resource.js","sourceRoot":"","sources":["../../../src/resources/resource.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,KAAK,EAAE,MAAM,qBAAqB,CAAC;AAe5C,MAAM,OAAgB,gBAAgB;IAYlC,YAAY,EACR,qBAAqB,EACrB,OAAO,EACP,OAAO,EACP,MAAM,EACN,SAAS,EACT,OAAO,GAQV;QA4BO,qBAAgB,GAAyB,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;YAC7D,QAAQ,EAAE;gBACN;oBACI,IAAI,EAAE,MAAM,IAAI,CAAC,QAAQ,EAAE;oBAC3B,QAAQ,EAAE,kBAAkB;oBAC5B,GAAG,EAAE,GAAG,CAAC,IAAI;iBAChB;aACJ;SACJ,CAAC,CAAC;QAnCC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAE3B,IAAI,CAAC,IAAI,GAAG,qBAAqB,CAAC,IAAI,CAAC;QACvC,IAAI,CAAC,GAAG,GAAG,qBAAqB,CAAC,GAAG,CAAC;QACrC,IAAI,CAAC,cAAc,GAAG,qBAAqB,CAAC,MAAM,CAAC;QACnD,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,IAAI,CAAC,OAAO,GAAG,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC;QAE1C,IAAI,CAAC,mBAAmB,EAAE,CAAC;IAC/B,CAAC;IAEO,mBAAmB;QACvB,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAC9B,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,GAAG,IAAiC,EAAE,EAAE;gBAC5D,IAAI,CAAC,WAAW,CAAC,KAAK,EAAG,IAAkB,CAAC,CAAC,CAA4B,CAAC,CAAC;gBAC3E,KAAK,IAAI,CAAC,aAAa,EAAE,CAAC;YAC9B,CAAC,CAAC,CAAC;QACP,CAAC;IACL,CAAC;IAEM,QAAQ,CAAC,MAAc;QAC1B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,gBAAgB,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC;IAC5G,CAAC;IAYO,aAAa;QACjB,IAAI,CAAC;YACD,IAAI,CAAC,MAAM,EAAE,uBAAuB,EAAE,CAAC;YACvC,IAAI,CAAC,MAAM,EAAE,mBAAmB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC/C,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACtB,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC;gBACxB,EAAE,EAAE,KAAK,CAAC,qBAAqB;gBAC/B,OAAO,EAAE,UAAU;gBACnB,OAAO,EAAE,sDAAsD,KAAe,EAAE;aACnF,CAAC,CAAC;QACP,CAAC;IACL,CAAC;IAEM,WAAW,CAAC,SAAiC,EAAE,GAAG,KAA0C;QAC/F,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,GAAG,KAAK,CAAC,CAAC;IACpD,CAAC;CAIJ"}

package/dist/esm/server.d.ts

@@ -4,7 +4,7 @@ import type { Transport } from "@modelcontextprotocol/sdk/shared/transport.js";
 import type { LogLevel } from "./common/logger.js";
 import type { Telemetry } from "./telemetry/telemetry.js";
 import type { UserConfig } from "./common/config.js";
-import type { ToolBase, ToolConstructorParams } from "./tools/tool.js";
+import type { ToolBase, ToolCategory, ToolConstructorParams } from "./tools/tool.js";
 import { type ConnectionErrorHandler } from "./common/connectionErrorHandler.js";
 import type { Elicitation } from "./elicitation.js";
 export interface ServerOptions {
@@ -33,6 +33,7 @@ export declare class Server {
     connect(transport: Transport): Promise<void>;
     close(): Promise<void>;
     sendResourceListChanged(): void;
+    isToolCategoryAvailable(name: ToolCategory): boolean;
     sendResourceUpdated(uri: string): void;
     private emitServerTelemetryEvent;
     private registerTools;

package/dist/esm/server.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../../src/server.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACzE,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,+CAA+C,CAAC;AAI/E,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AAEnD,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,0BAA0B,CAAC;AAC1D,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAWrD,OAAO,KAAK,EAAE,QAAQ,EAAE,qBAAqB,EAAE,MAAM,iBAAiB,CAAC;AAGvE,OAAO,EAAE,KAAK,sBAAsB,EAAE,MAAM,oCAAoC,CAAC;AACjF,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAEpD,MAAM,WAAW,aAAa;IAC1B,OAAO,EAAE,OAAO,CAAC;IACjB,UAAU,EAAE,UAAU,CAAC;IACvB,SAAS,EAAE,SAAS,CAAC;IACrB,SAAS,EAAE,SAAS,CAAC;IACrB,WAAW,EAAE,WAAW,CAAC;IACzB,sBAAsB,EAAE,sBAAsB,CAAC;IAC/C,gBAAgB,CAAC,EAAE,CAAC,KAAK,MAAM,EAAE,qBAAqB,KAAK,QAAQ,CAAC,EAAE,CAAC;CAC1E;AAED,qBAAa,MAAM;IACf,SAAgB,OAAO,EAAE,OAAO,CAAC;IACjC,SAAgB,SAAS,EAAE,SAAS,CAAC;IACrC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAY;IACtC,SAAgB,UAAU,EAAE,UAAU,CAAC;IACvC,SAAgB,WAAW,EAAE,WAAW,CAAC;IACzC,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAsD;IACvF,SAAgB,KAAK,EAAE,QAAQ,EAAE,CAAM;IACvC,SAAgB,sBAAsB,EAAE,sBAAsB,CAAC;IAE/D,OAAO,CAAC,YAAY,CAAqB;IAEzC,IAAW,WAAW,IAAI,QAAQ,CAEjC;IAED,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IACnC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAqB;gBAEvC,EACR,OAAO,EACP,SAAS,EACT,UAAU,EACV,SAAS,EACT,sBAAsB,EACtB,WAAW,EACX,gBAAgB,GACnB,EAAE,aAAa;IAWV,OAAO,CAAC,SAAS,EAAE,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC;IA4F5C,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAMrB,uBAAuB,IAAI,IAAI;IAI/B,mBAAmB,CAAC,GAAG,EAAE,MAAM,GAAG,IAAI;IAY7C,OAAO,CAAC,wBAAwB;IA8BhC,OAAO,CAAC,aAAa;IAcrB,OAAO,CAAC,iBAAiB;YAOX,cAAc;IAwC5B,OAAO,CAAC,eAAe;YAkBT,+BAA+B;CAqBhD"}
+{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../../src/server.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACzE,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,+CAA+C,CAAC;AAK/E,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AAEnD,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,0BAA0B,CAAC;AAC1D,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAWrD,OAAO,KAAK,EAAE,QAAQ,EAAE,YAAY,EAAE,qBAAqB,EAAE,MAAM,iBAAiB,CAAC;AAGrF,OAAO,EAAE,KAAK,sBAAsB,EAAE,MAAM,oCAAoC,CAAC;AACjF,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAEpD,MAAM,WAAW,aAAa;IAC1B,OAAO,EAAE,OAAO,CAAC;IACjB,UAAU,EAAE,UAAU,CAAC;IACvB,SAAS,EAAE,SAAS,CAAC;IACrB,SAAS,EAAE,SAAS,CAAC;IACrB,WAAW,EAAE,WAAW,CAAC;IACzB,sBAAsB,EAAE,sBAAsB,CAAC;IAC/C,gBAAgB,CAAC,EAAE,CAAC,KAAK,MAAM,EAAE,qBAAqB,KAAK,QAAQ,CAAC,EAAE,CAAC;CAC1E;AAED,qBAAa,MAAM;IACf,SAAgB,OAAO,EAAE,OAAO,CAAC;IACjC,SAAgB,SAAS,EAAE,SAAS,CAAC;IACrC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAY;IACtC,SAAgB,UAAU,EAAE,UAAU,CAAC;IACvC,SAAgB,WAAW,EAAE,WAAW,CAAC;IACzC,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAsD;IACvF,SAAgB,KAAK,EAAE,QAAQ,EAAE,CAAM;IACvC,SAAgB,sBAAsB,EAAE,sBAAsB,CAAC;IAE/D,OAAO,CAAC,YAAY,CAAqB;IAEzC,IAAW,WAAW,IAAI,QAAQ,CAEjC;IAED,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IACnC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAqB;gBAEvC,EACR,OAAO,EACP,SAAS,EACT,UAAU,EACV,SAAS,EACT,sBAAsB,EACtB,WAAW,EACX,gBAAgB,GACnB,EAAE,aAAa;IAWV,OAAO,CAAC,SAAS,EAAE,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC;IA4F5C,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAMrB,uBAAuB,IAAI,IAAI;IAI/B,uBAAuB,CAAC,IAAI,EAAE,YAAY,GAAG,OAAO;IAIpD,mBAAmB,CAAC,GAAG,EAAE,MAAM,GAAG,IAAI;IAY7C,OAAO,CAAC,wBAAwB;IA8BhC,OAAO,CAAC,aAAa;IAcrB,OAAO,CAAC,iBAAiB;YAOX,cAAc;IAwC5B,OAAO,CAAC,eAAe;YAkBT,+BAA+B;CAqBhD"}