tokenmeter 0.9.0

package/dist/instrumentation/proxy.js

@@ -0,0 +1,337 @@
+/**
+ * Proxy Engine
+ *
+ * Creates a recursive Proxy that intercepts method calls and creates OTel spans.
+ * Calculates costs and adds them to spans before export.
+ */
+import { trace, context, propagation, SpanKind, SpanStatusCode, } from "@opentelemetry/api";
+import { extractUsage } from "./strategies/index.js";
+import { getModelPricing, calculateCost, getCachedManifest, loadManifest, } from "../pricing/manifest.js";
+import { TM_ATTRIBUTES, GEN_AI_ATTRIBUTES } from "../types.js";
+import { logger } from "../logger.js";
+// The tracer instance
+let tracer = null;
+// Pricing manifest (loaded once)
+let pricingManifest = null;
+let manifestLoadPromise = null;
+/**
+ * Ensure pricing manifest is loaded
+ */
+async function ensureManifestLoaded() {
+    if (pricingManifest)
+        return;
+    if (manifestLoadPromise) {
+        await manifestLoadPromise;
+        return;
+    }
+    manifestLoadPromise = loadManifest()
+        .then((manifest) => {
+        pricingManifest = manifest;
+    })
+        .catch((err) => {
+        logger.warn("Failed to load pricing manifest:", err);
+    });
+    await manifestLoadPromise;
+}
+// Start loading manifest immediately
+ensureManifestLoaded();
+/**
+ * Get or create the TokenMeter tracer
+ */
+function getTracer() {
+    if (!tracer) {
+        tracer = trace.getTracer("tokenmeter", "5.0.0");
+    }
+    return tracer;
+}
+/**
+ * Get baggage attributes from current context
+ */
+function getBaggageAttributes() {
+    const currentContext = context.active();
+    const baggage = propagation.getBaggage(currentContext);
+    if (!baggage) {
+        return {};
+    }
+    const attributes = {};
+    baggage.getAllEntries().forEach(([key, entry]) => {
+        attributes[key] = entry.value;
+    });
+    return attributes;
+}
+/**
+ * Detect the provider from the client instance
+ */
+function detectProvider(client) {
+    if (!client || typeof client !== "object")
+        return "unknown";
+    const c = client;
+    // OpenAI: has chat.completions
+    if ("chat" in c &&
+        typeof c.chat === "object" &&
+        c.chat &&
+        "completions" in c.chat) {
+        return "openai";
+    }
+    // Anthropic: has messages.create
+    if ("messages" in c && typeof c.messages === "object") {
+        return "anthropic";
+    }
+    // fal.ai: has subscribe method
+    if ("subscribe" in c && typeof c.subscribe === "function") {
+        return "fal";
+    }
+    // ElevenLabs: has textToSpeech
+    if ("textToSpeech" in c && typeof c.textToSpeech === "object") {
+        return "elevenlabs";
+    }
+    return "unknown";
+}
+/**
+ * Check if a value is a Promise
+ */
+function isPromise(value) {
+    return (value !== null &&
+        typeof value === "object" &&
+        "then" in value &&
+        typeof value.then === "function");
+}
+/**
+ * Check if a value is an async iterator (stream)
+ */
+function isAsyncIterable(value) {
+    return (value !== null && typeof value === "object" && Symbol.asyncIterator in value);
+}
+/**
+ * Calculate cost for usage data
+ */
+function calculateUsageCost(usage) {
+    const manifest = pricingManifest || getCachedManifest();
+    if (!manifest) {
+        logger.warn("Pricing manifest not loaded, cost will be 0");
+        return 0;
+    }
+    const pricing = getModelPricing(usage.provider, usage.model, manifest);
+    if (!pricing) {
+        logger.warn(`Missing pricing for ${usage.provider}/${usage.model}`);
+        return 0;
+    }
+    return calculateCost({
+        inputUnits: usage.inputUnits,
+        outputUnits: usage.outputUnits,
+        cachedInputUnits: usage.cachedInputUnits,
+    }, pricing);
+}
+/**
+ * Add usage data and calculated cost to a span
+ */
+function addUsageToSpan(span, usage) {
+    const attributes = {
+        [TM_ATTRIBUTES.PROVIDER]: usage.provider,
+        [TM_ATTRIBUTES.MODEL]: usage.model,
+    };
+    if (usage.inputUnits !== undefined) {
+        attributes[TM_ATTRIBUTES.INPUT_UNITS] = usage.inputUnits;
+        attributes[GEN_AI_ATTRIBUTES.INPUT_TOKENS] = usage.inputUnits;
+    }
+    if (usage.outputUnits !== undefined) {
+        attributes[TM_ATTRIBUTES.OUTPUT_UNITS] = usage.outputUnits;
+        attributes[GEN_AI_ATTRIBUTES.OUTPUT_TOKENS] = usage.outputUnits;
+    }
+    attributes[GEN_AI_ATTRIBUTES.MODEL] = usage.model;
+    attributes[GEN_AI_ATTRIBUTES.SYSTEM] = usage.provider;
+    // Calculate and add cost BEFORE span.end()
+    const costUsd = calculateUsageCost(usage);
+    attributes[TM_ATTRIBUTES.COST_USD] = costUsd;
+    span.setAttributes(attributes);
+}
+/**
+ * Wrap an async iterator to track usage from streamed responses
+ */
+function wrapAsyncIterator(iterator, span, provider, methodPath, args) {
+    const originalIterator = iterator[Symbol.asyncIterator]();
+    let accumulatedUsage = null;
+    const wrappedIterator = {
+        async next() {
+            try {
+                const result = await originalIterator.next();
+                if (result.done) {
+                    // Stream completed - extract final usage if available
+                    if (accumulatedUsage) {
+                        addUsageToSpan(span, accumulatedUsage);
+                    }
+                    span.setStatus({ code: SpanStatusCode.OK });
+                    span.end();
+                }
+                else {
+                    // Accumulate usage from chunks (OpenAI streams include usage in final chunk)
+                    const chunk = result.value;
+                    if (chunk && typeof chunk === "object" && "usage" in chunk) {
+                        const usage = extractUsage(methodPath, chunk, args, provider);
+                        if (usage) {
+                            accumulatedUsage = usage;
+                        }
+                    }
+                }
+                return result;
+            }
+            catch (error) {
+                span.recordException(error);
+                span.setStatus({
+                    code: SpanStatusCode.ERROR,
+                    message: error instanceof Error ? error.message : String(error),
+                });
+                span.end();
+                throw error;
+            }
+        },
+        async return(value) {
+            // Stream was cancelled - still calculate cost for what we consumed
+            if (accumulatedUsage) {
+                addUsageToSpan(span, accumulatedUsage);
+            }
+            span.setStatus({ code: SpanStatusCode.OK });
+            span.end();
+            if (originalIterator.return) {
+                return originalIterator.return(value);
+            }
+            return { done: true, value: undefined };
+        },
+        async throw(error) {
+            // Stream errored - still calculate cost for what we consumed
+            if (accumulatedUsage) {
+                addUsageToSpan(span, accumulatedUsage);
+            }
+            span.recordException(error);
+            span.setStatus({
+                code: SpanStatusCode.ERROR,
+                message: error instanceof Error ? error.message : String(error),
+            });
+            span.end();
+            if (originalIterator.throw) {
+                return originalIterator.throw(error);
+            }
+            throw error;
+        },
+    };
+    return {
+        [Symbol.asyncIterator]: () => wrappedIterator,
+    };
+}
+/**
+ * Create a monitored proxy for a client instance
+ *
+ * @param client - The SDK client instance to wrap
+ * @param options - Configuration options
+ * @returns A proxied version of the client that creates spans for method calls
+ *
+ * @example
+ * ```typescript
+ * import OpenAI from 'openai';
+ * import { monitor } from 'tokenmeter';
+ *
+ * const openai = monitor(new OpenAI({ apiKey: '...' }));
+ * await openai.chat.completions.create({ model: 'gpt-4o', messages: [...] });
+ * ```
+ */
+export function monitor(client, options = {}) {
+    const provider = options.provider || detectProvider(client);
+    const clientName = options.name || provider;
+    const baseAttributes = options.attributes || {};
+    /**
+     * Create a recursive proxy handler
+     */
+    function createProxyHandler(path = []) {
+        return {
+            get(target, prop, receiver) {
+                const value = Reflect.get(target, prop, receiver);
+                // Skip symbols and internal properties
+                if (typeof prop === "symbol" || prop.startsWith("_")) {
+                    return value;
+                }
+                // If it's a function, wrap it
+                if (typeof value === "function") {
+                    return function (...args) {
+                        const methodPath = [...path, prop];
+                        const spanName = `${clientName}.${methodPath.join(".")}`;
+                        // Get baggage attributes from context (set by withAttributes)
+                        const baggageAttrs = getBaggageAttributes();
+                        // Start span with base + baggage attributes
+                        const currentTracer = getTracer();
+                        const span = currentTracer.startSpan(spanName, {
+                            kind: SpanKind.CLIENT,
+                            attributes: {
+                                ...baseAttributes,
+                                ...baggageAttrs, // Include baggage attributes (org.id, user.id, etc.)
+                                [TM_ATTRIBUTES.PROVIDER]: provider,
+                                "rpc.service": clientName,
+                                "rpc.method": methodPath.join("."),
+                            },
+                        });
+                        // Execute within span context
+                        return context.with(trace.setSpan(context.active(), span), () => {
+                            try {
+                                const result = value.apply(this === receiver ? target : this, args);
+                                // Handle async results
+                                if (isPromise(result)) {
+                                    return result.then((resolved) => {
+                                        // Check if result is a stream
+                                        if (isAsyncIterable(resolved)) {
+                                            return wrapAsyncIterator(resolved, span, provider, methodPath, args);
+                                        }
+                                        // Extract usage from result
+                                        const usage = extractUsage(methodPath, resolved, args, provider);
+                                        if (usage) {
+                                            addUsageToSpan(span, usage);
+                                        }
+                                        span.setStatus({ code: SpanStatusCode.OK });
+                                        span.end();
+                                        return resolved;
+                                    }, (error) => {
+                                        span.recordException(error);
+                                        span.setStatus({
+                                            code: SpanStatusCode.ERROR,
+                                            message: error instanceof Error
+                                                ? error.message
+                                                : String(error),
+                                        });
+                                        span.end();
+                                        throw error;
+                                    });
+                                }
+                                // Handle sync results (rare for API calls)
+                                if (isAsyncIterable(result)) {
+                                    return wrapAsyncIterator(result, span, provider, methodPath, args);
+                                }
+                                const usage = extractUsage(methodPath, result, args, provider);
+                                if (usage) {
+                                    addUsageToSpan(span, usage);
+                                }
+                                span.setStatus({ code: SpanStatusCode.OK });
+                                span.end();
+                                return result;
+                            }
+                            catch (error) {
+                                span.recordException(error);
+                                span.setStatus({
+                                    code: SpanStatusCode.ERROR,
+                                    message: error instanceof Error ? error.message : String(error),
+                                });
+                                span.end();
+                                throw error;
+                            }
+                        });
+                    };
+                }
+                // If it's an object, recursively proxy it
+                if (value !== null && typeof value === "object") {
+                    return new Proxy(value, createProxyHandler([...path, prop]));
+                }
+                return value;
+            },
+        };
+    }
+    return new Proxy(client, createProxyHandler());
+}
+export default monitor;
+//# sourceMappingURL=proxy.js.map

