Update document_generator.py

document_generator.py

@@ -246,7 +246,7 @@ async def generate_document_stream(document_generator: DocumentGenerator, docume
         content = document_generator.generate_content(section_title, content_instruction, section_number)
         section["Content"] = content
         yield json.dumps({
-            "type": "document_section", 
+            "type": "document_section",
             "content": {
                 "section_number": section_number,
                 "section_title": section_title,
@@ -262,7 +262,7 @@ async def generate_document_stream(document_generator: DocumentGenerator, docume
             content = document_generator.generate_content(subsection_title, subsection_content_instruction, subsection_number)
             subsection["Content"] = content
             yield json.dumps({
-                "type": "document_section", 
+                "type": "document_section",
                 "content": {
                     "section_number": subsection_number,
                     "section_title": subsection_title,
@@ -271,7 +271,10 @@ async def generate_document_stream(document_generator: DocumentGenerator, docume
             }) + "\n"
 
     markdown_document = MarkdownConverter.convert_to_markdown(document_generator.document_outline["Document"])
-    yield json.dumps({"type": "complete_document", "content": markdown_document}) + "\n"
+    yield json.dumps({
+        "type": "complete_document",
+        "content": markdown_document
+    }) + "\n"
 
 @cache(expire=600*24*7)
 @router.post("/generate-document/json", response_model=JsonDocumentResponse)
@@ -299,9 +302,12 @@ async def generate_markdown_document_stream_endpoint(request: MarkdownDocumentRe
             async for chunk in generate_document_stream(document_generator, request.json_document, request.query):
                 yield chunk
         except Exception as e:
-            yield json.dumps({"type": "error", "content": str(e)}) + "\n"
+            yield json.dumps({
+                "type": "error",
+                "content": str(e)
+            }) + "\n"
 
-    return StreamingResponse(stream_generator(), media_type="application/x-ndjson")
+    return StreamingResponse(stream_generator(), media_type="text/plain")
 
 ###########################################
 class MarkdownDocumentResponse(BaseModel):