@astrosheep/pi-context 0.26.0 → 0.26.1

package/dist/build-info.json

@@ -1,4 +1,4 @@
 {
-  "version": "0.26.0",
-  "sourceHash": "48dcba30fd66f729c046d9ec89214e69ada06592df8d464f3b8cc70562dd5174"
+  "version": "0.26.1",
+  "sourceHash": "fb313022cd1aab93e422144e8f336eced42b67c4d35b89a1133be2254f066093"
 }

package/dist/extension.js

@@ -1,5 +1,5 @@
 // <define:__PI_CONTEXT_BUILD__>
-var define_PI_CONTEXT_BUILD_default = { version: "0.26.0", sourceHash: "48dcba30fd66f729c046d9ec89214e69ada06592df8d464f3b8cc70562dd5174" };
+var define_PI_CONTEXT_BUILD_default = { version: "0.26.1", sourceHash: "fb313022cd1aab93e422144e8f336eced42b67c4d35b89a1133be2254f066093" };
 
 // src/index.ts
 import { VERSION as VERSION2 } from "@earendil-works/pi-coding-agent";
@@ -8,6 +8,17 @@ import { VERSION as VERSION2 } from "@earendil-works/pi-coding-agent";
 import { Type as Type2 } from "@earendil-works/pi-ai";
 import { defineTool } from "@earendil-works/pi-coding-agent";
 
+// src/text-match.ts
+function earliestMatchOffsetChars(text, queries) {
+  let earliest = -1;
+  for (const query of queries) {
+    const index = text.indexOf(query);
+    if (index < 0) continue;
+    if (earliest < 0 || index < earliest) earliest = index;
+  }
+  return earliest <= 0 ? 0 : Array.from(text.slice(0, earliest)).length;
+}
+
 // src/tool-output.ts
 var TOOL_OUTPUT_MAX_BYTES = 32 * 1024;
 var DEFAULT_READ_WINDOW_CHARS = 12e3;
