update

Paper2Poster/PosterAgent/parse_raw.py

@@ -1,8 +1,9 @@
 from dotenv import load_dotenv
-from utils.src.utils import get_json_from_response
+# from utils.src.utils import get_json_from_response
 from utils.src.model_utils import parse_pdf
 import json
 import random
+import json_repair
 
 from camel.models import ModelFactory
 from camel.agents import ChatAgent
@@ -46,7 +47,17 @@ def account_token(response):
     output_token = response.info['usage']['completion_tokens']
 
     return input_token, output_token
-
+def get_json_from_response(raw_response: str):
+    response = raw_response.strip()
+    l, r = response.rfind("```json"), response.rfind("```")
+    try:
+        if l == -1 or r == -1:
+            response = json_repair.loads(response)
+        else:
+            response = json_repair.loads(response[l + 7 : r].strip())
+        return response
+    except Exception as e:
+        raise RuntimeError("Failed to parse JSON from response", e)
 @retry(stop=stop_after_attempt(5))
 def parse_raw(args, actor_config, version=2):
     raw_source = args.poster_path