Update document_generator.py

document_generator.py

@@ -277,4 +277,26 @@ async def generate_document_endpoint(request: DocumentRequest):
             markdown_document=markdown_document
         )
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        raise HTTPException(status_code=500, detail=str(e))
+
+class CacheTestResponse(BaseModel):
+    result: str
+    execution_time: float
+
+@router.get("/test-cache/{test_id}", response_model=CacheTestResponse)
+@cache(expire=60)  # Cache for 1 minute
+async def test_cache(test_id: int):
+    start_time = time.time()
+    
+    # Simulate some time-consuming operation
+    await asyncio.sleep(2)
+    
+    result = f"Test result for ID: {test_id}"
+    
+    end_time = time.time()
+    execution_time = end_time - start_time
+    
+    return CacheTestResponse(
+        result=result,
+        execution_time=execution_time
+    )