pass collection name, db name, and validation rules

Author: kpamaran

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

@@ -547,7 +547,7 @@ export const generateFakerMappings = (
   connectionInfo: ConnectionInfo
 ): CollectionThunkAction<Promise<void>> => {
   return async (dispatch, getState, { logger, atlasAiService }) => {
-    const { schemaAnalysis, fakerSchemaGeneration } = getState();
+    const { schemaAnalysis, fakerSchemaGeneration, namespace } = getState();
     if (schemaAnalysis.status !== SCHEMA_ANALYSIS_STATE_COMPLETE) {
       logger.log.error(
         mongoLogId(1_001_000_305),
@@ -566,17 +566,23 @@ export const generateFakerMappings = (
 
     // todo: dedup/abort around requestId
     const requestId = 'some-request-id';
+    var what = 'what';
     try {
       logger.debug('Generating faker mappings');
 
+      const { database, collection } = toNS(namespace);
+
       dispatch({
         type: CollectionActions.FakerMappingGenerationStarted,
         requestId: requestId,
       });
 
-      const mockDataSchemaRequest = MockDataSchemaRequestShape.parse(
-        schemaAnalysis.processedSchema
-      );
+      const mockDataSchemaRequest = MockDataSchemaRequestShape.parse({
+        databaseName: database,
+        collectionName: collection,
+        schema: schemaAnalysis.processedSchema,
+        validationRules: schemaAnalysis.schemaMetadata.validationRules,
+      });
 
       const response = await atlasAiService.getMockDataSchema(
         mockDataSchemaRequest,
@@ -595,7 +601,9 @@ export const generateFakerMappings = (
       );
       dispatch({
         type: CollectionActions.FakerMappingGenerationFailed,
-        error: 'Failed to generate faker mappings',
+        error:
+          'Failed to generate faker mappings' +
+          String(e instanceof Error ? e.message : e),
         requestId: requestId,
       });
     }

packages/compass-collection/src/stores/collection-tab.spec.ts

@@ -291,29 +291,30 @@ describe('Collection Tab Content store', function () {
     it('can complete successfully', async function () {
       const dispatch = sandbox.spy();
       const getState = sandbox.stub().returns({
+        namespace: 'some_db.some_collection',
         schemaAnalysis: {
           status: SCHEMA_ANALYSIS_STATE_COMPLETE,
           processedSchema: {
-            collectionName: 'foo',
-            databaseName: 'test',
-            schema: {
-              name: {
-                type: 'String',
-                sampleValues: ['John', 'Jane', 'Bob'],
-                probability: 1.0,
-              },
-              age: {
-                type: 'Number',
-                sampleValues: [25, 30, 35],
-                probability: 0.9,
-              },
-              isActive: {
-                type: 'Boolean',
-                sampleValues: [true, false],
-                probability: 0.8,
-              },
+            name: {
+              type: 'String',
+              probability: 1.0,
+              sampleValues: ['John', 'Jane', 'Bob'],
+            },
+            age: {
+              type: 'Number',
+              probability: 0.9,
+              sampleValues: [25, 30],
+            },
+            isActive: {
+              type: 'Boolean',
+              probability: 0.8,
+              sampleValues: [true, false],
             },
           },
+          schemaMetadata: {
+            maxNestingDepth: 1,
+            validationRules: null,
+          },
         },
         fakerSchemaGeneration: { status: MOCK_DATA_GENERATOR_STATE_IDLE },
       });
@@ -353,9 +354,7 @@ describe('Collection Tab Content store', function () {
         },
       };
       const atlasAiService = {
-        getMockDataSchema: sandbox
-          .stub()
-          .returns(Promise.resolve(mockDataSchemaResponse)),
+        getMockDataSchema: sandbox.stub().resolves(mockDataSchemaResponse),
       };
 
       // Act
@@ -386,6 +385,7 @@ describe('Collection Tab Content store', function () {
     it('can dispatch a failure', async function () {
       const dispatch = sandbox.spy();
       const getState = sandbox.stub().returns({
+        namespace: 'some_db.some_collection',
         schemaAnalysis: {
           status: SCHEMA_ANALYSIS_STATE_COMPLETE,
           processedSchema: undefined,
@@ -398,7 +398,7 @@ describe('Collection Tab Content store', function () {
       };
 
       const atlasAiService = {
-        getMockDataSchema: sandbox.stub().returns(Promise.resolve({})),
+        getMockDataSchema: sandbox.stub().resolves({}),
       };
 
       // Act

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

@@ -228,7 +228,7 @@ export const MockDataSchemaRequestShape = z.object({
   collectionName: z.string(),
   databaseName: z.string(),
   schema: MockDataSchemaRawFieldMappingShape,
-  validationRules: z.record(z.string(), z.unknown()).optional(),
+  validationRules: z.record(z.string(), z.unknown()).nullable().optional(),
   includeSampleValues: z.boolean().default(false),
 });