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

fix(proxy_server.py): fix track cost callback

Browse source
This commit is contained in:
Krrish Dholakia 2024-03-02 19:52:45 -08:00
parent 80e3798a15
commit f5f12d204e
1 changed files with 10 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/proxy/proxy_server.py
View file

@ -1094,12 +1094,16 @@ async def update_database(
## CHECK IF USER PROJECTED SPEND > SOFT LIMIT ## CHECK IF USER PROJECTED SPEND > SOFT LIMIT
soft_budget_cooldown = existing_spend_obj.soft_budget_cooldown soft_budget_cooldown = existing_spend_obj.soft_budget_cooldown
if existing_spend_obj.soft_budget_cooldown == False and ( if (
existing_spend_obj.soft_budget_cooldown == False
and existing_spend_obj.litellm_budget_table is not None
and (
_is_projected_spend_over_limit( _is_projected_spend_over_limit(
current_spend=new_spend, current_spend=new_spend,
soft_budget_limit=existing_spend_obj.litellm_budget_table.soft_budget, soft_budget_limit=existing_spend_obj.litellm_budget_table.soft_budget,
) )
== True == True
)
): ):
key_alias = existing_spend_obj.key_alias key_alias = existing_spend_obj.key_alias
projected_spend, projected_exceeded_date = ( projected_spend, projected_exceeded_date = (