trace helpers

guglielmo-san · guglielmo-san · commit ba8e8dbd885a · 2026-04-17T07:41:10.000Z
diff --git a/src/a2a/server/tasks/task_manager.py b/src/a2a/server/tasks/task_manager.py
@@ -13,11 +13,13 @@
     TaskStatusUpdateEvent,
 )
 from a2a.utils.errors import InvalidParamsError
+from a2a.utils.telemetry import trace_function
 
 
 logger = logging.getLogger(__name__)
 
 
+@trace_function()
 def append_artifact_to_task(task: Task, event: TaskArtifactUpdateEvent) -> None:
     """Helper method for updating a Task object with new artifact data from an event.
 

Original file line number	Diff line number	Diff line change
`@@ -13,11 +13,13 @@`
`13`	`13`	`TaskStatusUpdateEvent,`
`14`	`14`	`)`
`15`	`15`	`from a2a.utils.errors import InvalidParamsError`
	`16`	`+from a2a.utils.telemetry import trace_function`
`16`	`17`
`17`	`18`
`18`	`19`	`logger = logging.getLogger(__name__)`
`19`	`20`
`20`	`21`
	`22`	`+@trace_function()`
`21`	`23`	`def append_artifact_to_task(task: Task, event: TaskArtifactUpdateEvent) -> None:`
`22`	`24`	`"""Helper method for updating a Task object with new artifact data from an event.`
`23`	`25`