From b7cba2b0c94a2829ecc4675bc527698ddf7815c6 Mon Sep 17 00:00:00 2001 From: a2a-cloud Date: Mon, 13 Jul 2026 11:17:34 +0000 Subject: [PATCH] a2a-source-edit: write agent.py --- agent.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/agent.py b/agent.py index 505ddbd..fef9b6f 100644 --- a/agent.py +++ b/agent.py @@ -440,7 +440,7 @@ class CustomerIntegrationEngineer(A2AAgent[CustomerIntegrationEngineerConfig, No ConsumerSetupField.config("ARGOCD_APP_REF", label="Argo CD app reference", description="Optional caller-owned Argo CD app reference.", required=False), ) - @a2a.tool(description="Capture requirements, produce architecture and field mapping artifacts", input_schema=_tool_input_schema(DesignIntegrationRequest), timeout_seconds=120, idempotent=True, max_retries=0, cost_class="deterministic") + @a2a.tool(description="Capture requirements, produce architecture and field mapping artifacts", input_schema=_tool_input_schema(DesignIntegrationRequest), output_schema=_DESIGN_OUTPUT_SCHEMA, timeout_seconds=120, idempotent=True, max_retries=0, cost_class="deterministic") async def design_integration(self, ctx: RunContext[NoAuth], request: DesignIntegrationRequest) -> Any: req = request.requirements.model_copy( update={"dry_run": _effective_dry_run(self, request.dry_run, request.requirements.dry_run)}