llm-kb 0.4.2 → 0.6.0

package/bin/cli.js

@@ -1,16 +1,21 @@
 #!/usr/bin/env node
 import {
-  buildIndex,
-  completeSimple,
-  continueKBSession,
-  createKBSession,
-  getNodeModulesPath
-} from "./chunk-DHOXVEIR.js";
+  buildIndex
+} from "./chunk-EZ7LPPEP.js";
+import {
+  appendToQueryLog,
+  buildTrace,
+  completeWithFallback,
+  createChat,
+  parseCitations,
+  query,
+  saveTrace,
+  updateWiki
+} from "./chunk-Y2764FFH.js";
+import "./chunk-3WBSKCCH.js";
 import "./chunk-3YMNGUZZ.js";
 import "./chunk-LDHOKBJA.js";
-import {
-  getModels
-} from "./chunk-5PYKQQLA.js";
+import "./chunk-5PYKQQLA.js";
 import "./chunk-EAQYK3U2.js";
 
 // src/cli.ts
@@ -126,6 +131,19 @@ ${p.text}`).join("\n\n---\n\n");
   };
   await writeFile(mdPath, markdown);
   await writeFile(jsonPath, JSON.stringify(bboxData, null, 2));
+  const pagesDir = join(outputDir, `${name}.pages`);
+  await mkdir(pagesDir, { recursive: true });
+  for (const pageData of bboxData.pages) {
+    const pageFile = join(pagesDir, `${pageData.page}.json`);
+    await writeFile(pageFile, JSON.stringify({
+      source: bboxData.source,
+      totalPages: bboxData.totalPages,
+      page: pageData.page,
+      width: pageData.width,
+      height: pageData.height,
+      textItems: pageData.textItems
+    }));
+  }
   return {
     name,
     mdPath,
@@ -140,7 +158,7 @@ ${p.text}`).join("\n\n---\n\n");
 import { watch } from "chokidar";
 import { extname as extname2, basename as basename2 } from "path";
 import chalk from "chalk";
