From b28109e89147a75f01d51d164b7915617c86a461 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Tue, 2 Apr 2024 21:33:28 -0700
Subject: [PATCH] ui fix

---
 .../src/components/view_key_table.tsx         | 33 ++++++++++++-------
 1 file changed, 21 insertions(+), 12 deletions(-)

diff --git a/ui/litellm-dashboard/src/components/view_key_table.tsx b/ui/litellm-dashboard/src/components/view_key_table.tsx
index f500def6a..8898a9ad1 100644
--- a/ui/litellm-dashboard/src/components/view_key_table.tsx
+++ b/ui/litellm-dashboard/src/components/view_key_table.tsx
@@ -152,18 +152,27 @@ const ViewKeyTable: React.FC<ViewKeyTableProps> = ({
                     All Team Models
                   </Option>                
                   {selectedTeam && selectedTeam.models ? (
-                    selectedTeam.models.map((model: string) => (
-                      <Option key={model} value={model}>
-                        {model}
-                      </Option>
-                    ))
-                  ) : (
-                    userModels.map((model: string) => (
-                      <Option key={model} value={model}>
-                        {model}
-                      </Option>
-                    ))
-                  )}
+  selectedTeam.models.includes("all-proxy-models") ? (
+    userModels.filter(model => model !== "all-proxy-models").map((model: string) => (
+      <Option key={model} value={model}>
+        {model}
+      </Option>
+    ))
+  ) : (
+    selectedTeam.models.map((model: string) => (
+      <Option key={model} value={model}>
+        {model}
+      </Option>
+    ))
+  )
+) : (
+  userModels.map((model: string) => (
+    <Option key={model} value={model}>
+      {model}
+    </Option>
+  ))
+)}
+
 
 
                 </Select>