Zekun Wu commited on
Commit
1981871
1 Parent(s): c1d9dc3
Files changed (1) hide show
  1. util/injection.py +1 -1
util/injection.py CHANGED
@@ -56,7 +56,7 @@ def invoke_retry(prompt, agent, parameters,string_input=False):
56
  score_json = json.loads(score_text)
57
  except json.JSONDecodeError:
58
  try:
59
- score_json = json_repair.repair_json(score_text, skip_json_loads=True, return_objects=False)
60
  except json.JSONDecodeError:
61
  raise Exception("Failed to decode JSON response even after repair attempt.")
62
  # score = re.search(r'\d+', score_text)
 
56
  score_json = json.loads(score_text)
57
  except json.JSONDecodeError:
58
  try:
59
+ score_json = json.loads(json_repair.repair_json(score_text, skip_json_loads=True, return_objects=False))
60
  except json.JSONDecodeError:
61
  raise Exception("Failed to decode JSON response even after repair attempt.")
62
  # score = re.search(r'\d+', score_text)