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

Browse source 
Tracks LIT-2879.
This commit is contained in:
Ishaan Jaffer 2026-05-06 19:30:11 -07:00
parent e5fd4855fe
commit 4618cba00e
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_hydrate.py → tests/test_litellm/managed_agents/vms/warm_pool/test_hydrate.py
View file

@ -1,4 +1,4 @@
"""Unit tests for `litellm/proxy/agent_session_endpoints/warm_pool/hydrate.py`.
"""Unit tests for `litellm/managed_agents/vms/warm_pool/hydrate.py`.
Verify the payload builder:
* respects scope filtering — out-of-scope secrets never land in the payload
@ -22,7 +22,7 @@ import pytest
os.environ.setdefault("LITELLM_SALT_KEY", "sk-test-salt-key-do-not-use-in-prod")
from litellm.proxy.agent_session_endpoints.warm_pool.hydrate import (
from litellm.managed_agents.vms.warm_pool.hydrate import (
build_hydrate_payload,
)
from litellm.proxy.agent_settings_endpoints.encryption import encrypt_optional
@ -248,7 +248,7 @@ async def test_decrypt_failure_skips_secret_does_not_crash(prisma, jwt_expires_a
raise ValueError("simulated decrypt fail")
with patch(
"litellm.proxy.agent_session_endpoints.warm_pool.hydrate.decrypt_optional",
"litellm.managed_agents.vms.warm_pool.hydrate.decrypt_optional",
new=fake_decrypt,
):
payload = await build_hydrate_payload(