scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-09 22:31:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 86

fix parallel request limiter tests

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-17 14:21:59 -07:00
parent 221e5b829b
commit 653d2e6ce0
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
litellm/tests/test_parallel_request_limiter.py
View file

@ -1046,7 +1046,9 @@ async def test_pre_call_hook_rpm_limits_per_model():
kwargs = {
"model": model,
"litellm_params": {"metadata": {"user_api_key": _api_key}},
"litellm_params": {
"metadata": {"user_api_key": _api_key, "model_group": model}
},
}
await parallel_request_handler.async_log_success_event(
@ -1121,7 +1123,9 @@ async def test_pre_call_hook_tpm_limits_per_model():
kwargs = {
"model": model,
"litellm_params": {"metadata": {"user_api_key": _api_key}},
"litellm_params": {
"metadata": {"user_api_key": _api_key, "model_group": model}
},
}
await parallel_request_handler.async_log_success_event(