ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat - email alerts

Browse Source
This commit is contained in:
Ishaan Jaff 2024-05-24 15:23:36 -07:00
parent 4f4607a4df
commit 06370d24b4
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
litellm/integrations/slack_alerting.py
View File

@ -663,10 +663,10 @@ class SlackAlerting(CustomLogger):
# check if crossed budget
if user_info.spend >= user_info.max_budget:
event = "budget_crossed"
event_message += "Budget Crossed"
event_message += f"Budget Crossed\n Total Budget:`{user_info.max_budget}`"
elif percent_left <= 0.05:
event = "threshold_crossed"
event_message += "5% Threshold Crossed"
event_message += "5% Threshold Crossed "
elif percent_left <= 0.15:
event = "threshold_crossed"
event_message += "15% Threshold Crossed"
@ -796,6 +796,10 @@ Model Info:
user_name = webhook_event.user_id
max_budget = webhook_event.max_budget
email_html_content = "Alert from LiteLLM Server"
if recipient_email is None:
verbose_proxy_logger.error(
"Trying to send email alert to no recipient", extra=webhook_event.dict()
)
if webhook_event.event == "budget_crossed":
email_html_content = f"""
@ -819,7 +823,6 @@ Model Info:
"subject": event_name,
"html": email_html_content,
}
headers = {"Content-type": "application/json"}
response = await send_email(
receiver_email=email_event["to"],