From 67e8a0115d9252b7df6f994dd3eb9fdee36428ef Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Mon, 29 Jan 2024 15:28:08 -0800 Subject: [PATCH] (feat) proxy return version in /health/readiness --- litellm/proxy/proxy_server.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index d9944e6e1..dd9b509d9 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -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")