mirror of
https://github.com/BerriAI/litellm.git
synced 2025-12-06 11:33:26 +08:00
Revert "Add license metadata to health/readiness endpoint. (#15997)"
This reverts commit d89990e0c5.
This commit is contained in:
@@ -1048,28 +1048,6 @@ async def health_readiness():
|
||||
index_info = "index does not exist - error: " + str(e)
|
||||
cache_type = {"type": cache_type, "index_info": index_info}
|
||||
|
||||
# build license metadata
|
||||
try:
|
||||
from litellm.proxy.proxy_server import _license_check # type: ignore
|
||||
|
||||
license_available: bool = _license_check.is_premium() if _license_check else False
|
||||
license_expiration: Optional[str] = None
|
||||
|
||||
if getattr(_license_check, "airgapped_license_data", None):
|
||||
license_expiration = _license_check.airgapped_license_data.get( # type: ignore[arg-type]
|
||||
"expiration_date"
|
||||
)
|
||||
|
||||
license_metadata = {
|
||||
"license": {
|
||||
"has_license": license_available,
|
||||
"expiration_date": license_expiration,
|
||||
}
|
||||
}
|
||||
except Exception:
|
||||
# fail closed: don't let license check break readiness
|
||||
license_metadata = {"license": {"has_license": False, "expiration_date": None}}
|
||||
|
||||
# check DB
|
||||
if prisma_client is not None: # if db passed in, check if it's connected
|
||||
db_health_status = await _db_health_readiness_check()
|
||||
@@ -1080,7 +1058,6 @@ async def health_readiness():
|
||||
"litellm_version": version,
|
||||
"success_callbacks": success_callback_names,
|
||||
"use_aiohttp_transport": AsyncHTTPHandler._should_use_aiohttp_transport(),
|
||||
**license_metadata,
|
||||
**db_health_status,
|
||||
}
|
||||
else:
|
||||
@@ -1091,7 +1068,6 @@ async def health_readiness():
|
||||
"litellm_version": version,
|
||||
"success_callbacks": success_callback_names,
|
||||
"use_aiohttp_transport": AsyncHTTPHandler._should_use_aiohttp_transport(),
|
||||
**license_metadata,
|
||||
}
|
||||
except Exception as e:
|
||||
raise HTTPException(status_code=503, detail=f"Service Unhealthy ({str(e)})")
|
||||
|
||||
@@ -25,7 +25,10 @@ async def test_health_and_chat_completion():
|
||||
readiness_response = await response.json()
|
||||
assert readiness_response["status"] == "connected"
|
||||
|
||||
<<<<<<< HEAD
|
||||
|
||||
=======
|
||||
>>>>>>> parent of d89990e0c5 (Add license metadata to health/readiness endpoint. (#15997))
|
||||
# Test liveness endpoint
|
||||
async with session.get("http://0.0.0.0:4000/health/liveness") as response:
|
||||
assert response.status == 200
|
||||
|
||||
Reference in New Issue
Block a user