@astrosheep/pi-context 0.25.1 → 0.25.2

package/dist/build-info.json

@@ -1,4 +1,4 @@
 {
-  "version": "0.25.1",
-  "sourceHash": "d9c843b30fa0f1426f40b6a054277e757d64b58ede6cb590aada253f34b75a81"
+  "version": "0.25.2",
+  "sourceHash": "e240c1946d0f2b6d7b1d83ed658c348c7daed5fce5a374844215f1eebddcb25f"
 }

package/dist/extension.js

@@ -1,5 +1,5 @@
 // <define:__PI_CONTEXT_BUILD__>
-var define_PI_CONTEXT_BUILD_default = { version: "0.25.1", sourceHash: "d9c843b30fa0f1426f40b6a054277e757d64b58ede6cb590aada253f34b75a81" };
+var define_PI_CONTEXT_BUILD_default = { version: "0.25.2", sourceHash: "e240c1946d0f2b6d7b1d83ed658c348c7daed5fce5a374844215f1eebddcb25f" };
 
 // src/index.ts
 import { VERSION as VERSION2 } from "@earendil-works/pi-coding-agent";
@@ -1389,6 +1389,7 @@ function registerBudget(pi, isEnabled, settingsManager) {
   const invalidateThresholds = () => {
     cachedPolicy = void 0;
   };
+  const formatRemaining = (remaining) => `${Math.max(0, Math.ceil(remaining / 1e3))}k`;
   let pendingGuidance;
   let pendingWarning;
   let pendingNotices = [];
@@ -1396,7 +1397,7 @@ function registerBudget(pi, isEnabled, settingsManager) {
     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 budget critical \u2014 final checkpoint warning recorded for the model." : "pi-context: context budget low \u2014 checkpoint reminder recorded for the model, kept out of the chat view.", "warning");
+      ctx.ui.notify(notice.customType === WARNING_TYPE ? `pi-context: Context almost full \u2014 ${formatRemaining(notice.remaining)} remaining` : `pi-context: Context running low \u2014 ${formatRemaining(notice.remaining)} remaining`, "warning");
     }
     pendingNotices = [];
   };
