Skip to content

Commit 071c1cb

Browse files
authored
Merge pull request #2218 from jrobertboos/lcore-3227
LCORE-3227: Add Shields to `/a2a`
2 parents 7c75b8d + dfafe44 commit 071c1cb

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)
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)