diff --git a/litellm/tests/test_config.py b/litellm/tests/test_config.py index e38187e0eac..cb7a9f484d9 100644 --- a/litellm/tests/test_config.py +++ b/litellm/tests/test_config.py @@ -13,7 +13,7 @@ sys.path.insert( 0, os.path.abspath("../..") ) # Adds the parent directory to the, system path import pytest, litellm -from pydantic import BaseModel +from pydantic import BaseModel, ConfigDict from litellm.proxy.proxy_server import ProxyConfig from litellm.proxy.utils import encrypt_value, ProxyLogging, DualCache from litellm.types.router import Deployment, LiteLLM_Params, ModelInfo @@ -26,8 +26,7 @@ class DBModel(BaseModel): model_info: dict litellm_params: dict - class Config: - protected_namespaces = () + config_dict: ConfigDict = ConfigDict(protected_namespaces=()) @pytest.mark.asyncio