nodetest

pinterest · Sep 1, 2023 · e2aac19 · e2aac19
1 parent 7459b1f
commit e2aac19
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/querybook/webapp/components/AIAssistant/QueryGenerationModal.tsx b/querybook/webapp/components/AIAssistant/QueryGenerationModal.tsx
@@ -58,7 +58,7 @@ const useTablesInQuery = (query, language) => {
 };
 
 const useSQLGeneration = (
-    onData: ({ type: string, data: object }) => void
+    onData: (data: { type?: string; data: { [key: string]: string } }) => void
 ): {
     generating: boolean;
     generateSQL: (data: {
@@ -106,7 +106,6 @@ export const QueryGenerationModal = ({
         }
     }, []);
 
-    // const socket = useAISocket(AICommandType.TEXT_TO_SQL, onData);
     const { generating, generateSQL, cancelGeneration } =
         useSQLGeneration(onData);