diff --git a/dataherald/sql_generator/dataherald_sqlagent.py b/dataherald/sql_generator/dataherald_sqlagent.py index 7d634138..857cdd08 100644 --- 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