Update document_generator.py

document_generator.py

@@ -456,10 +456,13 @@ class JsonDocumentResponse(BaseModel):
 class JsonDocumentRequest(BaseModel):
     query: str
     template: bool = False
-    images: Optional[List[UploadFile]] = File(None)
-    #documents: Optional[List[UploadFile]] = File(None)
+    images: Optional[List[UploadFile]] = None
+    documents: Optional[List[UploadFile]] = None
     conversation_id: str = ""
 
+    class Config:
+        arbitrary_types_allowed = True
+
 class MarkdownDocumentRequest(BaseModel):
     json_document: Dict
     query: str
@@ -553,9 +556,19 @@ async def generate_markdown_document_stream_endpoint(request: MarkdownDocumentRe
 @cache(expire=600*24*7)
 @router.post("/generate-document/json", response_model=JsonDocumentResponse)
 async def generate_document_outline_endpoint(
-    request: JsonDocumentRequest = Depends(),
-    images: List[UploadFile] = File(None)
+    query: str,
+    template: bool = False,
+    images: Optional[List[UploadFile]] = File(None),
+    documents: Optional[List[UploadFile]] = File(None),
+    conversation_id: str = ""
 ):
+    request = JsonDocumentRequest(
+        query=query,
+        template=template,
+        images=images,
+        documents=documents,
+        conversation_id=conversation_id
+    )
     ai_client = AIClient()
     document_generator = DocumentGenerator(ai_client)
     vision_tools = VisionTools(ai_client)