Update document_generator.py

document_generator.py

@@ -457,7 +457,7 @@ class JsonDocumentRequest(BaseModel):
     query: str
     template: bool = False
     images: Optional[List[UploadFile]] = File(None)
-    documents: Optional[List[UploadFile]] = File(None)
+    #documents: Optional[List[UploadFile]] = File(None)
     conversation_id: str = ""
 
 class MarkdownDocumentRequest(BaseModel):
@@ -554,6 +554,7 @@ async def generate_markdown_document_stream_endpoint(request: MarkdownDocumentRe
 @router.post("/generate-document/json", response_model=JsonDocumentResponse)
 async def generate_document_outline_endpoint(
     request: JsonDocumentRequest = Depends(),
+    images: List[UploadFile] = File(None)
 ):
     ai_client = AIClient()
     document_generator = DocumentGenerator(ai_client)