DH5682/fixing the wrong exception handling with tools (#451)

Dataherald · Apr 3, 2024 · 39bef01 · 39bef01
1 parent 809ac31
commit 39bef01
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