phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix use SLP status

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-26 18:10:03 -08:00
parent d2b6054f0c
commit 48c4b272f7
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/integrations/datadog/datadog.py
View file

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