devchain-cli 0.15.0 → 0.16.0

package/dist/server/modules/session-reader/parsers/copilot-jsonl.parser.js

@@ -0,0 +1,432 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.parseCopilotJsonl = parseCopilotJsonl;
+const fs = __importStar(require("node:fs"));
+const readline = __importStar(require("node:readline"));
+const logger_1 = require("../../../common/logging/logger");
+const estimate_content_tokens_1 = require("../adapters/utils/estimate-content-tokens");
+const copilot_model_pricing_1 = require("../readers/copilot-model-pricing");
+const logger = (0, logger_1.createLogger)('CopilotJsonlParser');
+const MAX_LINE_BYTES = 10 * 1024 * 1024;
+function numOr(v) {
+    return typeof v === 'number' && Number.isFinite(v) ? v : undefined;
+}
+const stubPricing = {
+    calculateMessageCost: () => 0,
+    getContextWindowSize: () => 200_000,
+};
+async function parseCopilotJsonl(filePath, options) {
+    const byteOffset = options?.byteOffset ?? 0;
+    const maxMessages = options?.maxMessages;
+    const includeToolCalls = options?.includeToolCalls ?? true;
+    const pricing = options?.pricingService;
+    const messages = [];
+    let messageIndex = 0;
+    let sessionId;
+    let primaryModel = '';
+    const modelsSet = new Set();
+    const authUsageByModel = new Map();
+    let nativeCredits = 0;
+    let seenShutdown = false;
+    let lastContextTokens = 0;
+    let lastContextBreakdown;
+    let liveOutputTokens = 0;
+    let firstTimestamp;
+    let lastTimestamp;
+    let visibleContextTokens = 0;
+    let assistantBuffer = null;
+    let oversizedLineCount = 0;
+    let bytesRead = byteOffset;
+    const stream = fs.createReadStream(filePath, {
+        start: byteOffset,
+        encoding: 'utf8',
+    });
+    const rl = readline.createInterface({
+        input: stream,
+        crlfDelay: Infinity,
+    });
+    function trackTimestamp(ts) {
+        if (!firstTimestamp || ts < firstTimestamp)
+            firstTimestamp = ts;
+        if (!lastTimestamp || ts > lastTimestamp)
+            lastTimestamp = ts;
+    }
+    function flushAssistantBuffer() {
+        if (!assistantBuffer)
+            return;
+        const buf = assistantBuffer;
+        assistantBuffer = null;
+        if (buf.content.length === 0 && buf.toolCalls.length === 0 && buf.toolResults.length === 0) {
+            return;
+        }
+        const msg = {
+            id: `copilot-asst-${messageIndex++}`,
+            parentId: messages.length > 0 ? messages[messages.length - 1].id : null,
+            role: 'assistant',
+            timestamp: buf.timestamp,
+            content: buf.content,
+            model: buf.model || undefined,
+            toolCalls: buf.toolCalls,
+            toolResults: buf.toolResults,
+            usage: { input: 0, output: buf.outputTokens, cacheRead: 0, cacheCreation: 0 },
+            isMeta: false,
+            isSidechain: false,
+            stopReason: 'end_turn',
+        };
+        messages.push(msg);
+        visibleContextTokens += (0, estimate_content_tokens_1.estimateMessageTokens)(msg.content);
+    }
+    function ensureAssistantBuffer(ts, model) {
+        if (!assistantBuffer) {
+            assistantBuffer = {
+                timestamp: ts,
+                model,
+                content: [],
+                toolCalls: [],
+                toolResults: [],
+                outputTokens: 0,
+            };
+        }
+        else if (model) {
+            assistantBuffer.model = model;
+        }
+        return assistantBuffer;
+    }
+    function attachToolResult(block, resultEntry, ts) {
+        if (assistantBuffer) {
+            assistantBuffer.content.push(block);
+            assistantBuffer.toolResults.push(resultEntry);
+            return;
+        }
+        const last = messages.length > 0 ? messages[messages.length - 1] : null;
+        if (last && last.role === 'assistant') {
+            last.content.push(block);
+            last.toolResults.push(resultEntry);
+            return;
+        }
+        const msg = {
+            id: `copilot-toolresult-${messageIndex++}`,
+            parentId: messages.length > 0 ? messages[messages.length - 1].id : null,
+            role: 'user',
+            timestamp: ts,
+            content: [block],
+            toolCalls: [],
+            toolResults: [resultEntry],
+            isMeta: true,
+            isSidechain: false,
+        };
+        messages.push(msg);
+    }
+    try {
+        for await (const line of rl) {
+            const lineBytes = Buffer.byteLength(line, 'utf8') + 1;
+            bytesRead += lineBytes;
+            if (!line.trim())
+                continue;
+            if (lineBytes > MAX_LINE_BYTES) {
+                logger.warn({ filePath, lineBytes, byteOffset: bytesRead - lineBytes, snippet: line.slice(0, 200) }, 'Skipping oversized JSONL line (>10MB)');
+                oversizedLineCount++;
+                continue;
+            }
+            let entry;
+            try {
+                entry = JSON.parse(line);
+            }
+            catch {
+                logger.warn({ filePath }, 'Skipping malformed JSONL line');
+                continue;
+            }
+            if (!entry.type)
+                continue;
+            const data = entry.data ?? {};
+            const ts = entry.timestamp ? new Date(entry.timestamp) : new Date();
+            trackTimestamp(ts);
+            switch (entry.type) {
+                case 'session.start': {
+                    const sid = data.sessionId;
+                    if (typeof sid === 'string')
+                        sessionId = sid;
+                    break;
+                }
+                case 'session.resume': {
+                    break;
+                }
+                case 'session.model_change': {
+                    const newModel = data.newModel;
+                    if (typeof newModel === 'string' && newModel !== 'auto') {
+                        primaryModel = newModel;
+                        modelsSet.add(newModel);
+                    }
+                    break;
+                }
+                case 'session.info': {
+                    break;
+                }
+                case 'session.shutdown': {
+                    flushAssistantBuffer();
+                    seenShutdown = true;
+                    const shutdown = data;
+                    const mm = shutdown.modelMetrics ?? {};
+                    for (const model of Object.keys(mm)) {
+                        modelsSet.add(model);
+                        const u = mm[model]?.usage ?? {};
+                        const prev = authUsageByModel.get(model) ?? {
+                            input: 0,
+                            output: 0,
+                            cacheRead: 0,
+                            cacheWrite: 0,
+                            reasoning: 0,
+                        };
+                        prev.input += u.inputTokens ?? 0;
+                        prev.output += u.outputTokens ?? 0;
+                        prev.cacheRead += u.cacheReadTokens ?? 0;
+                        prev.cacheWrite += u.cacheWriteTokens ?? 0;
+                        prev.reasoning += u.reasoningTokens ?? 0;
+                        authUsageByModel.set(model, prev);
+                        nativeCredits += mm[model]?.requests?.cost ?? 0;
+                    }
+                    if (typeof shutdown.currentModel === 'string' && shutdown.currentModel !== 'auto') {
+                        primaryModel = shutdown.currentModel;
+                        modelsSet.add(shutdown.currentModel);
+                    }
+                    if (typeof shutdown.currentTokens === 'number')
+                        lastContextTokens = shutdown.currentTokens;
+                    const sys = numOr(shutdown.systemTokens);
+                    const conv = numOr(shutdown.conversationTokens);
+                    const tools = numOr(shutdown.toolDefinitionsTokens);
+                    if (sys !== undefined || conv !== undefined || tools !== undefined) {
+                        lastContextBreakdown = {
+                            system: sys ?? 0,
+                            conversation: conv ?? 0,
+                            toolDefinitions: tools ?? 0,
+                        };
+                    }
+                    break;
+                }
+                case 'system.message': {
+                    break;
+                }
+                case 'user.message': {
+                    flushAssistantBuffer();
+                    const content = typeof data.content === 'string' ? data.content : '';
+                    if (!content)
+                        break;
+                    const msg = {
+                        id: `copilot-user-${messageIndex++}`,
+                        parentId: messages.length > 0 ? messages[messages.length - 1].id : null,
+                        role: 'user',
+                        timestamp: ts,
+                        content: [{ type: 'text', text: content }],
+                        toolCalls: [],
+                        toolResults: [],
+                        isMeta: false,
+                        isSidechain: false,
+                    };
+                    messages.push(msg);
+                    visibleContextTokens += (0, estimate_content_tokens_1.estimateMessageTokens)(msg.content);
+                    break;
+                }
+                case 'assistant.turn_start': {
+                    break;
+                }
+                case 'assistant.message': {
+                    const model = typeof data.model === 'string' ? data.model : '';
+                    if (model) {
+                        primaryModel = model;
+                        modelsSet.add(model);
+                    }
+                    const buf = ensureAssistantBuffer(ts, model);
+                    const reasoningText = data.reasoningText;
+                    if (typeof reasoningText === 'string' && reasoningText) {
+                        buf.content.push({ type: 'thinking', thinking: reasoningText });
+                    }
+                    const text = data.content;
+                    if (typeof text === 'string' && text) {
+                        buf.content.push({ type: 'text', text });
+                    }
+                    if (includeToolCalls && Array.isArray(data.toolRequests)) {
+                        for (const tr of data.toolRequests) {
+                            const callId = tr.toolCallId;
+                            const name = tr.name;
+                            if (!callId || !name)
+                                continue;
+                            const input = tr.arguments ?? {};
+                            buf.content.push({ type: 'tool_call', toolCallId: callId, toolName: name, input });
+                            buf.toolCalls.push({
+                                id: callId,
+                                name,
+                                input,
+                                isTask: name === 'Task',
+                            });
+                        }
+                    }
+                    const out = typeof data.outputTokens === 'number' ? data.outputTokens : 0;
+                    buf.outputTokens += out;
+                    liveOutputTokens += out;
+                    break;
+                }
+                case 'assistant.turn_end': {
+                    break;
+                }
+                case 'tool.execution_start': {
+                    break;
+                }
+                case 'tool.execution_complete': {
+                    if (!includeToolCalls)
+                        break;
+                    const callId = typeof data.toolCallId === 'string' ? data.toolCallId : '';
+                    if (!callId)
+                        break;
+                    const result = data.result ?? {};
+                    const rawContent = result.content;
+                    const resultContent = typeof rawContent === 'string' ? rawContent : JSON.stringify(rawContent ?? '');
+                    const isError = data.success === false;
+                    const block = {
+                        type: 'tool_result',
+                        toolCallId: callId,
+                        content: resultContent,
+                        isError,
+                    };
+                    const resultEntry = {
+                        toolCallId: callId,
+                        content: resultContent,
+                        isError,
+                    };
+                    attachToolResult(block, resultEntry, ts);
+                    break;
+                }
+                case 'hook.start':
+                case 'hook.end': {
+                    break;
+                }
+                default:
+                    logger.debug({ type: entry.type }, 'Skipping unknown Copilot event type');
+                    break;
+            }
+            if (maxMessages && messages.length >= maxMessages)
+                break;
+        }
+    }
+    finally {
+        rl.close();
+        stream.destroy();
+    }
+    flushAssistantBuffer();
+    const warnings = [];
+    let inputTokens = 0;
+    let outputTokens = 0;
+    let cacheReadTokens = 0;
+    let cacheCreationTokens = 0;
+    let costUsd = 0;
+    let totalContextTokens = 0;
+    let contextBreakdown;
+    let nativeCost;
+    if (seenShutdown) {
+        for (const [model, u] of authUsageByModel) {
+            inputTokens += u.input;
+            outputTokens += u.output;
+            cacheReadTokens += u.cacheRead;
+            cacheCreationTokens += u.cacheWrite;
+            costUsd += (0, copilot_model_pricing_1.calculateCopilotMessageCost)(model, {
+                inputTokens: u.input,
+                outputTokens: u.output,
+                cacheReadTokens: u.cacheRead,
+                cacheCreationTokens: u.cacheWrite,
+            }, pricing ?? stubPricing);
+        }
+        totalContextTokens = lastContextTokens;
+        contextBreakdown = lastContextBreakdown;
+        nativeCost = nativeCredits;
+    }
+    else {
+        outputTokens = liveOutputTokens;
+        costUsd =
+            liveOutputTokens > 0 && primaryModel
+                ? (0, copilot_model_pricing_1.calculateCopilotMessageCost)(primaryModel, {
+                    inputTokens: 0,
+                    outputTokens: liveOutputTokens,
+                    cacheReadTokens: 0,
+                    cacheCreationTokens: 0,
+                }, pricing ?? stubPricing)
+                : 0;
+        warnings.push('Partial metrics — session.shutdown not yet observed; input/cache/reasoning/cost pending (showing per-turn outputTokens + output-only lower-bound cost).');
+    }
+    const durationMs = firstTimestamp && lastTimestamp ? lastTimestamp.getTime() - firstTimestamp.getTime() : 0;
+    const isOngoing = !seenShutdown;
+    const contextWindowTokens = pricing && primaryModel ? pricing.getContextWindowSize(primaryModel) : 200_000;
+    const modelsUsed = modelsSet.size > 1 ? Array.from(modelsSet) : undefined;
+    const phases = [
+        {
+            phaseNumber: 1,
+            contribution: inputTokens + cacheReadTokens,
+            peakTokens: inputTokens + cacheReadTokens,
+        },
+    ];
+    const totalContextConsumption = phases.reduce((sum, p) => sum + p.contribution, 0);
+    const totalTokens = inputTokens + outputTokens + cacheReadTokens + cacheCreationTokens;
+    const metrics = {
+        inputTokens,
+        outputTokens,
+        cacheReadTokens,
+        cacheCreationTokens,
+        totalTokens,
+        totalContextConsumption,
+        compactionCount: 0,
+        phaseBreakdowns: phases,
+        visibleContextTokens,
+        totalContextTokens,
+        contextWindowTokens,
+        contextBreakdown,
+        costUsd,
+        nativeCost,
+        primaryModel,
+        modelsUsed,
+        durationMs,
+        messageCount: messages.length,
+        isOngoing,
+    };
+    if (oversizedLineCount > 0) {
+        warnings.push(`Skipped ${oversizedLineCount} oversized line${oversizedLineCount > 1 ? 's' : ''} (>10MB each)`);
+    }
+    return {
+        messages,
+        metrics,
+        bytesRead,
+        sessionId,
+        warnings: warnings.length > 0 ? warnings : undefined,
+    };
+}
+//# sourceMappingURL=copilot-jsonl.parser.js.map

