diff --git a/src/crewai_tools/tools/nl2sql/nl2sql_tool.py b/src/crewai_tools/tools/nl2sql/nl2sql_tool.py index c33411ca8..da70db8ac 100644 --- a/src/crewai_tools/tools/nl2sql/nl2sql_tool.py +++ b/src/crewai_tools/tools/nl2sql/nl2sql_tool.py @@ -63,8 +63,10 @@ class NL2SQL(BaseTool): return data else: return f"Query {sql_query} executed successfully" + except Exception as e: session.rollback() raise e + finally: session.close()