diff --git a/agent.py b/agent.py index fd7ddde..c20b655 100644 --- a/agent.py +++ b/agent.py @@ -635,8 +635,8 @@ class CustomerIntegrationEngineer(A2AAgent[CustomerIntegrationEngineerConfig, No network_calls_made=network_calls, ) - @a2a.tool(description="Prepare rollout, rollback, monitoring, and operator handoff documentation with approval-gated delivery planning", input_schema=_tool_input_schema(PrepareHandoffRequest), output_schema=_HANDOFF_OUTPUT_SCHEMA, timeout_seconds=120, idempotent=True, max_retries=0, cost_class="deterministic") - async def prepare_handoff(self, ctx: RunContext[NoAuth], request: PrepareHandoffRequest) -> Any: + @a2a.tool(description="Prepare rollout, rollback, monitoring, and operator handoff documentation with approval-gated delivery planning", input_schema=_tool_input_schema(PrepareHandoffRequest), timeout_seconds=120, idempotent=True, max_retries=0, cost_class="deterministic") + 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])