From 4a09507c58822f677c263ff3c1a3c999c147d6fa Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 29 Sep 2025 14:25:08 -0700 Subject: [PATCH] fix(auth_utils.py): add model specific 'tpm_limit' to team's on litellm --- litellm/proxy/auth/auth_utils.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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