diff --git a/litellm/utils.py b/litellm/utils.py index 36f4ad481..cac3ac865 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -19,7 +19,7 @@ from functools import wraps, lru_cache import datetime, time import tiktoken import uuid -from pydantic import BaseModel +from pydantic import BaseModel, ConfigDict import aiohttp import textwrap import logging @@ -332,9 +332,7 @@ class HiddenParams(OpenAIObject): model_id: Optional[str] = None # used in Router for individual deployments api_base: Optional[str] = None # returns api base used for making completion call - class Config: - extra = "allow" - protected_namespaces = () + model_config: ConfigDict = ConfigDict(extra="allow", protected_namespaces=()) def get(self, key, default=None): # Custom .get() method to access attributes with a default value if the attribute doesn't exist