@vshulcz/pi-deja 0.21.7 → 0.22.0

package/README.md

@@ -3,13 +3,13 @@
 English | [中文](https://github.com/vshulcz/deja-vu/blob/main/extensions/pi/docs/zh.md)
 
 pi remembers its own sessions. This extension answers the other question: what
-was done in the thirty-seven other coding agents on this machine — Claude Code,
+was done in the forty other coding agents on this machine — Claude Code,
 Codex, Cursor, Gemini, OpenClaw and Hermes among them — including the months
 before pi was installed.
 
 It runs [deja](https://github.com/vshulcz/deja-vu), a local Go binary that
 indexes the transcripts those agents already wrote to disk. No LLM, no
-embeddings, no network path unless you ask for one.
+embeddings, no network path besides a daily release check, which DEJA_OFFLINE=1 turns off.
 
 ## Install
 

package/index.ts

@@ -75,7 +75,10 @@ export default function (pi: any) {
     if (installerExtensionPaths(homedir()).some((p) => existsSync(p))) return;
   } catch {}
 
-  let injected = false;
+  // Once per session, not once per process: /new, a resume and a fork keep the
+  // process and switch the session, and a flag set on the first one left every
+  // later session with no digest.
+  const injected = new Set<string>();
   let toldBuilding = false;
   // The tool and compaction events carry no session id of their own, and
   // recall dedupes per session: without one it repeats itself and forgets
@@ -123,17 +126,17 @@ export default function (pi: any) {
 
   pi.on("before_agent_start", async (event: any, ctx: any) => {
     try {
-      if (!injected) {
+      const key = sessionKey(event, ctx);
+      if (key) session = key;
+      if (!injected.has(session)) {
         // The session goes with it: hook-context marks the one starting as
         // live, which keeps it out of its own MCP recall on this first turn
         // (#4394, as #4246 and #4273 did for Hermes and opencode).
-        const key = sessionKey(event, ctx);
-        if (key) session = key;
         const { context: digest, receipt } = contextText(
-          run(["hook-context"], JSON.stringify({ session_id: key, cwd: process.cwd() })),
+          run(["hook-context"], JSON.stringify({ session_id: session, cwd: process.cwd() })),
         );
         if (digest) {
-          injected = true;
+          injected.add(session);
           ctx.ui.setStatus("deja", "");
           // The receipt is what tells the user memory arrived; without it the
           // recall is invisible and reads as the model guessing.
@@ -152,9 +155,7 @@ export default function (pi: any) {
         }
         return;
       }
-      const key = sessionKey(event, ctx);
-      if (key) session = key;
-      const raw = run(["hook-prompt"], JSON.stringify({ prompt: event.prompt || "", session_id: key }));
+      const raw = run(["hook-prompt"], JSON.stringify({ prompt: event.prompt || "", session_id: session }));
       if (!raw) return;
       const resp = JSON.parse(raw);
       if (resp && resp.systemMessage) ctx.ui.notify(resp.systemMessage, "info");
@@ -183,15 +184,16 @@ export default function (pi: any) {
       // and pi has no handler that runs earlier whose return the model reads.
       // So the file's own history goes out here: what was decided about it,
       // from the sessions that decided it. deja answers once per session per
-      // fact, so re-reading the same file stays quiet.
-      if (event.toolName === "read") {
+      // fact, so re-reading the same file stays quiet. An edit or a write
+      // without a read before it gets the same line, for the edits after it.
+      if (event.toolName === "read" || event.toolName === "edit" || event.toolName === "write") {
         const path = String((event.input && (event.input.path || event.input.file_path)) || "");
         if (!path) return;
         const parts = Array.isArray(event.content) ? event.content : [];
-        const id = "read:" + String(event.toolCallId || "");
+        const id = "file:" + String(event.toolCallId || "");
         if (!(id in repaired)) {
           repaired[id] = run(["hook-tool", "--plain"], JSON.stringify({
-            tool_name: "read",
+            tool_name: event.toolName,
             tool_input: { file_path: path },
             session_id: session,
             cwd: process.cwd(),
@@ -230,12 +232,26 @@ export default function (pi: any) {
   // Compaction throws away the blocks this session was shown, and the list
   // that stops them repeating outlives them. session_compact fires once the
   // summary has replaced the history, which is exactly when the session should
-  // be allowed to see those blocks again.
+  // be allowed to see those blocks again. The session file keeps the turns the
+  // summary replaced, so deja reads them from it and the next prompt carries
+  // what the agent was in the middle of.
   pi.on("session_compact", async (_event: any, ctx: any) => {
     try {
       const key = sessionKey(_event, ctx);
       if (key) session = key;
-      run(["hook-precompact"], JSON.stringify({ session_id: session }));
+      const m = ctx && ctx.sessionManager;
+      const file = m && m.getSessionFile ? String(m.getSessionFile() || "") : "";
+      run(["hook-precompact"], JSON.stringify({ session_id: session, transcript_path: file, cwd: process.cwd() }));
+    } catch {}
+  });
+
+  // The session is over: quit, /new, /resume, /fork or a reload. Its live
+  // stamp goes with it, so the next session's MCP recall can answer with it
+  // now rather than twenty minutes from now. The id is the one held so far;
+  // on a switch ctx may already name the next session.
+  pi.on("session_shutdown", async (_event: any) => {
+    try {
+      if (session) run(["hook-session-end"], JSON.stringify({ session_id: session }), 2000);
     } catch {}
   });
 }

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@vshulcz/pi-deja",
-  "version": "0.21.7",
-  "description": "deja-vu memory for pi: the coding sessions already on this machine — pi, Claude Code, Codex, Cursor and 34 more — recalled at session start and on every prompt, searchable with /deja. A local index, no LLM.",
+  "version": "0.22.0",
+  "description": "deja-vu memory for pi: the coding sessions already on this machine — pi, Claude Code, Codex, Cursor and 37 more — recalled at session start and on every prompt, searchable with /deja. A local index, no LLM.",
   "type": "module",
   "files": [
     "index.ts",
@@ -45,7 +45,7 @@
     "node": ">=18"
   },
   "dependencies": {
-    "@vshulcz/deja-vu": "^0.21.7"
+    "@vshulcz/deja-vu": "^0.22.0"
   },
   "pi": {
     "extensions": [