From 9e0fb6bd17d1b347e331795e5c02a6087043d2a5 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Mon, 6 Apr 2026 17:25:45 -0700 Subject: [PATCH] fix(tests): set default_team_member_models=None on team mocks to match new field --- .../proxy/management_endpoints/test_team_endpoints.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py b/tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py index 232c698603a..c28e8f315a5 100644 --- a/tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py +++ b/tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py @@ -1033,6 +1033,7 @@ async def test_process_team_members_single_member(): mock_prisma_client = MagicMock() mock_team = MagicMock(spec=LiteLLM_TeamTable) mock_team.metadata = {"team_member_budget_id": "budget-123"} + mock_team.default_team_member_models = None # Mock user and membership objects mock_user = MagicMock(spec=LiteLLM_UserTable) @@ -1074,6 +1075,7 @@ async def test_process_team_members_single_member(): litellm_proxy_admin_name="admin", team_id="test-team-123", default_team_budget_id="budget-123", + allowed_models=None, ) @@ -1089,6 +1091,7 @@ async def test_process_team_members_multiple_members(): mock_prisma_client = MagicMock() mock_team = MagicMock(spec=LiteLLM_TeamTable) mock_team.metadata = None + mock_team.default_team_member_models = None # Create multiple members as dictionaries (they will be converted to Member objects) members = [ @@ -2840,6 +2843,7 @@ async def test_new_team_max_budget_within_user_limit(): mock_created_team.max_budget = 50.0 mock_created_team.members_with_roles = [] mock_created_team.metadata = None + mock_created_team.default_team_member_models = None mock_created_team.model_dump.return_value = { "team_id": "team-within-budget-789", "team_alias": "within-budget-team", @@ -2968,6 +2972,7 @@ async def test_new_team_org_scoped_budget_bypasses_user_limit(): mock_created_team.organization_id = "test-org-123" mock_created_team.members_with_roles = [] mock_created_team.metadata = None + mock_created_team.default_team_member_models = None mock_created_team.model_dump.return_value = { "team_id": "team-org-scoped-789", "team_alias": "org-scoped-team", @@ -3099,6 +3104,7 @@ async def test_new_team_org_scoped_models_bypasses_user_limit(): mock_created_team.models = ["gpt-4"] mock_created_team.members_with_roles = [] mock_created_team.metadata = None + mock_created_team.default_team_member_models = None mock_created_team.model_dump.return_value = { "team_id": "team-org-scoped-models-789", "team_alias": "org-scoped-models-team", @@ -5293,6 +5299,7 @@ async def test_new_team_soft_budget_validation( mock_created_team.max_budget = expected_max_budget mock_created_team.members_with_roles = [] mock_created_team.metadata = None + mock_created_team.default_team_member_models = None mock_created_team.model_dump.return_value = { "team_id": "test-team-123", "team_alias": "test-soft-budget-team",