@zhushanwen/pi-subagent-workflow 8.3.0 → 8.5.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (103) hide show
  1. package/package.json +18 -4
  2. package/relay/relay.mjs +390 -0
  3. package/skills/subagent-ext-config/SKILL.md +80 -0
  4. package/src/execution/__tests__/agent-registry.test.ts +110 -0
  5. package/src/execution/__tests__/chat-engine-routing.test.ts +597 -0
  6. package/src/execution/__tests__/execution-record.test.ts +127 -1
  7. package/src/execution/__tests__/pi-invocation.test.ts +62 -1
  8. package/src/execution/__tests__/relay-agent.test.ts +448 -0
  9. package/src/execution/__tests__/relay-env.test.ts +42 -0
  10. package/src/execution/__tests__/startup-config-declaration.test.ts +35 -0
  11. package/src/execution/__tests__/stream-sink-retirement.test.ts +261 -0
  12. package/src/execution/__tests__/subprocess-agent-runner-routing.test.ts +310 -0
  13. package/src/execution/__tests__/subprocess-agent-runner.test.ts +53 -5
  14. package/src/execution/agent-registry.ts +10 -0
  15. package/src/execution/config.ts +25 -2
  16. package/src/execution/engine/__tests__/common/data-dir.test.ts +53 -0
  17. package/src/execution/engine/__tests__/common/errors.test.ts +132 -0
  18. package/src/execution/engine/__tests__/common/event-journal.test.ts +177 -0
  19. package/src/execution/engine/__tests__/common/kill-chain.test.ts +192 -0
  20. package/src/execution/engine/__tests__/common/nesting-guard.test.ts +81 -0
  21. package/src/execution/engine/__tests__/common/persona-router.test.ts +123 -0
  22. package/src/execution/engine/__tests__/common/pool-manager.test.ts +154 -0
  23. package/src/execution/engine/__tests__/common/schema-emulation.test.ts +128 -0
  24. package/src/execution/engine/__tests__/conformance/__fixtures__/pi-golden-events.json +28 -0
  25. package/src/execution/engine/__tests__/conformance/agent-event-invariants.ts +141 -0
  26. package/src/execution/engine/__tests__/conformance/contract.abort.test.ts +109 -0
  27. package/src/execution/engine/__tests__/conformance/contract.agent-events.test.ts +101 -0
  28. package/src/execution/engine/__tests__/conformance/contract.probe.test.ts +77 -0
  29. package/src/execution/engine/__tests__/conformance/contract.read-degradation.test.ts +104 -0
  30. package/src/execution/engine/__tests__/conformance/contract.relay.test.ts +342 -0
  31. package/src/execution/engine/__tests__/conformance/engine-conformance.live.test.ts +201 -0
  32. package/src/execution/engine/__tests__/conformance/golden-replay.pi.test.ts +76 -0
  33. package/src/execution/engine/__tests__/conformance/golden-replay.zcode.test.ts +79 -0
  34. package/src/execution/engine/__tests__/engine-discovery.test.ts +87 -0
  35. package/src/execution/engine/__tests__/engines-declaration.test.ts +36 -0
  36. package/src/execution/engine/__tests__/model-prompt.test.ts +85 -0
  37. package/src/execution/engine/__tests__/paths.test.ts +39 -0
  38. package/src/execution/engine/__tests__/registry.test.ts +120 -0
  39. package/src/execution/engine/__tests__/routing.test.ts +231 -0
  40. package/src/execution/engine/common/data-dir.ts +62 -0
  41. package/src/execution/engine/common/errors.ts +183 -0
  42. package/src/execution/engine/common/event-journal.ts +254 -0
  43. package/src/execution/engine/common/journal-replay.ts +62 -0
  44. package/src/execution/engine/common/kill-chain.ts +221 -0
  45. package/src/execution/engine/common/nesting-guard.ts +50 -0
  46. package/src/execution/engine/common/persona-router.ts +108 -0
  47. package/src/execution/engine/common/pool-manager.ts +226 -0
  48. package/src/execution/engine/common/schema-emulation.ts +189 -0
  49. package/src/execution/engine/common/session-view-projection.ts +51 -0
  50. package/src/execution/engine/engine-discovery.ts +65 -0
  51. package/src/execution/engine/engines/pi/__tests__/pi-engine.test.ts +469 -0
  52. package/src/execution/engine/engines/pi/__tests__/reader.test.ts +155 -0
  53. package/src/execution/engine/engines/pi/__tests__/task-spec-mapper.test.ts +164 -0
  54. package/src/execution/engine/engines/pi/pi-engine.ts +415 -0
  55. package/src/execution/engine/engines/pi/reader.ts +48 -0
  56. package/src/execution/engine/engines/pi/registration.ts +35 -0
  57. package/src/execution/engine/engines/pi/task-spec-mapper.ts +100 -0
  58. package/src/execution/engine/engines/zcode/__tests__/__fixtures__/zcode-golden-spawn.json +39 -0
  59. package/src/execution/engine/engines/zcode/__tests__/launcher.test.ts +150 -0
  60. package/src/execution/engine/engines/zcode/__tests__/parser.test.ts +246 -0
  61. package/src/execution/engine/engines/zcode/__tests__/preparer.test.ts +228 -0
  62. package/src/execution/engine/engines/zcode/__tests__/reader.test.ts +210 -0
  63. package/src/execution/engine/engines/zcode/__tests__/registration.test.ts +64 -0
  64. package/src/execution/engine/engines/zcode/__tests__/zcode-engine.live.test.ts +127 -0
  65. package/src/execution/engine/engines/zcode/__tests__/zcode-engine.test.ts +567 -0
  66. package/src/execution/engine/engines/zcode/constants.ts +43 -0
  67. package/src/execution/engine/engines/zcode/golden-sample.ts +39 -0
  68. package/src/execution/engine/engines/zcode/launcher.ts +161 -0
  69. package/src/execution/engine/engines/zcode/parser.ts +436 -0
  70. package/src/execution/engine/engines/zcode/preparer.ts +363 -0
  71. package/src/execution/engine/engines/zcode/reader.ts +381 -0
  72. package/src/execution/engine/engines/zcode/registration.ts +37 -0
  73. package/src/execution/engine/engines/zcode/zcode-engine.ts +648 -0
  74. package/src/execution/engine/host-task-spec.ts +47 -0
  75. package/src/execution/engine/model-prompt.ts +59 -0
  76. package/src/execution/engine/paths.ts +42 -0
  77. package/src/execution/engine/port.ts +153 -0
  78. package/src/execution/engine/registry.ts +123 -0
  79. package/src/execution/engine/routing.ts +218 -0
  80. package/src/execution/engine/types.ts +304 -0
  81. package/src/execution/execute-options-mapper.ts +5 -1
  82. package/src/execution/execution-record.ts +6 -0
  83. package/src/execution/model-resolver.ts +6 -0
  84. package/src/execution/pi-invocation.ts +32 -2
  85. package/src/execution/record-entry.ts +14 -0
  86. package/src/execution/record-store.ts +34 -0
  87. package/src/execution/relay-env.ts +37 -0
  88. package/src/execution/session-runner.ts +24 -0
  89. package/src/execution/stream-sink.ts +26 -0
  90. package/src/execution/subagent-service.ts +249 -11
  91. package/src/execution/subprocess-agent-runner.ts +196 -14
  92. package/src/execution/types.ts +56 -0
  93. package/src/index.ts +46 -1
  94. package/src/interface/command-actions.ts +72 -8
  95. package/src/interface/subagent-actions.ts +8 -2
  96. package/src/interface/subagent-tool.ts +6 -0
  97. package/src/interface/subagents.ts +198 -30
  98. package/src/orchestration/__tests__/__fixtures__/worker-template.snapshot.txt +5 -2
  99. package/src/orchestration/__tests__/worker-script-template-snapshot.test.ts +3 -3
  100. package/src/orchestration/models/types.ts +7 -0
  101. package/src/orchestration/worker-script-builder.ts +5 -2
  102. package/src/shared/meta-parser.ts +5 -1
  103. package/src/shared/resource-meta.ts +5 -0
