pi-tool-duration 0.2.2 → 0.2.3

package/CHANGELOG.md

@@ -1,5 +1,11 @@
 # Changelog
 
+## [0.2.3] - 2026-09-18
+
+### Fixed
+
+- Preserve timing in native compaction input when long tool output is truncated, including split-turn summaries. Ordinary requests still append one duration marker, and recorded tool content and details remain unchanged.
+
 ## [0.2.2] - 2026-09-18
 
 ### Fixed

package/README.md

@@ -19,7 +19,7 @@ The extension measures from Pi's `tool_execution_start` through `tool_execution_
 [duration: 5.0s]
 ```
 
-Compaction summaries also receive annotated copies of tool results. Recorded tool output stays unchanged, including after reload, resume, and branch navigation. Timings remain available to the model across those transitions without duplicating Pi's native TUI timing such as `Took 5.0s`.
+Compaction input copies put timing before the tool output so it survives Pi's truncation of long results. Recorded tool output stays unchanged, including after reload, resume, and branch navigation. Timings remain available to the model across those transitions without duplicating Pi's native TUI timing such as `Took 5.0s`.
 
 Markers already saved as tool text by versions through 0.2.1 remain unchanged. They cannot be safely distinguished from genuine tool output with the same text.
 

package/extensions/tool-duration/index.ts

@@ -27,7 +27,11 @@ function thresholdMs(pi: ExtensionAPI): number {
   );
 }
 
-function withDurations(messages: ContextEvent["messages"], ctx: ExtensionContext) {
+function withDurations(
+  messages: ContextEvent["messages"],
+  ctx: ExtensionContext,
+  position: "append" | "prepend" = "append",
+) {
   const saved = new Map<string, string>();
   for (const entry of ctx.sessionManager.getBranch()) {
     if (entry.type !== "custom" || entry.customType !== TIMING_ENTRY) continue;
@@ -44,9 +48,10 @@ function withDurations(messages: ContextEvent["messages"], ctx: ExtensionContext
     if (message.role !== "toolResult") return message;
     const duration = saved.get(`${message.timestamp}:${message.toolCallId}`);
     if (!duration) return message;
+    const marker = { type: "text" as const, text: duration };
     return {
       ...message,
-      content: [...message.content, { type: "text" as const, text: duration }],
+      content: position === "prepend" ? [marker, ...message.content] : [...message.content, marker],
     };
   });
 }
@@ -92,9 +97,10 @@ export default function (pi: ExtensionAPI) {
   pi.on("context", (event, ctx) => ({ messages: withDurations(event.messages, ctx) }));
 
   pi.on("session_before_compact", ({ preparation }, ctx) => {
-    // Native summarization bypasses context hooks. Replace its inputs, never the saved messages.
-    preparation.messagesToSummarize = withDurations(preparation.messagesToSummarize, ctx);
-    preparation.turnPrefixMessages = withDurations(preparation.turnPrefixMessages, ctx);
+    // Native summarization bypasses context hooks and truncates tool text from the end.
+    // Put timing first in its input copies, never in saved messages or ordinary requests.
+    preparation.messagesToSummarize = withDurations(preparation.messagesToSummarize, ctx, "prepend");
+    preparation.turnPrefixMessages = withDurations(preparation.turnPrefixMessages, ctx, "prepend");
   });
 
   const clearTimings = () => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-tool-duration",
-  "version": "0.2.2",
+  "version": "0.2.3",
   "description": "Append model-visible durations to slow Pi tool results",
   "type": "module",
   "author": "Mitch Fultz (https://github.com/fitchmultz)",