diff --git a/automon/integrations/openTelemetryWrapper/client.py b/automon/integrations/openTelemetryWrapper/client.py index ea816ff1..a84893db 100644 --- a/automon/integrations/openTelemetryWrapper/client.py +++ b/automon/integrations/openTelemetryWrapper/client.py @@ -49,7 +49,7 @@ async def record_exception(self, exception: Exception): def start_as_current_span( self, name: str, - attributes: types.Attributes = None, + attributes: any = None, **kwargs ): logger.debug(dict(name=name, attributes=attributes, kwargs=kwargs)) @@ -71,7 +71,7 @@ async def start_producer(self): return async def test(self): - with self.config.tracer.start_as_current_span(name='rootSpan') as trace_root: + with self.start_as_current_span(name='rootSpan') as trace_root: self.add_event('AAAAAAAA') with self.config.tracer.start_as_current_span(name='childSpan') as trace_child: