mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-03 02:22:24 +00:00
fix(auth_utils.py): add model specific 'tpm_limit' to team's on litellm
This commit is contained in:
parent
117d5963d0
commit
4a09507c58
1 changed files with 3 additions and 1 deletions
|
|
@ -447,7 +447,9 @@ def get_key_model_tpm_limit(
|
|||
elif user_api_key_dict.model_max_budget:
|
||||
if "tpm_limit" in user_api_key_dict.model_max_budget:
|
||||
return user_api_key_dict.model_max_budget["tpm_limit"]
|
||||
|
||||
elif user_api_key_dict.team_metadata:
|
||||
if "model_tpm_limit" in user_api_key_dict.team_metadata:
|
||||
return user_api_key_dict.team_metadata["model_tpm_limit"]
|
||||
return None
|
||||
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue