From 51d658e878a6991efb30fb803bb2df764fcbe553 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 18 Mar 2024 20:31:23 -0700 Subject: [PATCH] (fix) if litellm-proxy-budget set use it --- litellm/proxy/proxy_server.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 3d9212149d..529ab8b7e1 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -4608,7 +4608,13 @@ 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}