diff --git a/litellm/proxy/_types.py b/litellm/proxy/_types.py index 293d06023..fbe914a26 100644 --- a/litellm/proxy/_types.py +++ b/litellm/proxy/_types.py @@ -422,6 +422,9 @@ class LiteLLM_ModelTable(LiteLLMBase): created_by: str updated_by: str + class Config: + protected_namespaces = () + class NewUserRequest(GenerateKeyRequest): max_budget: Optional[float] = None @@ -485,6 +488,9 @@ class TeamBase(LiteLLMBase): class NewTeamRequest(TeamBase): model_aliases: Optional[dict] = None + class Config: + protected_namespaces = () + class GlobalEndUsersSpend(LiteLLMBase): api_key: Optional[str] = None @@ -534,6 +540,9 @@ class LiteLLM_TeamTable(TeamBase): budget_reset_at: Optional[datetime] = None model_id: Optional[int] = None + class Config: + protected_namespaces = () + @root_validator(pre=True) def set_model_info(cls, values): dict_fields = [ @@ -570,6 +579,9 @@ class LiteLLM_BudgetTable(LiteLLMBase): model_max_budget: Optional[dict] = None budget_duration: Optional[str] = None + class Config: + protected_namespaces = () + class NewOrganizationRequest(LiteLLM_BudgetTable): organization_id: Optional[str] = None