Upload main.py

main.py

@@ -30,6 +30,7 @@ from starlette.middleware.gzip import GZipMiddleware
 from transformers import pipeline as hf_pipeline
 os.environ.setdefault("OMP_NUM_THREADS", "1")
 from fastapi.responses import PlainTextResponse
+from fastapi import Response
 
 import torch
 torch.set_num_threads(2)
@@ -1720,7 +1721,8 @@ def get_events(
     max_events: int = Query(15, ge=5, le=50),
     min_countries: int = Query(2, ge=1, le=50),
     min_articles: int = Query(2, ge=1, le=200),
-    speed: Speed = Query(Speed.balanced),
+    speed: Speed = Query(Speed.fast),
+    response: Response = None
 ):
 
     # always build cache on untranslated data
@@ -1742,6 +1744,7 @@ def get_events(
     events = [e for e in events if (e["country_count"] >= min_countries and e["article_count"] >= min_articles)]
     events.sort(key=lambda e: e["article_count"], reverse=True)
 
+    response.headers["Cache-Control"] = "public, max-age=30, s-maxage=60, stale-while-revalidate=300"
     return {"events": events[:max_events], "cache_key": "|".join(map(str, cache_key))}
 
 @app.get("/event/{event_id}")
@@ -1814,6 +1817,7 @@ def get_news(
     speed: Speed = Query(Speed.balanced),
     page: int = Query(1, ge=1),
     page_size: int = Query(120, ge=5, le=300),
+    response: Response = None
 ):
     enriched: List[Dict[str, Any]] = []
 
@@ -1881,6 +1885,7 @@ def get_news(
             for k in drop:
                 i.pop(k, None)
 
+    response.headers["Cache-Control"] = "public, max-age=30, s-maxage=60, stale-while-revalidate=300"
     return {
         "items": items,
         "total": total,