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_attach.py

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

10
tests/test_litellm/proxy/agent_session_endpoints/warm_pool/test_attach.py → tests/test_litellm/managed_agents/vms/warm_pool/test_attach.py
View file

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