diff --git a/litellm/types/completion.py b/litellm/types/completion.py index 78af7667bae..36d6cf994c5 100644 --- a/litellm/types/completion.py +++ b/litellm/types/completion.py @@ -1,6 +1,6 @@ from typing import List, Optional, Union, Iterable -from pydantic import BaseModel, validator +from pydantic import BaseModel, ConfigDict, validator from typing_extensions import Literal, Required, TypedDict @@ -191,6 +191,4 @@ class CompletionRequest(BaseModel): api_key: Optional[str] = None model_list: Optional[List[str]] = None - class Config: - extra = "allow" - protected_namespaces = () + model_config: ConfigDict = ConfigDict(protected_namespaces=(), extra="allow")