@cc-monitoring/proxy-core 0.1.8 → 0.1.9

package/dist/mcp-server/run.js

@@ -0,0 +1,14 @@
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import { loadConfig } from '../config/index.js';
+import { configureDbPath, initDatabase } from '../utils/db.js';
+import { createMcpServer } from './index.js';
+export async function runMcpServer() {
+    const config = await loadConfig();
+    if (config.storage?.path) {
+        configureDbPath(config.storage.path);
+    }
+    initDatabase();
+    const server = createMcpServer(config);
+    const transport = new StdioServerTransport();
+    await server.connect(transport);
+}

package/dist/mcp-server/tools.d.ts

@@ -0,0 +1,11 @@
+import { Server } from '@modelcontextprotocol/sdk/server/index.js';
+import type { AppConfig } from '../config/schema.js';
+export interface McpToolContext {
+    config: AppConfig;
+}
+export declare const MCP_TOOLS: {
+    name: string;
+    description: string;
+    inputSchema: Record<string, unknown>;
+}[];
+export declare function registerMcpTools(server: Server, ctx: McpToolContext): void;

package/dist/mcp-server/tools.js

@@ -0,0 +1,219 @@
+import fs from 'fs';
+import { CallToolRequestSchema } from '@modelcontextprotocol/sdk/types.js';
+import { getDatabase } from '../utils/db.js';
+import { listRequestDB } from '../utils/request-writer.js';
+import { listProviders } from '../utils/providers-db.js';
+import { knowledgeStore } from '../utils/knowledge-store.js';
+function ok(text) {
+    return { content: [{ type: 'text', text }] };
+}
+function err(e) {
+    const msg = e instanceof Error ? e.message : String(e);
+    return { content: [{ type: 'text', text: `error: ${msg}` }] };
+}
+// --- Tool implementations ---
+async function handleGetStatus(_args, ctx) {
+    const db = getDatabase();
+    const lastRow = db.prepare('SELECT MAX(timestamp) AS ts FROM requests').get();
+    const countRow = db.prepare('SELECT COUNT(*) AS n FROM requests').get();
+    let defaultProvider = null;
+    try {
+        const providers = listProviders(db);
+        const def = providers.find(p => p.default);
+        if (def) {
+            defaultProvider = { name: def.name, model: def.model, base_url: def.base_url };
+        }
+    }
+    catch {
+        // providers table may not exist on a fresh DB
+    }
+    return ok(JSON.stringify({
+        server: `${ctx.config.server.host}:${ctx.config.server.port}`,
+        proxyTarget: ctx.config.proxy.target,
+        defaultProvider,
+        lastRequestTimestamp: lastRow.ts,
+        requestCount: countRow.n,
+    }, null, 2));
+}
+async function handleListRequests(args) {
+    const limit = typeof args?.limit === 'number' && args.limit > 0 ? Math.min(args.limit, 200) : 20;
+    const before = typeof args?.before === 'string' ? args.before : undefined;
+    const { requests, hasMore } = await listRequestDB(limit, before);
+    const slim = requests.map(r => ({
+        requestId: r.requestId,
+        timestamp: r.timestamp,
+        path: r.path,
+        statusCode: r.statusCode,
+        responseTime: r.responseTime,
+        model: r.model,
+        usage: r.usage,
+    }));
+    return ok(JSON.stringify({ requests: slim, hasMore }, null, 2));
+}
+async function handleGetTokenStats(args) {
+    const hours = typeof args?.since_hours === 'number' && args.since_hours > 0 ? args.since_hours : 24;
+    const db = getDatabase();
+    const where = "timestamp >= datetime('now', ?)";
+    const param = `-${hours} hours`;
+    const totals = db.prepare(`
+    SELECT
+      COUNT(*) AS count,
+      COALESCE(SUM(input_tokens), 0) AS input_tokens,
+      COALESCE(SUM(output_tokens), 0) AS output_tokens,
+      COALESCE(SUM(cache_read_tokens), 0) AS cache_read_tokens,
+      COALESCE(SUM(cache_creation_tokens), 0) AS cache_creation_tokens
+    FROM requests
+    WHERE ${where}
+  `).get(param);
+    const byModel = db.prepare(`
+    SELECT COALESCE(model, '(null)') AS model, COUNT(*) AS count
+    FROM requests
+    WHERE ${where}
+    GROUP BY model
+    ORDER BY count DESC
+  `).all(param);
+    return ok(JSON.stringify({
+        sinceHours: hours,
+        ...totals,
+        byModel,
+    }, null, 2));
+}
+async function handleGetLogs(args, ctx) {
+    const lines = typeof args?.lines === 'number' && args.lines > 0 ? Math.min(args.lines, 500) : 50;
+    const file = ctx.config.logging?.file;
+    if (!file)
+        return ok('(no logging.file configured)');
+    if (!fs.existsSync(file))
+        return ok(`(log file not found: ${file})`);
+    const content = fs.readFileSync(file, 'utf-8');
+    const all = content.split('\n');
+    const tail = all.slice(-lines).join('\n');
+    return ok(tail);
+}
+// --- Tool registry: list metadata + dispatch ---
+export const MCP_TOOLS = [
+    {
+        name: 'get_status',
+        description: 'Get cc-monitoring status: server/proxy config, default provider, latest request timestamp, total request count.',
+        inputSchema: {
+            type: 'object',
+            properties: {},
+            additionalProperties: false,
+        },
+    },
+    {
+        name: 'list_requests',
+        description: 'List recent proxy requests (summary view).',
+        inputSchema: {
+            type: 'object',
+            properties: {
+                limit: { type: 'number', description: 'Max requests to return (default 20, max 200).' },
+                before: { type: 'string', description: 'ISO timestamp cursor: return requests older than this.' },
+            },
+            additionalProperties: false,
+        },
+    },
+    {
+        name: 'get_token_stats',
+        description: 'Aggregate token usage for requests in the last N hours (sums + per-model counts).',
+        inputSchema: {
+            type: 'object',
+            properties: {
+                since_hours: { type: 'number', description: 'Lookback window in hours (default 24).' },
+            },
+            additionalProperties: false,
+        },
+    },
+    {
+        name: 'get_logs',
+        description: 'Return the last N lines of the cc-monitoring log file.',
+        inputSchema: {
+            type: 'object',
+            properties: {
+                lines: { type: 'number', description: 'Number of trailing lines (default 50, max 500).' },
+            },
+            additionalProperties: false,
+        },
+    },
+    {
+        name: 'list_topics',
+        description: 'List all knowledge topics with one-line summary. Call this first to discover what topics exist.',
+        inputSchema: { type: 'object', properties: {}, additionalProperties: false },
+    },
+    {
+        name: 'get_topic',
+        description: 'Get a topic metadata and its chunk list (descriptions only, no body). Call this to see which chunks exist in a topic before reading one.',
+        inputSchema: {
+            type: 'object',
+            properties: { topic: { type: 'string', description: 'Topic slug, e.g. react-hooks' } },
+            required: ['topic'],
+            additionalProperties: false,
+        },
+    },
+    {
+        name: 'search_chunks',
+        description: 'Keyword search across all chunks. Returns matching chunk list (no body). Use when topic is unknown.',
+        inputSchema: {
+            type: 'object',
+            properties: { query: { type: 'string', description: 'Search keyword' } },
+            required: ['query'],
+            additionalProperties: false,
+        },
+    },
+    {
+        name: 'read_chunk',
+        description: 'Read a single knowledge chunk body by topic and chunk id. ONLY one chunk per call. To discover chunk ids, call get_topic first.',
+        inputSchema: {
+            type: 'object',
+            properties: {
+                topic: { type: 'string', description: 'Topic slug' },
+                chunk_id: { type: 'string', description: 'Chunk id, e.g. chunk-001' },
+            },
+            required: ['topic', 'chunk_id'],
+            additionalProperties: false,
+        },
+    },
+];
+const HANDLERS = {
+    get_status: handleGetStatus,
+    list_requests: handleListRequests,
+    get_token_stats: handleGetTokenStats,
+    get_logs: handleGetLogs,
+    list_topics: async () => ok(JSON.stringify(await knowledgeStore.listTopics(), null, 2)),
+    get_topic: async (args) => {
+        try {
+            return ok(JSON.stringify(await knowledgeStore.getTopic(args.topic), null, 2));
+        }
+        catch (e) {
+            const msg = e instanceof Error ? e.message : String(e);
+            if (msg.includes('ENOENT')) {
+                return ok(JSON.stringify({ error: `topic "${args.topic}" not found. Call list_topics to see available topics.` }, null, 2));
+            }
+            return err(e);
+        }
+    },
+    search_chunks: async (args) => ok(JSON.stringify(await knowledgeStore.searchChunks(args.query ?? ''), null, 2)),
+    read_chunk: async (args) => {
+        try {
+            const body = await knowledgeStore.readChunk(args.topic, args.chunk_id);
+            return ok(body);
+        }
+        catch (e) {
+            return err(e);
+        }
+    },
+};
+export function registerMcpTools(server, ctx) {
+    server.setRequestHandler(CallToolRequestSchema, async (req) => {
+        const name = req.params.name;
+        const handler = HANDLERS[name];
+        try {
+            if (!handler)
+                return { content: [{ type: 'text', text: `unknown tool: ${name}` }] };
+            return await handler(req.params.arguments ?? {}, ctx);
+        }
+        catch (e) {
+            return err(e);
+        }
+    });
+}

