mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-02 02:11:58 +00:00
39e31958f8
* test(proxy): move auth, hooks, policy_engine and client tests into tests/unit/proxy Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> * test(proxy): stub HIBP through respx by disabling the aiohttp transport Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> * test(proxy): share the httpx transport fixture across proxy unit tests Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> * test(proxy): restore proxy globals without a missing-value sentinel Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> * test(proxy): package moved dirs and stub the login breach check at the HTTP boundary Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> * test(proxy): isolate the mcp server manager per test Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --------- Co-authored-by: yuneng <yuneng@berri.ai> Co-authored-by: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
51 lines
1.9 KiB
Python
51 lines
1.9 KiB
Python
"""Regression guard for the enterprise hook registration / import cycle.
|
|
|
|
Python 3.13 is stricter about partially-initialized modules and surfaces
|
|
cycles that Python 3.12 silently tolerated. The previous bug:
|
|
|
|
litellm.proxy.hooks.__init__
|
|
-> enterprise.enterprise_hooks
|
|
-> litellm_enterprise.proxy.hooks.managed_files
|
|
-> litellm.llms.base_llm.managed_resources.isolation
|
|
-> litellm.proxy.management_endpoints.common_utils
|
|
-> litellm.proxy.utils (re-enters litellm.proxy.hooks mid-init)
|
|
|
|
silently swallowed the ImportError in `hooks/__init__.py`, leaving
|
|
``managed_files`` unregistered and the /files endpoint returning 500.
|
|
"""
|
|
|
|
import pytest
|
|
|
|
from litellm.proxy.hooks import PROXY_HOOKS, get_proxy_hook
|
|
|
|
|
|
def test_managed_files_hook_registered():
|
|
pytest.importorskip("litellm_enterprise")
|
|
assert "managed_files" in PROXY_HOOKS
|
|
hook_cls = get_proxy_hook("managed_files")
|
|
assert hook_cls.__name__ == "_PROXY_LiteLLMManagedFiles"
|
|
|
|
|
|
def test_managed_vector_stores_hook_registered():
|
|
pytest.importorskip("litellm_enterprise")
|
|
assert "managed_vector_stores" in PROXY_HOOKS
|
|
hook_cls = get_proxy_hook("managed_vector_stores")
|
|
assert hook_cls.__name__ == "_PROXY_LiteLLMManagedVectorStores"
|
|
|
|
|
|
def test_isolation_module_does_not_pull_in_proxy_utils():
|
|
"""Layering guard: litellm.llms.* must not transitively import
|
|
litellm.proxy.utils, which would reintroduce the import cycle."""
|
|
import importlib
|
|
import sys
|
|
|
|
for mod in [
|
|
"litellm.proxy.utils",
|
|
"litellm.proxy.management_endpoints.common_utils",
|
|
"litellm.llms.base_llm.managed_resources.isolation",
|
|
]:
|
|
sys.modules.pop(mod, None)
|
|
|
|
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
|