Update document_generator.py

document_generator.py

@@ -273,7 +273,8 @@ async def generate_document_stream(document_generator: DocumentGenerator, docume
     markdown_document = MarkdownConverter.convert_to_markdown(document_generator.document_outline["Document"])
     yield json.dumps({
         "type": "complete_document",
-        "content": markdown_document
+        "markdown": markdown_document,
+        "json": document_generator.document_outline
     }, ensure_ascii=False) + "\n"
 
 @cache(expire=600*24*7)