package/dist/server/docs-route.js

@@ -1,9 +1,8 @@
 import multipart from '@fastify/multipart';
-import fs from 'fs';
 import path from 'path';
 import { registry } from '../parsers/index.js';
-import { dataDir } from '../utils/paths.js';
-const docsDir = path.join(dataDir, 'docs');
+import { loadConfig } from '../config/index.js';
+import { createPiSession } from '../utils/pi-runtime.js';
 export async function setupDocsRoutes(fastify) {
     await fastify.register(multipart, { limits: { fileSize: 10 * 1024 * 1024 } });
     fastify.post('/ui/api/docs/upload', async (request, reply) => {
@@ -41,23 +40,53 @@ export async function setupDocsRoutes(fastify) {
                 error: err.message || 'File parsing failed'
             });
         }
-        // 确保 docs 目录存在
-        if (!fs.existsSync(docsDir)) {
-            fs.mkdirSync(docsDir, { recursive: true });
+        // 不直接落盘根目录，而是交给 pi learn agent 切块入库
+        let config;
+        try {
+            config = await loadConfig();
+        }
+        catch (err) {
+            request.log.error({ err }, 'Failed to load config');
+            return reply.code(500).send({
+                success: false,
+                error: err?.message || 'Failed to load config'
+            });
+        }
+        const port = config.server.port;
+        const prompt = `以下是上传文档《${filename}》的完整内容。请按知识库规范处理：
+1. 调 list_topics 查重，决定新建主题还是复用现有
+2. 把文档内容按结构切成 300-500 字 chunks，写入 <主题>/chunks/chunk-NNN.md
+3. 更新该主题 INDEX.md（片段清单 + 元数据）
+4. 若新建主题，更新根 INDEX.md 的主题行
+
+文档内容：
+${mdContent}`;
+        let session;
+        try {
+            session = await createPiSession({ port, mode: 'learn' });
+        }
+        catch (err) {
+            request.log.error({ err }, 'Failed to create pi learn session');
+            return reply.code(500).send({
+                success: false,
+                error: err?.message || 'Failed to initialize pi runtime'
+            });
         }
-        // 写入 md 文件
-        const mdFilename = filename.slice(0, dotIndex) + '.md';
-        const mdPath = path.join(docsDir, mdFilename);
         try {
-            fs.writeFileSync(mdPath, mdContent, 'utf-8');
+            await Promise.race([
+                session.prompt(prompt),
+                new Promise((_, rej) => setTimeout(() => rej(new Error('Execution timeout (5 minutes)')), 5 * 60 * 1000)),
+            ]);
+            await session.waitForIdle();
         }
         catch (err) {
-            request.log.error({ err, mdPath }, 'Failed to write md file');
+            await session.abort().catch(() => { });
+            request.log.error({ err, filename }, 'pi learn distill failed');
             return reply.code(500).send({
                 success: false,
-                error: 'Failed to save file'
+                error: err?.message || 'Failed to distill document into knowledge base'
             });
         }
-        return { success: true, path: `docs/${mdFilename}` };
+        return { success: true, path: `docs/` };
     });
 }

