From 603705661a6590d100516e8e741c256a2a7b0082 Mon Sep 17 00:00:00 2001 From: lj Date: Thu, 16 May 2024 16:51:36 +0800 Subject: [PATCH] Update model config in test_config.py --- litellm/tests/test_config.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) 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