Merge pull request #360 from scalabreseGD/add-client_session_keep_alive

[Scalabrese] Added client_session_keep_alive = True to Snowflake
vanna-ai · Apr 13, 2024 · bec5f66 · bec5f66
2 parents 570968d + f4a6f42
commit bec5f66
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/vanna/base/base.py b/src/vanna/base/base.py
@@ -654,6 +654,7 @@ def connect_to_snowflake(
             password=password,
             account=account,
             database=database,
+            client_session_keep_alive=True
         )
 
         def run_sql_snowflake(sql: str) -> pd.DataFrame: