From e62e83c42a8e78208418e9a1ba4484d644bd3f5e Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Sat, 30 Mar 2024 11:30:18 -0700
Subject: [PATCH] (ui) dont let prediction block spend view

---
 .../src/components/view_key_table.tsx         | 19 ++++++++++++-------
 1 file changed, 12 insertions(+), 7 deletions(-)

diff --git a/ui/litellm-dashboard/src/components/view_key_table.tsx b/ui/litellm-dashboard/src/components/view_key_table.tsx
index 4269bfc02..bf3affe14 100644
--- a/ui/litellm-dashboard/src/components/view_key_table.tsx
+++ b/ui/litellm-dashboard/src/components/view_key_table.tsx
@@ -84,15 +84,20 @@ const ViewKeyTable: React.FC<ViewKeyTableProps> = ({
       setSpendData(response);
 
       // predict spend based on response
-      const predictedSpend = await PredictedSpendLogsCall(accessToken, response);
-      console.log("Response2:", predictedSpend);
+      try {
+        const predictedSpend = await PredictedSpendLogsCall(accessToken, response);
+        console.log("Response2:", predictedSpend);
 
-      // append predictedSpend to data
-      const combinedData = [...response, ...predictedSpend.response];
-      setSpendData(combinedData);
-      setPredictedSpendString(predictedSpend.predicted_spend)
+        // append predictedSpend to data
+        const combinedData = [...response, ...predictedSpend.response];
+        setSpendData(combinedData);
+        setPredictedSpendString(predictedSpend.predicted_spend)
 
-      console.log("Combined Data:", combinedData);
+        console.log("Combined Data:", combinedData);
+      } catch (error) {
+        console.error("There was an error fetching the predicted data", error);
+      }
+      
       // setPredictedSpend(predictedSpend);
       
     } catch (error) {