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

Phase 2: warm pool is part of VM management — moves to
litellm/managed_agents/vms/warm_pool/ alongside the providers.

Tracks LIT-2879.
This commit is contained in:
Ishaan Jaffer 2026-05-06 19:30:01 -07:00
parent 7f289b6738
commit 9c82e97397
No known key found for this signature in database

View file

@ -10,15 +10,15 @@ Public surface:
- ``HydratePayload`` — wire format the daemon receives
"""
from litellm.proxy.agent_session_endpoints.warm_pool.attach import (
from litellm.managed_agents.vms.warm_pool.attach import (
AttachResult,
attach_warm_vm,
)
from litellm.proxy.agent_session_endpoints.warm_pool.hydrate import (
from litellm.managed_agents.vms.warm_pool.hydrate import (
build_hydrate_payload,
)
from litellm.proxy.agent_session_endpoints.warm_pool.manager import WarmPoolManager
from litellm.proxy.agent_session_endpoints.warm_pool.types import (
from litellm.managed_agents.vms.warm_pool.manager import WarmPoolManager
from litellm.managed_agents.vms.warm_pool.types import (
AgentConfig,
HydratePayload,
NetworkAccess,