context-lens 0.2.0 → 0.2.1

package/dist/http/headers.js

@@ -0,0 +1,54 @@
+/**
+ * Shared header redaction utilities.
+ *
+ * Context Lens captures and exports some headers for debugging and provenance,
+ * but must never persist secrets (API keys, cookies, auth challenges, etc.).
+ *
+ * Keep this as the single source of truth to avoid drift between "capture" and "export".
+ */
+/** Case-insensitive set of header names that must never be persisted/exported. */
+export const SENSITIVE_HEADERS = new Set([
+    "authorization",
+    "x-api-key",
+    "cookie",
+    "set-cookie",
+    "x-target-url",
+    "proxy-authorization",
+    "x-auth-token",
+    "x-forwarded-authorization",
+    "www-authenticate",
+    "proxy-authenticate",
+    "x-goog-api-key",
+]);
+/**
+ * Remove sensitive headers from a header map.
+ *
+ * @param headers - Header map (string -> string)
+ * @returns A new object with sensitive headers removed.
+ */
+export function redactHeaders(headers) {
+    const result = {};
+    for (const [key, val] of Object.entries(headers)) {
+        if (SENSITIVE_HEADERS.has(key.toLowerCase()))
+            continue;
+        result[key] = val;
+    }
+    return result;
+}
+/**
+ * Select a safe subset of request/response headers for capture.
+ *
+ * - Drops sensitive headers (see `SENSITIVE_HEADERS`)
+ * - Keeps only string-valued headers (Node can represent multi-valued headers as arrays)
+ */
+export function selectHeaders(headers) {
+    const result = {};
+    for (const [key, val] of Object.entries(headers)) {
+        if (SENSITIVE_HEADERS.has(key.toLowerCase()))
+            continue;
+        if (typeof val === "string")
+            result[key] = val;
+    }
+    return result;
+}
+//# sourceMappingURL=headers.js.map