package/dist/server/index.js

@@ -2,7 +2,6 @@ import Fastify from 'fastify';
 import { loadConfig } from '../config/index.js';
 import { createLogger } from '../logger/index.js';
 import { configureDbPath } from '../utils/db.js';
-import { getCurrentModuleUrl } from '../utils/esm-dirname.js';
 import { setupProxy } from './proxy.js';
 import { setupUiRoutes } from './ui-route.js';
 import { setupChatRoute } from './chat-route.js';
@@ -39,7 +38,9 @@ export async function startServer() {
 // PM2 compatibility: PM2 fork mode sets process.argv[1] to ProcessContainerFork.js wrapper,
 // not the user script. Check PM2-specific env vars instead.
 const _isPm2 = process.env.pm_id !== undefined || process.env.NODE_APP_INSTANCE !== undefined;
-const _currentModuleUrl = getCurrentModuleUrl();
+// 直接用本模块的 import.meta.url；getCurrentModuleUrl() 返回的是 util 模块自身的 URL，
+// 不能用来判断"本文件是否被直接执行"（否则 guard 永远 false，dev 模式从不启动 server）
+const _currentModuleUrl = import.meta.url;
 // For PM2: always start if PM2 env detected and this module is the configured script
 // For direct execution: check if this module is the main module
 import path from 'path';

package/dist/server/ui-route.js

@@ -1,14 +1,13 @@
 import staticPlugin from '@fastify/static';
 import path from 'path';
 import fs from 'fs';
-import { spawn } from 'child_process';
 import { WebSocketServer, WebSocket } from 'ws';
-import { randomUUID } from 'node:crypto';
 import { loadConfig, saveConfig } from '../config/index.js';
 import { getProcessStatus, startProcess, stopProcess, reloadProcess } from '../utils/pm2.js';
 import { getCallerDirname } from '../utils/esm-dirname.js';
 import { getConfigPath, getDisplayPath } from '../utils/paths.js';
-import { loadRequestDB, listRequestDB, sessionIdExists } from '../utils/request-writer.js';
+import { loadRequestDB, listRequestDB } from '../utils/request-writer.js';
+import { createPiSession } from '../utils/pi-runtime.js';
 import { statusHistory } from '../utils/status-history.js';
 import { addDataClient, removeDataClient, broadcastToDataClients, getDataClientCount } from './data-push.js';
 import { getVersionInfo } from '../utils/version.js';
@@ -222,7 +221,9 @@ export async function setupUiRoutes(fastify) {
             });
         }
     });
