scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

test(proxy): configure stable agent IDs in error tests

Browse source
This commit is contained in:
AlisinaDevelo 2026-09-27 06:41:07 +02:00
parent 0d0df83bfd
commit 3d42321bcc
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tests/unit/test_rate_limit_error_unification.py
View file

@ -498,6 +498,7 @@ class TestProxyHooksActuallyRaiseProxyRateLimitError:
"litellm.proxy.agent_endpoints.agent_registry.global_agent_registry"
) as mock_registry:
mock_registry.get_agent_by_id.return_value = agent
mock_registry.stable_agent_id.return_value = "agent-iter-1"
# First call within budget.
await handler.async_pre_call_hook(
user_api_key_dict=user_api_key_dict,
@ -854,6 +855,7 @@ class TestProxyHooksActuallyRaiseProxyRateLimitError:
"litellm.proxy.agent_endpoints.agent_registry.global_agent_registry"
) as mock_registry:
mock_registry.get_agent_by_id.return_value = agent
mock_registry.stable_agent_id.return_value = "agent-session-1"
with pytest.raises(ProxyRateLimitError) as exc_info:
await handler.async_pre_call_hook(
user_api_key_dict=user_api_key_dict,