fix: agenerate sql

vanna-ai · Jul 5, 2024 · 1c3a68d · 1c3a68d
1 parent 0c75730
commit 1c3a68d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/vanna/base/base.py b/src/vanna/base/base.py
@@ -201,7 +201,7 @@ async def agenerate_sql(
 
                 try:
                     self.log(title="Running Intermediate SQL", message=intermediate_sql)
-                    df = self.run_sql(intermediate_sql)
+                    df = await self.arun_sql(intermediate_sql)
 
                     prompt = self.get_sql_prompt(
                         initial_prompt=initial_prompt,