gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix linting

Browse Source
This commit is contained in:
yangdx
2025-05-21 16:46:36 +08:00
parent 45cebc71c5
commit 85bed30764
1 changed files with 16 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
lightrag/api/lightrag_server.py
View File

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