From dfafe44054f9234ac3ef95519127fc3d5934f10d Mon Sep 17 00:00:00 2001 From: JR Boos Date: Mon, 27 Jul 2026 11:04:26 -0400 Subject: [PATCH] feat(shields): add shields to `/a2a` --- src/app/endpoints/a2a.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/app/endpoints/a2a.py b/src/app/endpoints/a2a.py index 8b14e5a95..cef33ed47 100644 --- a/src/app/endpoints/a2a.py +++ b/src/app/endpoints/a2a.py @@ -372,7 +372,12 @@ async def _process_task_streaming( # pylint: disable=too-many-locals ) responses_params = compaction.params - agent = build_agent(client, responses_params, configuration) + agent = build_agent( + client, + responses_params, + configuration, + shields=query_request.shield_ids, + ) except (AgentRunError, APIStatusError, APIConnectionError, RuntimeError) as e: error_response = map_agent_inference_error(e, query_request.model or "") logger.error("Error preparing A2A agent: %s", str(e), exc_info=True)