fix: no cache for html file

opendatalab · Jan 7, 2025 · 9258822 · 9258822
1 parent 9202a65
commit 9258822
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 8 deletions.
diff --git a/.gitmodules b/.gitmodules
diff --git a/labelu/main.py b/labelu/main.py
@@ -103,9 +103,13 @@ def __init__(self, *args: Any, **kwargs: Any):
 
     def file_response(self, *args: Any, **kwargs: Any) -> Response:
         resp = super().file_response(*args, **kwargs)
-        resp.headers.setdefault("Cache-Control", self.cachecontrol)
-        resp.headers.setdefault("Pragma", self.pragma)
-        resp.headers.setdefault("Expires", self.expires)
+
+        # No cache for html files
+        if resp.media_type == "text/html":
+            resp.headers.setdefault("Cache-Control", self.cachecontrol)
+            resp.headers.setdefault("Pragma", self.pragma)
+            resp.headers.setdefault("Expires", self.expires)
+
         return resp
 
 app.mount("", NoCacheStaticFiles(packages=["labelu.internal"], html=True))