scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-08 03:08:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 114

fix(health_check): preserve resolved api_base and api_key in ahealth_check (#37321)

Browse source
This commit is contained in:
Rodrigoue9 2026-08-19 10:11:50 -03:00
parent 0b82b087fd
commit 8a90b93714
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
litellm/main.py
View file

@ -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")