From 32cfd1b00c72460d3672b13abcb28b1e9fed1011 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Wed, 6 May 2026 19:30:02 -0700 Subject: [PATCH] refactor(managed_agents/vms/warm_pool): move warm_pool/manager.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/manager.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) rename litellm/{proxy/agent_session_endpoints => managed_agents/vms}/warm_pool/manager.py (97%) diff --git a/litellm/proxy/agent_session_endpoints/warm_pool/manager.py b/litellm/managed_agents/vms/warm_pool/manager.py similarity index 97% rename from litellm/proxy/agent_session_endpoints/warm_pool/manager.py rename to litellm/managed_agents/vms/warm_pool/manager.py index d38a3c665df..f5a1cd67f13 100644 --- a/litellm/proxy/agent_session_endpoints/warm_pool/manager.py +++ b/litellm/managed_agents/vms/warm_pool/manager.py @@ -23,16 +23,16 @@ from datetime import datetime, timedelta, timezone from typing import Any, Dict, List, Optional from litellm._logging import verbose_proxy_logger -from litellm.proxy.agent_session_endpoints.vm_providers.base import ( +from litellm.managed_agents.vms.base import ( AwsCreds, Ec2Config, ProvisionContext, Repo, ) -from litellm.proxy.agent_session_endpoints.vm_providers.factory import ( +from litellm.managed_agents.vms.factory import ( build_vm_provider, ) -from litellm.proxy.agent_session_endpoints.vm_providers.team_config import ( +from litellm.managed_agents.vms.team_config import ( get_team_vm_config, ) @@ -330,7 +330,7 @@ class WarmPoolManager: provider = build_vm_provider( {"vm_provider": "ec2", "ec2": self._provider_settings} ) - from litellm.proxy.agent_session_endpoints.vm_providers.base import ( + from litellm.managed_agents.vms.base import ( VMHandle, )