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