From a27858c5c87d40a07b2f6e25c588654f67dc1696 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Tue, 30 Jan 2024 18:58:54 -0800
Subject: [PATCH] fix(_types.py): support datetime as a type for expires field

---
 litellm/proxy/_types.py | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/litellm/proxy/_types.py b/litellm/proxy/_types.py
index 22565eb2b7f..1431128aec6 100644
--- a/litellm/proxy/_types.py
+++ b/litellm/proxy/_types.py
@@ -325,17 +325,18 @@ class LiteLLM_VerificationToken(LiteLLMBase):
     key_alias: Optional[str] = None
     spend: float = 0.0
     max_budget: Optional[float] = None
-    expires: Union[str, None]
+    expires: Union[datetime, str, None]
     models: List[str]
-    aliases: Dict[str, str] = {}
-    config: Dict[str, str] = {}
-    user_id: Union[str, None]
-    max_parallel_requests: Union[int, None]
-    metadata: Dict[str, str] = {}
+    aliases: Dict = {}
+    config: Dict = {}
+    user_id: Optional[str] = None
+    max_parallel_requests: Optional[int] = None
+    metadata: Dict = {}
     tpm_limit: Optional[int] = None
     rpm_limit: Optional[int] = None
     budget_duration: Optional[str] = None
     budget_reset_at: Optional[datetime] = None
+    team_id: Optional[str] = None
 
 
 class LiteLLM_Config(LiteLLMBase):