diff --git a/src/vanna/base/base.py b/src/vanna/base/base.py index 56360adc..492516ea 100644 --- a/src/vanna/base/base.py +++ b/src/vanna/base/base.py @@ -182,7 +182,7 @@ def extract_sql(self, llm_response: str) -> str: """ # If the llm_response contains a CTE (with clause), extract the last sql between WITH and ; - sqls = re.findall(r"WITH.*?;", llm_response, re.DOTALL) + sqls = re.findall(r"\bWITH\b .*?;", llm_response, re.DOTALL) if sqls: sql = sqls[-1] self.log(title="Extracted SQL", message=f"{sql}")