'commit'

AI/AiService/CallDeepSeekStream.py

@@ -50,20 +50,31 @@ def extract_level2_and_level3(json_dict, level1_title=None):
     if level1_title and level1_title in json_dict:
         result = []
         for level2_title, level2_content in json_dict[level1_title].items():
-            result.append({"type": "transition", "data": {"title": level2_title, "text": ""}})
+            result.append({"type": "transition", "data": {"title": level2_title, "text": level2_title}})
             # 输出三级目录内容
             if isinstance(level2_content, dict):
                 for level3_title, level3_items in level2_content.items():
                     # 确保 level3_items 是列表
                     if isinstance(level3_items, list):
-                        items = [{"title": item, "text": item} for item in level3_items]
+                        items = []
+                        for item in level3_items:
+                            # 如果 item 是字符串，直接作为 title 和 text
+                            if isinstance(item, str):
+                                items.append({"title": item, "text": item})
+                            # 如果 item 是字典，提取 title 和 text
+                            elif isinstance(item, dict):
+                                items.append({
+                                    "title": item.get("title", ""),
+                                    "text": item.get("text", "")
+                                })
                     else:
-                        items = [{"title": str(level3_items), "text": ""}]
+                        # 如果 level3_items 不是列表，直接作为 title 和 text
+                        items = [{"title": str(level3_items), "text": str(level3_items)}]
                     result.append({
                         "type": "content",
                         "data": {
                             "title": level3_title,
-                            "text": "",
+                            "text": level3_title,
                             "items": items
                         }
                     })
@@ -93,17 +104,20 @@ async def ConvertMarkdownToJson(markdown_content):
 
     # 提取一级目录
     level1_json = extract_level1(json_dict)
+    print(level1_json)
     for item in level1_json:
         yield json.dumps(item, ensure_ascii=False)
         await asyncio.sleep(0.5)  # 控制逐行输出的速度
 
     # 生成目录部分
     contents_json = extract_contents(json_dict)
+    print(contents_json)
     yield json.dumps(contents_json, ensure_ascii=False)
     await asyncio.sleep(0.5)
 
     # 提取二级目录及其三级目录内容
     level2_and_level3_json = extract_level2_and_level3(json_dict)
+    print(level2_and_level3_json)
     for item in level2_and_level3_json:
         yield json.dumps(item, ensure_ascii=False)
         await asyncio.sleep(0.5)