From 0071abd9ea00a213558155f6b8419d4761a62d94 Mon Sep 17 00:00:00 2001
From: martimfasantos <72747170+martimfasantos@users.noreply.github.com>
Date: Tue, 20 May 2025 17:52:07 +0100
Subject: [PATCH 1/3] add final and metadata fields to new agent messages

---
 src/a2a/utils/message.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/a2a/utils/message.py b/src/a2a/utils/message.py
index 2e156f6e..5c99a830 100644
--- a/src/a2a/utils/message.py
+++ b/src/a2a/utils/message.py
@@ -1,6 +1,7 @@
 """Utility functions for creating and handling A2A Message objects."""
 
 import uuid
+from typing import Any
 
 from a2a.types import (
     Message,
@@ -14,6 +15,8 @@ def new_agent_text_message(
     text: str,
     context_id: str | None = None,
     task_id: str | None = None,
+    final: bool | None = None,
+    metadata: dict[str, Any] | None = None,
 ) -> Message:
     """Creates a new agent message containing a single TextPart.
 
@@ -33,6 +36,8 @@ def new_agent_text_message(
         messageId=str(uuid.uuid4()),
         taskId=task_id,
         contextId=context_id,
+        final=final,
+        metadata=metadata,
     )
 
 
@@ -40,6 +45,8 @@ def new_agent_parts_message(
     parts: list[Part],
     context_id: str | None = None,
     task_id: str | None = None,
+    final: bool | None = None,
+    metadata: dict[str, Any] | None = None,
 ):
     """Creates a new agent message containing a list of Parts.
 
@@ -59,6 +66,8 @@ def new_agent_parts_message(
         messageId=str(uuid.uuid4()),
         taskId=task_id,
         contextId=context_id,
+        final=final,
+        metadata=metadata,
     )
 
 

From 78197dc0403a5ed4ce296e9d9cfbccc5c40d818d Mon Sep 17 00:00:00 2001
From: martimfasantos <72747170+martimfasantos@users.noreply.github.com>
Date: Wed, 21 May 2025 10:03:52 +0100
Subject: [PATCH 2/3] removed incorrect docstrings

---
 src/a2a/utils/message.py | 11 -----------
 1 file changed, 11 deletions(-)

diff --git a/src/a2a/utils/message.py b/src/a2a/utils/message.py
index 5c99a830..283958ca 100644
--- a/src/a2a/utils/message.py
+++ b/src/a2a/utils/message.py
@@ -1,7 +1,6 @@
 """Utility functions for creating and handling A2A Message objects."""
 
 import uuid
-from typing import Any
 
 from a2a.types import (
     Message,
@@ -15,8 +14,6 @@ def new_agent_text_message(
     text: str,
     context_id: str | None = None,
     task_id: str | None = None,
-    final: bool | None = None,
-    metadata: dict[str, Any] | None = None,
 ) -> Message:
     """Creates a new agent message containing a single TextPart.
 
@@ -24,8 +21,6 @@ def new_agent_text_message(
         text: The text content of the message.
         context_id: The context ID for the message.
         task_id: The task ID for the message.
-        final: Optional boolean indicating if this is the final message.
-        metadata: Optional metadata for the message.
 
     Returns:
         A new `Message` object with role 'agent'.
@@ -36,8 +31,6 @@ def new_agent_text_message(
         messageId=str(uuid.uuid4()),
         taskId=task_id,
         contextId=context_id,
-        final=final,
-        metadata=metadata,
     )
 
 
@@ -45,8 +38,6 @@ def new_agent_parts_message(
     parts: list[Part],
     context_id: str | None = None,
     task_id: str | None = None,
-    final: bool | None = None,
-    metadata: dict[str, Any] | None = None,
 ):
     """Creates a new agent message containing a list of Parts.
 
@@ -66,8 +57,6 @@ def new_agent_parts_message(
         messageId=str(uuid.uuid4()),
         taskId=task_id,
         contextId=context_id,
-        final=final,
-        metadata=metadata,
     )
 
 

From 264f3b8de3f65e66e0deb7d1ce9303d483b72317 Mon Sep 17 00:00:00 2001
From: martimfasantos <72747170+martimfasantos@users.noreply.github.com>
Date: Wed, 21 May 2025 10:05:18 +0100
Subject: [PATCH 3/3] removed incorrect docstrings

---
 src/a2a/utils/message.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/src/a2a/utils/message.py b/src/a2a/utils/message.py
index 283958ca..fd58a2fa 100644
--- a/src/a2a/utils/message.py
+++ b/src/a2a/utils/message.py
@@ -45,8 +45,6 @@ def new_agent_parts_message(
         parts: The list of `Part` objects for the message content.
         context_id: The context ID for the message.
         task_id: The task ID for the message.
-        final: Optional boolean indicating if this is the final message.
-        metadata: Optional metadata for the message.
 
     Returns:
         A new `Message` object with role 'agent'.