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

(fix) test_parallel limiter fix

Browse source
This commit is contained in:
ishaan-jaff 2024-02-03 20:59:07 -08:00
parent 24aa304061
commit c353161456
1 changed files with 5 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
litellm/tests/test_parallel_request_limiter.py
View file

@ -371,14 +371,13 @@ async def test_normal_router_tpm_limit():
print(f"response: {response}")
try:
await parallel_request_handler.async_pre_call_hook(
user_api_key_dict=user_api_key_dict,
cache=local_cache,
data={},
call_type="",
assert (
parallel_request_handler.user_api_key_cache.get_cache(
key=request_count_api_key
)["current_tpm"]
> 0
)
pytest.fail(f"Expected call to fail")
except Exception as e:
assert e.status_code == 429