pi-goosedump 0.12.43 → 0.12.44

package/dist/index.ts

@@ -826,6 +826,8 @@ class CompactionRuntime {
   activeSessionFile = null;
   sessionActive = false;
   activePi = null;
+  expectedTurnBoundaryAbort = false;
+  suppressedAbortMessages = new WeakSet;
   constructor(deps) {
     this.overrideDefaultCompaction = deps.overrideDefaultCompaction;
     this.nativeExecutablePath = deps.nativeExecutablePath;
@@ -884,6 +886,7 @@ class CompactionRuntime {
     this.activeSessionFile = ctx.sessionManager.getSessionFile?.() ?? null;
     this.sessionActive = true;
     this.compactionState = { phase: "idle" };
+    this.expectedTurnBoundaryAbort = false;
     this.syncCompactCounters(ctx);
     this.updateStatus(ctx);
     this.queueTurnCompaction(ctx, 1);
@@ -895,6 +898,7 @@ class CompactionRuntime {
     this.activeSessionFile = null;
     this.sessionActive = false;
     this.compactionState = { phase: "idle" };
+    this.expectedTurnBoundaryAbort = false;
   }
   onAgentStart(ctx) {
     if (this.compactionState.phase !== "running" || this.compactionState.owner !== "pi")
@@ -904,7 +908,19 @@ class CompactionRuntime {
       this.syncCompactCounters(ctx);
     this.setCompactionState(ctx, request ? this.requeueRequest(request, this.settlementGeneration + 1) : { phase: "idle" });
   }
+  onMessageEnd(event, ctx) {
+    if (!this.expectedTurnBoundaryAbort || !this.contextIsCurrent(ctx) || event.message.role !== "assistant" || event.message.stopReason !== "aborted") {
+      return;
+    }
+    this.expectedTurnBoundaryAbort = false;
+    this.suppressedAbortMessages.add(event.message);
+    return {
+      message: { ...event.message, stopReason: "stop", errorMessage: undefined }
+    };
+  }
   onTurnEnd(event, ctx) {
+    if (this.suppressedAbortMessages.delete(event.message))
+      return;
     if (!isCompletedAssistant(event.message))
       return;
     const resumeAfter = event.toolResults.length > 0 && !ctx.hasPendingMessages();
@@ -917,6 +933,7 @@ class CompactionRuntime {
     this.interruptToolLoopForQueuedCompaction(ctx, resumeAfter);
   }
   onAgentSettled(ctx) {
+    this.expectedTurnBoundaryAbort = false;
     this.settlementGeneration += 1;
     this.syncCompactCounters(ctx);
     this.updateStatus(ctx);
@@ -1094,7 +1111,7 @@ class CompactionRuntime {
     const target = compactionEntry.details.target;
     if (typeof target !== "string")
       return;
-    this.enqueueRemember(target, (message) => {
+    this.enqueueRemember(target, ctx, (message) => {
       if (ctx.hasUI) {
         ctx.ui.notify(`goosedump memory learning skipped: ${message}`, "warning");
       }
@@ -1147,6 +1164,7 @@ class CompactionRuntime {
       phase: "queued",
       request: { ...request, resumeAfter: true }
     };
+    this.expectedTurnBoundaryAbort = true;
     ctx.abort();
   }
   queueThresholdCompaction(ctx, eligibleSettlement, resumeAfter) {
@@ -1258,6 +1276,38 @@ class CompactionRuntime {
   }
 }
 
+// src/memory-activity.ts
+import { truncateToWidth } from "@earendil-works/pi-tui";
+var MEMORY_ACTIVITY_WIDGET_KEY = "goosedump-memory-activity";
+var ACTIVITY_FRAMES = ["▰▱▱▱", "▱▰▱▱", "▱▱▰▱", "▱▱▱▰", "▱▱▰▱", "▱▰▱▱"];
+function showMemoryActivity(ctx) {
+  if (!ctx.hasUI)
+    return;
+  ctx.ui.setWidget(MEMORY_ACTIVITY_WIDGET_KEY, (tui, theme) => {
+    let frame = 0;
+    const timer = setInterval(() => {
+      frame = (frame + 1) % ACTIVITY_FRAMES.length;
+      tui.requestRender();
+    }, 140);
+    timer.unref();
+    return {
+      render(width) {
+        const bar = theme.fg("accent", ACTIVITY_FRAMES[frame] ?? ACTIVITY_FRAMES[0]);
+        const label = theme.fg("muted", "long-term memory");
+        return [truncateToWidth(`${bar} ${label}`, width, "")];
+      },
+      invalidate() {},
+      dispose() {
+        clearInterval(timer);
+      }
+    };
+  }, { placement: "belowEditor" });
+}
+function hideMemoryActivity(ctx) {
+  if (ctx.hasUI)
+    ctx.ui.setWidget(MEMORY_ACTIVITY_WIDGET_KEY, undefined);
+}
+
 // src/policy.ts
 var GOOSE_TOOL_NAMES = [
   "goose_remember",
@@ -1502,7 +1552,7 @@ function registerGooseTools(pi, deps) {
     const resolved = resolveHistoryTarget(ctx, "goose_remember", params);
     if (!resolved.ok)
       return resolved.result;
-    enqueueRemember(resolved.target, (message) => {
+    enqueueRemember(resolved.target, ctx, (message) => {
       if (ctx.hasUI) {
         ctx.ui.notify(`goosedump memory learning failed: ${message}`, "warning");
       }
@@ -1671,7 +1721,7 @@ import {
   Key,
   decodeKittyPrintable,
   matchesKey,
-  truncateToWidth,
+  truncateToWidth as truncateToWidth2,
   visibleWidth,
   wrapTextWithAnsi
 } from "@earendil-works/pi-tui";
@@ -1745,7 +1795,7 @@ class NumberInputSubmenu {
   }
 }
 function makeRow(content, innerW, border) {
-  const line = truncateToWidth(content, innerW);
+  const line = truncateToWidth2(content, innerW);
   const pad = Math.max(0, innerW - visibleWidth(line));
   return `${border} ${line}${" ".repeat(pad)} ${border}`;
 }
@@ -1873,7 +1923,7 @@ async function runSettingsEditor(ctx) {
         const { accent, bold, dim, muted, text } = colors;
         const innerW = overlayInnerWidth(width);
         const pad = (value, cellWidth) => {
-          const clipped = truncateToWidth(value, cellWidth);
+          const clipped = truncateToWidth2(value, cellWidth);
           return `${clipped}${" ".repeat(Math.max(0, cellWidth - visibleWidth(clipped)))}`;
         };
         const tabLabel = (label, active) => active ? accent(bold(`[${label}]`)) : muted(label);
@@ -1884,7 +1934,7 @@ async function runSettingsEditor(ctx) {
         } else if (tab === "help") {
           const shortcutWidth = Math.max(visibleWidth("Shortcut"), ...helpRows.map(([shortcut]) => visibleWidth(shortcut)));
           const descriptionWidth = Math.max(1, innerW - shortcutWidth - 4);
-          content.push(dim(`  ${pad("Shortcut", shortcutWidth)}  Description`), ...helpRows.map(([shortcut, description]) => `  ${accent(pad(shortcut, shortcutWidth))}  ${text(truncateToWidth(description, descriptionWidth))}`));
+          content.push(dim(`  ${pad("Shortcut", shortcutWidth)}  Description`), ...helpRows.map(([shortcut, description]) => `  ${accent(pad(shortcut, shortcutWidth))}  ${text(truncateToWidth2(description, descriptionWidth))}`));
         } else {
           for (const [index, spec] of specs.entries()) {
             const selected = index === selectedSetting;
@@ -2015,7 +2065,7 @@ async function runSearchPanel(ctx, query, executablePath) {
     }
     return {
       render(width) {
-        const header = overlayHeader(colors, width, `Goosedump search: ${truncateToWidth(query, 44)}`);
+        const header = overlayHeader(colors, width, `Goosedump search: ${truncateToWidth2(query, 44)}`);
         if (error) {
           return [header, ...overlayError(colors, width, error), overlayFooter(colors, width)];
         }
@@ -2375,12 +2425,12 @@ function createGoosedumpIntegration(options = {}) {
   const nativeExecutablePath = options.nativeExecutablePath ?? resolveNativeGoosedumpBinary();
   const rememberQueue = [];
   let rememberRunning = null;
-  function enqueueRemember(target, reportError) {
+  function enqueueRemember(target, ctx, reportError) {
     for (let i = rememberQueue.length - 1;i >= 0; i -= 1) {
       if (rememberQueue[i]?.target === target)
         rememberQueue.splice(i, 1);
     }
-    rememberQueue.push({ target, reportError });
+    rememberQueue.push({ target, ctx, reportError });
     ensureRememberWorker();
   }
   function ensureRememberWorker() {
@@ -2391,11 +2441,14 @@ function createGoosedumpIntegration(options = {}) {
         const job = rememberQueue.shift();
         if (!job)
           continue;
+        showMemoryActivity(job.ctx);
         try {
           await goosedumpMemoryRemember(job.target, undefined, nativeExecutablePath);
         } catch (err) {
           const message = err instanceof Error ? err.message : String(err);
           job.reportError(message);
+        } finally {
+          hideMemoryActivity(job.ctx);
         }
       }
     })().finally(() => {
@@ -2444,6 +2497,7 @@ ${renderToolRoutingPolicy()}`
     pi.on("agent_start", async (_event, ctx) => {
       compaction.onAgentStart(ctx);
     });
+    pi.on("message_end", (event, ctx) => compaction.onMessageEnd(event, ctx));
     pi.on("turn_end", async (event, ctx) => {
       compaction.onTurnEnd(event, ctx);
     });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-goosedump",
-  "version": "0.12.43",
+  "version": "0.12.44",
   "description": "Pi extension for goosedump session search, durable memory, session management, and compaction",
   "keywords": [
     "compaction",
@@ -38,7 +38,7 @@
   },
   "dependencies": {
     "@earendil-works/pi-tui": "^0.82.1",
-    "@jarkkojs/goosedump": "^0.12.43",
+    "@jarkkojs/goosedump": "^0.12.44",
     "@sinclair/typebox": "^0.34.49"
   },
   "devDependencies": {