From 64f40385203934e0238560d56b3394ebad6b60ff Mon Sep 17 00:00:00 2001 From: lj Date: Thu, 16 May 2024 16:52:21 +0800 Subject: [PATCH] Update model config in completion.py --- litellm/types/completion.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) 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")