phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(feat) proxy: protect health endpoint

Browse source
This commit is contained in:
ishaan-jaff 2023-12-06 18:14:54 -08:00
parent 1b43c7120c
commit 19b1deb200
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/proxy/proxy_server.py
View file

@ -1281,7 +1281,7 @@ async def retrieve_server_log(request: Request):
async def test_endpoint(request: Request): async def test_endpoint(request: Request):
return {"route": request.url.path} return {"route": request.url.path}
@router.get("/health", description="Check the health of all the endpoints in config.yaml", tags=["health"]) @router.get("/health", description="Check the health of all the endpoints in config.yaml", tags=["health"], dependencies=[Depends(user_api_key_auth)])
async def health_endpoint(request: Request, model: Optional[str] = fastapi.Query(None, description="Specify the model name (optional)")): async def health_endpoint(request: Request, model: Optional[str] = fastapi.Query(None, description="Specify the model name (optional)")):
global llm_model_list global llm_model_list