@cortexkit/pi-magic-context 0.31.1 → 0.31.3

package/dist/subagent-entry.js

@@ -159897,6 +159897,10 @@ var MagicContextConfigSchema = exports_external.object({
   }).describe("Embedding provider configuration"),
   temporal_awareness: exports_external.boolean().default(true).describe('Inject wall-clock gap markers (<!-- +Xm -->) between user messages where > 5 min elapsed since the previous message, and add start/end date attributes on compartments. Gives the agent a sense of session pacing and "how long ago" across multi-day sessions. Graduated from experimental.temporal_awareness; default: true (set false to opt out).'),
   keep_subagents: exports_external.boolean().default(false).describe("Debug: keep the child sessions Magic Context spawns for its own subagents (historian, dreamer, sidekick, memory-migration) instead of deleting them on success. Useful for short-term inspection/data collection — their full transcript (prompt, tool calls, token usage, output) stays in the host session store. Kept sessions accumulate until manually cleared; leave false for normal use. Requires a restart to take effect."),
+  todowrite: exports_external.object({
+    enabled: exports_external.boolean().default(true).describe("Pi only: register Magic Context's todowrite task-list tool. Disable if you use your own todo extension. OpenCode ships its own built-in todowrite; this setting has no effect there."),
+    overlay: exports_external.boolean().default(true).describe("Pi only: show the persistent todo overlay above the editor while tasks are active.")
+  }).default({ enabled: true, overlay: true }).describe("Pi-only todowrite tool and overlay controls. Pi registers tools and widgets at extension boot, so changing this after /cd requires /reload or restart."),
   smart_drops: exports_external.boolean().default(false).describe("Content-aware reclaim of provably-superseded tool output, layered on the existing execute-pass auto-drop. When on: superseded todowrite (keep newest 1), spent ctx_reduce (keep newest 5), and zero-value meta (bash_status, bash_kill, ctx_note read/dismiss) outputs are dropped; older edits to a file are compressed to a filePath-preserving marker while the newest edit per file stays full. Only acts on passes already busting the cache, so it never originates a cache bust. Honors the protected-tag reserve. Experimental: opt-in, default off until cache stability is proven; when off the wire is byte-identical to the positional-only reclaim. Requires a restart."),
   caveman_text_compression: exports_external.object({
     enabled: exports_external.boolean().default(false).describe("Apply deterministic caveman-style text compression to old conversation text. Active for primary sessions when enabled; never for subagents. Compresses user/assistant text in oldest-first tiers: ultra (oldest 20%), full, lite, untouched (newest 40%)."),
@@ -169120,16 +169124,16 @@ function synthesizeToolResultParts(msg) {
 // src/tools/ctx-expand.ts
 var ParamsSchema = exports_typebox.Object({
   start: exports_typebox.Optional(exports_typebox.Number({
-    description: "Start message ordinal (from compartment start attribute)"
+    description: `First message ordinal to expand — a compartment's start="N" attribute, or an ordinal from a ctx_search message hit`
   })),
   end: exports_typebox.Optional(exports_typebox.Number({
-    description: "End message ordinal (from compartment end attribute)"
+    description: `Last message ordinal to expand (inclusive) — a compartment's end="M" attribute`
   })),
   verbose: exports_typebox.Optional(exports_typebox.Boolean({
     description: "With start/end: list each message separately with its ordinal [N] and per-part preview, so you can recover one in full by ordinal."
   })),
   message: exports_typebox.Optional(exports_typebox.Number({
-    description: "Full untruncated recovery of ONE message by its ordinal (text + every tool call's full input/output). Recovers a tool output you dropped with ctx_reduce."
+    description: "Full untruncated recovery of ONE message by its ordinal (every text part + every tool call's complete input/output). Use an ordinal from a compartment, ctx_search hit, or verbose range. Recovers a tool output you dropped with ctx_reduce."
   }))
 });
 function ok(text) {
@@ -170105,9 +170109,11 @@ import path4 from "node:path";
 import { promisify } from "node:util";
 var execFileAsync = promisify(execFile);
 var GIT_TIMEOUT_MS2 = 1e4;
+var defaultExecFileForVerificationPaths = async (file2, args, options) => await execFileAsync(file2, [...args], options);
+var execFileForVerificationPaths = defaultExecFileForVerificationPaths;
 async function runGit(cwd, args) {
   try {
-    const result = await execFileAsync("git", [...args], {
+    const result = await execFileForVerificationPaths("git", args, {
       cwd,
       timeout: GIT_TIMEOUT_MS2,
       maxBuffer: 16 * 1024 * 1024,
@@ -171620,6 +171626,13 @@ function setSessionWorkMetrics(db, sessionId, newWorkTokens, totalInputTokens) {
          SET new_work_tokens = ?, total_input_tokens = ?
          WHERE session_id = ?`).run(Math.max(0, Math.floor(newWorkTokens)), Math.max(0, Math.floor(totalInputTokens)), sessionId);
 }
+function getSessionWorkMetrics(db, sessionId) {
+  const row = db.prepare("SELECT new_work_tokens, total_input_tokens FROM session_meta WHERE session_id = ?").get(sessionId);
+  return {
+    newWorkTokens: typeof row?.new_work_tokens === "number" ? row.new_work_tokens : 0,
+    totalInputTokens: typeof row?.total_input_tokens === "number" ? row.total_input_tokens : 0
+  };
+}
 // ../plugin/src/features/magic-context/storage-meta-session.ts
 import { Buffer as Buffer3 } from "node:buffer";
 
@@ -172941,25 +172954,45 @@ function formatMemoryList(memories) {
     id: String(m.id),
     category: m.category,
     status: m.status,
+    verification: m.verificationStatus,
     updated: new Date(m.updatedAt).toISOString(),
     content: m.content.replace(/\s+/g, " ").trim()
   }));
+  const headers = {
+    id: "ID",
+    category: "CATEGORY",
+    status: "STATUS",
+    verification: "VERIFY",
+    updated: "UPDATED",
+    content: "CONTENT"
+  };
   const widths = {
-    id: Math.max(2, ...rows.map((r) => r.id.length)),
-    category: Math.max(8, ...rows.map((r) => r.category.length)),
-    status: Math.max(6, ...rows.map((r) => r.status.length)),
-    updated: Math.max(7, ...rows.map((r) => r.updated.length))
+    id: Math.max(headers.id.length, ...rows.map((r) => r.id.length)),
+    category: Math.max(headers.category.length, ...rows.map((r) => r.category.length)),
+    status: Math.max(headers.status.length, ...rows.map((r) => r.status.length)),
+    verification: Math.max(headers.verification.length, ...rows.map((r) => r.verification.length)),
+    updated: Math.max(headers.updated.length, ...rows.map((r) => r.updated.length))
   };
   const fmt = (r) => [
     r.id.padEnd(widths.id),
     r.category.padEnd(widths.category),
     r.status.padEnd(widths.status),
+    r.verification.padEnd(widths.verification),
     r.updated.padEnd(widths.updated),
     r.content
   ].join(" | ");
   return [
     `Found ${rows.length} active ${rows.length === 1 ? "memory" : "memories"}:`,
     "",
+    fmt(headers),
+    [
+      "-".repeat(widths.id),
+      "-".repeat(widths.category),
+      "-".repeat(widths.status),
+      "-".repeat(widths.verification),
+      "-".repeat(widths.updated),
+      "-------"
+    ].join("-+-"),
     ...rows.map(fmt)
   ].join(`
 `);
@@ -173365,6 +173398,7 @@ function createCtxNoteTool(deps) {
     parameters: ParamsSchema3,
     async execute(_toolCallId, params, _signal, _onUpdate, ctx) {
       const sessionId = ctx.sessionManager.getSessionId();
+      const dreamerEnabled = deps.resolveDreamerEnabled?.(ctx) ?? deps.dreamerEnabled;
       const action2 = params.action ?? (params.content?.trim() ? "write" : "read");
       if (action2 === "write") {
         const content = params.content?.trim();
@@ -173373,7 +173407,7 @@ function createCtxNoteTool(deps) {
         const anchorOrdinal = captureAnchorOrdinal(deps.db, sessionId);
         const surfaceCondition = params.surface_condition?.trim();
         if (surfaceCondition) {
-          if (deps.dreamerEnabled !== true) {
+          if (dreamerEnabled !== true) {
             return err3("Error: Smart notes require dreamer to be enabled. Enable dreamer in magic-context.jsonc to use surface_condition.");
           }
           const projectIdentity = resolveProjectIdentity(ctx.cwd);
@@ -173382,6 +173416,7 @@ function createCtxNoteTool(deps) {
           }
           const note2 = addNote(deps.db, "smart", {
             content,
+            sessionId,
             projectPath: projectIdentity,
             surfaceCondition,
             anchorOrdinal
@@ -173633,6 +173668,7 @@ function createCtxReduceTool(deps) {
     parameters: ParamsSchema4,
     async execute(_toolCallId, params, _signal, _onUpdate, ctx) {
       const sessionId = ctx.sessionManager.getSessionId();
+      const protectedTags = Math.max(0, Math.floor(deps.resolveProtectedTags?.(ctx) ?? deps.protectedTags));
       if (!params.drop) {
         return err4("Error: 'drop' must be provided.");
       }
@@ -173650,7 +173686,7 @@ function createCtxReduceTool(deps) {
         return err4(`Error: Unknown tag(s) ${formatIds(unknownIds)}. Check available tags in conversation.`);
       }
       const activeTags = allTags.filter((tag) => tag.status === "active");
-      const protectedTagIds = activeTags.map((tag) => tag.tagNumber).sort((left, right) => right - left).slice(0, deps.protectedTags);
+      const protectedTagIds = activeTags.map((tag) => tag.tagNumber).sort((left, right) => right - left).slice(0, protectedTags);
       const protectedSet = new Set(protectedTagIds);
       const tagStatusMap = new Map(allTags.map((tag) => [tag.tagNumber, tag.status]));
       const pendingOps = getPendingOps(deps.db, sessionId);
@@ -175627,6 +175663,417 @@ function createCtxSearchTool(deps) {
   };
 }
 
+// src/tools/todo-view-pi.ts
+import {
+  truncateToWidth
+} from "@earendil-works/pi-tui";
+var TODO_TOOL_NAME = "todowrite";
+var TODOS_COMMAND_NAME = "todos";
+var WIDGET_KEY = "magic-context-todos";
+var MAX_OVERLAY_CONTENT_ROWS = 12;
+var TODO_STATUSES = [
+  "pending",
+  "in_progress",
+  "completed",
+  "cancelled"
+];
+var TODO_PRIORITIES = ["high", "medium", "low"];
+var STATUS_GLYPH = {
+  pending: "○",
+  in_progress: "◐",
+  completed: "✓",
+  cancelled: "✗"
+};
+var STATUS_COLOR = {
+  pending: "dim",
+  in_progress: "warning",
+  completed: "success",
+  cancelled: "error"
+};
+var snapshotsBySession = new Map;
+function isTodoStatus(value) {
+  return TODO_STATUSES.includes(value);
+}
+function isTodoPriority(value) {
+  return TODO_PRIORITIES.includes(value);
+}
+function cloneTodos(todos) {
+  return todos.map((todo) => ({ ...todo }));
+}
+function parseTodos(input) {
+  if (!Array.isArray(input))
+    return null;
+  const todos = [];
+  for (const item of input) {
+    if (item === null || typeof item !== "object")
+      return null;
+    const raw = item;
+    if (typeof raw.content !== "string" || !isTodoStatus(raw.status)) {
+      return null;
+    }
+    const todo = {
+      content: raw.content,
+      status: raw.status
+    };
+    if (isTodoPriority(raw.priority))
+      todo.priority = raw.priority;
+    if (typeof raw.id === "string" && raw.id.length > 0)
+      todo.id = raw.id;
+    todos.push(todo);
+  }
+  return todos;
+}
+function parseTodoStateJson(stateJson) {
+  if (!stateJson)
+    return null;
+  try {
+    return parseTodos(JSON.parse(stateJson));
+  } catch {
+    return null;
+  }
+}
+function setTodoSnapshot(sessionId, todos) {
+  const parsed = parseTodos(todos);
+  if (parsed === null)
+    return false;
+  snapshotsBySession.set(sessionId, { todos: parsed });
+  return true;
+}
+function seedTodoSnapshotFromStateJson(sessionId, stateJson) {
+  const parsed = parseTodoStateJson(stateJson);
+  if (parsed === null) {
+    snapshotsBySession.delete(sessionId);
+    return false;
+  }
+  snapshotsBySession.set(sessionId, { todos: parsed });
+  return true;
+}
+function getTodoSnapshot(sessionId) {
+  if (!sessionId)
+    return { todos: [] };
+  const snapshot = snapshotsBySession.get(sessionId);
+  return { todos: snapshot ? cloneTodos(snapshot.todos) : [] };
+}
+function clearTodoSnapshot(sessionId) {
+  snapshotsBySession.delete(sessionId);
+}
+function getSessionId(ctx) {
+  try {
+    const id = ctx.sessionManager?.getSessionId?.();
+    return typeof id === "string" && id.length > 0 ? id : undefined;
+  } catch {
+    return;
+  }
+}
+function countTodos(todos) {
+  const counts = {
+    total: todos.length,
+    pending: 0,
+    inProgress: 0,
+    completed: 0,
+    cancelled: 0
+  };
+  for (const todo of todos) {
+    switch (todo.status) {
+      case "pending":
+        counts.pending++;
+        break;
+      case "in_progress":
+        counts.inProgress++;
+        break;
+      case "completed":
+        counts.completed++;
+        break;
+      case "cancelled":
+        counts.cancelled++;
+        break;
+    }
+  }
+  return counts;
+}
+function activeTitleCount(todos) {
+  return todos.filter((todo) => todo.status !== "completed").length;
+}
+function formatCounts(counts) {
+  if (counts.total === 0)
+    return "No todos";
+  const parts = [`${counts.completed}/${counts.total} completed`];
+  if (counts.inProgress > 0)
+    parts.push(`${counts.inProgress} in progress`);
+  if (counts.pending > 0)
+    parts.push(`${counts.pending} pending`);
+  if (counts.cancelled > 0)
+    parts.push(`${counts.cancelled} cancelled`);
+  return parts.join(" · ");
+}
+function formatTodoLine(todo, theme, options = {}) {
+  const glyph = theme.fg(STATUS_COLOR[todo.status], STATUS_GLYPH[todo.status]);
+  const id = options.showId && todo.id ? `${theme.fg("accent", `#${todo.id}`)} ` : "";
+  const color = todo.status === "pending" || todo.status === "in_progress" ? "text" : "dim";
+  let content = theme.fg(color, todo.content);
+  if (todo.status === "completed" || todo.status === "cancelled") {
+    content = theme.strikethrough(content);
+  }
+  return `${glyph} ${id}${content}`;
+}
+function formatCommandLine(todo) {
+  const id = todo.id ? `#${todo.id} ` : "";
+  return `  ${STATUS_GLYPH[todo.status]} ${id}${todo.content}`;
+}
+function lineComponent(renderLines) {
+  return {
+    render: renderLines,
+    invalidate() {}
+  };
+}
+function renderTodowriteCall(args, theme) {
+  const todos = parseTodos(args.todos) ?? [];
+  const active = activeTitleCount(todos);
+  const activeColor = active > 0 ? "warning" : "success";
+  const line = `${theme.fg("toolTitle", theme.bold("Todos"))} ${theme.fg("muted", "—")} ${theme.fg(activeColor, `${active} active`)}`;
+  return lineComponent((width) => [truncateToWidth(line, width, "…")]);
+}
+function renderTodowriteResult(result, theme) {
+  const detailsTodos = result.details?.todos;
+  let todos = parseTodos(detailsTodos);
+  if (todos === null && Array.isArray(result.content)) {
+    const text = result.content.find((part) => part !== null && typeof part === "object" && part.type === "text" && typeof part.text === "string")?.text;
+    if (text) {
+      try {
+        todos = parseTodos(JSON.parse(text));
+      } catch {}
+    }
+  }
+  const renderedTodos = todos ?? [];
+  return lineComponent((width) => {
+    if (renderedTodos.length === 0)
+      return [theme.fg("dim", "No todos")];
+    return renderedTodos.map((todo) => truncateToWidth(formatTodoLine(todo, theme, { showId: true }), width, "…"));
+  });
+}
+function registerTodosCommand(pi) {
+  pi.registerCommand(TODOS_COMMAND_NAME, {
+    description: "Show the current Magic Context todo list",
+    handler: async (_args, ctx) => {
+      if (!ctx.hasUI) {
+        ctx.ui.notify("The /todos command requires interactive UI.", "error");
+        return;
+      }
+      const sessionId = getSessionId(ctx);
+      const todos = getTodoSnapshot(sessionId).todos;
+      if (todos.length === 0) {
+        ctx.ui.notify("No todos yet.", "info");
+        return;
+      }
+      const counts = countTodos(todos);
+      const lines = [formatCounts(counts)];
+      const groups = [
+        ["pending", "── Pending ──"],
+        ["in_progress", "── In Progress ──"],
+        ["completed", "── Completed ──"],
+        ["cancelled", "── Cancelled ──"]
+      ];
+      for (const [status, heading] of groups) {
+        const group = todos.filter((todo) => todo.status === status);
+        if (group.length === 0)
+          continue;
+        lines.push(heading);
+        for (const todo of group)
+          lines.push(formatCommandLine(todo));
+      }
+      ctx.ui.notify(lines.join(`
+`), "info");
+    }
+  });
+}
+function todoKey(todo, index) {
+  return todo.id ? `id:${todo.id}` : `pos:${index}:${todo.content}`;
+}
+function isOverlayLive(todo) {
+  return todo.status === "pending" || todo.status === "in_progress";
+}
+
+class TodoOverlay {
+  uiCtx;
+  sessionId;
+  widgetRegistered = false;
+  tui;
+  completedTaskIdsPendingHide = new Set;
+  hiddenCompletedTaskIds = new Set;
+  lastTodoKeys = new Set;
+  setUICtx(sessionId, ctx) {
+    if (ctx !== this.uiCtx || sessionId !== this.sessionId) {
+      this.uiCtx = ctx;
+      this.sessionId = sessionId;
+      this.widgetRegistered = false;
+      this.tui = undefined;
+      this.resetCompletedDisplayState();
+    }
+  }
+  update(sessionId = this.sessionId) {
+    if (!this.uiCtx || !this.sessionId || sessionId !== this.sessionId)
+      return;
+    const todos = getTodoSnapshot(this.sessionId).todos;
+    this.pruneCompletedDisplayState(todos);
+    const visible = this.selectOverlayTodos(todos);
+    if (visible.length === 0) {
+      if (this.widgetRegistered) {
+        this.uiCtx.setWidget(WIDGET_KEY, undefined);
+        this.widgetRegistered = false;
+        this.tui = undefined;
+      }
+      return;
+    }
+    if (!this.widgetRegistered) {
+      this.uiCtx.setWidget(WIDGET_KEY, (tui, theme) => {
+        this.tui = tui;
+        return {
+          render: (width) => this.renderWidget(theme, width),
+          invalidate: () => {
+            this.widgetRegistered = false;
+            this.tui = undefined;
+          }
+        };
+      }, { placement: "aboveEditor" });
+      this.widgetRegistered = true;
+    } else {
+      this.tui?.requestRender();
+    }
+  }
+  resetCompletedDisplayState() {
+    this.completedTaskIdsPendingHide.clear();
+    this.hiddenCompletedTaskIds.clear();
+    this.lastTodoKeys.clear();
+  }
+  hideCompletedTasksFromPreviousTurn() {
+    if (this.completedTaskIdsPendingHide.size === 0)
+      return;
+    for (const taskId of this.completedTaskIdsPendingHide) {
+      this.hiddenCompletedTaskIds.add(taskId);
+    }
+    this.completedTaskIdsPendingHide.clear();
+    this.update();
+  }
+  disposeSession(sessionId) {
+    if (this.sessionId !== sessionId)
+      return;
+    this.dispose();
+  }
+  dispose() {
+    if (this.uiCtx)
+      this.uiCtx.setWidget(WIDGET_KEY, undefined);
+    this.widgetRegistered = false;
+    this.tui = undefined;
+    this.uiCtx = undefined;
+    this.sessionId = undefined;
+    this.resetCompletedDisplayState();
+  }
+  pruneCompletedDisplayState(todos) {
+    const currentKeys = new Set(todos.map((todo, index) => todoKey(todo, index)));
+    const hasSharedKeys = [...currentKeys].some((key) => this.lastTodoKeys.has(key));
+    if (this.lastTodoKeys.size > 0 && currentKeys.size > 0 && !hasSharedKeys) {
+      this.resetCompletedDisplayState();
+    }
+    this.lastTodoKeys = currentKeys;
+    const completedKeys = new Set(todos.map((todo, index) => ({ todo, key: todoKey(todo, index) })).filter(({ todo }) => todo.status === "completed").map(({ key }) => key));
+    for (const taskId of this.completedTaskIdsPendingHide) {
+      if (!completedKeys.has(taskId))
+        this.completedTaskIdsPendingHide.delete(taskId);
+    }
+    for (const taskId of this.hiddenCompletedTaskIds) {
+      if (!completedKeys.has(taskId))
+        this.hiddenCompletedTaskIds.delete(taskId);
+    }
+  }
+  selectOverlayTodos(todos) {
+    return todos.map((todo, index) => ({ todo, key: todoKey(todo, index) })).filter(({ todo, key }) => {
+      if (isOverlayLive(todo))
+        return true;
+      return todo.status === "completed" && !this.hiddenCompletedTaskIds.has(key);
+    });
+  }
+  renderWidget(theme, width) {
+    if (!this.sessionId)
+      return [];
+    const todos = getTodoSnapshot(this.sessionId).todos;
+    this.pruneCompletedDisplayState(todos);
+    const overlayTodos = this.selectOverlayTodos(todos);
+    if (overlayTodos.length === 0)
+      return [];
+    const counts = countTodos(todos);
+    const hasInProgress = counts.inProgress > 0;
+    const headingColor = hasInProgress ? "accent" : "dim";
+    const headingIcon = hasInProgress ? "●" : "○";
+    const heading = `${theme.fg(headingColor, headingIcon)} ${theme.fg(headingColor, "Todos")} ${theme.fg("muted", "—")} ${theme.fg("dim", formatCounts(counts))}`;
+    const truncate2 = (line) => truncateToWidth(line, width, "…");
+    const lines = [truncate2(heading)];
+    const hasTruncatedTail = overlayTodos.length > MAX_OVERLAY_CONTENT_ROWS;
+    const visibleRows = hasTruncatedTail ? MAX_OVERLAY_CONTENT_ROWS - 1 : MAX_OVERLAY_CONTENT_ROWS;
+    const truncatedTail = Math.max(0, overlayTodos.length - visibleRows);
+    const visible = overlayTodos.slice(0, visibleRows);
+    for (const [index, { todo, key }] of visible.entries()) {
+      const isLast = index === visible.length - 1 && truncatedTail === 0;
+      const branch = theme.fg("dim", isLast ? "└─" : "├─");
+      lines.push(truncate2(`${branch} ${formatTodoLine(todo, theme, { showId: true })}`));
+      if (todo.status === "completed")
+        this.completedTaskIdsPendingHide.add(key);
+    }
+    if (truncatedTail > 0) {
+      lines.push(truncate2(`${theme.fg("dim", "└─")} ${theme.fg("dim", `+${truncatedTail} more`)}`));
+    }
+    return lines;
+  }
+}
+function registerTodoOverlay(pi, deps) {
+  const overlay = new TodoOverlay;
+  pi.on("session_start", async (_event, ctx) => {
+    const sessionId = getSessionId(ctx);
+    if (!sessionId)
+      return;
+    seedTodoSnapshotFromStateJson(sessionId, deps.readLastTodoState(sessionId));
+    if (!ctx.hasUI)
+      return;
+    overlay.setUICtx(sessionId, ctx.ui);
+    overlay.update(sessionId);
+  });
+  pi.on("agent_start", async () => {
+    overlay.hideCompletedTasksFromPreviousTurn();
+  });
+  pi.on("session_shutdown", async (_event, ctx) => {
+    const sessionId = getSessionId(ctx);
+    if (!sessionId)
+      return;
+    clearTodoSnapshot(sessionId);
+    overlay.disposeSession(sessionId);
+  });
+  pi.on("session_before_switch", async (_event, ctx) => {
+    const sessionId = getSessionId(ctx);
+    if (!sessionId)
+      return;
+    clearTodoSnapshot(sessionId);
+    overlay.disposeSession(sessionId);
+  });
+  return overlay;
+}
+function registerTodoStateLifecycle(pi, deps) {
+  pi.on("session_start", async (_event, ctx) => {
+    const sessionId = getSessionId(ctx);
+    if (!sessionId)
+      return;
+    seedTodoSnapshotFromStateJson(sessionId, deps.readLastTodoState(sessionId));
+  });
+  pi.on("session_shutdown", async (_event, ctx) => {
+    const sessionId = getSessionId(ctx);
+    if (sessionId)
+      clearTodoSnapshot(sessionId);
+  });
+  pi.on("session_before_switch", async (_event, ctx) => {
+    const sessionId = getSessionId(ctx);
+    if (sessionId)
+      clearTodoSnapshot(sessionId);
+  });
+}
+
 // src/tools/todowrite.ts
 var STATUS_VALUES = [
   "pending",
@@ -175646,29 +176093,20 @@ var TodowriteParams = exports_typebox.Object({
     description: "Replace the current task list with this complete set of todos. Include every task you intend to track this turn — pending, in_progress, completed, or cancelled — because the list overwrites previous state."
   })
 });
-var TOOL_DESCRIPTION = [
-  "Manage your task list for this session.",
-  "",
-  "Use this tool to plan multi-step work, track in-flight tasks, and",
-  "mark progress as you complete steps. Pass the COMPLETE updated list",
-  "of todos every time — this tool replaces the prior list rather than",
-  "appending to it.",
-  "",
-  "Task states:",
-  "  - pending: not started yet",
-  "  - in_progress: currently working on (limit to ONE task at a time)",
-  "  - completed: finished successfully",
-  "  - cancelled: no longer needed",
-  "",
-  "Use this tool proactively for non-trivial work spanning 3+ steps.",
-  "Skip it for single-shot answers or trivial 1-2 step tasks."
-].join(`
-`);
+var PROMPT_SNIPPET = "Manage a task list to track multi-step progress";
+var PROMPT_GUIDELINES = [
+  "Use `todowrite` for non-trivial work spanning 3+ steps, when the user gives you multiple tasks, or when you need to track progress across a verify/fix loop. Skip it for single-shot answers or trivial one-step work.",
+  "Pass the COMPLETE updated todo list every time. This tool replaces the prior list rather than appending to it, so include pending, in_progress, completed, and cancelled tasks that should remain visible.",
+  "When starting a task, mark exactly one todo `in_progress` before doing the work. Mark items `completed` immediately when done; use `cancelled` only for work that is no longer needed.",
+  "Never mark a todo completed if verification is failing, implementation is partial, or an unresolved blocker remains. Keep it `in_progress` and add or update a todo for the blocker instead."
+];
 function createTodowriteTool() {
   return {
-    name: "todowrite",
+    name: TODO_TOOL_NAME,
     label: "Todos",
-    description: TOOL_DESCRIPTION,
+    description: "Manage the session task list.",
+    promptSnippet: PROMPT_SNIPPET,
+    promptGuidelines: PROMPT_GUIDELINES,
     parameters: TodowriteParams,
     async execute(_toolCallId, params, _signal, _onUpdate, _ctx) {
       const todos = params.todos ?? [];
@@ -175687,6 +176125,12 @@ function createTodowriteTool() {
           truncated: false
         }
       };
+    },
+    renderCall(args, theme) {
+      return renderTodowriteCall(args, theme);
+    },
+    renderResult(result, _opts, theme) {
+      return renderTodowriteResult(result, theme);
     }
   };
 }
@@ -175712,15 +176156,22 @@ function registerMagicContextTools(pi, opts) {
   if (!opts.sessionScopedToolsDisabled) {
     pi.registerTool(createCtxNoteTool({
       db: opts.db,
-      dreamerEnabled: opts.dreamerEnabled ?? false
+      dreamerEnabled: opts.dreamerEnabled ?? false,
+      resolveDreamerEnabled: opts.resolveDreamerEnabled
     }));
     pi.registerTool(createCtxExpandTool({ db: opts.db }));
   }
-  pi.registerTool(createTodowriteTool());
+  if (opts.todowriteEnabled !== false) {
+    pi.registerTool(createTodowriteTool());
+    if (opts.todowriteCommandEnabled !== false) {
+      registerTodosCommand(pi);
+    }
+  }
   if (!opts.sessionScopedToolsDisabled) {
     pi.registerTool(createCtxReduceTool({
       db: opts.db,
-      protectedTags: opts.protectedTags ?? 20
+      protectedTags: opts.protectedTags ?? 20,
+      resolveProtectedTags: opts.resolveProtectedTags
     }));
   }
 }
@@ -175751,9 +176202,11 @@ function magicContextSubagentExtension(pi) {
         ensureProjectRegistered: ensureProjectRegisteredFromPiDirectory,
         memoryToolEnabled: dreamerActionsEnabled,
         allowDreamerActions: dreamerActionsEnabled,
-        sessionScopedToolsDisabled: true
+        sessionScopedToolsDisabled: true,
+        todowriteEnabled: cfg.todowrite.enabled !== false,
+        todowriteCommandEnabled: false
       });
-      log(`[pi-subagent] registered tools: ctx_search${dreamerActionsEnabled ? ", ctx_memory" : ""}` + ` (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 !== 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})`);
     } catch (err5) {
       const message = err5 instanceof Error ? err5.message : String(err5);
       log(`[pi-subagent] startup failed: ${message}`);

package/package.json

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