@monotykamary/pi-loop 0.1.12 → 0.1.14

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@monotykamary/pi-loop",
-  "version": "0.1.12",
+  "version": "0.1.14",
   "description": "A pi extension that closes the verification loop on task completion",
   "author": "monotykamary",
   "license": "MIT",

package/src/index.ts

@@ -85,6 +85,16 @@ export default function (pi: ExtensionAPI) {
   const widgetState = createInitialState();
   let currentCtx: ExtensionContext | undefined;
   let idleSteers = 0; // consecutive agent_end steers; reset on done/stop/new loop
+  let userInputEpoch = 0;
+
+  pi.on('input', (event) => {
+    if (event.source === 'interactive' || event.source === 'rpc') {
+      userInputEpoch++;
+    }
+  });
+  pi.on('before_agent_start', () => {
+    userInputEpoch++;
+  });
 
   // ---- Session lifecycle: restore state ----
 
@@ -212,6 +222,7 @@ export default function (pi: ExtensionAPI) {
   pi.on('agent_settled', async (_event, ctx) => {
     currentCtx = ctx;
     if (!state.isActive()) return;
+    const inputEpochAtStart = userInputEpoch;
 
     state.incrementTurnCount();
     const s = state.getState()!;
@@ -261,7 +272,7 @@ export default function (pi: ExtensionAPI) {
     const decision = await analyze(
       ctx,
       s,
-      true /* always idle at agent_end */,
+      true /* fully settled checkpoint */,
       ineffectivePattern,
       undefined,
       (accumulated) => {
@@ -275,6 +286,16 @@ export default function (pi: ExtensionAPI) {
       }
     );
 
+    // A real user prompt supersedes a decision computed from the preceding
+    // settled snapshot. Do not race that prompt or steer from stale context.
+    if (userInputEpoch !== inputEpochAtStart) {
+      updateUI(ctx, widgetState, state.getState(), {
+        type: 'watching',
+        reframeTier: state.getReframeTier(),
+      });
+      return;
+    }
+
     if (decision.action === 'steer' && decision.message) {
       idleSteers++;
       state.addIntervention({
@@ -289,7 +310,7 @@ export default function (pi: ExtensionAPI) {
         message: decision.message,
         reframeTier: state.getReframeTier(),
       });
-      pi.sendUserMessage(decision.message);
+      pi.sendUserMessage(decision.message, { deliverAs: 'followUp' });
     } else if (decision.action === 'done') {
       idleSteers = 0;
       state.resetReframeTier();
@@ -317,7 +338,7 @@ export default function (pi: ExtensionAPI) {
         message: steerMessage,
         reframeTier: state.getReframeTier(),
       });
-      pi.sendUserMessage(steerMessage);
+      pi.sendUserMessage(steerMessage, { deliverAs: 'followUp' });
     } else {
       updateUI(ctx, widgetState, state.getState(), {
         type: 'watching',

package/src/ui/renderer.ts

@@ -175,6 +175,8 @@ function renderWithState(
     return {
       render: (width: number) => {
         const paddedWidth = Math.max(0, width - 1);
+        const fitLines = (lines: string[]) =>
+          lines.map((line) => truncateToWidth(line, paddedWidth, ''));
         widgetState.lastRenderedWidth = paddedWidth;
         const suffix = suffixParts.length > 0 ? sep + suffixParts.join(sep) : '';
 
@@ -206,7 +208,7 @@ function renderWithState(
           const visibleLines = widgetState.lastThinkingLines
             .slice(0, visibleCount)
             .map((ln) => theme.fg('dim', ln));
-          return [l1, ...visibleLines];
+          return fitLines([l1, ...visibleLines]);
         }
 
         // For non-analyzing states (steering/done/waiting), preserve existing line breaks.
@@ -214,12 +216,12 @@ function renderWithState(
         const isAnalyzing = action.type === 'analyzing';
         if (!isAnalyzing && widgetState.lastThinkingLines.length > 0) {
           const visibleLines = widgetState.lastThinkingLines.map((ln) => theme.fg('dim', ln));
-          return [l1, ...visibleLines];
+          return fitLines([l1, ...visibleLines]);
         }
 
         // No thinking to display
         if (!rawThinking) {
-          return [l1];
+          return fitLines([l1]);
         }
 
         const thinkingIndent = stripAnsi(thinkingPrefix).length;
@@ -249,7 +251,7 @@ function renderWithState(
         }
 
         widgetState.lastThinkingLines = plainLines;
-        return [l1, ...thinkingLines];
+        return fitLines([l1, ...thinkingLines]);
       },
       invalidate: () => {},
     };

package/tests/width-regression.test.ts

@@ -0,0 +1,41 @@
+import { describe, expect, it, vi } from 'vitest';
+import { updateUI } from '../src/ui/status-widget.js';
+
+vi.mock('@earendil-works/pi-tui', () => ({
+  truncateToWidth: (text: string, width: number) => text.slice(0, Math.max(0, width)),
+}));
+
+const theme = { fg: (_color: string, text: string) => text };
+
+describe('loop widget width', () => {
+  it('bounds long words and cached lines after a terminal shrink', () => {
+    const setWidget = vi.fn();
+    const ctx = {
+      ui: { setWidget, setStatus: vi.fn(), notify: vi.fn(), select: vi.fn() },
+      sessionManager: { getBranch: vi.fn(() => []) },
+      model: { provider: 'test', id: 'model' },
+    } as any;
+    const state = {
+      active: true,
+      outcome: 'goal',
+      interventions: [],
+      turnCount: 1,
+      justSteered: false,
+      lastAnalyzedTurn: 0,
+      snapshotBuffer: [],
+      reframeTier: 0,
+      lastSteerTurn: -1,
+    } as any;
+
+    updateUI(ctx, state, { type: 'analyzing', turn: 1, thinking: 'x'.repeat(200) });
+    let factory = setWidget.mock.calls.at(-1)[1];
+    factory(null, theme).render(100);
+
+    updateUI(ctx, state, { type: 'done' });
+    factory = setWidget.mock.calls.at(-1)[1];
+    const width = 20;
+    const lines = factory(null, theme).render(width);
+
+    expect(lines.every((line: string) => line.length <= width - 1)).toBe(true);
+  });
+});