scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-12 08:42:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 207

test(proxy): annotate the hook registry test bindings as Final

Browse source
This commit is contained in:
mateo-berri 2026-10-20 21:07:52 -07:00
parent f57972b101
commit 6336e97328
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tests/unit/proxy/hooks/test_proxy_hooks_init.py
View file

@ -17,6 +17,7 @@ silently swallowed the ImportError in `hooks/__init__.py`, leaving
import importlib
import logging
import sys
from typing import Final
import pytest
@ -69,7 +70,7 @@ def test_managed_hooks_register_from_litellm_enterprise_without_the_repo_only_en
monkeypatch.delitem(sys.modules, "litellm.proxy.hooks")
monkeypatch.delattr(litellm.proxy, "hooks")
wheel_install_hooks = importlib.import_module("litellm.proxy.hooks")
wheel_install_hooks: Final = importlib.import_module("litellm.proxy.hooks")
assert wheel_install_hooks.get_proxy_hook("managed_files") is PROXY_LiteLLMManagedFiles
assert wheel_install_hooks.get_proxy_hook("managed_vector_stores") is PROXY_LiteLLMManagedVectorStores
@ -85,7 +86,7 @@ def test_proxy_hooks_still_import_and_warn_when_litellm_enterprise_cannot_load(
verbose_proxy_logger.addHandler(caplog.handler)
try:
with caplog.at_level(logging.WARNING, logger=verbose_proxy_logger.name):
hooks_without_enterprise = importlib.import_module("litellm.proxy.hooks")
hooks_without_enterprise: Final = importlib.import_module("litellm.proxy.hooks")
finally:
verbose_proxy_logger.removeHandler(caplog.handler)