scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-13 08:52:02 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 353

test(proxy): cover managed files hook registration without the repo-only enterprise package

Browse source
This commit is contained in:
mateo-berri 2026-10-20 19:23:31 -07:00
parent 41890bdf64
commit c912562672
1 changed files with 22 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -14,8 +14,12 @@ silently swallowed the ImportError in `hooks/__init__.py`, leaving
``managed_files`` unregistered and the /files endpoint returning 500.
"""
import importlib
import sys
import pytest
import litellm.proxy
from litellm.proxy.hooks import PROXY_HOOKS, get_proxy_hook
@ -49,3 +53,21 @@ def test_isolation_module_does_not_pull_in_proxy_utils():
importlib.import_module("litellm.llms.base_llm.managed_resources.isolation")
assert "litellm.proxy.utils" not in sys.modules
assert "litellm.proxy.management_endpoints.common_utils" not in sys.modules
def test_managed_hooks_register_from_litellm_enterprise_without_the_repo_only_enterprise_package(
monkeypatch: pytest.MonkeyPatch,
):
pytest.importorskip("litellm_enterprise")
from litellm_enterprise.proxy.hooks.managed_files import PROXY_LiteLLMManagedFiles
from litellm_enterprise.proxy.hooks.managed_vector_stores import PROXY_LiteLLMManagedVectorStores
monkeypatch.setitem(sys.modules, "enterprise", None)
monkeypatch.setitem(sys.modules, "enterprise.enterprise_hooks", None)
monkeypatch.delitem(sys.modules, "litellm.proxy.hooks")
monkeypatch.delattr(litellm.proxy, "hooks")
wheel_install_hooks = 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