diff --git a/litellm/proxy/_types.py b/litellm/proxy/_types.py
index e8e01602097..ed7237141e0 100644
--- a/litellm/proxy/_types.py
+++ b/litellm/proxy/_types.py
@@ -208,6 +208,7 @@ class UpdateUserRequest(GenerateRequestBase):
     spend: Optional[float] = None
     metadata: Optional[dict] = None
     user_role: Optional[str] = None
+    max_budget: Optional[float] = None
 
 
 class KeyManagementSystem(enum.Enum):