package/dist/server/modules/session-reader/parsers/copilot-jsonl.parser.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"copilot-jsonl.parser.js","sourceRoot":"","sources":["../../../../src/modules/session-reader/parsers/copilot-jsonl.parser.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA8HA,8CAwgBC;AAtoBD,4CAA8B;AAC9B,wDAA0C;AAC1C,2DAA8D;AAU9D,uFAAkF;AAClF,4EAA+E;AAE/E,MAAM,MAAM,GAAG,IAAA,qBAAY,EAAC,oBAAoB,CAAC,CAAC;AAGlD,MAAM,cAAc,GAAG,EAAE,GAAG,IAAI,GAAG,IAAI,CAAC;AAoDxC,SAAS,KAAK,CAAC,CAAU;IACvB,OAAO,OAAO,CAAC,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;AACrE,CAAC;AAOD,MAAM,WAAW,GAA4B;IAC3C,oBAAoB,EAAE,GAAG,EAAE,CAAC,CAAC;IAC7B,oBAAoB,EAAE,GAAG,EAAE,CAAC,OAAO;CACpC,CAAC;AA4CK,KAAK,UAAU,iBAAiB,CACrC,QAAgB,EAChB,OAA6B;IAE7B,MAAM,UAAU,GAAG,OAAO,EAAE,UAAU,IAAI,CAAC,CAAC;IAC5C,MAAM,WAAW,GAAG,OAAO,EAAE,WAAW,CAAC;IACzC,MAAM,gBAAgB,GAAG,OAAO,EAAE,gBAAgB,IAAI,IAAI,CAAC;IAC3D,MAAM,OAAO,GAAG,OAAO,EAAE,cAAc,CAAC;IAExC,MAAM,QAAQ,GAAqB,EAAE,CAAC;IACtC,IAAI,YAAY,GAAG,CAAC,CAAC;IAGrB,IAAI,SAA6B,CAAC;IAGlC,IAAI,YAAY,GAAG,EAAE,CAAC;IACtB,MAAM,SAAS,GAAG,IAAI,GAAG,EAAU,CAAC;IAOpC,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAG7B,CAAC;IAEJ,IAAI,aAAa,GAAG,CAAC,CAAC;IACtB,IAAI,YAAY,GAAG,KAAK,CAAC;IAEzB,IAAI,iBAAiB,GAAG,CAAC,CAAC;IAC1B,IAAI,oBAES,CAAC;IAGd,IAAI,gBAAgB,GAAG,CAAC,CAAC;IAGzB,IAAI,cAAgC,CAAC;IACrC,IAAI,aAA+B,CAAC;IAGpC,IAAI,oBAAoB,GAAG,CAAC,CAAC;IAG7B,IAAI,eAAe,GAA2B,IAAI,CAAC;IAGnD,IAAI,kBAAkB,GAAG,CAAC,CAAC;IAG3B,IAAI,SAAS,GAAG,UAAU,CAAC;IAE3B,MAAM,MAAM,GAAG,EAAE,CAAC,gBAAgB,CAAC,QAAQ,EAAE;QAC3C,KAAK,EAAE,UAAU;QACjB,QAAQ,EAAE,MAAM;KACjB,CAAC,CAAC;IAEH,MAAM,EAAE,GAAG,QAAQ,CAAC,eAAe,CAAC;QAClC,KAAK,EAAE,MAAM;QACb,SAAS,EAAE,QAAQ;KACpB,CAAC,CAAC;IAEH,SAAS,cAAc,CAAC,EAAQ;QAC9B,IAAI,CAAC,cAAc,IAAI,EAAE,GAAG,cAAc;YAAE,cAAc,GAAG,EAAE,CAAC;QAChE,IAAI,CAAC,aAAa,IAAI,EAAE,GAAG,aAAa;YAAE,aAAa,GAAG,EAAE,CAAC;IAC/D,CAAC;IAED,SAAS,oBAAoB;QAC3B,IAAI,CAAC,eAAe;YAAE,OAAO;QAC7B,MAAM,GAAG,GAAG,eAAe,CAAC;QAC5B,eAAe,GAAG,IAAI,CAAC;QAGvB,IAAI,GAAG,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC,IAAI,GAAG,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,IAAI,GAAG,CAAC,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC3F,OAAO;QACT,CAAC;QACD,MAAM,GAAG,GAAmB;YAC1B,EAAE,EAAE,gBAAgB,YAAY,EAAE,EAAE;YACpC,QAAQ,EAAE,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI;YACvE,IAAI,EAAE,WAAW;YACjB,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,OAAO,EAAE,GAAG,CAAC,OAAO;YACpB,KAAK,EAAE,GAAG,CAAC,KAAK,IAAI,SAAS;YAC7B,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,WAAW,EAAE,GAAG,CAAC,WAAW;YAE5B,KAAK,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,GAAG,CAAC,YAAY,EAAE,SAAS,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE;YAC7E,MAAM,EAAE,KAAK;YACb,WAAW,EAAE,KAAK;YAGlB,UAAU,EAAE,UAAU;SACvB,CAAC;QACF,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACnB,oBAAoB,IAAI,IAAA,+CAAqB,EAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IAC7D,CAAC;IAED,SAAS,qBAAqB,CAAC,EAAQ,EAAE,KAAa;QACpD,IAAI,CAAC,eAAe,EAAE,CAAC;YACrB,eAAe,GAAG;gBAChB,SAAS,EAAE,EAAE;gBACb,KAAK;gBACL,OAAO,EAAE,EAAE;gBACX,SAAS,EAAE,EAAE;gBACb,WAAW,EAAE,EAAE;gBACf,YAAY,EAAE,CAAC;aAChB,CAAC;QACJ,CAAC;aAAM,IAAI,KAAK,EAAE,CAAC;YACjB,eAAe,CAAC,KAAK,GAAG,KAAK,CAAC;QAChC,CAAC;QACD,OAAO,eAAe,CAAC;IACzB,CAAC;IAOD,SAAS,gBAAgB,CACvB,KAA0B,EAC1B,WAA8B,EAC9B,EAAQ;QAER,IAAI,eAAe,EAAE,CAAC;YACpB,eAAe,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACpC,eAAe,CAAC,WAAW,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAC9C,OAAO;QACT,CAAC;QACD,MAAM,IAAI,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QACxE,IAAI,IAAI,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YACtC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACzB,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACnC,OAAO;QACT,CAAC;QACD,MAAM,GAAG,GAAmB;YAC1B,EAAE,EAAE,sBAAsB,YAAY,EAAE,EAAE;YAC1C,QAAQ,EAAE,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI;YACvE,IAAI,EAAE,MAAM;YACZ,SAAS,EAAE,EAAE;YACb,OAAO,EAAE,CAAC,KAAK,CAAC;YAChB,SAAS,EAAE,EAAE;YACb,WAAW,EAAE,CAAC,WAAW,CAAC;YAC1B,MAAM,EAAE,IAAI;YACZ,WAAW,EAAE,KAAK;SACnB,CAAC;QACF,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACrB,CAAC;IAED,IAAI,CAAC;QACH,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,EAAE,EAAE,CAAC;YAC5B,MAAM,SAAS,GAAG,MAAM,CAAC,UAAU,CAAC,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC;YACtD,SAAS,IAAI,SAAS,CAAC;YAEvB,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;gBAAE,SAAS;YAE3B,IAAI,SAAS,GAAG,cAAc,EAAE,CAAC;gBAC/B,MAAM,CAAC,IAAI,CACT,EAAE,QAAQ,EAAE,SAAS,EAAE,UAAU,EAAE,SAAS,GAAG,SAAS,EAAE,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,EACvF,uCAAuC,CACxC,CAAC;gBACF,kBAAkB,EAAE,CAAC;gBACrB,SAAS;YACX,CAAC;YAED,IAAI,KAAsB,CAAC;YAC3B,IAAI,CAAC;gBACH,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAoB,CAAC;YAC9C,CAAC;YAAC,MAAM,CAAC;gBACP,MAAM,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,EAAE,+BAA+B,CAAC,CAAC;gBAC3D,SAAS;YACX,CAAC;YAED,IAAI,CAAC,KAAK,CAAC,IAAI;gBAAE,SAAS;YAC1B,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,IAAI,EAAE,CAAC;YAC9B,MAAM,EAAE,GAAG,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC;YACpE,cAAc,CAAC,EAAE,CAAC,CAAC;YAEnB,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;gBAEnB,KAAK,eAAe,CAAC,CAAC,CAAC;oBACrB,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC;oBAC3B,IAAI,OAAO,GAAG,KAAK,QAAQ;wBAAE,SAAS,GAAG,GAAG,CAAC;oBAC7C,MAAM;gBACR,CAAC;gBAED,KAAK,gBAAgB,CAAC,CAAC,CAAC;oBAGtB,MAAM;gBACR,CAAC;gBAED,KAAK,sBAAsB,CAAC,CAAC,CAAC;oBAC5B,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC;oBAE/B,IAAI,OAAO,QAAQ,KAAK,QAAQ,IAAI,QAAQ,KAAK,MAAM,EAAE,CAAC;wBACxD,YAAY,GAAG,QAAQ,CAAC;wBACxB,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;oBAC1B,CAAC;oBACD,MAAM;gBACR,CAAC;gBAED,KAAK,cAAc,CAAC,CAAC,CAAC;oBAEpB,MAAM;gBACR,CAAC;gBAED,KAAK,kBAAkB,CAAC,CAAC,CAAC;oBAIxB,oBAAoB,EAAE,CAAC;oBACvB,YAAY,GAAG,IAAI,CAAC;oBACpB,MAAM,QAAQ,GAAG,IAAkC,CAAC;oBACpD,MAAM,EAAE,GAAG,QAAQ,CAAC,YAAY,IAAI,EAAE,CAAC;oBACvC,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;wBACpC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;wBACrB,MAAM,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,EAAE,KAAK,IAAI,EAAE,CAAC;wBACjC,MAAM,IAAI,GAAG,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI;4BAC1C,KAAK,EAAE,CAAC;4BACR,MAAM,EAAE,CAAC;4BACT,SAAS,EAAE,CAAC;4BACZ,UAAU,EAAE,CAAC;4BACb,SAAS,EAAE,CAAC;yBACb,CAAC;wBAGF,IAAI,CAAC,KAAK,IAAI,CAAC,CAAC,WAAW,IAAI,CAAC,CAAC;wBACjC,IAAI,CAAC,MAAM,IAAI,CAAC,CAAC,YAAY,IAAI,CAAC,CAAC;wBACnC,IAAI,CAAC,SAAS,IAAI,CAAC,CAAC,eAAe,IAAI,CAAC,CAAC;wBACzC,IAAI,CAAC,UAAU,IAAI,CAAC,CAAC,gBAAgB,IAAI,CAAC,CAAC;wBAC3C,IAAI,CAAC,SAAS,IAAI,CAAC,CAAC,eAAe,IAAI,CAAC,CAAC;wBACzC,gBAAgB,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;wBAClC,aAAa,IAAI,EAAE,CAAC,KAAK,CAAC,EAAE,QAAQ,EAAE,IAAI,IAAI,CAAC,CAAC;oBAClD,CAAC;oBACD,IAAI,OAAO,QAAQ,CAAC,YAAY,KAAK,QAAQ,IAAI,QAAQ,CAAC,YAAY,KAAK,MAAM,EAAE,CAAC;wBAClF,YAAY,GAAG,QAAQ,CAAC,YAAY,CAAC;wBACrC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;oBACvC,CAAC;oBAED,IAAI,OAAO,QAAQ,CAAC,aAAa,KAAK,QAAQ;wBAC5C,iBAAiB,GAAG,QAAQ,CAAC,aAAa,CAAC;oBAC7C,MAAM,GAAG,GAAG,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;oBACzC,MAAM,IAAI,GAAG,KAAK,CAAC,QAAQ,CAAC,kBAAkB,CAAC,CAAC;oBAChD,MAAM,KAAK,GAAG,KAAK,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC;oBACpD,IAAI,GAAG,KAAK,SAAS,IAAI,IAAI,KAAK,SAAS,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;wBACnE,oBAAoB,GAAG;4BACrB,MAAM,EAAE,GAAG,IAAI,CAAC;4BAChB,YAAY,EAAE,IAAI,IAAI,CAAC;4BACvB,eAAe,EAAE,KAAK,IAAI,CAAC;yBAC5B,CAAC;oBACJ,CAAC;oBACD,MAAM;gBACR,CAAC;gBAGD,KAAK,gBAAgB,CAAC,CAAC,CAAC;oBAItB,MAAM;gBACR,CAAC;gBAED,KAAK,cAAc,CAAC,CAAC,CAAC;oBAEpB,oBAAoB,EAAE,CAAC;oBACvB,MAAM,OAAO,GAAG,OAAO,IAAI,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;oBACrE,IAAI,CAAC,OAAO;wBAAE,MAAM;oBACpB,MAAM,GAAG,GAAmB;wBAC1B,EAAE,EAAE,gBAAgB,YAAY,EAAE,EAAE;wBACpC,QAAQ,EAAE,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI;wBACvE,IAAI,EAAE,MAAM;wBACZ,SAAS,EAAE,EAAE;wBACb,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;wBAC1C,SAAS,EAAE,EAAE;wBACb,WAAW,EAAE,EAAE;wBACf,MAAM,EAAE,KAAK;wBACb,WAAW,EAAE,KAAK;qBACnB,CAAC;oBACF,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;oBACnB,oBAAoB,IAAI,IAAA,+CAAqB,EAAC,GAAG,CAAC,OAAO,CAAC,CAAC;oBAC3D,MAAM;gBACR,CAAC;gBAGD,KAAK,sBAAsB,CAAC,CAAC,CAAC;oBAG5B,MAAM;gBACR,CAAC;gBAED,KAAK,mBAAmB,CAAC,CAAC,CAAC;oBACzB,MAAM,KAAK,GAAG,OAAO,IAAI,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC/D,IAAI,KAAK,EAAE,CAAC;wBACV,YAAY,GAAG,KAAK,CAAC;wBACrB,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;oBACvB,CAAC;oBACD,MAAM,GAAG,GAAG,qBAAqB,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;oBAG7C,MAAM,aAAa,GAAG,IAAI,CAAC,aAAa,CAAC;oBACzC,IAAI,OAAO,aAAa,KAAK,QAAQ,IAAI,aAAa,EAAE,CAAC;wBACvD,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,aAAa,EAAE,CAAC,CAAC;oBAClE,CAAC;oBAGD,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC;oBAC1B,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,EAAE,CAAC;wBACrC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;oBAC3C,CAAC;oBAGD,IAAI,gBAAgB,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC;wBACzD,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,YAAgC,EAAE,CAAC;4BACvD,MAAM,MAAM,GAAG,EAAE,CAAC,UAAU,CAAC;4BAC7B,MAAM,IAAI,GAAG,EAAE,CAAC,IAAI,CAAC;4BACrB,IAAI,CAAC,MAAM,IAAI,CAAC,IAAI;gCAAE,SAAS;4BAC/B,MAAM,KAAK,GAAG,EAAE,CAAC,SAAS,IAAI,EAAE,CAAC;4BACjC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;4BACnF,GAAG,CAAC,SAAS,CAAC,IAAI,CAAC;gCACjB,EAAE,EAAE,MAAM;gCACV,IAAI;gCACJ,KAAK;gCACL,MAAM,EAAE,IAAI,KAAK,MAAM;6BACxB,CAAC,CAAC;wBACL,CAAC;oBACH,CAAC;oBAGD,MAAM,GAAG,GAAG,OAAO,IAAI,CAAC,YAAY,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC;oBAC1E,GAAG,CAAC,YAAY,IAAI,GAAG,CAAC;oBACxB,gBAAgB,IAAI,GAAG,CAAC;oBACxB,MAAM;gBACR,CAAC;gBAED,KAAK,oBAAoB,CAAC,CAAC,CAAC;oBAI1B,MAAM;gBACR,CAAC;gBAGD,KAAK,sBAAsB,CAAC,CAAC,CAAC;oBAG5B,MAAM;gBACR,CAAC;gBAED,KAAK,yBAAyB,CAAC,CAAC,CAAC;oBAC/B,IAAI,CAAC,gBAAgB;wBAAE,MAAM;oBAC7B,MAAM,MAAM,GAAG,OAAO,IAAI,CAAC,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC1E,IAAI,CAAC,MAAM;wBAAE,MAAM;oBACnB,MAAM,MAAM,GAAI,IAAI,CAAC,MAA4C,IAAI,EAAE,CAAC;oBACxE,MAAM,UAAU,GAAG,MAAM,CAAC,OAAO,CAAC;oBAClC,MAAM,aAAa,GACjB,OAAO,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC;oBACjF,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,KAAK,KAAK,CAAC;oBAIvC,MAAM,KAAK,GAAwB;wBACjC,IAAI,EAAE,aAAa;wBACnB,UAAU,EAAE,MAAM;wBAClB,OAAO,EAAE,aAAa;wBACtB,OAAO;qBACR,CAAC;oBACF,MAAM,WAAW,GAAsB;wBACrC,UAAU,EAAE,MAAM;wBAClB,OAAO,EAAE,aAAa;wBACtB,OAAO;qBACR,CAAC;oBACF,gBAAgB,CAAC,KAAK,EAAE,WAAW,EAAE,EAAE,CAAC,CAAC;oBACzC,MAAM;gBACR,CAAC;gBAED,KAAK,YAAY,CAAC;gBAClB,KAAK,UAAU,CAAC,CAAC,CAAC;oBAEhB,MAAM;gBACR,CAAC;gBAED;oBACE,MAAM,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,EAAE,qCAAqC,CAAC,CAAC;oBAC1E,MAAM;YACV,CAAC;YAED,IAAI,WAAW,IAAI,QAAQ,CAAC,MAAM,IAAI,WAAW;gBAAE,MAAM;QAC3D,CAAC;IACH,CAAC;YAAS,CAAC;QACT,EAAE,CAAC,KAAK,EAAE,CAAC;QACX,MAAM,CAAC,OAAO,EAAE,CAAC;IACnB,CAAC;IAGD,oBAAoB,EAAE,CAAC;IAWvB,MAAM,QAAQ,GAAa,EAAE,CAAC;IAE9B,IAAI,WAAW,GAAG,CAAC,CAAC;IACpB,IAAI,YAAY,GAAG,CAAC,CAAC;IACrB,IAAI,eAAe,GAAG,CAAC,CAAC;IACxB,IAAI,mBAAmB,GAAG,CAAC,CAAC;IAC5B,IAAI,OAAO,GAAG,CAAC,CAAC;IAChB,IAAI,kBAAkB,GAAG,CAAC,CAAC;IAC3B,IAAI,gBAAoD,CAAC;IACzD,IAAI,UAA8B,CAAC;IAEnC,IAAI,YAAY,EAAE,CAAC;QACjB,KAAK,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,IAAI,gBAAgB,EAAE,CAAC;YAC1C,WAAW,IAAI,CAAC,CAAC,KAAK,CAAC;YACvB,YAAY,IAAI,CAAC,CAAC,MAAM,CAAC;YACzB,eAAe,IAAI,CAAC,CAAC,SAAS,CAAC;YAC/B,mBAAmB,IAAI,CAAC,CAAC,UAAU,CAAC;YAEpC,OAAO,IAAI,IAAA,mDAA2B,EACpC,KAAK,EACL;gBACE,WAAW,EAAE,CAAC,CAAC,KAAK;gBACpB,YAAY,EAAE,CAAC,CAAC,MAAM;gBACtB,eAAe,EAAE,CAAC,CAAC,SAAS;gBAC5B,mBAAmB,EAAE,CAAC,CAAC,UAAU;aAClC,EAGD,OAAO,IAAI,WAAW,CACvB,CAAC;QACJ,CAAC;QACD,kBAAkB,GAAG,iBAAiB,CAAC;QACvC,gBAAgB,GAAG,oBAAoB,CAAC;QACxC,UAAU,GAAG,aAAa,CAAC;IAC7B,CAAC;SAAM,CAAC;QAEN,YAAY,GAAG,gBAAgB,CAAC;QAChC,OAAO;YACL,gBAAgB,GAAG,CAAC,IAAI,YAAY;gBAClC,CAAC,CAAC,IAAA,mDAA2B,EACzB,YAAY,EACZ;oBACE,WAAW,EAAE,CAAC;oBACd,YAAY,EAAE,gBAAgB;oBAC9B,eAAe,EAAE,CAAC;oBAClB,mBAAmB,EAAE,CAAC;iBACvB,EACD,OAAO,IAAI,WAAW,CACvB;gBACH,CAAC,CAAC,CAAC,CAAC;QACR,QAAQ,CAAC,IAAI,CACX,yJAAyJ,CAC1J,CAAC;IACJ,CAAC;IAED,MAAM,UAAU,GACd,cAAc,IAAI,aAAa,CAAC,CAAC,CAAC,aAAa,CAAC,OAAO,EAAE,GAAG,cAAc,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAE3F,MAAM,SAAS,GAAG,CAAC,YAAY,CAAC;IAEhC,MAAM,mBAAmB,GACvB,OAAO,IAAI,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC,oBAAoB,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;IAEjF,MAAM,UAAU,GAAG,SAAS,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAG1E,MAAM,MAAM,GAA0B;QACpC;YACE,WAAW,EAAE,CAAC;YACd,YAAY,EAAE,WAAW,GAAG,eAAe;YAC3C,UAAU,EAAE,WAAW,GAAG,eAAe;SAC1C;KACF,CAAC;IACF,MAAM,uBAAuB,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC;IACnF,MAAM,WAAW,GAAG,WAAW,GAAG,YAAY,GAAG,eAAe,GAAG,mBAAmB,CAAC;IAEvF,MAAM,OAAO,GAAmB;QAC9B,WAAW;QACX,YAAY;QACZ,eAAe;QACf,mBAAmB;QACnB,WAAW;QACX,uBAAuB;QACvB,eAAe,EAAE,CAAC;QAClB,eAAe,EAAE,MAAM;QACvB,oBAAoB;QACpB,kBAAkB;QAClB,mBAAmB;QACnB,gBAAgB;QAChB,OAAO;QACP,UAAU;QACV,YAAY;QACZ,UAAU;QACV,UAAU;QACV,YAAY,EAAE,QAAQ,CAAC,MAAM;QAC7B,SAAS;KACV,CAAC;IAEF,IAAI,kBAAkB,GAAG,CAAC,EAAE,CAAC;QAC3B,QAAQ,CAAC,IAAI,CACX,WAAW,kBAAkB,kBAAkB,kBAAkB,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,eAAe,CAChG,CAAC;IACJ,CAAC;IAED,OAAO;QACL,QAAQ;QACR,OAAO;QACP,SAAS;QACT,SAAS;QACT,QAAQ,EAAE,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS;KACrD,CAAC;AACJ,CAAC"}

