scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-14 09:03:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 214

Merge b83a135999 into 89cc395a88

Browse source
This commit is contained in:
Maha Rana Yadavalli 2026-11-05 18:54:15 +00:00 • committed by GitHub
commit 681b2f68bc
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 14 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/proxy/common_utils/callback_utils.py
View file

@ -409,7 +409,7 @@ def initialize_callbacks_on_proxy(
PrometheusLogger._mount_metrics_endpoint()
else:
litellm.callbacks = [
litellm.callbacks.append(
_loaded_callback_or_raise(
entry=value,
loaded=_LOADED_CALLBACK.validate_python(
@ -419,7 +419,7 @@ def initialize_callbacks_on_proxy(
)
),
)
]
)
verbose_proxy_logger.debug("%s Initialized Callbacks - %s %s", blue_color_code, litellm.callbacks, reset_color_code)

12
tests/unit/proxy/common_utils/test_callback_utils.py
View file

@ -775,3 +775,15 @@ def test_initialize_callbacks_on_proxy_accepts_instance_non_list_value(probe_con
assert len(litellm.callbacks) == 1
assert isinstance(litellm.callbacks[0], CustomLogger)
@pytest.mark.parametrize("as_list", [False, True], ids=["string_value", "list_value"])
def test_initialize_callbacks_on_proxy_keeps_previously_registered_callbacks(probe_config_path: str, as_list: bool):
entry: Final = f"{_PROBE_MODULE_NAME}.proxy_handler_instance"
internal_callback: Final = CustomLogger()
litellm.callbacks.append(internal_callback)
_load_callbacks([entry] if as_list else entry, probe_config_path)
assert litellm.callbacks[0] is internal_callback
assert [type(cb).__name__ for cb in litellm.callbacks[1:]] == ["FloorMaxTokens"]