pushed up

Author: nvs119

packages/compass-vector-embedding-visualizer/src/index.ts

@@ -6,36 +6,19 @@ import { dataServiceLocator } from '@mongodb-js/compass-connections/provider';
 import { createStore } from 'redux';
 
 import { VectorVisualizer } from './components/vector-visualizer';
+import { activateVectorPlugin } from './stores/store';
 
 function reducer(state = {}, _action: any) {
   return state;
 }
 
-export const CompassVectorPluginProvider = registerHadronPlugin<
-  { dataService: any; collection: any },
-  any,
-  any
->(
+export const CompassVectorPluginProvider = registerHadronPlugin(
   {
     name: 'CompassVectorEmbeddingVisualizer',
-    component: function VectorVisualizerProvider({
-      dataService,
-      collection,
-      children,
-    }) {
-      return React.createElement(
-        VectorVisualizer,
-        { dataService, collection },
-        children
-      );
-    },
-    activate: () => {
-      const store = createStore(reducer);
-      return {
-        store: () => store,
-        deactivate: () => {},
-      };
+    component: function VectorVisualizerProvider({ children }) {
+      return React.createElement(React.Fragment, null, children);
     },
+    activate: activateVectorPlugin,
   },
   {
     dataService: dataServiceLocator,
@@ -44,11 +27,18 @@ export const CompassVectorPluginProvider = registerHadronPlugin<
   }
 );
 
+const VectorVisualizerWrapper = (props: {
+  dataService: any;
+  collection: any;
+}) => {
+  return React.createElement(VectorVisualizer, props);
+};
+
 export const CompassVectorPlugin = {
-  name: 'VectorVisualizer',
+  name: 'Vector Visualizer' as const,
   type: 'CollectionTab' as const,
   provider: CompassVectorPluginProvider,
-  content: VectorVisualizer,
+  content: VectorVisualizerWrapper,
   header: () => React.createElement('div', null, 'Vector Embeddings'),
 };
 

packages/compass-vector-embedding-visualizer/src/stores/store.ts

@@ -0,0 +1,59 @@
+import type { Store } from 'redux';
+import { createStore } from 'redux';
+import type { DataService } from 'mongodb-data-service';
+import type { ConnectionInfoRef } from '@mongodb-js/compass-connections/provider';
+import type {
+  Collection,
+  MongoDBInstance,
+} from '@mongodb-js/compass-app-stores/provider';
+import type AppRegistry from 'hadron-app-registry';
+import type { Logger } from '@mongodb-js/compass-logging';
+import type { TrackFunction } from '@mongodb-js/compass-telemetry';
+import type { AtlasService } from '@mongodb-js/atlas-service/provider';
+import type { PreferencesAccess } from 'compass-preferences-model';
+import type { ActivateHelpers } from 'hadron-app-registry';
+
+export type VectorDataServiceProps =
+  | 'find'
+  // Required for collection model (fetching stats)
+  | 'collectionStats'
+  | 'collectionInfo'
+  | 'listCollections';
+export type VectorDataService = Pick<DataService, VectorDataServiceProps>;
+
+export type VectorPluginServices = {
+  dataService: VectorDataService;
+  connectionInfoRef: ConnectionInfoRef;
+  instance: MongoDBInstance;
+  localAppRegistry: Pick<AppRegistry, 'on' | 'emit' | 'removeListener'>;
+  globalAppRegistry: Pick<AppRegistry, 'on' | 'emit' | 'removeListener'>;
+  logger: Logger;
+  collection: Collection;
+  track: TrackFunction;
+  atlasService: AtlasService;
+  preferences: PreferencesAccess;
+};
+
+export type VectorPluginOptions = {
+  namespace: string;
+  serverVersion: string;
+  isReadonly: boolean;
+};
+
+export type VectorPluginStore = Store;
+
+function reducer(state = {}, _action: any) {
+  return state;
+}
+
+export function activateVectorPlugin(
+  _options: VectorPluginOptions,
+  _services: VectorPluginServices,
+  { cleanup }: ActivateHelpers
+) {
+  const store = createStore((state = {}) => state);
+  return {
+    store,
+    deactivate: () => cleanup(),
+  };
+}

packages/compass-workspaces/src/types.ts

@@ -4,7 +4,8 @@ export type CollectionSubtab =
   | 'Schema'
   | 'Indexes'
   | 'Validation'
-  | 'GlobalWrites';
+  | 'GlobalWrites'
+  | 'Vector Visualizer';
 
 export type WelcomeWorkspace = {
   type: 'Welcome';