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

fix(health): bridge litellm_metadata into logging object in _batch_health_check

Browse source
This commit is contained in:
pramod 2026-06-18 21:14:36 +05:30
parent 343e453c2a
commit b9cfe4cb19
1 changed files with 21 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
litellm/litellm_core_utils/health_check_helpers.py
View file

@ -101,6 +101,27 @@ class HealthCheckHelpers:
"""
import litellm
# Bridge identity/tracking metadata into the logging object so callback
# handlers receive user_api_key_alias, tags, etc. in model_call_details.
#
# completion() does this via update_environment_variables (main.py), but
# alist_batches never makes that call. The @client decorator skips
# function_setup() when litellm_logging_obj already exists in kwargs
# (injected by ahealth_check), leaving litellm_params["metadata"] empty.
_logging_obj = filtered_model_params.get("litellm_logging_obj")
_litellm_metadata = filtered_model_params.get("litellm_metadata")
if _logging_obj is not None and isinstance(_litellm_metadata, dict):
_logging_obj.update_environment_variables(
model=filtered_model_params.get("model"),
user="",
optional_params={},
litellm_params={
"api_base": "",
"metadata": _litellm_metadata.copy(),
"litellm_metadata": _litellm_metadata.copy(),
},
)
if custom_llm_provider in LIST_BATCHES_SUPPORTED_PROVIDERS:
return await litellm.alist_batches(**filtered_model_params)
else: