From 11628d6c40180f9be8f0fc7adea80ac41c988a8f Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Wed, 24 Apr 2024 15:16:37 -0700
Subject: [PATCH] ui cleanup filter by model group section

---
 ui/litellm-dashboard/src/components/model_dashboard.tsx | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ui/litellm-dashboard/src/components/model_dashboard.tsx b/ui/litellm-dashboard/src/components/model_dashboard.tsx
index c10ef3292d..40c3557474 100644
--- a/ui/litellm-dashboard/src/components/model_dashboard.tsx
+++ b/ui/litellm-dashboard/src/components/model_dashboard.tsx
@@ -639,7 +639,7 @@ const handleEditSubmit = async (formValues: Record<string, any>) => {
       <Select
               className="mb-4 mt-2 ml-2 w-50"
               defaultValue="all"
-              onValueChange={(value) => setSelectedModelGroup(value === "all" ? null : value)}
+              onValueChange={(value) => setSelectedModelGroup(value === "all" ? "all" : value)}
             >
               <SelectItem 
                   value={"all"}
@@ -685,7 +685,7 @@ const handleEditSubmit = async (formValues: Record<string, any>) => {
             <TableBody>
               { modelData.data
                   .filter((model: any) =>
-                    selectedModelGroup === null || model.model_name === selectedModelGroup
+                    selectedModelGroup === "all" || model.model_name === selectedModelGroup || selectedModelGroup === null || selectedModelGroup === undefined || selectedModelGroup === ""
                   )
                   .map((model: any, index: number) => (
                 <TableRow key={index}>