Merge pull request #2343 from BerriAI/litellm_fix_pydantic_warnings

[FIX] pydantic warnings
This commit is contained in:
Ishaan Jaff
2024-03-05 18:57:46 -08:00
committed by GitHub
+5 -1
View File
@@ -33,6 +33,9 @@ class LiteLLMBase(BaseModel):
# if using pydantic v1
return self.__fields_set__
class Config:
protected_namespaces = ()
######### Request Class Definition ######
class ProxyChatCompletionRequest(LiteLLMBase):
@@ -328,6 +331,7 @@ class TeamRequest(LiteLLMBase):
class LiteLLM_BudgetTable(LiteLLMBase):
"""Represents user-controllable params for a LiteLLM_BudgetTable record"""
soft_budget: Optional[float] = None
max_budget: Optional[float] = None
max_parallel_requests: Optional[int] = None
@@ -367,7 +371,7 @@ class OrganizationRequest(LiteLLMBase):
class BudgetRequest(LiteLLMBase):
budgets: List[str]
class KeyManagementSystem(enum.Enum):
GOOGLE_KMS = "google_kms"
AZURE_KEY_VAULT = "azure_key_vault"