@asterxsk/kiln 0.4.0 → 0.4.2

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 (259) hide show
  1. package/README.md +155 -155
  2. package/agent/AGENTS.md +67 -67
  3. package/agent/README.md +5 -5
  4. package/agent/extensions/ask-user/index.ts +418 -418
  5. package/agent/extensions/ask-user/install.ps1 +27 -27
  6. package/agent/extensions/ask-user/install.sh +24 -24
  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/install.ps1 +27 -27
  14. package/agent/extensions/background-terminals/install.sh +24 -24
  15. package/agent/extensions/background-terminals/manager.test.ts +735 -735
  16. package/agent/extensions/background-terminals/output.test.ts +109 -109
  17. package/agent/extensions/background-terminals/package-lock.json +769 -769
  18. package/agent/extensions/background-terminals/package.json +17 -17
  19. package/agent/extensions/background-terminals/prompt.test.ts +125 -125
  20. package/agent/extensions/background-terminals/ps.test.ts +82 -82
  21. package/agent/extensions/background-terminals/result-delivery.test.ts +44 -44
  22. package/agent/extensions/background-terminals/src/domain.ts +87 -87
  23. package/agent/extensions/background-terminals/src/manager.ts +907 -907
  24. package/agent/extensions/background-terminals/src/output.ts +84 -84
  25. package/agent/extensions/background-terminals/src/prompt.ts +142 -142
  26. package/agent/extensions/background-terminals/src/result-delivery.ts +27 -27
  27. package/agent/extensions/background-terminals/src/runtime.ts +36 -36
  28. package/agent/extensions/background-terminals/src/ui/output-view.ts +79 -79
  29. package/agent/extensions/background-terminals/src/ui/ps.ts +621 -621
  30. package/agent/extensions/background-terminals/tsconfig.json +7 -7
  31. package/agent/extensions/destructive/README.md +31 -31
  32. package/agent/extensions/destructive/index.ts +88 -88
  33. package/agent/extensions/destructive/install.ps1 +27 -27
  34. package/agent/extensions/destructive/install.sh +24 -24
  35. package/agent/extensions/file-search/index.spec.ts +443 -443
  36. package/agent/extensions/file-search/index.ts +459 -459
  37. package/agent/extensions/file-search/install.ps1 +27 -27
  38. package/agent/extensions/file-search/install.sh +24 -24
  39. package/agent/extensions/file-search/package-lock.json +2253 -2253
  40. package/agent/extensions/file-search/package.json +23 -23
  41. package/agent/extensions/file-search/src/args.ts +122 -122
  42. package/agent/extensions/file-search/src/binaries.ts +422 -422
  43. package/agent/extensions/file-search/src/output.ts +126 -126
  44. package/agent/extensions/file-search/src/process.ts +146 -146
  45. package/agent/extensions/file-search/src/prompt.ts +52 -52
  46. package/agent/extensions/file-search/tsconfig.json +7 -7
  47. package/agent/extensions/goal/README.md +50 -50
  48. package/agent/extensions/goal/index.ts +155 -155
  49. package/agent/extensions/goal/install.ps1 +27 -27
  50. package/agent/extensions/goal/install.sh +24 -24
  51. package/agent/extensions/kiln-update/README.md +38 -25
  52. package/agent/extensions/kiln-update/index.ts +177 -99
  53. package/agent/extensions/modelconf/PLAN.md +915 -915
  54. package/agent/extensions/modelconf/README.md +66 -66
  55. package/agent/extensions/modelconf/index.ts +296 -296
  56. package/agent/extensions/modelconf/install.ps1 +27 -27
  57. package/agent/extensions/modelconf/install.sh +24 -24
  58. package/agent/extensions/modelconf/package-lock.json +1809 -1809
  59. package/agent/extensions/modelconf/package.json +13 -13
  60. package/agent/extensions/modelconf/src/fuzzy.ts +26 -26
  61. package/agent/extensions/modelconf/src/glob.ts +13 -13
  62. package/agent/extensions/modelconf/src/persistence.test.ts +46 -46
  63. package/agent/extensions/modelconf/src/persistence.ts +164 -164
  64. package/agent/extensions/modelconf/src/ui/ModelConfView.test.ts +75 -75
  65. package/agent/extensions/modelconf/src/ui/ModelConfView.ts +1101 -1101
  66. package/agent/extensions/modelconf/tsconfig.json +15 -15
  67. package/agent/extensions/pi-web-access/CHANGELOG.md +690 -690
  68. package/agent/extensions/pi-web-access/LICENSE +21 -21
  69. package/agent/extensions/pi-web-access/README.md +470 -470
  70. package/agent/extensions/pi-web-access/SECURITY.md +5 -5
  71. package/agent/extensions/pi-web-access/activity.ts +101 -101
  72. package/agent/extensions/pi-web-access/auth-fetch.ts +148 -148
  73. package/agent/extensions/pi-web-access/brightdata-unlocker.ts +272 -272
  74. package/agent/extensions/pi-web-access/chrome-cookies.ts +669 -669
  75. package/agent/extensions/pi-web-access/content-find.ts +139 -139
  76. package/agent/extensions/pi-web-access/credential-source.ts +191 -191
  77. package/agent/extensions/pi-web-access/data-uri-sanitize.ts +406 -406
  78. package/agent/extensions/pi-web-access/datalab-pdf-extract.ts +568 -568
  79. package/agent/extensions/pi-web-access/declared-web-links.ts +173 -173
  80. package/agent/extensions/pi-web-access/evidence/CONTRACT-EVIDENCE.md +496 -496
  81. package/agent/extensions/pi-web-access/evidence/contract-probe.mjs +140 -140
  82. package/agent/extensions/pi-web-access/exa.ts +526 -526
  83. package/agent/extensions/pi-web-access/extract.ts +1196 -1196
  84. package/agent/extensions/pi-web-access/feature-config.ts +29 -29
  85. package/agent/extensions/pi-web-access/fetch-params.ts +111 -111
  86. package/agent/extensions/pi-web-access/gemini-adc.ts +298 -298
  87. package/agent/extensions/pi-web-access/gemini-api.ts +353 -353
  88. package/agent/extensions/pi-web-access/gemini-pdf-extract.ts +108 -108
  89. package/agent/extensions/pi-web-access/gemini-search.ts +21 -21
  90. package/agent/extensions/pi-web-access/gemini-url-context.ts +128 -128
  91. package/agent/extensions/pi-web-access/gemini-web-config.ts +101 -101
  92. package/agent/extensions/pi-web-access/gemini-web.ts +487 -487
  93. package/agent/extensions/pi-web-access/github-api.ts +197 -197
  94. package/agent/extensions/pi-web-access/github-extract.ts +746 -746
  95. package/agent/extensions/pi-web-access/github-issue-pr.ts +700 -700
  96. package/agent/extensions/pi-web-access/index.ts +1737 -1737
  97. package/agent/extensions/pi-web-access/package-lock.json +5808 -5808
  98. package/agent/extensions/pi-web-access/package.json +64 -64
  99. package/agent/extensions/pi-web-access/page-query.ts +96 -96
  100. package/agent/extensions/pi-web-access/pdf-extract.ts +409 -409
  101. package/agent/extensions/pi-web-access/promise-try.d.ts +7 -7
  102. package/agent/extensions/pi-web-access/query-rewrite.ts +51 -51
  103. package/agent/extensions/pi-web-access/render-search-error.ts +170 -170
  104. package/agent/extensions/pi-web-access/rsc-extract.ts +338 -338
  105. package/agent/extensions/pi-web-access/search-types.ts +20 -20
  106. package/agent/extensions/pi-web-access/source-check.ts +282 -282
  107. package/agent/extensions/pi-web-access/ssrf-protection.ts +526 -526
  108. package/agent/extensions/pi-web-access/storage.ts +521 -521
  109. package/agent/extensions/pi-web-access/summary-model-scope.ts +125 -125
  110. package/agent/extensions/pi-web-access/test/auth-fetch.test.mjs +208 -208
  111. package/agent/extensions/pi-web-access/test/brightdata-unlocker.test.mjs +840 -840
  112. package/agent/extensions/pi-web-access/test/chrome-cookie-extraction.test.mjs +441 -441
  113. package/agent/extensions/pi-web-access/test/config-path.test.mjs +283 -283
  114. package/agent/extensions/pi-web-access/test/content-find.test.mjs +25 -25
  115. package/agent/extensions/pi-web-access/test/credential-source.test.mjs +118 -118
  116. package/agent/extensions/pi-web-access/test/data-uri-sanitize.test.mjs +210 -210
  117. package/agent/extensions/pi-web-access/test/datalab-pdf-extract.test.mjs +552 -552
  118. package/agent/extensions/pi-web-access/test/declared-web-links.test.mjs +212 -212
  119. package/agent/extensions/pi-web-access/test/fetch-answer-storage.test.mjs +40 -40
  120. package/agent/extensions/pi-web-access/test/fetch-cache-storage.test.mjs +334 -334
  121. package/agent/extensions/pi-web-access/test/fetch-content-domain-policy.test.mjs +95 -95
  122. package/agent/extensions/pi-web-access/test/fetch-modes.test.mjs +53 -53
  123. package/agent/extensions/pi-web-access/test/fetch-not-found-guidance.test.mjs +92 -92
  124. package/agent/extensions/pi-web-access/test/fetch-params.test.mjs +86 -86
  125. package/agent/extensions/pi-web-access/test/fetch-render-call.test.mjs +34 -34
  126. package/agent/extensions/pi-web-access/test/fetch-routing.test.mjs +173 -173
  127. package/agent/extensions/pi-web-access/test/gemini-adc-auth.test.mjs +257 -257
  128. package/agent/extensions/pi-web-access/test/gemini-api-transport.test.mjs +170 -170
  129. package/agent/extensions/pi-web-access/test/gemini-pdf-extract.test.mjs +133 -133
  130. package/agent/extensions/pi-web-access/test/gemini-web-cookie-opt-in.test.mjs +178 -178
  131. package/agent/extensions/pi-web-access/test/gemini-web-header-overflow.test.mjs +148 -148
  132. package/agent/extensions/pi-web-access/test/get-search-content.test.mjs +223 -223
  133. package/agent/extensions/pi-web-access/test/github-extract.test.mjs +378 -378
  134. package/agent/extensions/pi-web-access/test/github-issue-pr.test.mjs +565 -565
  135. package/agent/extensions/pi-web-access/test/inline-content-config.test.mjs +99 -99
  136. package/agent/extensions/pi-web-access/test/lazy-extract-load.test.mjs +118 -118
  137. package/agent/extensions/pi-web-access/test/local-video-oversize.test.mjs +52 -52
  138. package/agent/extensions/pi-web-access/test/package-typebox-dependency.test.mjs +50 -50
  139. package/agent/extensions/pi-web-access/test/page-query.test.mjs +51 -51
  140. package/agent/extensions/pi-web-access/test/pdf-config.test.mjs +140 -140
  141. package/agent/extensions/pi-web-access/test/pdf-extract.test.mjs +500 -500
  142. package/agent/extensions/pi-web-access/test/proxy-transport.test.mjs +286 -286
  143. package/agent/extensions/pi-web-access/test/query-rewrite.test.mjs +52 -52
  144. package/agent/extensions/pi-web-access/test/rsc-fallback.test.mjs +102 -102
  145. package/agent/extensions/pi-web-access/test/search-error-render.test.mjs +152 -152
  146. package/agent/extensions/pi-web-access/test/search-providers.test.mjs +274 -274
  147. package/agent/extensions/pi-web-access/test/source-check.test.mjs +179 -179
  148. package/agent/extensions/pi-web-access/test/ssrf-allow-ranges-config.test.mjs +205 -205
  149. package/agent/extensions/pi-web-access/test/ssrf-protection.test.mjs +456 -456
  150. package/agent/extensions/pi-web-access/test/summary-model-scope.test.mjs +106 -106
  151. package/agent/extensions/pi-web-access/test/tool-registration-config.test.mjs +182 -182
  152. package/agent/extensions/pi-web-access/test/web-search-answer-render.test.mjs +66 -66
  153. package/agent/extensions/pi-web-access/test/youtube-extract-errors.test.mjs +64 -64
  154. package/agent/extensions/pi-web-access/tsconfig.json +11 -11
  155. package/agent/extensions/pi-web-access/utils.ts +451 -451
  156. package/agent/extensions/pi-web-access/video-extract.ts +392 -392
  157. package/agent/extensions/pi-web-access/youtube-extract.ts +328 -328
  158. package/agent/extensions/shared/activity-status.ts +31 -31
  159. package/agent/extensions/shared/child-session.test.ts +270 -270
  160. package/agent/extensions/shared/child-session.ts +148 -148
  161. package/agent/extensions/shared/context-utilization.test.ts +48 -48
  162. package/agent/extensions/shared/context-utilization.ts +47 -47
  163. package/agent/extensions/shared/dashboard-state.ts +99 -99
  164. package/agent/extensions/shared/install.ps1 +27 -27
  165. package/agent/extensions/shared/install.sh +24 -24
  166. package/agent/extensions/shared/tool-call-timeout.test.ts +117 -117
  167. package/agent/extensions/shared/tool-call-timeout.ts +104 -104
  168. package/agent/extensions/skillsconf/README.md +74 -74
  169. package/agent/extensions/skillsconf/index.ts +92 -92
  170. package/agent/extensions/skillsconf/install.ps1 +27 -27
  171. package/agent/extensions/skillsconf/install.sh +24 -24
  172. package/agent/extensions/skillsconf/package-lock.json +1809 -1809
  173. package/agent/extensions/skillsconf/package.json +18 -18
  174. package/agent/extensions/skillsconf/src/delete-skill.test.ts +64 -64
  175. package/agent/extensions/skillsconf/src/delete-skill.ts +45 -45
  176. package/agent/extensions/skillsconf/src/filter.test.ts +80 -80
  177. package/agent/extensions/skillsconf/src/filter.ts +74 -74
  178. package/agent/extensions/skillsconf/src/fuzzy.ts +26 -26
  179. package/agent/extensions/skillsconf/src/persistence.test.ts +72 -72
  180. package/agent/extensions/skillsconf/src/persistence.ts +169 -169
  181. package/agent/extensions/skillsconf/src/ui/SkillConfView.test.ts +337 -337
  182. package/agent/extensions/skillsconf/src/ui/SkillConfView.ts +758 -758
  183. package/agent/extensions/skillsconf/src/ui/text-input.ts +91 -91
  184. package/agent/extensions/skillsconf/src/ui/tui-helpers.ts +144 -144
  185. package/agent/extensions/skillsconf/tsconfig.json +15 -15
  186. package/agent/extensions/statusline/index.ts +273 -282
  187. package/agent/extensions/statusline/install.ps1 +27 -27
  188. package/agent/extensions/statusline/install.sh +24 -24
  189. package/agent/extensions/subagents/by-the-way.test.ts +29 -29
  190. package/agent/extensions/subagents/claude.test.ts +119 -119
  191. package/agent/extensions/subagents/codex.test.ts +102 -102
  192. package/agent/extensions/subagents/context-usage.test.ts +107 -107
  193. package/agent/extensions/subagents/docs/design-plan.md +568 -568
  194. package/agent/extensions/subagents/docs/effect-v4-extension-guide.md +354 -354
  195. package/agent/extensions/subagents/docs/effect-v4-notes.md +571 -571
  196. package/agent/extensions/subagents/index.ts +779 -779
  197. package/agent/extensions/subagents/install.ps1 +27 -27
  198. package/agent/extensions/subagents/install.sh +24 -24
  199. package/agent/extensions/subagents/manager.test.ts +276 -276
  200. package/agent/extensions/subagents/package-lock.json +2244 -2244
  201. package/agent/extensions/subagents/package.json +19 -19
  202. package/agent/extensions/subagents/result-delivery.test.ts +27 -27
  203. package/agent/extensions/subagents/src/backend.ts +73 -73
  204. package/agent/extensions/subagents/src/backends/claude.ts +701 -701
  205. package/agent/extensions/subagents/src/backends/codex.ts +1060 -1060
  206. package/agent/extensions/subagents/src/backends/pi.ts +575 -575
  207. package/agent/extensions/subagents/src/backends/stub.ts +300 -300
  208. package/agent/extensions/subagents/src/by-the-way.ts +21 -21
  209. package/agent/extensions/subagents/src/domain.ts +253 -253
  210. package/agent/extensions/subagents/src/format.ts +74 -74
  211. package/agent/extensions/subagents/src/manager.ts +736 -736
  212. package/agent/extensions/subagents/src/prompt.ts +92 -92
  213. package/agent/extensions/subagents/src/result-delivery.ts +20 -20
  214. package/agent/extensions/subagents/src/runtime.ts +53 -53
  215. package/agent/extensions/subagents/src/ui/takeover.ts +583 -583
  216. package/agent/extensions/subagents/src/ui/transcript.ts +201 -201
  217. package/agent/extensions/subagents/takeover.test.ts +29 -29
  218. package/agent/extensions/subagents/tsconfig.json +7 -7
  219. package/agent/extensions/todo/AGENTS.md +38 -38
  220. package/agent/extensions/todo/LICENSE +21 -21
  221. package/agent/extensions/todo/config.ts +55 -55
  222. package/agent/extensions/todo/index.ts +151 -151
  223. package/agent/extensions/todo/install.ps1 +27 -27
  224. package/agent/extensions/todo/install.sh +24 -24
  225. package/agent/extensions/todo/locales/de.json +17 -17
  226. package/agent/extensions/todo/locales/en.json +15 -15
  227. package/agent/extensions/todo/locales/es.json +17 -17
  228. package/agent/extensions/todo/locales/fr.json +17 -17
  229. package/agent/extensions/todo/locales/pt-BR.json +17 -17
  230. package/agent/extensions/todo/locales/pt.json +17 -17
  231. package/agent/extensions/todo/locales/ru.json +17 -17
  232. package/agent/extensions/todo/locales/uk.json +17 -17
  233. package/agent/extensions/todo/locales/zh.json +17 -17
  234. package/agent/extensions/todo/package-lock.json +3358 -3358
  235. package/agent/extensions/todo/package.json +67 -67
  236. package/agent/extensions/todo/state/i18n-bridge.ts +64 -64
  237. package/agent/extensions/todo/state/invariants.ts +20 -20
  238. package/agent/extensions/todo/state/replay.ts +38 -38
  239. package/agent/extensions/todo/state/selectors.ts +107 -107
  240. package/agent/extensions/todo/state/state-reducer.ts +326 -326
  241. package/agent/extensions/todo/state/state.ts +18 -18
  242. package/agent/extensions/todo/state/store.ts +82 -82
  243. package/agent/extensions/todo/state/task-graph.ts +57 -57
  244. package/agent/extensions/todo/todo-overlay.ts +200 -200
  245. package/agent/extensions/todo/todo.ts +155 -155
  246. package/agent/extensions/todo/tool/response-envelope.ts +109 -109
  247. package/agent/extensions/todo/tool/types.ts +206 -206
  248. package/agent/extensions/todo/view/format.ts +177 -177
  249. package/agent/extensions/trim-context/README.md +54 -54
  250. package/agent/extensions/trim-context/index.ts +487 -487
  251. package/agent/extensions/trim-context/install.ps1 +27 -27
  252. package/agent/extensions/trim-context/install.sh +24 -24
  253. package/agent/keybindings.json +7 -7
  254. package/agent/version.txt +1 -1
  255. package/bin/kiln.js +20 -13
  256. package/package.json +1 -1
  257. package/agent/extensions/taste/index.ts +0 -443
  258. package/agent/extensions/taste/install.ps1 +0 -27
  259. package/agent/extensions/taste/install.sh +0 -24
@@ -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
+ };