From eacf0e788f52f1e917aa14bd0fcde1ed926cbcbb Mon Sep 17 00:00:00 2001 From: a2a-cloud Date: Mon, 13 Jul 2026 11:21:05 +0000 Subject: [PATCH] a2a-source-edit: write agent.py --- agent.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/agent.py b/agent.py index 1b7d734..f2d14a6 100644 --- a/agent.py +++ b/agent.py @@ -440,8 +440,8 @@ 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), 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: + @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) -> DesignIntegrationResult: req = request.requirements.model_copy( update={"dry_run": _effective_dry_run(self, request.dry_run, request.requirements.dry_run)} )