From a49f2a5148e7b2cffb939520c40a4bfbbe7cab4e Mon Sep 17 00:00:00 2001 From: Lei Zhang Date: Thu, 12 Jun 2025 16:46:02 +0800 Subject: [PATCH 1/2] feat: Add append and last_chunk to add_artifact method on TaskUpdater --- src/a2a/server/tasks/task_updater.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/src/a2a/server/tasks/task_updater.py b/src/a2a/server/tasks/task_updater.py index 26b93ba8..95fd10ea 100644 --- a/src/a2a/server/tasks/task_updater.py +++ b/src/a2a/server/tasks/task_updater.py @@ -65,12 +65,14 @@ async def update_status( ) ) - async def add_artifact( + async def add_artifact( # noqa: PLR0913 self, parts: list[Part], artifact_id: str | None = None, name: str | None = None, metadata: dict[str, Any] | None = None, + append: bool | None = None, + last_chunk: bool | None = None, ) -> None: """Adds an artifact chunk to the task and publishes a `TaskArtifactUpdateEvent`. @@ -79,6 +81,8 @@ async def add_artifact( artifact_id: The ID of the artifact. A new UUID is generated if not provided. name: Optional name for the artifact. metadata: Optional metadata for the artifact. + append: Optional boolean indicating if this chunk appends to a previous one. + last_chunk: Optional boolean indicating if this is the last chunk. """ if not artifact_id: artifact_id = str(uuid.uuid4()) @@ -93,6 +97,8 @@ async def add_artifact( parts=parts, metadata=metadata, ), + append=append, + last_chunk=last_chunk ) ) From d2117726f46509c42066cb2b04156999a58cfb65 Mon Sep 17 00:00:00 2001 From: kthota-g Date: Tue, 17 Jun 2025 13:56:56 -0700 Subject: [PATCH 2/2] Update src/a2a/server/tasks/task_updater.py Co-authored-by: Holt Skinner <13262395+holtskinner@users.noreply.github.com> --- src/a2a/server/tasks/task_updater.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/a2a/server/tasks/task_updater.py b/src/a2a/server/tasks/task_updater.py index 95fd10ea..6f8aca73 100644 --- a/src/a2a/server/tasks/task_updater.py +++ b/src/a2a/server/tasks/task_updater.py @@ -98,7 +98,7 @@ async def add_artifact( # noqa: PLR0913 metadata=metadata, ), append=append, - last_chunk=last_chunk + lastChunk=last_chunk ) )