@blxzer/cursor-trellis 0.1.0 → 0.1.2

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (285) hide show
  1. package/README.md +124 -5
  2. package/README.zh-CN.md +170 -0
  3. package/dist/cli/index.d.ts.map +1 -1
  4. package/dist/cli/index.js +1 -33
  5. package/dist/cli/index.js.map +1 -1
  6. package/dist/commands/init.d.ts +0 -13
  7. package/dist/commands/init.d.ts.map +1 -1
  8. package/dist/commands/init.js +7 -14
  9. package/dist/commands/init.js.map +1 -1
  10. package/dist/commands/uninstall.d.ts.map +1 -1
  11. package/dist/commands/uninstall.js +3 -32
  12. package/dist/commands/uninstall.js.map +1 -1
  13. package/dist/commands/update.d.ts.map +1 -1
  14. package/dist/commands/update.js +7 -84
  15. package/dist/commands/update.js.map +1 -1
  16. package/dist/configurators/index.d.ts +7 -8
  17. package/dist/configurators/index.d.ts.map +1 -1
  18. package/dist/configurators/index.js +11 -236
  19. package/dist/configurators/index.js.map +1 -1
  20. package/dist/configurators/workflow.d.ts.map +1 -1
  21. package/dist/configurators/workflow.js +5 -1
  22. package/dist/configurators/workflow.js.map +1 -1
  23. package/dist/migrations/manifests/0.1.1.json +9 -0
  24. package/dist/migrations/manifests/0.1.2.json +9 -0
  25. package/dist/templates/common/bundled-skills/smart-search-cli/SKILL.md +3 -3
  26. package/dist/templates/common/bundled-skills/trellis-meta/SKILL.md +4 -2
  27. package/dist/templates/common/bundled-skills/trellis-meta/references/customize-local/change-agents.md +8 -15
  28. package/dist/templates/common/bundled-skills/trellis-meta/references/customize-local/change-hooks.md +7 -7
  29. package/dist/templates/common/bundled-skills/trellis-meta/references/customize-local/change-skills-or-commands.md +18 -21
  30. package/dist/templates/common/bundled-skills/trellis-meta/references/customize-local/overview.md +2 -2
  31. package/dist/templates/common/bundled-skills/trellis-meta/references/local-architecture/generated-files.md +10 -7
  32. package/dist/templates/common/bundled-skills/trellis-meta/references/local-architecture/overview.md +4 -4
  33. package/dist/templates/common/bundled-skills/trellis-meta/references/platform-files/agents.md +20 -31
  34. package/dist/templates/common/bundled-skills/trellis-meta/references/platform-files/hooks-and-settings.md +20 -31
  35. package/dist/templates/common/bundled-skills/trellis-meta/references/platform-files/overview.md +24 -40
  36. package/dist/templates/common/bundled-skills/trellis-meta/references/platform-files/platform-map.md +28 -59
  37. package/dist/templates/common/bundled-skills/trellis-meta/references/platform-files/skills-and-commands.md +12 -28
  38. package/dist/templates/common/bundled-skills/trellis-skill-creator/references/trellis-skill-locations.md +2 -2
  39. package/dist/templates/cursor/hooks.json +6 -0
  40. package/dist/templates/cursor/rules/retrieval-routing.mdc +57 -0
  41. package/dist/templates/extract.d.ts +0 -14
  42. package/dist/templates/extract.d.ts.map +1 -1
  43. package/dist/templates/extract.js +0 -34
  44. package/dist/templates/extract.js.map +1 -1
  45. package/dist/templates/markdown/agents.md +2 -6
  46. package/dist/templates/markdown/index.d.ts +1 -0
  47. package/dist/templates/markdown/index.d.ts.map +1 -1
  48. package/dist/templates/markdown/index.js +1 -0
  49. package/dist/templates/markdown/index.js.map +1 -1
  50. package/dist/templates/markdown/prompts/run-semantic-slice-12.md.txt +41 -0
  51. package/dist/templates/markdown/spec/guides/code-reuse-thinking-guide.md.txt +14 -14
  52. package/dist/templates/markdown/spec/guides/cross-layer-thinking-guide.md.txt +8 -20
  53. package/dist/templates/markdown/spec/guides/cross-platform-thinking-guide.md.txt +15 -14
  54. package/dist/templates/markdown/spec/guides/cursor-context-injection-guide.md.txt +4 -2
  55. package/dist/templates/markdown/spec/guides/cursor-semantic-compliance.md.txt +49 -0
  56. package/dist/templates/markdown/spec/guides/cursor-subagent-policy.md.txt +18 -5
  57. package/dist/templates/markdown/spec/guides/retrieval-daily-guide.md.txt +68 -20
  58. package/dist/templates/shared-hooks/index.d.ts +1 -1
  59. package/dist/templates/shared-hooks/index.d.ts.map +1 -1
  60. package/dist/templates/shared-hooks/index.js +1 -0
  61. package/dist/templates/shared-hooks/index.js.map +1 -1
  62. package/dist/templates/shared-hooks/inject-retrieval-plan.py +163 -0
  63. package/dist/templates/shared-hooks/inject-subagent-context.py +31 -567
  64. package/dist/templates/shared-hooks/research-end-retrieval-pack.py +2 -11
  65. package/dist/templates/trellis/index.d.ts +12 -0
  66. package/dist/templates/trellis/index.d.ts.map +1 -1
  67. package/dist/templates/trellis/index.js +24 -0
  68. package/dist/templates/trellis/index.js.map +1 -1
  69. package/dist/templates/trellis/scripts/aggregate_retrieval_telemetry.py +342 -0
  70. package/dist/templates/trellis/scripts/batch_plan_envelope.py +105 -0
  71. package/dist/templates/trellis/scripts/codegraph_session_smoke.py +76 -0
  72. package/dist/templates/trellis/scripts/common/active_task.py +116 -147
  73. package/dist/templates/trellis/scripts/common/cli_adapter.py +34 -710
  74. package/dist/templates/trellis/scripts/common/codebase_retrieval_router.py +49 -106
  75. package/dist/templates/trellis/scripts/common/git_context.py +17 -2
  76. package/dist/templates/trellis/scripts/common/project_file_stats.py +91 -0
  77. package/dist/templates/trellis/scripts/common/retrieval_adapter_metadata.py +9 -11
  78. package/dist/templates/trellis/scripts/common/retrieval_agent_instructions.py +313 -0
  79. package/dist/templates/trellis/scripts/common/retrieval_pack.py +0 -6
  80. package/dist/templates/trellis/scripts/common/retrieval_pack_context.py +0 -4
  81. package/dist/templates/trellis/scripts/common/retrieval_plan_gate.py +76 -0
  82. package/dist/templates/trellis/scripts/common/retrieval_result_ranking.py +253 -0
  83. package/dist/templates/trellis/scripts/common/retrieval_tool_classification.py +157 -0
  84. package/dist/templates/trellis/scripts/common/semantic_plan_gate.py +52 -0
  85. package/dist/templates/trellis/scripts/common/subagent_dispatch.py +527 -0
  86. package/dist/templates/trellis/scripts/common/task_store.py +32 -0
  87. package/dist/templates/trellis/scripts/common/workflow_phase.py +3 -24
  88. package/dist/templates/trellis/scripts/rank_retrieval_candidates.py +80 -0
  89. package/dist/templates/trellis/scripts/route_codebase_retrieval.py +38 -11
  90. package/dist/templates/trellis/scripts/score_evidence.py +112 -0
  91. package/dist/templates/trellis/scripts/task.py +26 -0
  92. package/dist/templates/trellis/workflow.md +844 -960
  93. package/dist/types/ai-tools.d.ts +13 -16
  94. package/dist/types/ai-tools.d.ts.map +1 -1
  95. package/dist/types/ai-tools.js +5 -238
  96. package/dist/types/ai-tools.js.map +1 -1
  97. package/dist/utils/codebase-retrieval-router.d.ts +1 -8
  98. package/dist/utils/codebase-retrieval-router.d.ts.map +1 -1
  99. package/dist/utils/codebase-retrieval-router.js +39 -180
  100. package/dist/utils/codebase-retrieval-router.js.map +1 -1
  101. package/dist/utils/project-capabilities.d.ts.map +1 -1
  102. package/dist/utils/project-capabilities.js +10 -24
  103. package/dist/utils/project-capabilities.js.map +1 -1
  104. package/dist/utils/retrieval-agent-instructions.d.ts +14 -0
  105. package/dist/utils/retrieval-agent-instructions.d.ts.map +1 -0
  106. package/dist/utils/retrieval-agent-instructions.js +221 -0
  107. package/dist/utils/retrieval-agent-instructions.js.map +1 -0
  108. package/dist/utils/retrieval-execution-telemetry.d.ts +43 -85
  109. package/dist/utils/retrieval-execution-telemetry.d.ts.map +1 -1
  110. package/dist/utils/retrieval-execution-telemetry.js +289 -22
  111. package/dist/utils/retrieval-execution-telemetry.js.map +1 -1
  112. package/dist/utils/retrieval-result-ranking.js +1 -1
  113. package/dist/utils/retrieval-result-ranking.js.map +1 -1
  114. package/dist/utils/retrieval-tool-classification.d.ts +32 -0
  115. package/dist/utils/retrieval-tool-classification.d.ts.map +1 -0
  116. package/dist/utils/retrieval-tool-classification.js +113 -0
  117. package/dist/utils/retrieval-tool-classification.js.map +1 -0
  118. package/dist/utils/uninstall-scrubbers.d.ts +0 -36
  119. package/dist/utils/uninstall-scrubbers.d.ts.map +1 -1
  120. package/dist/utils/uninstall-scrubbers.js +0 -184
  121. package/dist/utils/uninstall-scrubbers.js.map +1 -1
  122. package/package.json +4 -2
  123. package/dist/commands/mem.d.ts +0 -30
  124. package/dist/commands/mem.d.ts.map +0 -1
  125. package/dist/commands/mem.js +0 -424
  126. package/dist/commands/mem.js.map +0 -1
  127. package/dist/configurators/antigravity.d.ts +0 -7
  128. package/dist/configurators/antigravity.d.ts.map +0 -1
  129. package/dist/configurators/antigravity.js +0 -19
  130. package/dist/configurators/antigravity.js.map +0 -1
  131. package/dist/configurators/claude.d.ts +0 -9
  132. package/dist/configurators/claude.d.ts.map +0 -1
  133. package/dist/configurators/claude.js +0 -72
  134. package/dist/configurators/claude.js.map +0 -1
  135. package/dist/configurators/codebuddy.d.ts +0 -10
  136. package/dist/configurators/codebuddy.d.ts.map +0 -1
  137. package/dist/configurators/codebuddy.js +0 -30
  138. package/dist/configurators/codebuddy.js.map +0 -1
  139. package/dist/configurators/codex.d.ts +0 -8
  140. package/dist/configurators/codex.d.ts.map +0 -1
  141. package/dist/configurators/codex.js +0 -87
  142. package/dist/configurators/codex.js.map +0 -1
  143. package/dist/configurators/copilot.d.ts +0 -10
  144. package/dist/configurators/copilot.d.ts.map +0 -1
  145. package/dist/configurators/copilot.js +0 -51
  146. package/dist/configurators/copilot.js.map +0 -1
  147. package/dist/configurators/droid.d.ts +0 -10
  148. package/dist/configurators/droid.d.ts.map +0 -1
  149. package/dist/configurators/droid.js +0 -30
  150. package/dist/configurators/droid.js.map +0 -1
  151. package/dist/configurators/gemini.d.ts +0 -16
  152. package/dist/configurators/gemini.d.ts.map +0 -1
  153. package/dist/configurators/gemini.js +0 -38
  154. package/dist/configurators/gemini.js.map +0 -1
  155. package/dist/configurators/kilo.d.ts +0 -7
  156. package/dist/configurators/kilo.d.ts.map +0 -1
  157. package/dist/configurators/kilo.js +0 -19
  158. package/dist/configurators/kilo.js.map +0 -1
  159. package/dist/configurators/kiro.d.ts +0 -8
  160. package/dist/configurators/kiro.d.ts.map +0 -1
  161. package/dist/configurators/kiro.js +0 -24
  162. package/dist/configurators/kiro.js.map +0 -1
  163. package/dist/configurators/opencode.d.ts +0 -14
  164. package/dist/configurators/opencode.d.ts.map +0 -1
  165. package/dist/configurators/opencode.js +0 -96
  166. package/dist/configurators/opencode.js.map +0 -1
  167. package/dist/configurators/pi.d.ts +0 -3
  168. package/dist/configurators/pi.d.ts.map +0 -1
  169. package/dist/configurators/pi.js +0 -45
  170. package/dist/configurators/pi.js.map +0 -1
  171. package/dist/configurators/qoder.d.ts +0 -11
  172. package/dist/configurators/qoder.d.ts.map +0 -1
  173. package/dist/configurators/qoder.js +0 -31
  174. package/dist/configurators/qoder.js.map +0 -1
  175. package/dist/configurators/windsurf.d.ts +0 -7
  176. package/dist/configurators/windsurf.d.ts.map +0 -1
  177. package/dist/configurators/windsurf.js +0 -19
  178. package/dist/configurators/windsurf.js.map +0 -1
  179. package/dist/templates/claude/agents/trellis-check.md +0 -124
  180. package/dist/templates/claude/agents/trellis-implement.md +0 -110
  181. package/dist/templates/claude/agents/trellis-research.md +0 -137
  182. package/dist/templates/claude/index.d.ts +0 -22
  183. package/dist/templates/claude/index.d.ts.map +0 -1
  184. package/dist/templates/claude/index.js +0 -46
  185. package/dist/templates/claude/index.js.map +0 -1
  186. package/dist/templates/claude/settings.json +0 -84
  187. package/dist/templates/codebuddy/agents/trellis-check.md +0 -115
  188. package/dist/templates/codebuddy/agents/trellis-implement.md +0 -110
  189. package/dist/templates/codebuddy/agents/trellis-research.md +0 -137
  190. package/dist/templates/codebuddy/index.d.ts +0 -15
  191. package/dist/templates/codebuddy/index.d.ts.map +0 -1
  192. package/dist/templates/codebuddy/index.js +0 -15
  193. package/dist/templates/codebuddy/index.js.map +0 -1
  194. package/dist/templates/codebuddy/settings.json +0 -70
  195. package/dist/templates/codex/agents/trellis-check.toml +0 -67
  196. package/dist/templates/codex/agents/trellis-implement.toml +0 -40
  197. package/dist/templates/codex/agents/trellis-research.toml +0 -73
  198. package/dist/templates/codex/config.toml +0 -28
  199. package/dist/templates/codex/hooks/session-start.py +0 -575
  200. package/dist/templates/codex/hooks.json +0 -15
  201. package/dist/templates/codex/index.d.ts +0 -39
  202. package/dist/templates/codex/index.d.ts.map +0 -1
  203. package/dist/templates/codex/index.js +0 -85
  204. package/dist/templates/codex/index.js.map +0 -1
  205. package/dist/templates/codex/skills/before-dev/SKILL.md +0 -40
  206. package/dist/templates/codex/skills/brainstorm/SKILL.md +0 -112
  207. package/dist/templates/codex/skills/break-loop/SKILL.md +0 -130
  208. package/dist/templates/codex/skills/check/SKILL.md +0 -98
  209. package/dist/templates/codex/skills/check-cross-layer/SKILL.md +0 -158
  210. package/dist/templates/codex/skills/create-command/SKILL.md +0 -101
  211. package/dist/templates/codex/skills/finish-work/SKILL.md +0 -90
  212. package/dist/templates/codex/skills/improve-ut/SKILL.md +0 -69
  213. package/dist/templates/codex/skills/integrate-skill/SKILL.md +0 -221
  214. package/dist/templates/codex/skills/onboard/SKILL.md +0 -363
  215. package/dist/templates/codex/skills/record-session/SKILL.md +0 -67
  216. package/dist/templates/codex/skills/start/SKILL.md +0 -66
  217. package/dist/templates/codex/skills/update-spec/SKILL.md +0 -335
  218. package/dist/templates/copilot/hooks/session-start.py +0 -557
  219. package/dist/templates/copilot/hooks.json +0 -19
  220. package/dist/templates/copilot/index.d.ts +0 -23
  221. package/dist/templates/copilot/index.d.ts.map +0 -1
  222. package/dist/templates/copilot/index.js +0 -54
  223. package/dist/templates/copilot/index.js.map +0 -1
  224. package/dist/templates/copilot/prompts/before-dev.prompt.md +0 -39
  225. package/dist/templates/copilot/prompts/brainstorm.prompt.md +0 -111
  226. package/dist/templates/copilot/prompts/break-loop.prompt.md +0 -129
  227. package/dist/templates/copilot/prompts/check-cross-layer.prompt.md +0 -157
  228. package/dist/templates/copilot/prompts/check.prompt.md +0 -97
  229. package/dist/templates/copilot/prompts/create-command.prompt.md +0 -116
  230. package/dist/templates/copilot/prompts/finish-work.prompt.md +0 -99
  231. package/dist/templates/copilot/prompts/integrate-skill.prompt.md +0 -223
  232. package/dist/templates/copilot/prompts/onboard.prompt.md +0 -362
  233. package/dist/templates/copilot/prompts/parallel.prompt.md +0 -204
  234. package/dist/templates/copilot/prompts/record-session.prompt.md +0 -66
  235. package/dist/templates/copilot/prompts/start.prompt.md +0 -63
  236. package/dist/templates/copilot/prompts/update-spec.prompt.md +0 -358
  237. package/dist/templates/droid/droids/trellis-check.md +0 -107
  238. package/dist/templates/droid/droids/trellis-implement.md +0 -102
  239. package/dist/templates/droid/droids/trellis-research.md +0 -137
  240. package/dist/templates/droid/index.d.ts +0 -15
  241. package/dist/templates/droid/index.d.ts.map +0 -1
  242. package/dist/templates/droid/index.js +0 -15
  243. package/dist/templates/droid/index.js.map +0 -1
  244. package/dist/templates/droid/settings.json +0 -70
  245. package/dist/templates/gemini/agents/trellis-check.md +0 -107
  246. package/dist/templates/gemini/agents/trellis-implement.md +0 -102
  247. package/dist/templates/gemini/agents/trellis-research.md +0 -136
  248. package/dist/templates/gemini/index.d.ts +0 -13
  249. package/dist/templates/gemini/index.d.ts.map +0 -1
  250. package/dist/templates/gemini/index.js +0 -13
  251. package/dist/templates/gemini/index.js.map +0 -1
  252. package/dist/templates/gemini/settings.json +0 -28
  253. package/dist/templates/kiro/agents/trellis-check.json +0 -26
  254. package/dist/templates/kiro/agents/trellis-implement.json +0 -26
  255. package/dist/templates/kiro/agents/trellis-research.json +0 -30
  256. package/dist/templates/kiro/index.d.ts +0 -18
  257. package/dist/templates/kiro/index.d.ts.map +0 -1
  258. package/dist/templates/kiro/index.js +0 -18
  259. package/dist/templates/kiro/index.js.map +0 -1
  260. package/dist/templates/opencode/agents/trellis-check.md +0 -122
  261. package/dist/templates/opencode/agents/trellis-implement.md +0 -118
  262. package/dist/templates/opencode/agents/trellis-research.md +0 -145
  263. package/dist/templates/opencode/lib/session-utils.js +0 -667
  264. package/dist/templates/opencode/lib/trellis-context.js +0 -336
  265. package/dist/templates/opencode/package.json +0 -5
  266. package/dist/templates/opencode/plugins/inject-subagent-context.js +0 -495
  267. package/dist/templates/opencode/plugins/inject-workflow-state.js +0 -159
  268. package/dist/templates/opencode/plugins/session-start.js +0 -101
  269. package/dist/templates/pi/agents/trellis-check.md +0 -37
  270. package/dist/templates/pi/agents/trellis-implement.md +0 -42
  271. package/dist/templates/pi/agents/trellis-research.md +0 -25
  272. package/dist/templates/pi/extensions/trellis/index.ts.txt +0 -1600
  273. package/dist/templates/pi/index.d.ts +0 -5
  274. package/dist/templates/pi/index.d.ts.map +0 -1
  275. package/dist/templates/pi/index.js +0 -12
  276. package/dist/templates/pi/index.js.map +0 -1
  277. package/dist/templates/pi/settings.json +0 -12
  278. package/dist/templates/qoder/agents/trellis-check.md +0 -108
  279. package/dist/templates/qoder/agents/trellis-implement.md +0 -103
  280. package/dist/templates/qoder/agents/trellis-research.md +0 -137
  281. package/dist/templates/qoder/index.d.ts +0 -15
  282. package/dist/templates/qoder/index.d.ts.map +0 -1
  283. package/dist/templates/qoder/index.js +0 -15
  284. package/dist/templates/qoder/index.js.map +0 -1
  285. package/dist/templates/qoder/settings.json +0 -47
