@bolloon/bolloon-agent 0.2.12 → 0.2.14

package/dist/web/routes-llm-config.js

@@ -0,0 +1,279 @@
+/**
+ * routes-llm-config.ts — LLM/Video/Audio 配置路由 (2026-07-06 抽出)
+ *
+ * 从 src/web/server.ts 抽出 (~230 行).
+ * 包含 /api/llm-config/* /api/video-config/* /api/audio-config/* /api/ai-parse
+ */
+import { llmConfigStore } from '../llm/config-store.js';
+import { videoConfigStore } from '../llm/video-config-store.js';
+import { audioConfigStore } from '../llm/audio-config-store.js';
+import { initMinimax, getMinimax } from '../constraints/index.js';
+export function registerLlmConfigRoutes(app) {
+    // ==================== LLM 配置 API ====================
+    // 获取所有 LLM 配置
+    app.get('/api/llm-config', async (req, res) => {
+        try {
+            const config = await llmConfigStore.getConfig();
+            const providerInfo = llmConfigStore.getAllProviderInfo();
+            // 隐藏 API Key
+            const safeConfig = {
+                ...config,
+                providers: Object.fromEntries(Object.entries(config.providers).map(([key, val]) => [
+                    key,
+                    { ...val, apiKey: val.apiKey ? '***' + val.apiKey.slice(-4) : '' }
+                ])),
+                providerInfo
+            };
+            res.json(safeConfig);
+        }
+        catch (err) {
+            res.status(500).json({ error: err.message });
+        }
+    });
+    // 更新 LLM 配置
+    app.post('/api/llm-config', async (req, res) => {
+        try {
+            const { provider, config } = req.body;
+            if (!provider || !config) {
+                return res.status(400).json({ error: 'provider and config required' });
+            }
+            // 如果前端发的是掩码（***xxx），从当前配置里取真实 key
+            const currentConfig = await llmConfigStore.getProvider(provider);
+            if (currentConfig && config.apiKey && config.apiKey.startsWith('***')) {
+                config.apiKey = currentConfig.apiKey;
+            }
+            await llmConfigStore.updateProvider(provider, config);
+            // 如果是活跃供应商，重新初始化 Pi SDK
+            const currentActive = await llmConfigStore.getActiveProvider();
+            if (provider === currentActive) {
+                const newConfig = await llmConfigStore.getActiveProviderConfig();
+                if (newConfig) {
+                    initMinimax({
+                        provider,
+                        apiKey: newConfig.apiKey || undefined,
+                        baseUrl: newConfig.baseUrl || undefined,
+                        model: newConfig.model || undefined
+                    });
+                }
+            }
+            res.json({ ok: true });
+        }
+        catch (err) {
+            res.status(500).json({ error: err.message });
+        }
+    });
+    // 设置活跃供应商
+    app.post('/api/llm-provider', async (req, res) => {
+        try {
+            const { provider } = req.body;
+            if (!provider) {
+                return res.status(400).json({ error: 'provider required' });
+            }
+            await llmConfigStore.setActiveProvider(provider);
+            // 重新初始化 Pi SDK
+            const config = await llmConfigStore.getActiveProviderConfig();
+            if (config) {
+                initMinimax({
+                    provider: provider,
+                    apiKey: config.apiKey || undefined,
+                    baseUrl: config.baseUrl || undefined,
+                    model: config.model || undefined
+                });
+            }
+            res.json({ ok: true, provider });
+        }
+        catch (err) {
+            res.status(500).json({ error: err.message });
+        }
+    });
+    // 测试供应商连接
+    app.post('/api/llm-test', async (req, res) => {
+        try {
+            const { provider } = req.body;
+            if (!provider) {
+                return res.status(400).json({ error: 'provider required' });
+            }
+            const result = await llmConfigStore.testProvider(provider);
+            res.json(result);
+        }
+        catch (err) {
+            res.status(500).json({ error: err.message });
+        }
+    });
+    // ==================== 视频生成配置 (Seedance 等) ====================
+    // 获取视频生成配置
+    app.get('/api/video-config', async (req, res) => {
+        try {
+            const config = await videoConfigStore.getConfig();
+            const providerInfo = videoConfigStore.getAllProviderInfo();
+            // 脱敏：不返回 apiKey 明文
+            const masked = Object.fromEntries(Object.entries(config.providers).map(([key, val]) => [
+                key,
+                { ...val, apiKey: val.apiKey ? '***' + val.apiKey.slice(-4) : '' }
+            ]));
+            res.json({
+                activeProvider: config.activeProvider,
+                providers: masked,
+                providerInfo
+            });
+        }
+        catch (err) {
+            res.status(500).json({ error: err.message });
+        }
+    });
+    // 更新视频供应商配置
+    app.post('/api/video-config', async (req, res) => {
+        try {
+            const { provider, config } = req.body;
+            if (!provider || !config) {
+                return res.status(400).json({ error: 'provider and config required' });
+            }
+            // 如果前端发的是掩码（***xxx），从当前配置里取真实 key
+            const currentConfig = await videoConfigStore.getProvider(provider);
+            if (currentConfig && config.apiKey && config.apiKey.startsWith('***')) {
+                config.apiKey = currentConfig.apiKey;
+            }
+            await videoConfigStore.updateProvider(provider, config);
+            res.json({ ok: true });
+        }
+        catch (err) {
+            res.status(500).json({ error: err.message });
+        }
+    });
+    // 测试视频供应商连接
+    app.post('/api/video-test', async (req, res) => {
+        try {
+            const { provider } = req.body;
+            if (!provider) {
+                return res.status(400).json({ error: 'provider required' });
+            }
+            const result = await videoConfigStore.testProvider(provider);
+            res.json(result);
+        }
+        catch (err) {
+            res.status(500).json({ error: err.message });
+        }
+    });
+    // ==================== 音频生成配置 (TTS / Music) ====================
+    // 获取音频配置
+    app.get('/api/audio-config', async (req, res) => {
+        try {
+            const config = await audioConfigStore.getConfig();
+            const providerInfo = audioConfigStore.getAllProviderInfo();
+            const masked = Object.fromEntries(Object.entries(config.providers).map(([key, val]) => [
+                key,
+                { ...val, apiKey: val.apiKey ? '***' + val.apiKey.slice(-4) : '' }
+            ]));
+            res.json({
+                activeProvider: config.activeProvider,
+                providers: masked,
+                providerInfo
+            });
+        }
+        catch (err) {
+            res.status(500).json({ error: err.message });
+        }
+    });
+    // 更新音频供应商配置
+    app.post('/api/audio-config', async (req, res) => {
+        try {
+            const { provider, config } = req.body;
+            if (!provider || !config) {
+                return res.status(400).json({ error: 'provider and config required' });
+            }
+            // 掩码回写真实 key
+            const currentConfig = await audioConfigStore.getProvider(provider);
+            if (currentConfig && config.apiKey && config.apiKey.startsWith('***')) {
+                config.apiKey = currentConfig.apiKey;
+            }
+            await audioConfigStore.updateProvider(provider, config);
+            res.json({ ok: true });
+        }
+        catch (err) {
+            res.status(500).json({ error: err.message });
+        }
+    });
+    // 测试音频供应商连接
+    app.post('/api/audio-test', async (req, res) => {
+        try {
+            const { provider } = req.body;
+            if (!provider) {
+                return res.status(400).json({ error: 'provider required' });
+            }
+            const result = await audioConfigStore.testProvider(provider);
+            res.json(result);
+        }
+        catch (err) {
+            res.status(500).json({ error: err.message });
+        }
+    });
+    // 统一 AI 解析入口：CLI / 接收方节点 调这里完成 LLM + judgment + harness
+    // 入参: { text, mimeType, fileName, fromNodeId, source }
+    // 出参: { summary, qualityScore, judgmentId?, gateArtifact? }
+    app.post('/api/ai-parse', async (req, res) => {
+        try {
+            const { text, mimeType, fileName, fromNodeId, source } = req.body || {};
+            if (!text || !fileName) {
+                return res.status(400).json({ error: 'text and fileName required' });
+            }
+            const truncated = text.length > 6000 ? text.substring(0, 6000) + '...[截断]' : text;
+            const prompt = `请分析以下 ${mimeType || 'text'} 文档，并给出 (1) 一句话中文摘要 (2) 三个关键要点 (3) 质量评分(0-1)。\n\n文件名: ${fileName}\n\n内容:\n${truncated}`;
+            // 1. LLM 解析
+            const llm = getMinimax();
+            const t0 = Date.now();
+            const llmResult = await llm.summarize(prompt);
+            const dt = Date.now() - t0;
+            const out = {
+                ok: true,
+                summary: llmResult.summary,
+                qualityScore: llmResult.qualityScore,
+                latencyMs: dt,
+                mimeType: mimeType || 'text/plain',
+                fileName,
+            };
+            // 2. 蒸馏为 judgment (异步,失败不影响主返回)
+            try {
+                const judgmentMod = await import('../pi-ecosystem-judgment/index.js');
+                await judgmentMod.initializeJudgmentStore();
+                const j = await judgmentMod.createJudgment({
+                    type: 'trajectory',
+                    content: `AI 解析 ${fileName}: ${llmResult.summary.slice(0, 200)}`,
+                    source: 'agent',
+                    confidence: Math.min(1, llmResult.qualityScore),
+                    context: `ai-parse:${mimeType || 'text'}:${source || 'p2p'}`,
+                    evidence: {
+                        trajectory: [{
+                                timestamp: new Date().toISOString(),
+                                action: `parse:${fileName}`,
+                                outcome: `score=${llmResult.qualityScore.toFixed(2)}`,
+                                approved: true,
+                            }],
+                    },
+                });
+                out.judgmentId = j.id;
+            }
+            catch (e) {
+                out.judgmentError = e.message;
+            }
+            // 3. 在 harness 落产物 (异步,失败不影响)
+            try {
+                const harnessMod = await import('../bollharness-integration/index.js');
+                const gate = new harnessMod.GateStateMachine();
+                gate.submitArtifact(`ai-parse:${fileName}`, {
+                    summary: llmResult.summary,
+                    score: llmResult.qualityScore,
+                    fromNodeId: fromNodeId || null,
+                    parsedAt: Date.now(),
+                });
+                out.gateArtifact = `ai-parse:${fileName}`;
+            }
+            catch (e) {
+                out.gateError = e.message;
+            }
+            res.json(out);
+        }
+        catch (err) {
+            res.status(500).json({ error: err.message });
+        }
+    });
+}

