Skip to content

Commit b7ce5e1

Browse files
google-genai-botcopybara-github
authored andcommitted
fix: Yield event with error code when agent run raised A2AClientHTTPError
PiperOrigin-RevId: 840816236
1 parent 9d2388a commit b7ce5e1

File tree

1 file changed

+19
-1
lines changed

1 file changed

+19
-1
lines changed

src/google/adk/agents/remote_a2a_agent.py

Lines changed: 19 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@
3131
from a2a.client.card_resolver import A2ACardResolver
3232
from a2a.client.client import ClientConfig as A2AClientConfig
3333
from a2a.client.client_factory import ClientFactory as A2AClientFactory
34-
from a2a.client.errors import A2AClientError
34+
from a2a.client.errors import A2AClientHTTPError
3535
from a2a.types import AgentCard
3636
from a2a.types import Message as A2AMessage
3737
from a2a.types import Part as A2APart
@@ -558,6 +558,24 @@ async def _run_async_impl(
558558

559559
yield event
560560

561+
except A2AClientHTTPError as e:
562+
error_message = f"A2A request failed: {e}"
563+
logger.error(error_message)
564+
yield Event(
565+
author=self.name,
566+
error_message=error_message,
567+
invocation_id=ctx.invocation_id,
568+
branch=ctx.branch,
569+
custom_metadata={
570+
A2A_METADATA_PREFIX
571+
+ "request": a2a_request.model_dump(
572+
exclude_none=True, by_alias=True
573+
),
574+
A2A_METADATA_PREFIX + "error": error_message,
575+
A2A_METADATA_PREFIX + "status_code": str(e.status_code),
576+
},
577+
)
578+
561579
except Exception as e:
562580
error_message = f"A2A request failed: {e}"
563581
logger.error(error_message)

0 commit comments

Comments
 (0)