improve json parsing

landing-ai · Sep 17, 2024 · e1a862c · e1a862c
1 parent cabc88d
commit e1a862c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/vision_agent/agent/agent_utils.py b/vision_agent/agent/agent_utils.py
@@ -41,6 +41,8 @@ def _strip_markdown_code(inp_str: str) -> str:
 
 def extract_json(json_str: str) -> Dict[str, Any]:
  json_str = json_str.replace("\n", " ").strip()
+ json_str = json_str.replace("'", '"')
+ json_str = json_str.replace(": True", ": true").replace(": False", ": false")
 
  try:
  return json.loads(json_str) # type: ignore