@asterxsk/kiln 0.1.0 → 0.2.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 (203) hide show
  1. package/LICENSE +21 -21
  2. package/README.md +170 -170
  3. package/agent/AGENTS.md +67 -67
  4. package/agent/README.md +5 -5
  5. package/agent/extensions/AGENTS.md +68 -68
  6. package/agent/extensions/ask-user/index.ts +418 -418
  7. package/agent/extensions/ask-user/package-lock.json +769 -769
  8. package/agent/extensions/ask-user/package.json +19 -19
  9. package/agent/extensions/ask-user/prompt.ts +45 -45
  10. package/agent/extensions/ask-user/tsconfig.json +7 -7
  11. package/agent/extensions/background-terminals/docs/implementation-guide.md +942 -942
  12. package/agent/extensions/background-terminals/index.ts +627 -627
  13. package/agent/extensions/background-terminals/manager.test.ts +735 -735
  14. package/agent/extensions/background-terminals/output.test.ts +109 -109
  15. package/agent/extensions/background-terminals/package-lock.json +769 -769
  16. package/agent/extensions/background-terminals/package.json +17 -17
  17. package/agent/extensions/background-terminals/prompt.test.ts +125 -125
  18. package/agent/extensions/background-terminals/ps.test.ts +82 -82
  19. package/agent/extensions/background-terminals/result-delivery.test.ts +44 -44
  20. package/agent/extensions/background-terminals/src/domain.ts +87 -87
  21. package/agent/extensions/background-terminals/src/manager.ts +907 -907
  22. package/agent/extensions/background-terminals/src/output.ts +84 -84
  23. package/agent/extensions/background-terminals/src/prompt.ts +142 -142
  24. package/agent/extensions/background-terminals/src/result-delivery.ts +27 -27
  25. package/agent/extensions/background-terminals/src/runtime.ts +36 -36
  26. package/agent/extensions/background-terminals/src/ui/output-view.ts +79 -79
  27. package/agent/extensions/background-terminals/src/ui/ps.ts +621 -621
  28. package/agent/extensions/background-terminals/tsconfig.json +7 -7
  29. package/agent/extensions/file-search/index.spec.ts +443 -443
  30. package/agent/extensions/file-search/index.ts +459 -459
  31. package/agent/extensions/file-search/package-lock.json +2253 -2253
  32. package/agent/extensions/file-search/package.json +23 -23
  33. package/agent/extensions/file-search/src/args.ts +122 -122
  34. package/agent/extensions/file-search/src/binaries.ts +422 -422
  35. package/agent/extensions/file-search/src/output.ts +126 -126
  36. package/agent/extensions/file-search/src/process.ts +146 -146
  37. package/agent/extensions/file-search/src/prompt.ts +52 -52
  38. package/agent/extensions/file-search/tsconfig.json +7 -7
  39. package/agent/extensions/modelconf/PLAN.md +915 -915
  40. package/agent/extensions/modelconf/index.ts +296 -296
  41. package/agent/extensions/modelconf/src/ui/ModelConfView.ts +1101 -1101
  42. package/agent/extensions/pi-web-access/CHANGELOG.md +690 -690
  43. package/agent/extensions/pi-web-access/LICENSE +21 -21
  44. package/agent/extensions/pi-web-access/README.md +470 -470
  45. package/agent/extensions/pi-web-access/SECURITY.md +5 -5
  46. package/agent/extensions/pi-web-access/activity.ts +101 -101
  47. package/agent/extensions/pi-web-access/auth-fetch.ts +148 -148
  48. package/agent/extensions/pi-web-access/brightdata-unlocker.ts +272 -272
  49. package/agent/extensions/pi-web-access/chrome-cookies.ts +669 -669
  50. package/agent/extensions/pi-web-access/content-find.ts +139 -139
  51. package/agent/extensions/pi-web-access/credential-source.ts +191 -191
  52. package/agent/extensions/pi-web-access/data-uri-sanitize.ts +406 -406
  53. package/agent/extensions/pi-web-access/datalab-pdf-extract.ts +568 -568
  54. package/agent/extensions/pi-web-access/declared-web-links.ts +173 -173
  55. package/agent/extensions/pi-web-access/evidence/CONTRACT-EVIDENCE.md +496 -496
  56. package/agent/extensions/pi-web-access/evidence/contract-probe.mjs +140 -140
  57. package/agent/extensions/pi-web-access/exa.ts +526 -526
  58. package/agent/extensions/pi-web-access/extract.ts +1196 -1196
  59. package/agent/extensions/pi-web-access/feature-config.ts +29 -29
  60. package/agent/extensions/pi-web-access/fetch-params.ts +111 -111
  61. package/agent/extensions/pi-web-access/gemini-adc.ts +298 -298
  62. package/agent/extensions/pi-web-access/gemini-api.ts +353 -353
  63. package/agent/extensions/pi-web-access/gemini-pdf-extract.ts +108 -108
  64. package/agent/extensions/pi-web-access/gemini-url-context.ts +128 -128
  65. package/agent/extensions/pi-web-access/gemini-web-config.ts +101 -101
  66. package/agent/extensions/pi-web-access/gemini-web.ts +487 -487
  67. package/agent/extensions/pi-web-access/github-api.ts +197 -197
  68. package/agent/extensions/pi-web-access/github-extract.ts +746 -746
  69. package/agent/extensions/pi-web-access/github-issue-pr.ts +700 -700
  70. package/agent/extensions/pi-web-access/index.ts +1737 -1737
  71. package/agent/extensions/pi-web-access/package-lock.json +5808 -5808
  72. package/agent/extensions/pi-web-access/package.json +64 -64
  73. package/agent/extensions/pi-web-access/page-query.ts +96 -96
  74. package/agent/extensions/pi-web-access/pdf-extract.ts +409 -409
  75. package/agent/extensions/pi-web-access/promise-try.d.ts +7 -7
  76. package/agent/extensions/pi-web-access/query-rewrite.ts +51 -51
  77. package/agent/extensions/pi-web-access/render-search-error.ts +170 -170
  78. package/agent/extensions/pi-web-access/rsc-extract.ts +338 -338
  79. package/agent/extensions/pi-web-access/source-check.ts +282 -282
  80. package/agent/extensions/pi-web-access/ssrf-protection.ts +526 -526
  81. package/agent/extensions/pi-web-access/storage.ts +521 -521
  82. package/agent/extensions/pi-web-access/summary-model-scope.ts +125 -125
  83. package/agent/extensions/pi-web-access/test/auth-fetch.test.mjs +208 -208
  84. package/agent/extensions/pi-web-access/test/brightdata-unlocker.test.mjs +840 -840
  85. package/agent/extensions/pi-web-access/test/chrome-cookie-extraction.test.mjs +441 -441
  86. package/agent/extensions/pi-web-access/test/config-path.test.mjs +283 -283
  87. package/agent/extensions/pi-web-access/test/content-find.test.mjs +25 -25
  88. package/agent/extensions/pi-web-access/test/credential-source.test.mjs +118 -118
  89. package/agent/extensions/pi-web-access/test/data-uri-sanitize.test.mjs +210 -210
  90. package/agent/extensions/pi-web-access/test/datalab-pdf-extract.test.mjs +552 -552
  91. package/agent/extensions/pi-web-access/test/declared-web-links.test.mjs +212 -212
  92. package/agent/extensions/pi-web-access/test/fetch-answer-storage.test.mjs +40 -40
  93. package/agent/extensions/pi-web-access/test/fetch-cache-storage.test.mjs +334 -334
  94. package/agent/extensions/pi-web-access/test/fetch-content-domain-policy.test.mjs +95 -95
  95. package/agent/extensions/pi-web-access/test/fetch-modes.test.mjs +53 -53
  96. package/agent/extensions/pi-web-access/test/fetch-not-found-guidance.test.mjs +92 -92
  97. package/agent/extensions/pi-web-access/test/fetch-params.test.mjs +86 -86
  98. package/agent/extensions/pi-web-access/test/fetch-render-call.test.mjs +34 -34
  99. package/agent/extensions/pi-web-access/test/fetch-routing.test.mjs +173 -173
  100. package/agent/extensions/pi-web-access/test/gemini-adc-auth.test.mjs +257 -257
  101. package/agent/extensions/pi-web-access/test/gemini-api-transport.test.mjs +170 -170
  102. package/agent/extensions/pi-web-access/test/gemini-pdf-extract.test.mjs +133 -133
  103. package/agent/extensions/pi-web-access/test/gemini-web-cookie-opt-in.test.mjs +178 -178
  104. package/agent/extensions/pi-web-access/test/gemini-web-header-overflow.test.mjs +148 -148
  105. package/agent/extensions/pi-web-access/test/get-search-content.test.mjs +223 -223
  106. package/agent/extensions/pi-web-access/test/github-extract.test.mjs +378 -378
  107. package/agent/extensions/pi-web-access/test/github-issue-pr.test.mjs +565 -565
  108. package/agent/extensions/pi-web-access/test/inline-content-config.test.mjs +99 -99
  109. package/agent/extensions/pi-web-access/test/lazy-extract-load.test.mjs +118 -118
  110. package/agent/extensions/pi-web-access/test/local-video-oversize.test.mjs +52 -52
  111. package/agent/extensions/pi-web-access/test/package-typebox-dependency.test.mjs +50 -50
  112. package/agent/extensions/pi-web-access/test/page-query.test.mjs +51 -51
  113. package/agent/extensions/pi-web-access/test/pdf-config.test.mjs +140 -140
  114. package/agent/extensions/pi-web-access/test/pdf-extract.test.mjs +500 -500
  115. package/agent/extensions/pi-web-access/test/proxy-transport.test.mjs +286 -286
  116. package/agent/extensions/pi-web-access/test/query-rewrite.test.mjs +52 -52
  117. package/agent/extensions/pi-web-access/test/rsc-fallback.test.mjs +102 -102
  118. package/agent/extensions/pi-web-access/test/search-error-render.test.mjs +152 -152
  119. package/agent/extensions/pi-web-access/test/search-providers.test.mjs +274 -274
  120. package/agent/extensions/pi-web-access/test/source-check.test.mjs +179 -179
  121. package/agent/extensions/pi-web-access/test/ssrf-allow-ranges-config.test.mjs +205 -205
  122. package/agent/extensions/pi-web-access/test/ssrf-protection.test.mjs +456 -456
  123. package/agent/extensions/pi-web-access/test/tool-registration-config.test.mjs +182 -182
  124. package/agent/extensions/pi-web-access/test/youtube-extract-errors.test.mjs +64 -64
  125. package/agent/extensions/pi-web-access/tsconfig.json +11 -11
  126. package/agent/extensions/pi-web-access/utils.ts +451 -451
  127. package/agent/extensions/pi-web-access/video-extract.ts +392 -392
  128. package/agent/extensions/pi-web-access/youtube-extract.ts +328 -328
  129. package/agent/extensions/shared/activity-status.ts +31 -31
  130. package/agent/extensions/shared/child-session.test.ts +270 -270
  131. package/agent/extensions/shared/child-session.ts +148 -148
  132. package/agent/extensions/shared/context-utilization.test.ts +48 -48
  133. package/agent/extensions/shared/context-utilization.ts +47 -47
  134. package/agent/extensions/shared/dashboard-state.ts +99 -99
  135. package/agent/extensions/shared/tool-call-timeout.test.ts +117 -117
  136. package/agent/extensions/shared/tool-call-timeout.ts +104 -104
  137. package/agent/extensions/subagents/by-the-way.test.ts +29 -29
  138. package/agent/extensions/subagents/claude.test.ts +119 -119
  139. package/agent/extensions/subagents/codex.test.ts +102 -102
  140. package/agent/extensions/subagents/context-usage.test.ts +107 -107
  141. package/agent/extensions/subagents/docs/design-plan.md +568 -568
  142. package/agent/extensions/subagents/docs/effect-v4-extension-guide.md +354 -354
  143. package/agent/extensions/subagents/docs/effect-v4-notes.md +571 -571
  144. package/agent/extensions/subagents/index.ts +779 -779
  145. package/agent/extensions/subagents/manager.test.ts +276 -276
  146. package/agent/extensions/subagents/package-lock.json +2244 -2244
  147. package/agent/extensions/subagents/package.json +19 -19
  148. package/agent/extensions/subagents/result-delivery.test.ts +27 -27
  149. package/agent/extensions/subagents/src/backend.ts +73 -73
  150. package/agent/extensions/subagents/src/backends/claude.ts +701 -701
  151. package/agent/extensions/subagents/src/backends/codex.ts +1060 -1060
  152. package/agent/extensions/subagents/src/backends/pi.ts +575 -575
  153. package/agent/extensions/subagents/src/backends/stub.ts +300 -300
  154. package/agent/extensions/subagents/src/by-the-way.ts +21 -21
  155. package/agent/extensions/subagents/src/domain.ts +253 -253
  156. package/agent/extensions/subagents/src/format.ts +74 -74
  157. package/agent/extensions/subagents/src/manager.ts +736 -736
  158. package/agent/extensions/subagents/src/prompt.ts +92 -92
  159. package/agent/extensions/subagents/src/result-delivery.ts +20 -20
  160. package/agent/extensions/subagents/src/runtime.ts +53 -53
  161. package/agent/extensions/subagents/src/ui/takeover.ts +583 -583
  162. package/agent/extensions/subagents/src/ui/transcript.ts +201 -201
  163. package/agent/extensions/subagents/takeover.test.ts +29 -29
  164. package/agent/extensions/subagents/tsconfig.json +7 -7
  165. package/agent/extensions/todo/AGENTS.md +38 -38
  166. package/agent/extensions/todo/LICENSE +21 -21
  167. package/agent/extensions/todo/config.ts +55 -55
  168. package/agent/extensions/todo/index.ts +151 -151
  169. package/agent/extensions/todo/locales/de.json +17 -17
  170. package/agent/extensions/todo/locales/en.json +15 -15
  171. package/agent/extensions/todo/locales/es.json +17 -17
  172. package/agent/extensions/todo/locales/fr.json +17 -17
  173. package/agent/extensions/todo/locales/pt-BR.json +17 -17
  174. package/agent/extensions/todo/locales/pt.json +17 -17
  175. package/agent/extensions/todo/locales/ru.json +17 -17
  176. package/agent/extensions/todo/locales/uk.json +17 -17
  177. package/agent/extensions/todo/locales/zh.json +17 -17
  178. package/agent/extensions/todo/package-lock.json +3358 -3358
  179. package/agent/extensions/todo/package.json +67 -67
  180. package/agent/extensions/todo/state/i18n-bridge.ts +64 -64
  181. package/agent/extensions/todo/state/invariants.ts +20 -20
  182. package/agent/extensions/todo/state/replay.ts +38 -38
  183. package/agent/extensions/todo/state/selectors.ts +107 -107
  184. package/agent/extensions/todo/state/state-reducer.ts +326 -326
  185. package/agent/extensions/todo/state/state.ts +18 -18
  186. package/agent/extensions/todo/state/store.ts +82 -82
  187. package/agent/extensions/todo/state/task-graph.ts +57 -57
  188. package/agent/extensions/todo/todo-overlay.ts +200 -200
  189. package/agent/extensions/todo/todo.ts +155 -155
  190. package/agent/extensions/todo/tool/response-envelope.ts +109 -109
  191. package/agent/extensions/todo/tool/types.ts +206 -206
  192. package/agent/extensions/todo/view/format.ts +177 -177
  193. package/agent/install.ps1 +637 -527
  194. package/agent/install.sh +620 -511
  195. package/agent/keybindings.json +7 -7
  196. package/bin/kiln.js +124 -11
  197. package/package.json +8 -2
  198. package/agent/extensions/taste/index.ts +0 -443
  199. package/agent/extensions/taste/install.ps1 +0 -23
  200. package/agent/extensions/taste/install.sh +0 -21
  201. /package/agent/extensions/{status line → statusline}/index.ts +0 -0
  202. /package/agent/extensions/{status line → statusline}/install.ps1 +0 -0
  203. /package/agent/extensions/{status line → statusline}/install.sh +0 -0
