@bolloon/bolloon-agent 0.2.8 → 0.2.10

package/dist/agents/session-store.js

@@ -1,171 +0,0 @@
-/**
- * session-store — 持久化 Message[] (LLM 历史) 到 ~/.bolloon/sessions/cache/<key>.json
- *
- * 2026-06-30 抽出:
- *   - 之前 PiAgentSession 用 constraint-runtime 的 saveSession (存 string[]) + 独立的 hydrateMessageHistory
- *     读路径, 写路径不存在 — claude code / 外部 harness 想"接续 prompt"完全无入口.
- *   - 现在抽出 SessionStore 模块, 暴露纯 IO 接口, 完全可消融测试:
- *       saveMessages(key, msgs) — 写
- *       loadMessages(key)      — 读
- *       listKeys()              — 列全部 key
- *       deleteKey(key)          — 清
- *   - 任何目录位置都可注入 (默认 ~/.bolloon/sessions/cache/), 测试里指向临时 dir.
- *
- * 使用:
- *   import { sessionStore } from './session-store';
- *   sessionStore.saveMessages('cli:session-A', messageHistory);
- *   const restored = sessionStore.loadMessages('cli:session-A');
- */
-import * as fs from 'fs/promises';
-import * as path from 'path';
-import * as os from 'os';
-export class SessionStore {
-    cacheDir;
-    constructor(config = {}) {
-        this.cacheDir = config.cacheDir ?? path.join(os.homedir(), '.bolloon', 'sessions', 'cache');
-    }
-    /** 当前缓存目录 (只读). */
-    get dir() {
-        return this.cacheDir;
-    }
-    /** 单文件路径 — 暴露出来方便测试和外部读取.
-     *
-     * 2026-07-04 fix: Windows 文件名禁止 `:` (NTFS). web server 用 `channelId:currentSessionId`
-     * 拼 sessionKey (server.ts:1759 之类), 会含 `:`. 在 Linux/macOS 上能用, Windows 上
-     * fs.writeFile 抛 EINVAL. 修法: filename 层 escape `:` → `__`, key 保持不变
-     * (load/save/listKeys/deleteKey 全部透明).
-     */
-    pathFor(key) {
-        if (!key || key.includes('/') || key.includes('..')) {
-            throw new Error(`SessionStore: invalid key ${JSON.stringify(key)}`);
-        }
-        return path.join(this.cacheDir, `${SessionStore.filenameEscape(key)}.json`);
-    }
-    /** 把 session key 转换成跨平台安全的 filename (escape Windows 非法字符). */
-    static filenameEscape(key) {
-        return key.replace(/:/g, '__');
-    }
-    /** listKeys 反向: 把 filename 还原成 session key. */
-    static filenameUnescape(filenameKey) {
-        return filenameKey.replace(/__/g, ':');
-    }
-    /**
-     * 写 history to disk.
-     * - 自动 ensure 缓存目录
-     * - 失败抛错 (写失败不应静默 — 用户可能依赖持久化作为审计)
-     */
-    async saveMessages(key, messages) {
-        if (!key || key.includes('/') || key.includes('..')) {
-            throw new Error(`SessionStore: invalid key ${JSON.stringify(key)}`);
-        }
-        await fs.mkdir(this.cacheDir, { recursive: true });
-        const payload = {
-            key,
-            messages,
-            metadata: {
-                savedAt: Date.now(),
-                cwd: process.cwd(),
-                totalCount: messages.length,
-            },
-        };
-        const filePath = this.pathFor(key);
-        // 写临时文件 + rename — 防止半写损坏
-        const tmpPath = `${filePath}.tmp`;
-        await fs.writeFile(tmpPath, JSON.stringify(payload, null, 2), 'utf-8');
-        await fs.rename(tmpPath, filePath);
-    }
-    /**
-     * 同步版 (PiAgentSession 里多条路径用同步 — 避免 async 链)
-     * 失败抛错
-     */
-    saveMessagesSync(key, messages) {
-        if (!key || key.includes('/') || key.includes('..')) {
-            throw new Error(`SessionStore: invalid key ${JSON.stringify(key)}`);
-        }
-        fsSyncMkdir(this.cacheDir);
-        const payload = {
-            key,
-            messages,
-            metadata: {
-                savedAt: Date.now(),
-                cwd: process.cwd(),
-                totalCount: messages.length,
-            },
-        };
-        const filePath = this.pathFor(key);
-        const tmpPath = `${filePath}.tmp`;
-        fsSyncWrite(tmpPath, JSON.stringify(payload, null, 2));
-        fsSyncRename(tmpPath, filePath);
-    }
-    /**
-     * 读 history from disk.
-     * - key 不存在 → 返回 null (不抛错)
-     * - 文件存在但 JSON 损坏 → 抛错 (claude code 应该感知坏数据)
-     * - 文件 schema 不匹配 (新版格式) → 抛错含 details
-     */
-    async loadMessages(key) {
-        if (!key || key.includes('/') || key.includes('..')) {
-            throw new Error(`SessionStore: invalid key ${JSON.stringify(key)}`);
-        }
-        const filePath = this.pathFor(key);
-        let raw;
-        try {
-            raw = await fs.readFile(filePath, 'utf-8');
-        }
-        catch (e) {
-            if (e?.code === 'ENOENT')
-                return null;
-            throw e;
-        }
-        const parsed = JSON.parse(raw);
-        if (!parsed || !Array.isArray(parsed.messages)) {
-            throw new Error(`SessionStore: ${key} schema invalid (missing messages[])`);
-        }
-        return parsed.messages;
-    }
-    /** 列所有 keys — 用于调试 / claude code "看有哪些 session 可接续". */
-    async listKeys() {
-        try {
-            await fs.mkdir(this.cacheDir, { recursive: true });
-            const files = await fs.readdir(this.cacheDir);
-            return files
-                .filter(f => f.endsWith('.json') && !f.endsWith('.tmp'))
-                .map(f => SessionStore.filenameUnescape(f.slice(0, -'.json'.length)))
-                .sort();
-        }
-        catch {
-            return [];
-        }
-    }
-    /** 删一条 session — 失败抛错. */
-    async deleteKey(key) {
-        if (!key || key.includes('/') || key.includes('..')) {
-            throw new Error(`SessionStore: invalid key ${JSON.stringify(key)}`);
-        }
-        try {
-            await fs.unlink(this.pathFor(key));
-        }
-        catch (e) {
-            if (e?.code !== 'ENOENT')
-                throw e;
-        }
-    }
-}
-// --- sync fs helpers (避免 import 'fs' 二次) ---
-function fsSyncMkdir(dir) {
-    try {
-        require('fs').mkdirSync(dir, { recursive: true });
-    }
-    catch (e) {
-        if (e?.code !== 'EEXIST')
-            throw e;
-    }
-}
-function fsSyncWrite(filePath, content) {
-    require('fs').writeFileSync(filePath, content, 'utf-8');
-}
-function fsSyncRename(from, to) {
-    require('fs').renameSync(from, to);
-}
-/** 默认 store 实例 (用 ~/.bolloon/sessions/cache/). 业务代码用这个. */
-export const sessionStore = new SessionStore();