-function startWatcher({ folder, sourcesDir, debounceMs = 2e3, authStorage, indexModel }) {
+function startWatcher({ folder, sourcesDir, debounceMs = 2e3, authStorage, indexModel, onSourcesChanged }) {
   let pendingFiles = [];
   let debounceTimer = null;
   async function processBatch() {
@@ -166,6 +184,12 @@ function startWatcher({ folder, sourcesDir, debounceMs = 2e3, authStorage, index
     try {
       await buildIndex(folder, sourcesDir, void 0, authStorage, indexModel);
       console.log(chalk.green(` \u2713 index.md updated`));
+      if (onSourcesChanged) {
+        try {
+          await onSourcesChanged();
+        } catch {
+        }
+      }
     } catch (err) {
       console.log(chalk.red(` \u2717 ${err.message}`));
     }
@@ -207,250 +231,28 @@ function startWatcher({ folder, sourcesDir, debounceMs = 2e3, authStorage, index
 
 // src/session-watcher.ts
 import { watch as watch2 } from "chokidar";
-import { join as join5, basename as basename3 } from "path";
-import { readdir as readdir3, readFile as readFile3, writeFile as writeFile4, mkdir as mkdir4 } from "fs/promises";
-import { existsSync as existsSync3 } from "fs";
-
-// src/trace-builder.ts
-import { readFile, writeFile as writeFile2, mkdir as mkdir2, readdir as readdir2 } from "fs/promises";
+import { join as join3, basename as basename3 } from "path";
+import { readdir as readdir2, readFile, writeFile as writeFile2, mkdir as mkdir2 } from "fs/promises";
 import { existsSync } from "fs";
-import { join as join3, basename as pathBasename } from "path";
-async function buildTrace(sessionFile, sourcesDir) {
-  const raw = await readFile(sessionFile, "utf-8");
-  const lines = raw.trim().split("\n").filter(Boolean);
-  if (lines.length < 2) return null;
-  const entries = [];
-  let header = null;
-  for (const line of lines) {
-    try {
-      const obj = JSON.parse(line);
-      if (obj.type === "session") header = obj;
-      else entries.push(obj);
-    } catch {
-    }
-  }
-  if (!header) return null;
-  const messages = entries.filter((e) => e.type === "message");
-  const lastAssistant = [...messages].reverse().find(
-    (e) => e.message?.role === "assistant" && e.message?.stopReason === "stop"
-  );
-  if (!lastAssistant) return null;
-  const modelChange = entries.find((e) => e.type === "model_change");
-  const model = modelChange?.modelId ?? lastAssistant.message?.model ?? void 0;
-  const firstUser = messages.find((e) => e.message?.role === "user");
-  const question = extractText(firstUser?.message?.content);
-  const sessionInfo = entries.find((e) => e.type === "session_info");
-  const sessionName = sessionInfo?.name ?? "";
-  const mode = sessionName.startsWith("index:") ? "index" : sessionName.startsWith("query:") || question ? "query" : "unknown";
-  const answer = extractText(lastAssistant.message?.content);
-  const filesRead = [];
-  for (const entry of messages) {
-    if (entry.message?.role !== "assistant") continue;
-    for (const block of entry.message?.content ?? []) {
-      if (block.type === "toolCall" && block.name === "read") {
-        const p = block.arguments?.path ?? "";
-        if (p && !filesRead.includes(p)) filesRead.push(p);
-      }
-    }
-  }
-  let filesAvailable = [];
-  try {
-    const all = await readdir2(sourcesDir);
-    filesAvailable = all.filter((f) => f.endsWith(".md"));
-  } catch {
-  }
-  const filesSkipped = filesAvailable.filter(
-    (f) => !filesRead.some((r) => r.endsWith(f))
-  );
-  const firstMsg = messages[0];
-  const lastMsg = messages[messages.length - 1];
-  let durationMs;
-  if (firstMsg?.timestamp && lastMsg?.timestamp) {
-    durationMs = new Date(lastMsg.timestamp).getTime() - new Date(firstMsg.timestamp).getTime();
-  }
-  return {
-    sessionId: header.id,
-    sessionFile: pathBasename(sessionFile),
-    timestamp: header.timestamp,
-    mode,
-    question: question || void 0,
-    answer: answer || void 0,
-    filesRead,
-    filesAvailable,
-    filesSkipped,
-    model,
-    durationMs
-  };
-}
-async function saveTrace(kbRoot, trace) {
-  const tracesDir = join3(kbRoot, ".llm-kb", "traces");
-  await mkdir2(tracesDir, { recursive: true });
-  const outPath = join3(tracesDir, `${trace.sessionId}.json`);
-  await writeFile2(outPath, JSON.stringify(trace, null, 2) + "\n", "utf-8");
-}
-async function appendToQueryLog(kbRoot, trace) {
-  if (trace.mode !== "query" || !trace.question) return;
-  const wikiDir = join3(kbRoot, ".llm-kb", "wiki");
-  await mkdir2(wikiDir, { recursive: true });
-  const logPath = join3(wikiDir, "queries.md");
-  const date = new Date(trace.timestamp).toISOString().replace("T", " ").slice(0, 19);
-  const durationSec = trace.durationMs ? `${(trace.durationMs / 1e3).toFixed(1)}s` : "?";
-  const filesLine = trace.filesRead.length > 0 ? trace.filesRead.map((f) => pathBasename(f)).join(", ") : "_none_";
-  let header = "";
-  if (!existsSync(logPath)) {
-    header = `# Query Log
-
-All queries run against this knowledge base.
-
----
-
-`;
-  }
-  const entry = [
-    `## ${trace.question}`,
-    ``,
-    `- **Date:** ${date}`,
-    `- **Model:** ${trace.model ?? "unknown"}`,
-    `- **Duration:** ${durationSec}`,
-    `- **Files read:** ${filesLine}`,
-    trace.filesSkipped.length > 0 ? `- **Files skipped:** ${trace.filesSkipped.join(", ")}` : null,
-    ``,
-    trace.answer ? `### Answer
-
-${trace.answer}` : null,
-    ``,
-    `---`,
-    ``
-  ].filter((l) => l !== null).join("\n");
-  const existing = existsSync(logPath) ? await readFile(logPath, "utf-8") : "";
-  await writeFile2(logPath, header + entry + existing, "utf-8");
-}
-function extractText(content) {
-  if (!content) return "";
-  if (typeof content === "string") return content;
-  if (Array.isArray(content)) {
-    return content.filter((b) => b.type === "text").map((b) => b.text ?? "").join("").trim();
-  }
-  return "";
-}
-
-// src/wiki-updater.ts
-import { AuthStorage } from "@mariozechner/pi-coding-agent";
-import { existsSync as existsSync2 } from "fs";
-import { readFile as readFile2, writeFile as writeFile3, mkdir as mkdir3 } from "fs/promises";
-import { join as join4 } from "path";
-import { homedir } from "os";
-async function resolveApiKey(authStorage) {
-  if (authStorage) {
-    return authStorage.getApiKey("anthropic");
-  }
-  const piAuthPath = join4(homedir(), ".pi", "agent", "auth.json");
-  if (existsSync2(piAuthPath)) {
-    const storage = AuthStorage.create(piAuthPath);
-    return storage.getApiKey("anthropic");
-  }
-  return process.env.ANTHROPIC_API_KEY;
-}
-function buildPrompt(question, answer, sources, date, currentWiki) {
-  const rules = `Rules for wiki structure:
-- Use ## for CONCEPTS and TOPICS \u2014 NOT source file names
-  Good: "## Electronic Evidence", "## Mob Lynching", "## Burden of Proof"
-  Bad: "## Indian Evidence Act.md", "## indian penal code - new.md"
-- Use ### for subtopics within a concept
-- A concept can draw from MULTIPLE source files \u2014 synthesize, don't separate by file
-- If knowledge from this Q&A fits an existing concept, ADD to it \u2014 never duplicate
-- If it's a genuinely new concept, create a new ## section
-- Be concise: bullet points for lists, short prose for explanations
-- Include source citations inline: (Source: filename, p.X)
-- Add cross-references where concepts relate: See also: [[Other Concept]]
-- End each ## section with: *Sources: file1, file2 \xB7 date*
-- Separate ## sections with: ---`;
-  if (currentWiki.trim()) {
-    return `You are maintaining a concept-organized knowledge wiki.
-
-## Current wiki
-${currentWiki}
-
-## New Q&A to integrate
-**Question:** ${question}
-**Sources used:** ${sources}
-**Date:** ${date}
-
-**Answer:**
-${answer}
-
----
-
-Update the wiki to integrate this new knowledge.
-${rules}
-
-Return ONLY the complete updated wiki markdown. No explanation.`;
-  }
-  return `You are creating a concept-organized knowledge wiki.
-
-## First Q&A to add
-**Question:** ${question}
-**Sources used:** ${sources}
-**Date:** ${date}
-
-**Answer:**
-${answer}
-
----
-
-Create a clean wiki from this Q&A.
-- Start with: # Knowledge Wiki\\n\\n> Concept-organized knowledge base. Updated after each query.\\n\\n---
-${rules}
-
-Return ONLY the wiki markdown. No explanation.`;
-}
-async function updateWiki(kbRoot, trace, authStorage, indexModelId = "claude-haiku-4-5") {
-  if (trace.mode !== "query" || !trace.question || !trace.answer) return;
-  const wikiDir = join4(kbRoot, ".llm-kb", "wiki");
-  await mkdir3(wikiDir, { recursive: true });
-  const wikiPath = join4(wikiDir, "wiki.md");
-  const currentWiki = existsSync2(wikiPath) ? await readFile2(wikiPath, "utf-8").catch(() => "") : "";
-  const sources = trace.filesRead.map((f) => f.split(/[\\/]/).pop() ?? f).filter((f) => f.endsWith(".md") && f !== "index.md" && f !== "wiki.md").join(", ") || "unknown";
-  const date = new Date(trace.timestamp).toISOString().slice(0, 10);
-  const prompt = buildPrompt(trace.question, trace.answer, sources, date, currentWiki);
-  const apiKey = await resolveApiKey(authStorage);
-  if (!apiKey) return;
-  const model = getModels("anthropic").find((m) => m.id === indexModelId);
-  if (!model) return;
-  const result = await completeSimple(
-    model,
-    {
-      systemPrompt: "You are a precise knowledge librarian. Organize information by CONCEPT, not by source file. Synthesize knowledge from multiple sources into unified topic articles. Return only clean markdown.",
-      messages: [{ role: "user", content: prompt, timestamp: Date.now() }]
-    },
-    { apiKey }
-  );
-  const text = result.content.filter((b) => b.type === "text").map((b) => b.text).join("").trim();
-  if (text) {
-    await writeFile3(wikiPath, text + "\n", "utf-8");
-  }
-}
-
-// src/session-watcher.ts
 var PROCESSED_LOG = ".llm-kb/traces/.processed";
 async function loadProcessed(kbRoot) {
-  const path3 = join5(kbRoot, PROCESSED_LOG);
-  if (!existsSync3(path3)) return /* @__PURE__ */ new Set();
+  const path3 = join3(kbRoot, PROCESSED_LOG);
+  if (!existsSync(path3)) return /* @__PURE__ */ new Set();
   try {
-    const lines = (await readFile3(path3, "utf-8")).split("\n").filter(Boolean);
+    const lines = (await readFile(path3, "utf-8")).split("\n").filter(Boolean);
     return new Set(lines);
   } catch {
     return /* @__PURE__ */ new Set();
   }
 }
 async function markProcessed(kbRoot, sessionId) {
-  const path3 = join5(kbRoot, PROCESSED_LOG);
-  await mkdir4(join5(kbRoot, ".llm-kb", "traces"), { recursive: true });
-  await writeFile4(path3, sessionId + "\n", { flag: "a" });
+  const path3 = join3(kbRoot, PROCESSED_LOG);
+  await mkdir2(join3(kbRoot, ".llm-kb", "traces"), { recursive: true });
+  await writeFile2(path3, sessionId + "\n", { flag: "a" });
 }
 async function startSessionWatcher(kbRoot) {
-  const sessionsDir = join5(kbRoot, ".llm-kb", "sessions");
-  const sourcesDir = join5(kbRoot, ".llm-kb", "wiki", "sources");
+  const sessionsDir = join3(kbRoot, ".llm-kb", "sessions");
+  const sourcesDir = join3(kbRoot, ".llm-kb", "wiki", "sources");
   const processed = await loadProcessed(kbRoot);
   const timers = /* @__PURE__ */ new Map();
   async function processSession(filePath) {
@@ -478,11 +280,11 @@ async function startSessionWatcher(kbRoot) {
     }, 1500);
     timers.set(filePath, timer);
   }
-  if (existsSync3(sessionsDir)) {
+  if (existsSync(sessionsDir)) {
     try {
-      const files = (await readdir3(sessionsDir)).filter((f) => f.endsWith(".jsonl"));
+      const files = (await readdir2(sessionsDir)).filter((f) => f.endsWith(".jsonl"));
       for (const f of files) {
-        await processSession(join5(sessionsDir, f));
+        await processSession(join3(sessionsDir, f));
       }
     } catch {
     }
@@ -500,482 +302,23 @@ async function startSessionWatcher(kbRoot) {
   });
 }
 
-// src/query.ts
-import {
-  createAgentSession,
-  createBashTool,
-  createReadTool,
-  createWriteTool,
-  DefaultResourceLoader,
-  SettingsManager
-} from "@mariozechner/pi-coding-agent";
-import { readdir as readdir4, mkdir as mkdir5, readFile as readFile4 } from "fs/promises";
-import { existsSync as existsSync4 } from "fs";
-import { join as join6, basename as basename4 } from "path";
-import chalk3 from "chalk";
-
-// src/md-stream.ts
-import chalk2 from "chalk";
-var MarkdownStream = class {
-  buffer = "";
-  isTTY;
-  constructor(isTTY = false) {
-    this.isTTY = isTTY;
-  }
-  /** Feed a text_delta chunk. Returns styled string ready for stdout. */
-  push(chunk) {
-    if (!this.isTTY) return chunk;
-    this.buffer += chunk;
-    return this.drain(false);
-  }
-  /** Flush remaining buffer (call on text_end). */
-  end() {
-    if (!this.isTTY) return "";
-    const out = this.drain(true);
-    this.buffer = "";
-    return out;
-  }
-  drain(final) {
-    let out = "";
-    while (true) {
-      const nlIdx = this.buffer.indexOf("\n");
-      if (nlIdx === -1) {
-        if (final && this.buffer.length > 0) {
-          out += this.renderLine(this.buffer);
-          this.buffer = "";
-        }
-        break;
-      }
-      const line = this.buffer.slice(0, nlIdx);
-      this.buffer = this.buffer.slice(nlIdx + 1);
-      out += this.renderLine(line) + "\n";
-    }
-    return out;
-  }
-  /** Render a single complete line with block + inline styling. */
-  renderLine(line) {
-    const trimmed = line.trimStart();
-    if (/^-{3,}\s*$/.test(trimmed) || /^\*{3,}\s*$/.test(trimmed)) {
-      const cols = process.stdout.columns || 80;
-      return chalk2.dim("\u2500".repeat(Math.min(cols, 60)));
-    }
-    const headerMatch = trimmed.match(/^(#{1,6})\s+(.*)$/);
-    if (headerMatch) {
-      const text = this.inline(headerMatch[2]);
-      return "\n" + chalk2.bold(text);
-    }
-    const bulletMatch = trimmed.match(/^[-*+]\s+(.*)$/);
-    if (bulletMatch) {
-      const indent = line.length - trimmed.length;
-      return " ".repeat(indent) + chalk2.dim("\u2022") + " " + this.inline(bulletMatch[1]);
-    }
-    const numMatch = trimmed.match(/^(\d+)[.)]\s+(.*)$/);
-    if (numMatch) {
-      const indent = line.length - trimmed.length;
-      return " ".repeat(indent) + chalk2.dim(numMatch[1] + ".") + " " + this.inline(numMatch[2]);
-    }
-    if (/^\|[\s\-:|]+\|$/.test(trimmed)) {
-      return chalk2.dim(trimmed);
-    }
-    if (trimmed.startsWith("|") && trimmed.endsWith("|")) {
-      return this.inline(line);
-    }
-    if (trimmed.startsWith(">")) {
-      const content = trimmed.replace(/^>+\s*/, "");
-      return chalk2.dim("\u2502 ") + chalk2.italic(this.inline(content));
-    }
-    return this.inline(line);
-  }
-  /** Apply inline markdown styling to text. */
-  inline(text) {
-    text = text.replace(/`([^`]+)`/g, (_, c) => chalk2.cyan(c));
-    text = text.replace(/\*\*\*(.+?)\*\*\*/g, (_, t) => chalk2.bold.italic(t));
-    text = text.replace(/\*\*(.+?)\*\*/g, (_, t) => chalk2.bold(t));
-    text = text.replace(/(?<!\*)\*(.+?)\*(?!\*)/g, (_, t) => chalk2.italic(t));
-    text = text.replace(/~~(.+?)~~/g, (_, t) => chalk2.strikethrough(t));
-    text = text.replace(
-      /\[([^\]]+)\]\(([^)]+)\)/g,
-      (_, label, url) => `${label} ${chalk2.dim(`(${url})`)}`
-    );
-    return text;
-  }
-};
-
-// src/query.ts
-function extractAnswerText(content) {
-  return (content ?? []).filter((b) => b.type === "text").map((b) => b.text ?? "").join("").trim();
-}
-function extractFilesRead(messages) {
-  const paths = [];
-  for (const msg of messages) {
-    if (msg.role !== "assistant") continue;
-    for (const block of msg.content ?? []) {
-      if (block.type === "toolCall" && block.name === "read") {
-        const p = block.arguments?.path ?? "";
-        if (p && !paths.includes(p)) paths.push(p);
-      }
-    }
-  }
-  return paths;
-}
-function getToolLabel(toolName, args) {
-  if (toolName === "read" || toolName === "write" || toolName === "edit") {
-    const file = basename4(args?.path ?? "");
-    if (!file || !/\.[a-z0-9]{1,6}$/i.test(file)) return null;
-    const verb = toolName === "read" ? "Reading" : toolName === "write" ? "Writing" : "Editing";
-    return `${verb}  ${file}`;
-  }
-  if (toolName === "bash" && args?.command) {
-    return `Running  bash`;
-  }
-  return null;
-}
-function buildQueryAgents(sourceFiles, save, wikiContent) {
-  const sourceList = sourceFiles.map((f) => `  - ${f}`).join("\n");
-  const wikiSection = wikiContent ? `## Knowledge Wiki (use this first)
-
-The wiki below contains knowledge already extracted from this knowledge base.
-If the user's question is covered here, answer directly from it \u2014 no need to re-read source files.
-Always cite the original source files mentioned in the wiki.
-
-${wikiContent}
-
----
-
-` : "";
-  const sourceStep = wikiContent ? "If not covered in the wiki above: read the sources" : "How to answer";
-  const lines = [
-    `# llm-kb Knowledge Base \u2014 Query Mode`,
-    ``,
-    wikiSection,
-    `## ${sourceStep}`,
-    ``,
-    `1. Read .llm-kb/wiki/index.md to understand all available sources`,
-    `2. Select the most relevant source files (usually 2-5) and read them in full`,
-    `3. Answer with inline citations: (filename, page number)`,
-    `4. If you can't find the answer, say so \u2014 don't hallucinate`,
-    ``,
-    `## Available parsed sources`,
-    sourceList,
-    ``,
-    `## Non-PDF files (docx, xlsx, pptx)`,
-    `Use bash to run Node.js scripts. Libraries are pre-installed via require().`,
-    ``,
-    `### Word (.docx) \u2014 structured XML`,
-    `.docx files are ZIP archives containing word/document.xml.`,
-    `Read them SELECTIVELY \u2014 extract only what is relevant to the question:`,
-    ``,
-    "```javascript",
-    `const AdmZip = require('adm-zip');`,
-    `const zip = new AdmZip('file.docx');`,
-    `const xml = zip.readAsText('word/document.xml');`,
-    `// Parse XML to find specific paragraphs, headings, tables`,
-    "```",
-    ``,
-    `Strategy for large .docx files:`,
-    `1. First: extract headings/structure to understand the document layout`,
-    `2. Then: extract only the sections relevant to the user's question`,
-    `NEVER dump the entire document.`,
-    ``,
-    `### Excel (.xlsx) \u2014 use exceljs`,
-    `Read specific sheets and ranges, not the whole workbook:`,
-    ``,
-    "```javascript",
-    `const ExcelJS = require('exceljs');`,
-    `const wb = new ExcelJS.Workbook();`,
-    `await wb.xlsx.readFile('file.xlsx');`,
-    `const sheet = wb.getWorksheet(1);`,
-    `// Read specific rows/columns relevant to the question`,
-    "```",
-    ``,
-    `### PowerPoint (.pptx) \u2014 use officeparser`,
-    ``,
-    "```javascript",
-    `const officeparser = require('officeparser');`,
-    `const text = await officeparser.parseOfficeAsync('file.pptx');`,
-    "```",
-    ``,
-    `## Rules`,
-    `- Always cite sources with filename and page number`,
-    `- Read the FULL source file, not just the beginning (for .md sources)`,
-    `- For non-PDF files, extract ONLY relevant sections \u2014 never dump entire files`,
-    `- Prefer primary sources over previous analyses`,
-    ``,
-    `## Guidelines`,
-    `A guidelines file may exist at .llm-kb/guidelines.md with learned rules from`,
-    `past evaluations and user preferences. Read it when:`,
-    `- You're unsure about citation accuracy or format`,
-    `- You're about to read source files (guidelines may suggest using wiki instead)`,
-    `- The question touches a topic that may have had issues in past evaluations`
-  ];
-  if (save) {
-    lines.push(``, `## Research Mode`, `Save your analysis to .llm-kb/wiki/outputs/ with a descriptive filename.`, `Include the question at the top and all citations.`);
-  }
-  return lines.join("\n");
-}
-var WikiUpdateScheduler = class {
-  constructor(everyN, everyMin) {
-    this.everyN = everyN;
-    this.everyMin = everyMin;
-  }
-  everyN;
-  everyMin;
-  stopMsgCount = 0;
-  lastUpdateAt = 0;
-  chain = Promise.resolve();
-  shouldUpdate() {
-    return this.stopMsgCount > 0 && this.stopMsgCount % this.everyN === 0 || this.lastUpdateAt > 0 && Date.now() - this.lastUpdateAt > this.everyMin * 6e4;
-  }
-  enqueue(work) {
-    this.chain = this.chain.then(() => work().catch(() => {
-    }));
-  }
-  onMessageEnd(msg, snap, doUpdate) {
-    if (msg.role !== "assistant" || msg.stopReason !== "stop") return;
-    this.stopMsgCount++;
-    if (this.shouldUpdate()) {
-      this.lastUpdateAt = Date.now();
-      this.enqueue(() => doUpdate(snap().messages));
-    }
-  }
-  onAgentEnd(msgs, doUpdate) {
-    this.lastUpdateAt = Date.now();
-    this.enqueue(() => doUpdate(msgs));
-  }
-  flush() {
-    return this.chain;
-  }
-};
-function subscribeDisplay(session, opts) {
-  const ui = opts.tuiDisplay;
-  const dim = (s) => process.stdout.isTTY ? chalk3.dim(s) : s;
-  const thinLine = () => dim("\u2500".repeat(process.stdout.columns || 80));
-  let phase = "idle";
-  let filesReadCount = 0;
-  let shownToolCalls = /* @__PURE__ */ new Set();
-  let startTime = Date.now();
-  let md = new MarkdownStream(process.stdout.isTTY ?? false);
-  let lastQuestion = "";
-  const scheduler = new WikiUpdateScheduler(5, 3);
-  const buildTrace2 = (messages) => {
-    const last = [...messages].reverse().find((m) => m.role === "assistant" && m.stopReason === "stop");
-    if (!last) return null;
-    const filesRead = extractFilesRead(messages);
-    return {
-      sessionId: session.sessionId,
-      sessionFile: session.sessionFile ?? "",
-      timestamp: (/* @__PURE__ */ new Date()).toISOString(),
-      mode: "query",
-      question: lastQuestion,
-      answer: extractAnswerText(last.content),
-      filesRead,
-      filesAvailable: opts.mdFiles,
-      filesSkipped: opts.mdFiles.filter((f) => !filesRead.some((r) => r.endsWith(f))),
-      model: last.model
-    };
-  };
-  const doUpdate = async (messages) => {
-    const trace = buildTrace2(messages);
-    if (!trace) return;
-    await saveTrace(opts.folder, trace);
-    await appendToQueryLog(opts.folder, trace);
-    await updateWiki(opts.folder, trace, opts.authStorage);
-  };
-  session.subscribe((event) => {
-    if (event.type === "agent_start") {
-      phase = "idle";
-      filesReadCount = 0;
-      shownToolCalls = /* @__PURE__ */ new Set();
-      startTime = Date.now();
-      md = new MarkdownStream(process.stdout.isTTY ?? false);
-      const modelName = opts.modelId ?? "claude-sonnet-4-6";
-      if (ui) {
-        ui.disableInput();
-        ui.beginResponse(modelName);
-      } else process.stdout.write(dim(`\u27E1 ${modelName}`) + "\n");
-    }
-    if (event.type === "message_update") {
-      const ae = event.assistantMessageEvent;
-      if (ae.type === "thinking_start") {
-        if (!ui) process.stdout.write(dim("\n\u25B8 Thinking\n"));
-        phase = "thinking";
-      }
-      if (ae.type === "thinking_delta") {
-        if (ui) ui.appendThinking(ae.delta);
-        else process.stdout.write(dim(`  ${ae.delta}`));
-      }
-      if (ae.type === "thinking_end") {
-        if (ui) ui.endThinking();
-        else process.stdout.write("\n");
-      }
-    }
-    if (event.type === "message_update") {
-      const ae = event.assistantMessageEvent;
-      if (ae.type === "toolcall_end" && ae.toolCall) {
-        const label = getToolLabel(ae.toolCall.name, ae.toolCall.arguments);
-        if (label) {
-          if (!ui && phase !== "tools") process.stdout.write("\n");
-          phase = "tools";
-          if (ui) {
-            ui.addToolCall(ae.toolCall.id, label, ae.toolCall.name);
-            if (ae.toolCall.name === "bash" && ae.toolCall.arguments?.command) {
-              ui.addCodeBlock(ae.toolCall.arguments.command);
-            }
-          } else {
-            process.stdout.write(dim(`  \u25B8 ${label}`) + "\n");
-            if (ae.toolCall.name === "bash" && ae.toolCall.arguments?.command) {
-              const code = ae.toolCall.arguments.command;
-              process.stdout.write(dim(code.split("\n").map((l) => `    ${l}`).join("\n")) + "\n");
-            }
-            shownToolCalls.add(ae.toolCall.id);
-            if (ae.toolCall.name === "read") filesReadCount++;
-          }
-        }
-      }
-    }
-    if (event.type === "tool_execution_start") {
-      const { toolCallId, toolName, args } = event;
-      if (ui) {
-        const label = getToolLabel(toolName, args);
-        if (label) ui.addToolCall(toolCallId, label, toolName);
-      } else if (!shownToolCalls.has(toolCallId)) {
-        const label = getToolLabel(toolName, args);
-        if (label) {
-          if (phase !== "tools") process.stdout.write("\n");
-          phase = "tools";
-          process.stdout.write(dim(`  \u25B8 ${label}`) + "\n");
-          shownToolCalls.add(toolCallId);
-          if (toolName === "read") filesReadCount++;
-        }
-      }
-    }
-    if (event.type === "tool_execution_end") {
-      const { toolCallId, isError } = event;
-      if (ui) ui.addToolResult(toolCallId, isError);
-    }
-    if (event.type === "message_update") {
-      const ae = event.assistantMessageEvent;
-      if (ae.type === "text_start" && phase !== "answer") {
-        if (ui) ui.beginAnswer();
-        else if (phase === "thinking" || phase === "tools") {
-          process.stdout.write(`
-${thinLine()}
-
-`);
-        }
-        phase = "answer";
-      }
-      if (ae.type === "text_delta") {
-        if (ui) ui.appendAnswer(ae.delta);
-        else process.stdout.write(md.push(ae.delta));
-      }
-      if (ae.type === "text_end" && !ui) process.stdout.write(md.end());
-    }
-    if (event.type === "agent_end") {
-      if (ui) {
-        ui.showCompletion();
-        ui.enableInput();
-      } else {
-        const elapsed = ((Date.now() - startTime) / 1e3).toFixed(1);
-        const source = filesReadCount > 0 ? `${filesReadCount} file${filesReadCount !== 1 ? "s" : ""} read` : "wiki";
-        const stats = `${elapsed}s \xB7 ${source}`;
-        const cols = process.stdout.columns || 80;
-        const pad = Math.max(0, cols - stats.length - 4);
-        process.stdout.write(`
-
-${dim("\u2500\u2500 " + stats + " " + "\u2500".repeat(pad))}
-`);
-      }
-      scheduler.onAgentEnd(event.messages, doUpdate);
-    }
-    if (event.type === "message_end") {
-      scheduler.onMessageEnd(event.message, () => ({ messages: session.state.messages }), doUpdate);
-    }
-  });
-  return {
-    setQuestion(q) {
-      lastQuestion = q;
-    },
-    flush() {
-      return scheduler.flush();
-    }
-  };
-}
-async function createChat(folder, options) {
-  const sourcesDir = join6(folder, ".llm-kb", "wiki", "sources");
-  const files = await readdir4(sourcesDir);
-  const mdFiles = files.filter((f) => f.endsWith(".md"));
-  if (mdFiles.length === 0) throw new Error("No sources found. Run 'llm-kb run' first.");
-  if (options.save) await mkdir5(join6(folder, ".llm-kb", "wiki", "outputs"), { recursive: true });
-  process.env.NODE_PATH = getNodeModulesPath();
-  const wikiPath = join6(folder, ".llm-kb", "wiki", "wiki.md");
-  const wikiContent = existsSync4(wikiPath) ? await readFile4(wikiPath, "utf-8").catch(() => "") : "";
-  const agentsContent = buildQueryAgents(mdFiles, !!options.save, wikiContent);
-  const loader = new DefaultResourceLoader({
-    cwd: folder,
-    agentsFilesOverride: (current) => ({
-      agentsFiles: [...current.agentsFiles, { path: ".llm-kb/AGENTS.md", content: agentsContent }]
-    })
-  });
-  await loader.reload();
-  const tools = [
-    createReadTool(folder),
-    createBashTool(folder),
-    createWriteTool(folder)
-  ];
-  const model = options.modelId ? getModels("anthropic").find((m) => m.id === options.modelId) : void 0;
-  const { session } = await createAgentSession({
-    cwd: folder,
-    resourceLoader: loader,
-    tools,
-    sessionManager: options.save ? await createKBSession(folder) : await continueKBSession(folder),
-    settingsManager: SettingsManager.inMemory({ compaction: { enabled: false } }),
-    thinkingLevel: "low",
-    ...options.authStorage ? { authStorage: options.authStorage } : {},
-    ...model ? { model } : {}
-  });
-  const display = subscribeDisplay(session, {
-    modelId: options.modelId,
-    authStorage: options.authStorage,
-    folder,
-    mdFiles,
-    tuiDisplay: options.tuiDisplay
-  });
-  return { session, display };
-}
-async function query(folder, question, options) {
-  const { session, display } = await createChat(folder, options);
-  session.setSessionName(`query: ${question}`);
-  display.setQuestion(question);
-  await session.prompt(question);
-  await display.flush();
-  session.dispose();
-  if (options.save) {
-    const sourcesDir = join6(folder, ".llm-kb", "wiki", "sources");
-    const { buildIndex: buildIndex2 } = await import("./indexer-KSYRIVVN.js");
-    await buildIndex2(folder, sourcesDir, void 0, options.authStorage);
-  }
-}
-
 // src/eval.ts
-import { AuthStorage as AuthStorage3 } from "@mariozechner/pi-coding-agent";
-import { readFile as readFile5, readdir as readdir5, writeFile as writeFile5, mkdir as mkdir6 } from "fs/promises";
-import { existsSync as existsSync5 } from "fs";
-import { join as join7, basename as basename5 } from "path";
-import { homedir as homedir2 } from "os";
+import { readFile as readFile2, readdir as readdir3, writeFile as writeFile3, mkdir as mkdir3 } from "fs/promises";
+import { existsSync as existsSync2 } from "fs";
+import { join as join4, basename as basename4 } from "path";
 async function parseSessionsForEval(sessionsDir, sourcesDir, limit) {
-  if (!existsSync5(sessionsDir)) return [];
-  const sessionFiles = (await readdir5(sessionsDir)).filter((f) => f.endsWith(".jsonl")).sort().reverse();
+  if (!existsSync2(sessionsDir)) return [];
+  const sessionFiles = (await readdir3(sessionsDir)).filter((f) => f.endsWith(".jsonl")).sort().reverse();
   const files = limit ? sessionFiles.slice(0, limit) : sessionFiles;
   const qas = [];
   let filesAvailable = [];
   try {
-    filesAvailable = (await readdir5(sourcesDir)).filter((f) => f.endsWith(".md"));
+    filesAvailable = (await readdir3(sourcesDir)).filter((f) => f.endsWith(".md"));
   } catch {
   }
   for (const file of files) {
     try {
-      const raw = await readFile5(join7(sessionsDir, file), "utf-8");
+      const raw = await readFile2(join4(sessionsDir, file), "utf-8");
       const lines = raw.trim().split("\n").filter(Boolean);
       const entries = [];
       for (const line of lines) {
@@ -1000,6 +343,7 @@ async function parseSessionsForEval(sessionsDir, sourcesDir, limit) {
         if (!msg) continue;
         if (msg.role === "user") {
           if (currentQuestion && currentAnswer) {
+            const parsed = parseCitations(currentAnswer);
             qas.push({
               sessionFile: file,
               question: currentQuestion,
@@ -1011,10 +355,11 @@ async function parseSessionsForEval(sessionsDir, sourcesDir, limit) {
               ),
               answer: currentAnswer,
               model: currentModel,
-              durationMs: endTs - startTs
+              durationMs: endTs - startTs,
+              citations: parsed.citations
             });
           }
-          currentQuestion = extractText2(msg.content);
+          currentQuestion = extractText(msg.content);
           currentThinking = "";
           currentFilesRead = [];
           currentAnswer = "";
@@ -1035,7 +380,7 @@ async function parseSessionsForEval(sessionsDir, sourcesDir, limit) {
             for (const block of prev.message?.content ?? []) {
               if (block.type === "toolCall" && block.id === toolCallId && block.name === "read") {
                 const path3 = block.arguments?.path ?? "";
-                const content = extractText2(msg.content);
+                const content = extractText(msg.content);
                 if (path3 && content) {
                   currentFilesRead.push({ path: path3, content: content.slice(0, 2e3) });
                 }
@@ -1045,6 +390,7 @@ async function parseSessionsForEval(sessionsDir, sourcesDir, limit) {
         }
       }
       if (currentQuestion && currentAnswer) {
+        const parsed = parseCitations(currentAnswer);
         qas.push({
           sessionFile: file,
           question: currentQuestion,
@@ -1056,7 +402,8 @@ async function parseSessionsForEval(sessionsDir, sourcesDir, limit) {
           ),
           answer: currentAnswer,
           model: currentModel,
-          durationMs: endTs - startTs
+          durationMs: endTs - startTs,
+          citations: parsed.citations
         });
       }
     } catch {
@@ -1084,13 +431,37 @@ function calculateMetrics(qas) {
     }
     for (const f of sourceFilesRead) {
       totalFilesRead++;
-      const name = basename5(f.path);
+      const name = basename4(f.path);
       uniqueFiles.set(name, (uniqueFiles.get(name) ?? 0) + 1);
       if (!qa.answer.includes(name) && !qa.answer.includes(name.replace(".md", ""))) {
         wastedReads++;
       }
     }
   }
+  let totalCitations = 0;
+  let withBbox = 0;
+  let withoutBbox = 0;
+  let multiPage = 0;
+  let answersWithCitations = 0;
+  let answersWithoutCitations = 0;
+  for (const qa of qas) {
+    if (qa.citations.length > 0) {
+      answersWithCitations++;
+      for (const c of qa.citations) {
+        totalCitations++;
+        if (c.bbox || c.pages && c.pages.length > 0) {
+          withBbox++;
+        } else {
+          withoutBbox++;
+        }
+        if (c.pages && c.pages.length > 0) {
+          multiPage++;
+        }
+      }
+    } else {
+      answersWithoutCitations++;
+    }
+  }
   return {
     totalSessions: uniqueSessions.size,
     totalQAs: qas.length,
@@ -1099,23 +470,21 @@ function calculateMetrics(qas) {
     sourceReads,
     totalFilesRead,
     uniqueFilesRead: uniqueFiles,
-    wastedReads
+    wastedReads,
+    citations: {
+      totalCitations,
+      withBbox,
+      withoutBbox,
+      multiPage,
+      avgPerAnswer: qas.length > 0 ? totalCitations / qas.length : 0,
+      answersWithCitations,
+      answersWithoutCitations
+    }
   };
 }
-async function resolveApiKey2(authStorage) {
-  if (authStorage) return authStorage.getApiKey("anthropic");
-  const piAuthPath = join7(homedir2(), ".pi", "agent", "auth.json");
-  if (existsSync5(piAuthPath)) {
-    const storage = AuthStorage3.create(piAuthPath);
-    return storage.getApiKey("anthropic");
-  }
-  return process.env.ANTHROPIC_API_KEY;
-}
-async function judgeQA(qa, apiKey, modelId) {
+async function judgeQA(qa, modelId, authStorage) {
   const issues = [];
-  const model = getModels("anthropic").find((m) => m.id === modelId);
-  if (!model) return issues;
-  const filesSummary = qa.filesRead.map((f) => `File: ${basename5(f.path)}
+  const filesSummary = qa.filesRead.map((f) => `File: ${basename4(f.path)}
 Content (first 2000 chars):
 ${f.content}`).join("\n\n---\n\n");
   const skippedList = qa.filesSkipped.join(", ") || "none";
@@ -1148,13 +517,14 @@ Checks:
 Return ONLY a JSON array. If no issues found, return [].
 Example: [{"type":"wiki-gap","severity":"warning","detail":"Electronic evidence topic not in wiki","recommendation":"Add electronic evidence section to wiki"}]`;
   try {
-    const result = await completeSimple(
-      model,
+    const result = await completeWithFallback(
+      modelId,
+      authStorage,
+      "eval",
       {
         systemPrompt: "You are a precise QA evaluator. Return only valid JSON arrays. No explanation.",
         messages: [{ role: "user", content: prompt, timestamp: Date.now() }]
-      },
-      { apiKey }
+      }
     );
     const text = result.content.filter((b) => b.type === "text").map((b) => b.text).join("").trim();
     const jsonMatch = text.match(/\[[\s\S]*\]/);
@@ -1193,6 +563,18 @@ function buildReport(result) {
   lines.push(`| Total file reads | ${metrics.totalFilesRead} |`);
   lines.push(`| Wasted reads | ${metrics.wastedReads} |`);
   lines.push(``);
+  const cm = metrics.citations;
+  lines.push(`## Citations`);
+  lines.push(``);
+  lines.push(`| Metric | Value |`);
+  lines.push(`|---|---|`);
+  lines.push(`| Total citations | ${cm.totalCitations} |`);
+  lines.push(`| Avg per answer | ${cm.avgPerAnswer.toFixed(1)} |`);
+  lines.push(`| With bbox | ${cm.withBbox} (${cm.totalCitations > 0 ? Math.round(cm.withBbox / cm.totalCitations * 100) : 0}%) |`);
+  lines.push(`| Without bbox | ${cm.withoutBbox} |`);
+  lines.push(`| Multi-page | ${cm.multiPage} |`);
+  lines.push(`| Answers with citations | ${cm.answersWithCitations}/${metrics.totalQAs} (${metrics.totalQAs > 0 ? Math.round(cm.answersWithCitations / metrics.totalQAs * 100) : 0}%) |`);
+  lines.push(``);
   if (metrics.uniqueFilesRead.size > 0) {
     lines.push(`### Most Read Files`);
     lines.push(``);
@@ -1278,6 +660,20 @@ function buildAgentsInsights(result) {
       lines.push(``);
     }
   }
+  const cm = metrics.citations;
+  if (cm.totalCitations > 0) {
+    const bboxRate = Math.round(cm.withBbox / cm.totalCitations * 100);
+    lines.push(`### Citation Quality`);
+    lines.push(`- Bbox coverage: ${bboxRate}% (target: 100%)`);
+    lines.push(`- Avg citations per answer: ${cm.avgPerAnswer.toFixed(1)}`);
+    if (cm.withoutBbox > 0) {
+      lines.push(`- ${cm.withoutBbox} citations missing bbox \u2014 agent should always read .json files`);
+    }
+    if (cm.answersWithoutCitations > 0) {
+      lines.push(`- ${cm.answersWithoutCitations} answers had no citations \u2014 every answer must cite sources`);
+    }
+    lines.push(``);
+  }
   const hitRate = metrics.totalQAs > 0 ? Math.round(metrics.wikiHits / metrics.totalQAs * 100) : 0;
   lines.push(`### Performance`);
   lines.push(`- Wiki hit rate: ${hitRate}% (target: 80%+)`);
@@ -1286,8 +682,8 @@ function buildAgentsInsights(result) {
   return lines.join("\n");
 }
 async function runEval(kbRoot, options) {
-  const sessionsDir = join7(kbRoot, ".llm-kb", "sessions");
-  const sourcesDir = join7(kbRoot, ".llm-kb", "wiki", "sources");
+  const sessionsDir = join4(kbRoot, ".llm-kb", "sessions");
+  const sourcesDir = join4(kbRoot, ".llm-kb", "wiki", "sources");
   const log = options.onProgress ?? (() => {
   });
   log("Reading sessions...");
@@ -1295,7 +691,7 @@ async function runEval(kbRoot, options) {
   log(`Found ${qas.length} Q&A exchanges across sessions`);
   if (qas.length === 0) {
     return {
-      metrics: { totalSessions: 0, totalQAs: 0, avgDurationMs: 0, wikiHits: 0, sourceReads: 0, totalFilesRead: 0, uniqueFilesRead: /* @__PURE__ */ new Map(), wastedReads: 0 },
+      metrics: { totalSessions: 0, totalQAs: 0, avgDurationMs: 0, wikiHits: 0, sourceReads: 0, totalFilesRead: 0, uniqueFilesRead: /* @__PURE__ */ new Map(), wastedReads: 0, citations: { totalCitations: 0, withBbox: 0, withoutBbox: 0, multiPage: 0, avgPerAnswer: 0, answersWithCitations: 0, answersWithoutCitations: 0 } },
       issues: [],
       wikiGaps: [],
       timestamp: (/* @__PURE__ */ new Date()).toISOString()
@@ -1303,17 +699,16 @@ async function runEval(kbRoot, options) {
   }
   log("Calculating metrics...");
   const metrics = calculateMetrics(qas);
-  const apiKey = await resolveApiKey2(options.authStorage);
   const allIssues = [];
-  if (apiKey) {
-    const modelId = "claude-haiku-4-5";
+  const modelId = "claude-haiku-4-5";
+  try {
     for (let i = 0; i < qas.length; i++) {
       log(`Judging ${i + 1}/${qas.length}: "${qas[i].question.slice(0, 50)}..."`);
-      const issues = await judgeQA(qas[i], apiKey, modelId);
+      const issues = await judgeQA(qas[i], modelId, options.authStorage);
       allIssues.push(...issues);
     }
-  } else {
-    log("No API key \u2014 skipping LLM judge checks");
+  } catch (err) {
+    log(`LLM judge unavailable \u2014 skipping (${err instanceof Error ? err.message : String(err)})`);
   }
   const wikiGaps = allIssues.filter((i) => i.type === "wiki-gap").map((i) => i.detail);
   const result = {
@@ -1325,11 +720,11 @@ async function runEval(kbRoot, options) {
   };
   result.agentsInsights = buildAgentsInsights(result);
   log("Writing eval report + insights...");
-  const outputsDir = join7(kbRoot, ".llm-kb", "wiki", "outputs");
-  await mkdir6(outputsDir, { recursive: true });
+  const outputsDir = join4(kbRoot, ".llm-kb", "wiki", "outputs");
+  await mkdir3(outputsDir, { recursive: true });
   const report = buildReport(result);
-  await writeFile5(join7(outputsDir, "eval-report.md"), report, "utf-8");
-  const guidelinesPath = join7(kbRoot, ".llm-kb", "guidelines.md");
+  await writeFile3(join4(outputsDir, "eval-report.md"), report, "utf-8");
+  const guidelinesPath = join4(kbRoot, ".llm-kb", "guidelines.md");
   await writeGuidelines(guidelinesPath, result.agentsInsights);
   log("Insights saved to .llm-kb/guidelines.md (agent reads on-demand)");
   return result;
@@ -1338,20 +733,20 @@ var EVAL_SECTION_RE = /## Eval Insights[\s\S]*?(?=\n## |$)/;
 async function writeGuidelines(path3, evalSection) {
   let existing = "";
   try {
-    existing = await readFile5(path3, "utf-8");
+    existing = await readFile2(path3, "utf-8");
   } catch {
   }
   if (!existing) {
-    await writeFile5(path3, evalSection, "utf-8");
+    await writeFile3(path3, evalSection, "utf-8");
     return;
   }
   if (EVAL_SECTION_RE.test(existing)) {
-    await writeFile5(path3, existing.replace(EVAL_SECTION_RE, evalSection.trim()), "utf-8");
+    await writeFile3(path3, existing.replace(EVAL_SECTION_RE, evalSection.trim()), "utf-8");
   } else {
-    await writeFile5(path3, evalSection + "\n\n" + existing, "utf-8");
+    await writeFile3(path3, evalSection + "\n\n" + existing, "utf-8");
   }
 }
-function extractText2(content) {
+function extractText(content) {
   if (!content) return "";
   if (typeof content === "string") return content;
   if (Array.isArray(content)) {
@@ -5247,33 +4642,33 @@ var ProcessTerminal = class {
 };
 
 // src/tui-display.ts
-import chalk4 from "chalk";
+import chalk2 from "chalk";
 function createMarkdownTheme() {
   return {
-    heading: (t) => chalk4.bold(t),
-    link: (t) => chalk4.cyan(t),
-    linkUrl: (t) => chalk4.dim(t),
-    code: (t) => chalk4.cyan(t),
-    codeBlock: (t) => chalk4.dim(t),
-    codeBlockBorder: (t) => chalk4.dim(t),
-    quote: (t) => chalk4.italic(t),
-    quoteBorder: (t) => chalk4.dim(t),
-    hr: (t) => chalk4.dim(t),
-    listBullet: (t) => chalk4.dim(t),
-    bold: (t) => chalk4.bold(t),
-    italic: (t) => chalk4.italic(t),
-    underline: (t) => chalk4.underline(t),
-    strikethrough: (t) => chalk4.strikethrough(t)
+    heading: (t) => chalk2.bold(t),
+    link: (t) => chalk2.cyan(t),
+    linkUrl: (t) => chalk2.dim(t),
+    code: (t) => chalk2.cyan(t),
+    codeBlock: (t) => chalk2.dim(t),
+    codeBlockBorder: (t) => chalk2.dim(t),
+    quote: (t) => chalk2.italic(t),
+    quoteBorder: (t) => chalk2.dim(t),
+    hr: (t) => chalk2.dim(t),
+    listBullet: (t) => chalk2.dim(t),
+    bold: (t) => chalk2.bold(t),
+    italic: (t) => chalk2.italic(t),
+    underline: (t) => chalk2.underline(t),
+    strikethrough: (t) => chalk2.strikethrough(t)
   };
 }
 var mdTheme = createMarkdownTheme();
 function dimText(text, px = 1, py = 0) {
-  return new Text(chalk4.dim(text), px, py);
+  return new Text(chalk2.dim(text), px, py);
 }
 var HRule = class {
   colorFn;
   constructor(colorFn) {
-    this.colorFn = colorFn ?? chalk4.dim;
+    this.colorFn = colorFn ?? chalk2.dim;
   }
   invalidate() {
   }
@@ -5295,6 +4690,8 @@ var ChatDisplay = class {
   // active thinking block
   hadSeparator = false;
   // has a ─── line been drawn?
+  accumulatedAnswer = "";
+  // full answer text for citation stripping
   filesReadCount = 0;
   shownToolCalls = /* @__PURE__ */ new Set();
   startTime = Date.now();
@@ -5306,7 +4703,7 @@ var ChatDisplay = class {
     this.messageArea = new Container();
     this.tui.addChild(this.messageArea);
     this.inputArea = new Container();
-    this.inputArea.addChild(new HRule((s) => chalk4.hex("#c678dd")(s)));
+    this.inputArea.addChild(new HRule((s) => chalk2.hex("#c678dd")(s)));
     this.input = new Input();
     this.input.onSubmit = (text) => {
       if (text.trim() && this.onSubmit) {
@@ -5316,7 +4713,7 @@ var ChatDisplay = class {
       this.input.setValue("");
     };
     this.inputArea.addChild(this.input);
-    this.inputArea.addChild(new HRule((s) => chalk4.hex("#c678dd")(s)));
+    this.inputArea.addChild(new HRule((s) => chalk2.hex("#c678dd")(s)));
     this.tui.addChild(this.inputArea);
     this.tui.setFocus(this.input);
   }
@@ -5338,7 +4735,7 @@ var ChatDisplay = class {
   }
   addUserMessage(text) {
     this.messageArea.addChild(new Spacer(1));
-    this.messageArea.addChild(new Text(chalk4.bold(text), 1, 0));
+    this.messageArea.addChild(new Text(chalk2.bold(text), 1, 0));
     this.tui.requestRender();
   }
   // ── Per-prompt lifecycle (events arrive in any order) ───────────────────
@@ -5349,6 +4746,7 @@ var ChatDisplay = class {
     this.currentMd = null;
     this.currentThinking = null;
     this.hadSeparator = false;
+    this.accumulatedAnswer = "";
     this.currentResponse = new Container();
     this.currentResponse.addChild(new Spacer(1));
     this.currentResponse.addChild(dimText(`\u27E1 ${modelName}`));
@@ -5362,12 +4760,12 @@ var ChatDisplay = class {
     if (!this.currentThinking) {
       this.currentResponse.addChild(new Spacer(1));
       this.currentResponse.addChild(dimText("\u25B8 Thinking"));
-      this.currentThinking = new Text(chalk4.dim(chalk4.italic(text)), 2, 0);
+      this.currentThinking = new Text(chalk2.dim(chalk2.italic(text)), 2, 0);
       this.currentResponse.addChild(this.currentThinking);
     } else {
       const prev = this.currentThinking.text ?? "";
       this.currentThinking.setText(
-        chalk4.dim(chalk4.italic(prev.replace(/\x1b\[[0-9;]*m/g, "") + text))
+        chalk2.dim(chalk2.italic(prev.replace(/\x1b\[[0-9;]*m/g, "") + text))
       );
     }
     this.tui.requestRender();
@@ -5396,7 +4794,7 @@ var ChatDisplay = class {
   addToolResult(toolCallId, isError) {
     if (!this.currentResponse) return;
     if (isError) {
-      this.currentResponse.addChild(new Text(chalk4.red("    \u2717 failed"), 0, 0));
+      this.currentResponse.addChild(new Text(chalk2.red("    \u2717 failed"), 0, 0));
       this.tui.requestRender();
     }
   }
@@ -5427,19 +4825,61 @@ var ChatDisplay = class {
     }
     const prev = this.currentMd.text ?? "";
     this.currentMd.setText(prev + text);
+    this.accumulatedAnswer += text;
     this.tui.requestRender();
   }
-  showCompletion() {
+  /** Strip CITATIONS block from answer and show formatted citation footer */
+  showCitations(citations) {
+    if (!this.currentResponse || citations.length === 0) return;
+    if (this.currentMd) {
+      const citIdx = this.accumulatedAnswer.search(/^CITATIONS:\s*$/im);
+      if (citIdx >= 0) {
+        const clean = this.accumulatedAnswer.slice(0, citIdx).trimEnd();
+        this.currentMd.setText(clean);
+      }
+    }
+    this.currentResponse.addChild(new Spacer(1));
+    this.currentResponse.addChild(new HRule());
+    for (let i = 0; i < citations.length; i++) {
+      const c = citations[i];
+      const num = chalk2.bold(`  [${i + 1}]`);
+      const file = chalk2.cyan(c.file);
+      const pageStr = c.pages && c.pages.length > 0 ? `p.${c.pages.map((p) => p.page).join("-")}` : `p.${c.page}`;
+      let status;
+      if (c.bbox || c.pages && c.pages.length > 0) {
+        const bboxInfo = c.pages && c.pages.length > 0 ? `(${c.pages.length} pages)` : c.bbox ? `(${c.bbox.x},${c.bbox.y} \u2192 ${Math.round(c.bbox.x + c.bbox.width)},${Math.round(c.bbox.y + c.bbox.height)})` : "";
+        status = chalk2.green(`\u2705 bbox ${bboxInfo}`);
+      } else {
+        status = chalk2.yellow(`\u26A0\uFE0F  no bbox`);
+      }
+      const quote = c.quote.length > 60 ? c.quote.slice(0, 57) + "..." : c.quote;
+      this.currentResponse.addChild(
+        new Text(`${num} \u{1F4C4} ${file}, ${pageStr}`, 0, 0)
+      );
+      this.currentResponse.addChild(
+        new Text(chalk2.dim(`      "${quote}"`), 0, 0)
+      );
+      this.currentResponse.addChild(
+        new Text(`      ${status}`, 0, 0)
+      );
+    }
+    this.tui.requestRender();
+  }
+  showCompletion(citations) {
     if (!this.currentResponse) return;
+    if (citations && citations.length > 0) {
+      this.showCitations(citations);
+    }
     const elapsed = ((Date.now() - this.startTime) / 1e3).toFixed(1);
     const source = this.filesReadCount > 0 ? `${this.filesReadCount} file${this.filesReadCount !== 1 ? "s" : ""} read` : "wiki";
-    const stats = `\u2500\u2500 ${elapsed}s \xB7 ${source} `;
+    const citCount = citations && citations.length > 0 ? ` \xB7 ${citations.length} citation${citations.length !== 1 ? "s" : ""}` : "";
+    const stats = `\u2500\u2500 ${elapsed}s \xB7 ${source}${citCount} `;
     const completion = {
       invalidate() {
       },
       render(width) {
         const pad = Math.max(0, width - stats.length);
-        return [chalk4.dim(stats + "\u2500".repeat(pad))];
+        return [chalk2.dim(stats + "\u2500".repeat(pad))];
       }
     };
     this.currentResponse.addChild(new Spacer(1));
@@ -5459,12 +4899,12 @@ var ChatDisplay = class {
 };
 
 // src/resolve-kb.ts
-import { existsSync as existsSync6 } from "fs";
-import { resolve as resolve2, join as join10, dirname as dirname2 } from "path";
+import { existsSync as existsSync3 } from "fs";
+import { resolve as resolve2, join as join7, dirname as dirname2 } from "path";
 function resolveKnowledgeBase(startDir) {
   let dir = resolve2(startDir);
   while (true) {
-    if (existsSync6(join10(dir, ".llm-kb"))) {
+    if (existsSync3(join7(dir, ".llm-kb"))) {
       return dir;
     }
     const parent = dirname2(dir);
@@ -5474,39 +4914,53 @@ function resolveKnowledgeBase(startDir) {
 }
 
 // src/auth.ts
-import { existsSync as existsSync7 } from "fs";
-import { join as join11 } from "path";
-import { homedir as homedir4 } from "os";
-import { AuthStorage as AuthStorage4 } from "@mariozechner/pi-coding-agent";
-import chalk5 from "chalk";
+import { existsSync as existsSync4 } from "fs";
+import { join as join8 } from "path";
+import { homedir as homedir2 } from "os";
+import { AuthStorage } from "@mariozechner/pi-coding-agent";
+import chalk3 from "chalk";
 function checkAuth() {
-  const piAuthPath = join11(homedir4(), ".pi", "agent", "auth.json");
-  if (existsSync7(piAuthPath)) {
-    const authStorage = AuthStorage4.create();
-    return { ok: true, method: "pi-sdk", authStorage };
+  const piAuthPath = join8(homedir2(), ".pi", "agent", "auth.json");
+  if (existsSync4(piAuthPath)) {
+    const authStorage = AuthStorage.create();
+    return { ok: true, method: "pi-sdk", authStorage, providers: ["pi-sdk"] };
   }
+  const providers = {};
+  const names = [];
   if (process.env.ANTHROPIC_API_KEY) {
-    const authStorage = AuthStorage4.inMemory({
-      anthropic: { type: "api_key", key: process.env.ANTHROPIC_API_KEY }
-    });
-    return { ok: true, method: "api-key", authStorage };
+    providers.anthropic = { type: "api_key", key: process.env.ANTHROPIC_API_KEY };
+    names.push("anthropic");
+  }
+  if (process.env.OPENROUTER_API_KEY) {
+    providers.openrouter = { type: "api_key", key: process.env.OPENROUTER_API_KEY };
+    names.push("openrouter");
+  }
+  if (process.env.OPENAI_API_KEY) {
+    providers.openai = { type: "api_key", key: process.env.OPENAI_API_KEY };
+    names.push("openai");
+  }
+  if (names.length > 0) {
+    const authStorage = AuthStorage.inMemory(providers);
+    return { ok: true, method: "env", authStorage, providers: names };
   }
   return { ok: false };
 }
 function exitWithAuthError() {
-  console.error(chalk5.red("\n  No LLM authentication found.\n"));
-  console.error(`  ${chalk5.bold("Option 1:")} Install Pi SDK ${chalk5.dim("(recommended)")}`);
-  console.error(chalk5.dim("    npm install -g @mariozechner/pi-coding-agent"));
-  console.error(chalk5.dim("    pi\n"));
-  console.error(`  ${chalk5.bold("Option 2:")} Set your Anthropic API key`);
-  console.error(chalk5.dim("    export ANTHROPIC_API_KEY=sk-ant-...\n"));
+  console.error(chalk3.red("\n  No LLM authentication found.\n"));
+  console.error(`  ${chalk3.bold("Option 1:")} Install Pi SDK ${chalk3.dim("(recommended)")}`);
+  console.error(chalk3.dim("    npm install -g @mariozechner/pi-coding-agent"));
+  console.error(chalk3.dim("    pi\n"));
+  console.error(`  ${chalk3.bold("Option 2:")} Set one or more provider API keys`);
+  console.error(chalk3.dim("    export ANTHROPIC_API_KEY=sk-ant-..."));
+  console.error(chalk3.dim("    export OPENROUTER_API_KEY=sk-or-..."));
+  console.error(chalk3.dim("    export OPENAI_API_KEY=sk-...\n"));
   process.exit(1);
 }
 
 // src/config.ts
-import { existsSync as existsSync8 } from "fs";
-import { readFile as readFile6, writeFile as writeFile6, mkdir as mkdir7 } from "fs/promises";
-import { join as join12 } from "path";
+import { existsSync as existsSync5 } from "fs";
+import { readFile as readFile3, writeFile as writeFile4, mkdir as mkdir4 } from "fs/promises";
+import { join as join9 } from "path";
 var DEFAULT_INDEX_MODEL = "claude-haiku-4-5";
 var DEFAULT_QUERY_MODEL = "claude-sonnet-4-6";
 var DEFAULTS = {
@@ -5514,14 +4968,14 @@ var DEFAULTS = {
   queryModel: DEFAULT_QUERY_MODEL
 };
 function configPath(kbRoot) {
-  return join12(kbRoot, ".llm-kb", "config.json");
+  return join9(kbRoot, ".llm-kb", "config.json");
 }
 async function loadConfig(kbRoot) {
   let base = { ...DEFAULTS };
   const path3 = configPath(kbRoot);
-  if (existsSync8(path3)) {
+  if (existsSync5(path3)) {
     try {
-      const raw = await readFile6(path3, "utf-8");
+      const raw = await readFile3(path3, "utf-8");
       const parsed = JSON.parse(raw);
       if (parsed.indexModel) base.indexModel = parsed.indexModel;
       if (parsed.queryModel) base.queryModel = parsed.queryModel;
@@ -5534,29 +4988,35 @@ async function loadConfig(kbRoot) {
 }
 async function ensureConfig(kbRoot) {
   const path3 = configPath(kbRoot);
-  if (!existsSync8(path3)) {
-    await mkdir7(join12(kbRoot, ".llm-kb"), { recursive: true });
-    await writeFile6(path3, JSON.stringify(DEFAULTS, null, 2) + "\n", "utf-8");
+  if (!existsSync5(path3)) {
+    await mkdir4(join9(kbRoot, ".llm-kb"), { recursive: true });
+    await writeFile4(path3, JSON.stringify(DEFAULTS, null, 2) + "\n", "utf-8");
     return { ...DEFAULTS };
   }
   return loadConfig(kbRoot);
 }
 
 // src/cli.ts
-import { existsSync as existsSync9 } from "fs";
-import { mkdir as mkdir8, readdir as readdir6, stat as stat2 } from "fs/promises";
-import { resolve as resolve3, join as join13 } from "path";
-import chalk6 from "chalk";
+import { existsSync as existsSync6 } from "fs";
+import { readFileSync } from "fs";
+import { mkdir as mkdir5, readdir as readdir4, stat as stat2 } from "fs/promises";
+import { resolve as resolve3, join as join10, dirname as dirname3 } from "path";
+import { fileURLToPath } from "url";
+import chalk4 from "chalk";
+var __filename = fileURLToPath(import.meta.url);
+var __dirname = dirname3(__filename);
+var pkg = JSON.parse(readFileSync(join10(__dirname, "..", "package.json"), "utf-8"));
+var VERSION = pkg.version;
 var program = new Command();
-program.name("llm-kb").description("Drop files into a folder. Get a knowledge base you can query.").version("0.4.0");
+program.name("llm-kb").description("Drop files into a folder. Get a knowledge base you can query.").version(VERSION);
 program.command("run").description("Scan, parse, index, and watch a folder").argument("<folder>", "Path to your documents folder").action(async (folder) => {
   console.log(`
-${chalk6.bold("llm-kb")} v0.4.0
+${chalk4.bold("llm-kb")} v${VERSION}
 `);
   const auth = checkAuth();
   if (!auth.ok) exitWithAuthError();
-  if (!existsSync9(folder)) {
-    console.error(chalk6.red(`Error: Folder not found: ${folder}`));
+  if (!existsSync6(folder)) {
+    console.error(chalk4.red(`Error: Folder not found: ${folder}`));
     process.exit(1);
   }
   const root = resolve3(folder);
@@ -5564,72 +5024,95 @@ ${chalk6.bold("llm-kb")} v0.4.0
   console.log(`Scanning ${folder}...`);
   const files = await scan(folder);
   if (files.length === 0) {
-    console.log(chalk6.yellow("  No supported files found."));
+    console.log(chalk4.yellow("  No supported files found."));
     return;
   }
   const pdfs = files.filter((f) => f.ext === ".pdf");
-  console.log(`  Found ${chalk6.bold(files.length.toString())} files (${summarize(files)})`);
+  console.log(`  Found ${chalk4.bold(files.length.toString())} files (${summarize(files)})`);
   if (pdfs.length === 0) return;
-  const sourcesDir = join13(root, ".llm-kb", "wiki", "sources");
-  await mkdir8(sourcesDir, { recursive: true });
+  const sourcesDir = join10(root, ".llm-kb", "wiki", "sources");
+  await mkdir5(sourcesDir, { recursive: true });
   let parsed = 0;
   let skipped = 0;
   let failed = 0;
   const errors = [];
-  for (let i = 0; i < pdfs.length; i++) {
-    const pdf = pdfs[i];
-    const progress = `  Parsing... ${i + 1}/${pdfs.length} \u2014 ${pdf.name}`;
-    process.stdout.write(`\r${progress.padEnd(process.stdout.columns || 80)}`);
-    try {
-      const result = await parsePDF(join13(root, pdf.path), sourcesDir);
-      if (result.skipped) skipped++;
-      else parsed++;
-    } catch (err) {
-      failed++;
-      errors.push({ name: pdf.name, message: err.message });
+  const CONCURRENCY = Math.min(8, pdfs.length);
+  let nextIdx = 0;
+  let completed = 0;
+  const parseStart = Date.now();
+  let activeCount = 0;
+  function renderProgress(currentFile) {
+    const cols = process.stdout.columns || 80;
+    const pct = Math.round(completed / pdfs.length * 100);
+    const barWidth = Math.min(30, cols - 40);
+    const filled = Math.round(completed / pdfs.length * barWidth);
+    const active = Math.min(activeCount, barWidth - filled);
+    const bar = chalk4.green("\u2588".repeat(filled)) + chalk4.yellow("\u2593".repeat(active)) + chalk4.dim("\u2591".repeat(barWidth - filled - active));
+    const elapsed = ((Date.now() - parseStart) / 1e3).toFixed(0);
+    const line = `  ${bar} ${chalk4.bold(`${completed}/${pdfs.length}`)} ${chalk4.cyan(`(${pct}%)`)} ${chalk4.dim(`${elapsed}s`)} ${chalk4.yellow(`${activeCount} active`)}`;
+    process.stdout.write(`\r${line.padEnd(cols)}`);
+  }
+  async function worker() {
+    while (nextIdx < pdfs.length) {
+      const i = nextIdx++;
+      const pdf = pdfs[i];
+      activeCount++;
+      renderProgress(pdf.name);
+      try {
+        const result = await parsePDF(join10(root, pdf.path), sourcesDir);
+        if (result.skipped) skipped++;
+        else parsed++;
+      } catch (err) {
+        failed++;
+        errors.push({ name: pdf.name, message: err.message });
+      }
+      activeCount--;
+      completed++;
+      renderProgress(pdf.name);
     }
   }
+  await Promise.all(Array.from({ length: CONCURRENCY }, () => worker()));
   process.stdout.write(`\r${"".padEnd(process.stdout.columns || 80)}\r`);
   const parts = [];
-  if (parsed > 0) parts.push(chalk6.green(`${parsed} parsed`));
-  if (skipped > 0) parts.push(chalk6.dim(`${skipped} skipped (up to date)`));
-  if (failed > 0) parts.push(chalk6.red(`${failed} failed`));
+  if (parsed > 0) parts.push(chalk4.green(`${parsed} parsed`));
+  if (skipped > 0) parts.push(chalk4.dim(`${skipped} skipped (up to date)`));
+  if (failed > 0) parts.push(chalk4.red(`${failed} failed`));
   console.log(`  ${parts.join(", ")}`);
-  for (const err of errors) console.log(chalk6.red(`    \u2717 ${err.name} \u2014 ${err.message}`));
-  const indexFile = join13(root, ".llm-kb", "wiki", "index.md");
+  for (const err of errors) console.log(chalk4.red(`    \u2717 ${err.name} \u2014 ${err.message}`));
+  const indexFile = join10(root, ".llm-kb", "wiki", "index.md");
   let indexUpToDate = false;
-  if (parsed === 0 && existsSync9(indexFile)) {
+  if (parsed === 0 && existsSync6(indexFile)) {
     try {
       const indexMtime = (await stat2(indexFile)).mtimeMs;
-      const sourceFiles = await readdir6(sourcesDir);
-      const mtimes = await Promise.all(sourceFiles.map((f) => stat2(join13(sourcesDir, f)).then((s) => s.mtimeMs)));
+      const sourceFiles = await readdir4(sourcesDir);
+      const mtimes = await Promise.all(sourceFiles.map((f) => stat2(join10(sourcesDir, f)).then((s) => s.mtimeMs)));
       indexUpToDate = mtimes.every((mt) => indexMtime >= mt);
     } catch {
     }
   }
   if (indexUpToDate) {
-    console.log(chalk6.dim(`
+    console.log(chalk4.dim(`
   Index up to date.`));
   } else {
     console.log(`
-  Building index... ${chalk6.dim(`(${config.indexModel})`)}`);
+  Building index... ${chalk4.dim(`(${config.indexModel})`)}`);
     try {
       await buildIndex(root, sourcesDir, void 0, auth.authStorage, config.indexModel);
-      console.log(chalk6.green(`  Index built: .llm-kb/wiki/index.md`));
+      console.log(chalk4.green(`  Index built: .llm-kb/wiki/index.md`));
     } catch (err) {
-      console.error(chalk6.red(`  Index failed: ${err.message}`));
+      console.error(chalk4.red(`  Index failed: ${err.message}`));
     }
   }
   console.log(`
-  ${chalk6.dim("Output:")} ${sourcesDir}`);
-  startWatcher({ folder: root, sourcesDir, authStorage: auth.authStorage, indexModel: config.indexModel });
-  startSessionWatcher(root);
+  ${chalk4.dim("Output:")} ${sourcesDir}`);
   const chatUI = new ChatDisplay();
-  const { session, display } = await createChat(root, {
+  const { session, display, reloadSources } = await createChat(root, {
     authStorage: auth.authStorage,
     modelId: config.queryModel,
     tuiDisplay: chatUI
   });
+  startWatcher({ folder: root, sourcesDir, authStorage: auth.authStorage, indexModel: config.indexModel, onSourcesChanged: reloadSources });
+  startSessionWatcher(root);
   chatUI.onSubmit = (text) => {
     display.setQuestion(text);
     session.prompt(text).catch(() => {
@@ -5642,7 +5125,7 @@ ${chalk6.bold("llm-kb")} v0.4.0
     });
   };
   console.log(`
-${chalk6.bold("Ready.")} Ask a question or drop files in to re-index.
+${chalk4.bold("Ready.")} Ask a question or drop files in to re-index.
 `);
   chatUI.start();
 });
@@ -5651,7 +5134,7 @@ program.command("query").description("Ask a single question (non-interactive, st
   if (!auth.ok) exitWithAuthError();
   const root = resolveKnowledgeBase(options.folder || process.cwd());
   if (!root) {
-    console.error(chalk6.red("No knowledge base found. Run 'llm-kb run <folder>' first."));
+    console.error(chalk4.red("No knowledge base found. Run 'llm-kb run <folder>' first."));
     process.exit(1);
   }
   const config = await loadConfig(root);
@@ -5662,7 +5145,7 @@ program.command("query").description("Ask a single question (non-interactive, st
       modelId: config.queryModel
     });
   } catch (err) {
-    console.error(chalk6.red(err.message));
+    console.error(chalk4.red(err.message));
     process.exit(1);
   }
 });
@@ -5671,46 +5154,49 @@ program.command("eval").description("Analyze sessions for quality issues, wiki g
   if (!auth.ok) exitWithAuthError();
   const root = resolveKnowledgeBase(options.folder || process.cwd());
   if (!root) {
-    console.error(chalk6.red("No knowledge base found. Run 'llm-kb run <folder>' first."));
+    console.error(chalk4.red("No knowledge base found. Run 'llm-kb run <folder>' first."));
     process.exit(1);
   }
   console.log(`
-${chalk6.bold("llm-kb eval")}
+${chalk4.bold("llm-kb eval")}
 `);
   const result = await runEval(root, {
     authStorage: auth.authStorage,
     last: options.last,
-    onProgress: (msg) => console.log(chalk6.dim(`  ${msg}`))
+    onProgress: (msg) => console.log(chalk4.dim(`  ${msg}`))
   });
   const { metrics, issues, wikiGaps } = result;
   const errors = issues.filter((i) => i.severity === "error").length;
   const warnings = issues.filter((i) => i.severity === "warning").length;
   console.log();
-  console.log(`  ${chalk6.bold("Results:")}`);
+  console.log(`  ${chalk4.bold("Results:")}`);
   console.log(`  Queries analyzed:  ${metrics.totalQAs}`);
   console.log(`  Wiki hit rate:     ${metrics.totalQAs > 0 ? Math.round(metrics.wikiHits / metrics.totalQAs * 100) : 0}%`);
   console.log(`  Wasted reads:      ${metrics.wastedReads}`);
-  console.log(`  Issues:            ${errors > 0 ? chalk6.red(`${errors} errors`) : chalk6.green("0 errors")}  ${warnings > 0 ? chalk6.yellow(`${warnings} warnings`) : chalk6.dim("0 warnings")}`);
-  console.log(`  Wiki gaps:         ${wikiGaps.length > 0 ? chalk6.yellow(String(wikiGaps.length)) : chalk6.green("0")}`);
+  const cm = metrics.citations;
+  const bboxPct = cm.totalCitations > 0 ? Math.round(cm.withBbox / cm.totalCitations * 100) : 0;
+  console.log(`  Citations:         ${cm.totalCitations} total, ${chalk4.green(`${cm.withBbox} with bbox`)}${cm.withoutBbox > 0 ? chalk4.yellow(` ${cm.withoutBbox} without`) : ""} (${bboxPct}%)`);
+  console.log(`  Issues:            ${errors > 0 ? chalk4.red(`${errors} errors`) : chalk4.green("0 errors")}  ${warnings > 0 ? chalk4.yellow(`${warnings} warnings`) : chalk4.dim("0 warnings")}`);
+  console.log(`  Wiki gaps:         ${wikiGaps.length > 0 ? chalk4.yellow(String(wikiGaps.length)) : chalk4.green("0")}`);
   console.log();
-  console.log(chalk6.green(`  Report: .llm-kb/wiki/outputs/eval-report.md`));
+  console.log(chalk4.green(`  Report: .llm-kb/wiki/outputs/eval-report.md`));
   console.log();
 });
 program.command("status").description("Show knowledge base stats and current config").option("--folder <path>", "Path to document folder (auto-detects if omitted)").action(async (options) => {
   const root = resolveKnowledgeBase(options.folder || process.cwd());
   if (!root) {
-    console.error(chalk6.red("No knowledge base found. Run 'llm-kb run <folder>' first."));
+    console.error(chalk4.red("No knowledge base found. Run 'llm-kb run <folder>' first."));
     process.exit(1);
   }
   const auth = checkAuth();
   const config = await loadConfig(root);
-  const sourcesDir = join13(root, ".llm-kb", "wiki", "sources");
-  const indexFile = join13(root, ".llm-kb", "wiki", "index.md");
-  const articlesDir = join13(root, ".llm-kb", "wiki", "articles");
-  const outputsDir = join13(root, ".llm-kb", "wiki", "outputs");
+  const sourcesDir = join10(root, ".llm-kb", "wiki", "sources");
+  const indexFile = join10(root, ".llm-kb", "wiki", "index.md");
+  const articlesDir = join10(root, ".llm-kb", "wiki", "articles");
+  const outputsDir = join10(root, ".llm-kb", "wiki", "outputs");
   let sourceCount = 0;
   try {
-    sourceCount = (await readdir6(sourcesDir)).filter((f) => f.endsWith(".md")).length;
+    sourceCount = (await readdir4(sourcesDir)).filter((f) => f.endsWith(".md")).length;
   } catch {
   }
   let indexAge = "not built yet";
@@ -5721,23 +5207,48 @@ program.command("status").description("Show knowledge base stats and current con
   }
   let outputCount = 0;
   try {
-    outputCount = (await readdir6(outputsDir)).filter((f) => f.endsWith(".md")).length;
+    outputCount = (await readdir4(outputsDir)).filter((f) => f.endsWith(".md")).length;
   } catch {
   }
   console.log(`
-${chalk6.bold("Knowledge Base Status")}`);
-  console.log(`  ${chalk6.dim("Folder:")}  ${root}`);
-  console.log(`  ${chalk6.dim("Sources:")} ${sourceCount > 0 ? `${sourceCount} parsed source${sourceCount !== 1 ? "s" : ""}` : chalk6.yellow("none yet")}`);
-  console.log(`  ${chalk6.dim("Index:")}   ${indexAge}`);
+${chalk4.bold("Knowledge Base Status")}`);
+  console.log(`  ${chalk4.dim("Folder:")}  ${root}`);
+  console.log(`  ${chalk4.dim("Sources:")} ${sourceCount > 0 ? `${sourceCount} parsed source${sourceCount !== 1 ? "s" : ""}` : chalk4.yellow("none yet")}`);
+  console.log(`  ${chalk4.dim("Index:")}   ${indexAge}`);
   let articleCount = 0;
   try {
-    articleCount = (await readdir6(articlesDir)).filter((f) => f.endsWith(".md") && f !== "index.md").length;
+    articleCount = (await readdir4(articlesDir)).filter((f) => f.endsWith(".md") && f !== "index.md").length;
   } catch {
   }
-  if (articleCount > 0) console.log(`  ${chalk6.dim("Articles:")} ${articleCount} compiled`);
-  if (outputCount > 0) console.log(`  ${chalk6.dim("Outputs:")} ${outputCount} saved answer${outputCount !== 1 ? "s" : ""}`);
-  console.log(`  ${chalk6.dim("Models:")}  ${chalk6.cyan(config.queryModel)} ${chalk6.dim("(query)")}  ${chalk6.cyan(config.indexModel)} ${chalk6.dim("(index)")}`);
-  console.log(`  ${chalk6.dim("Auth:")}    ${auth.ok ? auth.method === "pi-sdk" ? "Pi SDK" : "ANTHROPIC_API_KEY" : chalk6.red("not configured")}`);
+  if (articleCount > 0) console.log(`  ${chalk4.dim("Articles:")} ${articleCount} compiled`);
+  if (outputCount > 0) console.log(`  ${chalk4.dim("Outputs:")} ${outputCount} saved answer${outputCount !== 1 ? "s" : ""}`);
+  console.log(`  ${chalk4.dim("Models:")}  ${chalk4.cyan(config.queryModel)} ${chalk4.dim("(query)")}  ${chalk4.cyan(config.indexModel)} ${chalk4.dim("(index)")}`);
+  console.log(`  ${chalk4.dim("Auth:")}    ${auth.ok ? auth.method === "pi-sdk" ? "Pi SDK" : `env (${auth.providers.join(", ")})` : chalk4.red("not configured")}`);
   console.log();
 });
+program.command("ui").description("Open the web UI with chat, citations, and source viewer").argument("<folder>", "Path to your documents folder").option("--port <n>", "Port number", parseInt, 3947).option("--no-open", "Don't auto-open the browser").action(async (folder, options) => {
+  console.log(`
+${chalk4.bold("llm-kb")} web UI
+`);
+  const auth = checkAuth();
+  if (!auth.ok) exitWithAuthError();
+  if (!existsSync6(folder)) {
+    console.error(chalk4.red(`Error: Folder not found: ${folder}`));
+    process.exit(1);
+  }
+  const root = resolve3(folder);
+  const config = await loadConfig(root);
+  if (!existsSync6(join10(root, ".llm-kb", "wiki", "sources"))) {
+    console.error(chalk4.red("No knowledge base found. Run 'llm-kb run <folder>' first."));
+    process.exit(1);
+  }
+  const { startWebUI } = await import("./server-QC5SN6T4.js");
+  await startWebUI({
+    folder: root,
+    port: options.port,
+    open: options.open,
+    authStorage: auth.authStorage,
+    modelId: config.queryModel
+  });
+});
 program.parse();