DH5682/fixing the wrong exception handling with tools

Dataherald · Apr 3, 2024 · 6e8f435 · 6e8f435
1 parent 5f2c7a4
commit 6e8f435
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/dataherald/sql_generator/dataherald_sqlagent.py b/dataherald/sql_generator/dataherald_sqlagent.py
@@ -90,6 +90,8 @@ def wrapper(*args: Any, **kwargs: Any) -> Any:  # noqa: PLR0911
                 return f"Google API returned an error: {e}"
             except SQLAlchemyError as e:
                 return f"Error: {e}"
+            except Exception as e:
+                return f"Error: {e}"
 
         return wrapper