wip

Author: guglielmo-san

src/a2a/helpers/__init__.py

@@ -1 +1,33 @@
 """Helper functions for the A2A Python SDK."""
+
+from a2a.helpers.agent_card import display_agent_card
+from a2a.helpers.types import (
+    get_artifact_text,
+    get_message_text,
+    get_stream_response_text,
+    get_text_parts,
+    new_artifact,
+    new_message,
+    new_task,
+    new_task_from_request,
+    new_text_artifact,
+    new_text_artifact_update_event,
+    new_text_message,
+    new_text_status_update_event,
+)
+
+__all__ = [
+    'display_agent_card',
+    'get_artifact_text',
+    'get_message_text',
+    'get_stream_response_text',
+    'get_text_parts',
+    'new_artifact',
+    'new_message',
+    'new_task',
+    'new_task_from_request',
+    'new_text_artifact',
+    'new_text_artifact_update_event',
+    'new_text_message',
+    'new_text_status_update_event',
+]

src/a2a/helpers/types.py

@@ -197,9 +197,8 @@ def get_stream_response_text(response: StreamResponse, delimiter: str = '\n') ->
     if response.HasField('message'):
         return get_message_text(response.message, delimiter)
     elif response.HasField('task'):
-        if response.task.status.HasField('message'):
-            return get_message_text(response.task.status.message, delimiter)
-        return ''
+        texts = [get_artifact_text(a, delimiter) for a in response.task.artifacts]
+        return delimiter.join(t for t in texts if t)
     elif response.HasField('status_update'):
         if response.status_update.status.HasField('message'):
             return get_message_text(response.status_update.status.message, delimiter)

src/a2a/server/agent_execution/context.py

@@ -12,7 +12,7 @@
     SendMessageRequest,
     Task,
 )
-from a2a.utils import get_message_text
+from a2a.helpers.types import get_message_text
 from a2a.utils.errors import InvalidParamsError
 
 

tests/client/transports/test_grpc_client.py

@@ -35,7 +35,7 @@
     TaskStatus,
     TaskStatusUpdateEvent,
 )
-from a2a.utils import get_text_parts
+from a2a.helpers.types import get_text_parts
 
 
 @pytest.fixture