From e5fd4855fe99cbdeef4d19e546130a6bea30b31c Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Wed, 6 May 2026 19:30:11 -0700 Subject: [PATCH] =?UTF-8?q?refactor(tests):=20mirror=20warm=5Fpool=20move?= =?UTF-8?q?=20=E2=80=94=20test=5Fattach.py?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Tracks LIT-2879. --- .../vms}/warm_pool/test_attach.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) rename tests/test_litellm/{proxy/agent_session_endpoints => managed_agents/vms}/warm_pool/test_attach.py (95%) diff --git a/tests/test_litellm/proxy/agent_session_endpoints/warm_pool/test_attach.py b/tests/test_litellm/managed_agents/vms/warm_pool/test_attach.py similarity index 95% rename from tests/test_litellm/proxy/agent_session_endpoints/warm_pool/test_attach.py rename to tests/test_litellm/managed_agents/vms/warm_pool/test_attach.py index a5b547a47db..3545582e649 100644 --- a/tests/test_litellm/proxy/agent_session_endpoints/warm_pool/test_attach.py +++ b/tests/test_litellm/managed_agents/vms/warm_pool/test_attach.py @@ -1,4 +1,4 @@ -"""Unit tests for `litellm/proxy/agent_session_endpoints/warm_pool/attach.py`. +"""Unit tests for `litellm/managed_agents/vms/warm_pool/attach.py`. Cover the race-safe attach path: * the CAS via `update_many(state='warm')` blocks double-attach @@ -22,9 +22,9 @@ import pytest os.environ.setdefault("LITELLM_SALT_KEY", "sk-test-salt-key-do-not-use-in-prod") -from litellm.proxy.agent_session_endpoints.vm_providers.base import AwsCreds, Ec2Config -from litellm.proxy.agent_session_endpoints.vm_providers.team_config import TeamVMConfig -from litellm.proxy.agent_session_endpoints.warm_pool.attach import ( +from litellm.managed_agents.vms.base import AwsCreds, Ec2Config +from litellm.managed_agents.vms.team_config import TeamVMConfig +from litellm.managed_agents.vms.warm_pool.attach import ( WarmPoolEmptyError, attach_warm_vm, ) @@ -127,7 +127,7 @@ def fake_team_creds(monkeypatch): ) monkeypatch.setattr( - "litellm.proxy.agent_session_endpoints.warm_pool.attach.get_team_vm_config", + "litellm.managed_agents.vms.warm_pool.attach.get_team_vm_config", fake, )