feat: Add get_data_parts() and get_file_parts() helper methods

Author: holtskinner

src/a2a/utils/__init__.py

@@ -17,6 +17,8 @@
     create_task_obj,
 )
 from a2a.utils.message import (
+    get_data_parts,
+    get_file_parts,
     get_message_text,
     get_text_parts,
     new_agent_parts_message,
@@ -37,6 +39,8 @@
     'build_text_artifact',
     'completed_task',
     'create_task_obj',
+    'get_data_parts',
+    'get_file_parts',
     'get_message_text',
     'get_text_parts',
     'new_agent_parts_message',

src/a2a/utils/message.py

@@ -2,7 +2,13 @@
 
 import uuid
 
+from typing import Any
+
 from a2a.types import (
+    DataPart,
+    FilePart,
+    FileWithBytes,
+    FileWithUri,
     Message,
     Part,
     Role,
@@ -70,6 +76,30 @@ def get_text_parts(parts: list[Part]) -> list[str]:
     return [part.root.text for part in parts if isinstance(part.root, TextPart)]
 
 
+def get_data_parts(parts: list[Part]) -> list[dict[str, Any]]:
+    """Extracts dictionary data from all DataPart objects in a list of Parts.
+
+    Args:
+        parts: A list of `Part` objects.
+
+    Returns:
+        A list of dictionaries containing the data from any `DataPart` objects found.
+    """
+    return [part.root.data for part in parts if isinstance(part.root, DataPart)]
+
+
+def get_file_parts(parts: list[Part]) -> list[FileWithBytes | FileWithUri]:
+    """Extracts file data from all FilePart objects in a list of Parts.
+
+    Args:
+        parts: A list of `Part` objects.
+
+    Returns:
+        A list of `FileWithBytes` or `FileWithUri` objects containing the file data from any `FilePart` objects found.
+    """
+    return [part.root.file for part in parts if isinstance(part.root, FilePart)]
+
+
 def get_message_text(message: Message, delimiter: str = '\n') -> str:
     """Extracts and joins all text content from a Message's parts.