@@ -76,15 +87,6 @@ chars: [${window.offset_chars},${end}) of ${window.total_chars}
 next_offset_chars: ${next}
 `;
 }
-function earliestMatchOffsetChars(text, queries) {
-  let earliest = -1;
-  for (const query of queries) {
-    const index = text.indexOf(query);
-    if (index < 0) continue;
-    if (earliest < 0 || index < earliest) earliest = index;
-  }
-  return earliest <= 0 ? 0 : Array.from(text.slice(0, earliest)).length;
-}
 function page(items, cursor2, key, limit, truncate) {
   const end = Math.min(items.length, cursor2 + (limit ?? items.length));
   const selected = [];
@@ -1003,15 +1005,6 @@ function matchLineNumbers(body, needle) {
   }
   return lines;
 }
-function earliestMatchOffsetChars2(text, queries) {
-  let earliest = -1;
-  for (const query of queries) {
-    const index = text.indexOf(query);
-    if (index < 0) continue;
-    if (earliest < 0 || index < earliest) earliest = index;
-  }
-  return earliest <= 0 ? 0 : Array.from(text.slice(0, earliest)).length;
-}
 async function atomicWrite(path, content) {
   await mkdir(dirname2(path), { recursive: true });
   const tmp = `${path}.${process.pid}.${randomUUID()}.tmp`;
@@ -1156,11 +1149,10 @@ function createNotesStore(input) {
       return { meta, applied: operations.length, resolvedScope: scope, change };
     });
   }
-  async function list(options = {}) {
-    const stableOptions = { ...options };
-    const matcher = matcherFor(normalizePattern(stableOptions.pattern, context));
-    const rows = [];
-    for (const home of await homesFor(context, stableOptions)) {
+  async function* scan(options) {
+    const matcher = matcherFor(normalizePattern(options.pattern, context));
+    const homes = await homesFor(context, options);
+    for (const home of homes) {
       const scope = home.scope;
       const root = scopeDir(scope, context, home.who);
       for (const path of await walkMarkdown(root)) {
@@ -1170,38 +1162,35 @@ function createNotesStore(input) {
         const raw = await withPathQueue(fullPath, () => readFile(fullPath, "utf8"));
         const { meta, body } = parseNote(raw);
         meta.scope = scope;
-        rows.push({ address, scope, path, meta, body, sizeBytes: Buffer.byteLength(body, "utf8") });
+        yield { address, scope, path, meta, body };
       }
     }
+  }
+  async function list(options = {}) {
+    const stableOptions = { ...options };
+    const rows = [];
+    for await (const row of scan(stableOptions)) {
+      rows.push({ ...row, sizeBytes: Buffer.byteLength(row.body, "utf8") });
+    }
     rows.sort((a, b) => b.meta.updatedAt - a.meta.updatedAt || a.address.localeCompare(b.address));
     return rows;
   }
   async function search(queries, options = {}) {
     const stableQueries = [...queries];
     const stableOptions = { ...options };
-    const matcher = matcherFor(normalizePattern(stableOptions.pattern, context));
     const rows = [];
-    for (const home of await homesFor(context, stableOptions)) {
-      const scope = home.scope;
-      const root = scopeDir(scope, context, home.who);
-      for (const path of await walkMarkdown(root)) {
-        const address = addressFor(context, scope, path, home.who);
-        if (matcher && !matcher.test(address)) continue;
-        const fullPath = join2(root, path);
-        const raw = await withPathQueue(fullPath, () => readFile(fullPath, "utf8"));
-        const { meta, body } = parseNote(raw);
-        meta.scope = scope;
-        const serializedBodyOffset = Array.from(serializeNote(accessedMeta(meta, scope, Date.now()), "")).length;
-        let baseChars = 0;
-        const matches = [];
-        for (const [index, line] of body.split("\n").entries()) {
-          if (stableQueries.some((query) => line.includes(query))) {
-            matches.push({ line: index + 1, text: line, offsetChars: serializedBodyOffset + baseChars + earliestMatchOffsetChars2(line, stableQueries) });
-          }
-          baseChars += Array.from(line).length + 1;
+    for await (const note of scan(stableOptions)) {
+      const { address, path, scope, meta, body } = note;
+      const serializedBodyOffset = Array.from(serializeNote(accessedMeta(meta, scope, Date.now()), "")).length;
+      let baseChars = 0;
+      const matches = [];
+      for (const [index, line] of body.split("\n").entries()) {
+        if (stableQueries.some((query) => line.includes(query))) {
+          matches.push({ line: index + 1, text: line, offsetChars: serializedBodyOffset + baseChars + earliestMatchOffsetChars(line, stableQueries) });
         }
-        if (matches.length > 0) rows.push({ address, path, scope, meta, matches });
+        baseChars += Array.from(line).length + 1;
       }
+      if (matches.length > 0) rows.push({ address, path, scope, meta, matches });
     }
     rows.sort((a, b) => a.address.localeCompare(b.address));
     return rows;
@@ -1556,14 +1545,29 @@ function registerBudget(pi, isEnabled, settingsManager, onCloseOut = () => {
   };
   let pendingGuidance;
   let pendingWarning;
-  let pendingNotices = [];
-  const notifyCommittedReminders = (ctx) => {
-    const windowId = currentWindowId(ctx);
-    for (const notice of pendingNotices) {
-      if (notice.windowId !== windowId || !hasWindowMessage(ctx, notice.customType)) continue;
-      ctx.ui.notify(notice.customType === WARNING_TYPE ? "pi-context: Context almost full; close out the current memory window." : "pi-context: Context running low; checkpoint your notes soon.", "warning");
+  let pendingNotices = /* @__PURE__ */ new Map();
+  const noticeKey = (sessionId, windowId) => `${sessionId}:${windowId}`;
+  const warningCommittedInWindow = (ctx, notice) => {
+    if (ctx.sessionManager.getSessionId() !== notice.sessionId) return false;
+    let windowId = rootWindowId(notice.sessionId);
+    for (const entry of ctx.sessionManager.getBranch()) {
+      if (isWindowMarker(entry)) {
+        windowId = entry.data.windowId;
+        continue;
+      }
+      if (windowId === notice.windowId && entry.type === "custom_message" && entry.customType === WARNING_TYPE) return true;
+    }
+    return false;
+  };
+  const notifyCommittedWarnings = (ctx, settled = false) => {
+    for (const [key, notice] of pendingNotices) {
+      if (warningCommittedInWindow(ctx, notice)) {
+        pendingNotices.delete(key);
+        ctx.ui.notify("pi-context: Context almost full; close out the current memory window.", "warning");
+      } else if (settled) {
+        pendingNotices.delete(key);
+      }
     }
-    pendingNotices = [];
   };
   const clearStaged = () => {
     pendingGuidance = void 0;
@@ -1571,7 +1575,7 @@ function registerBudget(pi, isEnabled, settingsManager, onCloseOut = () => {
   };
   const resetForTransition = () => {
     clearStaged();
-    pendingNotices = [];
+    pendingNotices.clear();
     invalidateThresholds();
     notifiedWarnings.clear();
   };
@@ -1583,7 +1587,10 @@ function registerBudget(pi, isEnabled, settingsManager, onCloseOut = () => {
     clearStaged();
     const windowId = currentWindowId(ctx);
     const drafts = staged.filter((draft) => draft !== void 0 && draft.windowId === windowId);
-    pendingNotices = drafts.map(({ windowId: windowId2, customType, remaining }) => ({ windowId: windowId2, customType, remaining }));
+    if (drafts.some((draft) => draft.customType === WARNING_TYPE)) {
+      const sessionId = ctx.sessionManager.getSessionId();
+      pendingNotices.set(noticeKey(sessionId, windowId), { sessionId, windowId });
+    }
     return drafts.map((draft) => ({
       type: "custom_message",
       customType: draft.customType,
@@ -1598,8 +1605,9 @@ function registerBudget(pi, isEnabled, settingsManager, onCloseOut = () => {
   pi.on("session_tree", resetForTransition);
   pi.on("model_select", resetForTransition);
   pi.on("session_shutdown", resetForTransition);
+  pi.on("turn_start", (_event, ctx) => notifyCommittedWarnings(ctx));
   pi.on("agent_settled", (_event, ctx) => {
-    notifyCommittedReminders(ctx);
+    notifyCommittedWarnings(ctx, true);
     clearStaged();
   });
   pi.on("context", (_event, ctx) => {
@@ -1647,7 +1655,7 @@ function registerBudget(pi, isEnabled, settingsManager, onCloseOut = () => {
     consumeTurnEnd,
     clear: () => {
       clearStaged();
-      pendingNotices = [];
+      pendingNotices.clear();
     }
   };
 }

package/dist/src/context/budget.js

@@ -2,7 +2,7 @@ import { Type } from "@earendil-works/pi-ai";
 import { defineTool } from "@earendil-works/pi-coding-agent";
 import { GUIDANCE_TYPE, WARNING_CONTENT, WARNING_TYPE } from "../protocol.js";
 import { readThresholdSettings } from "./thresholds.js";
-import { currentWindowId, hasWindowMessage, windowUsage } from "./context-window.js";
+import { currentWindowId, hasWindowMessage, isWindowMarker, rootWindowId, windowUsage } from "./context-window.js";
 import { tokenBudgetGuidance } from "./prompts.js";
 import { output } from "../tool-output.js";
 /** Remaining tokens in the provider's active window, or null without a usable estimate. */
@@ -42,17 +42,33 @@ export function registerBudget(pi, isEnabled, settingsManager, onCloseOut = () =
     const invalidateThresholds = () => { cachedPolicy = undefined; };
     let pendingGuidance;
     let pendingWarning;
-    let pendingNotices = [];
-    const notifyCommittedReminders = (ctx) => {
-        const windowId = currentWindowId(ctx);
-        for (const notice of pendingNotices) {
-            if (notice.windowId !== windowId || !hasWindowMessage(ctx, notice.customType))
+    let pendingNotices = new Map();
+    const noticeKey = (sessionId, windowId) => `${sessionId}:${windowId}`;
+    const warningCommittedInWindow = (ctx, notice) => {
+        if (ctx.sessionManager.getSessionId() !== notice.sessionId)
+            return false;
+        let windowId = rootWindowId(notice.sessionId);
+        for (const entry of ctx.sessionManager.getBranch()) {
+            if (isWindowMarker(entry)) {
+                windowId = entry.data.windowId;
                 continue;
-            ctx.ui.notify(notice.customType === WARNING_TYPE
-                ? "pi-context: Context almost full; close out the current memory window."
-                : "pi-context: Context running low; checkpoint your notes soon.", "warning");
+            }
+            if (windowId === notice.windowId && entry.type === "custom_message" && entry.customType === WARNING_TYPE)
+                return true;
+        }
+        return false;
+    };
+    const notifyCommittedWarnings = (ctx, settled = false) => {
+        for (const [key, notice] of pendingNotices) {
+            if (warningCommittedInWindow(ctx, notice)) {
+                pendingNotices.delete(key);
+                ctx.ui.notify("pi-context: Context almost full; close out the current memory window.", "warning");
+            }
+            else if (settled) {
+                // An uncommitted draft must not be matched to a later manual warning.
+                pendingNotices.delete(key);
+            }
         }
-        pendingNotices = [];
     };
     const clearStaged = () => {
         pendingGuidance = undefined;
@@ -60,7 +76,7 @@ export function registerBudget(pi, isEnabled, settingsManager, onCloseOut = () =
     };
     const resetForTransition = () => {
         clearStaged();
-        pendingNotices = [];
+        pendingNotices.clear();
         invalidateThresholds();
         notifiedWarnings.clear();
     };
@@ -72,7 +88,10 @@ export function registerBudget(pi, isEnabled, settingsManager, onCloseOut = () =
         clearStaged();
         const windowId = currentWindowId(ctx);
         const drafts = staged.filter((draft) => draft !== undefined && draft.windowId === windowId);
-        pendingNotices = drafts.map(({ windowId, customType, remaining }) => ({ windowId, customType, remaining }));
+        if (drafts.some((draft) => draft.customType === WARNING_TYPE)) {
+            const sessionId = ctx.sessionManager.getSessionId();
+            pendingNotices.set(noticeKey(sessionId, windowId), { sessionId, windowId });
+        }
         return drafts.map((draft) => ({
             type: "custom_message",
             customType: draft.customType,
@@ -84,12 +103,12 @@ export function registerBudget(pi, isEnabled, settingsManager, onCloseOut = () =
     pi.on("session_tree", resetForTransition);
     pi.on("model_select", resetForTransition);
     pi.on("session_shutdown", resetForTransition);
-    // A request can fail before Pi emits turn_end. agent_settled is the public
-    // lifecycle point that must discard an uncommitted draft before the next prompt.
-    // UI notices follow committed reminders. Aborted requests can retry their drafts
-    // without showing the same low-budget notification twice.
+    // A warning can be committed at turn_end, before a tool turn or a reset changes the
+    // active window. Observe the active branch at public lifecycle boundaries and match
+    // the candidate against its originating window segment, not only the current window.
+    pi.on("turn_start", (_event, ctx) => notifyCommittedWarnings(ctx));
     pi.on("agent_settled", (_event, ctx) => {
-        notifyCommittedReminders(ctx);
+        notifyCommittedWarnings(ctx, true);
         clearStaged();
     });
     pi.on("context", (_event, ctx) => {
@@ -149,6 +168,6 @@ export function registerBudget(pi, isEnabled, settingsManager, onCloseOut = () =
         automaticResetEnabled,
         hardReserveDue,
         consumeTurnEnd,
-        clear: () => { clearStaged(); pendingNotices = []; },
+        clear: () => { clearStaged(); pendingNotices.clear(); },
     };
 }

package/dist/src/notes/store.js

@@ -1,6 +1,7 @@
 import { randomUUID } from "node:crypto";
 import { mkdir, readdir, readFile, rename, rm, writeFile } from "node:fs/promises";
 import { dirname, join, resolve } from "node:path";
+import { earliestMatchOffsetChars } from "../text-match.js";
 import { assertAddress, assertGlobPattern, addressFor, globToRegExp } from "./address.js";
 import { snapshotNotesContext } from "./context.js";
 import { MAX_NOTE_BYTES, MAX_NOTE_PATH_BYTES } from "./constants.js";
@@ -149,18 +150,6 @@ function matchLineNumbers(body, needle) {
     }
     return lines;
 }
-/** Code-point offset of the earliest query occurrence, matching the serialized read text. */
-function earliestMatchOffsetChars(text, queries) {
-    let earliest = -1;
-    for (const query of queries) {
-        const index = text.indexOf(query);
-        if (index < 0)
-            continue;
-        if (earliest < 0 || index < earliest)
-            earliest = index;
-    }
-    return earliest <= 0 ? 0 : Array.from(text.slice(0, earliest)).length;
-}
 /** Every mutation uses a tmp file renamed into place in the same directory. */
 async function atomicWrite(path, content) {
     await mkdir(dirname(path), { recursive: true });
@@ -329,11 +318,10 @@ export function createNotesStore(input) {
             return { meta, applied: operations.length, resolvedScope: scope, change };
         });
     }
-    async function list(options = {}) {
-        const stableOptions = { ...options };
-        const matcher = matcherFor(normalizePattern(stableOptions.pattern, context));
-        const rows = [];
-        for (const home of await homesFor(context, stableOptions)) {
+    async function* scan(options) {
+        const matcher = matcherFor(normalizePattern(options.pattern, context));
+        const homes = await homesFor(context, options);
+        for (const home of homes) {
             const scope = home.scope;
             const root = scopeDir(scope, context, home.who);
             for (const path of await walkMarkdown(root)) {
@@ -344,40 +332,36 @@ export function createNotesStore(input) {
                 const raw = await withPathQueue(fullPath, () => readFile(fullPath, "utf8"));
                 const { meta, body } = parseNote(raw);
                 meta.scope = scope;
-                rows.push({ address, scope, path, meta, body, sizeBytes: Buffer.byteLength(body, "utf8") });
+                yield { address, scope, path, meta, body };
             }
         }
+    }
+    async function list(options = {}) {
+        const stableOptions = { ...options };
+        const rows = [];
+        for await (const row of scan(stableOptions)) {
+            rows.push({ ...row, sizeBytes: Buffer.byteLength(row.body, "utf8") });
+        }
         rows.sort((a, b) => b.meta.updatedAt - a.meta.updatedAt || a.address.localeCompare(b.address));
         return rows;
     }
     async function search(queries, options = {}) {
         const stableQueries = [...queries];
         const stableOptions = { ...options };
-        const matcher = matcherFor(normalizePattern(stableOptions.pattern, context));
         const rows = [];
-        for (const home of await homesFor(context, stableOptions)) {
-            const scope = home.scope;
-            const root = scopeDir(scope, context, home.who);
-            for (const path of await walkMarkdown(root)) {
-                const address = addressFor(context, scope, path, home.who);
-                if (matcher && !matcher.test(address))
-                    continue;
-                const fullPath = join(root, path);
-                const raw = await withPathQueue(fullPath, () => readFile(fullPath, "utf8"));
-                const { meta, body } = parseNote(raw);
-                meta.scope = scope;
-                const serializedBodyOffset = Array.from(serializeNote(accessedMeta(meta, scope, Date.now()), "")).length;
-                let baseChars = 0;
-                const matches = [];
-                for (const [index, line] of body.split("\n").entries()) {
-                    if (stableQueries.some((query) => line.includes(query))) {
-                        matches.push({ line: index + 1, text: line, offsetChars: serializedBodyOffset + baseChars + earliestMatchOffsetChars(line, stableQueries) });
-                    }
-                    baseChars += Array.from(line).length + 1;
+        for await (const note of scan(stableOptions)) {
+            const { address, path, scope, meta, body } = note;
+            const serializedBodyOffset = Array.from(serializeNote(accessedMeta(meta, scope, Date.now()), "")).length;
+            let baseChars = 0;
+            const matches = [];
+            for (const [index, line] of body.split("\n").entries()) {
+                if (stableQueries.some((query) => line.includes(query))) {
+                    matches.push({ line: index + 1, text: line, offsetChars: serializedBodyOffset + baseChars + earliestMatchOffsetChars(line, stableQueries) });
                 }
-                if (matches.length > 0)
-                    rows.push({ address, path, scope, meta, matches });
+                baseChars += Array.from(line).length + 1;
             }
+            if (matches.length > 0)
+                rows.push({ address, path, scope, meta, matches });
         }
         rows.sort((a, b) => a.address.localeCompare(b.address));
         return rows;

package/dist/src/text-match.d.ts

@@ -0,0 +1,5 @@
+/**
+ * Code-point offset of the earliest occurrence of any of `queries` in `text`, or 0 when
+ * none occurs. Shared by the two search tools so a match address is computed identically.
+ */
+export declare function earliestMatchOffsetChars(text: string, queries: string[]): number;

package/dist/src/text-match.js

@@ -0,0 +1,15 @@
+/**
+ * Code-point offset of the earliest occurrence of any of `queries` in `text`, or 0 when
+ * none occurs. Shared by the two search tools so a match address is computed identically.
+ */
+export function earliestMatchOffsetChars(text, queries) {
+    let earliest = -1;
+    for (const query of queries) {
+        const index = text.indexOf(query);
+        if (index < 0)
+            continue;
+        if (earliest < 0 || index < earliest)
+            earliest = index;
+    }
+    return earliest <= 0 ? 0 : Array.from(text.slice(0, earliest)).length;
+}

package/dist/src/tool-output.d.ts

@@ -1,3 +1,4 @@
+export { earliestMatchOffsetChars } from "./text-match.js";
 export declare const TOOL_OUTPUT_MAX_BYTES: number;
 export declare const DEFAULT_READ_WINDOW_CHARS = 12000;
 export declare const MAX_READ_WINDOW_CHARS = 50000;
@@ -55,11 +56,6 @@ export declare function readCharacterWindow<T>(text: string, offsetChars: number
  * source identity fields in wire order; range and continuation semantics are shared.
  */
 export declare function readWindowBlock(identity: ReadonlyArray<readonly [string, string]>, window: CharacterWindow): string;
-/**
- * Code-point offset of the earliest occurrence of any of `queries` in `text`, or 0 when
- * none occurs. Shared by the two search tools so a match address is computed identically.
- */
-export declare function earliestMatchOffsetChars(text: string, queries: string[]): number;
 /** Shrink a single page item to fit; only invoked when that item alone exceeds the budget. */
 export type ItemTruncator<T> = (item: T, fits: (candidate: T) => boolean) => T;
 /**

package/dist/src/tool-output.js

@@ -1,3 +1,4 @@
+export { earliestMatchOffsetChars } from "./text-match.js";
 export const TOOL_OUTPUT_MAX_BYTES = 32 * 1024;
 export const DEFAULT_READ_WINDOW_CHARS = 12000;
 export const MAX_READ_WINDOW_CHARS = 50000;
@@ -111,21 +112,6 @@ export function readWindowBlock(identity, window) {
     const fields = identity.map(([name, value]) => `${name}: ${value}`).join("\n");
     return `--- READ WINDOW ---\n${fields}\nchars: [${window.offset_chars},${end}) of ${window.total_chars}\nnext_offset_chars: ${next}\n`;
 }
-/**
- * Code-point offset of the earliest occurrence of any of `queries` in `text`, or 0 when
- * none occurs. Shared by the two search tools so a match address is computed identically.
- */
-export function earliestMatchOffsetChars(text, queries) {
-    let earliest = -1;
-    for (const query of queries) {
-        const index = text.indexOf(query);
-        if (index < 0)
-            continue;
-        if (earliest < 0 || index < earliest)
-            earliest = index;
-    }
-    return earliest <= 0 ? 0 : Array.from(text.slice(0, earliest)).length;
-}
 /**
  * Build a page without ever adding an item that would exceed the wire budget.
  *

package/dist/test/agent-loop.test.js

@@ -127,7 +127,7 @@ async function openFixture(options) {
                     assert.ok(sessionManager.getBranch().some((entry) => entry.type === "custom" && entry.customType === RESET_MARKER_TYPE && entry.data?.windowId === windowId), "notification follows the reset marker commit");
                     assert.ok(sessionManager.getBranch().some((entry) => entry.type === "custom_message" && entry.customType === BOOT_TYPE && entry.details?.windowId === windowId), "notification follows the reset boot commit");
                 }
-                if (type === "warning" && sessionManager.getBranch().some((entry) => entry.type === "custom_message" && (entry.customType === GUIDANCE_TYPE || entry.customType === WARNING_TYPE)))
+                if (type === "warning" && /^pi-context: Context (?:almost full|running low)/.test(message))
                     budgetNotices++;
                 notices.push(message);
             },
@@ -213,30 +213,98 @@ function assertFreshRequest(fixture, requestIndex, oldSentinel) {
     assert.ok(body.includes(CONTEXT_WINDOW_OPEN_TAG), "the new provider request includes the fresh context-window boot");
     assert.equal(body.split(CONTINUATION).length - 1, 1, "the fresh window carries exactly one reset message");
 }
-test("real AgentSession: aborted low-budget requests notify only after a retry commits the reminder", async () => {
+test("real AgentSession: aborted early guidance retries silently with automatic reset disabled", async () => {
     let fixture;
     fixture = await openFixture({
         compactionEnabled: false,
         script: (request) => assistant(fixture, [{ type: "text", text: `response ${request}` }], request === 2 ? "aborted" : "stop", { usage: usage(50_000) }),
     });
     try {
-        const reminders = () => fixture.sessionManager.getBranch().filter((entry) => entry.type === "custom_message" && entry.customType === GUIDANCE_TYPE);
+        const guidance = () => fixture.sessionManager.getBranch().filter((entry) => entry.type === "custom_message" && entry.customType === GUIDANCE_TYPE);
         const notices = () => fixture.budgetNotices();
         await fixture.session.prompt("Establish usage below the reminder line.");
         await fixture.session.waitForIdle();
-        assert.equal(reminders().length, 0);
+        assert.equal(guidance().length, 0);
         await fixture.session.prompt("Abort this low-budget request.");
         await fixture.session.waitForIdle();
-        assert.equal(reminders().length, 0, "an aborted turn does not commit its reminder");
-        assert.equal(notices(), 0, "an uncommitted reminder never notifies");
+        assert.equal(guidance().length, 0, "an aborted turn does not commit early guidance");
+        assert.equal(notices(), 0, "early guidance never notifies");
         await fixture.session.prompt("Retry successfully.");
         await fixture.session.waitForIdle();
-        assert.equal(reminders().length, 1);
-        assert.equal(notices(), 1, "the committed retry notifies once");
+        assert.equal(guidance().length, 1, "the successful retry commits the model guidance");
+        assert.equal(notices(), 0, "committed early guidance stays UI-silent");
         await fixture.session.prompt("Continue in the same window.");
         await fixture.session.waitForIdle();
-        assert.equal(reminders().length, 1);
-        assert.equal(notices(), 1, "later turns cannot repeat the notice");
+        assert.equal(guidance().length, 1);
+        assert.equal(notices(), 0, "later turns cannot toast the early reminder");
+    }
+    finally {
+        fixture.close();
+    }
+});
+test("real AgentSession: an aborted final warning stays silent until a successful retry commits it", async () => {
+    let fixture;
+    fixture = await openFixture({
+        compactionEnabled: true,
+        script: (request, context) => {
+            if (request === 1)
+                return assistant(fixture, [{ type: "toolCall", id: "abort-budget-probe-1", name: "get_context_remaining", arguments: {} }], "toolUse", { usage: usage(50_000) });
+            if (request === 2)
+                return assistant(fixture, [{ type: "toolCall", id: "abort-budget-probe-2", name: "get_context_remaining", arguments: {} }], "toolUse", { usage: usage(60_000) });
+            assert.ok(text(context).includes(WARNING_PROMPT), "the final warning reaches the model on both attempts");
+            return assistant(fixture, [{ type: "text", text: `response ${request}` }], request === 3 ? "aborted" : "stop", { usage: usage(60_000) });
+        },
+    });
+    try {
+        const warnings = () => fixture.sessionManager.getBranch().filter((entry) => entry.type === "custom_message" && entry.customType === WARNING_TYPE);
+        const notices = () => fixture.budgetNotices();
+        await fixture.session.prompt("Establish usage and reach the final runway.");
+        await fixture.session.waitForIdle();
+        assert.equal(fixture.requests.length, 3, "budget probes reach one final-warning attempt");
+        assert.equal(warnings().length, 0, "the aborted attempt leaves no durable warning");
+        assert.equal(notices(), 0, "an uncommitted warning never notifies");
+        await fixture.session.prompt("Retry successfully.");
+        await fixture.session.waitForIdle();
+        assert.equal(warnings().length, 1, "the successful retry commits one hidden warning");
+        assert.equal(resetMarkers(fixture).length, 1, "normal close-out resets after the committed warning");
+        assert.equal(notices(), 1, "the committed warning notifies once, including after reset");
+        assert.equal(fixture.notices.some((notice) => notice.includes("Context running low")), false, "early guidance remains UI-silent");
+    }
+    finally {
+        fixture.close();
+    }
+});
+test("real AgentSession: final warning survives its reset boundary and notifies once", { timeout: 20000 }, async () => {
+    let fixture;
+    fixture = await openFixture({
+        compactionEnabled: true,
+        script: (request, context) => {
+            if (request === 1)
+                return assistant(fixture, [{ type: "toolCall", id: "final-budget-probe-1", name: "get_context_remaining", arguments: {} }], "toolUse", { usage: usage(50_000) });
+            if (request === 2)
+                return assistant(fixture, [{ type: "toolCall", id: "final-budget-probe-2", name: "get_context_remaining", arguments: {} }], "toolUse", { usage: usage(60_000) });
+            if (request === 3) {
+                assert.ok(text(context).includes(WARNING_PROMPT), "the final warning reaches the model");
+                return assistant(fixture, [{ type: "toolCall", id: "final-budget-wipe", name: "wipe_memory", arguments: {} }], "toolUse", { usage: usage(60_000) });
+            }
+            return assistant(fixture, [{ type: "text", text: "fresh window ready" }]);
+        },
+    });
+    try {
+        assert.equal(fixture.notices.length, 0, "boot is silent");
+        await fixture.session.prompt("FINAL_WARNING_RESET_SENTINEL");
+        await fixture.session.waitForIdle();
+        assert.equal(fixture.requests.length, 4, "two budget probes, close-out tool, and fresh-window request run");
+        assertFreshRequest(fixture, 3, "FINAL_WARNING_RESET_SENTINEL");
+        assert.equal(fixture.sessionManager.getBranch().filter((entry) => entry.type === "custom_message" && entry.customType === WARNING_TYPE).length, 1);
+        assert.equal(resetMarkers(fixture).length, 1, "the reset commits in the same boundary as the final warning");
+        assert.equal(fixture.budgetNotices(), 1, "the old-window committed warning reaches the UI after reset");
+        assert.equal(fixture.notices.filter((notice) => notice.startsWith("pi-context: memory cleared · ")).length, 1, "reset notification remains intact");
+        assert.equal(fixture.notices.some((notice) => notice.includes("Context running low")), false, "early guidance stays silent");
+        await fixture.session.prompt("Continue after reset.");
+        await fixture.session.waitForIdle();
+        assert.equal(fixture.budgetNotices(), 1, "later turns cannot repeat the consumed final warning");
+        assert.equal(fixture.notices.filter((notice) => notice.startsWith("pi-context: memory cleared · ")).length, 1, "the prior reset notification is also deduplicated");
     }
     finally {
         fixture.close();

package/dist/test/boot.integration.test.js

@@ -4,11 +4,9 @@ import test from "node:test";
 import { historyFromSession, internal } from "../src/index.js";
 import { listNotes } from "./helpers/notes.js";
 import { CONTINUATION_TYPE, WARNING_TYPE } from "../src/protocol.js";
-import { appendText, call, commitTurnEndBoundary, context, installExtensionTestEnvironment, makeExtension, manager, noticesOf, resultJson, resultRead, runContextHook, runContextWithSystemHook, runHandlers, runManualCompact, runCommand, sentOf, } from "./helpers/extension.js";
-const testEnvironment = installExtensionTestEnvironment("pi-context-integration");
-test.beforeEach(() => testEnvironment.beforeEach());
-test.afterEach(() => testEnvironment.afterEach());
-test.after(() => testEnvironment.dispose());
+import { appendText, call, commitTurnEndBoundary, context, makeExtension, manager, noticesOf, resultJson, resultRead, runContextHook, runContextWithSystemHook, runHandlers, runManualCompact, runCommand, sentOf, } from "./helpers/extension.js";
+import { installExtensionTestHooks } from "./helpers/extension-test-environment.js";
+const testEnvironment = installExtensionTestHooks("pi-context-integration");
 test("custom reset marker removes old provider context while history remains searchable", async () => {
     const sessionManager = manager();
     const captured = makeExtension(sessionManager);
@@ -50,7 +48,7 @@ test("the root boot and reset boot carry durable window identity", async () => {
     const ctx = context(sessionManager);
     appendText(sessionManager, "user", "task before reset");
     appendText(sessionManager, "assistant", "working");
-    await call(captured, "notes_write", { path: "decisions.md", content: "use terra" }, ctx);
+    await call(captured, "notes_write", { address: "decisions.md", content: "use terra" }, ctx);
     // Root window: session_start persists the boot block without triggering a turn.
     await runHandlers(captured, "session_start", { reason: "startup" }, ctx);
     assert.equal(captured.sent.length, 1);