@@ -1418,7 +1419,7 @@ function registerBudget(pi, isEnabled, settingsManager) {
     clearStaged();
     const windowId = currentWindowId(ctx);
     const drafts = staged.filter((draft) => draft !== void 0 && draft.windowId === windowId);
-    pendingNotices = drafts.map(({ windowId: windowId2, customType }) => ({ windowId: windowId2, customType }));
+    pendingNotices = drafts.map(({ windowId: windowId2, customType, remaining }) => ({ windowId: windowId2, customType, remaining }));
     return drafts.map((draft) => ({
       type: "custom_message",
       customType: draft.customType,
@@ -1433,7 +1434,6 @@ function registerBudget(pi, isEnabled, settingsManager) {
   pi.on("session_tree", resetForTransition);
   pi.on("model_select", resetForTransition);
   pi.on("session_shutdown", resetForTransition);
-  pi.on("turn_start", (_event, ctx) => notifyCommittedReminders(ctx));
   pi.on("agent_settled", (_event, ctx) => {
     notifyCommittedReminders(ctx);
     clearStaged();
@@ -1450,7 +1450,7 @@ function registerBudget(pi, isEnabled, settingsManager) {
       const content = `${GUIDANCE_OPEN_TAG}
 ${WARNING_PROMPT}
 ${GUIDANCE_CLOSE_TAG}`;
-      pendingWarning = { windowId, content };
+      pendingWarning = { windowId, content, remaining };
       const warningMessage = {
         role: "custom",
         customType: WARNING_TYPE,
@@ -1463,7 +1463,7 @@ ${GUIDANCE_CLOSE_TAG}`;
     if (hasWindowMessage(ctx, GUIDANCE_TYPE) || pendingGuidance?.windowId === windowId) return void 0;
     if (remaining <= reminder) {
       const left = Math.max(0, remaining - warning);
-      pendingGuidance = { windowId, content: tokenBudgetGuidance(left) };
+      pendingGuidance = { windowId, content: tokenBudgetGuidance(left), remaining };
     }
     return void 0;
   });

package/dist/src/context/budget.js

@@ -40,6 +40,7 @@ export function registerBudget(pi, isEnabled, settingsManager) {
         return usage !== undefined && usage.tokens !== null && usage.contextWindow - usage.tokens <= thresholdsFor(ctx).reserve;
     };
     const invalidateThresholds = () => { cachedPolicy = undefined; };
+    const formatRemaining = (remaining) => `${Math.max(0, Math.ceil(remaining / 1000))}k`;
     let pendingGuidance;
     let pendingWarning;
     let pendingNotices = [];
@@ -49,8 +50,8 @@ export function registerBudget(pi, isEnabled, settingsManager) {
             if (notice.windowId !== windowId || !hasWindowMessage(ctx, notice.customType))
                 continue;
             ctx.ui.notify(notice.customType === WARNING_TYPE
-                ? "pi-context: context budget critical — final checkpoint warning recorded for the model."
-                : "pi-context: context budget low — checkpoint reminder recorded for the model, kept out of the chat view.", "warning");
+                ? `pi-context: Context almost full — ${formatRemaining(notice.remaining)} remaining`
+                : `pi-context: Context running low — ${formatRemaining(notice.remaining)} remaining`, "warning");
         }
         pendingNotices = [];
     };
@@ -72,7 +73,7 @@ export function registerBudget(pi, isEnabled, settingsManager) {
         clearStaged();
         const windowId = currentWindowId(ctx);
         const drafts = staged.filter((draft) => draft !== undefined && draft.windowId === windowId);
-        pendingNotices = drafts.map(({ windowId, customType }) => ({ windowId, customType }));
+        pendingNotices = drafts.map(({ windowId, customType, remaining }) => ({ windowId, customType, remaining }));
         return drafts.map((draft) => ({
             type: "custom_message",
             customType: draft.customType,
@@ -88,7 +89,6 @@ export function registerBudget(pi, isEnabled, settingsManager) {
     // 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.
-    pi.on("turn_start", (_event, ctx) => notifyCommittedReminders(ctx));
     pi.on("agent_settled", (_event, ctx) => {
         notifyCommittedReminders(ctx);
         clearStaged();
@@ -109,7 +109,7 @@ export function registerBudget(pi, isEnabled, settingsManager) {
             // A not-yet-committed shallow reminder is superseded by the final warning.
             pendingGuidance = undefined;
             const content = `${GUIDANCE_OPEN_TAG}\n${WARNING_PROMPT}\n${GUIDANCE_CLOSE_TAG}`;
-            pendingWarning = { windowId, content };
+            pendingWarning = { windowId, content, remaining };
             const warningMessage = {
                 role: "custom",
                 customType: WARNING_TYPE,
@@ -125,7 +125,7 @@ export function registerBudget(pi, isEnabled, settingsManager) {
             // Persist at turn_end, before any reset drafts. A queued sendMessage could
             // otherwise cross the marker and leak the old window's reminder forward.
             const left = Math.max(0, remaining - warning);
-            pendingGuidance = { windowId, content: tokenBudgetGuidance(left) };
+            pendingGuidance = { windowId, content: tokenBudgetGuidance(left), remaining };
         }
         return undefined;
     });

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

@@ -59,6 +59,7 @@ async function openFixture(options) {
     const streamSignals = [];
     const events = [];
     const notices = [];
+    let budgetNotices = 0;
     let session;
     const loader = new DefaultResourceLoader({
         cwd: dir,
@@ -120,16 +121,14 @@ async function openFixture(options) {
     };
     await session.bindExtensions({
         uiContext: {
-            notify(message) {
+            notify(message, type) {
                 if (message.startsWith("pi-context: memory cleared · ")) {
                     const windowId = message.split(" · ")[1];
                     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 (message.includes("context budget low") || message.includes("context budget critical")) {
-                    const customType = message.includes("context budget critical") ? WARNING_TYPE : GUIDANCE_TYPE;
-                    assert.ok(sessionManager.getBranch().some((entry) => entry.type === "custom_message" && entry.customType === customType), "budget notifications follow the reminder commit");
-                }
+                if (type === "warning" && sessionManager.getBranch().some((entry) => entry.type === "custom_message" && (entry.customType === GUIDANCE_TYPE || entry.customType === WARNING_TYPE)))
+                    budgetNotices++;
                 notices.push(message);
             },
         },
@@ -145,6 +144,7 @@ async function openFixture(options) {
         streamSignals,
         events,
         notices,
+        budgetNotices: () => budgetNotices,
         close: () => {
             session.dispose();
             if (managesEnvironment) {
@@ -221,22 +221,22 @@ test("real AgentSession: aborted low-budget requests notify only after a retry c
     });
     try {
         const reminders = () => fixture.sessionManager.getBranch().filter((entry) => entry.type === "custom_message" && entry.customType === GUIDANCE_TYPE);
-        const notices = () => fixture.notices.filter((message) => message.includes("context budget low"));
+        const notices = () => fixture.budgetNotices();
         await fixture.session.prompt("Establish usage below the reminder line.");
         await fixture.session.waitForIdle();
         assert.equal(reminders().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().length, 0, "an uncommitted reminder never notifies");
+        assert.equal(notices(), 0, "an uncommitted reminder never notifies");
         await fixture.session.prompt("Retry successfully.");
         await fixture.session.waitForIdle();
         assert.equal(reminders().length, 1);
-        assert.equal(notices().length, 1, "the committed retry notifies once");
+        assert.equal(notices(), 1, "the committed retry notifies once");
         await fixture.session.prompt("Continue in the same window.");
         await fixture.session.waitForIdle();
         assert.equal(reminders().length, 1);
-        assert.equal(notices().length, 1, "later turns cannot repeat the notice");
+        assert.equal(notices(), 1, "later turns cannot repeat the notice");
     }
     finally {
         fixture.close();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@astrosheep/pi-context",
-  "version": "0.25.1",
+  "version": "0.25.2",
   "type": "module",
   "description": "Codex-style context windows for Pi: durable reset windows, session history tools, and persistent notes.",
   "license": "MIT",

package/src/context/budget.ts

@@ -38,16 +38,17 @@ export function registerBudget(pi: ExtensionAPI, isEnabled: () => boolean, setti
 		return usage !== undefined && usage.tokens !== null && usage.contextWindow - usage.tokens <= thresholdsFor(ctx).reserve;
 	};
 	const invalidateThresholds = () => { cachedPolicy = undefined; };
-	let pendingGuidance: { windowId: string; content: string } | undefined;
-	let pendingWarning: { windowId: string; content: string } | undefined;
-	let pendingNotices: Array<{ windowId: string; customType: string }> = [];
+	const formatRemaining = (remaining: number): string => `${Math.max(0, Math.ceil(remaining / 1000))}k`;
+	let pendingGuidance: { windowId: string; content: string; remaining: number } | undefined;
+	let pendingWarning: { windowId: string; content: string; remaining: number } | undefined;
+	let pendingNotices: Array<{ windowId: string; customType: string; remaining: number }> = [];
 	const notifyCommittedReminders = (ctx: ExtensionContext) => {
 		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 budget critical — final checkpoint warning recorded for the model."
-				: "pi-context: context budget low — checkpoint reminder recorded for the model, kept out of the chat view.", "warning");
+				? `pi-context: Context almost full — ${formatRemaining(notice.remaining)} remaining`
+				: `pi-context: Context running low — ${formatRemaining(notice.remaining)} remaining`, "warning");
 		}
 		pendingNotices = [];
 	};
@@ -71,7 +72,7 @@ export function registerBudget(pi: ExtensionAPI, isEnabled: () => boolean, setti
 		clearStaged();
 		const windowId = currentWindowId(ctx);
 		const drafts = staged.filter((draft): draft is NonNullable<typeof draft> => draft !== undefined && draft.windowId === windowId);
-		pendingNotices = drafts.map(({ windowId, customType }) => ({ windowId, customType }));
+		pendingNotices = drafts.map(({ windowId, customType, remaining }) => ({ windowId, customType, remaining }));
 		return drafts.map((draft) => ({
 			type: "custom_message" as const,
 			customType: draft.customType,
@@ -88,7 +89,6 @@ export function registerBudget(pi: ExtensionAPI, isEnabled: () => boolean, setti
 	// 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.
-	pi.on("turn_start", (_event, ctx) => notifyCommittedReminders(ctx));
 	pi.on("agent_settled", (_event, ctx) => {
 		notifyCommittedReminders(ctx);
 		clearStaged();
@@ -106,7 +106,7 @@ export function registerBudget(pi: ExtensionAPI, isEnabled: () => boolean, setti
 			// A not-yet-committed shallow reminder is superseded by the final warning.
 			pendingGuidance = undefined;
 			const content = `${GUIDANCE_OPEN_TAG}\n${WARNING_PROMPT}\n${GUIDANCE_CLOSE_TAG}`;
-			pendingWarning = { windowId, content };
+			pendingWarning = { windowId, content, remaining };
 			const warningMessage = {
 				role: "custom" as const,
 				customType: WARNING_TYPE,
@@ -121,7 +121,7 @@ export function registerBudget(pi: ExtensionAPI, isEnabled: () => boolean, setti
 			// Persist at turn_end, before any reset drafts. A queued sendMessage could
 			// otherwise cross the marker and leak the old window's reminder forward.
 			const left = Math.max(0, remaining - warning);
-			pendingGuidance = { windowId, content: tokenBudgetGuidance(left) };
+			pendingGuidance = { windowId, content: tokenBudgetGuidance(left), remaining };
 		}
 		return undefined;
 	});