@zhushanwen/pi-subagent-workflow 8.4.0 → 8.6.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 (171) hide show
  1. package/package.json +22 -7
  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__/bg-notify-render.test.ts +73 -0
  6. package/src/execution/__tests__/chat-engine-routing.test.ts +601 -0
  7. package/src/execution/__tests__/delivery-methods.test.ts +38 -1
  8. package/src/execution/__tests__/execute-options-mapper.test.ts +11 -0
  9. package/src/execution/__tests__/execution-record.test.ts +237 -1
  10. package/src/execution/__tests__/explicit-agent-ref-guard.test.ts +171 -0
  11. package/src/execution/__tests__/format-schema-instruction.test.ts +63 -32
  12. package/src/execution/__tests__/helpers/spawn-mock.ts +4 -0
  13. package/src/execution/__tests__/index-session-start.test.ts +86 -7
  14. package/src/execution/__tests__/lifecycle-manager.test.ts +46 -0
  15. package/src/execution/__tests__/list-fields.test.ts +45 -14
  16. package/src/execution/__tests__/model-resolver.test.ts +57 -5
  17. package/src/execution/__tests__/notifier-flush.test.ts +64 -26
  18. package/src/execution/__tests__/notify-ledger.test.ts +826 -0
  19. package/src/execution/__tests__/output-collector.test.ts +299 -2
  20. package/src/execution/__tests__/pi-invocation.test.ts +62 -1
  21. package/src/execution/__tests__/relay-agent.test.ts +448 -0
  22. package/src/execution/__tests__/relay-env.test.ts +42 -0
  23. package/src/execution/__tests__/rpc-mode.test.ts +1 -1
  24. package/src/execution/__tests__/run-spawn-edges.test.ts +44 -1
  25. package/src/execution/__tests__/run-spawn-stdout-callback-throw.test.ts +199 -0
  26. package/src/execution/__tests__/session-runner-schema-env.test.ts +39 -0
  27. package/src/execution/__tests__/spawn-args.test.ts +37 -26
  28. package/src/execution/__tests__/start-sync-model-guard.test.ts +150 -0
  29. package/src/execution/__tests__/startup-config-declaration.test.ts +35 -0
  30. package/src/execution/__tests__/stream-sink-retirement.test.ts +261 -0
  31. package/src/execution/__tests__/subprocess-agent-runner-routing.test.ts +310 -0
  32. package/src/execution/__tests__/subprocess-agent-runner.test.ts +147 -6
  33. package/src/execution/__tests__/timeout-integration.test.ts +220 -2
  34. package/src/execution/__tests__/tool-action.test.ts +92 -1
  35. package/src/execution/agent-registry.ts +16 -0
  36. package/src/execution/argv-mirror.ts +5 -1
  37. package/src/execution/concurrency-pool.ts +1 -1
  38. package/src/execution/config.ts +25 -2
  39. package/src/execution/engine/__tests__/common/data-dir.test.ts +53 -0
  40. package/src/execution/engine/__tests__/common/errors.test.ts +132 -0
  41. package/src/execution/engine/__tests__/common/event-journal.test.ts +177 -0
  42. package/src/execution/engine/__tests__/common/kill-chain.test.ts +192 -0
  43. package/src/execution/engine/__tests__/common/nesting-guard.test.ts +81 -0
  44. package/src/execution/engine/__tests__/common/persona-router.test.ts +123 -0
  45. package/src/execution/engine/__tests__/common/pool-manager.test.ts +154 -0
  46. package/src/execution/engine/__tests__/common/schema-emulation.test.ts +128 -0
  47. package/src/execution/engine/__tests__/conformance/__fixtures__/pi-golden-events.json +28 -0
  48. package/src/execution/engine/__tests__/conformance/agent-event-invariants.ts +141 -0
  49. package/src/execution/engine/__tests__/conformance/contract.abort.test.ts +109 -0
  50. package/src/execution/engine/__tests__/conformance/contract.agent-events.test.ts +101 -0
  51. package/src/execution/engine/__tests__/conformance/contract.probe.test.ts +77 -0
  52. package/src/execution/engine/__tests__/conformance/contract.read-degradation.test.ts +104 -0
  53. package/src/execution/engine/__tests__/conformance/contract.relay.test.ts +342 -0
  54. package/src/execution/engine/__tests__/conformance/engine-conformance.live.test.ts +201 -0
  55. package/src/execution/engine/__tests__/conformance/golden-replay.pi.test.ts +76 -0
  56. package/src/execution/engine/__tests__/conformance/golden-replay.zcode.test.ts +79 -0
  57. package/src/execution/engine/__tests__/engine-discovery.test.ts +87 -0
  58. package/src/execution/engine/__tests__/engines-declaration.test.ts +36 -0
  59. package/src/execution/engine/__tests__/model-prompt.test.ts +85 -0
  60. package/src/execution/engine/__tests__/paths.test.ts +39 -0
  61. package/src/execution/engine/__tests__/registry.test.ts +120 -0
  62. package/src/execution/engine/__tests__/routing.test.ts +231 -0
  63. package/src/execution/engine/common/data-dir.ts +62 -0
  64. package/src/execution/engine/common/errors.ts +183 -0
  65. package/src/execution/engine/common/event-journal.ts +254 -0
  66. package/src/execution/engine/common/journal-replay.ts +62 -0
  67. package/src/execution/engine/common/kill-chain.ts +221 -0
  68. package/src/execution/engine/common/nesting-guard.ts +50 -0
  69. package/src/execution/engine/common/persona-router.ts +108 -0
  70. package/src/execution/engine/common/pool-manager.ts +226 -0
  71. package/src/execution/engine/common/schema-emulation.ts +189 -0
  72. package/src/execution/engine/common/session-view-projection.ts +51 -0
  73. package/src/execution/engine/engine-discovery.ts +65 -0
  74. package/src/execution/engine/engines/pi/__tests__/pi-engine.test.ts +469 -0
  75. package/src/execution/engine/engines/pi/__tests__/reader.test.ts +155 -0
  76. package/src/execution/engine/engines/pi/__tests__/task-spec-mapper.test.ts +164 -0
  77. package/src/execution/engine/engines/pi/pi-engine.ts +415 -0
  78. package/src/execution/engine/engines/pi/reader.ts +48 -0
  79. package/src/execution/engine/engines/pi/registration.ts +35 -0
  80. package/src/execution/engine/engines/pi/task-spec-mapper.ts +100 -0
  81. package/src/execution/engine/engines/zcode/__tests__/__fixtures__/zcode-golden-spawn.json +39 -0
  82. package/src/execution/engine/engines/zcode/__tests__/launcher.test.ts +150 -0
  83. package/src/execution/engine/engines/zcode/__tests__/parser.test.ts +246 -0
  84. package/src/execution/engine/engines/zcode/__tests__/preparer.test.ts +228 -0
  85. package/src/execution/engine/engines/zcode/__tests__/reader.test.ts +210 -0
  86. package/src/execution/engine/engines/zcode/__tests__/registration.test.ts +64 -0
  87. package/src/execution/engine/engines/zcode/__tests__/zcode-engine.live.test.ts +127 -0
  88. package/src/execution/engine/engines/zcode/__tests__/zcode-engine.test.ts +580 -0
  89. package/src/execution/engine/engines/zcode/constants.ts +43 -0
  90. package/src/execution/engine/engines/zcode/golden-sample.ts +39 -0
  91. package/src/execution/engine/engines/zcode/launcher.ts +161 -0
  92. package/src/execution/engine/engines/zcode/parser.ts +436 -0
  93. package/src/execution/engine/engines/zcode/preparer.ts +363 -0
  94. package/src/execution/engine/engines/zcode/reader.ts +381 -0
  95. package/src/execution/engine/engines/zcode/registration.ts +37 -0
  96. package/src/execution/engine/engines/zcode/zcode-engine.ts +658 -0
  97. package/src/execution/engine/host-task-spec.ts +47 -0
  98. package/src/execution/engine/model-prompt.ts +59 -0
  99. package/src/execution/engine/paths.ts +42 -0
  100. package/src/execution/engine/port.ts +153 -0
  101. package/src/execution/engine/registry.ts +123 -0
  102. package/src/execution/engine/routing.ts +218 -0
  103. package/src/execution/engine/types.ts +309 -0
  104. package/src/execution/execute-options-mapper.ts +13 -8
  105. package/src/execution/execution-record.ts +66 -1
  106. package/src/execution/lifecycle-manager.ts +23 -1
  107. package/src/execution/model-config-service.ts +16 -1
  108. package/src/execution/model-resolver.ts +37 -59
  109. package/src/execution/notifier.ts +105 -35
  110. package/src/execution/notify-ledger.ts +580 -0
  111. package/src/execution/output-collector.ts +143 -3
  112. package/src/execution/pi-invocation.ts +32 -2
  113. package/src/execution/record-entry.ts +14 -0
  114. package/src/execution/record-store.ts +34 -0
  115. package/src/execution/relay-env.ts +37 -0
  116. package/src/execution/session-runner.ts +328 -71
  117. package/src/execution/stream-sink.ts +26 -0
  118. package/src/execution/subagent-service.ts +273 -13
  119. package/src/execution/subprocess-agent-runner.ts +210 -14
  120. package/src/execution/types.ts +124 -5
  121. package/src/execution/ui-request-queue.ts +14 -4
  122. package/src/index.ts +99 -1
  123. package/src/interface/__tests__/subagent-tool-path-guard.test.ts +157 -0
  124. package/src/interface/__tests__/subagent-tool-prompt.test.ts +12 -0
  125. package/src/interface/bg-notify-render.ts +33 -12
  126. package/src/interface/helpers.ts +2 -2
  127. package/src/interface/subagent-actions.ts +29 -9
  128. package/src/interface/subagent-tool-schema.ts +156 -0
  129. package/src/interface/subagent-tool.ts +56 -119
  130. package/src/interface/subagents.ts +2 -2
  131. package/src/orchestration/__tests__/__fixtures__/worker-template.snapshot.txt +19 -3
  132. package/src/orchestration/__tests__/agent-call-catch-fallback.test.ts +0 -6
  133. package/src/orchestration/__tests__/agent-call-stream.test.ts +0 -5
  134. package/src/orchestration/__tests__/error-recovery-handlers.test.ts +89 -4
  135. package/src/orchestration/__tests__/execute-agent-call.test.ts +137 -0
  136. package/src/orchestration/__tests__/jsonl-run-store-corrupt-entry.test.ts +150 -0
  137. package/src/orchestration/__tests__/jsonl-run-store-retention.test.ts +202 -0
  138. package/src/orchestration/__tests__/launcher-nested-workflow.test.ts +326 -3
  139. package/src/orchestration/__tests__/lifecycle.test.ts +41 -7
  140. package/src/orchestration/__tests__/non-cloneable-return-e2e.test.ts +95 -0
  141. package/src/orchestration/__tests__/review-fix-loop-e2e.test.ts +57 -3
  142. package/src/orchestration/__tests__/skill-discovery.test.ts +44 -0
  143. package/src/orchestration/__tests__/worker-exit-without-result.test.ts +368 -0
  144. package/src/orchestration/__tests__/worker-script-builder-runtime.test.ts +43 -0
  145. package/src/orchestration/__tests__/worker-script-template-snapshot.test.ts +22 -3
  146. package/src/orchestration/agent-opts-resolver.ts +104 -23
  147. package/src/orchestration/error-recovery.ts +189 -33
  148. package/src/orchestration/execute-agent-call.ts +39 -0
  149. package/src/orchestration/jsonl-run-store.ts +121 -7
  150. package/src/orchestration/launcher.ts +60 -15
  151. package/src/orchestration/lifecycle.ts +10 -7
  152. package/src/orchestration/models/__tests__/budget.test.ts +1 -61
  153. package/src/orchestration/models/budget.ts +5 -35
  154. package/src/orchestration/models/run-runtime.ts +24 -9
  155. package/src/orchestration/models/types.ts +16 -0
  156. package/src/orchestration/script-lint.ts +1 -1
  157. package/src/orchestration/skill-discovery.ts +31 -8
  158. package/src/orchestration/worker-script-builder.ts +19 -3
  159. package/src/shared/__tests__/model-ref.test.ts +306 -0
  160. package/src/shared/__tests__/schema-jsonify.test.ts +1 -1
  161. package/src/shared/__tests__/timer-delay.test.ts +61 -0
  162. package/src/shared/meta-parser.ts +5 -1
  163. package/src/shared/model-ref.ts +286 -0
  164. package/src/shared/resource-meta.ts +5 -0
  165. package/src/shared/schema-env.ts +44 -0
  166. package/src/shared/schema-jsonify.ts +6 -4
  167. package/src/shared/timer-delay.ts +54 -0
  168. package/workflows/review-fix-loop-utils.cjs +9 -7
  169. package/workflows/review-fix-loop.js +20 -12
  170. package/src/orchestration/__tests__/concurrency-gate.test.ts +0 -125
  171. package/src/orchestration/concurrency-gate.ts +0 -69
