From b2b422dcd5a4a107d5c3aacc84128c0857a9e509 Mon Sep 17 00:00:00 2001 From: a2a-cloud Date: Mon, 13 Jul 2026 11:13:17 +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 02595dd..1c298d8 100644 --- a/agent.py +++ b/agent.py @@ -432,7 +432,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=_PUBLIC_REQUEST_SCHEMA, 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), 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)}