From 53d02b4a72991ee94c83f9edd3d98321611793f3 Mon Sep 17 00:00:00 2001 From: a2a-cloud Date: Mon, 13 Jul 2026 03:29:29 +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 f3fa4d5..ead88d8 100644 --- a/agent.py +++ b/agent.py @@ -608,7 +608,7 @@ class CustomerIntegrationEngineer(A2AAgent[CustomerIntegrationEngineerConfig, No ) @a2a.tool(description="Run dry-run or allowlisted acceptance checks using endpoint metadata and secret references", input_schema=_ACCEPTANCE_INPUT_SCHEMA, timeout_seconds=180, idempotent=True, max_retries=0, cost_class="deterministic", grant_outputs_prefix="outputs", grant_write_prefixes=("outputs/integrations",)) - async def run_acceptance(self, ctx: RunContext[NoAuth], request: RunAcceptanceRequest) -> RunAcceptanceResult: + async def run_acceptance(self, ctx: RunContext[NoAuth], request: RunAcceptanceRequest) -> Any: dry_run = _effective_dry_run(self, request.dry_run) live_network_block = _live_network_unavailable_reason(self) audit = [_audit("acceptance_started", request.integration_id, {"endpoint_tests": len(request.endpoint_tests), "dry_run": dry_run, "allow_network": request.allow_network})] @@ -669,7 +669,7 @@ class CustomerIntegrationEngineer(A2AAgent[CustomerIntegrationEngineerConfig, No ) @a2a.tool(description="Prepare rollout, rollback, monitoring, and operator handoff documentation with approval-gated delivery planning", input_schema=_HANDOFF_INPUT_SCHEMA, timeout_seconds=120, idempotent=True, max_retries=0, cost_class="deterministic", grant_outputs_prefix="outputs", grant_write_prefixes=("outputs/integrations",)) - async def prepare_handoff(self, ctx: RunContext[NoAuth], request: PrepareHandoffRequest) -> Any: + async def prepare_handoff(self, ctx: RunContext[NoAuth], request: PrepareHandoffRequest) -> PrepareHandoffResult: dry_run = _effective_dry_run(self, request.dry_run) audit = [_audit("handoff_started", request.integration_id, {"dry_run": dry_run, "delivery_actions": len(request.delivery_actions)})] await _emit(ctx, audit[-1])