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

fix init custom logger when init OTEL runs

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-09 16:03:39 -07:00
parent 12d8c0d0a4
commit b36f964217
1 changed files with 9 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

10
litellm/integrations/opentelemetry.py
View file

@ -71,7 +71,10 @@ class OpenTelemetryConfig:
class OpenTelemetry(CustomLogger): class OpenTelemetry(CustomLogger):
def __init__( def __init__(
self, config=OpenTelemetryConfig.from_env(), callback_name: Optional[str] = None self,
config=OpenTelemetryConfig.from_env(),
callback_name: Optional[str] = None,
**kwargs,
): ):
from opentelemetry import trace from opentelemetry import trace
from opentelemetry.sdk.resources import Resource from opentelemetry.sdk.resources import Resource
@ -101,6 +104,9 @@ class OpenTelemetry(CustomLogger):
otel_exporter_logger = logging.getLogger("opentelemetry.sdk.trace.export") otel_exporter_logger = logging.getLogger("opentelemetry.sdk.trace.export")
otel_exporter_logger.setLevel(logging.DEBUG) otel_exporter_logger.setLevel(logging.DEBUG)
# init CustomLogger params
super().__init__(**kwargs)
def log_success_event(self, kwargs, response_obj, start_time, end_time): def log_success_event(self, kwargs, response_obj, start_time, end_time):
self._handle_sucess(kwargs, response_obj, start_time, end_time) self._handle_sucess(kwargs, response_obj, start_time, end_time)
@ -261,6 +267,8 @@ class OpenTelemetry(CustomLogger):
if litellm.turn_off_message_logging is True: if litellm.turn_off_message_logging is True:
pass pass
elif self.message_logging is not True:
pass
else: else:
# Span 2: Raw Request / Response to LLM # Span 2: Raw Request / Response to LLM
raw_request_span = self.tracer.start_span( raw_request_span = self.tracer.start_span(