mirror of
https://github.com/BerriAI/litellm.git
synced 2026-11-09 08:18:12 +00:00
618d5aec71
* fix(mcp): load deprecated package aliases on demand * fix(mcp): preserve lazy child access and module reloads
31 lines
1.2 KiB
Python
31 lines
1.2 KiB
Python
import importlib
|
|
import sys
|
|
from collections.abc import Mapping
|
|
|
|
import pytest
|
|
|
|
import litellm.mcp_client
|
|
import litellm.mcp_client.client
|
|
import litellm.mcp_client.tools
|
|
|
|
OLD_PACKAGE = "litellm.experimental_mcp_client"
|
|
|
|
|
|
def test_old_package_warns_and_deep_imports_resolve_to_mcp_client(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
for name in tuple(sys.modules):
|
|
if name.startswith(OLD_PACKAGE):
|
|
monkeypatch.delitem(sys.modules, name)
|
|
|
|
with pytest.warns(DeprecationWarning, match="litellm.mcp_client"):
|
|
import litellm.experimental_mcp_client as old
|
|
|
|
assert old.client is litellm.mcp_client.client
|
|
assert old.tools is litellm.mcp_client.tools
|
|
assert old.__all__ == litellm.mcp_client.__all__
|
|
assert old.load_mcp_tools is litellm.mcp_client.load_mcp_tools
|
|
assert old.call_openai_tool is litellm.mcp_client.call_openai_tool
|
|
assert importlib.import_module(f"{OLD_PACKAGE}.client") is litellm.mcp_client.client
|
|
assert importlib.import_module(f"{OLD_PACKAGE}.tools") is litellm.mcp_client.tools
|
|
attrs: Mapping[str, object] = vars(old)
|
|
assert attrs["tools"] is litellm.mcp_client.tools
|
|
assert attrs["client"] is litellm.mcp_client.client
|