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

feat(health-check): add pipeline busy status to health endpoint

Browse Source
This commit is contained in:
yangdx
2025-03-26 12:41:49 +08:00
parent 57143fedcc
commit 1030068459
1 changed files with 37 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
lightrag/api/lightrag_server.py
View File

@@ -421,6 +421,9 @@ def create_app(args):
@app.get("/health", dependencies=[Depends(combined_auth)]) @app.get("/health", dependencies=[Depends(combined_auth)])
async def get_status(): async def get_status():
"""Get current system status""" """Get current system status"""
try:
pipeline_status = await get_namespace_data("pipeline_status")
username = os.getenv("AUTH_USERNAME") username = os.getenv("AUTH_USERNAME")
password = os.getenv("AUTH_PASSWORD") password = os.getenv("AUTH_PASSWORD")
if not (username and password): if not (username and password):
@@ -451,7 +454,11 @@ def create_app(args):
"core_version": core_version, "core_version": core_version,
"api_version": __api_version__, "api_version": __api_version__,
"auth_mode": auth_mode, "auth_mode": auth_mode,
"pipeline_busy": pipeline_status.get("busy", False)
} }
except Exception as e:
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 # Custom StaticFiles class to prevent caching of HTML files
class NoCacheStaticFiles(StaticFiles): class NoCacheStaticFiles(StaticFiles):