Merge pull request vanna-ai#512 from mahendrasinghbora/fix/extract-sq…

…l-cte-regex

Fix regex in `extract_sql` to correctly handle SQL containing CTEs (without matching WITHIN)

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}")