Update model config in utils.py
This commit is contained in:
parent
497f4bb38d
commit
6a60bfbd97
@ -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
|
||||
|
||||
Loading…
Reference in New Issue
Block a user