package/dist/server/modules/session-reader/readers/antigravity-metrics.reader.d.ts

@@ -0,0 +1,25 @@
+export interface AgyTokenMetrics {
+    inputTokens: number;
+    outputTokens: number;
+    cacheReadTokens: number;
+    cacheCreationTokens: number;
+    generationCount: number;
+    lastContextTokens: number;
+    modelId?: string;
+    displayName?: string;
+    warnings: string[];
+}
+export interface AntigravityMetricsReaderOptions {
+    busyTimeoutMs?: number;
+    openRetries?: number;
+}
+export declare class AntigravityMetricsReader {
+    private readonly busyTimeoutMs;
+    private readonly openRetries;
+    constructor(options?: AntigravityMetricsReaderOptions);
+    decode(dbPath: string, providerSessionId: string): AgyTokenMetrics;
+    private decodeRowUsage;
+    private withDb;
+    private hasTable;
+    private sleep;
+}

package/dist/server/modules/session-reader/readers/antigravity-metrics.reader.js

@@ -0,0 +1,142 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AntigravityMetricsReader = void 0;
+const better_sqlite3_1 = __importDefault(require("better-sqlite3"));
+const logger_1 = require("../../../common/logging/logger");
+const protobuf_wire_1 = require("./protobuf-wire");
+const logger = (0, logger_1.createLogger)('AntigravityMetricsReader');
+const DEFAULT_BUSY_TIMEOUT_MS = 5_000;
+const DEFAULT_OPEN_RETRIES = 3;
+const OPEN_RETRY_DELAY_MS = 50;
+const WRAPPER_FIELD = 1;
+const USAGE_FIELD = 4;
+const USAGE_INPUT = 2;
+const USAGE_OUTPUT_TOTAL = 3;
+const MODEL_ID_FIELD = 19;
+const DISPLAY_NAME_FIELD = 21;
+class AntigravityMetricsReader {
+    constructor(options) {
+        this.busyTimeoutMs = options?.busyTimeoutMs ?? DEFAULT_BUSY_TIMEOUT_MS;
+        this.openRetries = options?.openRetries ?? DEFAULT_OPEN_RETRIES;
+    }
+    decode(dbPath, providerSessionId) {
+        const result = emptyMetrics();
+        try {
+            this.withDb(dbPath, (db) => {
+                if (!this.hasTable(db, 'gen_metadata') || !this.hasTable(db, 'trajectory_meta')) {
+                    result.warnings.push('agy metrics: DB schema drift — gen_metadata/trajectory_meta missing (token usage unavailable)');
+                    return;
+                }
+                const cascade = db.prepare('SELECT cascade_id FROM trajectory_meta LIMIT 1').get();
+                if (cascade?.cascade_id && cascade.cascade_id !== providerSessionId) {
+                    result.warnings.push(`agy metrics: .db conversation id mismatch (cascade_id=${cascade.cascade_id} ≠ providerSessionId=${providerSessionId}) — token usage NOT trusted`);
+                    return;
+                }
+                const rows = db
+                    .prepare('SELECT idx, data FROM gen_metadata ORDER BY idx')
+                    .all();
+                if (rows.length === 0) {
+                    result.warnings.push('agy metrics: no generation metadata rows — token usage unavailable');
+                    return;
+                }
+                let decodedRows = 0;
+                for (const row of rows) {
+                    try {
+                        const usage = this.decodeRowUsage(row.data, result);
+                        if (!usage)
+                            continue;
+                        result.inputTokens += usage.input;
+                        result.outputTokens += usage.output;
+                        result.lastContextTokens = usage.input + usage.output;
+                        if (usage.modelId)
+                            result.modelId = usage.modelId;
+                        if (usage.displayName)
+                            result.displayName = usage.displayName;
+                        decodedRows++;
+                    }
+                    catch (error) {
+                        if (error instanceof protobuf_wire_1.ProtobufDecodeError) {
+                            result.warnings.push(`agy metrics: undecodable protobuf at gen_metadata[${row.idx}] (possible version drift): ${error.message}`);
+                            continue;
+                        }
+                        throw error;
+                    }
+                }
+                result.generationCount = decodedRows;
+                if (decodedRows === 0 && result.warnings.length === 0) {
+                    result.warnings.push('agy metrics: no Usage submessage found in any generation — token usage unavailable');
+                }
+            });
+        }
+        catch (error) {
+            result.warnings.push(`agy metrics: cannot read .db (${String(error?.message ?? error)})`);
+            logger.debug({ dbPath, error }, 'agy metrics decode failed');
+        }
+        return result;
+    }
+    decodeRowUsage(data, _result) {
+        const top = (0, protobuf_wire_1.decodeMessage)(data);
+        const wrapper = (0, protobuf_wire_1.getMessage)(top, WRAPPER_FIELD);
+        if (!wrapper)
+            return null;
+        const usage = (0, protobuf_wire_1.getMessage)(wrapper, USAGE_FIELD);
+        if (!usage)
+            return null;
+        return {
+            input: (0, protobuf_wire_1.getVarint)(usage, USAGE_INPUT) ?? 0,
+            output: (0, protobuf_wire_1.getVarint)(usage, USAGE_OUTPUT_TOTAL) ?? 0,
+            modelId: (0, protobuf_wire_1.getString)(wrapper, MODEL_ID_FIELD),
+            displayName: (0, protobuf_wire_1.getString)(wrapper, DISPLAY_NAME_FIELD),
+        };
+    }
+    withDb(dbPath, fn) {
+        let lastError;
+        for (let attempt = 0; attempt <= this.openRetries; attempt++) {
+            let db;
+            try {
+                db = new better_sqlite3_1.default(dbPath, { readonly: true, fileMustExist: true });
+                db.pragma(`busy_timeout = ${this.busyTimeoutMs}`);
+                return fn(db);
+            }
+            catch (error) {
+                lastError = error;
+                const code = error?.code;
+                const transient = code === 'SQLITE_BUSY' || code === 'SQLITE_LOCKED';
+                if (!transient || attempt === this.openRetries)
+                    throw error;
+                this.sleep(OPEN_RETRY_DELAY_MS * (attempt + 1));
+            }
+            finally {
+                db?.close();
+            }
+        }
+        throw lastError;
+    }
+    hasTable(db, name) {
+        const row = db
+            .prepare("SELECT name FROM sqlite_master WHERE type='table' AND name = ?")
+            .get(name);
+        return row !== undefined;
+    }
+    sleep(ms) {
+        const end = Date.now() + ms;
+        while (Date.now() < end) {
+        }
+    }
+}
+exports.AntigravityMetricsReader = AntigravityMetricsReader;
+function emptyMetrics() {
+    return {
+        inputTokens: 0,
+        outputTokens: 0,
+        cacheReadTokens: 0,
+        cacheCreationTokens: 0,
+        generationCount: 0,
+        lastContextTokens: 0,
+        warnings: [],
+    };
+}
+//# sourceMappingURL=antigravity-metrics.reader.js.map

