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)