scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-03 02:22:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 80

fix(proxy): add default=None to LiteLLM_TeamMembership.litellm_budget_table

Browse source 
In Pydantic v2, Optional[T] without a default is a required field. Any
row with budget_id=null triggered a validation error and returned 401.
This commit is contained in:
Florent Chenebault 2026-05-29 12:04:39 +02:00
parent f27df8d516
commit 921bfe6c70
No known key found for this signature in database
2 changed files with 21 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/proxy/_types.py
View file

@ -3892,7 +3892,9 @@ class LiteLLM_TeamMembership(LiteLLMPydanticObjectBase):
# Union so Pydantic picks Full when data has server-managed fields
# (/team/info) and Base when callers/tests construct with only
# user-settable fields.
litellm_budget_table: Optional[Union[LiteLLM_BudgetTableFull, LiteLLM_BudgetTable]]
litellm_budget_table: Optional[
Union[LiteLLM_BudgetTableFull, LiteLLM_BudgetTable]
] = None
def safe_get_team_member_rpm_limit(self) -> Optional[int]:
if self.litellm_budget_table is not None:

18
tests/test_litellm/proxy/test_proxy_types.py
View file

@ -47,6 +47,24 @@ def test_audit_log_masking():
assert json_before_value["key"] == "sk-1*****7890"
def test_team_membership_null_budget_table():
"""
Regression test for: LiteLLM_TeamMembership.litellm_budget_table missing = None.
In Pydantic v2, Optional[T] without a default is required; rows with budget_id=null
raised a validation error and returned 401.
Related: https://github.com/BerriAI/litellm/issues/28689
"""
from litellm.proxy._types import LiteLLM_TeamMembership
membership = LiteLLM_TeamMembership(user_id="u1", team_id="t1")
assert membership.litellm_budget_table is None
membership_explicit = LiteLLM_TeamMembership(
user_id="u1", team_id="t1", litellm_budget_table=None
)
assert membership_explicit.litellm_budget_table is None
def test_internal_jobs_user_has_proxy_admin_role():
"""
Test that the internal jobs system user has PROXY_ADMIN role.