Update model config in embedding.py

2024-05-16 16:52:44 +08:00 · 2024-05-16 16:52:44 +08:00 · 5945bb5ed2
commit 5945bb5ed2
parent 64f4038520
1 changed files with 2 additions and 4 deletions
--- 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")