From c9e2f977dd92ab1803d2c56e6eb4173c629f4acf Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 22 Jul 2024 18:21:50 -0700 Subject: [PATCH] feat - return team_metadata in user_api_key_auth --- litellm/proxy/_types.py | 1 + litellm/proxy/utils.py | 1 + 2 files changed, 2 insertions(+) diff --git a/litellm/proxy/_types.py b/litellm/proxy/_types.py index 190d4be271..0cd4af7e18 100644 --- a/litellm/proxy/_types.py +++ b/litellm/proxy/_types.py @@ -1256,6 +1256,7 @@ class LiteLLM_VerificationTokenView(LiteLLM_VerificationToken): soft_budget: Optional[float] = None team_model_aliases: Optional[Dict] = None team_member_spend: Optional[float] = None + team_metadata: Optional[Dict] = None # End User Params end_user_id: Optional[str] = None diff --git a/litellm/proxy/utils.py b/litellm/proxy/utils.py index 0f87e962ab..5846dd25a8 100644 --- a/litellm/proxy/utils.py +++ b/litellm/proxy/utils.py @@ -1313,6 +1313,7 @@ class PrismaClient: t.tpm_limit AS team_tpm_limit, t.rpm_limit AS team_rpm_limit, t.models AS team_models, + t.metadata AS team_metadata, t.blocked AS team_blocked, t.team_alias AS team_alias, tm.spend AS team_member_spend,