@cortexkit/pi-magic-context 0.44.0 → 0.44.2

package/dist/{message-time-backfill-3rp0w2z9.js → message-time-backfill-5yrp95cx.js}

@@ -1,7 +1,7 @@
-import"./index-tfzybajk.js";
+import"./index-59dmj4zs.js";
 import {
   logSlowWriteTransaction
-} from "./index-mmkdxa5j.js";
+} from "./index-97e5fzy3.js";
 
 // ../plugin/src/features/magic-context/message-time-backfill.ts
 var MESSAGE_TIME_BACKFILL_BATCH_SIZE = 500;

package/dist/{read-session-chunk-2s116ah5.js → read-session-chunk-zhtcx9f1.js}

@@ -10,6 +10,7 @@ import {
   readRawSessionMessages2,
   readRawSessionMessagePage2,
   getRawSessionMessageOrdinalCount2,
+  visitRawSessionMessages2,
   readRawSessionMessageRange2,
   primeTailRawMessageCache2,
   getCachedAbsoluteMessageCount2,
@@ -22,8 +23,8 @@ import {
   getRawSessionTagKeysThrough2,
   getLegacyProtectedTailStartOrdinal2,
   readSessionChunk2
-} from "./index-2kj55q4e.js";
-import"./index-tfzybajk.js";
+} from "./index-gsff513e.js";
+import"./index-59dmj4zs.js";
 export {
   cleanUserText2 as cleanUserText,
   compareRawSessionMessageOrder2 as compareRawSessionMessageOrder,
@@ -46,6 +47,7 @@ export {
   readSessionChunk2 as readSessionChunk,
   resolveHostServedBoundaryId2 as resolveHostServedBoundaryId,
   setRawMessageProvider2 as setRawMessageProvider,
+  visitRawSessionMessages2 as visitRawSessionMessages,
   withRawMessageProvider2 as withRawMessageProvider,
   withRawSessionMessageCache2 as withRawSessionMessageCache
 };

package/dist/{safe-notification-target-pd09ybr0.js → safe-notification-target-y3hg2m7p.js}

@@ -1,8 +1,8 @@
 import {
   isDefaultSessionTitle2,
   waitForSafeNotificationTarget2
-} from "./index-jscq7067.js";
-import"./index-tfzybajk.js";
+} from "./index-5fw91enm.js";
+import"./index-59dmj4zs.js";
 export {
   isDefaultSessionTitle2 as isDefaultSessionTitle,
   waitForSafeNotificationTarget2 as waitForSafeNotificationTarget

package/dist/subagent-entry.js

@@ -1,7 +1,10 @@
+import {
+  TOKEN_BUDGET_TOOL_REFUSAL
+} from "./index-gsff513e.js";
 import {
   setHarness,
   log
-} from "./index-tfzybajk.js";
+} from "./index-59dmj4zs.js";
 import {
   setStoragePrivatePermissionEnforcement,
   resolveProjectIdentityForSession,
@@ -10,11 +13,11 @@ import {
   ensureProjectRegisteredFromPiDirectory,
   resolvePiHarnessKind,
   registerMagicContextTools
-} from "./index-8ytss1hp.js";
-import"./index-2kj55q4e.js";
-import"./index-mmkdxa5j.js";
+} from "./index-esnhck8v.js";
+import"./index-97e5fzy3.js";
 
 // src/subagent-entry.ts
+import { readFileSync } from "node:fs";
 var SUBAGENT_DREAMER_ACTIONS_FLAG = "magic-context-dreamer-actions";
 var openedDb;
 function magicContextSubagentExtension(pi) {
@@ -24,6 +27,19 @@ function magicContextSubagentExtension(pi) {
     type: "boolean",
     default: false
   });
+  const budgetGuardFile = process.env.MAGIC_CONTEXT_SUBAGENT_BUDGET_GUARD_FILE;
+  if (budgetGuardFile) {
+    pi.on("tool_call", () => {
+      try {
+        if (readFileSync(budgetGuardFile, "utf8") === "finalize") {
+          return { block: true, reason: TOKEN_BUDGET_TOOL_REFUSAL };
+        }
+      } catch {
+        return { block: true, reason: TOKEN_BUDGET_TOOL_REFUSAL };
+      }
+      return;
+    });
+  }
   pi.on("session_start", async () => {
     try {
       const directory = process.cwd();
@@ -45,11 +61,11 @@ function magicContextSubagentExtension(pi) {
         memoryToolEnabled: dreamerActionsEnabled,
         allowDreamerActions: dreamerActionsEnabled,
         sessionScopedToolsDisabled: true,
-        todowriteEnabled: cfg.todowrite.enabled !== false,
+        todowriteEnabled: cfg.todowrite.enabled,
         todowriteCommandEnabled: false,
         promptSurface: registrationPromptSurface
       });
-      log(`[pi-subagent] registered tools: ctx_search${dreamerActionsEnabled ? ", ctx_memory" : ""}${cfg.todowrite.enabled !== false ? ", todowrite" : ""}` + ` (ctx_note/ctx_expand omitted: --no-session child;` + ` memory=${cfg.memory.enabled}, embedding=${cfg.embedding.provider !== "off"},` + ` git_commits=${cfg.memory.git_commit_indexing.enabled}, dreamer_actions=${dreamerActionsEnabled})`);
+      log(`[pi-subagent] registered tools: ctx_search${dreamerActionsEnabled ? ", ctx_memory" : ""}${cfg.todowrite.enabled ? ", todowrite" : ""}` + ` (ctx_note/ctx_expand omitted: --no-session child;` + ` memory=${cfg.memory.enabled}, embedding=${cfg.embedding.provider !== "off"},` + ` git_commits=${cfg.memory.git_commit_indexing.enabled}, dreamer_actions=${dreamerActionsEnabled})`);
     } catch (err) {
       const message = err instanceof Error ? err.message : String(err);
       log(`[pi-subagent] startup failed: ${message}`);

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@cortexkit/pi-magic-context",
-	"version": "0.44.0",
+	"version": "0.44.2",
 	"type": "module",
 	"description": "Pi and OMP coding agent extension for Magic Context — cross-session memory and context management",
 	"main": "dist/index.js",