scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-03 02:22:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 80

refactor(tests): mirror warm_pool move — test_manager.py

Browse source 
Tracks LIT-2879.
This commit is contained in:
Ishaan Jaffer 2026-05-06 19:30:11 -07:00
parent 4618cba00e
commit 781f8ef2a9
No known key found for this signature in database
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/test_litellm/proxy/agent_session_endpoints/warm_pool/test_manager.py → tests/test_litellm/managed_agents/vms/warm_pool/test_manager.py
View file

@ -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,
)