phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix getting user_id

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-26 17:36:02 -07:00
parent 17ec5f52e5
commit 5ef0ab5b2e
1 changed files with 1 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/proxy/hooks/parallel_request_limiter.py
View file

@ -322,9 +322,8 @@ class _PROXY_MaxParallelRequestsHandler(CustomLogger):
# check if REQUEST ALLOWED for user_id # check if REQUEST ALLOWED for user_id
user_id = user_api_key_dict.user_id user_id = user_api_key_dict.user_id
if user_id is not None: 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( _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, litellm_parent_otel_span=user_api_key_dict.parent_otel_span,
) )
# get user tpm/rpm limits # get user tpm/rpm limits