mirror of
https://gitcode.com/JianFeeeee/homeagent-sdk.git
synced 2026-10-31 06:17:25 +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:
@ -21,6 +21,7 @@ import (
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
func zcodeE2EReady(t *testing.T) (nodeBin, cliPath, dataDir string) {
|
||||
@ -144,11 +145,21 @@ func TestZcodeFullPipeline(t *testing.T) {
|
||||
if err := json.Unmarshal(resp.Result, &pr); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
sid, _ := pr["session_id"].(string)
|
||||
if !strings.HasPrefix(sid, "sess_") {
|
||||
t.Fatalf("★ 未回报真实 ZCode 会话 id: %q", sid)
|
||||
// ★ 异步投递(2026-10-17):prompt 立即回 "accepted",此时还没有真实
|
||||
// sess_ id —— 它要等 zcode 真正建好会话。真实 id 与任务结果都随
|
||||
// **完成通知**(id=0)推回。所以这里必须等通知,而不是看应答。
|
||||
if st, _ := pr["status"].(string); st != "accepted" {
|
||||
t.Fatalf("投递应回 status=accepted(异步),实际 %q", st)
|
||||
}
|
||||
t.Logf("zcode 会话: %s", sid)
|
||||
notif := waitPromptDone(t, rd)
|
||||
sid, _ := notif["session_id"].(string)
|
||||
if !strings.HasPrefix(sid, "sess_") {
|
||||
t.Fatalf("★ 完成通知未含真实 ZCode 会话 id: %q(通知=%v)", sid, notif)
|
||||
}
|
||||
if reply, _ := notif["reply"].(string); strings.TrimSpace(reply) == "" {
|
||||
t.Logf("提示:完成通知里 reply 为空(zcode 可能只回了工具动作)")
|
||||
}
|
||||
t.Logf("zcode 会话: %s(任务已完成)", sid)
|
||||
|
||||
// ★ 关键断言:真的改了**这个工程**的文件,且没破坏原代码
|
||||
b2, err := os.ReadFile(filepath.Join(workdir, "src", "calc.py"))
|
||||
@ -300,6 +311,12 @@ func TestZcodeOverSSH(t *testing.T) {
|
||||
if resp.Error != nil {
|
||||
t.Fatalf("远端投递失败: %s", resp.Error.Message)
|
||||
}
|
||||
// ★ 异步投递:必须等完成通知,否则读文件时远端任务还没开始跑
|
||||
// (实测:不等的话 calc.py 根本还没被创建)。
|
||||
notif := waitPromptDone(t, rd)
|
||||
if sid, _ := notif["session_id"].(string); !strings.HasPrefix(sid, "sess_") {
|
||||
t.Fatalf("远端完成通知未含真实会话 id: %q", sid)
|
||||
}
|
||||
b2, err := os.ReadFile(filepath.Join(remoteWork, "src", "calc.py"))
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
@ -493,12 +510,19 @@ func TestZcodeOverSSHWithPerHostEndpoints(t *testing.T) {
|
||||
"session_id": "br-" + tc.proj,
|
||||
"text": "在 src/calc.py 里新增 multiply(a, b) 返回 a*b,保留 add。只回一句完成。",
|
||||
})
|
||||
_ = rc.Process.Kill()
|
||||
_, _ = rc.Process.Wait()
|
||||
if resp.Error != nil {
|
||||
_ = rc.Process.Kill()
|
||||
_, _ = rc.Process.Wait()
|
||||
t.Fatalf("★ %s 投递失败(端点表里的 CLI 路径可能在该机器上不存在): %s",
|
||||
tc.host, resp.Error.Message)
|
||||
}
|
||||
// ★ 异步投递:等完成通知再读文件(并把会话 id 一并验掉)。
|
||||
notif := waitPromptDone(t, rrd)
|
||||
_ = rc.Process.Kill()
|
||||
_, _ = rc.Process.Wait()
|
||||
if sid, _ := notif["session_id"].(string); !strings.HasPrefix(sid, "sess_") {
|
||||
t.Fatalf("★ %s 完成通知未含真实会话 id: %q", tc.host, sid)
|
||||
}
|
||||
b2, err := os.ReadFile(filepath.Join(tc.wd, "src/calc.py"))
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
@ -512,3 +536,53 @@ func TestZcodeOverSSHWithPerHostEndpoints(t *testing.T) {
|
||||
_ = env
|
||||
}
|
||||
}
|
||||
|
||||
// waitPromptDone 等 sidecar 推来的「任务完成」通知(id=0)。
|
||||
//
|
||||
// 异步投递后,任务是否真的完成、结果是什么,只有这条通知说得清。
|
||||
// 超时给足 20 分钟:zcode 的默认单轮上限就是这个量级(见 sidecar.mjs)。
|
||||
func waitPromptDone(t *testing.T, rd *bufio.Reader) map[string]interface{} {
|
||||
t.Helper()
|
||||
type line struct{ s string }
|
||||
ch := make(chan line, 1)
|
||||
go func() {
|
||||
for {
|
||||
s, err := rd.ReadString('\n')
|
||||
if err != nil {
|
||||
ch <- line{""}
|
||||
return
|
||||
}
|
||||
ch <- line{s}
|
||||
}
|
||||
}()
|
||||
deadline := time.After(20 * time.Minute)
|
||||
for {
|
||||
select {
|
||||
case l := <-ch:
|
||||
if l.s == "" {
|
||||
t.Fatal("sidecar 输出结束,未收到完成通知")
|
||||
}
|
||||
var m map[string]interface{}
|
||||
if json.Unmarshal([]byte(l.s), &m) != nil {
|
||||
continue
|
||||
}
|
||||
if id, _ := m["id"].(float64); id != 0 {
|
||||
continue // 不是通知
|
||||
}
|
||||
res, _ := m["result"].(map[string]interface{})
|
||||
if res == nil {
|
||||
continue
|
||||
}
|
||||
ev, _ := res["event"].(string)
|
||||
if ev == "prompt_error" {
|
||||
e, _ := res["error"].(string)
|
||||
t.Fatalf("★ 任务失败(通知):%s", e)
|
||||
}
|
||||
if ev == "prompt_done" {
|
||||
return res
|
||||
}
|
||||
case <-deadline:
|
||||
t.Fatal("等任务完成通知超时(20 分钟)")
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user