Merge pull request #2893 from unclecode/main

Fix issue #2832: Add protected_namespaces to Config class within utils.py, router.py and completion.py to avoid the warning message.
This commit is contained in:
Ishaan Jaff 2024-04-09 08:51:41 -07:00 committed by GitHub
commit caec5374d9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 8 additions and 1 deletions

View file

@ -32,5 +32,5 @@ class CompletionRequest(BaseModel):
model_list: Optional[List[str]] = None model_list: Optional[List[str]] = None
class Config: class Config:
# allow kwargs
extra = "allow" extra = "allow"
protected_namespaces = ()

View file

@ -12,6 +12,9 @@ class ModelConfig(BaseModel):
tpm: int tpm: int
rpm: int rpm: int
class Config:
protected_namespaces = ()
class RouterConfig(BaseModel): class RouterConfig(BaseModel):
model_list: List[ModelConfig] model_list: List[ModelConfig]
@ -41,6 +44,8 @@ class RouterConfig(BaseModel):
"latency-based-routing", "latency-based-routing",
] = "simple-shuffle" ] = "simple-shuffle"
class Config:
protected_namespaces = ()
class ModelInfo(BaseModel): class ModelInfo(BaseModel):
id: Optional[ id: Optional[
@ -141,6 +146,7 @@ class Deployment(BaseModel):
class Config: class Config:
extra = "allow" extra = "allow"
protected_namespaces = ()
def __contains__(self, key): def __contains__(self, key):
# Define custom behavior for the 'in' operator # Define custom behavior for the 'in' operator

View file

@ -236,6 +236,7 @@ class HiddenParams(OpenAIObject):
class Config: class Config:
extra = "allow" extra = "allow"
protected_namespaces = ()
def get(self, key, default=None): def get(self, key, default=None):
# Custom .get() method to access attributes with a default value if the attribute doesn't exist # Custom .get() method to access attributes with a default value if the attribute doesn't exist