From ae6d3780ff321a6b1fa820549c41c6b326a97fa9 Mon Sep 17 00:00:00 2001
From: Ishaan Jaffer <ishaanjaffer0324@gmail.com>
Date: Wed, 6 May 2026 19:30:44 -0700
Subject: [PATCH] refactor(session_endpoints): update vm_providers + warm_pool
 import paths

Tracks LIT-2879.
---
 litellm/proxy/agent_session_endpoints/session_endpoints.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/litellm/proxy/agent_session_endpoints/session_endpoints.py b/litellm/proxy/agent_session_endpoints/session_endpoints.py
index ef2ea380b0b..beea8813170 100644
--- a/litellm/proxy/agent_session_endpoints/session_endpoints.py
+++ b/litellm/proxy/agent_session_endpoints/session_endpoints.py
@@ -62,13 +62,13 @@ from litellm.proxy.agent_session_endpoints.serialization import (
 from litellm.proxy.agent_session_endpoints.session_status import (
     refresh_session_status_from_runs,
 )
-from litellm.proxy.agent_session_endpoints.vm_providers import (
+from litellm.managed_agents.vms import (
     ProvisionContext,
     Repo,
     VMHandle,
     get_vm_provider,
 )
-from litellm.proxy.agent_session_endpoints.warm_pool.attach import (
+from litellm.managed_agents.vms.warm_pool.attach import (
     AttachResult,
     WarmPoolEmptyError,
     attach_warm_vm,