Update document_generator_v2.py

document_generator_v2.py

@@ -166,6 +166,7 @@ import functools
 from fastapi import APIRouter, HTTPException, Request, UploadFile, File, Depends
 from fastapi.responses import StreamingResponse
 from pydantic import BaseModel
+from fastapi_cache import FastAPICache
 from fastapi_cache.decorator import cache
 import psycopg2
 from datetime import datetime
@@ -638,4 +639,27 @@ async def get_last_observations(limit: int = 10, format: str = "json"):
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Failed to retrieve observations: {str(e)}")
 
+## TEST CACHE
 
+class CacheItem(BaseModel):
+    key: str
+    value: str
+
+@router.post("/set-cache")
+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)
+        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)}")
+
+@router.get("/get-cache/{key}")
+async def get_cache(key: str):
+    try:
+        value = await FastAPICache.get(key)
+        if value is None:
+            raise HTTPException(status_code=404, detail=f"No cache found for key: {key}")
+        return {"key": key, "value": value}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Failed to get cache: {str(e)}")