Merge pull request #420 from zyclove/main

fix for a known problem with presto db where an extra ; will cause a error
vanna-ai · May 7, 2024 · e2572af · e2572af
2 parents a41f8e2 + 2437cf3
commit e2572af
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/vanna/base/base.py b/src/vanna/base/base.py
@@ -1451,6 +1451,10 @@ def connect_to_presto(
       def run_sql_presto(sql: str) -> Union[pd.DataFrame, None]:
         if conn:
           try:
+            sql = sql.rstrip()
+            # fix for a known problem with presto db where an extra ; will cause an error.
+            if sql.endswith(';'):
+                sql = sql[:-1]
             cs = conn.cursor()
             cs.execute(sql)
             results = cs.fetchall()