Update model config in utils.py

This commit is contained in:
lj 2024-05-16 16:39:37 +08:00
parent 497f4bb38d
commit 6a60bfbd97
No known key found for this signature in database

View file

@ -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