package/dist/http/headers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"headers.js","sourceRoot":"","sources":["../../src/http/headers.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,kFAAkF;AAClF,MAAM,CAAC,MAAM,iBAAiB,GAAG,IAAI,GAAG,CAAC;IACvC,eAAe;IACf,WAAW;IACX,QAAQ;IACR,YAAY;IACZ,cAAc;IACd,qBAAqB;IACrB,cAAc;IACd,2BAA2B;IAC3B,kBAAkB;IAClB,oBAAoB;IACpB,gBAAgB;CACjB,CAAC,CAAC;AAEH;;;;;GAKG;AACH,MAAM,UAAU,aAAa,CAC3B,OAA+B;IAE/B,MAAM,MAAM,GAA2B,EAAE,CAAC;IAC1C,KAAK,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QACjD,IAAI,iBAAiB,CAAC,GAAG,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC;YAAE,SAAS;QACvD,MAAM,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC;IACpB,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,aAAa,CAC3B,OAA4B;IAE5B,MAAM,MAAM,GAA2B,EAAE,CAAC;IAC1C,KAAK,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QACjD,IAAI,iBAAiB,CAAC,GAAG,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC;YAAE,SAAS;QACvD,IAAI,OAAO,GAAG,KAAK,QAAQ;YAAE,MAAM,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC;IACjD,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/lhar-types.generated.d.ts

@@ -9,7 +9,7 @@
  */
 export type CompositionCategory = "system_prompt" | "tool_definitions" | "tool_results" | "tool_calls" | "assistant_text" | "user_text" | "thinking" | "system_injections" | "images" | "cache_markers" | "other";
 /**
- * JSON Schema for the LHAR format — what HAR is for web traffic, but for LLM API calls.
+ * JSON Schema for the LHAR format. What HAR is for web traffic, but for LLM API calls.
  */
 export interface LHARLLMHTTPArchiveFormat {
     [k: string]: unknown;

package/dist/lhar.d.ts

@@ -1,7 +1,7 @@
-import type { CompositionEntry, LharRecord, LharSessionLine, LharJsonWrapper } from './lhar-types.generated.js';
-import type { CapturedEntry, ContextInfo, Conversation } from './types.js';
-export declare const SENSITIVE_HEADERS: Set<string>;
-export declare function redactHeaders(headers: Record<string, string>): Record<string, string>;
+import { redactHeaders, SENSITIVE_HEADERS } from "./http/headers.js";
+import type { CompositionEntry, LharJsonWrapper, LharRecord, LharSessionLine } from "./lhar-types.generated.js";
+import type { CapturedEntry, ContextInfo, Conversation } from "./types.js";
+export { redactHeaders, SENSITIVE_HEADERS };
 export declare function analyzeComposition(contextInfo: ContextInfo, rawBody: Record<string, any> | undefined): CompositionEntry[];
 export interface ParsedResponseUsage {
     inputTokens: number;

package/dist/lhar.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"lhar.d.ts","sourceRoot":"","sources":["../src/lhar.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EACW,gBAAgB,EACrC,UAAU,EAAE,eAAe,EAAE,eAAe,EAC7C,MAAM,2BAA2B,CAAC;AACnC,OAAO,KAAK,EAAE,aAAa,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAQ3E,eAAO,MAAM,iBAAiB,aAI5B,CAAC;AAEH,wBAAgB,aAAa,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAOrF;AAID,wBAAgB,kBAAkB,CAChC,WAAW,EAAE,WAAW,EACxB,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,SAAS,GACvC,gBAAgB,EAAE,CAyDpB;AAwID,MAAM,WAAW,mBAAmB;IAClC,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,eAAe,EAAE,MAAM,CAAC;IACxB,gBAAgB,EAAE,MAAM,CAAC;IACzB,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;IACrB,aAAa,EAAE,MAAM,EAAE,CAAC;IACxB,MAAM,EAAE,OAAO,CAAC;CACjB;AAED,wBAAgB,kBAAkB,CAAC,YAAY,EAAE,GAAG,GAAG,mBAAmB,CAuCzE;AA+DD,wBAAgB,eAAe,CAC7B,KAAK,EAAE,aAAa,EACpB,WAAW,EAAE,aAAa,EAAE,GAC3B,UAAU,CAqIZ;AAID,wBAAgB,gBAAgB,CAC9B,cAAc,EAAE,MAAM,EACtB,YAAY,EAAE,YAAY,EAC1B,KAAK,EAAE,MAAM,GACZ,eAAe,CAQjB;AAID,wBAAgB,WAAW,CAAC,OAAO,EAAE,aAAa,EAAE,EAAE,aAAa,EAAE,GAAG,CAAC,MAAM,EAAE,YAAY,CAAC,GAAG,MAAM,CA2BtG;AAED,wBAAgB,UAAU,CACxB,OAAO,EAAE,aAAa,EAAE,EACxB,aAAa,EAAE,GAAG,CAAC,MAAM,EAAE,YAAY,CAAC,GACvC,eAAe,CAsCjB"}
+{"version":3,"file":"lhar.d.ts","sourceRoot":"","sources":["../src/lhar.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,aAAa,EAAE,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AACrE,OAAO,KAAK,EAEV,gBAAgB,EAChB,eAAe,EACf,UAAU,EACV,eAAe,EAChB,MAAM,2BAA2B,CAAC;AACnC,OAAO,KAAK,EAAE,aAAa,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAS3E,OAAO,EAAE,aAAa,EAAE,iBAAiB,EAAE,CAAC;AAI5C,wBAAgB,kBAAkB,CAChC,WAAW,EAAE,WAAW,EACxB,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,SAAS,GACvC,gBAAgB,EAAE,CA0EpB;AAgLD,MAAM,WAAW,mBAAmB;IAClC,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,eAAe,EAAE,MAAM,CAAC;IACxB,gBAAgB,EAAE,MAAM,CAAC;IACzB,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;IACrB,aAAa,EAAE,MAAM,EAAE,CAAC;IACxB,MAAM,EAAE,OAAO,CAAC;CACjB;AAED,wBAAgB,kBAAkB,CAAC,YAAY,EAAE,GAAG,GAAG,mBAAmB,CAoEzE;AAsFD,wBAAgB,eAAe,CAC7B,KAAK,EAAE,aAAa,EACpB,WAAW,EAAE,aAAa,EAAE,GAC3B,UAAU,CAkJZ;AAID,wBAAgB,gBAAgB,CAC9B,cAAc,EAAE,MAAM,EACtB,YAAY,EAAE,YAAY,EAC1B,KAAK,EAAE,MAAM,GACZ,eAAe,CAQjB;AAID,wBAAgB,WAAW,CACzB,OAAO,EAAE,aAAa,EAAE,EACxB,aAAa,EAAE,GAAG,CAAC,MAAM,EAAE,YAAY,CAAC,GACvC,MAAM,CAmCR;AAED,wBAAgB,UAAU,CACxB,OAAO,EAAE,aAAa,EAAE,EACxB,aAAa,EAAE,GAAG,CAAC,MAAM,EAAE,YAAY,CAAC,GACvC,eAAe,CAsCjB"}

package/dist/lhar.js

@@ -1,23 +1,13 @@
-import { randomUUID, randomBytes, createHash } from 'node:crypto';
-import { estimateTokens } from './core.js';
-const COLLECTOR_NAME = 'context-lens';
-const COLLECTOR_VERSION = '0.1.0';
-const LHAR_VERSION = '0.1.0';
+import { createHash, randomBytes, randomUUID } from "node:crypto";
+import { estimateTokens } from "./core.js";
+import { redactHeaders, SENSITIVE_HEADERS } from "./http/headers.js";
+import { VERSION } from "./version.generated.js";
+const COLLECTOR_NAME = "context-lens";
+const COLLECTOR_VERSION = VERSION;
+const LHAR_VERSION = "0.1.0";
 // --- Header Redaction ---
-export const SENSITIVE_HEADERS = new Set([
-    'authorization', 'x-api-key', 'cookie', 'set-cookie',
-    'x-target-url', 'proxy-authorization', 'x-auth-token',
-    'x-forwarded-authorization', 'www-authenticate', 'proxy-authenticate',
-]);
-export function redactHeaders(headers) {
-    const result = {};
-    for (const [key, val] of Object.entries(headers)) {
-        if (SENSITIVE_HEADERS.has(key.toLowerCase()))
-            continue;
-        result[key] = val;
-    }
-    return result;
-}
+// Kept as exports for backward compatibility, but implemented in `src/http/headers.ts`.
+export { redactHeaders, SENSITIVE_HEADERS };
 // --- Composition Analysis ---
 export function analyzeComposition(contextInfo, rawBody) {
     const counts = new Map();
@@ -34,90 +24,104 @@ export function analyzeComposition(contextInfo, rawBody) {
     if (!rawBody) {
         // Fallback to contextInfo aggregates
         if (contextInfo.systemTokens > 0)
-            add('system_prompt', contextInfo.systemTokens);
+            add("system_prompt", contextInfo.systemTokens);
         if (contextInfo.toolsTokens > 0)
-            add('tool_definitions', contextInfo.toolsTokens);
+            add("tool_definitions", contextInfo.toolsTokens);
         if (contextInfo.messagesTokens > 0)
-            add('other', contextInfo.messagesTokens);
+            add("other", contextInfo.messagesTokens);
         return buildCompositionArray(counts, contextInfo.totalTokens);
     }
     // System prompt(s)
     if (rawBody.system) {
-        if (typeof rawBody.system === 'string') {
-            add('system_prompt', estimateTokens(rawBody.system));
+        if (typeof rawBody.system === "string") {
+            add("system_prompt", estimateTokens(rawBody.system));
         }
         else if (Array.isArray(rawBody.system)) {
             for (const block of rawBody.system) {
                 if (block.cache_control) {
-                    add('cache_markers', estimateTokens(block.cache_control));
+                    add("cache_markers", estimateTokens(block.cache_control));
                 }
-                add('system_prompt', estimateTokens(block.text || block));
+                add("system_prompt", estimateTokens(block.text || block));
             }
         }
     }
     // Instructions (OpenAI Responses API / ChatGPT)
     if (rawBody.instructions) {
-        add('system_prompt', estimateTokens(rawBody.instructions));
+        add("system_prompt", estimateTokens(rawBody.instructions));
     }
     // Tool definitions
     if (rawBody.tools && Array.isArray(rawBody.tools)) {
-        add('tool_definitions', estimateTokens(JSON.stringify(rawBody.tools)));
+        add("tool_definitions", estimateTokens(JSON.stringify(rawBody.tools)));
     }
-    // Messages array (Anthropic, OpenAI chat completions)
-    const messages = rawBody.messages || rawBody.input;
+    // Gemini/Code Assist: unwrap .request wrapper if present
+    const geminiBody = rawBody.request || rawBody;
+    // Gemini systemInstruction
+    if (geminiBody.systemInstruction) {
+        const parts = geminiBody.systemInstruction.parts || [];
+        add("system_prompt", estimateTokens(parts.map((p) => p.text || "").join("\n")));
+    }
+    // Gemini contents[] or standard messages[]/input[]
+    const messages = geminiBody.contents || rawBody.messages || rawBody.input;
     if (Array.isArray(messages)) {
         for (const msg of messages) {
             classifyMessage(msg, add);
         }
     }
-    else if (typeof messages === 'string') {
-        add('user_text', estimateTokens(messages));
+    else if (typeof messages === "string") {
+        add("user_text", estimateTokens(messages));
     }
     const total = Array.from(counts.values()).reduce((s, e) => s + e.tokens, 0);
     return buildCompositionArray(counts, total);
 }
 function classifyMessage(msg, add) {
-    const type = msg.type || '';
+    const type = msg.type || "";
     // OpenAI Responses API typed items (no role field)
     if (!msg.role && type) {
-        if (type === 'function_call' || type === 'custom_tool_call') {
-            add('tool_calls', estimateTokens(msg));
+        if (type === "function_call" || type === "custom_tool_call") {
+            add("tool_calls", estimateTokens(msg));
             return;
         }
-        if (type === 'function_call_output' || type === 'custom_tool_call_output') {
-            add('tool_results', estimateTokens(msg.output || ''));
+        if (type === "function_call_output" || type === "custom_tool_call_output") {
+            add("tool_results", estimateTokens(msg.output || ""));
             return;
         }
-        if (type === 'reasoning') {
-            add('thinking', estimateTokens(msg));
+        if (type === "reasoning") {
+            add("thinking", estimateTokens(msg));
             return;
         }
-        if (type === 'output_text') {
-            add('assistant_text', estimateTokens(msg.text || ''));
+        if (type === "output_text") {
+            add("assistant_text", estimateTokens(msg.text || ""));
             return;
         }
-        if (type === 'input_text') {
-            add('user_text', estimateTokens(msg.text || ''));
+        if (type === "input_text") {
+            add("user_text", estimateTokens(msg.text || ""));
             return;
         }
     }
-    const role = msg.role || 'user';
+    const role = msg.role || "user";
     const content = msg.content;
     // System / developer messages
-    if (role === 'system' || role === 'developer') {
-        add('system_prompt', estimateTokens(content));
+    if (role === "system" || role === "developer") {
+        add("system_prompt", estimateTokens(content));
         return;
     }
     // String content
-    if (typeof content === 'string') {
-        if (content.includes('<system-reminder>')) {
-            add('system_injections', estimateTokens(content));
+    if (typeof content === "string") {
+        if (content.includes("<system-reminder>")) {
+            add("system_injections", estimateTokens(content));
         }
-        else if (role === 'assistant') {
-            add('assistant_text', estimateTokens(content));
+        else if (role === "assistant") {
+            add("assistant_text", estimateTokens(content));
         }
         else {
-            add('user_text', estimateTokens(content));
+            add("user_text", estimateTokens(content));
+        }
+        return;
+    }
+    // Gemini parts array (role + parts instead of role + content)
+    if (msg.parts && Array.isArray(msg.parts)) {
+        for (const part of msg.parts) {
+            classifyGeminiPart(part, role, add);
         }
         return;
     }
@@ -130,52 +134,80 @@ function classifyMessage(msg, add) {
     }
     // Fallback
     if (content) {
-        add('other', estimateTokens(content));
+        add("other", estimateTokens(content));
     }
 }
 function classifyBlock(block, role, add) {
-    const type = block.type || '';
-    if (type === 'tool_use') {
-        add('tool_calls', estimateTokens(block));
+    const type = block.type || "";
+    if (type === "tool_use") {
+        add("tool_calls", estimateTokens(block));
         return;
     }
-    if (type === 'tool_result') {
-        add('tool_results', estimateTokens(block.content || ''));
+    if (type === "tool_result") {
+        add("tool_results", estimateTokens(block.content || ""));
         return;
     }
-    if (type === 'thinking') {
-        add('thinking', estimateTokens(block.thinking || block.text || ''));
+    if (type === "thinking") {
+        add("thinking", estimateTokens(block.thinking || block.text || ""));
         return;
     }
-    if (type === 'image' || type === 'image_url') {
-        add('images', estimateTokens(block));
+    if (type === "image" || type === "image_url") {
+        add("images", estimateTokens(block));
         return;
     }
     // Text blocks
-    const text = block.text || '';
-    if (type === 'text' || type === 'input_text' || !type) {
-        if (text.includes('<system-reminder>')) {
-            add('system_injections', estimateTokens(text));
+    const text = block.text || "";
+    if (type === "text" || type === "input_text" || !type) {
+        if (text.includes("<system-reminder>")) {
+            add("system_injections", estimateTokens(text));
         }
         else if (block.cache_control) {
-            add('cache_markers', estimateTokens(block.cache_control));
+            add("cache_markers", estimateTokens(block.cache_control));
             // Still count the text content in its natural category
-            if (role === 'assistant') {
-                add('assistant_text', estimateTokens(text));
+            if (role === "assistant") {
+                add("assistant_text", estimateTokens(text));
             }
             else {
-                add('user_text', estimateTokens(text));
+                add("user_text", estimateTokens(text));
             }
         }
-        else if (role === 'assistant') {
-            add('assistant_text', estimateTokens(text));
+        else if (role === "assistant") {
+            add("assistant_text", estimateTokens(text));
         }
         else {
-            add('user_text', estimateTokens(text));
+            add("user_text", estimateTokens(text));
         }
         return;
     }
-    add('other', estimateTokens(block));
+    add("other", estimateTokens(block));
+}
+function classifyGeminiPart(part, role, add) {
+    if (part.text) {
+        if (role === "model") {
+            add("assistant_text", estimateTokens(part.text));
+        }
+        else {
+            add("user_text", estimateTokens(part.text));
+        }
+        return;
+    }
+    if (part.functionCall) {
+        add("tool_calls", estimateTokens(part.functionCall));
+        return;
+    }
+    if (part.functionResponse) {
+        add("tool_results", estimateTokens(part.functionResponse));
+        return;
+    }
+    if (part.inlineData || part.fileData) {
+        add("images", estimateTokens(part));
+        return;
+    }
+    if (part.executableCode || part.codeExecutionResult) {
+        add("assistant_text", estimateTokens(part));
+        return;
+    }
+    add("other", estimateTokens(part));
 }
 function buildCompositionArray(counts, total) {
     const result = [];
@@ -205,8 +237,8 @@ export function parseResponseUsage(responseData) {
     };
     if (!responseData)
         return result;
-    // Streaming response — scan SSE chunks for usage
-    if (responseData.streaming && typeof responseData.chunks === 'string') {
+    // Streaming response: scan SSE chunks for usage
+    if (responseData.streaming && typeof responseData.chunks === "string") {
         result.stream = true;
         return parseStreamingUsage(responseData.chunks, result);
     }
@@ -218,7 +250,24 @@ export function parseResponseUsage(responseData) {
         result.cacheReadTokens = u.cache_read_input_tokens || 0;
         result.cacheWriteTokens = u.cache_creation_input_tokens || 0;
     }
-    result.model = responseData.model || null;
+    // Gemini usageMetadata (direct or inside Code Assist wrapper .response)
+    const geminiResp = responseData.usageMetadata
+        ? responseData
+        : responseData.response;
+    if (geminiResp?.usageMetadata) {
+        const u = geminiResp.usageMetadata;
+        result.inputTokens = u.promptTokenCount || 0;
+        result.outputTokens =
+            u.candidatesTokenCount ||
+                u.totalTokenCount - (u.promptTokenCount || 0) ||
+                0;
+        result.cacheReadTokens = u.cachedContentTokenCount || 0;
+    }
+    result.model =
+        responseData.model ||
+            responseData.modelVersion ||
+            geminiResp?.modelVersion ||
+            null;
     if (responseData.stop_reason) {
         result.finishReasons = [responseData.stop_reason];
     }
@@ -227,45 +276,76 @@ export function parseResponseUsage(responseData) {
             .map((c) => c.finish_reason)
             .filter(Boolean);
     }
+    else if (responseData.candidates &&
+        Array.isArray(responseData.candidates)) {
+        result.finishReasons = responseData.candidates
+            .map((c) => c.finishReason)
+            .filter(Boolean);
+    }
+    else if (geminiResp?.candidates && Array.isArray(geminiResp.candidates)) {
+        result.finishReasons = geminiResp.candidates
+            .map((c) => c.finishReason)
+            .filter(Boolean);
+    }
     return result;
 }
 function parseStreamingUsage(chunks, result) {
     // Parse SSE events looking for usage data
-    const lines = chunks.split('\n');
+    const lines = chunks.split("\n");
     for (const line of lines) {
-        if (!line.startsWith('data: '))
+        if (!line.startsWith("data: "))
             continue;
         const data = line.slice(6).trim();
-        if (data === '[DONE]')
+        if (data === "[DONE]")
             continue;
         try {
             const parsed = JSON.parse(data);
             // Anthropic message_start: contains model
-            if (parsed.type === 'message_start' && parsed.message) {
+            if (parsed.type === "message_start" && parsed.message) {
                 result.model = parsed.message.model || result.model;
                 if (parsed.message.usage) {
                     result.inputTokens = parsed.message.usage.input_tokens || 0;
-                    result.cacheReadTokens = parsed.message.usage.cache_read_input_tokens || 0;
-                    result.cacheWriteTokens = parsed.message.usage.cache_creation_input_tokens || 0;
+                    result.cacheReadTokens =
+                        parsed.message.usage.cache_read_input_tokens || 0;
+                    result.cacheWriteTokens =
+                        parsed.message.usage.cache_creation_input_tokens || 0;
                 }
             }
             // Anthropic message_delta: contains stop_reason and output token count
-            if (parsed.type === 'message_delta') {
+            if (parsed.type === "message_delta") {
                 if (parsed.delta?.stop_reason) {
                     result.finishReasons = [parsed.delta.stop_reason];
                 }
                 if (parsed.usage) {
-                    result.outputTokens = parsed.usage.output_tokens || result.outputTokens;
+                    result.outputTokens =
+                        parsed.usage.output_tokens || result.outputTokens;
                 }
             }
             // OpenAI streaming: final chunk with usage
             if (parsed.usage && parsed.choices) {
                 result.inputTokens = parsed.usage.prompt_tokens || result.inputTokens;
-                result.outputTokens = parsed.usage.completion_tokens || result.outputTokens;
+                result.outputTokens =
+                    parsed.usage.completion_tokens || result.outputTokens;
             }
             if (parsed.choices?.[0]?.finish_reason) {
                 result.finishReasons = [parsed.choices[0].finish_reason];
             }
+            // Gemini streaming: usageMetadata in chunks
+            if (parsed.usageMetadata) {
+                result.inputTokens =
+                    parsed.usageMetadata.promptTokenCount || result.inputTokens;
+                result.outputTokens =
+                    parsed.usageMetadata.candidatesTokenCount || result.outputTokens;
+                result.cacheReadTokens =
+                    parsed.usageMetadata.cachedContentTokenCount ||
+                        result.cacheReadTokens;
+            }
+            if (parsed.candidates?.[0]?.finishReason) {
+                result.finishReasons = [parsed.candidates[0].finishReason];
+            }
+            if (parsed.modelVersion) {
+                result.model = parsed.modelVersion;
+            }
             if (parsed.model) {
                 result.model = parsed.model;
             }
@@ -278,13 +358,13 @@ function parseStreamingUsage(chunks, result) {
 }
 // --- LHAR Record Builder ---
 function hexId(bytes) {
-    return randomBytes(bytes).toString('hex');
+    return randomBytes(bytes).toString("hex");
 }
 function traceIdFromConversation(conversationId) {
     if (!conversationId)
         return hexId(16);
     // Deterministic: hash the conversationId to a 32-hex-char trace ID
-    return createHash('sha256').update(conversationId).digest('hex').slice(0, 32);
+    return createHash("sha256").update(conversationId).digest("hex").slice(0, 32);
 }
 export function buildLharRecord(entry, prevEntries) {
     const ci = entry.contextInfo;
@@ -296,11 +376,11 @@ export function buildLharRecord(entry, prevEntries) {
     // Sequence + growth must be derived from a stable ordering.
     // Use oldest-first timestamp ordering within the conversation; tie-break by id.
     let convoEntries = entry.conversationId
-        ? prevEntries.filter(e => e.conversationId === entry.conversationId)
+        ? prevEntries.filter((e) => e.conversationId === entry.conversationId)
         : [entry];
-    // Make buildLharRecord robust even if the caller doesn't include `entry` in `prevEntries`.
+    // Make buildLharRecord work even if the caller doesn't include `entry` in `prevEntries`.
     if (entry.conversationId) {
-        const found = convoEntries.some(e => e.id === entry.id && e.timestamp === entry.timestamp);
+        const found = convoEntries.some((e) => e.id === entry.id && e.timestamp === entry.timestamp);
         if (!found)
             convoEntries = [...convoEntries, entry];
     }
@@ -310,9 +390,9 @@ export function buildLharRecord(entry, prevEntries) {
             return dt;
         return a.id - b.id;
     });
-    let convoIndex = convoEntries.findIndex(e => e.id === entry.id && e.timestamp === entry.timestamp);
+    let convoIndex = convoEntries.findIndex((e) => e.id === entry.id && e.timestamp === entry.timestamp);
     if (convoIndex < 0)
-        convoIndex = convoEntries.findIndex(e => e.id === entry.id);
+        convoIndex = convoEntries.findIndex((e) => e.id === entry.id);
     if (convoIndex < 0)
         convoIndex = 0;
     const sequence = convoIndex + 1;
@@ -322,18 +402,22 @@ export function buildLharRecord(entry, prevEntries) {
     const tokensAdded = prevEntry ? ci.totalTokens - prevTokens : null;
     const compactionDetected = tokensAdded !== null && tokensAdded < 0;
     // Agent role
-    const agentRole = entry.agentKey ? 'subagent' : 'main';
+    const agentRole = entry.agentKey ? "subagent" : "main";
     // Tokens per second
     let tokensPerSecond = null;
     if (entry.timings && entry.timings.receive_ms > 0 && usage.outputTokens > 0) {
-        tokensPerSecond = Math.round((usage.outputTokens / entry.timings.receive_ms) * 1000 * 10) / 10;
+        tokensPerSecond =
+            Math.round((usage.outputTokens / entry.timings.receive_ms) * 1000 * 10) /
+                10;
     }
-    const timings = entry.timings ? {
-        ...entry.timings,
-        tokens_per_second: tokensPerSecond,
-    } : null;
+    const timings = entry.timings
+        ? {
+            ...entry.timings,
+            tokens_per_second: tokensPerSecond,
+        }
+        : null;
     return {
-        type: 'entry',
+        type: "entry",
         id: randomUUID(),
         trace_id: traceIdFromConversation(entry.conversationId),
         span_id: hexId(8),
@@ -341,7 +425,7 @@ export function buildLharRecord(entry, prevEntries) {
         timestamp: entry.timestamp,
         sequence,
         source: {
-            tool: entry.source || 'unknown',
+            tool: entry.source || "unknown",
             tool_version: null,
             agent_role: agentRole,
             collector: COLLECTOR_NAME,
@@ -372,7 +456,7 @@ export function buildLharRecord(entry, prevEntries) {
             cost_usd: entry.costUsd,
         },
         http: {
-            method: 'POST',
+            method: "POST",
             url: entry.targetUrl,
             status_code: entry.httpStatus,
             api_format: ci.apiFormat,
@@ -410,7 +494,7 @@ export function buildLharRecord(entry, prevEntries) {
 // --- Session Line ---
 export function buildSessionLine(conversationId, conversation, model) {
     return {
-        type: 'session',
+        type: "session",
         trace_id: traceIdFromConversation(conversationId),
         started_at: conversation.firstSeen,
         tool: conversation.source,
@@ -437,11 +521,11 @@ export function toLharJsonl(entries, conversations) {
         }
         lines.push(JSON.stringify(record));
     }
-    return lines.join('\n') + '\n';
+    return `${lines.join("\n")}\n`;
 }
 export function toLharJson(entries, conversations) {
     const sorted = [...entries].sort((a, b) => new Date(a.timestamp).getTime() - new Date(b.timestamp).getTime());
-    const records = sorted.map(entry => buildLharRecord(entry, entries));
+    const records = sorted.map((entry) => buildLharRecord(entry, entries));
     // Build sessions from conversations map
     const sessions = [];
     const seenTraces = new Set();
@@ -449,7 +533,7 @@ export function toLharJson(entries, conversations) {
         if (!seenTraces.has(record.trace_id)) {
             seenTraces.add(record.trace_id);
             const convo = record.trace_id
-                ? Array.from(conversations.values()).find(c => traceIdFromConversation(c.id) === record.trace_id)
+                ? Array.from(conversations.values()).find((c) => traceIdFromConversation(c.id) === record.trace_id)
                 : undefined;
             sessions.push({
                 trace_id: record.trace_id,