feat: enhanced query execution with an error management

climateqa/engine/talk_to_data/query.py

@@ -40,15 +40,19 @@ async def execute_sql_query(sql_query: str) -> pd.DataFrame:
     """
     def _execute_query():
         # Execute the query
-        con = duckdb.connect()
-        HF_TTD_TOKEN = os.getenv("HF_TTD_TOKEN")
-        con.execute(f"""CREATE SECRET hf_token (
-            TYPE huggingface,
-            TOKEN '{HF_TTD_TOKEN}'
-        );""")
-        results = con.execute(sql_query).fetchdf()
-        # return fetched data
-        return results
+        try:
+            con = duckdb.connect()
+            HF_TTD_TOKEN = os.getenv("HF_TTD_TOKEN")
+            con.execute(f"""CREATE SECRET hf_token (
+                TYPE huggingface,
+                TOKEN '{HF_TTD_TOKEN}'
+            );""")
+            results = con.execute(sql_query).fetchdf()
+        except Exception as e:
+            print(f"Error during tehe execution of the query : {sql_query}\n Error : {e}")
+        finally:
+            # return fetched data
+            return results
 
     # Run the query in a thread pool to avoid blocking
     loop = asyncio.get_event_loop()