package/dist/agents/tool-registry.js

@@ -1,136 +0,0 @@
-/**
- * ToolRegistry — 工具注册表 + 别名解析器
- *
- * 2026-06-30 抽出:
- *   - 原 PiAgentSession.tools Map + resolveToolName 是 private 方法, 测试无法直接验证 alias 映射
- *   - 现在抽到独立模块, 完全可消融测试.
- *
- * 设计目标:
- *   1. claude code / 外部 harness 直接 import, 用同一份 alias 表
- *   2. 注册/反注册/查表都是纯方法, 无副作用
- *   3. 默认 alias 表跟 parse-tool-call.ts 保持一致 (read→read_file, bash→shell_exec 等)
- *   4. 自定义 alias 可注入 (业务可扩)
- *
- * 用法:
- *   import { ToolRegistry, DEFAULT_ALIASES } from './tool-registry';
- *   const reg = new ToolRegistry();
- *   reg.register({ name: 'shell_exec', description: '...', parameters: {}, execute });
- *   reg.has('bash');                  // true (alias 命中)
- *   reg.resolve('bash');              // 'shell_exec'
- *   reg.get('shell_exec')?.execute({ command: 'ls' });
- */
-/** 默认 alias 表 — Claude Code 风格 → bolloon 工具名 */
-export const DEFAULT_ALIASES = Object.freeze({
-    // file ops
-    read: 'read_file',
-    edit: 'edit_file',
-    write: 'write_file',
-    rm: 'delete_file',
-    mv: 'move_file',
-    cat: 'read_file',
-    // shell
-    bash: 'shell_exec',
-    shell: 'shell_exec',
-    sh: 'shell_exec',
-    // test/build
-    test: 'vitest_run',
-    vitest: 'vitest_run',
-    typecheck: 'tsc_check',
-    tsc: 'tsc_check',
-    // git
-    log: 'git_log',
-    show: 'git_show',
-    diff: 'git_diff',
-    commit: 'git_commit',
-    push: 'git_push',
-    branch: 'git_branch',
-    checkout: 'git_branch',
-    stash: 'git_stash',
-    // task
-    todo_write: 'create_task',
-    todowrite: 'create_task',
-    task: 'create_task',
-});
-export class ToolRegistry {
-    tools = new Map();
-    aliases;
-    constructor(aliases = DEFAULT_ALIASES) {
-        this.aliases = new Map();
-        for (const [k, v] of Object.entries(aliases)) {
-            this.aliases.set(k.toLowerCase(), v);
-        }
-    }
-    /** 注册一个 tool (alias 字典不动) */
-    register(tool) {
-        if (!tool.name)
-            throw new Error('ToolRegistry.register: tool.name 必填');
-        this.tools.set(tool.name, tool);
-    }
-    /** 批量注册 */
-    registerAll(tools) {
-        for (const t of tools)
-            this.register(t);
-    }
-    /** 反注册 — 名字 + 对应 alias (如果有指向它) 全部清掉. */
-    unregister(name) {
-        const existed = this.tools.delete(name);
-        // 反向清 alias
-        for (const [alias, target] of this.aliases.entries()) {
-            if (target === name)
-                this.aliases.delete(alias);
-        }
-        return existed;
-    }
-    /** 查 tool (只查主名, alias 走 resolve). */
-    get(name) {
-        return this.tools.get(name);
-    }
-    /** 是否注册 (主名 + alias 命中都算). */
-    has(name) {
-        if (this.tools.has(name))
-            return true;
-        const aliased = this.aliases.get(name.toLowerCase());
-        return aliased !== undefined && this.tools.has(aliased);
-    }
-    /** 把 LLM 输出的名字 (可能是 alias / 大小写不一致) 解析为标准主名. */
-    resolve(name) {
-        if (this.tools.has(name))
-            return name;
-        const lower = name.toLowerCase();
-        const aliased = this.aliases.get(lower);
-        if (aliased && this.tools.has(aliased))
-            return aliased;
-        if (this.tools.has(lower))
-            return lower;
-        return null;
-    }
-    /** 列出所有主名. 顺序按注册时间. */
-    list() {
-        return Array.from(this.tools.keys());
-    }
-    /** 当前 alias 表快照 (调试用). */
-    get aliasMap() {
-        const out = {};
-        for (const [k, v] of this.aliases.entries())
-            out[k] = v;
-        return out;
-    }
-    /** 个数 */
-    get size() {
-        return this.tools.size;
-    }
-    /** 清空 */
-    clear() {
-        this.tools.clear();
-        // alias 表保留 (除非调用方显式 clear, alias 表是稳定的)
-    }
-    /** 调一个 tool, 名字可能是 alias. throw if 未知. */
-    async invoke(name, args) {
-        const canonical = this.resolve(name);
-        if (!canonical) {
-            throw new Error(`ToolRegistry.invoke: 未知工具名 ${JSON.stringify(name)}`);
-        }
-        const tool = this.tools.get(canonical);
-        return tool.execute(args);
-    }
-}

