phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

Merge pull request #2343 from BerriAI/litellm_fix_pydantic_warnings

Browse source 
[FIX] pydantic warnings
This commit is contained in:
Ishaan Jaff 2024-03-05 18:57:46 -08:00 • committed by GitHub
commit 6b6775f68e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/proxy/_types.py
View file

@ -33,6 +33,9 @@ class LiteLLMBase(BaseModel):
# if using pydantic v1 # if using pydantic v1
return self.__fields_set__ return self.__fields_set__
class Config:
protected_namespaces = ()
######### Request Class Definition ###### ######### Request Class Definition ######
class ProxyChatCompletionRequest(LiteLLMBase): class ProxyChatCompletionRequest(LiteLLMBase):
@ -328,6 +331,7 @@ class TeamRequest(LiteLLMBase):
class LiteLLM_BudgetTable(LiteLLMBase): class LiteLLM_BudgetTable(LiteLLMBase):
"""Represents user-controllable params for a LiteLLM_BudgetTable record""" """Represents user-controllable params for a LiteLLM_BudgetTable record"""
soft_budget: Optional[float] = None soft_budget: Optional[float] = None
max_budget: Optional[float] = None max_budget: Optional[float] = None
max_parallel_requests: Optional[int] = None max_parallel_requests: Optional[int] = None
@ -367,7 +371,7 @@ class OrganizationRequest(LiteLLMBase):
class BudgetRequest(LiteLLMBase): class BudgetRequest(LiteLLMBase):
budgets: List[str] budgets: List[str]
class KeyManagementSystem(enum.Enum): class KeyManagementSystem(enum.Enum):
GOOGLE_KMS = "google_kms" GOOGLE_KMS = "google_kms"
AZURE_KEY_VAULT = "azure_key_vault" AZURE_KEY_VAULT = "azure_key_vault"