mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-12 03:48:56 +00:00
58258409c9
* refactor(proxy): expose public names for private proxy helpers Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> * fix(proxy): keep original class names behind public aliases Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> * fix(proxy): preserve internal callback filtering Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> * fix(proxy): keep _PROXY_ class names for managed files hooks Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> * fix(proxy): keep old private names in package exports Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> * fix(proxy): preserve recursive auth helper name Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> * test(proxy): align MCP limiter tests with server enforcement Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> * test(proxy): match main's MCP limiter tests Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> * fix(proxy): keep old private names bound in importing modules Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> * fix(proxy): preserve compatibility imports through strict lint Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> * test(proxy): use exact pyright suppression in password helper test Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> * style(proxy): add reasons to compatibility import noqa comments Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> * test(proxy): update IN-list baseline for renamed helpers Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --------- 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
|