From 77ca095412687ccd53eb9741c679bfe8c29cfceb Mon Sep 17 00:00:00 2001
From: Ishaan Jaffer <ishaanjaffer0324@gmail.com>
Date: Wed, 1 Apr 2026 15:37:03 -0700
Subject: [PATCH] EditMembership: add Allowed Models multi-select field

---
 .../src/components/team/EditMembership.tsx        | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/ui/litellm-dashboard/src/components/team/EditMembership.tsx b/ui/litellm-dashboard/src/components/team/EditMembership.tsx
index e6d9ccad60b..cc655ba52b9 100644
--- a/ui/litellm-dashboard/src/components/team/EditMembership.tsx
+++ b/ui/litellm-dashboard/src/components/team/EditMembership.tsx
@@ -21,7 +21,7 @@ interface ModalConfig {
   additionalFields?: Array<{
     name: string;
     label: string | React.ReactNode;
-    type: "input" | "select" | "numerical";
+    type: "input" | "select" | "numerical" | "multi-select";
     options?: Array<{ label: string; value: string }>;
     rules?: any[];
     step?: number;
@@ -65,6 +65,8 @@ const MemberModal = <T extends BaseMember>({
           max_budget_in_team: (initialData as any).max_budget_in_team || null,
           tpm_limit: (initialData as any).tpm_limit || null,
           rpm_limit: (initialData as any).rpm_limit || null,
+          // Keep array values for multi-select fields
+          allowed_models: (initialData as any).allowed_models || [],
         };
         console.log("Setting form values:", formValues);
         form.setFieldsValue(formValues);
@@ -115,7 +117,7 @@ const MemberModal = <T extends BaseMember>({
   const renderField = (field: {
     name: string;
     label: string | React.ReactNode;
-    type: "input" | "select" | "numerical";
+    type: "input" | "select" | "numerical" | "multi-select";
     options?: Array<{ label: string; value: string }>;
     rules?: any[];
     step?: number;
@@ -144,6 +146,15 @@ const MemberModal = <T extends BaseMember>({
             ))}
           </Select>
         );
+      case "multi-select":
+        return (
+          <Select
+            mode="multiple"
+            placeholder={field.placeholder || "Select options"}
+            options={field.options}
+            allowClear
+          />
+        );
       default:
         return null;
     }