scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-06 08:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 79

Merge d30747c88c into 11a02b9581

Browse source
This commit is contained in:
devin-ai-integration[bot] 2026-09-03 08:32:15 -07:00 • committed by GitHub
commit 08ba714cb9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 25 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/proxy/proxy_server.py
View file

@ -5318,6 +5318,7 @@ class ProxyConfig:
# these are litellm callbacks - "langfuse", "sentry", "wandb"
else:
litellm.logging_callback_manager.add_litellm_success_callback(callback)
_add_custom_logger_callback_to_specific_event(callback, "success")
if "prometheus" in callback:
from litellm.integrations.prometheus import (
PrometheusLogger,
@ -5345,6 +5346,7 @@ class ProxyConfig:
# these are litellm callbacks - "langfuse", "sentry", "wandb"
else:
litellm.logging_callback_manager.add_litellm_failure_callback(callback)
_add_custom_logger_callback_to_specific_event(callback, "failure")
print( # noqa: T201
f"{blue_color_code} Initialized Failure Callbacks - {litellm.failure_callback} {reset_color_code}"
)

23
tests/test_litellm/proxy/test_proxy_server.py
View file

@ -3153,6 +3153,29 @@ async def test_custom_ui_sso_sign_in_handler_config_loading():
os.unlink(config_file_path)
@pytest.mark.asyncio
async def test_load_config_eagerly_initializes_string_success_callback(tmp_path, monkeypatch):
from litellm.integrations.generic_api.generic_api_callback import GenericAPILogger
from litellm.proxy.proxy_server import ProxyConfig
config_file = tmp_path / "config.yaml"
config_file.write_text(
"model_list: []\n"
"litellm_settings:\n"
" success_callback:\n"
" - generic_api\n"
)
monkeypatch.setenv("GENERIC_LOGGER_ENDPOINT", "http://127.0.0.1:8899/")
monkeypatch.setattr(litellm, "success_callback", [])
monkeypatch.setattr(litellm, "_async_success_callback", [])
monkeypatch.setattr(litellm, "failure_callback", [])
monkeypatch.setattr(litellm, "_async_failure_callback", [])
await ProxyConfig().load_config(router=MagicMock(), config_file_path=str(config_file))
assert any(isinstance(callback, GenericAPILogger) for callback in litellm._async_success_callback)
assert "generic_api" not in litellm.success_callback
@pytest.mark.asyncio
async def test_load_config_max_budget_env_var_coerced_to_float(tmp_path, monkeypatch):
"""