@@ -1,1600 +0,0 @@
1
- import { existsSync, readFileSync, readdirSync, statSync } from "node:fs";
2
- import { createHash, randomBytes } from "node:crypto";
3
- import { delimiter, dirname, isAbsolute, join, resolve } from "node:path";
4
- import { spawn, spawnSync } from "node:child_process";
5
-
6
- // ── Types ──────────────────────────────────────────────────────────────
7
- type JsonObject = Record<string, unknown>;
8
- type TextContent = { type: "text"; text: string };
9
- interface PiToolResult {
10
- content: TextContent[];
11
- details?: unknown;
12
- }
13
- interface PiExtensionContext {
14
- hasUI?: boolean;
15
- sessionManager?: {
16
- getSessionId?: () => string;
17
- getSessionFile?: () => string | undefined;
18
- };
19
- ui?: {
20
- notify?: (msg: string, type?: "info" | "warning" | "error") => void;
21
- };
22
- }
23
- interface SubagentInput {
24
- agent?: string;
25
- prompt?: string;
26
- mode?: "single" | "parallel" | "chain";
27
- prompts?: string[];
28
- model?: string;
29
- thinking?: string;
30
- }
31
- interface AgentConfig {
32
- model?: string;
33
- thinking?: string;
34
- fallbackModels: string[];
35
- }
36
- interface PiRunConfig {
37
- model?: string;
38
- thinking?: string;
39
- }
40
-
41
- // ── Lazy-load pi-tui (avoid failing top-level imports) ─────────────────
42
- let _piTui: {
43
- visibleWidth?: (s: string) => number;
44
- truncateToWidth?: (s: string, w: number, ellipsis?: string) => string;
45
- } | null = null;
46
- function piTui() {
47
- if (!_piTui) {
48
- try {
49
- _piTui = require("@earendil-works/pi-tui");
50
- } catch {
51
- _piTui = {};
52
- }
53
- }
54
- return _piTui;
55
- }
56
- function trunc(s: string, w: number) {
57
- const t = piTui();
58
- return t.truncateToWidth
59
- ? t.truncateToWidth(s, w, "…")
60
- : s.length <= w
61
- ? s
62
- : w > 1
63
- ? s.slice(0, w - 1) + "…"
64
- : s.slice(0, w);
65
- }
66
-
67
- // ── Constants ─────────────────────────────────────────────────────────
68
- const TRELLIS_AGENT_JSONL: Record<string, string> = {
69
- "trellis-implement": "implement.jsonl",
70
- implement: "implement.jsonl",
71
- "trellis-check": "check.jsonl",
72
- check: "check.jsonl",
73
- };
74
- const MAX_STDOUT = 8 * 1024 * 1024;
75
- const MAX_STDERR = 1024 * 1024;
76
- const MAX_TAIL = 256 * 1024;
77
- const MAX_LINE_BUFFER = 1024 * 1024;
78
- const MAX_TOOL_ARG_CHARS = 2048;
79
- const MAX_TOOLS = 256;
80
- const MAX_PARALLEL_PROMPTS = 6;
81
- const ABORT_KILL_GRACE_MS = 1500;
82
- const SESSION_OVERVIEW_TIMEOUT_MS = 1500;
83
- const THROTTLE_MS = 500;
84
-
85
- // ── State types ───────────────────────────────────────────────────────
86
- type RunStatus = "pending" | "running" | "succeeded" | "failed" | "cancelled";
87
- type ToolStatus = "running" | "succeeded" | "failed";
88
-
89
- interface Usage {
90
- input: number;
91
- output: number;
92
- cacheRead: number;
93
- cacheWrite: number;
94
- cost: number;
95
- ctxTokens: number;
96
- turns: number;
97
- }
98
- interface ToolTrace {
99
- id: string;
100
- name: string;
101
- args: string;
102
- status: ToolStatus;
103
- startedAt: number;
104
- finishedAt?: number;
105
- }
106
- interface RunState {
107
- id: string;
108
- agent: string;
109
- prompt: string;
110
- step?: number;
111
- status: RunStatus;
112
- startedAt?: number;
113
- finishedAt?: number;
114
- finalText: string;
115
- textTail: string;
116
- thinkingTail: string;
117
- stderrTail: string;
118
- tools: ToolTrace[];
119
- usage: Usage;
120
- model?: string;
121
- thinking?: string;
122
- errorMessage?: string;
123
- }
124
- interface ProgressDetails {
125
- kind: "trellis-subagent-progress";
126
- agent: string;
127
- mode: "single" | "parallel" | "chain";
128
- startedAt: number;
129
- updatedAt: number;
130
- final: boolean;
131
- runs: RunState[];
132
- }
133
-
134
- // ── Native partial-update card state ──────────────────────────────────
135
- interface NativeCardHandle {
136
- state: JsonObject;
137
- invalidate: () => void;
138
- updatedAt: number;
139
- }
140
- const MAX_NATIVE_CARDS = 20;
141
- const nativeCards = new Map<string, NativeCardHandle>();
142
- let activeSubagentToolCallId: string | null = null;
143
- function rememberNativeCard(id: string, card: NativeCardHandle) {
144
- nativeCards.set(id, card);
145
- const active = activeSubagentToolCallId
146
- ? nativeCards.get(activeSubagentToolCallId)
147
- : undefined;
148
- if (!active || card.updatedAt >= active.updatedAt)
149
- activeSubagentToolCallId = id;
150
- for (const key of nativeCards.keys()) {
151
- if (nativeCards.size <= MAX_NATIVE_CARDS) break;
152
- if (key !== activeSubagentToolCallId) nativeCards.delete(key);
153
- }
154
- }
155
- function totalUsage(d: ProgressDetails): Usage {
156
- const u: Usage = {
157
- input: 0,
158
- output: 0,
159
- cacheRead: 0,
160
- cacheWrite: 0,
161
- cost: 0,
162
- ctxTokens: 0,
163
- turns: 0,
164
- };
165
- for (const r of d.runs) {
166
- u.input += r.usage.input;
167
- u.output += r.usage.output;
168
- u.cacheRead += r.usage.cacheRead;
169
- u.cacheWrite += r.usage.cacheWrite;
170
- u.cost += r.usage.cost;
171
- u.ctxTokens = Math.max(u.ctxTokens, r.usage.ctxTokens);
172
- u.turns += r.usage.turns;
173
- }
174
- return u;
175
- }
176
- function activeRun(d: ProgressDetails) {
177
- return d.runs.find((r) => r.status === "running") ?? d.runs.at(-1);
178
- }
179
- function toolArgs(t: ToolTrace) {
180
- try {
181
- return JSON.parse(t.args) as Record<string, unknown>;
182
- } catch {
183
- return {};
184
- }
185
- }
186
- function bashCommand(t: ToolTrace) {
187
- const a = toolArgs(t);
188
- return String(a.command || "").toLowerCase();
189
- }
190
- function isSearchTool(t: ToolTrace) {
191
- return t.name === "read" || t.name === "grep" || t.name === "find";
192
- }
193
- function isMutationTool(t: ToolTrace) {
194
- return t.name === "edit" || t.name === "write";
195
- }
196
- function isValidationCommand(t: ToolTrace) {
197
- const c = bashCommand(t);
198
- return /\b(test|typecheck|lint|build|gofmt|go test|npm run|pnpm|vitest|tsc)\b/.test(
199
- c,
200
- );
201
- }
202
- function isInspectionCommand(t: ToolTrace) {
203
- const c = bashCommand(t);
204
- return /\b(rg|grep|find|git diff|git status|ls|tree)\b/.test(c);
205
- }
206
- function thinkingIntent(text: string) {
207
- const s = text.toLowerCase();
208
- if (/error|failed|failure|panic|exception|报错|失败|错误|异常/.test(s))
209
- return "Analyzing failure cause";
210
- if (/test|verify|check|typecheck|lint|验证|测试|检查/.test(s))
211
- return "Planning verification steps";
212
- if (/plan|approach|design|strategy|方案|计划|思路|设计/.test(s))
213
- return "Structuring the implementation approach";
214
- if (/implement|change|edit|modify|refactor|实现|修改|重构/.test(s))
215
- return "Reasoning through code changes";
216
- if (/inspect|search|locate|read|context|定位|搜索|阅读|上下文/.test(s))
217
- return "Locating relevant context";
218
- return "";
219
- }
220
- function behaviorSummary(r: RunState) {
221
- if (r.status === "succeeded") return "Task completed and result returned";
222
- if (r.status === "failed")
223
- return "Task failed and error details were retained";
224
-
225
- const runningTool = r.tools.findLast((t) => t.status === "running");
226
- if (runningTool) {
227
- if (isMutationTool(runningTool)) return "Applying the plan to code";
228
- if (runningTool.name === "bash" && isValidationCommand(runningTool))
229
- return "Verifying whether the implementation passes";
230
- if (runningTool.name === "bash" && isInspectionCommand(runningTool))
231
- return "Inspecting current code state";
232
- if (isSearchTool(runningTool)) return "Locating relevant code and context";
233
- if (runningTool.name === "bash")
234
- return "Validating assumptions with commands";
235
- return "Using tools to advance the task";
236
- }
237
-
238
- const recent = r.tools.slice(-5);
239
- if (recent.some((t) => t.status === "failed"))
240
- return "Investigating tool or command failure";
241
- if (recent.some(isMutationTool)) return "Reviewing recent changes";
242
- if (recent.some((t) => t.name === "bash" && isValidationCommand(t)))
243
- return "Analyzing verification results";
244
- if (
245
- recent.length >= 2 &&
246
- recent.every(
247
- (t) => isSearchTool(t) || (t.name === "bash" && isInspectionCommand(t)),
248
- )
249
- )
250
- return "Mapping code structure and impact";
251
-
252
- const intent = thinkingIntent(`${r.thinkingTail}\n${r.textTail}`);
253
- if (intent) return intent;
254
- if (!r.tools.length) return "Understanding the task and planning execution";
255
- return "Advancing the task and preparing next steps";
256
- }
257
- function progressState(d: ProgressDetails) {
258
- const running = d.runs.filter((r) => r.status === "running").length;
259
- const failed = d.runs.some((r) => r.status === "failed");
260
- return failed
261
- ? "failed"
262
- : d.final
263
- ? "completed"
264
- : running
265
- ? `${running} running`
266
- : "pending";
267
- }
268
- function progressDone(d: ProgressDetails) {
269
- return d.runs.filter((r) => r.status !== "pending" && r.status !== "running")
270
- .length;
271
- }
272
- function summaryText(text: string) {
273
- return `${text.trim().replace(/[。.!?…]+$/u, "")}...`;
274
- }
275
- function splitModelThinking(model?: string, fallbackThinking?: string) {
276
- const m = model?.match(/^(.*):(off|minimal|low|medium|high|xhigh)$/i);
277
- return {
278
- model: m ? m[1] : model,
279
- thinking: (m?.[2] ?? fallbackThinking)?.toLowerCase(),
280
- };
281
- }
282
- function modelLabel(r: RunState) {
283
- const { model, thinking } = splitModelThinking(r.model, r.thinking);
284
- if (!model) return undefined;
285
- return thinking && thinking !== "off" ? `${model}(${thinking})` : model;
286
- }
287
- function applyRunConfig(r: RunState, cfg: PiRunConfig) {
288
- const parsed = splitModelThinking(cfg.model, cfg.thinking);
289
- r.model = parsed.model;
290
- r.thinking = parsed.thinking;
291
- }
292
- function runElapsed(d: ProgressDetails, r: RunState) {
293
- const start = r.startedAt ?? d.startedAt;
294
- const end =
295
- r.finishedAt ?? (r.status === "running" ? Date.now() : d.updatedAt);
296
- return fmtDur(Math.max(0, end - start));
297
- }
298
- function runHeader(d: ProgressDetails, r: RunState) {
299
- const usage = fmtUsage(r.usage, modelLabel(r)) || fmtUsage(totalUsage(d));
300
- return `${r.agent} · ${progressDone(d)}/${d.runs.length} done · ${progressState(d)} · ${runElapsed(d, r)}${usage ? ` · ${usage}` : ""}`;
301
- }
302
- function renderRunBlock(
303
- lines: string[],
304
- d: ProgressDetails,
305
- run: RunState,
306
- expanded: boolean,
307
- ) {
308
- const step = run.step ? `step ${run.step} · ` : "";
309
- lines.push(` - ${step}${runHeader(d, run)}`);
310
- const summary = behaviorSummary(run);
311
- if (summary) lines.push(` › ${summaryText(summary)}`);
312
- const visibleTools = expanded ? run.tools.slice(-8) : run.tools.slice(-1);
313
- for (const t of visibleTools)
314
- lines.push(` ${toolIcon(t.status)} ${toolBrief(t)}`);
315
- if (expanded && run.errorMessage) {
316
- lines.push(` ✗ ${oneLine(run.errorMessage, 120)}`);
317
- }
318
- }
319
- function renderProgressCard(
320
- d: ProgressDetails,
321
- expanded: boolean,
322
- w: number,
323
- ): string[] {
324
- const r = activeRun(d);
325
- if (!r) return [];
326
- const spinner = ["◐", "◓", "◑", "◒"][Math.floor(Date.now() / 250) % 4]!;
327
- const icon = d.final
328
- ? d.runs.some((x) => x.status === "failed")
329
- ? "✗"
330
- : "✓"
331
- : spinner;
332
- const totalElapsed = fmtDur(
333
- (d.final ? d.updatedAt : Date.now()) - d.startedAt,
334
- );
335
- const lines: string[] = [
336
- `${icon} subagent ${d.mode} · total ${totalElapsed}`,
337
- ];
338
-
339
- if (!expanded) {
340
- renderRunBlock(lines, d, r, false);
341
- lines.push(" Alt+O expand latest subagent card");
342
- return lines.map((l) => trunc(l, w));
343
- }
344
-
345
- for (const run of d.runs) renderRunBlock(lines, d, run, true);
346
- lines.push(" Alt+O collapse latest subagent card");
347
- const max = 48;
348
- const shown =
349
- lines.length > max
350
- ? [
351
- ...lines.slice(0, max - 1),
352
- ` … ${lines.length - max + 1} lines hidden`,
353
- ]
354
- : lines;
355
- return shown.map((l) => trunc(l, w));
356
- }
357
- function progressKey(d: ProgressDetails) {
358
- return d.runs
359
- .map((r) => {
360
- const t = r.tools.at(-1);
361
- return [
362
- r.id,
363
- r.status,
364
- r.tools.length,
365
- t?.id ?? "",
366
- t?.status ?? "",
367
- r.usage.turns,
368
- r.usage.input,
369
- r.usage.output,
370
- r.usage.cacheRead,
371
- r.usage.cacheWrite,
372
- r.usage.ctxTokens,
373
- r.model ?? "",
374
- r.thinking ?? "",
375
- r.errorMessage ?? "",
376
- ].join("~");
377
- })
378
- .join("|");
379
- }
380
-
381
- // ── Utilities ─────────────────────────────────────────────────────────
382
- function isObj(v: unknown): v is JsonObject {
383
- return typeof v === "object" && v !== null && !Array.isArray(v);
384
- }
385
- function str(v: unknown): string | null {
386
- return typeof v === "string" && v.trim() ? v.trim() : null;
387
- }
388
- function num(v: unknown): number {
389
- return typeof v === "number" && Number.isFinite(v) ? v : 0;
390
- }
391
- function hash(s: string) {
392
- return createHash("sha256").update(s).digest("hex").slice(0, 24);
393
- }
394
- function readText(p: string) {
395
- try {
396
- return readFileSync(p, "utf-8");
397
- } catch {
398
- return "";
399
- }
400
- }
401
- function exists(p: string) {
402
- try {
403
- return statSync(p).isFile();
404
- } catch {
405
- return false;
406
- }
407
- }
408
- function shellQuote(v: string) {
409
- return `'${v.replace(/'/g, `'\\''`)}'`;
410
- }
411
- function callStr(cb: (() => string | undefined) | undefined): string | null {
412
- if (!cb) return null;
413
- try {
414
- return str(cb());
415
- } catch {
416
- return null;
417
- }
418
- }
419
- function lookupStr(data: unknown, keys: string[]): string | null {
420
- if (!isObj(data)) return null;
421
- for (const k of keys) {
422
- const v = str(data[k]);
423
- if (v) return v;
424
- }
425
- for (const nk of [
426
- "input",
427
- "properties",
428
- "event",
429
- "hook_input",
430
- "hookInput",
431
- ]) {
432
- const nested = data[nk];
433
- const v = lookupStr(nested, keys);
434
- if (v) return v;
435
- }
436
- return null;
437
- }
438
- function cmdHasTrellisCtx(cmd: string) {
439
- const t = cmd.trimStart();
440
- return (
441
- /^export\s+TRELLIS_CONTEXT_ID=/.test(t) ||
442
- /^TRELLIS_CONTEXT_ID=/.test(t) ||
443
- /^env\s+.*TRELLIS_CONTEXT_ID=/.test(t)
444
- );
445
- }
446
- function fmtDur(ms: number) {
447
- if (ms < 1000) return `${ms}ms`;
448
- const s = Math.floor(ms / 1000);
449
- if (s < 60) return `${s}s`;
450
- return `${Math.floor(s / 60)}m${s % 60}s`;
451
- }
452
- function fmtNum(n: number) {
453
- if (!n) return "0";
454
- if (Math.abs(n) < 1000) return `${n}`;
455
- if (Math.abs(n) < 1000000) return `${(n / 1000).toFixed(1)}k`;
456
- return `${(n / 1000000).toFixed(1)}m`;
457
- }
458
- function fmtUsage(u: Usage, m?: string) {
459
- const p: string[] = [];
460
- if (u.turns) p.push(`${u.turns}t`);
461
- if (u.input) p.push(`↑${fmtNum(u.input)}`);
462
- if (u.output) p.push(`↓${fmtNum(u.output)}`);
463
- if (u.cost) p.push(`$${u.cost.toFixed(3)}`);
464
- if (u.ctxTokens) p.push(`ctx:${fmtNum(u.ctxTokens)}`);
465
- if (m) p.push(m);
466
- return p.join(" ");
467
- }
468
- function statusIcon(s: RunStatus) {
469
- return s === "pending"
470
- ? "○"
471
- : s === "running"
472
- ? "●"
473
- : s === "succeeded"
474
- ? "✓"
475
- : s === "failed"
476
- ? "✗"
477
- : "⊘";
478
- }
479
- function toolIcon(s: ToolStatus) {
480
- return s === "running" ? "•" : s === "succeeded" ? "✓" : "✗";
481
- }
482
- function latest(text: string, n: number) {
483
- return text
484
- .split(/\r?\n/)
485
- .map((l) => l.trimEnd())
486
- .filter((l) => l.trim())
487
- .slice(-n);
488
- }
489
- function appendTail(cur: string, next: string, max: number) {
490
- if (!next) return cur;
491
- const c = cur + next;
492
- return c.length <= max ? c : c.slice(-max);
493
- }
494
- function extractText(content: unknown): string {
495
- if (typeof content === "string") return content;
496
- if (!Array.isArray(content)) return "";
497
- return content
498
- .map((b) =>
499
- isObj(b) && b.type === "text" && typeof b.text === "string" ? b.text : "",
500
- )
501
- .join("");
502
- }
503
- function extractThinking(content: unknown): string {
504
- if (!Array.isArray(content)) return "";
505
- return content
506
- .map((b) =>
507
- isObj(b) && b.type === "thinking" && typeof b.thinking === "string"
508
- ? b.thinking
509
- : "",
510
- )
511
- .join("\n");
512
- }
513
- function newUsage(): Usage {
514
- return {
515
- input: 0,
516
- output: 0,
517
- cacheRead: 0,
518
- cacheWrite: 0,
519
- cost: 0,
520
- ctxTokens: 0,
521
- turns: 0,
522
- };
523
- }
524
- function newRun(
525
- id: string,
526
- agent: string,
527
- prompt: string,
528
- step?: number,
529
- ): RunState {
530
- return {
531
- id,
532
- agent,
533
- prompt: trunc(prompt.replace(/\s+/g, " ").trim(), 120) || "(empty)",
534
- step,
535
- status: "pending",
536
- finalText: "",
537
- textTail: "",
538
- thinkingTail: "",
539
- stderrTail: "",
540
- tools: [],
541
- usage: newUsage(),
542
- };
543
- }
544
- function cloneProgress(d: ProgressDetails): ProgressDetails {
545
- return {
546
- ...d,
547
- runs: d.runs.map((r) => ({
548
- ...r,
549
- tools: r.tools.map((t) => ({ ...t })),
550
- usage: { ...r.usage },
551
- })),
552
- };
553
- }
554
-
555
- function oneLine(v: unknown, max = 80) {
556
- return String(v || "...")
557
- .replace(/\s+/g, " ")
558
- .trim()
559
- .slice(0, max);
560
- }
561
- function summarizeToolArgs(name: string, args: unknown): string {
562
- const a = isObj(args) ? args : {};
563
- const summary: JsonObject = {};
564
- if ("path" in a) summary.path = oneLine(a.path, 240);
565
- if ("file_path" in a) summary.file_path = oneLine(a.file_path, 240);
566
- if ("command" in a) summary.command = oneLine(a.command, 240);
567
- if ("pattern" in a) summary.pattern = oneLine(a.pattern, 120);
568
- if ("limit" in a) summary.limit = a.limit;
569
- if ("offset" in a) summary.offset = a.offset;
570
- if (name === "edit" && Array.isArray(a.edits))
571
- summary.edits = `${a.edits.length} edit(s)`;
572
- if (name === "write" && "content" in a)
573
- summary.content = `<${String(a.content ?? "").length} chars>`;
574
- const json = JSON.stringify(
575
- Object.keys(summary).length ? summary : { tool: name },
576
- );
577
- return json.length <= MAX_TOOL_ARG_CHARS
578
- ? json
579
- : json.slice(0, MAX_TOOL_ARG_CHARS);
580
- }
581
- function toolBrief(t: ToolTrace): string {
582
- const a = toolArgs(t);
583
- if (t.name === "read") return `read: ${oneLine(a.path || a.file_path, 80)}`;
584
- if (t.name === "bash") return `bash: ${oneLine(a.command, 60)}`;
585
- if (t.name === "write") return `write: ${oneLine(a.path || a.file_path, 80)}`;
586
- if (t.name === "edit") return `edit: ${oneLine(a.path || a.file_path, 80)}`;
587
- if (t.name === "grep") return `grep: ${oneLine(a.pattern, 50)}`;
588
- if (t.name === "find") return `find: ${oneLine(a.pattern || "*", 50)}`;
589
- return oneLine(t.name, 50);
590
- }
591
-
592
- // ── Pi CLI path resolution ────────────────────────────────────────────
593
- const PI_CLI_SEGMENTS = [
594
- ["node_modules", "@earendil-works", "pi-coding-agent", "dist", "cli.js"],
595
- ["node_modules", "@mariozechner", "pi-coding-agent", "dist", "cli.js"],
596
- ];
597
-
598
- function resolvePiCli(): { command: string; args: string[] } {
599
- const envCli = str(process.env.TRELLIS_PI_CLI_JS);
600
- if (envCli) {
601
- const p = resolve(envCli);
602
- if (!exists(p)) throw new Error(`TRELLIS_PI_CLI_JS missing: ${p}`);
603
- return { command: process.execPath, args: [p] };
604
- }
605
- const candidates: string[] = [];
606
- for (const arg of process.argv)
607
- if (/pi-coding-agent[\\/]dist[\\/]cli\.js$/i.test(arg))
608
- candidates.push(resolve(arg));
609
- const prefix =
610
- str(process.env.npm_config_prefix) ?? str(process.env.NPM_CONFIG_PREFIX);
611
- const appData = str(process.env.APPDATA);
612
- const pathVal = process.env.PATH ?? process.env.Path ?? "";
613
- const addBase = (base: string) => {
614
- for (const seg of PI_CLI_SEGMENTS) candidates.push(join(base, ...seg));
615
- };
616
- if (prefix) {
617
- addBase(prefix);
618
- addBase(join(prefix, "lib"));
619
- }
620
- if (appData) addBase(join(appData, "npm"));
621
- for (const entry of pathVal.split(delimiter)) {
622
- const e = entry.trim();
623
- if (!e) continue;
624
- addBase(e);
625
- addBase(dirname(e));
626
- addBase(join(dirname(e), "lib"));
627
- }
628
- for (const c of [...new Set(candidates)])
629
- if (exists(c)) return { command: process.execPath, args: [c] };
630
- return { command: "pi", args: [] };
631
- }
632
-
633
- function resolveRunCfg(
634
- input: SubagentInput,
635
- agentCfg: AgentConfig,
636
- inheritedThinking?: string,
637
- ): PiRunConfig {
638
- const THINKING_LEVELS = ["off", "minimal", "low", "medium", "high", "xhigh"];
639
- const normalize = (v: unknown): string | undefined => {
640
- const s = typeof v === "string" && v.trim() ? v.trim().toLowerCase() : "";
641
- return THINKING_LEVELS.includes(s) ? s : undefined;
642
- };
643
- const suffixRe = /:(off|minimal|low|medium|high|xhigh)$/i;
644
- const inputModel = str(input.model);
645
- const agentModel = agentCfg.model;
646
- const rawModel = inputModel ?? agentModel;
647
- const inputSuffixThinking = normalize(inputModel?.match(suffixRe)?.[1]);
648
- const agentSuffixThinking = normalize(agentModel?.match(suffixRe)?.[1]);
649
- const baseModel = rawModel?.replace(suffixRe, "");
650
- const thinking =
651
- normalize(input.thinking) ??
652
- inputSuffixThinking ??
653
- normalize(agentCfg.thinking) ??
654
- agentSuffixThinking ??
655
- normalize(inheritedThinking);
656
- if (baseModel && thinking && thinking !== "off")
657
- return { model: `${baseModel}:${thinking}`, thinking };
658
- return { model: baseModel || rawModel, thinking };
659
- }
660
-
661
- function buildPiArgs(cfg: PiRunConfig): string[] {
662
- const args = ["--mode", "json", "-p", "--no-session"];
663
- if (cfg.model)
664
- args.push(
665
- "--model",
666
- cfg.thinking && cfg.thinking !== "off" && !cfg.model.includes(":")
667
- ? `${cfg.model}:${cfg.thinking}`
668
- : cfg.model,
669
- );
670
- else if (cfg.thinking && cfg.thinking !== "off")
671
- args.push("--thinking", cfg.thinking);
672
- return args;
673
- }
674
-
675
- // ── BoundedBufferCollector ─────────────────────────────────────────────
676
- class BBC {
677
- private c: Buffer[] = [];
678
- private len = 0;
679
- private trunc = 0;
680
- constructor(private max: number) {}
681
- append(b: Buffer) {
682
- if (b.length >= this.max) {
683
- this.trunc += this.len + b.length - this.max;
684
- this.c = [b.subarray(b.length - this.max)];
685
- this.len = this.max;
686
- return;
687
- }
688
- this.c.push(b);
689
- this.len += b.length;
690
- while (this.len > this.max) {
691
- const f = this.c[0]!;
692
- if (f.length <= this.len - this.max) {
693
- this.c.shift();
694
- this.len -= f.length;
695
- this.trunc += f.length;
696
- } else {
697
- const ov = this.len - this.max;
698
- this.c[0] = f.subarray(ov);
699
- this.len -= ov;
700
- this.trunc += ov;
701
- break;
702
- }
703
- }
704
- }
705
- toString() {
706
- const body = Buffer.concat(this.c, this.len).toString("utf-8");
707
- return this.trunc ? `[${this.trunc} bytes truncated]\n${body}` : body;
708
- }
709
- }
710
-
711
- // ── Trellis Context ────────────────────────────────────────────────────
712
- function findRoot(start: string): string {
713
- let c = resolve(start);
714
- while (true) {
715
- if (existsSync(join(c, ".trellis")) || existsSync(join(c, ".pi"))) return c;
716
- const p = dirname(c);
717
- if (p === c) return resolve(start);
718
- c = p;
719
- }
720
- }
721
- function splitFM(c: string) {
722
- const m = c.replace(/^\uFEFF/, "").match(/^---\r?\n([\s\S]*?)\r?\n---\r?\n?/);
723
- return m
724
- ? { fm: m[1] ?? "", body: c.slice(m[0].length) }
725
- : { fm: "", body: c };
726
- }
727
- function stripFM(c: string) {
728
- return splitFM(c).body.trimStart();
729
- }
730
- function parseAgentFM(c: string): AgentConfig {
731
- const cfg: AgentConfig = { fallbackModels: [] };
732
- const { fm } = splitFM(c);
733
- const lines = fm.split(/\r?\n/);
734
- for (let i = 0; i < lines.length; i++) {
735
- const m = (lines[i] ?? "").match(/^([A-Za-z][A-Za-z0-9_-]*)\s*:\s*(.*)$/);
736
- if (!m) continue;
737
- const k = m[1] ?? "",
738
- v = m[2] ?? "";
739
- if (k === "model")
740
- cfg.model = v.trim().replace(/^["']|["']$/g, "") || undefined;
741
- else if (k === "thinking")
742
- cfg.thinking = (v.trim().replace(/^["']|["']$/g, "") || undefined) as
743
- | string
744
- | undefined;
745
- else if (k === "fallbackModels" || k === "fallback_models") {
746
- if (v.trim()) {
747
- cfg.fallbackModels = v
748
- .trim()
749
- .replace(/^\[|\]$/g, "")
750
- .split(",")
751
- .map((s) => s.trim().replace(/^["']|["']$/g, ""))
752
- .filter(Boolean);
753
- } else {
754
- i++;
755
- while (i < lines.length && /^\s+-\s/.test(lines[i] ?? "")) {
756
- const item = (lines[i] ?? "")
757
- .trim()
758
- .replace(/^-\s+/, "")
759
- .replace(/^["']|["']$/g, "");
760
- if (item) cfg.fallbackModels.push(item);
761
- i++;
762
- }
763
- i--;
764
- }
765
- }
766
- }
767
- return cfg;
768
- }
769
-
770
- function contextKey(input?: unknown, ctx?: PiExtensionContext): string | null {
771
- const ov = str(process.env.TRELLIS_CONTEXT_ID);
772
- if (ov) return ov.replace(/[^A-Za-z0-9._-]+/g, "_").slice(0, 160) || hash(ov);
773
- const sessionId =
774
- callStr(ctx?.sessionManager?.getSessionId) ??
775
- str(process.env.PI_SESSION_ID) ??
776
- str(process.env.PI_SESSIONID) ??
777
- lookupStr(input, ["session_id", "sessionId", "sessionID"]);
778
- if (sessionId)
779
- return `pi_${sessionId.replace(/[^A-Za-z0-9._-]+/g, "_") || hash(sessionId)}`;
780
- const transcriptPath =
781
- callStr(ctx?.sessionManager?.getSessionFile) ??
782
- lookupStr(input, ["transcript_path", "transcriptPath", "transcript"]);
783
- if (transcriptPath) return `pi_transcript_${hash(transcriptPath)}`;
784
- return null;
785
- }
786
-
787
- function readTaskDir(root: string, key: string | null): string | null {
788
- if (!key) return null;
789
- try {
790
- const ctx = JSON.parse(
791
- readText(join(root, ".trellis", ".runtime", "sessions", `${key}.json`)),
792
- ) as JsonObject;
793
- let ref = str(ctx.current_task);
794
- if (!ref) return null;
795
- ref = ref;
796
- ref = ref.replace(/\\/g, "/").replace(/^\.\//, "");
797
- if (ref.startsWith("tasks/")) ref = `.trellis/${ref}`;
798
- return ref.startsWith(".trellis/")
799
- ? join(root, ref)
800
- : isAbsolute(ref)
801
- ? ref
802
- : join(root, ".trellis", "tasks", ref);
803
- } catch {
804
- return null;
805
- }
806
- }
807
- function sessionHasTask(root: string, key: string): boolean {
808
- try {
809
- const ctx = JSON.parse(
810
- readText(join(root, ".trellis", ".runtime", "sessions", `${key}.json`)),
811
- ) as JsonObject;
812
- return !!str(ctx.current_task);
813
- } catch {
814
- return false;
815
- }
816
- }
817
- function adoptKey(root: string, key: string): string {
818
- if (sessionHasTask(root, key)) return key;
819
- try {
820
- const dir = join(root, ".trellis", ".runtime", "sessions");
821
- const keys = readdirSync(dir)
822
- .filter(
823
- (f) => f.endsWith(".json") && sessionHasTask(root, f.slice(0, -5)),
824
- )
825
- .map((f) => f.slice(0, -5));
826
- const proc = keys.filter((k) => k.startsWith("pi_process_"));
827
- const cands = proc.length ? proc : keys;
828
- return cands.length === 1 ? cands[0]! : key;
829
- } catch {
830
- return key;
831
- }
832
- }
833
-
834
- // ── Workflow State Breadcrumb ─────────────────────────────────────────
835
- const WF_RE =
836
- /\[workflow-state:([A-Za-z0-9_-]+)\]\s*\n([\s\S]*?)\n\s*\[\/workflow-state:\1\]/g;
837
- function workflowBreadcrumb(root: string, key: string | null): string {
838
- const wf = readText(join(root, ".trellis", "workflow.md"));
839
- if (!wf) return "";
840
- const templates: Record<string, string> = {};
841
- for (const m of wf.matchAll(WF_RE)) {
842
- const s = m[1] ?? "",
843
- b = (m[2] ?? "").trim();
844
- if (s && b) templates[s] = b;
845
- }
846
- const dir = readTaskDir(root, key);
847
- let header = "Status: no_task",
848
- lookup = "no_task";
849
- if (dir) {
850
- try {
851
- const d = JSON.parse(readText(join(dir, "task.json"))) as JsonObject;
852
- const status = str(d.status) ?? "";
853
- const id = str(d.id) ?? dir.split(/[\\/]/).pop() ?? "";
854
- if (status) {
855
- header = `Task: ${id} (${status})`;
856
- lookup = status;
857
- }
858
- } catch {}
859
- }
860
- const body = templates[lookup] ?? "Refer to workflow.md for current step.";
861
- return `<workflow-state>\n${header}\n${body}\n</workflow-state>`;
862
- }
863
-
864
- // ── Session Overview ───────────────────────────────────────────────────
865
- function sessionOverview(root: string, key: string | null): string {
866
- const script = join(root, ".trellis", "scripts", "get_context.py");
867
- if (!exists(script)) return "";
868
- try {
869
- const py = process.platform === "win32" ? "python" : "python3";
870
- const result = spawnSync(py, [script], {
871
- cwd: root,
872
- env: key ? { ...process.env, TRELLIS_CONTEXT_ID: key } : process.env,
873
- encoding: "utf-8",
874
- timeout: SESSION_OVERVIEW_TIMEOUT_MS,
875
- windowsHide: true,
876
- });
877
- if (result.status !== 0) return "";
878
- const stdout = (result.stdout ?? "").trim();
879
- return stdout ? `<session-overview>\n${stdout}\n</session-overview>` : "";
880
- } catch {
881
- return "";
882
- }
883
- }
884
-
885
- function buildContext(root: string, agent: string, key: string | null): string {
886
- const dir = readTaskDir(root, key);
887
- if (!dir)
888
- return "No active Trellis task found. Read .trellis/ before proceeding.";
889
- const prd = readText(join(dir, "prd.md"));
890
- const design = readText(join(dir, "design.md"));
891
- const impl = readText(join(dir, "implement.md"));
892
- const jsonlName = TRELLIS_AGENT_JSONL[agent] ?? "";
893
- let spec = "";
894
- if (jsonlName) {
895
- const chunks: string[] = [];
896
- for (const line of readText(join(dir, jsonlName)).split(/\r?\n/)) {
897
- const t = line.trim();
898
- if (!t) continue;
899
- try {
900
- const r = JSON.parse(t) as JsonObject;
901
- const f = typeof r.file === "string" ? r.file : "";
902
- if (f) {
903
- const c = readText(join(root, f));
904
- if (c) chunks.push(`## ${f}\n\n${c}`);
905
- }
906
- } catch {}
907
- }
908
- spec = chunks.join("\n\n---\n\n");
909
- }
910
- return [
911
- `## Trellis Task Context`,
912
- `Task directory: ${dir}`,
913
- "",
914
- "### prd.md",
915
- prd || "(missing)",
916
- design ? "\n### design.md\n" + design : "",
917
- impl ? "\n### implement.md\n" + impl : "",
918
- spec ? "\n### Curated Spec / Research Context\n" + spec : "",
919
- ].join("\n");
920
- }
921
-
922
- function normalizeAgent(agent: string | undefined): string {
923
- const name = agent ?? "trellis-implement";
924
- return name.startsWith("trellis-") ? name : `trellis-${name}`;
925
- }
926
-
927
- function isTrellisAgent(root: string, agent: string): boolean {
928
- return existsSync(join(root, ".pi", "agents", `${agent}.md`));
929
- }
930
-
931
- function buildPrompt(
932
- root: string,
933
- input: SubagentInput,
934
- key: string | null,
935
- ): string {
936
- const agent = normalizeAgent(input.agent);
937
- const raw = readText(join(root, ".pi", "agents", `${agent}.md`));
938
- const def = stripFM(raw);
939
- const ctx = buildContext(root, agent, key);
940
- return [
941
- "## Trellis Agent Definition",
942
- def || "(missing)",
943
- "",
944
- ctx,
945
- "",
946
- "## Delegated Task",
947
- input.prompt ?? "",
948
- ].join("\n");
949
- }
950
-
951
- // ── Event parsing ─────────────────────────────────────────────────────
952
- function parseJsonEvent(line: string): JsonObject | null {
953
- const t = line.trim();
954
- if (!t) return null;
955
- const i = t.indexOf("{");
956
- if (i < 0) return null;
957
- try {
958
- const p = JSON.parse(t.slice(i));
959
- return isObj(p) ? p : null;
960
- } catch {
961
- return null;
962
- }
963
- }
964
-
965
- function applyEvent(r: RunState, evt: JsonObject): boolean {
966
- const type = typeof evt.type === "string" ? evt.type : "";
967
- if (!type) return false;
968
- if (type === "agent_start" || type === "turn_start") {
969
- r.status = "running";
970
- r.startedAt ??= Date.now();
971
- return true;
972
- }
973
- if (type === "message_update") {
974
- const ae = isObj(evt.assistantMessageEvent)
975
- ? evt.assistantMessageEvent
976
- : null;
977
- if (!ae || typeof ae.delta !== "string") return false;
978
- if (ae.type === "thinking_delta") {
979
- r.thinkingTail = appendTail(r.thinkingTail, ae.delta, MAX_TAIL);
980
- return true;
981
- }
982
- if (ae.type === "text_delta") {
983
- r.textTail = appendTail(r.textTail, ae.delta, MAX_TAIL);
984
- return true;
985
- }
986
- return false;
987
- }
988
- if (type === "message_end" && isObj(evt.message)) {
989
- const msg = evt.message;
990
- if (msg.role !== "assistant") return false;
991
- r.usage.turns += 1;
992
- const u = isObj(msg.usage) ? msg.usage : null;
993
- const cost = isObj(u?.cost) ? u.cost : null;
994
- r.usage.input += num(u?.input);
995
- r.usage.output += num(u?.output);
996
- r.usage.cacheRead += num(u?.cacheRead);
997
- r.usage.cacheWrite += num(u?.cacheWrite);
998
- r.usage.cost += num(cost?.total);
999
- r.usage.ctxTokens = num(u?.totalTokens);
1000
- const thinking = extractThinking(msg.content);
1001
- if (thinking) r.thinkingTail = appendTail("", thinking, MAX_TAIL);
1002
- const text = extractText(msg.content);
1003
- if (text) {
1004
- r.finalText = text;
1005
- r.textTail = appendTail("", text, MAX_TAIL);
1006
- }
1007
- if (typeof msg.model === "string") {
1008
- const parsed = splitModelThinking(msg.model, r.thinking);
1009
- r.model = parsed.model;
1010
- r.thinking = parsed.thinking;
1011
- }
1012
- if (typeof msg.errorMessage === "string") r.errorMessage = msg.errorMessage;
1013
- return true;
1014
- }
1015
- if (type === "tool_execution_start") {
1016
- const id =
1017
- typeof evt.toolCallId === "string"
1018
- ? evt.toolCallId
1019
- : hash(`${Date.now()}`);
1020
- const name = typeof evt.toolName === "string" ? evt.toolName : "tool";
1021
- const args = summarizeToolArgs(name, evt.args);
1022
- const existing = r.tools.findIndex((t) => t.id === id);
1023
- if (existing >= 0)
1024
- r.tools[existing] = { ...r.tools[existing]!, args, status: "running" };
1025
- else
1026
- r.tools.push({
1027
- id,
1028
- name,
1029
- args,
1030
- status: "running",
1031
- startedAt: Date.now(),
1032
- });
1033
- if (r.tools.length > MAX_TOOLS)
1034
- r.tools.splice(0, r.tools.length - MAX_TOOLS);
1035
- return true;
1036
- }
1037
- if (type === "tool_execution_end") {
1038
- const id = typeof evt.toolCallId === "string" ? evt.toolCallId : "";
1039
- const idx = r.tools.findIndex((t) => t.id === id);
1040
- if (idx >= 0)
1041
- r.tools[idx] = {
1042
- ...r.tools[idx]!,
1043
- status: evt.isError ? "failed" : "succeeded",
1044
- finishedAt: Date.now(),
1045
- };
1046
- return true;
1047
- }
1048
- if (type === "agent_end") {
1049
- r.finishedAt = Date.now();
1050
- if (r.status === "running" || r.status === "pending")
1051
- r.status = "succeeded";
1052
- return true;
1053
- }
1054
- return false;
1055
- }
1056
-
1057
- function finalize(r: RunState, fallback: string): string {
1058
- return r.finalText || fallback.trim() || r.stderrTail.trim();
1059
- }
1060
- function formatPiOutput(stdout: string, stderr: string): string {
1061
- let ft = "";
1062
- for (const line of stdout.split(/\r?\n/)) {
1063
- const t = line.trim();
1064
- if (!t) continue;
1065
- try {
1066
- const evt = JSON.parse(t) as JsonObject;
1067
- const msg = isObj(evt.message) ? evt.message : null;
1068
- if (msg?.role === "assistant") {
1069
- const txt = extractText(msg.content);
1070
- if (txt) ft = txt;
1071
- }
1072
- } catch {}
1073
- }
1074
- return ft || stdout || stderr;
1075
- }
1076
-
1077
- // ── runPi: subprocess execution + event processing ───────────────────
1078
- function runPi(
1079
- root: string,
1080
- prompt: string,
1081
- cfg: PiRunConfig,
1082
- state: RunState,
1083
- emit: () => void,
1084
- key?: string | null,
1085
- signal?: AbortSignal,
1086
- ): Promise<{ output: string; failed: boolean }> {
1087
- return new Promise((resolve) => {
1088
- if (signal?.aborted) {
1089
- state.status = "cancelled";
1090
- state.errorMessage = "cancelled";
1091
- state.finishedAt = Date.now();
1092
- emit();
1093
- resolve({ output: "cancelled", failed: true });
1094
- return;
1095
- }
1096
- const inv = resolvePiCli();
1097
- const childEnv = {
1098
- ...process.env,
1099
- TRELLIS_SUBAGENT_CHILD: "1",
1100
- ...(key ? { TRELLIS_CONTEXT_ID: key } : {}),
1101
- };
1102
- const cli = spawn(inv.command, [...inv.args, ...buildPiArgs(cfg)], {
1103
- cwd: root,
1104
- env: childEnv,
1105
- stdio: ["pipe", "pipe", "pipe"],
1106
- windowsHide: true,
1107
- });
1108
- const stdout = new BBC(MAX_STDOUT);
1109
- const stderr = new BBC(MAX_STDERR);
1110
- let buf = "";
1111
- let settled = false;
1112
- let aborted = false;
1113
- let killTimer: ReturnType<typeof setTimeout> | null = null;
1114
- const abort = () => {
1115
- aborted = true;
1116
- cli.kill();
1117
- killTimer = setTimeout(() => {
1118
- if (!settled && cli.exitCode === null) cli.kill("SIGKILL");
1119
- }, ABORT_KILL_GRACE_MS);
1120
- killTimer?.unref?.();
1121
- };
1122
- const done = (v: { output: string; failed: boolean }) => {
1123
- if (settled) return;
1124
- settled = true;
1125
- if (killTimer) clearTimeout(killTimer);
1126
- signal?.removeEventListener("abort", abort);
1127
- emit();
1128
- resolve(v);
1129
- };
1130
- signal?.addEventListener("abort", abort, { once: true });
1131
- state.status = "running";
1132
- state.startedAt = Date.now();
1133
- emit();
1134
- const processLine = (line: string) => {
1135
- const evt = parseJsonEvent(line);
1136
- if (evt && applyEvent(state, evt)) emit();
1137
- };
1138
- cli.stdout?.on("data", (d: Buffer) => {
1139
- stdout.append(d);
1140
- buf += d.toString("utf-8");
1141
- if (buf.length > MAX_LINE_BUFFER) buf = buf.slice(-MAX_LINE_BUFFER);
1142
- const lines = buf.split(/\r?\n/);
1143
- buf = lines.pop() ?? "";
1144
- for (const l of lines) processLine(l);
1145
- });
1146
- cli.stderr?.on("data", (d: Buffer) => {
1147
- stderr.append(d);
1148
- state.stderrTail = appendTail(
1149
- state.stderrTail,
1150
- d.toString("utf-8"),
1151
- MAX_TAIL,
1152
- );
1153
- });
1154
- cli.stdin?.on("error", (e: Error & { code?: string }) => {
1155
- if (!aborted && e.code !== "EPIPE")
1156
- done({ output: e.message, failed: true });
1157
- });
1158
- cli.on("error", (e) => {
1159
- state.status = aborted ? "cancelled" : "failed";
1160
- state.errorMessage = e instanceof Error ? e.message : String(e);
1161
- state.finishedAt = Date.now();
1162
- done({ output: finalize(state, state.errorMessage), failed: true });
1163
- });
1164
- cli.on("close", (code) => {
1165
- if (buf.trim()) processLine(buf);
1166
- const out = stdout.toString();
1167
- const err = stderr.toString();
1168
- state.stderrTail = appendTail("", err, MAX_TAIL);
1169
- state.finishedAt = Date.now();
1170
- if (aborted) {
1171
- state.status = "cancelled";
1172
- state.errorMessage = "cancelled";
1173
- done({ output: finalize(state, "cancelled"), failed: true });
1174
- return;
1175
- }
1176
- if (code === 0) {
1177
- if (state.status === "pending" || state.status === "running")
1178
- state.status = "succeeded";
1179
- done({
1180
- output: finalize(state, formatPiOutput(out, err)),
1181
- failed: false,
1182
- });
1183
- return;
1184
- }
1185
- state.status = "failed";
1186
- state.errorMessage = err || out || `exit ${code ?? "?"}`;
1187
- done({ output: finalize(state, state.errorMessage), failed: true });
1188
- });
1189
- cli.stdin?.end(prompt);
1190
- });
1191
- }
1192
-
1193
- // ── runSubagent: orchestrate single/parallel/chain via native partial updates ──
1194
- async function runSubagent(
1195
- root: string,
1196
- input: SubagentInput,
1197
- key: string | null,
1198
- signal?: AbortSignal,
1199
- onUpdate?: (r: PiToolResult) => void,
1200
- inheritedThinking?: string,
1201
- ): Promise<{ output: string; details: ProgressDetails; failed: boolean }> {
1202
- const agentName = normalizeAgent(input.agent);
1203
- const agentRaw = readText(join(root, ".pi", "agents", `${agentName}.md`));
1204
- const agentCfg = parseAgentFM(agentRaw);
1205
- const runCfg = resolveRunCfg(input, agentCfg, inheritedThinking);
1206
- const mode = input.mode ?? "single";
1207
- const startedAt = Date.now();
1208
- const details: ProgressDetails = {
1209
- kind: "trellis-subagent-progress",
1210
- agent: agentName,
1211
- mode,
1212
- startedAt,
1213
- updatedAt: startedAt,
1214
- final: false,
1215
- runs: [],
1216
- };
1217
- let lastEmit = 0;
1218
- let lastPartialKey = "";
1219
- let closed = false;
1220
- const pushPartial = (force = false) => {
1221
- if (closed || !onUpdate) return;
1222
- const key = progressKey(details);
1223
- if (!force && key === lastPartialKey) return;
1224
- lastPartialKey = key;
1225
- onUpdate({
1226
- // Keep native partial content stable; renderResult owns the visible progress UI.
1227
- content: [{ type: "text", text: "subagent running" }],
1228
- details: cloneProgress(details),
1229
- });
1230
- };
1231
- const emit = (force = false) => {
1232
- const now = Date.now();
1233
- if (!force && now - lastEmit < THROTTLE_MS) return;
1234
- lastEmit = now;
1235
- details.updatedAt = now;
1236
- pushPartial(force);
1237
- };
1238
- const finish = (output: string, failed: boolean) => {
1239
- closed = true;
1240
- details.final = true;
1241
- details.updatedAt = Date.now();
1242
- return { output, details: cloneProgress(details), failed };
1243
- };
1244
-
1245
- try {
1246
- if (mode === "parallel") {
1247
- const prompts = input.prompts ?? (input.prompt ? [input.prompt] : []);
1248
- details.runs = prompts.map((p, i) => {
1249
- const r = newRun(`${agentName}-${i + 1}`, agentName, p);
1250
- applyRunConfig(r, runCfg);
1251
- return r;
1252
- });
1253
- emit(true);
1254
- const results = await Promise.all(
1255
- prompts.map((p, i) =>
1256
- runPi(
1257
- root,
1258
- buildPrompt(root, { ...input, prompt: p }, key),
1259
- runCfg,
1260
- details.runs[i]!,
1261
- emit,
1262
- key,
1263
- signal,
1264
- ),
1265
- ),
1266
- );
1267
- return finish(
1268
- results.map((r) => r.output).join("\n\n---\n\n"),
1269
- results.some((r) => r.failed),
1270
- );
1271
- }
1272
- if (mode === "chain") {
1273
- let prev = "";
1274
- let failed = false;
1275
- for (let i = 0; i < (input.prompts?.length ?? 1); i++) {
1276
- const p = input.prompts?.[i] ?? input.prompt ?? "";
1277
- const rs = newRun(`${agentName}-${i + 1}`, agentName, p, i + 1);
1278
- applyRunConfig(rs, runCfg);
1279
- details.runs.push(rs);
1280
- emit(true);
1281
- const result = await runPi(
1282
- root,
1283
- buildPrompt(
1284
- root,
1285
- {
1286
- ...input,
1287
- prompt: prev ? `${p}\n\nPrevious output:\n${prev}` : p,
1288
- },
1289
- key,
1290
- ),
1291
- runCfg,
1292
- rs,
1293
- emit,
1294
- key,
1295
- signal,
1296
- );
1297
- prev = result.output;
1298
- failed = failed || result.failed;
1299
- if (result.failed) break;
1300
- }
1301
- return finish(prev, failed);
1302
- }
1303
- const rs = newRun(`${agentName}-1`, agentName, input.prompt ?? "");
1304
- applyRunConfig(rs, runCfg);
1305
- details.runs = [rs];
1306
- emit(true);
1307
- const result = await runPi(
1308
- root,
1309
- buildPrompt(root, input, key),
1310
- runCfg,
1311
- rs,
1312
- emit,
1313
- key,
1314
- signal,
1315
- );
1316
- return finish(result.output, result.failed);
1317
- } catch (e) {
1318
- const message = e instanceof Error ? e.message : String(e);
1319
- const r = activeRun(details);
1320
- if (r) {
1321
- r.status = "failed";
1322
- r.errorMessage = message;
1323
- r.finishedAt = Date.now();
1324
- }
1325
- return finish(message, true);
1326
- }
1327
- }
1328
-
1329
- // ── Extension ──────────────────────────────────────────────────────────
1330
- export default function trellisExtension(pi: {
1331
- registerTool?: (tool: JsonObject) => void;
1332
- registerShortcut?: (
1333
- key: string,
1334
- opts: {
1335
- description?: string;
1336
- handler: (ctx: PiExtensionContext) => unknown;
1337
- },
1338
- ) => void;
1339
- on?: (
1340
- event: string,
1341
- handler: (event: unknown, ctx?: PiExtensionContext) => unknown,
1342
- ) => void;
1343
- getThinkingLevel?: () => string;
1344
- }): void {
1345
- if (process.env.TRELLIS_SUBAGENT_CHILD === "1") return;
1346
- const root = findRoot(process.cwd());
1347
- const procKey = `pi_process_${hash([root, process.pid, Date.now(), randomBytes(8).toString("hex")].join(":"))}`;
1348
- let curKey: string | null = null;
1349
-
1350
- const getKey = (input?: unknown, ctx?: PiExtensionContext) => {
1351
- const k = adoptKey(root, contextKey(input, ctx) ?? curKey ?? procKey);
1352
- curKey = k;
1353
- return k;
1354
- };
1355
-
1356
- // Per-turn cache to avoid double-spawning python
1357
- let turnCache: {
1358
- key: string | null;
1359
- ts: number;
1360
- wf: string;
1361
- ov: string;
1362
- } | null = null;
1363
- const getTurnCtx = (k: string | null) => {
1364
- const now = Date.now();
1365
- if (turnCache && turnCache.key === k && now - turnCache.ts < 1500)
1366
- return turnCache;
1367
- turnCache = {
1368
- key: k,
1369
- ts: now,
1370
- wf: workflowBreadcrumb(root, k),
1371
- ov: sessionOverview(root, k),
1372
- };
1373
- return turnCache;
1374
- };
1375
-
1376
- // Toggle only the latest subagent native card; do not use Pi global tool expansion.
1377
- const toggleDetail = (ctx: PiExtensionContext) => {
1378
- const id = activeSubagentToolCallId;
1379
- const card = id ? nativeCards.get(id) : undefined;
1380
- if (!card) {
1381
- ctx.ui?.notify?.("No subagent card to toggle yet.", "warning");
1382
- return;
1383
- }
1384
- card.state.localExpanded = card.state.localExpanded !== true;
1385
- card.invalidate();
1386
- };
1387
-
1388
- pi.registerShortcut?.("alt+o", {
1389
- description: "Toggle latest subagent card details",
1390
- handler: async (ctx: PiExtensionContext) => toggleDetail(ctx),
1391
- });
1392
-
1393
- // Tool registration
1394
- pi.registerTool?.({
1395
- name: "trellis_subagent",
1396
- label: "Trellis Subagent",
1397
- description: "Run a Trellis project sub-agent with selected task context.",
1398
- promptSnippet:
1399
- 'Sub-agent dispatch protocol (Trellis): your dispatch prompt MUST start with one line "Selected task: <task path from `task.py selected`>" before any other instructions.',
1400
- promptGuidelines: [
1401
- 'Use subagent for task delegation. Your dispatch prompt MUST start with "Selected task: <task path from `task.py selected`>".',
1402
- ],
1403
- parameters: {
1404
- type: "object",
1405
- properties: {
1406
- agent: {
1407
- type: "string",
1408
- description:
1409
- "Agent name, such as trellis-implement or trellis-check.",
1410
- },
1411
- prompt: {
1412
- type: "string",
1413
- description: "Task prompt for the sub-agent.",
1414
- },
1415
- mode: { type: "string", enum: ["single", "parallel", "chain"] },
1416
- prompts: {
1417
- type: "array",
1418
- items: { type: "string" },
1419
- maxItems: MAX_PARALLEL_PROMPTS,
1420
- },
1421
- model: {
1422
- type: "string",
1423
- description:
1424
- "Optional Pi model override for the child sub-agent process.",
1425
- },
1426
- thinking: {
1427
- type: "string",
1428
- description:
1429
- "Optional Pi thinking level override for the child sub-agent process.",
1430
- enum: ["off", "minimal", "low", "medium", "high", "xhigh"],
1431
- },
1432
- },
1433
- },
1434
- execute: async (
1435
- id: string,
1436
- input: SubagentInput,
1437
- signal?: AbortSignal,
1438
- onUpdate?: (r: PiToolResult) => void,
1439
- ctx?: PiExtensionContext,
1440
- ) => {
1441
- activeSubagentToolCallId = id;
1442
- const agentName = normalizeAgent(input.agent);
1443
- if (!isTrellisAgent(root, agentName)) {
1444
- return {
1445
- content: [
1446
- {
1447
- type: "text",
1448
- text:
1449
- "`trellis_subagent` is only for Trellis workflow agents with a definition file in .pi/agents/.\n\n" +
1450
- `No definition found for: ${agentName}\n\n` +
1451
- "For general-purpose sub-agents, use one of these community tools:\n" +
1452
- "- `subagent` tool from npm:pi-subagents (nicobailon/pi-subagents)\n" +
1453
- "- `Agent` tool from npm:@tintinweb/pi-subagents\n\n" +
1454
- "If neither is installed, ask the user to either:\n" +
1455
- `- Create .pi/agents/${agentName}.md for your custom Trellis agent\n` +
1456
- "- Install a community subagent package: pi install -l npm:@tintinweb/pi-subagents",
1457
- },
1458
- ],
1459
- details: { agent: agentName, error: "not a trellis workflow agent" },
1460
- };
1461
- }
1462
- const mode = input.mode ?? "single";
1463
- const prompt = input.prompt?.trim();
1464
- const prompts = input.prompts?.map((p) => p.trim()).filter(Boolean);
1465
- if (mode === "single" && !prompt)
1466
- throw new Error("subagent prompt is required for single mode");
1467
- if (
1468
- (mode === "parallel" || mode === "chain") &&
1469
- !prompt &&
1470
- !prompts?.length
1471
- )
1472
- throw new Error(
1473
- "subagent prompt or prompts are required for parallel/chain mode",
1474
- );
1475
- if (
1476
- mode === "parallel" &&
1477
- prompts &&
1478
- prompts.length > MAX_PARALLEL_PROMPTS
1479
- )
1480
- throw new Error(
1481
- `subagent parallel mode supports at most ${MAX_PARALLEL_PROMPTS} prompts`,
1482
- );
1483
- const cleanInput: SubagentInput = {
1484
- ...input,
1485
- prompt,
1486
- prompts: prompts?.length ? prompts : undefined,
1487
- };
1488
- const key = getKey(cleanInput, ctx);
1489
- const inheritedThinking = pi.getThinkingLevel?.();
1490
- const result = await runSubagent(
1491
- root,
1492
- cleanInput,
1493
- key,
1494
- signal,
1495
- onUpdate,
1496
- inheritedThinking,
1497
- );
1498
- return {
1499
- content: [{ type: "text", text: result.output }],
1500
- details: result.details,
1501
- };
1502
- },
1503
- // Hide the call renderer so the native card only shows result/progress once.
1504
- renderCall: () => ({
1505
- render() {
1506
- return [];
1507
- },
1508
- invalidate() {},
1509
- }),
1510
- renderResult: (
1511
- result: PiToolResult,
1512
- _opts?: { expanded?: boolean; isPartial?: boolean },
1513
- _theme?: unknown,
1514
- context?: unknown,
1515
- ) => {
1516
- const ctxObj = isObj(context) ? context : null;
1517
- const toolCallId = str(ctxObj?.toolCallId);
1518
- const state = isObj(ctxObj?.state) ? (ctxObj.state as JsonObject) : null;
1519
- const invalidate =
1520
- typeof ctxObj?.invalidate === "function"
1521
- ? (ctxObj.invalidate as () => void)
1522
- : null;
1523
- const isProgress =
1524
- isObj(result.details) &&
1525
- result.details.kind === "trellis-subagent-progress";
1526
- if (toolCallId && state && invalidate) {
1527
- const updatedAt = isProgress
1528
- ? (result.details as ProgressDetails).updatedAt
1529
- : Date.now();
1530
- rememberNativeCard(toolCallId, { state, invalidate, updatedAt });
1531
- }
1532
- return {
1533
- render(w: number) {
1534
- if (isProgress) {
1535
- const expanded = state?.localExpanded === true;
1536
- return renderProgressCard(
1537
- result.details as ProgressDetails,
1538
- expanded,
1539
- w,
1540
- );
1541
- }
1542
- return [trunc(result.content?.[0]?.text ?? "(no output)", w)];
1543
- },
1544
- invalidate() {},
1545
- };
1546
- },
1547
- });
1548
-
1549
- // Events
1550
- pi.on?.("session_start", (event, ctx) => {
1551
- getKey(event, ctx);
1552
- ctx?.ui?.notify?.(
1553
- "Trellis project context is available. Use /trellis-continue to resume the current task.",
1554
- "info",
1555
- );
1556
- });
1557
- pi.on?.("session_shutdown", () => {
1558
- nativeCards.clear();
1559
- activeSubagentToolCallId = null;
1560
- });
1561
- pi.on?.("tool_call", (event, ctx) => {
1562
- const k = getKey(event, ctx);
1563
- const ev = event as { toolName?: string; input?: JsonObject };
1564
- if (
1565
- ev.toolName === "bash" &&
1566
- isObj(ev.input) &&
1567
- typeof ev.input.command === "string" &&
1568
- !cmdHasTrellisCtx(ev.input.command)
1569
- )
1570
- ev.input.command = `export TRELLIS_CONTEXT_ID=${shellQuote(k)}; ${ev.input.command}`;
1571
- });
1572
- // Preserve progress details from execute(); mark failed subagent results through
1573
- // the official tool_result patch hook instead of throwing away renderer details.
1574
- pi.on?.("tool_result", (event) => {
1575
- const ev = event as { toolName?: string; details?: unknown };
1576
- if (
1577
- ev.toolName === "trellis_subagent" &&
1578
- isObj(ev.details) &&
1579
- ev.details.kind === "trellis-subagent-progress" &&
1580
- Array.isArray(ev.details.runs) &&
1581
- ev.details.runs.some(
1582
- (r) => isObj(r) && (r.status === "failed" || r.status === "cancelled"),
1583
- )
1584
- )
1585
- return { isError: true };
1586
- return undefined;
1587
- });
1588
- pi.on?.("before_agent_start", (event, ctx) => {
1589
- const k = getKey(event, ctx);
1590
- const cur = (event as { systemPrompt?: string }).systemPrompt ?? "";
1591
- const ctxText = buildContext(root, "trellis-implement", k);
1592
- const { wf, ov } = getTurnCtx(k);
1593
- return {
1594
- systemPrompt: [cur, ctxText, wf, ov].filter(Boolean).join("\n\n"),
1595
- };
1596
- });
1597
- pi.on?.("context", (event, ctx) => {
1598
- getKey(event, ctx);
1599
- });
1600
- }