Browse Source

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

John Wang 1 year ago
parent
commit
19b5cb1e10
1 changed files with 6 additions and 0 deletions
  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("{"):