pi-doc-injector 0.2.0 → 0.2.1

package/README.md

@@ -101,11 +101,18 @@ Injection is also skipped if the current context usage exceeds 80% of the token
 
 The extension uses a per-session injection model:
 
-- On `session_start`, the registry is rebuilt from scratch, resetting all `injected` flags.
+- On `session_start`, the registry scans `docs/` and indexes all valid documents.
 - Within a session, once a document is injected, it won't be re-injected automatically.
 - Use `/doc-inject reset` to manually reset all flags and allow docs to be injected again.
 - Use `/doc-inject list` to see which docs have been injected (✅) and which are pending (⬜).
 
+### Injection Timing
+
+- **User messages**: matched via the `input` event, injected before the assistant
+  responds — **same turn**, no delay.
+- **Assistant streaming**: if the assistant mentions a NEW keyword mid-response,
+  generation is aborted and restarted with the doc injected immediately.
+
 ### System Prompt Lifecycle
 
 Pi **reconstructs the system prompt from source files each turn** (verified against pi v0.70.6).

package/index.ts

@@ -66,6 +66,7 @@ export default async function docInjectorExtension(pi: ExtensionAPI) {
   // ---- State ----
   let config = loadConfig(process.cwd());
   let registry: DocRegistry | null = null;
+  let initRegistryPromise: Promise<void> | null = null;
   let enabled = true;
   let textBuffer = "";
   let pendingMatches = new Map<string, string[]>(); // filePath → matchedKeywords
@@ -98,17 +99,24 @@ export default async function docInjectorExtension(pi: ExtensionAPI) {
     );
   };
 
-  let lastInitTime = 0;
-
   // ---- Event: session_start ----
-  // Pi fires session_start twice on startup (both with reason "startup").
-  // Use a 2-second dedup window to skip the duplicate. Real session changes
-  // (/new, /resume, /fork) happen well outside this window.
-  pi.on("session_start", async (_event, ctx) => {
-    const now = Date.now();
-    if (now - lastInitTime < 100) return;
-    lastInitTime = now;
-    await initRegistry(ctx.cwd);
+  // Pi emits session_start for startup, reload, and real session transitions.
+  // Skip the reload variant because resources_discover will rebuild docs right
+  // after it, and deduplicate any overlapping non-reload inits.
+  pi.on("session_start", async (event, ctx) => {
+    if (event.reason === "reload") return;
+
+    if (initRegistryPromise) {
+      await initRegistryPromise;
+      return;
+    }
+
+    initRegistryPromise = initRegistry(ctx.cwd);
+    try {
+      await initRegistryPromise;
+    } finally {
+      initRegistryPromise = null;
+    }
   });
 
   const reloadRegistry = async (): Promise<number> => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-doc-injector",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "description": "Auto-inject relevant project documentation into Pi's LLM context based on keyword matching",
   "type": "module",
   "main": "./index.ts",