scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-29 01:42:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 79

test(proxy): cover non-compatible string and failure-callback branches

Browse source
This commit is contained in:
wadrzl55333 2026-08-28 16:06:50 +08:00
parent 817b0a9837
commit 2bafe2c978
1 changed files with 15 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
tests/proxy_unit_tests/test_proxy_config_unit_test.py
View file

@ -178,8 +178,15 @@ async def test_config_file_success_callback_registers_custom_logger_instance(mon
monkeypatch.setattr(litellm, "success_callback", [])
monkeypatch.setattr(litellm, "_async_success_callback", [])
monkeypatch.setattr(litellm, "failure_callback", [])
monkeypatch.setattr(litellm, "_async_failure_callback", [])
config_content = {"litellm_settings": {"success_callback": ["langfuse"]}}
config_content = {
"litellm_settings": {
"success_callback": ["langfuse", "sentry"],
"failure_callback": ["langfuse"],
}
}
with tempfile.NamedTemporaryFile(mode="w", suffix=".yaml", delete=False) as temp_file:
yaml.dump(config_content, temp_file)
temp_file_path = temp_file.name
@ -191,11 +198,18 @@ async def test_config_file_success_callback_registers_custom_logger_instance(mon
config_file_path=temp_file_path,
)
# custom-logger-compatible names are registered as instances, not strings
assert "langfuse" not in litellm.success_callback
# non-compatible names still take the string path
assert "sentry" in litellm.success_callback
num_langfuse_instances = sum(
isinstance(callback, LangfusePromptManagement) for callback in litellm._async_success_callback
)
assert num_langfuse_instances == 1
num_failure_instances = sum(
isinstance(callback, LangfusePromptManagement) for callback in litellm._async_failure_callback
)
assert num_failure_instances == 1
finally:
os.unlink(temp_file_path)