diff --git a/src/a2a/server/tasks/task_updater.py b/src/a2a/server/tasks/task_updater.py index 529cc4fd..d4f3662e 100644 --- a/src/a2a/server/tasks/task_updater.py +++ b/src/a2a/server/tasks/task_updater.py @@ -49,6 +49,7 @@ async def update_status( message: Message | None = None, final: bool = False, timestamp: str | None = None, + metadata: dict[str, Any] | None = None, ) -> None: """Updates the status of the task and publishes a `TaskStatusUpdateEvent`. @@ -57,6 +58,7 @@ async def update_status( message: An optional message associated with the status update. final: If True, indicates this is the final status update for the task. timestamp: Optional ISO 8601 datetime string. Defaults to current time. + metadata: Optional metadata for extensions. """ async with self._lock: if self._terminal_state_reached: @@ -77,6 +79,7 @@ async def update_status( task_id=self.task_id, context_id=self.context_id, final=final, + metadata=metadata, status=TaskStatus( state=state, message=message,