Merge pull request #692 from brochure/main

Update base.py

Author: zainhoda

src/vanna/base/base.py

@@ -689,6 +689,9 @@ def generate_question(self, sql: str, **kwargs) -> str:
         return response
 
     def _extract_python_code(self, markdown_string: str) -> str:
+        # Strip whitespace to avoid indentation errors in LLM-generated code
+        markdown_string = markdown_string.strip()
+
         # Regex pattern to match Python code blocks
         pattern = r"```[\w\s]*python\n([\s\S]*?)```|```([\s\S]*?)```"