@wahack/pi-coding-agent 15.11.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 (1152) hide show
  1. package/CHANGELOG.md +10031 -0
  2. package/README.md +36 -0
  3. package/examples/README.md +21 -0
  4. package/examples/custom-tools/README.md +104 -0
  5. package/examples/custom-tools/hello/index.ts +20 -0
  6. package/examples/extensions/README.md +142 -0
  7. package/examples/extensions/api-demo.ts +79 -0
  8. package/examples/extensions/chalk-logger.ts +25 -0
  9. package/examples/extensions/hello.ts +31 -0
  10. package/examples/extensions/pirate.ts +43 -0
  11. package/examples/extensions/plan-mode.ts +549 -0
  12. package/examples/extensions/reload-runtime.ts +38 -0
  13. package/examples/extensions/thinking-note.ts +13 -0
  14. package/examples/extensions/tools.ts +145 -0
  15. package/examples/extensions/with-deps/index.ts +36 -0
  16. package/examples/extensions/with-deps/package-lock.json +31 -0
  17. package/examples/extensions/with-deps/package.json +17 -0
  18. package/examples/hooks/README.md +56 -0
  19. package/examples/hooks/auto-commit-on-exit.ts +48 -0
  20. package/examples/hooks/confirm-destructive.ts +58 -0
  21. package/examples/hooks/custom-compaction.ts +115 -0
  22. package/examples/hooks/dirty-repo-guard.ts +51 -0
  23. package/examples/hooks/file-trigger.ts +40 -0
  24. package/examples/hooks/git-checkpoint.ts +52 -0
  25. package/examples/hooks/handoff.ts +149 -0
  26. package/examples/hooks/permission-gate.ts +33 -0
  27. package/examples/hooks/protected-paths.ts +29 -0
  28. package/examples/hooks/qna.ts +118 -0
  29. package/examples/hooks/status-line.ts +39 -0
  30. package/examples/sdk/01-minimal.ts +21 -0
  31. package/examples/sdk/02-custom-model.ts +49 -0
  32. package/examples/sdk/03-custom-prompt.ts +46 -0
  33. package/examples/sdk/04-skills.ts +43 -0
  34. package/examples/sdk/06-extensions.ts +82 -0
  35. package/examples/sdk/06-hooks.ts +61 -0
  36. package/examples/sdk/07-context-files.ts +35 -0
  37. package/examples/sdk/08-prompt-templates.ts +41 -0
  38. package/examples/sdk/08-slash-commands.ts +46 -0
  39. package/examples/sdk/09-api-keys-and-oauth.ts +54 -0
  40. package/examples/sdk/11-sessions.ts +47 -0
  41. package/examples/sdk/12-redis-sessions.ts +54 -0
  42. package/examples/sdk/13-sql-sessions.ts +61 -0
  43. package/examples/sdk/README.md +172 -0
  44. package/package.json +554 -0
  45. package/scripts/build-binary.ts +100 -0
  46. package/scripts/bundle-dist.ts +90 -0
  47. package/scripts/format-prompts.ts +68 -0
  48. package/scripts/generate-docs-index.ts +40 -0
  49. package/scripts/generate-template.ts +33 -0
  50. package/scripts/omp +42 -0
  51. package/scripts/omp.ts +19 -0
  52. package/src/async/index.ts +1 -0
  53. package/src/async/job-manager.ts +625 -0
  54. package/src/auto-thinking/classifier.ts +185 -0
  55. package/src/autoresearch/command-resume.md +14 -0
  56. package/src/autoresearch/dashboard.ts +436 -0
  57. package/src/autoresearch/git.ts +319 -0
  58. package/src/autoresearch/helpers.ts +218 -0
  59. package/src/autoresearch/index.ts +536 -0
  60. package/src/autoresearch/prompt-setup.md +43 -0
  61. package/src/autoresearch/prompt.md +103 -0
  62. package/src/autoresearch/resume-message.md +10 -0
  63. package/src/autoresearch/state.ts +273 -0
  64. package/src/autoresearch/storage.ts +699 -0
  65. package/src/autoresearch/tools/init-experiment.ts +272 -0
  66. package/src/autoresearch/tools/log-experiment.ts +524 -0
  67. package/src/autoresearch/tools/run-experiment.ts +407 -0
  68. package/src/autoresearch/tools/update-notes.ts +109 -0
  69. package/src/autoresearch/types.ts +168 -0
  70. package/src/bun-imports.d.ts +28 -0
  71. package/src/capability/context-file.ts +44 -0
  72. package/src/capability/extension-module.ts +34 -0
  73. package/src/capability/extension.ts +47 -0
  74. package/src/capability/fs.ts +117 -0
  75. package/src/capability/hook.ts +40 -0
  76. package/src/capability/index.ts +436 -0
  77. package/src/capability/instruction.ts +37 -0
  78. package/src/capability/mcp.ts +74 -0
  79. package/src/capability/prompt.ts +35 -0
  80. package/src/capability/rule-buckets.ts +66 -0
  81. package/src/capability/rule.ts +261 -0
  82. package/src/capability/settings.ts +34 -0
  83. package/src/capability/skill.ts +63 -0
  84. package/src/capability/slash-command.ts +40 -0
  85. package/src/capability/ssh.ts +41 -0
  86. package/src/capability/system-prompt.ts +34 -0
  87. package/src/capability/tool.ts +38 -0
  88. package/src/capability/types.ts +168 -0
  89. package/src/cli/agents-cli.ts +138 -0
  90. package/src/cli/args.ts +340 -0
  91. package/src/cli/auth-broker-cli.ts +895 -0
  92. package/src/cli/auth-gateway-cli.ts +611 -0
  93. package/src/cli/classify-install-target.ts +76 -0
  94. package/src/cli/claude-trace-cli.ts +795 -0
  95. package/src/cli/commands/init-xdg.ts +27 -0
  96. package/src/cli/completion-gen.ts +550 -0
  97. package/src/cli/config-cli.ts +418 -0
  98. package/src/cli/dry-balance-cli.ts +856 -0
  99. package/src/cli/extension-flags.ts +48 -0
  100. package/src/cli/file-processor.ts +133 -0
  101. package/src/cli/gallery-cli.ts +230 -0
  102. package/src/cli/gallery-fixtures/agentic.ts +407 -0
  103. package/src/cli/gallery-fixtures/codeintel.ts +187 -0
  104. package/src/cli/gallery-fixtures/edit.ts +194 -0
  105. package/src/cli/gallery-fixtures/fs.ts +220 -0
  106. package/src/cli/gallery-fixtures/index.ts +40 -0
  107. package/src/cli/gallery-fixtures/interaction.ts +49 -0
  108. package/src/cli/gallery-fixtures/memory.ts +81 -0
  109. package/src/cli/gallery-fixtures/misc.ts +250 -0
  110. package/src/cli/gallery-fixtures/search.ts +213 -0
  111. package/src/cli/gallery-fixtures/shell.ts +167 -0
  112. package/src/cli/gallery-fixtures/types.ts +57 -0
  113. package/src/cli/gallery-fixtures/web.ts +158 -0
  114. package/src/cli/gallery-screenshot.ts +279 -0
  115. package/src/cli/grep-cli.ts +160 -0
  116. package/src/cli/grievances-cli.ts +256 -0
  117. package/src/cli/initial-message.ts +58 -0
  118. package/src/cli/list-models.ts +194 -0
  119. package/src/cli/plugin-cli.ts +996 -0
  120. package/src/cli/read-cli.ts +57 -0
  121. package/src/cli/session-picker.ts +79 -0
  122. package/src/cli/setup-cli.ts +231 -0
  123. package/src/cli/shell-cli.ts +176 -0
  124. package/src/cli/ssh-cli.ts +179 -0
  125. package/src/cli/startup-cwd.ts +68 -0
  126. package/src/cli/stats-cli.ts +238 -0
  127. package/src/cli/tiny-models-cli.ts +127 -0
  128. package/src/cli/update-cli.ts +611 -0
  129. package/src/cli/usage-cli.ts +603 -0
  130. package/src/cli/web-search-cli.ts +132 -0
  131. package/src/cli/worktree-cli.ts +291 -0
  132. package/src/cli-commands.ts +79 -0
  133. package/src/cli.ts +200 -0
  134. package/src/commands/acp.ts +24 -0
  135. package/src/commands/agents.ts +57 -0
  136. package/src/commands/auth-broker.ts +99 -0
  137. package/src/commands/auth-gateway.ts +69 -0
  138. package/src/commands/commit.ts +46 -0
  139. package/src/commands/complete.ts +66 -0
  140. package/src/commands/completions.ts +60 -0
  141. package/src/commands/config.ts +51 -0
  142. package/src/commands/dry-balance.ts +43 -0
  143. package/src/commands/gallery.ts +52 -0
  144. package/src/commands/grep.ts +48 -0
  145. package/src/commands/grievances.ts +51 -0
  146. package/src/commands/install.ts +107 -0
  147. package/src/commands/launch.ts +169 -0
  148. package/src/commands/plugin.ts +78 -0
  149. package/src/commands/read.ts +38 -0
  150. package/src/commands/setup.ts +67 -0
  151. package/src/commands/shell.ts +29 -0
  152. package/src/commands/ssh.ts +60 -0
  153. package/src/commands/stats.ts +29 -0
  154. package/src/commands/tiny-models.ts +36 -0
  155. package/src/commands/update.ts +21 -0
  156. package/src/commands/usage.ts +35 -0
  157. package/src/commands/web-search.ts +42 -0
  158. package/src/commands/worktree.ts +56 -0
  159. package/src/commit/agentic/agent.ts +317 -0
  160. package/src/commit/agentic/fallback.ts +96 -0
  161. package/src/commit/agentic/index.ts +355 -0
  162. package/src/commit/agentic/prompts/analyze-file.md +22 -0
  163. package/src/commit/agentic/prompts/session-user.md +25 -0
  164. package/src/commit/agentic/prompts/split-confirm.md +1 -0
  165. package/src/commit/agentic/prompts/system.md +38 -0
  166. package/src/commit/agentic/state.ts +60 -0
  167. package/src/commit/agentic/tools/analyze-file.ts +146 -0
  168. package/src/commit/agentic/tools/git-file-diff.ts +191 -0
  169. package/src/commit/agentic/tools/git-hunk.ts +50 -0
  170. package/src/commit/agentic/tools/git-overview.ts +81 -0
  171. package/src/commit/agentic/tools/index.ts +54 -0
  172. package/src/commit/agentic/tools/propose-changelog.ts +144 -0
  173. package/src/commit/agentic/tools/propose-commit.ts +109 -0
  174. package/src/commit/agentic/tools/recent-commits.ts +81 -0
  175. package/src/commit/agentic/tools/schemas.ts +23 -0
  176. package/src/commit/agentic/tools/split-commit.ts +245 -0
  177. package/src/commit/agentic/topo-sort.ts +44 -0
  178. package/src/commit/agentic/trivial.ts +51 -0
  179. package/src/commit/agentic/validation.ts +183 -0
  180. package/src/commit/analysis/conventional.ts +64 -0
  181. package/src/commit/analysis/index.ts +4 -0
  182. package/src/commit/analysis/scope.ts +242 -0
  183. package/src/commit/analysis/summary.ts +105 -0
  184. package/src/commit/analysis/validation.ts +66 -0
  185. package/src/commit/changelog/detect.ts +40 -0
  186. package/src/commit/changelog/generate.ts +97 -0
  187. package/src/commit/changelog/index.ts +234 -0
  188. package/src/commit/changelog/parse.ts +44 -0
  189. package/src/commit/cli.ts +85 -0
  190. package/src/commit/git/diff.ts +148 -0
  191. package/src/commit/index.ts +5 -0
  192. package/src/commit/map-reduce/index.ts +69 -0
  193. package/src/commit/map-reduce/map-phase.ts +193 -0
  194. package/src/commit/map-reduce/reduce-phase.ts +49 -0
  195. package/src/commit/map-reduce/utils.ts +9 -0
  196. package/src/commit/message.ts +11 -0
  197. package/src/commit/model-selection.ts +92 -0
  198. package/src/commit/pipeline.ts +243 -0
  199. package/src/commit/prompts/analysis-system.md +148 -0
  200. package/src/commit/prompts/analysis-user.md +38 -0
  201. package/src/commit/prompts/changelog-system.md +50 -0
  202. package/src/commit/prompts/changelog-user.md +18 -0
  203. package/src/commit/prompts/file-observer-system.md +24 -0
  204. package/src/commit/prompts/file-observer-user.md +8 -0
  205. package/src/commit/prompts/reduce-system.md +50 -0
  206. package/src/commit/prompts/reduce-user.md +17 -0
  207. package/src/commit/prompts/summary-retry.md +3 -0
  208. package/src/commit/prompts/summary-system.md +38 -0
  209. package/src/commit/prompts/summary-user.md +13 -0
  210. package/src/commit/prompts/types-description.md +2 -0
  211. package/src/commit/shared-llm.ts +77 -0
  212. package/src/commit/types.ts +118 -0
  213. package/src/commit/utils/exclusions.ts +42 -0
  214. package/src/commit/utils.ts +58 -0
  215. package/src/config/api-key-resolver.ts +60 -0
  216. package/src/config/append-only-context-mode.ts +31 -0
  217. package/src/config/config-file.ts +317 -0
  218. package/src/config/file-lock.ts +164 -0
  219. package/src/config/keybindings.ts +628 -0
  220. package/src/config/mcp-schema.json +230 -0
  221. package/src/config/model-discovery.ts +554 -0
  222. package/src/config/model-registry.ts +2090 -0
  223. package/src/config/model-resolver.ts +1502 -0
  224. package/src/config/model-roles.ts +74 -0
  225. package/src/config/models-config-schema.ts +226 -0
  226. package/src/config/models-config.ts +129 -0
  227. package/src/config/prompt-templates.ts +185 -0
  228. package/src/config/resolve-config-value.ts +94 -0
  229. package/src/config/settings-schema.ts +3530 -0
  230. package/src/config/settings.ts +1178 -0
  231. package/src/config.ts +242 -0
  232. package/src/cursor.ts +340 -0
  233. package/src/dap/client.ts +760 -0
  234. package/src/dap/config.ts +189 -0
  235. package/src/dap/defaults.json +212 -0
  236. package/src/dap/index.ts +4 -0
  237. package/src/dap/session.ts +1441 -0
  238. package/src/dap/types.ts +610 -0
  239. package/src/debug/index.ts +515 -0
  240. package/src/debug/log-formatting.ts +58 -0
  241. package/src/debug/log-viewer.ts +908 -0
  242. package/src/debug/profiler.ts +162 -0
  243. package/src/debug/protocol-probe.ts +267 -0
  244. package/src/debug/raw-sse-buffer.ts +273 -0
  245. package/src/debug/raw-sse.ts +292 -0
  246. package/src/debug/report-bundle.ts +374 -0
  247. package/src/debug/system-info.ts +111 -0
  248. package/src/debug/terminal-info.ts +124 -0
  249. package/src/discovery/agents-md.ts +67 -0
  250. package/src/discovery/agents.ts +230 -0
  251. package/src/discovery/at-imports.ts +273 -0
  252. package/src/discovery/builtin-defaults.ts +39 -0
  253. package/src/discovery/builtin-rules/index.ts +54 -0
  254. package/src/discovery/builtin-rules/rs-box-leak.md +48 -0
  255. package/src/discovery/builtin-rules/rs-future-prelude.md +23 -0
  256. package/src/discovery/builtin-rules/rs-lazylock.md +51 -0
  257. package/src/discovery/builtin-rules/rs-match-ergonomics.md +67 -0
  258. package/src/discovery/builtin-rules/rs-parking-lot.md +44 -0
  259. package/src/discovery/builtin-rules/rs-result-type.md +19 -0
  260. package/src/discovery/builtin-rules/ts-bare-catch.md +38 -0
  261. package/src/discovery/builtin-rules/ts-import-type.md +42 -0
  262. package/src/discovery/builtin-rules/ts-no-any.md +56 -0
  263. package/src/discovery/builtin-rules/ts-no-deprecated-leftovers.md +44 -0
  264. package/src/discovery/builtin-rules/ts-no-dynamic-import.md +39 -0
  265. package/src/discovery/builtin-rules/ts-no-return-type.md +45 -0
  266. package/src/discovery/builtin-rules/ts-no-test-timers.md +55 -0
  267. package/src/discovery/builtin-rules/ts-no-tiny-functions.md +51 -0
  268. package/src/discovery/builtin-rules/ts-promise-with-resolvers.md +65 -0
  269. package/src/discovery/builtin-rules/ts-redundant-clear-guard.md +75 -0
  270. package/src/discovery/builtin-rules/ts-set-map.md +28 -0
  271. package/src/discovery/builtin.ts +906 -0
  272. package/src/discovery/claude-plugins.ts +386 -0
  273. package/src/discovery/claude.ts +584 -0
  274. package/src/discovery/cline.ts +83 -0
  275. package/src/discovery/codex.ts +522 -0
  276. package/src/discovery/cursor.ts +220 -0
  277. package/src/discovery/gemini.ts +383 -0
  278. package/src/discovery/github.ts +154 -0
  279. package/src/discovery/helpers.ts +1016 -0
  280. package/src/discovery/index.ts +81 -0
  281. package/src/discovery/mcp-json.ts +171 -0
  282. package/src/discovery/omp-extension-roots.ts +190 -0
  283. package/src/discovery/omp-plugins.ts +383 -0
  284. package/src/discovery/opencode.ts +398 -0
  285. package/src/discovery/plugin-dir-roots.ts +28 -0
  286. package/src/discovery/ssh.ts +153 -0
  287. package/src/discovery/substitute-plugin-root.ts +29 -0
  288. package/src/discovery/vscode.ts +105 -0
  289. package/src/discovery/windsurf.ts +147 -0
  290. package/src/edit/apply-patch/index.ts +87 -0
  291. package/src/edit/apply-patch/parser.ts +174 -0
  292. package/src/edit/diff.ts +999 -0
  293. package/src/edit/file-snapshot-store.ts +91 -0
  294. package/src/edit/hashline/block-resolver.ts +33 -0
  295. package/src/edit/hashline/diff.ts +290 -0
  296. package/src/edit/hashline/execute.ts +242 -0
  297. package/src/edit/hashline/filesystem.ts +130 -0
  298. package/src/edit/hashline/index.ts +5 -0
  299. package/src/edit/hashline/noop-loop-guard.ts +99 -0
  300. package/src/edit/hashline/params.ts +18 -0
  301. package/src/edit/index.ts +571 -0
  302. package/src/edit/modes/apply-patch.lark +19 -0
  303. package/src/edit/modes/apply-patch.ts +53 -0
  304. package/src/edit/modes/patch.ts +1891 -0
  305. package/src/edit/modes/replace.ts +1137 -0
  306. package/src/edit/normalize.ts +345 -0
  307. package/src/edit/notebook.ts +242 -0
  308. package/src/edit/read-file.ts +25 -0
  309. package/src/edit/renderer.ts +769 -0
  310. package/src/edit/streaming.ts +517 -0
  311. package/src/eval/__tests__/agent-bridge.test.ts +708 -0
  312. package/src/eval/__tests__/bridge-timeout.test.ts +64 -0
  313. package/src/eval/__tests__/budget-bridge.test.ts +69 -0
  314. package/src/eval/__tests__/completion-bridge.test.ts +412 -0
  315. package/src/eval/__tests__/helpers-local-roots.test.ts +58 -0
  316. package/src/eval/__tests__/idle-timeout.test.ts +80 -0
  317. package/src/eval/__tests__/js-context-manager.test.ts +241 -0
  318. package/src/eval/__tests__/kernel-spawn.test.ts +103 -0
  319. package/src/eval/agent-bridge.ts +319 -0
  320. package/src/eval/backend.ts +71 -0
  321. package/src/eval/bridge-timeout.ts +44 -0
  322. package/src/eval/budget-bridge.ts +48 -0
  323. package/src/eval/completion-bridge.ts +207 -0
  324. package/src/eval/concurrency-bridge.ts +34 -0
  325. package/src/eval/idle-timeout.ts +91 -0
  326. package/src/eval/index.ts +4 -0
  327. package/src/eval/js/context-manager.ts +502 -0
  328. package/src/eval/js/executor.ts +173 -0
  329. package/src/eval/js/index.ts +51 -0
  330. package/src/eval/js/shared/helpers.ts +283 -0
  331. package/src/eval/js/shared/indirect-eval.ts +30 -0
  332. package/src/eval/js/shared/local-module-loader.ts +342 -0
  333. package/src/eval/js/shared/prelude.ts +2 -0
  334. package/src/eval/js/shared/prelude.txt +246 -0
  335. package/src/eval/js/shared/rewrite-imports.ts +532 -0
  336. package/src/eval/js/shared/runtime.ts +352 -0
  337. package/src/eval/js/shared/types.ts +18 -0
  338. package/src/eval/js/tool-bridge.ts +162 -0
  339. package/src/eval/js/worker-core.ts +132 -0
  340. package/src/eval/js/worker-entry.ts +30 -0
  341. package/src/eval/js/worker-protocol.ts +47 -0
  342. package/src/eval/py/__tests__/prelude.test.ts +19 -0
  343. package/src/eval/py/display.ts +71 -0
  344. package/src/eval/py/executor.ts +742 -0
  345. package/src/eval/py/index.ts +68 -0
  346. package/src/eval/py/kernel.ts +748 -0
  347. package/src/eval/py/prelude.py +658 -0
  348. package/src/eval/py/prelude.ts +3 -0
  349. package/src/eval/py/runner.py +1133 -0
  350. package/src/eval/py/runtime.ts +276 -0
  351. package/src/eval/py/spawn-options.ts +126 -0
  352. package/src/eval/py/tool-bridge.ts +182 -0
  353. package/src/eval/session-id.ts +8 -0
  354. package/src/eval/types.ts +48 -0
  355. package/src/exa/index.ts +2 -0
  356. package/src/exa/mcp-client.ts +370 -0
  357. package/src/exa/types.ts +69 -0
  358. package/src/exec/bash-executor.ts +419 -0
  359. package/src/exec/exec.ts +53 -0
  360. package/src/exec/non-interactive-env.ts +48 -0
  361. package/src/export/custom-share.ts +65 -0
  362. package/src/export/html/index.ts +164 -0
  363. package/src/export/html/template.css +1051 -0
  364. package/src/export/html/template.generated.ts +2 -0
  365. package/src/export/html/template.html +46 -0
  366. package/src/export/html/template.js +2271 -0
  367. package/src/export/html/template.macro.ts +25 -0
  368. package/src/export/html/vendor/highlight.min.js +1213 -0
  369. package/src/export/html/vendor/marked.min.js +6 -0
  370. package/src/export/ttsr.ts +583 -0
  371. package/src/extensibility/custom-commands/bundled/ci-green/index.ts +54 -0
  372. package/src/extensibility/custom-commands/bundled/review/index.ts +489 -0
  373. package/src/extensibility/custom-commands/index.ts +2 -0
  374. package/src/extensibility/custom-commands/loader.ts +238 -0
  375. package/src/extensibility/custom-commands/types.ts +113 -0
  376. package/src/extensibility/custom-tools/index.ts +7 -0
  377. package/src/extensibility/custom-tools/loader.ts +269 -0
  378. package/src/extensibility/custom-tools/types.ts +270 -0
  379. package/src/extensibility/custom-tools/wrapper.ts +47 -0
  380. package/src/extensibility/extensions/compact-handler.ts +40 -0
  381. package/src/extensibility/extensions/get-commands-handler.ts +78 -0
  382. package/src/extensibility/extensions/index.ts +16 -0
  383. package/src/extensibility/extensions/loader.ts +572 -0
  384. package/src/extensibility/extensions/runner.ts +922 -0
  385. package/src/extensibility/extensions/types.ts +1322 -0
  386. package/src/extensibility/extensions/wrapper.ts +223 -0
  387. package/src/extensibility/hooks/index.ts +5 -0
  388. package/src/extensibility/hooks/loader.ts +257 -0
  389. package/src/extensibility/hooks/runner.ts +425 -0
  390. package/src/extensibility/hooks/tool-wrapper.ts +107 -0
  391. package/src/extensibility/hooks/types.ts +606 -0
  392. package/src/extensibility/legacy-pi-ai-shim.ts +24 -0
  393. package/src/extensibility/legacy-pi-coding-agent-shim.ts +15 -0
  394. package/src/extensibility/plugins/doctor.ts +65 -0
  395. package/src/extensibility/plugins/git-url.ts +367 -0
  396. package/src/extensibility/plugins/index.ts +9 -0
  397. package/src/extensibility/plugins/installer.ts +192 -0
  398. package/src/extensibility/plugins/legacy-pi-compat.ts +682 -0
  399. package/src/extensibility/plugins/loader.ts +313 -0
  400. package/src/extensibility/plugins/manager.ts +827 -0
  401. package/src/extensibility/plugins/marketplace/cache.ts +136 -0
  402. package/src/extensibility/plugins/marketplace/fetcher.ts +317 -0
  403. package/src/extensibility/plugins/marketplace/index.ts +6 -0
  404. package/src/extensibility/plugins/marketplace/manager.ts +770 -0
  405. package/src/extensibility/plugins/marketplace/registry.ts +196 -0
  406. package/src/extensibility/plugins/marketplace/source-resolver.ts +147 -0
  407. package/src/extensibility/plugins/marketplace/types.ts +191 -0
  408. package/src/extensibility/plugins/marketplace-auto-update.ts +49 -0
  409. package/src/extensibility/plugins/parser.ts +105 -0
  410. package/src/extensibility/plugins/types.ts +194 -0
  411. package/src/extensibility/shared-events.ts +343 -0
  412. package/src/extensibility/skills.ts +312 -0
  413. package/src/extensibility/slash-commands.ts +227 -0
  414. package/src/extensibility/tool-proxy.ts +25 -0
  415. package/src/extensibility/typebox.ts +418 -0
  416. package/src/extensibility/utils.ts +44 -0
  417. package/src/goals/index.ts +3 -0
  418. package/src/goals/runtime.ts +528 -0
  419. package/src/goals/state.ts +37 -0
  420. package/src/goals/tools/goal-tool.ts +251 -0
  421. package/src/hindsight/backend.ts +354 -0
  422. package/src/hindsight/bank.ts +156 -0
  423. package/src/hindsight/client.ts +598 -0
  424. package/src/hindsight/config.ts +175 -0
  425. package/src/hindsight/content.ts +210 -0
  426. package/src/hindsight/index.ts +8 -0
  427. package/src/hindsight/mental-models.ts +429 -0
  428. package/src/hindsight/seeds.json +32 -0
  429. package/src/hindsight/state.ts +488 -0
  430. package/src/hindsight/transcript.ts +71 -0
  431. package/src/index.ts +59 -0
  432. package/src/internal-urls/agent-protocol.ts +146 -0
  433. package/src/internal-urls/artifact-protocol.ts +107 -0
  434. package/src/internal-urls/docs-index.generated.ts +106 -0
  435. package/src/internal-urls/history-protocol.ts +113 -0
  436. package/src/internal-urls/index.ts +25 -0
  437. package/src/internal-urls/issue-pr-protocol.ts +584 -0
  438. package/src/internal-urls/json-query.ts +126 -0
  439. package/src/internal-urls/local-protocol.ts +287 -0
  440. package/src/internal-urls/mcp-protocol.ts +151 -0
  441. package/src/internal-urls/memory-protocol.ts +169 -0
  442. package/src/internal-urls/omp-protocol.ts +93 -0
  443. package/src/internal-urls/parse.ts +72 -0
  444. package/src/internal-urls/registry-helpers.ts +25 -0
  445. package/src/internal-urls/router.ts +105 -0
  446. package/src/internal-urls/rule-protocol.ts +45 -0
  447. package/src/internal-urls/skill-protocol.ts +96 -0
  448. package/src/internal-urls/types.ts +152 -0
  449. package/src/internal-urls/vault-protocol.ts +936 -0
  450. package/src/irc/bus.ts +292 -0
  451. package/src/lib/xai-http.ts +124 -0
  452. package/src/lsp/client.ts +1193 -0
  453. package/src/lsp/clients/biome-client.ts +264 -0
  454. package/src/lsp/clients/index.ts +50 -0
  455. package/src/lsp/clients/lsp-linter-client.ts +93 -0
  456. package/src/lsp/clients/swiftlint-client.ts +120 -0
  457. package/src/lsp/config.ts +502 -0
  458. package/src/lsp/defaults.json +493 -0
  459. package/src/lsp/diagnostics-ledger.ts +51 -0
  460. package/src/lsp/edits.ts +267 -0
  461. package/src/lsp/index.ts +2477 -0
  462. package/src/lsp/lspmux.ts +233 -0
  463. package/src/lsp/render.ts +694 -0
  464. package/src/lsp/startup-events.ts +13 -0
  465. package/src/lsp/types.ts +455 -0
  466. package/src/lsp/utils.ts +718 -0
  467. package/src/main.ts +1325 -0
  468. package/src/mcp/client.ts +484 -0
  469. package/src/mcp/config-writer.ts +225 -0
  470. package/src/mcp/config.ts +365 -0
  471. package/src/mcp/index.ts +29 -0
  472. package/src/mcp/json-rpc.ts +122 -0
  473. package/src/mcp/loader.ts +124 -0
  474. package/src/mcp/manager.ts +1275 -0
  475. package/src/mcp/oauth-discovery.ts +442 -0
  476. package/src/mcp/oauth-flow.ts +442 -0
  477. package/src/mcp/render.ts +132 -0
  478. package/src/mcp/smithery-auth.ts +104 -0
  479. package/src/mcp/smithery-connect.ts +145 -0
  480. package/src/mcp/smithery-registry.ts +477 -0
  481. package/src/mcp/timeout.ts +59 -0
  482. package/src/mcp/tool-bridge.ts +426 -0
  483. package/src/mcp/tool-cache.ts +117 -0
  484. package/src/mcp/transports/http.ts +519 -0
  485. package/src/mcp/transports/index.ts +6 -0
  486. package/src/mcp/transports/stdio.ts +528 -0
  487. package/src/mcp/types.ts +423 -0
  488. package/src/memories/index.ts +1150 -0
  489. package/src/memories/storage.ts +577 -0
  490. package/src/memory-backend/index.ts +18 -0
  491. package/src/memory-backend/local-backend.ts +39 -0
  492. package/src/memory-backend/off-backend.ts +25 -0
  493. package/src/memory-backend/resolve.ts +25 -0
  494. package/src/memory-backend/runtime.ts +66 -0
  495. package/src/memory-backend/types.ts +166 -0
  496. package/src/mnemopi/backend.ts +547 -0
  497. package/src/mnemopi/config.ts +160 -0
  498. package/src/mnemopi/index.ts +3 -0
  499. package/src/mnemopi/state.ts +584 -0
  500. package/src/modes/acp/acp-agent.ts +2407 -0
  501. package/src/modes/acp/acp-client-bridge.ts +154 -0
  502. package/src/modes/acp/acp-event-mapper.ts +929 -0
  503. package/src/modes/acp/acp-mode.ts +23 -0
  504. package/src/modes/acp/index.ts +2 -0
  505. package/src/modes/acp/terminal-auth.ts +37 -0
  506. package/src/modes/components/agent-dashboard.ts +1206 -0
  507. package/src/modes/components/agent-hub.ts +1071 -0
  508. package/src/modes/components/assistant-message.ts +307 -0
  509. package/src/modes/components/bash-execution.ts +220 -0
  510. package/src/modes/components/bordered-loader.ts +41 -0
  511. package/src/modes/components/branch-summary-message.ts +45 -0
  512. package/src/modes/components/btw-panel.ts +104 -0
  513. package/src/modes/components/chat-block.ts +111 -0
  514. package/src/modes/components/compaction-summary-message.ts +87 -0
  515. package/src/modes/components/copy-selector.ts +206 -0
  516. package/src/modes/components/countdown-timer.ts +75 -0
  517. package/src/modes/components/custom-editor.ts +398 -0
  518. package/src/modes/components/custom-message.ts +63 -0
  519. package/src/modes/components/diff.ts +277 -0
  520. package/src/modes/components/dynamic-border.ts +34 -0
  521. package/src/modes/components/error-banner.ts +33 -0
  522. package/src/modes/components/eval-execution.ts +158 -0
  523. package/src/modes/components/execution-shared.ts +101 -0
  524. package/src/modes/components/extensions/extension-dashboard.ts +399 -0
  525. package/src/modes/components/extensions/extension-list.ts +502 -0
  526. package/src/modes/components/extensions/index.ts +9 -0
  527. package/src/modes/components/extensions/inspector-panel.ts +317 -0
  528. package/src/modes/components/extensions/state-manager.ts +627 -0
  529. package/src/modes/components/extensions/types.ts +186 -0
  530. package/src/modes/components/footer.ts +274 -0
  531. package/src/modes/components/history-search.ts +280 -0
  532. package/src/modes/components/hook-editor.ts +167 -0
  533. package/src/modes/components/hook-input.ts +87 -0
  534. package/src/modes/components/hook-message.ts +66 -0
  535. package/src/modes/components/hook-selector.ts +660 -0
  536. package/src/modes/components/index.ts +38 -0
  537. package/src/modes/components/keybinding-hints.ts +65 -0
  538. package/src/modes/components/late-diagnostics-message.ts +60 -0
  539. package/src/modes/components/login-dialog.ts +164 -0
  540. package/src/modes/components/mcp-add-wizard.ts +1340 -0
  541. package/src/modes/components/message-frame.ts +88 -0
  542. package/src/modes/components/model-selector.ts +1271 -0
  543. package/src/modes/components/oauth-selector.ts +368 -0
  544. package/src/modes/components/omfg-panel.ts +141 -0
  545. package/src/modes/components/overlay-box.ts +108 -0
  546. package/src/modes/components/plan-review-overlay.ts +820 -0
  547. package/src/modes/components/plan-toc.ts +138 -0
  548. package/src/modes/components/plugin-selector.ts +95 -0
  549. package/src/modes/components/plugin-settings.ts +722 -0
  550. package/src/modes/components/queue-mode-selector.ts +56 -0
  551. package/src/modes/components/read-tool-group.ts +670 -0
  552. package/src/modes/components/segment-track.ts +52 -0
  553. package/src/modes/components/session-selector.ts +625 -0
  554. package/src/modes/components/settings-defs.ts +189 -0
  555. package/src/modes/components/settings-selector.ts +651 -0
  556. package/src/modes/components/show-images-selector.ts +45 -0
  557. package/src/modes/components/skill-message.ts +89 -0
  558. package/src/modes/components/status-line/component.ts +869 -0
  559. package/src/modes/components/status-line/context-thresholds.ts +79 -0
  560. package/src/modes/components/status-line/git-utils.ts +42 -0
  561. package/src/modes/components/status-line/index.ts +5 -0
  562. package/src/modes/components/status-line/presets.ts +106 -0
  563. package/src/modes/components/status-line/segments.ts +584 -0
  564. package/src/modes/components/status-line/separators.ts +55 -0
  565. package/src/modes/components/status-line/token-rate.ts +66 -0
  566. package/src/modes/components/status-line/types.ts +108 -0
  567. package/src/modes/components/theme-selector.ts +63 -0
  568. package/src/modes/components/thinking-selector.ts +52 -0
  569. package/src/modes/components/tiny-title-download-progress.ts +90 -0
  570. package/src/modes/components/tips.txt +19 -0
  571. package/src/modes/components/todo-reminder.ts +38 -0
  572. package/src/modes/components/tool-execution.ts +1024 -0
  573. package/src/modes/components/transcript-container.ts +608 -0
  574. package/src/modes/components/tree-selector.ts +978 -0
  575. package/src/modes/components/ttsr-notification.ts +122 -0
  576. package/src/modes/components/user-message-selector.ts +227 -0
  577. package/src/modes/components/user-message.ts +66 -0
  578. package/src/modes/components/visual-truncate.ts +63 -0
  579. package/src/modes/components/welcome.ts +493 -0
  580. package/src/modes/controllers/btw-controller.ts +105 -0
  581. package/src/modes/controllers/command-controller-shared.ts +109 -0
  582. package/src/modes/controllers/command-controller.ts +1566 -0
  583. package/src/modes/controllers/event-controller.ts +1054 -0
  584. package/src/modes/controllers/extension-ui-controller.ts +886 -0
  585. package/src/modes/controllers/input-controller.ts +1073 -0
  586. package/src/modes/controllers/mcp-command-controller.ts +2017 -0
  587. package/src/modes/controllers/omfg-controller.ts +283 -0
  588. package/src/modes/controllers/omfg-rule.ts +647 -0
  589. package/src/modes/controllers/selector-controller.ts +1108 -0
  590. package/src/modes/controllers/ssh-command-controller.ts +384 -0
  591. package/src/modes/controllers/streaming-reveal.ts +279 -0
  592. package/src/modes/controllers/tan-command-controller.ts +173 -0
  593. package/src/modes/controllers/todo-command-controller.ts +485 -0
  594. package/src/modes/data/emojis.json +1 -0
  595. package/src/modes/emoji-autocomplete.ts +285 -0
  596. package/src/modes/gradient-highlight.ts +87 -0
  597. package/src/modes/image-references.ts +117 -0
  598. package/src/modes/index.ts +17 -0
  599. package/src/modes/interactive-mode.ts +3370 -0
  600. package/src/modes/internal-url-autocomplete.ts +143 -0
  601. package/src/modes/loop-limit.ts +140 -0
  602. package/src/modes/magic-keywords.ts +20 -0
  603. package/src/modes/markdown-prose.ts +247 -0
  604. package/src/modes/oauth-manual-input.ts +69 -0
  605. package/src/modes/orchestrate.ts +42 -0
  606. package/src/modes/print-mode.ts +126 -0
  607. package/src/modes/prompt-action-autocomplete.ts +260 -0
  608. package/src/modes/rpc/host-tools.ts +186 -0
  609. package/src/modes/rpc/host-uris.ts +235 -0
  610. package/src/modes/rpc/rpc-client.ts +963 -0
  611. package/src/modes/rpc/rpc-mode.ts +947 -0
  612. package/src/modes/rpc/rpc-subagents.ts +265 -0
  613. package/src/modes/rpc/rpc-types.ts +458 -0
  614. package/src/modes/runtime-init.ts +116 -0
  615. package/src/modes/session-observer-registry.ts +146 -0
  616. package/src/modes/setup-version.ts +11 -0
  617. package/src/modes/setup-wizard/index.ts +99 -0
  618. package/src/modes/setup-wizard/lazy.ts +16 -0
  619. package/src/modes/setup-wizard/scenes/glyph.ts +96 -0
  620. package/src/modes/setup-wizard/scenes/outro.ts +35 -0
  621. package/src/modes/setup-wizard/scenes/providers.ts +69 -0
  622. package/src/modes/setup-wizard/scenes/sign-in.ts +205 -0
  623. package/src/modes/setup-wizard/scenes/splash.ts +201 -0
  624. package/src/modes/setup-wizard/scenes/theme.ts +299 -0
  625. package/src/modes/setup-wizard/scenes/types.ts +48 -0
  626. package/src/modes/setup-wizard/scenes/web-search.ts +129 -0
  627. package/src/modes/setup-wizard/wizard-overlay.ts +275 -0
  628. package/src/modes/shared.ts +47 -0
  629. package/src/modes/theme/dark.json +95 -0
  630. package/src/modes/theme/defaults/alabaster.json +93 -0
  631. package/src/modes/theme/defaults/amethyst.json +96 -0
  632. package/src/modes/theme/defaults/anthracite.json +93 -0
  633. package/src/modes/theme/defaults/basalt.json +91 -0
  634. package/src/modes/theme/defaults/birch.json +95 -0
  635. package/src/modes/theme/defaults/dark-abyss.json +91 -0
  636. package/src/modes/theme/defaults/dark-arctic.json +104 -0
  637. package/src/modes/theme/defaults/dark-aurora.json +95 -0
  638. package/src/modes/theme/defaults/dark-catppuccin.json +107 -0
  639. package/src/modes/theme/defaults/dark-cavern.json +91 -0
  640. package/src/modes/theme/defaults/dark-copper.json +95 -0
  641. package/src/modes/theme/defaults/dark-cosmos.json +90 -0
  642. package/src/modes/theme/defaults/dark-cyberpunk.json +102 -0
  643. package/src/modes/theme/defaults/dark-dracula.json +98 -0
  644. package/src/modes/theme/defaults/dark-eclipse.json +91 -0
  645. package/src/modes/theme/defaults/dark-ember.json +95 -0
  646. package/src/modes/theme/defaults/dark-equinox.json +90 -0
  647. package/src/modes/theme/defaults/dark-forest.json +96 -0
  648. package/src/modes/theme/defaults/dark-github.json +105 -0
  649. package/src/modes/theme/defaults/dark-gruvbox.json +112 -0
  650. package/src/modes/theme/defaults/dark-lavender.json +95 -0
  651. package/src/modes/theme/defaults/dark-lunar.json +89 -0
  652. package/src/modes/theme/defaults/dark-midnight.json +95 -0
  653. package/src/modes/theme/defaults/dark-monochrome.json +94 -0
  654. package/src/modes/theme/defaults/dark-monokai.json +98 -0
  655. package/src/modes/theme/defaults/dark-nebula.json +90 -0
  656. package/src/modes/theme/defaults/dark-nord.json +97 -0
  657. package/src/modes/theme/defaults/dark-ocean.json +101 -0
  658. package/src/modes/theme/defaults/dark-one.json +100 -0
  659. package/src/modes/theme/defaults/dark-poimandres.json +142 -0
  660. package/src/modes/theme/defaults/dark-rainforest.json +91 -0
  661. package/src/modes/theme/defaults/dark-reef.json +91 -0
  662. package/src/modes/theme/defaults/dark-retro.json +92 -0
  663. package/src/modes/theme/defaults/dark-rose-pine.json +96 -0
  664. package/src/modes/theme/defaults/dark-sakura.json +95 -0
  665. package/src/modes/theme/defaults/dark-slate.json +95 -0
  666. package/src/modes/theme/defaults/dark-solarized.json +97 -0
  667. package/src/modes/theme/defaults/dark-solstice.json +90 -0
  668. package/src/modes/theme/defaults/dark-starfall.json +91 -0
  669. package/src/modes/theme/defaults/dark-sunset.json +99 -0
  670. package/src/modes/theme/defaults/dark-swamp.json +90 -0
  671. package/src/modes/theme/defaults/dark-synthwave.json +103 -0
  672. package/src/modes/theme/defaults/dark-taiga.json +91 -0
  673. package/src/modes/theme/defaults/dark-terminal.json +95 -0
  674. package/src/modes/theme/defaults/dark-tokyo-night.json +101 -0
  675. package/src/modes/theme/defaults/dark-tundra.json +91 -0
  676. package/src/modes/theme/defaults/dark-twilight.json +91 -0
  677. package/src/modes/theme/defaults/dark-volcanic.json +91 -0
  678. package/src/modes/theme/defaults/graphite.json +92 -0
  679. package/src/modes/theme/defaults/index.ts +199 -0
  680. package/src/modes/theme/defaults/light-arctic.json +107 -0
  681. package/src/modes/theme/defaults/light-aurora-day.json +91 -0
  682. package/src/modes/theme/defaults/light-canyon.json +91 -0
  683. package/src/modes/theme/defaults/light-catppuccin.json +106 -0
  684. package/src/modes/theme/defaults/light-cirrus.json +90 -0
  685. package/src/modes/theme/defaults/light-coral.json +95 -0
  686. package/src/modes/theme/defaults/light-cyberpunk.json +96 -0
  687. package/src/modes/theme/defaults/light-dawn.json +90 -0
  688. package/src/modes/theme/defaults/light-dunes.json +91 -0
  689. package/src/modes/theme/defaults/light-eucalyptus.json +95 -0
  690. package/src/modes/theme/defaults/light-forest.json +100 -0
  691. package/src/modes/theme/defaults/light-frost.json +95 -0
  692. package/src/modes/theme/defaults/light-github.json +115 -0
  693. package/src/modes/theme/defaults/light-glacier.json +91 -0
  694. package/src/modes/theme/defaults/light-gruvbox.json +108 -0
  695. package/src/modes/theme/defaults/light-haze.json +90 -0
  696. package/src/modes/theme/defaults/light-honeycomb.json +95 -0
  697. package/src/modes/theme/defaults/light-lagoon.json +91 -0
  698. package/src/modes/theme/defaults/light-lavender.json +95 -0
  699. package/src/modes/theme/defaults/light-meadow.json +91 -0
  700. package/src/modes/theme/defaults/light-mint.json +95 -0
  701. package/src/modes/theme/defaults/light-monochrome.json +101 -0
  702. package/src/modes/theme/defaults/light-ocean.json +99 -0
  703. package/src/modes/theme/defaults/light-one.json +99 -0
  704. package/src/modes/theme/defaults/light-opal.json +91 -0
  705. package/src/modes/theme/defaults/light-orchard.json +91 -0
  706. package/src/modes/theme/defaults/light-paper.json +95 -0
  707. package/src/modes/theme/defaults/light-poimandres.json +142 -0
  708. package/src/modes/theme/defaults/light-prism.json +90 -0
  709. package/src/modes/theme/defaults/light-retro.json +98 -0
  710. package/src/modes/theme/defaults/light-sand.json +95 -0
  711. package/src/modes/theme/defaults/light-savanna.json +91 -0
  712. package/src/modes/theme/defaults/light-solarized.json +102 -0
  713. package/src/modes/theme/defaults/light-soleil.json +90 -0
  714. package/src/modes/theme/defaults/light-sunset.json +99 -0
  715. package/src/modes/theme/defaults/light-synthwave.json +98 -0
  716. package/src/modes/theme/defaults/light-tokyo-night.json +111 -0
  717. package/src/modes/theme/defaults/light-wetland.json +91 -0
  718. package/src/modes/theme/defaults/light-zenith.json +89 -0
  719. package/src/modes/theme/defaults/limestone.json +94 -0
  720. package/src/modes/theme/defaults/mahogany.json +97 -0
  721. package/src/modes/theme/defaults/marble.json +93 -0
  722. package/src/modes/theme/defaults/obsidian.json +91 -0
  723. package/src/modes/theme/defaults/onyx.json +91 -0
  724. package/src/modes/theme/defaults/pearl.json +93 -0
  725. package/src/modes/theme/defaults/porcelain.json +91 -0
  726. package/src/modes/theme/defaults/quartz.json +96 -0
  727. package/src/modes/theme/defaults/sandstone.json +95 -0
  728. package/src/modes/theme/defaults/titanium.json +90 -0
  729. package/src/modes/theme/light.json +93 -0
  730. package/src/modes/theme/mermaid-cache.ts +29 -0
  731. package/src/modes/theme/shimmer.ts +235 -0
  732. package/src/modes/theme/theme-schema.json +459 -0
  733. package/src/modes/theme/theme.ts +2676 -0
  734. package/src/modes/turn-budget.ts +31 -0
  735. package/src/modes/types.ts +359 -0
  736. package/src/modes/ultrathink.ts +41 -0
  737. package/src/modes/utils/context-usage.ts +339 -0
  738. package/src/modes/utils/copy-targets.ts +360 -0
  739. package/src/modes/utils/hotkeys-markdown.ts +61 -0
  740. package/src/modes/utils/keybinding-matchers.ts +51 -0
  741. package/src/modes/utils/tools-markdown.ts +27 -0
  742. package/src/modes/utils/ui-helpers.ts +801 -0
  743. package/src/modes/workflow.ts +42 -0
  744. package/src/plan-mode/approved-plan.ts +186 -0
  745. package/src/plan-mode/plan-handoff.ts +37 -0
  746. package/src/plan-mode/plan-protection.ts +31 -0
  747. package/src/plan-mode/state.ts +6 -0
  748. package/src/priority.json +41 -0
  749. package/src/prompts/agents/designer.md +66 -0
  750. package/src/prompts/agents/explore.md +58 -0
  751. package/src/prompts/agents/frontmatter.md +11 -0
  752. package/src/prompts/agents/init.md +33 -0
  753. package/src/prompts/agents/librarian.md +119 -0
  754. package/src/prompts/agents/oracle.md +55 -0
  755. package/src/prompts/agents/plan.md +48 -0
  756. package/src/prompts/agents/reviewer.md +140 -0
  757. package/src/prompts/agents/task.md +16 -0
  758. package/src/prompts/ci-green-request.md +36 -0
  759. package/src/prompts/dry-balance-bench.md +8 -0
  760. package/src/prompts/goals/goal-budget-limit.md +16 -0
  761. package/src/prompts/goals/goal-continuation.md +28 -0
  762. package/src/prompts/goals/goal-mode-active.md +23 -0
  763. package/src/prompts/memories/consolidation.md +30 -0
  764. package/src/prompts/memories/read-path.md +11 -0
  765. package/src/prompts/memories/stage_one_input.md +6 -0
  766. package/src/prompts/memories/stage_one_system.md +21 -0
  767. package/src/prompts/review-custom-request.md +22 -0
  768. package/src/prompts/review-headless-request.md +16 -0
  769. package/src/prompts/review-request.md +69 -0
  770. package/src/prompts/steering/user-interjection.md +10 -0
  771. package/src/prompts/system/agent-creation-architect.md +50 -0
  772. package/src/prompts/system/agent-creation-user.md +6 -0
  773. package/src/prompts/system/auto-continue.md +1 -0
  774. package/src/prompts/system/auto-thinking-difficulty-local.md +14 -0
  775. package/src/prompts/system/auto-thinking-difficulty.md +12 -0
  776. package/src/prompts/system/background-tan-dispatch.md +8 -0
  777. package/src/prompts/system/btw-user.md +8 -0
  778. package/src/prompts/system/commit-message-system.md +14 -0
  779. package/src/prompts/system/custom-system-prompt.md +64 -0
  780. package/src/prompts/system/eager-todo.md +13 -0
  781. package/src/prompts/system/empty-stop-retry.md +6 -0
  782. package/src/prompts/system/irc-incoming.md +7 -0
  783. package/src/prompts/system/manual-continue.md +7 -0
  784. package/src/prompts/system/memory-consolidation-system.md +8 -0
  785. package/src/prompts/system/memory-extraction-system.md +26 -0
  786. package/src/prompts/system/omfg-user.md +50 -0
  787. package/src/prompts/system/orchestrate-notice.md +40 -0
  788. package/src/prompts/system/plan-mode-active.md +109 -0
  789. package/src/prompts/system/plan-mode-approved.md +25 -0
  790. package/src/prompts/system/plan-mode-compact-instructions.md +16 -0
  791. package/src/prompts/system/plan-mode-reference.md +11 -0
  792. package/src/prompts/system/plan-mode-subagent.md +33 -0
  793. package/src/prompts/system/plan-mode-tool-decision-reminder.md +9 -0
  794. package/src/prompts/system/project-prompt.md +52 -0
  795. package/src/prompts/system/subagent-system-prompt.md +64 -0
  796. package/src/prompts/system/subagent-user-prompt.md +3 -0
  797. package/src/prompts/system/subagent-yield-reminder.md +12 -0
  798. package/src/prompts/system/system-prompt.md +258 -0
  799. package/src/prompts/system/tiny-title-system.md +8 -0
  800. package/src/prompts/system/title-system.md +16 -0
  801. package/src/prompts/system/ttsr-interrupt.md +7 -0
  802. package/src/prompts/system/ttsr-tool-reminder.md +5 -0
  803. package/src/prompts/system/ultrathink-notice.md +3 -0
  804. package/src/prompts/system/web-search.md +25 -0
  805. package/src/prompts/system/workflow-notice.md +70 -0
  806. package/src/prompts/tools/apply-patch.md +65 -0
  807. package/src/prompts/tools/ask.md +30 -0
  808. package/src/prompts/tools/ast-edit.md +39 -0
  809. package/src/prompts/tools/ast-grep.md +42 -0
  810. package/src/prompts/tools/async-result.md +8 -0
  811. package/src/prompts/tools/bash.md +46 -0
  812. package/src/prompts/tools/browser.md +73 -0
  813. package/src/prompts/tools/checkpoint.md +16 -0
  814. package/src/prompts/tools/debug.md +34 -0
  815. package/src/prompts/tools/eval.md +92 -0
  816. package/src/prompts/tools/find.md +36 -0
  817. package/src/prompts/tools/github.md +21 -0
  818. package/src/prompts/tools/goal.md +18 -0
  819. package/src/prompts/tools/image-gen.md +7 -0
  820. package/src/prompts/tools/inspect-image-system.md +20 -0
  821. package/src/prompts/tools/inspect-image.md +32 -0
  822. package/src/prompts/tools/irc.md +59 -0
  823. package/src/prompts/tools/job.md +19 -0
  824. package/src/prompts/tools/lsp-late-diagnostic.md +8 -0
  825. package/src/prompts/tools/lsp.md +42 -0
  826. package/src/prompts/tools/memory-edit.md +8 -0
  827. package/src/prompts/tools/patch.md +70 -0
  828. package/src/prompts/tools/read.md +84 -0
  829. package/src/prompts/tools/recall.md +5 -0
  830. package/src/prompts/tools/reflect.md +5 -0
  831. package/src/prompts/tools/render-mermaid.md +9 -0
  832. package/src/prompts/tools/replace.md +30 -0
  833. package/src/prompts/tools/resolve.md +9 -0
  834. package/src/prompts/tools/retain.md +6 -0
  835. package/src/prompts/tools/rewind.md +13 -0
  836. package/src/prompts/tools/search-tool-bm25.md +32 -0
  837. package/src/prompts/tools/search.md +24 -0
  838. package/src/prompts/tools/ssh.md +31 -0
  839. package/src/prompts/tools/task-summary.md +17 -0
  840. package/src/prompts/tools/task.md +88 -0
  841. package/src/prompts/tools/todo.md +62 -0
  842. package/src/prompts/tools/web-search.md +10 -0
  843. package/src/prompts/tools/write.md +14 -0
  844. package/src/registry/agent-lifecycle.ts +218 -0
  845. package/src/registry/agent-registry.ts +151 -0
  846. package/src/sdk.ts +2558 -0
  847. package/src/secrets/index.ts +123 -0
  848. package/src/secrets/obfuscator.ts +298 -0
  849. package/src/secrets/regex.ts +21 -0
  850. package/src/session/agent-session.ts +10121 -0
  851. package/src/session/agent-storage.ts +455 -0
  852. package/src/session/artifacts.ts +135 -0
  853. package/src/session/auth-broker-config.ts +131 -0
  854. package/src/session/auth-storage.ts +29 -0
  855. package/src/session/blob-store.ts +255 -0
  856. package/src/session/client-bridge.ts +85 -0
  857. package/src/session/history-storage.ts +348 -0
  858. package/src/session/indexed-session-storage.ts +430 -0
  859. package/src/session/messages.ts +541 -0
  860. package/src/session/redis-session-storage.ts +170 -0
  861. package/src/session/session-dump-format.ts +209 -0
  862. package/src/session/session-history-format.ts +246 -0
  863. package/src/session/session-manager.ts +3676 -0
  864. package/src/session/session-storage.ts +529 -0
  865. package/src/session/shake-types.ts +43 -0
  866. package/src/session/sql-session-storage.ts +314 -0
  867. package/src/session/streaming-output.ts +1330 -0
  868. package/src/session/tool-choice-queue.ts +213 -0
  869. package/src/session/yield-queue.ts +173 -0
  870. package/src/slash-commands/acp-builtins.ts +70 -0
  871. package/src/slash-commands/builtin-registry.ts +1798 -0
  872. package/src/slash-commands/helpers/context-report.ts +39 -0
  873. package/src/slash-commands/helpers/format.ts +46 -0
  874. package/src/slash-commands/helpers/marketplace-manager.ts +25 -0
  875. package/src/slash-commands/helpers/mcp.ts +532 -0
  876. package/src/slash-commands/helpers/parse.ts +85 -0
  877. package/src/slash-commands/helpers/ssh.ts +195 -0
  878. package/src/slash-commands/helpers/stats-dashboard.ts +85 -0
  879. package/src/slash-commands/helpers/todo.ts +279 -0
  880. package/src/slash-commands/helpers/usage-report.ts +95 -0
  881. package/src/slash-commands/marketplace-install-parser.ts +99 -0
  882. package/src/slash-commands/types.ts +135 -0
  883. package/src/ssh/config-writer.ts +183 -0
  884. package/src/ssh/connection-manager.ts +509 -0
  885. package/src/ssh/ssh-executor.ts +189 -0
  886. package/src/ssh/sshfs-mount.ts +140 -0
  887. package/src/ssh/utils.ts +8 -0
  888. package/src/stt/downloader.ts +71 -0
  889. package/src/stt/index.ts +3 -0
  890. package/src/stt/recorder.ts +351 -0
  891. package/src/stt/setup.ts +52 -0
  892. package/src/stt/stt-controller.ts +160 -0
  893. package/src/stt/transcribe.py +70 -0
  894. package/src/stt/transcriber.ts +91 -0
  895. package/src/stubs/natives/index.ts +814 -0
  896. package/src/stubs/natives/package.json +7 -0
  897. package/src/stubs/tui/index.ts +282 -0
  898. package/src/stubs/tui/package.json +7 -0
  899. package/src/system-prompt.ts +611 -0
  900. package/src/task/agents.ts +167 -0
  901. package/src/task/commands.ts +132 -0
  902. package/src/task/discovery.ts +122 -0
  903. package/src/task/executor.ts +2133 -0
  904. package/src/task/index.ts +1419 -0
  905. package/src/task/name-generator.ts +1577 -0
  906. package/src/task/omp-command.ts +26 -0
  907. package/src/task/output-manager.ts +88 -0
  908. package/src/task/parallel.ts +116 -0
  909. package/src/task/render.ts +1381 -0
  910. package/src/task/repair-args.ts +129 -0
  911. package/src/task/subprocess-tool-registry.ts +88 -0
  912. package/src/task/types.ts +336 -0
  913. package/src/task/worktree.ts +514 -0
  914. package/src/telemetry-export.ts +144 -0
  915. package/src/thinking.ts +167 -0
  916. package/src/tiny/compiled-runtime.ts +179 -0
  917. package/src/tiny/device.ts +111 -0
  918. package/src/tiny/dtype.ts +101 -0
  919. package/src/tiny/models.ts +242 -0
  920. package/src/tiny/text.ts +165 -0
  921. package/src/tiny/title-client.ts +543 -0
  922. package/src/tiny/title-protocol.ts +56 -0
  923. package/src/tiny/worker.ts +568 -0
  924. package/src/tool-discovery/mode.ts +24 -0
  925. package/src/tool-discovery/tool-index.ts +256 -0
  926. package/src/tools/approval.ts +189 -0
  927. package/src/tools/archive-reader.ts +721 -0
  928. package/src/tools/ask.ts +928 -0
  929. package/src/tools/ast-edit.ts +642 -0
  930. package/src/tools/ast-grep.ts +452 -0
  931. package/src/tools/auto-generated-guard.ts +322 -0
  932. package/src/tools/bash-command-fixup.ts +37 -0
  933. package/src/tools/bash-interactive.ts +408 -0
  934. package/src/tools/bash-interceptor.ts +67 -0
  935. package/src/tools/bash-pty-selection.ts +14 -0
  936. package/src/tools/bash-skill-urls.ts +248 -0
  937. package/src/tools/bash.ts +1386 -0
  938. package/src/tools/browser/attach.ts +175 -0
  939. package/src/tools/browser/launch.ts +660 -0
  940. package/src/tools/browser/readable.ts +112 -0
  941. package/src/tools/browser/registry.ts +197 -0
  942. package/src/tools/browser/render.ts +216 -0
  943. package/src/tools/browser/tab-protocol.ts +105 -0
  944. package/src/tools/browser/tab-supervisor.ts +628 -0
  945. package/src/tools/browser/tab-worker-entry.ts +21 -0
  946. package/src/tools/browser/tab-worker.ts +1226 -0
  947. package/src/tools/browser.ts +343 -0
  948. package/src/tools/checkpoint.ts +136 -0
  949. package/src/tools/conflict-detect.ts +718 -0
  950. package/src/tools/context.ts +39 -0
  951. package/src/tools/debug.ts +1067 -0
  952. package/src/tools/eval-backends.ts +27 -0
  953. package/src/tools/eval-render.ts +752 -0
  954. package/src/tools/eval.ts +577 -0
  955. package/src/tools/fetch.ts +1926 -0
  956. package/src/tools/file-recorder.ts +35 -0
  957. package/src/tools/find.ts +609 -0
  958. package/src/tools/fs-cache-invalidation.ts +28 -0
  959. package/src/tools/gh-cache-invalidation.ts +255 -0
  960. package/src/tools/gh-format.ts +12 -0
  961. package/src/tools/gh-renderer.ts +481 -0
  962. package/src/tools/gh.ts +3720 -0
  963. package/src/tools/github-cache.ts +637 -0
  964. package/src/tools/grouped-file-output.ts +210 -0
  965. package/src/tools/image-gen.ts +1517 -0
  966. package/src/tools/index.ts +599 -0
  967. package/src/tools/inspect-image-renderer.ts +132 -0
  968. package/src/tools/inspect-image.ts +174 -0
  969. package/src/tools/irc.ts +723 -0
  970. package/src/tools/job.ts +557 -0
  971. package/src/tools/json-tree.ts +243 -0
  972. package/src/tools/jtd-to-json-schema.ts +219 -0
  973. package/src/tools/jtd-to-typescript.ts +136 -0
  974. package/src/tools/jtd-utils.ts +102 -0
  975. package/src/tools/list-limit.ts +40 -0
  976. package/src/tools/match-line-format.ts +20 -0
  977. package/src/tools/memory-edit.ts +59 -0
  978. package/src/tools/memory-recall.ts +100 -0
  979. package/src/tools/memory-reflect.ts +88 -0
  980. package/src/tools/memory-render.ts +202 -0
  981. package/src/tools/memory-retain.ts +91 -0
  982. package/src/tools/output-meta.ts +754 -0
  983. package/src/tools/output-schema-validator.ts +132 -0
  984. package/src/tools/path-utils.ts +1054 -0
  985. package/src/tools/plan-mode-guard.ts +108 -0
  986. package/src/tools/puppeteer/00_stealth_tampering.txt +63 -0
  987. package/src/tools/puppeteer/01_stealth_activity.txt +20 -0
  988. package/src/tools/puppeteer/02_stealth_hairline.txt +11 -0
  989. package/src/tools/puppeteer/03_stealth_botd.txt +384 -0
  990. package/src/tools/puppeteer/04_stealth_iframe.txt +81 -0
  991. package/src/tools/puppeteer/05_stealth_webgl.txt +75 -0
  992. package/src/tools/puppeteer/06_stealth_screen.txt +72 -0
  993. package/src/tools/puppeteer/07_stealth_fonts.txt +97 -0
  994. package/src/tools/puppeteer/08_stealth_audio.txt +51 -0
  995. package/src/tools/puppeteer/09_stealth_locale.txt +46 -0
  996. package/src/tools/puppeteer/10_stealth_plugins.txt +206 -0
  997. package/src/tools/puppeteer/11_stealth_hardware.txt +8 -0
  998. package/src/tools/puppeteer/12_stealth_codecs.txt +40 -0
  999. package/src/tools/puppeteer/13_stealth_worker.txt +74 -0
  1000. package/src/tools/read.ts +2929 -0
  1001. package/src/tools/render-mermaid.ts +69 -0
  1002. package/src/tools/render-utils.ts +838 -0
  1003. package/src/tools/renderers.ts +77 -0
  1004. package/src/tools/report-tool-issue.ts +534 -0
  1005. package/src/tools/resolve.ts +276 -0
  1006. package/src/tools/review.ts +253 -0
  1007. package/src/tools/search-tool-bm25.ts +351 -0
  1008. package/src/tools/search.ts +1580 -0
  1009. package/src/tools/sqlite-reader.ts +828 -0
  1010. package/src/tools/ssh.ts +349 -0
  1011. package/src/tools/todo.ts +982 -0
  1012. package/src/tools/tool-errors.ts +62 -0
  1013. package/src/tools/tool-result.ts +94 -0
  1014. package/src/tools/tool-timeouts.ts +30 -0
  1015. package/src/tools/tts.ts +133 -0
  1016. package/src/tools/write.ts +1217 -0
  1017. package/src/tools/yield.ts +269 -0
  1018. package/src/tui/code-cell.ts +216 -0
  1019. package/src/tui/file-list.ts +55 -0
  1020. package/src/tui/hyperlink.ts +175 -0
  1021. package/src/tui/index.ts +12 -0
  1022. package/src/tui/output-block.ts +240 -0
  1023. package/src/tui/status-line.ts +54 -0
  1024. package/src/tui/tree-list.ts +84 -0
  1025. package/src/tui/types.ts +15 -0
  1026. package/src/tui/utils.ts +103 -0
  1027. package/src/utils/block-context.ts +312 -0
  1028. package/src/utils/changelog.ts +132 -0
  1029. package/src/utils/clipboard.ts +193 -0
  1030. package/src/utils/command-args.ts +76 -0
  1031. package/src/utils/commit-message-generator.ts +151 -0
  1032. package/src/utils/edit-mode.ts +41 -0
  1033. package/src/utils/enhanced-paste.ts +230 -0
  1034. package/src/utils/event-bus.ts +33 -0
  1035. package/src/utils/external-editor.ts +65 -0
  1036. package/src/utils/file-display-mode.ts +45 -0
  1037. package/src/utils/file-mentions.ts +281 -0
  1038. package/src/utils/git.ts +1833 -0
  1039. package/src/utils/image-loading.ts +132 -0
  1040. package/src/utils/image-resize.ts +309 -0
  1041. package/src/utils/jj.ts +248 -0
  1042. package/src/utils/lang-from-path.ts +239 -0
  1043. package/src/utils/markit.ts +89 -0
  1044. package/src/utils/open.ts +55 -0
  1045. package/src/utils/session-color.ts +68 -0
  1046. package/src/utils/shell-snapshot.ts +187 -0
  1047. package/src/utils/sixel.ts +69 -0
  1048. package/src/utils/title-generator.ts +373 -0
  1049. package/src/utils/tool-choice.ts +33 -0
  1050. package/src/utils/tools-manager.ts +363 -0
  1051. package/src/web/kagi.ts +305 -0
  1052. package/src/web/parallel.ts +353 -0
  1053. package/src/web/scrapers/artifacthub.ts +207 -0
  1054. package/src/web/scrapers/arxiv.ts +83 -0
  1055. package/src/web/scrapers/aur.ts +162 -0
  1056. package/src/web/scrapers/biorxiv.ts +133 -0
  1057. package/src/web/scrapers/bluesky.ts +262 -0
  1058. package/src/web/scrapers/brew.ts +172 -0
  1059. package/src/web/scrapers/cheatsh.ts +68 -0
  1060. package/src/web/scrapers/chocolatey.ts +196 -0
  1061. package/src/web/scrapers/choosealicense.ts +95 -0
  1062. package/src/web/scrapers/cisa-kev.ts +87 -0
  1063. package/src/web/scrapers/clojars.ts +154 -0
  1064. package/src/web/scrapers/coingecko.ts +177 -0
  1065. package/src/web/scrapers/crates-io.ts +97 -0
  1066. package/src/web/scrapers/crossref.ts +136 -0
  1067. package/src/web/scrapers/devto.ts +147 -0
  1068. package/src/web/scrapers/discogs.ts +306 -0
  1069. package/src/web/scrapers/discourse.ts +197 -0
  1070. package/src/web/scrapers/dockerhub.ts +138 -0
  1071. package/src/web/scrapers/docs-rs.ts +653 -0
  1072. package/src/web/scrapers/fdroid.ts +134 -0
  1073. package/src/web/scrapers/firefox-addons.ts +191 -0
  1074. package/src/web/scrapers/flathub.ts +223 -0
  1075. package/src/web/scrapers/github-gist.ts +58 -0
  1076. package/src/web/scrapers/github.ts +704 -0
  1077. package/src/web/scrapers/gitlab.ts +401 -0
  1078. package/src/web/scrapers/go-pkg.ts +266 -0
  1079. package/src/web/scrapers/hackage.ts +140 -0
  1080. package/src/web/scrapers/hackernews.ts +189 -0
  1081. package/src/web/scrapers/hex.ts +105 -0
  1082. package/src/web/scrapers/huggingface.ts +321 -0
  1083. package/src/web/scrapers/iacr.ts +89 -0
  1084. package/src/web/scrapers/index.ts +252 -0
  1085. package/src/web/scrapers/jetbrains-marketplace.ts +159 -0
  1086. package/src/web/scrapers/lemmy.ts +203 -0
  1087. package/src/web/scrapers/lobsters.ts +175 -0
  1088. package/src/web/scrapers/mastodon.ts +292 -0
  1089. package/src/web/scrapers/maven.ts +138 -0
  1090. package/src/web/scrapers/mdn.ts +173 -0
  1091. package/src/web/scrapers/metacpan.ts +222 -0
  1092. package/src/web/scrapers/musicbrainz.ts +250 -0
  1093. package/src/web/scrapers/npm.ts +98 -0
  1094. package/src/web/scrapers/nuget.ts +183 -0
  1095. package/src/web/scrapers/nvd.ts +222 -0
  1096. package/src/web/scrapers/ollama.ts +239 -0
  1097. package/src/web/scrapers/open-vsx.ts +106 -0
  1098. package/src/web/scrapers/opencorporates.ts +292 -0
  1099. package/src/web/scrapers/openlibrary.ts +336 -0
  1100. package/src/web/scrapers/orcid.ts +286 -0
  1101. package/src/web/scrapers/osv.ts +176 -0
  1102. package/src/web/scrapers/packagist.ts +160 -0
  1103. package/src/web/scrapers/pub-dev.ts +143 -0
  1104. package/src/web/scrapers/pubmed.ts +211 -0
  1105. package/src/web/scrapers/pypi.ts +112 -0
  1106. package/src/web/scrapers/rawg.ts +110 -0
  1107. package/src/web/scrapers/readthedocs.ts +120 -0
  1108. package/src/web/scrapers/reddit.ts +95 -0
  1109. package/src/web/scrapers/repology.ts +251 -0
  1110. package/src/web/scrapers/rfc.ts +201 -0
  1111. package/src/web/scrapers/rubygems.ts +103 -0
  1112. package/src/web/scrapers/searchcode.ts +189 -0
  1113. package/src/web/scrapers/sec-edgar.ts +261 -0
  1114. package/src/web/scrapers/semantic-scholar.ts +171 -0
  1115. package/src/web/scrapers/snapcraft.ts +187 -0
  1116. package/src/web/scrapers/sourcegraph.ts +336 -0
  1117. package/src/web/scrapers/spdx.ts +108 -0
  1118. package/src/web/scrapers/spotify.ts +198 -0
  1119. package/src/web/scrapers/stackoverflow.ts +120 -0
  1120. package/src/web/scrapers/terraform.ts +277 -0
  1121. package/src/web/scrapers/tldr.ts +47 -0
  1122. package/src/web/scrapers/twitter.ts +94 -0
  1123. package/src/web/scrapers/types.ts +397 -0
  1124. package/src/web/scrapers/utils.ts +109 -0
  1125. package/src/web/scrapers/vimeo.ts +133 -0
  1126. package/src/web/scrapers/vscode-marketplace.ts +187 -0
  1127. package/src/web/scrapers/w3c.ts +156 -0
  1128. package/src/web/scrapers/wikidata.ts +344 -0
  1129. package/src/web/scrapers/wikipedia.ts +84 -0
  1130. package/src/web/scrapers/youtube.ts +325 -0
  1131. package/src/web/search/index.ts +292 -0
  1132. package/src/web/search/provider.ts +157 -0
  1133. package/src/web/search/providers/anthropic.ts +318 -0
  1134. package/src/web/search/providers/base.ts +89 -0
  1135. package/src/web/search/providers/brave.ts +152 -0
  1136. package/src/web/search/providers/codex.ts +591 -0
  1137. package/src/web/search/providers/exa.ts +400 -0
  1138. package/src/web/search/providers/gemini.ts +460 -0
  1139. package/src/web/search/providers/jina.ts +111 -0
  1140. package/src/web/search/providers/kagi.ts +86 -0
  1141. package/src/web/search/providers/kimi.ts +196 -0
  1142. package/src/web/search/providers/parallel.ts +225 -0
  1143. package/src/web/search/providers/perplexity.ts +730 -0
  1144. package/src/web/search/providers/searxng.ts +313 -0
  1145. package/src/web/search/providers/synthetic.ts +114 -0
  1146. package/src/web/search/providers/tavily.ts +176 -0
  1147. package/src/web/search/providers/utils.ts +128 -0
  1148. package/src/web/search/providers/zai.ts +333 -0
  1149. package/src/web/search/render.ts +262 -0
  1150. package/src/web/search/types.ts +482 -0
  1151. package/src/web/search/utils.ts +17 -0
  1152. package/src/workspace-tree.ts +286 -0