package/dist/instrumentation/proxy.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"proxy.js","sourceRoot":"","sources":["../../src/instrumentation/proxy.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EACL,KAAK,EACL,OAAO,EACP,WAAW,EACX,QAAQ,EACR,cAAc,GAIf,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AACrD,OAAO,EACL,eAAe,EACf,aAAa,EACb,iBAAiB,EACjB,YAAY,GACb,MAAM,wBAAwB,CAAC;AAEhC,OAAO,EAAE,aAAa,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAC/D,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAEtC,sBAAsB;AACtB,IAAI,MAAM,GAAkB,IAAI,CAAC;AAEjC,iCAAiC;AACjC,IAAI,eAAe,GAA2B,IAAI,CAAC;AACnD,IAAI,mBAAmB,GAAyB,IAAI,CAAC;AAErD;;GAEG;AACH,KAAK,UAAU,oBAAoB;IACjC,IAAI,eAAe;QAAE,OAAO;IAC5B,IAAI,mBAAmB,EAAE,CAAC;QACxB,MAAM,mBAAmB,CAAC;QAC1B,OAAO;IACT,CAAC;IAED,mBAAmB,GAAG,YAAY,EAAE;SACjC,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE;QACjB,eAAe,GAAG,QAAQ,CAAC;IAC7B,CAAC,CAAC;SACD,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;QACb,MAAM,CAAC,IAAI,CAAC,kCAAkC,EAAE,GAAG,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEL,MAAM,mBAAmB,CAAC;AAC5B,CAAC;AAED,qCAAqC;AACrC,oBAAoB,EAAE,CAAC;AAEvB;;GAEG;AACH,SAAS,SAAS;IAChB,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,GAAG,KAAK,CAAC,SAAS,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;IAClD,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;GAEG;AACH,SAAS,oBAAoB;IAC3B,MAAM,cAAc,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IACxC,MAAM,OAAO,GAAG,WAAW,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC;IAEvD,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,MAAM,UAAU,GAA2B,EAAE,CAAC;IAC9C,OAAO,CAAC,aAAa,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE;QAC/C,UAAU,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,OAAO,UAAU,CAAC;AACpB,CAAC;AAED;;GAEG;AACH,SAAS,cAAc,CAAC,MAAe;IACrC,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ;QAAE,OAAO,SAAS,CAAC;IAE5D,MAAM,CAAC,GAAG,MAAiC,CAAC;IAE5C,+BAA+B;IAC/B,IACE,MAAM,IAAI,CAAC;QACX,OAAO,CAAC,CAAC,IAAI,KAAK,QAAQ;QAC1B,CAAC,CAAC,IAAI;QACN,aAAa,IAAI,CAAC,CAAC,IAAI,EACvB,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,iCAAiC;IACjC,IAAI,UAAU,IAAI,CAAC,IAAI,OAAO,CAAC,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;QACtD,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,+BAA+B;IAC/B,IAAI,WAAW,IAAI,CAAC,IAAI,OAAO,CAAC,CAAC,SAAS,KAAK,UAAU,EAAE,CAAC;QAC1D,OAAO,KAAK,CAAC;IACf,CAAC;IAED,+BAA+B;IAC/B,IAAI,cAAc,IAAI,CAAC,IAAI,OAAO,CAAC,CAAC,YAAY,KAAK,QAAQ,EAAE,CAAC;QAC9D,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;GAEG;AACH,SAAS,SAAS,CAAC,KAAc;IAC/B,OAAO,CACL,KAAK,KAAK,IAAI;QACd,OAAO,KAAK,KAAK,QAAQ;QACzB,MAAM,IAAI,KAAK;QACf,OAAQ,KAA2B,CAAC,IAAI,KAAK,UAAU,CACxD,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,eAAe,CAAC,KAAc;IACrC,OAAO,CACL,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,aAAa,IAAI,KAAK,CAC7E,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,kBAAkB,CAAC,KAAgB;IAC1C,MAAM,QAAQ,GAAG,eAAe,IAAI,iBAAiB,EAAE,CAAC;IACxD,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,MAAM,CAAC,IAAI,CAAC,6CAA6C,CAAC,CAAC;QAC3D,OAAO,CAAC,CAAC;IACX,CAAC;IAED,MAAM,OAAO,GAAG,eAAe,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;IACvE,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,MAAM,CAAC,IAAI,CAAC,uBAAuB,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC;QACpE,OAAO,CAAC,CAAC;IACX,CAAC;IAED,OAAO,aAAa,CAClB;QACE,UAAU,EAAE,KAAK,CAAC,UAAU;QAC5B,WAAW,EAAE,KAAK,CAAC,WAAW;QAC9B,gBAAgB,EAAE,KAAK,CAAC,gBAAgB;KACzC,EACD,OAAO,CACR,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,cAAc,CAAC,IAAU,EAAE,KAAgB;IAClD,MAAM,UAAU,GAAe;QAC7B,CAAC,aAAa,CAAC,QAAQ,CAAC,EAAE,KAAK,CAAC,QAAQ;QACxC,CAAC,aAAa,CAAC,KAAK,CAAC,EAAE,KAAK,CAAC,KAAK;KACnC,CAAC;IAEF,IAAI,KAAK,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;QACnC,UAAU,CAAC,aAAa,CAAC,WAAW,CAAC,GAAG,KAAK,CAAC,UAAU,CAAC;QACzD,UAAU,CAAC,iBAAiB,CAAC,YAAY,CAAC,GAAG,KAAK,CAAC,UAAU,CAAC;IAChE,CAAC;IAED,IAAI,KAAK,CAAC,WAAW,KAAK,SAAS,EAAE,CAAC;QACpC,UAAU,CAAC,aAAa,CAAC,YAAY,CAAC,GAAG,KAAK,CAAC,WAAW,CAAC;QAC3D,UAAU,CAAC,iBAAiB,CAAC,aAAa,CAAC,GAAG,KAAK,CAAC,WAAW,CAAC;IAClE,CAAC;IAED,UAAU,CAAC,iBAAiB,CAAC,KAAK,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC;IAClD,UAAU,CAAC,iBAAiB,CAAC,MAAM,CAAC,GAAG,KAAK,CAAC,QAAQ,CAAC;IAEtD,2CAA2C;IAC3C,MAAM,OAAO,GAAG,kBAAkB,CAAC,KAAK,CAAC,CAAC;IAC1C,UAAU,CAAC,aAAa,CAAC,QAAQ,CAAC,GAAG,OAAO,CAAC;IAE7C,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;AACjC,CAAC;AAED;;GAEG;AACH,SAAS,iBAAiB,CACxB,QAA0B,EAC1B,IAAU,EACV,QAAgB,EAChB,UAAoB,EACpB,IAAe;IAEf,MAAM,gBAAgB,GAAG,QAAQ,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC;IAC1D,IAAI,gBAAgB,GAAqB,IAAI,CAAC;IAE9C,MAAM,eAAe,GAAqB;QACxC,KAAK,CAAC,IAAI;YACR,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,gBAAgB,CAAC,IAAI,EAAE,CAAC;gBAE7C,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;oBAChB,sDAAsD;oBACtD,IAAI,gBAAgB,EAAE,CAAC;wBACrB,cAAc,CAAC,IAAI,EAAE,gBAAgB,CAAC,CAAC;oBACzC,CAAC;oBACD,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,cAAc,CAAC,EAAE,EAAE,CAAC,CAAC;oBAC5C,IAAI,CAAC,GAAG,EAAE,CAAC;gBACb,CAAC;qBAAM,CAAC;oBACN,6EAA6E;oBAC7E,MAAM,KAAK,GAAG,MAAM,CAAC,KAAgC,CAAC;oBACtD,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,IAAI,KAAK,EAAE,CAAC;wBAC3D,MAAM,KAAK,GAAG,YAAY,CAAC,UAAU,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC;wBAC9D,IAAI,KAAK,EAAE,CAAC;4BACV,gBAAgB,GAAG,KAAK,CAAC;wBAC3B,CAAC;oBACH,CAAC;gBACH,CAAC;gBAED,OAAO,MAAM,CAAC;YAChB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,eAAe,CAAC,KAAc,CAAC,CAAC;gBACrC,IAAI,CAAC,SAAS,CAAC;oBACb,IAAI,EAAE,cAAc,CAAC,KAAK;oBAC1B,OAAO,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;iBAChE,CAAC,CAAC;gBACH,IAAI,CAAC,GAAG,EAAE,CAAC;gBACX,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;QAED,KAAK,CAAC,MAAM,CAAC,KAAe;YAC1B,mEAAmE;YACnE,IAAI,gBAAgB,EAAE,CAAC;gBACrB,cAAc,CAAC,IAAI,EAAE,gBAAgB,CAAC,CAAC;YACzC,CAAC;YACD,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,cAAc,CAAC,EAAE,EAAE,CAAC,CAAC;YAC5C,IAAI,CAAC,GAAG,EAAE,CAAC;YAEX,IAAI,gBAAgB,CAAC,MAAM,EAAE,CAAC;gBAC5B,OAAO,gBAAgB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACxC,CAAC;YACD,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,SAAc,EAAE,CAAC;QAC/C,CAAC;QAED,KAAK,CAAC,KAAK,CAAC,KAAe;YACzB,6DAA6D;YAC7D,IAAI,gBAAgB,EAAE,CAAC;gBACrB,cAAc,CAAC,IAAI,EAAE,gBAAgB,CAAC,CAAC;YACzC,CAAC;YACD,IAAI,CAAC,eAAe,CAAC,KAAc,CAAC,CAAC;YACrC,IAAI,CAAC,SAAS,CAAC;gBACb,IAAI,EAAE,cAAc,CAAC,KAAK;gBAC1B,OAAO,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;aAChE,CAAC,CAAC;YACH,IAAI,CAAC,GAAG,EAAE,CAAC;YAEX,IAAI,gBAAgB,CAAC,KAAK,EAAE,CAAC;gBAC3B,OAAO,gBAAgB,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACvC,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;KACF,CAAC;IAEF,OAAO;QACL,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,GAAG,EAAE,CAAC,eAAe;KAC9C,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;;;;;GAeG;AACH,MAAM,UAAU,OAAO,CACrB,MAAS,EACT,UAA0B,EAAE;IAE5B,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,cAAc,CAAC,MAAM,CAAC,CAAC;IAC5D,MAAM,UAAU,GAAG,OAAO,CAAC,IAAI,IAAI,QAAQ,CAAC;IAC5C,MAAM,cAAc,GAAG,OAAO,CAAC,UAAU,IAAI,EAAE,CAAC;IAEhD;;OAEG;IACH,SAAS,kBAAkB,CAAC,OAAiB,EAAE;QAC7C,OAAO;YACL,GAAG,CAAC,MAAc,EAAE,IAAqB,EAAE,QAAiB;gBAC1D,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC;gBAElD,uCAAuC;gBACvC,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;oBACrD,OAAO,KAAK,CAAC;gBACf,CAAC;gBAED,8BAA8B;gBAC9B,IAAI,OAAO,KAAK,KAAK,UAAU,EAAE,CAAC;oBAChC,OAAO,UAAyB,GAAG,IAAe;wBAChD,MAAM,UAAU,GAAG,CAAC,GAAG,IAAI,EAAE,IAAI,CAAC,CAAC;wBACnC,MAAM,QAAQ,GAAG,GAAG,UAAU,IAAI,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;wBAEzD,8DAA8D;wBAC9D,MAAM,YAAY,GAAG,oBAAoB,EAAE,CAAC;wBAE5C,4CAA4C;wBAC5C,MAAM,aAAa,GAAG,SAAS,EAAE,CAAC;wBAClC,MAAM,IAAI,GAAG,aAAa,CAAC,SAAS,CAAC,QAAQ,EAAE;4BAC7C,IAAI,EAAE,QAAQ,CAAC,MAAM;4BACrB,UAAU,EAAE;gCACV,GAAG,cAAc;gCACjB,GAAG,YAAY,EAAE,qDAAqD;gCACtE,CAAC,aAAa,CAAC,QAAQ,CAAC,EAAE,QAAQ;gCAClC,aAAa,EAAE,UAAU;gCACzB,YAAY,EAAE,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC;6BACnC;yBACF,CAAC,CAAC;wBAEH,8BAA8B;wBAC9B,OAAO,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC,EAAE,GAAG,EAAE;4BAC9D,IAAI,CAAC;gCACH,MAAM,MAAM,GAAG,KAAK,CAAC,KAAK,CACxB,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,EACjC,IAAI,CACL,CAAC;gCAEF,uBAAuB;gCACvB,IAAI,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;oCACtB,OAAO,MAAM,CAAC,IAAI,CAChB,CAAC,QAAQ,EAAE,EAAE;wCACX,8BAA8B;wCAC9B,IAAI,eAAe,CAAC,QAAQ,CAAC,EAAE,CAAC;4CAC9B,OAAO,iBAAiB,CACtB,QAAkC,EAClC,IAAI,EACJ,QAAQ,EACR,UAAU,EACV,IAAI,CACL,CAAC;wCACJ,CAAC;wCAED,4BAA4B;wCAC5B,MAAM,KAAK,GAAG,YAAY,CACxB,UAAU,EACV,QAAQ,EACR,IAAI,EACJ,QAAQ,CACT,CAAC;wCACF,IAAI,KAAK,EAAE,CAAC;4CACV,cAAc,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;wCAC9B,CAAC;wCAED,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,cAAc,CAAC,EAAE,EAAE,CAAC,CAAC;wCAC5C,IAAI,CAAC,GAAG,EAAE,CAAC;wCACX,OAAO,QAAQ,CAAC;oCAClB,CAAC,EACD,CAAC,KAAK,EAAE,EAAE;wCACR,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;wCAC5B,IAAI,CAAC,SAAS,CAAC;4CACb,IAAI,EAAE,cAAc,CAAC,KAAK;4CAC1B,OAAO,EACL,KAAK,YAAY,KAAK;gDACpB,CAAC,CAAC,KAAK,CAAC,OAAO;gDACf,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;yCACpB,CAAC,CAAC;wCACH,IAAI,CAAC,GAAG,EAAE,CAAC;wCACX,MAAM,KAAK,CAAC;oCACd,CAAC,CACF,CAAC;gCACJ,CAAC;gCAED,2CAA2C;gCAC3C,IAAI,eAAe,CAAC,MAAM,CAAC,EAAE,CAAC;oCAC5B,OAAO,iBAAiB,CACtB,MAAgC,EAChC,IAAI,EACJ,QAAQ,EACR,UAAU,EACV,IAAI,CACL,CAAC;gCACJ,CAAC;gCAED,MAAM,KAAK,GAAG,YAAY,CAAC,UAAU,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC;gCAC/D,IAAI,KAAK,EAAE,CAAC;oCACV,cAAc,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;gCAC9B,CAAC;gCAED,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,cAAc,CAAC,EAAE,EAAE,CAAC,CAAC;gCAC5C,IAAI,CAAC,GAAG,EAAE,CAAC;gCACX,OAAO,MAAM,CAAC;4BAChB,CAAC;4BAAC,OAAO,KAAK,EAAE,CAAC;gCACf,IAAI,CAAC,eAAe,CAAC,KAAc,CAAC,CAAC;gCACrC,IAAI,CAAC,SAAS,CAAC;oCACb,IAAI,EAAE,cAAc,CAAC,KAAK;oCAC1B,OAAO,EACL,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;iCACzD,CAAC,CAAC;gCACH,IAAI,CAAC,GAAG,EAAE,CAAC;gCACX,MAAM,KAAK,CAAC;4BACd,CAAC;wBACH,CAAC,CAAC,CAAC;oBACL,CAAC,CAAC;gBACJ,CAAC;gBAED,0CAA0C;gBAC1C,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;oBAChD,OAAO,IAAI,KAAK,CACd,KAAe,EACf,kBAAkB,CAAC,CAAC,GAAG,IAAI,EAAE,IAAI,CAAC,CAAC,CACpC,CAAC;gBACJ,CAAC;gBAED,OAAO,KAAK,CAAC;YACf,CAAC;SACF,CAAC;IACJ,CAAC;IAED,OAAO,IAAI,KAAK,CAAC,MAAM,EAAE,kBAAkB,EAAE,CAAM,CAAC;AACtD,CAAC;AAED,eAAe,OAAO,CAAC"}

