@llmtracer/sdk 2.1.0 → 2.5.0

package/dist/cjs/transport.js

@@ -1,5 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+exports._testExports = void 0;
 exports.start = start;
 exports.enqueue = enqueue;
 exports.flush = flush;
@@ -9,18 +10,34 @@ const config_1 = require("./config");
 let eventQueue = [];
 let flushTimer = null;
 let firstBatch = true;
+let exitHandlersRegistered = false;
+// Retry state: tracks consecutive failures and next retry time
+let retryAttempt = 0;
+let backoffUntil = 0;
 function start() {
     // Reset state for fresh start
     firstBatch = true;
+    retryAttempt = 0;
+    backoffUntil = 0;
+    if (!config_1.config.skipExitHandlers) {
+        registerExitHandlers();
+    }
 }
 function enqueue(event) {
+    // sampleRate check: skip event if it doesn't pass the sampling threshold
+    if (config_1.config.sampleRate < 1.0) {
+        if (Math.random() >= config_1.config.sampleRate)
+            return;
+    }
+    if (eventQueue.length >= config_1.config.maxQueueSize)
+        eventQueue.shift();
     eventQueue.push(event);
     if (firstBatch) {
         firstBatch = false;
         // Flush immediately on first event
         setImmediate(() => doFlush());
-        // Start regular 5s interval
-        flushTimer = setInterval(() => doFlush(), 5000);
+        // Start regular interval using configured flushIntervalMs
+        flushTimer = setInterval(() => doFlush(), config_1.config.flushIntervalMs);
         flushTimer.unref();
     }
 }
@@ -34,16 +51,23 @@ function stop() {
     }
     eventQueue = [];
     firstBatch = true;
+    retryAttempt = 0;
+    backoffUntil = 0;
 }
 function getQueueLength() {
     return eventQueue.length;
 }
