From 6e09b1f869f6cd700152bd57d82afce90795f011 Mon Sep 17 00:00:00 2001 From: Dillon Laird Date: Mon, 13 May 2024 14:31:55 -0700 Subject: [PATCH] add some extra parsing for code snippets --- vision_agent/agent/vision_agent_v2.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/vision_agent/agent/vision_agent_v2.py b/vision_agent/agent/vision_agent_v2.py index 8c4fd9cd..59ec7a2d 100644 --- 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