package/dist/bootstrap/chat-archiver.js

@@ -1,276 +0,0 @@
-/**
- * chat-archiver.ts — 按 peer 月度归档聊天记录 + 接 memory 摘要
- *
- * 设计目的 (2026-07-05):
- *   解决"对方不在线也能看到历史"的核心问题.
- *   sessions/cache/<channelId>.json 单文件无限增长, 超过 50MB 直接拒绝加载.
- *
- *   新方案:
- *     - 每次 /message 处理后, 同时写到 3 个地方:
- *       ① sessions/cache/<channelId>.json (原有, 最近窗口用)
- *       ② ~/.bolloon/peers/<pk>/chat-<YYYY-MM>.md (按 peer 月度归档)
- *       ③ ~/.bolloon/memory/<agentId>/peers/<pk>/<YYYY-MM>.summary.md (月度摘要)
- *
- *   触发时机:
- *     - appendChatArchive(): 每次保存远端/本地 user + ai 消息后立即调用 (高频)
- *     - compressMonthlyArchive(): 月底 / 90 天滚动 / 显式调用 (低频, 调 LLM 摘要)
- *
- *   peer 维度 而不是 channel 维度: 同一个远端节点可能跨多个 channel 协作,
- *   按 peer 归档才能完整还原"与对方的对话历史".
- */
-import * as fs from 'fs/promises';
-import * as fsSync from 'fs';
-import * as path from 'path';
-import * as os from 'os';
-import * as peerFs from '../network/peer-fs.js';
-// ============== 路径 ==============
-const HOME = process.env.BOLLOON_HOME || path.join(os.homedir(), '.bolloon');
-export function getPeerSummaryDir(agentId, publicKey) {
-    const safeAgent = sanitizeId(agentId);
-    const safePeer = sanitizeId(publicKey);
-    return path.join(HOME, 'memory', safeAgent, 'peers', safePeer);
-}
-export function getPeerSummaryPath(agentId, publicKey, yearMonth) {
-    return path.join(getPeerSummaryDir(agentId, publicKey), `${yearMonth}.summary.md`);
-}
-export function getPeerSummaryCursorPath(agentId, publicKey, yearMonth) {
-    return path.join(getPeerSummaryDir(agentId, publicKey), `${yearMonth}.cursor`);
-}
-function sanitizeId(s) {
-    return s.replace(/[^a-zA-Z0-9_-]/g, '_').slice(0, 64);
-}
-// ============== appendChatArchive — 高频, 每次 message 都调 ==============
-/**
- * 追加一条 chat entry 到 peer 的月度归档.
- * 不依赖 LLM, 纯文件 IO; 失败静默不阻塞主对话.
- */
-export async function appendChatArchive(opts) {
-    try {
-        await peerFs.appendChat(opts.publicKey, opts.entry);
-        return { ok: true, path: peerFs.getPeerChatPath(opts.publicKey) };
-    }
-    catch (err) {
-        return { ok: false, error: err?.message || String(err) };
-    }
-}
-// ============== compressMonthlyArchive — 低频, 月底/显式调 ==============
-/**
- * 读取月度 markdown 归档, 调 LLM 摘要, append 写到 memory.
- * 用 cursor 跟踪已摘要到第几行, 增量压缩.
- */
-async function readArchiveLines(publicKey, yearMonth) {
-    const file = peerFs.getPeerChatPath(publicKey, yearMonth);
-    try {
-        const raw = await fs.readFile(file, 'utf-8');
-        return raw.split('\n');
-    }
-    catch {
-        return [];
-    }
-}
-async function readSummaryCursor(cursorPath) {
-    try {
-        const raw = await fs.readFile(cursorPath, 'utf-8');
-        const n = parseInt(raw.trim(), 10);
-        return Number.isFinite(n) && n >= 0 ? n : 0;
-    }
-    catch {
-        return 0;
-    }
-}
-async function writeSummaryCursor(cursorPath, value) {
-    await fs.mkdir(path.dirname(cursorPath), { recursive: true });
-    await fs.writeFile(cursorPath, String(value), 'utf-8');
-}
-/**
- * 尝试调 LLM 生成月度摘要. 失败 → 抛错, 由 caller fallback 到模板.
- */
-async function tryMonthlyLlm(systemPrompt, userPrompt) {
-    try {
-        const piAi = await import('../llm/pi-ai.js');
-        const generateText = piAi.generateText;
-        if (typeof generateText !== 'function')
-            throw new Error('generateText not exported');
-        const result = await generateText({
-            messages: [
-                { role: 'system', content: systemPrompt },
-                { role: 'user', content: userPrompt },
-            ],
-            temperature: 0.2,
-            maxTokens: 1000,
-        });
-        const text = result?.reply || result?.text || '';
-        if (typeof text !== 'string' || text.length < 20)
-            throw new Error('LLM returned too-short text');
-        return text;
-    }
-    catch (e) {
-        throw new Error(`LLM call failed: ${e?.message || String(e)}`);
-    }
-}
-/**
- * 模板 fallback: 简单统计 user/ai 数量, 列前 5 条关键内容.
- */
-function templateMonthlySummary(opts) {
-    const user = opts.entries.filter(e => e.msgType === 'user' || e.source === 'local' || e.source === 'remote');
-    const ai = opts.entries.filter(e => e.msgType === 'ai' || e.source?.startsWith('ai-mention'));
-    const channelSet = new Set(opts.entries.map(e => e.channelName).filter(Boolean));
-    const sampleUser = user.slice(0, 5).map(e => `  - [${e.ts}] ${e.text.slice(0, 80)}`).join('\n');
-    const sampleAi = ai.slice(0, 5).map(e => `  - [${e.ts}] ${e.text.slice(0, 80)}`).join('\n');
-    return `# 月度对话摘要 — ${opts.yearMonth} (peer ${opts.publicKey.slice(0, 12)}…)
-
-时间: ${opts.timestamp}
-对话轮次: ${opts.entries.length} (user=${user.length}, ai=${ai.length})
-涉及 channel: ${Array.from(channelSet).slice(0, 10).join(', ') || '(无)'}
-
-## 用户关键提问
-${sampleUser || '  (无)'}
-
-## AI 关键回答
-${sampleAi || '  (无)'}
-
-## 备注
-- 此摘要由模板生成 (LLM 调用失败 fallback)
-`;
-}
-/**
- * 解析 markdown 归档为 entries (用 "### " 切分)
- */
-function parseArchiveEntries(lines) {
-    const out = [];
-    let i = 0;
-    while (i < lines.length) {
-        if (lines[i].startsWith('### ')) {
-            const header = lines[i];
-            // 解析 "### 2026-07-05 10:30:45 UTC [channelName] — source"
-            const m = header.match(/^### (.+?)\s*(?:\[(.+?)\])?\s*—\s*(\S+?)(?:\s+\(.+?\))?$/);
-            if (m) {
-                const ts = m[1].replace(/\s+UTC$/, '').trim();
-                const channelName = m[2];
-                const source = m[3];
-                // 找下一段 "### " 或 "---" 之间的内容
-                let j = i + 1;
-                const bodyLines = [];
-                while (j < lines.length && !lines[j].startsWith('### ') && !lines[j].startsWith('---')) {
-                    bodyLines.push(lines[j]);
-                    j++;
-                }
-                const text = bodyLines.join('\n').trim();
-                out.push({
-                    startLine: i,
-                    entry: {
-                        ts: ts.includes('T') ? ts : ts.replace(' ', 'T') + 'Z',
-                        source,
-                        channelName,
-                        text,
-                    }
-                });
-                i = j;
-                continue;
-            }
-        }
-        i++;
-    }
-    return out;
-}
-/**
- * 主入口: 压缩月度归档到 memory.
- */
-export async function compressMonthlyArchive(opts) {
-    const agentId = sanitizeId(opts.agentId);
-    const publicKey = opts.publicKey;
-    const yearMonth = opts.yearMonth || peerFs.currentYearMonth();
-    const home = opts.home || HOME;
-    const summaryPath = getPeerSummaryPath(agentId, publicKey, yearMonth);
-    const cursorPath = getPeerSummaryCursorPath(agentId, publicKey, yearMonth);
-    const archivePath = peerFs.getPeerChatPath(publicKey, yearMonth);
-    // 读归档
-    const lines = await readArchiveLines(publicKey, yearMonth);
-    if (lines.length === 0) {
-        return { summaryPath, cursorPath, entriesCount: 0, bytesWritten: 0, skipped: 'no-archive' };
-    }
-    // 解析 entries
-    const allEntries = parseArchiveEntries(lines);
-    const cursor = await readSummaryCursor(cursorPath);
-    const newEntries = allEntries.slice(cursor);
-    const minNew = opts.minNewEntries ?? 20;
-    if (newEntries.length < minNew) {
-        return { summaryPath, cursorPath, entriesCount: newEntries.length, bytesWritten: 0, skipped: 'too-few-entries' };
-    }
-    const timestamp = new Date().toISOString();
-    let summaryBody;
-    try {
-        const sysPrompt = '你是 bolloon 月度记忆压缩助手. 输入是按 peer 月度归档的聊天记录 markdown, 输出 400-800 字中文摘要, 包含: 1) 关键合作主题; 2) 共同完成的成果; 3) 待跟进事项; 4) 对方能力偏好. 不要寒暄, 不复述已知.';
-        const recentSnippet = newEntries.slice(-30).map(e => {
-            const ts = e.entry.ts;
-            const chan = e.entry.channelName ? `[${e.entry.channelName}]` : '';
-            return `[${ts}]${chan} ${e.entry.source}: ${e.entry.text}`;
-        }).join('\n').slice(0, 8000);
-        const userPrompt = `Peer: ${publicKey}\n月份: ${yearMonth}\n时间: ${timestamp}\n新增条数: ${newEntries.length}\n\n最近对话:\n${recentSnippet}`;
-        summaryBody = await tryMonthlyLlm(sysPrompt, userPrompt);
-    }
-    catch (e) {
-        summaryBody = templateMonthlySummary({
-            publicKey,
-            yearMonth,
-            entries: newEntries.map(e => e.entry),
-            timestamp,
-        });
-    }
-    const block = `\n\n---\n\n## 增量摘要 @ ${timestamp} (${newEntries.length} entries)\n\n${summaryBody.trim()}\n`;
-    await fs.mkdir(path.dirname(summaryPath), { recursive: true });
-    await fs.appendFile(summaryPath, block, 'utf-8');
-    await writeSummaryCursor(cursorPath, allEntries.length);
-    return {
-        summaryPath,
-        cursorPath,
-        entriesCount: newEntries.length,
-        bytesWritten: Buffer.byteLength(block, 'utf-8'),
-    };
-}
-/**
- * 把 session.json 中的所有消息按 peer 切分, 一次性 archive 到各 peer 月度文件.
- * 用于: ① 启动时迁移已有 session 历史到 peer archive; ② 兜底补档.
- */
-export async function archiveSessionMessagesForPeer(opts) {
-    let written = 0;
-    let errors = 0;
-    for (const m of opts.messages) {
-        const ts = m.timestamp || new Date().toISOString();
-        const entry = {
-            ts,
-            source: m.source || (m.type === 'user' ? 'remote' : 'ai-mention'),
-            channelId: m.channelId,
-            channelName: m.channelName,
-            text: m.content,
-            fromPublicKey: m.fromPublicKey || opts.publicKey,
-            msgType: m.type,
-        };
-        const r = await appendChatArchive({ publicKey: opts.publicKey, entry });
-        if (r.ok)
-            written++;
-        else
-            errors++;
-    }
-    return { written, errors };
-}
-/**
- * 列出指定 peer 所有可用的月度摘要 (按月份倒序)
- */
-export async function listPeerSummaries(agentId, publicKey) {
-    const dir = getPeerSummaryDir(agentId, publicKey);
-    try {
-        const entries = await fs.readdir(dir);
-        return entries
-            .filter(e => /^\d{4}-\d{2}\.summary\.md$/.test(e))
-            .map(e => ({
-            yearMonth: e.replace(/\.summary\.md$/, ''),
-            path: path.join(dir, e),
-            size: fsSync.statSync(path.join(dir, e)).size,
-        }))
-            .sort((a, b) => b.yearMonth.localeCompare(a.yearMonth));
-    }
-    catch {
-        return [];
-    }
-}