forked from phoenix/litellm-mirror
fix init custom logger when init OTEL runs
This commit is contained in:
parent
12d8c0d0a4
commit
b36f964217
1 changed files with 9 additions and 1 deletions
|
@ -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(
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue