fix(auth_utils.py): add model specific 'tpm_limit' to team's on litellm

This commit is contained in:
Krrish Dholakia 2025-09-29 14:25:08 -07:00
parent 117d5963d0
commit 4a09507c58

View file

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