Skip to content

Commit db852ba

Browse files
committed
feat(shields): add shields to /a2a
1 parent b2ab480 commit db852ba

1 file changed

Lines changed: 6 additions & 1 deletion

File tree

src/app/endpoints/a2a.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -372,7 +372,12 @@ async def _process_task_streaming( # pylint: disable=too-many-locals
372372
)
373373
responses_params = compaction.params
374374

375-
agent = build_agent(client, responses_params, configuration.skills)
375+
agent = build_agent(
376+
client,
377+
responses_params,
378+
configuration,
379+
shields=query_request.shield_ids,
380+
)
376381
except (AgentRunError, APIStatusError, APIConnectionError, RuntimeError) as e:
377382
error_response = map_agent_inference_error(e, query_request.model or "")
378383
logger.error("Error preparing A2A agent: %s", str(e), exc_info=True)

0 commit comments

Comments
 (0)