@@ -0,0 +1,469 @@
1
+ // src/execution/engine/engines/pi/__tests__/pi-engine.test.ts
2
+ //
3
+ // PiEngine 适配层测试(P1 回填):capabilities 链路接通口径(D3)、run 委托与
4
+ // 中立声明→ExecuteOptions 映射、interact 三 action 直通、read 三级降级、probe 形状
5
+ // (C1:ok=false 时 error.recovery 非空)。fake PiEngineService 注入,不 spawn 真进程。
6
+
7
+ import * as fs from "node:fs";
8
+ import * as os from "node:os";
9
+ import * as path from "node:path";
10
+
11
+ import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
12
+
13
+ import { createRecord } from "../../../../execution-record.ts";
14
+ import type { ExecuteOptions, SubagentRecord } from "../../../../types.ts";
15
+ import { IDENTITY_CUSTOM_TYPE } from "../../../../session-reconstructor.ts";
16
+ import { SubagentStream } from "../../../../stream-sink.ts";
17
+ import type { RunContext } from "../../../port.ts";
18
+ import type { AgentTaskSpec, EngineHandle } from "../../../types.ts";
19
+ import {
20
+ PI_ADAPTER_VERSION,
21
+ PI_ENGINE_ID,
22
+ PiEngine,
23
+ type PiEngineService,
24
+ } from "../pi-engine.ts";
25
+
26
+ // ── fake 编排服务(结构子集,记录调用) ──
27
+
28
+ interface ServiceCalls {
29
+ executeOpts?: ExecuteOptions;
30
+ signal?: AbortSignal;
31
+ onEvent?: (event: unknown) => void;
32
+ stream?: unknown;
33
+ deliverMessage?: { recordId: string; text: string; interrupt: boolean };
34
+ close?: { recordId: string; force: boolean };
35
+ cancelId?: string;
36
+ getRecordId?: string;
37
+ }
38
+
39
+ /** 构造最小完整 SubagentRecord(collectRecords 返回项——resolveRecordId 只读 id/sessionFile)。 */
40
+ function makeListedRecord(id: string, sessionFile: string | undefined): SubagentRecord {
41
+ return {
42
+ id,
43
+ agent: "reviewer",
44
+ task: "t",
45
+ slug: "s",
46
+ status: "running",
47
+ mode: "background",
48
+ startedAt: 1,
49
+ rootSessionId: undefined,
50
+ parentRecordId: undefined,
51
+ depth: 0,
52
+ endedAt: undefined,
53
+ turns: 0,
54
+ totalTokens: 0,
55
+ model: "p/m",
56
+ thinkingLevel: undefined,
57
+ eventLog: [],
58
+ displayItems: [],
59
+ ...(sessionFile !== undefined ? { sessionFile } : {}),
60
+ };
61
+ }
62
+
63
+ function makeFakeService(overrides?: {
64
+ executeResult?: Awaited<ReturnType<PiEngineService["executeAndAwait"]>>;
65
+ executeReject?: Error;
66
+ recordsById?: Map<string, ReturnType<typeof createRecord>>;
67
+ listed?: Array<{ id: string; sessionFile?: string }>;
68
+ cancelResult?: boolean;
69
+ deliverReject?: Error;
70
+ }): { service: PiEngineService; calls: ServiceCalls } {
71
+ const calls: ServiceCalls = {};
72
+ const record = createRecord("sa-fake", {
73
+ agent: "reviewer",
74
+ model: "p/m",
75
+ thinkingLevel: "high",
76
+ mode: "background",
77
+ task: "t",
78
+ slug: "s",
79
+ startedAt: 1,
80
+ controller: new AbortController(),
81
+ });
82
+ const service: PiEngineService = {
83
+ executeAndAwait: async (opts, signal, onEvent, stream) => {
84
+ calls.executeOpts = opts;
85
+ calls.signal = signal;
86
+ calls.onEvent = onEvent;
87
+ calls.stream = stream;
88
+ if (overrides?.executeReject) throw overrides.executeReject;
89
+ return (
90
+ overrides?.executeResult ?? {
91
+ content: "done",
92
+ durationMs: 42,
93
+ toolCalls: [],
94
+ }
95
+ );
96
+ },
97
+ getRecordForAction: (id) => {
98
+ calls.getRecordId = id;
99
+ // recordsById 提供时严格查表(模拟「record 不存在 → throw」路径);缺省返回默认 record
100
+ if (overrides?.recordsById) {
101
+ const found = overrides.recordsById.get(id);
102
+ if (!found) {
103
+ throw new Error(`subagent not found or not owned: ${id}`);
104
+ }
105
+ return found;
106
+ }
107
+ return record;
108
+ },
109
+ deliverMessage: async (rec, text, interrupt) => {
110
+ calls.deliverMessage = { recordId: rec.id, text, interrupt };
111
+ if (overrides?.deliverReject) throw overrides.deliverReject;
112
+ },
113
+ closeSubagent: async (rec, force) => {
114
+ calls.close = { recordId: rec.id, force };
115
+ },
116
+ cancel: (id) => {
117
+ calls.cancelId = id;
118
+ return overrides?.cancelResult ?? true;
119
+ },
120
+ collectRecords: () =>
121
+ (overrides?.listed ?? []).map(({ id, sessionFile }) => makeListedRecord(id, sessionFile)),
122
+ };
123
+ return { service, calls };
124
+ }
125
+
126
+ /** 全字段任务声明(覆盖全部泛化点)。 */
127
+ function makeSpec(): AgentTaskSpec {
128
+ return {
129
+ task: "do the thing",
130
+ slug: "do-thing",
131
+ agent: "reviewer",
132
+ model: "zai-coding-cn/glm-5.2",
133
+ effort: "high",
134
+ persona: { skillPath: "/skills/r/SKILL.md", appendSystemPrompt: ["extra"] },
135
+ schema: { type: "object", properties: { ok: { type: "boolean" } } },
136
+ maxTurns: 5,
137
+ graceTurns: 2,
138
+ fork: true,
139
+ worktree: false,
140
+ cwd: "/work",
141
+ conversation: true,
142
+ idleTimeoutMs: 99,
143
+ };
144
+ }
145
+
146
+ function makeHandle(sessionRef: Record<string, string>): EngineHandle {
147
+ return { data: { v: 1, engineId: PI_ENGINE_ID, sessionRef, poolKey: "shared", adapterVersion: PI_ADAPTER_VERSION } };
148
+ }
149
+
150
+ function makeRunCtx(overrides?: Partial<RunContext>): RunContext {
151
+ return { taskId: "sa-test", poolKey: "shared", ...overrides };
152
+ }
153
+
154
+ // ── capabilities ──
155
+
156
+ describe("PiEngine.capabilities(D3 链路接通口径)", () => {
157
+ it("id = 'pi';声明本仓 subagent 链路实际接通的能力(非 RPC 理论能力)", () => {
158
+ const engine = new PiEngine({ getService: () => null });
159
+ expect(engine.id).toBe(PI_ENGINE_ID);
160
+ expect(engine.capabilities()).toEqual({
161
+ schemaEnforcement: "native", // PI_WORKFLOW_SCHEMA env 注入链路(方案 A 唯一权威)
162
+ steer: "unsupported", // RPC 有 steer 但 spawn 链路未接通(session-runner no-op)
163
+ conversation: "native", // chatMode idle 复用 + message/close/cancel 已接通
164
+ personaInjection: "flag", // --skill / --append-system-prompt
165
+ eventGranularity: "stream", // 30+ 事件流
166
+ sandbox: "emulated", // 无 OS sandbox;worktree 隔离
167
+ sessionRead: "full", // pi JSONL 完整重建
168
+ resume: "native", // 同进程 idle 复用 + --session 冷续写
169
+ interrupt: "kill-only", // 现链路 abort = SIGTERM,rpc abort 未接通
170
+ permissionMode: "native", // argv-mirror 镜像主进程 flag
171
+ });
172
+ });
173
+ });
174
+
175
+ // ── run ──
176
+
177
+ describe("PiEngine.run", () => {
178
+ it("中立声明 → ExecuteOptions 映射 + signal/onEvent/stream/ctxModel 直通(行为零变化)", async () => {
179
+ const { service, calls } = makeFakeService();
180
+ const engine = new PiEngine({ getService: () => service });
181
+ const spec = makeSpec();
182
+ const ctxModel = { id: "m", name: "M", provider: "p", reasoning: false };
183
+ const onEvent = (e: unknown) => void e;
184
+ const stream = new SubagentStream("sa-test", { setWidget: () => {} });
185
+ const signal = new AbortController().signal;
186
+
187
+ const { handle, outcome } = await engine.run(spec, makeRunCtx({ signal, onEvent, ctxModel, stream }));
188
+
189
+ // 传参恒等(SAR 接线前的直接调用形态)
190
+ expect(calls.signal).toBe(signal);
191
+ expect(calls.onEvent).toBe(onEvent);
192
+ expect(calls.stream).toBe(stream);
193
+ expect(calls.executeOpts).toEqual({
194
+ task: "do the thing",
195
+ slug: "do-thing",
196
+ agent: "reviewer",
197
+ model: "zai-coding-cn/glm-5.2",
198
+ thinkingLevel: "high", // effort 恒等映射回
199
+ skillPath: "/skills/r/SKILL.md", // persona 还原
200
+ appendSystemPrompt: ["extra"],
201
+ schema: spec.schema,
202
+ schemaEnv: JSON.stringify(spec.schema), // schema 派生(D-A6 bridge)
203
+ maxTurns: 5,
204
+ graceTurns: 2,
205
+ ctxModel,
206
+ fork: true,
207
+ worktree: false,
208
+ cwd: "/work",
209
+ conversation: true,
210
+ idleTimeoutMs: 99,
211
+ engine: "pi", // P4 引擎留痕(D9①):实际执行引擎 id 进 record 投影链
212
+ });
213
+
214
+ // outcome 字段全集映射(缺字段在此转红——SAR outcomeToRunnerResult 的完整性依赖)
215
+ expect(outcome).toEqual({
216
+ content: "done",
217
+ parsedOutput: undefined,
218
+ usage: undefined,
219
+ durationMs: 42,
220
+ error: undefined,
221
+ sessionId: undefined,
222
+ sessionFile: undefined,
223
+ worktreePath: undefined,
224
+ toolCalls: [],
225
+ engineId: "pi",
226
+ });
227
+
228
+ // handle 自描述(engineId + sessionRef + poolKey + adapter 版本)
229
+ expect(handle.data).toMatchObject({
230
+ v: 1,
231
+ engineId: "pi",
232
+ poolKey: "shared",
233
+ adapterVersion: PI_ADAPTER_VERSION,
234
+ });
235
+ });
236
+
237
+ it("sessionFile 回填进 handle.sessionRef;ctx.poolKey 透传(空串兜底 'shared')", async () => {
238
+ const { service } = makeFakeService({
239
+ executeResult: { content: "x", sessionFile: "/tmp/s.jsonl", sessionId: "sess-1", toolCalls: [] },
240
+ });
241
+ const engine = new PiEngine({ getService: () => service });
242
+ const r1 = await engine.run({ task: "t", slug: "s" }, makeRunCtx({ poolKey: "agent-reviewer" }));
243
+ expect(r1.handle.data.sessionRef).toEqual({ sessionFile: "/tmp/s.jsonl" });
244
+ expect(r1.handle.data.poolKey).toBe("agent-reviewer");
245
+ expect(r1.outcome.sessionId).toBe("sess-1");
246
+ const r2 = await engine.run({ task: "t", slug: "s" }, makeRunCtx({ poolKey: "" }));
247
+ expect(r2.handle.data.poolKey).toBe("shared");
248
+ });
249
+
250
+ it("服务不可用(prepare 期)→ reject 且不产生 handle", async () => {
251
+ const engine = new PiEngine({ getService: () => null });
252
+ await expect(engine.run({ task: "t", slug: "s" }, makeRunCtx())).rejects.toThrow(/SubagentService unavailable/);
253
+ });
254
+
255
+ it("executeAndAwait throw(嵌套超限等创建期异常)→ 向上传播(SAR catch 兜底不变)", async () => {
256
+ const { service } = makeFakeService({ executeReject: new Error("nesting depth exceeded") });
257
+ const engine = new PiEngine({ getService: () => service });
258
+ await expect(engine.run({ task: "t", slug: "s" }, makeRunCtx())).rejects.toThrow("nesting depth exceeded");
259
+ });
260
+ });
261
+
262
+ // ── interact ──
263
+
264
+ describe("PiEngine.interact(chatMode 交互面直通)", () => {
265
+ it("message → deliverMessage(record, text, false)(interrupt 中立 action 未携带,P1 恒 followUp)", async () => {
266
+ const { service, calls } = makeFakeService();
267
+ const engine = new PiEngine({ getService: () => service });
268
+ const res = await engine.interact(makeHandle({ recordId: "sa-fake" }), { kind: "message", payload: "next round" });
269
+ expect(res).toEqual({ ok: true, delivered: true });
270
+ expect(calls.deliverMessage).toEqual({ recordId: "sa-fake", text: "next round", interrupt: false });
271
+ });
272
+
273
+ it("close → closeSubagent(record, force);payload 缺省 force=false", async () => {
274
+ const { service, calls } = makeFakeService();
275
+ const engine = new PiEngine({ getService: () => service });
276
+ await engine.interact(makeHandle({ recordId: "sa-fake" }), { kind: "close", payload: { force: true } });
277
+ expect(calls.close).toEqual({ recordId: "sa-fake", force: true });
278
+ await engine.interact(makeHandle({ recordId: "sa-fake" }), { kind: "close" });
279
+ expect(calls.close).toEqual({ recordId: "sa-fake", force: false });
280
+ });
281
+
282
+ it("cancel → service.cancel(recordId)", async () => {
283
+ const { service, calls } = makeFakeService();
284
+ const engine = new PiEngine({ getService: () => service });
285
+ const res = await engine.interact(makeHandle({ recordId: "sa-fake" }), { kind: "cancel" });
286
+ expect(res).toEqual({ ok: true, delivered: true });
287
+ expect(calls.cancelId).toBe("sa-fake");
288
+ });
289
+
290
+ it("cancel 返回 false → engine_interact_failed", async () => {
291
+ const { service } = makeFakeService({ cancelResult: false });
292
+ const engine = new PiEngine({ getService: () => service });
293
+ const res = await engine.interact(makeHandle({ recordId: "sa-fake" }), { kind: "cancel" });
294
+ expect(res.ok).toBe(false);
295
+ if (!res.ok) {
296
+ expect(res.code).toBe("engine_interact_failed");
297
+ expect(res.message).toContain("sa-fake");
298
+ }
299
+ });
300
+
301
+ it("recordId 缺席时按 sessionFile 扫描兜底定位(collectRecords 匹配)", async () => {
302
+ const { service, calls } = makeFakeService({ listed: [{ id: "sa-scan-hit", sessionFile: "/x/s.jsonl" }] });
303
+ const engine = new PiEngine({ getService: () => service });
304
+ await engine.interact(makeHandle({ sessionFile: "/x/s.jsonl" }), { kind: "message", payload: "hi" });
305
+ expect(calls.getRecordId).toBe("sa-scan-hit");
306
+ });
307
+
308
+ it("死 handle(无定位符 / record 不存在)→ engine_session_not_resumable(含 cold resume 指引)", async () => {
309
+ const { service } = makeFakeService();
310
+ const engine = new PiEngine({ getService: () => service });
311
+ const res = await engine.interact(makeHandle({}), { kind: "message", payload: "hi" });
312
+ expect(res.ok).toBe(false);
313
+ if (!res.ok) {
314
+ expect(res.code).toBe("engine_session_not_resumable");
315
+ expect(res.message).toContain("--session");
316
+ }
317
+ });
318
+
319
+ it("交互面 throw → engine_interact_failed(message 文案透传行动语言)", async () => {
320
+ const { service } = makeFakeService({
321
+ deliverReject: new Error("subagent sa-fake is not ready for a new message"),
322
+ });
323
+ const engine = new PiEngine({ getService: () => service });
324
+ const res = await engine.interact(makeHandle({ recordId: "sa-fake" }), { kind: "message", payload: "hi" });
325
+ expect(res).toEqual({
326
+ ok: false,
327
+ code: "engine_interact_failed",
328
+ message: "subagent sa-fake is not ready for a new message",
329
+ });
330
+ });
331
+
332
+ it("服务不可用 → reject", async () => {
333
+ const engine = new PiEngine({ getService: () => null });
334
+ await expect(
335
+ engine.interact(makeHandle({ recordId: "x" }), { kind: "message", payload: "hi" }),
336
+ ).rejects.toThrow(/SubagentService unavailable/);
337
+ });
338
+ });
339
+
340
+ // ── read ──
341
+
342
+ describe("PiEngine.read(D6 降级链)", () => {
343
+ let tmpDir: string;
344
+
345
+ beforeEach(() => {
346
+ tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), "pi-engine-read-"));
347
+ });
348
+ afterEach(() => {
349
+ fs.rmSync(tmpDir, { recursive: true, force: true });
350
+ });
351
+
352
+ it("无 sessionRef → outcome-only(turns 空)", async () => {
353
+ const engine = new PiEngine({ getService: () => null });
354
+ const view = await engine.read(makeHandle({}));
355
+ expect(view).toEqual({ engineId: "pi", turns: [], source: "outcome-only" });
356
+ });
357
+
358
+ it("sessionFile 不存在/损坏 → outcome-only(不 throw)", async () => {
359
+ const engine = new PiEngine({ getService: () => null });
360
+ const view = await engine.read(makeHandle({ sessionFile: path.join(tmpDir, "missing.jsonl") }));
361
+ expect(view.source).toBe("outcome-only");
362
+ });
363
+
364
+ it("pi JSONL 原生读取(第①级):turns 重建 + usage 聚合 + 内部态剥离", async () => {
365
+ const file = path.join(tmpDir, "s.jsonl");
366
+ const lines: string[] = [
367
+ JSON.stringify({ type: "session", version: 3, id: "sess-uuid", timestamp: "2026-01-01T00:00:00.000Z", cwd: tmpDir }),
368
+ JSON.stringify({
369
+ type: "custom", id: "id-1", parentId: null, timestamp: "2026-01-01T00:00:00.000Z",
370
+ customType: IDENTITY_CUSTOM_TYPE,
371
+ data: { id: "bg-1", agent: "worker", mode: "background", task: "t", startedAt: 500 },
372
+ }),
373
+ JSON.stringify({
374
+ type: "message", id: "msg-1", parentId: "id-1", timestamp: new Date(1000).toISOString(),
375
+ message: {
376
+ role: "assistant",
377
+ content: [
378
+ { type: "text", text: "hello" },
379
+ { type: "toolCall", id: "tc-1", name: "bash", arguments: { cmd: "ls" } },
380
+ ],
381
+ usage: { input: 10, output: 20, cacheRead: 1, cacheWrite: 2, totalTokens: 33, cost: { total: 0.5 } },
382
+ stopReason: "stop",
383
+ timestamp: 1000,
384
+ },
385
+ }),
386
+ JSON.stringify({
387
+ type: "message", id: "tr-1", parentId: "msg-1", timestamp: new Date(2000).toISOString(),
388
+ message: {
389
+ role: "toolResult", toolCallId: "tc-1", toolName: "bash",
390
+ content: [{ type: "text", text: "ok" }], timestamp: 2000,
391
+ },
392
+ }),
393
+ ];
394
+ fs.writeFileSync(file, lines.join("\n") + "\n");
395
+
396
+ const engine = new PiEngine({ getService: () => null });
397
+ const view = await engine.read(makeHandle({ sessionFile: file }));
398
+ expect(view.source).toBe("native");
399
+ expect(view.engineId).toBe("pi");
400
+ expect(view.sessionId).toBe("bg-1");
401
+ expect(view.turns).toHaveLength(1);
402
+ expect(view.turns[0].text).toBe("hello");
403
+ expect(view.turns[0].closed).toBe(true);
404
+ // toolCall 纯净形状(无 _status/startedTs 内部态)
405
+ expect(view.turns[0].toolCalls[0]).toMatchObject({ toolName: "bash" });
406
+ expect("_status" in view.turns[0].toolCalls[0]).toBe(false);
407
+ // usage 聚合(turn usageDelta → AgentUsageTotal)
408
+ expect(view.usage).toEqual({ input: 10, output: 20, cacheRead: 1, cacheWrite: 2, cost: 0.5, total: 33 });
409
+ });
410
+ });
411
+
412
+ // ── probe ──
413
+
414
+ describe("PiEngine.probe(D7 + C1 形状)", () => {
415
+ const originalArgv = process.argv;
416
+ const originalExecPath = process.execPath;
417
+ const originalPath = process.env.PATH;
418
+
419
+ afterEach(() => {
420
+ Object.defineProperty(process, "argv", { value: originalArgv, configurable: true });
421
+ Object.defineProperty(process, "execPath", { value: originalExecPath, configurable: true });
422
+ process.env.PATH = originalPath;
423
+ });
424
+
425
+ it("探针成功:invocation + version 两级 check,engineVersion 实测", async () => {
426
+ const probeVersion = vi.fn(async () => "pi 0.84.1");
427
+ const engine = new PiEngine({ getService: () => null, probeVersion });
428
+ const report = await engine.probe();
429
+ expect(report.ok).toBe(true);
430
+ expect(report.engineVersion).toBe("pi 0.84.1");
431
+ expect(report.checks.map((c) => c.name)).toEqual(["invocation", "version"]);
432
+ expect(report.error).toBeUndefined();
433
+ expect(probeVersion).toHaveBeenCalledTimes(1);
434
+ });
435
+
436
+ it("版本探测失败 → ok=false 且 error.recovery 非空(C1:恢复指引必填)", async () => {
437
+ const engine = new PiEngine({ getService: () => null, probeVersion: async () => undefined });
438
+ const report = await engine.probe();
439
+ expect(report.ok).toBe(false);
440
+ expect(report.error?.code).toBe("engine_probe_failed");
441
+ expect(report.error?.recovery.length).toBeGreaterThan(0);
442
+ });
443
+
444
+ it("结果缓存:二次 probe 不重跑;force 强探重跑", async () => {
445
+ const probeVersion = vi.fn(async () => "pi 1.0");
446
+ const engine = new PiEngine({ getService: () => null, probeVersion });
447
+ const first = await engine.probe();
448
+ const second = await engine.probe();
449
+ expect(second).toBe(first); // 缓存命中(同对象)
450
+ expect(probeVersion).toHaveBeenCalledTimes(1);
451
+ await engine.probe({ force: true });
452
+ expect(probeVersion).toHaveBeenCalledTimes(2);
453
+ });
454
+
455
+ it("invocation 不可解析(PATH 无 pi)→ 第 1 check 失败,不再尝试版本探测", async () => {
456
+ // bun 虚拟脚本 + 通用 runtime → 退化为 PATH 依赖形态(command === "pi")
457
+ Object.defineProperty(process, "argv", { value: ["bun", "/$bunfs/root/pi"], configurable: true });
458
+ Object.defineProperty(process, "execPath", { value: "/usr/bin/bun", configurable: true });
459
+ process.env.PATH = path.join(os.tmpdir(), `no-pi-here-${Date.now()}`);
460
+ const probeVersion = vi.fn(async () => "should-not-run");
461
+ const engine = new PiEngine({ getService: () => null, probeVersion });
462
+ const report = await engine.probe();
463
+ expect(report.ok).toBe(false);
464
+ expect(report.checks).toHaveLength(1);
465
+ expect(report.checks[0].name).toBe("invocation");
466
+ expect(report.checks[0].ok).toBe(false);
467
+ expect(probeVersion).not.toHaveBeenCalled();
468
+ });
469
+ });
@@ -0,0 +1,155 @@
1
+ // reader.test.ts —— PiEngine reader(P5):readPiSessionView 对 session-reconstructor
2
+ // 的投影包装。断言三件事:①合法 JSONL → SessionView(turns/toolCalls strip 内部态/
3
+ // usage 聚合/source='native');②reconstructor 降级条件(文件缺失/缺 identity/无
4
+ // assistant)→ undefined 不 throw;③不泄漏 InternalToolCall 内部态(_status/startedTs)。
5
+
6
+ import * as fs from "node:fs";
7
+ import * as os from "node:os";
8
+ import * as path from "node:path";
9
+
10
+ import { beforeAll, describe, expect, it } from "vitest";
11
+
12
+ import { readPiSessionView } from "../reader.ts";
13
+
14
+ let tmpDir: string;
15
+
16
+ beforeAll(() => {
17
+ tmpDir = fs.mkdtempSync(join(os.tmpdir(), "pi-reader-"));
18
+ });
19
+
20
+ // node:path 的 join 在 ESM 测试里直接用(避免顶部 import 与 beforeAll 顺序问题)
21
+ function join(...segs: string[]): string {
22
+ return path.join(...segs);
23
+ }
24
+
25
+ /** 最小合法 subagent JSONL:header + identity + 一轮 assistant(text/thinking/toolCall)+ toolResult。 */
26
+ function writeValidSession(file: string): void {
27
+ const lines = [
28
+ JSON.stringify({ type: "session", id: "pi-sess-1", cwd: "/proj", timestamp: "2026-08-25T10:00:00Z" }),
29
+ JSON.stringify({
30
+ type: "custom",
31
+ customType: "subagent-identity",
32
+ data: { id: "bg-1-abc", agent: "reviewer", mode: "background", task: "review code", slug: "rev", startedAt: 1756000000000 },
33
+ }),
34
+ JSON.stringify({
35
+ type: "message",
36
+ timestamp: "2026-08-25T10:00:01Z",
37
+ message: {
38
+ role: "assistant",
39
+ timestamp: 1756000001000,
40
+ content: [
41
+ { type: "thinking", thinking: "let me look" },
42
+ { type: "toolCall", id: "tc-1", name: "bash", arguments: { command: "ls" } },
43
+ ],
44
+ usage: { input: 10, output: 5, cacheRead: 1, cacheWrite: 2, cost: { total: 0.1 } },
45
+ stopReason: "toolUse",
46
+ },
47
+ }),
48
+ JSON.stringify({
49
+ type: "message",
50
+ timestamp: "2026-08-25T10:00:02Z",
51
+ message: { role: "toolResult", toolCallId: "tc-1", toolName: "bash", content: [{ type: "text", text: "file-a" }], timestamp: 1756000002000 },
52
+ }),
53
+ JSON.stringify({
54
+ type: "message",
55
+ timestamp: "2026-08-25T10:00:03Z",
56
+ message: {
57
+ role: "assistant",
58
+ timestamp: 1756000003000,
59
+ content: [{ type: "text", text: "all good" }],
60
+ usage: { input: 20, output: 8, cacheRead: 0, cacheWrite: 0, cost: { total: 0.05 } },
61
+ stopReason: "stop",
62
+ },
63
+ }),
64
+ ];
65
+ fs.writeFileSync(file, `${lines.join("\n")}\n`, "utf8");
66
+ }
67
+
68
+ describe("readPiSessionView", () => {
69
+ it("projects a valid subagent JSONL to a native SessionView", async () => {
70
+ const file = join(tmpDir, "valid.jsonl");
71
+ writeValidSession(file);
72
+
73
+ const view = await readPiSessionView(file);
74
+
75
+ expect(view).toBeDefined();
76
+ expect(view?.engineId).toBe("pi");
77
+ expect(view?.source).toBe("native");
78
+ // 两轮 assistant message → 两个 turn(toolResult 不产 turn)
79
+ expect(view?.turns).toHaveLength(2);
80
+
81
+ const [turn1, turn2] = view?.turns ?? [];
82
+ expect(turn1?.text).toBe("");
83
+ expect(turn1?.thinking).toBe("let me look");
84
+ expect(turn1?.toolCalls).toHaveLength(1);
85
+ const tc = turn1?.toolCalls[0];
86
+ expect(tc?.toolName).toBe("bash");
87
+ expect(tc?.args).toEqual({ command: "ls" });
88
+ expect(tc?.result).toEqual({ content: [{ type: "text", text: "file-a" }] });
89
+ // InternalToolCall 内部态不泄漏(导出纯净形状)
90
+ expect(tc).not.toHaveProperty("_status");
91
+ expect(tc).not.toHaveProperty("startedTs");
92
+ expect(turn1?.closed).toBe(true);
93
+
94
+ expect(turn2?.text).toBe("all good");
95
+ expect(turn2?.closed).toBe(true);
96
+
97
+ // usage 聚合:两轮 usageDelta 之和(cost 浮点累加有精度噪声,用 toBeCloseTo)
98
+ expect(view?.usage?.input).toBe(30)
99
+ expect(view?.usage?.output).toBe(13)
100
+ expect(view?.usage?.cacheRead).toBe(1)
101
+ expect(view?.usage?.cacheWrite).toBe(2)
102
+ expect(view?.usage?.cost).toBeCloseTo(0.15, 10)
103
+ expect(view?.usage?.total).toBe(46);
104
+ });
105
+
106
+ it("returns undefined (no throw) when file is missing", async () => {
107
+ await expect(readPiSessionView(join(tmpDir, "nope.jsonl"))).resolves.toBeUndefined();
108
+ });
109
+
110
+ it("returns undefined when identity entry is missing", async () => {
111
+ const file = join(tmpDir, "no-identity.jsonl");
112
+ fs.writeFileSync(
113
+ file,
114
+ `${JSON.stringify({ type: "session", id: "s" })}\n${JSON.stringify({
115
+ type: "message",
116
+ message: { role: "assistant", content: [{ type: "text", text: "hi" }], timestamp: 1 },
117
+ })}\n`,
118
+ "utf8",
119
+ );
120
+ await expect(readPiSessionView(file)).resolves.toBeUndefined();
121
+ });
122
+
123
+ it("returns undefined when no assistant message exists", async () => {
124
+ const file = join(tmpDir, "no-assistant.jsonl");
125
+ fs.writeFileSync(
126
+ file,
127
+ `${JSON.stringify({ type: "session", id: "s" })}\n${JSON.stringify({
128
+ type: "custom",
129
+ customType: "subagent-identity",
130
+ data: { id: "bg-2", agent: "w", mode: "background", task: "t", startedAt: 1 },
131
+ })}\n`,
132
+ "utf8",
133
+ );
134
+ await expect(readPiSessionView(file)).resolves.toBeUndefined();
135
+ });
136
+
137
+ it("omits usage when no message carries it", async () => {
138
+ const file = join(tmpDir, "no-usage.jsonl");
139
+ fs.writeFileSync(
140
+ file,
141
+ `${JSON.stringify({ type: "session", id: "s" })}\n${JSON.stringify({
142
+ type: "custom",
143
+ customType: "subagent-identity",
144
+ data: { id: "bg-3", agent: "w", mode: "background", task: "t", startedAt: 1 },
145
+ })}\n${JSON.stringify({
146
+ type: "message",
147
+ message: { role: "assistant", content: [{ type: "text", text: "hi" }], stopReason: "stop", timestamp: 1 },
148
+ })}\n`,
149
+ "utf8",
150
+ );
151
+ const view = await readPiSessionView(file);
152
+ expect(view?.turns).toHaveLength(1);
153
+ expect(view?.usage).toBeUndefined();
154
+ });
155
+ });