Update document_generator.py

document_generator.py

@@ -254,8 +254,12 @@ class MarkdownConverter:
 
 router = APIRouter()
 
-class DocumentRequest(BaseModel):
-    query: str
+class DocumentRequest(BaseModel): 
+    query: str class 
+
+class DocumentResponse(BaseModel): 
+    json_document: Dict 
+    markdown_document: str 
 
 class JsonDocumentResponse(BaseModel):
     json_document: Dict