From f72cbdbffe18e7f34a29e175ec09e2e92a440de2 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Fri, 29 Mar 2024 07:52:34 -0700
Subject: [PATCH] (ui) teams models

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

diff --git a/ui/litellm-dashboard/src/components/teams.tsx b/ui/litellm-dashboard/src/components/teams.tsx
index 0f497595c..8a8d0b037 100644
--- a/ui/litellm-dashboard/src/components/teams.tsx
+++ b/ui/litellm-dashboard/src/components/teams.tsx
@@ -186,11 +186,17 @@ const Team: React.FC<TeamProps> = ({
                           <TableCell style={{ maxWidth: "8-x", whiteSpace: "pre-wrap", overflow: "hidden" }}>
                             {Array.isArray(team.models) ? (
                               <div style={{ display: "flex", flexDirection: "column" }}>
-                                {team.models.map((model: string, index: number) => (
-                                  <Badge key={index} size={"xs"} className="mb-1" color="blue">
-                                    {model.length > 30 ? `${model.slice(0, 30)}...` : model}
+                                {team.models.length === 0 ? (
+                                  <Badge size={"xs"} className="mb-1" color="purple">
+                                    <Text>All Models</Text>
                                   </Badge>
-                                ))}
+                                ) : (
+                                  team.models.map((model: string, index: number) => (
+                                    <Badge key={index} size={"xs"} className="mb-1" color="blue">
+                                      <Text>{model.length > 30 ? `${model.slice(0, 30)}...` : model}</Text>
+                                    </Badge>
+                                  ))
+                                )}
                               </div>
                             ) : null}
                         </TableCell>