phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 11:43:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat - add key name and team alias to slack alerts

Browse source
This commit is contained in:
Ishaan Jaff 2024-05-06 14:30:10 -07:00
parent 1d814a7234
commit edfed32d1c
1 changed files with 2 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

15
litellm/integrations/slack_alerting.py
View file

@ -182,7 +182,7 @@ class SlackAlerting:
and "metadata" in kwargs["litellm_params"] and "metadata" in kwargs["litellm_params"]
): ):
_metadata = kwargs["litellm_params"]["metadata"] _metadata = kwargs["litellm_params"]["metadata"]
request_info = self._add_key_name_and_team_to_alert( request_info = litellm.utils._add_key_name_and_team_to_alert(
request_info=request_info, metadata=_metadata request_info=request_info, metadata=_metadata
) )
@ -259,7 +259,7 @@ class SlackAlerting:
_metadata = request_data["metadata"] _metadata = request_data["metadata"]
_api_base = _metadata.get("api_base", "") _api_base = _metadata.get("api_base", "")
request_info = self._add_key_name_and_team_to_alert( request_info = litellm.utils._add_key_name_and_team_to_alert(
request_info=request_info, metadata=_metadata request_info=request_info, metadata=_metadata
) )
@ -476,14 +476,3 @@ class SlackAlerting:
pass pass
else: else:
print("Error sending slack alert. Error=", response.text) # noqa print("Error sending slack alert. Error=", response.text) # noqa
def _add_key_name_and_team_to_alert(self, request_info: str, metadata: dict) -> str:
_api_key_name = metadata.get("user_api_key_alias", "")
_user_api_key_team_alias = metadata.get("user_api_key_team_alias", "")
if _api_key_name is not None:
request_info = (
f"\n\n\n*API Key Info* \n- Key Name: `{_api_key_name}`\n- Team: `{_user_api_key_team_alias}`\n\n"
+ request_info
)
return request_info