Update document_generator.py

document_generator.py

@@ -453,15 +453,17 @@ router = APIRouter()
 class JsonDocumentResponse(BaseModel):
     json_document: Dict
 
-
 class JsonDocumentRequest(BaseModel):
     query: str
     template: bool = False
     images: Optional[List[UploadFile]] = File(None)
+    documents: Optional[List[UploadFile]] = File(None)
+    conversation_id: str = ""
 
 class MarkdownDocumentRequest(BaseModel):
     json_document: Dict
     query: str
+    conversation_id: str = ""
 
 MESSAGE_DELIMITER = b"\n---DELIMITER---\n"
 
@@ -552,7 +554,6 @@ 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)