pvanand commited on
Commit
74ee173
1 Parent(s): 0181978

Update document_generator.py

Browse files
Files changed (1) hide show
  1. document_generator.py +2 -1
document_generator.py CHANGED
@@ -457,7 +457,7 @@ class JsonDocumentRequest(BaseModel):
457
  query: str
458
  template: bool = False
459
  images: Optional[List[UploadFile]] = File(None)
460
- documents: Optional[List[UploadFile]] = File(None)
461
  conversation_id: str = ""
462
 
463
  class MarkdownDocumentRequest(BaseModel):
@@ -554,6 +554,7 @@ async def generate_markdown_document_stream_endpoint(request: MarkdownDocumentRe
554
  @router.post("/generate-document/json", response_model=JsonDocumentResponse)
555
  async def generate_document_outline_endpoint(
556
  request: JsonDocumentRequest = Depends(),
 
557
  ):
558
  ai_client = AIClient()
559
  document_generator = DocumentGenerator(ai_client)
 
457
  query: str
458
  template: bool = False
459
  images: Optional[List[UploadFile]] = File(None)
460
+ #documents: Optional[List[UploadFile]] = File(None)
461
  conversation_id: str = ""
462
 
463
  class MarkdownDocumentRequest(BaseModel):
 
554
  @router.post("/generate-document/json", response_model=JsonDocumentResponse)
555
  async def generate_document_outline_endpoint(
556
  request: JsonDocumentRequest = Depends(),
557
+ images: List[UploadFile] = File(None)
558
  ):
559
  ai_client = AIClient()
560
  document_generator = DocumentGenerator(ai_client)