@cortexkit/pi-magic-context 0.49.27 → 0.49.28

package/dist/index.js

@@ -3549,7 +3549,7 @@ function startHostStallProfilerSwitch() {
 // package.json
 var package_default = {
   name: "@cortexkit/pi-magic-context",
-  version: "0.49.27",
+  version: "0.49.28",
   type: "module",
   description: "Pi and OMP coding agent extension for Magic Context — cross-session memory and context management",
   main: "dist/index.js",
@@ -14927,8 +14927,8 @@ async function mapMemories(args) {
         result.failureStage = outcome.failure.stage;
       }
       const committed = outcome.mapped + outcome.independent;
+      const committedIds = new Set(outcome.committedIds);
       runLeaseGuardedWrite(args.db, args.holderId, args.leaseKey, () => {
-        const committedIds = new Set(outcome.committedIds);
         for (const memory of batch.inputs) {
           if (committedIds.has(memory.id)) {
             delete retries[memory.id];
@@ -14963,7 +14963,22 @@ async function mapMemories(args) {
           log2(`[dreamer] map-memories: targeted retry still omitted ${outcome.requeue.length} id(s); leaving them unmapped for the next run`);
         }
       }
-      if (outcome.failure?.class === "timeout" && committed === 0) {
+      if (outcome.failure?.class === "timeout" && batch.inputs.length > 1) {
+        const unmapped = batch.inputs.filter((memory) => !committedIds.has(memory.id));
+        const splitSize = Math.floor(batch.inputs.length / 2);
+        const splits = [];
+        for (let offset = 0;offset < unmapped.length; offset += splitSize) {
+          splits.push({
+            inputs: unmapped.slice(offset, offset + splitSize),
+            isOmissionRetry: batch.isOmissionRetry
+          });
+        }
+        if (splits.length > 0) {
+          batches.splice(i + 1, 0, ...splits);
+          log2(`[dreamer] map-memories: splitting batch [${batch.inputs.map((memory) => memory.id).join(",")}] into ${splits.map((split) => split.inputs.length).join("+")} after timeout; retrying ${unmapped.length} uncommitted mapping(s)`);
+        }
+      }
+      if (outcome.failure?.class === "timeout" && batch.inputs.length === 1 && committed === 0) {
         consecutiveTimeouts += 1;
         timeoutStreakElapsedMs.push(outcome.failure.elapsedMs);
         if (consecutiveTimeouts >= CONSECUTIVE_TIMEOUT_LIMIT2) {
@@ -14971,7 +14986,7 @@ async function mapMemories(args) {
           log2(`[dreamer] map-memories starvation: circuit breaker tripped after ${consecutiveTimeouts} consecutive batch timeouts with no committed mappings (investigation made no durable progress); per-batch elapsed [${timeoutStreakElapsedMs.join("ms, ")}ms] vs ${sliceMs}ms slice; stopping with ${result.remaining} mapping(s) remaining`);
           break;
         }
-      } else {
+      } else if (committed > 0 || outcome.failure?.class !== "timeout") {
         consecutiveTimeouts = 0;
         timeoutStreakElapsedMs = [];
       }
@@ -22630,6 +22645,8 @@ async function awaitInFlightDreamers(registrationOwner) {
     return;
   await Promise.allSettled(runs);
 }
+var RUNNER_TIMEOUT_HEADROOM_MINUTES = 5;
+var UNKNOWN_TASK_TIMEOUT_MINUTES = 30;
 function createPiDreamerClient(opts, onAdjunctsRefreshNeeded = opts.onAdjunctsRefreshNeeded, isRegistrationOwnerActive = () => true, getMaxOutputTokens = () => opts.config.maxTokens, getDreamerConfig = () => opts.config) {
   const runner = piSubagentRunnerFactory();
   const budgetStates = new Map;
@@ -22683,7 +22700,7 @@ function createPiDreamerClient(opts, onAdjunctsRefreshNeeded = opts.onAdjunctsRe
         userMessage,
         model: perTaskModel,
         fallbackModels: undefined,
-        timeoutMs: 30 * 60 * 1000,
+        timeoutMs: ((taskConfig?.timeoutMinutes ?? UNKNOWN_TASK_TIMEOUT_MINUTES) + RUNNER_TIMEOUT_HEADROOM_MINUTES) * 60 * 1000,
         cwd: dreamSession.directory,
         signal: args.signal ? AbortSignal.any([args.signal, controller.signal]) : controller.signal,
         thinkingLevel: extractBodyVariant(args),
@@ -22792,7 +22809,11 @@ function accountingTaskFromTitle(title) {
   const task = title.slice("magic-context-dream-".length);
   if (!title.startsWith("magic-context-dream-") || task.length === 0)
     return null;
-  return task === "classify" ? "classify-memories" : task;
+  if (task === "classify")
+    return "classify-memories";
+  if (task === "user-memories")
+    return "review-user-memories";
+  return task;
 }
 function readSessionTitle(args) {
   const body = args.body;

package/package.json

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