diff --git a/ui/litellm-dashboard/src/components/team/TeamInfo.tsx b/ui/litellm-dashboard/src/components/team/TeamInfo.tsx index a4c7ae2bbbe..ca84b470b6e 100644 --- a/ui/litellm-dashboard/src/components/team/TeamInfo.tsx +++ b/ui/litellm-dashboard/src/components/team/TeamInfo.tsx @@ -66,6 +66,7 @@ export interface TeamMembership { rpm_limit: number | null; model_max_budget: Record | null; budget_duration: string | null; + allowed_models?: string[] | null; }; } @@ -95,6 +96,7 @@ export interface TeamData { } | null; created_at: string; access_group_ids?: string[]; + default_team_member_models?: string[]; guardrails?: string[]; policies?: string[]; object_permission?: { @@ -366,6 +368,7 @@ const TeamInfoView: React.FC = ({ max_budget_in_team: values.max_budget_in_team, tpm_limit: values.tpm_limit, rpm_limit: values.rpm_limit, + allowed_models: values.allowed_models, }; MessageManager.destroy(); // Remove all existing toasts @@ -556,6 +559,11 @@ const TeamInfoView: React.FC = ({ updateData.access_group_ids = values.access_group_ids; } + // Pass default_team_member_models to the update request + if (values.default_team_member_models !== undefined) { + updateData.default_team_member_models = values.default_team_member_models; + } + const response = await teamUpdateCall(accessToken, updateData); NotificationsManager.success("Team settings updated successfully"); @@ -833,6 +841,7 @@ const TeamInfoView: React.FC = ({ accessGroups: info.object_permission?.agent_access_groups || [], }, access_group_ids: info.access_group_ids || [], + default_team_member_models: info.default_team_member_models || [], }} layout="vertical" > @@ -864,6 +873,26 @@ const TeamInfoView: React.FC = ({ /> + + Default Member Models{" "} + + + + + } + name="default_team_member_models" + > +