update create_text_message_object function to handle all inputs

Author: yolagil

src/a2a/client/helpers.py

@@ -1,22 +1,22 @@
 """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.
-
-    Returns:
-        A `Message` object with a new UUID message_id.
-    """
+    """Create a Message object containing a single TextPart."""
     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 {},
     )