@hk_net/pi-timestamp 0.1.4 → 0.1.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@hk_net/pi-timestamp",
-  "version": "0.1.4",
+  "version": "0.1.5",
   "description": "Pi extension that shows timestamps for user input and agent completion timing",
   "type": "module",
   "private": false,
@@ -26,8 +26,8 @@
     "README.md"
   ],
   "peerDependencies": {
-    "@earendil-works/pi-coding-agent": ">=0.80.1",
-    "@earendil-works/pi-tui": ">=0.80.1"
+    "@earendil-works/pi-coding-agent": ">=0.80.2",
+    "@earendil-works/pi-tui": ">=0.80.2"
   },
   "pi": {
     "extensions": [

package/timestamp.ts

@@ -32,11 +32,18 @@ function formatDuration(ms: number): string {
     return `${Math.floor(hrs)}h ${Math.floor(totalMins % 60)}m`;
 }
 
+function isTimeoutErrorMessage(message: string | undefined): boolean {
+    return /timed? out|timeout/i.test(message ?? "");
+}
+
 export default function (pi: ExtensionAPI) {
     let taskStartTime: number | undefined;
+    let waitingForRetryAfterTimeout = false;
 
-    // Track when the agent starts processing
+    // Track when the agent starts processing. If Pi is auto-retrying after a timeout,
+    // keep the original start time so the eventual completion covers the full task.
     pi.on("agent_start", async () => {
+        if (waitingForRetryAfterTimeout) return;
         taskStartTime = Date.now();
     });
 
@@ -46,16 +53,33 @@ export default function (pi: ExtensionAPI) {
     pi.on("message_end", async (event, ctx) => {
         if (event.message.role !== "user") return;
 
+        if (waitingForRetryAfterTimeout) {
+            taskStartTime = undefined;
+            waitingForRetryAfterTimeout = false;
+        }
+
         const ts = event.message.timestamp;
         if (!ts) return;
 
         ctx.ui.notify(`Sent ${formatTime(ts)}`, "info");
     });
 
-    // Show completion timing after agent finishes
-    pi.on("agent_end", async (_event, ctx) => {
+    // Show completion timing after the whole task finishes. If Pi ended this agent loop
+    // with a timeout error and will likely auto-retry, wait for the later completion.
+    pi.on("agent_end", async (event, ctx) => {
+        const lastAssistantMessage = [...event.messages].reverse().find((message) => message.role === "assistant");
+        if (
+            lastAssistantMessage?.role === "assistant" &&
+            lastAssistantMessage.stopReason === "error" &&
+            isTimeoutErrorMessage(lastAssistantMessage.errorMessage)
+        ) {
+            waitingForRetryAfterTimeout = true;
+            return;
+        }
+
         const startTime = taskStartTime;
         taskStartTime = undefined;
+        waitingForRetryAfterTimeout = false;
 
         if (startTime === undefined) return;