@ganziliang/kb 0.1.7 → 0.3.0

package/dist/cli.js

@@ -1,204 +1,470 @@
 #!/usr/bin/env node
-import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
-import { useEffect, useRef, useState } from "react";
-import { Box, Text, render, useApp, useInput } from "ink";
 import { mkdirSync } from "node:fs";
+import { Container, Editor, Loader, ProcessTerminal, SelectList, Spacer, TuiMainScreen, matchesKey, } from "@earendil-works/pi-tui";
 import { ensureModelConfig, saveModels } from "./config.js";
-import { Agent, describeImage, fetchTransport } from "./model.js";
-import { copyOriginal, defaultDataRoot, ensureKnowledgeBase, isImageFile, KnowledgeStore, readImage, readLocalFile, listKnowledgeBases, deleteKnowledgeBase, backupKnowledgeBase, restoreKnowledgeBase } from "./storage.js";
-const banner = [
-    "  _  __ ____  ",
-    " | |/ // __ ) ",
-    " |   // __  | ",
-    " |   |/ /_/ / ",
-    " |_|\\_\\____/ ",
-    " Welcome to KB Knowledge Base",
+import { SAVE_KNOWLEDGE_TOOL, describeImage, fetchTransport } from "./model.js";
+import { backupKnowledgeBase, copyOriginal, defaultDataRoot, deleteKnowledgeBase, ensureKnowledgeBase, isImageFile, KnowledgeStore, listKnowledgeBases, readImage, readLocalFile, restoreKnowledgeBase, writeNote, } from "./storage.js";
+import { color, editorTheme, selectListTheme } from "./theme.js";
+import { AssistantMessage, HelpPanel, ImageNotice, Notice, Rule, Splash, UserMessage } from "./ui.js";
+const errText = (error) => (error instanceof Error ? error.message : String(error));
+/**
+ * 显式录入命令，仅作为模型不可用时的兜底。
+ * 正常情况交给模型按语义判断（见 buildSystemPrompt），
+ * 因此这里只保留 / 开头的命令，不硬匹配「记住」这类自然语言词。
+ */
+const NOTE_PATTERN = /^\/(?:add|note)(?:\s+([\s\S]*))?$/;
+/** 从录入内容里推导一个标题：优先取 Markdown 标题，否则取首个非空行。 */
+function deriveTitle(content) {
+    const firstLine = content.split("\n").map((l) => l.trim()).find(Boolean) ?? "笔记";
+    const heading = firstLine.match(/^#{1,6}\s+(.+)$/);
+    const base = (heading ? heading[1] : firstLine).replace(/[#*`>_]/g, "").trim();
+    return base.slice(0, 40) || "���记";
+}
+/** 多行内容回显时只显示首行，避免刷屏。 */
+function shortPreview(text) {
+    const lines = text.split("\n");
+    return lines.length > 1 ? `${lines[0]} …（共 ${lines.length} 行）` : text;
+}
+/**
+ * 把「回答」和「录入」放进同一轮对话：模型自己决定是回答还是调用工具，
+ * 因此判断意图不会带来额外的一次模型调用。
+ */
+function buildSystemPrompt(context) {
+    return [
+        "你是本地知识库助手。每一轮输入，先判断用户意图：",
+        "",
+        "A) 提问 —— 用户想从知识库获取信息。",
+        "   用「知识库内容」回答，并标注来源文件名与版本。",
+        "   内容不足以回答时明确说明没有找到，不要编造。",
+        "",
+        "B) 录入 —— 用户想把一段内容长期存进知识库。",
+        "   调用 save_knowledge 工具，title 用一句话概括，content 用规范 Markdown 并完整保留用户信息。",
+        "   此时不要回答内容本身，也不要反问确认。",
+        "",
+        "判断依据是用户表达的意图，不要依赖固定关键词。",
+        "",
+        "知识库内容：",
+        "---",
+        context,
+        "---",
+    ].join("\n");
+}
+const HELP_ITEMS = [
+    ["/help", "显示本帮助"],
+    ["/clear", "清空对话上下文"],
+    ["/sources", "列出知识来源与版本"],
+    ["/add <内容>", "显式录入（通常直接说「记住…」即可）"],
+    ["/model", "查看 / 切换模型"],
+    ["/kb", "多知识库管理"],
+    ["/backup", "备份当前知识库"],
+    ["/restore", "��备份恢复"],
+    ["/cleanup", "清理历史版本"],
+    ["/quit", "退出"],
+    ["", ""],
+    ["导入 <路径>", "录入 md / txt / json / csv / xlsx / png / jpg"],
 ];
-function App({ config, root, knowledgeBase }) {
-    const { exit } = useApp();
-    const [input, setInput] = useState("");
-    const [lines, setLines] = useState([...banner, `knowledge base: ${knowledgeBase.name}`, "Type a question, or /help."]);
-    const [busy, setBusy] = useState(false);
-    const [messages, setMessages] = useState([]);
-    const [modelIndex, setModelIndex] = useState(0);
-    const [currentBase, setCurrentBase] = useState(knowledgeBase);
-    const storeRef = useRef(new KnowledgeStore(knowledgeBase.path));
-    const store = storeRef.current;
-    useEffect(() => () => storeRef.current.close(), []);
-    useInput((char, key) => {
-        if (key.ctrl && char === "c") {
-            store.close();
-            exit();
-            return;
+const ROLE = { user: "❯", assistant: "⏺", ok: "✔", err: "✖", warn: "!", image: "★" };
+export class KbApp {
+    root;
+    tui;
+    stream = new Container();
+    editor;
+    store;
+    base;
+    models;
+    modelIndex = 0;
+    messages = [];
+    turns = 0;
+    busy = false;
+    loader = null;
+    overlay = null;
+    closed = false;
+    constructor(root, base, models) {
+        this.root = root;
+        this.base = base;
+        this.models = models;
+        this.store = new KnowledgeStore(base.path);
+        this.tui = new TuiMainScreen(new ProcessTerminal());
+        this.editor = new Editor(this.tui, editorTheme, { paddingX: 1 });
+        this.editor.onSubmit = (text) => { void this.submit(text.trim()); };
+    }
+    // ---------- 启动 ----------
+    start() {
+        this.tui.addChild(new Splash({
+            base: this.base.name,
+            model: this.model?.model ?? "-",
+            dataPath: this.root,
+            sources: this.store.sources().length,
+        }));
+        this.tui.addChild(this.stream);
+        this.tui.addChild(new Spacer(1));
+        this.tui.addChild(new Rule());
+        this.tui.addChild(this.editor);
+        this.tui.setFocus(this.editor);
+        this.tui.addInputListener((data) => {
+            if (matchesKey(data, "ctrl+c")) {
+                this.shutdown();
+                return { consume: true };
+            }
+            if (matchesKey(data, "escape") && this.tui.hasOverlay()) {
+                this.closeOverlay();
+                return { consume: true };
+            }
+            return {};
+        });
+        this.tui.start();
+    }
+    get model() { return this.models[this.modelIndex]; }
+    // ---------- 渲染原语 ----------
+    add(component) {
+        this.stream.addChild(component);
+        this.tui.requestRender();
+    }
+    setBusy(on, message = "") {
+        if (on) {
+            this.editor.disableSubmit = true;
+            if (this.loader) {
+                this.loader.setMessage(message);
+            }
+            else {
+                this.loader = new Loader(this.tui, color.accent, color.muted, message);
+                this.stream.addChild(this.loader);
+                this.loader.start();
+            }
         }
-        if (key.return && input.trim() && !busy) {
-            void submit(input.trim());
-            setInput("");
-            return;
+        else {
+            this.editor.disableSubmit = false;
+            if (this.loader) {
+                this.loader.stop();
+                this.stream.removeChild(this.loader);
+                this.loader = null;
+            }
         }
-        if (key.backspace || key.delete) {
-            setInput((value) => value.slice(0, -1));
-            return;
+        this.tui.requestRender();
+    }
+    showOverlay(component, width) {
+        this.closeOverlay();
+        this.overlay = this.tui.showOverlay(component, { width, anchor: "center", margin: 1 });
+    }
+    closeOverlay() {
+        if (this.overlay) {
+            this.overlay.hide();
+            this.overlay = null;
         }
-        if (!key.ctrl && !key.meta && char)
-            setInput((value) => value + char);
-    });
-    async function submit(value) {
-        if (value === "/quit" || value === "/exit") {
-            store.close();
-            exit();
+        this.tui.setFocus(this.editor);
+    }
+    shutdown() {
+        if (this.closed)
             return;
+        this.closed = true;
+        try {
+            this.store.close();
         }
-        if (value === "/clear") {
-            setLines(["Conversation cleared."]);
-            setMessages([]);
+        catch { /* 已关闭 */ }
+        this.tui.stop();
+        process.exit(0);
+    }
+    // ---------- 输入入口 ----------
+    async submit(value) {
+        if (this.busy || !value)
+            return;
+        this.editor.setText("");
+        this.tui.requestRender();
+        const note = value.match(NOTE_PATTERN);
+        if (note) {
+            const content = (note[1] ?? "").trim();
+            if (!content) {
+                this.add(new Notice(ROLE.warn, color.warn, "用法：/add <内容>；也可以直接说「记住 ……」，模型会自动识别为录入。"));
+                return;
+            }
+            this.add(new UserMessage(shortPreview(`/add ${content}`)));
+            await this.addNote(content);
             return;
         }
-        if (value === "/help") {
-            setLines((old) => [...old, "/help /clear /sources /backup /restore /cleanup /model /models /kb /quit", "导入：输入「导入 <文件路径>」，支持 .md/.txt/.json/.csv/.xlsx/.xls 及 .png/.jpg/.jpeg/.gif/.webp/.bmp", "图片入库时自动生成中文描述用于检索，提问命中后会把原图一并发给模型"]);
+        if (value.startsWith("/")) {
+            await this.command(value);
             return;
         }
-        if (value === "/sources") {
-            setLines((old) => [...old, ...store.sources().map((s) => `${s.name} | ${s.path} | v${s.version} | ${s.updatedAt}`)]);
+        this.turns += 1;
+        this.add(new UserMessage(value));
+        await this.run(value);
+    }
+    async run(value) {
+        const pathMatch = value.match(/(?:录入|导入|整理|ingest|import)\s+(.+)$/i);
+        if (pathMatch) {
+            await this.ingest(pathMatch[1].trim().replace(/^['"]|['"]$/g, ""));
             return;
         }
-        if (value === "/model" || value === "/models") {
-            setLines((old) => [...old, ...config.map((item, i) => `${i === modelIndex ? "*" : " "} ${i}: ${item.provider}/${item.model} (${item.api})`)]);
+        await this.ask(value);
+    }
+    // ---------- 直接录入文本 ----------
+    async addNote(content, explicitTitle = "") {
+        const body = content.trim();
+        if (!body) {
+            this.add(new Notice(ROLE.warn, color.warn, "没有可录入的内容。"));
             return;
         }
-        if (value.startsWith("/model ")) {
-            const index = Number(value.slice(7));
-            if (config[index]) {
-                setModelIndex(index);
-                saveModels([config[index], ...config.filter((_, i) => i !== index)]);
-                setLines((old) => [...old, `Using ${config[index].model}`]);
+        try {
+            const title = explicitTitle.trim() || deriveTitle(body);
+            const file = writeNote(this.store.root, title, body);
+            const result = this.store.addOrUpdateSource(file, title, body);
+            this.add(new Notice(ROLE.ok, color.ok, `已录入「${result.title}」（v${result.version}，${body.length} 字符）`));
+        }
+        catch (error) {
+            this.add(new Notice(ROLE.err, color.err, `录入失败：${errText(error)}`));
+        }
+    }
+    // ---------- 导入 ----------
+    async ingest(path) {
+        try {
+            this.setBusy(true, "读取文件…");
+            const source = readLocalFile(path);
+            if (!source.hash) {
+                this.add(new Notice(ROLE.err, color.err, `${source.title}：无法解析（${source.content}）`));
+                return;
             }
-            else
-                setLines((old) => [...old, "Unknown model index."]);
+            let content = source.content;
+            let note = "";
+            if (source.image) {
+                this.setBusy(true, "识别图片并生成描述…");
+                try {
+                    const caption = await describeImage(this.model, fetchTransport, source.image);
+                    if (caption) {
+                        content = `${source.content}\n\n${caption}`;
+                        note = "，已生成图片描述";
+                    }
+                }
+                catch {
+                    note = "，图片描述生成失败（仅按文件名检索）";
+                }
+            }
+            const copied = copyOriginal(source.originalPath, this.store.root);
+            const result = this.store.addOrUpdateSource(copied, source.title, content);
+            this.add(new Notice(ROLE.ok, color.ok, `${result.isNew ? "已导入" : "已创建新版本"} ${result.title}（v${result.version}）${note}`));
+        }
+        catch (error) {
+            this.add(new Notice(ROLE.err, color.err, `导入失败：${errText(error)}`));
+        }
+        finally {
+            this.setBusy(false);
+        }
+    }
+    // ---------- 问答 ----------
+    async ask(question) {
+        const config = this.model;
+        if (!config) {
+            this.add(new Notice(ROLE.err, color.err, "没有可用模型，请先配置模型。"));
             return;
         }
-        if (value.startsWith("/kb ")) {
-            const [command, ...args] = value.slice(4).trim().split(/\s+/);
-            if (command === "list") {
-                setLines((old) => [...old, ...listKnowledgeBases(root).map((item) => `${item.id}${item.id === currentBase.id ? " *" : ""}`)]);
-                return;
+        const started = Date.now();
+        try {
+            this.setBusy(true, "检索知识库…");
+            const results = this.store.search(question);
+            const context = results.length
+                ? results.map((item) => `[${item.id}] ${item.title} (source: ${item.source}, v${item.version})\n${item.content}`).join("\n\n")
+                : "（没有检索到相关内容）";
+            const attached = [];
+            const questionBlocks = [{ type: "text", text: question }];
+            for (const item of results.filter((hit) => hit.originalPath && isImageFile(hit.originalPath)).slice(0, 4)) {
+                try {
+                    const image = readImage(item.originalPath);
+                    questionBlocks.push({ type: "text", text: `[附图] ${item.source} v${item.version}` });
+                    questionBlocks.push({ type: "image", mediaType: image.mediaType, base64: image.base64 });
+                    attached.push({ name: item.source, version: item.version });
+                }
+                catch { /* 跳过无法读取的图片 */ }
             }
-            if (command === "current") {
-                setLines((old) => [...old, `${currentBase.name} (${currentBase.id})`]);
+            this.setBusy(true, attached.length ? `命中 ${results.length} 条，附图 ${attached.length} 张，正在处理…` : `检索到 ${results.length} 条，正在理解意图…`);
+            const reply = await fetchTransport.complete(config, [
+                { role: "system", content: buildSystemPrompt(context) },
+                ...this.messages,
+                { role: "user", content: questionBlocks.length > 1 ? questionBlocks : question },
+            ], [SAVE_KNOWLEDGE_TOOL]);
+            const save = reply.toolCalls?.find((call) => call.name === SAVE_KNOWLEDGE_TOOL.name);
+            if (save) {
+                this.setBusy(true, "正在录入知识库…");
+                await this.addNote(String(save.arguments.content ?? ""), String(save.arguments.title ?? ""));
                 return;
             }
-            if (command === "create" && args.length) {
-                const created = ensureKnowledgeBase(root, args.join(" "));
-                setLines((old) => [...old, `Created ${created.name} (${created.id})`]);
+            const answer = (reply.text ?? "").trim();
+            this.messages = [...this.messages, { role: "user", content: question }, { role: "assistant", content: answer }];
+            this.store.saveMessage("user", question);
+            this.store.saveMessage("assistant", answer);
+            for (const item of attached)
+                this.add(new ImageNotice(item.name, item.version));
+            const seconds = ((Date.now() - started) / 1000).toFixed(1);
+            const meta = `检索 ${results.length} 条 · 用时 ${seconds}s` + (attached.length ? ` · 附图 ${attached.length} 张` : "");
+            this.add(new AssistantMessage(answer || "（模型返回空回答）", meta));
+        }
+        catch (error) {
+            this.add(new Notice(ROLE.err, color.err, `模型请求失败：${errText(error)}`));
+        }
+        finally {
+            this.setBusy(false);
+        }
+    }
+    // ---------- 命令 ----------
+    async command(value) {
+        const [cmd, ...rest] = value.split(/\s+/);
+        const arg = rest.join(" ");
+        switch (cmd) {
+            case "/quit":
+            case "/exit":
+                this.shutdown();
                 return;
-            }
-            if (command === "use" && args[0]) {
-                const selected = listKnowledgeBases(root).find((item) => item.id === args[0] || item.name === args.join(" "));
-                if (!selected) {
-                    setLines((old) => [...old, "Knowledge base not found."]);
+            case "/help":
+                this.showOverlay(new HelpPanel(HELP_ITEMS), 66);
+                return;
+            case "/clear":
+                this.stream.clear();
+                this.messages = [];
+                this.turns = 0;
+                this.add(new Notice(ROLE.ok, color.ok, "已清空对话上下文。"));
+                return;
+            case "/sources": return this.showSources();
+            case "/model":
+            case "/models":
+                if (!rest.length) {
+                    this.showModelPicker();
                     return;
                 }
-                store.close();
-                storeRef.current = new KnowledgeStore(selected.path);
-                setCurrentBase(selected);
-                setMessages([]);
-                setLines((old) => [...old, `Using ${selected.name}`]);
+                this.switchModel(Number(rest[0]));
                 return;
-            }
-            if (command === "delete" && args[0] && args[1] === "confirm") {
-                deleteKnowledgeBase(root, args[0]);
-                setLines((old) => [...old, `Deleted ${args[0]}; backups were retained.`]);
+            case "/kb": return this.kbCommand(rest, arg);
+            case "/backup": {
+                const target = backupKnowledgeBase(this.base, arg || undefined);
+                this.add(new Notice(ROLE.ok, color.ok, `已备份到 ${target}`));
                 return;
             }
-            setLines((old) => [...old, "Usage: /kb list | /kb create <name> | /kb use <id> | /kb current | /kb delete <id> confirm"]);
-            return;
+            case "/restore":
+                if (!arg) {
+                    this.add(new Notice(ROLE.warn, color.warn, "用法：/restore <备份目录>"));
+                    return;
+                }
+                restoreKnowledgeBase(this.base, arg);
+                this.add(new Notice(ROLE.ok, color.ok, "已恢复。重启 kb 后重新打开数据库。"));
+                return;
+            case "/cleanup": return this.cleanup(rest);
+            default:
+                this.add(new Notice(ROLE.warn, color.warn, `未知命令 ${cmd}，输入 /help 查看可用命令。`));
         }
-        if (value === "/backup" || value.startsWith("/backup ")) {
-            setLines((old) => [...old, `Backup created at ${backupKnowledgeBase(currentBase, value.slice(7).trim() || undefined)}`]);
+    }
+    showSources() {
+        const sources = this.store.sources();
+        if (!sources.length) {
+            this.add(new Notice(ROLE.warn, color.warn, "知识库还是空的，用「导入 <文件路径>」录入内容。"));
             return;
         }
-        if (value.startsWith("/restore ")) {
-            restoreKnowledgeBase(currentBase, value.slice(9).trim());
-            setLines((old) => [...old, "Knowledge base restored. Restart the session to reopen the database."]);
+        this.add(new Notice(ROLE.ok, color.ok, `共 ${sources.length} 个来源：`));
+        for (const source of sources) {
+            this.add(new Notice(" ", color.muted, `${String(source.name)}  v${String(source.version)}  ${String(source.updatedAt)}`));
+        }
+    }
+    showModelPicker() {
+        if (!this.models.length) {
+            this.add(new Notice(ROLE.err, color.err, "没有已配置的模型。"));
             return;
         }
-        if (value === "/cleanup") {
-            const oldVersions = store.oldVersions();
-            setLines((old) => [...old, oldVersions.length ? `Old versions (delete with /cleanup <id> confirm): ${oldVersions.map((item) => `${item.id} ${item.title} v${item.version}`).join("; ")}` : "No old versions."]);
+        const items = this.models.map((m, index) => ({
+            value: String(index),
+            label: `${m.provider}/${m.model}`,
+            description: `${m.api}${index === this.modelIndex ? "   ← 当前" : ""}`,
+        }));
+        const list = new SelectList(items, Math.min(items.length, 10), selectListTheme);
+        list.onSelect = (item) => { this.switchModel(Number(item.value)); this.closeOverlay(); };
+        list.onCancel = () => { this.closeOverlay(); };
+        this.showOverlay(list, 68);
+    }
+    switchModel(index) {
+        const next = this.models[index];
+        if (!next) {
+            this.add(new Notice(ROLE.err, color.err, `没有编号为 ${index} 的模型。`));
             return;
         }
-        if (value.startsWith("/cleanup ")) {
-            const [id, confirmation] = value.slice(9).trim().split(/\s+/);
-            if (confirmation === "confirm") {
-                store.removeVersions([id]);
-                setLines((old) => [...old, `Deleted old version ${id}.`]);
+        this.modelIndex = index;
+        this.models = [next, ...this.models.filter((_, i) => i !== index)];
+        saveModels(this.models);
+        this.add(new Notice(ROLE.ok, color.ok, `已切换到 ${next.provider}/${next.model}`));
+    }
+    kbCommand(rest, arg) {
+        const [sub, ...args] = rest;
+        if (!sub || sub === "list") {
+            const bases = listKnowledgeBases(this.root);
+            this.add(new Notice(ROLE.ok, color.ok, `共 ${bases.length} 个知识库：`));
+            for (const item of bases) {
+                this.add(new Notice(" ", color.muted, `${item.id}${item.id === this.base.id ? "   ← 当前" : ""}`));
             }
-            else
-                setLines((old) => [...old, "Usage: /cleanup <version-id> confirm"]);
             return;
         }
-        setBusy(true);
-        setLines((old) => [...old, `> ${value}`]);
-        try {
-            const result = await ingestOrAsk(value, store, config[modelIndex], messages);
-            setMessages(result.messages);
-            store.saveMessage("user", value);
-            store.saveMessage("assistant", result.text);
-            setLines((old) => [...old, result.text]);
+        if (sub === "current") {
+            this.add(new Notice(ROLE.ok, color.ok, `${this.base.name}（${this.base.id}）`));
+            return;
         }
-        catch (error) {
-            setLines((old) => [...old, `Error: ${error instanceof Error ? error.message : String(error)}`]);
+        if (sub === "create") {
+            if (!args.length) {
+                this.add(new Notice(ROLE.warn, color.warn, "用法：/kb create <名称>"));
+                return;
+            }
+            const created = ensureKnowledgeBase(this.root, args.join(" "));
+            this.add(new Notice(ROLE.ok, color.ok, `已创建 ${created.name}（${created.id}）`));
+            return;
         }
-        finally {
-            setBusy(false);
+        if (sub === "use") {
+            const selected = listKnowledgeBases(this.root).find((item) => item.id === args[0] || item.name === args.join(" "));
+            if (!selected) {
+                this.add(new Notice(ROLE.err, color.err, `未找到知识库 ${arg}`));
+                return;
+            }
+            this.store.close();
+            this.store = new KnowledgeStore(selected.path);
+            this.base = selected;
+            this.messages = [];
+            this.turns = 0;
+            this.add(new Notice(ROLE.ok, color.ok, `已切换到知识库 ${selected.name}（${this.store.sources().length} 个来源）`));
+            return;
         }
-    }
-    return _jsxs(Box, { flexDirection: "column", padding: 1, children: [lines.slice(-18).map((line, i) => _jsx(Text, { children: line }, `${i}-${line}`)), _jsxs(Text, { color: busy ? "yellow" : "cyan", children: [busy ? "Searching..." : "> ", input] })] });
-}
-async function ingestOrAsk(value, store, modelConfig, previous) {
-    const pathMatch = value.match(/(?:录入|导入|整理|ingest|import)\s+(.+)$/i);
-    if (pathMatch) {
-        const source = readLocalFile(pathMatch[1].trim().replace(/^['"]|['"]$/g, ""));
-        if (!source.hash)
-            return { text: `${source.title}: unable to parse (${source.content})`, messages: previous };
-        let content = source.content;
-        let note = "";
-        if (source.image) {
+        if (sub === "delete") {
+            if (!args[0] || args[1] !== "confirm") {
+                this.add(new Notice(ROLE.warn, color.warn, "用法：/kb delete <id> confirm"));
+                return;
+            }
             try {
-                const caption = await describeImage(modelConfig, fetchTransport, source.image);
-                if (caption) {
-                    content = `${source.content}\n\n${caption}`;
-                    note = "，已生成图片描述";
-                }
+                deleteKnowledgeBase(this.root, args[0]);
+                this.add(new Notice(ROLE.ok, color.ok, `已删除 ${args[0]}，备份已保留。`));
             }
             catch (error) {
-                note = `，图片描述生成失败（仅按文件名检索）: ${error instanceof Error ? error.message : String(error)}`;
+                this.add(new Notice(ROLE.err, color.err, errText(error)));
             }
+            return;
         }
-        const copied = copyOriginal(source.originalPath, store.root);
-        const result = store.addOrUpdateSource(copied, source.title, content);
-        return { text: `${result.isNew ? "Imported" : "New version created for"} ${result.title} (v${result.version})${note}`, messages: previous };
+        this.add(new Notice(ROLE.warn, color.warn, "用法：/kb list | /kb create <名称> | /kb use <id> | /kb current | /kb delete <id> confirm"));
     }
-    const results = store.search(value);
-    const context = results.length ? results.map((item) => `[${item.id}] ${item.title} (source: ${item.source}, v${item.version})\n${item.content}`).join("\n\n") : "No matching knowledge was found.";
-    const prompt = `Answer only from the local knowledge below. If it is insufficient, say no knowledge was found. Include source filename and version.\n\n${context}\n\nQuestion: ${value}`;
-    const blocks = [{ type: "text", text: prompt }];
-    for (const item of results.filter((hit) => hit.originalPath && isImageFile(hit.originalPath)).slice(0, 4)) {
-        try {
-            const image = readImage(item.originalPath);
-            blocks.push({ type: "text", text: `[附图] ${item.source} v${item.version}` });
-            blocks.push({ type: "image", mediaType: image.mediaType, base64: image.base64 });
+    cleanup(rest) {
+        if (!rest.length) {
+            const old = this.store.oldVersions();
+            if (!old.length) {
+                this.add(new Notice(ROLE.ok, color.ok, "没有历史版本。"));
+                return;
+            }
+            this.add(new Notice(ROLE.warn, color.warn, `历史版本 ${old.length} 个，用 /cleanup <版本ID> confirm 删除：`));
+            for (const item of old)
+                this.add(new Notice(" ", color.muted, `${String(item.id)}  ${String(item.title)} v${String(item.version)}`));
+            return;
+        }
+        const [id, confirmation] = rest;
+        if (confirmation !== "confirm") {
+            this.add(new Notice(ROLE.warn, color.warn, "用法：/cleanup <版本ID> confirm"));
+            return;
         }
-        catch { /* 忽略无法读取的图片 */ }
+        this.store.removeVersions([id]);
+        this.add(new Notice(ROLE.ok, color.ok, `已删除历史版本 ${id}。`));
     }
-    const agent = new Agent(store, modelConfig, fetchTransport);
-    return agent.answer(blocks.length > 1 ? blocks : prompt, previous);
 }
 export async function main() {
     const root = defaultDataRoot();
     mkdirSync(root, { recursive: true });
-    const knowledgeRoot = root;
     const base = ensureKnowledgeBase(root, "default");
-    const configs = await ensureModelConfig();
-    render(_jsx(App, { config: configs, root: knowledgeRoot, knowledgeBase: base }));
+    const models = await ensureModelConfig();
+    const app = new KbApp(root, base, models);
+    app.start();
 }