From d4afe8a3bc7480436a505e0cbb702b78f1072d22 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Fri, 3 May 2024 21:59:22 -0700
Subject: [PATCH] ui - view logs by model group and time

---
 .../src/components/model_dashboard.tsx             | 14 +++++++-------
 ui/litellm-dashboard/src/components/networking.tsx | 12 ++++++------
 2 files changed, 13 insertions(+), 13 deletions(-)

diff --git a/ui/litellm-dashboard/src/components/model_dashboard.tsx b/ui/litellm-dashboard/src/components/model_dashboard.tsx
index 1a9da3b4e9..5d2effa1f5 100644
--- a/ui/litellm-dashboard/src/components/model_dashboard.tsx
+++ b/ui/litellm-dashboard/src/components/model_dashboard.tsx
@@ -474,7 +474,7 @@ const handleEditSubmit = async (formValues: Record<string, any>) => {
           accessToken,
           userID,
           userRole,
-          null,
+          selectedModelGroup,
           dateValue.from?.toISOString(),
           dateValue.to?.toISOString()
         );
@@ -491,7 +491,7 @@ const handleEditSubmit = async (formValues: Record<string, any>) => {
           accessToken,
           userID,
           userRole,
-          null,
+          selectedModelGroup,
           dateValue.from?.toISOString(),
           dateValue.to?.toISOString()
         )
@@ -504,7 +504,7 @@ const handleEditSubmit = async (formValues: Record<string, any>) => {
           accessToken,
           userID,
           userRole,
-          null,
+          selectedModelGroup,
           dateValue.from?.toISOString(),
           dateValue.to?.toISOString()
         )
@@ -666,9 +666,9 @@ const handleEditSubmit = async (formValues: Record<string, any>) => {
   };
 
 
-  const updateModelMetrics = async (modelGroup: string | null, startTime: Date, endTime: Date) => {
+  const updateModelMetrics = async (modelGroup: string | null, startTime: Date | undefined, endTime: Date | undefined) => {
     console.log("Updating model metrics for group:", modelGroup);
-    if (!accessToken || !userID || !userRole) {
+    if (!accessToken || !userID || !userRole || !startTime || !endTime) {
       return
     }
     console.log("inside updateModelMetrics - startTime:", startTime, "endTime:", endTime)
@@ -1127,8 +1127,8 @@ const handleEditSubmit = async (formValues: Record<string, any>) => {
               <Col>
                 <Select
                 className="mb-4 mt-2"
-                defaultValue={selectedModelGroup}
-                value={selectedModelGroup}
+                defaultValue={selectedModelGroup? selectedModelGroup : availableModelGroups[0]}
+                value={selectedModelGroup ? selectedModelGroup : availableModelGroups[0]}
               >
                 {availableModelGroups.map((group, idx) => (
                   <SelectItem 
diff --git a/ui/litellm-dashboard/src/components/networking.tsx b/ui/litellm-dashboard/src/components/networking.tsx
index 32d13fa1f7..e29e0e372f 100644
--- a/ui/litellm-dashboard/src/components/networking.tsx
+++ b/ui/litellm-dashboard/src/components/networking.tsx
@@ -441,8 +441,8 @@ export const modelMetricsCall = async (
   userID: String,
   userRole: String, 
   modelGroup: String | null,
-  startTime: String,
-  endTime: String
+  startTime: String | null,
+  endTime: String | null
 ) => {
   /**
    * Get all models on proxy
@@ -483,8 +483,8 @@ export const modelMetricsSlowResponsesCall = async (
   userID: String,
   userRole: String, 
   modelGroup: String | null,
-  startTime: String,
-  endTime: String
+  startTime: String | null,
+  endTime: String | null
 ) => {
   /**
    * Get all models on proxy
@@ -525,8 +525,8 @@ export const modelExceptionsCall = async (
   userID: String,
   userRole: String, 
   modelGroup: String | null,
-  startTime: String,
-  endTime: String
+  startTime: String | null,
+  endTime: String | null
 ) => {
   /**
    * Get all models on proxy