llm-kb 0.4.0 → 0.4.2

package/src/md-stream.ts

@@ -1,133 +0,0 @@
-import chalk from "chalk";
-
-/**
- * Streaming markdown renderer for terminal output.
- *
- * Processes text_delta chunks and applies ANSI styling as patterns complete.
- * Handles: **bold**, *italic*, `code`, ## headers, --- hr, • bullets,
- * > blockquotes, [links](url), ~~strikethrough~~, | tables.
- */
-export class MarkdownStream {
-  private buffer = "";
-  private isTTY: boolean;
-
-  constructor(isTTY = false) {
-    this.isTTY = isTTY;
-  }
-
-  /** Feed a text_delta chunk. Returns styled string ready for stdout. */
-  push(chunk: string): string {
-    if (!this.isTTY) return chunk;
-
-    this.buffer += chunk;
-    return this.drain(false);
-  }
-
-  /** Flush remaining buffer (call on text_end). */
-  end(): string {
-    if (!this.isTTY) return "";
-    const out = this.drain(true);
-    this.buffer = "";
-    return out;
-  }
-
-  private drain(final: boolean): string {
-    let out = "";
-
-    while (true) {
-      const nlIdx = this.buffer.indexOf("\n");
-
-      if (nlIdx === -1) {
-        if (final && this.buffer.length > 0) {
-          // Final flush — render whatever's left
-          out += this.renderLine(this.buffer);
-          this.buffer = "";
-        }
-        // else: wait for more data (incomplete line)
-        break;
-      }
-
-      // Complete line found — render it
-      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. */
-  private renderLine(line: string): string {
-    const trimmed = line.trimStart();
-
-    // Horizontal rule
-    if (/^-{3,}\s*$/.test(trimmed) || /^\*{3,}\s*$/.test(trimmed)) {
-      const cols = process.stdout.columns || 80;
-      return chalk.dim("\u2500".repeat(Math.min(cols, 60)));
-    }
-
-    // Headers — strip # prefix, render bold
-    const headerMatch = trimmed.match(/^(#{1,6})\s+(.*)$/);
-    if (headerMatch) {
-      const text = this.inline(headerMatch[2]);
-      return "\n" + chalk.bold(text);
-    }
-
-    // Bullet points
-    const bulletMatch = trimmed.match(/^[-*+]\s+(.*)$/);
-    if (bulletMatch) {
-      const indent = line.length - trimmed.length;
-      return " ".repeat(indent) + chalk.dim("\u2022") + " " + this.inline(bulletMatch[1]);
-    }
-
-    // Numbered lists
-    const numMatch = trimmed.match(/^(\d+)[.)]\s+(.*)$/);
-    if (numMatch) {
-      const indent = line.length - trimmed.length;
-      return " ".repeat(indent) + chalk.dim(numMatch[1] + ".") + " " + this.inline(numMatch[2]);
-    }
-
-    // Table separator row
-    if (/^\|[\s\-:|]+\|$/.test(trimmed)) {
-      return chalk.dim(trimmed);
-    }
-
-    // Table data row
-    if (trimmed.startsWith("|") && trimmed.endsWith("|")) {
-      return this.inline(line);
-    }
-
-    // Block quotes — support nested > > and inline formatting
-    if (trimmed.startsWith(">")) {
-      const content = trimmed.replace(/^>+\s*/, "");
-      return chalk.dim("\u2502 ") + chalk.italic(this.inline(content));
-    }
-
-    return this.inline(line);
-  }
-
-  /** Apply inline markdown styling to text. */
-  private inline(text: string): string {
-    // Code spans (before bold/italic to avoid conflicts inside backticks)
-    text = text.replace(/`([^`]+)`/g, (_, c) => chalk.cyan(c));
-
-    // Bold + italic
-    text = text.replace(/\*\*\*(.+?)\*\*\*/g, (_, t) => chalk.bold.italic(t));
-
-    // Bold
-    text = text.replace(/\*\*(.+?)\*\*/g, (_, t) => chalk.bold(t));
-
-    // Italic (single * not adjacent to another *)
-    text = text.replace(/(?<!\*)\*(.+?)\*(?!\*)/g, (_, t) => chalk.italic(t));
-
-    // Strikethrough
-    text = text.replace(/~~(.+?)~~/g, (_, t) => chalk.strikethrough(t));
-
-    // Links
-    text = text.replace(/\[([^\]]+)\]\(([^)]+)\)/g, (_, label, url) =>
-      `${label} ${chalk.dim(`(${url})`)}`
-    );
-
-    return text;
-  }
-}

package/src/pdf.ts

@@ -1,119 +0,0 @@
-import { LiteParse } from "@llamaindex/liteparse";
-import { writeFile, mkdir, stat } from "node:fs/promises";
-import { join, basename } from "node:path";
-import { cpus } from "node:os";
-
-export interface ParsedPDF {
-  name: string;
-  mdPath: string;
-  jsonPath: string;
-  totalPages: number;
-  textLength: number;
-  skipped: boolean;
-}
-
-/**
- * Check if source PDF is newer than the parsed output.
- * Returns true if we can skip parsing.
- */
-async function isUpToDate(
-  pdfPath: string,
-  mdPath: string,
-  jsonPath: string
-): Promise<boolean> {
-  try {
-    const [pdfStat, mdStat, jsonStat] = await Promise.all([
-      stat(pdfPath),
-      stat(mdPath),
-      stat(jsonPath),
-    ]);
-    return pdfStat.mtimeMs <= mdStat.mtimeMs && pdfStat.mtimeMs <= jsonStat.mtimeMs;
-  } catch {
-    return false;
-  }
-}
-
-/**
- * Suppress stderr temporarily to hide noisy library warnings.
- */
-function suppressStderr(): () => void {
-  const originalWrite = process.stderr.write.bind(process.stderr);
-  process.stderr.write = (() => true) as any;
-  return () => {
-    process.stderr.write = originalWrite;
-  };
-}
-
-export async function parsePDF(
-  pdfPath: string,
-  outputDir: string
-): Promise<ParsedPDF> {
-  const name = basename(pdfPath, ".pdf");
-  await mkdir(outputDir, { recursive: true });
-
-  const mdPath = join(outputDir, `${name}.md`);
-  const jsonPath = join(outputDir, `${name}.json`);
-
-  // Skip if already parsed and source hasn't changed
-  if (await isUpToDate(pdfPath, mdPath, jsonPath)) {
-    return { name, mdPath, jsonPath, totalPages: 0, textLength: 0, skipped: true };
-  }
-
-  const ocrServerUrl = process.env.OCR_SERVER_URL;
-  const ocrEnabled = ocrServerUrl ? true : process.env.OCR_ENABLED === "true";
-
-  const parser = new LiteParse({
-    ocrEnabled,
-    outputFormat: "json",
-    numWorkers: cpus().length,
-    ...(ocrServerUrl ? { ocrServerUrl } : {}),
-  });
-
-  // Suppress noisy Tesseract/PDF.js warnings during parse
-  const restore = suppressStderr();
-  let result;
-  try {
-    result = await parser.parse(pdfPath, true);
-  } finally {
-    restore();
-  }
-
-  // Build markdown — spatial text per page
-  const markdown = result.pages
-    .map((p: any) => `# Page ${p.pageNum}\n\n${p.text}`)
-    .join("\n\n---\n\n");
-
-  // Build bounding box JSON
-  const bboxData = {
-    source: basename(pdfPath),
-    totalPages: result.pages.length,
-    pages: result.pages.map((p: any) => ({
-      page: p.pageNum,
-      width: p.width,
-      height: p.height,
-      textItems: p.textItems.map((item: any) => ({
-        text: (item.str ?? item.text ?? "").trim(),
-        x: Math.round(item.x * 100) / 100,
-        y: Math.round(item.y * 100) / 100,
-        width: Math.round((item.width ?? item.w ?? 0) * 100) / 100,
-        height: Math.round((item.height ?? item.h ?? 0) * 100) / 100,
-        fontName: item.fontName,
-        fontSize: item.fontSize
-          ? Math.round(item.fontSize * 100) / 100
-          : undefined,
-      })),
-    })),
-  };
-
-  await writeFile(mdPath, markdown);
-  await writeFile(jsonPath, JSON.stringify(bboxData, null, 2));
-
-  return {
-    name,
-    mdPath,
-    jsonPath,
-    totalPages: result.pages.length,
-    textLength: markdown.length,
-    skipped: false,
-  };
-}

package/src/query.ts

@@ -1,408 +0,0 @@
-import {
-  createAgentSession,
-  createBashTool,
-  createReadTool,
-  createWriteTool,
-  DefaultResourceLoader,
-  SettingsManager,
-  AuthStorage,
-} from "@mariozechner/pi-coding-agent";
-import type { AgentSession } from "@mariozechner/pi-coding-agent";
-import { getModels } from "@mariozechner/pi-ai";
-import { readdir, mkdir, readFile } from "node:fs/promises";
-import { existsSync } from "node:fs";
-import { createKBSession, continueKBSession } from "./session-store.js";
-import { saveTrace, appendToQueryLog, KBTrace } from "./trace-builder.js";
-import { updateWiki } from "./wiki-updater.js";
-import { join, basename } from "node:path";
-import chalk from "chalk";
-import { getNodeModulesPath } from "./utils.js";
-import { MarkdownStream } from "./md-stream.js";
-import type { ChatDisplay } from "./tui-display.js";
-
-// ── Helpers ─────────────────────────────────────────────────────────────────
-
-function extractAnswerText(content: any[]): string {
-  return (content ?? [])
-    .filter((b: any) => b.type === "text")
-    .map((b: any) => b.text ?? "")
-    .join("")
-    .trim();
-}
-
-function extractFilesRead(messages: any[]): string[] {
-  const paths: string[] = [];
-  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: string = block.arguments?.path ?? "";
-        if (p && !paths.includes(p)) paths.push(p);
-      }
-    }
-  }
-  return paths;
-}
-
-function getToolLabel(toolName: string, args: any): string | null {
-  if (toolName === "read" || toolName === "write" || toolName === "edit") {
-    const file = basename((args?.path as string) ?? "");
-    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;
-}
-
-// ── AGENTS.md ───────────────────────────────────────────────────────────────
-
-function buildQueryAgents(sourceFiles: string[], save: boolean, wikiContent: string): string {
-  const sourceList = sourceFiles.map((f) => `  - ${f}`).join("\n");
-  const wikiSection = wikiContent
-    ? `## Knowledge Wiki (use this first)\n\nThe wiki below contains knowledge already extracted from this knowledge base.\nIf the user's question is covered here, answer directly from it — no need to re-read source files.\nAlways cite the original source files mentioned in the wiki.\n\n${wikiContent}\n\n---\n\n`
-    : "";
-  const sourceStep = wikiContent ? "If not covered in the wiki above: read the sources" : "How to answer";
-
-  const lines = [
-    `# llm-kb Knowledge Base — 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 — 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) — structured XML`,
-    `.docx files are ZIP archives containing word/document.xml.`,
-    `Read them SELECTIVELY — 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) — 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) — 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 — never dump entire files`,
-    `- Prefer primary sources over previous analyses`,
-  ];
-
-  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");
-}
-
-// ── Wiki update scheduler ───────────────────────────────────────────────────
-
-class WikiUpdateScheduler {
-  private stopMsgCount = 0;
-  private lastUpdateAt = 0;
-  private chain: Promise<void> = Promise.resolve();
-  constructor(private readonly everyN: number, private readonly everyMin: number) {}
-  private shouldUpdate() {
-    return (this.stopMsgCount > 0 && this.stopMsgCount % this.everyN === 0) ||
-      (this.lastUpdateAt > 0 && Date.now() - this.lastUpdateAt > this.everyMin * 60_000);
-  }
-  private enqueue(work: () => Promise<void>) { this.chain = this.chain.then(() => work().catch(() => {})); }
-  onMessageEnd(msg: any, snap: () => { messages: any[] }, doUpdate: (m: any[]) => Promise<void>) {
-    if (msg.role !== "assistant" || msg.stopReason !== "stop") return;
-    this.stopMsgCount++;
-    if (this.shouldUpdate()) { this.lastUpdateAt = Date.now(); this.enqueue(() => doUpdate(snap().messages)); }
-  }
-  onAgentEnd(msgs: any[], doUpdate: (m: any[]) => Promise<void>) {
-    this.lastUpdateAt = Date.now(); this.enqueue(() => doUpdate(msgs));
-  }
-  flush() { return this.chain; }
-}
-
-// ── Display subscriber ──────────────────────────────────────────────────────
-// Routes events to either TUI components (interactive) or stdout (one-shot)
-
-function subscribeDisplay(
-  session: AgentSession,
-  opts: {
-    modelId?: string;
-    authStorage?: AuthStorage;
-    folder: string;
-    mdFiles: string[];
-    tuiDisplay?: ChatDisplay;
-  }
-) {
-  const ui = opts.tuiDisplay;
-  const dim = (s: string) => process.stdout.isTTY ? chalk.dim(s) : s;
-  const thinLine = () => dim("\u2500".repeat(process.stdout.columns || 80));
-
-  let phase: "idle" | "thinking" | "tools" | "answer" = "idle";
-  let filesReadCount = 0;
-  let shownToolCalls = new Set<string>();
-  let startTime = Date.now();
-  let md = new MarkdownStream(process.stdout.isTTY ?? false);
-  let lastQuestion = "";
-
-  const scheduler = new WikiUpdateScheduler(5, 3);
-
-  const buildTrace = (messages: any[]): KBTrace | null => {
-    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: 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: any[]) => {
-    const trace = buildTrace(messages);
-    if (!trace) return;
-    await saveTrace(opts.folder, trace);
-    await appendToQueryLog(opts.folder, trace);
-    await updateWiki(opts.folder, trace, opts.authStorage);
-  };
-
-  session.subscribe((event) => {
-
-    // ── Reset ────────────────────────────────────────────────────────────
-    if (event.type === "agent_start") {
-      phase = "idle";
-      filesReadCount = 0;
-      shownToolCalls = 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");
-    }
-
-    // ── Thinking ─────────────────────────────────────────────────────────
-    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");
-      }
-    }
-
-    // ── Tool calls ───────────────────────────────────────────────────────
-    if (event.type === "message_update") {
-      const ae = event.assistantMessageEvent as any;
-      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);
-            // Show the actual bash code the agent wrote
-            if (ae.toolCall.name === "bash" && ae.toolCall.arguments?.command) {
-              ui.addCodeBlock(ae.toolCall.arguments.command);
-            }
-          } else {
-            process.stdout.write(dim(`  \u25b8 ${label}`) + "\n");
-            // Show bash code in stdout mode too
-            if (ae.toolCall.name === "bash" && ae.toolCall.arguments?.command) {
-              const code = ae.toolCall.arguments.command as string;
-              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 as any;
-      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++;
-        }
-      }
-    }
-
-    // tool result (show errors)
-    if (event.type === "tool_execution_end") {
-      const { toolCallId, isError } = event as any;
-      if (ui) ui.addToolResult(toolCallId, isError);
-    }
-
-    // ── Answer ───────────────────────────────────────────────────────────
-    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(`\n${thinLine()}\n\n`);
-        }
-        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());
-    }
-
-    // ── Completion ───────────────────────────────────────────────────────
-    if (event.type === "agent_end") {
-      if (ui) { ui.showCompletion(); ui.enableInput(); }
-      else {
-        const elapsed = ((Date.now() - startTime) / 1000).toFixed(1);
-        const source = filesReadCount > 0
-          ? `${filesReadCount} file${filesReadCount !== 1 ? "s" : ""} read` : "wiki";
-        const stats = `${elapsed}s \u00b7 ${source}`;
-        const cols = process.stdout.columns || 80;
-        const pad = Math.max(0, cols - stats.length - 4);
-        process.stdout.write(`\n\n${dim("\u2500\u2500 " + stats + " " + "\u2500".repeat(pad))}\n`);
-      }
-      scheduler.onAgentEnd(event.messages as any[], doUpdate);
-    }
-
-    // ── Wiki throttle ────────────────────────────────────────────────────
-    if (event.type === "message_end") {
-      scheduler.onMessageEnd(event.message, () => ({ messages: session.state.messages as any[] }), doUpdate);
-    }
-  });
-
-  return {
-    setQuestion(q: string) { lastQuestion = q; },
-    flush() { return scheduler.flush(); },
-  };
-}
-
-// ── Session factory ─────────────────────────────────────────────────────────
-
-export interface ChatSession {
-  session: AgentSession;
-  display: ReturnType<typeof subscribeDisplay>;
-}
-
-export async function createChat(
-  folder: string,
-  options: { save?: boolean; authStorage?: AuthStorage; modelId?: string; tuiDisplay?: ChatDisplay }
-): Promise<ChatSession> {
-  const sourcesDir = join(folder, ".llm-kb", "wiki", "sources");
-  const files = await readdir(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 mkdir(join(folder, ".llm-kb", "wiki", "outputs"), { recursive: true });
-
-  process.env.NODE_PATH = getNodeModulesPath();
-
-  const wikiPath = join(folder, ".llm-kb", "wiki", "wiki.md");
-  const wikiContent = existsSync(wikiPath) ? await readFile(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();
-
-  // Always include all tools — agent needs bash for .docx/.xlsx reading
-  const tools = [
-    createReadTool(folder),
-    createBashTool(folder),
-    createWriteTool(folder),
-  ];
-
-  const model = options.modelId ? getModels("anthropic").find((m) => m.id === options.modelId) : undefined;
-
-  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 };
-}
-
-// ── One-shot query (stdout mode, for `llm-kb query` command) ────────────────
-
-export async function query(
-  folder: string,
-  question: string,
-  options: { save?: boolean; authStorage?: AuthStorage; modelId?: string }
-): Promise<void> {
-  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 = join(folder, ".llm-kb", "wiki", "sources");
-    const { buildIndex } = await import("./indexer.js");
-    await buildIndex(folder, sourcesDir, undefined, options.authStorage);
-  }
-}

package/src/resolve-kb.ts

@@ -1,19 +0,0 @@
-import { existsSync } from "node:fs";
-import { resolve, join, dirname } from "node:path";
-
-/**
- * Walk up from startDir looking for a .llm-kb/ directory.
- * Returns the folder containing .llm-kb/, or null if not found.
- */
-export function resolveKnowledgeBase(startDir: string): string | null {
-  let dir = resolve(startDir);
-
-  while (true) {
-    if (existsSync(join(dir, ".llm-kb"))) {
-      return dir;
-    }
-    const parent = dirname(dir);
-    if (parent === dir) return null;
-    dir = parent;
-  }
-}