用 Go 编写可取消的 Agent Runtime

context 能把超时与取消传播至模型、检索和工具。

问题背景

生成式模型擅长处理模糊语言和提出候选方案，却不天然保证状态一致、权限正确或动作可恢复。讨论“用 Go 编写可取消的 Agent Runtime”时，重点不应只是跑通演示，而是系统失败后是否知道发生了什么、能否安全继续，以及结果能否重复验证。

设计方案

所有 I/O 接收 context，errgroup 管理并发，取消后清理再提交状态。

在“可取消运行循环”进入实现前，先标注输入来源、调用身份和数据等级；输出通过结构与领域校验后才允许进入下一步。涉及外部副作用时，执行记录必须包含稳定业务键和脱敏证据。

实施产物：可取消运行循环

所有模型和工具调用传递同一 context；每步 select 监听取消；资源用 defer 回收；终止原因写入结果。先在测试环境保存一条完整运行证据，再接入真实写操作；每次变更都必须能定位到配置、代码和数据版本。

type GoAgentRuntimeConfig struct {
    Timeout time.Duration
    MaxInFlight int
}

func RunGoAgentRuntime(ctx context.Context, cfg GoAgentRuntimeConfig) error {
    return executeWithPolicy(ctx, cfg.Timeout, cfg.MaxInFlight)
}

示例只表达边界，不是可直接复制的完整框架配置。落地时应使用当前依赖版本的官方 API，并把超时、身份和审计接到现有基础设施。

实施步骤

选一个可回放的小任务，写清输入、成功条件和禁止发生的副作用。
实现“可取消运行循环”，把模型建议与真正执行隔离。
人工走通成功、拒绝和恢复三条路径，保存 run_id 与关键事件。
加入自动化回归，再用小比例非关键流量观察指标。
达到门槛后逐步放量；任一安全约束失败立即停止新任务并回退版本。

失败演练与验收

首要反例是：创建 context.Background 导致上游取消失效。不要只验证顺利完成；还要注入超时、重复请求、取消、无权限输入和依赖返回畸形数据。

本篇的硬性验收是：压测中随机取消，任务在截止时间内退出且无泄漏。同时记录 取消延迟、活跃 goroutine、超时率。验收脚本应按结构化字段判定，不以“回答看起来合理”代替测试。

上线与回滚

“可取消运行循环”上线前要固定模型、提示、规则、工具 schema 与数据版本，并用影子或灰度流量比较旧版。若 取消延迟、活跃 goroutine、超时率 越过门槛，先停止新 Run，再处理在途任务，恢复旧组合后逐项核对未决副作用。

适用边界

当场景是“无外部调用的短计算”时，这套设计的维护成本通常高于收益。权限、金额、删除和一致性约束始终交给确定性代码，不能交给模型自行判断。

取消必须从入口传播到最深工具

层级	接收的边界	退出动作
HTTP/队列入口	父 `context.Context`	不再接新步骤
Run	总 deadline、预算	标记终止原因
Step	子 deadline	取消模型/工具请求
外部进程	context + process group	回收整个进程树

所有可能阻塞的函数把 ctx 作为首参数，不在内部用 context.Background() 切断链路。循环在规划前、排队前和重试退避时检查取消；向 channel 发送也用 select，避免接收者退出后永久阻塞。

取消不是失败重试信号。错误使用 errors.Is(err, context.Canceled/DeadlineExceeded) 分类，终态记录触发层级与剩余预算。启动的 goroutine 由 errgroup、WaitGroup 或明确 owner 回收，后台缓存刷新等独立生命周期也由服务级 context 管理。

压测基线 goroutine 数后随机取消数千个 Run，等待宽限期再比较；同时开启 race detector、检查队列和连接池许可归还。优雅停机先停止入口、取消超出窗口的 Run、等待 worker，最后关闭共享客户端，顺序反过来会制造大量伪错误。

第二轮实践复核

围绕“可取消运行循环”建立一个不接触生产数据的最小项目，固定运行时、框架、模型、提示、规则和工具 schema 版本。正文代码块若缺少完整依赖、导入或领域类型，应视为架构伪代码；只有放入示例仓库并通过下述命令的代码才标记为可运行。

项目验证：go test -race ./...
回归夹具：把输入、结构化输出、工具轨迹和副作用记录为 JSON，用固定版本执行正常、拒绝、超时、取消与重复请求五组回归样本。
必留证据：run_id、版本组合、终止原因、动作幂等键，以及 取消延迟、活跃 goroutine、超时率。
通过条件：未批准写操作为零副作用；相同幂等键重复执行只产生一个业务结果；超时或恢复后状态可解释。
失败处理：先停新任务，再核对在途动作；回滚模型、提示、规则、工具与状态迁移的完整组合，不能只换回模型。

复用顺序

为“用 Go 编写可取消的 Agent Runtime”建立确定性假工具，先跑契约与失败用例，不接触真实凭据。
围绕“可取消运行循环”接入只读能力，核对权限裁剪、脱敏字段和 Trace 完整性。
需要写能力时先输出 dry-run 差异，再绑定审批结果与一次性动作哈希。
以本文验收指标比较影子流量；达到门槛才放量，越线立即执行文中的组合回滚。

实践与资料复核日期：2026-08-22；框架 API 以锁定版本的官方文档为准。

参考资料

官方文档

框架 API 会持续演进，复制示例前应再次确认当前版本。可持续的 AI 工程不依赖某个模型偶尔表现出色，而依赖清晰契约、可靠 Harness 与可重复验证。