From f554d17e424db069cda0b84d82338d64f0fd706f Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Wed, 6 May 2026 19:30:44 -0700 Subject: [PATCH] refactor(tests/conftest): update vm_providers import path Tracks LIT-2879. --- .../test_litellm/proxy/agent_session_endpoints/conftest.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_litellm/proxy/agent_session_endpoints/conftest.py b/tests/test_litellm/proxy/agent_session_endpoints/conftest.py index dc3a01d0c16..047922e0829 100644 --- a/tests/test_litellm/proxy/agent_session_endpoints/conftest.py +++ b/tests/test_litellm/proxy/agent_session_endpoints/conftest.py @@ -312,7 +312,7 @@ class _RecordingNoopProvider: name = "noop" def __init__(self) -> None: - from litellm.proxy.agent_session_endpoints.vm_providers import NoopProvider + from litellm.managed_agents.vms import NoopProvider self._inner = NoopProvider() self.provision_calls: list = [] @@ -342,10 +342,10 @@ class _RecordingNoopProvider: @pytest.fixture def noop_provider(monkeypatch): """Reset the VM provider registry to a fresh recording noop.""" - from litellm.proxy.agent_session_endpoints.vm_providers import ( + from litellm.managed_agents.vms import ( register_vm_provider, ) - from litellm.proxy.agent_session_endpoints.vm_providers.registry import ( + from litellm.managed_agents.vms.registry import ( reset_vm_provider_registry, )