diff --git a/litellm/proxy/_types.py b/litellm/proxy/_types.py
index b8e0aa0d128..8ecc575e2c3 100644
--- a/litellm/proxy/_types.py
+++ b/litellm/proxy/_types.py
@@ -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 = [