@zosmaai/pi-llm-wiki 0.9.2 → 0.9.3

package/extensions/llm-wiki/index.ts

@@ -2,6 +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 { appendWikiStatus } from "./lib/inject.js";
 import { registerWikiModelCommand } from "./lib/model-command.js";
 import {
   buildSessionNotice,
@@ -295,8 +296,7 @@ Then call wiki_bootstrap with the inferred topic and mode to finalize the setup.
 
     // Always inject a visible wiki status footer, even when empty
     // This ensures the model knows the wiki is active and can use it
-    injectedContext +=
-      "\n\n<wiki_status>LLM Wiki active — use wiki_recall for deeper search, wiki_observe to record observations, wiki_retro to save insights.</wiki_status>";
+    injectedContext = appendWikiStatus(injectedContext);
 
     if (injectedContext === event.systemPrompt) return;
     return { systemPrompt: injectedContext };

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

@@ -0,0 +1,27 @@
+/**
+ * System-prompt context injection primitives (issue #87).
+ *
+ * The `before_agent_start` hook augments the chained system prompt with a
+ * visible wiki-status footer. This module isolates that append so it can be
+ * unit-tested for idempotency — a turn that aborts (network error / ESC) and is
+ * retried can carry the prior injection forward, and a naive append stacks the
+ * footer 2x, 3x, ...
+ */
+
+/** The always-injected wiki-status footer (sans surrounding whitespace). */
+export const WIKI_STATUS_BLOCK =
+  "<wiki_status>LLM Wiki active — use wiki_recall for deeper search, wiki_observe to record observations, wiki_retro to save insights.</wiki_status>";
+
+/**
+ * Append the wiki-status footer to a system prompt — idempotently (issue #87).
+ *
+ * Strips any already-present footer (with its leading blank line) before
+ * appending exactly one. This makes the injection safe across aborted/retried
+ * agent starts that carry the prior injection forward in the chained system
+ * prompt, so the footer never stacks (2x, 3x, ...).
+ * See test/inject-idempotent.test.ts.
+ */
+export function appendWikiStatus(systemPrompt: string): string {
+  const base = systemPrompt.split(`\n\n${WIKI_STATUS_BLOCK}`).join("");
+  return `${base}\n\n${WIKI_STATUS_BLOCK}`;
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@zosmaai/pi-llm-wiki",
-  "version": "0.9.2",
+  "version": "0.9.3",
   "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",