@@ -0,0 +1,597 @@
1
+ // src/execution/__tests__/chat-engine-routing.test.ts
2
+ //
3
+ // U0 chat 工具域引擎路由分叉测试。设计权威源:
4
+ // docs/architecture/subagent-engine-gui-visibility.md §3.3 D4(chat 入口路由分叉)/
5
+ // D5(pi 缺省字节级零变化)/ D10(zcode 分支终止链)。
6
+ //
7
+ // 覆盖:
8
+ // 1. 三层路由优先级(opts.engine > agent frontmatter engine > config defaultEngine;
9
+ // 全缺省 → pi 原路径)
10
+ // 2. pi 缺省守护(record.engine === undefined;entry 序列化产物不含 "engine" 键)
11
+ // 3. unsupported 参数预检(conversation/fork/worktree 同步拒绝,不产生 record)
12
+ // 4. 未注册 engine id → engine_not_found
13
+ // 5. 引擎分支骨架(record 创建+盖章 / taskSpec 字段 / detached run / done+failed
14
+ // 终态迁移 / spawnedChildren 注册 / abort signal 触达引擎 kill-chain)
15
+ // 6. U2:probe 兜底两态(默认路由兜底回 pi / 显式 engine 守卫报错)+ JournalWriter
16
+ // 接线(taskId=record.id + onPoolResolved retarget)+ engineHandle 完整回填
17
+ //
18
+ // mock 策略:只 mock node:child_process.spawn(pi 原路径的 FakeChild,见
19
+ // execute-nesting.test.ts 同款范式)——非 pi 引擎分支用假 EnginePort(registerEngine
20
+ // 注入),不 spawn 任何进程;fs 用真实 os.tmpdir()(engine 分支的 manifest 落盘无妨)。
21
+
22
+ import type { ChildProcess } from "node:child_process";
23
+ import { EventEmitter } from "node:events";
24
+ import * as fs from "node:fs";
25
+ import * as os from "node:os";
26
+ import * as path from "node:path";
27
+
28
+ import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
29
+
30
+ // pi 原路径 spawn mock(FakeChild 挂起不推进——本文件只断言路由归属与 record 形态,
31
+ // 不驱动 pi session 事件流)。
32
+ vi.mock("node:child_process", async () => {
33
+ const { EventEmitter: EE } = await import("node:events");
34
+ const { PassThrough: PT } = await import("node:stream");
35
+ class FakeChild extends EE {
36
+ pid = 12345;
37
+ stdout = new PT();
38
+ stderr = new PT();
39
+ stdin = new PT();
40
+ killed = false;
41
+ kill(sig?: string): boolean {
42
+ this.killed = true;
43
+ return sig !== undefined;
44
+ }
45
+ }
46
+ return {
47
+ spawn: vi.fn(() => new FakeChild()),
48
+ execFile: vi.fn(
49
+ (
50
+ _cmd: string,
51
+ _args: readonly string[],
52
+ _opts: unknown,
53
+ cb: (err: Error | null, stdout?: string, stderr?: string) => void,
54
+ ) => cb(new Error("execFile not configured in this test")),
55
+ ),
56
+ };
57
+ });
58
+
59
+ import { spawn } from "node:child_process";
60
+
61
+ import type { EnginePort, RunContext } from "../engine/port.ts";
62
+ import { resolveJournalPath } from "../engine/paths.ts";
63
+ import { clearEngines, registerEngine } from "../engine/registry.ts";
64
+ import type {
65
+ AgentEvent,
66
+ AgentOutcome,
67
+ AgentTaskSpec,
68
+ EngineCapabilities,
69
+ EngineHandle,
70
+ ProbeReport,
71
+ } from "../engine/types.ts";
72
+ import { ModelConfigService } from "../model-config-service.ts";
73
+ import type { ModelInfo, ModelRegistryLike } from "../model-resolver.ts";
74
+ import { toSubagentRecordEntry } from "../record-entry.ts";
75
+ import { getChildByRecord } from "../session-runner.ts";
76
+ import { SubagentService } from "../subagent-service.ts";
77
+ import type { ExecuteOptions } from "../types.ts";
78
+
79
+ const mockSpawn = vi.mocked(spawn);
80
+
81
+ // ============================================================
82
+ // 假引擎(EnginePort 最小实现;run 行为由每个用例注入)
83
+ // ============================================================
84
+
85
+ /** zcode 形态 capabilities(conversation/steer unsupported、sandbox none)。 */
86
+ const ZCODE_LIKE_CAPS: EngineCapabilities = {
87
+ schemaEnforcement: "emulated",
88
+ steer: "unsupported",
89
+ conversation: "unsupported",
90
+ personaInjection: "prompt",
91
+ eventGranularity: "coarse",
92
+ sandbox: "none",
93
+ sessionRead: "full",
94
+ resume: "cold",
95
+ interrupt: "kill-only",
96
+ permissionMode: "native",
97
+ };
98
+
99
+ interface CapturedRun {
100
+ task: AgentTaskSpec;
101
+ ctx: RunContext;
102
+ }
103
+
104
+ class FakeEngine implements EnginePort {
105
+ readonly id: string;
106
+ readonly runs: CapturedRun[] = [];
107
+ /** U2:probe 结果注入(路由期 routeEngine 消费;缺省 ok——probe 通过的常态)。 */
108
+ probeFailed = false;
109
+ /** run 实现注入(缺省:挂起不 resolve——record 保持 running 便于内存断言)。 */
110
+ runImpl: (task: AgentTaskSpec, ctx: RunContext) => Promise<{ handle: EngineHandle; outcome: AgentOutcome }>;
111
+
112
+ constructor(id: string) {
113
+ this.id = id;
114
+ // 缺省挂起:never settle(用例不驱动时 record 停留 running,便于内存态断言)
115
+ this.runImpl = () => new Promise(() => {});
116
+ }
117
+
118
+ capabilities(): EngineCapabilities {
119
+ return ZCODE_LIKE_CAPS;
120
+ }
121
+ async probe(): Promise<ProbeReport> {
122
+ return this.probeFailed
123
+ ? {
124
+ ok: false,
125
+ engineVersion: "",
126
+ checks: [{ name: "bin", ok: false }],
127
+ error: { code: "engine_probe_failed", recovery: "fix the binary" },
128
+ }
129
+ : { ok: true, engineVersion: "fake", checks: [{ name: "bin", ok: true }] };
130
+ }
131
+ async run(task: AgentTaskSpec, ctx: RunContext): Promise<{ handle: EngineHandle; outcome: AgentOutcome }> {
132
+ this.runs.push({ task, ctx });
133
+ return this.runImpl(task, ctx);
134
+ }
135
+ async interact(): Promise<{ ok: false; code: string; message: string }> {
136
+ return { ok: false, code: "engine_capability_unsupported", message: "fake" };
137
+ }
138
+ async read(): Promise<{ engineId: string; turns: never[]; source: "outcome-only" }> {
139
+ return { engineId: this.id, turns: [], source: "outcome-only" };
140
+ }
141
+ }
142
+
143
+ /** 构造最小合法 AgentOutcome(done 形态)。 */
144
+ function doneOutcome(content: string): AgentOutcome {
145
+ return { content, engineId: "zcode", durationMs: 10, usage: { input: 1, output: 1, cacheRead: 0, cacheWrite: 0, cost: 0, contextTokens: 2, turns: 1 } };
146
+ }
147
+
148
+ // ============================================================
149
+ // 环境:tmp agentDir + 假引擎注册 + service 装配
150
+ // ============================================================
151
+
152
+ function makeTmpAgentDir(): string {
153
+ return fs.mkdtempSync(path.join(os.tmpdir(), "chat-engine-routing-"));
154
+ }
155
+
156
+ function writeGlobalConfig(agentDir: string, defaultEngine?: string): void {
157
+ fs.mkdirSync(path.join(agentDir, "subagents"), { recursive: true });
158
+ fs.writeFileSync(
159
+ path.join(agentDir, "subagents", "config.json"),
160
+ JSON.stringify({ version: 1, maxConcurrent: 6, ...(defaultEngine !== undefined ? { defaultEngine } : {}) }),
161
+ );
162
+ }
163
+
164
+ /** 写 agent .md(frontmatter engine 字段 = 第二层路由输入;须在引擎注册后调用——解析期校验注册表)。 */
165
+ function writeAgentMd(dir: string, engine: string): string {
166
+ const file = path.join(dir, `agent-${engine}.md`);
167
+ fs.writeFileSync(file, `---\nname: agent-${engine}\ndescription: test agent\nengine: ${engine}\n---\nbody\n`);
168
+ return file;
169
+ }
170
+
171
+ function makePi() {
172
+ return { sendMessage: vi.fn(), appendEntry: vi.fn(), events: { emit: vi.fn() } };
173
+ }
174
+
175
+ const CTX_MODEL: ModelInfo = { id: "m", name: "M", provider: "p", reasoning: false };
176
+
177
+ /** registry:可解析 "zcode/glm"(taskSpec 字段用例的显式 model),其余未配置。 */
178
+ function fakeRegistry(): ModelRegistryLike {
179
+ return {
180
+ getAvailable: () => [],
181
+ find: (provider: string, id: string) =>
182
+ provider === "zcode" && id === "glm" ? { id: "glm", name: "GLM", provider: "zcode", reasoning: true } : undefined,
183
+ hasConfiguredAuth: () => true,
184
+ };
185
+ }
186
+
187
+ interface SetupResult {
188
+ service: SubagentService;
189
+ zcode: FakeEngine;
190
+ piEngine: FakeEngine;
191
+ pi: ReturnType<typeof makePi>;
192
+ }
193
+
194
+ function setup(agentDir: string): SetupResult {
195
+ const zcode = new FakeEngine("zcode");
196
+ const piEngine = new FakeEngine("pi");
197
+ registerEngine("zcode", () => zcode);
198
+ registerEngine("pi", () => piEngine);
199
+ const modelService = new ModelConfigService({ agentDir, cwd: agentDir });
200
+ modelService.initModel({
201
+ modelRegistry: fakeRegistry(),
202
+ sessionId: "test-session",
203
+ ctxModel: CTX_MODEL,
204
+ });
205
+ const pi = makePi();
206
+ const service = new SubagentService({ cwd: agentDir, modelService });
207
+ service.initSession({ pi, sessionId: "test-session" });
208
+ return { service, zcode, piEngine, pi };
209
+ }
210
+
211
+ function baseOpts(agentDir: string, extra: Partial<ExecuteOptions> = {}): ExecuteOptions {
212
+ return { task: "do work", slug: "routing-test", cwd: agentDir, ctxModel: CTX_MODEL, ...extra };
213
+ }
214
+
215
+ /** 挂起 run 的引擎路径下取内存 running record 的 engine 盖章(collectRecords 内存源投影)。 */
216
+ function runningEngineTag(service: SubagentService, id: string): string | undefined {
217
+ const rec = service.collectRecords(10, "running").find((r) => r.id === id);
218
+ return rec?.engine;
219
+ }
220
+
221
+ describe("chat 工具域引擎路由分叉(U0:D4/D5/D10)", () => {
222
+ let agentDir: string;
223
+
224
+ beforeEach(() => {
225
+ agentDir = makeTmpAgentDir();
226
+ writeGlobalConfig(agentDir);
227
+ });
228
+
229
+ afterEach(() => {
230
+ // registry 是 globalThis 进程单例——必须清空,防假引擎泄漏进其他测试文件
231
+ clearEngines();
232
+ fs.rmSync(agentDir, { recursive: true, force: true });
233
+ vi.clearAllMocks();
234
+ });
235
+
236
+ // ============================================================
237
+ // 1. 三层路由优先级
238
+ // ============================================================
239
+
240
+ it("[层1] opts.engine='zcode' 最优先(覆盖 agent frontmatter 与全局默认)", async () => {
241
+ writeAgentMd(agentDir, "pi");
242
+ writeGlobalConfig(agentDir, "pi");
243
+ const { service, zcode, piEngine } = setup(agentDir);
244
+
245
+ const handle = await service.execute(baseOpts(agentDir, { engine: "zcode", agent: path.join(agentDir, "agent-pi.md") }));
246
+ await vi.waitFor(() => expect(zcode.runs.length).toBe(1));
247
+
248
+ expect(piEngine.runs.length).toBe(0);
249
+ expect(runningEngineTag(service, handle.subagentId)).toBe("zcode");
250
+ });
251
+
252
+ it("[层2] 无调用参数时 agent frontmatter engine 生效", async () => {
253
+ writeAgentMd(agentDir, "zcode");
254
+ const { service, zcode, piEngine } = setup(agentDir);
255
+
256
+ const handle = await service.execute(baseOpts(agentDir, { agent: path.join(agentDir, "agent-zcode.md") }));
257
+ await vi.waitFor(() => expect(zcode.runs.length).toBe(1));
258
+
259
+ expect(piEngine.runs.length).toBe(0);
260
+ expect(runningEngineTag(service, handle.subagentId)).toBe("zcode");
261
+ });
262
+
263
+ it("[层3] 两者皆无时 config.json defaultEngine 生效", async () => {
264
+ writeGlobalConfig(agentDir, "zcode");
265
+ const { service, zcode, piEngine } = setup(agentDir);
266
+
267
+ await service.execute(baseOpts(agentDir));
268
+ await vi.waitFor(() => expect(zcode.runs.length).toBe(1));
269
+ expect(piEngine.runs.length).toBe(0);
270
+ expect(mockSpawn).not.toHaveBeenCalled();
271
+ });
272
+
273
+ it("[缺省] 全缺省 → pi 原路径(spawn 启动,不经引擎注册表)", async () => {
274
+ // 不注册任何引擎也照跑:pi 缺省免探免注册表校验(D4「pi 恒免探」)
275
+ clearEngines();
276
+ const modelService = new ModelConfigService({ agentDir, cwd: agentDir });
277
+ modelService.initModel({
278
+ modelRegistry: { getAvailable: () => [], find: () => undefined, hasConfiguredAuth: () => true } satisfies ModelRegistryLike,
279
+ sessionId: "test-session",
280
+ ctxModel: CTX_MODEL,
281
+ });
282
+ const service = new SubagentService({ cwd: agentDir, modelService });
283
+ service.initSession({ pi: makePi(), sessionId: "test-session" });
284
+
285
+ await service.execute(baseOpts(agentDir));
286
+ await vi.waitFor(() => expect(mockSpawn).toHaveBeenCalled());
287
+ });
288
+
289
+ // ============================================================
290
+ // 2. pi 缺省守护(D5 字节级)
291
+ // ============================================================
292
+
293
+ it("[D5] 全缺省 pi record:engine===undefined 且 entry JSON 不含 engine 键", async () => {
294
+ const { service } = setup(agentDir);
295
+ const handle = await service.execute(baseOpts(agentDir));
296
+ await vi.waitFor(() => expect(mockSpawn).toHaveBeenCalled());
297
+
298
+ const rec = service.collectRecords(10, "running").find((r) => r.id === handle.subagentId);
299
+ expect(rec).toBeDefined();
300
+ expect(rec?.engine).toBeUndefined();
301
+ expect(JSON.stringify(toSubagentRecordEntry(rec!))).not.toContain("engine");
302
+ });
303
+
304
+ it("[D5] 显式 engine:'pi' 路由回 pi:record 不盖章 engine", async () => {
305
+ const { service, piEngine, zcode } = setup(agentDir);
306
+ const handle = await service.execute(baseOpts(agentDir, { engine: "pi" }));
307
+ await vi.waitFor(() => expect(mockSpawn).toHaveBeenCalled());
308
+
309
+ expect(zcode.runs.length).toBe(0);
310
+ // piEngine 只是注册表占位——pi 缺省路径不查注册表(上一用例已验证零注册场景),
311
+ // 此处断言 piEngine 也未被消费(pi 原路径不走 EnginePort)
312
+ expect(piEngine.runs.length).toBe(0);
313
+ const rec = service.collectRecords(10, "running").find((r) => r.id === handle.subagentId);
314
+ expect(rec?.engine).toBeUndefined();
315
+ });
316
+
317
+ // ============================================================
318
+ // 3. unsupported 参数预检(record 创建前同步拒绝)
319
+ // ============================================================
320
+
321
+ it("[预检] engine='zcode' + conversation:true → throw 且不产生 record", async () => {
322
+ const { service, zcode } = setup(agentDir);
323
+ const err = await service
324
+ .execute(baseOpts(agentDir, { engine: "zcode", conversation: true }))
325
+ .catch((e: unknown) => e);
326
+ expect(err).toBeInstanceOf(Error);
327
+ expect((err as Error).message).toContain("不支持 conversation");
328
+ // 恢复指引(EngineError.recovery)含「改用 engine: pi 或不传该参数」
329
+ expect((err as { recovery?: string }).recovery).toContain("engine: pi");
330
+ expect((err as { recovery?: string }).recovery).toContain("不传该参数");
331
+ expect(zcode.runs.length).toBe(0);
332
+ expect(service.collectRecords(10, "all")).toHaveLength(0);
333
+ });
334
+
335
+ it("[预检] fork:true / worktree:true 同步拒绝", async () => {
336
+ const { service, zcode } = setup(agentDir);
337
+ await expect(service.execute(baseOpts(agentDir, { engine: "zcode", fork: true }))).rejects.toThrow(
338
+ /engine_capability_unsupported/,
339
+ );
340
+ await expect(service.execute(baseOpts(agentDir, { engine: "zcode", worktree: true }))).rejects.toThrow(
341
+ /engine_capability_unsupported/,
342
+ );
343
+ expect(zcode.runs.length).toBe(0);
344
+ expect(service.collectRecords(10, "all")).toHaveLength(0);
345
+ });
346
+
347
+ // ============================================================
348
+ // 4. 未注册 engine id
349
+ // ============================================================
350
+
351
+ it("[未注册] engine='claude' → engine_not_found(含注册清单),不产生 record", async () => {
352
+ const { service } = setup(agentDir);
353
+ const err = await service.execute(baseOpts(agentDir, { engine: "claude" })).catch((e: unknown) => e);
354
+ expect(err).toBeInstanceOf(Error);
355
+ expect((err as Error).message).toContain("engine_not_found");
356
+ expect((err as Error).message).toContain("zcode");
357
+ expect(service.collectRecords(10, "all")).toHaveLength(0);
358
+ });
359
+
360
+ // ============================================================
361
+ // 5. 引擎分支骨架(D10 终止链 + 终态迁移)
362
+ // ============================================================
363
+
364
+ it("[骨架] taskSpec 字段正确(task/cwd/model/effort/schema/persona)", async () => {
365
+ const { service, zcode } = setup(agentDir);
366
+ await service.execute(
367
+ baseOpts(agentDir, {
368
+ engine: "zcode",
369
+ model: "zcode/glm",
370
+ thinkingLevel: "high",
371
+ schema: { type: "object" },
372
+ skillPath: "/tmp/skill.md",
373
+ appendSystemPrompt: ["extra"],
374
+ }),
375
+ );
376
+ await vi.waitFor(() => expect(zcode.runs.length).toBe(1));
377
+ const { task, ctx } = zcode.runs[0];
378
+ expect(task.task).toBe("do work");
379
+ expect(task.slug).toBe("routing-test");
380
+ expect(task.model).toBe("zcode/glm");
381
+ expect(task.effort).toBe("high");
382
+ expect(task.schema).toEqual({ type: "object" });
383
+ expect(task.persona).toEqual({ skillPath: "/tmp/skill.md", appendSystemPrompt: ["extra"] });
384
+ expect(ctx.poolKey).toBe("shared");
385
+ });
386
+
387
+ it("[骨架] run resolve 成功 → record 终态 done + result=content", async () => {
388
+ const { service, zcode, pi } = setup(agentDir);
389
+ zcode.runImpl = () => Promise.resolve({ handle: fakeHandle(), outcome: doneOutcome("hello result") });
390
+ const handle = await service.execute(baseOpts(agentDir, { engine: "zcode" }));
391
+
392
+ // record 终态化即 archive(无 sessionFile 不可磁盘重建)→ 内存查询落空
393
+ await vi.waitFor(() => expect(service.findRecord(handle.subagentId)).toBeUndefined());
394
+ // 完成通知(chat 域宿主职责):notifier 立即 flush(无其他 running)→ result 进 sendMessage 正文
395
+ await vi.waitFor(() => {
396
+ const sent = pi.sendMessage.mock.calls.some((c) => String(c[0]?.content).includes("hello result"));
397
+ expect(sent).toBe(true);
398
+ });
399
+ });
400
+
401
+ it("[骨架] outcome.error → record 终态 failed", async () => {
402
+ const { service, zcode, pi } = setup(agentDir);
403
+ zcode.runImpl = () =>
404
+ Promise.resolve({ handle: fakeHandle(), outcome: { ...doneOutcome(""), error: "engine_run_failed: boom", engineId: "zcode" } });
405
+ const handle = await service.execute(baseOpts(agentDir, { engine: "zcode" }));
406
+
407
+ await vi.waitFor(() => expect(service.findRecord(handle.subagentId)).toBeUndefined());
408
+ await vi.waitFor(() => {
409
+ const sent = pi.sendMessage.mock.calls.some((c) => String(c[0]?.content).includes("boom"));
410
+ expect(sent).toBe(true);
411
+ });
412
+ });
413
+
414
+ it("[D10] onChildSpawned 注册进 spawnedChildren + cancel abort 后 signal 触达引擎", async () => {
415
+ const { service, zcode } = setup(agentDir);
416
+ class FakeProc extends EventEmitter {
417
+ pid = 4321;
418
+ killed = false;
419
+ kill(sig?: string): boolean {
420
+ this.killed = true;
421
+ return sig !== undefined;
422
+ }
423
+ }
424
+ const child = new FakeProc() as unknown as ChildProcess;
425
+ const handle = await service.execute(baseOpts(agentDir, { engine: "zcode" }));
426
+ await vi.waitFor(() => expect(zcode.runs.length).toBe(1));
427
+
428
+ // 引擎经 RunContext.onChildSpawned 上报子进程 → 宿主记账(kill-chain 数据源)
429
+ zcode.runs[0].ctx.onChildSpawned?.(child);
430
+ expect(getChildByRecord(handle.subagentId)).toBe(child);
431
+
432
+ // cancel → controller.abort → engine 收到的 signal aborted(kill-chain 两级的第一级)
433
+ expect(zcode.runs[0].ctx.signal?.aborted).toBe(false);
434
+ service.cancel(handle.subagentId);
435
+ expect(zcode.runs[0].ctx.signal?.aborted).toBe(true);
436
+ // 子进程退出后记账按句移除
437
+ child.emit("close", 0, null);
438
+ expect(getChildByRecord(handle.subagentId)).toBeUndefined();
439
+ });
440
+ });
441
+
442
+ // ============================================================
443
+ // U2:probe/守卫兜底 + JournalWriter + engineHandle 回填
444
+ // 设计权威源:docs/architecture/subagent-engine-gui-visibility.md §3.3 D4/D6、§5 U2 行
445
+ // ============================================================
446
+
447
+ describe("chat 引擎分支 U2:probe 兜底 / journal / engineHandle", () => {
448
+ let agentDir: string;
449
+
450
+ beforeEach(() => {
451
+ agentDir = makeTmpAgentDir();
452
+ writeGlobalConfig(agentDir);
453
+ });
454
+
455
+ afterEach(() => {
456
+ clearEngines();
457
+ delete process.env.XYZ_AGENT_DATA_DIR;
458
+ fs.rmSync(agentDir, { recursive: true, force: true });
459
+ vi.clearAllMocks();
460
+ });
461
+
462
+ it("[兜底] 默认路由 zcode + probe 失败 → 回退 pi runSpawn + engineFallback 留痕", async () => {
463
+ writeGlobalConfig(agentDir, "zcode");
464
+ const { service, zcode, piEngine } = setup(agentDir);
465
+ zcode.probeFailed = true;
466
+
467
+ const handle = await service.execute(baseOpts(agentDir));
468
+ // 兜底 = 走 pi runSpawn 路径(spawn 被调),非 pi EnginePort 未被消费
469
+ await vi.waitFor(() => expect(mockSpawn).toHaveBeenCalled());
470
+ expect(piEngine.runs.length).toBe(0);
471
+
472
+ const rec = service.collectRecords(10, "running").find((r) => r.id === handle.subagentId);
473
+ expect(rec?.engine).toBe("pi");
474
+ expect(rec?.engineFallback).toEqual({ from: "zcode", reason: "engine_probe_failed" });
475
+ // entry(register 写点)含 engineFallback——兜底路径允许新增键(D5 只约束纯缺省路径)
476
+ expect(JSON.stringify(toSubagentRecordEntry(rec!))).toContain("engine_probe_failed");
477
+ });
478
+
479
+ it("[守卫] 显式 engine='zcode' + probe 失败 → engine_probe_failed 报错不兜底", async () => {
480
+ const { service, zcode } = setup(agentDir);
481
+ zcode.probeFailed = true;
482
+
483
+ await expect(service.execute(baseOpts(agentDir, { engine: "zcode" }))).rejects.toThrow(
484
+ /engine_probe_failed/,
485
+ );
486
+ expect(mockSpawn).not.toHaveBeenCalled();
487
+ expect(zcode.runs.length).toBe(0);
488
+ expect(service.collectRecords(10, "all")).toHaveLength(0);
489
+ });
490
+
491
+ it("[journal] taskId=record.id + onPoolResolved retarget 落 engines/zcode/<poolKey>/,journalPath 同源回填", async () => {
492
+ process.env.XYZ_AGENT_DATA_DIR = agentDir;
493
+ const { service, zcode, pi } = setup(agentDir);
494
+ const POOL = "zcode-p-glm";
495
+ zcode.runImpl = (task, ctx) => {
496
+ ctx.onPoolResolved?.(POOL);
497
+ ctx.onEvent?.({ type: "message_end" } as AgentEvent);
498
+ return Promise.resolve({
499
+ handle: {
500
+ data: {
501
+ v: 1,
502
+ engineId: "zcode",
503
+ sessionRef: { dbPath: "sessions.db", sessionId: "sess-1" },
504
+ poolKey: POOL,
505
+ adapterVersion: "test",
506
+ },
507
+ },
508
+ outcome: doneOutcome("ok"),
509
+ });
510
+ };
511
+ const handle = await service.execute(baseOpts(agentDir, { engine: "zcode" }));
512
+ await vi.waitFor(() => expect(service.findRecord(handle.subagentId)).toBeUndefined());
513
+
514
+ // journal 文件名 = journal-<record.id>.jsonl,落在 engines/zcode/<poolKey>/ 下
515
+ const journalPath = resolveJournalPath(agentDir, "zcode", POOL, handle.subagentId);
516
+ expect(fs.existsSync(journalPath)).toBe(true);
517
+ const firstLine = JSON.parse(fs.readFileSync(journalPath, "utf8").split("\n")[0]);
518
+ expect(firstLine.taskId).toBe(handle.subagentId);
519
+
520
+ // archive entry 的 engineHandle:三键齐全,journalPath 与实际落盘路径一致(同源)
521
+ const entry = lastRecordEntry(pi);
522
+ expect(entry?.engineHandle).toEqual({
523
+ sessionRef: { dbPath: "sessions.db", sessionId: "sess-1" },
524
+ poolKey: POOL,
525
+ journalPath,
526
+ });
527
+ });
528
+
529
+ it("[engineHandle] 失败终态 sessionId 缺失 → 仍回填 dbPath/poolKey(①级降②级防御形态)", async () => {
530
+ process.env.XYZ_AGENT_DATA_DIR = agentDir;
531
+ const { service, zcode, pi } = setup(agentDir);
532
+ const POOL = "zcode-p-glm";
533
+ zcode.runImpl = (task, ctx) => {
534
+ ctx.onPoolResolved?.(POOL);
535
+ return Promise.resolve({
536
+ handle: {
537
+ data: { v: 1, engineId: "zcode", sessionRef: { dbPath: "sessions.db" }, poolKey: POOL, adapterVersion: "test" },
538
+ },
539
+ outcome: { ...doneOutcome(""), error: "engine_run_failed: boom", engineId: "zcode" },
540
+ });
541
+ };
542
+ const handle = await service.execute(baseOpts(agentDir, { engine: "zcode" }));
543
+ await vi.waitFor(() => expect(service.findRecord(handle.subagentId)).toBeUndefined());
544
+
545
+ const h = lastRecordEntry(pi)?.engineHandle;
546
+ expect(h?.sessionRef).toEqual({ dbPath: "sessions.db" });
547
+ expect(h?.poolKey).toBe(POOL);
548
+ expect(h?.journalPath).toBe(resolveJournalPath(agentDir, "zcode", POOL, handle.subagentId));
549
+ });
550
+
551
+ it("[D5 回归] pi 纯缺省路径 entry 不含 engine/engineFallback/engineHandle 键", async () => {
552
+ const { service, pi } = setup(agentDir);
553
+ const handle = await service.execute(baseOpts(agentDir));
554
+ await vi.waitFor(() => expect(mockSpawn).toHaveBeenCalled());
555
+
556
+ const rec = service.collectRecords(10, "running").find((r) => r.id === handle.subagentId);
557
+ const entryJson = JSON.stringify(toSubagentRecordEntry(rec!));
558
+ expect(entryJson).not.toContain("engineFallback");
559
+ expect(entryJson).not.toContain("engineHandle");
560
+ expect(entryJson).not.toMatch(/"engine"/);
561
+ // pi.appendEntry 上的 record entry 同形态(register 写点)
562
+ const entry = lastRecordEntry(pi);
563
+ expect(entry?.engine).toBeUndefined();
564
+ expect(entry?.engineFallback).toBeUndefined();
565
+ expect(entry?.engineHandle).toBeUndefined();
566
+ });
567
+
568
+ it("[池槽回归] 引擎 run 完成后 release 并发槽:连续 7 次(> maxConcurrent=6)后第 7 次不被永久阻塞", async () => {
569
+ // review MF1:kickOffEngineRun 旧实现 acquire 后无 release——每次引擎后台 run 泄漏
570
+ // 一个槽,累计 maxConcurrent(6) 次后全部 background subagent 在 acquire 队列挂死
571
+ process.env.XYZ_AGENT_DATA_DIR = agentDir;
572
+ const { service, zcode } = setup(agentDir);
573
+ zcode.runImpl = () => Promise.resolve({ handle: fakeHandle(), outcome: doneOutcome("ok") });
574
+
575
+ for (let i = 0; i < 7; i++) {
576
+ const h = await service.execute(baseOpts(agentDir, { engine: "zcode" }));
577
+ // 等本条终态化(record 离开内存)再发下一条——泄漏形态下第 7 条 acquire 永久
578
+ // 排队、record 永卡 running,本 waitFor 即超时失败
579
+ await vi.waitFor(() => expect(service.findRecord(h.subagentId)).toBeUndefined());
580
+ }
581
+ // 7 次引擎 run 全部真实执行(第 7 次未被泄漏槽阻塞)
582
+ expect(zcode.runs.length).toBe(7);
583
+ }, 10_000);
584
+
585
+ /** 最后一条 subagent-record entry(register→archive 双写点取终态侧)。 */
586
+ function lastRecordEntry(pi: ReturnType<typeof makePi>): Record<string, unknown> | undefined {
587
+ const calls = pi.appendEntry.mock.calls.filter((c) => c[0] === "subagent-record");
588
+ return calls.length > 0 ? (calls[calls.length - 1][1] as Record<string, unknown>) : undefined;
589
+ }
590
+ });
591
+
592
+ /** 假 EngineHandle(骨架路径不消费 handle 内容——U2 journal/handle 回填才用)。 */
593
+ function fakeHandle(): EngineHandle {
594
+ return {
595
+ data: { v: 1, engineId: "zcode", sessionRef: {}, poolKey: "shared", adapterVersion: "test" },
596
+ };
597
+ }