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

test(logging): cover the logger registry without the enterprise package
Some checks are pending
LiteLLM Rust / rust-lint (push) Waiting to run
Details
LiteLLM Rust / rust-test (push) Waiting to run
Details
LiteLLM Rust / rust-wheel (push) Waiting to run
Details

Browse source
This commit is contained in:
nate-berri 2026-10-11 18:35:16 +00:00
parent afaf339471
commit aaa41a35cb
No known key found for this signature in database
1 changed files with 23 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
tests/unit/litellm_core_utils/test_custom_logger_registry.py
View file

@ -474,3 +474,26 @@ def test_registry_maps_every_callback_name_to_a_class() -> None:
]
assert non_classes == []
def test_registry_builds_without_enterprise_callbacks_when_enterprise_is_missing(
monkeypatch: pytest.MonkeyPatch,
) -> None:
registry_module_name: Final = "litellm.litellm_core_utils.custom_logger_registry"
monkeypatch.setitem(sys.modules, "litellm_enterprise.enterprise_callbacks.pagerduty.pagerduty", None)
monkeypatch.setattr(
litellm.litellm_core_utils, "custom_logger_registry", importlib.import_module(registry_module_name)
)
monkeypatch.delitem(sys.modules, registry_module_name)
fresh_mapping: Final = importlib.import_module(
registry_module_name
).CustomLoggerRegistry.CALLBACK_CLASS_STR_TO_CLASS_TYPE
assert fresh_mapping.keys() == CustomLoggerRegistry.CALLBACK_CLASS_STR_TO_CLASS_TYPE.keys() - {
"pagerduty",
"generic_api",
"resend_email",
"sendgrid_email",
"smtp_email",
}