From c613a47c66ff55fda42d64998b3ec1adc77aa242 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 17 Apr 2024 18:10:18 -0700 Subject: [PATCH] test(test_max_tpm_rpm_limiter.py): fix test to use hashed api key in cache check --- litellm/tests/test_max_tpm_rpm_limiter.py | 1 + 1 file changed, 1 insertion(+) diff --git a/litellm/tests/test_max_tpm_rpm_limiter.py b/litellm/tests/test_max_tpm_rpm_limiter.py index 3dd6fb49f1..fbaf30c591 100644 --- a/litellm/tests/test_max_tpm_rpm_limiter.py +++ b/litellm/tests/test_max_tpm_rpm_limiter.py @@ -87,6 +87,7 @@ async def test_pre_call_hook_team_rpm_limits( "team_id": _team_id, } user_api_key_dict = UserAPIKeyAuth(**_user_api_key_dict) # type: ignore + _api_key = hash_token(_api_key) local_cache = DualCache() local_cache.set_cache(key=_api_key, value=_user_api_key_dict) internal_cache = DualCache(redis_cache=_redis_usage_cache)