pi-tool-duration 0.1.0 → 0.1.2

package/CHANGELOG.md

@@ -1,5 +1,20 @@
 # Changelog
 
+## Unreleased
+
+## [0.1.2] - 2026-07-11
+
+### Changed
+
+- Measure from Pi's `tool_execution_start` event so timing matches the full current execution lifecycle, including preflight hooks.
+- Clear abandoned timing state when the agent fully settles.
+
+## [0.1.1] - 2026-07-03
+
+### Changed
+
+- Always append duration markers for tool results that report a non-zero exit code, even below the normal duration threshold.
+
 ## [0.1.0] - 2026-06-24
 
 ### Added

package/README.md

@@ -13,7 +13,7 @@ Pi already shows tool timing in the TUI (`Took Xs`), but that timing is UI-only.
 
 ## How it works
 
-The extension matches Pi `tool_call` and `tool_result` events by `toolCallId`. When elapsed time is at or above the configured threshold, it appends one text block:
+The extension matches Pi `tool_execution_start` and `tool_result` events by `toolCallId`. When elapsed time is at or above the configured threshold, or a result reports a non-zero exit code, it appends one text block:
 
 ```text
 [duration: 5.0s]
@@ -65,7 +65,7 @@ hi
 [duration: 5.0s]
 ```
 
-A fast command below the threshold stays unchanged.
+A fast successful command below the threshold stays unchanged. A non-zero exit code is always annotated, even below the threshold.
 
 ## License
 

package/extensions/tool-duration/index.ts

@@ -30,6 +30,19 @@ function alreadyAnnotated(content: Array<{ type: string; text?: string }>): bool
   return last?.type === "text" && typeof last.text === "string" && DURATION_RE.test(last.text);
 }
 
+function nonZeroExitCode(event: { content: Array<{ type: string; text?: string }>; details?: unknown }): boolean {
+  const details = event.details as Record<string, unknown> | null;
+  const code = details && Number(details.exitCode ?? details.code ?? details.status);
+  if (code !== null && Number.isFinite(code)) return code !== 0;
+
+  return event.content.some(
+    (item) =>
+      item.type === "text" &&
+      typeof item.text === "string" &&
+      /(?:exited with code|exit code)\s+(-?[1-9]\d*)\b/i.test(item.text),
+  );
+}
+
 export default function (pi: ExtensionAPI) {
   const starts = new Map<string, number>();
 
@@ -38,7 +51,7 @@ export default function (pi: ExtensionAPI) {
     type: "string",
   });
 
-  pi.on("tool_call", async (event) => {
+  pi.on("tool_execution_start", async (event) => {
     starts.set(event.toolCallId, performance.now());
   });
 
@@ -48,7 +61,8 @@ export default function (pi: ExtensionAPI) {
     if (startedAt === undefined) return;
 
     const ms = performance.now() - startedAt;
-    if (ms < thresholdMs(pi) || alreadyAnnotated(event.content)) return;
+    if (!nonZeroExitCode(event) && ms < thresholdMs(pi)) return;
+    if (alreadyAnnotated(event.content)) return;
 
     return {
       content: [
@@ -62,4 +76,5 @@ export default function (pi: ExtensionAPI) {
   pi.on("session_start", clearStarts);
   pi.on("session_shutdown", clearStarts);
   pi.on("agent_end", clearStarts);
+  pi.on("agent_settled", clearStarts);
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-tool-duration",
-  "version": "0.1.0",
+  "version": "0.1.2",
   "description": "Append model-visible durations to slow Pi tool results",
   "type": "module",
   "author": "Mitch Fultz (https://github.com/fitchmultz)",
@@ -40,7 +40,7 @@
     ]
   },
   "devDependencies": {
-    "@earendil-works/pi-coding-agent": "^0.80.2",
+    "@earendil-works/pi-coding-agent": "^0.80.6",
     "typescript": "^5.9.3"
   },
   "peerDependencies": {