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

(feat) proxy return version in /health/readiness

Browse source
This commit is contained in:
ishaan-jaff 2024-01-29 15:28:08 -08:00
parent 6097102c21
commit 67e8a0115d
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/proxy/proxy_server.py
View file

@ -3617,6 +3617,8 @@ async def health_readiness():
cache_type = None
if litellm.cache is not None:
cache_type = litellm.cache.type
from litellm._version import version
if prisma_client is not None: # if db passed in, check if it's connected
if prisma_client.db.is_connected() == True:
response_object = {"db": "connected"}
@ -3625,6 +3627,7 @@ async def health_readiness():
"status": "healthy",
"db": "connected",
"cache": cache_type,
"litellm_version": version,
"success_callbacks": litellm.success_callback,
}
else:
@ -3632,6 +3635,7 @@ async def health_readiness():
"status": "healthy",
"db": "Not connected",
"cache": cache_type,
"litellm_version": version,
"success_callbacks": litellm.success_callback,
}
raise HTTPException(status_code=503, detail="Service Unhealthy")