@zosmaai/pi-llm-wiki 0.11.5 → 0.11.7

package/dist/extensions/llm-wiki/lib/tools.js

@@ -6,11 +6,11 @@ import { launchEmbedPages, reindexEmbeddings, resolveEmbedder } from "./embeddin
 import { scheduleReindex } from "./indexing.js";
 import { runIngestSynthesis } from "./ingest-worker.js";
 import { createKnowledgeDocument, serializeKnowledgeDocument, writeKnowledgeDocumentFile, } from "./knowledge-document.js";
-import { buildResolvedBacklinks } from "./knowledge-links.js";
+import { applyWikilinkGate, buildResolvedBacklinks, buildWikilinkIndex, } from "./knowledge-links.js";
 import { repairLegacyKnowledgeDocuments } from "./legacy-repair.js";
 import { appendEvent, rebuildMetadata, rebuildMetadataLight } from "./metadata.js";
 import { captureFile, captureText, captureUrl } from "./source-packet.js";
-import { parseModelRef } from "./task-config.js";
+import { loadTaskConfig, parseModelRef, resolveWikilinkValidation } from "./task-config.js";
 import { detectVaultFormat, fmtDate, getVaultPaths, readJson, resolveVaultPaths, slugify, writeJson, } from "./utils.js";
 import { assertWritableVault, compareCodePoint, discoverKnowledgeDocuments, inspectVaultFormat, inspectWritableVault, } from "./vault-format.js";
 import { getWikiStatus, searchRegistry } from "./wiki-service.js";
@@ -328,6 +328,7 @@ export function registerWikiIngest(pi, runtime) {
                                 manifest: s.manifest,
                                 extracted: s.extracted,
                                 synthesisLanguage: runtime.config.synthesisLanguage,
+                                wikilinkValidation: runtime.config.wikilinkValidation,
                             });
                             if (committed) {
                                 // Background semantic embeddings (#66): embed the pages this
@@ -341,8 +342,10 @@ export function registerWikiIngest(pi, runtime) {
                                 ];
                                 launchEmbedPages(runtime, launchCtx, paths, pageIds, `embed:ingest:${s.id}`);
                             }
+                            const wl = committed?.wikilinkDiagnostics?.length ?? 0;
+                            const wlNote = wl > 0 ? `, ${wl} wikilink issue${wl === 1 ? "" : "s"}` : "";
                             const summary = committed
