diff --git a/tests/test_litellm/proxy/auth/test_team_grants.py b/tests/test_litellm/proxy/auth/test_team_grants.py index 447fc1c93a1..c0e8626dba8 100644 --- a/tests/test_litellm/proxy/auth/test_team_grants.py +++ b/tests/test_litellm/proxy/auth/test_team_grants.py @@ -27,6 +27,7 @@ def _full_team(model_aliases=ALIASES) -> LiteLLM_TeamTable: team_alias="grants-team", tpm_limit=1000, rpm_limit=10, + tpd_limit=200000, max_budget=50.0, soft_budget=25.0, spend=12.5, diff --git a/tests/test_litellm/proxy/management_endpoints/test_model_management_endpoints.py b/tests/test_litellm/proxy/management_endpoints/test_model_management_endpoints.py index b9affeccebb..54e9872c8d2 100644 --- a/tests/test_litellm/proxy/management_endpoints/test_model_management_endpoints.py +++ b/tests/test_litellm/proxy/management_endpoints/test_model_management_endpoints.py @@ -4488,6 +4488,8 @@ class TestTeamMemberAutoRouterWrites: assert row.litellm_params["complexity_router_config"] == stored_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: from litellm.proxy.management_endpoints.model_management_endpoints import patch_model, update_model