fix init custom logger when init OTEL runs

This commit is contained in:
Ishaan Jaff 2024-09-09 16:03:39 -07:00
parent 12d8c0d0a4
commit b36f964217

View file

@ -71,7 +71,10 @@ class OpenTelemetryConfig:
class OpenTelemetry(CustomLogger):
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.sdk.resources import Resource
@ -101,6 +104,9 @@ class OpenTelemetry(CustomLogger):
otel_exporter_logger = logging.getLogger("opentelemetry.sdk.trace.export")
otel_exporter_logger.setLevel(logging.DEBUG)
# init CustomLogger params
super().__init__(**kwargs)
def log_success_event(self, 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:
pass
elif self.message_logging is not True:
pass
else:
# Span 2: Raw Request / Response to LLM
raw_request_span = self.tracer.start_span(