From dfc020ca5f43f28b072cb6fb01c05c76120a50fe Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Tue, 2 Apr 2024 15:27:50 -0700
Subject: [PATCH] ui - support all-models alias

---
 .../src/components/create_key_button.tsx                 | 3 ---
 ui/litellm-dashboard/src/components/teams.tsx            | 7 +++++--
 ui/litellm-dashboard/src/components/view_key_table.tsx   | 9 +++------
 3 files changed, 8 insertions(+), 11 deletions(-)

diff --git a/ui/litellm-dashboard/src/components/create_key_button.tsx b/ui/litellm-dashboard/src/components/create_key_button.tsx
index 2b1a199c2..99cbd7b6e 100644
--- a/ui/litellm-dashboard/src/components/create_key_button.tsx
+++ b/ui/litellm-dashboard/src/components/create_key_button.tsx
@@ -142,9 +142,6 @@ const CreateKey: React.FC<CreateKeyProps> = ({
                   placeholder="Select models"
                   style={{ width: "100%" }}
                 >
-                  <Option key="all-models" value="all-models">
-                    All Models
-                  </Option>
                   {team && team.models ? (
                     team.models.map((model: string) => (
                       <Option key={model} value={model}>
diff --git a/ui/litellm-dashboard/src/components/teams.tsx b/ui/litellm-dashboard/src/components/teams.tsx
index af7d59df9..9d4ebe708 100644
--- a/ui/litellm-dashboard/src/components/teams.tsx
+++ b/ui/litellm-dashboard/src/components/teams.tsx
@@ -123,6 +123,9 @@ const Team: React.FC<TeamProps> = ({
                     placeholder="Select models"
                     style={{ width: "100%" }}
                   >
+                    <Select2.Option key="all-models" value="all-models">
+                      All Models
+                    </Select2.Option>
                     {userModels && userModels.map((model) => (
                       <Select2.Option key={model} value={model}>
                         {model}
@@ -360,8 +363,8 @@ const handleEditSubmit = async (formValues: Record<string, any>) => {
                             {Array.isArray(team.models) ? (
                               <div style={{ display: "flex", flexDirection: "column" }}>
                                 {team.models.length === 0 ? (
-                                  <Badge size={"xs"} className="mb-1" color="purple">
-                                    <Text>All Models</Text>
+                                  <Badge size={"xs"} className="mb-1" color="blue">
+                                    <Text>all-models</Text>
                                   </Badge>
                                 ) : (
                                   team.models.map((model: string, index: number) => (
diff --git a/ui/litellm-dashboard/src/components/view_key_table.tsx b/ui/litellm-dashboard/src/components/view_key_table.tsx
index 5d7bdf2d0..6784211c3 100644
--- a/ui/litellm-dashboard/src/components/view_key_table.tsx
+++ b/ui/litellm-dashboard/src/components/view_key_table.tsx
@@ -147,10 +147,7 @@ const ViewKeyTable: React.FC<ViewKeyTableProps> = ({
                   mode="multiple"
                   placeholder="Select models"
                   style={{ width: "100%" }}
-                >
-                  <Option key="all_models" value="all_models">
-                    All Models
-                  </Option>
+                >                
                   {selectedTeam && selectedTeam.models ? (
                     selectedTeam.models.map((model: string) => (
                       <Option key={model} value={model}>
@@ -420,8 +417,8 @@ const handleEditSubmit = async (formValues: Record<string, any>) => {
                             ))
                           ) : (
                             // If selected team is None or selected team's models are empty, show all models
-                            <Badge size={"xs"} className="mb-1" color="purple">
-                              <Text>All Models</Text>
+                            <Badge size={"xs"} className="mb-1" color="blue">
+                              <Text>all-models</Text>
                             </Badge>
                           )}
                         </>