phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 10:44:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix trace hierarchy on otel

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-06 15:37:23 -07:00
parent f4eb04640c
commit 196884a779
2 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/proxy/common_utils/init_callbacks.py
View file

@ -31,10 +31,12 @@ def initialize_callbacks_on_proxy(
imported_list.append(callback)
elif isinstance(callback, str) and callback == "otel":
from litellm.integrations.opentelemetry import OpenTelemetry
from litellm.proxy import proxy_server
open_telemetry_logger = OpenTelemetry()
imported_list.append(open_telemetry_logger)
setattr(proxy_server, "open_telemetry_logger", open_telemetry_logger)
elif isinstance(callback, str) and callback == "presidio":
from litellm.proxy.hooks.presidio_pii_masking import (
_OPTIONAL_PresidioPIIMasking,