Fix MSSQL connection close

heresandyboy · May 2, 2024 · 7b962d5 · 7b962d5
1 parent c85fe13
commit 7b962d5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/vanna/base/base.py b/src/vanna/base/base.py
@@ -1279,6 +1279,7 @@ def run_sql_mssql(sql: str):
             # Execute the SQL statement and return the result as a pandas DataFrame
             with engine.begin() as conn:
                 df = pd.read_sql_query(sa.text(sql), conn)
+                conn.close()
                 return df
 
             raise Exception("Couldn't run sql")