From 781f8ef2a9f27eff163571dcc144a7e839917dbe 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=5Fmanager.py?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Tracks LIT-2879. --- .../vms}/warm_pool/test_manager.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) rename tests/test_litellm/{proxy/agent_session_endpoints => managed_agents/vms}/warm_pool/test_manager.py (97%) diff --git a/tests/test_litellm/proxy/agent_session_endpoints/warm_pool/test_manager.py b/tests/test_litellm/managed_agents/vms/warm_pool/test_manager.py similarity index 97% rename from tests/test_litellm/proxy/agent_session_endpoints/warm_pool/test_manager.py rename to tests/test_litellm/managed_agents/vms/warm_pool/test_manager.py index 6592fb364f7..3a0aec4d590 100644 --- a/tests/test_litellm/proxy/agent_session_endpoints/warm_pool/test_manager.py +++ b/tests/test_litellm/managed_agents/vms/warm_pool/test_manager.py @@ -1,4 +1,4 @@ -"""Unit tests for `litellm/proxy/agent_session_endpoints/warm_pool/manager.py`. +"""Unit tests for `litellm/managed_agents/vms/warm_pool/manager.py`. Cover the maintenance loop logic without hitting AWS or the database: * refill spawns the deficit when fewer warm VMs exist than configured @@ -17,7 +17,7 @@ from unittest.mock import AsyncMock, patch import pytest -from litellm.proxy.agent_session_endpoints.warm_pool.manager import WarmPoolManager +from litellm.managed_agents.vms.warm_pool.manager import WarmPoolManager # --------------------------------------------------------------------------- @@ -187,7 +187,7 @@ async def test_refill_spawns_deficit(fake_prisma, manager): @pytest.mark.asyncio async def test_refill_caps_at_max_concurrent(fake_prisma, manager): """Pool size 100, 0 warm -> spawn at most MAX_CONCURRENT_SPAWNS_PER_TEAM.""" - from litellm.proxy.agent_session_endpoints.warm_pool.manager import ( + from litellm.managed_agents.vms.warm_pool.manager import ( MAX_CONCURRENT_SPAWNS_PER_TEAM, )