scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-21 00:21:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 74

fix(proxy): reject non-finite temp budget increases on team member update

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
yassin 2026-09-17 20:07:33 +00:00
parent 42429500be
commit 67c522fe73
2 changed files with 8 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
litellm/proxy/_types.py
View file

@ -4400,6 +4400,7 @@ class TeamMemberUpdateRequest(TeamMemberDeleteRequest):
temp_budget_increase: float | None = Field(
default=None,
ge=0,
allow_inf_nan=False,
description="Temporary additive budget increase for this team member, active until temp_budget_expiry",
)
temp_budget_expiry: datetime | None = Field(

10
tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py
View file

@ -15447,8 +15447,12 @@ def test_team_member_update_request_temp_budget_fields_must_be_set_together() ->
TeamMemberUpdateRequest(team_id="team-1", user_id="user-1", temp_budget_expiry="2030-01-01T00:00:00Z")
def test_team_member_update_request_rejects_negative_temp_budget_increase() -> None:
with pytest.raises(ValidationError, match="greater than or equal to 0"):
@pytest.mark.parametrize(
("increase", "message"),
[(-1.0, "greater than or equal to 0"), (float("inf"), "finite number")],
)
def test_team_member_update_request_rejects_unusable_temp_budget_increase(increase: float, message: str) -> None:
with pytest.raises(ValidationError, match=message):
TeamMemberUpdateRequest(
team_id="team-1", user_id="user-1", temp_budget_increase=-1.0, temp_budget_expiry="2030-01-01T00:00:00Z"
team_id="team-1", user_id="user-1", temp_budget_increase=increase, temp_budget_expiry="2030-01-01T00:00:00Z"
)