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,