From 920dc1754b7d8a1107447c1e182b4a6b75f2988a Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Sat, 25 May 2024 16:50:10 -0700
Subject: [PATCH] fix ResetTeamBudgetRequest

---
 litellm/proxy/_types.py | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)

diff --git a/litellm/proxy/_types.py b/litellm/proxy/_types.py
index 0ea6fff8e80..5a0cbdf6ce1 100644
--- a/litellm/proxy/_types.py
+++ b/litellm/proxy/_types.py
@@ -602,6 +602,21 @@ class TeamMemberDeleteRequest(LiteLLMBase):
 
 
 class UpdateTeamRequest(LiteLLMBase):
+    """
+    UpdateTeamRequest, used by /team/update when you need to update a team
+
+    team_id: str
+    team_alias: Optional[str] = None
+    organization_id: Optional[str] = None
+    metadata: Optional[dict] = None
+    tpm_limit: Optional[int] = None
+    rpm_limit: Optional[int] = None
+    max_budget: Optional[float] = None
+    models: Optional[list] = None
+    blocked: Optional[bool] = None
+    budget_duration: Optional[str] = None
+    """
+
     team_id: str  # required
     team_alias: Optional[str] = None
     organization_id: Optional[str] = None
@@ -614,6 +629,21 @@ class UpdateTeamRequest(LiteLLMBase):
     budget_duration: Optional[str] = None
 
 
+class ResetTeamBudgetRequest(LiteLLMBase):
+    """
+    internal type used to reset the budget on a team
+    used by reset_budget()
+
+    team_id: str
+    spend: float
+    budget_reset_at: datetime
+    """
+
+    team_id: str
+    spend: float
+    budget_reset_at: datetime
+
+
 class DeleteTeamRequest(LiteLLMBase):
     team_ids: List[str]  # required