diff --git a/litellm/tests/test_config.py b/litellm/tests/test_config.py index ed68694039a..800f0693e6c 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 09965bb8a9d..87608a71cc3 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,