diff --git a/src/vanna/base/base.py b/src/vanna/base/base.py index 1baca34ee..fa7822a56 100644 --- 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,