diff --git a/agent.py b/agent.py index 475b074..cd88133 100644 --- a/agent.py +++ b/agent.py @@ -426,7 +426,9 @@ class CustomerIntegrationEngineer(A2AAgent[CustomerIntegrationEngineerConfig, No @a2a.tool(description="Capture requirements, produce architecture and field mapping artifacts", 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": request.dry_run or request.requirements.dry_run}) + req = request.requirements.model_copy( + update={"dry_run": _effective_dry_run(self, request.dry_run, request.requirements.dry_run)} + ) audit = [_audit("design_started", req.integration_id, {"dry_run": req.dry_run})] await _emit(ctx, audit[-1]) warnings = _requirements_warnings(req)