diff --git a/litellm/proxy/auth/auth_utils.py b/litellm/proxy/auth/auth_utils.py index 9cd57844fc6..c400c2d0d86 100644 --- a/litellm/proxy/auth/auth_utils.py +++ b/litellm/proxy/auth/auth_utils.py @@ -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