mongodb-mcp-server 1.1.0 → 1.2.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +8 -0
- package/dist/cjs/common/atlas/cluster.d.ts +7 -2
- package/dist/cjs/common/atlas/cluster.d.ts.map +1 -1
- package/dist/cjs/common/atlas/cluster.js +18 -4
- package/dist/cjs/common/atlas/cluster.js.map +1 -1
- package/dist/cjs/common/atlasLocal.d.ts +4 -0
- package/dist/cjs/common/atlasLocal.d.ts.map +1 -0
- package/dist/cjs/common/atlasLocal.js +57 -0
- package/dist/cjs/common/atlasLocal.js.map +1 -0
- package/dist/cjs/common/config.d.ts +84 -15
- package/dist/cjs/common/config.d.ts.map +1 -1
- package/dist/cjs/common/config.js +153 -10
- package/dist/cjs/common/config.js.map +1 -1
- package/dist/cjs/common/connectionErrorHandler.d.ts.map +1 -1
- package/dist/cjs/common/connectionErrorHandler.js +14 -5
- package/dist/cjs/common/connectionErrorHandler.js.map +1 -1
- package/dist/cjs/common/connectionManager.d.ts.map +1 -1
- package/dist/cjs/common/connectionManager.js +5 -4
- package/dist/cjs/common/connectionManager.js.map +1 -1
- package/dist/cjs/common/errors.d.ts +5 -1
- package/dist/cjs/common/errors.d.ts.map +1 -1
- package/dist/cjs/common/errors.js +4 -0
- package/dist/cjs/common/errors.js.map +1 -1
- package/dist/cjs/common/packageInfo.js +1 -1
- package/dist/cjs/common/search/embeddingsProvider.d.ts +41 -0
- package/dist/cjs/common/search/embeddingsProvider.d.ts.map +1 -0
- package/dist/cjs/common/search/embeddingsProvider.js +53 -0
- package/dist/cjs/common/search/embeddingsProvider.js.map +1 -0
- package/dist/cjs/common/search/vectorSearchEmbeddingsManager.d.ts +58 -0
- package/dist/cjs/common/search/vectorSearchEmbeddingsManager.d.ts.map +1 -0
- package/dist/cjs/common/search/vectorSearchEmbeddingsManager.js +200 -0
- package/dist/cjs/common/search/vectorSearchEmbeddingsManager.js.map +1 -0
- package/dist/cjs/common/session.d.ts +8 -1
- package/dist/cjs/common/session.d.ts.map +1 -1
- package/dist/cjs/common/session.js +12 -4
- package/dist/cjs/common/session.js.map +1 -1
- package/dist/cjs/server.d.ts +2 -1
- package/dist/cjs/server.d.ts.map +1 -1
- package/dist/cjs/server.js +6 -2
- package/dist/cjs/server.js.map +1 -1
- package/dist/cjs/telemetry/types.d.ts +1 -0
- package/dist/cjs/telemetry/types.d.ts.map +1 -1
- package/dist/cjs/tools/args.d.ts +1 -0
- package/dist/cjs/tools/args.d.ts.map +1 -1
- package/dist/cjs/tools/args.js +1 -0
- package/dist/cjs/tools/args.js.map +1 -1
- package/dist/cjs/tools/atlas/atlasTool.d.ts +1 -1
- package/dist/cjs/tools/atlas/atlasTool.d.ts.map +1 -1
- package/dist/cjs/tools/atlas/atlasTool.js +1 -1
- package/dist/cjs/tools/atlas/atlasTool.js.map +1 -1
- package/dist/cjs/tools/atlas/connect/connectCluster.d.ts +3 -1
- package/dist/cjs/tools/atlas/connect/connectCluster.d.ts.map +1 -1
- package/dist/cjs/tools/atlas/connect/connectCluster.js +11 -6
- package/dist/cjs/tools/atlas/connect/connectCluster.js.map +1 -1
- package/dist/cjs/tools/atlas/create/createProject.d.ts +0 -4
- package/dist/cjs/tools/atlas/create/createProject.d.ts.map +1 -1
- package/dist/cjs/tools/atlas/create/createProject.js +3 -6
- package/dist/cjs/tools/atlas/create/createProject.js.map +1 -1
- package/dist/cjs/tools/atlas/read/inspectCluster.js +1 -1
- package/dist/cjs/tools/atlas/read/inspectCluster.js.map +1 -1
- package/dist/cjs/tools/atlas/read/listClusters.js +1 -1
- package/dist/cjs/tools/atlas/read/listClusters.js.map +1 -1
- package/dist/cjs/tools/atlas/read/listProjects.d.ts +0 -3
- package/dist/cjs/tools/atlas/read/listProjects.d.ts.map +1 -1
- package/dist/cjs/tools/atlas/read/listProjects.js +15 -20
- package/dist/cjs/tools/atlas/read/listProjects.js.map +1 -1
- package/dist/cjs/tools/atlasLocal/atlasLocalTool.d.ts +19 -0
- package/dist/cjs/tools/atlasLocal/atlasLocalTool.d.ts.map +1 -0
- package/dist/cjs/tools/atlasLocal/atlasLocalTool.js +109 -0
- package/dist/cjs/tools/atlasLocal/atlasLocalTool.js.map +1 -0
- package/dist/cjs/tools/atlasLocal/connect/connectDeployment.d.ts +14 -0
- package/dist/cjs/tools/atlasLocal/connect/connectDeployment.d.ts.map +1 -0
- package/dist/cjs/tools/atlasLocal/connect/connectDeployment.js +35 -0
- package/dist/cjs/tools/atlasLocal/connect/connectDeployment.js.map +1 -0
- package/dist/cjs/tools/atlasLocal/create/createDeployment.d.ts +14 -0
- package/dist/cjs/tools/atlasLocal/create/createDeployment.d.ts.map +1 -0
- package/dist/cjs/tools/atlasLocal/create/createDeployment.js +41 -0
- package/dist/cjs/tools/atlasLocal/create/createDeployment.js.map +1 -0
- package/dist/cjs/tools/atlasLocal/delete/deleteDeployment.d.ts +14 -0
- package/dist/cjs/tools/atlasLocal/delete/deleteDeployment.d.ts.map +1 -0
- package/dist/cjs/tools/atlasLocal/delete/deleteDeployment.js +32 -0
- package/dist/cjs/tools/atlasLocal/delete/deleteDeployment.js.map +1 -0
- package/dist/cjs/tools/atlasLocal/read/listDeployments.d.ts +13 -0
- package/dist/cjs/tools/atlasLocal/read/listDeployments.d.ts.map +1 -0
- package/dist/cjs/tools/atlasLocal/read/listDeployments.js +42 -0
- package/dist/cjs/tools/atlasLocal/read/listDeployments.js.map +1 -0
- package/dist/cjs/tools/atlasLocal/tools.d.ts +6 -0
- package/dist/cjs/tools/atlasLocal/tools.d.ts.map +1 -0
- package/dist/cjs/tools/atlasLocal/tools.js +9 -0
- package/dist/cjs/tools/atlasLocal/tools.js.map +1 -0
- package/dist/cjs/tools/mongodb/create/createIndex.d.ts +1 -1
- package/dist/cjs/tools/mongodb/create/createIndex.d.ts.map +1 -1
- package/dist/cjs/tools/mongodb/create/createIndex.js +6 -21
- package/dist/cjs/tools/mongodb/create/createIndex.js.map +1 -1
- package/dist/cjs/tools/mongodb/create/insertMany.d.ts +1 -1
- package/dist/cjs/tools/mongodb/create/insertMany.d.ts.map +1 -1
- package/dist/cjs/tools/mongodb/create/insertMany.js +14 -10
- package/dist/cjs/tools/mongodb/create/insertMany.js.map +1 -1
- package/dist/cjs/tools/mongodb/delete/dropIndex.d.ts +5 -2
- package/dist/cjs/tools/mongodb/delete/dropIndex.d.ts.map +1 -1
- package/dist/cjs/tools/mongodb/delete/dropIndex.js +36 -3
- package/dist/cjs/tools/mongodb/delete/dropIndex.js.map +1 -1
- package/dist/cjs/tools/mongodb/metadata/collectionIndexes.d.ts +15 -0
- package/dist/cjs/tools/mongodb/metadata/collectionIndexes.d.ts.map +1 -1
- package/dist/cjs/tools/mongodb/metadata/collectionIndexes.js +30 -5
- package/dist/cjs/tools/mongodb/metadata/collectionIndexes.js.map +1 -1
- package/dist/cjs/tools/mongodb/metadata/explain.d.ts +191 -9
- package/dist/cjs/tools/mongodb/metadata/explain.d.ts.map +1 -1
- package/dist/cjs/tools/mongodb/metadata/listDatabases.d.ts.map +1 -1
- package/dist/cjs/tools/mongodb/metadata/listDatabases.js +1 -3
- package/dist/cjs/tools/mongodb/metadata/listDatabases.js.map +1 -1
- package/dist/cjs/tools/mongodb/mongodbTool.d.ts +2 -1
- package/dist/cjs/tools/mongodb/mongodbTool.d.ts.map +1 -1
- package/dist/cjs/tools/mongodb/mongodbTool.js +19 -0
- package/dist/cjs/tools/mongodb/mongodbTool.js.map +1 -1
- package/dist/cjs/tools/mongodb/read/aggregate.d.ts +215 -2
- package/dist/cjs/tools/mongodb/read/aggregate.d.ts.map +1 -1
- package/dist/cjs/tools/mongodb/read/aggregate.js +82 -8
- package/dist/cjs/tools/mongodb/read/aggregate.js.map +1 -1
- package/dist/cjs/tools/mongodb/read/export.d.ts +191 -9
- package/dist/cjs/tools/mongodb/read/export.d.ts.map +1 -1
- package/dist/cjs/tools/mongodb/read/find.js +1 -1
- package/dist/cjs/tools/mongodb/read/find.js.map +1 -1
- package/dist/cjs/tools/mongodb/tools.d.ts +1 -2
- package/dist/cjs/tools/mongodb/tools.d.ts.map +1 -1
- package/dist/cjs/tools/mongodb/tools.js +0 -2
- package/dist/cjs/tools/mongodb/tools.js.map +1 -1
- package/dist/cjs/tools/tool.d.ts +5 -4
- package/dist/cjs/tools/tool.d.ts.map +1 -1
- package/dist/cjs/tools/tool.js +8 -5
- package/dist/cjs/tools/tool.js.map +1 -1
- package/dist/cjs/transports/base.d.ts +4 -1
- package/dist/cjs/transports/base.d.ts.map +1 -1
- package/dist/cjs/transports/base.js +6 -1
- package/dist/cjs/transports/base.js.map +1 -1
- package/dist/esm/common/atlas/cluster.d.ts +7 -2
- package/dist/esm/common/atlas/cluster.d.ts.map +1 -1
- package/dist/esm/common/atlas/cluster.js +17 -4
- package/dist/esm/common/atlas/cluster.js.map +1 -1
- package/dist/esm/common/atlasLocal.d.ts +4 -0
- package/dist/esm/common/atlasLocal.d.ts.map +1 -0
- package/dist/esm/common/atlasLocal.js +20 -0
- package/dist/esm/common/atlasLocal.js.map +1 -0
- package/dist/esm/common/config.d.ts +84 -15
- package/dist/esm/common/config.d.ts.map +1 -1
- package/dist/esm/common/config.js +112 -2
- package/dist/esm/common/config.js.map +1 -1
- package/dist/esm/common/connectionErrorHandler.d.ts.map +1 -1
- package/dist/esm/common/connectionErrorHandler.js +14 -5
- package/dist/esm/common/connectionErrorHandler.js.map +1 -1
- package/dist/esm/common/connectionManager.d.ts.map +1 -1
- package/dist/esm/common/connectionManager.js +5 -4
- package/dist/esm/common/connectionManager.js.map +1 -1
- package/dist/esm/common/errors.d.ts +5 -1
- package/dist/esm/common/errors.d.ts.map +1 -1
- package/dist/esm/common/errors.js +4 -0
- package/dist/esm/common/errors.js.map +1 -1
- package/dist/esm/common/packageInfo.js +1 -1
- package/dist/esm/common/search/embeddingsProvider.d.ts +41 -0
- package/dist/esm/common/search/embeddingsProvider.d.ts.map +1 -0
- package/dist/esm/common/search/embeddingsProvider.js +46 -0
- package/dist/esm/common/search/embeddingsProvider.js.map +1 -0
- package/dist/esm/common/search/vectorSearchEmbeddingsManager.d.ts +58 -0
- package/dist/esm/common/search/vectorSearchEmbeddingsManager.d.ts.map +1 -0
- package/dist/esm/common/search/vectorSearchEmbeddingsManager.js +193 -0
- package/dist/esm/common/search/vectorSearchEmbeddingsManager.js.map +1 -0
- package/dist/esm/common/session.d.ts +8 -1
- package/dist/esm/common/session.d.ts.map +1 -1
- package/dist/esm/common/session.js +12 -4
- package/dist/esm/common/session.js.map +1 -1
- package/dist/esm/server.d.ts +2 -1
- package/dist/esm/server.d.ts.map +1 -1
- package/dist/esm/server.js +5 -1
- package/dist/esm/server.js.map +1 -1
- package/dist/esm/telemetry/types.d.ts +1 -0
- package/dist/esm/telemetry/types.d.ts.map +1 -1
- package/dist/esm/tools/args.d.ts +1 -0
- package/dist/esm/tools/args.d.ts.map +1 -1
- package/dist/esm/tools/args.js +1 -0
- package/dist/esm/tools/args.js.map +1 -1
- package/dist/esm/tools/atlas/atlasTool.d.ts +1 -1
- package/dist/esm/tools/atlas/atlasTool.d.ts.map +1 -1
- package/dist/esm/tools/atlas/atlasTool.js +1 -1
- package/dist/esm/tools/atlas/atlasTool.js.map +1 -1
- package/dist/esm/tools/atlas/connect/connectCluster.d.ts +3 -1
- package/dist/esm/tools/atlas/connect/connectCluster.d.ts.map +1 -1
- package/dist/esm/tools/atlas/connect/connectCluster.js +12 -7
- package/dist/esm/tools/atlas/connect/connectCluster.js.map +1 -1
- package/dist/esm/tools/atlas/create/createProject.d.ts +0 -4
- package/dist/esm/tools/atlas/create/createProject.d.ts.map +1 -1
- package/dist/esm/tools/atlas/create/createProject.js +2 -5
- package/dist/esm/tools/atlas/create/createProject.js.map +1 -1
- package/dist/esm/tools/atlas/read/inspectCluster.js +1 -1
- package/dist/esm/tools/atlas/read/inspectCluster.js.map +1 -1
- package/dist/esm/tools/atlas/read/listClusters.js +1 -1
- package/dist/esm/tools/atlas/read/listClusters.js.map +1 -1
- package/dist/esm/tools/atlas/read/listProjects.d.ts +0 -3
- package/dist/esm/tools/atlas/read/listProjects.d.ts.map +1 -1
- package/dist/esm/tools/atlas/read/listProjects.js +14 -19
- package/dist/esm/tools/atlas/read/listProjects.js.map +1 -1
- package/dist/esm/tools/atlasLocal/atlasLocalTool.d.ts +19 -0
- package/dist/esm/tools/atlasLocal/atlasLocalTool.d.ts.map +1 -0
- package/dist/esm/tools/atlasLocal/atlasLocalTool.js +105 -0
- package/dist/esm/tools/atlasLocal/atlasLocalTool.js.map +1 -0
- package/dist/esm/tools/atlasLocal/connect/connectDeployment.d.ts +14 -0
- package/dist/esm/tools/atlasLocal/connect/connectDeployment.d.ts.map +1 -0
- package/dist/esm/tools/atlasLocal/connect/connectDeployment.js +31 -0
- package/dist/esm/tools/atlasLocal/connect/connectDeployment.js.map +1 -0
- package/dist/esm/tools/atlasLocal/create/createDeployment.d.ts +14 -0
- package/dist/esm/tools/atlasLocal/create/createDeployment.d.ts.map +1 -0
- package/dist/esm/tools/atlasLocal/create/createDeployment.js +37 -0
- package/dist/esm/tools/atlasLocal/create/createDeployment.js.map +1 -0
- package/dist/esm/tools/atlasLocal/delete/deleteDeployment.d.ts +14 -0
- package/dist/esm/tools/atlasLocal/delete/deleteDeployment.d.ts.map +1 -0
- package/dist/esm/tools/atlasLocal/delete/deleteDeployment.js +28 -0
- package/dist/esm/tools/atlasLocal/delete/deleteDeployment.js.map +1 -0
- package/dist/esm/tools/atlasLocal/read/listDeployments.d.ts +13 -0
- package/dist/esm/tools/atlasLocal/read/listDeployments.d.ts.map +1 -0
- package/dist/esm/tools/atlasLocal/read/listDeployments.js +38 -0
- package/dist/esm/tools/atlasLocal/read/listDeployments.js.map +1 -0
- package/dist/esm/tools/atlasLocal/tools.d.ts +6 -0
- package/dist/esm/tools/atlasLocal/tools.d.ts.map +1 -0
- package/dist/esm/tools/atlasLocal/tools.js +6 -0
- package/dist/esm/tools/atlasLocal/tools.js.map +1 -0
- package/dist/esm/tools/mongodb/create/createIndex.d.ts +1 -1
- package/dist/esm/tools/mongodb/create/createIndex.d.ts.map +1 -1
- package/dist/esm/tools/mongodb/create/createIndex.js +6 -21
- package/dist/esm/tools/mongodb/create/createIndex.js.map +1 -1
- package/dist/esm/tools/mongodb/create/insertMany.d.ts +1 -1
- package/dist/esm/tools/mongodb/create/insertMany.d.ts.map +1 -1
- package/dist/esm/tools/mongodb/create/insertMany.js +14 -10
- package/dist/esm/tools/mongodb/create/insertMany.js.map +1 -1
- package/dist/esm/tools/mongodb/delete/dropIndex.d.ts +5 -2
- package/dist/esm/tools/mongodb/delete/dropIndex.d.ts.map +1 -1
- package/dist/esm/tools/mongodb/delete/dropIndex.js +36 -3
- package/dist/esm/tools/mongodb/delete/dropIndex.js.map +1 -1
- package/dist/esm/tools/mongodb/metadata/collectionIndexes.d.ts +15 -0
- package/dist/esm/tools/mongodb/metadata/collectionIndexes.d.ts.map +1 -1
- package/dist/esm/tools/mongodb/metadata/collectionIndexes.js +30 -5
- package/dist/esm/tools/mongodb/metadata/collectionIndexes.js.map +1 -1
- package/dist/esm/tools/mongodb/metadata/explain.d.ts +191 -9
- package/dist/esm/tools/mongodb/metadata/explain.d.ts.map +1 -1
- package/dist/esm/tools/mongodb/metadata/listDatabases.d.ts.map +1 -1
- package/dist/esm/tools/mongodb/metadata/listDatabases.js +1 -3
- package/dist/esm/tools/mongodb/metadata/listDatabases.js.map +1 -1
- package/dist/esm/tools/mongodb/mongodbTool.d.ts +2 -1
- package/dist/esm/tools/mongodb/mongodbTool.d.ts.map +1 -1
- package/dist/esm/tools/mongodb/mongodbTool.js +19 -0
- package/dist/esm/tools/mongodb/mongodbTool.js.map +1 -1
- package/dist/esm/tools/mongodb/read/aggregate.d.ts +215 -2
- package/dist/esm/tools/mongodb/read/aggregate.d.ts.map +1 -1
- package/dist/esm/tools/mongodb/read/aggregate.js +82 -8
- package/dist/esm/tools/mongodb/read/aggregate.js.map +1 -1
- package/dist/esm/tools/mongodb/read/export.d.ts +191 -9
- package/dist/esm/tools/mongodb/read/export.d.ts.map +1 -1
- package/dist/esm/tools/mongodb/read/find.js +1 -1
- package/dist/esm/tools/mongodb/read/find.js.map +1 -1
- package/dist/esm/tools/mongodb/tools.d.ts +1 -2
- package/dist/esm/tools/mongodb/tools.d.ts.map +1 -1
- package/dist/esm/tools/mongodb/tools.js +0 -2
- package/dist/esm/tools/mongodb/tools.js.map +1 -1
- package/dist/esm/tools/tool.d.ts +5 -4
- package/dist/esm/tools/tool.d.ts.map +1 -1
- package/dist/esm/tools/tool.js +8 -5
- package/dist/esm/tools/tool.js.map +1 -1
- package/dist/esm/transports/base.d.ts +4 -1
- package/dist/esm/transports/base.d.ts.map +1 -1
- package/dist/esm/transports/base.js +6 -1
- package/dist/esm/transports/base.js.map +1 -1
- package/package.json +15 -11
- package/dist/cjs/tools/mongodb/search/listSearchIndexes.d.ts +0 -30
- package/dist/cjs/tools/mongodb/search/listSearchIndexes.d.ts.map +0 -1
- package/dist/cjs/tools/mongodb/search/listSearchIndexes.js +0 -65
- package/dist/cjs/tools/mongodb/search/listSearchIndexes.js.map +0 -1
- package/dist/esm/tools/mongodb/search/listSearchIndexes.d.ts +0 -30
- package/dist/esm/tools/mongodb/search/listSearchIndexes.d.ts.map +0 -1
- package/dist/esm/tools/mongodb/search/listSearchIndexes.js +0 -61
- package/dist/esm/tools/mongodb/search/listSearchIndexes.js.map +0 -1
|
@@ -6,12 +6,21 @@ const connectionErrorHandler = (error, { availableTools, connectionState }) => {
|
|
|
6
6
|
const connectTools = availableTools
|
|
7
7
|
.filter((t) => t.operationType === "connect")
|
|
8
8
|
.sort((a, b) => a.category.localeCompare(b.category)); // Sort Atlas tools before MongoDB tools
|
|
9
|
-
// Find
|
|
10
|
-
// Note: if we ever have multiple Atlas connect tools, we may want to refine this logic to select the most appropriate one.
|
|
9
|
+
// Find what Atlas connect tools are available and suggest when the LLM should to use each. If no Atlas tools are found, return a suggestion for the MongoDB connect tool.
|
|
11
10
|
const atlasConnectTool = connectTools?.find((t) => t.category === "atlas");
|
|
12
|
-
const
|
|
13
|
-
|
|
14
|
-
|
|
11
|
+
const atlasLocalConnectTool = connectTools?.find((t) => t.category === "atlas-local");
|
|
12
|
+
const llmConnectHint = (() => {
|
|
13
|
+
const hints = [];
|
|
14
|
+
if (atlasConnectTool) {
|
|
15
|
+
hints.push(`Note to LLM: prefer using the "${atlasConnectTool.name}" tool to connect to an Atlas cluster over using a connection string. Make sure to ask the user to specify a cluster name they want to connect to or ask them if they want to use the "list-clusters" tool to list all their clusters. Do not invent cluster names or connection strings unless the user has explicitly specified them. If they've previously connected to MongoDB using MCP, you can ask them if they want to reconnect using the same cluster/connection.`);
|
|
16
|
+
}
|
|
17
|
+
if (atlasLocalConnectTool) {
|
|
18
|
+
hints.push(`Note to LLM: For MongoDB Atlas Local deployments, ask the user to either provide a connection string, specify a deployment name, or use "atlas-local-list-deployments" to show available local deployments. If a deployment name is provided, prefer using the "${atlasLocalConnectTool.name}" tool. If a connection string is provided, prefer using the "connect" tool. Do not invent deployment names or connection strings unless the user has explicitly specified them. If they've previously connected to a MongoDB Atlas Local deployment using MCP, you can ask them if they want to reconnect using the same deployment.`);
|
|
19
|
+
}
|
|
20
|
+
return hints.length > 0
|
|
21
|
+
? hints.join("\n")
|
|
22
|
+
: "Note to LLM: do not invent connection strings and explicitly ask the user to provide one. If they have previously connected to MongoDB using MCP, you can ask them if they want to reconnect using the same connection string.";
|
|
23
|
+
})();
|
|
15
24
|
const connectToolsNames = connectTools?.map((t) => `"${t.name}"`).join(", ");
|
|
16
25
|
const additionalPromptForConnectivity = [];
|
|
17
26
|
if (connectionState.tag === "connecting" && connectionState.oidcConnectionType) {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"connectionErrorHandler.js","sourceRoot":"","sources":["../../../src/common/connectionErrorHandler.ts"],"names":[],"mappings":";;;AACA,2CAA4D;AAarD,MAAM,sBAAsB,GAA2B,CAAC,KAAK,EAAE,EAAE,cAAc,EAAE,eAAe,EAAE,EAAE,EAAE;IACzG,MAAM,YAAY,GAAG,cAAc;SAC9B,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa,KAAK,SAAS,CAAC;SAC5C,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,wCAAwC;IAEnG,
|
|
1
|
+
{"version":3,"file":"connectionErrorHandler.js","sourceRoot":"","sources":["../../../src/common/connectionErrorHandler.ts"],"names":[],"mappings":";;;AACA,2CAA4D;AAarD,MAAM,sBAAsB,GAA2B,CAAC,KAAK,EAAE,EAAE,cAAc,EAAE,eAAe,EAAE,EAAE,EAAE;IACzG,MAAM,YAAY,GAAG,cAAc;SAC9B,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa,KAAK,SAAS,CAAC;SAC5C,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,wCAAwC;IAEnG,0KAA0K;IAC1K,MAAM,gBAAgB,GAAG,YAAY,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,OAAO,CAAC,CAAC;IAC3E,MAAM,qBAAqB,GAAG,YAAY,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,aAAa,CAAC,CAAC;IAEtF,MAAM,cAAc,GAAG,CAAC,GAAW,EAAE;QACjC,MAAM,KAAK,GAAa,EAAE,CAAC;QAE3B,IAAI,gBAAgB,EAAE,CAAC;YACnB,KAAK,CAAC,IAAI,CACN,kCAAkC,gBAAgB,CAAC,IAAI,6cAA6c,CACvgB,CAAC;QACN,CAAC;QAED,IAAI,qBAAqB,EAAE,CAAC;YACxB,KAAK,CAAC,IAAI,CACN,mQAAmQ,qBAAqB,CAAC,IAAI,uUAAuU,CACvmB,CAAC;QACN,CAAC;QAED,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC;YACnB,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC;YAClB,CAAC,CAAC,gOAAgO,CAAC;IAC3O,CAAC,CAAC,EAAE,CAAC;IAEL,MAAM,iBAAiB,GAAG,YAAY,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC7E,MAAM,+BAA+B,GAAqC,EAAE,CAAC;IAE7E,IAAI,eAAe,CAAC,GAAG,KAAK,YAAY,IAAI,eAAe,CAAC,kBAAkB,EAAE,CAAC;QAC7E,+BAA+B,CAAC,IAAI,CAAC;YACjC,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,8DAA8D,eAAe,CAAC,YAAY,sDAAsD,eAAe,CAAC,YAAY,GAAG;SACxL,CAAC,CAAC;IACP,CAAC;SAAM,CAAC;QACJ,+BAA+B,CAAC,IAAI,CAAC;YACjC,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,iBAAiB;gBACnB,CAAC,CAAC,0CAA0C,iBAAiB,4GAA4G,cAAc,EAAE;gBACzL,CAAC,CAAC,iIAAiI;SAC1I,CAAC,CAAC;IACP,CAAC;IAED,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;QACjB,KAAK,sBAAU,CAAC,qBAAqB;YACjC,OAAO;gBACH,YAAY,EAAE,IAAI;gBAClB,MAAM,EAAE;oBACJ,OAAO,EAAE;wBACL;4BACI,IAAI,EAAE,MAAM;4BACZ,IAAI,EAAE,2EAA2E;yBACpF;wBACD,GAAG,+BAA+B;qBACrC;oBACD,OAAO,EAAE,IAAI;iBAChB;aACJ,CAAC;QACN,KAAK,sBAAU,CAAC,6BAA6B;YACzC,OAAO;gBACH,YAAY,EAAE,IAAI;gBAClB,MAAM,EAAE;oBACJ,OAAO,EAAE;wBACL;4BACI,IAAI,EAAE,MAAM;4BACZ,IAAI,EAAE,sIAAsI;yBAC/I;wBACD;4BACI,IAAI,EAAE,MAAM;4BACZ,IAAI,EAAE,YAAY;gCACd,CAAC,CAAC,0DAA0D,iBAAiB,sCAAsC,cAAc,EAAE;gCACnI,CAAC,CAAC,0FAA0F;yBACnG;qBACJ;oBACD,OAAO,EAAE,IAAI;iBAChB;aACJ,CAAC;QAEN;YACI,OAAO,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC;IACvC,CAAC;AACL,CAAC,CAAC;AApFW,QAAA,sBAAsB,0BAoFjC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"connectionManager.d.ts","sourceRoot":"","sources":["../../../src/common/connectionManager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AAGtC,OAAO,EAAE,yBAAyB,EAAE,MAAM,uCAAuC,CAAC;AAElF,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,wBAAwB,CAAC;AACvD,OAAO,EAA2C,KAAK,aAAa,EAAE,KAAK,UAAU,EAAE,MAAM,aAAa,CAAC;AAE3G,OAAO,EAAE,KAAK,UAAU,EAAS,MAAM,aAAa,CAAC;AAIrD,MAAM,WAAW,0BAA0B;IACvC,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,IAAI,CAAC;CACpB;AAED,MAAM,WAAW,kBAAkB;IAC/B,gBAAgB,EAAE,MAAM,CAAC;IACzB,KAAK,CAAC,EAAE,0BAA0B,CAAC;CACtC;AAED,KAAK,aAAa,GAAG,WAAW,GAAG,YAAY,GAAG,cAAc,GAAG,SAAS,CAAC;AAC7E,KAAK,sBAAsB,GAAG,gBAAgB,GAAG,kBAAkB,CAAC;AACpE,MAAM,MAAM,wBAAwB,GAAG,OAAO,GAAG,MAAM,GAAG,UAAU,GAAG,sBAAsB,GAAG,OAAO,CAAC;AAExG,MAAM,WAAW,eAAe;IAC5B,GAAG,EAAE,aAAa,CAAC;IACnB,wBAAwB,CAAC,EAAE,wBAAwB,CAAC;IACpD,qBAAqB,CAAC,EAAE,0BAA0B,CAAC;CACtD;
|
|
1
|
+
{"version":3,"file":"connectionManager.d.ts","sourceRoot":"","sources":["../../../src/common/connectionManager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AAGtC,OAAO,EAAE,yBAAyB,EAAE,MAAM,uCAAuC,CAAC;AAElF,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,wBAAwB,CAAC;AACvD,OAAO,EAA2C,KAAK,aAAa,EAAE,KAAK,UAAU,EAAE,MAAM,aAAa,CAAC;AAE3G,OAAO,EAAE,KAAK,UAAU,EAAS,MAAM,aAAa,CAAC;AAIrD,MAAM,WAAW,0BAA0B;IACvC,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,IAAI,CAAC;CACpB;AAED,MAAM,WAAW,kBAAkB;IAC/B,gBAAgB,EAAE,MAAM,CAAC;IACzB,KAAK,CAAC,EAAE,0BAA0B,CAAC;CACtC;AAED,KAAK,aAAa,GAAG,WAAW,GAAG,YAAY,GAAG,cAAc,GAAG,SAAS,CAAC;AAC7E,KAAK,sBAAsB,GAAG,gBAAgB,GAAG,kBAAkB,CAAC;AACpE,MAAM,MAAM,wBAAwB,GAAG,OAAO,GAAG,MAAM,GAAG,UAAU,GAAG,sBAAsB,GAAG,OAAO,CAAC;AAExG,MAAM,WAAW,eAAe;IAC5B,GAAG,EAAE,aAAa,CAAC;IACnB,wBAAwB,CAAC,EAAE,wBAAwB,CAAC;IACpD,qBAAqB,CAAC,EAAE,0BAA0B,CAAC;CACtD;AAGD,qBAAa,wBAAyB,YAAW,eAAe;IAIjD,eAAe,EAAE,yBAAyB;IAC1C,wBAAwB,CAAC,EAAE,wBAAwB;IACnD,qBAAqB,CAAC,EAAE,0BAA0B;IALtD,GAAG,EAAG,WAAW,CAAU;gBAGvB,eAAe,EAAE,yBAAyB,EAC1C,wBAAwB,CAAC,EAAE,wBAAwB,YAAA,EACnD,qBAAqB,CAAC,EAAE,0BAA0B,YAAA;IAG7D,OAAO,CAAC,kBAAkB,CAAC,CAAU;IAExB,iBAAiB,IAAI,OAAO,CAAC,OAAO,CAAC;CAgBrD;AAED,MAAM,WAAW,yBAA0B,SAAQ,eAAe;IAC9D,GAAG,EAAE,YAAY,CAAC;IAClB,eAAe,EAAE,OAAO,CAAC,yBAAyB,CAAC,CAAC;IACpD,kBAAkB,EAAE,sBAAsB,CAAC;IAC3C,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,YAAY,CAAC,EAAE,MAAM,CAAC;CACzB;AAED,MAAM,WAAW,2BAA4B,SAAQ,eAAe;IAChE,GAAG,EAAE,cAAc,CAAC;CACvB;AAED,MAAM,WAAW,sBAAuB,SAAQ,eAAe;IAC3D,GAAG,EAAE,SAAS,CAAC;IACf,WAAW,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,MAAM,kBAAkB,GACxB,wBAAwB,GACxB,yBAAyB,GACzB,2BAA2B,GAC3B,sBAAsB,CAAC;AAE7B,MAAM,WAAW,uBAAuB;IACpC,oBAAoB,EAAE,CAAC,kBAAkB,CAAC,CAAC;IAC3C,oBAAoB,EAAE,CAAC,wBAAwB,CAAC,CAAC;IACjD,qBAAqB,EAAE,CAAC,sBAAsB,CAAC,CAAC;IAChD,kBAAkB,EAAE,CAAC,2BAA2B,CAAC,CAAC;IAClD,kBAAkB,EAAE,CAAC,sBAAsB,CAAC,CAAC;IAC7C,KAAK,EAAE,CAAC,kBAAkB,CAAC,CAAC;CAC/B;AAED,8BAAsB,iBAAiB;IAC5B,UAAU,EAAE,MAAM,CAAC;IAC1B,SAAS,CAAC,QAAQ,CAAC,OAAO,EAAE,YAAY,CAAC,uBAAuB,CAAC,CAAC;IAClE,QAAQ,CAAC,MAAM,EAAE,IAAI,CAAC,YAAY,CAAC,uBAAuB,CAAC,EAAE,IAAI,GAAG,KAAK,GAAG,MAAM,CAAC,CAAC;IACpF,OAAO,CAAC,KAAK,CAAqB;;IAQlC,IAAI,sBAAsB,IAAI,kBAAkB,CAE/C;IAED,SAAS,CAAC,WAAW,CAAC,KAAK,SAAS,MAAM,uBAAuB,EAAE,KAAK,SAAS,uBAAuB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAC9G,KAAK,EAAE,KAAK,EACZ,QAAQ,EAAE,KAAK,GAChB,KAAK;IAQR,aAAa,CAAC,UAAU,EAAE,MAAM,GAAG,IAAI;IAIvC,QAAQ,CAAC,OAAO,CAAC,QAAQ,EAAE,kBAAkB,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAC3E,QAAQ,CAAC,UAAU,IAAI,OAAO,CAAC,2BAA2B,GAAG,sBAAsB,CAAC;IACpF,QAAQ,CAAC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAClC;AAED,qBAAa,oBAAqB,SAAQ,iBAAiB;IAKnD,OAAO,CAAC,UAAU;IAClB,OAAO,CAAC,aAAa;IACrB,OAAO,CAAC,MAAM;IANlB,OAAO,CAAC,QAAQ,CAAW;IAC3B,OAAO,CAAC,GAAG,CAAe;gBAGd,UAAU,EAAE,UAAU,EACtB,aAAa,EAAE,aAAa,EAC5B,MAAM,EAAE,UAAU,EAC1B,QAAQ,EAAE,QAAQ,EAClB,GAAG,CAAC,EAAE,YAAY;IAUP,OAAO,CAAC,QAAQ,EAAE,kBAAkB,GAAG,OAAO,CAAC,kBAAkB,CAAC;IA2FlE,UAAU,IAAI,OAAO,CAAC,2BAA2B,GAAG,sBAAsB,CAAC;IAwB3E,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAerC,OAAO,CAAC,gBAAgB;YASV,mBAAmB;IAsBjC,OAAO,CAAC,sBAAsB;IAqB9B,MAAM,CAAC,+BAA+B,CAClC,MAAM,EAAE,UAAU,EAClB,QAAQ,EAAE;QAAE,gBAAgB,EAAE,MAAM,CAAA;KAAE,GACvC,wBAAwB;YAiCb,qBAAqB;CAatC;AAED;;;;GAIG;AACH,MAAM,MAAM,0BAA0B,GAAG,CAAC,YAAY,EAAE;IACpD,MAAM,EAAE,UAAU,CAAC;IACnB,QAAQ,EAAE,QAAQ,CAAC;IACnB,UAAU,EAAE,UAAU,CAAC;CAC1B,KAAK,OAAO,CAAC,iBAAiB,CAAC,CAAC;AAEjC,eAAO,MAAM,0BAA0B,EAAE,0BAOxC,CAAC"}
|
|
@@ -10,6 +10,7 @@ const errors_js_1 = require("./errors.js");
|
|
|
10
10
|
const logger_js_1 = require("./logger.js");
|
|
11
11
|
const packageInfo_js_1 = require("./packageInfo.js");
|
|
12
12
|
const connectionOptions_js_1 = require("../helpers/connectionOptions.js");
|
|
13
|
+
const MCP_TEST_DATABASE = "#mongodb-mcp";
|
|
13
14
|
class ConnectionStateConnected {
|
|
14
15
|
constructor(serviceProvider, connectionStringAuthType, connectedAtlasCluster) {
|
|
15
16
|
this.serviceProvider = serviceProvider;
|
|
@@ -20,11 +21,11 @@ class ConnectionStateConnected {
|
|
|
20
21
|
async isSearchSupported() {
|
|
21
22
|
if (this._isSearchSupported === undefined) {
|
|
22
23
|
try {
|
|
23
|
-
const dummyDatabase = "test";
|
|
24
|
-
const dummyCollection = "test";
|
|
25
24
|
// If a cluster supports search indexes, the call below will succeed
|
|
26
|
-
// with a cursor otherwise will throw an Error
|
|
27
|
-
|
|
25
|
+
// with a cursor otherwise will throw an Error.
|
|
26
|
+
// the Search Index Management Service might not be ready yet, but
|
|
27
|
+
// we assume that the agent can retry in that situation.
|
|
28
|
+
await this.serviceProvider.getSearchIndexes(MCP_TEST_DATABASE, "test");
|
|
28
29
|
this._isSearchSupported = true;
|
|
29
30
|
}
|
|
30
31
|
catch {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"connectionManager.js","sourceRoot":"","sources":["../../../src/common/connectionManager.ts"],"names":[],"mappings":";;;AAAA,mCAAsC;AAEtC,iFAAiE;AACjE,wFAAkF;AAClF,oDAA6F;AAE7F,2CAA2G;AAC3G,2CAAuD;AACvD,2CAAqD;AACrD,qDAA+C;AAC/C,0EAAmG;AAwBnG,MAAa,wBAAwB;IAGjC,YACW,eAA0C,EAC1C,wBAAmD,EACnD,qBAAkD;QAFlD,oBAAe,GAAf,eAAe,CAA2B;QAC1C,6BAAwB,GAAxB,wBAAwB,CAA2B;QACnD,0BAAqB,GAArB,qBAAqB,CAA6B;QALtD,QAAG,GAAG,WAAoB,CAAC;IAM/B,CAAC;IAIG,KAAK,CAAC,iBAAiB;QAC1B,IAAI,IAAI,CAAC,kBAAkB,KAAK,SAAS,EAAE,CAAC;YACxC,IAAI,CAAC;gBACD,
|
|
1
|
+
{"version":3,"file":"connectionManager.js","sourceRoot":"","sources":["../../../src/common/connectionManager.ts"],"names":[],"mappings":";;;AAAA,mCAAsC;AAEtC,iFAAiE;AACjE,wFAAkF;AAClF,oDAA6F;AAE7F,2CAA2G;AAC3G,2CAAuD;AACvD,2CAAqD;AACrD,qDAA+C;AAC/C,0EAAmG;AAwBnG,MAAM,iBAAiB,GAAG,cAAc,CAAC;AACzC,MAAa,wBAAwB;IAGjC,YACW,eAA0C,EAC1C,wBAAmD,EACnD,qBAAkD;QAFlD,oBAAe,GAAf,eAAe,CAA2B;QAC1C,6BAAwB,GAAxB,wBAAwB,CAA2B;QACnD,0BAAqB,GAArB,qBAAqB,CAA6B;QALtD,QAAG,GAAG,WAAoB,CAAC;IAM/B,CAAC;IAIG,KAAK,CAAC,iBAAiB;QAC1B,IAAI,IAAI,CAAC,kBAAkB,KAAK,SAAS,EAAE,CAAC;YACxC,IAAI,CAAC;gBACD,oEAAoE;gBACpE,+CAA+C;gBAC/C,kEAAkE;gBAClE,wDAAwD;gBACxD,MAAM,IAAI,CAAC,eAAe,CAAC,gBAAgB,CAAC,iBAAiB,EAAE,MAAM,CAAC,CAAC;gBACvE,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC;YACnC,CAAC;YAAC,MAAM,CAAC;gBACL,IAAI,CAAC,kBAAkB,GAAG,KAAK,CAAC;YACpC,CAAC;QACL,CAAC;QAED,OAAO,IAAI,CAAC,kBAAkB,CAAC;IACnC,CAAC;CACJ;AA3BD,4DA2BC;AAkCD,MAAsB,iBAAiB;IAMnC;QACI,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;QAC5B,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,OAAO,GAAG,IAAI,qBAAY,EAA2B,CAAC;QACzE,IAAI,CAAC,KAAK,GAAG,EAAE,GAAG,EAAE,cAAc,EAAE,CAAC;IACzC,CAAC;IAED,IAAI,sBAAsB;QACtB,OAAO,IAAI,CAAC,KAAK,CAAC;IACtB,CAAC;IAES,WAAW,CACjB,KAAY,EACZ,QAAe;QAEf,IAAI,CAAC,KAAK,GAAG,QAAQ,CAAC;QACtB,4EAA4E;QAC5E,2BAA2B;QAC3B,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,GAAI,CAAC,QAAQ,CAAS,CAAC,CAAC;QACjD,OAAO,QAAQ,CAAC;IACpB,CAAC;IAED,aAAa,CAAC,UAAkB;QAC5B,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;IACjC,CAAC;CAKJ;AAlCD,8CAkCC;AAED,MAAa,oBAAqB,SAAQ,iBAAiB;IAIvD,YACY,UAAsB,EACtB,aAA4B,EAC5B,MAAkB,EAC1B,QAAkB,EAClB,GAAkB;QAElB,KAAK,EAAE,CAAC;QANA,eAAU,GAAV,UAAU,CAAY;QACtB,kBAAa,GAAb,aAAa,CAAe;QAC5B,WAAM,GAAN,MAAM,CAAY;QAK1B,IAAI,CAAC,GAAG,GAAG,GAAG,IAAI,IAAI,qBAAY,EAAE,CAAC;QACrC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,iCAAiC,EAAE,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QACjF,kEAAkE;QAClE,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,oCAAoC,EAAE,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QACvF,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC7B,CAAC;IAEQ,KAAK,CAAC,OAAO,CAAC,QAA4B;;QAC/C,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,oBAAoB,EAAE,IAAI,CAAC,sBAAsB,CAAC,CAAC;QAErE,IAAI,IAAI,CAAC,sBAAsB,CAAC,GAAG,KAAK,WAAW,IAAI,IAAI,CAAC,sBAAsB,CAAC,GAAG,KAAK,YAAY,EAAE,CAAC;YACtG,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAC5B,CAAC;QAED,IAAI,eAAmD,CAAC;QACxD,IAAI,cAA8B,CAAC;QACnC,IAAI,wBAAwB,GAA6B,OAAO,CAAC;QAEjE,IAAI,CAAC;YACD,QAAQ,GAAG,EAAE,GAAG,QAAQ,EAAE,CAAC;YAC3B,MAAM,iBAAiB,GAAsB;gBACzC,OAAO,EAAE,GAAG,4BAAW,CAAC,aAAa,IAAI,4BAAW,CAAC,OAAO,EAAE;gBAC9D,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,GAAG,EAAE;gBAC7B,UAAU,EAAE,IAAI,CAAC,UAAU;aAC9B,CAAC;YAEF,QAAQ,CAAC,gBAAgB,GAAG,MAAM,IAAA,+CAAwB,EAAC;gBACvD,gBAAgB,EAAE,QAAQ,CAAC,gBAAgB;gBAC3C,UAAU,EAAE,iBAAiB;aAChC,CAAC,CAAC;YAEH,cAAc,GAAG,IAAA,8CAAiC,EAAC;gBAC/C,GAAG,IAAI,CAAC,UAAU;gBAClB,GAAG,IAAI,CAAC,aAAa;gBACrB,mBAAmB,EAAE,QAAQ,CAAC,gBAAgB;aACjD,CAAC,CAAC;YAEH,IAAI,cAAc,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC;gBACpC,MAAA,cAAc,CAAC,aAAa,CAAC,IAAI,EAAC,YAAY,QAAZ,YAAY,GAAK,CAAC,WAAW,CAAC,EAAC;gBACjE,MAAA,cAAc,CAAC,aAAa,CAAC,IAAI,EAAC,gBAAgB,QAAhB,gBAAgB,GAAK,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,IAAI,CAAC,EAAC;YAClG,CAAC;YAED,MAAA,cAAc,CAAC,aAAa,EAAC,KAAK,QAAL,KAAK,GAAK,EAAE,6BAA6B,EAAE,IAAI,EAAE,EAAC;YAC/E,MAAA,cAAc,CAAC,aAAa,EAAC,gBAAgB,QAAhB,gBAAgB,GAAK,IAAI,EAAC;YAEvD,wBAAwB,GAAG,oBAAoB,CAAC,+BAA+B,CAC3E,IAAI,CAAC,UAAU,EACf,cAAc,CACjB,CAAC;YAEF,eAAe,GAAG,wDAAyB,CAAC,OAAO,CAC/C,cAAc,CAAC,gBAAgB,EAC/B;gBACI,eAAe,EAAE,mDAAmD;gBACpE,WAAW,EAAE,aAAa;gBAC1B,GAAG,cAAc,CAAC,aAAa;aAClC,EACD,SAAS,EACT,IAAI,CAAC,GAAG,CACX,CAAC;QACN,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACtB,MAAM,WAAW,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,KAAe,EAAE,CAAC;YAClF,IAAI,CAAC,WAAW,CAAC,kBAAkB,EAAE;gBACjC,GAAG,EAAE,SAAS;gBACd,WAAW;gBACX,wBAAwB;gBACxB,qBAAqB,EAAE,QAAQ,CAAC,KAAK;aACxC,CAAC,CAAC;YACH,MAAM,IAAI,wBAAY,CAAC,sBAAU,CAAC,6BAA6B,EAAE,WAAW,CAAC,CAAC;QAClF,CAAC;QAED,IAAI,CAAC;YACD,IAAI,wBAAwB,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC9C,OAAO,IAAI,CAAC,WAAW,CAAC,oBAAoB,EAAE;oBAC1C,GAAG,EAAE,YAAY;oBACjB,eAAe;oBACf,qBAAqB,EAAE,QAAQ,CAAC,KAAK;oBACrC,wBAAwB;oBACxB,kBAAkB,EAAE,wBAAkD;iBACzE,CAAC,CAAC;YACP,CAAC;YAED,OAAO,IAAI,CAAC,WAAW,CACnB,oBAAoB,EACpB,IAAI,wBAAwB,CAAC,MAAM,eAAe,EAAE,wBAAwB,EAAE,QAAQ,CAAC,KAAK,CAAC,CAChG,CAAC;QACN,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACtB,MAAM,WAAW,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,KAAe,EAAE,CAAC;YAClF,IAAI,CAAC,WAAW,CAAC,kBAAkB,EAAE;gBACjC,GAAG,EAAE,SAAS;gBACd,WAAW;gBACX,wBAAwB;gBACxB,qBAAqB,EAAE,QAAQ,CAAC,KAAK;aACxC,CAAC,CAAC;YACH,MAAM,IAAI,wBAAY,CAAC,sBAAU,CAAC,qBAAqB,EAAE,WAAW,CAAC,CAAC;QAC1E,CAAC;IACL,CAAC;IAEQ,KAAK,CAAC,UAAU;QACrB,IAAI,IAAI,CAAC,sBAAsB,CAAC,GAAG,KAAK,cAAc,IAAI,IAAI,CAAC,sBAAsB,CAAC,GAAG,KAAK,SAAS,EAAE,CAAC;YACtG,OAAO,IAAI,CAAC,sBAAsB,CAAC;QACvC,CAAC;QAED,IAAI,IAAI,CAAC,sBAAsB,CAAC,GAAG,KAAK,WAAW,IAAI,IAAI,CAAC,sBAAsB,CAAC,GAAG,KAAK,YAAY,EAAE,CAAC;YACtG,IAAI,CAAC;gBACD,IAAI,IAAI,CAAC,sBAAsB,CAAC,GAAG,KAAK,WAAW,EAAE,CAAC;oBAClD,MAAM,IAAI,CAAC,sBAAsB,CAAC,eAAe,EAAE,KAAK,EAAE,CAAC;gBAC/D,CAAC;gBACD,IAAI,IAAI,CAAC,sBAAsB,CAAC,GAAG,KAAK,YAAY,EAAE,CAAC;oBACnD,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,sBAAsB,CAAC,eAAe,CAAC;oBAC1E,MAAM,eAAe,CAAC,KAAK,EAAE,CAAC;gBAClC,CAAC;YACL,CAAC;oBAAS,CAAC;gBACP,IAAI,CAAC,WAAW,CAAC,kBAAkB,EAAE;oBACjC,GAAG,EAAE,cAAc;iBACtB,CAAC,CAAC;YACP,CAAC;QACL,CAAC;QAED,OAAO,EAAE,GAAG,EAAE,cAAc,EAAE,CAAC;IACnC,CAAC;IAEQ,KAAK,CAAC,KAAK;QAChB,IAAI,CAAC;YACD,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAC5B,CAAC;QAAC,OAAO,GAAY,EAAE,CAAC;YACpB,MAAM,KAAK,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;YAClE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;gBACd,EAAE,EAAE,iBAAK,CAAC,wBAAwB;gBAClC,OAAO,EAAE,mBAAmB;gBAC5B,OAAO,EAAE,yCAAyC,KAAK,CAAC,OAAO,EAAE;aACpE,CAAC,CAAC;QACP,CAAC;gBAAS,CAAC;YACP,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,sBAAsB,CAAC,CAAC;QAC5D,CAAC;IACL,CAAC;IAEO,gBAAgB,CAAC,KAAc;QACnC,IACI,IAAI,CAAC,sBAAsB,CAAC,GAAG,KAAK,YAAY;YAChD,IAAI,CAAC,sBAAsB,CAAC,wBAAwB,EAAE,UAAU,CAAC,MAAM,CAAC,EAC1E,CAAC;YACC,KAAK,IAAI,CAAC,qBAAqB,CAAC,KAAK,CAAC,CAAC;QAC3C,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,mBAAmB;QAC7B,IACI,IAAI,CAAC,sBAAsB,CAAC,GAAG,KAAK,YAAY;YAChD,IAAI,CAAC,sBAAsB,CAAC,wBAAwB,EAAE,UAAU,CAAC,MAAM,CAAC,EAC1E,CAAC;YACC,IAAI,CAAC,WAAW,CACZ,oBAAoB,EACpB,IAAI,wBAAwB,CACxB,MAAM,IAAI,CAAC,sBAAsB,CAAC,eAAe,EACjD,IAAI,CAAC,sBAAsB,CAAC,wBAAwB,EACpD,IAAI,CAAC,sBAAsB,CAAC,qBAAqB,CACpD,CACJ,CAAC;QACN,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;YACb,EAAE,EAAE,iBAAK,CAAC,QAAQ;YAClB,OAAO,EAAE,oCAAoC;YAC7C,OAAO,EAAE,6BAA6B;SACzC,CAAC,CAAC;IACP,CAAC;IAEO,sBAAsB,CAAC,QAAuD;QAClF,IACI,IAAI,CAAC,sBAAsB,CAAC,GAAG,KAAK,YAAY;YAChD,IAAI,CAAC,sBAAsB,CAAC,wBAAwB,EAAE,UAAU,CAAC,MAAM,CAAC,EAC1E,CAAC;YACC,IAAI,CAAC,WAAW,CAAC,oBAAoB,EAAE;gBACnC,GAAG,IAAI,CAAC,sBAAsB;gBAC9B,GAAG,EAAE,YAAY;gBACjB,wBAAwB,EAAE,kBAAkB;gBAC5C,YAAY,EAAE,QAAQ,CAAC,eAAe;gBACtC,YAAY,EAAE,QAAQ,CAAC,QAAQ;aAClC,CAAC,CAAC;QACP,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;YACb,EAAE,EAAE,iBAAK,CAAC,QAAQ;YAClB,OAAO,EAAE,wCAAwC;YACjD,OAAO,EAAE,iDAAiD;SAC7D,CAAC,CAAC;IACP,CAAC;IAED,MAAM,CAAC,+BAA+B,CAClC,MAAkB,EAClB,QAAsC;QAEtC,MAAM,UAAU,GAAG,IAAI,gDAAgB,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAC;QACnE,MAAM,YAAY,GAAG,UAAU,CAAC,iBAAiB,EAAsB,CAAC;QAExE,QAAQ,YAAY,CAAC,GAAG,CAAC,eAAe,CAAC,EAAE,CAAC;YACxC,KAAK,cAAc,CAAC,CAAC,CAAC;gBAClB,IAAI,MAAM,CAAC,SAAS,KAAK,OAAO,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;oBACjD,OAAO,gBAAgB,CAAC;gBAC5B,CAAC;gBAED,IAAI,MAAM,CAAC,SAAS,KAAK,MAAM,IAAI,MAAM,CAAC,QAAQ,KAAK,WAAW,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;oBACnF,OAAO,gBAAgB,CAAC;gBAC5B,CAAC;gBAED,OAAO,kBAAkB,CAAC;YAC9B,CAAC;YACD,KAAK,cAAc;gBACf,OAAO,OAAO,CAAC;YACnB,KAAK,QAAQ;gBACT,OAAO,UAAU,CAAC;YACtB,KAAK,OAAO;gBACR,IAAI,YAAY,CAAC,GAAG,CAAC,YAAY,CAAC,KAAK,WAAW,EAAE,CAAC;oBACjD,OAAO,MAAM,CAAC;gBAClB,CAAC;gBACD,OAAO,OAAO,CAAC;YACnB,uDAAuD;YACvD,YAAY;YACZ,KAAK,IAAI,CAAC;YACV;gBACI,OAAO,OAAO,CAAC;QACvB,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,qBAAqB,CAAC,KAAc;QAC9C,IAAI,CAAC;YACD,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAC5B,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACtB,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC;gBAChB,EAAE,EAAE,iBAAK,CAAC,QAAQ;gBAClB,OAAO,EAAE,uBAAuB;gBAChC,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC;aACzB,CAAC,CAAC;QACP,CAAC;gBAAS,CAAC;YACP,IAAI,CAAC,WAAW,CAAC,kBAAkB,EAAE,EAAE,GAAG,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACzF,CAAC;IACL,CAAC;CACJ;AA1PD,oDA0PC;AAaM,MAAM,0BAA0B,GAA+B,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE;IACvG,MAAM,aAAa,GAAG,IAAA,6BAAiB,EAAC;QACpC,MAAM,EAAE,UAAU;QAClB,QAAQ,EAAE,gCAAoB;KACjC,CAAC,CAAC;IAEH,OAAO,OAAO,CAAC,OAAO,CAAC,IAAI,oBAAoB,CAAC,UAAU,EAAE,aAAa,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC,CAAC;AAClG,CAAC,CAAC;AAPW,QAAA,0BAA0B,8BAOrC"}
|
|
@@ -2,7 +2,11 @@ export declare enum ErrorCodes {
|
|
|
2
2
|
NotConnectedToMongoDB = 1000000,
|
|
3
3
|
MisconfiguredConnectionString = 1000001,
|
|
4
4
|
ForbiddenCollscan = 1000002,
|
|
5
|
-
ForbiddenWriteOperation = 1000003
|
|
5
|
+
ForbiddenWriteOperation = 1000003,
|
|
6
|
+
AtlasSearchNotSupported = 1000004,
|
|
7
|
+
NoEmbeddingsProviderConfigured = 1000005,
|
|
8
|
+
AtlasVectorSearchIndexNotFound = 1000006,
|
|
9
|
+
AtlasVectorSearchInvalidQuery = 1000007
|
|
6
10
|
}
|
|
7
11
|
export declare class MongoDBError<ErrorCode extends ErrorCodes = ErrorCodes> extends Error {
|
|
8
12
|
code: ErrorCode;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"errors.d.ts","sourceRoot":"","sources":["../../../src/common/errors.ts"],"names":[],"mappings":"AAAA,oBAAY,UAAU;IAClB,qBAAqB,UAAY;IACjC,6BAA6B,UAAY;IACzC,iBAAiB,UAAY;IAC7B,uBAAuB,UAAY;
|
|
1
|
+
{"version":3,"file":"errors.d.ts","sourceRoot":"","sources":["../../../src/common/errors.ts"],"names":[],"mappings":"AAAA,oBAAY,UAAU;IAClB,qBAAqB,UAAY;IACjC,6BAA6B,UAAY;IACzC,iBAAiB,UAAY;IAC7B,uBAAuB,UAAY;IACnC,uBAAuB,UAAY;IACnC,8BAA8B,UAAY;IAC1C,8BAA8B,UAAY;IAC1C,6BAA6B,UAAY;CAC5C;AAED,qBAAa,YAAY,CAAC,SAAS,SAAS,UAAU,GAAG,UAAU,CAAE,SAAQ,KAAK;IAEnE,IAAI,EAAE,SAAS;gBAAf,IAAI,EAAE,SAAS,EACtB,OAAO,EAAE,MAAM;CAItB"}
|
|
@@ -7,6 +7,10 @@ var ErrorCodes;
|
|
|
7
7
|
ErrorCodes[ErrorCodes["MisconfiguredConnectionString"] = 1000001] = "MisconfiguredConnectionString";
|
|
8
8
|
ErrorCodes[ErrorCodes["ForbiddenCollscan"] = 1000002] = "ForbiddenCollscan";
|
|
9
9
|
ErrorCodes[ErrorCodes["ForbiddenWriteOperation"] = 1000003] = "ForbiddenWriteOperation";
|
|
10
|
+
ErrorCodes[ErrorCodes["AtlasSearchNotSupported"] = 1000004] = "AtlasSearchNotSupported";
|
|
11
|
+
ErrorCodes[ErrorCodes["NoEmbeddingsProviderConfigured"] = 1000005] = "NoEmbeddingsProviderConfigured";
|
|
12
|
+
ErrorCodes[ErrorCodes["AtlasVectorSearchIndexNotFound"] = 1000006] = "AtlasVectorSearchIndexNotFound";
|
|
13
|
+
ErrorCodes[ErrorCodes["AtlasVectorSearchInvalidQuery"] = 1000007] = "AtlasVectorSearchInvalidQuery";
|
|
10
14
|
})(ErrorCodes || (exports.ErrorCodes = ErrorCodes = {}));
|
|
11
15
|
class MongoDBError extends Error {
|
|
12
16
|
constructor(code, message) {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"errors.js","sourceRoot":"","sources":["../../../src/common/errors.ts"],"names":[],"mappings":";;;AAAA,IAAY,
|
|
1
|
+
{"version":3,"file":"errors.js","sourceRoot":"","sources":["../../../src/common/errors.ts"],"names":[],"mappings":";;;AAAA,IAAY,UASX;AATD,WAAY,UAAU;IAClB,mFAAiC,CAAA;IACjC,mGAAyC,CAAA;IACzC,2EAA6B,CAAA;IAC7B,uFAAmC,CAAA;IACnC,uFAAmC,CAAA;IACnC,qGAA0C,CAAA;IAC1C,qGAA0C,CAAA;IAC1C,mGAAyC,CAAA;AAC7C,CAAC,EATW,UAAU,0BAAV,UAAU,QASrB;AAED,MAAa,YAAwD,SAAQ,KAAK;IAC9E,YACW,IAAe,EACtB,OAAe;QAEf,KAAK,CAAC,OAAO,CAAC,CAAC;QAHR,SAAI,GAAJ,IAAI,CAAW;IAI1B,CAAC;CACJ;AAPD,oCAOC"}
|
|
@@ -3,7 +3,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
|
|
|
3
3
|
exports.packageInfo = void 0;
|
|
4
4
|
// This file was generated by scripts/updatePackageVersion.ts - Do not edit it manually.
|
|
5
5
|
exports.packageInfo = {
|
|
6
|
-
version: "1.
|
|
6
|
+
version: "1.2.0",
|
|
7
7
|
mcpServerName: "MongoDB MCP Server",
|
|
8
8
|
};
|
|
9
9
|
//# sourceMappingURL=packageInfo.js.map
|
|
@@ -0,0 +1,41 @@
|
|
|
1
|
+
import type { UserConfig } from "../config.js";
|
|
2
|
+
import { z } from "zod";
|
|
3
|
+
type EmbeddingsInput = string;
|
|
4
|
+
type Embeddings = number[];
|
|
5
|
+
export type EmbeddingParameters = {
|
|
6
|
+
inputType: "query" | "document";
|
|
7
|
+
};
|
|
8
|
+
export interface EmbeddingsProvider<SupportedModels extends string, SupportedEmbeddingParameters extends EmbeddingParameters> {
|
|
9
|
+
embed(modelId: SupportedModels, content: EmbeddingsInput[], parameters: SupportedEmbeddingParameters): Promise<Embeddings[]>;
|
|
10
|
+
}
|
|
11
|
+
export declare const zVoyageModels: z.ZodDefault<z.ZodEnum<["voyage-3-large", "voyage-3.5", "voyage-3.5-lite", "voyage-code-3"]>>;
|
|
12
|
+
export declare const zVoyageEmbeddingParameters: z.ZodObject<{
|
|
13
|
+
outputDimension: z.ZodDefault<z.ZodOptional<z.ZodUnion<[z.ZodLiteral<256>, z.ZodLiteral<512>, z.ZodLiteral<1024>, z.ZodLiteral<2048>, z.ZodLiteral<4096>]>>>;
|
|
14
|
+
outputDType: z.ZodDefault<z.ZodOptional<z.ZodEnum<["float", "int8", "uint8", "binary", "ubinary"]>>>;
|
|
15
|
+
}, "strip", z.ZodTypeAny, {
|
|
16
|
+
outputDimension: 256 | 512 | 1024 | 2048 | 4096;
|
|
17
|
+
outputDType: "binary" | "float" | "int8" | "uint8" | "ubinary";
|
|
18
|
+
}, {
|
|
19
|
+
outputDimension?: 256 | 512 | 1024 | 2048 | 4096 | undefined;
|
|
20
|
+
outputDType?: "binary" | "float" | "int8" | "uint8" | "ubinary" | undefined;
|
|
21
|
+
}>;
|
|
22
|
+
type VoyageModels = z.infer<typeof zVoyageModels>;
|
|
23
|
+
type VoyageEmbeddingParameters = z.infer<typeof zVoyageEmbeddingParameters> & EmbeddingParameters;
|
|
24
|
+
export declare function getEmbeddingsProvider(userConfig: UserConfig): EmbeddingsProvider<VoyageModels, VoyageEmbeddingParameters> | undefined;
|
|
25
|
+
export declare const zSupportedEmbeddingParameters: z.ZodObject<{
|
|
26
|
+
outputDimension: z.ZodDefault<z.ZodOptional<z.ZodUnion<[z.ZodLiteral<256>, z.ZodLiteral<512>, z.ZodLiteral<1024>, z.ZodLiteral<2048>, z.ZodLiteral<4096>]>>>;
|
|
27
|
+
outputDType: z.ZodDefault<z.ZodOptional<z.ZodEnum<["float", "int8", "uint8", "binary", "ubinary"]>>>;
|
|
28
|
+
} & {
|
|
29
|
+
model: z.ZodDefault<z.ZodEnum<["voyage-3-large", "voyage-3.5", "voyage-3.5-lite", "voyage-code-3"]>>;
|
|
30
|
+
}, "strip", z.ZodTypeAny, {
|
|
31
|
+
outputDimension: 256 | 512 | 1024 | 2048 | 4096;
|
|
32
|
+
outputDType: "binary" | "float" | "int8" | "uint8" | "ubinary";
|
|
33
|
+
model: "voyage-3-large" | "voyage-3.5" | "voyage-3.5-lite" | "voyage-code-3";
|
|
34
|
+
}, {
|
|
35
|
+
outputDimension?: 256 | 512 | 1024 | 2048 | 4096 | undefined;
|
|
36
|
+
outputDType?: "binary" | "float" | "int8" | "uint8" | "ubinary" | undefined;
|
|
37
|
+
model?: "voyage-3-large" | "voyage-3.5" | "voyage-3.5-lite" | "voyage-code-3" | undefined;
|
|
38
|
+
}>;
|
|
39
|
+
export type SupportedEmbeddingParameters = z.infer<typeof zSupportedEmbeddingParameters>;
|
|
40
|
+
export {};
|
|
41
|
+
//# sourceMappingURL=embeddingsProvider.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
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"}
|
|
@@ -0,0 +1,53 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
3
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
4
|
+
};
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
exports.zSupportedEmbeddingParameters = exports.zVoyageEmbeddingParameters = exports.zVoyageModels = void 0;
|
|
7
|
+
exports.getEmbeddingsProvider = getEmbeddingsProvider;
|
|
8
|
+
const voyage_ai_provider_1 = require("voyage-ai-provider");
|
|
9
|
+
const ai_1 = require("ai");
|
|
10
|
+
const assert_1 = __importDefault(require("assert"));
|
|
11
|
+
const devtools_proxy_support_1 = require("@mongodb-js/devtools-proxy-support");
|
|
12
|
+
const zod_1 = require("zod");
|
|
13
|
+
exports.zVoyageModels = zod_1.z
|
|
14
|
+
.enum(["voyage-3-large", "voyage-3.5", "voyage-3.5-lite", "voyage-code-3"])
|
|
15
|
+
.default("voyage-3-large");
|
|
16
|
+
exports.zVoyageEmbeddingParameters = zod_1.z.object({
|
|
17
|
+
outputDimension: zod_1.z
|
|
18
|
+
.union([zod_1.z.literal(256), zod_1.z.literal(512), zod_1.z.literal(1024), zod_1.z.literal(2048), zod_1.z.literal(4096)])
|
|
19
|
+
.optional()
|
|
20
|
+
.default(1024),
|
|
21
|
+
outputDType: zod_1.z.enum(["float", "int8", "uint8", "binary", "ubinary"]).optional().default("float"),
|
|
22
|
+
});
|
|
23
|
+
class VoyageEmbeddingsProvider {
|
|
24
|
+
constructor({ voyageApiKey }, providedFetch) {
|
|
25
|
+
(0, assert_1.default)(voyageApiKey, "The VoyageAI API Key does not exist. This is likely a bug.");
|
|
26
|
+
// We should always use, by default, any enterprise proxy that the user has configured.
|
|
27
|
+
// Direct requests to VoyageAI might get blocked by the network if they don't go through
|
|
28
|
+
// the provided proxy.
|
|
29
|
+
const customFetch = (providedFetch ??
|
|
30
|
+
(0, devtools_proxy_support_1.createFetch)({ useEnvironmentVariableProxies: true }));
|
|
31
|
+
this.voyage = (0, voyage_ai_provider_1.createVoyage)({ apiKey: voyageApiKey, fetch: customFetch });
|
|
32
|
+
}
|
|
33
|
+
static isConfiguredIn({ voyageApiKey }) {
|
|
34
|
+
return !!voyageApiKey;
|
|
35
|
+
}
|
|
36
|
+
async embed(modelId, content, parameters) {
|
|
37
|
+
const model = this.voyage.textEmbeddingModel(modelId);
|
|
38
|
+
const { embeddings } = await (0, ai_1.embedMany)({
|
|
39
|
+
model,
|
|
40
|
+
values: content,
|
|
41
|
+
providerOptions: { voyage: parameters },
|
|
42
|
+
});
|
|
43
|
+
return embeddings;
|
|
44
|
+
}
|
|
45
|
+
}
|
|
46
|
+
function getEmbeddingsProvider(userConfig) {
|
|
47
|
+
if (VoyageEmbeddingsProvider.isConfiguredIn(userConfig)) {
|
|
48
|
+
return new VoyageEmbeddingsProvider(userConfig);
|
|
49
|
+
}
|
|
50
|
+
return undefined;
|
|
51
|
+
}
|
|
52
|
+
exports.zSupportedEmbeddingParameters = exports.zVoyageEmbeddingParameters.extend({ model: exports.zVoyageModels });
|
|
53
|
+
//# sourceMappingURL=embeddingsProvider.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"embeddingsProvider.js","sourceRoot":"","sources":["../../../../src/common/search/embeddingsProvider.ts"],"names":[],"mappings":";;;;;;AA2EA,sDAQC;AAnFD,2DAAkD;AAElD,2BAA+B;AAE/B,oDAA4B;AAC5B,+EAAiE;AACjE,6BAAwB;AAmBX,QAAA,aAAa,GAAG,OAAC;KACzB,IAAI,CAAC,CAAC,gBAAgB,EAAE,YAAY,EAAE,iBAAiB,EAAE,eAAe,CAAC,CAAC;KAC1E,OAAO,CAAC,gBAAgB,CAAC,CAAC;AAElB,QAAA,0BAA0B,GAAG,OAAC,CAAC,MAAM,CAAC;IAC/C,eAAe,EAAE,OAAC;SACb,KAAK,CAAC,CAAC,OAAC,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,OAAC,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,OAAC,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,OAAC,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,OAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;SAC1F,QAAQ,EAAE;SACV,OAAO,CAAC,IAAI,CAAC;IAClB,WAAW,EAAE,OAAC,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,IAAA,gBAAM,EAAC,YAAY,EAAE,4DAA4D,CAAC,CAAC;QAEnF,uFAAuF;QACvF,wFAAwF;QACxF,sBAAsB;QACtB,MAAM,WAAW,GAAiB,CAAC,aAAa;YAC5C,IAAA,oCAAW,EAAC,EAAE,6BAA6B,EAAE,IAAI,EAAE,CAAC,CAA4B,CAAC;QAErF,IAAI,CAAC,MAAM,GAAG,IAAA,iCAAY,EAAC,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,IAAA,cAAS,EAAC;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,SAAgB,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;AAEY,QAAA,6BAA6B,GAAG,kCAA0B,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,qBAAa,EAAE,CAAC,CAAC"}
|
|
@@ -0,0 +1,58 @@
|
|
|
1
|
+
import { type Document } from "bson";
|
|
2
|
+
import type { UserConfig } from "../config.js";
|
|
3
|
+
import type { ConnectionManager } from "../connectionManager.js";
|
|
4
|
+
import z from "zod";
|
|
5
|
+
import { getEmbeddingsProvider } from "./embeddingsProvider.js";
|
|
6
|
+
import type { EmbeddingParameters, SupportedEmbeddingParameters } from "./embeddingsProvider.js";
|
|
7
|
+
export declare const similarityEnum: z.ZodEnum<["cosine", "euclidean", "dotProduct"]>;
|
|
8
|
+
export type Similarity = z.infer<typeof similarityEnum>;
|
|
9
|
+
export declare const quantizationEnum: z.ZodEnum<["none", "scalar", "binary"]>;
|
|
10
|
+
export type Quantization = z.infer<typeof quantizationEnum>;
|
|
11
|
+
export type VectorFieldIndexDefinition = {
|
|
12
|
+
type: "vector";
|
|
13
|
+
path: string;
|
|
14
|
+
numDimensions: number;
|
|
15
|
+
quantization: Quantization;
|
|
16
|
+
similarity: Similarity;
|
|
17
|
+
};
|
|
18
|
+
export type VectorFieldValidationError = {
|
|
19
|
+
path: string;
|
|
20
|
+
expectedNumDimensions: number;
|
|
21
|
+
expectedQuantization: Quantization;
|
|
22
|
+
actualNumDimensions: number | "unknown";
|
|
23
|
+
actualQuantization: Quantization | "unknown";
|
|
24
|
+
error: "dimension-mismatch" | "quantization-mismatch" | "not-a-vector" | "not-numeric";
|
|
25
|
+
};
|
|
26
|
+
export type EmbeddingNamespace = `${string}.${string}`;
|
|
27
|
+
export declare class VectorSearchEmbeddingsManager {
|
|
28
|
+
private readonly config;
|
|
29
|
+
private readonly connectionManager;
|
|
30
|
+
private readonly embeddings;
|
|
31
|
+
private readonly embeddingsProvider;
|
|
32
|
+
constructor(config: UserConfig, connectionManager: ConnectionManager, embeddings?: Map<EmbeddingNamespace, VectorFieldIndexDefinition[]>, embeddingsProvider?: typeof getEmbeddingsProvider);
|
|
33
|
+
cleanupEmbeddingsForNamespace({ database, collection }: {
|
|
34
|
+
database: string;
|
|
35
|
+
collection: string;
|
|
36
|
+
}): void;
|
|
37
|
+
embeddingsForNamespace({ database, collection, }: {
|
|
38
|
+
database: string;
|
|
39
|
+
collection: string;
|
|
40
|
+
}): Promise<VectorFieldIndexDefinition[]>;
|
|
41
|
+
findFieldsWithWrongEmbeddings({ database, collection, }: {
|
|
42
|
+
database: string;
|
|
43
|
+
collection: string;
|
|
44
|
+
}, document: Document): Promise<VectorFieldValidationError[]>;
|
|
45
|
+
private atlasSearchEnabledProvider;
|
|
46
|
+
private isVectorFieldIndexDefinition;
|
|
47
|
+
private getValidationErrorForDocument;
|
|
48
|
+
generateEmbeddings({ database, collection, path, rawValues, embeddingParameters, inputType, }: {
|
|
49
|
+
database: string;
|
|
50
|
+
collection: string;
|
|
51
|
+
path: string;
|
|
52
|
+
rawValues: string[];
|
|
53
|
+
embeddingParameters: SupportedEmbeddingParameters;
|
|
54
|
+
inputType: EmbeddingParameters["inputType"];
|
|
55
|
+
}): Promise<unknown[]>;
|
|
56
|
+
private isANumber;
|
|
57
|
+
}
|
|
58
|
+
//# sourceMappingURL=vectorSearchEmbeddingsManager.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
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"}
|
|
@@ -0,0 +1,200 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
3
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
4
|
+
};
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
exports.VectorSearchEmbeddingsManager = exports.quantizationEnum = exports.similarityEnum = void 0;
|
|
7
|
+
const bson_1 = require("bson");
|
|
8
|
+
const zod_1 = __importDefault(require("zod"));
|
|
9
|
+
const errors_js_1 = require("../errors.js");
|
|
10
|
+
const embeddingsProvider_js_1 = require("./embeddingsProvider.js");
|
|
11
|
+
exports.similarityEnum = zod_1.default.enum(["cosine", "euclidean", "dotProduct"]);
|
|
12
|
+
exports.quantizationEnum = zod_1.default.enum(["none", "scalar", "binary"]);
|
|
13
|
+
class VectorSearchEmbeddingsManager {
|
|
14
|
+
constructor(config, connectionManager, embeddings = new Map(), embeddingsProvider = embeddingsProvider_js_1.getEmbeddingsProvider) {
|
|
15
|
+
this.config = config;
|
|
16
|
+
this.connectionManager = connectionManager;
|
|
17
|
+
this.embeddings = embeddings;
|
|
18
|
+
this.embeddingsProvider = embeddingsProvider;
|
|
19
|
+
connectionManager.events.on("connection-close", () => {
|
|
20
|
+
this.embeddings.clear();
|
|
21
|
+
});
|
|
22
|
+
}
|
|
23
|
+
cleanupEmbeddingsForNamespace({ database, collection }) {
|
|
24
|
+
const embeddingDefKey = `${database}.${collection}`;
|
|
25
|
+
this.embeddings.delete(embeddingDefKey);
|
|
26
|
+
}
|
|
27
|
+
async embeddingsForNamespace({ database, collection, }) {
|
|
28
|
+
const provider = await this.atlasSearchEnabledProvider();
|
|
29
|
+
if (!provider) {
|
|
30
|
+
return [];
|
|
31
|
+
}
|
|
32
|
+
// We only need the embeddings for validation now, so don't query them if
|
|
33
|
+
// validation is disabled.
|
|
34
|
+
if (this.config.disableEmbeddingsValidation) {
|
|
35
|
+
return [];
|
|
36
|
+
}
|
|
37
|
+
const embeddingDefKey = `${database}.${collection}`;
|
|
38
|
+
const definition = this.embeddings.get(embeddingDefKey);
|
|
39
|
+
if (!definition) {
|
|
40
|
+
const allSearchIndexes = await provider.getSearchIndexes(database, collection);
|
|
41
|
+
const vectorSearchIndexes = allSearchIndexes.filter((index) => index.type === "vectorSearch");
|
|
42
|
+
const vectorFields = vectorSearchIndexes
|
|
43
|
+
// eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
|
|
44
|
+
.flatMap((index) => index.latestDefinition?.fields ?? [])
|
|
45
|
+
.filter((field) => this.isVectorFieldIndexDefinition(field));
|
|
46
|
+
this.embeddings.set(embeddingDefKey, vectorFields);
|
|
47
|
+
return vectorFields;
|
|
48
|
+
}
|
|
49
|
+
return definition;
|
|
50
|
+
}
|
|
51
|
+
async findFieldsWithWrongEmbeddings({ database, collection, }, document) {
|
|
52
|
+
const provider = await this.atlasSearchEnabledProvider();
|
|
53
|
+
if (!provider) {
|
|
54
|
+
return [];
|
|
55
|
+
}
|
|
56
|
+
// While we can do our best effort to ensure that the embedding validation is correct
|
|
57
|
+
// based on https://www.mongodb.com/docs/atlas/atlas-vector-search/vector-quantization/
|
|
58
|
+
// it's a complex process so we will also give the user the ability to disable this validation
|
|
59
|
+
if (this.config.disableEmbeddingsValidation) {
|
|
60
|
+
return [];
|
|
61
|
+
}
|
|
62
|
+
const embeddings = await this.embeddingsForNamespace({ database, collection });
|
|
63
|
+
return embeddings
|
|
64
|
+
.map((emb) => this.getValidationErrorForDocument(emb, document))
|
|
65
|
+
.filter((e) => e !== undefined);
|
|
66
|
+
}
|
|
67
|
+
async atlasSearchEnabledProvider() {
|
|
68
|
+
const connectionState = this.connectionManager.currentConnectionState;
|
|
69
|
+
if (connectionState.tag === "connected" && (await connectionState.isSearchSupported())) {
|
|
70
|
+
return connectionState.serviceProvider;
|
|
71
|
+
}
|
|
72
|
+
return null;
|
|
73
|
+
}
|
|
74
|
+
isVectorFieldIndexDefinition(doc) {
|
|
75
|
+
return doc["type"] === "vector";
|
|
76
|
+
}
|
|
77
|
+
getValidationErrorForDocument(definition, document) {
|
|
78
|
+
const fieldPath = definition.path.split(".");
|
|
79
|
+
let fieldRef = document;
|
|
80
|
+
const constructError = (details) => ({
|
|
81
|
+
path: definition.path,
|
|
82
|
+
expectedNumDimensions: definition.numDimensions,
|
|
83
|
+
expectedQuantization: definition.quantization,
|
|
84
|
+
actualNumDimensions: details.actualNumDimensions ?? "unknown",
|
|
85
|
+
actualQuantization: details.actualQuantization ?? "unknown",
|
|
86
|
+
error: details.error ?? "not-a-vector",
|
|
87
|
+
});
|
|
88
|
+
for (const field of fieldPath) {
|
|
89
|
+
if (fieldRef && typeof fieldRef === "object" && field in fieldRef) {
|
|
90
|
+
fieldRef = fieldRef[field];
|
|
91
|
+
}
|
|
92
|
+
else {
|
|
93
|
+
return undefined;
|
|
94
|
+
}
|
|
95
|
+
}
|
|
96
|
+
switch (definition.quantization) {
|
|
97
|
+
// Because quantization is not defined by the user
|
|
98
|
+
// we have to trust them in the format they use.
|
|
99
|
+
case "none":
|
|
100
|
+
return undefined;
|
|
101
|
+
case "scalar":
|
|
102
|
+
case "binary":
|
|
103
|
+
if (fieldRef instanceof bson_1.BSON.Binary) {
|
|
104
|
+
try {
|
|
105
|
+
const elements = fieldRef.toFloat32Array();
|
|
106
|
+
if (elements.length !== definition.numDimensions) {
|
|
107
|
+
return constructError({
|
|
108
|
+
actualNumDimensions: elements.length,
|
|
109
|
+
actualQuantization: "binary",
|
|
110
|
+
error: "dimension-mismatch",
|
|
111
|
+
});
|
|
112
|
+
}
|
|
113
|
+
return undefined;
|
|
114
|
+
}
|
|
115
|
+
catch {
|
|
116
|
+
// bits are also supported
|
|
117
|
+
try {
|
|
118
|
+
const bits = fieldRef.toBits();
|
|
119
|
+
if (bits.length !== definition.numDimensions) {
|
|
120
|
+
return constructError({
|
|
121
|
+
actualNumDimensions: bits.length,
|
|
122
|
+
actualQuantization: "binary",
|
|
123
|
+
error: "dimension-mismatch",
|
|
124
|
+
});
|
|
125
|
+
}
|
|
126
|
+
return undefined;
|
|
127
|
+
}
|
|
128
|
+
catch {
|
|
129
|
+
return constructError({
|
|
130
|
+
actualQuantization: "binary",
|
|
131
|
+
error: "not-a-vector",
|
|
132
|
+
});
|
|
133
|
+
}
|
|
134
|
+
}
|
|
135
|
+
}
|
|
136
|
+
else {
|
|
137
|
+
if (!Array.isArray(fieldRef)) {
|
|
138
|
+
return constructError({
|
|
139
|
+
error: "not-a-vector",
|
|
140
|
+
});
|
|
141
|
+
}
|
|
142
|
+
if (fieldRef.length !== definition.numDimensions) {
|
|
143
|
+
return constructError({
|
|
144
|
+
actualNumDimensions: fieldRef.length,
|
|
145
|
+
actualQuantization: "scalar",
|
|
146
|
+
error: "dimension-mismatch",
|
|
147
|
+
});
|
|
148
|
+
}
|
|
149
|
+
if (!fieldRef.every((e) => this.isANumber(e))) {
|
|
150
|
+
return constructError({
|
|
151
|
+
actualNumDimensions: fieldRef.length,
|
|
152
|
+
actualQuantization: "scalar",
|
|
153
|
+
error: "not-numeric",
|
|
154
|
+
});
|
|
155
|
+
}
|
|
156
|
+
}
|
|
157
|
+
break;
|
|
158
|
+
}
|
|
159
|
+
return undefined;
|
|
160
|
+
}
|
|
161
|
+
async generateEmbeddings({ database, collection, path, rawValues, embeddingParameters, inputType, }) {
|
|
162
|
+
const provider = await this.atlasSearchEnabledProvider();
|
|
163
|
+
if (!provider) {
|
|
164
|
+
throw new errors_js_1.MongoDBError(errors_js_1.ErrorCodes.AtlasSearchNotSupported, "Atlas Search is not supported in this cluster.");
|
|
165
|
+
}
|
|
166
|
+
const embeddingsProvider = this.embeddingsProvider(this.config);
|
|
167
|
+
if (!embeddingsProvider) {
|
|
168
|
+
throw new errors_js_1.MongoDBError(errors_js_1.ErrorCodes.NoEmbeddingsProviderConfigured, "No embeddings provider configured.");
|
|
169
|
+
}
|
|
170
|
+
if (this.config.disableEmbeddingsValidation) {
|
|
171
|
+
return await embeddingsProvider.embed(embeddingParameters.model, rawValues, {
|
|
172
|
+
inputType,
|
|
173
|
+
...embeddingParameters,
|
|
174
|
+
});
|
|
175
|
+
}
|
|
176
|
+
const embeddingInfoForCollection = await this.embeddingsForNamespace({ database, collection });
|
|
177
|
+
const embeddingInfoForPath = embeddingInfoForCollection.find((definition) => definition.path === path);
|
|
178
|
+
if (!embeddingInfoForPath) {
|
|
179
|
+
throw new errors_js_1.MongoDBError(errors_js_1.ErrorCodes.AtlasVectorSearchIndexNotFound, `No Vector Search index found for path "${path}" in namespace "${database}.${collection}"`);
|
|
180
|
+
}
|
|
181
|
+
return await embeddingsProvider.embed(embeddingParameters.model, rawValues, {
|
|
182
|
+
inputType,
|
|
183
|
+
...embeddingParameters,
|
|
184
|
+
});
|
|
185
|
+
}
|
|
186
|
+
isANumber(value) {
|
|
187
|
+
if (typeof value === "number") {
|
|
188
|
+
return true;
|
|
189
|
+
}
|
|
190
|
+
if (value instanceof bson_1.BSON.Int32 ||
|
|
191
|
+
value instanceof bson_1.BSON.Decimal128 ||
|
|
192
|
+
value instanceof bson_1.BSON.Double ||
|
|
193
|
+
value instanceof bson_1.BSON.Long) {
|
|
194
|
+
return true;
|
|
195
|
+
}
|
|
196
|
+
return false;
|
|
197
|
+
}
|
|
198
|
+
}
|
|
199
|
+
exports.VectorSearchEmbeddingsManager = VectorSearchEmbeddingsManager;
|
|
200
|
+
//# sourceMappingURL=vectorSearchEmbeddingsManager.js.map
|
|
@@ -0,0 +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"}
|
|
@@ -5,7 +5,9 @@ import EventEmitter from "events";
|
|
|
5
5
|
import type { AtlasClusterConnectionInfo, ConnectionManager, ConnectionSettings, ConnectionStateErrored } from "./connectionManager.js";
|
|
6
6
|
import type { NodeDriverServiceProvider } from "@mongosh/service-provider-node-driver";
|
|
7
7
|
import type { ExportsManager } from "./exportsManager.js";
|
|
8
|
+
import type { Client } from "@mongodb-js/atlas-local";
|
|
8
9
|
import type { Keychain } from "./keychain.js";
|
|
10
|
+
import type { VectorSearchEmbeddingsManager } from "./search/vectorSearchEmbeddingsManager.js";
|
|
9
11
|
export interface SessionOptions {
|
|
10
12
|
apiBaseUrl: string;
|
|
11
13
|
apiClientId?: string;
|
|
@@ -14,6 +16,8 @@ export interface SessionOptions {
|
|
|
14
16
|
exportsManager: ExportsManager;
|
|
15
17
|
connectionManager: ConnectionManager;
|
|
16
18
|
keychain: Keychain;
|
|
19
|
+
atlasLocalClient?: Client;
|
|
20
|
+
vectorSearchEmbeddingsManager: VectorSearchEmbeddingsManager;
|
|
17
21
|
}
|
|
18
22
|
export type SessionEvents = {
|
|
19
23
|
connect: [];
|
|
@@ -26,20 +30,23 @@ export declare class Session extends EventEmitter<SessionEvents> {
|
|
|
26
30
|
readonly exportsManager: ExportsManager;
|
|
27
31
|
readonly connectionManager: ConnectionManager;
|
|
28
32
|
readonly apiClient: ApiClient;
|
|
33
|
+
readonly atlasLocalClient?: Client;
|
|
29
34
|
readonly keychain: Keychain;
|
|
35
|
+
readonly vectorSearchEmbeddingsManager: VectorSearchEmbeddingsManager;
|
|
30
36
|
mcpClient?: {
|
|
31
37
|
name?: string;
|
|
32
38
|
version?: string;
|
|
33
39
|
title?: string;
|
|
34
40
|
};
|
|
35
41
|
logger: CompositeLogger;
|
|
36
|
-
constructor({ apiBaseUrl, apiClientId, apiClientSecret, logger, connectionManager, exportsManager, keychain, }: SessionOptions);
|
|
42
|
+
constructor({ apiBaseUrl, apiClientId, apiClientSecret, logger, connectionManager, exportsManager, keychain, atlasLocalClient, vectorSearchEmbeddingsManager, }: SessionOptions);
|
|
37
43
|
setMcpClient(mcpClient: Implementation | undefined): void;
|
|
38
44
|
disconnect(): Promise<void>;
|
|
39
45
|
close(): Promise<void>;
|
|
40
46
|
connectToMongoDB(settings: ConnectionSettings): Promise<void>;
|
|
41
47
|
get isConnectedToMongoDB(): boolean;
|
|
42
48
|
isSearchSupported(): Promise<boolean>;
|
|
49
|
+
assertSearchSupported(): Promise<void>;
|
|
43
50
|
get serviceProvider(): NodeDriverServiceProvider;
|
|
44
51
|
get connectedAtlasCluster(): AtlasClusterConnectionInfo | undefined;
|
|
45
52
|
}
|
|
@@ -1 +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;
|
|
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"}
|