Update model config in embedding.py

This commit is contained in:
lj 2024-05-16 16:52:44 +08:00
parent 64f4038520
commit 5945bb5ed2
No known key found for this signature in database

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")