diff --git a/litellm/proxy/agent_session_endpoints/warm_pool/hydrate.py b/litellm/managed_agents/vms/warm_pool/hydrate.py
similarity index 98%
rename from litellm/proxy/agent_session_endpoints/warm_pool/hydrate.py
rename to litellm/managed_agents/vms/warm_pool/hydrate.py
index 93be410dd9f..45da97f6f73 100644
--- a/litellm/proxy/agent_session_endpoints/warm_pool/hydrate.py
+++ b/litellm/managed_agents/vms/warm_pool/hydrate.py
@@ -17,7 +17,7 @@ from datetime import datetime, timezone
 from typing import Any, Dict, List, Optional
 
 from litellm._logging import verbose_proxy_logger
-from litellm.proxy.agent_session_endpoints.warm_pool.types import (
+from litellm.managed_agents.vms.warm_pool.types import (
     AgentConfig,
     HydratePayload,
     NetworkAccess,