add some extra parsing for code snippets

landing-ai · May 13, 2024 · 6e09b1f · 6e09b1f
1 parent 3bfab15
commit 6e09b1f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/vision_agent/agent/vision_agent_v2.py b/vision_agent/agent/vision_agent_v2.py
@@ -46,6 +46,8 @@ def extract_code(code: str) -> str:
  if "```python" in code:
  code = code[code.find("```python") + len("```python") :]
  code = code[: code.find("```")]
+ if code.startswith("python\n"):
+ code = code[len("python\n") :]
  return code