fix: ensure failed states are handled for push, poll

lib/crewai/src/crewai/a2a/updates/polling/handler.py

@@ -5,13 +5,18 @@ from __future__ import annotations
 import asyncio
 import time
 from typing import TYPE_CHECKING, Any
+import uuid
 
 from a2a.client import Client
+from a2a.client.errors import A2AClientHTTPError
 from a2a.types import (
     AgentCard,
     Message,
+    Part,
+    Role,
     TaskQueryParams,
     TaskState,
+    TextPart,
 )
 from typing_extensions import Unpack
 
@@ -28,6 +33,7 @@ from crewai.events.event_bus import crewai_event_bus
 from crewai.events.types.a2a_events import (
     A2APollingStartedEvent,
     A2APollingStatusEvent,
+    A2AResponseReceivedEvent,
 )
 
 
@@ -129,7 +135,10 @@ class PollingHandler:
         agent_role = kwargs.get("agent_role")
         history_length = kwargs.get("history_length", 100)
         max_polls = kwargs.get("max_polls")
+        context_id = kwargs.get("context_id")
+        task_id = kwargs.get("task_id")
 
+        try:
             result_or_task_id = await send_message_and_get_task_id(
                 event_stream=client.send_message(message),
                 new_messages=new_messages,
@@ -179,3 +188,31 @@ class PollingHandler:
                 error=f"Unexpected task state: {final_task.status.state}",
                 history=new_messages,
             )
+
+        except A2AClientHTTPError as e:
+            error_msg = f"HTTP Error {e.status_code}: {e!s}"
+
+            error_message = Message(
+                role=Role.agent,
+                message_id=str(uuid.uuid4()),
+                parts=[Part(root=TextPart(text=error_msg))],
+                context_id=context_id,
+                task_id=task_id,
+            )
+            new_messages.append(error_message)
+
+            crewai_event_bus.emit(
+                agent_branch,
+                A2AResponseReceivedEvent(
+                    response=error_msg,
+                    turn_number=turn_number,
+                    is_multiturn=is_multiturn,
+                    status="failed",
+                    agent_role=agent_role,
+                ),
+            )
+            return TaskStateResult(
+                status=TaskState.failed,
+                error=error_msg,
+                history=new_messages,
+            )

lib/crewai/src/crewai/a2a/updates/push_notifications/handler.py

@@ -4,12 +4,17 @@ from __future__ import annotations
 
 import logging
 from typing import TYPE_CHECKING, Any
+import uuid
 
 from a2a.client import Client
+from a2a.client.errors import A2AClientHTTPError
 from a2a.types import (
     AgentCard,
     Message,
+    Part,
+    Role,
     TaskState,
+    TextPart,
 )
 from typing_extensions import Unpack
 
@@ -26,6 +31,7 @@ from crewai.events.event_bus import crewai_event_bus
 from crewai.events.types.a2a_events import (
     A2APushNotificationRegisteredEvent,
     A2APushNotificationTimeoutEvent,
+    A2AResponseReceivedEvent,
 )
 
 
@@ -107,6 +113,8 @@ class PushNotificationHandler:
         turn_number = kwargs.get("turn_number", 0)
         is_multiturn = kwargs.get("is_multiturn", False)
         agent_role = kwargs.get("agent_role")
+        context_id = kwargs.get("context_id")
+        task_id = kwargs.get("task_id")
 
         if config is None:
             return TaskStateResult(
@@ -122,10 +130,7 @@ class PushNotificationHandler:
                 history=new_messages,
             )
 
-        # Note: Push notification config is now included in the initial send_message
-        # request via ClientConfig.push_notification_configs, so no separate
-        # set_task_callback call is needed. This avoids race conditions where
-        # the task completes before the callback is registered.
+        try:
             result_or_task_id = await send_message_and_get_task_id(
                 event_stream=client.send_message(message),
                 new_messages=new_messages,
@@ -185,3 +190,31 @@ class PushNotificationHandler:
                 error=f"Unexpected task state: {final_task.status.state}",
                 history=new_messages,
             )
+
+        except A2AClientHTTPError as e:
+            error_msg = f"HTTP Error {e.status_code}: {e!s}"
+
+            error_message = Message(
+                role=Role.agent,
+                message_id=str(uuid.uuid4()),
+                parts=[Part(root=TextPart(text=error_msg))],
+                context_id=context_id,
+                task_id=task_id,
+            )
+            new_messages.append(error_message)
+
+            crewai_event_bus.emit(
+                agent_branch,
+                A2AResponseReceivedEvent(
+                    response=error_msg,
+                    turn_number=turn_number,
+                    is_multiturn=is_multiturn,
+                    status="failed",
+                    agent_role=agent_role,
+                ),
+            )
+            return TaskStateResult(
+                status=TaskState.failed,
+                error=error_msg,
+                history=new_messages,
+            )