-async function doFlush() {
+async function doFlush(timeoutMs = 10000) {
     if (eventQueue.length === 0)
         return;
-    const events = eventQueue.splice(0);
+    // If we're in backoff, skip this flush cycle
+    if (backoffUntil > 0 && Date.now() < backoffUntil)
+        return;
+    // Take up to maxBatchSize events from the queue
+    const batchSize = Math.min(eventQueue.length, config_1.config.maxBatchSize);
+    const events = eventQueue.splice(0, batchSize);
     try {
-        const payload = JSON.stringify({ events });
+        const payload = JSON.stringify({ schema_version: '1.0', events });
         const response = await fetch(config_1.config.endpoint, {
             method: 'POST',
             headers: {
@@ -51,19 +75,101 @@ async function doFlush() {
                 'Authorization': `Bearer ${config_1.config.apiKey}`,
             },
             body: payload,
-            signal: AbortSignal.timeout(10000),
+            signal: AbortSignal.timeout(timeoutMs),
         });
         if (config_1.config.debug) {
             console.log(`[llmtracer] Flushed ${events.length} events \u2192 ${response.status}`);
         }
+        // Success — reset retry state
+        retryAttempt = 0;
+        backoffUntil = 0;
     }
     catch (e) {
         if (config_1.config.debug) {
             console.log(`[llmtracer] Flush failed: ${e}`);
         }
-        // Re-queue up to 100 events for retry
-        const requeue = events.slice(0, 100);
-        eventQueue.unshift(...requeue);
+        if (retryAttempt < config_1.config.maxRetries) {
+            // Re-queue events for retry (prepend so they go out first)
+            const requeue = events.slice(0, config_1.config.maxBatchSize);
+            eventQueue.unshift(...requeue);
+            // Set exponential backoff: delay = min(1000 * 2^attempt, 30000) + jitter(0-1000ms)
+            retryAttempt++;
+            const baseDelay = Math.min(1000 * Math.pow(2, retryAttempt), 30000);
+            const jitter = Math.floor(Math.random() * 1000);
+            backoffUntil = Date.now() + baseDelay + jitter;
+            if (config_1.config.debug) {
+                console.log(`[llmtracer] Retry ${retryAttempt}/${config_1.config.maxRetries} in ${baseDelay + jitter}ms`);
+            }
+        }
+        else {
+            // Max retries exceeded — drop the batch
+            if (config_1.config.debug) {
+                console.log(`[llmtracer] Dropping ${events.length} events after ${config_1.config.maxRetries} retries`);
+            }
+            retryAttempt = 0;
+            backoffUntil = 0;
+        }
+    }
+}
+function exitFlush() {
+    try {
+        if (eventQueue.length === 0)
+            return;
+        const events = eventQueue.splice(0);
+        const payload = JSON.stringify({ schema_version: '1.0', events });
+        // Best-effort flush with short timeout — never log, never throw
+        fetch(config_1.config.endpoint, {
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json',
+                'Authorization': `Bearer ${config_1.config.apiKey}`,
+            },
+            body: payload,
+            signal: AbortSignal.timeout(3000),
+        }).catch(() => { });
+    }
+    catch {
+        // crash-proof: swallow everything
+    }
+}
+function registerExitHandlers() {
+    if (exitHandlersRegistered)
+        return;
+    exitHandlersRegistered = true;
+    try {
+        process.on('beforeExit', () => {
+            try {
+                exitFlush();
+            }
+            catch { /* swallow */ }
+        });
+        process.on('SIGINT', () => {
+            try {
+                exitFlush();
+            }
+            catch { /* swallow */ }
+            process.exit(0);
+        });
+        process.on('SIGTERM', () => {
+            try {
+                exitFlush();
+            }
+            catch { /* swallow */ }
+            process.exit(0);
+        });
+    }
+    catch {
+        // crash-proof: if process.on fails, just skip
     }
 }
+// Export for testing only
+exports._testExports = {
+    exitFlush,
+    registerExitHandlers,
+    get exitHandlersRegistered() { return exitHandlersRegistered; },
+    resetExitFlag() { exitHandlersRegistered = false; },
+    get retryAttempt() { return retryAttempt; },
+    get backoffUntil() { return backoffUntil; },
+    resetRetryState() { retryAttempt = 0; backoffUntil = 0; },
+};
 //# sourceMappingURL=transport.js.map

package/dist/cjs/transport.js.map

@@ -1 +1 @@
-{"version":3,"file":"transport.js","sourceRoot":"","sources":["../../src/transport.ts"],"names":[],"mappings":";;AAMA,sBAGC;AAED,0BAWC;AAED,sBAEC;AAED,oBAOC;AAED,wCAEC;AAvCD,qCAAkC;AAElC,IAAI,UAAU,GAA0B,EAAE,CAAC;AAC3C,IAAI,UAAU,GAA0C,IAAI,CAAC;AAC7D,IAAI,UAAU,GAAG,IAAI,CAAC;AAEtB,SAAgB,KAAK;IACnB,8BAA8B;IAC9B,UAAU,GAAG,IAAI,CAAC;AACpB,CAAC;AAED,SAAgB,OAAO,CAAC,KAA0B;IAChD,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAEvB,IAAI,UAAU,EAAE,CAAC;QACf,UAAU,GAAG,KAAK,CAAC;QACnB,mCAAmC;QACnC,YAAY,CAAC,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;QAC9B,4BAA4B;QAC5B,UAAU,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC,OAAO,EAAE,EAAE,IAAI,CAAC,CAAC;QAChD,UAAU,CAAC,KAAK,EAAE,CAAC;IACrB,CAAC;AACH,CAAC;AAEM,KAAK,UAAU,KAAK;IACzB,MAAM,OAAO,EAAE,CAAC;AAClB,CAAC;AAED,SAAgB,IAAI;IAClB,IAAI,UAAU,EAAE,CAAC;QACf,aAAa,CAAC,UAAU,CAAC,CAAC;QAC1B,UAAU,GAAG,IAAI,CAAC;IACpB,CAAC;IACD,UAAU,GAAG,EAAE,CAAC;IAChB,UAAU,GAAG,IAAI,CAAC;AACpB,CAAC;AAED,SAAgB,cAAc;IAC5B,OAAO,UAAU,CAAC,MAAM,CAAC;AAC3B,CAAC;AAED,KAAK,UAAU,OAAO;IACpB,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO;IAEpC,MAAM,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;IAEpC,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;QAE3C,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,eAAM,CAAC,QAAQ,EAAE;YAC5C,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,eAAe,EAAE,UAAU,eAAM,CAAC,MAAM,EAAE;aAC3C;YACD,IAAI,EAAE,OAAO;YACb,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,KAAK,CAAC;SACnC,CAAC,CAAC;QAEH,IAAI,eAAM,CAAC,KAAK,EAAE,CAAC;YACjB,OAAO,CAAC,GAAG,CAAC,uBAAuB,MAAM,CAAC,MAAM,kBAAkB,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QACvF,CAAC;IACH,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,IAAI,eAAM,CAAC,KAAK,EAAE,CAAC;YACjB,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC,EAAE,CAAC,CAAC;QAChD,CAAC;QACD,sCAAsC;QACtC,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;QACrC,UAAU,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC,CAAC;IACjC,CAAC;AACH,CAAC"}
+{"version":3,"file":"transport.js","sourceRoot":"","sources":["../../src/transport.ts"],"names":[],"mappings":";;;AAWA,sBAQC;AAED,0BAiBC;AAED,sBAEC;AAED,oBASC;AAED,wCAEC;AAzDD,qCAAkC;AAElC,IAAI,UAAU,GAA0B,EAAE,CAAC;AAC3C,IAAI,UAAU,GAA0C,IAAI,CAAC;AAC7D,IAAI,UAAU,GAAG,IAAI,CAAC;AACtB,IAAI,sBAAsB,GAAG,KAAK,CAAC;AAEnC,+DAA+D;AAC/D,IAAI,YAAY,GAAG,CAAC,CAAC;AACrB,IAAI,YAAY,GAAG,CAAC,CAAC;AAErB,SAAgB,KAAK;IACnB,8BAA8B;IAC9B,UAAU,GAAG,IAAI,CAAC;IAClB,YAAY,GAAG,CAAC,CAAC;IACjB,YAAY,GAAG,CAAC,CAAC;IACjB,IAAI,CAAC,eAAM,CAAC,gBAAgB,EAAE,CAAC;QAC7B,oBAAoB,EAAE,CAAC;IACzB,CAAC;AACH,CAAC;AAED,SAAgB,OAAO,CAAC,KAA0B;IAChD,yEAAyE;IACzE,IAAI,eAAM,CAAC,UAAU,GAAG,GAAG,EAAE,CAAC;QAC5B,IAAI,IAAI,CAAC,MAAM,EAAE,IAAI,eAAM,CAAC,UAAU;YAAE,OAAO;IACjD,CAAC;IAED,IAAI,UAAU,CAAC,MAAM,IAAI,eAAM,CAAC,YAAY;QAAE,UAAU,CAAC,KAAK,EAAE,CAAC;IACjE,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAEvB,IAAI,UAAU,EAAE,CAAC;QACf,UAAU,GAAG,KAAK,CAAC;QACnB,mCAAmC;QACnC,YAAY,CAAC,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;QAC9B,0DAA0D;QAC1D,UAAU,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC,OAAO,EAAE,EAAE,eAAM,CAAC,eAAe,CAAC,CAAC;QAClE,UAAU,CAAC,KAAK,EAAE,CAAC;IACrB,CAAC;AACH,CAAC;AAEM,KAAK,UAAU,KAAK;IACzB,MAAM,OAAO,EAAE,CAAC;AAClB,CAAC;AAED,SAAgB,IAAI;IAClB,IAAI,UAAU,EAAE,CAAC;QACf,aAAa,CAAC,UAAU,CAAC,CAAC;QAC1B,UAAU,GAAG,IAAI,CAAC;IACpB,CAAC;IACD,UAAU,GAAG,EAAE,CAAC;IAChB,UAAU,GAAG,IAAI,CAAC;IAClB,YAAY,GAAG,CAAC,CAAC;IACjB,YAAY,GAAG,CAAC,CAAC;AACnB,CAAC;AAED,SAAgB,cAAc;IAC5B,OAAO,UAAU,CAAC,MAAM,CAAC;AAC3B,CAAC;AAED,KAAK,UAAU,OAAO,CAAC,YAAoB,KAAK;IAC9C,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO;IAEpC,6CAA6C;IAC7C,IAAI,YAAY,GAAG,CAAC,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,YAAY;QAAE,OAAO;IAE1D,gDAAgD;IAChD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,MAAM,EAAE,eAAM,CAAC,YAAY,CAAC,CAAC;IACnE,MAAM,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;IAE/C,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,cAAc,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;QAElE,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,eAAM,CAAC,QAAQ,EAAE;YAC5C,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,eAAe,EAAE,UAAU,eAAM,CAAC,MAAM,EAAE;aAC3C;YACD,IAAI,EAAE,OAAO;YACb,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,SAAS,CAAC;SACvC,CAAC,CAAC;QAEH,IAAI,eAAM,CAAC,KAAK,EAAE,CAAC;YACjB,OAAO,CAAC,GAAG,CAAC,uBAAuB,MAAM,CAAC,MAAM,kBAAkB,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QACvF,CAAC;QAED,8BAA8B;QAC9B,YAAY,GAAG,CAAC,CAAC;QACjB,YAAY,GAAG,CAAC,CAAC;IACnB,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,IAAI,eAAM,CAAC,KAAK,EAAE,CAAC;YACjB,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC,EAAE,CAAC,CAAC;QAChD,CAAC;QAED,IAAI,YAAY,GAAG,eAAM,CAAC,UAAU,EAAE,CAAC;YACrC,2DAA2D;YAC3D,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,eAAM,CAAC,YAAY,CAAC,CAAC;YACrD,UAAU,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC,CAAC;YAE/B,mFAAmF;YACnF,YAAY,EAAE,CAAC;YACf,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,YAAY,CAAC,EAAE,KAAK,CAAC,CAAC;YACpE,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC,CAAC;YAChD,YAAY,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,GAAG,MAAM,CAAC;YAE/C,IAAI,eAAM,CAAC,KAAK,EAAE,CAAC;gBACjB,OAAO,CAAC,GAAG,CAAC,qBAAqB,YAAY,IAAI,eAAM,CAAC,UAAU,OAAO,SAAS,GAAG,MAAM,IAAI,CAAC,CAAC;YACnG,CAAC;QACH,CAAC;aAAM,CAAC;YACN,wCAAwC;YACxC,IAAI,eAAM,CAAC,KAAK,EAAE,CAAC;gBACjB,OAAO,CAAC,GAAG,CAAC,wBAAwB,MAAM,CAAC,MAAM,iBAAiB,eAAM,CAAC,UAAU,UAAU,CAAC,CAAC;YACjG,CAAC;YACD,YAAY,GAAG,CAAC,CAAC;YACjB,YAAY,GAAG,CAAC,CAAC;QACnB,CAAC;IACH,CAAC;AACH,CAAC;AAED,SAAS,SAAS;IAChB,IAAI,CAAC;QACH,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAEpC,MAAM,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QACpC,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,cAAc,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;QAElE,gEAAgE;QAChE,KAAK,CAAC,eAAM,CAAC,QAAQ,EAAE;YACrB,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,eAAe,EAAE,UAAU,eAAM,CAAC,MAAM,EAAE;aAC3C;YACD,IAAI,EAAE,OAAO;YACb,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC;SAClC,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;IACrB,CAAC;IAAC,MAAM,CAAC;QACP,kCAAkC;IACpC,CAAC;AACH,CAAC;AAED,SAAS,oBAAoB;IAC3B,IAAI,sBAAsB;QAAE,OAAO;IACnC,sBAAsB,GAAG,IAAI,CAAC;IAE9B,IAAI,CAAC;QACH,OAAO,CAAC,EAAE,CAAC,YAAY,EAAE,GAAG,EAAE;YAC5B,IAAI,CAAC;gBAAC,SAAS,EAAE,CAAC;YAAC,CAAC;YAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE;YACxB,IAAI,CAAC;gBAAC,SAAS,EAAE,CAAC;YAAC,CAAC;YAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC;YAC5C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC,CAAC,CAAC;QAEH,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE;YACzB,IAAI,CAAC;gBAAC,SAAS,EAAE,CAAC;YAAC,CAAC;YAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC;YAC5C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC,CAAC,CAAC;IACL,CAAC;IAAC,MAAM,CAAC;QACP,8CAA8C;IAChD,CAAC;AACH,CAAC;AAED,0BAA0B;AACb,QAAA,YAAY,GAAG;IAC1B,SAAS;IACT,oBAAoB;IACpB,IAAI,sBAAsB,KAAK,OAAO,sBAAsB,CAAC,CAAC,CAAC;IAC/D,aAAa,KAAK,sBAAsB,GAAG,KAAK,CAAC,CAAC,CAAC;IACnD,IAAI,YAAY,KAAK,OAAO,YAAY,CAAC,CAAC,CAAC;IAC3C,IAAI,YAAY,KAAK,OAAO,YAAY,CAAC,CAAC,CAAC;IAC3C,eAAe,KAAK,YAAY,GAAG,CAAC,CAAC,CAAC,YAAY,GAAG,CAAC,CAAC,CAAC,CAAC;CAC1D,CAAC"}

package/dist/cjs/wrapper.d.ts

@@ -1,3 +1,3 @@
 import { ResponseData } from './schema';
-export declare function makeAsyncWrapper(originalFn: Function, provider: string, parseResponseFn: (result: any) => ResponseData, wrapStreamFn: (result: any, context: any, elapsedMs: number, args: any[]) => any): Function;
+export declare function makeAsyncWrapper(originalFn: Function, provider: string, parseResponseFn: (result: any, args?: any[]) => ResponseData, wrapStreamFn: ((result: any, context: any, elapsedMs: number, args: any[]) => any) | null): Function;
 //# sourceMappingURL=wrapper.d.ts.map

package/dist/cjs/wrapper.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"wrapper.d.ts","sourceRoot":"","sources":["../../src/wrapper.ts"],"names":[],"mappings":"AACA,OAAO,EAA0C,YAAY,EAAE,MAAM,UAAU,CAAC;AAGhF,wBAAgB,gBAAgB,CAC9B,UAAU,EAAE,QAAQ,EACpB,QAAQ,EAAE,MAAM,EAChB,eAAe,EAAE,CAAC,MAAM,EAAE,GAAG,KAAK,YAAY,EAC9C,YAAY,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,GAAG,EAAE,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,KAAK,GAAG,GAC/E,QAAQ,CAgDV"}
+{"version":3,"file":"wrapper.d.ts","sourceRoot":"","sources":["../../src/wrapper.ts"],"names":[],"mappings":"AACA,OAAO,EAA0C,YAAY,EAAE,MAAM,UAAU,CAAC;AAGhF,wBAAgB,gBAAgB,CAC9B,UAAU,EAAE,QAAQ,EACpB,QAAQ,EAAE,MAAM,EAChB,eAAe,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,IAAI,CAAC,EAAE,GAAG,EAAE,KAAK,YAAY,EAC5D,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,GAAG,EAAE,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,KAAK,GAAG,CAAC,GAAG,IAAI,GACxF,QAAQ,CA2DV"}

