mirror of
https://gitcode.com/JianFeeeee/homeagent-sdk.git
synced 2026-10-31 14:27:30 +00:00
fix(commondev): 投递任务改成真异步,不再同步等 pi 跑完
## 缺陷(用户质疑"投递不就是发一条消息"后实测确认)
`deliverToPi` 的注释写着"把给 pi 发消息包装一下",但实际是:
await s.prompt(text) // 同步等 pi 把**整个任务**跑完
而 pi/zcode 开工程线干活(克隆/改代码/跑测试)要几十秒到 20 分钟。
生产日志实测两次:
23:04:03 executing tool: output_send__commondev_br-...
23:05:03 ★ timed out after 60s
通道的语义是"**投递**任务"(见 outputch 描述),不是"等完成"。
把它实现成同步等待,就必然撞上内核的 60s 工具超时。
★ 我一开始的诊断是错的:以为"超时值太小",给 output_send 声明了 3m
—— 那是掩盖症状。投递本该瞬时返回,慢是 bug 不是需求。已撤回。
## 改动
投递 → 立即回 status=accepted(瞬时)
→ pi 在后台跑
→ 跑完推**主动通知**(id=0)
→ Go 侧 onNotify 经 inputch 注入回 agent
`sidecar.go` 的 readLoop 里早就留好了这个扩展位:
if resp.ID == 0 { // 主动通知(暂未使用,保留扩展位)
现在接上了 `makePromptNotifier`。
pi 与 zcode 两份 sidecar.mjs 都改了(rawfile/ 里那两份同步更新)。
## 判据抓到我三次(都是真问题)
1. `TestSidecarScriptShipsWithPackage`:rawfile/ 与工程根不一致
—— 我只改了根那份
2. `TestZcodeMissingCLIErrorIsActionable`:`inflight is not defined`
—— zcode 版用了未声明变量
3. 同一条判据随后**卡死 2 分钟**:CLI 不存在本该立刻报,异步后错误
被吞进后台通知 ⇒ 用户先收到"已受理"、几秒后才知道配置错。
改成**受理阶段就校验 CLI**(配置错是立刻可判定的,不该推迟)。
判据还逼我保住语义:原 `TestZcodeSidecarArgvBehavior` 断言首轮拿到
真实 `sess_` id,异步后要到完成通知里才有 —— 断言保留、改从通知取,
没有削弱。
## 真机 e2e(三条全过,都要求真改工程文件)
TestZcodeFullPipeline 42.1s
TestZcodeOverSSH 72.4s
TestZcodeOverSSHWithPerHostEndpoints 75.1s
受理现在是瞬时的 ⇒ 内核工具不会再 60s 超时。
This commit is contained in:
@ -64,10 +64,21 @@ type sidecarProcess struct {
|
||||
mu sync.Mutex
|
||||
nextID int
|
||||
pending map[int]chan []byte
|
||||
closed bool
|
||||
stopped bool
|
||||
exited chan struct{}
|
||||
waitErr error
|
||||
// onNotify 处理 sidecar **主动推送**的帧(resp.ID == 0)。
|
||||
//
|
||||
// ★ 为什么需要它(2026-10-17 实测缺陷):
|
||||
// deliverToPi 原来 `await s.prompt(text)` **同步等 pi 把任务跑完**,
|
||||
// 而 pi 开工程线干活要几十秒到几分钟 ⇒ 内核 60s 工具超时先到,
|
||||
// 报「执行超时(60秒)」。
|
||||
// 但通道语义是"**投递**任务"(见 outputch 的描述),不是"等完成"。
|
||||
// ⇒ 改成:prompt 立即返回"已受理",pi 跑完后用主动通知回投结果。
|
||||
//
|
||||
// 回调在 readLoop 里调用(已持锁外),实现方不得再进 readLoop。
|
||||
onNotify func(branchID string, payload map[string]interface{})
|
||||
closed bool
|
||||
stopped bool
|
||||
exited chan struct{}
|
||||
waitErr error
|
||||
|
||||
timeout time.Duration
|
||||
|
||||
@ -114,7 +125,21 @@ func (s *sidecarProcess) readLoop(r io.Reader) {
|
||||
continue
|
||||
}
|
||||
if resp.ID == 0 {
|
||||
// 主动通知(暂未使用,保留扩展位)
|
||||
// 主动通知:sidecar 不等请求就推过来的帧。
|
||||
//
|
||||
// 当前用途:异步投递的**完成回投**(见 deliverToPi)。
|
||||
// 在独立 goroutine 里跑回调 —— 回调可能调 InjectText,那会
|
||||
// 走到内核(可能阻塞);在 readLoop 里同步跑会堵住 stdout,
|
||||
// 而 stdout 一堵,pi 就写不进来(RPC 类协议最常见的死法)。
|
||||
if s.onNotify != nil {
|
||||
var payload map[string]interface{}
|
||||
if resp.Result != nil && len(*resp.Result) > 0 {
|
||||
_ = json.Unmarshal(*resp.Result, &payload)
|
||||
}
|
||||
branch := s.branchID
|
||||
fn := s.onNotify
|
||||
go fn(branch, payload)
|
||||
}
|
||||
continue
|
||||
}
|
||||
s.mu.Lock()
|
||||
@ -413,6 +438,12 @@ func (p *Plugin) startSidecar(b *Branch, sessionID string) (*sidecarProcess, err
|
||||
|
||||
sp := newSidecarProcess(b.ID, b.Workdir, cmd, bufio.NewWriter(stdin), stdout, timeout)
|
||||
|
||||
// ★ 接上"完成回投":pi 跑完任务后主动推通知,我们把它注入回 agent。
|
||||
//
|
||||
// 这是**异步投递**的另一半(见 deliverToPi 与 sidecar.mjs 的 prompt)。
|
||||
// 没有它,任务虽然完成了但模型永远不知道结果 —— 比同步超时更糟。
|
||||
sp.onNotify = p.makePromptNotifier()
|
||||
|
||||
// ★ 握手必须短且确定 —— 绝不能用默认 timeout(那是 60s,正好撞上内核的
|
||||
// 工具调用 60s 硬上限 ⇒ 工具超时但进程已起,状态"做了一半")。
|
||||
sp.handshakeTimeout = 10 * time.Second
|
||||
@ -588,3 +619,53 @@ func (p *Plugin) locateSidecarScript(drv driverSpec, cliDir, cli string) (string
|
||||
"若要自带 CLI,请把 driver.json 与脚本放到 <插件数据>/runtime/%s/",
|
||||
cli, drv.Script, filepath.Join(cliDir, drv.Script), cli, drv.Script, drv.Script, cli)
|
||||
}
|
||||
|
||||
// makePromptNotifier 构造"任务完成回投"的回调。
|
||||
//
|
||||
// ## 它解决什么问题
|
||||
//
|
||||
// 异步投递后,pi 在后台跑完任务,结果必须回到 agent 的上下文里 ——
|
||||
// 否则模型只知道"投递成功了",永远不知道 pi 干了什么、成功还是失败。
|
||||
//
|
||||
// ## 为什么走 inputch(而不是 return)
|
||||
//
|
||||
// 投递那一刻 tool 调用早已返回(这是异步的意义)。完成事件到时,
|
||||
// 已经没有"谁在等这个返回值"了。唯一能把信息送进 agent 的通道是
|
||||
// **注入输入**(InjectTextTo),而 commondev 已经为每条工程线注册了
|
||||
// inputch(`commondev/<id>`,NoMemory=true —— 控制流不进记忆)。
|
||||
//
|
||||
// ## 失败不静默
|
||||
//
|
||||
// 注入失败要记日志:否则表现是"任务跑完了、模型毫无反应",
|
||||
// 排查时看不出任何线索。
|
||||
func (p *Plugin) makePromptNotifier() func(branchID string, payload map[string]interface{}) {
|
||||
return func(branchID string, payload map[string]interface{}) {
|
||||
if payload == nil {
|
||||
return
|
||||
}
|
||||
event, _ := payload["event"].(string)
|
||||
sess, _ := payload["session_id"].(string)
|
||||
|
||||
var text string
|
||||
switch event {
|
||||
case "prompt_done":
|
||||
reply, _ := payload["reply"].(string)
|
||||
if strings.TrimSpace(reply) == "" {
|
||||
reply = "(pi 未返回文本回复)"
|
||||
}
|
||||
text = fmt.Sprintf("[工程线 %s 任务完成 · session=%s]\n%s", branchID, sess, reply)
|
||||
case "prompt_error":
|
||||
e, _ := payload["error"].(string)
|
||||
text = fmt.Sprintf("[工程线 %s 任务失败 · session=%s]\n错误:%s", branchID, sess, e)
|
||||
default:
|
||||
log.Printf("[commondev] 未知的通知事件 %q(branch=%s),忽略", event, branchID)
|
||||
return
|
||||
}
|
||||
|
||||
// source 用插件名(内核据此识别来源),channel 是该线的 inputch。
|
||||
inCh := inputChannelName(branchID)
|
||||
log.Printf("[commondev] 回投任务结果到 %s(event=%s)", inCh, event)
|
||||
// InjectText 无返回值(内部已做错误日志),调用即注入。
|
||||
p.sdk.InjectText("commondev", inCh, text)
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user