phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(feat) show monthly predicted spend

Browse source
This commit is contained in:
ishaan-jaff 2024-03-01 09:54:12 -08:00
parent 49051b3f40
commit a491704201
2 changed files with 19 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

18
enterprise/utils.py
View file

@ -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}")

8
ui/litellm-dashboard/src/components/view_key_spend_report.tsx
View file

@ -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 && (