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(