From 48c4b272f7625005a702c5a9ff1340322716fead Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Tue, 26 Nov 2024 18:10:03 -0800
Subject: [PATCH] fix use SLP status

---
 litellm/integrations/datadog/datadog.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/litellm/integrations/datadog/datadog.py b/litellm/integrations/datadog/datadog.py
index 01283ac2d..a100c2863 100644
--- a/litellm/integrations/datadog/datadog.py
+++ b/litellm/integrations/datadog/datadog.py
@@ -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