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

(fix) show correct spend on ui

Browse source
This commit is contained in:
Ishaan Jaff 2024-03-29 09:41:00 -07:00
parent 6233351b01
commit 7df2d7cb33
4 changed files with 6 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/proxy/proxy_server.py
View file

@ -4881,13 +4881,7 @@ async def global_spend():
if user_response is not None:
if isinstance(user_response, list) and len(user_response) > 0:
total_proxy_budget = user_response[0].get("max_budget", 0.0)
litellm_proxy_budget_spend = user_response[0].get("spend", None)
if (
litellm_proxy_budget_spend is not None
and litellm_proxy_budget_spend > 0.0
):
# if the user has spend on litellm-proxy-budget, use that to show $spend/max budget
total_spend = litellm_proxy_budget_spend
return {"spend": total_spend, "max_budget": total_proxy_budget}