-                                ? `LLM Wiki: ingested ${s.id} → ${committed.entitiesCreated.length} entit${committed.entitiesCreated.length === 1 ? "y" : "ies"}, ${committed.conceptsCreated.length} concept${committed.conceptsCreated.length === 1 ? "" : "s"}`
+                                ? `LLM Wiki: ingested ${s.id} → ${committed.entitiesCreated.length} entit${committed.entitiesCreated.length === 1 ? "y" : "ies"}, ${committed.conceptsCreated.length} concept${committed.conceptsCreated.length === 1 ? "" : "s"}${wlNote}`
                                 : `LLM Wiki: ${s.id} produced no synthesis`;
                             if (ctx.hasUI) {
                                 ctx.ui.notify(summary, committed ? "info" : "warning");
@@ -419,7 +422,7 @@ export function registerWikiEnsurePage(pi, runtime) {
         ],
         parameters: Type.Object({
             type: Type.String({
-                description: "Page type: entity | concept | synthesis | analysis | requirement | skill | case",
+                description: "Page type: entity | concept | synthesis | analysis | requirement | skill | case (built-in) or any user-defined type from llm-wiki.customTypes config",
             }),
             title: Type.String({ description: "Page title" }),
             content: Type.Optional(Type.String({ description: "Optional initial content (otherwise uses template)" })),
@@ -439,9 +442,8 @@ export function registerWikiEnsurePage(pi, runtime) {
                     isError: true,
                 };
             }
-            const type = params.type;
-            const slug = slugify(params.title);
-            const folderMap = {
+            const config = loadTaskConfig(ctx.cwd);
+            const builtInFolderMap = {
                 entity: "entities",
                 concept: "concepts",
                 synthesis: "syntheses",
@@ -450,6 +452,9 @@ export function registerWikiEnsurePage(pi, runtime) {
                 skill: "skills",
                 case: "cases",
             };
+            const folderMap = { ...builtInFolderMap, ...config.customTypes };
+            const type = params.type;
+            const slug = slugify(params.title);
             const folder = folderMap[type] || "concepts";
             const pagePath = join(paths.wiki, folder, `${slug}.md`);
             if (existsSync(pagePath)) {
@@ -459,7 +464,31 @@ export function registerWikiEnsurePage(pi, runtime) {
                 };
             }
             const today = fmtDate();
-            const body = params.content ?? buildPageBody(type, params.title);
+            let body = params.content ?? buildPageBody(type, params.title);
+            // Pre-write wikilink gate (#172): validate/normalize caller-supplied content.
+            const mode = resolveWikilinkValidation(loadTaskConfig(ctx.cwd));
+            let wikilinkIssues = [];
+            if (mode !== "off") {
+                const registry = readJson(join(paths.meta, "registry.json"), { pages: {} });
+                const gate = applyWikilinkGate(body, buildWikilinkIndex(Object.keys(registry.pages)), `${folder}/${slug}`, mode);
+                wikilinkIssues = gate.diagnostics.map((d) => d.message);
+                if (!gate.ok) {
+                    return {
+                        content: [
+                            {
+                                type: "text",
+                                text: `Rejected write — unresolved/ambiguous wikilinks:\n${wikilinkIssues
+                                    .map((m) => `- ${m}`)
+                                    .join("\n")}`,
+                            },
+                        ],
+                        details: { error: "link_validation", issues: wikilinkIssues },
+                        isError: true,
+                    };
+                }
+                if (mode === "normalize")
+                    body = gate.body;
+            }
             const doc = createKnowledgeDocument(`${folder}/${slug}.md`, {
                 type,
                 title: params.title,
@@ -483,9 +512,16 @@ export function registerWikiEnsurePage(pi, runtime) {
             else {
                 rebuildMetadataLight(paths);
             }
+            const gateNote = wikilinkIssues.length
+                ? `\n\n⚠️ ${wikilinkIssues.length} wikilink issue(s):\n${wikilinkIssues.map((m) => `- ${m}`).join("\n")}`
+                : "";
             return {
-                content: [{ type: "text", text: `✅ Created ${type} page: \`${pagePath}\`` }],
-                details: { path: pagePath, created: true },
+                content: [{ type: "text", text: `✅ Created ${type} page: \`${pagePath}\`${gateNote}` }],
+                details: {
+                    path: pagePath,
+                    created: true,
+                    wikilinkIssues,
+                },
             };
         },
     });
@@ -740,14 +776,14 @@ function runWikiLint(paths, autoFix) {
     }
     const discovery = discoverKnowledgeDocuments(paths);
     const pages = discovery.documents;
-    const knownIds = new Set(pages.map((page) => page.id));
+    const wikilinkIndex = buildWikilinkIndex(pages.map((page) => page.id));
     const inbound = Object.fromEntries(pages.map((page) => [page.id, 0]));
     const gapSources = new Map();
     const findings = [];
     let missingPages = 0;
     let contradictions = 0;
     for (const page of pages) {
-        const resolved = buildResolvedBacklinks(page.id, page.body, knownIds);
+        const resolved = buildResolvedBacklinks(page.id, page.body, wikilinkIndex);
         for (const target of resolved.targets)
             inbound[target]++;
         for (const unresolved of resolved.unresolved) {
@@ -757,6 +793,11 @@ function runWikiLint(paths, autoFix) {
             missingPages++;
             findings.push(`Missing page: ${unresolved.target} (in ${page.id})`);
         }
+        for (const d of resolved.diagnostics) {
+            if (d.code === "link_ambiguous") {
+                findings.push(d.message.replace("Ambiguous wikilink: ", "Ambiguous: "));
+            }
+        }
     }
     let orphans = 0;
     for (const page of pages) {

package/dist/mcp/index.js

@@ -13,6 +13,7 @@ import { join } from "node:path";
 import { McpServer } from "@modelcontextprotocol/server";
 import { StdioServerTransport } from "@modelcontextprotocol/server/stdio";
 import * as z from "zod/v4";
+import { loadTaskConfig, resolveWikilinkValidation, } from "../extensions/llm-wiki/lib/task-config.js";
 import { getVaultPaths, resolveVaultPaths } from "../extensions/llm-wiki/lib/utils.js";
 import { createExecApi } from "./exec.js";
 import { bootstrapOperation, captureSourceOperation, recallOperation, retroOperation, searchOperation, statusOperation, } from "./operations.js";
@@ -206,7 +207,7 @@ server.registerTool("wiki_retro", {
         };
     }
     const paths = getPaths();
-    const result = await retroOperation(paths, slug, title, body, category);
+    const result = await retroOperation(paths, slug, title, body, category, resolveWikilinkValidation(loadTaskConfig(process.cwd())));
     if (!result.ok) {
         return {
             content: [

package/dist/mcp/operations.js

@@ -5,11 +5,15 @@
  * used by Pi tools. No operation parses YAML, scans files, scores
  * registry entries, or builds page strings itself.
  */
+import { join } from "node:path";
 import { bootstrapVault } from "../extensions/llm-wiki/lib/bootstrap.js";
+import { applyWikilinkGate, buildWikilinkIndex, } from "../extensions/llm-wiki/lib/knowledge-links.js";
 import { rebuildMetadata } from "../extensions/llm-wiki/lib/metadata.js";
 import { searchWikiLayered } from "../extensions/llm-wiki/lib/recall.js";
 import { saveInsight } from "../extensions/llm-wiki/lib/retro.js";
 import { captureFile, captureText, captureUrl } from "../extensions/llm-wiki/lib/source-packet.js";
+import { resolveWikilinkValidation } from "../extensions/llm-wiki/lib/task-config.js";
+import { readJson } from "../extensions/llm-wiki/lib/utils.js";
 import { VaultWriteError, inspectVaultFormat, inspectWritableVault, } from "../extensions/llm-wiki/lib/vault-format.js";
 import { getWikiStatus, searchRegistry } from "../extensions/llm-wiki/lib/wiki-service.js";
 function projectionOutcome(projection) {
@@ -87,7 +91,7 @@ export async function statusOperation(paths) {
     };
 }
 /** Shared retro operation: validates vault then delegates to saveInsight. */
-export async function retroOperation(paths, slug, title, body, category) {
+export async function retroOperation(paths, slug, title, body, category, wikilinkValidation) {
     const vaultCheck = inspectWritableVault(paths);
     if (!vaultCheck.ok) {
         return {
@@ -95,8 +99,23 @@ export async function retroOperation(paths, slug, title, body, category) {
             diagnostics: vaultCheck.diagnostics.map((d) => ({ code: d.code, message: d.message })),
         };
     }
+    // Pre-write wikilink gate (#172): validate/normalize caller-supplied body.
+    const mode = resolveWikilinkValidation({ wikilinkValidation });
+    let gateBody = body;
+    if (mode !== "off") {
+        const registry = readJson(join(paths.meta, "registry.json"), { pages: {} });
+        const gate = applyWikilinkGate(body, buildWikilinkIndex(Object.keys(registry.pages)), `sources/${slug}`, mode);
+        if (!gate.ok) {
+            return {
+                ok: false,
+                diagnostics: gate.diagnostics.map((d) => ({ code: "link_validation", message: d.message })),
+            };
+        }
+        if (mode === "normalize")
+            gateBody = gate.body;
+    }
     try {
-        const result = saveInsight(paths, slug, title, body, category, { rebuild: false });
+        const result = saveInsight(paths, slug, title, gateBody, category, { rebuild: false });
         const projection = projectionOutcome(rebuildMetadata(paths));
         if (!projection.ok)
             return projection;

package/docs/configuration.md

@@ -58,6 +58,7 @@ All of the above are viewable and editable in the `/wiki-settings` TUI (persists
 | `taskModel`            | —          | Model for background tasks (`{ provider: "openai", id: "gpt-4o" }`) |
 | `synthesisLanguage`    | —          | BCP 47 language tag for ingest synthesis (e.g. `"ru"`, `"fr"`). When unset, synthesis defaults to English. |
 | `synthesisMaxTokens`     | 16384    | Max output tokens for ingest/synthesis runs (stored as a plain number)                        |
+| `wikilinkValidation`   | warn       | Pre-write wikilink gate for page writes (ingest, `wiki_ensure_page`, `wiki_retro`, MCP `wiki_retro`). `off` ignore, `warn` report, `normalize` rewrite resolvable links, `strict` block writes with unresolved/ambiguous links |
 | `trajectories`         | false      | Enable agent-trajectory working-memory                       |
 | `notices`              | true       | Show wiki activity notices in chat                           |
 | `ambientPersonalVault` | host-dependent | Let the personal vault act as the ambient vault in projects that have no wiki. `true` under pi, `false` under oh-my-pi — see below. |

package/docs/obsidian.md

@@ -2,20 +2,20 @@
 
 ## Setup
 
-1. Open `.llm-wiki/wiki/` as an Obsidian vault
-2. The extension generates `.llm-wiki/meta/index.md` as a browsable catalog
-3. `.llm-wiki/meta/backlinks.json` is available for graph plugins
+1. Open `.llm-wiki/` as an Obsidian vault
+2. Your wiki pages live in `wiki/` — Graph View and Backlinks work on these automatically
+3. The extension generates `meta/index.md` (browsable catalog) and `meta/backlinks.json` (link map) — both are read-only, regenerated on each rebuild
 
 ## Recommended Plugins
 
 - [Dataview](https://github.com/blacksmithgu/obsidian-dataview) — Query pages by frontmatter
-- [Graph View](https://obsidian.md) (built-in) — Visualize `[[wikilink]]` connections
+- [Graph View](https://obsidian.md) (built-in) — Visualize `[[wikilink]]` connections in `wiki/`
 - [Backlinks](https://obsidian.md) (built-in) — See inbound links
 
 ## Web Clipper
 
-Use [Obsidian Web Clipper](https://obsidian.md/clipper) to save articles directly into `.llm-wiki/raw/articles/`.
+Use [Obsidian Web Clipper](https://obsidian.md/clipper) to save articles directly into `raw/articles/`.
 
 ## Dataview Dashboard
 
-The extension creates `.llm-wiki/meta/index.md` with page listings. For custom dashboards, use Dataview queries against frontmatter fields like `type`, `domain`, `category`, `sources`.
+For custom dashboards, use Dataview queries against frontmatter fields like `type`, `domain`, `category`, `sources`. Query the `wiki/` directory for knowledge pages.