pi-brain 0.1.5 → 0.1.6

package/README.md

@@ -6,7 +6,7 @@
 
 Versioned memory for the [pi coding agent](https://github.com/badlogic/pi-mono). Agents commit decisions and reasoning to a `.memory/` directory, preserving context across sessions, compactions, and model switches.
 
-Credit: This project is my implementation for Pi of this paper [Git Context Controller: Manage the Context of LLM-based Agents like Git](https://arxiv.org/html/2508.00031v1) 
+Credit: This project is my implementation for Pi of this paper [Git Context Controller: Manage the Context of LLM-based Agents like Git](https://arxiv.org/html/2508.00031v1)
 
 ## Getting Started
 
@@ -16,7 +16,7 @@ pi install npm:pi-brain
 
 Open pi in any project and say "initialize Brain" (or run `/skill:brain`). The agent creates `.memory/` and starts remembering.
 
-That's it. The agent decides when to commit, branch, and merge — you don't **need** to manage anything. However, you can always prompt the agent to remember something specific if you'd like. 
+That's it. The agent decides when to commit, branch, and merge — you don't **need** to manage anything. However, you can always prompt the agent to remember something specific if you'd like.
 
 ## How It Works
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-brain",
-  "version": "0.1.5",
+  "version": "0.1.6",
   "description": "Versioned memory extension for the pi coding agent",
   "keywords": [
     "agent-memory",

package/skills/brain/SKILL.md

@@ -78,13 +78,21 @@ then produces the structured commit entry. You just provide a good `summary` str
 
 ## After Every Commit
 
-**Update `.memory/main.md` to reflect the current state.** The roadmap is the first
+**Re-read `.memory/main.md` and rewrite stale sections.** The roadmap is the first
 thing a new session reads — if it's stale, every future session starts with a wrong
-picture. After `memory_commit` returns, review and update:
+picture. This means genuine review, not mechanical edits.
 
-- **Current State** section — reflect what's actually true now
-- **Key Decisions Made** — add any new decisions from this commit
-- **Milestones** — move completed items, add new planned ones
+After `memory_commit` returns:
+
+1. Read `.memory/main.md` in full.
+2. Rewrite the **Current State** section to describe what is true _right now_ — remove
+   historical context that has migrated to Key Decisions or Milestones.
+3. Add any new entries to **Key Decisions Made**.
+4. Move completed items in **Milestones** and add new planned ones.
+5. Remove or rewrite anything a new reader would find misleading.
+
+The goal is curation, not accumulation. A reader of Current State should understand
+the project as it exists today without wading through history.
 
 For trivial commits that don't change the project's state, decisions, or milestones
 (e.g., minor refactors, typo fixes), you can pass `update_roadmap: false` to skip

package/skills/brain/templates/agents-md.md

@@ -58,4 +58,4 @@ Call `memory_commit` when one of these is true:
 - **Rolling summaries**: Each commit re-synthesizes all prior progress
 - **No direct log.md writes**: The extension maintains log.md automatically
 - **Status is automatic**: Memory status is injected at session start and appended to tool results (compact/truncated when large; use `read .memory/main.md` for full roadmap)
-- **Keep main.md current**: After every commit, update `.memory/main.md` to reflect the new state — current state, decisions, and milestones. The roadmap is the first thing new sessions read; stale roadmaps cause wrong orientation. For trivial commits that don't change project state (e.g., minor refactors), pass `update_roadmap: false` to skip the reminder.
+- **Keep main.md current**: After every commit, re-read `.memory/main.md` in full and rewrite stale sections. Current State should describe what is true _right now_ — remove historical context that belongs in Key Decisions or Milestones. The goal is curation, not accumulation. For trivial commits that don't change project state, pass `update_roadmap: false` to skip the reminder.

package/src/index.ts

@@ -101,7 +101,7 @@ function resolveSkillPath(): string {
 export default function activate(pi: ExtensionAPI) {
   let state: MemoryState | null = null;
   let branchManager: BranchManager | null = null;
-  let statusInjected = false;
+  let frozenStatusSnapshot: string | null = null;
 
   function tryLoad(ctx: ExtensionContext): boolean {
     if (isMemoryReady(state, branchManager)) {
@@ -230,7 +230,7 @@ export default function activate(pi: ExtensionAPI) {
     state = new MemoryState(ctx.cwd);
     state.load();
     branchManager = new BranchManager(ctx.cwd);
-    statusInjected = false;
+    frozenStatusSnapshot = null;
 
     if (!state.isInitialized) {
       setBrainFooterStatus(ctx, state, branchManager);
@@ -253,7 +253,7 @@ export default function activate(pi: ExtensionAPI) {
   });
 
   pi.on("session_switch", (_event, ctx) => {
-    statusInjected = false;
+    frozenStatusSnapshot = null;
 
     state = new MemoryState(ctx.cwd);
     state.load();
@@ -266,11 +266,7 @@ export default function activate(pi: ExtensionAPI) {
     setBrainFooterStatus(ctx, state, branchManager);
   });
 
-  pi.on("before_agent_start", (_event: BeforeAgentStartEvent, ctx) => {
-    if (statusInjected) {
-      return;
-    }
-
+  pi.on("before_agent_start", (event: BeforeAgentStartEvent, ctx) => {
     if (
       !tryLoad(ctx) ||
       !isMemoryReady(state, branchManager) ||
@@ -279,25 +275,21 @@ export default function activate(pi: ExtensionAPI) {
       return;
     }
 
-    statusInjected = true;
+    if (frozenStatusSnapshot === null) {
+      frozenStatusSnapshot = buildStatusView(state, branchManager, ctx.cwd, {
+        compact: true,
+        roadmapCharLimit: BEFORE_AGENT_START_ROADMAP_CHAR_LIMIT,
+        branchLimit: BEFORE_AGENT_START_BRANCH_LIMIT,
+      });
+    }
 
-    const status = buildStatusView(state, branchManager, ctx.cwd, {
-      compact: true,
-      roadmapCharLimit: BEFORE_AGENT_START_ROADMAP_CHAR_LIMIT,
-      branchLimit: BEFORE_AGENT_START_BRANCH_LIMIT,
-    });
     return {
-      message: {
-        customType: "brain-status",
-        content: status,
-        display: true,
-        details: {},
-      },
+      systemPrompt: `${event.systemPrompt}\n\n${frozenStatusSnapshot}`,
     };
   });
 
   pi.on("session_compact", () => {
-    statusInjected = false;
+    frozenStatusSnapshot = null;
   });
 
   pi.on("resources_discover", () => ({

package/src/memory-commit.ts

@@ -65,7 +65,7 @@ export function finalizeMemoryCommit(
   const roadmapReminder =
     updateRoadmap === false
       ? ""
-      : "\n\n**Action required:** Update `.memory/main.md` to reflect this commit — current state, decisions, and milestones. The roadmap is the first thing new sessions read.";
+      : "\n\n**Action required:** Re-read `.memory/main.md` in full and rewrite stale sections. Current State should describe what is true right now — curate, don't just append.";
 
   return `${resultText}\n\n${status}${roadmapReminder}`;
 }