From 9c82e973976003be6eafd930e9b226fcb0c646f4 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Wed, 6 May 2026 19:30:01 -0700 Subject: [PATCH] refactor(managed_agents/vms/warm_pool): move warm_pool/__init__.py out of proxy MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Phase 2: warm pool is part of VM management — moves to litellm/managed_agents/vms/warm_pool/ alongside the providers. Tracks LIT-2879. --- .../vms}/warm_pool/__init__.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) rename litellm/{proxy/agent_session_endpoints => managed_agents/vms}/warm_pool/__init__.py (73%) diff --git a/litellm/proxy/agent_session_endpoints/warm_pool/__init__.py b/litellm/managed_agents/vms/warm_pool/__init__.py similarity index 73% rename from litellm/proxy/agent_session_endpoints/warm_pool/__init__.py rename to litellm/managed_agents/vms/warm_pool/__init__.py index eec467b0ddc..2a7df77ec7d 100644 --- a/litellm/proxy/agent_session_endpoints/warm_pool/__init__.py +++ b/litellm/managed_agents/vms/warm_pool/__init__.py @@ -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,