Update document_generator_v2.py

document_generator_v2.py

@@ -649,7 +649,7 @@ class CacheItem(BaseModel):
 async def set_cache(item: CacheItem):
     try:
         # Set the cache with a default expiration of 1 hour (3600 seconds)
-        await FastAPICache.set(item.key, item.value, expire=3600)
+        await FastAPICache.get_backend().set(item.key, item.value, expire=3600)
         return {"message": f"Cache set for key: {item.key}"}
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Failed to set cache: {str(e)}")
@@ -657,7 +657,7 @@ async def set_cache(item: CacheItem):
 @router.get("/get-cache/{key}")
 async def get_cache(key: str):
     try:
-        value = await FastAPICache.get(key)
+        value = await FastAPICache.get_backend().get(key)
         if value is None:
             raise HTTPException(status_code=404, detail=f"No cache found for key: {key}")
         return {"key": key, "value": value}