refactor(managed_agents/vms): move vm_providers/__init__.py out of proxy

Phase 2 architecture rule: VM management code lives in
litellm/managed_agents/vms/, not litellm/proxy/. The proxy is control
plane only.

Tracks LIT-2879.
This commit is contained in:
Ishaan Jaffer 2026-05-06 19:29:53 -07:00
parent f070f4ff12
commit 9c7eec479f
No known key found for this signature in database

View file

@ -11,7 +11,7 @@ Public re-exports:
- `ProvisionError` / `InvalidCredentialsError` — error types
"""
from litellm.proxy.agent_session_endpoints.vm_providers.base import (
from litellm.managed_agents.vms.base import (
AgentVMProvider,
AwsCreds,
Ec2Config,
@ -23,13 +23,13 @@ from litellm.proxy.agent_session_endpoints.vm_providers.base import (
VMState,
VMStatus,
)
from litellm.proxy.agent_session_endpoints.vm_providers.ec2 import EC2Provider
from litellm.proxy.agent_session_endpoints.vm_providers.factory import (
from litellm.managed_agents.vms.ec2 import EC2Provider
from litellm.managed_agents.vms.factory import (
SUPPORTED_PROVIDERS,
build_vm_provider,
)
from litellm.proxy.agent_session_endpoints.vm_providers.noop import NoopProvider
from litellm.proxy.agent_session_endpoints.vm_providers.registry import (
from litellm.managed_agents.vms.noop import NoopProvider
from litellm.managed_agents.vms.registry import (
get_vm_provider,
register_vm_provider,
reset_vm_provider_registry,