From 79859f6a823de38176a10e8e1db41caedd3cad6f Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Mon, 3 Jun 2024 17:43:56 -0700
Subject: [PATCH] ui - basic filter by api key,

---
 ui/litellm-dashboard/src/app/page.tsx         |  1 +
 .../src/components/model_dashboard.tsx        | 73 ++++++++++++++++++-
 2 files changed, 70 insertions(+), 4 deletions(-)

diff --git a/ui/litellm-dashboard/src/app/page.tsx b/ui/litellm-dashboard/src/app/page.tsx
index 9c56abdf73c..c218a8aad60 100644
--- a/ui/litellm-dashboard/src/app/page.tsx
+++ b/ui/litellm-dashboard/src/app/page.tsx
@@ -138,6 +138,7 @@ const CreateKeyPage = () => {
               userID={userID}
               userRole={userRole}
               token={token}
+              keys={keys}
               accessToken={accessToken}
               modelData={modelData}
               setModelData={setModelData}
diff --git a/ui/litellm-dashboard/src/components/model_dashboard.tsx b/ui/litellm-dashboard/src/components/model_dashboard.tsx
index 1504e251a96..8a7c3980cb2 100644
--- a/ui/litellm-dashboard/src/components/model_dashboard.tsx
+++ b/ui/litellm-dashboard/src/components/model_dashboard.tsx
@@ -56,6 +56,7 @@ import { BarChart, AreaChart } from "@tremor/react";
 import {
   Button as Button2,
   Modal,
+  Popover,
   Form,
   Input,
   Select as Select2,
@@ -80,6 +81,7 @@ import {
   RefreshIcon,
   CheckCircleIcon,
   XCircleIcon,
+  FilterIcon,
 } from "@heroicons/react/outline";
 import DeleteModelButton from "./delete_model_button";
 const { Title: Title2, Link } = Typography;
@@ -96,6 +98,7 @@ interface ModelDashboardProps {
   userRole: string | null;
   userID: string | null;
   modelData: any;
+  keys: any[] | null;
   setModelData: any;
   premiumUser: boolean;
 }
@@ -260,6 +263,7 @@ const ModelDashboard: React.FC<ModelDashboardProps> = ({
   userRole,
   userID,
   modelData = { data: [] },
+  keys,
   setModelData,
   premiumUser,
 }) => {
@@ -313,6 +317,8 @@ const ModelDashboard: React.FC<ModelDashboardProps> = ({
   const [globalExceptionData, setGlobalExceptionData] =  useState<GlobalExceptionActivityData>({} as GlobalExceptionActivityData);
   const [globalExceptionPerDeployment, setGlobalExceptionPerDeployment] = useState<any[]>([]);
 
+  const [showAdvancedFilters, setShowAdvancedFilters] = useState<boolean>(false);
+
   function formatCreatedAt(createdAt: string | null) {
     if (createdAt) {
       const date = new Date(createdAt);
@@ -969,6 +975,53 @@ const ModelDashboard: React.FC<ModelDashboardProps> = ({
     }
   };
 
+
+  const FilterByContent = (
+      <div >
+        <Text className="mb-1">Select API Key Name</Text>
+        
+
+        <Select defaultValue="all-keys">
+                  <SelectItem
+                    key="all-keys"
+                    value="all-keys"
+                    // onClick={() => {
+                    //   updateEndUserData(dateValue.from, dateValue.to, null);
+                    // }}
+                  >
+                    All Keys
+                  </SelectItem>
+                    {keys?.map((key: any, index: number) => {
+                      if (
+                        key &&
+                        key["key_alias"] !== null &&
+                        key["key_alias"].length > 0
+                      ) {
+                        return (
+                          
+                          <SelectItem
+                            key={index}
+                            value={String(index)}
+                            // onClick={() => {
+                            //   updateEndUserData(dateValue.from, dateValue.to, key["token"]);
+                            // }}
+                          >
+                            {key["key_alias"]}
+                          </SelectItem>
+                        );
+                      }
+                      return null; // Add this line to handle the case when the condition is not met
+                    })}
+                  </Select>
+
+        {/* <Text className="mt-2 mb-1">Select Customer</Text>
+        <Select>
+        </Select> */}
+
+        </div>
+
+  );
+
   const customTooltip = (props: any) => {
     const { payload, active } = props;
     if (!active || !payload) return null;
@@ -1722,9 +1775,7 @@ const ModelDashboard: React.FC<ModelDashboardProps> = ({
             </Card>
           </TabPanel>
           <TabPanel>
-            {/* <p style={{fontSize: '0.85rem', color: '#808080'}}>View how requests were load balanced within a model group</p> */}
-
-            <Grid numItems={2} className="mt-2">
+            <Grid numItems={3} className="mt-2">
               <Col>
                 <Text>Select Time Range</Text>
                 <DateRangePicker
@@ -1768,6 +1819,20 @@ const ModelDashboard: React.FC<ModelDashboardProps> = ({
                   ))}
                 </Select>
               </Col>
+              <Col>
+
+                  <Popover 
+                    trigger="click" content={FilterByContent}
+                    >
+              <Button
+              icon={FilterIcon}
+              size="md"
+              className="mb-4 mt-6 ml-2"
+              onClick={() => setShowAdvancedFilters(true)}
+                >
+              </Button>      
+              </Popover>
+              </Col>
             </Grid>
 
             <Grid numItems={2}>
@@ -2035,7 +2100,7 @@ const ModelDashboard: React.FC<ModelDashboardProps> = ({
           </TabPanel>
         </TabPanels>
       </TabGroup>
-    </div>
+    </div>  
   );
 };