From 5ef0ab5b2e282902f40859a9722e7a58d27d5d8d Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Thu, 26 Sep 2024 17:36:02 -0700 Subject: [PATCH] fix getting user_id --- litellm/proxy/hooks/parallel_request_limiter.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/litellm/proxy/hooks/parallel_request_limiter.py b/litellm/proxy/hooks/parallel_request_limiter.py index 7b3c0960f..f5c45f3bd 100644 --- a/litellm/proxy/hooks/parallel_request_limiter.py +++ b/litellm/proxy/hooks/parallel_request_limiter.py @@ -322,9 +322,8 @@ class _PROXY_MaxParallelRequestsHandler(CustomLogger): # check if REQUEST ALLOWED for user_id user_id = user_api_key_dict.user_id if user_id is not None: - request_count_api_key = f"{user_id}::{precise_minute}::request_count" _user_id_rate_limits = await self.internal_usage_cache.async_get_cache( - key=request_count_api_key, + key=user_id, litellm_parent_otel_span=user_api_key_dict.parent_otel_span, ) # get user tpm/rpm limits