Update model config in test_config.py

This commit is contained in:
lj 2024-05-16 16:51:36 +08:00
parent 665b224226
commit 603705661a
No known key found for this signature in database

View file

@ -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