From 2ac8f1c6ec73e9f6d7cec6750a0c7489122b3ba4 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Mon, 3 Jun 2024 14:21:35 -0700
Subject: [PATCH] get all tags on ui

---
 litellm/proxy/proxy_server.py                 |  7 +++----
 ui/litellm-dashboard/src/components/usage.tsx | 16 ++++++++++------
 2 files changed, 13 insertions(+), 10 deletions(-)

diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py
index a222f7d27c..82f665a822 100644
--- a/litellm/proxy/proxy_server.py
+++ b/litellm/proxy/proxy_server.py
@@ -8246,10 +8246,9 @@ async def global_get_all_tag_names():
             )
 
         sql_query = """
-        SELECT
-        jsonb_array_elements_text(request_tags) AS individual_request_tag
-        FROM "LiteLLM_SpendLogs"
-        GROUP BY individual_request_tag
+        SELECT DISTINCT
+            jsonb_array_elements_text(request_tags) AS individual_request_tag
+        FROM "LiteLLM_SpendLogs";
         """
 
         db_response = await prisma_client.db.query_raw(sql_query)
diff --git a/ui/litellm-dashboard/src/components/usage.tsx b/ui/litellm-dashboard/src/components/usage.tsx
index 90405f08d0..e3425aa8de 100644
--- a/ui/litellm-dashboard/src/components/usage.tsx
+++ b/ui/litellm-dashboard/src/components/usage.tsx
@@ -299,7 +299,7 @@ const UsagePage: React.FC<UsagePageProps> = ({
 
             // all_tag_names
             const all_tag_names = await allTagNamesCall(accessToken);
-            setAllTagNames(all_tag_names);
+            setAllTagNames(all_tag_names.tag_names);
 
             // get spend per end-user
             let spend_user_call = await adminTopEndUsersCall(accessToken, null, undefined, undefined);
@@ -762,16 +762,20 @@ const UsagePage: React.FC<UsagePageProps> = ({
 
               <Col>
 
-              <MultiSelect>
+              <MultiSelect
+                defaultValue={["all-tags"]}>
+              <MultiSelectItem
+                  key={"all-tags"}
+                  value={"All Tags"}
+                >
+                  All Tags
+                </MultiSelectItem>
                 {
-                  allTagNames?.map((tag: any, index: number) => {
+                  allTagNames && allTagNames?.map((tag: any, index: number) => {
                     return (
                       <MultiSelectItem
                         key={index}
                         value={String(index)}
-                        onClick={() => {
-                          updateTagSpendData(dateValue.from, dateValue.to);
-                        }}
                       >
                         {tag}
                       </MultiSelectItem>