From 1d82f37fc55ccd563dea908539cf642e7cf8fffc Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Arthur=20Mar=C3=A7al?= Date: Wed, 13 Nov 2024 18:55:52 -0300 Subject: [PATCH] Add autocommit as a parameter to connect_to_mysql function --- src/vanna/base/base.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/vanna/base/base.py b/src/vanna/base/base.py index 696ad681..06226467 100644 --- a/src/vanna/base/base.py +++ b/src/vanna/base/base.py @@ -992,6 +992,7 @@ def connect_to_mysql( user: str = None, password: str = None, port: int = None, + autocommit: bool = False, **kwargs ): @@ -1043,6 +1044,7 @@ def connect_to_mysql( database=dbname, port=port, cursorclass=pymysql.cursors.DictCursor, + autocommit=autocommit, **kwargs ) except pymysql.Error as e: @@ -1079,7 +1081,7 @@ def connect_to_clickhouse( dbname: str = None, user: str = None, password: str = None, - port: int = None, + port: int = None, **kwargs ):