package/src/sdk.ts ADDED
@@ -0,0 +1,2558 @@
1
+ import {
2
+ Agent,
3
+ type AgentEvent,
4
+ type AgentMessage,
5
+ type AgentTelemetryConfig,
6
+ type AgentTool,
7
+ AppendOnlyContextManager,
8
+ INTENT_FIELD,
9
+ type ThinkingLevel,
10
+ } from "@oh-my-pi/pi-agent-core";
11
+ import {
12
+ type CredentialDisabledEvent,
13
+ type Message,
14
+ type Model,
15
+ type SimpleStreamOptions,
16
+ streamSimple,
17
+ } from "@oh-my-pi/pi-ai";
18
+ import {
19
+ getOpenAICodexTransportDetails,
20
+ prewarmOpenAICodexResponses,
21
+ } from "@oh-my-pi/pi-ai/providers/openai-codex-responses";
22
+ import { DEFAULT_MODEL_PER_PROVIDER } from "@oh-my-pi/pi-catalog/provider-models";
23
+ import type { Component } from './stubs/tui/index.ts';
24
+ import {
25
+ $env,
26
+ $flag,
27
+ getAgentDbPath,
28
+ getAgentDir,
29
+ getAuthBrokerSnapshotCachePath,
30
+ getProjectDir,
31
+ logger,
32
+ postmortem,
33
+ prompt,
34
+ Snowflake,
35
+ } from "@oh-my-pi/pi-utils";
36
+ import chalk from "chalk";
37
+ import { type AsyncJob, AsyncJobManager } from "./async";
38
+ import { loadCapability } from "./capability";
39
+ import { type Rule, ruleCapability, setActiveRules } from "./capability/rule";
40
+ import { bucketRules } from "./capability/rule-buckets";
41
+ import { createApiKeyResolver } from "./config/api-key-resolver";
42
+ import { shouldEnableAppendOnlyContext } from "./config/append-only-context-mode";
43
+ import { ModelRegistry } from "./config/model-registry";
44
+ import {
45
+ formatModelString,
46
+ getModelMatchPreferences,
47
+ parseModelPattern,
48
+ parseModelString,
49
+ resolveAllowedModels,
50
+ resolveModelRoleValue,
51
+ } from "./config/model-resolver";
52
+ import { loadPromptTemplates as loadPromptTemplatesInternal, type PromptTemplate } from "./config/prompt-templates";
53
+ import { Settings, type SkillsSettings } from "./config/settings";
54
+ import { CursorExecHandlers } from "./cursor";
55
+ import "./discovery";
56
+ import { resolveConfigValue } from "./config/resolve-config-value";
57
+ import { initializeWithSettings } from "./discovery";
58
+ import { disposeAllKernelSessions, disposeKernelSessionsByOwner } from "./eval/py/executor";
59
+ import { defaultEvalSessionId } from "./eval/session-id";
60
+ import {
61
+ type CustomCommandsLoadResult,
62
+ type LoadedCustomCommand,
63
+ loadCustomCommands as loadCustomCommandsInternal,
64
+ } from "./extensibility/custom-commands";
65
+ import { discoverCustomToolPaths, loadCustomTools, type ToolPathWithSource } from "./extensibility/custom-tools";
66
+ import type { CustomTool, CustomToolContext, CustomToolSessionEvent } from "./extensibility/custom-tools/types";
67
+ import {
68
+ discoverAndLoadExtensions,
69
+ discoverExtensionPaths,
70
+ type ExtensionContext,
71
+ type ExtensionFactory,
72
+ ExtensionRunner,
73
+ ExtensionToolWrapper,
74
+ type ExtensionUIContext,
75
+ type LoadExtensionsResult,
76
+ loadExtensionFromFactory,
77
+ loadExtensions,
78
+ type ToolDefinition,
79
+ wrapRegisteredTools,
80
+ } from "./extensibility/extensions";
81
+ import {
82
+ loadSkills as loadSkillsInternal,
83
+ type Skill,
84
+ type SkillWarning,
85
+ setActiveSkills,
86
+ } from "./extensibility/skills";
87
+ import { type FileSlashCommand, loadSlashCommands as loadSlashCommandsInternal } from "./extensibility/slash-commands";
88
+ import type { HindsightSessionState } from "./hindsight/state";
89
+ import { LocalProtocolHandler, type LocalProtocolOptions } from "./internal-urls";
90
+ import { LSP_STARTUP_EVENT_CHANNEL, type LspStartupEvent } from "./lsp/startup-events";
91
+ import { discoverAndLoadMCPTools, MCPManager, type MCPToolsLoadResult } from "./mcp";
92
+ import { createSessionMemoryRuntimeContext, resolveMemoryBackend } from "./memory-backend";
93
+ import type { MnemopiSessionState } from "./mnemopi/state";
94
+ import asyncResultTemplate from "./prompts/tools/async-result.md" with { type: "text" };
95
+ import lateDiagnosticTemplate from "./prompts/tools/lsp-late-diagnostic.md" with { type: "text" };
96
+ import { AgentLifecycleManager } from "./registry/agent-lifecycle";
97
+ import { AgentRegistry, MAIN_AGENT_ID } from "./registry/agent-registry";
98
+ import {
99
+ collectEnvSecrets,
100
+ deobfuscateSessionContext,
101
+ loadSecrets,
102
+ obfuscateMessages,
103
+ obfuscateProviderContext,
104
+ SecretObfuscator,
105
+ } from "./secrets";
106
+ import { AgentSession } from "./session/agent-session";
107
+ import { resolveAuthBrokerConfig } from "./session/auth-broker-config";
108
+ import {
109
+ AuthBrokerClient,
110
+ AuthStorage,
111
+ DEFAULT_SNAPSHOT_CACHE_TTL_MS,
112
+ RemoteAuthCredentialStore,
113
+ readAuthBrokerSnapshotCache,
114
+ type SnapshotResponse,
115
+ writeAuthBrokerSnapshotCache,
116
+ } from "./session/auth-storage";
117
+ import {
118
+ type CustomMessage,
119
+ convertToLlm,
120
+ LSP_LATE_DIAGNOSTIC_MESSAGE_TYPE,
121
+ wrapSteeringForModel,
122
+ } from "./session/messages";
123
+ import { getRestorableSessionModels, SessionManager } from "./session/session-manager";
124
+ import { closeAllConnections } from "./ssh/connection-manager";
125
+ import { unmountAll } from "./ssh/sshfs-mount";
126
+ import {
127
+ type BuildSystemPromptResult,
128
+ buildSystemPrompt as buildSystemPromptInternal,
129
+ buildSystemPromptToolMetadata,
130
+ loadProjectContextFiles as loadContextFilesInternal,
131
+ } from "./system-prompt";
132
+ import { AgentOutputManager } from "./task/output-manager";
133
+ import {
134
+ AUTO_THINKING,
135
+ type ConfiguredThinkingLevel,
136
+ parseThinkingLevel,
137
+ resolveProvisionalAutoLevel,
138
+ resolveThinkingLevelForModel,
139
+ shouldDisableReasoning,
140
+ toReasoningEffort,
141
+ } from "./thinking";
142
+ import { countToolsForAutoDiscovery, resolveEffectiveToolDiscoveryMode } from "./tool-discovery/mode";
143
+ import {
144
+ collectDiscoverableTools,
145
+ type DiscoverableTool,
146
+ filterBySource,
147
+ formatDiscoverableToolServerSummary,
148
+ selectDiscoverableToolNamesByServer,
149
+ summarizeDiscoverableTools,
150
+ } from "./tool-discovery/tool-index";
151
+ import {
152
+ BashTool,
153
+ BUILTIN_TOOLS,
154
+ computeEssentialBuiltinNames,
155
+ createTools,
156
+ type DeferredDiagnosticsEntry,
157
+ discoverStartupLspServers,
158
+ EditTool,
159
+ EvalTool,
160
+ FindTool,
161
+ filterInitialToolsForDiscoveryAll,
162
+ getSearchTools,
163
+ HIDDEN_TOOLS,
164
+ isImageProviderPreference,
165
+ isSearchProviderPreference,
166
+ type LspStartupServerInfo,
167
+ loadSshTool,
168
+ ReadTool,
169
+ ResolveTool,
170
+ renderSearchToolBm25Description,
171
+ SearchTool,
172
+ SearchToolBm25Tool,
173
+ setPreferredImageProvider,
174
+ setPreferredSearchProvider,
175
+ type Tool,
176
+ type ToolSession,
177
+ WebSearchTool,
178
+ WriteTool,
179
+ warmupLspServers,
180
+ } from "./tools";
181
+ import { ToolContextStore } from "./tools/context";
182
+ import { getImageGenTools } from "./tools/image-gen";
183
+ import { wrapToolWithMetaNotice } from "./tools/output-meta";
184
+ import { queueResolveHandler } from "./tools/resolve";
185
+ import { ttsTool } from "./tools/tts";
186
+ import { EventBus } from "./utils/event-bus";
187
+ import { buildNamedToolChoice } from "./utils/tool-choice";
188
+ import { buildWorkspaceTree, type WorkspaceTree } from "./workspace-tree";
189
+
190
+ type AsyncResultEntry = {
191
+ jobId: string;
192
+ result: string;
193
+ job: AsyncJob | undefined;
194
+ durationMs: number | undefined;
195
+ };
196
+
197
+ type AsyncResultJobDetails = {
198
+ jobId: string;
199
+ type?: "bash" | "task";
200
+ label?: string;
201
+ durationMs?: number;
202
+ };
203
+
204
+ type AsyncResultDetails = {
205
+ jobs: AsyncResultJobDetails[];
206
+ };
207
+
208
+ type McpNotificationEntry = {
209
+ serverName: string;
210
+ uri: string;
211
+ };
212
+
213
+ function buildAsyncResultBatchMessage(entries: AsyncResultEntry[]): CustomMessage<AsyncResultDetails> | null {
214
+ if (entries.length === 0) return null;
215
+ const jobs = entries.map(entry => ({
216
+ jobId: entry.jobId,
217
+ result: entry.result,
218
+ type: entry.job?.type,
219
+ label: entry.job?.label,
220
+ durationMs: entry.durationMs,
221
+ }));
222
+ const details: AsyncResultDetails = {
223
+ jobs: jobs.map(job => ({
224
+ jobId: job.jobId,
225
+ type: job.type,
226
+ label: job.label,
227
+ durationMs: job.durationMs,
228
+ })),
229
+ };
230
+ return {
231
+ role: "custom",
232
+ customType: "async-result",
233
+ content: prompt.render(asyncResultTemplate, {
234
+ multiple: jobs.length > 1,
235
+ jobs,
236
+ }),
237
+ display: true,
238
+ attribution: "agent",
239
+ details,
240
+ timestamp: Date.now(),
241
+ };
242
+ }
243
+
244
+ type LateDiagnosticsDetails = {
245
+ files: Array<{ path: string; summary: string; errored: boolean; messages: string[] }>;
246
+ };
247
+
248
+ function buildLateDiagnosticsBatchMessage(
249
+ entries: DeferredDiagnosticsEntry[],
250
+ ): CustomMessage<LateDiagnosticsDetails> | null {
251
+ if (entries.length === 0) return null;
252
+ const files = entries.map(entry => ({
253
+ path: entry.path,
254
+ summary: entry.summary,
255
+ messages: entry.messages,
256
+ errored: entry.errored,
257
+ }));
258
+ const details: LateDiagnosticsDetails = {
259
+ files: files.map(file => ({
260
+ path: file.path,
261
+ summary: file.summary,
262
+ errored: file.errored,
263
+ messages: file.messages,
264
+ })),
265
+ };
266
+ return {
267
+ role: "custom",
268
+ customType: LSP_LATE_DIAGNOSTIC_MESSAGE_TYPE,
269
+ content: prompt.render(lateDiagnosticTemplate, {
270
+ multiple: files.length > 1,
271
+ files,
272
+ }),
273
+ display: true,
274
+ attribution: "agent",
275
+ details,
276
+ timestamp: Date.now(),
277
+ };
278
+ }
279
+
280
+ function buildMcpNotificationBatchMessage(entries: McpNotificationEntry[]): AgentMessage | null {
281
+ const resources: McpNotificationEntry[] = [];
282
+ const seen = new Set<string>();
283
+ for (const entry of entries) {
284
+ const key = `${entry.serverName}\0${entry.uri}`;
285
+ if (seen.has(key)) continue;
286
+ seen.add(key);
287
+ resources.push(entry);
288
+ }
289
+ if (resources.length === 0) return null;
290
+ const lines = [`[MCP notification] ${resources.length} resource(s) updated:`];
291
+ for (const resource of resources) {
292
+ lines.push(`- server="${resource.serverName}" uri=${resource.uri}`);
293
+ }
294
+ lines.push('Use read(path="mcp://<uri>") to inspect if relevant.');
295
+ return {
296
+ role: "user",
297
+ content: [{ type: "text", text: lines.join("\n") }],
298
+ attribution: "agent",
299
+ timestamp: Date.now(),
300
+ };
301
+ }
302
+
303
+ // Types
304
+ export interface CreateAgentSessionOptions {
305
+ /** Working directory for project-local discovery. Default: getProjectDir() */
306
+ cwd?: string;
307
+ /** Global config directory. Default: ~/.omp/agent */
308
+ agentDir?: string;
309
+ /** Spawns to allow. Default: "*" */
310
+ spawns?: string;
311
+
312
+ /** Auth storage for credentials. Default: discoverAuthStorage(agentDir) */
313
+ authStorage?: AuthStorage;
314
+ /** Model registry. Default: discoverModels(authStorage, agentDir) */
315
+ modelRegistry?: ModelRegistry;
316
+
317
+ /** Model to use. Default: from settings, else first available */
318
+ model?: Model;
319
+ /** Raw model pattern string (e.g. from --model CLI flag) to resolve after extensions load.
320
+ * Used when model lookup is deferred because extension-provided models aren't registered yet. */
321
+ modelPattern?: string;
322
+ /** Thinking selector. Default: from settings, else unset */
323
+ thinkingLevel?: ConfiguredThinkingLevel;
324
+ /** Models available for cycling (Ctrl+P in interactive mode) */
325
+ scopedModels?: Array<{ model: Model; thinkingLevel?: ThinkingLevel }>;
326
+
327
+ /** System prompt blocks. Array replaces default, function receives default blocks and returns final blocks. */
328
+ systemPrompt?: string[] | ((defaultPrompt: string[]) => string[]);
329
+ /** Optional provider-facing session identifier for prompt caches and sticky auth selection.
330
+ * Keeps persisted session files isolated while reusing provider-side caches. */
331
+ providerSessionId?: string;
332
+ /** Optional provider-facing prompt cache key, distinct from request lineage. */
333
+ providerPromptCacheKey?: string;
334
+
335
+ /** Custom tools to register (in addition to built-in tools). Accepts both CustomTool and ToolDefinition. */
336
+ customTools?: (CustomTool | ToolDefinition)[];
337
+ /** Inline extensions (merged with discovery). */
338
+ extensions?: ExtensionFactory[];
339
+ /** Additional extension paths to load (merged with discovery). */
340
+ additionalExtensionPaths?: string[];
341
+ /** Disable extension discovery (explicit paths still load). */
342
+ disableExtensionDiscovery?: boolean;
343
+ /**
344
+ * Pre-loaded extensions (skips file discovery and the per-session factory
345
+ * call). Used by the CLI when extensions are loaded early to parse custom
346
+ * flags — the same process owns the returned instances, so reusing them is
347
+ * safe.
348
+ *
349
+ * NEVER pass this across session boundaries (e.g. parent → subagent).
350
+ * `Extension` instances close over a parent-bound `ExtensionAPI` (cwd,
351
+ * eventBus, runtime), and reusing them would route tools/handlers/commands
352
+ * back through the parent. For subagents, forward
353
+ * {@link preloadedExtensionPaths} instead.
354
+ *
355
+ * @internal
356
+ */
357
+ preloadedExtensions?: LoadExtensionsResult;
358
+ /**
359
+ * Pre-discovered extension source paths. When provided, the filesystem-scan
360
+ * inside `discoverExtensionPaths()` is skipped — the session still calls
361
+ * `loadExtensions()` itself so each `Extension` is bound to THIS session's
362
+ * `ExtensionAPI` (cwd, eventBus, runtime).
363
+ *
364
+ * This is the safe pass-through for parent → subagent forwarding.
365
+ */
366
+ preloadedExtensionPaths?: string[];
367
+ /**
368
+ * Pre-discovered custom-tool source paths from `.omp/tools/`, `.claude/tools/`,
369
+ * plugins, etc. When provided, the filesystem-scan inside
370
+ * `discoverCustomToolPaths()` is skipped — subagents inherit the parent's
371
+ * scan result and call `loadCustomTools()` themselves so each session binds
372
+ * tools to its OWN `CustomToolAPI` (cwd, exec, pushPendingAction, UI).
373
+ *
374
+ * Forwarding the loaded `LoadedCustomTool[]` instances directly would reuse
375
+ * the parent's session-bound API and route tool execution back through the
376
+ * parent — wrong for isolated tasks and for pending-action routing.
377
+ */
378
+ preloadedCustomToolPaths?: ToolPathWithSource[];
379
+
380
+ /** Shared event bus for tool/extension communication. Default: creates new bus. */
381
+ eventBus?: EventBus;
382
+
383
+ /** Skills. Default: discovered from multiple locations */
384
+ skills?: Skill[];
385
+ /** Rules. Default: discovered from multiple locations */
386
+ rules?: Rule[];
387
+ /** Context files (AGENTS.md content). Default: discovered walking up from cwd */
388
+ contextFiles?: Array<{ path: string; content: string }>;
389
+ /** Pre-built workspace tree (skips re-scanning; passed by parents to subagents). */
390
+ workspaceTree?: WorkspaceTree;
391
+ /** Prompt templates. Default: discovered from cwd/.omp/prompts/ + agentDir/prompts/ */
392
+ promptTemplates?: PromptTemplate[];
393
+ /** File-based slash commands. Default: discovered from commands/ directories */
394
+ slashCommands?: FileSlashCommand[];
395
+
396
+ /** Enable MCP server discovery from .mcp.json files. Default: true */
397
+ enableMCP?: boolean;
398
+ /** Existing MCP manager to reuse (skips discovery, propagates to toolSession). */
399
+ mcpManager?: MCPManager;
400
+
401
+ /** Enable LSP integration (tool, formatting, diagnostics, warmup). Default: true */
402
+ enableLsp?: boolean;
403
+ /** Skip Python kernel availability check and prelude warmup */
404
+ skipPythonPreflight?: boolean;
405
+ /** Tool names explicitly requested (enables disabled-by-default tools) */
406
+ toolNames?: string[];
407
+
408
+ /** Output schema for structured completion (subagents) */
409
+ outputSchema?: unknown;
410
+ /** Whether to include the yield tool by default */
411
+ requireYieldTool?: boolean;
412
+ /** Task recursion depth (for subagent sessions). Default: 0 */
413
+ taskDepth?: number;
414
+ /** Parent Hindsight state to alias for subagent memory tools. */
415
+ parentHindsightSessionState?: HindsightSessionState;
416
+ /** Parent Mnemopi state to alias for subagent memory tools. */
417
+ parentMnemopiSessionState?: MnemopiSessionState;
418
+ /** Pre-allocated agent identity for IRC routing. Default: "Main" for top-level, parentTaskPrefix-derived for sub. */
419
+ agentId?: string;
420
+ /** Display name for the agent in IRC. Default: "main" or "sub". */
421
+ agentDisplayName?: string;
422
+ /** Optional shared agent registry for IRC routing. Default: AgentRegistry.global(). */
423
+ agentRegistry?: AgentRegistry;
424
+ /** Parent task ID prefix for nested artifact naming (e.g., "Extensions") */
425
+ parentTaskPrefix?: string;
426
+ /** Inherited eval executor session id for subagents sharing parent eval state. */
427
+ parentEvalSessionId?: string;
428
+
429
+ /** Session manager. Default: session stored under the configured agentDir sessions root */
430
+ sessionManager?: SessionManager;
431
+
432
+ /** Override local:// protocol options for subagent local:// sharing. Default: uses the session's own artifacts dir and session ID. */
433
+ localProtocolOptions?: LocalProtocolOptions;
434
+
435
+ /** Settings instance. Default: Settings.init({ cwd, agentDir }) */
436
+ settings?: Settings;
437
+
438
+ /** Whether UI is available (enables interactive tools like ask). Default: false */
439
+ hasUI?: boolean;
440
+
441
+ /**
442
+ * Opt-in OpenTelemetry instrumentation forwarded to the underlying Agent.
443
+ * Passing `{}` enables the loop's GenAI-semantic-convention spans. See
444
+ * {@link AgentTelemetryConfig} for the full surface (hooks, content capture,
445
+ * cost estimator, agent identity).
446
+ *
447
+ * Safe to enable without an OTEL SDK registered in the host: the
448
+ * `@opentelemetry/api` package returns a no-op tracer in that case.
449
+ */
450
+ telemetry?: AgentTelemetryConfig;
451
+
452
+ /** Whether to auto-approve all tool calls (--auto-approve CLI flag). Default: false */
453
+ autoApprove?: boolean;
454
+ }
455
+
456
+ /** Result from createAgentSession */
457
+ export interface CreateAgentSessionResult {
458
+ /** The created session */
459
+ session: AgentSession;
460
+ /** Extensions result (loaded extensions + runtime) */
461
+ extensionsResult: LoadExtensionsResult;
462
+ /** Update tool UI context (interactive mode) */
463
+ setToolUIContext: (uiContext: ExtensionUIContext, hasUI: boolean) => void;
464
+ /** MCP manager for server lifecycle management (undefined if MCP disabled) */
465
+ mcpManager?: MCPManager;
466
+ /** Warning if session was restored with a different model than saved */
467
+ modelFallbackMessage?: string;
468
+ /** LSP servers detected for startup; warmup may continue in the background */
469
+ lspServers?: LspStartupServerInfo[];
470
+ /** Shared event bus for tool/extension communication */
471
+ eventBus: EventBus;
472
+ }
473
+
474
+ // Re-exports
475
+
476
+ export type { PromptTemplate } from "./config/prompt-templates";
477
+ export { Settings, type SkillsSettings } from "./config/settings";
478
+ export type { CustomCommand, CustomCommandFactory } from "./extensibility/custom-commands/types";
479
+ export type { CustomTool, CustomToolFactory } from "./extensibility/custom-tools/types";
480
+ export type * from "./extensibility/extensions";
481
+ export type { Skill } from "./extensibility/skills";
482
+ export type { FileSlashCommand } from "./extensibility/slash-commands";
483
+ export type { MCPManager, MCPServerConfig, MCPServerConnection, MCPToolsLoadResult } from "./mcp";
484
+ export type { Tool } from "./tools";
485
+ export { buildDirectoryTree, buildWorkspaceTree, type DirectoryTree, type WorkspaceTree } from "./workspace-tree";
486
+
487
+ export {
488
+ // Individual tool classes (for custom usage)
489
+ BashTool,
490
+ // Tool classes and factories
491
+ BUILTIN_TOOLS,
492
+ createTools,
493
+ EditTool,
494
+ EvalTool,
495
+ FindTool,
496
+ HIDDEN_TOOLS,
497
+ loadSshTool,
498
+ ReadTool,
499
+ ResolveTool,
500
+ SearchTool,
501
+ type ToolSession,
502
+ WebSearchTool,
503
+ WriteTool,
504
+ };
505
+
506
+ // Helper Functions
507
+
508
+ function getDefaultAgentDir(): string {
509
+ return getAgentDir();
510
+ }
511
+
512
+ function resolveSnapshotTtlMs(): number {
513
+ const raw = process.env.OMP_AUTH_BROKER_SNAPSHOT_TTL_MS;
514
+ if (raw === undefined) return DEFAULT_SNAPSHOT_CACHE_TTL_MS;
515
+ const value = raw.trim();
516
+ if (value === "") return DEFAULT_SNAPSHOT_CACHE_TTL_MS;
517
+ const ttlMs = Number(value);
518
+ if (Number.isFinite(ttlMs) && ttlMs >= 0) return ttlMs;
519
+ logger.warn("Invalid OMP_AUTH_BROKER_SNAPSHOT_TTL_MS; using default", { value: raw });
520
+ return DEFAULT_SNAPSHOT_CACHE_TTL_MS;
521
+ }
522
+
523
+ // Discovery Functions
524
+
525
+ /**
526
+ * Create an AuthStorage instance.
527
+ *
528
+ * Default: local SQLite store at `<agentDir>/agent.db`.
529
+ *
530
+ * Broker mode: when `OMP_AUTH_BROKER_URL` is set, credentials are pulled from
531
+ * a remote auth-broker over the wire. Refresh tokens never leave the broker;
532
+ * the client receives access tokens with `refresh = "__remote__"` and calls
533
+ * back into the broker through the {@link AuthStorageOptions.refreshOAuthCredential}
534
+ * override to re-mint access tokens when needed.
535
+ */
536
+ export async function discoverAuthStorage(agentDir: string = getDefaultAgentDir()): Promise<AuthStorage> {
537
+ const brokerConfigPromise = resolveAuthBrokerConfig();
538
+ const cachePath = getAuthBrokerSnapshotCachePath();
539
+ // Warm the encrypted snapshot cache into the page cache while the broker
540
+ // config resolves (it may shell out for a `!command` token). Decryption
541
+ // needs the resolved token, so the real cache read cannot start earlier.
542
+ void Bun.file(cachePath)
543
+ .arrayBuffer()
544
+ .catch(() => undefined);
545
+ const brokerConfig = await brokerConfigPromise;
546
+ if (brokerConfig) {
547
+ const client = new AuthBrokerClient({ url: brokerConfig.url, token: brokerConfig.token });
548
+ const ttlMs = resolveSnapshotTtlMs();
549
+ const persist =
550
+ ttlMs > 0
551
+ ? (snapshot: SnapshotResponse): void => {
552
+ void writeAuthBrokerSnapshotCache({
553
+ path: cachePath,
554
+ token: brokerConfig.token,
555
+ url: brokerConfig.url,
556
+ snapshot,
557
+ }).catch(error => {
558
+ logger.debug("auth-broker snapshot cache write failed", { error: String(error) });
559
+ });
560
+ }
561
+ : undefined;
562
+
563
+ let initialSnapshot: SnapshotResponse | undefined;
564
+ if (ttlMs > 0) {
565
+ initialSnapshot =
566
+ (await readAuthBrokerSnapshotCache({
567
+ path: cachePath,
568
+ token: brokerConfig.token,
569
+ url: brokerConfig.url,
570
+ ttlMs,
571
+ }).catch(error => {
572
+ logger.debug("auth-broker snapshot cache read failed", { error: String(error) });
573
+ return null;
574
+ })) ?? undefined;
575
+ }
576
+ if (!initialSnapshot) {
577
+ const initialResult = await client.fetchSnapshot();
578
+ if (initialResult.status !== 200) throw new Error("Auth broker returned no initial snapshot");
579
+ initialSnapshot = initialResult.snapshot;
580
+ persist?.(initialSnapshot);
581
+ }
582
+ const store = new RemoteAuthCredentialStore({ client, initialSnapshot, onSnapshot: persist });
583
+ // Refresh + usage hooks live on RemoteAuthCredentialStore; AuthStorage
584
+ // discovers them automatically when no explicit option overrides them.
585
+ const storage = new AuthStorage(store, {
586
+ configValueResolver: resolveConfigValue,
587
+ sourceLabel: `broker ${brokerConfig.url}`,
588
+ });
589
+ await storage.reload();
590
+ return storage;
591
+ }
592
+ const dbPath = getAgentDbPath(agentDir);
593
+ const storage = await AuthStorage.create(dbPath, {
594
+ configValueResolver: resolveConfigValue,
595
+ sourceLabel: `local ${dbPath}`,
596
+ });
597
+ await storage.reload();
598
+ return storage;
599
+ }
600
+
601
+ /**
602
+ * Discover extensions from cwd.
603
+ */
604
+ export async function discoverExtensions(cwd?: string): Promise<LoadExtensionsResult> {
605
+ const resolvedCwd = cwd ?? getProjectDir();
606
+
607
+ return discoverAndLoadExtensions([], resolvedCwd);
608
+ }
609
+
610
+ /**
611
+ * Path-only counterpart of {@link loadSessionExtensions}: the FS-heavy scan
612
+ * without the per-session module load. Subagents reuse the parent's path list
613
+ * (cached on {@link ToolSession.extensionPaths}) and rebuild Extension
614
+ * instances themselves so each session's `ExtensionAPI` (cwd, eventBus,
615
+ * runtime) is its own.
616
+ */
617
+ export async function discoverSessionExtensionPaths(
618
+ options: Pick<CreateAgentSessionOptions, "disableExtensionDiscovery" | "additionalExtensionPaths">,
619
+ cwd: string,
620
+ settings: Settings,
621
+ ): Promise<string[]> {
622
+ if (options.disableExtensionDiscovery) {
623
+ return options.additionalExtensionPaths ?? [];
624
+ }
625
+ const configuredPaths = [...(options.additionalExtensionPaths ?? []), ...(settings.get("extensions") ?? [])];
626
+ const disabledExtensionIds = settings.get("disabledExtensions") ?? [];
627
+ return discoverExtensionPaths(configuredPaths, cwd, disabledExtensionIds);
628
+ }
629
+
630
+ /**
631
+ * Load the discovered/configured extensions for a session — everything {@link
632
+ * createAgentSession} would load except the inline factory extensions it appends
633
+ * itself. Extracted so the CLI can resolve extension-registered flags (and thus
634
+ * classify `@file` arguments extension-aware) *before* a session — and its
635
+ * terminal breadcrumb — is created, then hand the result back through
636
+ * {@link CreateAgentSessionOptions.preloadedExtensions} so the work is not
637
+ * repeated. Keep this the single source of the discovery branch logic.
638
+ */
639
+ export async function loadSessionExtensions(
640
+ options: Pick<CreateAgentSessionOptions, "disableExtensionDiscovery" | "additionalExtensionPaths">,
641
+ cwd: string,
642
+ settings: Settings,
643
+ eventBus: EventBus,
644
+ ): Promise<LoadExtensionsResult> {
645
+ const paths = await discoverSessionExtensionPaths(options, cwd, settings);
646
+ const result = await logger.time("loadExtensions", loadExtensions, paths, cwd, eventBus);
647
+ for (const { path, error } of result.errors) {
648
+ logger.error("Failed to load extension", { path, error });
649
+ }
650
+ return result;
651
+ }
652
+
653
+ /**
654
+ * Discover skills from cwd and agentDir.
655
+ */
656
+ export async function discoverSkills(
657
+ cwd?: string,
658
+ _agentDir?: string,
659
+ settings?: SkillsSettings,
660
+ ): Promise<{ skills: Skill[]; warnings: SkillWarning[] }> {
661
+ return await loadSkillsInternal({
662
+ ...settings,
663
+ cwd: cwd ?? getProjectDir(),
664
+ });
665
+ }
666
+
667
+ /**
668
+ * Discover context files (AGENTS.md) walking up from cwd.
669
+ * Returns files sorted by depth (farther from cwd first, so closer files appear last/more prominent).
670
+ */
671
+ export async function discoverContextFiles(
672
+ cwd?: string,
673
+ _agentDir?: string,
674
+ ): Promise<Array<{ path: string; content: string; depth?: number }>> {
675
+ return await loadContextFilesInternal({
676
+ cwd: cwd ?? getProjectDir(),
677
+ });
678
+ }
679
+
680
+ /**
681
+ * Discover prompt templates from cwd and agentDir.
682
+ */
683
+ export async function discoverPromptTemplates(cwd?: string, agentDir?: string): Promise<PromptTemplate[]> {
684
+ return await loadPromptTemplatesInternal({
685
+ cwd: cwd ?? getProjectDir(),
686
+ agentDir: agentDir ?? getDefaultAgentDir(),
687
+ });
688
+ }
689
+
690
+ /**
691
+ * Discover file-based slash commands from commands/ directories.
692
+ */
693
+ export async function discoverSlashCommands(cwd?: string): Promise<FileSlashCommand[]> {
694
+ return loadSlashCommandsInternal({ cwd: cwd ?? getProjectDir() });
695
+ }
696
+
697
+ /**
698
+ * Discover custom commands (TypeScript slash commands) from cwd and agentDir.
699
+ */
700
+ export async function discoverCustomTSCommands(cwd?: string, agentDir?: string): Promise<CustomCommandsLoadResult> {
701
+ const resolvedCwd = cwd ?? getProjectDir();
702
+ const resolvedAgentDir = agentDir ?? getDefaultAgentDir();
703
+
704
+ return loadCustomCommandsInternal({
705
+ cwd: resolvedCwd,
706
+ agentDir: resolvedAgentDir,
707
+ });
708
+ }
709
+
710
+ /**
711
+ * Discover MCP servers from .mcp.json files.
712
+ * Returns the manager and loaded tools.
713
+ */
714
+ export async function discoverMCPServers(cwd?: string): Promise<MCPToolsLoadResult> {
715
+ const resolvedCwd = cwd ?? getProjectDir();
716
+ return discoverAndLoadMCPTools(resolvedCwd);
717
+ }
718
+
719
+ // API Key Helpers
720
+
721
+ // System Prompt
722
+
723
+ export interface BuildSystemPromptOptions {
724
+ tools?: Tool[];
725
+ skills?: Skill[];
726
+ contextFiles?: Array<{ path: string; content: string }>;
727
+ cwd?: string;
728
+ appendPrompt?: string;
729
+ repeatToolDescriptions?: boolean;
730
+ }
731
+
732
+ /**
733
+ * Build the default provider-facing system prompt blocks.
734
+ *
735
+ * The returned `systemPrompt` preserves the stable harness prompt and dynamic project context
736
+ * as separate entries so providers can cache prompt prefixes without concatenating blocks.
737
+ */
738
+ export async function buildSystemPrompt(options: BuildSystemPromptOptions = {}): Promise<BuildSystemPromptResult> {
739
+ return await buildSystemPromptInternal({
740
+ cwd: options.cwd,
741
+ skills: options.skills,
742
+ contextFiles: options.contextFiles,
743
+ appendSystemPrompt: options.appendPrompt,
744
+ repeatToolDescriptions: options.repeatToolDescriptions,
745
+ });
746
+ }
747
+
748
+ // Internal Helpers
749
+
750
+ function createCustomToolContext(ctx: ExtensionContext): CustomToolContext {
751
+ return {
752
+ sessionManager: ctx.sessionManager,
753
+ modelRegistry: ctx.modelRegistry,
754
+ model: ctx.model,
755
+ isIdle: ctx.isIdle,
756
+ hasQueuedMessages: ctx.hasPendingMessages,
757
+ abort: ctx.abort,
758
+ };
759
+ }
760
+
761
+ function isCustomTool(tool: CustomTool | ToolDefinition): tool is CustomTool {
762
+ // To distinguish, we mark converted tools with a hidden symbol property.
763
+ // If the tool doesn't have this marker, it's a CustomTool that needs conversion.
764
+ return !(tool as any).__isToolDefinition;
765
+ }
766
+
767
+ const TOOL_DEFINITION_MARKER = Symbol("__isToolDefinition");
768
+
769
+ /** Matches the truncation applied to per-server instructions inside `rebuildSystemPrompt`. */
770
+ const MAX_MCP_INSTRUCTIONS_LENGTH = 4000;
771
+
772
+ let sshCleanupRegistered = false;
773
+
774
+ async function cleanupSshResources(): Promise<void> {
775
+ const results = await Promise.allSettled([closeAllConnections(), unmountAll()]);
776
+ for (const result of results) {
777
+ if (result.status === "rejected") {
778
+ logger.warn("SSH cleanup failed", { error: String(result.reason) });
779
+ }
780
+ }
781
+ }
782
+
783
+ function registerSshCleanup(): void {
784
+ if (sshCleanupRegistered) return;
785
+ sshCleanupRegistered = true;
786
+ postmortem.register("ssh-cleanup", cleanupSshResources);
787
+ }
788
+
789
+ let pythonCleanupRegistered = false;
790
+
791
+ function registerPythonCleanup(): void {
792
+ if (pythonCleanupRegistered) return;
793
+ pythonCleanupRegistered = true;
794
+ postmortem.register("python-cleanup", disposeAllKernelSessions);
795
+ }
796
+
797
+ function customToolToDefinition(tool: CustomTool): ToolDefinition {
798
+ const definition: ToolDefinition & { [TOOL_DEFINITION_MARKER]: true } = {
799
+ name: tool.name,
800
+ label: tool.label,
801
+ description: tool.description,
802
+ parameters: tool.parameters,
803
+ hidden: tool.hidden,
804
+ deferrable: tool.deferrable,
805
+ approval: typeof tool.approval === "function" ? tool.approval.bind(tool) : tool.approval,
806
+ mcpServerName: tool.mcpServerName,
807
+ mcpToolName: tool.mcpToolName,
808
+ execute: (toolCallId, params, signal, onUpdate, ctx) =>
809
+ tool.execute(toolCallId, params, onUpdate, createCustomToolContext(ctx), signal),
810
+ onSession: tool.onSession ? (event, ctx) => tool.onSession?.(event, createCustomToolContext(ctx)) : undefined,
811
+ renderCall: tool.renderCall,
812
+ renderResult: tool.renderResult
813
+ ? (result, options, theme): Component => {
814
+ const component = tool.renderResult?.(
815
+ result,
816
+ { expanded: options.expanded, isPartial: options.isPartial, spinnerFrame: options.spinnerFrame },
817
+ theme,
818
+ );
819
+ // Return empty component if undefined to match Component type requirement
820
+ return component ?? ({ render: () => [] } as unknown as Component);
821
+ }
822
+ : undefined,
823
+ [TOOL_DEFINITION_MARKER]: true,
824
+ };
825
+ return definition;
826
+ }
827
+
828
+ function createCustomToolsExtension(tools: CustomTool[]): ExtensionFactory {
829
+ return api => {
830
+ for (const tool of tools) {
831
+ api.registerTool(customToolToDefinition(tool));
832
+ }
833
+
834
+ const runOnSession = async (event: CustomToolSessionEvent, ctx: ExtensionContext) => {
835
+ for (const tool of tools) {
836
+ if (!tool.onSession) continue;
837
+ try {
838
+ await tool.onSession(event, createCustomToolContext(ctx));
839
+ } catch (err) {
840
+ logger.warn("Custom tool onSession error", { tool: tool.name, error: String(err) });
841
+ }
842
+ }
843
+ };
844
+
845
+ api.on("session_start", async (_event, ctx) =>
846
+ runOnSession({ reason: "start", previousSessionFile: undefined }, ctx),
847
+ );
848
+ api.on("session_switch", async (event, ctx) =>
849
+ runOnSession({ reason: "switch", previousSessionFile: event.previousSessionFile }, ctx),
850
+ );
851
+ api.on("session_branch", async (event, ctx) =>
852
+ runOnSession({ reason: "branch", previousSessionFile: event.previousSessionFile }, ctx),
853
+ );
854
+ api.on("session_tree", async (_event, ctx) =>
855
+ runOnSession({ reason: "tree", previousSessionFile: undefined }, ctx),
856
+ );
857
+ api.on("session_shutdown", async (_event, ctx) =>
858
+ runOnSession({ reason: "shutdown", previousSessionFile: undefined }, ctx),
859
+ );
860
+ api.on("auto_compaction_start", async (event, ctx) =>
861
+ runOnSession({ reason: "auto_compaction_start", trigger: event.reason, action: event.action }, ctx),
862
+ );
863
+ api.on("auto_compaction_end", async (event, ctx) =>
864
+ runOnSession(
865
+ {
866
+ reason: "auto_compaction_end",
867
+ action: event.action,
868
+ result: event.result,
869
+ aborted: event.aborted,
870
+ willRetry: event.willRetry,
871
+ errorMessage: event.errorMessage,
872
+ },
873
+ ctx,
874
+ ),
875
+ );
876
+ api.on("auto_retry_start", async (event, ctx) =>
877
+ runOnSession(
878
+ {
879
+ reason: "auto_retry_start",
880
+ attempt: event.attempt,
881
+ maxAttempts: event.maxAttempts,
882
+ delayMs: event.delayMs,
883
+ errorMessage: event.errorMessage,
884
+ },
885
+ ctx,
886
+ ),
887
+ );
888
+ api.on("auto_retry_end", async (event, ctx) =>
889
+ runOnSession(
890
+ {
891
+ reason: "auto_retry_end",
892
+ success: event.success,
893
+ attempt: event.attempt,
894
+ finalError: event.finalError,
895
+ },
896
+ ctx,
897
+ ),
898
+ );
899
+ api.on("ttsr_triggered", async (event, ctx) =>
900
+ runOnSession({ reason: "ttsr_triggered", rules: event.rules }, ctx),
901
+ );
902
+ api.on("todo_reminder", async (event, ctx) =>
903
+ runOnSession(
904
+ {
905
+ reason: "todo_reminder",
906
+ todos: event.todos,
907
+ attempt: event.attempt,
908
+ maxAttempts: event.maxAttempts,
909
+ },
910
+ ctx,
911
+ ),
912
+ );
913
+ };
914
+ }
915
+
916
+ // Factory
917
+
918
+ /**
919
+ * Build LoadedCustomCommand entries for all MCP prompts across connected servers.
920
+ * These are re-created whenever prompts change (setOnPromptsChanged callback).
921
+ */
922
+ function buildMCPPromptCommands(manager: MCPManager): LoadedCustomCommand[] {
923
+ const commands: LoadedCustomCommand[] = [];
924
+ for (const serverName of manager.getConnectedServers()) {
925
+ const prompts = manager.getServerPrompts(serverName);
926
+ if (!prompts?.length) continue;
927
+ for (const prompt of prompts) {
928
+ const commandName = `${serverName}:${prompt.name}`;
929
+ commands.push({
930
+ path: `mcp:${commandName}`,
931
+ resolvedPath: `mcp:${commandName}`,
932
+ source: "bundled",
933
+ command: {
934
+ name: commandName,
935
+ description: prompt.description ?? `MCP prompt from ${serverName}`,
936
+ async execute(args: string[]) {
937
+ const promptArgs: Record<string, string> = {};
938
+ for (const arg of args) {
939
+ const eqIdx = arg.indexOf("=");
940
+ if (eqIdx > 0) {
941
+ promptArgs[arg.slice(0, eqIdx)] = arg.slice(eqIdx + 1);
942
+ }
943
+ }
944
+ const result = await manager.executePrompt(serverName, prompt.name, promptArgs);
945
+ if (!result) return "";
946
+ const parts: string[] = [];
947
+ for (const msg of result.messages) {
948
+ const contentItems = Array.isArray(msg.content) ? msg.content : [msg.content];
949
+ for (const item of contentItems) {
950
+ if (item.type === "text") {
951
+ parts.push(item.text);
952
+ } else if (item.type === "resource") {
953
+ const resource = item.resource;
954
+ if (resource.text) parts.push(resource.text);
955
+ }
956
+ }
957
+ }
958
+ return parts.join("\n\n");
959
+ },
960
+ },
961
+ });
962
+ }
963
+ }
964
+ return commands;
965
+ }
966
+ /**
967
+ * Create an AgentSession with the specified options.
968
+ *
969
+ * @example
970
+ * ```typescript
971
+ * // Minimal - uses defaults
972
+ * const { session } = await createAgentSession();
973
+ *
974
+ * // With explicit model
975
+ * import { getModel } from '@oh-my-pi/pi-ai';
976
+ * const { session } = await createAgentSession({
977
+ * model: getModel('anthropic', 'claude-opus-4-5'),
978
+ * thinkingLevel: 'high',
979
+ * });
980
+ *
981
+ * // Continue previous session
982
+ * const { session, modelFallbackMessage } = await createAgentSession({
983
+ * continueSession: true,
984
+ * });
985
+ *
986
+ * // Full control
987
+ * const { session } = await createAgentSession({
988
+ * model: myModel,
989
+ * getApiKey: async () => Bun.env.MY_KEY,
990
+ * systemPrompt: ['You are helpful.'],
991
+ * tools: codingTools({ cwd: getProjectDir() }),
992
+ * skills: [],
993
+ * sessionManager: SessionManager.inMemory(),
994
+ * });
995
+ * ```
996
+ */
997
+ export async function createAgentSession(options: CreateAgentSessionOptions = {}): Promise<CreateAgentSessionResult> {
998
+ const cwd = options.cwd ?? getProjectDir();
999
+ const agentDir = options.agentDir ?? getDefaultAgentDir();
1000
+ const eventBus = options.eventBus ?? new EventBus();
1001
+
1002
+ registerSshCleanup();
1003
+ registerPythonCleanup();
1004
+
1005
+ // Pin authStorage to modelRegistry.authStorage: ModelRegistry.getApiKey() routes refresh
1006
+ // failures through that instance, so any divergent storage handed to the bridge / mcpManager
1007
+ // / session would silently miss credential_disabled events.
1008
+ const modelRegistry =
1009
+ options.modelRegistry ??
1010
+ new ModelRegistry(options.authStorage ?? (await logger.time("discoverModels", discoverAuthStorage, agentDir)));
1011
+ const authStorage = modelRegistry.authStorage;
1012
+ if (options.authStorage && options.authStorage !== authStorage) {
1013
+ throw new Error(
1014
+ "options.authStorage and options.modelRegistry.authStorage must be the same instance when both are provided",
1015
+ );
1016
+ }
1017
+ // Subscribe before any getApiKey() call so startup model probes can't fire a
1018
+ // credential_disabled event past us. An embedder's constructor handler makes the
1019
+ // listener set non-empty from construction, which defeats AuthStorage's no-listener
1020
+ // buffer — so we can't rely on it to catch startup events for the extension runner.
1021
+ const startupCredentialDisabledEvents: CredentialDisabledEvent[] = [];
1022
+ let credentialDisabledTarget: ExtensionRunner | undefined;
1023
+ const unsubscribeCredentialDisabled: (() => void) | undefined = authStorage.onCredentialDisabled(event => {
1024
+ if (credentialDisabledTarget) {
1025
+ // Discard return: any handler error is routed through runner.onError listeners.
1026
+ void credentialDisabledTarget.emitCredentialDisabled(event);
1027
+ } else {
1028
+ startupCredentialDisabledEvents.push(event);
1029
+ }
1030
+ });
1031
+ const settings = options.settings ?? (await logger.time("settings", Settings.init, { cwd, agentDir }));
1032
+ logger.time("initializeWithSettings", initializeWithSettings, settings);
1033
+ if (!options.modelRegistry) {
1034
+ modelRegistry.refreshInBackground();
1035
+ }
1036
+ // Kick off workspace tree discovery early. The native workspace scan returns
1037
+ // both the rendered-tree input and the AGENTS.md directory-context index, so
1038
+ // startup does not perform a second recursive filesystem search. Subagents
1039
+ // inherit the parent's resolved values via options.
1040
+ const STARTUP_SCAN_DEADLINE_MS = 5000;
1041
+ const workspaceTreePromise: Promise<WorkspaceTree> = options.workspaceTree
1042
+ ? Promise.resolve(options.workspaceTree)
1043
+ : logger.time("buildWorkspaceTree", () => buildWorkspaceTree(cwd, { timeoutMs: STARTUP_SCAN_DEADLINE_MS }));
1044
+ workspaceTreePromise.catch(() => {});
1045
+
1046
+ // Independent discoveries that depend only on cwd/agentDir — kicked off in parallel and awaited
1047
+ // at their respective consumer sites. Their work can overlap with model resolution, secret loading,
1048
+ // session-context build, tool creation, MCP discovery, and extension discovery.
1049
+ const contextFilesPromise = options.contextFiles
1050
+ ? Promise.resolve(options.contextFiles)
1051
+ : logger.time("discoverContextFiles", discoverContextFiles, cwd, agentDir);
1052
+ contextFilesPromise.catch(() => {});
1053
+ const promptTemplatesPromise = options.promptTemplates
1054
+ ? Promise.resolve(options.promptTemplates)
1055
+ : logger.time("discoverPromptTemplates", discoverPromptTemplates, cwd, agentDir);
1056
+ promptTemplatesPromise.catch(() => {});
1057
+ const slashCommandsPromise = options.slashCommands
1058
+ ? Promise.resolve(options.slashCommands)
1059
+ : logger.time("discoverSlashCommands", discoverSlashCommands, cwd);
1060
+ slashCommandsPromise.catch(() => {});
1061
+ const skillsSettings = settings.getGroup("skills");
1062
+ const disabledExtensionIds = settings.get("disabledExtensions") ?? [];
1063
+ const discoveredSkillsPromise =
1064
+ options.skills === undefined
1065
+ ? logger.time("discoverSkills", discoverSkills, cwd, agentDir, {
1066
+ ...skillsSettings,
1067
+ disabledExtensions: disabledExtensionIds,
1068
+ })
1069
+ : undefined;
1070
+ discoveredSkillsPromise?.catch(() => {});
1071
+
1072
+ // Initialize provider preferences from settings
1073
+ const webSearchProvider = settings.get("providers.webSearch");
1074
+ if (typeof webSearchProvider === "string" && isSearchProviderPreference(webSearchProvider)) {
1075
+ setPreferredSearchProvider(webSearchProvider);
1076
+ }
1077
+
1078
+ const imageProvider = settings.get("providers.image");
1079
+ if (isImageProviderPreference(imageProvider)) {
1080
+ setPreferredImageProvider(imageProvider);
1081
+ }
1082
+
1083
+ const sessionManager =
1084
+ options.sessionManager ??
1085
+ logger.time("sessionManager", () =>
1086
+ SessionManager.create(cwd, SessionManager.getDefaultSessionDir(cwd, agentDir)),
1087
+ );
1088
+ const providerSessionId = options.providerSessionId ?? sessionManager.getSessionId();
1089
+ const modelApiKeyAvailability = new Map<string, boolean>();
1090
+ const getModelAvailabilityKey = (candidate: Model): string =>
1091
+ `${candidate.provider}\u0000${candidate.baseUrl ?? ""}`;
1092
+ const hasModelApiKey = async (candidate: Model): Promise<boolean> => {
1093
+ const availabilityKey = getModelAvailabilityKey(candidate);
1094
+ const cached = modelApiKeyAvailability.get(availabilityKey);
1095
+ if (cached !== undefined) {
1096
+ return cached;
1097
+ }
1098
+
1099
+ const hasKey = !!(await modelRegistry.getApiKey(candidate, providerSessionId));
1100
+ modelApiKeyAvailability.set(availabilityKey, hasKey);
1101
+ return hasKey;
1102
+ };
1103
+
1104
+ // Load and create secret obfuscator early so resumed session state and prompt warnings
1105
+ // reflect actual loaded secrets, not just the setting toggle.
1106
+ let obfuscator: SecretObfuscator | undefined;
1107
+ if (settings.get("secrets.enabled")) {
1108
+ const fileEntries = await logger.time("loadSecrets", loadSecrets, cwd, agentDir);
1109
+ const envEntries = collectEnvSecrets();
1110
+ const allEntries = [...envEntries, ...fileEntries];
1111
+ if (allEntries.length > 0) {
1112
+ obfuscator = new SecretObfuscator(allEntries);
1113
+ }
1114
+ }
1115
+ const secretsEnabled = obfuscator?.hasSecrets() === true;
1116
+
1117
+ // Check if session has existing data to restore
1118
+ const existingSession = logger.time("loadSessionContext", () =>
1119
+ deobfuscateSessionContext(sessionManager.buildSessionContext(), obfuscator),
1120
+ );
1121
+ const existingBranch = logger.time("getSessionBranch", () => sessionManager.getBranch());
1122
+ const hasExistingSession = existingBranch.length > 0;
1123
+ const hasThinkingEntry = existingBranch.some(entry => entry.type === "thinking_level_change");
1124
+ const hasServiceTierEntry = existingBranch.some(entry => entry.type === "service_tier_change");
1125
+
1126
+ const hasExplicitModel = options.model !== undefined || options.modelPattern !== undefined;
1127
+ const modelMatchPreferences = getModelMatchPreferences(settings);
1128
+ const allowedModels = await logger.time("resolveAllowedModels", () =>
1129
+ resolveAllowedModels(modelRegistry, settings, modelMatchPreferences),
1130
+ );
1131
+ const defaultRoleSpec = logger.time("resolveDefaultModelRole", () =>
1132
+ resolveModelRoleValue(settings.getModelRole("default"), allowedModels, {
1133
+ settings,
1134
+ matchPreferences: modelMatchPreferences,
1135
+ modelRegistry,
1136
+ }),
1137
+ );
1138
+ let model = options.model;
1139
+ let modelFallbackMessage: string | undefined;
1140
+ // Identify session model strings to restore in fallback order. We do an
1141
+ // initial pass here so model-dependent setup (thinking-level resolution,
1142
+ // host preconnect) can use the restored model; extension-registered
1143
+ // providers aren't visible yet, so we retry the preferred candidates once
1144
+ // extensions register below.
1145
+ const sessionModelStrings =
1146
+ !hasExplicitModel && hasExistingSession
1147
+ ? getRestorableSessionModels(existingSession.models, sessionManager.getLastModelChangeRole())
1148
+ : [];
1149
+ let restoredSessionModelIndex = -1;
1150
+ if (!hasExplicitModel && !model && sessionModelStrings.length > 0) {
1151
+ await logger.time("restoreSessionModel", async () => {
1152
+ let failedSessionModel: string | undefined;
1153
+ for (let i = 0; i < sessionModelStrings.length; i++) {
1154
+ const sessionModelStr = sessionModelStrings[i];
1155
+ const parsedModel = parseModelString(sessionModelStr);
1156
+ if (!parsedModel) {
1157
+ failedSessionModel ??= sessionModelStr;
1158
+ continue;
1159
+ }
1160
+
1161
+ const restoredModel = modelRegistry.find(parsedModel.provider, parsedModel.id);
1162
+ if (restoredModel && (await hasModelApiKey(restoredModel))) {
1163
+ model = restoredModel;
1164
+ restoredSessionModelIndex = i;
1165
+ break;
1166
+ }
1167
+ failedSessionModel ??= sessionModelStr;
1168
+ }
1169
+ if (failedSessionModel) {
1170
+ modelFallbackMessage = `Could not restore model ${failedSessionModel}`;
1171
+ }
1172
+ });
1173
+ }
1174
+
1175
+ // If still no model, try settings default.
1176
+ // Skip settings fallback when an explicit model was requested.
1177
+ if (!hasExplicitModel && !model && defaultRoleSpec.model) {
1178
+ const settingsDefaultModel = defaultRoleSpec.model;
1179
+ logger.time("resolveSettingsDefaultModel", () => {
1180
+ // defaultRoleSpec.model already comes from modelRegistry.getAvailable(),
1181
+ // so re-validating auth here just repeats the expensive lookup path.
1182
+ model = settingsDefaultModel;
1183
+ });
1184
+ }
1185
+
1186
+ const taskDepth = options.taskDepth ?? 0;
1187
+
1188
+ // Resolves the session/agent thinking level using the same precedence we
1189
+ // apply at startup: explicit option → persisted session entry → default
1190
+ // role's explicit selector → selected model's defaultLevel → global
1191
+ // settings default. Run again after extension role reclaim so the final
1192
+ // model's own defaults aren't masked by an earlier fallback model's.
1193
+ const pickInitialThinkingLevel = (selectedModel: Model | undefined): ConfiguredThinkingLevel | undefined => {
1194
+ let level = options.thinkingLevel;
1195
+ if (level === undefined && hasExistingSession && hasThinkingEntry) {
1196
+ level = parseThinkingLevel(existingSession.thinkingLevel);
1197
+ }
1198
+ if (level === undefined && !hasExplicitModel && !hasThinkingEntry && defaultRoleSpec.explicitThinkingLevel) {
1199
+ level = defaultRoleSpec.thinkingLevel;
1200
+ }
1201
+ if (level === undefined && selectedModel?.thinking?.defaultLevel !== undefined) {
1202
+ level = selectedModel.thinking.defaultLevel;
1203
+ }
1204
+ if (level === undefined) {
1205
+ level = settings.get("defaultThinkingLevel");
1206
+ }
1207
+ return level;
1208
+ };
1209
+ let thinkingLevel = pickInitialThinkingLevel(model);
1210
+ let autoThinking = thinkingLevel === AUTO_THINKING;
1211
+ // Concrete level the agent/session start with. With `auto` this is the
1212
+ // provisional level shown until the first per-turn classification resolves;
1213
+ // `auto` itself stays a session-only concept handled by AgentSession.
1214
+ let effectiveThinkingLevel: ThinkingLevel | undefined = thinkingLevel === AUTO_THINKING ? undefined : thinkingLevel;
1215
+ if (model) {
1216
+ const resolvedModel = model;
1217
+ effectiveThinkingLevel = logger.time("resolveThinkingLevelForModel", () =>
1218
+ autoThinking
1219
+ ? resolveProvisionalAutoLevel(resolvedModel)
1220
+ : resolveThinkingLevelForModel(resolvedModel, effectiveThinkingLevel),
1221
+ );
1222
+ // Fire-and-forget TLS+H2 handshake to the model's host so it overlaps
1223
+ // with the rest of session setup (extension/skill load, tool registry,
1224
+ // system prompt build). Without this, the first `fetch(...)` pays the
1225
+ // full handshake serially — 100–300 ms transcontinental for
1226
+ // api.anthropic.com from a residential IP. Every mode benefits
1227
+ // (interactive, print, rpc, acp).
1228
+ preconnectModelHost(model.baseUrl);
1229
+ }
1230
+
1231
+ let skills: Skill[];
1232
+ let skillWarnings: SkillWarning[];
1233
+ if (options.skills !== undefined) {
1234
+ skills = options.skills;
1235
+ skillWarnings = [];
1236
+ } else {
1237
+ const discovered = await (discoveredSkillsPromise ?? Promise.resolve({ skills: [], warnings: [] }));
1238
+ skills = discovered.skills;
1239
+ skillWarnings = discovered.warnings;
1240
+ }
1241
+
1242
+ // Discover rules and bucket them in one pass to avoid repeated scans over large rule sets.
1243
+ const { ttsrManager, rulebookRules, alwaysApplyRules, allRules } = await logger.time(
1244
+ "discoverTtsrRules",
1245
+ async () => {
1246
+ const { TtsrManager } = await import("./export/ttsr");
1247
+ const ttsrSettings = settings.getGroup("ttsr");
1248
+ const ttsrManager = new TtsrManager(ttsrSettings);
1249
+ const rulesResult =
1250
+ options.rules !== undefined
1251
+ ? { items: options.rules, warnings: undefined }
1252
+ : await loadCapability<Rule>(ruleCapability.id, { cwd });
1253
+ const { rulebookRules, alwaysApplyRules } = bucketRules(rulesResult.items, ttsrManager, {
1254
+ builtinRules: ttsrSettings.builtinRules,
1255
+ disabledRules: ttsrSettings.disabledRules,
1256
+ });
1257
+ if (existingSession.injectedTtsrRules.length > 0) {
1258
+ ttsrManager.restoreInjected(existingSession.injectedTtsrRules);
1259
+ }
1260
+ return { ttsrManager, rulebookRules, alwaysApplyRules, allRules: rulesResult.items };
1261
+ },
1262
+ );
1263
+
1264
+ // Resolve contextFiles up-front (it's needed before tool creation). The
1265
+ // workspace tree scan is slow on large repos and we MUST NOT block startup on
1266
+ // it. On timeout we forward `undefined` to ToolSession; buildSystemPromptInternal
1267
+ // will re-race the same promise through its own withDeadline path. Background
1268
+ // work continues so caches still warm.
1269
+ const raceWithDeadline = async <T>(name: string, work: Promise<T>): Promise<T | undefined> => {
1270
+ let timedOut = false;
1271
+ const result = await Promise.race([
1272
+ work,
1273
+ Bun.sleep(STARTUP_SCAN_DEADLINE_MS).then(() => {
1274
+ timedOut = true;
1275
+ return undefined;
1276
+ }),
1277
+ ]);
1278
+ if (timedOut) {
1279
+ logger.warn("Startup scan exceeded deadline; deferring to system prompt fallback", {
1280
+ name,
1281
+ timeoutMs: STARTUP_SCAN_DEADLINE_MS,
1282
+ cwd,
1283
+ });
1284
+ }
1285
+ return result;
1286
+ };
1287
+ const [contextFiles, resolvedWorkspaceTree] = await Promise.all([
1288
+ contextFilesPromise,
1289
+ raceWithDeadline("buildWorkspaceTree", workspaceTreePromise),
1290
+ ]);
1291
+
1292
+ let agent: Agent;
1293
+ let session!: AgentSession;
1294
+ let hasSession = false;
1295
+ let hasRegistered = false;
1296
+ const enableLsp = options.enableLsp ?? true;
1297
+ const asyncMaxJobs = Math.min(100, Math.max(1, settings.get("async.maxJobs") ?? 100));
1298
+ const ASYNC_INLINE_RESULT_MAX_CHARS = 12_000;
1299
+ const ASYNC_PREVIEW_MAX_CHARS = 4_000;
1300
+ const formatAsyncResultForFollowUp = async (result: string): Promise<string> => {
1301
+ if (result.length <= ASYNC_INLINE_RESULT_MAX_CHARS) {
1302
+ return result;
1303
+ }
1304
+
1305
+ const preview = `${result.slice(0, ASYNC_PREVIEW_MAX_CHARS)}\n\n[Output truncated. Showing first ${ASYNC_PREVIEW_MAX_CHARS.toLocaleString()} characters.]`;
1306
+ try {
1307
+ const { path: artifactPath, id: artifactId } = await sessionManager.allocateArtifactPath("async");
1308
+ if (artifactPath && artifactId) {
1309
+ await Bun.write(artifactPath, result);
1310
+ return `${preview}\nFull output: artifact://${artifactId}`;
1311
+ }
1312
+ } catch (error) {
1313
+ logger.warn("Failed to persist async follow-up artifact", {
1314
+ error: error instanceof Error ? error.message : String(error),
1315
+ });
1316
+ }
1317
+
1318
+ return preview;
1319
+ };
1320
+ // Only the first top-level session in a process owns an AsyncJobManager.
1321
+ // Subagents inherit the parent's manager via `AsyncJobManager.instance()`
1322
+ // (set below), and any additional top-level session spun up in-process
1323
+ // (e.g. the agent-creation architect in `agent-dashboard.ts`) must share
1324
+ // the live singleton — otherwise its dispose path would clobber the
1325
+ // owning session's manager and break the `task`/`bash` async paths
1326
+ // (issue #1923). The `instance()` guard means later sessions also skip
1327
+ // constructing an orphaned manager that nothing would ever route to.
1328
+ const asyncJobManager =
1329
+ !options.parentTaskPrefix && !AsyncJobManager.instance()
1330
+ ? new AsyncJobManager({
1331
+ maxRunningJobs: asyncMaxJobs,
1332
+ onJobComplete: async (jobId, result, job) => {
1333
+ if (!session || asyncJobManager!.isDeliverySuppressed(jobId)) return;
1334
+ const formattedResult = await formatAsyncResultForFollowUp(result);
1335
+ if (asyncJobManager!.isDeliverySuppressed(jobId)) return;
1336
+
1337
+ const durationMs = job ? Math.max(0, Date.now() - job.startTime) : undefined;
1338
+ session.yieldQueue.enqueue<AsyncResultEntry>("async-result", {
1339
+ jobId,
1340
+ result: formattedResult,
1341
+ job,
1342
+ durationMs,
1343
+ });
1344
+ },
1345
+ })
1346
+ : undefined;
1347
+
1348
+ const scopedAsyncJobManager = asyncJobManager ?? (options.parentTaskPrefix ? AsyncJobManager.instance() : undefined);
1349
+
1350
+ const agentRegistry = options.agentRegistry ?? AgentRegistry.global();
1351
+ const resolvedAgentId = options.agentId ?? options.parentTaskPrefix ?? MAIN_AGENT_ID;
1352
+ const resolvedAgentDisplayName =
1353
+ options.agentDisplayName ?? ((options.taskDepth ?? 0) > 0 || options.parentTaskPrefix ? "sub" : "main");
1354
+ const agentKind = (options.taskDepth ?? 0) > 0 || options.parentTaskPrefix ? ("sub" as const) : ("main" as const);
1355
+ /**
1356
+ * Forget the agent ref on teardown — unless the agent is being parked (or is
1357
+ * already parked). Parking disposes the session but keeps the ref addressable
1358
+ * (history://, revive); only process teardown / explicit kill unregisters.
1359
+ */
1360
+ const unregisterUnlessParked = (): void => {
1361
+ if (agentRegistry.get(resolvedAgentId)?.status === "parked") return;
1362
+ if (AgentLifecycleManager.global().isParking(resolvedAgentId)) return;
1363
+ agentRegistry.unregister(resolvedAgentId);
1364
+ };
1365
+ const evalKernelOwnerId = `agent-session:${Snowflake.next()}`;
1366
+
1367
+ try {
1368
+ const getActiveModelString = (): string | undefined => {
1369
+ const activeModel = agent?.state.model;
1370
+ if (activeModel) return formatModelString(activeModel);
1371
+ if (model) return formatModelString(model);
1372
+ return undefined;
1373
+ };
1374
+ // Per-path mutation counter shared across edit/write tools. Late-diagnostics
1375
+ // entries capture it at fetch time and are dropped at injection if a newer
1376
+ // mutation (any tool) bumped it in the meantime.
1377
+ const fileMutationVersions = new Map<string, number>();
1378
+ const toolSession: ToolSession = {
1379
+ get cwd() {
1380
+ return sessionManager.getCwd();
1381
+ },
1382
+ hasUI: options.hasUI ?? false,
1383
+ enableLsp,
1384
+ get hasEditTool() {
1385
+ const requestedToolNames = options.toolNames
1386
+ ? [...new Set(options.toolNames.map(name => name.toLowerCase()))]
1387
+ : undefined;
1388
+ return !requestedToolNames || requestedToolNames.includes("edit");
1389
+ },
1390
+ skipPythonPreflight: options.skipPythonPreflight,
1391
+ contextFiles,
1392
+ workspaceTree: resolvedWorkspaceTree,
1393
+ skills,
1394
+ rules: allRules,
1395
+ eventBus,
1396
+ outputSchema: options.outputSchema,
1397
+ requireYieldTool: options.requireYieldTool,
1398
+ taskDepth: options.taskDepth ?? 0,
1399
+ getSessionFile: () => sessionManager.getSessionFile() ?? null,
1400
+ getEvalKernelOwnerId: () => evalKernelOwnerId,
1401
+ getEvalSessionId: () =>
1402
+ session?.getEvalSessionId() ?? options.parentEvalSessionId ?? defaultEvalSessionId(toolSession),
1403
+ assertEvalExecutionAllowed: () => session?.assertEvalExecutionAllowed(),
1404
+ trackEvalExecution: (execution, abortController) =>
1405
+ session ? session.trackEvalExecution(execution, abortController) : execution,
1406
+ getSessionId: () => sessionManager.getSessionId?.() ?? null,
1407
+ getHindsightSessionState: () => session?.getHindsightSessionState(),
1408
+ getMnemopiSessionState: () => session?.getMnemopiSessionState(),
1409
+ getAgentId: () => resolvedAgentId,
1410
+ getToolByName: name => session?.getToolByName(name),
1411
+ agentRegistry,
1412
+ getSessionSpawns: () => options.spawns ?? "*",
1413
+ getModelString: () => (hasExplicitModel && model ? formatModelString(model) : undefined),
1414
+ getActiveModelString,
1415
+ getPlanModeState: () => session?.getPlanModeState(),
1416
+ getPlanReferencePath: () => session?.getPlanReferencePath() ?? "local://PLAN.md",
1417
+ getGoalModeState: () => session?.getGoalModeState(),
1418
+ getGoalRuntime: () => session?.goalRuntime,
1419
+ getUsageStatistics: () => sessionManager.getUsageStatistics(),
1420
+ getTurnBudget: () => sessionManager.getTurnBudget(),
1421
+ recordEvalSubagentUsage: output => sessionManager.recordEvalSubagentOutput(output),
1422
+ getClientBridge: () => session?.clientBridge,
1423
+ queueDeferredDiagnostics: entry => session?.yieldQueue.enqueue(LSP_LATE_DIAGNOSTIC_MESSAGE_TYPE, entry),
1424
+ bumpFileMutationVersion: path => {
1425
+ const next = (fileMutationVersions.get(path) ?? 0) + 1;
1426
+ fileMutationVersions.set(path, next);
1427
+ return next;
1428
+ },
1429
+ getFileMutationVersion: path => fileMutationVersions.get(path) ?? 0,
1430
+ getTodoPhases: () => session.getTodoPhases(),
1431
+ setTodoPhases: phases => session.setTodoPhases(phases),
1432
+ isMCPDiscoveryEnabled: () => session.isMCPDiscoveryEnabled(),
1433
+ getSelectedMCPToolNames: () => session.getSelectedMCPToolNames(),
1434
+ activateDiscoveredMCPTools: toolNames => session.activateDiscoveredMCPTools(toolNames),
1435
+ // Generic tool discovery (unified — covers built-in + MCP + extension)
1436
+ isToolDiscoveryEnabled: () => session.isToolDiscoveryEnabled(),
1437
+ getDiscoverableTools: filter => session.getDiscoverableTools(filter),
1438
+ getDiscoverableToolSearchIndex: () => session.getDiscoverableToolSearchIndex(),
1439
+ getSelectedDiscoveredToolNames: () => session.getSelectedDiscoveredToolNames(),
1440
+ activateDiscoveredTools: toolNames => session.activateDiscoveredTools(toolNames),
1441
+ getCheckpointState: () => session.getCheckpointState(),
1442
+ setCheckpointState: state => session.setCheckpointState(state ?? undefined),
1443
+ getToolChoiceQueue: () => session.toolChoiceQueue,
1444
+ buildToolChoice: name => {
1445
+ const m = session.model;
1446
+ return m ? buildNamedToolChoice(name, m) : undefined;
1447
+ },
1448
+ steer: msg =>
1449
+ session.agent.steer({
1450
+ role: "custom",
1451
+ customType: msg.customType,
1452
+ content: msg.content,
1453
+ display: false,
1454
+ details: msg.details,
1455
+ attribution: "agent",
1456
+ timestamp: Date.now(),
1457
+ }),
1458
+ peekQueueInvoker: () => session.peekQueueInvoker(),
1459
+ peekStandingResolveHandler: () => session.peekStandingResolveHandler(),
1460
+ setStandingResolveHandler: handler => session.setStandingResolveHandler(handler),
1461
+ allocateOutputArtifact: async toolType => {
1462
+ try {
1463
+ return await sessionManager.allocateArtifactPath(toolType);
1464
+ } catch {
1465
+ return {};
1466
+ }
1467
+ },
1468
+ getArtifactManager: () => sessionManager.getArtifactManager(),
1469
+ settings,
1470
+ authStorage,
1471
+ modelRegistry,
1472
+ getTelemetry: () => agent?.telemetry,
1473
+ // Subagents inherit the singleton (the parent's manager) so their bash/task
1474
+ // completions still flow into the spawning conversation's yieldQueue.
1475
+ // Secondary in-process top-level sessions (no parentTaskPrefix, no
1476
+ // constructed manager because the singleton was already installed) leave
1477
+ // this undefined so tools and session job snapshots refuse async work
1478
+ // instead of silently routing into the owning session (issue #1923).
1479
+ asyncJobManager: scopedAsyncJobManager,
1480
+ };
1481
+
1482
+ // Wire process-wide internal URL singletons owned by their real classes.
1483
+ // Top-level sessions install the active snapshots; subagents inherit them.
1484
+ // Artifact and agent-output URLs resolve via `AgentRegistry.global()` —
1485
+ // the protocol handlers walk each ref's `sessionManager.getArtifactsDir()`,
1486
+ // which collapses to the parent's dir for subagents (they adopt the
1487
+ // parent's ArtifactManager) so one lookup hits everything.
1488
+ const getArtifactsDir = () => sessionManager.getArtifactsDir();
1489
+ if (!options.parentTaskPrefix) {
1490
+ setActiveSkills(skills);
1491
+ // Include TTSR rules so `rule://<name>` can resolve them too. They are
1492
+ // registered with the manager and bucketed out before rulebook/always,
1493
+ // so without this a TTSR-only rule (e.g. a triggered builtin) is not
1494
+ // addressable and `rule://` reports "Available: none".
1495
+ setActiveRules([...rulebookRules, ...alwaysApplyRules, ...ttsrManager.getRules()]);
1496
+ if (asyncJobManager) AsyncJobManager.setInstance(asyncJobManager);
1497
+ }
1498
+ const localProtocolOptions = options.localProtocolOptions ?? {
1499
+ getArtifactsDir,
1500
+ getSessionId: () => sessionManager.getSessionId?.() ?? null,
1501
+ };
1502
+ if (options.localProtocolOptions) {
1503
+ LocalProtocolHandler.setOverride(options.localProtocolOptions);
1504
+ }
1505
+ toolSession.getArtifactsDir = getArtifactsDir;
1506
+ toolSession.localProtocolOptions = localProtocolOptions;
1507
+ toolSession.agentOutputManager = new AgentOutputManager(
1508
+ getArtifactsDir,
1509
+ options.parentTaskPrefix ? { parentPrefix: options.parentTaskPrefix } : undefined,
1510
+ );
1511
+
1512
+ // Create built-in tools (already wrapped with meta notice formatting)
1513
+ const builtinTools = await logger.time("createAllTools", createTools, toolSession, options.toolNames);
1514
+
1515
+ // Discover MCP tools from .mcp.json files
1516
+ let mcpManager: MCPManager | undefined = options.mcpManager;
1517
+ toolSession.mcpManager = mcpManager;
1518
+ const enableMCP = options.enableMCP ?? true;
1519
+ const customTools: CustomTool[] = [];
1520
+ if (enableMCP && !mcpManager) {
1521
+ const mcpResult = await logger.time("discoverAndLoadMCPTools", discoverAndLoadMCPTools, cwd, {
1522
+ onConnecting: serverNames => {
1523
+ if (options.hasUI && serverNames.length > 0) {
1524
+ process.stderr.write(`${chalk.gray(`Connecting to MCP servers: ${serverNames.join(", ")}…`)}\n`);
1525
+ }
1526
+ },
1527
+ enableProjectConfig: settings.get("mcp.enableProjectConfig") ?? true,
1528
+ // Always filter Exa - we have native integration
1529
+ filterExa: true,
1530
+ // Filter browser MCP servers when builtin browser tool is active
1531
+ filterBrowser: settings.get("browser.enabled") ?? false,
1532
+ cacheStorage: settings.getStorage(),
1533
+ authStorage,
1534
+ });
1535
+ mcpManager = mcpResult.manager;
1536
+ toolSession.mcpManager = mcpManager;
1537
+
1538
+ if (settings.get("mcp.notifications")) {
1539
+ mcpManager.setNotificationsEnabled(true);
1540
+ }
1541
+ // If we extracted Exa API keys from MCP configs and EXA_API_KEY isn't set, use the first one
1542
+ if (mcpResult.exaApiKeys.length > 0 && !$env.EXA_API_KEY) {
1543
+ Bun.env.EXA_API_KEY = mcpResult.exaApiKeys[0];
1544
+ }
1545
+
1546
+ // Log MCP errors
1547
+ for (const { path, error } of mcpResult.errors) {
1548
+ logger.error("MCP tool load failed", { path, error });
1549
+ }
1550
+
1551
+ if (mcpResult.tools.length > 0) {
1552
+ // MCP tools are LoadedCustomTool, extract the tool property
1553
+ customTools.push(...mcpResult.tools.map(loaded => loaded.tool));
1554
+ }
1555
+ }
1556
+ // Only top-level sessions own the global MCPManager. Subagents already
1557
+ // receive the parent's manager via `options.mcpManager`, and reassigning
1558
+ // the singleton to the same value is a no-op \u2014 keep the gate explicit
1559
+ // to mirror the AsyncJobManager ownership rule.
1560
+ if (mcpManager && !options.parentTaskPrefix) MCPManager.setInstance(mcpManager);
1561
+
1562
+ // Add image tools when the active model or configured image providers can generate images.
1563
+ const imageGenTools = await logger.time("getImageGenTools", () => getImageGenTools(modelRegistry, model));
1564
+ if (imageGenTools.length > 0) {
1565
+ customTools.push(...(imageGenTools as unknown as CustomTool[]));
1566
+ }
1567
+
1568
+ if (settings.get("tts.enabled")) {
1569
+ customTools.push(ttsTool as unknown as CustomTool);
1570
+ }
1571
+
1572
+ // Add web search tools
1573
+ if (options.toolNames?.includes("web_search")) {
1574
+ customTools.push(...getSearchTools());
1575
+ }
1576
+
1577
+ // Discover custom tools from `.omp/tools/`, `.claude/tools/`, plugins, etc.
1578
+ // Subagents reuse the parent's scan via `preloadedCustomToolPaths` to skip
1579
+ // the FS walk, but ALWAYS re-call `loadCustomTools` here so factories bind
1580
+ // to THIS session's `CustomToolAPI` (cwd, exec, pushPendingAction, UI).
1581
+ // Forwarding the parent's `LoadedCustomTool[]` directly would route tool
1582
+ // execution back through the parent — wrong for isolated tasks and for
1583
+ // pending-action queueing.
1584
+ const builtInToolNames = builtinTools.map(t => t.name);
1585
+ const customToolPaths: ToolPathWithSource[] =
1586
+ options.preloadedCustomToolPaths ??
1587
+ (await logger.time("discoverCustomToolPaths", () => discoverCustomToolPaths([], cwd)));
1588
+ const customToolsLoadResult = await logger.time("loadCustomTools", () =>
1589
+ loadCustomTools(customToolPaths, cwd, builtInToolNames, action => queueResolveHandler(toolSession, action)),
1590
+ );
1591
+ for (const { path, error } of customToolsLoadResult.errors) {
1592
+ logger.error("Custom tool load failed", { path, error });
1593
+ }
1594
+ if (customToolsLoadResult.tools.length > 0) {
1595
+ customTools.push(...customToolsLoadResult.tools.map(loaded => loaded.tool));
1596
+ }
1597
+ // Forward the path list (NOT the loaded tools) to subagents so they
1598
+ // re-bind under their own `CustomToolAPI` while skipping the FS scan.
1599
+ toolSession.customToolPaths = customToolPaths;
1600
+
1601
+ const inlineExtensions: ExtensionFactory[] = options.extensions ? [...options.extensions] : [];
1602
+ inlineExtensions.push((await import("./autoresearch")).createAutoresearchExtension);
1603
+ if (customTools.length > 0) {
1604
+ inlineExtensions.push(createCustomToolsExtension(customTools));
1605
+ }
1606
+
1607
+ // Load extensions. Three paths:
1608
+ // 1. `preloadedExtensions` (CLI): caller already loaded — reuse the
1609
+ // Extension instances. Shallow-clone `extensions` so the inline
1610
+ // push below cannot mutate the caller's array. `runtime` is shared
1611
+ // so flag values set pre-creation flow into the live session.
1612
+ // 2. `preloadedExtensionPaths` (subagent): caller resolved paths;
1613
+ // skip the FS scan but always re-call `loadExtensions` here so
1614
+ // each `Extension` binds to THIS session's `ExtensionAPI`
1615
+ // (cwd, eventBus, runtime).
1616
+ // 3. No preload: run the full session discovery.
1617
+ // `disableExtensionDiscovery` is honored implicitly: a caller that set
1618
+ // the flag and pre-resolved the result already reflects that choice.
1619
+ let extensionPaths: string[];
1620
+ let extensionsResult: LoadExtensionsResult;
1621
+ if (options.preloadedExtensions) {
1622
+ extensionsResult = {
1623
+ ...options.preloadedExtensions,
1624
+ extensions: [...options.preloadedExtensions.extensions],
1625
+ };
1626
+ // Capture paths for downstream forwarding; filter inline-factory
1627
+ // entries (`<inline-N>`) — those are per-session, not source paths.
1628
+ extensionPaths = extensionsResult.extensions
1629
+ .map(ext => ext.resolvedPath)
1630
+ .filter(p => !p.startsWith("<inline"));
1631
+ } else if (options.preloadedExtensionPaths) {
1632
+ extensionPaths = options.preloadedExtensionPaths;
1633
+ extensionsResult = await logger.time("loadExtensions", loadExtensions, extensionPaths, cwd, eventBus);
1634
+ for (const { path, error } of extensionsResult.errors) {
1635
+ logger.error("Failed to load extension", { path, error });
1636
+ }
1637
+ } else {
1638
+ extensionPaths = await logger.time("discoverSessionExtensionPaths", () =>
1639
+ discoverSessionExtensionPaths(options, cwd, settings),
1640
+ );
1641
+ extensionsResult = await logger.time("loadExtensions", loadExtensions, extensionPaths, cwd, eventBus);
1642
+ for (const { path, error } of extensionsResult.errors) {
1643
+ logger.error("Failed to load extension", { path, error });
1644
+ }
1645
+ }
1646
+ // Forward the source-path list (NOT the loaded instances) so subagents
1647
+ // rebuild their own session-scoped extensions.
1648
+ toolSession.extensionPaths = extensionPaths;
1649
+
1650
+ // Load inline extensions from factories
1651
+ if (inlineExtensions.length > 0) {
1652
+ for (let i = 0; i < inlineExtensions.length; i++) {
1653
+ const factory = inlineExtensions[i];
1654
+ const loaded = await loadExtensionFromFactory(
1655
+ factory,
1656
+ cwd,
1657
+ eventBus,
1658
+ extensionsResult.runtime,
1659
+ `<inline-${i}>`,
1660
+ );
1661
+ extensionsResult.extensions.push(loaded);
1662
+ }
1663
+ }
1664
+
1665
+ // Process provider registrations queued during extension loading.
1666
+ // This must happen before the runner is created so that models registered by
1667
+ // extensions are available for model selection on session resume / fallback.
1668
+ const activeExtensionSources = extensionsResult.extensions.map(extension => extension.path);
1669
+ modelRegistry.syncExtensionSources(activeExtensionSources);
1670
+ for (const sourceId of new Set(activeExtensionSources)) {
1671
+ modelRegistry.clearSourceRegistrations(sourceId);
1672
+ }
1673
+ if (extensionsResult.runtime.pendingProviderRegistrations.length > 0) {
1674
+ for (const { name, config, sourceId } of extensionsResult.runtime.pendingProviderRegistrations) {
1675
+ modelRegistry.registerProvider(name, config, sourceId);
1676
+ }
1677
+ extensionsResult.runtime.pendingProviderRegistrations = [];
1678
+ }
1679
+ // Discover runtime (extension) provider catalogs now that they are
1680
+ // registered. The startup refreshInBackground() ran before extensions
1681
+ // loaded, so dynamic extension providers are only discovered here. Runs in
1682
+ // the background (cache-aware) so startup is never blocked on the fetch; the
1683
+ // model list re-renders when the catalog arrives, like other dynamic providers.
1684
+ void modelRegistry.refreshRuntimeProviders().catch(error => {
1685
+ logger.warn("runtime provider discovery failed", {
1686
+ error: error instanceof Error ? error.message : String(error),
1687
+ });
1688
+ });
1689
+
1690
+ // Retry session-model candidates now that extension providers are
1691
+ // registered. The initial restore runs before extensions load, so a role
1692
+ // model supplied by an extension would have either fallen back to the
1693
+ // saved default (`restoredSessionModelIndex > 0`) or failed entirely
1694
+ // (`restoredSessionModelIndex === -1`, with the settings default or
1695
+ // downstream fallback filling `model`). Reclaim it here so resume
1696
+ // honors the last active role in either case.
1697
+ const sessionRetryLimit = restoredSessionModelIndex >= 0 ? restoredSessionModelIndex : sessionModelStrings.length;
1698
+ if (!hasExplicitModel && sessionRetryLimit > 0) {
1699
+ for (let i = 0; i < sessionRetryLimit; i++) {
1700
+ const sessionModelStr = sessionModelStrings[i];
1701
+ const parsedModel = parseModelString(sessionModelStr);
1702
+ if (!parsedModel) continue;
1703
+ const restoredModel = modelRegistry.find(parsedModel.provider, parsedModel.id);
1704
+ if (restoredModel && (await hasModelApiKey(restoredModel))) {
1705
+ model = restoredModel;
1706
+ modelFallbackMessage = undefined;
1707
+ restoredSessionModelIndex = i;
1708
+ // Recompute thinking-level from scratch against the reclaimed
1709
+ // model: any value derived from the earlier fallback model's
1710
+ // `thinking.defaultLevel` must not become sticky.
1711
+ thinkingLevel = pickInitialThinkingLevel(restoredModel);
1712
+ autoThinking = thinkingLevel === AUTO_THINKING;
1713
+ effectiveThinkingLevel = thinkingLevel === AUTO_THINKING ? undefined : thinkingLevel;
1714
+ effectiveThinkingLevel = logger.time("resolveThinkingLevelForModel", () =>
1715
+ autoThinking
1716
+ ? resolveProvisionalAutoLevel(restoredModel)
1717
+ : resolveThinkingLevelForModel(restoredModel, effectiveThinkingLevel),
1718
+ );
1719
+ preconnectModelHost(restoredModel.baseUrl);
1720
+ break;
1721
+ }
1722
+ }
1723
+ }
1724
+ // Resolve deferred --model pattern now that extension models are registered.
1725
+ if (!model && options.modelPattern) {
1726
+ const availableModels = modelRegistry.getAll();
1727
+ const matchPreferences = getModelMatchPreferences(settings);
1728
+ const { model: resolved } = parseModelPattern(options.modelPattern, availableModels, matchPreferences, {
1729
+ modelRegistry,
1730
+ });
1731
+ if (resolved) {
1732
+ model = resolved;
1733
+ modelFallbackMessage = undefined;
1734
+ } else {
1735
+ modelFallbackMessage = `Model "${options.modelPattern}" not found`;
1736
+ }
1737
+ }
1738
+
1739
+ // Fall back to first available model with a valid API key, honoring the
1740
+ // path-scoped `enabledModels` allow-list when configured. Skip when the
1741
+ // user explicitly requested a model via --model that wasn't found.
1742
+ if (!model && !options.modelPattern) {
1743
+ // Re-resolve the allowed set: extension factories above may have
1744
+ // registered providers/models that weren't visible at startup.
1745
+ const fallbackCandidates = await resolveAllowedModels(modelRegistry, settings, modelMatchPreferences);
1746
+ // Prefer each provider's configured default model
1747
+ // (DEFAULT_MODEL_PER_PROVIDER) over raw catalog order. Without this the
1748
+ // first-run fallback picks whatever model sorts first in models.json for
1749
+ // the winning provider (e.g. anthropic's claude-3-5-sonnet-20240620)
1750
+ // instead of the intended provider default (claude-sonnet-4-6). Mirrors
1751
+ // findInitialModel's precedence.
1752
+ for (const [provider, defaultId] of Object.entries(DEFAULT_MODEL_PER_PROVIDER)) {
1753
+ const preferred = fallbackCandidates.find(
1754
+ candidate => candidate.provider === provider && candidate.id === defaultId,
1755
+ );
1756
+ if (preferred && (await hasModelApiKey(preferred))) {
1757
+ model = preferred;
1758
+ break;
1759
+ }
1760
+ }
1761
+ // Otherwise, first available model with a valid API key.
1762
+ if (!model) {
1763
+ for (const candidate of fallbackCandidates) {
1764
+ if (await hasModelApiKey(candidate)) {
1765
+ model = candidate;
1766
+ break;
1767
+ }
1768
+ }
1769
+ }
1770
+ if (model) {
1771
+ if (modelFallbackMessage) {
1772
+ modelFallbackMessage += `. Using ${model.provider}/${model.id}`;
1773
+ }
1774
+ } else {
1775
+ const patterns = settings.get("enabledModels");
1776
+ modelFallbackMessage =
1777
+ patterns && patterns.length > 0
1778
+ ? `No model available matching enabledModels (${patterns.join(", ")}) with usable credentials. Configure auth for an allowed provider or adjust enabledModels.`
1779
+ : "No models available. Use /login or set an API key environment variable. Then use /model to select a model.";
1780
+ }
1781
+ }
1782
+
1783
+ // Discover custom commands (TypeScript slash commands)
1784
+ const customCommandsResult: CustomCommandsLoadResult = options.disableExtensionDiscovery
1785
+ ? { commands: [], errors: [] }
1786
+ : await logger.time("discoverCustomCommands", loadCustomCommandsInternal, { cwd, agentDir });
1787
+ if (!options.disableExtensionDiscovery) {
1788
+ for (const { path, error } of customCommandsResult.errors) {
1789
+ logger.error("Failed to load custom command", { path, error });
1790
+ }
1791
+ }
1792
+
1793
+ // The runner is created unconditionally — even with zero extensions loaded — because the
1794
+ // `ExtensionToolWrapper` installed below is the only place the per-tool approval gate runs.
1795
+ // A conditional runner means the approval system silently disappears for users with no
1796
+ // extensions, contradicting non-yolo `tools.approvalMode` settings without feedback.
1797
+ // (The builtin autoresearch extension is unconditionally loaded above, so this scenario
1798
+ // is unreachable; unconditional runner construction keeps that invariant explicit and
1799
+ // prevents future optional extensions from silently re-opening the hole.)
1800
+ const extensionRunner: ExtensionRunner = new ExtensionRunner(
1801
+ extensionsResult.extensions,
1802
+ extensionsResult.runtime,
1803
+ cwd,
1804
+ sessionManager,
1805
+ modelRegistry,
1806
+ () => (hasSession ? createSessionMemoryRuntimeContext(session, agentDir, cwd) : undefined),
1807
+ );
1808
+
1809
+ credentialDisabledTarget = extensionRunner;
1810
+ for (const event of startupCredentialDisabledEvents.splice(0)) {
1811
+ // Discard return: any handler error is routed through runner.onError listeners.
1812
+ void extensionRunner.emitCredentialDisabled(event);
1813
+ }
1814
+
1815
+ const getSessionContext = () => ({
1816
+ sessionManager,
1817
+ modelRegistry,
1818
+ model: agent.state.model,
1819
+ isIdle: () => !session.isStreaming,
1820
+ hasQueuedMessages: () => session.queuedMessageCount > 0,
1821
+ abort: () => {
1822
+ session.abort();
1823
+ },
1824
+ settings,
1825
+ autoApprove: options.autoApprove ?? false,
1826
+ });
1827
+ const toolContextStore = new ToolContextStore(getSessionContext);
1828
+
1829
+ const registeredTools = extensionRunner.getAllRegisteredTools();
1830
+ const allCustomTools = [
1831
+ ...registeredTools,
1832
+ ...(options.customTools?.map(tool => {
1833
+ const definition = isCustomTool(tool) ? customToolToDefinition(tool) : tool;
1834
+ return { definition, extensionPath: "<sdk>" };
1835
+ }) ?? []),
1836
+ ];
1837
+ const wrappedExtensionTools: Tool[] = wrapRegisteredTools(allCustomTools, extensionRunner);
1838
+
1839
+ // All built-in tools are active (conditional tools like git/ask return null from factory if disabled)
1840
+ const toolRegistry = new Map<string, Tool>();
1841
+ for (const tool of builtinTools) {
1842
+ toolRegistry.set(tool.name, tool);
1843
+ }
1844
+ if (!toolRegistry.has("goal") && settings.get("goal.enabled")) {
1845
+ const goalTool = await logger.time("createTools:goal:session", HIDDEN_TOOLS.goal, toolSession);
1846
+ if (goalTool) {
1847
+ toolRegistry.set(goalTool.name, wrapToolWithMetaNotice(goalTool));
1848
+ }
1849
+ }
1850
+ for (const tool of wrappedExtensionTools) {
1851
+ toolRegistry.set(tool.name, tool);
1852
+ }
1853
+ // Wrap every tool with `ExtensionToolWrapper` so the per-tool approval gate runs on every
1854
+ // call site, regardless of whether any user extensions are loaded. See the runner-construction
1855
+ // comment above for the safety invariant this enforces.
1856
+ for (const tool of toolRegistry.values()) {
1857
+ toolRegistry.set(tool.name, new ExtensionToolWrapper(tool, extensionRunner));
1858
+ }
1859
+ if (model?.provider === "cursor") {
1860
+ toolRegistry.delete("edit");
1861
+ }
1862
+
1863
+ // `resolve` is hidden but must stay in the registry whenever any code path can invoke it:
1864
+ // either a deferrable tool stages a preview action, or plan mode installs a standing handler
1865
+ // that consumes `resolve { action: "apply" }` to submit the plan for approval (issue #1428).
1866
+ // Dropping it on read-only sessions (e.g. plan-mode toolset `read`, `search`, `find`,
1867
+ // `web_search`) leaves plan mode unable to exit through the intended path.
1868
+ const hasDeferrableTools = Array.from(toolRegistry.values()).some(tool => tool.deferrable === true);
1869
+ const planModeAvailable = settings.get("plan.enabled");
1870
+ const needsResolveTool = hasDeferrableTools || planModeAvailable;
1871
+ if (!needsResolveTool) {
1872
+ toolRegistry.delete("resolve");
1873
+ } else if (!toolRegistry.has("resolve")) {
1874
+ const resolveTool = await logger.time("createTools:resolve:session", HIDDEN_TOOLS.resolve, toolSession);
1875
+ if (resolveTool) {
1876
+ toolRegistry.set(resolveTool.name, wrapToolWithMetaNotice(resolveTool));
1877
+ }
1878
+ }
1879
+
1880
+ const effectiveDiscoveryMode = resolveEffectiveToolDiscoveryMode(
1881
+ settings,
1882
+ countToolsForAutoDiscovery(toolRegistry.keys()),
1883
+ );
1884
+ if (effectiveDiscoveryMode !== "off" && !toolRegistry.has("search_tool_bm25")) {
1885
+ const searchTool: Tool = new SearchToolBm25Tool(toolSession);
1886
+ toolRegistry.set(
1887
+ searchTool.name,
1888
+ new ExtensionToolWrapper(wrapToolWithMetaNotice(searchTool), extensionRunner) as Tool,
1889
+ );
1890
+ }
1891
+ const mcpDiscoveryEnabled = effectiveDiscoveryMode !== "off"; // back-compat: true when any discovery active
1892
+
1893
+ const reloadSshTool = async (): Promise<AgentTool | null> => {
1894
+ if (!requestedToolNameSet.has("ssh")) return null;
1895
+ const sshTool = (await loadSshTool({
1896
+ ...toolSession,
1897
+ cwd: sessionManager.getCwd(),
1898
+ })) as unknown as AgentTool | null;
1899
+ if (!sshTool) return null;
1900
+ const wrapped = wrapToolWithMetaNotice(sshTool);
1901
+ return new ExtensionToolWrapper(wrapped, extensionRunner) as AgentTool;
1902
+ };
1903
+
1904
+ let cursorEventEmitter: ((event: AgentEvent) => void) | undefined;
1905
+ const cursorExecHandlers = new CursorExecHandlers({
1906
+ cwd,
1907
+ tools: toolRegistry,
1908
+ getToolContext: () => toolContextStore.getContext(),
1909
+ emitEvent: event => cursorEventEmitter?.(event),
1910
+ });
1911
+
1912
+ const repeatToolDescriptions = settings.get("repeatToolDescriptions");
1913
+ const eagerTasks = settings.get("task.eager");
1914
+ const intentField = $flag("PI_INTENT_TRACING", settings.get("tools.intentTracing")) ? INTENT_FIELD : undefined;
1915
+ const rebuildSystemPrompt = async (
1916
+ toolNames: string[],
1917
+ tools: Map<string, AgentTool>,
1918
+ ): Promise<BuildSystemPromptResult> => {
1919
+ toolContextStore.setToolNames(toolNames);
1920
+ const discoverableMCPTools: DiscoverableTool[] = mcpDiscoveryEnabled
1921
+ ? filterBySource(collectDiscoverableTools(tools.values()), "mcp")
1922
+ : [];
1923
+ const activeToolNames = new Set(toolNames);
1924
+ const discoverableBuiltinTools: DiscoverableTool[] =
1925
+ effectiveDiscoveryMode === "all"
1926
+ ? collectDiscoverableTools(
1927
+ Array.from(tools.values()).filter(
1928
+ tool => tool.loadMode === "discoverable" && !activeToolNames.has(tool.name),
1929
+ ),
1930
+ { source: "builtin" },
1931
+ )
1932
+ : [];
1933
+ const discoverableToolsForDesc: DiscoverableTool[] = [...discoverableBuiltinTools, ...discoverableMCPTools];
1934
+ const discoverableToolSummary = summarizeDiscoverableTools(discoverableToolsForDesc);
1935
+ const hasDiscoverableTools =
1936
+ mcpDiscoveryEnabled && toolNames.includes("search_tool_bm25") && discoverableToolsForDesc.length > 0;
1937
+ const promptTools = buildSystemPromptToolMetadata(tools, {
1938
+ search_tool_bm25: { description: renderSearchToolBm25Description(discoverableToolsForDesc) },
1939
+ });
1940
+ const memoryBackend = await resolveMemoryBackend(settings);
1941
+ const memoryInstructions = await memoryBackend.buildDeveloperInstructions(agentDir, settings, session);
1942
+
1943
+ // Build combined append prompt: memory instructions + MCP server instructions
1944
+ const serverInstructions = mcpManager?.getServerInstructions();
1945
+ let appendPrompt: string | undefined = memoryInstructions ?? undefined;
1946
+ if (serverInstructions && serverInstructions.size > 0) {
1947
+ const parts: string[] = [];
1948
+ if (appendPrompt) parts.push(appendPrompt);
1949
+ parts.push(
1950
+ "## MCP Server Instructions\n\nThe following instructions are provided by connected MCP servers. They are server-controlled and may not be verified.",
1951
+ );
1952
+ for (const [srvName, srvInstructions] of serverInstructions) {
1953
+ const truncated =
1954
+ srvInstructions.length > MAX_MCP_INSTRUCTIONS_LENGTH
1955
+ ? `${srvInstructions.slice(0, MAX_MCP_INSTRUCTIONS_LENGTH)}\n[truncated]`
1956
+ : srvInstructions;
1957
+ parts.push(`### ${srvName}\n${truncated}`);
1958
+ }
1959
+ appendPrompt = parts.join("\n\n");
1960
+ }
1961
+ const defaultPrompt = await buildSystemPromptInternal({
1962
+ cwd,
1963
+ skills,
1964
+ contextFiles,
1965
+ tools: promptTools,
1966
+ toolNames,
1967
+ rules: rulebookRules,
1968
+ alwaysApplyRules,
1969
+ skillsSettings: settings.getGroup("skills"),
1970
+ appendSystemPrompt: appendPrompt,
1971
+ repeatToolDescriptions,
1972
+ intentField,
1973
+ mcpDiscoveryMode: hasDiscoverableTools,
1974
+ mcpDiscoveryServerSummaries: discoverableToolSummary.servers.map(formatDiscoverableToolServerSummary),
1975
+ eagerTasks,
1976
+ secretsEnabled,
1977
+ workspaceTree: workspaceTreePromise,
1978
+ memoryRootEnabled: memoryBackend.id === "local",
1979
+ model: settings.get("includeModelInPrompt") ? getActiveModelString() : undefined,
1980
+ });
1981
+
1982
+ if (options.systemPrompt === undefined) {
1983
+ return defaultPrompt;
1984
+ }
1985
+ if (Array.isArray(options.systemPrompt)) {
1986
+ return { systemPrompt: options.systemPrompt };
1987
+ }
1988
+ return {
1989
+ systemPrompt: options.systemPrompt(defaultPrompt.systemPrompt),
1990
+ };
1991
+ };
1992
+
1993
+ const toolNamesFromRegistry = Array.from(toolRegistry.keys());
1994
+ const explicitlyRequestedToolNames = options.toolNames
1995
+ ? [...new Set(options.toolNames.map(name => name.toLowerCase()))]
1996
+ : undefined;
1997
+ // When `requireYieldTool` is set, the subagent's prompts and idle-reminders demand a
1998
+ // `yield` call to terminate. The tool registry already includes `yield` (see
1999
+ // `createTools`), but an explicit `toolNames` list would otherwise drop it from the
2000
+ // active set — leaving the model unable to satisfy the contract. Mirror the same
2001
+ // invariant `parseAgentFields` enforces on frontmatter `tools`.
2002
+ if (
2003
+ options.requireYieldTool === true &&
2004
+ explicitlyRequestedToolNames &&
2005
+ !explicitlyRequestedToolNames.includes("yield")
2006
+ ) {
2007
+ explicitlyRequestedToolNames.push("yield");
2008
+ }
2009
+ const requestedToolNames = explicitlyRequestedToolNames ?? toolNamesFromRegistry;
2010
+ const normalizedRequested = requestedToolNames.filter(name => toolRegistry.has(name));
2011
+ const requestedToolNameSet = new Set(normalizedRequested);
2012
+ // Effective discovery mode is resolved after the full registry exists so auto mode can count MCP/extension tools.
2013
+ const defaultInactiveToolNames = new Set(
2014
+ registeredTools.filter(tool => tool.definition.defaultInactive).map(tool => tool.definition.name),
2015
+ );
2016
+ const requestedActiveToolNames = normalizedRequested.filter(name => name !== "goal");
2017
+ const initialRequestedActiveToolNames = options.toolNames
2018
+ ? requestedActiveToolNames
2019
+ : requestedActiveToolNames.filter(name => !defaultInactiveToolNames.has(name));
2020
+ const explicitlyRequestedMCPToolNames = options.toolNames
2021
+ ? requestedActiveToolNames.filter(name => name.startsWith("mcp__"))
2022
+ : [];
2023
+ const discoveryDefaultServers = new Set(
2024
+ (settings.get("mcp.discoveryDefaultServers") ?? []).map(serverName => serverName.trim()).filter(Boolean),
2025
+ );
2026
+ const discoveryDefaultServerToolNames = mcpDiscoveryEnabled
2027
+ ? selectDiscoverableToolNamesByServer(
2028
+ filterBySource(collectDiscoverableTools(toolRegistry.values()), "mcp"),
2029
+ discoveryDefaultServers,
2030
+ )
2031
+ : [];
2032
+ let initialSelectedMCPToolNames: string[] = [];
2033
+ let defaultSelectedMCPToolNames: string[] = [];
2034
+ let initialToolNames = [...initialRequestedActiveToolNames];
2035
+ if (mcpDiscoveryEnabled) {
2036
+ const restoredSelectedMCPToolNames = existingSession.selectedMCPToolNames.filter(name =>
2037
+ toolRegistry.has(name),
2038
+ );
2039
+ defaultSelectedMCPToolNames = [
2040
+ ...new Set([...discoveryDefaultServerToolNames, ...explicitlyRequestedMCPToolNames]),
2041
+ ];
2042
+ initialSelectedMCPToolNames = existingSession.hasPersistedMCPToolSelection
2043
+ ? restoredSelectedMCPToolNames
2044
+ : [...new Set([...restoredSelectedMCPToolNames, ...defaultSelectedMCPToolNames])];
2045
+ initialToolNames = [
2046
+ ...new Set([
2047
+ ...initialRequestedActiveToolNames.filter(name => !name.startsWith("mcp__")),
2048
+ ...initialSelectedMCPToolNames,
2049
+ ]),
2050
+ ];
2051
+ }
2052
+
2053
+ // Custom tools and extension-registered tools are always included regardless of toolNames filter
2054
+ const alwaysInclude: string[] = [
2055
+ ...(options.customTools?.map(t => (isCustomTool(t) ? t.name : t.name)) ?? []),
2056
+ ...registeredTools.filter(t => !t.definition.defaultInactive).map(t => t.definition.name),
2057
+ ];
2058
+ for (const name of alwaysInclude) {
2059
+ if (mcpDiscoveryEnabled && name.startsWith("mcp__")) {
2060
+ continue;
2061
+ }
2062
+ if (toolRegistry.has(name) && !initialToolNames.includes(name)) {
2063
+ initialToolNames.push(name);
2064
+ }
2065
+ }
2066
+
2067
+ // When tools.discoveryMode === "all", hide non-essential built-in discoverable tools
2068
+ // from the initial set unless they were explicitly requested or restored from persistence.
2069
+ // The model finds them via search_tool_bm25 and activates them on demand.
2070
+ if (effectiveDiscoveryMode === "all") {
2071
+ // Tools a forced tool_choice will target must stay active, or the named
2072
+ // choice references a tool absent from the request (provider 400). Eager
2073
+ // todos force a named `todo` choice on the first turn.
2074
+ const forceActive = new Set<string>();
2075
+ if (settings.get("todo.eager") && settings.get("todo.enabled") && toolRegistry.has("todo")) {
2076
+ forceActive.add("todo");
2077
+ }
2078
+ initialToolNames = filterInitialToolsForDiscoveryAll(initialToolNames, {
2079
+ loadModeOf: name => toolRegistry.get(name)?.loadMode,
2080
+ essentialNames: new Set(computeEssentialBuiltinNames(settings)),
2081
+ explicitlyRequested: new Set(options.toolNames?.map(name => name.toLowerCase()) ?? []),
2082
+ // Back-compat: persisted activations live under selectedMCPToolNames today (built-in
2083
+ // activation persistence is a follow-up). MCP names won't collide with built-in names.
2084
+ restored: new Set(existingSession.selectedMCPToolNames),
2085
+ forceActive,
2086
+ });
2087
+ }
2088
+
2089
+ // Pre-register in the global agent registry BEFORE building the system prompt,
2090
+ // so that subagents launched in the same parallel batch can see each other in
2091
+ // their initial `# IRC Peers` block (rendered inside `rebuildSystemPrompt`).
2092
+ // The session reference is attached after construction below.
2093
+ agentRegistry.register({
2094
+ id: resolvedAgentId,
2095
+ displayName: resolvedAgentDisplayName,
2096
+ kind: agentKind,
2097
+ parentId: options.parentTaskPrefix,
2098
+ session: null,
2099
+ sessionFile: sessionManager.getSessionFile() ?? null,
2100
+ status: "running",
2101
+ });
2102
+ hasRegistered = true;
2103
+
2104
+ const { systemPrompt } = await logger.time(
2105
+ "buildSystemPrompt",
2106
+ rebuildSystemPrompt,
2107
+ initialToolNames,
2108
+ toolRegistry,
2109
+ );
2110
+
2111
+ const promptTemplates = await promptTemplatesPromise;
2112
+ toolSession.promptTemplates = promptTemplates;
2113
+
2114
+ const slashCommands = await slashCommandsPromise;
2115
+
2116
+ // Create convertToLlm wrapper that filters images if blockImages is enabled (defense-in-depth)
2117
+ const convertToLlmWithBlockImages = (messages: AgentMessage[]): Message[] => {
2118
+ const converted = convertToLlm(messages);
2119
+ // Check setting dynamically so mid-session changes take effect
2120
+ if (!settings.get("images.blockImages")) {
2121
+ return converted;
2122
+ }
2123
+ // Filter out ImageContent from all messages, replacing with text placeholder
2124
+ return converted.map(msg => {
2125
+ if (msg.role === "user" || msg.role === "toolResult") {
2126
+ const content = msg.content;
2127
+ if (Array.isArray(content)) {
2128
+ const hasImages = content.some(c => c.type === "image");
2129
+ if (hasImages) {
2130
+ const filteredContent = content
2131
+ .map(c =>
2132
+ c.type === "image" ? { type: "text" as const, text: "Image reading is disabled." } : c,
2133
+ )
2134
+ .filter((c, i, arr) => {
2135
+ // Dedupe consecutive "Image reading is disabled." texts
2136
+ if (!(c.type === "text" && c.text === "Image reading is disabled." && i > 0)) return true;
2137
+ const prev = arr[i - 1];
2138
+ return !(prev.type === "text" && prev.text === "Image reading is disabled.");
2139
+ });
2140
+ return { ...msg, content: filteredContent };
2141
+ }
2142
+ }
2143
+ }
2144
+ return msg;
2145
+ });
2146
+ };
2147
+
2148
+ // Final convertToLlm: chain block-images filter with secret obfuscation
2149
+ const convertToLlmFinal = (messages: AgentMessage[]): Message[] => {
2150
+ const converted = convertToLlmWithBlockImages(messages);
2151
+ if (!obfuscator?.hasSecrets()) return converted;
2152
+ return obfuscateMessages(obfuscator, converted);
2153
+ };
2154
+
2155
+ const transformContext = async (messages: AgentMessage[], _signal?: AbortSignal) => {
2156
+ const withContext = await extensionRunner.emitContext(messages);
2157
+ return wrapSteeringForModel(withContext);
2158
+ };
2159
+ const onPayload = async (payload: unknown, _model?: Model) => {
2160
+ return await extensionRunner.emitBeforeProviderRequest(payload);
2161
+ };
2162
+ const onResponse: SimpleStreamOptions["onResponse"] = async (response, model) => {
2163
+ await extensionRunner.emitAfterProviderResponse(response, model);
2164
+ };
2165
+
2166
+ const setToolUIContext = (uiContext: ExtensionUIContext, hasUI: boolean) => {
2167
+ toolContextStore.setUIContext(uiContext, hasUI);
2168
+ };
2169
+
2170
+ const initialTools = initialToolNames
2171
+ .map(name => toolRegistry.get(name))
2172
+ .filter((tool): tool is AgentTool => tool !== undefined);
2173
+
2174
+ const openaiWebsocketSetting = settings.get("providers.openaiWebsockets") ?? "off";
2175
+ const preferOpenAICodexWebsockets =
2176
+ openaiWebsocketSetting === "on" ? true : openaiWebsocketSetting === "off" ? false : undefined;
2177
+ const serviceTierSetting = settings.get("serviceTier");
2178
+
2179
+ const initialServiceTier = hasServiceTierEntry
2180
+ ? existingSession.serviceTier
2181
+ : serviceTierSetting === "none"
2182
+ ? undefined
2183
+ : serviceTierSetting;
2184
+
2185
+ agent = new Agent({
2186
+ initialState: {
2187
+ systemPrompt,
2188
+ model,
2189
+ thinkingLevel: toReasoningEffort(effectiveThinkingLevel),
2190
+ disableReasoning: shouldDisableReasoning(effectiveThinkingLevel),
2191
+ tools: initialTools,
2192
+ },
2193
+ convertToLlm: convertToLlmFinal,
2194
+ onPayload,
2195
+ onResponse,
2196
+ sessionId: providerSessionId,
2197
+ promptCacheKey: options.providerPromptCacheKey,
2198
+ transformContext,
2199
+ transformProviderContext: obfuscator ? context => obfuscateProviderContext(obfuscator, context) : undefined,
2200
+ steeringMode: settings.get("steeringMode") ?? "one-at-a-time",
2201
+ followUpMode: settings.get("followUpMode") ?? "one-at-a-time",
2202
+ interruptMode: settings.get("interruptMode") ?? "immediate",
2203
+ thinkingBudgets: settings.getGroup("thinkingBudgets"),
2204
+ temperature: settings.get("temperature") >= 0 ? settings.get("temperature") : undefined,
2205
+ topP: settings.get("topP") >= 0 ? settings.get("topP") : undefined,
2206
+ topK: settings.get("topK") >= 0 ? settings.get("topK") : undefined,
2207
+ minP: settings.get("minP") >= 0 ? settings.get("minP") : undefined,
2208
+ presencePenalty: settings.get("presencePenalty") >= 0 ? settings.get("presencePenalty") : undefined,
2209
+ repetitionPenalty: settings.get("repetitionPenalty") >= 0 ? settings.get("repetitionPenalty") : undefined,
2210
+ serviceTier: initialServiceTier,
2211
+ hideThinkingSummary: settings.get("hideThinkingBlock"),
2212
+ kimiApiFormat: settings.get("providers.kimiApiFormat") ?? "anthropic",
2213
+ preferWebsockets: preferOpenAICodexWebsockets,
2214
+ getToolContext: tc => toolContextStore.getContext(tc),
2215
+ getApiKey: async (provider, ctx) => {
2216
+ // Read agent.sessionId at call time so credential selection stays aligned
2217
+ // with metadataResolver after /new, fork, resume, or branch switches.
2218
+ // Retry steps (ctx carries an auth error) drive the central a/b/c
2219
+ // policy — force-refresh the same account, then rotate to a sibling —
2220
+ // and may legitimately yield no key when every account is exhausted.
2221
+ if (ctx?.error !== undefined) {
2222
+ return createApiKeyResolver(modelRegistry, provider, { sessionId: agent.sessionId })(ctx);
2223
+ }
2224
+ const key = await modelRegistry.getApiKeyForProvider(provider, agent.sessionId);
2225
+ if (!key) {
2226
+ throw new Error(`No API key found for provider "${provider}"`);
2227
+ }
2228
+ return key;
2229
+ },
2230
+ streamFn: (streamModel, context, streamOptions) => {
2231
+ const openrouterRoutingPreset = settings.get("providers.openrouterVariant");
2232
+ const openrouterVariant =
2233
+ openrouterRoutingPreset && openrouterRoutingPreset !== "default" ? openrouterRoutingPreset : undefined;
2234
+ return streamSimple(streamModel, context, {
2235
+ ...streamOptions,
2236
+ openrouterVariant: streamOptions?.openrouterVariant ?? openrouterVariant,
2237
+ });
2238
+ },
2239
+ cursorExecHandlers,
2240
+ transformToolCallArguments: (args, _toolName) => {
2241
+ let result = args;
2242
+ const maxTimeout = settings.get("tools.maxTimeout");
2243
+ if (maxTimeout > 0 && typeof result.timeout === "number") {
2244
+ result = { ...result, timeout: Math.min(result.timeout, maxTimeout) };
2245
+ }
2246
+ if (obfuscator?.hasSecrets()) {
2247
+ result = obfuscator.deobfuscateObject(result);
2248
+ }
2249
+ return result;
2250
+ },
2251
+ intentTracing: !!intentField,
2252
+ getToolChoice: () => session?.nextToolChoice(),
2253
+ telemetry: options.telemetry,
2254
+ appendOnlyContext: model
2255
+ ? shouldEnableAppendOnlyContext(settings.get("provider.appendOnlyContext"), model)
2256
+ ? new AppendOnlyContextManager()
2257
+ : undefined
2258
+ : undefined,
2259
+ });
2260
+
2261
+ cursorEventEmitter = event => agent.emitExternalEvent(event);
2262
+
2263
+ // Restore messages if session has existing data
2264
+ if (hasExistingSession) {
2265
+ agent.replaceMessages(existingSession.messages);
2266
+ } else {
2267
+ // Save initial model, thinking level, and service tier for new sessions so they can be restored on resume.
2268
+ if (model) {
2269
+ sessionManager.appendModelChange(`${model.provider}/${model.id}`);
2270
+ }
2271
+ if (!autoThinking) {
2272
+ // Do not write the `auto` selector before the first turn resolves; auto
2273
+ // classification persists its concrete effort once a real user turn runs.
2274
+ sessionManager.appendThinkingLevelChange(effectiveThinkingLevel);
2275
+ }
2276
+ if (initialServiceTier) {
2277
+ sessionManager.appendServiceTierChange(initialServiceTier);
2278
+ }
2279
+ }
2280
+
2281
+ session = new AgentSession({
2282
+ agent,
2283
+ thinkingLevel: autoThinking ? AUTO_THINKING : effectiveThinkingLevel,
2284
+ sessionManager,
2285
+ settings,
2286
+ autoApprove: options.autoApprove,
2287
+ evalKernelOwnerId,
2288
+ // Defined only for top-level sessions (creation is gated above).
2289
+ // AgentSession uses this to decide whether it may dispose the global
2290
+ // AsyncJobManager on teardown; subagents inherit the parent's and
2291
+ // **MUST NOT** tear it down.
2292
+ ownedAsyncJobManager: asyncJobManager,
2293
+ asyncJobManager: scopedAsyncJobManager,
2294
+ scopedModels: options.scopedModels,
2295
+ promptTemplates,
2296
+ slashCommands,
2297
+ extensionRunner,
2298
+ customCommands: customCommandsResult.commands,
2299
+ skills,
2300
+ skillWarnings,
2301
+ skillsSettings: settings.getGroup("skills"),
2302
+ modelRegistry,
2303
+ toolRegistry,
2304
+ transformContext,
2305
+ onPayload,
2306
+ onResponse,
2307
+ convertToLlm: convertToLlmFinal,
2308
+ rebuildSystemPrompt,
2309
+ reloadSshTool,
2310
+ requestedToolNames: requestedToolNameSet,
2311
+ getMcpServerInstructions: mcpManager
2312
+ ? () => {
2313
+ const raw = mcpManager.getServerInstructions();
2314
+ if (!raw || raw.size === 0) return raw;
2315
+ const out = new Map<string, string>();
2316
+ for (const [name, text] of raw) {
2317
+ out.set(
2318
+ name,
2319
+ text.length > MAX_MCP_INSTRUCTIONS_LENGTH ? text.slice(0, MAX_MCP_INSTRUCTIONS_LENGTH) : text,
2320
+ );
2321
+ }
2322
+ return out;
2323
+ }
2324
+ : undefined,
2325
+ mcpDiscoveryEnabled,
2326
+ initialSelectedMCPToolNames,
2327
+ defaultSelectedMCPToolNames,
2328
+ persistInitialMCPToolSelection: !hasExistingSession,
2329
+ defaultSelectedMCPServerNames: [...discoveryDefaultServers],
2330
+ ttsrManager,
2331
+ obfuscator,
2332
+ agentId: resolvedAgentId,
2333
+ providerSessionId: options.providerSessionId,
2334
+ parentEvalSessionId: options.parentEvalSessionId,
2335
+ });
2336
+ hasSession = true;
2337
+ if (asyncJobManager) {
2338
+ session.yieldQueue.register<AsyncResultEntry>("async-result", {
2339
+ isStale: entry => asyncJobManager.isDeliverySuppressed(entry.jobId),
2340
+ build: buildAsyncResultBatchMessage,
2341
+ });
2342
+ }
2343
+ session.yieldQueue.register<McpNotificationEntry>("mcp-notification", {
2344
+ build: buildMcpNotificationBatchMessage,
2345
+ });
2346
+ session.yieldQueue.register<DeferredDiagnosticsEntry>(LSP_LATE_DIAGNOSTIC_MESSAGE_TYPE, {
2347
+ isStale: entry => entry.isStale(),
2348
+ build: buildLateDiagnosticsBatchMessage,
2349
+ });
2350
+
2351
+ // Attach the live session to the pre-registered ref so peers can route IRC
2352
+ // messages here. Refresh sessionFile in case it was unavailable at pre-register
2353
+ // time. The dispose wrapper below unregisters on teardown (unless parked).
2354
+ agentRegistry.attachSession(resolvedAgentId, session, sessionManager.getSessionFile() ?? null);
2355
+ {
2356
+ const originalDispose = session.dispose.bind(session);
2357
+ session.dispose = async () => {
2358
+ try {
2359
+ // Reject new session work (Python/eval starts) the moment disposal
2360
+ // begins — the lifecycle await below opens an async gap before
2361
+ // AgentSession.dispose() would otherwise set its guards.
2362
+ session.beginDispose();
2363
+ if (agentKind === "main") {
2364
+ // Top-level teardown owns the global agent lifecycle: park timers,
2365
+ // adopted subagent sessions, revivers. Tear it down while shared
2366
+ // resources (kernels, MCP, LSP) are still live. Subagent disposal
2367
+ // must NOT touch the global lifecycle.
2368
+ await AgentLifecycleManager.global().dispose();
2369
+ }
2370
+ await originalDispose();
2371
+ } finally {
2372
+ unregisterUnlessParked();
2373
+ unsubscribeCredentialDisabled?.();
2374
+ }
2375
+ };
2376
+ }
2377
+
2378
+ if (model?.api === "openai-codex-responses") {
2379
+ // `.api` equality doesn't narrow the generic; the guard makes this cast sound.
2380
+ const codexModel = model as Model<"openai-codex-responses">;
2381
+ const codexTransport = getOpenAICodexTransportDetails(codexModel, {
2382
+ sessionId: providerSessionId,
2383
+ baseUrl: codexModel.baseUrl,
2384
+ preferWebsockets: preferOpenAICodexWebsockets,
2385
+ providerSessionState: session.providerSessionState,
2386
+ });
2387
+ if (codexTransport.websocketPreferred) {
2388
+ void (async () => {
2389
+ try {
2390
+ const codexPrewarmApiKey = await modelRegistry.getApiKey(codexModel, providerSessionId);
2391
+ if (!codexPrewarmApiKey) return;
2392
+ await logger.time("prewarmOpenAICodexResponses", prewarmOpenAICodexResponses, codexModel, {
2393
+ apiKey: codexPrewarmApiKey,
2394
+ sessionId: providerSessionId,
2395
+ preferWebsockets: preferOpenAICodexWebsockets,
2396
+ providerSessionState: session.providerSessionState,
2397
+ });
2398
+ } catch (error) {
2399
+ const errorMessage = error instanceof Error ? error.message : String(error);
2400
+ logger.debug("Codex websocket prewarm failed", {
2401
+ error: errorMessage,
2402
+ provider: codexModel.provider,
2403
+ model: codexModel.id,
2404
+ });
2405
+ }
2406
+ })();
2407
+ }
2408
+ }
2409
+
2410
+ // Start LSP warmup in the background so startup does not block on language server initialization.
2411
+ // With `lsp.lazy` (the default) the warmup is skipped: recognized servers are still discovered and
2412
+ // surfaced in the UI as "available", but cold-start on first use — the lsp tool or an edit/write
2413
+ // touching a matching file type — through `getOrCreateClient`.
2414
+ // Print/script invocations (`hasUI=false`) skip it regardless: they don't render the warmup status
2415
+ // indicator AND typically finish before LSP servers would have stabilized — warming them just spends
2416
+ // CPU parsing big `initialize` responses concurrently with the LLM stream consumer, jittering
2417
+ // perceived latency.
2418
+ let lspServers: CreateAgentSessionResult["lspServers"];
2419
+ if (enableLsp && options.hasUI && settings.get("lsp.lazy")) {
2420
+ lspServers = discoverStartupLspServers(cwd, "available");
2421
+ } else if (enableLsp && options.hasUI) {
2422
+ lspServers = discoverStartupLspServers(cwd);
2423
+ if (lspServers.length > 0) {
2424
+ void (async () => {
2425
+ try {
2426
+ const result = await logger.time("warmupLspServers", warmupLspServers, cwd);
2427
+ const serversByName = new Map(result.servers.map(server => [server.name, server] as const));
2428
+ for (const server of lspServers ?? []) {
2429
+ const next = serversByName.get(server.name);
2430
+ if (!next) continue;
2431
+ server.status = next.status;
2432
+ server.fileTypes = next.fileTypes;
2433
+ server.error = next.error;
2434
+ }
2435
+ const event: LspStartupEvent = {
2436
+ type: "completed",
2437
+ servers: result.servers,
2438
+ };
2439
+ eventBus.emit(LSP_STARTUP_EVENT_CHANNEL, event);
2440
+ } catch (error) {
2441
+ const errorMessage = error instanceof Error ? error.message : String(error);
2442
+ logger.warn("LSP server warmup failed", { cwd, error: errorMessage });
2443
+ for (const server of lspServers ?? []) {
2444
+ server.status = "error";
2445
+ server.error = errorMessage;
2446
+ }
2447
+ const event: LspStartupEvent = {
2448
+ type: "failed",
2449
+ error: errorMessage,
2450
+ };
2451
+ eventBus.emit(LSP_STARTUP_EVENT_CHANNEL, event);
2452
+ }
2453
+ })();
2454
+ }
2455
+ }
2456
+
2457
+ logger.time("startMemoryStartupTask", async () => {
2458
+ const memoryBackend = await resolveMemoryBackend(settings);
2459
+ await memoryBackend.start({
2460
+ session,
2461
+ settings,
2462
+ modelRegistry,
2463
+ agentDir,
2464
+ taskDepth,
2465
+ parentHindsightSessionState: options.parentHindsightSessionState,
2466
+ parentMnemopiSessionState: options.parentMnemopiSessionState,
2467
+ });
2468
+ });
2469
+
2470
+ // Wire MCP manager callbacks to session for reactive tool updates.
2471
+ // Skip when reusing a parent's manager — the parent owns the callbacks.
2472
+ if (mcpManager && !options.mcpManager) {
2473
+ mcpManager.setOnToolsChanged(tools => {
2474
+ void session.refreshMCPTools(tools);
2475
+ });
2476
+ // Wire prompt refresh → rebuild MCP prompt slash commands
2477
+ mcpManager.setOnPromptsChanged(serverName => {
2478
+ const promptCommands = buildMCPPromptCommands(mcpManager);
2479
+ session.setMCPPromptCommands(promptCommands);
2480
+ logger.debug("MCP prompt commands refreshed", { path: `mcp:${serverName}` });
2481
+ });
2482
+ const notificationDebounceTimers = new Map<string, Timer>();
2483
+ const clearDebounceTimers = () => {
2484
+ for (const timer of notificationDebounceTimers.values()) clearTimeout(timer);
2485
+ notificationDebounceTimers.clear();
2486
+ };
2487
+ postmortem.register("mcp-notification-cleanup", clearDebounceTimers);
2488
+ mcpManager.setOnResourcesChanged((serverName, uri) => {
2489
+ logger.debug("MCP resources changed", { path: `mcp:${serverName}`, uri });
2490
+ if (!settings.get("mcp.notifications")) return;
2491
+ const debounceMs = settings.get("mcp.notificationDebounceMs");
2492
+ const key = `${serverName}:${uri}`;
2493
+ const existing = notificationDebounceTimers.get(key);
2494
+ if (existing) clearTimeout(existing);
2495
+ notificationDebounceTimers.set(
2496
+ key,
2497
+ setTimeout(() => {
2498
+ notificationDebounceTimers.delete(key);
2499
+ // Re-check: user may have disabled notifications during the debounce window
2500
+ if (!settings.get("mcp.notifications")) return;
2501
+ session.yieldQueue.enqueue<McpNotificationEntry>("mcp-notification", { serverName, uri });
2502
+ }, debounceMs),
2503
+ );
2504
+ });
2505
+ }
2506
+
2507
+ return {
2508
+ session,
2509
+ extensionsResult,
2510
+ setToolUIContext,
2511
+ mcpManager,
2512
+ modelFallbackMessage,
2513
+ lspServers,
2514
+ eventBus,
2515
+ };
2516
+ } catch (error) {
2517
+ // Release the subscription if the throw happened after install but before the
2518
+ // dispose-wrap took ownership. Idempotent with dispose() — Set.delete is a no-op
2519
+ // for already-removed listeners.
2520
+ unsubscribeCredentialDisabled?.();
2521
+ try {
2522
+ if (hasSession) {
2523
+ await session.dispose();
2524
+ } else {
2525
+ if (hasRegistered) unregisterUnlessParked();
2526
+ if (asyncJobManager) {
2527
+ if (AsyncJobManager.instance() === asyncJobManager) {
2528
+ AsyncJobManager.setInstance(undefined);
2529
+ }
2530
+ await asyncJobManager.dispose({ timeoutMs: 3_000 });
2531
+ }
2532
+ await disposeKernelSessionsByOwner(evalKernelOwnerId);
2533
+ }
2534
+ } catch (cleanupError) {
2535
+ logger.warn("Failed to clean up createAgentSession resources after startup error", {
2536
+ error: cleanupError instanceof Error ? cleanupError.message : String(cleanupError),
2537
+ });
2538
+ }
2539
+ throw error;
2540
+ }
2541
+ }
2542
+
2543
+ /**
2544
+ * Best-effort preconnect to the model's API host. Bun's `fetch.preconnect`
2545
+ * primes DNS + TCP + TLS + H2 so the first real request reuses the warm
2546
+ * connection. Errors are swallowed: preconnect is an optimization, never a
2547
+ * hard dependency.
2548
+ */
2549
+ function preconnectModelHost(baseUrl: string | undefined): void {
2550
+ if (!baseUrl) return;
2551
+ const preconnect = (globalThis.fetch as typeof fetch & { preconnect?: (url: string) => void }).preconnect;
2552
+ if (typeof preconnect !== "function") return;
2553
+ try {
2554
+ preconnect(baseUrl);
2555
+ } catch {
2556
+ // Best effort.
2557
+ }
2558
+ }