From d6c1282f88a8f5c6cca3c0197e6e6bd69d0efe3d Mon Sep 17 00:00:00 2001
From: lj <lj@wego.com>
Date: Fri, 31 May 2024 23:55:28 +0800
Subject: [PATCH] Update model config definition to use v2 style

---
 litellm/tests/test_config.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/litellm/tests/test_config.py b/litellm/tests/test_config.py
index 60a5de494..cd61101a3 100644
--- a/litellm/tests/test_config.py
+++ b/litellm/tests/test_config.py
@@ -26,9 +26,7 @@ class DBModel(BaseModel):
     model_info: dict
     litellm_params: dict
 
-    class Config:
-        protected_namespaces = ()
-
+    model_config = ConfigDict(protected_namespaces=())
 
 @pytest.mark.asyncio
 async def test_delete_deployment():