diff --git a/litellm/proxy/agent_session_endpoints/vm_providers/registry.py b/litellm/managed_agents/vms/registry.py
similarity index 90%
rename from litellm/proxy/agent_session_endpoints/vm_providers/registry.py
rename to litellm/managed_agents/vms/registry.py
index 3127ed3767c..b14b545dd1e 100644
--- a/litellm/proxy/agent_session_endpoints/vm_providers/registry.py
+++ b/litellm/managed_agents/vms/registry.py
@@ -12,8 +12,8 @@ a setup hook just to use the noop.
 
 from typing import Dict
 
-from litellm.proxy.agent_session_endpoints.vm_providers.base import AgentVMProvider
-from litellm.proxy.agent_session_endpoints.vm_providers.noop import NoopProvider
+from litellm.managed_agents.vms.base import AgentVMProvider
+from litellm.managed_agents.vms.noop import NoopProvider
 
 _REGISTRY: Dict[str, AgentVMProvider] = {}