scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 108

test(proxy): restore the member-scope parametrization and cover team tpd_limit

Browse source 
The merged TestTeamMemberAutoRouterWrites class dropped the parametrize
decorator on test_admin_router_changes_release_member_scope, leaving the
test with an unsatisfiable endpoint fixture; this restores main's
patch/legacy x config/strategy/unrelated matrix. _full_team now sets
tpd_limit so the every-team-field grant coverage assertion sees the new
LiteLLM_VerificationTokenView field populated.
This commit is contained in:
devin-ai-integration[bot] 2026-09-23 06:07:40 +00:00
parent 54b42e3f05
commit 4864227716
2 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
tests/test_litellm/proxy/auth/test_team_grants.py
View file

@ -27,6 +27,7 @@ def _full_team(model_aliases=ALIASES) -> LiteLLM_TeamTable:
team_alias="grants-team", team_alias="grants-team",
tpm_limit=1000, tpm_limit=1000,
rpm_limit=10, rpm_limit=10,
tpd_limit=200000,
max_budget=50.0, max_budget=50.0,
soft_budget=25.0, soft_budget=25.0,
spend=12.5, spend=12.5,

2
tests/test_litellm/proxy/management_endpoints/test_model_management_endpoints.py
View file

@ -4488,6 +4488,8 @@ class TestTeamMemberAutoRouterWrites:
assert row.litellm_params["complexity_router_config"] == stored_config assert row.litellm_params["complexity_router_config"] == stored_config
assert request.litellm_params.complexity_router_config == config assert request.litellm_params.complexity_router_config == config
@pytest.mark.asyncio
@pytest.mark.parametrize("endpoint,change", [("patch", "config"), ("legacy", "strategy"), ("patch", "unrelated")])
async def test_admin_router_changes_release_member_scope(self, endpoint: str, change: str) -> None: async def test_admin_router_changes_release_member_scope(self, endpoint: str, change: str) -> None:
from litellm.proxy.management_endpoints.model_management_endpoints import patch_model, update_model from litellm.proxy.management_endpoints.model_management_endpoints import patch_model, update_model