better json parsing

landing-ai · Oct 6, 2024 · 10c6332 · 10c6332
1 parent dda1e64
commit 10c6332
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/vision_agent/agent/agent_utils.py b/vision_agent/agent/agent_utils.py
@@ -48,11 +48,16 @@ def _strip_markdown_code(inp_str: str) -> str:
 
 def extract_json(json_str: str) -> Dict[str, Any]:
  json_str_mod = json_str.replace("\n", " ").strip()
- json_str_mod = json_str_mod.replace("'", '"')
  json_str_mod = json_str_mod.replace(": True", ": true").replace(
  ": False", ": false"
  )
 
+ # sometimes the json is in single quotes
+ try:
+ return json.loads(json_str_mod.replace("'", '"')) # type: ignore
+ except json.JSONDecodeError:
+ pass
+
  try:
  return json.loads(json_str_mod) # type: ignore
  except json.JSONDecodeError: