fix use SLP status

This commit is contained in:
Ishaan Jaff 2024-11-26 18:10:03 -08:00
parent d2b6054f0c
commit 48c4b272f7

View file

@ -259,6 +259,10 @@ class DataDogLogger(CustomBatchLogger):
if standard_logging_object is None:
raise ValueError("standard_logging_object not found in kwargs")
status = DataDogStatus.INFO
if standard_logging_object.get("status") == "failure":
status = DataDogStatus.WARN
# Build the initial payload
make_json_serializable(standard_logging_object)
json_payload = json.dumps(standard_logging_object)
@ -271,7 +275,7 @@ class DataDogLogger(CustomBatchLogger):
hostname="",
message=json_payload,
service="litellm-server",
status=DataDogStatus.INFO,
status=status,
)
return dd_payload