scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-02 02:11:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 101

refactor(tests/conftest): update vm_providers import path

Browse source 
Tracks LIT-2879.
This commit is contained in:
Ishaan Jaffer 2026-05-06 19:30:44 -07:00
parent 4033702a53
commit f554d17e42
No known key found for this signature in database
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/test_litellm/proxy/agent_session_endpoints/conftest.py
View file

@ -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,
)