diff --git a/docs/my-website/docs/proxy/ui_team_soft_budget_alerts.md b/docs/my-website/docs/proxy/ui_team_soft_budget_alerts.md index 17c42e57c9a..413457ccb82 100644 --- a/docs/my-website/docs/proxy/ui_team_soft_budget_alerts.md +++ b/docs/my-website/docs/proxy/ui_team_soft_budget_alerts.md @@ -2,6 +2,16 @@ import Image from '@theme/IdealImage'; # Team Soft Budget Alerts +:::info + +✨ This is an Enterprise feature. Email budget alerts require an enterprise license. + +[Enterprise Pricing](https://www.litellm.ai/#pricing) + +[Get free 7-day trial key](https://www.litellm.ai/enterprise#trial) + +::: + Set a soft budget on a team and get email alerts when spending crosses the threshold — without blocking any requests. ## Overview diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 5bcf912a4af..2d789b982da 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -1917,26 +1917,20 @@ async def update_cache( # noqa: PLR0915 ## CHECK IF USER PROJECTED SPEND > SOFT LIMIT if ( existing_spend_obj.soft_budget_cooldown is False - and existing_spend_obj.litellm_budget_table is not None + and existing_spend_obj.soft_budget is not None and ( _is_projected_spend_over_limit( current_spend=new_spend, - soft_budget_limit=existing_spend_obj.litellm_budget_table[ - "soft_budget" - ], + soft_budget_limit=existing_spend_obj.soft_budget, ) is True ) ): projected_spend, projected_exceeded_date = _get_projected_spend_over_limit( current_spend=new_spend, - soft_budget_limit=existing_spend_obj.litellm_budget_table.get( - "soft_budget", None - ), + soft_budget_limit=existing_spend_obj.soft_budget, ) # type: ignore - soft_limit = existing_spend_obj.litellm_budget_table.get( - "soft_budget", float("inf") - ) + soft_limit = existing_spend_obj.soft_budget call_info = CallInfo( token=existing_spend_obj.token or "", spend=new_spend, @@ -1944,7 +1938,7 @@ async def update_cache( # noqa: PLR0915 max_budget=soft_limit, user_id=existing_spend_obj.user_id, projected_spend=projected_spend, - projected_exceeded_date=projected_exceeded_date, + projected_exceeded_date=str(projected_exceeded_date), event_group=Litellm_EntityType.KEY, ) # alert user