@ganziliang/kb 0.1.7 → 0.3.0

package/dist/model.d.ts

@@ -1,9 +1,15 @@
 import type { ModelConfig } from "@ganziliang/kb-model-setup";
 import type { SearchResult, KnowledgeStore } from "./storage.js";
-export type ToolCall = {
-    name: "search" | "read" | "write";
+export type NativeToolCall = {
+    id: string;
+    name: string;
     arguments: Record<string, unknown>;
 };
+export type ToolDefinition = {
+    name: string;
+    description: string;
+    inputSchema: Record<string, unknown>;
+};
 export type ContentBlock = {
     type: "text";
     text: string;
@@ -19,13 +25,13 @@ export type ModelMessage = {
 };
 export type ModelReply = {
     text?: string;
-    toolCalls?: ToolCall[];
+    toolCalls?: NativeToolCall[];
 };
 export declare function toAnthropicContent(content: string | ContentBlock[]): unknown;
 export declare function toOpenAIContent(content: string | ContentBlock[]): unknown;
 export declare function contentToText(content: string | ContentBlock[]): string;
 export type ModelTransport = {
-    complete(config: ModelConfig, messages: ModelMessage[]): Promise<ModelReply>;
+    complete(config: ModelConfig, messages: ModelMessage[], tools?: ToolDefinition[]): Promise<ModelReply>;
 };
 export declare const fetchTransport: ModelTransport;
 export declare class Agent {
@@ -40,6 +46,11 @@ export declare class Agent {
     private execute;
 }
 export declare function formatResults(results: SearchResult[]): string;
+/**
+ * 让模型在一轮对话里自己决定「回答」还是「录入」，
+ * 避免为了判断意图而多跑一次模型调用。
+ */
+export declare const SAVE_KNOWLEDGE_TOOL: ToolDefinition;
 type VisionTransport = {
     complete: (config: ModelConfig, messages: ModelMessage[]) => Promise<ModelReply>;
 };

package/dist/model.js

@@ -23,7 +23,7 @@ function endpoint(config) {
     return `${config.baseURL.replace(/\/$/, "")}${config.api === "anthropic-messages" ? "/v1/messages" : "/v1/responses"}`;
 }
 export const fetchTransport = {
-    async complete(config, messages) {
+    async complete(config, messages, tools) {
         const headers = { "content-type": "application/json" };
         if (config.api === "anthropic-messages") {
             headers["x-api-key"] = config.apiKey;
@@ -31,19 +31,46 @@ export const fetchTransport = {
         }
         else
             headers.authorization = `Bearer ${config.apiKey}`;
+        const anthropicTools = tools?.length
+            ? { tools: tools.map((tool) => ({ name: tool.name, description: tool.description, input_schema: tool.inputSchema })) }
+            : {};
+        const openaiTools = tools?.length
+            ? { tools: tools.map((tool) => ({ type: "function", name: tool.name, description: tool.description, parameters: tool.inputSchema })) }
+            : {};
         const response = await fetch(endpoint(config), {
             method: "POST",
             headers,
             body: JSON.stringify(config.api === "anthropic-messages"
-                ? { model: config.model, max_tokens: 4096, messages: messages.filter((m) => m.role !== "system").map((m) => ({ role: m.role, content: toAnthropicContent(m.content) })), system: messages.find((m) => m.role === "system")?.content }
-                : { model: config.model, input: messages.map((m) => ({ role: m.role, content: toOpenAIContent(m.content) })) }),
+                ? { model: config.model, max_tokens: 4096, messages: messages.filter((m) => m.role !== "system").map((m) => ({ role: m.role, content: toAnthropicContent(m.content) })), system: messages.find((m) => m.role === "system")?.content, ...anthropicTools }
+                : { model: config.model, input: messages.map((m) => ({ role: m.role, content: toOpenAIContent(m.content) })), ...openaiTools }),
         });
         if (!response.ok)
             throw new Error(`Model request failed (${response.status}): ${await response.text()}`);
         const data = await response.json();
-        if (config.api === "anthropic-messages")
-            return { text: data.content?.map((item) => item.text ?? "").join("") };
-        return { text: data.output_text ?? data.output?.map((item) => item.content?.map((part) => part.text ?? "").join("")).join("") ?? "" };
+        if (config.api === "anthropic-messages") {
+            let text = "";
+            const toolCalls = [];
+            for (const item of (data.content ?? [])) {
+                if (item.type === "text")
+                    text += item.text ?? "";
+                else if (item.type === "tool_use")
+                    toolCalls.push({ id: String(item.id ?? ""), name: String(item.name), arguments: (item.input ?? {}) });
+            }
+            return toolCalls.length ? { text, toolCalls } : { text };
+        }
+        const toolCalls = [];
+        for (const item of (data.output ?? [])) {
+            if (item.type !== "function_call")
+                continue;
+            let parsed = {};
+            try {
+                parsed = JSON.parse(item.arguments ?? "{}");
+            }
+            catch { /* 保留空参数 */ }
+            toolCalls.push({ id: String(item.call_id ?? item.id ?? ""), name: String(item.name), arguments: parsed });
+        }
+        const text = data.output_text ?? (data.output ?? []).map((item) => (item.content ?? []).map((part) => part.text ?? "").join("")).join("") ?? "";
+        return toolCalls.length ? { text, toolCalls } : { text };
     },
 };
 export class Agent {
@@ -72,12 +99,34 @@ export class Agent {
             return this.store.search(String(call.arguments.query ?? ""));
         if (call.name === "read")
             return this.store.read(String(call.arguments.id ?? "")) ?? { error: "Knowledge record not found" };
-        return { error: "write requires CLI ingestion flow" };
+        return { error: `Unknown tool: ${call.name}` };
     }
 }
 export function formatResults(results) {
     return results.map((result) => `[${result.id}] ${result.title} (source: ${result.source}, v${result.version})\n${result.content}`).join("\n\n");
 }
+/**
+ * 让模型在一轮对话里自己决定「回答」还是「录入」，
+ * 避免为了判断意图而多跑一次模型调用。
+ */
+export const SAVE_KNOWLEDGE_TOOL = {
+    name: "save_knowledge",
+    description: [
+        "把用户提供的内容保存到本地知识库。",
+        "当用户意图是「记录 / 保存 / 录入 / 收藏 / 备忘 / 以后要用」某段内容时调用，",
+        "例如：记住…、记一下…、记录一下…、把这段存起来、帮我存个配置、以后就按这个来。",
+        "只有用户确实在提供一段希望长期保存的知识时才调用。",
+        "如果用户在询问、查询、确认、闲聊，不要调用，直接回答。",
+    ].join(""),
+    inputSchema: {
+        type: "object",
+        properties: {
+            title: { type: "string", description: "对内容的一句话概括，作为知识条目标题，20 字以内" },
+            content: { type: "string", description: "要保存的完整内容，规范 Markdown。必须完整保留用户给出的全部信息，不要自行增删或改写事实" },
+        },
+        required: ["title", "content"],
+    },
+};
 const IMAGE_QUESTION = "\u8bf7\u7528\u4e2d\u6587\u8be6\u7ec6\u63cf\u8ff0\u8fd9\u5f20\u56fe\u7247\u7684\u5185\u5bb9\uff0c\u5199\u6e05\u6240\u6709\u53ef\u89c1\u7684\u6587\u5b57\u3001\u6570\u5b57\u3001\u8868\u683c\u3001\u754c\u9762\u5143\u7d20\u548c\u5173\u952e\u7ec6\u8282\uff0c\u4ee5\u53ca\u56fe\u7247\u6574\u4f53\u5728\u8bb2\u4ec0\u4e48\u3002\u53ea\u8f93\u51fa\u63cf\u8ff0\u6b63\u6587\uff0c\u4e0d\u8981\u5ba2\u5957\u8bdd\u3002";
 export async function describeImage(config, transport, image) {
     const reply = await transport.complete(config, [{

package/dist/storage.d.ts

@@ -54,6 +54,8 @@ type KnowledgeStoreResult = {
 };
 export declare function readLocalFile(filePath: string): LocalFile;
 export declare function copyOriginal(filePath: string, root: string): string;
+/** \u628a\u624b\u52a8\u5f55\u5165\u7684\u6587\u672c\u843d\u76d8\u4e3a originals/ \u4e0b\u7684 Markdown\uff0c\u8fd4\u56de\u8be5\u6587\u4ef6\u8def\u5f84\u3002 */
+export declare function writeNote(root: string, title: string, content: string): string;
 export declare function defaultDataRoot(): string;
 export declare function knowledgeBasePath(root: string, id: string): string;
 export declare function listKnowledgeBases(root: string): KnowledgeBase[];

package/dist/storage.js

@@ -1,4 +1,4 @@
-import { mkdirSync, copyFileSync, existsSync, readFileSync, statSync, cpSync, rmSync, readdirSync } from "node:fs";
+import { mkdirSync, copyFileSync, existsSync, readFileSync, statSync, cpSync, rmSync, readdirSync, writeFileSync } from "node:fs";
 import { createHash } from "node:crypto";
 import { join, resolve, extname, basename } from "node:path";
 import { DatabaseSync } from "node:sqlite";
@@ -115,6 +115,15 @@ export function copyOriginal(filePath, root) {
     copyFileSync(filePath, destination);
     return destination;
 }
+/** \u628a\u624b\u52a8\u5f55\u5165\u7684\u6587\u672c\u843d\u76d8\u4e3a originals/ \u4e0b\u7684 Markdown\uff0c\u8fd4\u56de\u8be5\u6587\u4ef6\u8def\u5f84\u3002 */
+export function writeNote(root, title, content) {
+    const dir = join(root, "originals");
+    mkdirSync(dir, { recursive: true });
+    const slug = title.replace(/[^\p{L}\p{N}]+/gu, "-").replace(/^-+|-+$/g, "").slice(0, 40) || "note";
+    const target = join(dir, `${Date.now()}-${slug}.md`);
+    writeFileSync(target, content, "utf8");
+    return target;
+}
 export function defaultDataRoot() {
     const home = process.env.HOME ?? process.env.USERPROFILE ?? process.cwd();
     return process.env.KB_DATA_DIR ?? join(home, ".kb");

package/dist/theme.d.ts

@@ -0,0 +1,27 @@
+import type { EditorTheme, ImageTheme, MarkdownTheme, SelectListTheme } from "@earendil-works/pi-tui";
+export declare const fg: (r: number, g: number, b: number) => (s: string) => string;
+export declare const bg: (r: number, g: number, b: number) => (s: string) => string;
+export declare const bold: (s: string) => string;
+export declare const dim: (s: string) => string;
+export declare const italic: (s: string) => string;
+export declare const underline: (s: string) => string;
+export declare const strike: (s: string) => string;
+/** 全局配色：slate/sky/teal 冷色调，深浅终端下都有足够对比度。 */
+export declare const color: {
+    accent: (s: string) => string;
+    accent2: (s: string) => string;
+    user: (s: string) => string;
+    text: (s: string) => string;
+    muted: (s: string) => string;
+    dim: (s: string) => string;
+    border: (s: string) => string;
+    ok: (s: string) => string;
+    warn: (s: string) => string;
+    err: (s: string) => string;
+    chip: (s: string) => string;
+    userBg: (s: string) => string;
+};
+export declare const markdownTheme: MarkdownTheme;
+export declare const selectListTheme: SelectListTheme;
+export declare const editorTheme: EditorTheme;
+export declare const imageTheme: ImageTheme;

package/dist/theme.js

@@ -0,0 +1,52 @@
+export const fg = (r, g, b) => (s) => `\x1b[38;2;${r};${g};${b}m${s}\x1b[39m`;
+export const bg = (r, g, b) => (s) => `\x1b[48;2;${r};${g};${b}m${s}\x1b[49m`;
+export const bold = (s) => `\x1b[1m${s}\x1b[22m`;
+export const dim = (s) => `\x1b[2m${s}\x1b[22m`;
+export const italic = (s) => `\x1b[3m${s}\x1b[23m`;
+export const underline = (s) => `\x1b[4m${s}\x1b[24m`;
+export const strike = (s) => `\x1b[9m${s}\x1b[29m`;
+/** 全局配色：slate/sky/teal 冷色调，深浅终端下都有足够对比度。 */
+export const color = {
+    accent: fg(56, 189, 248),
+    accent2: fg(45, 212, 191),
+    user: fg(226, 232, 240),
+    text: fg(203, 213, 225),
+    muted: fg(100, 116, 139),
+    dim: fg(71, 85, 105),
+    border: fg(51, 65, 85),
+    ok: fg(74, 222, 128),
+    warn: fg(251, 191, 36),
+    err: fg(248, 113, 113),
+    chip: bg(30, 41, 59),
+    userBg: bg(15, 23, 42),
+};
+export const markdownTheme = {
+    heading: (t) => bold(color.accent2(t)),
+    link: (t) => color.accent(t),
+    linkUrl: (t) => dim(color.muted(t)),
+    code: (t) => color.warn(t),
+    codeBlock: (t) => color.text(t),
+    codeBlockBorder: (t) => color.border(t),
+    quote: (t) => dim(color.muted(t)),
+    quoteBorder: (t) => color.accent2(t),
+    hr: (t) => color.border(t),
+    listBullet: (t) => color.accent(t),
+    bold: (t) => bold(color.user(t)),
+    italic: (t) => italic(color.text(t)),
+    strikethrough: (t) => strike(color.text(t)),
+    underline: (t) => underline(color.text(t)),
+};
+export const selectListTheme = {
+    selectedPrefix: (t) => color.accent(bold(t)),
+    selectedText: (t) => bold(color.user(t)),
+    description: (t) => color.muted(t),
+    scrollInfo: (t) => dim(color.muted(t)),
+    noMatch: (t) => dim(color.muted(t)),
+};
+export const editorTheme = {
+    borderColor: (t) => color.border(t),
+    selectList: selectListTheme,
+};
+export const imageTheme = {
+    fallbackColor: (t) => color.muted(t),
+};

package/dist/ui.d.ts

@@ -0,0 +1,88 @@
+import { type Component } from "@earendil-works/pi-tui";
+import { dim, selectListTheme } from "./theme.js";
+/** 把文本按可见宽度换行；保证每行都不超过 width。 */
+export declare function wrap(text: string, width: number): string[];
+/** 通用多行文本组件，按渲染宽度截断，保证永不超宽。 */
+export declare class Lines implements Component {
+    private lines;
+    constructor(lines?: string[]);
+    setLines(lines: string[]): void;
+    invalidate(): void;
+    render(width: number): string[];
+}
+/** 启动屏：logo + 环境信息。 */
+export declare class Splash implements Component {
+    private info;
+    constructor(info: {
+        base: string;
+        model: string;
+        dataPath: string;
+        sources: number;
+    });
+    invalidate(): void;
+    render(width: number): string[];
+}
+/** 顶部状态条：带圆角边框的一行摘要。 */ export declare class StatusBar implements Component {
+    private info;
+    constructor(info: {
+        base: string;
+        model: string;
+        sources: number;
+        turns: number;
+    });
+    setInfo(info: {
+        base: string;
+        model: string;
+        sources: number;
+        turns: number;
+    }): void;
+    invalidate(): void;
+    render(width: number): string[];
+}
+/** 用户提问。 */
+export declare class UserMessage implements Component {
+    private text;
+    constructor(text: string);
+    invalidate(): void;
+    render(width: number): string[];
+}
+/** 助手回答：图标 + Markdown 正文 + 元信息脚注。 */
+export declare class AssistantMessage implements Component {
+    private meta;
+    private md;
+    constructor(text: string, meta?: string);
+    setMeta(meta: string): void;
+    invalidate(): void;
+    render(width: number): string[];
+}
+/** 单行提示（成功 / 信息 / 警告 / 错误）。 */
+export declare class Notice implements Component {
+    private icon;
+    private style;
+    private text;
+    constructor(icon: string, style: (s: string) => string, text: string);
+    invalidate(): void;
+    render(width: number): string[];
+}
+/** 检索命中的附图标记。 */
+export declare class ImageNotice implements Component {
+    private name;
+    private version;
+    constructor(name: string, version: number);
+    invalidate(): void;
+    render(width: number): string[];
+}
+/** 水平分隔线。 */
+export declare class Rule implements Component {
+    invalidate(): void;
+    render(width: number): string[];
+}
+/** 帮助面板（作为 overlay 内容）。 */
+export declare class HelpPanel implements Component {
+    private items;
+    constructor(items: Array<[string, string]>);
+    invalidate(): void;
+    render(width: number): string[];
+}
+/** 供 SelectList 复用的主题。 */
+export { selectListTheme, dim };

package/dist/ui.js

@@ -0,0 +1,171 @@
+import { Markdown, truncateToWidth, visibleWidth, wrapTextWithAnsi } from "@earendil-works/pi-tui";
+import { bold, color, dim, markdownTheme, selectListTheme } from "./theme.js";
+/** 把文本按可见宽度换行；保证每行都不超过 width。 */
+export function wrap(text, width) {
+    if (width <= 0)
+        return [""];
+    const out = [];
+    for (const raw of text.split("\n")) {
+        if (raw === "") {
+            out.push("");
+            continue;
+        }
+        for (const piece of wrapTextWithAnsi(raw, width))
+            out.push(piece);
+    }
+    return out;
+}
+const pad = (s, w) => s + " ".repeat(Math.max(0, w - visibleWidth(s)));
+/** 统一裁剪：保证组件输出永不超过给定宽度（pi-tui 会因超宽报错）。 */
+const clip = (lines, width) => lines.map((l) => truncateToWidth(l, Math.max(0, width)));
+/** 通用多行文本组件，按渲染宽度截断，保证永不超宽。 */
+export class Lines {
+    lines = [];
+    constructor(lines = []) { this.lines = lines; }
+    setLines(lines) { this.lines = lines; }
+    invalidate() { }
+    render(width) {
+        return this.lines.map((l) => truncateToWidth(l, Math.max(0, width)));
+    }
+}
+/** 启动屏：logo + 环境信息。 */
+export class Splash {
+    info;
+    constructor(info) {
+        this.info = info;
+    }
+    invalidate() { }
+    render(width) {
+        const logo = ["██╗  ██╗ ██████╗", "██║ ██╔╝ ██╔══██╗", "█████╔╝  ██████╔╝", "██╔═██╗  ██╔══██╗", "██║  ██╗ ██████╔╝", "╚═╝  ╚═╝ ╚═════╝ "];
+        const field = (k, v, val) => `   ${color.muted(pad(k, 8))}${val(v)}`;
+        const lines = [
+            "",
+            ...logo.map((l) => "   " + color.accent(bold(l))),
+            "",
+            "   " + color.text(bold("知识库")) + color.border(" · ") + color.muted("Knowledge Base"),
+            "",
+            field("知识库", this.info.base, color.user),
+            field("模型", this.info.model, color.user),
+            field("来源", String(this.info.sources), color.user),
+            field("数据", this.info.dataPath, (s) => color.dim(s)),
+            "",
+            "   " + color.dim("输入问题开始，或键入 ") + color.accent("/help") + color.dim(" 查看命令"),
+            "",
+        ];
+        return lines.map((l) => truncateToWidth(l, Math.max(0, width)));
+    }
+}
+/** 顶部状态条：带圆角边框的一行摘要。 */ export class StatusBar {
+    info;
+    constructor(info) {
+        this.info = info;
+    }
+    setInfo(info) { this.info = info; }
+    invalidate() { }
+    render(width) {
+        const w = Math.max(0, width);
+        const inner = Math.max(0, w - 4);
+        const label = " KB ";
+        const top = color.border("╭─") + color.accent(bold(label)) + color.border("─".repeat(Math.max(0, w - 3 - visibleWidth(label))) + "╮");
+        const bot = color.border("╰" + "─".repeat(Math.max(0, w - 2)) + "╯");
+        const stat = (k, v) => `${color.muted(k)} ${color.user(v)}`;
+        const content = "  " + [
+            stat("知识库", this.info.base),
+            stat("模型", this.info.model),
+            stat("来源", String(this.info.sources)),
+            stat("会话", String(this.info.turns)),
+        ].join(color.border("  ·  "));
+        const row = color.border("│") + " " + pad(truncateToWidth(content, inner), inner) + " " + color.border("│");
+        return clip([top, row, bot], w);
+    }
+}
+/** 用户提问。 */
+export class UserMessage {
+    text;
+    constructor(text) {
+        this.text = text;
+    }
+    invalidate() { }
+    render(width) {
+        const prefix = `  ${color.accent2(bold("❯"))} `;
+        const body = wrap(this.text, Math.max(1, width - 4));
+        return body.map((l, i) => (i === 0 ? prefix + color.user(l) : "    " + color.user(l)));
+    }
+}
+/** 助手回答：图标 + Markdown 正文 + 元信息脚注。 */
+export class AssistantMessage {
+    meta;
+    md;
+    constructor(text, meta = "") {
+        this.meta = meta;
+        this.md = new Markdown(text, 0, 0, markdownTheme);
+    }
+    setMeta(meta) { this.meta = meta; }
+    invalidate() { this.md.invalidate(); }
+    render(width) {
+        const indent = "    ";
+        const bodyWidth = Math.max(10, width - indent.length);
+        const lines = [`  ${color.ok(bold("⏺"))}`, ""];
+        for (const l of this.md.render(bodyWidth))
+            lines.push(indent + l.replace(/\s+$/, ""));
+        if (this.meta)
+            lines.push("", `${indent}${color.dim("·")} ${color.muted(this.meta)}`);
+        return lines.map((l) => truncateToWidth(l, Math.max(0, width)));
+    }
+}
+/** 单行提示（成功 / 信息 / 警告 / 错误）。 */
+export class Notice {
+    icon;
+    style;
+    text;
+    constructor(icon, style, text) {
+        this.icon = icon;
+        this.style = style;
+        this.text = text;
+    }
+    invalidate() { }
+    render(width) {
+        const prefix = `  ${this.style(this.icon)} `;
+        return wrap(this.text, Math.max(1, width - 4)).map((l, i) => (i === 0 ? prefix + color.text(l) : "    " + color.text(l)));
+    }
+}
+/** 检索命中的附图标记。 */
+export class ImageNotice {
+    name;
+    version;
+    constructor(name, version) {
+        this.name = name;
+        this.version = version;
+    }
+    invalidate() { }
+    render(width) {
+        const line = `  ${color.accent2("★")} ${color.muted("附图")} ${color.chip(color.text(` ${this.name} `))} ${color.dim("v" + this.version)}`;
+        return [truncateToWidth(line, Math.max(0, width))];
+    }
+}
+/** 水平分隔线。 */
+export class Rule {
+    invalidate() { }
+    render(width) {
+        return [color.dim("─".repeat(Math.max(0, width)))];
+    }
+}
+/** 帮助面板（作为 overlay 内容）。 */
+export class HelpPanel {
+    items;
+    constructor(items) { this.items = items; }
+    invalidate() { }
+    render(width) {
+        const w = Math.min(Math.max(0, width), 72);
+        const inner = Math.max(0, w - 4);
+        const title = " 命令 ";
+        const keyWidth = Math.max(10, ...this.items.map(([k]) => visibleWidth(k))) + 2;
+        const top = color.border("╭─") + color.accent(bold(title)) + color.border("─".repeat(Math.max(0, w - 3 - visibleWidth(title))) + "╮");
+        const bot = color.border("╰" + "─".repeat(Math.max(0, w - 2)) + "╯");
+        const row = (c) => color.border("│") + " " + pad(truncateToWidth(c, inner), inner) + " " + color.border("│");
+        const body = this.items.map(([k, d]) => row(k ? "  " + color.accent(k) + " ".repeat(Math.max(1, keyWidth - visibleWidth(k))) + color.muted(d) : ""));
+        return clip([top, row(""), ...body, row(""), bot], w);
+    }
+}
+/** 供 SelectList 复用的主题。 */
+export { selectListTheme, dim };

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@ganziliang/kb",
-  "version": "0.1.7",
-  "description": "Local knowledge base agent CLI",
+  "version": "0.3.0",
+  "description": "Local knowledge base agent CLI with a pi-tui interface",
   "type": "module",
   "bin": {
     "kb": "dist/entry.js"
@@ -17,17 +17,15 @@
     "test": "npm run build && node --test"
   },
   "engines": {
-    "node": ">=22.5.0"
+    "node": ">=22.19.0"
   },
   "dependencies": {
+    "@earendil-works/pi-tui": "^0.85.1",
     "@ganziliang/kb-model-setup": "^0.1.2",
-    "ink": "^5.1.0",
-    "react": "^18.3.1",
     "xlsx": "^0.18.5"
   },
   "devDependencies": {
     "@types/node": "^22.10.0",
-    "@types/react": "^18.3.12",
     "typescript": "^5.7.2"
   }
 }