chore: add query to ai assistant context logging (#1342)

pinterest · Sep 28, 2023 · 9c322ae · 9c322ae
1 parent bac499c
commit 9c322ae
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/querybook/webapp/components/AIAssistant/QueryGenerationModal.tsx b/querybook/webapp/components/AIAssistant/QueryGenerationModal.tsx
@@ -240,6 +240,18 @@ export const QueryGenerationModal = ({
             <Button
                 title="Cancel"
                 onClick={() => {
+                    if (newQuery) {
+                        trackClick({
+                            component: ComponentType.AI_ASSISTANT,
+                            element: ElementType.QUERY_GENERATION_REJECT_BUTTON,
+                            aux: {
+                                mode: textToSQLMode,
+                                question,
+                                tables,
+                                query: newQuery,
+                            },
+                        });
+                    }
                     onHide();
                 }}
             />
@@ -255,6 +267,7 @@ export const QueryGenerationModal = ({
                             mode: textToSQLMode,
                             question,
                             tables,
+                            query: newQuery,
                         },
                     });
                     onHide();
@@ -273,6 +286,7 @@ export const QueryGenerationModal = ({
                             mode: textToSQLMode,
                             question,
                             tables,
+                            query: newQuery,
                         },
                     });
                     onHide();
@@ -367,6 +381,7 @@ export const QueryGenerationModal = ({
                                                     mode: textToSQLMode,
                                                     question,
                                                     tables,
+                                                    query: newQuery,
                                                 },
                                             });
                                         }}

diff --git a/querybook/webapp/const/analytics.ts b/querybook/webapp/const/analytics.ts
@@ -103,6 +103,7 @@ export enum ElementType {
     QUERY_ERROR_AUTO_FIX_APPLY_AND_RUN_BUTTON = 'QUERY_ERROR_AUTO_FIX_APPLY_AND_RUN_BUTTON',
     QUERY_GENERATION_BUTTON = 'QUERY_GENERATION_BUTTON',
     QUERY_GENERATION_KEEP_BUTTON = 'QUERY_GENERATION_KEEP_BUTTON',
+    QUERY_GENERATION_REJECT_BUTTON = 'QUERY_GENERATION_REJECT_BUTTON',
     QUERY_GENERATION_APPLY_BUTTON = 'QUERY_GENERATION_APPLY_BUTTON',
     QUERY_GENERATION_APPLY_AND_RUN_BUTTON = 'QUERY_GENERATION_APPLY_AND_RUN_BUTTON',
 }