From 6e4098fa26c261b64ec3c01dce19531d99757c70 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Mon, 5 Aug 2024 20:21:26 -0700
Subject: [PATCH] otel log service errors

---
 litellm/integrations/opentelemetry.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/litellm/integrations/opentelemetry.py b/litellm/integrations/opentelemetry.py
index 72031e58c..379c41880 100644
--- a/litellm/integrations/opentelemetry.py
+++ b/litellm/integrations/opentelemetry.py
@@ -165,6 +165,7 @@ class OpenTelemetry(CustomLogger):
     async def async_service_failure_hook(
         self,
         payload: ServiceLoggerPayload,
+        error: Optional[str] = "",
         parent_otel_span: Optional[Span] = None,
         start_time: Optional[Union[datetime, float]] = None,
         end_time: Optional[Union[float, datetime]] = None,
@@ -199,6 +200,8 @@ class OpenTelemetry(CustomLogger):
             service_logging_span.set_attribute(
                 key="service", value=payload.service.value
             )
+            if error:
+                service_logging_span.set_attribute(key="error", value=error)
             if event_metadata:
                 for key, value in event_metadata.items():
                     if isinstance(value, dict):