@selesai/code 0.9.6 → 0.9.9

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 (281) hide show
  1. package/CHANGELOG.md +18 -0
  2. package/dist/cli/credential-print.js +4 -4
  3. package/dist/cli.js +0 -0
  4. package/dist/core/agent-session-runtime.js +1 -4
  5. package/dist/core/agent-session.d.ts +2 -0
  6. package/dist/core/agent-session.js +26 -0
  7. package/dist/core/compaction/branch-summarization.js +30 -30
  8. package/dist/core/compaction/compaction.js +81 -81
  9. package/dist/core/compaction/utils.js +2 -2
  10. package/dist/core/export-html/template.css +1066 -1066
  11. package/dist/core/export-html/template.html +55 -55
  12. package/dist/core/export-html/template.js +1864 -1864
  13. package/dist/core/export-html/vendor/highlight.min.js +1212 -1212
  14. package/dist/core/export-html/vendor/marked.min.js +78 -78
  15. package/dist/core/handoff.d.ts +11 -0
  16. package/dist/core/handoff.js +49 -0
  17. package/dist/core/messages.js +7 -7
  18. package/dist/extensions/context-compaction-reminder.test.ts +82 -82
  19. package/dist/extensions/context-compaction-reminder.ts +28 -28
  20. package/dist/extensions/handoff-new.ts +14 -75
  21. package/dist/extensions/pi-intercom/LICENSE +21 -21
  22. package/dist/extensions/pi-intercom/broker/client.test.ts +83 -83
  23. package/dist/extensions/pi-intercom/broker/extension.test.ts +387 -387
  24. package/dist/extensions/pi-intercom/broker/framing.test.ts +114 -114
  25. package/dist/extensions/pi-intercom/broker/paths.test.ts +153 -153
  26. package/dist/extensions/pi-intercom/broker/paths.ts +134 -134
  27. package/dist/extensions/pi-intercom/broker/runtime-claim.test.ts +34 -34
  28. package/dist/extensions/pi-intercom/broker/runtime-claim.ts +21 -21
  29. package/dist/extensions/pi-intercom/cwd.test.ts +40 -40
  30. package/dist/extensions/pi-intercom/cwd.ts +31 -31
  31. package/dist/extensions/pi-intercom/extension-api.ts +44 -44
  32. package/dist/extensions/pi-intercom/format-context.test.ts +31 -31
  33. package/dist/extensions/pi-intercom/format-context.ts +32 -32
  34. package/dist/extensions/pi-intercom/test/overlay-width.test.ts +66 -66
  35. package/dist/extensions/pi-intercom/ui/compose.ts +143 -143
  36. package/dist/extensions/pi-intercom/ui/session-list.ts +166 -166
  37. package/dist/extensions/pi-powerline-footer/bash-mode/shell-session.ts +286 -286
  38. package/dist/extensions/pi-powerline-footer/bash-mode/transcript.ts +108 -108
  39. package/dist/extensions/pi-powerline-footer/separators.ts +57 -57
  40. package/dist/extensions/pi-powerline-footer/tests/session-usage.test.ts +47 -47
  41. package/dist/extensions/pi-powerline-footer/tests/tps.test.ts +39 -39
  42. package/dist/extensions/pi-powerline-footer/theme.example.json +24 -24
  43. package/dist/extensions/pi-powerline-footer/theme.json +12 -12
  44. package/dist/extensions/pi-powerline-footer/tps.ts +345 -345
  45. package/dist/extensions/pi-rewind-hook/README.md +245 -245
  46. package/dist/extensions/pi-rewind-hook/index.ts +1445 -1445
  47. package/dist/extensions/pi-rewind-hook/package.json +29 -29
  48. package/dist/extensions/pi-subagents/install.mjs +0 -0
  49. package/dist/extensions/pi-web-agent/package.json +31 -31
  50. package/dist/extensions/pi-web-agent/src/backends/config.ts +205 -205
  51. package/dist/extensions/pi-web-agent/src/backends/doctor.ts +136 -136
  52. package/dist/extensions/pi-web-agent/src/backends/factory.ts +152 -152
  53. package/dist/extensions/pi-web-agent/src/backends/settings-reader.ts +25 -25
  54. package/dist/extensions/pi-web-agent/src/cache/ttl-cache.ts +28 -28
  55. package/dist/extensions/pi-web-agent/src/changelog-notice.ts +136 -136
  56. package/dist/extensions/pi-web-agent/src/commands/web-agent-config.ts +946 -946
  57. package/dist/extensions/pi-web-agent/src/extension.ts +126 -126
  58. package/dist/extensions/pi-web-agent/src/extract/readability.ts +118 -118
  59. package/dist/extensions/pi-web-agent/src/fetch/browser-resolution.ts +199 -199
  60. package/dist/extensions/pi-web-agent/src/fetch/firecrawl-fetch.ts +100 -100
  61. package/dist/extensions/pi-web-agent/src/fetch/headless-fetch.ts +117 -117
  62. package/dist/extensions/pi-web-agent/src/fetch/http-fetch.ts +67 -67
  63. package/dist/extensions/pi-web-agent/src/orchestration/answer-synthesizer.ts +60 -60
  64. package/dist/extensions/pi-web-agent/src/orchestration/candidate-selector.ts +50 -50
  65. package/dist/extensions/pi-web-agent/src/orchestration/direct-url.ts +52 -52
  66. package/dist/extensions/pi-web-agent/src/orchestration/evidence-quality.ts +105 -105
  67. package/dist/extensions/pi-web-agent/src/orchestration/evidence-ranker.ts +45 -45
  68. package/dist/extensions/pi-web-agent/src/orchestration/index.ts +28 -28
  69. package/dist/extensions/pi-web-agent/src/orchestration/query-planner.ts +47 -47
  70. package/dist/extensions/pi-web-agent/src/orchestration/research-orchestrator.ts +376 -376
  71. package/dist/extensions/pi-web-agent/src/orchestration/research-types.ts +64 -64
  72. package/dist/extensions/pi-web-agent/src/orchestration/research-worker.ts +181 -181
  73. package/dist/extensions/pi-web-agent/src/orchestration/source-profile.ts +101 -101
  74. package/dist/extensions/pi-web-agent/src/orchestration/stop-decider.ts +81 -81
  75. package/dist/extensions/pi-web-agent/src/presentation/config-store.ts +210 -210
  76. package/dist/extensions/pi-web-agent/src/presentation/config.ts +75 -75
  77. package/dist/extensions/pi-web-agent/src/presentation/explore-presentation.ts +61 -61
  78. package/dist/extensions/pi-web-agent/src/presentation/fetch-presentation.ts +54 -54
  79. package/dist/extensions/pi-web-agent/src/presentation/search-presentation.ts +41 -41
  80. package/dist/extensions/pi-web-agent/src/presentation/select-view.ts +20 -20
  81. package/dist/extensions/pi-web-agent/src/presentation/types.ts +63 -63
  82. package/dist/extensions/pi-web-agent/src/search/brave.ts +114 -114
  83. package/dist/extensions/pi-web-agent/src/search/duckduckgo.ts +72 -72
  84. package/dist/extensions/pi-web-agent/src/search/searxng.ts +96 -96
  85. package/dist/extensions/pi-web-agent/src/tools/web-explore.ts +71 -71
  86. package/dist/extensions/pi-web-agent/src/tools/web-fetch-headless.ts +31 -31
  87. package/dist/extensions/pi-web-agent/src/tools/web-fetch.ts +31 -31
  88. package/dist/extensions/pi-web-agent/src/tools/web-search.ts +157 -157
  89. package/dist/extensions/pi-web-agent/src/types.ts +88 -88
  90. package/dist/extensions/ponytail/package.json +8 -8
  91. package/dist/extensions/question/batch.ts +103 -103
  92. package/dist/extensions/question/constants.ts +30 -30
  93. package/dist/extensions/question/helpers.ts +58 -58
  94. package/dist/extensions/question/navigation.ts +14 -14
  95. package/dist/extensions/question/package.json +19 -19
  96. package/dist/extensions/question/schemas.ts +43 -43
  97. package/dist/extensions/question/selection-mode.ts +46 -46
  98. package/dist/extensions/question/shortcuts.ts +44 -44
  99. package/dist/extensions/question/types.ts +132 -132
  100. package/dist/extensions/test-resolve-hook-impl.mjs +6 -6
  101. package/dist/extensions/test-resolve-hook.mjs +6 -6
  102. package/dist/extensions/web-agent-onboarding.ts +222 -222
  103. package/dist/extensions/workflow/package.json +17 -17
  104. package/dist/modes/rpc/rpc-mode.js +24 -0
  105. package/dist/modes/rpc/rpc-types.d.ts +19 -0
  106. package/dist/package-manager-cli.js +71 -71
  107. package/dist/rpc-entry.js +0 -0
  108. package/dist/skills/agent-browser/SKILL.md +52 -52
  109. package/dist/skills/batch-grill-me/SKILL.md +19 -19
  110. package/dist/skills/grill-me/SKILL.md +10 -10
  111. package/dist/skills/handoff/SKILL.md +16 -16
  112. package/dist/skills/handoff-text/SKILL.md +14 -14
  113. package/dist/skills/implanger/SKILL.md +69 -69
  114. package/dist/skills/improve-codebase/REFERENCE.md +78 -78
  115. package/dist/skills/improve-codebase/SKILL.md +178 -178
  116. package/dist/skills/planger/SKILL.md +166 -166
  117. package/dist/skills/ponytail/SKILL.md +116 -116
  118. package/dist/skills/ponytail-audit/SKILL.md +42 -42
  119. package/dist/skills/ponytail-debt/SKILL.md +45 -45
  120. package/dist/skills/ponytail-gain/SKILL.md +51 -51
  121. package/dist/skills/ponytail-help/SKILL.md +70 -70
  122. package/dist/skills/ponytail-review/SKILL.md +58 -58
  123. package/dist/skills/selesai-handoff/SKILL.md +20 -20
  124. package/dist/skills/workflow-creation/SKILL.md +73 -73
  125. package/dist/themes/powerline-footer/theme.json +33 -33
  126. package/docs/compaction.md +396 -396
  127. package/docs/containerization.md +111 -111
  128. package/docs/development.md +71 -71
  129. package/docs/docs.json +164 -164
  130. package/docs/environment-variables.md +86 -86
  131. package/docs/index.md +83 -83
  132. package/docs/json.md +82 -82
  133. package/docs/models.md +502 -502
  134. package/docs/packages.md +227 -227
  135. package/docs/plans/subagent-delegation/phase-0-correctness.md +264 -264
  136. package/docs/plans/subagent-delegation/phase-1-behavioral-contract.md +485 -485
  137. package/docs/plans/subagent-delegation/phase-2-context-controls.md +281 -281
  138. package/docs/plans/subagent-delegation/phase-3-advisory-routing.md +361 -361
  139. package/docs/plans/subagent-delegation/phase-4-optional-enforcement.md +380 -380
  140. package/docs/prompt-templates.md +95 -95
  141. package/docs/providers.md +293 -293
  142. package/docs/sdk.md +1143 -1143
  143. package/docs/security.md +59 -59
  144. package/docs/session-format.md +414 -414
  145. package/docs/sessions.md +145 -145
  146. package/docs/shared-host-extensions.md +109 -109
  147. package/docs/shell-aliases.md +13 -13
  148. package/docs/skills.md +231 -231
  149. package/docs/terminal-setup.md +142 -142
  150. package/docs/termux.md +127 -127
  151. package/docs/themes.md +295 -295
  152. package/docs/tmux.md +63 -63
  153. package/docs/tui.md +927 -927
  154. package/docs/windows.md +17 -17
  155. package/examples/README.md +25 -25
  156. package/examples/extensions/README.md +211 -211
  157. package/examples/extensions/auto-commit-on-exit.ts +49 -49
  158. package/examples/extensions/bash-spawn-hook.ts +30 -30
  159. package/examples/extensions/bookmark.ts +50 -50
  160. package/examples/extensions/border-status-editor.ts +150 -150
  161. package/examples/extensions/built-in-tool-renderer.ts +249 -249
  162. package/examples/extensions/claude-rules.ts +86 -86
  163. package/examples/extensions/commands.ts +72 -72
  164. package/examples/extensions/confirm-destructive.ts +59 -59
  165. package/examples/extensions/custom-compaction.ts +130 -130
  166. package/examples/extensions/custom-footer.ts +64 -64
  167. package/examples/extensions/custom-header.ts +73 -73
  168. package/examples/extensions/custom-provider-anthropic/index.ts +610 -610
  169. package/examples/extensions/custom-provider-anthropic/package-lock.json +24 -24
  170. package/examples/extensions/custom-provider-anthropic/package.json +19 -19
  171. package/examples/extensions/custom-provider-gitlab-duo/index.ts +404 -404
  172. package/examples/extensions/custom-provider-gitlab-duo/package.json +16 -16
  173. package/examples/extensions/custom-provider-gitlab-duo/test.ts +82 -82
  174. package/examples/extensions/dirty-repo-guard.ts +56 -56
  175. package/examples/extensions/doom-overlay/README.md +46 -46
  176. package/examples/extensions/doom-overlay/doom/build/doom.js +21 -21
  177. package/examples/extensions/doom-overlay/doom/build/doom.wasm +0 -0
  178. package/examples/extensions/doom-overlay/doom/build.sh +152 -152
  179. package/examples/extensions/doom-overlay/doom/doomgeneric_pi.c +72 -72
  180. package/examples/extensions/doom-overlay/doom-component.ts +132 -132
  181. package/examples/extensions/doom-overlay/doom-engine.ts +173 -173
  182. package/examples/extensions/doom-overlay/doom-keys.ts +104 -104
  183. package/examples/extensions/doom-overlay/index.ts +74 -74
  184. package/examples/extensions/doom-overlay/wad-finder.ts +51 -51
  185. package/examples/extensions/dynamic-resources/SKILL.md +8 -8
  186. package/examples/extensions/dynamic-resources/dynamic.json +79 -79
  187. package/examples/extensions/dynamic-resources/dynamic.md +5 -5
  188. package/examples/extensions/dynamic-resources/index.ts +15 -15
  189. package/examples/extensions/dynamic-tools.ts +74 -74
  190. package/examples/extensions/event-bus.ts +43 -43
  191. package/examples/extensions/file-trigger.ts +41 -41
  192. package/examples/extensions/git-checkpoint.ts +53 -53
  193. package/examples/extensions/git-merge-and-resolve.ts +115 -115
  194. package/examples/extensions/github-issue-autocomplete.ts +185 -185
  195. package/examples/extensions/gondolin/index.ts +531 -531
  196. package/examples/extensions/gondolin/package-lock.json +185 -185
  197. package/examples/extensions/gondolin/package.json +19 -19
  198. package/examples/extensions/handoff.ts +199 -199
  199. package/examples/extensions/hello.ts +26 -26
  200. package/examples/extensions/hidden-thinking-label.ts +53 -53
  201. package/examples/extensions/inline-bash.ts +94 -94
  202. package/examples/extensions/input-transform-streaming.ts +39 -39
  203. package/examples/extensions/input-transform.ts +43 -43
  204. package/examples/extensions/interactive-shell.ts +196 -196
  205. package/examples/extensions/mac-system-theme.ts +47 -47
  206. package/examples/extensions/message-renderer.ts +59 -59
  207. package/examples/extensions/minimal-mode.ts +426 -426
  208. package/examples/extensions/modal-editor.ts +85 -85
  209. package/examples/extensions/model-status.ts +31 -31
  210. package/examples/extensions/notify.ts +55 -55
  211. package/examples/extensions/overlay-qa-tests.ts +1450 -1450
  212. package/examples/extensions/overlay-test.ts +153 -153
  213. package/examples/extensions/permission-gate.ts +34 -34
  214. package/examples/extensions/pirate.ts +47 -47
  215. package/examples/extensions/plan-mode/README.md +66 -66
  216. package/examples/extensions/plan-mode/index.ts +390 -390
  217. package/examples/extensions/plan-mode/utils.ts +168 -168
  218. package/examples/extensions/preset.ts +436 -436
  219. package/examples/extensions/project-trust.ts +64 -64
  220. package/examples/extensions/prompt-customizer.ts +97 -97
  221. package/examples/extensions/protected-paths.ts +30 -30
  222. package/examples/extensions/provider-payload.ts +18 -18
  223. package/examples/extensions/qna.ts +122 -122
  224. package/examples/extensions/question.ts +285 -285
  225. package/examples/extensions/questionnaire.ts +448 -448
  226. package/examples/extensions/rainbow-editor.ts +88 -88
  227. package/examples/extensions/reload-runtime.ts +37 -37
  228. package/examples/extensions/rpc-demo.ts +118 -118
  229. package/examples/extensions/sandbox/index.ts +321 -321
  230. package/examples/extensions/sandbox/package-lock.json +92 -92
  231. package/examples/extensions/sandbox/package.json +19 -19
  232. package/examples/extensions/send-user-message.ts +97 -97
  233. package/examples/extensions/session-name.ts +27 -27
  234. package/examples/extensions/shutdown-command.ts +63 -63
  235. package/examples/extensions/snake.ts +343 -343
  236. package/examples/extensions/space-invaders.ts +560 -560
  237. package/examples/extensions/ssh.ts +220 -220
  238. package/examples/extensions/status-line.ts +32 -32
  239. package/examples/extensions/structured-output.ts +65 -65
  240. package/examples/extensions/subagent/README.md +175 -175
  241. package/examples/extensions/subagent/agents/planner.md +37 -37
  242. package/examples/extensions/subagent/agents/reviewer.md +35 -35
  243. package/examples/extensions/subagent/agents/scout.md +50 -50
  244. package/examples/extensions/subagent/agents/worker.md +24 -24
  245. package/examples/extensions/subagent/agents.ts +126 -126
  246. package/examples/extensions/subagent/index.ts +1015 -1015
  247. package/examples/extensions/subagent/prompts/implement-and-review.md +10 -10
  248. package/examples/extensions/subagent/prompts/implement.md +10 -10
  249. package/examples/extensions/subagent/prompts/scout-and-plan.md +9 -9
  250. package/examples/extensions/summarize.ts +209 -209
  251. package/examples/extensions/system-prompt-header.ts +17 -17
  252. package/examples/extensions/tic-tac-toe.ts +1008 -1008
  253. package/examples/extensions/timed-confirm.ts +70 -70
  254. package/examples/extensions/titlebar-spinner.ts +58 -58
  255. package/examples/extensions/todo.ts +297 -297
  256. package/examples/extensions/tool-override.ts +144 -144
  257. package/examples/extensions/tools.ts +146 -146
  258. package/examples/extensions/trigger-compact.ts +50 -50
  259. package/examples/extensions/truncated-tool.ts +195 -195
  260. package/examples/extensions/widget-placement.ts +9 -9
  261. package/examples/extensions/with-deps/index.ts +32 -32
  262. package/examples/extensions/with-deps/package-lock.json +31 -31
  263. package/examples/extensions/with-deps/package.json +22 -22
  264. package/examples/extensions/working-indicator.ts +123 -123
  265. package/examples/extensions/working-message-test.ts +25 -25
  266. package/examples/rpc-extension-ui.ts +632 -632
  267. package/examples/sdk/01-minimal.ts +26 -26
  268. package/examples/sdk/02-custom-model.ts +53 -53
  269. package/examples/sdk/03-custom-prompt.ts +75 -75
  270. package/examples/sdk/04-skills.ts +55 -55
  271. package/examples/sdk/05-tools.ts +48 -48
  272. package/examples/sdk/06-extensions.ts +99 -99
  273. package/examples/sdk/07-context-files.ts +47 -47
  274. package/examples/sdk/08-prompt-templates.ts +51 -51
  275. package/examples/sdk/09-api-keys-and-oauth.ts +52 -52
  276. package/examples/sdk/10-settings.ts +53 -53
  277. package/examples/sdk/11-sessions.ts +52 -52
  278. package/examples/sdk/12-full-control.ts +79 -79
  279. package/examples/sdk/13-session-runtime.ts +67 -67
  280. package/examples/sdk/README.md +144 -144
  281. package/package.json +1 -1
