diff --git a/litellm/proxy/agent_session_endpoints/warm_pool/transports/__init__.py b/litellm/managed_agents/vms/warm_pool/transports/__init__.py
similarity index 88%
rename from litellm/proxy/agent_session_endpoints/warm_pool/transports/__init__.py
rename to litellm/managed_agents/vms/warm_pool/transports/__init__.py
index 7de11fcf929..f99c0c52fa8 100644
--- a/litellm/proxy/agent_session_endpoints/warm_pool/transports/__init__.py
+++ b/litellm/managed_agents/vms/warm_pool/transports/__init__.py
@@ -10,7 +10,7 @@ Public surface:
 - ``SSMHydrateTransport`` — the production transport
 """
 
-from litellm.proxy.agent_session_endpoints.warm_pool.transports.ssm import (
+from litellm.managed_agents.vms.warm_pool.transports.ssm import (
     HydrateTransport,
     HydrateTransportError,
     SSMHydrateTransport,