package/dist/instrumentation/strategies/index.d.ts

@@ -0,0 +1,55 @@
+/**
+ * Extraction Strategies
+ *
+ * Each strategy knows how to extract usage data from a specific provider's API responses.
+ */
+import type { ExtractionStrategy, UsageData } from "../../types.js";
+/**
+ * OpenAI extraction strategy
+ */
+export declare const openaiStrategy: ExtractionStrategy;
+/**
+ * Anthropic extraction strategy
+ */
+export declare const anthropicStrategy: ExtractionStrategy;
+/**
+ * fal.ai extraction strategy
+ */
+export declare const falStrategy: ExtractionStrategy;
+/**
+ * ElevenLabs extraction strategy
+ */
+export declare const elevenlabsStrategy: ExtractionStrategy;
+/**
+ * AWS Bedrock extraction strategy
+ */
+export declare const bedrockStrategy: ExtractionStrategy;
+/**
+ * Google Vertex AI / Gemini extraction strategy
+ */
+export declare const vertexAIStrategy: ExtractionStrategy;
+/**
+ * BFL (Black Forest Labs) extraction strategy
+ *
+ * Note: BFL responses are similar to fal.ai but use "id" (not "requestId")
+ * and have "sample" field for images. fal.ai uses "requestId" (camelCase).
+ */
+export declare const bflStrategy: ExtractionStrategy;
+/**
+ * Vercel AI SDK extraction strategy
+ * Handles generateText, streamText, generateObject, streamObject results
+ */
+export declare const vercelAIStrategy: ExtractionStrategy;
+/**
+ * All registered strategies
+ */
+export declare const strategies: ExtractionStrategy[];
+/**
+ * Find the appropriate strategy for a given result
+ */
+export declare function findStrategy(methodPath: string[], result: unknown): ExtractionStrategy | null;
+/**
+ * Extract usage data using the appropriate strategy
+ */
+export declare function extractUsage(methodPath: string[], result: unknown, args: unknown[], providerHint?: string): UsageData | null;
+//# sourceMappingURL=index.d.ts.map