package/dist/server/modules/session-reader/readers/antigravity-metrics.reader.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"antigravity-metrics.reader.js","sourceRoot":"","sources":["../../../../src/modules/session-reader/readers/antigravity-metrics.reader.ts"],"names":[],"mappings":";;;;;;AA8BA,oEAAsC;AACtC,2DAA8D;AAC9D,mDAMyB;AAEzB,MAAM,MAAM,GAAG,IAAA,qBAAY,EAAC,0BAA0B,CAAC,CAAC;AAExD,MAAM,uBAAuB,GAAG,KAAK,CAAC;AACtC,MAAM,oBAAoB,GAAG,CAAC,CAAC;AAC/B,MAAM,mBAAmB,GAAG,EAAE,CAAC;AAG/B,MAAM,aAAa,GAAG,CAAC,CAAC;AACxB,MAAM,WAAW,GAAG,CAAC,CAAC;AACtB,MAAM,WAAW,GAAG,CAAC,CAAC;AACtB,MAAM,kBAAkB,GAAG,CAAC,CAAC;AAC7B,MAAM,cAAc,GAAG,EAAE,CAAC;AAC1B,MAAM,kBAAkB,GAAG,EAAE,CAAC;AA4B9B,MAAa,wBAAwB;IAInC,YAAY,OAAyC;QACnD,IAAI,CAAC,aAAa,GAAG,OAAO,EAAE,aAAa,IAAI,uBAAuB,CAAC;QACvE,IAAI,CAAC,WAAW,GAAG,OAAO,EAAE,WAAW,IAAI,oBAAoB,CAAC;IAClE,CAAC;IAQD,MAAM,CAAC,MAAc,EAAE,iBAAyB;QAC9C,MAAM,MAAM,GAAG,YAAY,EAAE,CAAC;QAC9B,IAAI,CAAC;YACH,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,EAAE;gBACzB,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,iBAAiB,CAAC,EAAE,CAAC;oBAChF,MAAM,CAAC,QAAQ,CAAC,IAAI,CAClB,+FAA+F,CAChG,CAAC;oBACF,OAAO;gBACT,CAAC;gBAID,MAAM,OAAO,GAAG,EAAE,CAAC,OAAO,CAAC,gDAAgD,CAAC,CAAC,GAAG,EAEnE,CAAC;gBACd,IAAI,OAAO,EAAE,UAAU,IAAI,OAAO,CAAC,UAAU,KAAK,iBAAiB,EAAE,CAAC;oBACpE,MAAM,CAAC,QAAQ,CAAC,IAAI,CAClB,yDAAyD,OAAO,CAAC,UAAU,wBAAwB,iBAAiB,6BAA6B,CAClJ,CAAC;oBACF,OAAO;gBACT,CAAC;gBAED,MAAM,IAAI,GAAG,EAAE;qBACZ,OAAO,CAAC,iDAAiD,CAAC;qBAC1D,GAAG,EAAc,CAAC;gBACrB,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBACtB,MAAM,CAAC,QAAQ,CAAC,IAAI,CAClB,oEAAoE,CACrE,CAAC;oBACF,OAAO;gBACT,CAAC;gBAED,IAAI,WAAW,GAAG,CAAC,CAAC;gBACpB,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;oBACvB,IAAI,CAAC;wBACH,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;wBACpD,IAAI,CAAC,KAAK;4BAAE,SAAS;wBACrB,MAAM,CAAC,WAAW,IAAI,KAAK,CAAC,KAAK,CAAC;wBAClC,MAAM,CAAC,YAAY,IAAI,KAAK,CAAC,MAAM,CAAC;wBACpC,MAAM,CAAC,iBAAiB,GAAG,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC;wBACtD,IAAI,KAAK,CAAC,OAAO;4BAAE,MAAM,CAAC,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;wBAClD,IAAI,KAAK,CAAC,WAAW;4BAAE,MAAM,CAAC,WAAW,GAAG,KAAK,CAAC,WAAW,CAAC;wBAC9D,WAAW,EAAE,CAAC;oBAChB,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACf,IAAI,KAAK,YAAY,mCAAmB,EAAE,CAAC;4BACzC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAClB,qDAAqD,GAAG,CAAC,GAAG,+BAA+B,KAAK,CAAC,OAAO,EAAE,CAC3G,CAAC;4BACF,SAAS;wBACX,CAAC;wBACD,MAAM,KAAK,CAAC;oBACd,CAAC;gBACH,CAAC;gBACD,MAAM,CAAC,eAAe,GAAG,WAAW,CAAC;gBACrC,IAAI,WAAW,KAAK,CAAC,IAAI,MAAM,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBACtD,MAAM,CAAC,QAAQ,CAAC,IAAI,CAClB,oFAAoF,CACrF,CAAC;gBACJ,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,QAAQ,CAAC,IAAI,CAClB,iCAAiC,MAAM,CAAE,KAAe,EAAE,OAAO,IAAI,KAAK,CAAC,GAAG,CAC/E,CAAC;YACF,MAAM,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,2BAA2B,CAAC,CAAC;QAC/D,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAGO,cAAc,CACpB,IAAY,EACZ,OAAwB;QAExB,MAAM,GAAG,GAAG,IAAA,6BAAa,EAAC,IAAI,CAAC,CAAC;QAChC,MAAM,OAAO,GAAG,IAAA,0BAAU,EAAC,GAAG,EAAE,aAAa,CAAC,CAAC;QAC/C,IAAI,CAAC,OAAO;YAAE,OAAO,IAAI,CAAC;QAC1B,MAAM,KAAK,GAAG,IAAA,0BAAU,EAAC,OAAO,EAAE,WAAW,CAAC,CAAC;QAC/C,IAAI,CAAC,KAAK;YAAE,OAAO,IAAI,CAAC;QACxB,OAAO;YACL,KAAK,EAAE,IAAA,yBAAS,EAAC,KAAK,EAAE,WAAW,CAAC,IAAI,CAAC;YACzC,MAAM,EAAE,IAAA,yBAAS,EAAC,KAAK,EAAE,kBAAkB,CAAC,IAAI,CAAC;YACjD,OAAO,EAAE,IAAA,yBAAS,EAAC,OAAO,EAAE,cAAc,CAAC;YAC3C,WAAW,EAAE,IAAA,yBAAS,EAAC,OAAO,EAAE,kBAAkB,CAAC;SACpD,CAAC;IACJ,CAAC;IAMO,MAAM,CAAI,MAAc,EAAE,EAAgC;QAChE,IAAI,SAAkB,CAAC;QACvB,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,IAAI,CAAC,WAAW,EAAE,OAAO,EAAE,EAAE,CAAC;YAC7D,IAAI,EAAiC,CAAC;YACtC,IAAI,CAAC;gBACH,EAAE,GAAG,IAAI,wBAAQ,CAAC,MAAM,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;gBACnE,EAAE,CAAC,MAAM,CAAC,kBAAkB,IAAI,CAAC,aAAa,EAAE,CAAC,CAAC;gBAClD,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;YAChB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,SAAS,GAAG,KAAK,CAAC;gBAClB,MAAM,IAAI,GAAI,KAAkC,EAAE,IAAI,CAAC;gBACvD,MAAM,SAAS,GAAG,IAAI,KAAK,aAAa,IAAI,IAAI,KAAK,eAAe,CAAC;gBACrE,IAAI,CAAC,SAAS,IAAI,OAAO,KAAK,IAAI,CAAC,WAAW;oBAAE,MAAM,KAAK,CAAC;gBAC5D,IAAI,CAAC,KAAK,CAAC,mBAAmB,GAAG,CAAC,OAAO,GAAG,CAAC,CAAC,CAAC,CAAC;YAClD,CAAC;oBAAS,CAAC;gBACT,EAAE,EAAE,KAAK,EAAE,CAAC;YACd,CAAC;QACH,CAAC;QACD,MAAM,SAAS,CAAC;IAClB,CAAC;IAEO,QAAQ,CAAC,EAAqB,EAAE,IAAY;QAClD,MAAM,GAAG,GAAG,EAAE;aACX,OAAO,CAAC,gEAAgE,CAAC;aACzE,GAAG,CAAC,IAAI,CAAC,CAAC;QACb,OAAO,GAAG,KAAK,SAAS,CAAC;IAC3B,CAAC;IAGO,KAAK,CAAC,EAAU;QACtB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC;QAC5B,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,GAAG,EAAE,CAAC;QAE1B,CAAC;IACH,CAAC;CACF;AA9ID,4DA8IC;AAED,SAAS,YAAY;IACnB,OAAO;QACL,WAAW,EAAE,CAAC;QACd,YAAY,EAAE,CAAC;QACf,eAAe,EAAE,CAAC;QAClB,mBAAmB,EAAE,CAAC;QACtB,eAAe,EAAE,CAAC;QAClB,iBAAiB,EAAE,CAAC;QACpB,QAAQ,EAAE,EAAE;KACb,CAAC;AACJ,CAAC"}

package/dist/server/modules/session-reader/readers/antigravity-model-pricing.d.ts

@@ -0,0 +1,6 @@
+export interface AgyPricingMapping {
+    pricingKey: string | null;
+    isFree: boolean;
+    freeContextWindow?: number;
+}
+export declare function mapAgyModelToPricing(displayName?: string, modelId?: string): AgyPricingMapping | null;