Update document_generator.py

document_generator.py

@@ -257,27 +257,35 @@ router = APIRouter()
 class DocumentRequest(BaseModel):
     query: str
 
-class DocumentResponse(BaseModel):
+class JsonDocumentResponse(BaseModel):
     json_document: Dict
+
+class MarkdownDocumentResponse(BaseModel):
     markdown_document: str
 
 @cache(expire=600*24*7)
-@router.post("/generate-document", response_model=DocumentResponse)
-async def generate_document_endpoint(request: DocumentRequest):
+@router.post("/generate-document/json", response_model=JsonDocumentResponse)
+async def generate_json_document_endpoint(request: DocumentRequest):
     ai_client = AIClient()
     document_generator = DocumentGenerator(ai_client)
-    
     try:
-        # Generate the document
+        # Generate the JSON document
+        json_document = document_generator.generate_document(request.query)
+        return JsonDocumentResponse(json_document=json_document)
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+@cache(expire=600*24*7)
+@router.post("/generate-document/markdown", response_model=MarkdownDocumentResponse)
+async def generate_markdown_document_endpoint(request: DocumentRequest):
+    ai_client = AIClient()
+    document_generator = DocumentGenerator(ai_client)
+    try:
+        # Generate the JSON document
         json_document = document_generator.generate_document(request.query)
-        
         # Convert to Markdown
         markdown_document = MarkdownConverter.convert_to_markdown(json_document["Document"])
-        
-        return DocumentResponse(
-            json_document=json_document,
-            markdown_document=markdown_document
-        )
+        return MarkdownDocumentResponse(markdown_document=markdown_document)
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))