@zosmaai/pi-llm-wiki 0.10.6 → 0.10.7

package/extensions/llm-wiki/lib/guardrails.ts

@@ -1,3 +1,4 @@
+import { resolve, sep } from "node:path";
 import { isToolCallEventType } from "@mariozechner/pi-coding-agent";
 import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
 import { scheduleReindex } from "./indexing.js";
@@ -11,6 +12,39 @@ import { isProtectedPath, resolveVaultPaths } from "./utils.js";
 
 let pendingRebuild = false;
 
+const PATCH_HEADER = /^\[([^#\r\n]+)#[0-9A-F]{4}\]$/gm;
+
+function collectMutationPaths(input: unknown, seen: WeakSet<object>): string[] {
+  if (typeof input === "string") {
+    return Array.from(input.matchAll(PATCH_HEADER), ([, target]) => target);
+  }
+  if (!input || typeof input !== "object" || seen.has(input)) return [];
+
+  seen.add(input);
+  if (Array.isArray(input)) return input.flatMap((value) => collectMutationPaths(value, seen));
+
+  const { path, ...nested } = input as Record<string, unknown>;
+  if (typeof path === "string" && path.length > 0) return [path];
+
+  return Object.values(nested).flatMap((value) => collectMutationPaths(value, seen));
+}
+
+/** Return every file path targeted by a write or patch-shaped edit input. */
+export function extractMutationPaths(input: unknown): string[] {
+  return [...new Set(collectMutationPaths(input, new WeakSet()))];
+}
+
+/** True when a write or patch-shaped edit targets a page in the wiki directory. */
+export function hasWikiMutation(input: unknown, wikiPath: string): boolean {
+  const resolvedWikiPath = resolve(wikiPath);
+  return extractMutationPaths(input).some((path) => {
+    const resolvedPath = resolve(path);
+    return (
+      resolvedPath === resolvedWikiPath || resolvedPath.startsWith(`${resolvedWikiPath}${sep}`)
+    );
+  });
+}
+
 /** Install guardrails on the extension API. */
 export function installGuardrails(pi: ExtensionAPI, runtime?: Runtime): void {
   // Block direct edits to raw/ and meta/
@@ -25,11 +59,17 @@ export function installGuardrails(pi: ExtensionAPI, runtime?: Runtime): void {
     }
 
     if (isToolCallEventType("edit", event)) {
-      const path = event.input.path as string;
+      const targetPaths = extractMutationPaths(event.input);
+      if (targetPaths.length === 0) {
+        return { block: true, reason: "Cannot determine the files targeted by this edit." };
+      }
+
       const paths = resolveVaultPaths(process.cwd());
-      const check = isProtectedPath(path, paths);
-      if (check.protected) {
-        return { block: true, reason: check.reason };
+      for (const path of targetPaths) {
+        const check = isProtectedPath(path, paths);
+        if (check.protected) {
+          return { block: true, reason: check.reason };
+        }
       }
     }
   });
@@ -37,10 +77,8 @@ export function installGuardrails(pi: ExtensionAPI, runtime?: Runtime): void {
   // Track wiki edits for auto-rebuild
   pi.on("tool_result", async (event) => {
     if (event.toolName === "write" || event.toolName === "edit") {
-      const path = event.input.path as string;
       const paths = resolveVaultPaths(process.cwd());
-      const wikiPath = `${paths.wiki}/`;
-      if (path?.startsWith(wikiPath)) {
+      if (hasWikiMutation(event.input, paths.wiki)) {
         pendingRebuild = true;
       }
     }

package/extensions/llm-wiki/lib/tools.ts

@@ -1,5 +1,5 @@
 import { existsSync, mkdirSync, readFileSync, readdirSync, writeFileSync } from "node:fs";
-import { join } from "node:path";
+import { join, relative } from "node:path";
 import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
 import { Type } from "typebox";
 import { launchEmbedPages, reindexEmbeddings, resolveEmbedder } from "./embeddings.js";
@@ -390,7 +390,11 @@ export function registerWikiIngest(pi: ExtensionAPI, runtime?: Runtime): void {
         const manifestPath = join(paths.rawSources, id, "manifest.json");
         const extracted = existsSync(extractedPath) ? readFileSync(extractedPath, "utf-8") : "";
         const manifest = readJson<Record<string, unknown>>(manifestPath, {});
-        return { id, extracted, manifest };
+        // Vault-relative path used in tool messages so the read tool can open
+        // the file from the vault root (fix #101: agent previously got
+        // "raw/sources/..." and failed on new-layout vaults).
+        const relRaw = relative(paths.root, paths.rawSources);
+        return { id, extracted, manifest, relRaw };
       });
 
       // ── Background synthesis (issue #65) ──────────────────
@@ -474,7 +478,7 @@ export function registerWikiIngest(pi: ExtensionAPI, runtime?: Runtime): void {
                 [
                   `- **${s.id}**: ${s.manifest.title || s.id}`,
                   `  - Extracted: ${s.extracted.length} chars`,
-                  `  - Read: \`raw/sources/${s.id}/extracted.md\``,
+                  `  - Read: \`${s.relRaw}/${s.id}/extracted.md\``,
                 ].join("\n"),
               ),
               "",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@zosmaai/pi-llm-wiki",
-  "version": "0.10.6",
+  "version": "0.10.7",
   "description": "Self-maintaining LLM Wiki for Pi — Karpathy-pattern knowledge base with immutable source capture, automated ingestion, search, linting, and Obsidian-compatible vault. auto-updating personal & company wiki.",
   "keywords": [
     "pi",

package/prompts/wiki-ingest.md

@@ -19,7 +19,7 @@ $ARGUMENTS
 2. If the tool reports "All sources ingested", inform the user and stop.
 3. **If the tool reports it is ingesting in the background**, the synthesis sub-agent is handling those sources on the configured task model. Do NOT synthesize them yourself — just report which sources were dispatched and stop. (You'll be notified as each completes.)
 4. **Otherwise** (the tool returned extracted content — background unavailable or `background=false`), for each source in the returned batch:
-   a. Read the extracted text from `raw/sources/<SOURCE_ID>/extracted.md`
+   a. Read the extracted text from `.llm-wiki/raw/sources/<SOURCE_ID>/extracted.md`
    b. Update the skeleton source page in `wiki/sources/` with a proper summary, key entities, and concepts
    c. Use `wiki_ensure_page(type=entity, title=<name>)` for each new entity (people, orgs, tools, products)
    d. Use `wiki_ensure_page(type=concept, title=<name>)` for each new concept (ideas, patterns, frameworks)