package/dist/instrumentation/strategies/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/instrumentation/strategies/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,kBAAkB,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAEpE;;GAEG;AACH,eAAO,MAAM,cAAc,EAAE,kBAwD5B,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,iBAAiB,EAAE,kBAqD/B,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,WAAW,EAAE,kBA4DzB,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,kBAAkB,EAAE,kBAyDhC,CAAC;AAsBF;;GAEG;AACH,eAAO,MAAM,eAAe,EAAE,kBA8D7B,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,gBAAgB,EAAE,kBAwD9B,CAAC;AAEF;;;;;GAKG;AACH,eAAO,MAAM,WAAW,EAAE,kBAgEzB,CAAC;AAEF;;;GAGG;AACH,eAAO,MAAM,gBAAgB,EAAE,kBA0E9B,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,UAAU,EAAE,kBAAkB,EAS1C,CAAC;AAEF;;GAEG;AACH,wBAAgB,YAAY,CAC1B,UAAU,EAAE,MAAM,EAAE,EACpB,MAAM,EAAE,OAAO,GACd,kBAAkB,GAAG,IAAI,CAO3B;AAED;;GAEG;AACH,wBAAgB,YAAY,CAC1B,UAAU,EAAE,MAAM,EAAE,EACpB,MAAM,EAAE,OAAO,EACf,IAAI,EAAE,OAAO,EAAE,EACf,YAAY,CAAC,EAAE,MAAM,GACpB,SAAS,GAAG,IAAI,CAgBlB"}