taskplane 0.23.9 → 0.23.11

package/extensions/taskplane/engine-worker-entry.mjs

@@ -14,8 +14,21 @@ import { fileURLToPath } from "node:url";
 import { dirname, join } from "node:path";
 
 const __dirname = dirname(fileURLToPath(import.meta.url));
-// TP-115: Disable jiti filesystem cache to prevent stale compiled code
-// after npm update. Without this, jiti serves old cached .mjs even when
-// the .ts source files have been updated by a new package version.
-const jiti = createJiti(import.meta.url, { cache: false });
+
+// TP-115: Purge jiti cache for taskplane modules before loading.
+// Pi's main process caches compiled .ts → .mjs at $TEMP/jiti/.
+// The engine-worker fork must compile fresh from current source files,
+// not use stale cached versions from Pi's startup.
+import { readdirSync, unlinkSync } from "node:fs";
+import { tmpdir } from "node:os";
+try {
+  const cacheDir = join(tmpdir(), "jiti");
+  for (const f of readdirSync(cacheDir)) {
+    if (f.startsWith("taskplane-") && f.endsWith(".mjs")) {
+      try { unlinkSync(join(cacheDir, f)); } catch {}
+    }
+  }
+} catch { /* cache dir may not exist */ }
+
+const jiti = createJiti(import.meta.url);
 await jiti.import(join(__dirname, "engine-worker.ts"));

package/extensions/taskplane/engine.ts

@@ -2267,9 +2267,8 @@ export async function executeOrchBatch(
 						const snap = JSON.parse(readFileSync(join(lanesDir, f), "utf-8"));
 						const w = snap.worker || {};
 						const r = snap.reviewer || {};
-						// Key by session name (match lane record) for per-task lookup
-						const laneRec = batchState.lanes.find((l: { laneNumber: number }) => l.laneNumber === snap.laneNumber);
-						const key = laneRec?.tmuxSessionName || `lane-${snap.laneNumber}`;
+						// Key by lane number for reliable per-task lookup (TP-115)
+						const key = `lane-${snap.laneNumber}`;
 						laneTokens.set(key, {
 							input: (w.inputTokens || 0) + (r.inputTokens || 0),
 							output: (w.outputTokens || 0) + (r.outputTokens || 0),
@@ -2282,6 +2281,7 @@ export async function executeOrchBatch(
 			}
 		} catch { /* runtime dir may not exist */ }
 
+
 		// Legacy fallback: lane-state-*.json sidecars (only if V2 found nothing)
 		if (laneTokens.size === 0) {
 			try {
@@ -2320,9 +2320,9 @@ export async function executeOrchBatch(
 			const durationMs = (to.startTime && to.endTime) ? (to.endTime - to.startTime) : 0;
 
 			// Get tokens for this lane (cumulative — shared across tasks in same lane)
-			// TP-115: V2 sessionName includes role suffix (-worker/-reviewer) but
-			// laneTokens is keyed by tmuxSessionName (no suffix). Try both.
-			const tokens = laneTokens.get(to.sessionName)
+			// TP-115: Try lane number key first (V2), then sessionName variants (legacy).
+			const tokens = laneTokens.get(`lane-${lane}`)
+				|| laneTokens.get(to.sessionName)
 				|| laneTokens.get(to.sessionName?.replace(/-(?:worker|reviewer)$/, ""))
 				|| { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, costUsd: 0 };
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "taskplane",
-  "version": "0.23.9",
+  "version": "0.23.11",
   "description": "AI agent orchestration for pi — parallel task execution with checkpoint discipline",
   "keywords": [
     "pi-package",