evolcore 0.0.1 → 0.0.3

package/dist/stats/query.js

@@ -0,0 +1,763 @@
+/**
+ * query.ts — 聚合查询（只读，供 CLI 和 ecweb 共用）。
+ * 支持跨年查询：自动附加归档库。
+ */
+import { openReadonlyDb, getDbPath, listArchivePaths } from './db.js';
+import { calcCost } from './billing.js'; // 保留用于尚未迁移的函数
+const GRAN_FMT = {
+    hour: '%Y-%m-%d %H:00',
+    day: '%Y-%m-%d',
+    week: '%Y-W%W',
+    month: '%Y-%m',
+};
+// 非时间维度：按字段分组
+const GRAN_GROUP_COL = {
+    model: 'model',
+    peer: 'peer_key',
+    agent: 'agent_aid',
+};
+function _buildWhere(f) {
+    const conds = [];
+    const params = [];
+    if (f.from_ts) {
+        conds.push('ts >= ?');
+        params.push(f.from_ts);
+    }
+    if (f.to_ts) {
+        conds.push('ts < ?');
+        params.push(f.to_ts);
+    }
+    if (f.agent_aid) {
+        conds.push('agent_aid = ?');
+        params.push(f.agent_aid);
+    }
+    if (f.peer_key) {
+        conds.push('peer_key = ?');
+        params.push(f.peer_key);
+    }
+    if (f.model) {
+        conds.push('model = ?');
+        params.push(f.model);
+    }
+    if (f.session_id) {
+        conds.push('session_id = ?');
+        params.push(f.session_id);
+    }
+    if (f.billing_fn) {
+        conds.push('billing_fn = ?');
+        params.push(f.billing_fn);
+    }
+    return { clause: conds.length ? 'WHERE ' + conds.join(' AND ') : '', params };
+}
+/** 查询哪些归档库需要参与（给定时间范围）。 */
+function _relevantDbs(evolcoreHome, f) {
+    const dbPaths = [getDbPath(evolcoreHome)];
+    if (!f.from_ts)
+        return dbPaths; // 无时间下界，只查主库（最近数据）
+    const fromYear = new Date(f.from_ts).getUTCFullYear();
+    const toYear = f.to_ts ? new Date(f.to_ts).getUTCFullYear() : new Date().getUTCFullYear();
+    for (const { year, path: p } of listArchivePaths(evolcoreHome)) {
+        if (year >= fromYear && year <= toYear)
+            dbPaths.push(p);
+    }
+    return dbPaths;
+}
+function _sumRows(rows) {
+    if (!rows.length)
+        return null;
+    const sum = { ...rows[0] };
+    for (let i = 1; i < rows.length; i++) {
+        for (const k of Object.keys(rows[i])) {
+            if (k === 'period')
+                continue;
+            sum[k] = (sum[k] ?? 0) + (rows[i][k] ?? 0);
+        }
+    }
+    return sum;
+}
+/**
+ * 聚合统计（按粒度分组）。
+ * hour 粒度仍扫明细 usage_events（只查近期，本就快，且明细才有小时粒度）。
+ * 其余粒度（day/week/month/model/peer/agent）改读预聚合表 usage_daily：
+ *   工作量从"全表明细"降到"天×维度"小表。rollup 永留主库，无需跨归档库 union。
+ * 成本仍按 (model, billing_fn) 分组现算 calcCost，口径不变。
+ */
+export function queryAggregated(evolcoreHome, granularity, filter) {
+    if (granularity === 'hour')
+        return _queryAggregatedEvents(evolcoreHome, granularity, filter);
+    return _queryAggregatedDaily(evolcoreHome, granularity, filter);
+}
+// rollup 时间粒度的 period 表达式（day 列是 'YYYY-MM-DD' 字符串，strftime 可直接解析）。
+const DAILY_PERIOD_EXPR = {
+    day: 'day',
+    week: `strftime('%Y-W%W', day)`,
+    month: `substr(day,1,7)`,
+};
+/** 针对 usage_daily 构造 WHERE：时间过滤转成对 day 列的字符串比较，维度过滤直接相等。 */
+function _buildDailyWhere(f) {
+    const conds = [];
+    const params = [];
+    // from_ts/to_ts（ms，含 to_ts 排他）转成 localtime 日期串，与 day 列同口径比较。
+    if (f.from_ts) {
+        conds.push(`day >= strftime('%Y-%m-%d', ?/1000, 'unixepoch', 'localtime')`);
+        params.push(f.from_ts);
+    }
+    if (f.to_ts) {
+        conds.push(`day <  strftime('%Y-%m-%d', ?/1000, 'unixepoch', 'localtime')`);
+        params.push(f.to_ts);
+    }
+    if (f.agent_aid) {
+        conds.push('agent_aid = ?');
+        params.push(f.agent_aid);
+    }
+    if (f.peer_key) {
+        conds.push('peer_key = ?');
+        params.push(f.peer_key);
+    }
+    if (f.model) {
+        conds.push('model = ?');
+        params.push(f.model);
+    }
+    if (f.session_id) {
+        conds.push('session_id = ?');
+        params.push(f.session_id);
+    }
+    if (f.billing_fn) {
+        conds.push('billing_fn = ?');
+        params.push(f.billing_fn);
+    }
+    return { clause: conds.length ? 'WHERE ' + conds.join(' AND ') : '', params };
+}
+/** 走预聚合表 usage_daily 的聚合（day/week/month/model/peer/agent）。 */
+function _queryAggregatedDaily(evolcoreHome, granularity, filter) {
+    const { clause, params } = _buildDailyWhere(filter);
+    const groupCol = GRAN_GROUP_COL[granularity];
+    const periodExpr = groupCol || DAILY_PERIOD_EXPR[granularity] || 'day';
+    const db = openReadonlyDb(getDbPath(evolcoreHome));
+    if (!db)
+        return [];
+    try {
+        const sql = `
+      SELECT
+        ${periodExpr} AS period,
+        SUM(input_tokens)          AS input_tokens,
+        SUM(output_tokens)         AS output_tokens,
+        SUM(cache_creation_tokens) AS cache_creation_tokens,
+        SUM(cache_read_tokens)     AS cache_read_tokens,
+        SUM(cache_hit_tokens)      AS cache_hit_tokens,
+        SUM(cache_miss_tokens)     AS cache_miss_tokens,
+        SUM(image_tokens)          AS image_tokens,
+        SUM(total_context_tokens)  AS total_context_tokens,
+        SUM(turns)                 AS turns,
+        SUM(calls)                 AS call_count,
+        COALESCE(SUM(cost_official_usd), 0) AS cost_official_usd,
+        COALESCE(SUM(cost_official_cny), 0) AS cost_official_cny,
+        COALESCE(SUM(cost_gateway_usd), 0)  AS cost_gateway_usd,
+        COALESCE(SUM(cost_gateway_cny), 0)  AS cost_gateway_cny
+      FROM usage_daily ${clause}
+      GROUP BY ${periodExpr}
+    `;
+        const rows = db.prepare(sql).all(...params);
+        if (groupCol) {
+            rows.sort((a, b) => ((b.input_tokens ?? 0) + (b.output_tokens ?? 0)) - ((a.input_tokens ?? 0) + (a.output_tokens ?? 0)));
+        }
+        else {
+            rows.sort((a, b) => a.period.localeCompare(b.period));
+        }
+        return rows.map(r => _enrichRow(evolcoreHome, r));
+    }
+    finally {
+        db.close();
+    }
+}
+/**
+ * 走明细 usage_events 的聚合（仅 hour 粒度）。
+ * 跨年时合并多个 DB 的结果，再按 period 聚合。
+ */
+function _queryAggregatedEvents(evolcoreHome, granularity, filter) {
+    const { clause, params } = _buildWhere(filter);
+    const groupCol = GRAN_GROUP_COL[granularity];
+    let sql;
+    if (groupCol) {
+        // 非时间维度：按字段分组
+        sql = `
+      SELECT
+        ${groupCol} AS period,
+        SUM(input_tokens)           AS input_tokens,
+        SUM(output_tokens)          AS output_tokens,
+        SUM(cache_creation_tokens)  AS cache_creation_tokens,
+        SUM(cache_read_tokens)      AS cache_read_tokens,
+        SUM(COALESCE(cache_hit_tokens,0))  AS cache_hit_tokens,
+        SUM(COALESCE(cache_miss_tokens,0)) AS cache_miss_tokens,
+        SUM(COALESCE(image_tokens,0))      AS image_tokens,
+        SUM(COALESCE(total_context_tokens,0)) AS total_context_tokens,
+        SUM(turns)                  AS turns,
+        COUNT(*)                    AS call_count
+      FROM usage_events ${clause}
+      GROUP BY ${groupCol} ORDER BY (input_tokens + output_tokens) DESC
+    `;
+    }
+    else {
+        // 时间维度：strftime 分组
+        const fmt = GRAN_FMT[granularity] || GRAN_FMT.day;
+        sql = `
+      SELECT
+        strftime('${fmt}', ts/1000, 'unixepoch', 'localtime') AS period,
+        SUM(input_tokens)           AS input_tokens,
+        SUM(output_tokens)          AS output_tokens,
+        SUM(cache_creation_tokens)  AS cache_creation_tokens,
+        SUM(cache_read_tokens)      AS cache_read_tokens,
+        SUM(COALESCE(cache_hit_tokens,0))  AS cache_hit_tokens,
+        SUM(COALESCE(cache_miss_tokens,0)) AS cache_miss_tokens,
+        SUM(COALESCE(image_tokens,0))      AS image_tokens,
+        SUM(COALESCE(total_context_tokens,0)) AS total_context_tokens,
+        SUM(turns)                  AS turns,
+        COUNT(*)                    AS call_count,
+        COALESCE(SUM(cost_official_usd), 0) AS cost_official_usd,
+        COALESCE(SUM(cost_official_cny), 0) AS cost_official_cny,
+        COALESCE(SUM(cost_gateway_usd), 0)  AS cost_gateway_usd,
+        COALESCE(SUM(cost_gateway_cny), 0)  AS cost_gateway_cny
+      FROM usage_events ${clause}
+      GROUP BY period ORDER BY period
+    `;
+    }
+    // 收集所有相关 DB 的原始行，再按 period 合并
+    const periodMap = new Map();
+    for (const dbPath of _relevantDbs(evolcoreHome, filter)) {
+        const db = openReadonlyDb(dbPath);
+        if (!db)
+            continue;
+        try {
+            const rows = db.prepare(sql).all(...params);
+            for (const r of rows) {
+                const existing = periodMap.get(r.period);
+                periodMap.set(r.period, existing ? _sumRows([existing, r]) : r);
+            }
+        }
+        finally {
+            db.close();
+        }
+    }
+    const sorted = Array.from(periodMap.entries());
+    if (groupCol) {
+        // 非时间维度按 total tokens 降序
+        sorted.sort((a, b) => ((b[1].input_tokens ?? 0) + (b[1].output_tokens ?? 0)) - ((a[1].input_tokens ?? 0) + (a[1].output_tokens ?? 0)));
+    }
+    else {
+        sorted.sort((a, b) => a[0].localeCompare(b[0]));
+    }
+    return sorted.map(([, r]) => _enrichRow(evolcoreHome, r));
+}
+/** 查今日概览（单行汇总）。 */
+export function queryTodaySummary(evolcoreHome, agentAid) {
+    const now = new Date();
+    const from_ts = Date.UTC(now.getUTCFullYear(), now.getUTCMonth(), now.getUTCDate());
+    const rows = queryAggregated(evolcoreHome, 'day', { from_ts, agent_aid: agentAid });
+    return rows[0] ?? null;
+}
+/** 查单个会话的每轮明细。 */
+export function querySessionTurns(evolcoreHome, sessionId) {
+    const { clause, params } = _buildWhere({ session_id: sessionId });
+    const sql = `SELECT * FROM usage_events ${clause} ORDER BY ts`;
+    const result = [];
+    for (const dbPath of _relevantDbs(evolcoreHome, {})) {
+        const db = openReadonlyDb(dbPath);
+        if (!db)
+            continue;
+        try {
+            const rows = db.prepare(sql).all(...params);
+            for (const r of rows) {
+                result.push({
+                    ...r,
+                    cost_official_usd: r.cost_official_usd ?? 0,
+                    cost_official_cny: r.cost_official_cny ?? 0,
+                    cost_gateway_usd: r.cost_gateway_usd ?? 0,
+                    cost_gateway_cny: r.cost_gateway_cny ?? 0,
+                    usd: r.cost_gateway_usd ?? 0,
+                    cny: r.cost_gateway_cny ?? 0
+                });
+            }
+        }
+        finally {
+            db.close();
+        }
+    }
+    return result.sort((a, b) => a.ts - b.ts);
+}
+/**
+ * 单个 session 的累计汇总，单行。直接 SUM 已落库的 official/gateway 成本列，
+ * 不再逐行 calcCost（避免重复计算 + 全表扫描）。跨归档库累加。
+ */
+export function querySessionSummary(evolcoreHome, sessionId) {
+    const empty = {
+        input_tokens: 0, output_tokens: 0, cache_creation_tokens: 0, cache_read_tokens: 0,
+        total_tokens: 0, calls: 0,
+        cost_official_usd: 0, cost_official_cny: 0, cost_gateway_usd: 0, cost_gateway_cny: 0,
+        usd: 0, cny: 0,
+    };
+    const sql = `
+    SELECT
+      COALESCE(SUM(input_tokens), 0)          AS input_tokens,
+      COALESCE(SUM(output_tokens), 0)         AS output_tokens,
+      COALESCE(SUM(cache_creation_tokens), 0) AS cache_creation_tokens,
+      COALESCE(SUM(cache_read_tokens), 0)     AS cache_read_tokens,
+      COUNT(*)                                AS calls,
+      COALESCE(SUM(cost_official_usd), 0)     AS cost_official_usd,
+      COALESCE(SUM(cost_official_cny), 0)     AS cost_official_cny,
+      COALESCE(SUM(cost_gateway_usd), 0)      AS cost_gateway_usd,
+      COALESCE(SUM(cost_gateway_cny), 0)      AS cost_gateway_cny
+    FROM usage_events WHERE session_id = ?`;
+    const acc = { ...empty };
+    for (const dbPath of _relevantDbs(evolcoreHome, {})) {
+        const db = openReadonlyDb(dbPath);
+        if (!db)
+            continue;
+        try {
+            const r = db.prepare(sql).get(sessionId);
+            if (!r)
+                continue;
+            acc.input_tokens += r.input_tokens ?? 0;
+            acc.output_tokens += r.output_tokens ?? 0;
+            acc.cache_creation_tokens += r.cache_creation_tokens ?? 0;
+            acc.cache_read_tokens += r.cache_read_tokens ?? 0;
+            acc.calls += r.calls ?? 0;
+            acc.cost_official_usd += r.cost_official_usd ?? 0;
+            acc.cost_official_cny += r.cost_official_cny ?? 0;
+            acc.cost_gateway_usd += r.cost_gateway_usd ?? 0;
+            acc.cost_gateway_cny += r.cost_gateway_cny ?? 0;
+        }
+        finally {
+            db.close();
+        }
+    }
+    acc.total_tokens = acc.input_tokens + acc.output_tokens + acc.cache_read_tokens + acc.cache_creation_tokens;
+    acc.usd = acc.cost_gateway_usd;
+    acc.cny = acc.cost_gateway_cny;
+    return acc;
+}
+/** 查会话 context_breakdown 细目（每轮各段 token）。 */
+export function queryContextBreakdown(evolcoreHome, sessionId) {
+    const sql = `SELECT * FROM context_breakdown WHERE session_id = ? ORDER BY ts`;
+    const result = [];
+    for (const dbPath of _relevantDbs(evolcoreHome, {})) {
+        const db = openReadonlyDb(dbPath);
+        if (!db)
+            continue;
+        try {
+            result.push(...db.prepare(sql).all(sessionId));
+        }
+        finally {
+            db.close();
+        }
+    }
+    return result.sort((a, b) => a.ts - b.ts);
+}
+/** Top-N 对端（按 token 总量排序）。读预聚合表 usage_daily。 */
+export function queryTopPeers(evolcoreHome, filter, limit = 10) {
+    const { clause, params } = _buildDailyWhere(filter);
+    const sql = `
+    SELECT peer_key, SUM(input_tokens+output_tokens) AS total_tokens,
+           SUM(calls) AS call_count
+    FROM usage_daily ${clause}
+    GROUP BY peer_key ORDER BY total_tokens DESC LIMIT ${limit}
+  `;
+    const db = openReadonlyDb(getDbPath(evolcoreHome));
+    if (!db)
+        return [];
+    try {
+        return db.prepare(sql).all(...params).map(r => ({ ...r }));
+    }
+    finally {
+        db.close();
+    }
+}
+/** Top-N 模型（按 token 总量排序）。读预聚合表 usage_daily。 */
+export function queryTopModels(evolcoreHome, filter, limit = 10) {
+    const { clause, params } = _buildDailyWhere(filter);
+    const sql = `
+    SELECT model, SUM(input_tokens+output_tokens) AS total_tokens,
+           SUM(calls) AS call_count
+    FROM usage_daily ${clause}
+    GROUP BY model ORDER BY total_tokens DESC LIMIT ${limit}
+  `;
+    const db = openReadonlyDb(getDbPath(evolcoreHome));
+    if (!db)
+        return [];
+    try {
+        return db.prepare(sql).all(...params).map(r => ({ ...r }));
+    }
+    finally {
+        db.close();
+    }
+}
+// ── 私聊/群聊列表 + 汇总 + 对端按天明细（owner 前端用）────────────────────────
+/** 解析 peer_key 末段为裸 peer_id（对端 AID / 群 ID）。
+ *  peer_key = aun#<selfAID>#main#<encodeURIComponent(peer)>，取第 4 段起 decode。 */
+function _parsePeerId(peerKey) {
+    const parts = peerKey.split('#');
+    if (parts.length < 4) {
+        // 兼容两段式 aun#peer：取末段
+        try {
+            return decodeURIComponent(parts[parts.length - 1] ?? '');
+        }
+        catch {
+            return parts[parts.length - 1] ?? '';
+        }
+    }
+    try {
+        return decodeURIComponent(parts.slice(3).join('#'));
+    }
+    catch {
+        return parts.slice(3).join('#');
+    }
+}
+/** 对一组按 (model,billing_fn) 分组的行调用 calcCost 并累加，返回 {usd,cny}。 */
+function _accumCost(evolcoreHome, rows) {
+    let usd = 0, cny = 0;
+    for (const r of rows) {
+        const c = calcCost(evolcoreHome, {
+            model: r.model || 'unknown',
+            billing_fn: r.billing_fn || 'per_token_v1',
+            ts: Date.now(),
+            input_tokens: r.input_tokens ?? 0,
+            output_tokens: r.output_tokens ?? 0,
+            cache_creation_tokens: r.cache_creation_tokens ?? 0,
+            cache_read_tokens: r.cache_read_tokens ?? 0,
+            cache_hit_tokens: r.cache_hit_tokens ?? 0,
+            cache_miss_tokens: r.cache_miss_tokens ?? 0,
+            image_tokens: r.image_tokens ?? 0,
+            total_context_tokens: r.total_context_tokens ?? 0,
+        });
+        usd += c.usd ?? 0;
+        cny += c.cny ?? 0;
+    }
+    return { usd, cny };
+}
+/** 私聊（peer_type='private'）或群聊（'group'）列表，每项带累计汇总。读 usage_daily。 */
+export function queryPeerList(evolcoreHome, opts) {
+    const limit = opts.limit ?? 50;
+    const { clause, params } = _buildDailyWhere({
+        from_ts: opts.from_ts, to_ts: opts.to_ts, agent_aid: opts.agent_aid,
+    });
+    // peer_type 过滤拼到 WHERE 上（_buildDailyWhere 不含该字段）。
+    const peerCond = clause ? `${clause} AND peer_type = ?` : 'WHERE peer_type = ?';
+    const listParams = [...params, opts.peer_type];
+    const db = openReadonlyDb(getDbPath(evolcoreHome));
+    if (!db)
+        return [];
+    try {
+        const listSql = `
+      SELECT peer_key,
+        SUM(input_tokens) AS input_tokens, SUM(output_tokens) AS output_tokens,
+        SUM(cache_creation_tokens) AS cache_creation_tokens,
+        SUM(cache_read_tokens) AS cache_read_tokens,
+        SUM(input_tokens+output_tokens+cache_read_tokens+cache_creation_tokens) AS total_tokens,
+        SUM(calls) AS calls,
+        COUNT(DISTINCT session_id) AS session_count,
+        MIN(day) AS first_day, MAX(day) AS last_day,
+        MAX(peer_type) AS peer_type
+      FROM usage_daily ${peerCond}
+      GROUP BY peer_key ORDER BY total_tokens DESC LIMIT ${limit}
+    `;
+        const rows = db.prepare(listSql).all(...listParams);
+        // 每个 peer 的成本：按 (peer_key, model, billing_fn) 分组算 calcCost 累加。
+        const costSql = `
+      SELECT peer_key, model, COALESCE(billing_fn,'') AS billing_fn,
+        SUM(input_tokens) AS input_tokens, SUM(output_tokens) AS output_tokens,
+        SUM(cache_creation_tokens) AS cache_creation_tokens, SUM(cache_read_tokens) AS cache_read_tokens,
+        SUM(cache_hit_tokens) AS cache_hit_tokens, SUM(cache_miss_tokens) AS cache_miss_tokens,
+        SUM(image_tokens) AS image_tokens, SUM(total_context_tokens) AS total_context_tokens
+      FROM usage_daily ${peerCond}
+      GROUP BY peer_key, model, billing_fn
+    `;
+        const costByPeer = new Map();
+        for (const r of db.prepare(costSql).all(...listParams)) {
+            const arr = costByPeer.get(r.peer_key) ?? [];
+            arr.push(r);
+            costByPeer.set(r.peer_key, arr);
+        }
+        return rows.map(r => {
+            const cost = _accumCost(evolcoreHome, costByPeer.get(r.peer_key) ?? []);
+            return {
+                peer_key: r.peer_key,
+                peer_id: _parsePeerId(r.peer_key),
+                peer_type: r.peer_type ?? opts.peer_type,
+                input_tokens: r.input_tokens ?? 0,
+                output_tokens: r.output_tokens ?? 0,
+                cache_creation_tokens: r.cache_creation_tokens ?? 0,
+                cache_read_tokens: r.cache_read_tokens ?? 0,
+                total_tokens: r.total_tokens ?? 0,
+                calls: r.calls ?? 0,
+                session_count: r.session_count ?? 0,
+                first_day: r.first_day ?? '',
+                last_day: r.last_day ?? '',
+                cost_official_usd: cost.usd,
+                cost_official_cny: cost.cny,
+                cost_gateway_usd: cost.usd,
+                cost_gateway_cny: cost.cny,
+                usd: cost.usd,
+                cny: cost.cny,
+            };
+        });
+    }
+    finally {
+        db.close();
+    }
+}
+/** 指定时间范围（可选对端）的总消耗汇总，单行。读 usage_daily。 */
+export function querySummary(evolcoreHome, opts) {
+    const { clause, params } = _buildDailyWhere({
+        from_ts: opts.from_ts, to_ts: opts.to_ts, agent_aid: opts.agent_aid, peer_key: opts.peer_key,
+    });
+    const empty = {
+        input_tokens: 0, output_tokens: 0, cache_creation_tokens: 0, cache_read_tokens: 0,
+        total_tokens: 0, calls: 0,
+        cost_official_usd: 0, cost_official_cny: 0, cost_gateway_usd: 0, cost_gateway_cny: 0,
+        usd: 0, cny: 0,
+    };
+    const db = openReadonlyDb(getDbPath(evolcoreHome));
+    if (!db)
+        return empty;
+    try {
+        const row = db.prepare(`
+      SELECT
+        SUM(input_tokens) AS input_tokens, SUM(output_tokens) AS output_tokens,
+        SUM(cache_creation_tokens) AS cache_creation_tokens,
+        SUM(cache_read_tokens) AS cache_read_tokens,
+        SUM(calls) AS calls,
+        COALESCE(SUM(cost_official_usd), 0) AS cost_official_usd,
+        COALESCE(SUM(cost_official_cny), 0) AS cost_official_cny,
+        COALESCE(SUM(cost_gateway_usd), 0) AS cost_gateway_usd,
+        COALESCE(SUM(cost_gateway_cny), 0) AS cost_gateway_cny
+      FROM usage_daily ${clause}
+    `).get(...params);
+        if (!row || row.calls == null)
+            return empty;
+        const inTok = row.input_tokens ?? 0;
+        const cacheRead = row.cache_read_tokens ?? 0;
+        return {
+            input_tokens: inTok,
+            output_tokens: row.output_tokens ?? 0,
+            cache_creation_tokens: row.cache_creation_tokens ?? 0,
+            cache_read_tokens: cacheRead,
+            total_tokens: inTok + (row.output_tokens ?? 0) + cacheRead + (row.cache_creation_tokens ?? 0),
+            calls: row.calls ?? 0,
+            cost_official_usd: row.cost_official_usd ?? 0,
+            cost_official_cny: row.cost_official_cny ?? 0,
+            cost_gateway_usd: row.cost_gateway_usd ?? 0,
+            cost_gateway_cny: row.cost_gateway_cny ?? 0,
+            usd: row.cost_gateway_usd ?? 0,
+            cny: row.cost_gateway_cny ?? 0,
+        };
+    }
+    finally {
+        db.close();
+    }
+}
+/** 指定对端（peer_key 或 peer_id），按天返回消耗明细。读 usage_daily。 */
+export function queryPeerDaily(evolcoreHome, opts) {
+    const { clause, params } = _buildDailyWhere({
+        from_ts: opts.from_ts, to_ts: opts.to_ts, agent_aid: opts.agent_aid,
+        peer_key: opts.peer_key,
+    });
+    // peer_id：按末段 LIKE 收窄（与 peer_key 精确匹配二选一）。
+    let where = clause;
+    const qParams = [...params];
+    if (!opts.peer_key && opts.peer_id) {
+        where = where ? `${where} AND peer_key LIKE ?` : 'WHERE peer_key LIKE ?';
+        qParams.push(`aun#%#main#${encodeURIComponent(opts.peer_id)}`);
+    }
+    const db = openReadonlyDb(getDbPath(evolcoreHome));
+    if (!db)
+        return [];
+    try {
+        const sql = `
+      SELECT day AS period,
+        SUM(input_tokens) AS input_tokens, SUM(output_tokens) AS output_tokens,
+        SUM(cache_creation_tokens) AS cache_creation_tokens,
+        SUM(cache_read_tokens) AS cache_read_tokens,
+        SUM(cache_hit_tokens) AS cache_hit_tokens, SUM(cache_miss_tokens) AS cache_miss_tokens,
+        SUM(image_tokens) AS image_tokens, SUM(total_context_tokens) AS total_context_tokens,
+        SUM(turns) AS turns, SUM(calls) AS call_count
+      FROM usage_daily ${where}
+      GROUP BY day ORDER BY day
+    `;
+        const periodMap = new Map();
+        for (const r of db.prepare(sql).all(...qParams))
+            periodMap.set(r.period, r);
+        const costSql = `
+      SELECT day AS period, model, COALESCE(billing_fn,'') AS billing_fn,
+        SUM(input_tokens) AS input_tokens, SUM(output_tokens) AS output_tokens,
+        SUM(cache_creation_tokens) AS cache_creation_tokens, SUM(cache_read_tokens) AS cache_read_tokens,
+        SUM(cache_hit_tokens) AS cache_hit_tokens, SUM(cache_miss_tokens) AS cache_miss_tokens,
+        SUM(image_tokens) AS image_tokens, SUM(total_context_tokens) AS total_context_tokens
+      FROM usage_daily ${where}
+      GROUP BY day, model, billing_fn
+    `;
+        const costMap = new Map();
+        const byPeriod = new Map();
+        for (const r of db.prepare(costSql).all(...qParams)) {
+            const arr = byPeriod.get(r.period) ?? [];
+            arr.push(r);
+            byPeriod.set(r.period, arr);
+        }
+        for (const [period, rows] of byPeriod) {
+            const cost = _accumCost(evolcoreHome, rows);
+            const existing = periodMap.get(period);
+            if (existing) {
+                existing.cost_official_usd = cost.usd;
+                existing.cost_official_cny = cost.cny;
+                existing.cost_gateway_usd = cost.usd;
+                existing.cost_gateway_cny = cost.cny;
+            }
+        }
+        return Array.from(periodMap.entries())
+            .sort((a, b) => a[0].localeCompare(b[0]))
+            .map(([, r]) => _enrichRow(evolcoreHome, r));
+    }
+    finally {
+        db.close();
+    }
+}
+/** 查一个 task 的所有大模型调用明细，按 call_index 排序。 */
+export function queryTaskModelCalls(evolcoreHome, taskId) {
+    const db = openReadonlyDb(getDbPath(evolcoreHome));
+    if (!db)
+        return [];
+    try {
+        return db.prepare(`SELECT * FROM model_calls WHERE task_id = ? ORDER BY call_index`).all(taskId);
+    }
+    finally {
+        db.close();
+    }
+}
+/** 查一个 evolcore session 的所有大模型调用明细，按时间 + call_index 排序。 */
+export function querySessionModelCalls(evolcoreHome, sessionId) {
+    const db = openReadonlyDb(getDbPath(evolcoreHome));
+    if (!db)
+        return [];
+    try {
+        return db.prepare(`SELECT * FROM model_calls WHERE session_id = ? ORDER BY ts, call_index`).all(sessionId);
+    }
+    finally {
+        db.close();
+    }
+}
+function _enrichRow(evolcoreHome, r) {
+    const totalIn = (r.input_tokens ?? 0) + (r.cache_read_tokens ?? 0);
+    const cacheHitRate = totalIn > 0 ? (r.cache_read_tokens ?? 0) / totalIn : 0;
+    return {
+        period: r.period,
+        input_tokens: r.input_tokens ?? 0,
+        output_tokens: r.output_tokens ?? 0,
+        cache_creation_tokens: r.cache_creation_tokens ?? 0,
+        cache_read_tokens: r.cache_read_tokens ?? 0,
+        cache_hit_tokens: r.cache_hit_tokens ?? 0,
+        cache_miss_tokens: r.cache_miss_tokens ?? 0,
+        image_tokens: r.image_tokens ?? 0,
+        total_context_tokens: r.total_context_tokens ?? 0,
+        turns: r.turns ?? 0,
+        call_count: r.call_count ?? 0,
+        cost_official_usd: r.cost_official_usd ?? 0,
+        cost_official_cny: r.cost_official_cny ?? 0,
+        cost_gateway_usd: r.cost_gateway_usd ?? 0,
+        cost_gateway_cny: r.cost_gateway_cny ?? 0,
+        usd: r.cost_gateway_usd ?? 0, // 向后兼容
+        cny: r.cost_gateway_cny ?? 0, // 向后兼容
+        cache_hit_rate: cacheHitRate,
+    };
+}
+/** 流量聚合统计（按时间/agent/peer 分组）。 */
+export function queryMessageAggregated(evolcoreHome, granularity, filter) {
+    const groupCol = GRAN_GROUP_COL[granularity];
+    const { clause, params } = _buildMsgWhere(filter);
+    let sql;
+    if (groupCol) {
+        const col = groupCol === 'model' ? 'agent_aid' : groupCol; // model 无意义，降级 agent
+        sql = `
+      SELECT ${col} AS period,
+        SUM(CASE WHEN direction='in' THEN 1 ELSE 0 END) AS msg_in,
+        SUM(CASE WHEN direction='out' THEN 1 ELSE 0 END) AS msg_out,
+        SUM(CASE WHEN direction='in' THEN bytes ELSE 0 END) AS bytes_in,
+        SUM(CASE WHEN direction='out' THEN bytes ELSE 0 END) AS bytes_out
+      FROM message_events ${clause}
+      GROUP BY ${col} ORDER BY (bytes_in + bytes_out) DESC
+    `;
+    }
+    else {
+        const fmt = GRAN_FMT[granularity] || GRAN_FMT.day;
+        sql = `
+      SELECT strftime('${fmt}', ts/1000, 'unixepoch', 'localtime') AS period,
+        SUM(CASE WHEN direction='in' THEN 1 ELSE 0 END) AS msg_in,
+        SUM(CASE WHEN direction='out' THEN 1 ELSE 0 END) AS msg_out,
+        SUM(CASE WHEN direction='in' THEN bytes ELSE 0 END) AS bytes_in,
+        SUM(CASE WHEN direction='out' THEN bytes ELSE 0 END) AS bytes_out
+      FROM message_events ${clause}
+      GROUP BY period ORDER BY period
+    `;
+    }
+    const periodMap = new Map();
+    for (const dbPath of _relevantDbs(evolcoreHome, filter)) {
+        const db = openReadonlyDb(dbPath);
+        if (!db)
+            continue;
+        try {
+            const rows = db.prepare(sql).all(...params);
+            for (const r of rows) {
+                const existing = periodMap.get(r.period);
+                if (existing) {
+                    existing.msg_in += r.msg_in ?? 0;
+                    existing.msg_out += r.msg_out ?? 0;
+                    existing.bytes_in += r.bytes_in ?? 0;
+                    existing.bytes_out += r.bytes_out ?? 0;
+                }
+                else {
+                    periodMap.set(r.period, {
+                        period: r.period,
+                        msg_in: r.msg_in ?? 0, msg_out: r.msg_out ?? 0,
+                        bytes_in: r.bytes_in ?? 0, bytes_out: r.bytes_out ?? 0,
+                    });
+                }
+            }
+        }
+        finally {
+            db.close();
+        }
+    }
+    const sorted = Array.from(periodMap.values());
+    if (groupCol) {
+        sorted.sort((a, b) => (b.bytes_in + b.bytes_out) - (a.bytes_in + a.bytes_out));
+    }
+    else {
+        sorted.sort((a, b) => a.period.localeCompare(b.period));
+    }
+    return sorted;
+}
+/** 今日流量概览。 */
+export function queryMessageTodaySummary(evolcoreHome, agentAid) {
+    const now = new Date();
+    const from_ts = Date.UTC(now.getUTCFullYear(), now.getUTCMonth(), now.getUTCDate());
+    const rows = queryMessageAggregated(evolcoreHome, 'day', { from_ts, agent_aid: agentAid });
+    return rows[0] ?? null;
+}
+function _buildMsgWhere(f) {
+    const conds = [];
+    const params = [];
+    if (f.from_ts) {
+        conds.push('ts >= ?');
+        params.push(f.from_ts);
+    }
+    if (f.to_ts) {
+        conds.push('ts < ?');
+        params.push(f.to_ts);
+    }
+    if (f.agent_aid) {
+        conds.push('agent_aid = ?');
+        params.push(f.agent_aid);
+    }
+    if (f.peer_key) {
+        conds.push('peer_key = ?');
+        params.push(f.peer_key);
+    }
+    return { clause: conds.length ? 'WHERE ' + conds.join(' AND ') : '', params };
+}