document events

mongodb-js · Aug 19, 2024 · 89c2cc5 · 89c2cc5
1 parent 818dff8
commit 89c2cc5
Show file tree

Hide file tree

Showing 4 changed files with 565 additions and 17 deletions.
diff --git a/packages/compass-aggregations/src/modules/pipeline-builder/pipeline-ai.ts b/packages/compass-aggregations/src/modules/pipeline-builder/pipeline-ai.ts
@@ -233,7 +233,9 @@ export const runAIPipelineGeneration = (
     const provideSampleDocuments =
       preferences.getPreferences().enableGenAISampleDocumentPassing;
 
-    const editor_view_type = pipelineMode === 'builder-ui' ? 'stages' : 'text';
+    const editor_view_type: 'stages' | 'text' =
+      pipelineMode === 'builder-ui' ? 'stages' : 'text';
+
     if (existingRequestId !== null) {
       // Cancel the active request as this one will override.
       abort(existingRequestId);

diff --git a/packages/compass-query-bar/src/stores/ai-query-reducer.ts b/packages/compass-query-bar/src/stores/ai-query-reducer.ts
@@ -139,7 +139,7 @@ function trackAndLogFailed({
   track(
     'AI Response Failed',
     () => ({
-      editor_view_type: 'find',
+      editor_view_type: 'find' as const,
       error_name: errorName,
       status_code: statusCode,
       error_code: errorCode ?? '',
@@ -178,7 +178,7 @@ export const runAIQuery = (
     track(
       'AI Prompt Submitted',
       () => ({
-        editor_view_type: 'find',
+        editor_view_type: 'find' as const,
         user_input_length: userInput.length,
         has_sample_documents: provideSampleDocuments,
         request_id: requestId,

diff --git a/packages/compass-shell/src/modules/worker-runtime.ts b/packages/compass-shell/src/modules/worker-runtime.ts
@@ -64,8 +64,8 @@ export function createWorkerRuntime(
       // already check whether Compass telemetry is enabled or not.
       track: ({ event, properties }) => {
         return track(
-          'Shell Event',
-          { ...properties, shell_event_name: event },
+          `Shell ${event}`,
+          { ...properties },
           connectionInfo.getCurrentConnectionInfo()
         );
       },