@zosmaai/pi-llm-wiki 0.10.5 → 0.10.7

package/README.md

@@ -440,6 +440,13 @@ Thanks to everyone who has contributed! This list is regenerated automatically b
                     <sub><b>Akshay</b></sub>
                 </a>
             </td>
+            <td align="center">
+                <a href="https://github.com/danielnaab">
+                    <img src="https://avatars.githubusercontent.com/u/136512?v=4" width="64;" alt="danielnaab"/>
+                    <br />
+                    <sub><b>Daniel Naab</b></sub>
+                </a>
+            </td>
             <td align="center">
                 <a href="https://github.com/mystery4f">
                     <img src="https://avatars.githubusercontent.com/u/40482524?v=4" width="64;" alt="mystery4f"/>

package/extensions/llm-wiki/index.ts

@@ -2,7 +2,7 @@ import { existsSync, readFileSync, writeFileSync } from "node:fs";
 import { basename, join } from "node:path";
 import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
 import { installGuardrails } from "./lib/guardrails.js";
-import { buildAgentStartInjection } from "./lib/inject.js";
+import { buildAgentStartInjection, normalizeSystemPrompt } from "./lib/inject.js";
 import { registerWikiModelCommand } from "./lib/model-command.js";
 import {
   buildSessionNotice,
@@ -303,13 +303,12 @@ Then call wiki_bootstrap with the inferred topic and mode to finalize the setup.
 
     // Split into a cache-stable system prompt (static footer only) and a
     // volatile tail message (issue #92). See lib/inject.ts for the contract.
-    const { systemPrompt, message } = buildAgentStartInjection(event.systemPrompt || "", [
-      dynamicContext,
-    ]);
+    const priorSystemPrompt = normalizeSystemPrompt(event.systemPrompt);
+    const { systemPrompt, message } = buildAgentStartInjection(priorSystemPrompt, [dynamicContext]);
 
     // Only claim a systemPrompt change when the footer actually altered the
     // string (a carry-forward turn already carries it, so this no-ops).
-    const systemPromptChanged = systemPrompt !== event.systemPrompt;
+    const systemPromptChanged = systemPrompt !== priorSystemPrompt;
     if (!systemPromptChanged && !message) return;
     return {
       ...(systemPromptChanged ? { systemPrompt } : {}),

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/inject.ts

@@ -33,6 +33,14 @@ export function appendWikiStatus(systemPrompt: string): string {
   return `${base}\n\n${WIKI_STATUS_BLOCK}`;
 }
 
+/** Normalize upstream Pi and OMP system-prompt representations. */
+export function normalizeSystemPrompt(
+  systemPrompt: string | readonly string[] | null | undefined,
+): string {
+  if (Array.isArray(systemPrompt)) return systemPrompt.join("\n\n");
+  return typeof systemPrompt === "string" ? systemPrompt : "";
+}
+
 /** customType of the hidden tail message carrying volatile per-turn context. */
 export const WIKI_RECALL_MESSAGE_TYPE = "wiki-recall-context";
 
@@ -69,10 +77,10 @@ export interface AgentStartInjection {
  * Pure and side-effect free — see test/agent-start-injection.test.ts.
  */
 export function buildAgentStartInjection(
-  baseSystemPrompt: string,
+  baseSystemPrompt: string | readonly string[] | null | undefined,
   dynamicBlocks: Array<string | undefined>,
 ): AgentStartInjection {
-  const systemPrompt = appendWikiStatus(baseSystemPrompt);
+  const systemPrompt = appendWikiStatus(normalizeSystemPrompt(baseSystemPrompt));
   const content = dynamicBlocks
     .map((b) => b?.trim())
     .filter((b): b is string => Boolean(b))

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.5",
+  "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)