-    // Exec endpoint: run claude -p <prompt>
+    // pi AgentSession 进程内注册表：key = sessionId（pi 自己生成）
+    const piSessions = new Map();
+    // Exec endpoint: 通过嵌入的 pi SDK 跑 prompt，保持原 SSE/JSON 契约
     fastify.post('/ui/api/exec', async (request, reply) => {
         if (!checkAuth(request, config)) {
             return reply.code(401).send({
@@ -230,140 +231,197 @@ export async function setupUiRoutes(fastify) {
                 error: 'Unauthorized: missing or invalid Authorization header'
             });
         }
-        const { tool, prompt, stream = false, sessionId } = request.body;
-        if (tool !== 'claude') {
-            return reply.code(400).send({
-                success: false,
-                error: 'Unsupported tool: only "claude" is supported'
-            });
-        }
+        const { prompt, stream = false, sessionId, mode } = request.body;
+        // mode 只认 'learn'，其余一律按 chat 处理（默认只读问答）
+        const resolvedMode = mode === 'learn' ? 'learn' : 'chat';
         if (!prompt || typeof prompt !== 'string' || prompt.trim().length === 0) {
             return reply.code(400).send({
                 success: false,
                 error: 'prompt is required and must be a non-empty string'
             });
         }
+        const port = config.server.port;
+        // piSessions 按 `${mode}:${piSessionId}` 隔离：chat / learn 各自独立 session。
+        // 前端回传的 sessionId 不带 mode，先按 resolvedMode 精确查，找不到再按后缀兜底。
+        const sessionKey = (m, id) => `${m}:${id}`;
+        let existing;
+        if (sessionId) {
+            existing = piSessions.get(sessionKey(resolvedMode, sessionId));
+            if (!existing) {
+                // 兜底：mode 不一致或旧数据，按 `:${sessionId}` 后缀反查
+                for (const [key, s] of piSessions) {
+                    if (key.endsWith(`:${sessionId}`)) {
+                        existing = s;
+                        break;
+                    }
+                }
+            }
+        }
+        const isNewSession = !existing;
+        let session;
+        if (isNewSession) {
+            try {
+                session = await createPiSession({ port, mode: resolvedMode });
+            }
+            catch (err) {
+                request.log.error({ err }, 'Failed to create pi session');
+                return reply.code(500).send({
+                    success: false,
+                    error: err?.message || 'Failed to initialize pi runtime'
+                });
+            }
+            piSessions.set(sessionKey(resolvedMode, session.sessionId), session);
+        }
+        else {
+            // isNewSession 为 false 时 existing 必非 undefined（TS 窄化在循环赋值后丢失，这里显式断言）
+            session = existing;
+        }
+        const resolvedSessionId = session.sessionId;
+        // 前端透传 thinkingLevel（默认 off）；若 session 支持则设置
+        const level = request.body.thinkingLevel ?? 'off';
+        if (typeof session.supportsThinking === 'function' && session.supportsThinking()) {
+            session.setThinkingLevel(level);
+        }
         if (stream) {
-            // Streaming mode: use SSE
-            const newSessionId = sessionId ?? randomUUID();
+            // Streaming mode: SSE（与前端 chat.ts 解析契约严格对齐）
             reply.hijack();
             const res = reply.raw;
             res.writeHead(200, {
                 'Content-Type': 'text/event-stream',
                 'Cache-Control': 'no-cache',
-                'Connection': 'keep-alive'
+                'Connection': 'keep-alive',
+                // pi 真实 sessionId 通过 response header 下发；新建/复用都带，前端后续回传
+                'x-claude-code-session-id': resolvedSessionId
             });
             const sendSSE = (event, data) => {
                 if (res.destroyed)
                     return;
                 res.write(`event: ${event}\ndata: ${data}\n\n`);
             };
-            // 新对话时发送 session event
-            if (!sessionId) {
-                sendSSE('session', newSessionId);
-            }
-            // 判断：如果 sessionId 存在于数据库，用 -r；否则用 --session-id
-            const spawnArgs = sessionId
-                ? (sessionIdExists(sessionId)
-                    ? ['-r', sessionId, '-p', prompt, '--allowedTools', 'Bash,Write,Edit,Read']
-                    : ['--session-id', sessionId, '-p', prompt, '--allowedTools', 'Bash,Write,Edit,Read'])
-                : ['-p', prompt, '--allowedTools', 'Bash,Write,Edit,Read'];
-            const child = spawn('claude', spawnArgs, {
-                env: { ...process.env }
-            });
-            const TIMEOUT = 5 * 60 * 1000; // 5 minutes
-            const timeoutId = setTimeout(() => {
-                child.kill('SIGTERM');
-                if (!res.destroyed) {
-                    sendSSE('error', JSON.stringify({ message: 'Execution timeout (5 minutes)' }));
-                    sendSSE('done', JSON.stringify({ code: 1 }));
-                    res.end();
-                }
-            }, TIMEOUT);
-            child.stdout?.on('data', (chunk) => {
-                sendSSE('data', JSON.stringify({ type: 'stdout', content: chunk.toString() }));
-            });
-            child.stderr?.on('data', (chunk) => {
-                sendSSE('data', JSON.stringify({ type: 'stderr', content: chunk.toString() }));
-            });
-            child.on('error', (err) => {
+            let finished = false;
+            const finish = (code) => {
+                if (finished)
+                    return;
+                finished = true;
                 clearTimeout(timeoutId);
                 if (!res.destroyed) {
-                    sendSSE('error', JSON.stringify({ message: err.message }));
-                    sendSSE('done', JSON.stringify({ code: 1 }));
+                    sendSSE('done', JSON.stringify({ code }));
                     res.end();
                 }
+            };
+            const unsub = session.subscribe((ev) => {
+                try {
+                    if (ev.type === 'message_update') {
+                        // assistantMessageEvent 是底层 pi-ai 流事件；按 delta 类型结构化转发
+                        const ae = ev.assistantMessageEvent;
+                        if (ae?.type === 'text_delta' && typeof ae.delta === 'string') {
+                            sendSSE('data', JSON.stringify({ type: 'text_delta', content: ae.delta }));
+                        }
+                        else if (ae?.type === 'thinking_delta' && typeof ae.delta === 'string') {
+                            sendSSE('data', JSON.stringify({ type: 'thinking_delta', content: ae.delta }));
+                        }
+                        // thinking_start / text_start / thinking_end / text_end 不发：前端按 delta 累积
+                    }
+                    else if (ev.type === 'tool_execution_start') {
+                        const e = ev;
+                        sendSSE('data', JSON.stringify({
+                            type: 'tool_start',
+                            toolCallId: e.toolCallId,
+                            toolName: e.toolName,
+                            args: e.args,
+                        }));
+                    }
+                    else if (ev.type === 'tool_execution_end') {
+                        const e = ev;
+                        const resultText = Array.isArray(e.result?.content)
+                            ? e.result.content.map((c) => (c?.type === 'text' ? c.text : '')).join('\n')
+                            : '';
+                        sendSSE('data', JSON.stringify({
+                            type: 'tool_end',
+                            toolCallId: e.toolCallId,
+                            toolName: e.toolName,
+                            isError: !!e.isError,
+                            durationMs: e.durationMs,
+                            result: resultText,
+                        }));
+                    }
+                    else if (ev.type === 'agent_end') {
+                        finish(0);
+                    }
+                }
+                catch (e) {
+                    request.log.error({ err: e }, 'pi event mapping failed');
+                }
             });
-            child.on('close', (code) => {
-                clearTimeout(timeoutId);
+            const TIMEOUT = 5 * 60 * 1000;
+            const timeoutId = setTimeout(() => {
                 if (!res.destroyed) {
-                    sendSSE('done', JSON.stringify({ code: code ?? 0 }));
-                    res.end();
+                    sendSSE('error', JSON.stringify({ message: 'Execution timeout (5 minutes)' }));
                 }
-            });
-            // Handle client disconnect
+                session.abort().catch(() => { });
+                finish(1);
+            }, TIMEOUT);
             res.on('close', () => {
                 clearTimeout(timeoutId);
-                if (!child.killed) {
-                    child.kill('SIGTERM');
-                }
+                unsub();
+                session.abort().catch(() => { });
             });
+            try {
+                await session.prompt(prompt);
+            }
+            catch (err) {
+                unsub();
+                if (!res.destroyed) {
+                    sendSSE('error', JSON.stringify({ message: err?.message || 'Prompt failed' }));
+                }
+                finish(1);
+                return;
+            }
+            // prompt 已返回；agent_end 会触发 finish。若未触发（异常路径），waitForIdle 兜底
+            session.waitForIdle().catch(() => { });
         }
         else {
-            // Non-streaming mode: buffer all output
-            const newSessionId = sessionId ?? randomUUID();
-            // 判断：如果 sessionId 存在于数据库，用 -r；否则用 --session-id
-            const spawnArgs = sessionId
-                ? (sessionIdExists(sessionId)
-                    ? ['-r', sessionId, '-p', prompt, '--allowedTools', 'Bash,Write,Edit,Read']
-                    : ['--session-id', sessionId, '-p', prompt, '--allowedTools', 'Bash,Write,Edit,Read'])
-                : ['-p', prompt, '--allowedTools', 'Bash,Write,Edit,Read'];
-            const child = spawn('claude', spawnArgs, {
-                env: { ...process.env }
-            });
-            const TIMEOUT = 5 * 60 * 1000; // 5 minutes
-            const timeoutId = setTimeout(() => {
-                child.kill('SIGTERM');
-            }, TIMEOUT);
+            // Non-streaming mode: buffer 文本后返回 ExecResponse
             let stdout = '';
             let stderr = '';
-            child.stdout?.on('data', (chunk) => {
-                stdout += chunk.toString();
-            });
-            child.stderr?.on('data', (chunk) => {
-                stderr += chunk.toString();
-            });
-            const result = await new Promise((resolve) => {
-                child.on('error', (err) => {
-                    clearTimeout(timeoutId);
-                    resolve({
-                        success: false,
-                        error: err.message
-                    });
-                });
-                child.on('close', (code) => {
-                    clearTimeout(timeoutId);
-                    const output = stdout + stderr;
-                    if (code === 0) {
-                        resolve({
-                            success: true,
-                            output,
-                            sessionId: newSessionId
-                        });
-                    }
-                    else {
-                        resolve({
-                            success: false,
-                            error: output || `Process exited with code ${code}`,
-                            sessionId: newSessionId
-                        });
+            const unsub = session.subscribe((ev) => {
+                if (ev.type === 'message_update') {
+                    const ae = ev.assistantMessageEvent;
+                    if (ae?.type === 'text_delta' && typeof ae.delta === 'string') {
+                        stdout += ae.delta;
                     }
-                });
+                }
+                else if (ev.type === 'tool_execution_start' ||
+                    ev.type === 'tool_execution_update' ||
+                    ev.type === 'tool_execution_end') {
+                    const toolName = ev.toolName;
+                    if (toolName)
+                        stderr += `[tool:${toolName}]\n`;
+                }
             });
-            if (!result.success) {
-                return reply.code(500).send(result);
+            try {
+                await Promise.race([
+                    session.prompt(prompt),
+                    new Promise((_, rej) => setTimeout(() => rej(new Error('Execution timeout (5 minutes)')), 5 * 60 * 1000)),
+                ]);
+                await session.waitForIdle();
+                unsub();
+                return {
+                    success: true,
+                    output: stdout + stderr,
+                    sessionId: resolvedSessionId,
+                };
+            }
+            catch (err) {
+                unsub();
+                await session.abort().catch(() => { });
+                return reply.code(500).send({
+                    success: false,
+                    error: err?.message || 'Prompt failed',
+                    output: stdout + stderr,
+                    sessionId: resolvedSessionId,
+                });
             }
-            return result;
         }
     });
     // Handle /ui root path (redirect or serve index.html)

package/dist/ui/assets/{1c-DPsrzNHI.js → 1c-CK-kMhtR.js}

@@ -1 +1 @@
-import{aD as f}from"./index-DSGZBcd1.js";import{r as s}from"./index-Dlp8uGHu.js";function i(t,c){for(var o=0;o<c.length;o++){const e=c[o];if(typeof e!="string"&&!Array.isArray(e)){for(const r in e)if(r!=="default"&&!(r in t)){const a=Object.getOwnPropertyDescriptor(e,r);a&&Object.defineProperty(t,r,a.get?a:{enumerable:!0,get:()=>e[r]})}}}return Object.freeze(Object.defineProperty(t,Symbol.toStringTag,{value:"Module"}))}var n=s();const p=f(n),_=i({__proto__:null,default:p},[n]);export{_};
+import{aD as f}from"./index-Dsue7VlR.js";import{r as s}from"./index-DfWIGcNl.js";function i(t,c){for(var o=0;o<c.length;o++){const e=c[o];if(typeof e!="string"&&!Array.isArray(e)){for(const r in e)if(r!=="default"&&!(r in t)){const a=Object.getOwnPropertyDescriptor(e,r);a&&Object.defineProperty(t,r,a.get?a:{enumerable:!0,get:()=>e[r]})}}}return Object.freeze(Object.defineProperty(t,Symbol.toStringTag,{value:"Module"}))}var n=s();const p=f(n),_=i({__proto__:null,default:p},[n]);export{_};