@@ -1,50 +1,50 @@
1
- import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
2
-
3
- const COMPACT_THRESHOLD_TOKENS = 100_000;
4
-
5
- export default function (pi: ExtensionAPI) {
6
- let previousTokens: number | null | undefined;
7
-
8
- const triggerCompaction = (ctx: ExtensionContext, customInstructions?: string) => {
9
- if (ctx.hasUI) {
10
- ctx.ui.notify("Compaction started", "info");
11
- }
12
- ctx.compact({
13
- customInstructions,
14
- onComplete: () => {
15
- if (ctx.hasUI) {
16
- ctx.ui.notify("Compaction completed", "info");
17
- }
18
- },
19
- onError: (error) => {
20
- if (ctx.hasUI) {
21
- ctx.ui.notify(`Compaction failed: ${error.message}`, "error");
22
- }
23
- },
24
- });
25
- };
26
-
27
- pi.on("turn_end", (_event, ctx) => {
28
- const usage = ctx.getContextUsage();
29
- const currentTokens = usage?.tokens ?? null;
30
- if (currentTokens === null) {
31
- return;
32
- }
33
-
34
- const crossedThreshold =
35
- previousTokens !== undefined && previousTokens !== null && previousTokens <= COMPACT_THRESHOLD_TOKENS;
36
- previousTokens = currentTokens;
37
- if (!crossedThreshold || currentTokens <= COMPACT_THRESHOLD_TOKENS) {
38
- return;
39
- }
40
- triggerCompaction(ctx);
41
- });
42
-
43
- pi.registerCommand("trigger-compact", {
44
- description: "Trigger compaction immediately",
45
- handler: async (args, ctx) => {
46
- const instructions = args.trim() || undefined;
47
- triggerCompaction(ctx, instructions);
48
- },
49
- });
50
- }
1
+ import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
2
+
3
+ const COMPACT_THRESHOLD_TOKENS = 100_000;
4
+
5
+ export default function (pi: ExtensionAPI) {
6
+ let previousTokens: number | null | undefined;
7
+
8
+ const triggerCompaction = (ctx: ExtensionContext, customInstructions?: string) => {
9
+ if (ctx.hasUI) {
10
+ ctx.ui.notify("Compaction started", "info");
11
+ }
12
+ ctx.compact({
13
+ customInstructions,
14
+ onComplete: () => {
15
+ if (ctx.hasUI) {
16
+ ctx.ui.notify("Compaction completed", "info");
17
+ }
18
+ },
19
+ onError: (error) => {
20
+ if (ctx.hasUI) {
21
+ ctx.ui.notify(`Compaction failed: ${error.message}`, "error");
22
+ }
23
+ },
24
+ });
25
+ };
26
+
27
+ pi.on("turn_end", (_event, ctx) => {
28
+ const usage = ctx.getContextUsage();
29
+ const currentTokens = usage?.tokens ?? null;
30
+ if (currentTokens === null) {
31
+ return;
32
+ }
33
+
34
+ const crossedThreshold =
35
+ previousTokens !== undefined && previousTokens !== null && previousTokens <= COMPACT_THRESHOLD_TOKENS;
36
+ previousTokens = currentTokens;
37
+ if (!crossedThreshold || currentTokens <= COMPACT_THRESHOLD_TOKENS) {
38
+ return;
39
+ }
40
+ triggerCompaction(ctx);
41
+ });
42
+
43
+ pi.registerCommand("trigger-compact", {
44
+ description: "Trigger compaction immediately",
45
+ handler: async (args, ctx) => {
46
+ const instructions = args.trim() || undefined;
47
+ triggerCompaction(ctx, instructions);
48
+ },
49
+ });
50
+ }
@@ -1,195 +1,195 @@
1
- /**
2
- * Truncated Tool Example - Demonstrates proper output truncation for custom tools
3
- *
4
- * Custom tools MUST truncate their output to avoid overwhelming the LLM context.
5
- * The built-in limit is 50KB (~10k tokens) and 2000 lines, whichever is hit first.
6
- *
7
- * This example shows how to:
8
- * 1. Use the built-in truncation utilities
9
- * 2. Write full output to a temp file when truncated
10
- * 3. Inform the LLM where to find the complete output
11
- * 4. Custom rendering of tool calls and results
12
- *
13
- * The `rg` tool here wraps ripgrep with proper truncation. Compare this to the
14
- * built-in `grep` tool in src/core/tools/grep.ts for a more complete implementation.
15
- */
16
-
17
- import { mkdtemp, writeFile } from "node:fs/promises";
18
- import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
19
- import {
20
- DEFAULT_MAX_BYTES,
21
- DEFAULT_MAX_LINES,
22
- formatSize,
23
- type TruncationResult,
24
- truncateHead,
25
- withFileMutationQueue,
26
- } from "@earendil-works/pi-coding-agent";
27
- import { Text } from "@earendil-works/pi-tui";
28
- import { execSync } from "child_process";
29
- import { tmpdir } from "os";
30
- import { join } from "path";
31
- import { Type } from "typebox";
32
-
33
- const RgParams = Type.Object({
34
- pattern: Type.String({ description: "Search pattern (regex)" }),
35
- path: Type.Optional(Type.String({ description: "Directory to search (default: current directory)" })),
36
- glob: Type.Optional(Type.String({ description: "File glob pattern, e.g. '*.ts'" })),
37
- });
38
-
39
- interface RgDetails {
40
- pattern: string;
41
- path?: string;
42
- glob?: string;
43
- matchCount: number;
44
- truncation?: TruncationResult;
45
- fullOutputPath?: string;
46
- }
47
-
48
- export default function (pi: ExtensionAPI) {
49
- pi.registerTool({
50
- name: "rg",
51
- label: "ripgrep",
52
- // Document the truncation limits in the tool description so the LLM knows
53
- description: `Search file contents using ripgrep. Output is truncated to ${DEFAULT_MAX_LINES} lines or ${formatSize(DEFAULT_MAX_BYTES)} (whichever is hit first). If truncated, full output is saved to a temp file.`,
54
- parameters: RgParams,
55
-
56
- async execute(_toolCallId, params, _signal, _onUpdate, ctx) {
57
- const { pattern, path: searchPath, glob } = params;
58
-
59
- // Build the ripgrep command
60
- const args = ["rg", "--line-number", "--color=never"];
61
- if (glob) args.push("--glob", glob);
62
- args.push(pattern);
63
- args.push(searchPath || ".");
64
-
65
- let output: string;
66
- try {
67
- output = execSync(args.join(" "), {
68
- cwd: ctx.cwd,
69
- encoding: "utf-8",
70
- maxBuffer: 100 * 1024 * 1024, // 100MB buffer to capture full output
71
- });
72
- } catch (err: any) {
73
- // ripgrep exits with 1 when no matches found
74
- if (err.status === 1) {
75
- return {
76
- content: [{ type: "text", text: "No matches found" }],
77
- details: { pattern, path: searchPath, glob, matchCount: 0 } as RgDetails,
78
- };
79
- }
80
- throw new Error(`ripgrep failed: ${err.message}`);
81
- }
82
-
83
- if (!output.trim()) {
84
- return {
85
- content: [{ type: "text", text: "No matches found" }],
86
- details: { pattern, path: searchPath, glob, matchCount: 0 } as RgDetails,
87
- };
88
- }
89
-
90
- // Apply truncation using built-in utilities
91
- // truncateHead keeps the first N lines/bytes (good for search results)
92
- // truncateTail keeps the last N lines/bytes (good for logs/command output)
93
- const truncation = truncateHead(output, {
94
- maxLines: DEFAULT_MAX_LINES,
95
- maxBytes: DEFAULT_MAX_BYTES,
96
- });
97
-
98
- // Count matches (each non-empty line with a match)
99
- const matchCount = output.split("\n").filter((line) => line.trim()).length;
100
-
101
- const details: RgDetails = {
102
- pattern,
103
- path: searchPath,
104
- glob,
105
- matchCount,
106
- };
107
-
108
- let resultText = truncation.content;
109
-
110
- if (truncation.truncated) {
111
- // Save full output to a temp file so LLM can access it if needed
112
- const tempDir = await mkdtemp(join(tmpdir(), "pi-rg-"));
113
- const tempFile = join(tempDir, "output.txt");
114
- await withFileMutationQueue(tempFile, async () => {
115
- await writeFile(tempFile, output, "utf8");
116
- });
117
-
118
- details.truncation = truncation;
119
- details.fullOutputPath = tempFile;
120
-
121
- // Add truncation notice - this helps the LLM understand the output is incomplete
122
- const truncatedLines = truncation.totalLines - truncation.outputLines;
123
- const truncatedBytes = truncation.totalBytes - truncation.outputBytes;
124
-
125
- resultText += `\n\n[Output truncated: showing ${truncation.outputLines} of ${truncation.totalLines} lines`;
126
- resultText += ` (${formatSize(truncation.outputBytes)} of ${formatSize(truncation.totalBytes)}).`;
127
- resultText += ` ${truncatedLines} lines (${formatSize(truncatedBytes)}) omitted.`;
128
- resultText += ` Full output saved to: ${tempFile}]`;
129
- }
130
-
131
- return {
132
- content: [{ type: "text", text: resultText }],
133
- details,
134
- };
135
- },
136
-
137
- // Custom rendering of the tool call (shown before/during execution)
138
- renderCall(args, theme, _context) {
139
- let text = theme.fg("toolTitle", theme.bold("rg "));
140
- text += theme.fg("accent", `"${args.pattern}"`);
141
- if (args.path) {
142
- text += theme.fg("muted", ` in ${args.path}`);
143
- }
144
- if (args.glob) {
145
- text += theme.fg("dim", ` --glob ${args.glob}`);
146
- }
147
- return new Text(text, 0, 0);
148
- },
149
-
150
- // Custom rendering of the tool result
151
- renderResult(result, { expanded, isPartial }, theme, _context) {
152
- const details = result.details as RgDetails | undefined;
153
-
154
- // Handle streaming/partial results
155
- if (isPartial) {
156
- return new Text(theme.fg("warning", "Searching..."), 0, 0);
157
- }
158
-
159
- // No matches
160
- if (!details || details.matchCount === 0) {
161
- return new Text(theme.fg("dim", "No matches found"), 0, 0);
162
- }
163
-
164
- // Build result display
165
- let text = theme.fg("success", `${details.matchCount} matches`);
166
-
167
- // Show truncation warning if applicable
168
- if (details.truncation?.truncated) {
169
- text += theme.fg("warning", " (truncated)");
170
- }
171
-
172
- // In expanded view, show the actual matches
173
- if (expanded) {
174
- const content = result.content[0];
175
- if (content?.type === "text") {
176
- // Show first 20 lines in expanded view, or all if fewer
177
- const lines = content.text.split("\n").slice(0, 20);
178
- for (const line of lines) {
179
- text += `\n${theme.fg("dim", line)}`;
180
- }
181
- if (content.text.split("\n").length > 20) {
182
- text += `\n${theme.fg("muted", "... (use read tool to see full output)")}`;
183
- }
184
- }
185
-
186
- // Show temp file path if truncated
187
- if (details.fullOutputPath) {
188
- text += `\n${theme.fg("dim", `Full output: ${details.fullOutputPath}`)}`;
189
- }
190
- }
191
-
192
- return new Text(text, 0, 0);
193
- },
194
- });
195
- }
1
+ /**
2
+ * Truncated Tool Example - Demonstrates proper output truncation for custom tools
3
+ *
4
+ * Custom tools MUST truncate their output to avoid overwhelming the LLM context.
5
+ * The built-in limit is 50KB (~10k tokens) and 2000 lines, whichever is hit first.
6
+ *
7
+ * This example shows how to:
8
+ * 1. Use the built-in truncation utilities
9
+ * 2. Write full output to a temp file when truncated
10
+ * 3. Inform the LLM where to find the complete output
11
+ * 4. Custom rendering of tool calls and results
12
+ *
13
+ * The `rg` tool here wraps ripgrep with proper truncation. Compare this to the
14
+ * built-in `grep` tool in src/core/tools/grep.ts for a more complete implementation.
15
+ */
16
+
17
+ import { mkdtemp, writeFile } from "node:fs/promises";
18
+ import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
19
+ import {
20
+ DEFAULT_MAX_BYTES,
21
+ DEFAULT_MAX_LINES,
22
+ formatSize,
23
+ type TruncationResult,
24
+ truncateHead,
25
+ withFileMutationQueue,
26
+ } from "@earendil-works/pi-coding-agent";
27
+ import { Text } from "@earendil-works/pi-tui";
28
+ import { execSync } from "child_process";
29
+ import { tmpdir } from "os";
30
+ import { join } from "path";
31
+ import { Type } from "typebox";
32
+
33
+ const RgParams = Type.Object({
34
+ pattern: Type.String({ description: "Search pattern (regex)" }),
35
+ path: Type.Optional(Type.String({ description: "Directory to search (default: current directory)" })),
36
+ glob: Type.Optional(Type.String({ description: "File glob pattern, e.g. '*.ts'" })),
37
+ });
38
+
39
+ interface RgDetails {
40
+ pattern: string;
41
+ path?: string;
42
+ glob?: string;
43
+ matchCount: number;
44
+ truncation?: TruncationResult;
45
+ fullOutputPath?: string;
46
+ }
47
+
48
+ export default function (pi: ExtensionAPI) {
49
+ pi.registerTool({
50
+ name: "rg",
51
+ label: "ripgrep",
52
+ // Document the truncation limits in the tool description so the LLM knows
53
+ description: `Search file contents using ripgrep. Output is truncated to ${DEFAULT_MAX_LINES} lines or ${formatSize(DEFAULT_MAX_BYTES)} (whichever is hit first). If truncated, full output is saved to a temp file.`,
54
+ parameters: RgParams,
55
+
56
+ async execute(_toolCallId, params, _signal, _onUpdate, ctx) {
57
+ const { pattern, path: searchPath, glob } = params;
58
+
59
+ // Build the ripgrep command
60
+ const args = ["rg", "--line-number", "--color=never"];
61
+ if (glob) args.push("--glob", glob);
62
+ args.push(pattern);
63
+ args.push(searchPath || ".");
64
+
65
+ let output: string;
66
+ try {
67
+ output = execSync(args.join(" "), {
68
+ cwd: ctx.cwd,
69
+ encoding: "utf-8",
70
+ maxBuffer: 100 * 1024 * 1024, // 100MB buffer to capture full output
71
+ });
72
+ } catch (err: any) {
73
+ // ripgrep exits with 1 when no matches found
74
+ if (err.status === 1) {
75
+ return {
76
+ content: [{ type: "text", text: "No matches found" }],
77
+ details: { pattern, path: searchPath, glob, matchCount: 0 } as RgDetails,
78
+ };
79
+ }
80
+ throw new Error(`ripgrep failed: ${err.message}`);
81
+ }
82
+
83
+ if (!output.trim()) {
84
+ return {
85
+ content: [{ type: "text", text: "No matches found" }],
86
+ details: { pattern, path: searchPath, glob, matchCount: 0 } as RgDetails,
87
+ };
88
+ }
89
+
90
+ // Apply truncation using built-in utilities
91
+ // truncateHead keeps the first N lines/bytes (good for search results)
92
+ // truncateTail keeps the last N lines/bytes (good for logs/command output)
93
+ const truncation = truncateHead(output, {
94
+ maxLines: DEFAULT_MAX_LINES,
95
+ maxBytes: DEFAULT_MAX_BYTES,
96
+ });
97
+
98
+ // Count matches (each non-empty line with a match)
99
+ const matchCount = output.split("\n").filter((line) => line.trim()).length;
100
+
101
+ const details: RgDetails = {
102
+ pattern,
103
+ path: searchPath,
104
+ glob,
105
+ matchCount,
106
+ };
107
+
108
+ let resultText = truncation.content;
109
+
110
+ if (truncation.truncated) {
111
+ // Save full output to a temp file so LLM can access it if needed
112
+ const tempDir = await mkdtemp(join(tmpdir(), "pi-rg-"));
113
+ const tempFile = join(tempDir, "output.txt");
114
+ await withFileMutationQueue(tempFile, async () => {
115
+ await writeFile(tempFile, output, "utf8");
116
+ });
117
+
118
+ details.truncation = truncation;
119
+ details.fullOutputPath = tempFile;
120
+
121
+ // Add truncation notice - this helps the LLM understand the output is incomplete
122
+ const truncatedLines = truncation.totalLines - truncation.outputLines;
123
+ const truncatedBytes = truncation.totalBytes - truncation.outputBytes;
124
+
125
+ resultText += `\n\n[Output truncated: showing ${truncation.outputLines} of ${truncation.totalLines} lines`;
126
+ resultText += ` (${formatSize(truncation.outputBytes)} of ${formatSize(truncation.totalBytes)}).`;
127
+ resultText += ` ${truncatedLines} lines (${formatSize(truncatedBytes)}) omitted.`;
128
+ resultText += ` Full output saved to: ${tempFile}]`;
129
+ }
130
+
131
+ return {
132
+ content: [{ type: "text", text: resultText }],
133
+ details,
134
+ };
135
+ },
136
+
137
+ // Custom rendering of the tool call (shown before/during execution)
138
+ renderCall(args, theme, _context) {
139
+ let text = theme.fg("toolTitle", theme.bold("rg "));
140
+ text += theme.fg("accent", `"${args.pattern}"`);
141
+ if (args.path) {
142
+ text += theme.fg("muted", ` in ${args.path}`);
143
+ }
144
+ if (args.glob) {
145
+ text += theme.fg("dim", ` --glob ${args.glob}`);
146
+ }
147
+ return new Text(text, 0, 0);
148
+ },
149
+
150
+ // Custom rendering of the tool result
151
+ renderResult(result, { expanded, isPartial }, theme, _context) {
152
+ const details = result.details as RgDetails | undefined;
153
+
154
+ // Handle streaming/partial results
155
+ if (isPartial) {
156
+ return new Text(theme.fg("warning", "Searching..."), 0, 0);
157
+ }
158
+
159
+ // No matches
160
+ if (!details || details.matchCount === 0) {
161
+ return new Text(theme.fg("dim", "No matches found"), 0, 0);
162
+ }
163
+
164
+ // Build result display
165
+ let text = theme.fg("success", `${details.matchCount} matches`);
166
+
167
+ // Show truncation warning if applicable
168
+ if (details.truncation?.truncated) {
169
+ text += theme.fg("warning", " (truncated)");
170
+ }
171
+
172
+ // In expanded view, show the actual matches
173
+ if (expanded) {
174
+ const content = result.content[0];
175
+ if (content?.type === "text") {
176
+ // Show first 20 lines in expanded view, or all if fewer
177
+ const lines = content.text.split("\n").slice(0, 20);
178
+ for (const line of lines) {
179
+ text += `\n${theme.fg("dim", line)}`;
180
+ }
181
+ if (content.text.split("\n").length > 20) {
182
+ text += `\n${theme.fg("muted", "... (use read tool to see full output)")}`;
183
+ }
184
+ }
185
+
186
+ // Show temp file path if truncated
187
+ if (details.fullOutputPath) {
188
+ text += `\n${theme.fg("dim", `Full output: ${details.fullOutputPath}`)}`;
189
+ }
190
+ }
191
+
192
+ return new Text(text, 0, 0);
193
+ },
194
+ });
195
+ }
@@ -1,9 +1,9 @@
1
- import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
2
-
3
- export default function widgetPlacementExtension(pi: ExtensionAPI) {
4
- pi.on("session_start", (_event, ctx) => {
5
- if (!ctx.hasUI) return;
6
- ctx.ui.setWidget("widget-above", ["Above editor widget"]);
7
- ctx.ui.setWidget("widget-below", ["Below editor widget"], { placement: "belowEditor" });
8
- });
9
- }
1
+ import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
2
+
3
+ export default function widgetPlacementExtension(pi: ExtensionAPI) {
4
+ pi.on("session_start", (_event, ctx) => {
5
+ if (!ctx.hasUI) return;
6
+ ctx.ui.setWidget("widget-above", ["Above editor widget"]);
7
+ ctx.ui.setWidget("widget-below", ["Below editor widget"], { placement: "belowEditor" });
8
+ });
9
+ }
@@ -1,32 +1,32 @@
1
- /**
2
- * Example extension with its own npm dependencies.
3
- * Tests that jiti resolves modules from the extension's own node_modules.
4
- *
5
- * Requires: npm install in this directory
6
- */
7
-
8
- import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
9
- import ms from "ms";
10
- import { Type } from "typebox";
11
-
12
- export default function (pi: ExtensionAPI) {
13
- // Register a tool that uses ms
14
- pi.registerTool({
15
- name: "parse_duration",
16
- label: "Parse Duration",
17
- description: "Parse a human-readable duration string (e.g., '2 days', '1h', '5m') to milliseconds",
18
- parameters: Type.Object({
19
- duration: Type.String({ description: "Duration string like '2 days', '1h', '5m'" }),
20
- }),
21
- execute: async (_toolCallId, params) => {
22
- const result = ms(params.duration as ms.StringValue);
23
- if (result === undefined) {
24
- throw new Error(`Invalid duration: "${params.duration}"`);
25
- }
26
- return {
27
- content: [{ type: "text", text: `${params.duration} = ${result} milliseconds` }],
28
- details: {},
29
- };
30
- },
31
- });
32
- }
1
+ /**
2
+ * Example extension with its own npm dependencies.
3
+ * Tests that jiti resolves modules from the extension's own node_modules.
4
+ *
5
+ * Requires: npm install in this directory
6
+ */
7
+
8
+ import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
9
+ import ms from "ms";
10
+ import { Type } from "typebox";
11
+
12
+ export default function (pi: ExtensionAPI) {
13
+ // Register a tool that uses ms
14
+ pi.registerTool({
15
+ name: "parse_duration",
16
+ label: "Parse Duration",
17
+ description: "Parse a human-readable duration string (e.g., '2 days', '1h', '5m') to milliseconds",
18
+ parameters: Type.Object({
19
+ duration: Type.String({ description: "Duration string like '2 days', '1h', '5m'" }),
20
+ }),
21
+ execute: async (_toolCallId, params) => {
22
+ const result = ms(params.duration as ms.StringValue);
23
+ if (result === undefined) {
24
+ throw new Error(`Invalid duration: "${params.duration}"`);
25
+ }
26
+ return {
27
+ content: [{ type: "text", text: `${params.duration} = ${result} milliseconds` }],
28
+ details: {},
29
+ };
30
+ },
31
+ });
32
+ }
@@ -1,31 +1,31 @@
1
- {
2
- "name": "pi-extension-with-deps",
3
- "version": "0.83.0",
4
- "lockfileVersion": 3,
5
- "requires": true,
6
- "packages": {
7
- "": {
8
- "name": "pi-extension-with-deps",
9
- "version": "0.83.0",
10
- "dependencies": {
11
- "ms": "^2.1.3"
12
- },
13
- "devDependencies": {
14
- "@types/ms": "^2.1.0"
15
- }
16
- },
17
- "node_modules/@types/ms": {
18
- "version": "2.1.0",
19
- "resolved": "https://registry.npmjs.org/@types/ms/-/ms-2.1.0.tgz",
20
- "integrity": "sha512-GsCCIZDE/p3i96vtEqx+7dBUGXrc7zeSK3wwPHIaRThS+9OhWIXRqzs4d6k1SVU8g91DrNRWxWUGhp5KXQb2VA==",
21
- "dev": true,
22
- "license": "MIT"
23
- },
24
- "node_modules/ms": {
25
- "version": "2.1.3",
26
- "resolved": "https://registry.npmjs.org/ms/-/ms-2.1.3.tgz",
27
- "integrity": "sha512-6FlzubTLZG3J2a/NVCAleEhjzq5oxgHyaCU9yYXvcLsvoVaHJq/s5xXI6/XXP6tz7R9xAOtHnSO/tXtF3WRTlA==",
28
- "license": "MIT"
29
- }
30
- }
31
- }
1
+ {
2
+ "name": "pi-extension-with-deps",
3
+ "version": "0.83.0",
4
+ "lockfileVersion": 3,
5
+ "requires": true,
6
+ "packages": {
7
+ "": {
8
+ "name": "pi-extension-with-deps",
9
+ "version": "0.83.0",
10
+ "dependencies": {
11
+ "ms": "^2.1.3"
12
+ },
13
+ "devDependencies": {
14
+ "@types/ms": "^2.1.0"
15
+ }
16
+ },
17
+ "node_modules/@types/ms": {
18
+ "version": "2.1.0",
19
+ "resolved": "https://registry.npmjs.org/@types/ms/-/ms-2.1.0.tgz",
20
+ "integrity": "sha512-GsCCIZDE/p3i96vtEqx+7dBUGXrc7zeSK3wwPHIaRThS+9OhWIXRqzs4d6k1SVU8g91DrNRWxWUGhp5KXQb2VA==",
21
+ "dev": true,
22
+ "license": "MIT"
23
+ },
24
+ "node_modules/ms": {
25
+ "version": "2.1.3",
26
+ "resolved": "https://registry.npmjs.org/ms/-/ms-2.1.3.tgz",
27
+ "integrity": "sha512-6FlzubTLZG3J2a/NVCAleEhjzq5oxgHyaCU9yYXvcLsvoVaHJq/s5xXI6/XXP6tz7R9xAOtHnSO/tXtF3WRTlA==",
28
+ "license": "MIT"
29
+ }
30
+ }
31
+ }