Update document_generator.py

document_generator.py

@@ -222,7 +222,7 @@ class MarkdownDocumentRequest(BaseModel):
     json_document: Dict
     query: str
 
-async def generate_document_stream(document_generator: DocumentGenerator, document_outline: Dict, query: str):
+def generate_document_stream(document_generator: DocumentGenerator, document_outline: Dict, query: str):
     document_generator.document_outline = document_outline
     
     overall_objective = query
@@ -252,7 +252,7 @@ async def generate_document_stream(document_generator: DocumentGenerator, docume
                 "section_title": section_title,
                 "content": content
             }
-        }) + "\n"
+        }, ensure_ascii=False) + "\n"
 
         for subsection in section.get("Subsections", []):
             subsection_title = subsection.get("Title", "")
@@ -268,13 +268,13 @@ async def generate_document_stream(document_generator: DocumentGenerator, docume
                     "section_title": subsection_title,
                     "content": content
                 }
-            }) + "\n"
+            }, ensure_ascii=False) + "\n"
 
     markdown_document = MarkdownConverter.convert_to_markdown(document_generator.document_outline["Document"])
     yield json.dumps({
         "type": "complete_document",
         "content": markdown_document
-    }) + "\n"
+    }, ensure_ascii=False) + "\n"
 
 @cache(expire=600*24*7)
 @router.post("/generate-document/json", response_model=JsonDocumentResponse)