From e2aac193419d1204001924cf1f56f599ea45572a Mon Sep 17 00:00:00 2001
From: "J.C. Zhong" <exzjc@hotmail.com>
Date: Fri, 1 Sep 2023 05:12:23 +0000
Subject: [PATCH] nodetest

---
 .../webapp/components/AIAssistant/QueryGenerationModal.tsx     | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/querybook/webapp/components/AIAssistant/QueryGenerationModal.tsx b/querybook/webapp/components/AIAssistant/QueryGenerationModal.tsx
index 696cb6ec3..2f91302b2 100644
--- 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);