diff --git a/src/a2a/server/events/event_consumer.py b/src/a2a/server/events/event_consumer.py index 8414e2d17..c220e96c8 100644 --- a/src/a2a/server/events/event_consumer.py +++ b/src/a2a/server/events/event_consumer.py @@ -18,7 +18,10 @@ logger = logging.getLogger(__name__) -@trace_class(kind=SpanKind.SERVER) +@trace_class( + kind=SpanKind.SERVER, + exclude_list=['consume_all'], +) class EventConsumer: """Consumer to read events from the agent event queue.""" diff --git a/src/a2a/server/events/event_queue.py b/src/a2a/server/events/event_queue.py index bb4d7b9b4..9e1ebd84f 100644 --- a/src/a2a/server/events/event_queue.py +++ b/src/a2a/server/events/event_queue.py @@ -93,7 +93,10 @@ async def enqueue_event(self, event: Event) -> None: """ -@trace_class(kind=SpanKind.SERVER) +@trace_class( + kind=SpanKind.SERVER, + exclude_list=['enqueue_event', 'dequeue_event', 'task_done', 'is_closed'], +) class EventQueueLegacy(EventQueue): """Event queue for A2A responses from agent. diff --git a/src/a2a/server/events/event_queue_v2.py b/src/a2a/server/events/event_queue_v2.py index 224cb8e56..9c29cb063 100644 --- a/src/a2a/server/events/event_queue_v2.py +++ b/src/a2a/server/events/event_queue_v2.py @@ -20,7 +20,10 @@ logger = logging.getLogger(__name__) -@trace_class(kind=SpanKind.SERVER) +@trace_class( + kind=SpanKind.SERVER, + exclude_list=['enqueue_event', 'dequeue_event', 'task_done', 'is_closed'], +) class EventQueueSource(EventQueue): """The Parent EventQueue. diff --git a/src/a2a/server/events/in_memory_queue_manager.py b/src/a2a/server/events/in_memory_queue_manager.py index 0beb354f9..365ed525a 100644 --- a/src/a2a/server/events/in_memory_queue_manager.py +++ b/src/a2a/server/events/in_memory_queue_manager.py @@ -9,7 +9,10 @@ from a2a.utils.telemetry import SpanKind, trace_class -@trace_class(kind=SpanKind.SERVER) +@trace_class( + kind=SpanKind.SERVER, + exclude_list=['add', 'get', 'create_or_tap'], +) class InMemoryQueueManager(QueueManager): """InMemoryQueueManager is used for a single binary management.