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

Added special endpoint API key checks

Browse Source
This commit is contained in:
yangdx
2025-03-24 15:02:10 +08:00
parent d05cf286f4
commit deef7182b9
1 changed files with 24 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
lightrag/api/utils_api.py
View File

@@ -113,6 +113,20 @@ def get_combined_auth_dependency(api_key: Optional[str] = None):
): ):
return # API key validation successful return # API key validation successful
# Specail endpoint not accept token
if ( api_key_configured and is_special_endpoint ):
# Special endpoint but API key validation failed, return 403 error
if api_key_header_value:
raise HTTPException(
status_code=HTTP_403_FORBIDDEN,
detail="Invalid API Key",
)
else:
raise HTTPException(
status_code=HTTP_403_FORBIDDEN,
detail="API Key required",
)
# 4. Validate token # 4. Validate token
if token: if token:
try: try:
@@ -143,10 +157,16 @@ def get_combined_auth_dependency(api_key: Optional[str] = None):
# 5. No token and API key validation failed, return 403 error # 5. No token and API key validation failed, return 403 error
if api_key_configured: if api_key_configured:
if (api_key_header_value is None):
raise HTTPException( raise HTTPException(
status_code=HTTP_403_FORBIDDEN, status_code=HTTP_403_FORBIDDEN,
detail="API Key required or login authentication required.", detail="API Key required or login authentication required.",
) )
else:
raise HTTPException(
status_code=HTTP_403_FORBIDDEN,
detail="Invalid API Key or login authentication required.",
)
else: else:
raise HTTPException( raise HTTPException(
status_code=HTTP_403_FORBIDDEN, detail="Login authentication required." status_code=HTTP_403_FORBIDDEN, detail="Login authentication required."