From 8a90b93714dc57b93ec0ee8ca65d6a6808732776 Mon Sep 17 00:00:00 2001 From: Rodrigoue9 Date: Wed, 19 Aug 2026 10:11:50 -0300 Subject: [PATCH] fix(health_check): preserve resolved api_base and api_key in ahealth_check (#37321) --- litellm/main.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/litellm/main.py b/litellm/main.py index cc27da830d8..0b5e159bea4 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -8395,12 +8395,16 @@ async def ahealth_check( api_base_from_params: Final = model_params.get("api_base", None) api_key_from_params: Final = model_params.get("api_key", None) - model, custom_llm_provider, _, _ = get_llm_provider( + model, custom_llm_provider, dynamic_api_key, returned_api_base = get_llm_provider( model=model, custom_llm_provider=custom_llm_provider_from_params, api_base=api_base_from_params, api_key=api_key_from_params, ) + if returned_api_base is not None and not model_params.get("api_base"): + model_params["api_base"] = returned_api_base + if dynamic_api_key is not None and not model_params.get("api_key"): + model_params["api_key"] = dynamic_api_key if model in litellm.model_cost and mode is None: mode = litellm.model_cost[model].get("mode")