From e1786848cbfa685948c0b6d869af58771283853d Mon Sep 17 00:00:00 2001 From: CyanideByte Date: Sat, 27 Apr 2024 13:08:45 -0700 Subject: [PATCH] protected_namespaces fixed for model_info --- litellm/tests/test_config.py | 3 +++ litellm/types/router.py | 6 ++++++ 2 files changed, 9 insertions(+) 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,