diff --git a/litellm/proxy/_types.py b/litellm/proxy/_types.py index 293d06023a99..fbe914a26f0f 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 diff --git a/litellm/tests/test_config.py b/litellm/tests/test_config.py index ed68694039ae..800f0693e6ca 100644 --- a/litellm/tests/test_config.py +++ b/litellm/tests/test_config.py @@ -26,6 +26,9 @@ class DBModel(BaseModel): model_info: dict litellm_params: dict + class Config: + protected_namespaces = () + @pytest.mark.asyncio async def test_delete_deployment(): diff --git a/litellm/types/router.py b/litellm/types/router.py index 09965bb8a9dc..87608a71cc32 100644 --- a/litellm/types/router.py +++ b/litellm/types/router.py @@ -202,12 +202,18 @@ class updateDeployment(BaseModel): litellm_params: Optional[updateLiteLLMParams] = None model_info: Optional[ModelInfo] = None + class Config: + protected_namespaces = () + class Deployment(BaseModel): model_name: str litellm_params: LiteLLM_Params model_info: ModelInfo + class Config: + protected_namespaces = () + def __init__( self, model_name: str,