Explorar o código

feat: fix json end with `` (#285)

John Wang hai 1 ano
pai
achega
19b5cb1e10
Modificáronse 1 ficheiros con 6 adicións e 0 borrados
  1. 6 0
      api/libs/json_in_md_parser.py

+ 6 - 0
api/libs/json_in_md_parser.py

@@ -13,6 +13,12 @@ def parse_json_markdown(json_string: str) -> dict:
     if start_index != -1 and end_index != -1:
         extracted_content = json_string[start_index + len("```json"):end_index].strip()
 
+        # Parse the JSON string into a Python dictionary
+        parsed = json.loads(extracted_content)
+    elif start_index != -1 and end_index == -1 and json_string.endswith("``"):
+        end_index = json_string.find("``", start_index + len("```json"))
+        extracted_content = json_string[start_index + len("```json"):end_index].strip()
+
         # Parse the JSON string into a Python dictionary
         parsed = json.loads(extracted_content)
     elif json_string.startswith("{"):