fix: add back text2sql survey (#1452)

* fix: add back text2sql survey * fix linter
pinterest · May 16, 2024 · b8d9a95 · b8d9a95
1 parent d81d582
commit b8d9a95
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
     "name": "querybook",
-    "version": "3.34.0",
+    "version": "3.34.1",
     "description": "A Big Data Webapp",
     "private": true,
     "scripts": {

diff --git a/querybook/webapp/components/AIAssistant/AICommandResultView.tsx b/querybook/webapp/components/AIAssistant/AICommandResultView.tsx
@@ -7,6 +7,8 @@ import { IQueryCellCommand } from 'const/command';
 import useNonEmptyState from 'hooks/useNonEmptyState';
 import { trackClick } from 'lib/analytics';
 import { Button } from 'ui/Button/Button';
+import { SurveySurfaceType } from 'const/survey';
+import { useSurveyTrigger } from 'hooks/ui/useSurveyTrigger';
 
 interface IAICommandResultViewProps {
     command: IQueryCellCommand;
@@ -59,6 +61,19 @@ export const AICommandResultView = ({
         }
     }, [commandResult]);
 
+    const triggerSurvey = useSurveyTrigger();
+    useEffect(() => {
+        if (!newQuery || isStreaming) {
+            return;
+        }
+        triggerSurvey(SurveySurfaceType.TEXT_TO_SQL, {
+            question: commandKwargs.question,
+            tables,
+            query: newQuery,
+        });
+        // eslint-disable-next-line react-hooks/exhaustive-deps
+    }, [newQuery, triggerSurvey, isStreaming]);
+
     const handleAccept = useCallback(() => {
         onAccept(newQuery);
         trackClick({