scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-10 03:28:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 85

Merge pull request #2770 from BerriAI/ui_dont_let_failed_prediction_block_spend_view

Browse source 
(ui) dont let failed prediction block spend view
This commit is contained in:
Ishaan Jaff 2024-03-30 19:10:29 -07:00 • committed by GitHub
commit 8ee3975901
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 17 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
enterprise/utils.py
View file

@ -251,6 +251,11 @@ def _forecast_daily_cost(data: list):
import requests
from datetime import datetime, timedelta
if len(data) == 0:
return {
"response": [],
"predicted_spend": "Current Spend = $0, Predicted = $0",
}
first_entry = data[0]
last_entry = data[-1]

19
ui/litellm-dashboard/src/components/view_key_table.tsx
View file

@ -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) {