fix(slack_alerting.py): if 'turn_off_message_logging' enabled, do not log the message to logging integration

This commit is contained in:
Krrish Dholakia 2024-05-13 14:02:43 -07:00
parent 7fa203c810
commit bd2f46fd75

View file

@ -238,6 +238,8 @@ class SlackAlerting(CustomLogger):
end_time=end_time,
)
)
if litellm.turn_off_message_logging:
messages = "Message not logged. `litellm.turn_off_message_logging=True`."
request_info = f"\nRequest Model: `{model}`\nAPI Base: `{api_base}`\nMessages: `{messages}`"
slow_message = f"`Responses are slow - {round(time_difference_float,2)}s response time > Alerting threshold: {self.alerting_threshold}s`"
if time_difference_float > self.alerting_threshold:
@ -460,6 +462,11 @@ class SlackAlerting(CustomLogger):
messages = messages[:100]
except:
messages = ""
if litellm.turn_off_message_logging:
messages = (
"Message not logged. `litellm.turn_off_message_logging=True`."
)
request_info = f"\nRequest Model: `{model}`\nMessages: `{messages}`"
else:
request_info = ""