fix(ask-seer): Send correct analytics (#97511)

This PR fixes up some issues around the analytics for the trace
explorer.

Author: nsdeschenes

static/app/utils/analytics/tracingEventMap.tsx

@@ -15,6 +15,9 @@ export type TracingEventParameters = {
     query: string;
     visualize_count: number;
   };
+  'trace.explorer.ai_query_example_clicked': {
+    example_query: string;
+  };
   'trace.explorer.ai_query_feedback': {
     correct_query_results: 'yes' | 'no';
     natural_language_query: string;
@@ -217,6 +220,7 @@ export const tracingEventMap: Record<TracingEventKey, string | null> = {
   'trace.explorer.ai_query_submitted': 'Trace Explorer: AI Query Submitted',
   'trace.explorer.ai_query_interface': 'Trace Explorer: AI Query Interface',
   'trace.explorer.ai_query_feedback': 'Trace Explorer: AI Query Feedback',
+  'trace.explorer.ai_query_example_clicked': 'Trace Explorer: AI Query Example Clicked',
   'trace.explorer.metadata': 'Improved Trace Explorer Pageload Metadata',
   'trace.explorer.schema_hints_click':
     'Improved Trace Explorer: Schema Hints Click Events',

static/app/views/explore/components/seerComboBox/hooks.ts

@@ -3,6 +3,7 @@ import {useCallback, useState} from 'react';
 import {addErrorMessage} from 'sentry/actionCreators/indicator';
 import {useSearchQueryBuilder} from 'sentry/components/searchQueryBuilder/context';
 import {t} from 'sentry/locale';
+import {trackAnalytics} from 'sentry/utils/analytics';
 import {fetchMutation, useMutation} from 'sentry/utils/queryClient';
 import type RequestError from 'sentry/utils/requestError/requestError';
 import {useNavigate} from 'sentry/utils/useNavigate';
@@ -119,8 +120,7 @@ export const useApplySeerSearchQuery = () => {
   const pageFilters = usePageFilters();
   const organization = useOrganization();
 
-  const {setDisplayAskSeer, setDisplayAskSeerFeedback, askSeerSuggestedQueryRef} =
-    useSearchQueryBuilder();
+  const {askSeerSuggestedQueryRef} = useSearchQueryBuilder();
 
   return useCallback(
     (result: SeerSearchQuery) => {
@@ -178,17 +178,14 @@ export const useApplySeerSearchQuery = () => {
         sort,
         mode,
       });
+      trackAnalytics('trace.explorer.ai_query_applied', {
+        organization,
+        query,
+        group_by_count: groupBys.length,
+        visualize_count: visualizations.length,
+      });
       navigate(url, {replace: true, preventScrollReset: true});
-      setDisplayAskSeerFeedback(true);
-      setDisplayAskSeer(false);
     },
-    [
-      askSeerSuggestedQueryRef,
-      navigate,
-      organization,
-      pageFilters.selection,
-      setDisplayAskSeer,
-      setDisplayAskSeerFeedback,
-    ]
+    [askSeerSuggestedQueryRef, navigate, organization, pageFilters.selection]
   );
 };

static/app/views/explore/components/seerComboBox/seerComboBox.tsx

@@ -225,12 +225,10 @@ export function SeerComboBox({initialQuery, ...props}: SeerComboBoxProps) {
         return;
       }
 
-      trackAnalytics('trace.explorer.ai_query_submitted', {
-        organization,
-        natural_language_query: searchQuery.trim(),
-      });
       askSeerNLQueryRef.current = searchQuery.trim();
       applySeerSearchQuery(item);
+      setDisplayAskSeerFeedback(true);
+      setDisplayAskSeer(false);
       state.close();
     },
     children: item => {
@@ -296,6 +294,10 @@ export function SeerComboBox({initialQuery, ...props}: SeerComboBoxProps) {
         switch (e.key) {
           case 'Escape':
             if (!state.isOpen) {
+              trackAnalytics('trace.explorer.ai_query_interface', {
+                organization,
+                action: 'closed',
+              });
               setDisplayAskSeerFeedback(false);
               setDisplayAskSeer(false);
             }
@@ -342,8 +344,12 @@ export function SeerComboBox({initialQuery, ...props}: SeerComboBoxProps) {
                 addErrorMessage(t('Failed to find AI query to apply'));
                 return;
               }
-              state.close();
+
+              askSeerNLQueryRef.current = searchQuery.trim();
               applySeerSearchQuery(item);
+              setDisplayAskSeerFeedback(true);
+              setDisplayAskSeer(false);
+              state.close();
               return;
             }