From a576bca73c764c79d89a551427c5e6d61212bb2f Mon Sep 17 00:00:00 2001
From: Sergey Petushkov <petushkov.sergey@gmail.com>
Date: Wed, 11 Jun 2025 12:10:17 +0200
Subject: [PATCH] fix(aggregations): pass dataService to redux thunk arg, fix
 types

---
 packages/compass-aggregations/src/modules/index.ts | 6 +++---
 packages/compass-aggregations/src/stores/store.ts  | 5 ++---
 2 files changed, 5 insertions(+), 6 deletions(-)

diff --git a/packages/compass-aggregations/src/modules/index.ts b/packages/compass-aggregations/src/modules/index.ts
index 59686b726d0..1f5de34002b 100644
--- a/packages/compass-aggregations/src/modules/index.ts
+++ b/packages/compass-aggregations/src/modules/index.ts
@@ -95,10 +95,10 @@ const rootReducer = combineReducers({
 export type RootState = ReturnType<typeof rootReducer>;
 
 export type PipelineBuilderExtraArgs = {
-  globalAppRegistry: AppRegistry;
-  localAppRegistry: AppRegistry;
+  globalAppRegistry: Pick<AppRegistry, 'on' | 'emit' | 'removeListener'>;
+  localAppRegistry: Pick<AppRegistry, 'on' | 'emit' | 'removeListener'>;
   pipelineBuilder: PipelineBuilder;
-  pipelineStorage: PipelineStorage;
+  pipelineStorage: PipelineStorage | undefined;
   workspaces: WorkspacesService;
   preferences: PreferencesAccess;
   logger: Logger;
diff --git a/packages/compass-aggregations/src/stores/store.ts b/packages/compass-aggregations/src/stores/store.ts
index 7ee9ebedb01..cbefaaa7dae 100644
--- a/packages/compass-aggregations/src/stores/store.ts
+++ b/packages/compass-aggregations/src/stores/store.ts
@@ -135,9 +135,7 @@ export function activateAggregationsPlugin(
 
   const stagesIdAndType = mapStoreStagesToStageIdAndType(stages);
 
-  const store: Store<RootState> & {
-    dispatch: PipelineBuilderThunkDispatch;
-  } = createStore(
+  const store = createStore(
     reducer,
     {
       // TODO: move this to thunk extra arg
@@ -189,6 +187,7 @@ export function activateAggregationsPlugin(
         atlasAiService,
         connectionInfoRef,
         connectionScopedAppRegistry,
+        dataService,
       })
     )
   );