feat: update create_text_message_object() to accept other input parameters

src/a2a/client/helpers.py

@@ -1,22 +1,32 @@
 """Helper functions for the A2A client."""
 
+from typing import Any
 from uuid import uuid4
 
 from a2a.types import Message, Part, Role, TextPart
 
 
 def create_text_message_object(
-    role: Role = Role.user, content: str = ''
+    role: Role = Role.user,
+    content: str = '',
+    extensions: list[Any] | None = None,
+    metadata: dict[Any, Any] | None = None,
 ) -> Message:
     """Create a Message object containing a single TextPart.
 
     Args:
         role: The role of the message sender (user or agent). Defaults to Role.user.
         content: The text content of the message. Defaults to an empty string.
+        extensions: The extensions of the message. Defaults to an empty list.
+        metadata: The metadata of the message. Defaults to an empty dictionary.
 
     Returns:
         A `Message` object with a new UUID message_id.
     """
     return Message(
-        role=role, parts=[Part(TextPart(text=content))], message_id=str(uuid4())
+        role=role,
+        parts=[Part(TextPart(text=content or ''))],
+        message_id=str(uuid4()),
+        extensions=extensions or [],
+        metadata=metadata or {},
     )