fix ResetTeamBudgetRequest

This commit is contained in:
Ishaan Jaff 2024-05-25 16:50:10 -07:00
parent 6ec0efab50
commit 920dc1754b

View file

@ -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