feat: add team_member_model_max_budget field to NewTeamRequest and UpdateTeamRequest

This commit is contained in:
Ishaan Jaffer 2026-04-24 14:24:23 -07:00
parent 70a986e689
commit 41a9bfc94f
No known key found for this signature in database

View file

@ -1750,6 +1750,7 @@ class NewTeamRequest(TeamBase):
)
team_member_key_duration: Optional[str] = None # e.g. "1d", "1w", "1m"
team_member_budget_duration: Optional[str] = None # e.g. "30d", "1mo"
team_member_model_max_budget: Optional[dict] = None # e.g. {"gpt-4": {"max_budget": 10.0, "budget_duration": "30d"}}
allowed_vector_store_indexes: Optional[List[AllowedVectorStoreIndexItem]] = None
enforced_batch_output_expires_after: Optional[dict] = None
enforced_file_expires_after: Optional[dict] = None
@ -1818,6 +1819,7 @@ class UpdateTeamRequest(LiteLLMPydanticObjectBase):
default_team_member_models: Optional[List[str]] = (
None # default allowed_models seeded onto new team members
)
team_member_model_max_budget: Optional[dict] = None # e.g. {"gpt-4": {"max_budget": 10.0, "budget_duration": "30d"}}
class ResetTeamBudgetRequest(LiteLLMPydanticObjectBase):
@ -4109,6 +4111,7 @@ LiteLLM_ManagementEndpoint_MetadataFields = [
"allowed_vector_store_indexes",
"enforced_batch_output_expires_after",
"enforced_file_expires_after",
"team_member_model_max_budget",
]
LiteLLM_ManagementEndpoint_MetadataFields_Premium = [