package/dist/cjs/wrapper.js

@@ -52,12 +52,23 @@ function makeAsyncWrapper(originalFn, provider, parseResponseFn, wrapStreamFn) {
         }
         // 2. Detect streaming
         const isStream = args[0]?.stream === true;
+        // 2b. For OpenAI streams, inject stream_options to get usage data
+        if (isStream && provider === 'openai') {
+            try {
+                if (!args[0].stream_options?.include_usage) {
+                    args[0] = { ...args[0], stream_options: { ...args[0].stream_options, include_usage: true } };
+                }
+            }
+            catch {
+                // swallow — never modify args if something goes wrong
+            }
+        }
         // 3. ALWAYS call original — never blocked, never modified
         const start = performance.now();
         const result = await originalFn.apply(this, args);
         const elapsedMs = Math.round(performance.now() - start);
         // 4. Handle streaming result
-        if (isStream && context) {
+        if (isStream && wrapStreamFn && context) {
             try {
                 return wrapStreamFn(result, context, elapsedMs, args);
             }
@@ -68,7 +79,7 @@ function makeAsyncWrapper(originalFn, provider, parseResponseFn, wrapStreamFn) {
         // 5. Handle non-streaming result
         if (context) {
             try {
-                const event = (0, schema_1.buildEvent)(context, parseResponseFn(result), elapsedMs);
+                const event = (0, schema_1.buildEvent)(context, parseResponseFn(result, args), elapsedMs);
                 transport.enqueue(event);
                 if (config_1.config.debug)
                     (0, schema_1.printCapture)(event);

package/dist/cjs/wrapper.js.map

@@ -1 +1 @@
-{"version":3,"file":"wrapper.js","sourceRoot":"","sources":["../../src/wrapper.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAIA,4CAqDC;AAzDD,qCAAkC;AAClC,qCAAgF;AAChF,uDAAyC;AAEzC,SAAgB,gBAAgB,CAC9B,UAAoB,EACpB,QAAgB,EAChB,eAA8C,EAC9C,YAAgF;IAEhF,MAAM,OAAO,GAAG,KAAK,WAAsB,GAAG,IAAW;QACvD,IAAI,CAAC,eAAM,CAAC,OAAO,EAAE,CAAC;YACpB,OAAO,UAAU,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QACtC,CAAC;QAED,sCAAsC;QACtC,IAAI,OAAO,GAAQ,IAAI,CAAC;QACxB,IAAI,CAAC;YACH,OAAO,GAAG,IAAA,qBAAY,EAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QACzC,CAAC;QAAC,MAAM,CAAC;YACP,qBAAqB;QACvB,CAAC;QAED,sBAAsB;QACtB,MAAM,QAAQ,GAAG,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,KAAK,IAAI,CAAC;QAE1C,0DAA0D;QAC1D,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QAChC,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QAClD,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,CAAC;QAExD,6BAA6B;QAC7B,IAAI,QAAQ,IAAI,OAAO,EAAE,CAAC;YACxB,IAAI,CAAC;gBACH,OAAO,YAAY,CAAC,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC;YACxD,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,MAAM,CAAC;YAChB,CAAC;QACH,CAAC;QAED,iCAAiC;QACjC,IAAI,OAAO,EAAE,CAAC;YACZ,IAAI,CAAC;gBACH,MAAM,KAAK,GAAG,IAAA,mBAAU,EAAC,OAAO,EAAE,eAAe,CAAC,MAAM,CAAC,EAAE,SAAS,CAAC,CAAC;gBACtE,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;gBACzB,IAAI,eAAM,CAAC,KAAK;oBAAE,IAAA,qBAAY,EAAC,KAAK,CAAC,CAAC;YACxC,CAAC;YAAC,MAAM,CAAC;gBACP,mBAAmB;YACrB,CAAC;QACH,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC;IAEF,uCAAuC;IACvC,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,UAAU,CAAC,IAAI,EAAE,CAAC,CAAC;IACnE,OAAO,OAAO,CAAC;AACjB,CAAC"}
+{"version":3,"file":"wrapper.js","sourceRoot":"","sources":["../../src/wrapper.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAIA,4CAgEC;AApED,qCAAkC;AAClC,qCAAgF;AAChF,uDAAyC;AAEzC,SAAgB,gBAAgB,CAC9B,UAAoB,EACpB,QAAgB,EAChB,eAA4D,EAC5D,YAAyF;IAEzF,MAAM,OAAO,GAAG,KAAK,WAAsB,GAAG,IAAW;QACvD,IAAI,CAAC,eAAM,CAAC,OAAO,EAAE,CAAC;YACpB,OAAO,UAAU,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QACtC,CAAC;QAED,sCAAsC;QACtC,IAAI,OAAO,GAAQ,IAAI,CAAC;QACxB,IAAI,CAAC;YACH,OAAO,GAAG,IAAA,qBAAY,EAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QACzC,CAAC;QAAC,MAAM,CAAC;YACP,qBAAqB;QACvB,CAAC;QAED,sBAAsB;QACtB,MAAM,QAAQ,GAAG,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,KAAK,IAAI,CAAC;QAE1C,kEAAkE;QAClE,IAAI,QAAQ,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;YACtC,IAAI,CAAC;gBACH,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,cAAc,EAAE,aAAa,EAAE,CAAC;oBAC3C,IAAI,CAAC,CAAC,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,EAAE,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,cAAc,EAAE,aAAa,EAAE,IAAI,EAAE,EAAE,CAAC;gBAC/F,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,sDAAsD;YACxD,CAAC;QACH,CAAC;QAED,0DAA0D;QAC1D,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QAChC,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QAClD,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,CAAC;QAExD,6BAA6B;QAC7B,IAAI,QAAQ,IAAI,YAAY,IAAI,OAAO,EAAE,CAAC;YACxC,IAAI,CAAC;gBACH,OAAO,YAAY,CAAC,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC;YACxD,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,MAAM,CAAC;YAChB,CAAC;QACH,CAAC;QAED,iCAAiC;QACjC,IAAI,OAAO,EAAE,CAAC;YACZ,IAAI,CAAC;gBACH,MAAM,KAAK,GAAG,IAAA,mBAAU,EAAC,OAAO,EAAE,eAAe,CAAC,MAAM,EAAE,IAAI,CAAC,EAAE,SAAS,CAAC,CAAC;gBAC5E,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;gBACzB,IAAI,eAAM,CAAC,KAAK;oBAAE,IAAA,qBAAY,EAAC,KAAK,CAAC,CAAC;YACxC,CAAC;YAAC,MAAM,CAAC;gBACP,mBAAmB;YACrB,CAAC;QACH,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC;IAEF,uCAAuC;IACvC,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,UAAU,CAAC,IAAI,EAAE,CAAC,CAAC;IACnE,OAAO,OAAO,CAAC;AACjB,CAAC"}

package/dist/esm/config.d.ts

@@ -1,7 +1,15 @@
 export declare const config: {
     apiKey: string;
+    sessionId: string;
     endpoint: string;
     debug: boolean;
     enabled: boolean;
+    appName: string;
+    skipExitHandlers: boolean;
+    maxBatchSize: number;
+    flushIntervalMs: number;
+    maxQueueSize: number;
+    maxRetries: number;
+    sampleRate: number;
 };
 //# sourceMappingURL=config.d.ts.map

package/dist/esm/config.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../../src/config.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,MAAM;;;;;CAKlB,CAAC"}
+{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../../src/config.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,MAAM;;;;;;;;;;;;;CAalB,CAAC"}

package/dist/esm/config.js

@@ -1,7 +1,15 @@
 export const config = {
     apiKey: '',
+    sessionId: '',
     endpoint: 'https://us-central1-llmtracer-alt.cloudfunctions.net/v1Events',
     debug: false,
     enabled: false,
+    appName: '',
+    skipExitHandlers: false,
+    maxBatchSize: 50,
+    flushIntervalMs: 5000,
+    maxQueueSize: 1000,
+    maxRetries: 3,
+    sampleRate: 1.0,
 };
 //# sourceMappingURL=config.js.map

package/dist/esm/config.js.map

@@ -1 +1 @@
-{"version":3,"file":"config.js","sourceRoot":"","sources":["../../src/config.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,MAAM,GAAG;IACpB,MAAM,EAAE,EAAE;IACV,QAAQ,EAAE,+DAA+D;IACzE,KAAK,EAAE,KAAK;IACZ,OAAO,EAAE,KAAK;CACf,CAAC"}
+{"version":3,"file":"config.js","sourceRoot":"","sources":["../../src/config.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,MAAM,GAAG;IACpB,MAAM,EAAE,EAAE;IACV,SAAS,EAAE,EAAE;IACb,QAAQ,EAAE,+DAA+D;IACzE,KAAK,EAAE,KAAK;IACZ,OAAO,EAAE,KAAK;IACd,OAAO,EAAE,EAAE;IACX,gBAAgB,EAAE,KAAK;IACvB,YAAY,EAAE,EAAE;IAChB,eAAe,EAAE,IAAI;IACrB,YAAY,EAAE,IAAI;IAClB,UAAU,EAAE,CAAC;IACb,UAAU,EAAE,GAAG;CAChB,CAAC"}

package/dist/esm/index.d.ts

@@ -1,12 +1,20 @@
 import { config } from './config';
 import * as transport from './transport';
 import * as patcher from './patcher';
-export declare const __version__ = "2.0.0";
-export declare function init(options?: {
+export declare const __version__ = "2.4.0";
+export interface InitOptions {
     apiKey?: string;
     debug?: boolean;
     endpoint?: string;
-}): void;
+    appName?: string;
+    skipExitHandlers?: boolean;
+    maxBatchSize?: number;
+    flushIntervalMs?: number;
+    maxQueueSize?: number;
+    maxRetries?: number;
+    sampleRate?: number;
+}
+export declare function init(options?: InitOptions): void;
 export declare function trace<T>(tags: Record<string, any>, fn: () => T | Promise<T>): T | Promise<T>;
 export declare function flush(): Promise<void>;
 export { patcher as _patcher, transport as _transport, config as _config };

package/dist/esm/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC;AAElC,OAAO,KAAK,SAAS,MAAM,aAAa,CAAC;AACzC,OAAO,KAAK,OAAO,MAAM,WAAW,CAAC;AAErC,eAAO,MAAM,WAAW,UAAU,CAAC;AAEnC,wBAAgB,IAAI,CAAC,OAAO,CAAC,EAAE;IAAE,MAAM,CAAC,EAAE,MAAM,CAAC;IAAC,KAAK,CAAC,EAAE,OAAO,CAAC;IAAC,QAAQ,CAAC,EAAE,MAAM,CAAA;CAAE,GAAG,IAAI,CAsB5F;AAED,wBAAgB,KAAK,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,EAAE,EAAE,MAAM,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAS5F;AAED,wBAAsB,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAM3C;AAGD,OAAO,EAAE,OAAO,IAAI,QAAQ,EAAE,SAAS,IAAI,UAAU,EAAE,MAAM,IAAI,OAAO,EAAE,CAAC;AAG3E,QAAA,MAAM,SAAS;;;;;CAAsC,CAAC;AACtD,eAAe,SAAS,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC;AAElC,OAAO,KAAK,SAAS,MAAM,aAAa,CAAC;AACzC,OAAO,KAAK,OAAO,MAAM,WAAW,CAAC;AAGrC,eAAO,MAAM,WAAW,UAAU,CAAC;AAEnC,MAAM,WAAW,WAAW;IAC1B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,gBAAgB,CAAC,EAAE,OAAO,CAAC;IAC3B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAUD,wBAAgB,IAAI,CAAC,OAAO,CAAC,EAAE,WAAW,GAAG,IAAI,CA6ChD;AAED,wBAAgB,KAAK,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,EAAE,EAAE,MAAM,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAS5F;AAED,wBAAsB,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAM3C;AAGD,OAAO,EAAE,OAAO,IAAI,QAAQ,EAAE,SAAS,IAAI,UAAU,EAAE,MAAM,IAAI,OAAO,EAAE,CAAC;AAG3E,QAAA,MAAM,SAAS;;;;;CAAsC,CAAC;AACtD,eAAe,SAAS,CAAC"}

package/dist/esm/index.js

@@ -2,13 +2,42 @@ import { config } from './config';
 import * as context from './context';
 import * as transport from './transport';
 import * as patcher from './patcher';
-export const __version__ = '2.0.0';
+import { randomUUID } from 'crypto';
+export const __version__ = '2.4.0';
+function clampWithWarning(name, value, min, max, defaultVal, debug) {
+    if (typeof value !== 'number' || isNaN(value) || value < min || value > max) {
+        if (debug)
+            console.log(`[llmtracer] ${name}=${value} out of range [${min}, ${max}]. Using default ${defaultVal}.`);
+        return defaultVal;
+    }
+    return value;
+}
 export function init(options) {
     try {
         config.apiKey = options?.apiKey || process.env.LLMTRACER_API_KEY || '';
         config.debug = options?.debug || false;
         config.endpoint = options?.endpoint || 'https://us-central1-llmtracer-alt.cloudfunctions.net/v1Events';
+        config.appName = options?.appName || process.env.LLMTRACER_APP_NAME || '';
+        config.skipExitHandlers = options?.skipExitHandlers || false;
         config.enabled = Boolean(config.apiKey);
+        // Validate and apply new config options
+        const d = config.debug;
+        config.maxBatchSize = options?.maxBatchSize !== undefined
+            ? clampWithWarning('maxBatchSize', options.maxBatchSize, 1, 500, 50, d) : 50;
+        config.flushIntervalMs = options?.flushIntervalMs !== undefined
+            ? clampWithWarning('flushIntervalMs', options.flushIntervalMs, 1000, 60000, 5000, d) : 5000;
+        config.maxQueueSize = options?.maxQueueSize !== undefined
+            ? clampWithWarning('maxQueueSize', options.maxQueueSize, 100, 10000, 1000, d) : 1000;
+        config.maxRetries = options?.maxRetries !== undefined
+            ? clampWithWarning('maxRetries', options.maxRetries, 0, 10, 3, d) : 3;
+        config.sampleRate = options?.sampleRate !== undefined
+            ? clampWithWarning('sampleRate', options.sampleRate, 0.0, 1.0, 1.0, d) : 1.0;
+        try {
+            config.sessionId = randomUUID();
+        }
+        catch {
+            config.sessionId = '';
+        }
         if (!config.enabled) {
             if (config.debug)
                 console.log('[llmtracer] No API key found. Tracing disabled.');
@@ -17,6 +46,9 @@ export function init(options) {
         transport.start();
         patcher.patchOpenAI(config.debug);
         patcher.patchAnthropic(config.debug);
+        patcher.patchGoogle(config.debug);
+        if (config.debug)
+            console.log(`[llmtracer] Session: ${config.sessionId}`);
         if (config.debug)
             console.log('[llmtracer] Ready. Events \u2192 dashboard.llmtracer.com');
     }

package/dist/esm/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC;AAClC,OAAO,KAAK,OAAO,MAAM,WAAW,CAAC;AACrC,OAAO,KAAK,SAAS,MAAM,aAAa,CAAC;AACzC,OAAO,KAAK,OAAO,MAAM,WAAW,CAAC;AAErC,MAAM,CAAC,MAAM,WAAW,GAAG,OAAO,CAAC;AAEnC,MAAM,UAAU,IAAI,CAAC,OAAiE;IACpF,IAAI,CAAC;QACH,MAAM,CAAC,MAAM,GAAG,OAAO,EAAE,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,iBAAiB,IAAI,EAAE,CAAC;QACvE,MAAM,CAAC,KAAK,GAAG,OAAO,EAAE,KAAK,IAAI,KAAK,CAAC;QACvC,MAAM,CAAC,QAAQ,GAAG,OAAO,EAAE,QAAQ,IAAI,+DAA+D,CAAC;QACvG,MAAM,CAAC,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAExC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACpB,IAAI,MAAM,CAAC,KAAK;gBAAE,OAAO,CAAC,GAAG,CAAC,iDAAiD,CAAC,CAAC;YACjF,OAAO;QACT,CAAC;QAED,SAAS,CAAC,KAAK,EAAE,CAAC;QAElB,OAAO,CAAC,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAClC,OAAO,CAAC,cAAc,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAErC,IAAI,MAAM,CAAC,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,0DAA0D,CAAC,CAAC;IAC5F,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,MAAM,CAAC,OAAO,GAAG,KAAK,CAAC;QACvB,IAAI,MAAM,CAAC,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,8BAA8B,CAAC,qBAAqB,CAAC,CAAC;IACtF,CAAC;AACH,CAAC;AAED,MAAM,UAAU,KAAK,CAAI,IAAyB,EAAE,EAAwB;IAC1E,IAAI,CAAC;QACH,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACpB,OAAO,EAAE,EAAE,CAAC;QACd,CAAC;QACD,OAAO,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;IAC/B,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,EAAE,CAAC;IACd,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,KAAK;IACzB,IAAI,CAAC;QACH,MAAM,SAAS,CAAC,KAAK,EAAE,CAAC;IAC1B,CAAC;IAAC,MAAM,CAAC;QACP,2BAA2B;IAC7B,CAAC;AACH,CAAC;AAED,4CAA4C;AAC5C,OAAO,EAAE,OAAO,IAAI,QAAQ,EAAE,SAAS,IAAI,UAAU,EAAE,MAAM,IAAI,OAAO,EAAE,CAAC;AAE3E,4DAA4D;AAC5D,MAAM,SAAS,GAAG,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC;AACtD,eAAe,SAAS,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC;AAClC,OAAO,KAAK,OAAO,MAAM,WAAW,CAAC;AACrC,OAAO,KAAK,SAAS,MAAM,aAAa,CAAC;AACzC,OAAO,KAAK,OAAO,MAAM,WAAW,CAAC;AACrC,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAEpC,MAAM,CAAC,MAAM,WAAW,GAAG,OAAO,CAAC;AAenC,SAAS,gBAAgB,CAAC,IAAY,EAAE,KAAa,EAAE,GAAW,EAAE,GAAW,EAAE,UAAkB,EAAE,KAAc;IACjH,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,KAAK,CAAC,IAAI,KAAK,GAAG,GAAG,IAAI,KAAK,GAAG,GAAG,EAAE,CAAC;QAC5E,IAAI,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,IAAI,KAAK,kBAAkB,GAAG,KAAK,GAAG,oBAAoB,UAAU,GAAG,CAAC,CAAC;QACnH,OAAO,UAAU,CAAC;IACpB,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,UAAU,IAAI,CAAC,OAAqB;IACxC,IAAI,CAAC;QACH,MAAM,CAAC,MAAM,GAAG,OAAO,EAAE,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,iBAAiB,IAAI,EAAE,CAAC;QACvE,MAAM,CAAC,KAAK,GAAG,OAAO,EAAE,KAAK,IAAI,KAAK,CAAC;QACvC,MAAM,CAAC,QAAQ,GAAG,OAAO,EAAE,QAAQ,IAAI,+DAA+D,CAAC;QACvG,MAAM,CAAC,OAAO,GAAG,OAAO,EAAE,OAAO,IAAI,OAAO,CAAC,GAAG,CAAC,kBAAkB,IAAI,EAAE,CAAC;QAC1E,MAAM,CAAC,gBAAgB,GAAG,OAAO,EAAE,gBAAgB,IAAI,KAAK,CAAC;QAC7D,MAAM,CAAC,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAExC,wCAAwC;QACxC,MAAM,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC;QACvB,MAAM,CAAC,YAAY,GAAG,OAAO,EAAE,YAAY,KAAK,SAAS;YACvD,CAAC,CAAC,gBAAgB,CAAC,cAAc,EAAE,OAAO,CAAC,YAAY,EAAE,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/E,MAAM,CAAC,eAAe,GAAG,OAAO,EAAE,eAAe,KAAK,SAAS;YAC7D,CAAC,CAAC,gBAAgB,CAAC,iBAAiB,EAAE,OAAO,CAAC,eAAe,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QAC9F,MAAM,CAAC,YAAY,GAAG,OAAO,EAAE,YAAY,KAAK,SAAS;YACvD,CAAC,CAAC,gBAAgB,CAAC,cAAc,EAAE,OAAO,CAAC,YAAY,EAAE,GAAG,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QACvF,MAAM,CAAC,UAAU,GAAG,OAAO,EAAE,UAAU,KAAK,SAAS;YACnD,CAAC,CAAC,gBAAgB,CAAC,YAAY,EAAE,OAAO,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACxE,MAAM,CAAC,UAAU,GAAG,OAAO,EAAE,UAAU,KAAK,SAAS;YACnD,CAAC,CAAC,gBAAgB,CAAC,YAAY,EAAE,OAAO,CAAC,UAAU,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC;QAE/E,IAAI,CAAC;YACH,MAAM,CAAC,SAAS,GAAG,UAAU,EAAE,CAAC;QAClC,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,CAAC,SAAS,GAAG,EAAE,CAAC;QACxB,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACpB,IAAI,MAAM,CAAC,KAAK;gBAAE,OAAO,CAAC,GAAG,CAAC,iDAAiD,CAAC,CAAC;YACjF,OAAO;QACT,CAAC;QAED,SAAS,CAAC,KAAK,EAAE,CAAC;QAElB,OAAO,CAAC,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAClC,OAAO,CAAC,cAAc,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACrC,OAAO,CAAC,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAElC,IAAI,MAAM,CAAC,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,wBAAwB,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC;QAC1E,IAAI,MAAM,CAAC,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,0DAA0D,CAAC,CAAC;IAC5F,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,MAAM,CAAC,OAAO,GAAG,KAAK,CAAC;QACvB,IAAI,MAAM,CAAC,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,8BAA8B,CAAC,qBAAqB,CAAC,CAAC;IACtF,CAAC;AACH,CAAC;AAED,MAAM,UAAU,KAAK,CAAI,IAAyB,EAAE,EAAwB;IAC1E,IAAI,CAAC;QACH,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACpB,OAAO,EAAE,EAAE,CAAC;QACd,CAAC;QACD,OAAO,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;IAC/B,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,EAAE,CAAC;IACd,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,KAAK;IACzB,IAAI,CAAC;QACH,MAAM,SAAS,CAAC,KAAK,EAAE,CAAC;IAC1B,CAAC;IAAC,MAAM,CAAC;QACP,2BAA2B;IAC7B,CAAC;AACH,CAAC;AAED,4CAA4C;AAC5C,OAAO,EAAE,OAAO,IAAI,QAAQ,EAAE,SAAS,IAAI,UAAU,EAAE,MAAM,IAAI,OAAO,EAAE,CAAC;AAE3E,4DAA4D;AAC5D,MAAM,SAAS,GAAG,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC;AACtD,eAAe,SAAS,CAAC"}

package/dist/esm/patcher.d.ts

@@ -1,4 +1,5 @@
 export declare function patchOpenAI(debug: boolean): void;
 export declare function patchAnthropic(debug: boolean): void;
+export declare function patchGoogle(debug: boolean): void;
 export declare function unpatchAll(): void;
 //# sourceMappingURL=patcher.d.ts.map

package/dist/esm/patcher.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"patcher.d.ts","sourceRoot":"","sources":["../../src/patcher.ts"],"names":[],"mappings":"AA4BA,wBAAgB,WAAW,CAAC,KAAK,EAAE,OAAO,GAAG,IAAI,CAyBhD;AAED,wBAAgB,cAAc,CAAC,KAAK,EAAE,OAAO,GAAG,IAAI,CAyBnD;AAED,wBAAgB,UAAU,IAAI,IAAI,CAMjC"}
+{"version":3,"file":"patcher.d.ts","sourceRoot":"","sources":["../../src/patcher.ts"],"names":[],"mappings":"AA6BA,wBAAgB,WAAW,CAAC,KAAK,EAAE,OAAO,GAAG,IAAI,CA6FhD;AAED,wBAAgB,cAAc,CAAC,KAAK,EAAE,OAAO,GAAG,IAAI,CAyBnD;AAED,wBAAgB,WAAW,CAAC,KAAK,EAAE,OAAO,GAAG,IAAI,CAqChD;AAED,wBAAgB,UAAU,IAAI,IAAI,CAMjC"}

package/dist/esm/patcher.js

@@ -1,6 +1,7 @@
 import { makeAsyncWrapper } from './wrapper';
-import { parseOpenAIResponse, wrapOpenAIStream } from './providers/openai';
+import { parseOpenAIResponse, wrapOpenAIStream, parseTranscriptionResponse, parseSpeechResponse, parseEmbeddingResponse, parseImageResponse, parseModerationResponse } from './providers/openai';
 import { parseAnthropicResponse, wrapAnthropicStream } from './providers/anthropic';
+import { parseGoogleResponse, wrapGoogleStream } from './providers/google';
 const originals = new Map();
 function getPackageVersion(packageName) {
     try {
@@ -42,9 +43,76 @@ export function patchOpenAI(debug) {
             methodName: 'create',
             original: originalCreate,
         });
+        // ── Multi-modal endpoints ──────────────────────────────────
+        const capabilities = ['chat'];
+        // STT (Whisper)
+        try {
+            const sttModule = require('openai/resources/audio/transcriptions');
+            const TranscriptionsProto = sttModule.Transcriptions?.prototype;
+            if (TranscriptionsProto && !TranscriptionsProto._llmtracer_patched) {
+                const orig = TranscriptionsProto.create;
+                TranscriptionsProto.create = makeAsyncWrapper(orig, 'openai', parseTranscriptionResponse, null);
+                TranscriptionsProto._llmtracer_patched = true;
+                originals.set('openai.Transcriptions.create', { target: TranscriptionsProto, methodName: 'create', original: orig });
+                capabilities.push('stt');
+            }
+        }
+        catch { /* module not available */ }
+        // TTS
+        try {
+            const ttsModule = require('openai/resources/audio/speech');
+            const SpeechProto = ttsModule.Speech?.prototype;
+            if (SpeechProto && !SpeechProto._llmtracer_patched) {
+                const orig = SpeechProto.create;
+                SpeechProto.create = makeAsyncWrapper(orig, 'openai', parseSpeechResponse, null);
+                SpeechProto._llmtracer_patched = true;
+                originals.set('openai.Speech.create', { target: SpeechProto, methodName: 'create', original: orig });
+                capabilities.push('tts');
+            }
+        }
+        catch { /* module not available */ }
+        // Embeddings
+        try {
+            const embedModule = require('openai/resources/embeddings');
+            const EmbeddingsProto = embedModule.Embeddings?.prototype;
+            if (EmbeddingsProto && !EmbeddingsProto._llmtracer_patched) {
+                const orig = EmbeddingsProto.create;
+                EmbeddingsProto.create = makeAsyncWrapper(orig, 'openai', parseEmbeddingResponse, null);
+                EmbeddingsProto._llmtracer_patched = true;
+                originals.set('openai.Embeddings.create', { target: EmbeddingsProto, methodName: 'create', original: orig });
+                capabilities.push('embeddings');
+            }
+        }
+        catch { /* module not available */ }
+        // Images (DALL-E)
+        try {
+            const imgModule = require('openai/resources/images');
+            const ImagesProto = imgModule.Images?.prototype;
+            if (ImagesProto && !ImagesProto._llmtracer_patched) {
+                const orig = ImagesProto.generate;
+                ImagesProto.generate = makeAsyncWrapper(orig, 'openai', parseImageResponse, null);
+                ImagesProto._llmtracer_patched = true;
+                originals.set('openai.Images.generate', { target: ImagesProto, methodName: 'generate', original: orig });
+                capabilities.push('images');
+            }
+        }
+        catch { /* module not available */ }
+        // Moderations
+        try {
+            const modModule = require('openai/resources/moderations');
+            const ModerationsProto = modModule.Moderations?.prototype;
+            if (ModerationsProto && !ModerationsProto._llmtracer_patched) {
+                const orig = ModerationsProto.create;
+                ModerationsProto.create = makeAsyncWrapper(orig, 'openai', parseModerationResponse, null);
+                ModerationsProto._llmtracer_patched = true;
+                originals.set('openai.Moderations.create', { target: ModerationsProto, methodName: 'create', original: orig });
+                capabilities.push('moderation');
+            }
+        }
+        catch { /* module not available */ }
         if (debug) {
             const version = getPackageVersion('openai');
-            console.log(`[llmtracer] \u2713 Patched openai (v${version})`);
+            console.log(`[llmtracer] \u2713 Patched openai (v${version}) \u2014 ${capabilities.join(' + ')}`);
         }
     }
     catch (e) {
@@ -78,6 +146,42 @@ export function patchAnthropic(debug) {
             console.log(`[llmtracer] \u2717 anthropic not found or patch failed: ${e}`);
     }
 }
+export function patchGoogle(debug) {
+    try {
+        const googleModule = require('@google/genai');
+        const GenModel = googleModule.GenerativeModel?.prototype;
+        if (!GenModel)
+            throw new Error('GenerativeModel prototype not found');
+        if (GenModel._llmtracer_patched)
+            return;
+        const originalGenerate = GenModel.generateContent;
+        GenModel.generateContent = makeAsyncWrapper(originalGenerate, 'google', parseGoogleResponse, wrapGoogleStream);
+        GenModel._llmtracer_patched = true;
+        originals.set('google.GenerativeModel.generateContent', {
+            target: GenModel,
+            methodName: 'generateContent',
+            original: originalGenerate,
+        });
+        // Also patch generateContentStream if it exists
+        if (typeof GenModel.generateContentStream === 'function') {
+            const originalStream = GenModel.generateContentStream;
+            GenModel.generateContentStream = makeAsyncWrapper(originalStream, 'google', parseGoogleResponse, wrapGoogleStream);
+            originals.set('google.GenerativeModel.generateContentStream', {
+                target: GenModel,
+                methodName: 'generateContentStream',
+                original: originalStream,
+            });
+        }
+        if (debug) {
+            const version = getPackageVersion('@google/genai');
+            console.log(`[llmtracer] \u2713 Patched google (v${version})`);
+        }
+    }
+    catch (e) {
+        if (debug)
+            console.log(`[llmtracer] \u2717 google not found or patch failed: ${e}`);
+    }
+}
 export function unpatchAll() {
     for (const [, entry] of originals) {
         entry.target[entry.methodName] = entry.original;

package/dist/esm/patcher.js.map

@@ -1 +1 @@
-{"version":3,"file":"patcher.js","sourceRoot":"","sources":["../../src/patcher.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,WAAW,CAAC;AAC7C,OAAO,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAC3E,OAAO,EAAE,sBAAsB,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AAEpF,MAAM,SAAS,GAAyE,IAAI,GAAG,EAAE,CAAC;AAElG,SAAS,iBAAiB,CAAC,WAAmB;IAC5C,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;QAC7B,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;QACzB,MAAM,UAAU,GAAG,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QAChD,IAAI,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QACnC,OAAO,GAAG,KAAK,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;YACjC,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,cAAc,CAAC,CAAC;YAC/C,IAAI,CAAC;gBACH,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC;gBACzD,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW;oBAAE,OAAO,GAAG,CAAC,OAAO,IAAI,SAAS,CAAC;YAChE,CAAC;YAAC,MAAM,CAAC;gBACP,wCAAwC;YAC1C,CAAC;YACD,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,SAAS;IACX,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,MAAM,UAAU,WAAW,CAAC,KAAc;IACxC,IAAI,CAAC;QACH,MAAM,iBAAiB,GAAG,OAAO,CAAC,mCAAmC,CAAC,CAAC;QACvE,MAAM,gBAAgB,GAAG,iBAAiB,CAAC,WAAW,EAAE,SAAS,CAAC;QAElE,IAAI,CAAC,gBAAgB;YAAE,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;QAC1E,IAAK,gBAAwB,CAAC,kBAAkB;YAAE,OAAO;QAEzD,MAAM,cAAc,GAAG,gBAAgB,CAAC,MAAM,CAAC;QAC/C,gBAAgB,CAAC,MAAM,GAAG,gBAAgB,CAAC,cAAc,EAAE,QAAQ,EAAE,mBAAmB,EAAE,gBAAgB,CAAC,CAAC;QAC3G,gBAAwB,CAAC,kBAAkB,GAAG,IAAI,CAAC;QAEpD,SAAS,CAAC,GAAG,CAAC,2BAA2B,EAAE;YACzC,MAAM,EAAE,gBAAgB;YACxB,UAAU,EAAE,QAAQ;YACpB,QAAQ,EAAE,cAAc;SACzB,CAAC,CAAC;QAEH,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,OAAO,GAAG,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YAC5C,OAAO,CAAC,GAAG,CAAC,uCAAuC,OAAO,GAAG,CAAC,CAAC;QACjE,CAAC;IACH,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,IAAI,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,wDAAwD,CAAC,EAAE,CAAC,CAAC;IACtF,CAAC;AACH,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,KAAc;IAC3C,IAAI,CAAC;QACH,MAAM,cAAc,GAAG,OAAO,CAAC,sCAAsC,CAAC,CAAC;QACvE,MAAM,aAAa,GAAG,cAAc,CAAC,QAAQ,EAAE,SAAS,CAAC;QAEzD,IAAI,CAAC,aAAa;YAAE,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;QACpE,IAAK,aAAqB,CAAC,kBAAkB;YAAE,OAAO;QAEtD,MAAM,cAAc,GAAG,aAAa,CAAC,MAAM,CAAC;QAC5C,aAAa,CAAC,MAAM,GAAG,gBAAgB,CAAC,cAAc,EAAE,WAAW,EAAE,sBAAsB,EAAE,mBAAmB,CAAC,CAAC;QACjH,aAAqB,CAAC,kBAAkB,GAAG,IAAI,CAAC;QAEjD,SAAS,CAAC,GAAG,CAAC,2BAA2B,EAAE;YACzC,MAAM,EAAE,aAAa;YACrB,UAAU,EAAE,QAAQ;YACpB,QAAQ,EAAE,cAAc;SACzB,CAAC,CAAC;QAEH,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,OAAO,GAAG,iBAAiB,CAAC,mBAAmB,CAAC,CAAC;YACvD,OAAO,CAAC,GAAG,CAAC,0CAA0C,OAAO,GAAG,CAAC,CAAC;QACpE,CAAC;IACH,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,IAAI,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,2DAA2D,CAAC,EAAE,CAAC,CAAC;IACzF,CAAC;AACH,CAAC;AAED,MAAM,UAAU,UAAU;IACxB,KAAK,MAAM,CAAC,EAAE,KAAK,CAAC,IAAI,SAAS,EAAE,CAAC;QAClC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,GAAG,KAAK,CAAC,QAAQ,CAAC;QAChD,OAAO,KAAK,CAAC,MAAM,CAAC,kBAAkB,CAAC;IACzC,CAAC;IACD,SAAS,CAAC,KAAK,EAAE,CAAC;AACpB,CAAC"}
+{"version":3,"file":"patcher.js","sourceRoot":"","sources":["../../src/patcher.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,WAAW,CAAC;AAC7C,OAAO,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,0BAA0B,EAAE,mBAAmB,EAAE,sBAAsB,EAAE,kBAAkB,EAAE,uBAAuB,EAAE,MAAM,oBAAoB,CAAC;AACjM,OAAO,EAAE,sBAAsB,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AACpF,OAAO,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAE3E,MAAM,SAAS,GAAyE,IAAI,GAAG,EAAE,CAAC;AAElG,SAAS,iBAAiB,CAAC,WAAmB;IAC5C,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;QAC7B,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;QACzB,MAAM,UAAU,GAAG,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QAChD,IAAI,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QACnC,OAAO,GAAG,KAAK,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;YACjC,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,cAAc,CAAC,CAAC;YAC/C,IAAI,CAAC;gBACH,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC;gBACzD,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW;oBAAE,OAAO,GAAG,CAAC,OAAO,IAAI,SAAS,CAAC;YAChE,CAAC;YAAC,MAAM,CAAC;gBACP,wCAAwC;YAC1C,CAAC;YACD,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,SAAS;IACX,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,MAAM,UAAU,WAAW,CAAC,KAAc;IACxC,IAAI,CAAC;QACH,MAAM,iBAAiB,GAAG,OAAO,CAAC,mCAAmC,CAAC,CAAC;QACvE,MAAM,gBAAgB,GAAG,iBAAiB,CAAC,WAAW,EAAE,SAAS,CAAC;QAElE,IAAI,CAAC,gBAAgB;YAAE,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;QAC1E,IAAK,gBAAwB,CAAC,kBAAkB;YAAE,OAAO;QAEzD,MAAM,cAAc,GAAG,gBAAgB,CAAC,MAAM,CAAC;QAC/C,gBAAgB,CAAC,MAAM,GAAG,gBAAgB,CAAC,cAAc,EAAE,QAAQ,EAAE,mBAAmB,EAAE,gBAAgB,CAAC,CAAC;QAC3G,gBAAwB,CAAC,kBAAkB,GAAG,IAAI,CAAC;QAEpD,SAAS,CAAC,GAAG,CAAC,2BAA2B,EAAE;YACzC,MAAM,EAAE,gBAAgB;YACxB,UAAU,EAAE,QAAQ;YACpB,QAAQ,EAAE,cAAc;SACzB,CAAC,CAAC;QAEH,8DAA8D;QAC9D,MAAM,YAAY,GAAG,CAAC,MAAM,CAAC,CAAC;QAE9B,gBAAgB;QAChB,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,OAAO,CAAC,uCAAuC,CAAC,CAAC;YACnE,MAAM,mBAAmB,GAAG,SAAS,CAAC,cAAc,EAAE,SAAS,CAAC;YAChE,IAAI,mBAAmB,IAAI,CAAE,mBAA2B,CAAC,kBAAkB,EAAE,CAAC;gBAC5E,MAAM,IAAI,GAAG,mBAAmB,CAAC,MAAM,CAAC;gBACxC,mBAAmB,CAAC,MAAM,GAAG,gBAAgB,CAAC,IAAI,EAAE,QAAQ,EAAE,0BAA0B,EAAE,IAAI,CAAC,CAAC;gBAC/F,mBAA2B,CAAC,kBAAkB,GAAG,IAAI,CAAC;gBACvD,SAAS,CAAC,GAAG,CAAC,8BAA8B,EAAE,EAAE,MAAM,EAAE,mBAAmB,EAAE,UAAU,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;gBACrH,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC3B,CAAC;QACH,CAAC;QAAC,MAAM,CAAC,CAAC,0BAA0B,CAAC,CAAC;QAEtC,MAAM;QACN,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,OAAO,CAAC,+BAA+B,CAAC,CAAC;YAC3D,MAAM,WAAW,GAAG,SAAS,CAAC,MAAM,EAAE,SAAS,CAAC;YAChD,IAAI,WAAW,IAAI,CAAE,WAAmB,CAAC,kBAAkB,EAAE,CAAC;gBAC5D,MAAM,IAAI,GAAG,WAAW,CAAC,MAAM,CAAC;gBAChC,WAAW,CAAC,MAAM,GAAG,gBAAgB,CAAC,IAAI,EAAE,QAAQ,EAAE,mBAAmB,EAAE,IAAI,CAAC,CAAC;gBAChF,WAAmB,CAAC,kBAAkB,GAAG,IAAI,CAAC;gBAC/C,SAAS,CAAC,GAAG,CAAC,sBAAsB,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,UAAU,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;gBACrG,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC3B,CAAC;QACH,CAAC;QAAC,MAAM,CAAC,CAAC,0BAA0B,CAAC,CAAC;QAEtC,aAAa;QACb,IAAI,CAAC;YACH,MAAM,WAAW,GAAG,OAAO,CAAC,6BAA6B,CAAC,CAAC;YAC3D,MAAM,eAAe,GAAG,WAAW,CAAC,UAAU,EAAE,SAAS,CAAC;YAC1D,IAAI,eAAe,IAAI,CAAE,eAAuB,CAAC,kBAAkB,EAAE,CAAC;gBACpE,MAAM,IAAI,GAAG,eAAe,CAAC,MAAM,CAAC;gBACpC,eAAe,CAAC,MAAM,GAAG,gBAAgB,CAAC,IAAI,EAAE,QAAQ,EAAE,sBAAsB,EAAE,IAAI,CAAC,CAAC;gBACvF,eAAuB,CAAC,kBAAkB,GAAG,IAAI,CAAC;gBACnD,SAAS,CAAC,GAAG,CAAC,0BAA0B,EAAE,EAAE,MAAM,EAAE,eAAe,EAAE,UAAU,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;gBAC7G,YAAY,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YAClC,CAAC;QACH,CAAC;QAAC,MAAM,CAAC,CAAC,0BAA0B,CAAC,CAAC;QAEtC,kBAAkB;QAClB,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,OAAO,CAAC,yBAAyB,CAAC,CAAC;YACrD,MAAM,WAAW,GAAG,SAAS,CAAC,MAAM,EAAE,SAAS,CAAC;YAChD,IAAI,WAAW,IAAI,CAAE,WAAmB,CAAC,kBAAkB,EAAE,CAAC;gBAC5D,MAAM,IAAI,GAAG,WAAW,CAAC,QAAQ,CAAC;gBAClC,WAAW,CAAC,QAAQ,GAAG,gBAAgB,CAAC,IAAI,EAAE,QAAQ,EAAE,kBAAkB,EAAE,IAAI,CAAC,CAAC;gBACjF,WAAmB,CAAC,kBAAkB,GAAG,IAAI,CAAC;gBAC/C,SAAS,CAAC,GAAG,CAAC,wBAAwB,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,UAAU,EAAE,UAAU,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;gBACzG,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC9B,CAAC;QACH,CAAC;QAAC,MAAM,CAAC,CAAC,0BAA0B,CAAC,CAAC;QAEtC,cAAc;QACd,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,OAAO,CAAC,8BAA8B,CAAC,CAAC;YAC1D,MAAM,gBAAgB,GAAG,SAAS,CAAC,WAAW,EAAE,SAAS,CAAC;YAC1D,IAAI,gBAAgB,IAAI,CAAE,gBAAwB,CAAC,kBAAkB,EAAE,CAAC;gBACtE,MAAM,IAAI,GAAG,gBAAgB,CAAC,MAAM,CAAC;gBACrC,gBAAgB,CAAC,MAAM,GAAG,gBAAgB,CAAC,IAAI,EAAE,QAAQ,EAAE,uBAAuB,EAAE,IAAI,CAAC,CAAC;gBACzF,gBAAwB,CAAC,kBAAkB,GAAG,IAAI,CAAC;gBACpD,SAAS,CAAC,GAAG,CAAC,2BAA2B,EAAE,EAAE,MAAM,EAAE,gBAAgB,EAAE,UAAU,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;gBAC/G,YAAY,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YAClC,CAAC;QACH,CAAC;QAAC,MAAM,CAAC,CAAC,0BAA0B,CAAC,CAAC;QAEtC,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,OAAO,GAAG,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YAC5C,OAAO,CAAC,GAAG,CAAC,uCAAuC,OAAO,YAAY,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACpG,CAAC;IACH,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,IAAI,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,wDAAwD,CAAC,EAAE,CAAC,CAAC;IACtF,CAAC;AACH,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,KAAc;IAC3C,IAAI,CAAC;QACH,MAAM,cAAc,GAAG,OAAO,CAAC,sCAAsC,CAAC,CAAC;QACvE,MAAM,aAAa,GAAG,cAAc,CAAC,QAAQ,EAAE,SAAS,CAAC;QAEzD,IAAI,CAAC,aAAa;YAAE,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;QACpE,IAAK,aAAqB,CAAC,kBAAkB;YAAE,OAAO;QAEtD,MAAM,cAAc,GAAG,aAAa,CAAC,MAAM,CAAC;QAC5C,aAAa,CAAC,MAAM,GAAG,gBAAgB,CAAC,cAAc,EAAE,WAAW,EAAE,sBAAsB,EAAE,mBAAmB,CAAC,CAAC;QACjH,aAAqB,CAAC,kBAAkB,GAAG,IAAI,CAAC;QAEjD,SAAS,CAAC,GAAG,CAAC,2BAA2B,EAAE;YACzC,MAAM,EAAE,aAAa;YACrB,UAAU,EAAE,QAAQ;YACpB,QAAQ,EAAE,cAAc;SACzB,CAAC,CAAC;QAEH,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,OAAO,GAAG,iBAAiB,CAAC,mBAAmB,CAAC,CAAC;YACvD,OAAO,CAAC,GAAG,CAAC,0CAA0C,OAAO,GAAG,CAAC,CAAC;QACpE,CAAC;IACH,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,IAAI,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,2DAA2D,CAAC,EAAE,CAAC,CAAC;IACzF,CAAC;AACH,CAAC;AAED,MAAM,UAAU,WAAW,CAAC,KAAc;IACxC,IAAI,CAAC;QACH,MAAM,YAAY,GAAG,OAAO,CAAC,eAAe,CAAC,CAAC;QAC9C,MAAM,QAAQ,GAAG,YAAY,CAAC,eAAe,EAAE,SAAS,CAAC;QAEzD,IAAI,CAAC,QAAQ;YAAE,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;QACtE,IAAK,QAAgB,CAAC,kBAAkB;YAAE,OAAO;QAEjD,MAAM,gBAAgB,GAAG,QAAQ,CAAC,eAAe,CAAC;QAClD,QAAQ,CAAC,eAAe,GAAG,gBAAgB,CAAC,gBAAgB,EAAE,QAAQ,EAAE,mBAAmB,EAAE,gBAAgB,CAAC,CAAC;QAC9G,QAAgB,CAAC,kBAAkB,GAAG,IAAI,CAAC;QAE5C,SAAS,CAAC,GAAG,CAAC,wCAAwC,EAAE;YACtD,MAAM,EAAE,QAAQ;YAChB,UAAU,EAAE,iBAAiB;YAC7B,QAAQ,EAAE,gBAAgB;SAC3B,CAAC,CAAC;QAEH,gDAAgD;QAChD,IAAI,OAAO,QAAQ,CAAC,qBAAqB,KAAK,UAAU,EAAE,CAAC;YACzD,MAAM,cAAc,GAAG,QAAQ,CAAC,qBAAqB,CAAC;YACtD,QAAQ,CAAC,qBAAqB,GAAG,gBAAgB,CAAC,cAAc,EAAE,QAAQ,EAAE,mBAAmB,EAAE,gBAAgB,CAAC,CAAC;YAEnH,SAAS,CAAC,GAAG,CAAC,8CAA8C,EAAE;gBAC5D,MAAM,EAAE,QAAQ;gBAChB,UAAU,EAAE,uBAAuB;gBACnC,QAAQ,EAAE,cAAc;aACzB,CAAC,CAAC;QACL,CAAC;QAED,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,OAAO,GAAG,iBAAiB,CAAC,eAAe,CAAC,CAAC;YACnD,OAAO,CAAC,GAAG,CAAC,uCAAuC,OAAO,GAAG,CAAC,CAAC;QACjE,CAAC;IACH,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,IAAI,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,wDAAwD,CAAC,EAAE,CAAC,CAAC;IACtF,CAAC;AACH,CAAC;AAED,MAAM,UAAU,UAAU;IACxB,KAAK,MAAM,CAAC,EAAE,KAAK,CAAC,IAAI,SAAS,EAAE,CAAC;QAClC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,GAAG,KAAK,CAAC,QAAQ,CAAC;QAChD,OAAO,KAAK,CAAC,MAAM,CAAC,kBAAkB,CAAC;IACzC,CAAC;IACD,SAAS,CAAC,KAAK,EAAE,CAAC;AACpB,CAAC"}

package/dist/esm/providers/anthropic.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"anthropic.d.ts","sourceRoot":"","sources":["../../../src/providers/anthropic.ts"],"names":[],"mappings":"AACA,OAAO,EAA4B,WAAW,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AAGhF,wBAAgB,sBAAsB,CAAC,MAAM,EAAE,GAAG,GAAG,YAAY,CAQhE;AAED,wBAAgB,mBAAmB,CAAC,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,GAAG,CA4D1G"}
+{"version":3,"file":"anthropic.d.ts","sourceRoot":"","sources":["../../../src/providers/anthropic.ts"],"names":[],"mappings":"AACA,OAAO,EAA4B,WAAW,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AAGhF,wBAAgB,sBAAsB,CAAC,MAAM,EAAE,GAAG,GAAG,YAAY,CA0BhE;AAED,wBAAgB,mBAAmB,CAAC,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,GAAG,CA2E1G"}

package/dist/esm/providers/anthropic.js

@@ -2,25 +2,45 @@ import { config } from '../config';
 import { buildEvent, printCapture } from '../schema';
 import * as transport from '../transport';
 export function parseAnthropicResponse(result) {
+    const usage = result?.usage;
+    const baseInput = usage?.input_tokens || 0;
+    const baseOutput = usage?.output_tokens || 0;
+    const cacheCreation = usage?.cache_creation_input_tokens || 0;
+    const cacheRead = usage?.cache_read_input_tokens || 0;
+    if (config.debug) {
+        console.log(`[llmtracer] Anthropic tokens — in=${baseInput} out=${baseOutput} cache_create=${cacheCreation} cache_read=${cacheRead}`);
+    }
+    const toolsInvoked = (result?.content || [])
+        .filter((b) => b.type === 'tool_use')
+        .map((b) => b.name)
+        .filter(Boolean);
     return {
         provider: 'anthropic',
         model: result?.model || 'unknown',
-        input_tokens: result?.usage?.input_tokens || 0,
-        output_tokens: result?.usage?.output_tokens || 0,
+        input_tokens: baseInput + cacheCreation + cacheRead,
+        output_tokens: baseOutput,
         status: result?.stop_reason ? 'success' : 'error',
+        tools_invoked: toolsInvoked,
     };
 }
 export function wrapAnthropicStream(result, context, elapsedMs, args) {
     let model = 'unknown';
     let inputTokens = 0;
     let outputTokens = 0;
+    let cacheCreation = 0;
+    let cacheRead = 0;
+    const toolsInvoked = [];
     function record() {
+        if (config.debug) {
+            console.log(`[llmtracer] Anthropic tokens — in=${inputTokens} out=${outputTokens} cache_create=${cacheCreation} cache_read=${cacheRead}`);
+        }
         const event = buildEvent(context, {
             provider: 'anthropic',
             model,
-            input_tokens: inputTokens,
+            input_tokens: inputTokens + cacheCreation + cacheRead,
             output_tokens: outputTokens,
             status: 'success',
+            tools_invoked: toolsInvoked,
         }, elapsedMs);
         transport.enqueue(event);
         if (config.debug)
@@ -43,6 +63,11 @@ export function wrapAnthropicStream(result, context, elapsedMs, args) {
                                 if (event?.type === 'message_start' && event.message) {
                                     model = event.message.model || model;
                                     inputTokens = event.message.usage?.input_tokens || inputTokens;
+                                    cacheCreation = event.message.usage?.cache_creation_input_tokens || cacheCreation;
+                                    cacheRead = event.message.usage?.cache_read_input_tokens || cacheRead;
+                                }
+                                if (event?.type === 'content_block_start' && event.content_block?.type === 'tool_use') {
+                                    toolsInvoked.push(event.content_block.name);
                                 }
                                 if (event?.type === 'message_delta' && event.usage) {
                                     outputTokens = event.usage.output_tokens || outputTokens;