chore(generative-ai): add url endpoints for compass web usage COMPASS…

…-8376 (#6423)
mongodb-js · Nov 4, 2024 · 8b88964 · 8b88964
1 parent 6c4d534
commit 8b88964
Show file tree

Hide file tree

Showing 10 changed files with 488 additions and 316 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/packages/atlas-service/src/atlas-service.ts b/packages/atlas-service/src/atlas-service.ts
@@ -29,12 +29,8 @@ export class AtlasService {
   ) {
     this.config = getAtlasConfig(preferences);
   }
-  adminApiEndpoint(path?: string, requestId?: string): string {
-    const uri = `${this.config.atlasApiBaseUrl}${normalizePath(path)}`;
-    const query = requestId
-      ? `?request_id=${encodeURIComponent(requestId)}`
-      : '';
-    return `${uri}${query}`;
+  adminApiEndpoint(path?: string): string {
+    return `${this.config.atlasApiBaseUrl}${normalizePath(path)}`;
   }
   cloudEndpoint(path?: string): string {
     return `${this.config.cloudBaseUrl}${normalizePath(path)}`;

diff --git a/packages/compass-aggregations/src/modules/pipeline-builder/pipeline-ai.ts b/packages/compass-aggregations/src/modules/pipeline-builder/pipeline-ai.ts
@@ -281,20 +281,23 @@ export const runAIPipelineGeneration = (
 
       const { collection: collectionName, database: databaseName } =
         toNS(namespace);
-      jsonResponse = await atlasAiService.getAggregationFromUserInput({
-        signal: abortController.signal,
-        userInput,
-        collectionName,
-        databaseName,
-        schema,
-        // Provide sample documents when the user has opted in in their settings.
-        ...(provideSampleDocuments
-          ? {
-              sampleDocuments,
-            }
-          : undefined),
-        requestId,
-      });
+      jsonResponse = await atlasAiService.getAggregationFromUserInput(
+        {
+          signal: abortController.signal,
+          userInput,
+          collectionName,
+          databaseName,
+          schema,
+          // Provide sample documents when the user has opted in in their settings.
+          ...(provideSampleDocuments
+            ? {
+                sampleDocuments,
+              }
+            : undefined),
+          requestId,
+        },
+        connectionInfo
+      );
     } catch (err: any) {
       if (signal.aborted) {
         // If we already aborted so we ignore the error.

diff --git a/packages/compass-generative-ai/package.json b/packages/compass-generative-ai/package.json
@@ -54,6 +54,7 @@
   "dependencies": {
     "@mongodb-js/atlas-service": "^0.31.0",
     "@mongodb-js/compass-components": "^1.31.0",
+    "@mongodb-js/compass-connections": "^1.45.0",
     "@mongodb-js/compass-intercom": "^0.13.2",
     "@mongodb-js/compass-logging": "^1.4.9",
     "bson": "^6.8.0",
@@ -64,6 +65,7 @@
     "react": "^17.0.2"
   },
   "devDependencies": {
+    "@mongodb-js/connection-info": "^0.9.2",
     "@mongodb-js/eslint-config-compass": "^1.1.7",
     "@mongodb-js/mocha-config-compass": "^1.4.2",
     "@mongodb-js/prettier-config-compass": "^1.0.2",