package/dist/web/routes-tasks.js

@@ -0,0 +1,222 @@
+/**
+ * routes-tasks.ts — 任务队列 CRUD 路由 (2026-07-06 抽出)
+ *
+ * 从 src/web/server.ts 抽出 (~160 行).
+ * 包含 /api/tasks/* (GET, POST, PATCH, DELETE, execute, execute-next)
+ */
+import { loadTaskQueue, saveTaskQueue, isTaskExecuting } from './server-storage.js';
+import { documentReader } from '../documents/reader.js';
+export function registerTaskRoutes(app, opts) {
+    const { broadcast, getAgentForChannel } = opts;
+    // 获取所有任务
+    app.get('/api/tasks', async (req, res) => {
+        try {
+            const tasks = await loadTaskQueue();
+            res.json(tasks);
+        }
+        catch (err) {
+            res.status(500).json({ error: err.message });
+        }
+    });
+    // 创建新任务
+    app.post('/api/tasks', async (req, res) => {
+        try {
+            const { type, title, description, steps } = req.body;
+            if (!type || !title) {
+                return res.status(400).json({ error: 'type and title required' });
+            }
+            const tasks = await loadTaskQueue();
+            const task = {
+                id: `task_${Date.now()}_${Math.random().toString(36).substring(2, 8)}`,
+                type,
+                title,
+                description,
+                status: 'pending',
+                progress: 0,
+                createdAt: new Date().toISOString(),
+                updatedAt: new Date().toISOString(),
+                steps: steps?.map((s, i) => ({
+                    id: `step_${i}`,
+                    name: s,
+                    status: 'pending'
+                }))
+            };
+            tasks.push(task);
+            await saveTaskQueue(tasks);
+            res.json(task);
+        }
+        catch (err) {
+            res.status(500).json({ error: err.message });
+        }
+    });
+    // 获取单个任务
+    app.get('/api/tasks/:taskId', async (req, res) => {
+        try {
+            const { taskId } = req.params;
+            const tasks = await loadTaskQueue();
+            const task = tasks.find(t => t.id === taskId);
+            if (!task) {
+                return res.status(404).json({ error: 'Task not found' });
+            }
+            res.json(task);
+        }
+        catch (err) {
+            res.status(500).json({ error: err.message });
+        }
+    });
+    // 更新任务
+    app.patch('/api/tasks/:taskId', async (req, res) => {
+        try {
+            const { taskId } = req.params;
+            const { status, currentStep } = req.body;
+            const tasks = await loadTaskQueue();
+            const taskIndex = tasks.findIndex(t => t.id === taskId);
+            if (taskIndex === -1) {
+                return res.status(404).json({ error: 'Task not found' });
+            }
+            if (status) {
+                tasks[taskIndex].status = status;
+            }
+            if (currentStep !== undefined) {
+                tasks[taskIndex].currentStep = currentStep;
+            }
+            tasks[taskIndex].updatedAt = new Date().toISOString();
+            await saveTaskQueue(tasks);
+            res.json(tasks[taskIndex]);
+        }
+        catch (err) {
+            res.status(500).json({ error: err.message });
+        }
+    });
+    // 删除任务
+    app.delete('/api/tasks/:taskId', async (req, res) => {
+        try {
+            const { taskId } = req.params;
+            const tasks = await loadTaskQueue();
+            const filtered = tasks.filter(t => t.id !== taskId);
+            await saveTaskQueue(filtered);
+            res.json({ ok: true });
+        }
+        catch (err) {
+            res.status(500).json({ error: err.message });
+        }
+    });
+    // 执行任务（自动执行下一步）
+    app.post('/api/tasks/:taskId/execute', async (req, res) => {
+        try {
+            const { taskId } = req.params;
+            const { channelId } = req.body;
+            if (!channelId) {
+                return res.status(400).json({ error: 'channelId required' });
+            }
+            const tasks = await loadTaskQueue();
+            const task = tasks.find(t => t.id === taskId);
+            if (!task) {
+                return res.status(404).json({ error: 'Task not found' });
+            }
+            if (isTaskExecuting()) {
+                return res.status(409).json({ error: 'Another task is currently executing' });
+            }
+            // 异步执行任务
+            executeTask(task, channelId, getAgentForChannel, broadcast);
+            res.json({ ok: true, taskId: task.id });
+        }
+        catch (err) {
+            res.status(500).json({ error: err.message });
+        }
+    });
+    // 执行下一个待处理任务
+    app.post('/api/tasks/execute-next', async (req, res) => {
+        try {
+            const { channelId } = req.body;
+            if (!channelId) {
+                return res.status(400).json({ error: 'channelId required' });
+            }
+            const tasks = await loadTaskQueue();
+            const nextTask = tasks.find(t => t.status === 'pending');
+            if (!nextTask) {
+                return res.json({ ok: false, message: 'No pending tasks' });
+            }
+            if (isTaskExecuting()) {
+                return res.status(409).json({ error: 'Another task is currently executing' });
+            }
+            // 异步执行任务
+            executeTask(nextTask, channelId, getAgentForChannel, broadcast);
+            res.json({ ok: true, taskId: nextTask.id });
+        }
+        catch (err) {
+            res.status(500).json({ error: err.message });
+        }
+    });
+}
+// ==================== Task Execution ====================
+async function executeTask(task, channelId, getAgentForChannel, broadcast) {
+    // 使用 server-storage 的 startTaskExecution 做锁
+    const { startTaskExecution, endTaskExecution } = await import('./server-storage.js');
+    if (!startTaskExecution(task.id))
+        return;
+    const agent = await getAgentForChannel(channelId);
+    const tasks = await loadTaskQueue();
+    const taskIndex = tasks.findIndex(t => t.id === task.id);
+    if (taskIndex >= 0) {
+        tasks[taskIndex].status = 'running';
+        tasks[taskIndex].updatedAt = new Date().toISOString();
+        await saveTaskQueue(tasks);
+    }
+    broadcast({ type: 'task_status', taskId: task.id, status: 'running', progress: 0 }, channelId);
+    try {
+        let result = '';
+        switch (task.type) {
+            case 'chat':
+                if (task.description) {
+                    broadcast({ type: 'status', content: `执行任务: ${task.title}` }, channelId);
+                    result = await agent.prompt(task.description);
+                }
+                break;
+            case 'read':
+                if (task.description) {
+                    broadcast({ type: 'status', content: `读取文档: ${task.description}` }, channelId);
+                    const content = await documentReader.read(task.description);
+                    result = `📄 文档读取完成\n\n${content.text.substring(0, 500)}${content.text.length > 500 ? '...' : ''}`;
+                }
+                break;
+            case 'summarize':
+                if (task.description) {
+                    broadcast({ type: 'status', content: `总结文档: ${task.description}` }, channelId);
+                    const content = await documentReader.read(task.description);
+                    const { getMinimax } = await import('../constraints/index.js');
+                    const llm = getMinimax();
+                    const summary = await llm.summarize(content.text);
+                    result = `📝 文档总结:\n\n${summary.summary}`;
+                }
+                break;
+            default:
+                result = '未知任务类型';
+        }
+        // 更新任务状态
+        const tasks = await loadTaskQueue();
+        const idx = tasks.findIndex(t => t.id === task.id);
+        if (idx >= 0) {
+            tasks[idx].status = 'completed';
+            tasks[idx].progress = 100;
+            tasks[idx].result = result;
+            tasks[idx].updatedAt = new Date().toISOString();
+            await saveTaskQueue(tasks);
+        }
+        broadcast({ type: 'task_status', taskId: task.id, status: 'completed', progress: 100, result }, channelId);
+        broadcast({ type: 'ai', content: result }, channelId);
+    }
+    catch (error) {
+        const tasks = await loadTaskQueue();
+        const idx = tasks.findIndex(t => t.id === task.id);
+        if (idx >= 0) {
+            tasks[idx].status = 'failed';
+            tasks[idx].error = error.message;
+            tasks[idx].updatedAt = new Date().toISOString();
+            await saveTaskQueue(tasks);
+        }
+        broadcast({ type: 'task_status', taskId: task.id, status: 'failed', error: error.message }, channelId);
+        broadcast({ type: 'error', content: `任务执行失败: ${error.message}` }, channelId);
+    }
+    endTaskExecution();
+}

