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

include user_api_key_dict debug info on OTEL post

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-26 20:39:37 -08:00
parent fe151db27c
commit ef7f796d03
1 changed files with 13 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

13
litellm/integrations/opentelemetry.py
View file

@ -264,6 +264,8 @@ class OpenTelemetry(CustomLogger):
original_exception: Exception, original_exception: Exception,
user_api_key_dict: UserAPIKeyAuth, user_api_key_dict: UserAPIKeyAuth,
): ):
import traceback
from opentelemetry import trace from opentelemetry import trace
from opentelemetry.trace import Status, StatusCode from opentelemetry.trace import Status, StatusCode
@ -282,6 +284,17 @@ class OpenTelemetry(CustomLogger):
key="exception", key="exception",
value=str(original_exception), value=str(original_exception),
) )
self.safe_set_attribute(
span=exception_logging_span,
key="exception.traceback",
value=traceback.format_exc(),
)
for key, value in user_api_key_dict.model_dump().items():
self.safe_set_attribute(
span=exception_logging_span,
key=f"user_api_key_dict.{key}",
value=value,
)
exception_logging_span.set_status(Status(StatusCode.ERROR)) exception_logging_span.set_status(Status(StatusCode.ERROR))
exception_logging_span.end(end_time=self._to_ns(datetime.now())) exception_logging_span.end(end_time=self._to_ns(datetime.now()))