Pass requestId for tracking

Author: kpamaran

packages/compass-collection/src/modules/collection-tab.ts

@@ -1,5 +1,6 @@
 import type { Reducer, AnyAction, Action } from 'redux';
 import { analyzeDocuments } from 'mongodb-schema';
+import { UUID } from 'bson';
 
 import type { CollectionMetadata } from 'mongodb-collection-model';
 import type { ThunkAction } from 'redux-thunk';
@@ -15,7 +16,6 @@ import { type PreferencesAccess } from 'compass-preferences-model/provider';
 import type {
   MockDataSchemaRequest,
   MockDataSchemaResponse,
-  MockDataSchemaRawField,
 } from '@mongodb-js/compass-generative-ai';
 import { isInternalFieldPath } from 'hadron-document';
 import toNS from 'mongodb-ns';
@@ -584,8 +584,8 @@ export const generateFakerMappings = (
       return;
     }
 
-    // todo: dedup/abort around requestId
-    const requestId = 'some-request-id';
+    // todo: dedup/abort requests using requestId (CLOUDP-333850)
+    const requestId = new UUID().toString();
 
     try {
       logger.debug('Generating faker mappings');

packages/compass-generative-ai/src/atlas-ai-service.spec.ts

@@ -409,6 +409,7 @@ describe('AtlasAiService', function () {
             },
           },
           includeSampleValues: false,
+          requestId: 'test-request-id',
         };
 
         if (apiURLPreset === 'admin-api') {
@@ -467,7 +468,7 @@ describe('AtlasAiService', function () {
             expect(fetchStub).to.have.been.calledOnce;
             const { args } = fetchStub.firstCall;
             expect(args[0]).to.eq(
-              '/cloud/ai/v1/groups/testProject/mock-data-schema'
+              '/cloud/ai/v1/groups/testProject/mock-data-schema?request_id=test-request-id'
             );
             expect(result).to.deep.equal(mockResponse);
           });

packages/compass-generative-ai/src/atlas-ai-service.ts

@@ -227,6 +227,7 @@ export interface MockDataSchemaRequest {
   schema: Record<string, MockDataSchemaRawField>;
   validationRules?: Record<string, unknown> | null;
   includeSampleValues?: boolean;
+  requestId: string;
 }
 
 export const MockDataSchemaResponseShape = z.object({
@@ -466,7 +467,10 @@ export class AtlasAiService {
     const { collectionName, databaseName } = input;
     let schema = input.schema;
 
-    const url = this.getUrlForEndpoint('mock-data-schema', connectionInfo);
+    const url = `${this.getUrlForEndpoint(
+      'mock-data-schema',
+      connectionInfo
+    )}?request_id=${encodeURIComponent(input.requestId)}`;
 
     if (!input.includeSampleValues) {
       const newSchema: Record<