package/dist/web/server-sse.js

@@ -0,0 +1,121 @@
+/**
+ * server-sse.ts — server.ts 拆出的 SSE 实时推送层
+ *
+ * 包含:
+ *   - sseClients: 客户端连接 Set
+ *   - broadcast(): 推 SSE 事件 + 加 seq/msgId envelope
+ *   - nextEventSeq / nextMsgId: 用于客户端断线重连 resume
+ *   - installChatBusHook / installSelfImproveHook: 桥接 chat event bus / self-improve bus → SSE
+ *
+ * 从 src/web/server.ts 抽出 (2026-07-06).
+ * SSEClient type 来自 ./server-types.ts.
+ */
+import * as crypto from 'crypto';
+const sseClients = new Set();
+export function getSseClients() {
+    return sseClients;
+}
+export function addSseClient(client) {
+    sseClients.add(client);
+}
+export function removeSseClient(client) {
+    sseClients.delete(client);
+}
+// 2026-07-06: 每个 channelId 维护递增 sequence + msgId, 让前端能去重 + 重连后 resume
+const channelEventSeq = new Map();
+const channelMsgIds = new Map(); // channelId -> 上一条 msgId (uuid)
+export function nextEventSeq(channelId) {
+    if (!channelId)
+        return 0;
+    const cur = channelEventSeq.get(channelId) || 0;
+    const next = cur + 1;
+    channelEventSeq.set(channelId, next);
+    return next;
+}
+export function nextMsgId(channelId) {
+    const id = `msg_${Date.now()}_${crypto.randomBytes(4).toString('hex')}`;
+    if (channelId)
+        channelMsgIds.set(channelId, id);
+    return id;
+}
+/**
+ * 给 watchdog 喂活动, 让 broadcast() / 模块级业务函数能埋点喂活动
+ * 之前在 createWebServer 闭包内, 闭包外的 broadcast() 拿不到 → 误判 30min 无活动 → 自杀.
+ */
+let watchdogRef = null;
+export function setWatchdogRef(ref) {
+    watchdogRef = ref;
+}
+export function broadcast(data, channelId) {
+    watchdogRef?.recordActivity?.();
+    const seq = nextEventSeq(channelId);
+    const msgId = (data.type === 'ai' || data.type === 'user')
+        ? nextMsgId(channelId)
+        : `evt_${Date.now()}_${crypto.randomBytes(4).toString('hex')}`;
+    const envelope = { ...data, channelId, seq, msgId };
+    const message = `data: ${JSON.stringify(envelope)}\n\n`;
+    console.log(`[broadcast] type=${data.type}, channelId=${channelId}, seq=${seq}, msgId=${msgId}, clients=${sseClients.size}`);
+    for (const client of sseClients) {
+        if (!channelId || client.channelId === channelId) {
+            try {
+                client.res.write(message);
+            }
+            catch (e) {
+                console.error(`[broadcast] 写入失败:`, e.message);
+            }
+        }
+    }
+}
+// ============================================================================
+// Chat 事件总线 -> SSE 桥 (供前端 inbox UI 用)
+// ============================================================================
+let chatBusHookInstalled = false;
+export async function installChatBusHook() {
+    if (chatBusHookInstalled)
+        return;
+    chatBusHookInstalled = true;
+    try {
+        const { chatEventBus } = await import('../agents/p2p-chat-tools.js');
+        chatEventBus.on('chat', (ev) => {
+            broadcast({ type: 'chat_event', chatKind: ev.kind, payload: ev }, undefined);
+        });
+        console.log('[chat-bus] SSE bridge installed');
+    }
+    catch (e) {
+        console.warn('[chat-bus] install failed:', e.message);
+    }
+}
+// ============================================================================
+// Self-Improve Bus -> SSE 桥 (供前端 / 用户看到自改触发)
+// ============================================================================
+let selfImproveHookInstalled = false;
+export async function installSelfImproveHook() {
+    if (selfImproveHookInstalled)
+        return;
+    selfImproveHookInstalled = true;
+    try {
+        const { onSelfImproveTrigger } = await import('../heartbeat/self-improve-bus.js');
+        const { runSelfImproveLoop } = await import('../agents/pi-sdk.js');
+        onSelfImproveTrigger(async (event, goal) => {
+            broadcast({
+                type: 'self_improve_triggered',
+                eventKind: event.kind,
+                details: event.details,
+                goal,
+                ts: Date.now()
+            }, undefined);
+            const result = await runSelfImproveLoop(goal);
+            broadcast({
+                type: 'self_improve_result',
+                success: result.success,
+                output: result.output,
+                error: result.error,
+                ts: Date.now()
+            }, undefined);
+        });
+        console.log('[self-improve] SSE bridge installed');
+    }
+    catch (e) {
+        console.warn('[self-improve] install failed:', e.message);
+    }
+}