From 5945bb5ed28d0da04b3f7e7f8114cdcaddba481b Mon Sep 17 00:00:00 2001
From: lj <lj@wego.com>
Date: Thu, 16 May 2024 16:52:44 +0800
Subject: [PATCH] Update model config in embedding.py

---
 litellm/types/embedding.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/litellm/types/embedding.py b/litellm/types/embedding.py
index 9db0ef2907..c0af79b996 100644
--- a/litellm/types/embedding.py
+++ b/litellm/types/embedding.py
@@ -1,6 +1,6 @@
 from typing import List, Optional, Union
 
-from pydantic import BaseModel, validator
+from pydantic import BaseModel, ConfigDict
 
 
 class EmbeddingRequest(BaseModel):
@@ -18,6 +18,4 @@ class EmbeddingRequest(BaseModel):
     litellm_logging_obj: Optional[dict] = None
     logger_fn: Optional[str] = None
 
-    class Config:
-        # allow kwargs
-        extra = "allow"
+    model_config: ConfigDict = ConfigDict(extra="allow")