add final and metadata fields to new agent messages

Author: martimfasantos

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,13 +36,17 @@ def new_agent_text_message(
         messageId=str(uuid.uuid4()),
         taskId=task_id,
         contextId=context_id,
+        final=final,
+        metadata=metadata,
     )
 
 
 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,
     )