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("{}")
 }