@@ -1,575 +1,575 @@
1
- /**
2
- * pi backend — real implementation over the pi SDK.
3
- *
4
- * Each subagent is an in-process `AgentSession` (a port of v1
5
- * subagents/manager.ts + shared/child-session.ts):
6
- * - real session files visible in /resume, child resources loaded per-cwd
7
- * with trust gating, and the child tool denylist;
8
- * - `session.subscribe()` events translated to normalized SubagentEvents;
9
- * - send() steers a streaming run or starts a fresh prompt() when idle;
10
- * - interrupt clears the queue and aborts; closing the session scope emits
11
- * the child session_shutdown hook and disposes the session.
12
- */
13
-
14
- import type { AssistantMessage, Message, Model } from "@earendil-works/pi-ai";
15
- import type {
16
- AgentSession,
17
- AgentSessionEvent,
18
- ModelRegistry,
19
- } from "@earendil-works/pi-coding-agent";
20
- import {
21
- createAgentSession,
22
- DefaultResourceLoader,
23
- getAgentDir,
24
- SessionManager,
25
- SettingsManager,
26
- } from "@earendil-works/pi-coding-agent";
27
- import type { Cause, Scope } from "effect";
28
- import { Effect, Queue, Stream } from "effect";
29
- import type { SubagentBackend, SubagentSession } from "../backend.ts";
30
- import type {
31
- SpawnTask,
32
- SubagentEvent,
33
- SubagentMeta,
34
- TranscriptPart,
35
- } from "../domain.ts";
36
- import { SendError, SpawnError } from "../domain.ts";
37
- import { createToolCallTimeoutGuard } from "../../../shared/tool-call-timeout.ts";
38
-
39
- const CHILD_SHUTDOWN_TIMEOUT_MS = 5_000;
40
-
41
- /** Tools that headless children must not receive. Everything else stays enabled. */
42
- const CHILD_EXCLUDED_TOOL_NAMES = [
43
- "subagent_spawn",
44
- "subagent_wait",
45
- "subagent_cancel",
46
- "subagent_check",
47
- "subagent_list",
48
- "workflow",
49
- "ask_user",
50
- ] as const;
51
-
52
- // --- Model + effort resolution -----------------------------------------------
53
-
54
- type ThinkingLevel = NonNullable<
55
- NonNullable<Parameters<typeof createAgentSession>[0]>["thinkingLevel"]
56
- >;
57
-
58
- /**
59
- * Resolve the generic model hint against the parent registry (v1 semantics):
60
- * "provider/model-id" is exact; a bare id prefers the inherited provider,
61
- * then must be unambiguous across providers. No hint inherits the parent
62
- * model; with nothing to inherit, the SDK default applies.
63
- */
64
- function resolvePiModel(
65
- registry: ModelRegistry,
66
- hint: string | undefined,
67
- inherited: { provider: string; id: string } | undefined,
68
- ): Model<any> | undefined {
69
- if (!hint) {
70
- if (!inherited) return undefined;
71
- return registry.find(inherited.provider, inherited.id) ?? undefined;
72
- }
73
- const slash = hint.indexOf("/");
74
- if (slash > 0) {
75
- const provider = hint.slice(0, slash);
76
- const id = hint.slice(slash + 1);
77
- const found = registry.find(provider, id);
78
- if (found) return found;
79
- throw new Error(`Unknown model "${hint}".`);
80
- }
81
- if (inherited) {
82
- const found = registry.find(inherited.provider, hint);
83
- if (found) return found;
84
- }
85
- const matches = registry.getAll().filter((m) => m.id === hint);
86
- if (matches.length === 1) return matches[0];
87
- if (matches.length > 1) {
88
- throw new Error(
89
- `Model "${hint}" exists in multiple providers (${matches.map((m) => m.provider).join(", ")}). Use "provider/${hint}".`,
90
- );
91
- }
92
- throw new Error(`Unknown model "${hint}".`);
93
- }
94
-
95
- // --- Child session helpers (ported from v1 shared/child-session.ts) -----------
96
-
97
- /** Load normal global/package resources and trust-gated project resources. */
98
- async function createChildResources(cwd: string, projectTrusted: boolean) {
99
- const agentDir = getAgentDir();
100
- const settingsManager = SettingsManager.create(cwd, agentDir, {
101
- projectTrusted,
102
- });
103
- const loader = new DefaultResourceLoader({ cwd, agentDir, settingsManager });
104
- await loader.reload();
105
- return { loader, settingsManager };
106
- }
107
-
108
- function waitBounded(operation: Promise<unknown>, timeoutMs: number) {
109
- let timer: ReturnType<typeof setTimeout> | undefined;
110
- const timeout = new Promise<void>((resolve) => {
111
- timer = setTimeout(resolve, timeoutMs);
112
- });
113
- return Promise.race([
114
- operation.then(
115
- () => undefined,
116
- () => undefined,
117
- ),
118
- timeout,
119
- ])
120
- .catch(() => {})
121
- .finally(() => {
122
- if (timer) clearTimeout(timer);
123
- });
124
- }
125
-
126
- /** Emit child session_shutdown (bounded), then dispose. Never throws. */
127
- async function shutdownAndDisposeChildSession(session: AgentSession) {
128
- try {
129
- if (session.extensionRunner.hasHandlers("session_shutdown")) {
130
- await waitBounded(
131
- session.extensionRunner.emit({
132
- type: "session_shutdown",
133
- reason: "quit",
134
- }),
135
- CHILD_SHUTDOWN_TIMEOUT_MS,
136
- );
137
- }
138
- } catch {
139
- // Extension runner inspection/emission is best-effort during teardown.
140
- } finally {
141
- try {
142
- session.dispose();
143
- } catch {
144
- // Disposal is terminal and must remain idempotent for callers.
145
- }
146
- }
147
- }
148
-
149
- // --- Event translation ----------------------------------------------------------
150
-
151
- function messageRole(msg: unknown): Message["role"] | undefined {
152
- const role = (msg as { role?: string } | undefined)?.role;
153
- if (role === "user" || role === "assistant" || role === "toolResult")
154
- return role;
155
- return undefined;
156
- }
157
-
158
- function lastAssistantMessage(
159
- session: AgentSession,
160
- ): AssistantMessage | undefined {
161
- const messages = session.messages;
162
- for (let i = messages.length - 1; i >= 0; i--) {
163
- const msg = messages[i];
164
- if (messageRole(msg) === "assistant") return msg as AssistantMessage;
165
- }
166
- return undefined;
167
- }
168
-
169
- /** Final assistant text output (last assistant message with text), v1 semantics. */
170
- function finalOutput(session: AgentSession): string {
171
- const messages = session.messages;
172
- for (let i = messages.length - 1; i >= 0; i--) {
173
- const msg = messages[i];
174
- if (messageRole(msg) !== "assistant") continue;
175
- const text = (msg as AssistantMessage).content
176
- .filter((part) => part.type === "text")
177
- .map((part) => part.text)
178
- .join("\n")
179
- .trim();
180
- if (text) return text;
181
- }
182
- return "";
183
- }
184
-
185
- function safeJson(value: unknown): string | undefined {
186
- try {
187
- const text = JSON.stringify(value);
188
- return text === "{}" ? undefined : text.slice(0, 4_096);
189
- } catch {
190
- return undefined;
191
- }
192
- }
193
-
194
- /** First non-empty line of a tool result-ish value (v1 liveToolPreview). */
195
- function toolPreview(value: unknown): string | undefined {
196
- if (typeof value === "string") {
197
- return value
198
- .split("\n")
199
- .find((line) => line.trim())
200
- ?.trim();
201
- }
202
- if (!value || typeof value !== "object") return undefined;
203
- const content = (value as { content?: unknown }).content;
204
- if (!Array.isArray(content)) return undefined;
205
- for (const part of content) {
206
- if (!part || typeof part !== "object") continue;
207
- const record = part as { type?: unknown; text?: unknown };
208
- if (record.type !== "text" || typeof record.text !== "string") continue;
209
- const firstLine = record.text.split("\n").find((line) => line.trim());
210
- if (firstLine) return firstLine.trim();
211
- }
212
- return undefined;
213
- }
214
-
215
- function assistantParts(msg: AssistantMessage): TranscriptPart[] {
216
- const parts: TranscriptPart[] = [];
217
- for (const part of msg.content) {
218
- if (part.type === "text") {
219
- parts.push({ type: "text", text: part.text });
220
- } else if (part.type === "thinking") {
221
- parts.push({
222
- type: "thinking",
223
- text: part.redacted ? "" : part.thinking,
224
- redacted: part.redacted,
225
- });
226
- } else if (part.type === "toolCall") {
227
- parts.push({
228
- type: "toolCall",
229
- toolId: part.id,
230
- name: part.name,
231
- argsPreview: safeJson(part.arguments),
232
- });
233
- }
234
- }
235
- return parts;
236
- }
237
-
238
- function userText(msg: Message): string {
239
- const content = (msg as { content: unknown }).content;
240
- if (typeof content === "string") return content;
241
- if (!Array.isArray(content)) return "";
242
- return content
243
- .filter(
244
- (part): part is { type: "text"; text: string } =>
245
- !!part &&
246
- typeof part === "object" &&
247
- (part as { type?: unknown }).type === "text",
248
- )
249
- .map((part) => part.text)
250
- .join("\n");
251
- }
252
-
253
- // --- The session ------------------------------------------------------------------
254
-
255
- function boundedError(error: unknown) {
256
- return (error instanceof Error ? error.message : String(error)).slice(
257
- 0,
258
- 4096,
259
- );
260
- }
261
-
262
- const makePiSession = (
263
- task: SpawnTask,
264
- ): Effect.Effect<SubagentSession, SpawnError, Scope.Scope> =>
265
- Effect.gen(function* () {
266
- const registry = task.parent.modelRegistry;
267
- if (!registry) {
268
- return yield* new SpawnError({
269
- message: "pi backend requires the parent session's model registry.",
270
- });
271
- }
272
-
273
- const model = yield* Effect.try({
274
- try: () =>
275
- resolvePiModel(registry, task.model, task.parent.inheritedModel),
276
- catch: (error) => new SpawnError({ message: boundedError(error) }),
277
- });
278
- // pi's thinking levels ARE the shared reasoning-effort scale.
279
- const thinkingLevel = (task.reasoningEffort ??
280
- task.parent.inheritedThinkingLevel) as ThinkingLevel | undefined;
281
-
282
- const session = yield* Effect.tryPromise({
283
- try: async () => {
284
- const { loader, settingsManager } = await createChildResources(
285
- task.cwd,
286
- task.parent.projectTrusted,
287
- );
288
- const { session } = await createAgentSession({
289
- cwd: task.cwd,
290
- sessionManager: SessionManager.create(task.cwd),
291
- settingsManager,
292
- resourceLoader: loader,
293
- model,
294
- thinkingLevel,
295
- excludeTools: [...CHILD_EXCLUDED_TOOL_NAMES],
296
- });
297
- // Start child extension session hooks/resources in headless mode.
298
- // A rejection here would otherwise leak the freshly created session:
299
- // the scope finalizer that owns cleanup is only registered later.
300
- try {
301
- await session.bindExtensions({ mode: "print" });
302
- } catch (error) {
303
- await shutdownAndDisposeChildSession(session);
304
- throw error;
305
- }
306
- return session;
307
- },
308
- catch: (error) => new SpawnError({ message: boundedError(error) }),
309
- });
310
-
311
- const state = {
312
- closed: false,
313
- /** prompt() rejection for the active run; folded into RunSettled. */
314
- runError: undefined as string | undefined,
315
- /** One terminal event per run: lifecycle, prompt-rejection, and abort
316
- * fallbacks can all race to settle; the first wins. */
317
- settled: false,
318
- };
319
-
320
- const events = yield* Queue.make<SubagentEvent, Cause.Done>();
321
- const emit = (event: SubagentEvent) => {
322
- Queue.offerUnsafe(events, event);
323
- };
324
-
325
- const toolTimeout = createToolCallTimeoutGuard();
326
- toolTimeout.apply(session);
327
-
328
- const activeModel = (): Model<any> | undefined => {
329
- const sessionModel = session.model;
330
- const last = lastAssistantMessage(session);
331
- if (!last) return sessionModel;
332
- if (
333
- sessionModel &&
334
- (last.provider !== sessionModel.provider ||
335
- last.model !== sessionModel.id)
336
- ) {
337
- // The session changed models after this assistant response.
338
- return sessionModel;
339
- }
340
- return (
341
- registry.find(last.provider, last.responseModel ?? last.model) ??
342
- sessionModel
343
- );
344
- };
345
-
346
- const currentMeta = (): SubagentMeta => {
347
- const m = activeModel();
348
- return {
349
- backend: "pi",
350
- modelLabel: m ? `${m.provider}/${m.id}` : undefined,
351
- contextWindow: m?.contextWindow,
352
- sessionFilePath: session.sessionFile,
353
- };
354
- };
355
-
356
- const emitUsage = () => {
357
- const usage = session.getContextUsage();
358
- emit({
359
- _tag: "UsageChanged",
360
- tokens: usage?.tokens ?? undefined,
361
- contextWindow: activeModel()?.contextWindow ?? usage?.contextWindow,
362
- });
363
- };
364
-
365
- const settle = () => {
366
- if (state.settled) return;
367
- state.settled = true;
368
- const last = lastAssistantMessage(session);
369
- const partialText = finalOutput(session) || undefined;
370
- if (last?.stopReason === "aborted") {
371
- emit({
372
- _tag: "RunSettled",
373
- outcome: { _tag: "Interrupted", partialText },
374
- });
375
- return;
376
- }
377
- const errorText =
378
- state.runError ??
379
- (last?.stopReason === "error"
380
- ? (last.errorMessage ?? "Run failed")
381
- : undefined);
382
- if (errorText !== undefined) {
383
- emit({
384
- _tag: "RunSettled",
385
- outcome: {
386
- _tag: "Failed",
387
- errorText: boundedError(errorText),
388
- partialText,
389
- },
390
- });
391
- return;
392
- }
393
- emit({
394
- _tag: "RunSettled",
395
- outcome: { _tag: "Completed", finalText: finalOutput(session) },
396
- });
397
- };
398
-
399
- const handleEvent = (event: AgentSessionEvent) => {
400
- if (state.closed) return;
401
- switch (event.type) {
402
- case "agent_start":
403
- // Extensions may register tools between runs; guard new ones too.
404
- toolTimeout.apply(session);
405
- state.settled = false;
406
- emit({ _tag: "RunStarted" });
407
- break;
408
- case "message_update": {
409
- const streamEvent = event.assistantMessageEvent;
410
- if (streamEvent.type === "text_delta") {
411
- emit({
412
- _tag: "AssistantDelta",
413
- kind: "text",
414
- delta: streamEvent.delta,
415
- });
416
- } else if (streamEvent.type === "thinking_delta") {
417
- emit({
418
- _tag: "AssistantDelta",
419
- kind: "thinking",
420
- delta: streamEvent.delta,
421
- });
422
- }
423
- break;
424
- }
425
- case "message_end": {
426
- const role = messageRole(event.message);
427
- if (role === "user") {
428
- const text = userText(event.message as Message);
429
- if (text.trim()) emit({ _tag: "UserMessage", text });
430
- } else if (role === "assistant") {
431
- emit({
432
- _tag: "AssistantMessage",
433
- parts: assistantParts(event.message as AssistantMessage),
434
- });
435
- emitUsage();
436
- emit({ _tag: "MetaChanged", meta: currentMeta() });
437
- }
438
- // toolResult messages are covered by tool_execution_end.
439
- break;
440
- }
441
- case "tool_execution_start":
442
- emit({
443
- _tag: "ToolStart",
444
- toolId: event.toolCallId,
445
- name: event.toolName,
446
- argsPreview: safeJson(event.args),
447
- });
448
- break;
449
- case "tool_execution_update":
450
- emit({
451
- _tag: "ToolUpdate",
452
- toolId: event.toolCallId,
453
- outputPreview: toolPreview(event.partialResult),
454
- });
455
- break;
456
- case "tool_execution_end":
457
- emit({
458
- _tag: "ToolEnd",
459
- toolId: event.toolCallId,
460
- name: event.toolName,
461
- isError: event.isError,
462
- outputPreview: toolPreview(event.result),
463
- });
464
- break;
465
- case "queue_update":
466
- emit({
467
- _tag: "QueueChanged",
468
- queued: [
469
- ...event.steering.map((text) => ({
470
- text,
471
- kind: "steer" as const,
472
- })),
473
- ...event.followUp.map((text) => ({
474
- text,
475
- kind: "follow-up" as const,
476
- })),
477
- ],
478
- });
479
- break;
480
- case "agent_settled":
481
- settle();
482
- break;
483
- }
484
- };
485
- const unsubscribe = session.subscribe(handleEvent);
486
-
487
- yield* Effect.addFinalizer(() =>
488
- Effect.promise(async () => {
489
- state.closed = true;
490
- unsubscribe();
491
- try {
492
- session.clearQueue();
493
- } catch {
494
- // Continue with abort/dispose.
495
- }
496
- await waitBounded(session.abort(), CHILD_SHUTDOWN_TIMEOUT_MS);
497
- await shutdownAndDisposeChildSession(session);
498
- Queue.endUnsafe(events);
499
- }),
500
- );
501
-
502
- /** Start a fresh run (v1 manager.run): fire-and-forget, errors -> events. */
503
- const startRun = (text: string) => {
504
- state.runError = undefined;
505
- state.settled = false;
506
- emit({ _tag: "RunStarted" });
507
- void session.prompt(text).catch((error) => {
508
- state.runError = boundedError(error);
509
- // Preflight failures may never start the agent lifecycle, so no
510
- // agent_settled will arrive for them.
511
- if (!session.isStreaming) settle();
512
- });
513
- };
514
-
515
- // Session naming is best-effort.
516
- yield* Effect.try(() =>
517
- session.sessionManager.appendSessionInfo(
518
- `${task.origin === "btw" ? "btw" : "subagent"}: ${task.title}`,
519
- ),
520
- ).pipe(Effect.ignore);
521
-
522
- emit({ _tag: "MetaChanged", meta: currentMeta() });
523
- startRun(task.prompt);
524
-
525
- return {
526
- meta: Effect.sync(currentMeta),
527
- events: Stream.fromQueue(events),
528
- send: (text) =>
529
- Effect.suspend((): Effect.Effect<void, SendError> => {
530
- if (state.closed) {
531
- return new SendError({ message: "Subagent session is closed." });
532
- }
533
- if (session.isStreaming) {
534
- // Steer the active run via the SDK's queue; queue_update events
535
- // render it, message_end(user) lands it in the transcript. A
536
- // rejected steer is a real send failure, not a diagnostic.
537
- return Effect.tryPromise({
538
- try: () => session.steer(text),
539
- catch: (error) => new SendError({ message: boundedError(error) }),
540
- }).pipe(Effect.asVoid);
541
- }
542
- return Effect.sync(() => startRun(text));
543
- }),
544
- interrupt: Effect.promise(async () => {
545
- if (state.closed) return;
546
- try {
547
- session.clearQueue();
548
- } catch {
549
- // Abort regardless.
550
- }
551
- await session.abort().catch(() => undefined);
552
- // Only resolve once streaming has actually stopped: reporting the
553
- // interrupt as complete while the run keeps working would let the
554
- // manager settle a run that is still mutating the workspace. The
555
- // manager bounds this effect at 5s and force-disposes on timeout.
556
- while (!state.closed && session.isStreaming) {
557
- await new Promise((resolve) => setTimeout(resolve, 50));
558
- }
559
- // No streaming run means no agent_settled will arrive; emit the
560
- // terminal event (once) so the run cannot look running forever.
561
- if (!state.closed && !state.settled) {
562
- state.settled = true;
563
- emit({ _tag: "RunSettled", outcome: { _tag: "Interrupted" } });
564
- }
565
- }),
566
- } satisfies SubagentSession;
567
- });
568
-
569
- export const piBackend: SubagentBackend = {
570
- name: "pi",
571
- capabilities: { steering: true, modelSelection: true, reasoningEffort: true },
572
- // In-process SDK: always available.
573
- available: Effect.succeed(true),
574
- spawn: makePiSession,
575
- };
1
+ /**
2
+ * pi backend — real implementation over the pi SDK.
3
+ *
4
+ * Each subagent is an in-process `AgentSession` (a port of v1
5
+ * subagents/manager.ts + shared/child-session.ts):
6
+ * - real session files visible in /resume, child resources loaded per-cwd
7
+ * with trust gating, and the child tool denylist;
8
+ * - `session.subscribe()` events translated to normalized SubagentEvents;
9
+ * - send() steers a streaming run or starts a fresh prompt() when idle;
10
+ * - interrupt clears the queue and aborts; closing the session scope emits
11
+ * the child session_shutdown hook and disposes the session.
12
+ */
13
+
14
+ import type { AssistantMessage, Message, Model } from "@earendil-works/pi-ai";
15
+ import type {
16
+ AgentSession,
17
+ AgentSessionEvent,
18
+ ModelRegistry,
19
+ } from "@earendil-works/pi-coding-agent";
20
+ import {
21
+ createAgentSession,
22
+ DefaultResourceLoader,
23
+ getAgentDir,
24
+ SessionManager,
25
+ SettingsManager,
26
+ } from "@earendil-works/pi-coding-agent";
27
+ import type { Cause, Scope } from "effect";
28
+ import { Effect, Queue, Stream } from "effect";
29
+ import type { SubagentBackend, SubagentSession } from "../backend.ts";
30
+ import type {
31
+ SpawnTask,
32
+ SubagentEvent,
33
+ SubagentMeta,
34
+ TranscriptPart,
35
+ } from "../domain.ts";
36
+ import { SendError, SpawnError } from "../domain.ts";
37
+ import { createToolCallTimeoutGuard } from "../../../shared/tool-call-timeout.ts";
38
+
39
+ const CHILD_SHUTDOWN_TIMEOUT_MS = 5_000;
40
+
41
+ /** Tools that headless children must not receive. Everything else stays enabled. */
42
+ const CHILD_EXCLUDED_TOOL_NAMES = [
43
+ "subagent_spawn",
44
+ "subagent_wait",
45
+ "subagent_cancel",
46
+ "subagent_check",
47
+ "subagent_list",
48
+ "workflow",
49
+ "ask_user",
50
+ ] as const;
51
+
52
+ // --- Model + effort resolution -----------------------------------------------
53
+
54
+ type ThinkingLevel = NonNullable<
55
+ NonNullable<Parameters<typeof createAgentSession>[0]>["thinkingLevel"]
56
+ >;
57
+
58
+ /**
59
+ * Resolve the generic model hint against the parent registry (v1 semantics):
60
+ * "provider/model-id" is exact; a bare id prefers the inherited provider,
61
+ * then must be unambiguous across providers. No hint inherits the parent
62
+ * model; with nothing to inherit, the SDK default applies.
63
+ */
64
+ function resolvePiModel(
65
+ registry: ModelRegistry,
66
+ hint: string | undefined,
67
+ inherited: { provider: string; id: string } | undefined,
68
+ ): Model<any> | undefined {
69
+ if (!hint) {
70
+ if (!inherited) return undefined;
71
+ return registry.find(inherited.provider, inherited.id) ?? undefined;
72
+ }
73
+ const slash = hint.indexOf("/");
74
+ if (slash > 0) {
75
+ const provider = hint.slice(0, slash);
76
+ const id = hint.slice(slash + 1);
77
+ const found = registry.find(provider, id);
78
+ if (found) return found;
79
+ throw new Error(`Unknown model "${hint}".`);
80
+ }
81
+ if (inherited) {
82
+ const found = registry.find(inherited.provider, hint);
83
+ if (found) return found;
84
+ }
85
+ const matches = registry.getAll().filter((m) => m.id === hint);
86
+ if (matches.length === 1) return matches[0];
87
+ if (matches.length > 1) {
88
+ throw new Error(
89
+ `Model "${hint}" exists in multiple providers (${matches.map((m) => m.provider).join(", ")}). Use "provider/${hint}".`,
90
+ );
91
+ }
92
+ throw new Error(`Unknown model "${hint}".`);
93
+ }
94
+
95
+ // --- Child session helpers (ported from v1 shared/child-session.ts) -----------
96
+
97
+ /** Load normal global/package resources and trust-gated project resources. */
98
+ async function createChildResources(cwd: string, projectTrusted: boolean) {
99
+ const agentDir = getAgentDir();
100
+ const settingsManager = SettingsManager.create(cwd, agentDir, {
101
+ projectTrusted,
102
+ });
103
+ const loader = new DefaultResourceLoader({ cwd, agentDir, settingsManager });
104
+ await loader.reload();
105
+ return { loader, settingsManager };
106
+ }
107
+
108
+ function waitBounded(operation: Promise<unknown>, timeoutMs: number) {
109
+ let timer: ReturnType<typeof setTimeout> | undefined;
110
+ const timeout = new Promise<void>((resolve) => {
111
+ timer = setTimeout(resolve, timeoutMs);
112
+ });
113
+ return Promise.race([
114
+ operation.then(
115
+ () => undefined,
116
+ () => undefined,
117
+ ),
118
+ timeout,
119
+ ])
120
+ .catch(() => {})
121
+ .finally(() => {
122
+ if (timer) clearTimeout(timer);
123
+ });
124
+ }
125
+
126
+ /** Emit child session_shutdown (bounded), then dispose. Never throws. */
127
+ async function shutdownAndDisposeChildSession(session: AgentSession) {
128
+ try {
129
+ if (session.extensionRunner.hasHandlers("session_shutdown")) {
130
+ await waitBounded(
131
+ session.extensionRunner.emit({
132
+ type: "session_shutdown",
133
+ reason: "quit",
134
+ }),
135
+ CHILD_SHUTDOWN_TIMEOUT_MS,
136
+ );
137
+ }
138
+ } catch {
139
+ // Extension runner inspection/emission is best-effort during teardown.
140
+ } finally {
141
+ try {
142
+ session.dispose();
143
+ } catch {
144
+ // Disposal is terminal and must remain idempotent for callers.
145
+ }
146
+ }
147
+ }
148
+
149
+ // --- Event translation ----------------------------------------------------------
150
+
151
+ function messageRole(msg: unknown): Message["role"] | undefined {
152
+ const role = (msg as { role?: string } | undefined)?.role;
153
+ if (role === "user" || role === "assistant" || role === "toolResult")
154
+ return role;
155
+ return undefined;
156
+ }
157
+
158
+ function lastAssistantMessage(
159
+ session: AgentSession,
160
+ ): AssistantMessage | undefined {
161
+ const messages = session.messages;
162
+ for (let i = messages.length - 1; i >= 0; i--) {
163
+ const msg = messages[i];
164
+ if (messageRole(msg) === "assistant") return msg as AssistantMessage;
165
+ }
166
+ return undefined;
167
+ }
168
+
169
+ /** Final assistant text output (last assistant message with text), v1 semantics. */
170
+ function finalOutput(session: AgentSession): string {
171
+ const messages = session.messages;
172
+ for (let i = messages.length - 1; i >= 0; i--) {
173
+ const msg = messages[i];
174
+ if (messageRole(msg) !== "assistant") continue;
175
+ const text = (msg as AssistantMessage).content
176
+ .filter((part) => part.type === "text")
177
+ .map((part) => part.text)
178
+ .join("\n")
179
+ .trim();
180
+ if (text) return text;
181
+ }
182
+ return "";
183
+ }
184
+
185
+ function safeJson(value: unknown): string | undefined {
186
+ try {
187
+ const text = JSON.stringify(value);
188
+ return text === "{}" ? undefined : text.slice(0, 4_096);
189
+ } catch {
190
+ return undefined;
191
+ }
192
+ }
193
+
194
+ /** First non-empty line of a tool result-ish value (v1 liveToolPreview). */
195
+ function toolPreview(value: unknown): string | undefined {
196
+ if (typeof value === "string") {
197
+ return value
198
+ .split("\n")
199
+ .find((line) => line.trim())
200
+ ?.trim();
201
+ }
202
+ if (!value || typeof value !== "object") return undefined;
203
+ const content = (value as { content?: unknown }).content;
204
+ if (!Array.isArray(content)) return undefined;
205
+ for (const part of content) {
206
+ if (!part || typeof part !== "object") continue;
207
+ const record = part as { type?: unknown; text?: unknown };
208
+ if (record.type !== "text" || typeof record.text !== "string") continue;
209
+ const firstLine = record.text.split("\n").find((line) => line.trim());
210
+ if (firstLine) return firstLine.trim();
211
+ }
212
+ return undefined;
213
+ }
214
+
215
+ function assistantParts(msg: AssistantMessage): TranscriptPart[] {
216
+ const parts: TranscriptPart[] = [];
217
+ for (const part of msg.content) {
218
+ if (part.type === "text") {
219
+ parts.push({ type: "text", text: part.text });
220
+ } else if (part.type === "thinking") {
221
+ parts.push({
222
+ type: "thinking",
223
+ text: part.redacted ? "" : part.thinking,
224
+ redacted: part.redacted,
225
+ });
226
+ } else if (part.type === "toolCall") {
227
+ parts.push({
228
+ type: "toolCall",
229
+ toolId: part.id,
230
+ name: part.name,
231
+ argsPreview: safeJson(part.arguments),
232
+ });
233
+ }
234
+ }
235
+ return parts;
236
+ }
237
+
238
+ function userText(msg: Message): string {
239
+ const content = (msg as { content: unknown }).content;
240
+ if (typeof content === "string") return content;
241
+ if (!Array.isArray(content)) return "";
242
+ return content
243
+ .filter(
244
+ (part): part is { type: "text"; text: string } =>
245
+ !!part &&
246
+ typeof part === "object" &&
247
+ (part as { type?: unknown }).type === "text",
248
+ )
249
+ .map((part) => part.text)
250
+ .join("\n");
251
+ }
252
+
253
+ // --- The session ------------------------------------------------------------------
254
+
255
+ function boundedError(error: unknown) {
256
+ return (error instanceof Error ? error.message : String(error)).slice(
257
+ 0,
258
+ 4096,
259
+ );
260
+ }
261
+
262
+ const makePiSession = (
263
+ task: SpawnTask,
264
+ ): Effect.Effect<SubagentSession, SpawnError, Scope.Scope> =>
265
+ Effect.gen(function* () {
266
+ const registry = task.parent.modelRegistry;
267
+ if (!registry) {
268
+ return yield* new SpawnError({
269
+ message: "pi backend requires the parent session's model registry.",
270
+ });
271
+ }
272
+
273
+ const model = yield* Effect.try({
274
+ try: () =>
275
+ resolvePiModel(registry, task.model, task.parent.inheritedModel),
276
+ catch: (error) => new SpawnError({ message: boundedError(error) }),
277
+ });
278
+ // pi's thinking levels ARE the shared reasoning-effort scale.
279
+ const thinkingLevel = (task.reasoningEffort ??
280
+ task.parent.inheritedThinkingLevel) as ThinkingLevel | undefined;
281
+
282
+ const session = yield* Effect.tryPromise({
283
+ try: async () => {
284
+ const { loader, settingsManager } = await createChildResources(
285
+ task.cwd,
286
+ task.parent.projectTrusted,
287
+ );
288
+ const { session } = await createAgentSession({
289
+ cwd: task.cwd,
290
+ sessionManager: SessionManager.create(task.cwd),
291
+ settingsManager,
292
+ resourceLoader: loader,
293
+ model,
294
+ thinkingLevel,
295
+ excludeTools: [...CHILD_EXCLUDED_TOOL_NAMES],
296
+ });
297
+ // Start child extension session hooks/resources in headless mode.
298
+ // A rejection here would otherwise leak the freshly created session:
299
+ // the scope finalizer that owns cleanup is only registered later.
300
+ try {
301
+ await session.bindExtensions({ mode: "print" });
302
+ } catch (error) {
303
+ await shutdownAndDisposeChildSession(session);
304
+ throw error;
305
+ }
306
+ return session;
307
+ },
308
+ catch: (error) => new SpawnError({ message: boundedError(error) }),
309
+ });
310
+
311
+ const state = {
312
+ closed: false,
313
+ /** prompt() rejection for the active run; folded into RunSettled. */
314
+ runError: undefined as string | undefined,
315
+ /** One terminal event per run: lifecycle, prompt-rejection, and abort
316
+ * fallbacks can all race to settle; the first wins. */
317
+ settled: false,
318
+ };
319
+
320
+ const events = yield* Queue.make<SubagentEvent, Cause.Done>();
321
+ const emit = (event: SubagentEvent) => {
322
+ Queue.offerUnsafe(events, event);
323
+ };
324
+
325
+ const toolTimeout = createToolCallTimeoutGuard();
326
+ toolTimeout.apply(session);
327
+
328
+ const activeModel = (): Model<any> | undefined => {
329
+ const sessionModel = session.model;
330
+ const last = lastAssistantMessage(session);
331
+ if (!last) return sessionModel;
332
+ if (
333
+ sessionModel &&
334
+ (last.provider !== sessionModel.provider ||
335
+ last.model !== sessionModel.id)
336
+ ) {
337
+ // The session changed models after this assistant response.
338
+ return sessionModel;
339
+ }
340
+ return (
341
+ registry.find(last.provider, last.responseModel ?? last.model) ??
342
+ sessionModel
343
+ );
344
+ };
345
+
346
+ const currentMeta = (): SubagentMeta => {
347
+ const m = activeModel();
348
+ return {
349
+ backend: "pi",
350
+ modelLabel: m ? `${m.provider}/${m.id}` : undefined,
351
+ contextWindow: m?.contextWindow,
352
+ sessionFilePath: session.sessionFile,
353
+ };
354
+ };
355
+
356
+ const emitUsage = () => {
357
+ const usage = session.getContextUsage();
358
+ emit({
359
+ _tag: "UsageChanged",
360
+ tokens: usage?.tokens ?? undefined,
361
+ contextWindow: activeModel()?.contextWindow ?? usage?.contextWindow,
362
+ });
363
+ };
364
+
365
+ const settle = () => {
366
+ if (state.settled) return;
367
+ state.settled = true;
368
+ const last = lastAssistantMessage(session);
369
+ const partialText = finalOutput(session) || undefined;
370
+ if (last?.stopReason === "aborted") {
371
+ emit({
372
+ _tag: "RunSettled",
373
+ outcome: { _tag: "Interrupted", partialText },
374
+ });
375
+ return;
376
+ }
377
+ const errorText =
378
+ state.runError ??
379
+ (last?.stopReason === "error"
380
+ ? (last.errorMessage ?? "Run failed")
381
+ : undefined);
382
+ if (errorText !== undefined) {
383
+ emit({
384
+ _tag: "RunSettled",
385
+ outcome: {
386
+ _tag: "Failed",
387
+ errorText: boundedError(errorText),
388
+ partialText,
389
+ },
390
+ });
391
+ return;
392
+ }
393
+ emit({
394
+ _tag: "RunSettled",
395
+ outcome: { _tag: "Completed", finalText: finalOutput(session) },
396
+ });
397
+ };
398
+
399
+ const handleEvent = (event: AgentSessionEvent) => {
400
+ if (state.closed) return;
401
+ switch (event.type) {
402
+ case "agent_start":
403
+ // Extensions may register tools between runs; guard new ones too.
404
+ toolTimeout.apply(session);
405
+ state.settled = false;
406
+ emit({ _tag: "RunStarted" });
407
+ break;
408
+ case "message_update": {
409
+ const streamEvent = event.assistantMessageEvent;
410
+ if (streamEvent.type === "text_delta") {
411
+ emit({
412
+ _tag: "AssistantDelta",
413
+ kind: "text",
414
+ delta: streamEvent.delta,
415
+ });
416
+ } else if (streamEvent.type === "thinking_delta") {
417
+ emit({
418
+ _tag: "AssistantDelta",
419
+ kind: "thinking",
420
+ delta: streamEvent.delta,
421
+ });
422
+ }
423
+ break;
424
+ }
425
+ case "message_end": {
426
+ const role = messageRole(event.message);
427
+ if (role === "user") {
428
+ const text = userText(event.message as Message);
429
+ if (text.trim()) emit({ _tag: "UserMessage", text });
430
+ } else if (role === "assistant") {
431
+ emit({
432
+ _tag: "AssistantMessage",
433
+ parts: assistantParts(event.message as AssistantMessage),
434
+ });
435
+ emitUsage();
436
+ emit({ _tag: "MetaChanged", meta: currentMeta() });
437
+ }
438
+ // toolResult messages are covered by tool_execution_end.
439
+ break;
440
+ }
441
+ case "tool_execution_start":
442
+ emit({
443
+ _tag: "ToolStart",
444
+ toolId: event.toolCallId,
445
+ name: event.toolName,
446
+ argsPreview: safeJson(event.args),
447
+ });
448
+ break;
449
+ case "tool_execution_update":
450
+ emit({
451
+ _tag: "ToolUpdate",
452
+ toolId: event.toolCallId,
453
+ outputPreview: toolPreview(event.partialResult),
454
+ });
455
+ break;
456
+ case "tool_execution_end":
457
+ emit({
458
+ _tag: "ToolEnd",
459
+ toolId: event.toolCallId,
460
+ name: event.toolName,
461
+ isError: event.isError,
462
+ outputPreview: toolPreview(event.result),
463
+ });
464
+ break;
465
+ case "queue_update":
466
+ emit({
467
+ _tag: "QueueChanged",
468
+ queued: [
469
+ ...event.steering.map((text) => ({
470
+ text,
471
+ kind: "steer" as const,
472
+ })),
473
+ ...event.followUp.map((text) => ({
474
+ text,
475
+ kind: "follow-up" as const,
476
+ })),
477
+ ],
478
+ });
479
+ break;
480
+ case "agent_settled":
481
+ settle();
482
+ break;
483
+ }
484
+ };
485
+ const unsubscribe = session.subscribe(handleEvent);
486
+
487
+ yield* Effect.addFinalizer(() =>
488
+ Effect.promise(async () => {
489
+ state.closed = true;
490
+ unsubscribe();
491
+ try {
492
+ session.clearQueue();
493
+ } catch {
494
+ // Continue with abort/dispose.
495
+ }
496
+ await waitBounded(session.abort(), CHILD_SHUTDOWN_TIMEOUT_MS);
497
+ await shutdownAndDisposeChildSession(session);
498
+ Queue.endUnsafe(events);
499
+ }),
500
+ );
501
+
502
+ /** Start a fresh run (v1 manager.run): fire-and-forget, errors -> events. */
503
+ const startRun = (text: string) => {
504
+ state.runError = undefined;
505
+ state.settled = false;
506
+ emit({ _tag: "RunStarted" });
507
+ void session.prompt(text).catch((error) => {
508
+ state.runError = boundedError(error);
509
+ // Preflight failures may never start the agent lifecycle, so no
510
+ // agent_settled will arrive for them.
511
+ if (!session.isStreaming) settle();
512
+ });
513
+ };
514
+
515
+ // Session naming is best-effort.
516
+ yield* Effect.try(() =>
517
+ session.sessionManager.appendSessionInfo(
518
+ `${task.origin === "btw" ? "btw" : "subagent"}: ${task.title}`,
519
+ ),
520
+ ).pipe(Effect.ignore);
521
+
522
+ emit({ _tag: "MetaChanged", meta: currentMeta() });
523
+ startRun(task.prompt);
524
+
525
+ return {
526
+ meta: Effect.sync(currentMeta),
527
+ events: Stream.fromQueue(events),
528
+ send: (text) =>
529
+ Effect.suspend((): Effect.Effect<void, SendError> => {
530
+ if (state.closed) {
531
+ return new SendError({ message: "Subagent session is closed." });
532
+ }
533
+ if (session.isStreaming) {
534
+ // Steer the active run via the SDK's queue; queue_update events
535
+ // render it, message_end(user) lands it in the transcript. A
536
+ // rejected steer is a real send failure, not a diagnostic.
537
+ return Effect.tryPromise({
538
+ try: () => session.steer(text),
539
+ catch: (error) => new SendError({ message: boundedError(error) }),
540
+ }).pipe(Effect.asVoid);
541
+ }
542
+ return Effect.sync(() => startRun(text));
543
+ }),
544
+ interrupt: Effect.promise(async () => {
545
+ if (state.closed) return;
546
+ try {
547
+ session.clearQueue();
548
+ } catch {
549
+ // Abort regardless.
550
+ }
551
+ await session.abort().catch(() => undefined);
552
+ // Only resolve once streaming has actually stopped: reporting the
553
+ // interrupt as complete while the run keeps working would let the
554
+ // manager settle a run that is still mutating the workspace. The
555
+ // manager bounds this effect at 5s and force-disposes on timeout.
556
+ while (!state.closed && session.isStreaming) {
557
+ await new Promise((resolve) => setTimeout(resolve, 50));
558
+ }
559
+ // No streaming run means no agent_settled will arrive; emit the
560
+ // terminal event (once) so the run cannot look running forever.
561
+ if (!state.closed && !state.settled) {
562
+ state.settled = true;
563
+ emit({ _tag: "RunSettled", outcome: { _tag: "Interrupted" } });
564
+ }
565
+ }),
566
+ } satisfies SubagentSession;
567
+ });
568
+
569
+ export const piBackend: SubagentBackend = {
570
+ name: "pi",
571
+ capabilities: { steering: true, modelSelection: true, reasoningEffort: true },
572
+ // In-process SDK: always available.
573
+ available: Effect.succeed(true),
574
+ spawn: makePiSession,
575
+ };