Update base.py

move .strip() as part of the _extract_python_code()

Author: chatball

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]*?)```"
 
@@ -734,7 +737,7 @@ def generate_plotly_code(
 
         plotly_code = self.submit_prompt(message_log, kwargs=kwargs)
 
-        return self._sanitize_plotly_code(self._extract_python_code(plotly_code.strip()))
+        return self._sanitize_plotly_code(self._extract_python_code(plotly_code))
 
     # ----------------- Connect to Any Database to run the Generated SQL ----------------- #