From 653d2e6ce0277deee1644c638eab27318c54e21e Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 17 Aug 2024 14:21:59 -0700 Subject: [PATCH] fix parallel request limiter tests --- litellm/tests/test_parallel_request_limiter.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/litellm/tests/test_parallel_request_limiter.py b/litellm/tests/test_parallel_request_limiter.py index 6b272fa8cfd..38b85bf7f24 100644 --- a/litellm/tests/test_parallel_request_limiter.py +++ b/litellm/tests/test_parallel_request_limiter.py @@ -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(