scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-21 00:21:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 73

Fix litellm/tests/proxy_unit_tests/test_proxy_server.py

Browse source
This commit is contained in:
Sameer Kankute 2026-04-02 18:19:46 +05:30
parent 8bf927e3a3
commit fc155c7bfa
No known key found for this signature in database
1 changed files with 1 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
litellm/proxy/health_check.py
View file

@ -22,6 +22,7 @@ ILLEGAL_DISPLAY_PARAMS = [
"aws_access_key_id",
"aws_secret_access_key",
"exception", # internal; not JSON-serializable, never for display
"litellm_metadata", # internal tracking metadata with auth objects; not for display
]
MINIMAL_DISPLAY_PARAMS = ["model", "mode_error"]
@ -224,15 +225,11 @@ async def _perform_health_check(
cleaned = _clean_endpoint_data({**litellm_params, **is_healthy}, details)
if _model_id:
cleaned["model_id"] = _model_id
if "exception" in is_healthy:
cleaned["exception"] = is_healthy["exception"]
unhealthy_endpoints.append(cleaned)
else:
cleaned = _clean_endpoint_data(litellm_params, details)
if _model_id:
cleaned["model_id"] = _model_id
if isinstance(is_healthy, Exception):
cleaned["exception"] = is_healthy
unhealthy_endpoints.append(cleaned)
return healthy_endpoints, unhealthy_endpoints