Merge pull request #3600 from msabramo/msabramo/fix-pydantic-warnings

Update pydantic code to fix warnings
This commit is contained in:
Krish Dholakia 2024-05-13 22:00:39 -07:00 committed by GitHub
commit 2c867ea9a5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
7 changed files with 129 additions and 82 deletions

View file

@ -20,7 +20,7 @@ from functools import wraps
import datetime, time
import tiktoken
import uuid
from pydantic import BaseModel
from pydantic import ConfigDict, BaseModel
import aiohttp
import textwrap
import logging
@ -328,10 +328,7 @@ class HiddenParams(OpenAIObject):
original_response: Optional[str] = None
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(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