Merge pull request #1605 from HKUDS/fix-mime-for-windows

lightrag/api/lightrag_server.py

@@ -478,16 +478,31 @@ def create_app(args):
             logger.error(f"Error getting health status: {str(e)}")
             raise HTTPException(status_code=500, detail=str(e))
 
-    # Custom StaticFiles class to prevent caching of HTML files
-    class NoCacheStaticFiles(StaticFiles):
+    # Custom StaticFiles class for smart caching
+    class SmartStaticFiles(StaticFiles):  # Renamed from NoCacheStaticFiles
         async def get_response(self, path: str, scope):
             response = await super().get_response(path, scope)
+
             if path.endswith(".html"):
                 response.headers["Cache-Control"] = (
                     "no-cache, no-store, must-revalidate"
                 )
                 response.headers["Pragma"] = "no-cache"
                 response.headers["Expires"] = "0"
+            elif (
+                "/assets/" in path
+            ):  # Assets (JS, CSS, images, fonts) generated by Vite with hash in filename
+                response.headers["Cache-Control"] = (
+                    "public, max-age=31536000, immutable"
+                )
+            # Add other rules here if needed for non-HTML, non-asset files
+
+            # Ensure correct Content-Type
+            if path.endswith(".js"):
+                response.headers["Content-Type"] = "application/javascript"
+            elif path.endswith(".css"):
+                response.headers["Content-Type"] = "text/css"
+
             return response
 
     # Webui mount webui/index.html
@@ -495,7 +510,9 @@ def create_app(args):
     static_dir.mkdir(exist_ok=True)
     app.mount(
         "/webui",
-        NoCacheStaticFiles(directory=static_dir, html=True, check_dir=True),
+        SmartStaticFiles(
+            directory=static_dir, html=True, check_dir=True
+        ),  # Use SmartStaticFiles
         name="webui",
     )