Skip to content

Commit 793c966

Browse files
committed
feat: added alert type to alert messate to slack for easier handling on slack side.
1 parent 359780b commit 793c966

File tree

1 file changed

+9
-10
lines changed

1 file changed

+9
-10
lines changed

litellm/integrations/SlackAlerting/slack_alerting.py

Lines changed: 9 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -805,9 +805,9 @@ async def region_outage_alerts(
805805
### UNIQUE CACHE KEY ###
806806
cache_key = provider + region_name
807807

808-
outage_value: Optional[ProviderRegionOutageModel] = (
809-
await self.internal_usage_cache.async_get_cache(key=cache_key)
810-
)
808+
outage_value: Optional[
809+
ProviderRegionOutageModel
810+
] = await self.internal_usage_cache.async_get_cache(key=cache_key)
811811

812812
if (
813813
getattr(exception, "status_code", None) is None
@@ -1367,12 +1367,11 @@ async def send_alert(
13671367
# Get the current timestamp
13681368
current_time = datetime.now().strftime("%H:%M:%S")
13691369
_proxy_base_url = os.getenv("PROXY_BASE_URL", None)
1370+
alert_type_formatted = f"Alert type: `{alert_type}`\n"
13701371
if alert_type == "daily_reports" or alert_type == "new_model_added":
1371-
formatted_message = message
1372+
formatted_message = alert_type_formatted + message
13721373
else:
1373-
formatted_message = (
1374-
f"Level: `{level}`\nTimestamp: `{current_time}`\n\nMessage: {message}"
1375-
)
1374+
formatted_message = f"{alert_type_formatted}\n Level: `{level}`\nTimestamp: `{current_time}`\n\nMessage: {message}"
13761375

13771376
if kwargs:
13781377
for key, value in kwargs.items():
@@ -1388,9 +1387,9 @@ async def send_alert(
13881387
self.alert_to_webhook_url is not None
13891388
and alert_type in self.alert_to_webhook_url
13901389
):
1391-
slack_webhook_url: Optional[Union[str, List[str]]] = (
1392-
self.alert_to_webhook_url[alert_type]
1393-
)
1390+
slack_webhook_url: Optional[
1391+
Union[str, List[str]]
1392+
] = self.alert_to_webhook_url[alert_type]
13941393
elif self.default_webhook_url is not None:
13951394
slack_webhook_url = self.default_webhook_url
13961395
else:

0 commit comments

Comments
 (0)