Update model config in embedding.py

This commit is contained in:
lj
2024-05-16 16:52:44 +08:00
parent 64f4038520
commit 5945bb5ed2
+2 -4
View File
@@ -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")