diff --git a/packages/compass-collection/src/components/collection-header-actions/collection-header-actions.spec.tsx b/packages/compass-collection/src/components/collection-header-actions/collection-header-actions.spec.tsx
index 27edfa32e3d..54f18833a16 100644
--- a/packages/compass-collection/src/components/collection-header-actions/collection-header-actions.spec.tsx
+++ b/packages/compass-collection/src/components/collection-header-actions/collection-header-actions.spec.tsx
@@ -1,6 +1,10 @@
 import { expect } from 'chai';
 import React, { type ComponentProps } from 'react';
-import { render, screen, cleanup } from '@mongodb-js/testing-library-compass';
+import {
+  renderWithActiveConnection,
+  screen,
+  cleanup,
+} from '@mongodb-js/testing-library-compass';
 import sinon from 'sinon';
 import {
   WorkspacesServiceProvider,
@@ -9,38 +13,62 @@ import {
 import type { PreferencesAccess } from 'compass-preferences-model';
 import { createSandboxFromDefaultPreferences } from 'compass-preferences-model';
 import { PreferencesProvider } from 'compass-preferences-model/provider';
+import { ExperimentTestName } from '@mongodb-js/compass-telemetry/provider';
+import { CompassExperimentationProvider } from '@mongodb-js/compass-telemetry';
+import type { ConnectionInfo } from '@mongodb-js/compass-connections/provider';
 
 import CollectionHeaderActions from '../collection-header-actions';
 
 describe('CollectionHeaderActions [Component]', function () {
   let preferences: PreferencesAccess;
+  let mockUseAssignment: sinon.SinonStub;
+
   beforeEach(async function () {
     preferences = await createSandboxFromDefaultPreferences();
+    mockUseAssignment = sinon.stub();
+    mockUseAssignment.returns({
+      assignment: {
+        assignmentData: {
+          variant: 'mockDataGeneratorControl',
+        },
+      },
+    });
   });
+
   afterEach(function () {
     sinon.restore();
   });
 
   const renderCollectionHeaderActions = (
     props: Partial<ComponentProps<typeof CollectionHeaderActions>> = {},
-    workspaceService: Partial<WorkspacesService> = {}
+    workspaceService: Partial<WorkspacesService> = {},
+    connectionInfo?: ConnectionInfo
   ) => {
-    return render(
-      <WorkspacesServiceProvider value={workspaceService as WorkspacesService}>
-        <PreferencesProvider value={preferences}>
-          <CollectionHeaderActions
-            namespace="test.test"
-            isReadonly={false}
-            {...props}
-          />
-        </PreferencesProvider>
-      </WorkspacesServiceProvider>
+    return renderWithActiveConnection(
+      <CompassExperimentationProvider
+        useAssignment={mockUseAssignment}
+        assignExperiment={sinon.stub()}
+      >
+        <WorkspacesServiceProvider
+          value={workspaceService as WorkspacesService}
+        >
+          <PreferencesProvider value={preferences}>
+            <CollectionHeaderActions
+              namespace="test.test"
+              isReadonly={false}
+              onOpenMockDataModal={sinon.stub()}
+              {...props}
+            />
+          </PreferencesProvider>
+        </WorkspacesServiceProvider>
+      </CompassExperimentationProvider>,
+      connectionInfo
     );
   };
 
   context('when the collection is not readonly', function () {
-    beforeEach(function () {
-      renderCollectionHeaderActions({
+    beforeEach(async function () {
+      await renderCollectionHeaderActions({
         isReadonly: false,
         namespace: 'db.coll2',
         sourceName: 'db.coll',
@@ -63,7 +91,7 @@ describe('CollectionHeaderActions [Component]', function () {
     it('does not render edit view buttons when in readonly mode', async function () {
       await preferences.savePreferences({ readOnly: true });
 
-      renderCollectionHeaderActions({
+      await renderCollectionHeaderActions({
         isReadonly: true,
         namespace: 'db.coll2',
         sourceName: 'db.someSource',
@@ -78,8 +106,8 @@ describe('CollectionHeaderActions [Component]', function () {
       ).to.not.exist;
     });
 
-    it('renders edit view buttons when not in readonly mode', function () {
-      renderCollectionHeaderActions({
+    it('renders edit view buttons when not in readonly mode', async function () {
+      await renderCollectionHeaderActions({
         isReadonly: true,
         namespace: 'db.coll2',
         sourceName: 'db.someSource',
@@ -94,9 +122,9 @@ describe('CollectionHeaderActions [Component]', function () {
 
   context('when the collection is a view', function () {
     let openEditViewWorkspaceStub: sinon.SinonStub;
-    beforeEach(function () {
+    beforeEach(async function () {
       openEditViewWorkspaceStub = sinon.stub();
-      renderCollectionHeaderActions(
+      await renderCollectionHeaderActions(
         {
           isReadonly: true,
           namespace: 'db.coll2',
@@ -135,9 +163,9 @@ describe('CollectionHeaderActions [Component]', function () {
 
   context('when the collection is editing a view', function () {
     let openCollectionWorkspaceStub: sinon.SinonStub;
-    beforeEach(function () {
+    beforeEach(async function () {
       openCollectionWorkspaceStub = sinon.stub();
-      renderCollectionHeaderActions(
+      await renderCollectionHeaderActions(
         {
           isReadonly: false,
           namespace: 'db.coll2',
@@ -168,4 +196,183 @@ describe('CollectionHeaderActions [Component]', function () {
       );
     });
   });
+
+  context('Mock Data Generator Button', function () {
+    const atlasConnectionInfo: ConnectionInfo = {
+      id: 'test-atlas-connection',
+      connectionOptions: {
+        connectionString: 'mongodb://localhost:27017',
+      },
+      atlasMetadata: {
+        orgId: 'test-org',
+        projectId: 'test-project',
+        clusterName: 'test-cluster',
+        clusterUniqueId: 'test-cluster-unique-id',
+        clusterType: 'REPLICASET',
+        clusterState: 'IDLE',
+        metricsId: 'test-metrics-id',
+        metricsType: 'replicaSet',
+        regionalBaseUrl: null,
+        instanceSize: 'M10',
+        supports: {
+          globalWrites: false,
+          rollingIndexes: true,
+        },
+      },
+    };
+
+    it('should not show Mock Data Generator button when user is in control group', async function () {
+      mockUseAssignment.returns({
+        assignment: {
+          assignmentData: {
+            variant: 'mockDataGeneratorControl',
+          },
+        },
+      });
+
+      await renderCollectionHeaderActions(
+        {
+          namespace: 'test.collection',
+          isReadonly: false,
+        },
+        {},
+        atlasConnectionInfo
+      );
+
+      expect(
+        screen.queryByTestId('collection-header-generate-mock-data-button')
+      ).to.not.exist;
+    });
+
+    it('should not show Mock Data Generator button when not in Atlas', async function () {
+      mockUseAssignment.returns({
+        assignment: {
+          assignmentData: {
+            variant: 'treatment',
+          },
+        },
+      });
+
+      await renderCollectionHeaderActions({
+        namespace: 'test.collection',
+        isReadonly: false,
+        // Don't pass atlasConnectionInfo, to simulate not being in Atlas
+      });
+
+      expect(
+        screen.queryByTestId('collection-header-generate-mock-data-button')
+      ).to.not.exist;
+    });
+
+    it('should not show Mock Data Generator button for readonly collections', async function () {
+      mockUseAssignment.returns({
+        assignment: {
+          assignmentData: {
+            variant: 'treatment',
+          },
+        },
+      });
+
+      await renderCollectionHeaderActions(
+        {
+          namespace: 'test.collection',
+          isReadonly: true,
+        },
+        {},
+        atlasConnectionInfo
+      );
+
+      expect(
+        screen.queryByTestId('collection-header-generate-mock-data-button')
+      ).to.not.exist;
+    });
+
+    it('should not show Mock Data Generator button for views (sourceName present)', async function () {
+      mockUseAssignment.returns({
+        assignment: {
+          assignmentData: {
+            variant: 'treatment',
+          },
+        },
+      });
+
+      await renderCollectionHeaderActions(
+        {
+          namespace: 'test.collection',
+          isReadonly: false,
+          sourceName: 'source-collection',
+        },
+        {},
+        atlasConnectionInfo
+      );
+
+      expect(
+        screen.queryByTestId('collection-header-generate-mock-data-button')
+      ).to.not.exist;
+    });
+
+    it('should show Mock Data Generator button when user is in treatment group and in Atlas', async function () {
+      mockUseAssignment.returns({
+        assignment: {
+          assignmentData: {
+            variant: 'mockDataGeneratorVariant',
+          },
+        },
+      });
+
+      await renderCollectionHeaderActions(
+        {
+          namespace: 'test.collection',
+          isReadonly: false,
+        },
+        {},
+        atlasConnectionInfo
+      );
+
+      expect(
+        screen.getByTestId('collection-header-generate-mock-data-button')
+      ).to.exist;
+    });
+
+    it('should call useAssignment with correct parameters', async function () {
+      await renderCollectionHeaderActions({
+        namespace: 'test.collection',
+        isReadonly: false,
+      });
+
+      expect(mockUseAssignment).to.have.been.calledWith(
+        ExperimentTestName.mockDataGenerator,
+        true // trackIsInSample - Experiment viewed analytics event
+      );
+    });
+
+    it('should call onOpenMockDataModal when CTA button is clicked', async function () {
+      const onOpenMockDataModal = sinon.stub();
+
+      mockUseAssignment.returns({
+        assignment: {
+          assignmentData: {
+            variant: 'mockDataGeneratorVariant',
+          },
+        },
+      });
+
+      await renderCollectionHeaderActions(
+        {
+          namespace: 'test.collection',
+          isReadonly: false,
+          onOpenMockDataModal,
+        },
+        {},
+        atlasConnectionInfo
+      );
+
+      const button = screen.getByTestId(
+        'collection-header-generate-mock-data-button'
+      );
+      button.click();
+
+      expect(onOpenMockDataModal).to.have.been.calledOnce;
+    });
+  });
 });
diff --git a/packages/compass-collection/src/components/collection-header-actions/collection-header-actions.tsx b/packages/compass-collection/src/components/collection-header-actions/collection-header-actions.tsx
index 64e36dabede..e6f06f6525e 100644
--- a/packages/compass-collection/src/components/collection-header-actions/collection-header-actions.tsx
+++ b/packages/compass-collection/src/components/collection-header-actions/collection-header-actions.tsx
@@ -4,6 +4,7 @@ import {
   Icon,
   css,
   spacing,
+  Tooltip,
 } from '@mongodb-js/compass-components';
 import { useConnectionInfo } from '@mongodb-js/compass-connections/provider';
 import { useOpenWorkspace } from '@mongodb-js/compass-workspaces/provider';
@@ -11,7 +12,12 @@ import React from 'react';
 import { usePreferences } from 'compass-preferences-model/provider';
 import toNS from 'mongodb-ns';
 import { wrapField } from '@mongodb-js/mongodb-constants';
-import { useTelemetry } from '@mongodb-js/compass-telemetry/provider';
+import {
+  useTelemetry,
+  useAssignment,
+  ExperimentTestName,
+  ExperimentTestGroup,
+} from '@mongodb-js/compass-telemetry/provider';
 
 const collectionHeaderActionsStyles = css({
   display: 'flex',
@@ -40,6 +46,7 @@ type CollectionHeaderActionsProps = {
   editViewName?: string;
   sourceName?: string;
   sourcePipeline?: unknown[];
+  onOpenMockDataModal: () => void;
 };
 
 const CollectionHeaderActions: React.FunctionComponent<
@@ -50,6 +57,7 @@ const CollectionHeaderActions: React.FunctionComponent<
   editViewName,
   sourceName,
   sourcePipeline,
+  onOpenMockDataModal,
 }: CollectionHeaderActionsProps) => {
   const connectionInfo = useConnectionInfo();
   const { id: connectionId, atlasMetadata } = connectionInfo;
@@ -59,8 +67,28 @@ const CollectionHeaderActions: React.FunctionComponent<
     usePreferences(['readOnly', 'enableShell']);
   const track = useTelemetry();
 
+  // Get experiment assignment for Mock Data Generator
+  const mockDataGeneratorAssignment = useAssignment(
+    ExperimentTestName.mockDataGenerator,
+    true // trackIsInSample - this will fire the "Experiment Viewed" event
+  );
+
   const { database, collection } = toNS(namespace);
 
+  // Check if user is in treatment group for Mock Data Generator experiment
+  const isInMockDataTreatmentVariant =
+    mockDataGeneratorAssignment?.assignment?.assignmentData?.variant ===
+    ExperimentTestGroup.mockDataGeneratorVariant;
+
+  const shouldShowMockDataButton =
+    isInMockDataTreatmentVariant &&
+    atlasMetadata && // Only show in Atlas
+    !isReadonly && // Don't show for readonly collections (views)
+    !sourceName; // sourceName indicates it's a view
+  // TODO: CLOUDP-337090: also filter out overly nested collections
+
+  const hasData = true; // TODO: CLOUDP-337090
+
   return (
     <div
       className={collectionHeaderActionsStyles}
@@ -81,6 +109,26 @@ const CollectionHeaderActions: React.FunctionComponent<
           Open MongoDB shell
         </Button>
       )}
+      {shouldShowMockDataButton && (
+        <Tooltip
+          enabled={!hasData}
+          trigger={
+            <div>
+              <Button
+                data-testid="collection-header-generate-mock-data-button"
+                size={ButtonSize.Small}
+                disabled={!hasData}
+                onClick={onOpenMockDataModal}
+                leftGlyph={<Icon glyph="Sparkle" />}
+              >
+                Generate Mock Data
+              </Button>
+            </div>
+          }
+        >
+          Please add data to your collection to generate similar mock documents
+        </Tooltip>
+      )}
       {atlasMetadata && (
         <Button
           data-testid="collection-header-visualize-your-data"
diff --git a/packages/compass-collection/src/components/collection-header/collection-header.spec.tsx b/packages/compass-collection/src/components/collection-header/collection-header.spec.tsx
index ba64513eb99..20c6a30ce01 100644
--- a/packages/compass-collection/src/components/collection-header/collection-header.spec.tsx
+++ b/packages/compass-collection/src/components/collection-header/collection-header.spec.tsx
@@ -8,11 +8,12 @@ import {
   within,
   userEvent,
 } from '@mongodb-js/testing-library-compass';
-import { CollectionHeader } from './collection-header';
+import { UnconnectedCollectionHeader as CollectionHeader } from './collection-header';
 import {
   WorkspacesServiceProvider,
   type WorkspacesService,
 } from '@mongodb-js/compass-workspaces/provider';
+import { MockDataGeneratorStep } from '../mock-data-generator-modal/types';
 import Sinon from 'sinon';
 
 function renderCollectionHeader(
@@ -28,6 +29,11 @@ function renderCollectionHeader(
         isClustered={false}
         isFLE={false}
         namespace="test.test"
+        isMockDataModalOpen={false}
+        currentStep={MockDataGeneratorStep.AI_DISCLAIMER}
+        onOpenMockDataModal={() => {}}
+        onCloseMockDataModal={() => {}}
+        onSetMockDataGeneratorStep={() => {}}
         {...props}
       />
     </WorkspacesServiceProvider>
@@ -281,4 +287,34 @@ describe('CollectionHeader [Component]', function () {
       });
     });
   });
+
+  context('Mock Data Generator Modal', function () {
+    it('should close modal when cancel button is clicked', function () {
+      const onCloseMockDataModal = Sinon.stub();
+
+      renderCollectionHeader({
+        isMockDataModalOpen: true,
+        onCloseMockDataModal,
+      });
+
+      const cancelButton = screen.getByText('Cancel');
+      userEvent.click(cancelButton);
+
+      expect(onCloseMockDataModal).to.have.been.calledOnce;
+    });
+
+    it('should close modal when X button is clicked', function () {
+      const onCloseMockDataModal = Sinon.stub();
+
+      renderCollectionHeader({
+        isMockDataModalOpen: true,
+        onCloseMockDataModal,
+      });
+
+      const closeButton = screen.getByLabelText('Close modal');
+      userEvent.click(closeButton);
+
+      expect(onCloseMockDataModal).to.have.been.calledOnce;
+    });
+  });
 });
diff --git a/packages/compass-collection/src/components/collection-header/collection-header.tsx b/packages/compass-collection/src/components/collection-header/collection-header.tsx
index 02eb416e623..9eecd5e7f4b 100644
--- a/packages/compass-collection/src/components/collection-header/collection-header.tsx
+++ b/packages/compass-collection/src/components/collection-header/collection-header.tsx
@@ -18,6 +18,15 @@ import { CollectionBadge } from './badges';
 import { useOpenWorkspace } from '@mongodb-js/compass-workspaces/provider';
 import { useConnectionInfo } from '@mongodb-js/compass-connections/provider';
 import { getConnectionTitle } from '@mongodb-js/connection-info';
+import MockDataGeneratorModal from '../mock-data-generator-modal/mock-data-generator-modal';
+import type { MockDataGeneratorStep } from '../mock-data-generator-modal/types';
+import { connect } from 'react-redux';
+import type { CollectionState } from '../../modules/collection-tab';
+import {
+  mockDataGeneratorModalOpened,
+  mockDataGeneratorModalClosed,
+  mockDataGeneratorStepChanged,
+} from '../../modules/collection-tab';
 
 const collectionHeaderStyles = css({
   padding: spacing[400],
@@ -58,6 +67,11 @@ type CollectionHeaderProps = {
   sourceName?: string;
   editViewName?: string;
   sourcePipeline?: unknown[];
+  isMockDataModalOpen: boolean;
+  currentStep: MockDataGeneratorStep;
+  onOpenMockDataModal: () => void;
+  onCloseMockDataModal: () => void;
+  onSetMockDataGeneratorStep: (step: MockDataGeneratorStep) => void;
 };
 
 const getInsightsForPipeline = (pipeline: any[], isAtlas: boolean) => {
@@ -82,9 +96,7 @@ const getInsightsForPipeline = (pipeline: any[], isAtlas: boolean) => {
   return Array.from(insights);
 };
 
-export const CollectionHeader: React.FunctionComponent<
-  CollectionHeaderProps
-> = ({
+const CollectionHeader: React.FunctionComponent<CollectionHeaderProps> = ({
   namespace,
   isReadonly,
   isTimeSeries,
@@ -94,6 +106,11 @@ export const CollectionHeader: React.FunctionComponent<
   sourceName,
   editViewName,
   sourcePipeline,
+  isMockDataModalOpen,
+  currentStep,
+  onOpenMockDataModal,
+  onCloseMockDataModal,
+  onSetMockDataGeneratorStep,
 }) => {
   const darkMode = useDarkMode();
   const showInsights = usePreference('showInsights');
@@ -170,8 +187,40 @@ export const CollectionHeader: React.FunctionComponent<
           namespace={namespace}
           sourceName={sourceName}
           sourcePipeline={sourcePipeline}
+          onOpenMockDataModal={onOpenMockDataModal}
         />
       </div>
+      <MockDataGeneratorModal
+        isOpen={isMockDataModalOpen}
+        onClose={onCloseMockDataModal}
+        currentStep={currentStep}
+        onCurrentStepChange={onSetMockDataGeneratorStep}
+      />
     </div>
   );
 };
+
+const mapStateToProps = (state: CollectionState) => ({
+  isMockDataModalOpen: state.mockDataGenerator.isModalOpen,
+  currentStep: state.mockDataGenerator.currentStep,
+});
+
+type ConnectedCollectionHeaderProps = Omit<
+  CollectionHeaderProps,
+  | 'isMockDataModalOpen'
+  | 'currentStep'
+  | 'onOpenMockDataModal'
+  | 'onCloseMockDataModal'
+  | 'onSetMockDataGeneratorStep'
+>;
+
+const ConnectedCollectionHeader = connect(mapStateToProps, {
+  onOpenMockDataModal: mockDataGeneratorModalOpened,
+  onCloseMockDataModal: mockDataGeneratorModalClosed,
+  onSetMockDataGeneratorStep: mockDataGeneratorStepChanged,
+})(CollectionHeader) as React.ComponentType<ConnectedCollectionHeaderProps>;
+
+export {
+  ConnectedCollectionHeader as CollectionHeader,
+  CollectionHeader as UnconnectedCollectionHeader, // For tests
+};
diff --git a/packages/compass-collection/src/components/mock-data-generator-modal/mock-data-generator-modal.spec.tsx b/packages/compass-collection/src/components/mock-data-generator-modal/mock-data-generator-modal.spec.tsx
index 88d94c6e778..75fc890e1dc 100644
--- a/packages/compass-collection/src/components/mock-data-generator-modal/mock-data-generator-modal.spec.tsx
+++ b/packages/compass-collection/src/components/mock-data-generator-modal/mock-data-generator-modal.spec.tsx
@@ -8,10 +8,10 @@ import { StepButtonLabelMap } from './constants';
 
 describe('MockDataGeneratorModal', () => {
   const sandbox = Sinon.createSandbox();
-  let onOpenChange: Sinon.SinonSpy;
+  let onClose: Sinon.SinonSpy;
 
   beforeEach(() => {
-    onOpenChange = sandbox.spy();
+    onClose = sandbox.spy();
   });
 
   afterEach(() => {
@@ -28,7 +28,7 @@ describe('MockDataGeneratorModal', () => {
       return (
         <MockDataGeneratorModal
           isOpen={isOpen}
-          onOpenChange={onOpenChange}
+          onClose={onClose}
           currentStep={currentStepStateMock}
           onCurrentStepChange={(step) => {
             onCurrentStepChangeStateMock(step);
@@ -51,20 +51,20 @@ describe('MockDataGeneratorModal', () => {
     expect(screen.queryByTestId('generate-mock-data-modal')).to.not.exist;
   });
 
-  it('calls onOpenChange(false) when the modal is closed', () => {
+  it('calls onClose when the modal is closed', () => {
     renderModal();
 
     screen.getByLabelText('Close modal').click();
 
-    expect(onOpenChange.calledOnceWith(false)).to.be.true;
+    expect(onClose.calledOnce).to.be.true;
   });
 
-  it('calls onOpenChange(false) when the cancel button is clicked', () => {
+  it('calls onClose when the cancel button is clicked', () => {
     renderModal();
 
     screen.getByText('Cancel').click();
 
-    expect(onOpenChange.calledOnceWith(false)).to.be.true;
+    expect(onClose.calledOnce).to.be.true;
   });
 
   it('disables the Back button on the first step', () => {
diff --git a/packages/compass-collection/src/components/mock-data-generator-modal/mock-data-generator-modal.tsx b/packages/compass-collection/src/components/mock-data-generator-modal/mock-data-generator-modal.tsx
index b7723530d61..053194ff051 100644
--- a/packages/compass-collection/src/components/mock-data-generator-modal/mock-data-generator-modal.tsx
+++ b/packages/compass-collection/src/components/mock-data-generator-modal/mock-data-generator-modal.tsx
@@ -30,14 +30,14 @@ const rightButtonsStyles = css`
 
 interface Props {
   isOpen: boolean;
-  onOpenChange: (isOpen: boolean) => void;
+  onClose: () => void;
   currentStep: MockDataGeneratorStep;
   onCurrentStepChange: (step: MockDataGeneratorStep) => void;
 }
 
 const MockDataGeneratorModal = ({
   isOpen,
-  onOpenChange,
+  onClose,
   currentStep,
   onCurrentStepChange,
 }: Props) => {
@@ -51,14 +51,14 @@ const MockDataGeneratorModal = ({
     onCurrentStepChange(previousStep);
   };
 
-  const onCancel = () => {
-    onOpenChange(false);
-  };
-
   return (
     <Modal
       open={isOpen}
-      setOpen={(open) => onOpenChange(open)}
+      setOpen={(open) => {
+        if (!open) {
+          onClose();
+        }
+      }}
       data-testid="generate-mock-data-modal"
     >
       <ModalHeader title="Generate Mock Data" />
@@ -74,7 +74,7 @@ const MockDataGeneratorModal = ({
           Back
         </Button>
         <div className={rightButtonsStyles}>
-          <Button onClick={onCancel}>Cancel</Button>
+          <Button onClick={onClose}>Cancel</Button>
           <Button
             variant={ButtonVariant.Primary}
             onClick={onNext}
diff --git a/packages/compass-collection/src/modules/collection-tab.ts b/packages/compass-collection/src/modules/collection-tab.ts
index 12655bc0dce..7e97b0498ed 100644
--- a/packages/compass-collection/src/modules/collection-tab.ts
+++ b/packages/compass-collection/src/modules/collection-tab.ts
@@ -6,6 +6,7 @@ import type { workspacesServiceLocator } from '@mongodb-js/compass-workspaces/pr
 import type { CollectionSubtab } from '@mongodb-js/compass-workspaces';
 import type { DataService } from '@mongodb-js/compass-connections/provider';
 import type { experimentationServiceLocator } from '@mongodb-js/compass-telemetry/provider';
+import { MockDataGeneratorStep } from '../components/mock-data-generator-modal/types';
 
 function isAction<A extends AnyAction>(
   action: AnyAction,
@@ -31,10 +32,17 @@ export type CollectionState = {
   namespace: string;
   metadata: CollectionMetadata | null;
   editViewName?: string;
+  mockDataGenerator: {
+    isModalOpen: boolean;
+    currentStep: MockDataGeneratorStep;
+  };
 };
 
 enum CollectionActions {
   CollectionMetadataFetched = 'compass-collection/CollectionMetadataFetched',
+  MockDataGeneratorModalOpened = 'compass-collection/MockDataGeneratorModalOpened',
+  MockDataGeneratorModalClosed = 'compass-collection/MockDataGeneratorModalClosed',
+  MockDataGeneratorStepChanged = 'compass-collection/MockDataGeneratorStepChanged',
 }
 
 interface CollectionMetadataFetchedAction {
@@ -42,12 +50,29 @@ interface CollectionMetadataFetchedAction {
   metadata: CollectionMetadata;
 }
 
+interface MockDataGeneratorModalOpenedAction {
+  type: CollectionActions.MockDataGeneratorModalOpened;
+}
+
+interface MockDataGeneratorModalClosedAction {
+  type: CollectionActions.MockDataGeneratorModalClosed;
+}
+
+interface MockDataGeneratorStepChangedAction {
+  type: CollectionActions.MockDataGeneratorStepChanged;
+  step: MockDataGeneratorStep;
+}
+
 const reducer: Reducer<CollectionState, Action> = (
   state = {
     // TODO(COMPASS-7782): use hook to get the workspace tab id instead
     workspaceTabId: '',
     namespace: '',
     metadata: null,
+    mockDataGenerator: {
+      isModalOpen: false,
+      currentStep: MockDataGeneratorStep.AI_DISCLAIMER,
+    },
   },
   action
 ) => {
@@ -62,6 +87,54 @@ const reducer: Reducer<CollectionState, Action> = (
       metadata: action.metadata,
     };
   }
+
+  if (
+    isAction<MockDataGeneratorModalOpenedAction>(
+      action,
+      CollectionActions.MockDataGeneratorModalOpened
+    )
+  ) {
+    return {
+      ...state,
+      mockDataGenerator: {
+        ...state.mockDataGenerator,
+        isModalOpen: true,
+        currentStep: MockDataGeneratorStep.AI_DISCLAIMER,
+      },
+    };
+  }
+
+  if (
+    isAction<MockDataGeneratorModalClosedAction>(
+      action,
+      CollectionActions.MockDataGeneratorModalClosed
+    )
+  ) {
+    return {
+      ...state,
+      mockDataGenerator: {
+        ...state.mockDataGenerator,
+        isModalOpen: false,
+        currentStep: MockDataGeneratorStep.AI_DISCLAIMER,
+      },
+    };
+  }
+
+  if (
+    isAction<MockDataGeneratorStepChangedAction>(
+      action,
+      CollectionActions.MockDataGeneratorStepChanged
+    )
+  ) {
+    return {
+      ...state,
+      mockDataGenerator: {
+        ...state.mockDataGenerator,
+        currentStep: action.step,
+      },
+    };
+  }
+
   return state;
 };
 
@@ -71,6 +144,22 @@ export const collectionMetadataFetched = (
   return { type: CollectionActions.CollectionMetadataFetched, metadata };
 };
 
+export const mockDataGeneratorModalOpened =
+  (): MockDataGeneratorModalOpenedAction => {
+    return { type: CollectionActions.MockDataGeneratorModalOpened };
+  };
+
+export const mockDataGeneratorModalClosed =
+  (): MockDataGeneratorModalClosedAction => {
+    return { type: CollectionActions.MockDataGeneratorModalClosed };
+  };
+
+export const mockDataGeneratorStepChanged = (
+  step: MockDataGeneratorStep
+): MockDataGeneratorStepChangedAction => {
+  return { type: CollectionActions.MockDataGeneratorStepChanged, step };
+};
+
 export const selectTab = (
   tabName: CollectionSubtab
 ): CollectionThunkAction<void> => {
diff --git a/packages/compass-collection/src/stores/collection-tab.ts b/packages/compass-collection/src/stores/collection-tab.ts
index 925572bf7e3..c3a1e120ba8 100644
--- a/packages/compass-collection/src/stores/collection-tab.ts
+++ b/packages/compass-collection/src/stores/collection-tab.ts
@@ -6,6 +6,8 @@ import reducer, {
   selectTab,
   collectionMetadataFetched,
 } from '../modules/collection-tab';
+import { MockDataGeneratorStep } from '../components/mock-data-generator-modal/types';
+
 import type { Collection } from '@mongodb-js/compass-app-stores/provider';
 import type { ActivateHelpers } from '@mongodb-js/compass-app-registry';
 import type { workspacesServiceLocator } from '@mongodb-js/compass-workspaces/provider';
@@ -77,6 +79,10 @@ export function activatePlugin(
       namespace,
       metadata: null,
       editViewName,
+      mockDataGenerator: {
+        isModalOpen: false,
+        currentStep: MockDataGeneratorStep.AI_DISCLAIMER,
+      },
     },
     applyMiddleware(
       thunk.withExtraArgument({
diff --git a/packages/compass-telemetry/src/growth-experiments.ts b/packages/compass-telemetry/src/growth-experiments.ts
index 87437d07e5b..6b242a921fd 100644
--- a/packages/compass-telemetry/src/growth-experiments.ts
+++ b/packages/compass-telemetry/src/growth-experiments.ts
@@ -2,3 +2,8 @@ export enum ExperimentTestName {
   earlyJourneyIndexesGuidance = 'EARLY_JOURNEY_INDEXES_GUIDANCE_20250328',
   mockDataGenerator = 'MOCK_DATA_GENERATOR_20251001',
 }
+
+export enum ExperimentTestGroup {
+  mockDataGeneratorVariant = 'mockDataGeneratorVariant',
+  mockDataGeneratorControl = 'mockDataGeneratorControl',
+}
diff --git a/packages/compass-telemetry/src/index.ts b/packages/compass-telemetry/src/index.ts
index 6aa1bd54741..3fdacb2c506 100644
--- a/packages/compass-telemetry/src/index.ts
+++ b/packages/compass-telemetry/src/index.ts
@@ -7,4 +7,4 @@ export type {
 } from './types';
 
 export { CompassExperimentationProvider } from './experimentation-provider';
-export { ExperimentTestName } from './growth-experiments';
+export { ExperimentTestName, ExperimentTestGroup } from './growth-experiments';
diff --git a/packages/compass-telemetry/src/provider.tsx b/packages/compass-telemetry/src/provider.tsx
index 9eea405ad2e..301222d3838 100644
--- a/packages/compass-telemetry/src/provider.tsx
+++ b/packages/compass-telemetry/src/provider.tsx
@@ -3,7 +3,7 @@ import { createServiceLocator } from '@mongodb-js/compass-app-registry';
 import { createTrack, type TelemetryServiceOptions } from './generic-track';
 import { useLogger } from '@mongodb-js/compass-logging/provider';
 import type { TrackFunction } from './types';
-import { ExperimentTestName } from './growth-experiments';
+import type { ExperimentTestName } from './growth-experiments';
 import { ExperimentationContext } from './experimentation-provider';
 import type { types } from '@mongodb-js/mdb-experiment-js';
 
@@ -154,4 +154,5 @@ export const useFireExperimentViewed = ({
 };
 
 export type { TrackFunction };
-export { ExperimentTestName };
+export { ExperimentTestName, ExperimentTestGroup } from './growth-experiments';
+export { useAssignment } from './experimentation-provider';