pvanand commited on
Commit
3c03878
·
verified ·
1 Parent(s): 5e2ed5c

Update document_generator.py

Browse files
Files changed (1) hide show
  1. document_generator.py +3 -0
document_generator.py CHANGED
@@ -266,6 +266,9 @@ class MarkdownDocumentRequest(BaseModel):
266
  json_document: Dict
267
  query: str
268
 
 
 
 
269
  @cache(expire=600*24*7)
270
  @router.post("/generate-document/json", response_model=JsonDocumentResponse)
271
  async def generate_document_outline_endpoint(request: DocumentRequest):
 
266
  json_document: Dict
267
  query: str
268
 
269
+ class MarkdownDocumentResponse(BaseModel):
270
+ markdown_document: str
271
+
272
  @cache(expire=600*24*7)
273
  @router.post("/generate-document/json", response_model=JsonDocumentResponse)
274
  async def generate_document_outline_endpoint(request: DocumentRequest):