From da531564f6bd9a804eb12b628218a05a4d3e9948 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/transports/ssm.py out of proxy Tracks LIT-2879. --- .../vms}/warm_pool/transports/ssm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) rename litellm/{proxy/agent_session_endpoints => managed_agents/vms}/warm_pool/transports/ssm.py (97%) diff --git a/litellm/proxy/agent_session_endpoints/warm_pool/transports/ssm.py b/litellm/managed_agents/vms/warm_pool/transports/ssm.py similarity index 97% rename from litellm/proxy/agent_session_endpoints/warm_pool/transports/ssm.py rename to litellm/managed_agents/vms/warm_pool/transports/ssm.py index a7f141c7313..62d160d846e 100644 --- a/litellm/proxy/agent_session_endpoints/warm_pool/transports/ssm.py +++ b/litellm/managed_agents/vms/warm_pool/transports/ssm.py @@ -26,8 +26,8 @@ from concurrent.futures import ThreadPoolExecutor from typing import Any, Dict, Optional, Protocol from litellm._logging import verbose_proxy_logger -from litellm.proxy.agent_session_endpoints.vm_providers.base import AwsCreds -from litellm.proxy.agent_session_endpoints.warm_pool.types import HydratePayload +from litellm.managed_agents.vms.base import AwsCreds +from litellm.managed_agents.vms.warm_pool.types import HydratePayload # Silence boto3 debug logging (same defensive measure as ec2.py). for _name in ("boto3", "botocore", "urllib3.connectionpool"):