diff --git a/litellm/proxy/_types.py b/litellm/proxy/_types.py index e8b3e6572..1f951605c 100644 --- a/litellm/proxy/_types.py +++ b/litellm/proxy/_types.py @@ -561,7 +561,12 @@ class TeamBase(LiteLLMBase): metadata: Optional[dict] = None tpm_limit: Optional[int] = None rpm_limit: Optional[int] = None + + # Budget fields max_budget: Optional[float] = None + budget_duration: Optional[str] = None + budget_reset_at: Optional[datetime] = None + models: list = [] blocked: bool = False diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 904a414c2..189b6cb88 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -8033,12 +8033,9 @@ async def new_team( _model_id = model_dict.id - ## ADD TO TEAM TABLE + ## ADD TO TEAM TABLE -> the user api key settings are overriding the data we passed i complete_team_data = LiteLLM_TeamTable( **data.json(), - max_parallel_requests=user_api_key_dict.max_parallel_requests, - budget_duration=user_api_key_dict.budget_duration, - budget_reset_at=user_api_key_dict.budget_reset_at, model_id=_model_id, )