scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-06 08:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 79

test(test_max_tpm_rpm_limiter.py): fix test to use hashed api key in cache check

Browse source
This commit is contained in:
Krrish Dholakia 2024-04-17 18:10:18 -07:00
parent 473e667bdf
commit c613a47c66
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
litellm/tests/test_max_tpm_rpm_limiter.py
View file

@ -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)