Skip to content

[internal copy of #29277] fix(proxy): add default=None to LiteLLM_TeamMembership.litellm_budget_table #29684

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
mateo-berri merged 2 commits into from
Jun 4, 2026
Merged

[internal copy of #29277] fix(proxy): add default=None to LiteLLM_TeamMembership.litellm_budget_table #29684

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
921bfe6
fix(proxy): add default=None to LiteLLM_TeamMembership.litellm_budget…
fchenebault May 29, 2026
1a0ad62
Merge litellm_internal_staging into litellm_fix_team_membership_budge…
mateo-berri Jun 4, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 18 additions & 0 deletions tests/test_litellm/proxy/test_proxy_types.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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.
Expand Down
Loading