diff --git a/agent.py b/agent.py index f0d02e6..fd7ddde 100644 --- a/agent.py +++ b/agent.py @@ -564,8 +564,8 @@ class CustomerIntegrationEngineer(A2AAgent[CustomerIntegrationEngineerConfig, No drift_detected=drift, ) - @a2a.tool(description="Run dry-run or allowlisted acceptance checks using endpoint metadata and secret references", input_schema=_tool_input_schema(RunAcceptanceRequest), output_schema=_ACCEPTANCE_OUTPUT_SCHEMA, timeout_seconds=180, idempotent=True, max_retries=0, cost_class="deterministic") - async def run_acceptance(self, ctx: RunContext[NoAuth], request: RunAcceptanceRequest) -> Any: + @a2a.tool(description="Run dry-run or allowlisted acceptance checks using endpoint metadata and secret references", input_schema=_tool_input_schema(RunAcceptanceRequest), timeout_seconds=180, idempotent=True, max_retries=0, cost_class="deterministic") + async def run_acceptance(self, ctx: RunContext[NoAuth], request: RunAcceptanceRequest) -> RunAcceptanceResult: 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})]