@llmtracer/sdk 1.0.0 → 2.0.0

package/README.md

@@ -37,6 +37,64 @@ const response = await openai.chat.completions.create({
 await tracer.flush();
 ```
 
+## Tagging Guide
+
+Tags let you slice costs by any dimension in the dashboard. Global tags (set in the constructor) apply to all calls. Per-call tags (set via the `llmtracer` property) override globals for that specific call.
+
+### Track cost by feature
+
+```javascript
+await openai.chat.completions.create({
+  model: "gpt-4o",
+  messages: [...],
+  llmtracer: { tags: { feature: "chat" } }
+});
+```
+
+### Track cost by user (for B2B apps)
+
+```javascript
+await openai.chat.completions.create({
+  model: "gpt-4o",
+  messages: [...],
+  llmtracer: { tags: { feature: "chat", user_id: req.user.id } }
+});
+```
+
+### Track cost by customer/tenant
+
+```javascript
+await openai.chat.completions.create({
+  model: "gpt-4o",
+  messages: [...],
+  llmtracer: { tags: { customer: req.tenant.name, feature: "search" } }
+});
+```
+
+### Track cost by conversation
+
+```javascript
+await openai.chat.completions.create({
+  model: "gpt-4o",
+  messages: [...],
+  llmtracer: { tags: { conversation_id: sessionId, feature: "chat" } }
+});
+```
+
+### Track environment (global tag)
+
+```javascript
+const tracer = new LLMTracer({
+  apiKey: "lt_...",
+});
+
+tracer.instrumentOpenAI(openai, {
+  tags: { env: process.env.NODE_ENV }  // applies to all calls
+});
+```
+
+Tags appear in the dashboard's Breakdown page and Top Tags card. Use them to answer questions like "which customer costs the most?" or "which feature should I optimize?"
+
 ## Serverless Usage
 
 In environments like AWS Lambda or Google Cloud Functions, call `flush()` before your function returns to ensure all events are sent:

package/dist/cjs/caller.d.ts

@@ -0,0 +1,6 @@
+export declare function getCallerInfo(): {
+    caller_file?: string;
+    caller_function?: string;
+    caller_line?: number;
+};
+//# sourceMappingURL=caller.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"caller.d.ts","sourceRoot":"","sources":["../../src/caller.ts"],"names":[],"mappings":"AAAA,wBAAgB,aAAa,IAAI;IAAE,WAAW,CAAC,EAAE,MAAM,CAAC;IAAC,eAAe,CAAC,EAAE,MAAM,CAAC;IAAC,WAAW,CAAC,EAAE,MAAM,CAAA;CAAE,CA0CxG"}

package/dist/cjs/caller.js

@@ -0,0 +1,62 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getCallerInfo = getCallerInfo;
+function getCallerInfo() {
+    try {
+        const err = new Error();
+        const stack = err.stack?.split('\n');
+        if (!stack)
+            return {};
+        for (let i = 1; i < stack.length; i++) {
+            const line = stack[i];
+            // Skip our own SDK frames (dist/ or src/ within the SDK package)
+            if (isInternalFrame(line)) {
+                continue;
+            }
+            // Skip node internals
+            if (line.includes('node:') || line.includes('(<anonymous>)')) {
+                continue;
+            }
+            // "    at functionName (/path/to/file.js:line:col)"
+            const matchWithFn = line.match(/at\s+(.+?)\s+\((.+):(\d+):(\d+)\)/);
+            if (matchWithFn) {
+                return {
+                    caller_function: matchWithFn[1],
+                    caller_file: matchWithFn[2],
+                    caller_line: parseInt(matchWithFn[3], 10),
+                };
+            }
+            // "    at /path/to/file.js:line:col"
+            const matchNoFn = line.match(/at\s+(.+):(\d+):(\d+)/);
+            if (matchNoFn) {
+                return {
+                    caller_file: matchNoFn[1],
+                    caller_line: parseInt(matchNoFn[2], 10),
+                };
+            }
+        }
+        return {};
+    }
+    catch {
+        return {};
+    }
+}
+function isInternalFrame(line) {
+    // Match frames from our compiled output or source
+    // e.g. /path/to/@llmtracer/sdk/dist/cjs/wrapper.js
+    // or   /path/to/llmtracer-sdk/dist/cjs/schema.js
+    // or   /path/to/llmtracer-sdk/src/wrapper.ts
+    // Skip if it's within our dist/ or src/ directories
+    if (line.includes('node_modules/@llmtracer/') ||
+        line.includes('node_modules\\@llmtracer\\')) {
+        return true;
+    }
+    // When running from the repo itself (dev/test), skip dist/ and src/ paths
+    // but NOT test/ paths
+    const distOrSrc = /llmtracer-sdk[/\\](?:dist|src)[/\\]/;
+    if (distOrSrc.test(line)) {
+        return true;
+    }
+    return false;
+}
+//# sourceMappingURL=caller.js.map

package/dist/cjs/caller.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"caller.js","sourceRoot":"","sources":["../../src/caller.ts"],"names":[],"mappings":";;AAAA,sCA0CC;AA1CD,SAAgB,aAAa;IAC3B,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,IAAI,KAAK,EAAE,CAAC;QACxB,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;QACrC,IAAI,CAAC,KAAK;YAAE,OAAO,EAAE,CAAC;QAEtB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACtC,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;YAEtB,iEAAiE;YACjE,IAAI,eAAe,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC1B,SAAS;YACX,CAAC;YAED,sBAAsB;YACtB,IAAI,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;gBAC7D,SAAS;YACX,CAAC;YAED,oDAAoD;YACpD,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,mCAAmC,CAAC,CAAC;YACpE,IAAI,WAAW,EAAE,CAAC;gBAChB,OAAO;oBACL,eAAe,EAAE,WAAW,CAAC,CAAC,CAAC;oBAC/B,WAAW,EAAE,WAAW,CAAC,CAAC,CAAC;oBAC3B,WAAW,EAAE,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;iBAC1C,CAAC;YACJ,CAAC;YAED,qCAAqC;YACrC,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,uBAAuB,CAAC,CAAC;YACtD,IAAI,SAAS,EAAE,CAAC;gBACd,OAAO;oBACL,WAAW,EAAE,SAAS,CAAC,CAAC,CAAC;oBACzB,WAAW,EAAE,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;iBACxC,CAAC;YACJ,CAAC;QACH,CAAC;QACD,OAAO,EAAE,CAAC;IACZ,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED,SAAS,eAAe,CAAC,IAAY;IACnC,kDAAkD;IAClD,mDAAmD;IACnD,iDAAiD;IACjD,6CAA6C;IAE7C,oDAAoD;IACpD,IACE,IAAI,CAAC,QAAQ,CAAC,0BAA0B,CAAC;QACzC,IAAI,CAAC,QAAQ,CAAC,4BAA4B,CAAC,EAC3C,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,0EAA0E;IAC1E,sBAAsB;IACtB,MAAM,SAAS,GAAG,qCAAqC,CAAC;IACxD,IAAI,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QACzB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC"}

package/dist/cjs/config.d.ts

@@ -0,0 +1,7 @@
+export declare const config: {
+    apiKey: string;
+    endpoint: string;
+    debug: boolean;
+    enabled: boolean;
+};
+//# sourceMappingURL=config.d.ts.map

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

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

package/dist/cjs/config.js

@@ -0,0 +1,10 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.config = void 0;
+exports.config = {
+    apiKey: '',
+    endpoint: 'https://us-central1-llmtracer-alt.cloudfunctions.net/v1Events',
+    debug: false,
+    enabled: false,
+};
+//# sourceMappingURL=config.js.map

package/dist/cjs/config.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.js","sourceRoot":"","sources":["../../src/config.ts"],"names":[],"mappings":";;;AAAa,QAAA,MAAM,GAAG;IACpB,MAAM,EAAE,EAAE;IACV,QAAQ,EAAE,+DAA+D;IACzE,KAAK,EAAE,KAAK;IACZ,OAAO,EAAE,KAAK;CACf,CAAC"}

package/dist/cjs/context.d.ts

@@ -0,0 +1,9 @@
+export interface TraceContext {
+    traceId: string;
+    spanId: string;
+    tags: Record<string, any>;
+    parentSpanId?: string;
+}
+export declare function getCurrent(): TraceContext | undefined;
+export declare function run<T>(tags: Record<string, any>, fn: () => T): T;
+//# sourceMappingURL=context.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"context.d.ts","sourceRoot":"","sources":["../../src/context.ts"],"names":[],"mappings":"AAGA,MAAM,WAAW,YAAY;IAC3B,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC1B,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAID,wBAAgB,UAAU,IAAI,YAAY,GAAG,SAAS,CAErD;AAED,wBAAgB,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,CAWhE"}

package/dist/cjs/context.js

@@ -0,0 +1,21 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getCurrent = getCurrent;
+exports.run = run;
+const async_hooks_1 = require("async_hooks");
+const crypto_1 = require("crypto");
+const store = new async_hooks_1.AsyncLocalStorage();
+function getCurrent() {
+    return store.getStore();
+}
+function run(tags, fn) {
+    const parent = store.getStore();
+    const ctx = {
+        traceId: parent?.traceId || (0, crypto_1.randomUUID)(),
+        spanId: (0, crypto_1.randomUUID)(),
+        tags: { ...(parent?.tags || {}), ...tags },
+        parentSpanId: parent?.spanId,
+    };
+    return store.run(ctx, fn);
+}
+//# sourceMappingURL=context.js.map

package/dist/cjs/context.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"context.js","sourceRoot":"","sources":["../../src/context.ts"],"names":[],"mappings":";;AAYA,gCAEC;AAED,kBAWC;AA3BD,6CAAgD;AAChD,mCAAoC;AASpC,MAAM,KAAK,GAAG,IAAI,+BAAiB,EAAgB,CAAC;AAEpD,SAAgB,UAAU;IACxB,OAAO,KAAK,CAAC,QAAQ,EAAE,CAAC;AAC1B,CAAC;AAED,SAAgB,GAAG,CAAI,IAAyB,EAAE,EAAW;IAC3D,MAAM,MAAM,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC;IAEhC,MAAM,GAAG,GAAiB;QACxB,OAAO,EAAE,MAAM,EAAE,OAAO,IAAI,IAAA,mBAAU,GAAE;QACxC,MAAM,EAAE,IAAA,mBAAU,GAAE;QACpB,IAAI,EAAE,EAAE,GAAG,CAAC,MAAM,EAAE,IAAI,IAAI,EAAE,CAAC,EAAE,GAAG,IAAI,EAAE;QAC1C,YAAY,EAAE,MAAM,EAAE,MAAM;KAC7B,CAAC;IAEF,OAAO,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;AAC5B,CAAC"}

package/dist/cjs/index.d.ts

@@ -0,0 +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?: {
+    apiKey?: string;
+    debug?: boolean;
+    endpoint?: string;
+}): 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 };
+declare const llmtracer: {
+    init: typeof init;
+    trace: typeof trace;
+    flush: typeof flush;
+    __version__: string;
+};
+export default llmtracer;
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +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"}

package/dist/cjs/index.js

@@ -0,0 +1,93 @@
+"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._config = exports._transport = exports._patcher = exports.__version__ = void 0;
+exports.init = init;
+exports.trace = trace;
+exports.flush = flush;
+const config_1 = require("./config");
+Object.defineProperty(exports, "_config", { enumerable: true, get: function () { return config_1.config; } });
+const context = __importStar(require("./context"));
+const transport = __importStar(require("./transport"));
+exports._transport = transport;
+const patcher = __importStar(require("./patcher"));
+exports._patcher = patcher;
+exports.__version__ = '2.0.0';
+function init(options) {
+    try {
+        config_1.config.apiKey = options?.apiKey || process.env.LLMTRACER_API_KEY || '';
+        config_1.config.debug = options?.debug || false;
+        config_1.config.endpoint = options?.endpoint || 'https://us-central1-llmtracer-alt.cloudfunctions.net/v1Events';
+        config_1.config.enabled = Boolean(config_1.config.apiKey);
+        if (!config_1.config.enabled) {
+            if (config_1.config.debug)
+                console.log('[llmtracer] No API key found. Tracing disabled.');
+            return;
+        }
+        transport.start();
+        patcher.patchOpenAI(config_1.config.debug);
+        patcher.patchAnthropic(config_1.config.debug);
+        if (config_1.config.debug)
+            console.log('[llmtracer] Ready. Events \u2192 dashboard.llmtracer.com');
+    }
+    catch (e) {
+        config_1.config.enabled = false;
+        if (config_1.config.debug)
+            console.log(`[llmtracer] init() failed: ${e}. Tracing disabled.`);
+    }
+}
+function trace(tags, fn) {
+    try {
+        if (!config_1.config.enabled) {
+            return fn();
+        }
+        return context.run(tags, fn);
+    }
+    catch {
+        return fn();
+    }
+}
+async function flush() {
+    try {
+        await transport.flush();
+    }
+    catch {
+        // crash-proof: never throw
+    }
+}
+// Default export for ESM: import llmtracer from 'llmtracer'
+const llmtracer = { init, trace, flush, __version__: exports.__version__ };
+exports.default = llmtracer;
+//# sourceMappingURL=index.js.map

package/dist/cjs/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAOA,oBAsBC;AAED,sBASC;AAED,sBAMC;AAhDD,qCAAkC;AAmD+B,wFAnDxD,eAAM,OAmDyD;AAlDxE,mDAAqC;AACrC,uDAAyC;AAiDE,+BAAU;AAhDrD,mDAAqC;AAgDjB,2BAAQ;AA9Cf,QAAA,WAAW,GAAG,OAAO,CAAC;AAEnC,SAAgB,IAAI,CAAC,OAAiE;IACpF,IAAI,CAAC;QACH,eAAM,CAAC,MAAM,GAAG,OAAO,EAAE,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,iBAAiB,IAAI,EAAE,CAAC;QACvE,eAAM,CAAC,KAAK,GAAG,OAAO,EAAE,KAAK,IAAI,KAAK,CAAC;QACvC,eAAM,CAAC,QAAQ,GAAG,OAAO,EAAE,QAAQ,IAAI,+DAA+D,CAAC;QACvG,eAAM,CAAC,OAAO,GAAG,OAAO,CAAC,eAAM,CAAC,MAAM,CAAC,CAAC;QAExC,IAAI,CAAC,eAAM,CAAC,OAAO,EAAE,CAAC;YACpB,IAAI,eAAM,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,eAAM,CAAC,KAAK,CAAC,CAAC;QAClC,OAAO,CAAC,cAAc,CAAC,eAAM,CAAC,KAAK,CAAC,CAAC;QAErC,IAAI,eAAM,CAAC,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,0DAA0D,CAAC,CAAC;IAC5F,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,eAAM,CAAC,OAAO,GAAG,KAAK,CAAC;QACvB,IAAI,eAAM,CAAC,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,8BAA8B,CAAC,qBAAqB,CAAC,CAAC;IACtF,CAAC;AACH,CAAC;AAED,SAAgB,KAAK,CAAI,IAAyB,EAAE,EAAwB;IAC1E,IAAI,CAAC;QACH,IAAI,CAAC,eAAM,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;AAEM,KAAK,UAAU,KAAK;IACzB,IAAI,CAAC;QACH,MAAM,SAAS,CAAC,KAAK,EAAE,CAAC;IAC1B,CAAC;IAAC,MAAM,CAAC;QACP,2BAA2B;IAC7B,CAAC;AACH,CAAC;AAKD,4DAA4D;AAC5D,MAAM,SAAS,GAAG,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,WAAW,EAAX,mBAAW,EAAE,CAAC;AACtD,kBAAe,SAAS,CAAC"}

package/dist/cjs/patcher.d.ts

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

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

@@ -0,0 +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"}

package/dist/cjs/patcher.js

@@ -0,0 +1,93 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.patchOpenAI = patchOpenAI;
+exports.patchAnthropic = patchAnthropic;
+exports.unpatchAll = unpatchAll;
+const wrapper_1 = require("./wrapper");
+const openai_1 = require("./providers/openai");
+const anthropic_1 = require("./providers/anthropic");
+const originals = new Map();
+function getPackageVersion(packageName) {
+    try {
+        const path = require('path');
+        const fs = require('fs');
+        const entryPoint = require.resolve(packageName);
+        let dir = path.dirname(entryPoint);
+        while (dir !== path.dirname(dir)) {
+            const pkgPath = path.join(dir, 'package.json');
+            try {
+                const pkg = JSON.parse(fs.readFileSync(pkgPath, 'utf8'));
+                if (pkg.name === packageName)
+                    return pkg.version || 'unknown';
+            }
+            catch {
+                // no package.json here, keep walking up
+            }
+            dir = path.dirname(dir);
+        }
+    }
+    catch {
+        // ignore
+    }
+    return 'unknown';
+}
+function patchOpenAI(debug) {
+    try {
+        const completionsModule = require('openai/resources/chat/completions');
+        const CompletionsProto = completionsModule.Completions?.prototype;
+        if (!CompletionsProto)
+            throw new Error('Completions prototype not found');
+        if (CompletionsProto._llmtracer_patched)
+            return;
+        const originalCreate = CompletionsProto.create;
+        CompletionsProto.create = (0, wrapper_1.makeAsyncWrapper)(originalCreate, 'openai', openai_1.parseOpenAIResponse, openai_1.wrapOpenAIStream);
+        CompletionsProto._llmtracer_patched = true;
+        originals.set('openai.Completions.create', {
+            target: CompletionsProto,
+            methodName: 'create',
+            original: originalCreate,
+        });
+        if (debug) {
+            const version = getPackageVersion('openai');
+            console.log(`[llmtracer] \u2713 Patched openai (v${version})`);
+        }
+    }
+    catch (e) {
+        if (debug)
+            console.log(`[llmtracer] \u2717 openai not found or patch failed: ${e}`);
+    }
+}
+function patchAnthropic(debug) {
+    try {
+        const messagesModule = require('@anthropic-ai/sdk/resources/messages');
+        const MessagesProto = messagesModule.Messages?.prototype;
+        if (!MessagesProto)
+            throw new Error('Messages prototype not found');
+        if (MessagesProto._llmtracer_patched)
+            return;
+        const originalCreate = MessagesProto.create;
+        MessagesProto.create = (0, wrapper_1.makeAsyncWrapper)(originalCreate, 'anthropic', anthropic_1.parseAnthropicResponse, anthropic_1.wrapAnthropicStream);
+        MessagesProto._llmtracer_patched = true;
+        originals.set('anthropic.Messages.create', {
+            target: MessagesProto,
+            methodName: 'create',
+            original: originalCreate,
+        });
+        if (debug) {
+            const version = getPackageVersion('@anthropic-ai/sdk');
+            console.log(`[llmtracer] \u2713 Patched anthropic (v${version})`);
+        }
+    }
+    catch (e) {
+        if (debug)
+            console.log(`[llmtracer] \u2717 anthropic not found or patch failed: ${e}`);
+    }
+}
+function unpatchAll() {
+    for (const [, entry] of originals) {
+        entry.target[entry.methodName] = entry.original;
+        delete entry.target._llmtracer_patched;
+    }
+    originals.clear();
+}
+//# sourceMappingURL=patcher.js.map

package/dist/cjs/patcher.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"patcher.js","sourceRoot":"","sources":["../../src/patcher.ts"],"names":[],"mappings":";;AA4BA,kCAyBC;AAED,wCAyBC;AAED,gCAMC;AAxFD,uCAA6C;AAC7C,+CAA2E;AAC3E,qDAAoF;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,SAAgB,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,IAAA,0BAAgB,EAAC,cAAc,EAAE,QAAQ,EAAE,4BAAmB,EAAE,yBAAgB,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,SAAgB,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,IAAA,0BAAgB,EAAC,cAAc,EAAE,WAAW,EAAE,kCAAsB,EAAE,+BAAmB,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,SAAgB,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/cjs/providers/anthropic.d.ts

@@ -0,0 +1,4 @@
+import { ContextData, ResponseData } from '../schema';
+export declare function parseAnthropicResponse(result: any): ResponseData;
+export declare function wrapAnthropicStream(result: any, context: ContextData, elapsedMs: number, args: any[]): any;
+//# sourceMappingURL=anthropic.d.ts.map

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

@@ -0,0 +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"}

package/dist/cjs/providers/anthropic.js

@@ -0,0 +1,111 @@
+"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.parseAnthropicResponse = parseAnthropicResponse;
+exports.wrapAnthropicStream = wrapAnthropicStream;
+const config_1 = require("../config");
+const schema_1 = require("../schema");
+const transport = __importStar(require("../transport"));
+function parseAnthropicResponse(result) {
+    return {
+        provider: 'anthropic',
+        model: result?.model || 'unknown',
+        input_tokens: result?.usage?.input_tokens || 0,
+        output_tokens: result?.usage?.output_tokens || 0,
+        status: result?.stop_reason ? 'success' : 'error',
+    };
+}
+function wrapAnthropicStream(result, context, elapsedMs, args) {
+    let model = 'unknown';
+    let inputTokens = 0;
+    let outputTokens = 0;
+    function record() {
+        const event = (0, schema_1.buildEvent)(context, {
+            provider: 'anthropic',
+            model,
+            input_tokens: inputTokens,
+            output_tokens: outputTokens,
+            status: 'success',
+        }, elapsedMs);
+        transport.enqueue(event);
+        if (config_1.config.debug)
+            (0, schema_1.printCapture)(event);
+    }
+    return new Proxy(result, {
+        get(target, prop) {
+            if (prop === Symbol.asyncIterator) {
+                return function () {
+                    const originalIterator = target[Symbol.asyncIterator]();
+                    return {
+                        async next() {
+                            const iterResult = await originalIterator.next();
+                            if (!iterResult.done) {
+                                const event = iterResult.value;
+                                // Anthropic stream events:
+                                // - message_start: contains model and input token count
+                                // - message_delta: contains output token count
+                                // - content_block_delta: contains text deltas
+                                if (event?.type === 'message_start' && event.message) {
+                                    model = event.message.model || model;
+                                    inputTokens = event.message.usage?.input_tokens || inputTokens;
+                                }
+                                if (event?.type === 'message_delta' && event.usage) {
+                                    outputTokens = event.usage.output_tokens || outputTokens;
+                                }
+                            }
+                            if (iterResult.done) {
+                                try {
+                                    record();
+                                }
+                                catch {
+                                    // swallow
+                                }
+                            }
+                            return iterResult;
+                        },
+                        return: originalIterator.return?.bind(originalIterator),
+                        throw: originalIterator.throw?.bind(originalIterator),
+                    };
+                };
+            }
+            const value = target[prop];
+            if (typeof value === 'function') {
+                return value.bind(target);
+            }
+            return value;
+        },
+    });
+}
+//# sourceMappingURL=anthropic.js.map

package/dist/cjs/providers/anthropic.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["../../../src/providers/anthropic.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAIA,wDAQC;AAED,kDA4DC;AA1ED,sCAAmC;AACnC,sCAAgF;AAChF,wDAA0C;AAE1C,SAAgB,sBAAsB,CAAC,MAAW;IAChD,OAAO;QACL,QAAQ,EAAE,WAAW;QACrB,KAAK,EAAE,MAAM,EAAE,KAAK,IAAI,SAAS;QACjC,YAAY,EAAE,MAAM,EAAE,KAAK,EAAE,YAAY,IAAI,CAAC;QAC9C,aAAa,EAAE,MAAM,EAAE,KAAK,EAAE,aAAa,IAAI,CAAC;QAChD,MAAM,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO;KAClD,CAAC;AACJ,CAAC;AAED,SAAgB,mBAAmB,CAAC,MAAW,EAAE,OAAoB,EAAE,SAAiB,EAAE,IAAW;IACnG,IAAI,KAAK,GAAG,SAAS,CAAC;IACtB,IAAI,WAAW,GAAG,CAAC,CAAC;IACpB,IAAI,YAAY,GAAG,CAAC,CAAC;IAErB,SAAS,MAAM;QACb,MAAM,KAAK,GAAG,IAAA,mBAAU,EAAC,OAAO,EAAE;YAChC,QAAQ,EAAE,WAAW;YACrB,KAAK;YACL,YAAY,EAAE,WAAW;YACzB,aAAa,EAAE,YAAY;YAC3B,MAAM,EAAE,SAAS;SAClB,EAAE,SAAS,CAAC,CAAC;QACd,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACzB,IAAI,eAAM,CAAC,KAAK;YAAE,IAAA,qBAAY,EAAC,KAAK,CAAC,CAAC;IACxC,CAAC;IAED,OAAO,IAAI,KAAK,CAAC,MAAM,EAAE;QACvB,GAAG,CAAC,MAAM,EAAE,IAAI;YACd,IAAI,IAAI,KAAK,MAAM,CAAC,aAAa,EAAE,CAAC;gBAClC,OAAO;oBACL,MAAM,gBAAgB,GAAG,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC;oBACxD,OAAO;wBACL,KAAK,CAAC,IAAI;4BACR,MAAM,UAAU,GAAG,MAAM,gBAAgB,CAAC,IAAI,EAAE,CAAC;4BACjD,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;gCACrB,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC;gCAC/B,2BAA2B;gCAC3B,wDAAwD;gCACxD,+CAA+C;gCAC/C,8CAA8C;gCAC9C,IAAI,KAAK,EAAE,IAAI,KAAK,eAAe,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;oCACrD,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,IAAI,KAAK,CAAC;oCACrC,WAAW,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,YAAY,IAAI,WAAW,CAAC;gCACjE,CAAC;gCACD,IAAI,KAAK,EAAE,IAAI,KAAK,eAAe,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;oCACnD,YAAY,GAAG,KAAK,CAAC,KAAK,CAAC,aAAa,IAAI,YAAY,CAAC;gCAC3D,CAAC;4BACH,CAAC;4BACD,IAAI,UAAU,CAAC,IAAI,EAAE,CAAC;gCACpB,IAAI,CAAC;oCACH,MAAM,EAAE,CAAC;gCACX,CAAC;gCAAC,MAAM,CAAC;oCACP,UAAU;gCACZ,CAAC;4BACH,CAAC;4BACD,OAAO,UAAU,CAAC;wBACpB,CAAC;wBACD,MAAM,EAAE,gBAAgB,CAAC,MAAM,EAAE,IAAI,CAAC,gBAAgB,CAAC;wBACvD,KAAK,EAAE,gBAAgB,CAAC,KAAK,EAAE,IAAI,CAAC,gBAAgB,CAAC;qBACtD,CAAC;gBACJ,CAAC,CAAC;YACJ,CAAC;YACD,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;YAC3B,IAAI,OAAO,KAAK,KAAK,UAAU,EAAE,CAAC;gBAChC,OAAO,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC5B,CAAC;YACD,OAAO,KAAK,CAAC;QACf,CAAC;KACF,CAAC,CAAC;AACL,CAAC"}

package/dist/cjs/providers/index.d.ts

@@ -0,0 +1,3 @@
+export { parseOpenAIResponse, wrapOpenAIStream } from './openai';
+export { parseAnthropicResponse, wrapAnthropicStream } from './anthropic';
+//# sourceMappingURL=index.d.ts.map

package/dist/cjs/providers/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/providers/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC;AACjE,OAAO,EAAE,sBAAsB,EAAE,mBAAmB,EAAE,MAAM,aAAa,CAAC"}

package/dist/cjs/providers/index.js

@@ -0,0 +1,10 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.wrapAnthropicStream = exports.parseAnthropicResponse = exports.wrapOpenAIStream = exports.parseOpenAIResponse = void 0;
+var openai_1 = require("./openai");
+Object.defineProperty(exports, "parseOpenAIResponse", { enumerable: true, get: function () { return openai_1.parseOpenAIResponse; } });
+Object.defineProperty(exports, "wrapOpenAIStream", { enumerable: true, get: function () { return openai_1.wrapOpenAIStream; } });
+var anthropic_1 = require("./anthropic");
+Object.defineProperty(exports, "parseAnthropicResponse", { enumerable: true, get: function () { return anthropic_1.parseAnthropicResponse; } });
+Object.defineProperty(exports, "wrapAnthropicStream", { enumerable: true, get: function () { return anthropic_1.wrapAnthropicStream; } });
+//# sourceMappingURL=index.js.map

package/dist/cjs/providers/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/providers/index.ts"],"names":[],"mappings":";;;AAAA,mCAAiE;AAAxD,6GAAA,mBAAmB,OAAA;AAAE,0GAAA,gBAAgB,OAAA;AAC9C,yCAA0E;AAAjE,mHAAA,sBAAsB,OAAA;AAAE,gHAAA,mBAAmB,OAAA"}

package/dist/cjs/providers/openai.d.ts

@@ -0,0 +1,4 @@
+import { ContextData, ResponseData } from '../schema';
+export declare function parseOpenAIResponse(result: any): ResponseData;
+export declare function wrapOpenAIStream(result: any, context: ContextData, elapsedMs: number, args: any[]): any;
+//# sourceMappingURL=openai.d.ts.map

package/dist/cjs/providers/openai.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../../src/providers/openai.ts"],"names":[],"mappings":"AACA,OAAO,EAA4B,WAAW,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AAGhF,wBAAgB,mBAAmB,CAAC,MAAM,EAAE,GAAG,GAAG,YAAY,CAQ7D;AAED,wBAAgB,gBAAgB,CAAC,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,GAAG,CA0DvG"}