mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-12 03:48:56 +00:00
c3a23fe499
* refactor: expose core private symbols with compatibility aliases Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> * types: narrow core migration diagnostics Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> * fix: preserve runtime behavior in core symbol migration Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> * fix: preserve core private usage migration behavior Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> * fix: preserve private value rebinding compatibility Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> * fix: restore optional imports and cover public helpers Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> * test: exempt router property from call coverage Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> * test: align recursive detector ignore names Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --------- Co-authored-by: mateo <mateo@berri.ai> Co-authored-by: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
51 lines
1.8 KiB
Python
51 lines
1.8 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
|