From a491704201aa0ac0cba745705f7deaa6f4f8ae7a Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Fri, 1 Mar 2024 09:54:12 -0800
Subject: [PATCH] (feat) show monthly predicted spend

---
 enterprise/utils.py                            | 18 +++++++++++++++---
 .../src/components/view_key_spend_report.tsx   |  8 ++++----
 2 files changed, 19 insertions(+), 7 deletions(-)

diff --git a/enterprise/utils.py b/enterprise/utils.py
index ad8148f38..d762cd56c 100644
--- a/enterprise/utils.py
+++ b/enterprise/utils.py
@@ -274,14 +274,19 @@ def _forecast_daily_cost(data: list):
     ) - timedelta(days=1)
 
     # Calculate the remaining days in the month
-    remaining_days = (last_day_of_todays_month - last_entry_date).days + 1
+    remaining_days = (last_day_of_todays_month - last_entry_date).days
 
+    current_spend_this_month = 0
     series = {}
     for entry in data:
         date = entry["date"]
         spend = entry["spend"]
         series[date] = spend
 
+        # check if the date is in this month
+        if datetime.strptime(date, "%Y-%m-%d").month == today_day_month:
+            current_spend_this_month += spend
+
     if len(series) < 10:
         num_items_to_fill = 11 - len(series)
 
@@ -314,15 +319,22 @@ def _forecast_daily_cost(data: list):
     # print("Forecast Data:", forecast_data)
 
     response_data = []
+    total_predicted_spend = current_spend_this_month
     for date in forecast_data:
         spend = forecast_data[date]
         entry = {
             "date": date,
             "predicted_spend": spend,
         }
+        total_predicted_spend += spend
         response_data.append(entry)
-    # print("Response Data:", response_data)
-    return response_data
+
+    # get month as a string, Jan, Feb, etc.
+    today_month = today_date.strftime("%B")
+    predicted_spend = (
+        f"Predicted Spend for { today_month } 2024, ${total_predicted_spend}"
+    )
+    return {"response": response_data, "predicted_spend": predicted_spend}
 
     # print(f"Date: {entry['date']}, Spend: {entry['spend']}, Response: {response.text}")
 
diff --git a/ui/litellm-dashboard/src/components/view_key_spend_report.tsx b/ui/litellm-dashboard/src/components/view_key_spend_report.tsx
index fb19fcdee..4b1e45400 100644
--- a/ui/litellm-dashboard/src/components/view_key_spend_report.tsx
+++ b/ui/litellm-dashboard/src/components/view_key_spend_report.tsx
@@ -48,9 +48,7 @@ const ViewKeySpendReport: React.FC<ViewKeySpendReportProps> = ({
   const [data, setData] = useState<{ day: string; spend: number }[] | null>(
     null
   );
-  const [predictedSpend, setPredictedSpend] = useState<{ day: string; spend: number }[] | null>(
-    null
-  );
+  const [predictedSpendString, setPredictedSpendString] = useState("");
   const [userData, setUserData] = useState<
     { name: string; value: number }[] | null
   >(null);
@@ -88,8 +86,9 @@ const ViewKeySpendReport: React.FC<ViewKeySpendReportProps> = ({
       console.log("Response2:", predictedSpend);
 
       // append predictedSpend to data
-      const combinedData = [...response, ...predictedSpend];
+      const combinedData = [...response, ...predictedSpend.response];
       setData(combinedData);
+      setPredictedSpendString(predictedSpend.predicted_spend)
 
       console.log("Combined Data:", combinedData);
       // setPredictedSpend(predictedSpend);
@@ -119,6 +118,7 @@ const ViewKeySpendReport: React.FC<ViewKeySpendReportProps> = ({
         <Title style={{ textAlign: "left" }}>Key Name: {keyName}</Title>
 
         <Metric>Monthly Spend ${keySpend}</Metric>
+        <Title>{predictedSpendString}</Title>
 
         <Card className="mt-6 mb-6">
           {data && (