From ea32a8757bd4f984ca0e3cd58c1d9d84f426cad0 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Thu, 18 Jan 2024 13:34:51 -0800 Subject: [PATCH] (feat) set team_id on virtual_keys --- litellm/proxy/_types.py | 1 + litellm/proxy/schema.prisma | 1 + schema.prisma | 1 + 3 files changed, 3 insertions(+) diff --git a/litellm/proxy/_types.py b/litellm/proxy/_types.py index ad90173a4..e258582ef 100644 --- a/litellm/proxy/_types.py +++ b/litellm/proxy/_types.py @@ -129,6 +129,7 @@ class GenerateKeyRequest(LiteLLMBase): config: Optional[dict] = {} spend: Optional[float] = 0 user_id: Optional[str] = None + team_id: Optional[str] = None max_parallel_requests: Optional[int] = None metadata: Optional[dict] = {} diff --git a/litellm/proxy/schema.prisma b/litellm/proxy/schema.prisma index aa45a8818..1ed76140e 100644 --- a/litellm/proxy/schema.prisma +++ b/litellm/proxy/schema.prisma @@ -24,6 +24,7 @@ model LiteLLM_VerificationToken { aliases Json @default("{}") config Json @default("{}") user_id String? + team_id String? max_parallel_requests Int? metadata Json @default("{}") } diff --git a/schema.prisma b/schema.prisma index 704ada42c..3c2bf22cb 100644 --- a/schema.prisma +++ b/schema.prisma @@ -24,6 +24,7 @@ model LiteLLM_VerificationToken { aliases Json @default("{}") config Json @default("{}") user_id String? + team_id String? max_parallel_requests Int? metadata Json @default("{}") }