From 03a43b99a5ff8b57a117edab27b2d7f322c2fed1 Mon Sep 17 00:00:00 2001 From: CyanideByte Date: Sat, 27 Apr 2024 20:42:54 -0700 Subject: [PATCH] Added _types.py cases from edwinjosegeorge PR#3340 --- litellm/proxy/_types.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) 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