tokenmeter 0.9.0

package/dist/instrumentation/strategies/index.js

@@ -0,0 +1,429 @@
+/**
+ * Extraction Strategies
+ *
+ * Each strategy knows how to extract usage data from a specific provider's API responses.
+ */
+/**
+ * OpenAI extraction strategy
+ */
+export const openaiStrategy = {
+    provider: "openai",
+    canHandle(methodPath, result) {
+        // Handle chat.completions.create, completions.create, embeddings.create, etc.
+        if (!result || typeof result !== "object")
+            return false;
+        const r = result;
+        // Check for OpenAI response structure
+        return ("usage" in r &&
+            typeof r.usage === "object" &&
+            r.usage !== null &&
+            ("prompt_tokens" in r.usage ||
+                "total_tokens" in r.usage));
+    },
+    extract(methodPath, result, args) {
+        const r = result;
+        if (!r.usage)
+            return null;
+        // Extract model from result or from request args
+        let model = r.model || "unknown";
+        if (!r.model && args.length > 0) {
+            const params = args[0];
+            if (params?.model) {
+                model = params.model;
+            }
+        }
+        return {
+            provider: "openai",
+            model,
+            inputUnits: r.usage.prompt_tokens,
+            outputUnits: r.usage.completion_tokens,
+            cachedInputUnits: r.usage.cached_tokens,
+            metadata: {
+                totalTokens: r.usage.total_tokens,
+            },
+        };
+    },
+};
+/**
+ * Anthropic extraction strategy
+ */
+export const anthropicStrategy = {
+    provider: "anthropic",
+    canHandle(methodPath, result) {
+        if (!result || typeof result !== "object")
+            return false;
+        const r = result;
+        // Anthropic responses have usage with input_tokens/output_tokens
+        return ("usage" in r &&
+            typeof r.usage === "object" &&
+            r.usage !== null &&
+            "input_tokens" in r.usage);
+    },
+    extract(methodPath, result, args) {
+        const r = result;
+        if (!r.usage)
+            return null;
+        let model = r.model || "unknown";
+        if (!r.model && args.length > 0) {
+            const params = args[0];
+            if (params?.model) {
+                model = params.model;
+            }
+        }
+        return {
+            provider: "anthropic",
+            model,
+            inputUnits: r.usage.input_tokens,
+            outputUnits: r.usage.output_tokens,
+            cachedInputUnits: r.usage.cache_read_input_tokens,
+            metadata: {
+                cacheCreationTokens: r.usage.cache_creation_input_tokens,
+            },
+        };
+    },
+};
+/**
+ * fal.ai extraction strategy
+ */
+export const falStrategy = {
+    provider: "fal",
+    canHandle(methodPath, result) {
+        if (!result || typeof result !== "object")
+            return false;
+        const r = result;
+        // fal.ai responses typically have requestId and data
+        return "requestId" in r || "request_id" in r;
+    },
+    extract(methodPath, result, args) {
+        const r = result;
+        // Extract endpoint ID from args
+        let model = "unknown";
+        if (args.length > 0 && typeof args[0] === "string") {
+            // fal.subscribe("fal-ai/flux-pro", {...})
+            model = args[0].replace("fal-ai/", "");
+        }
+        const data = r.data || r;
+        let outputUnits = 1; // Default to 1 for request-based pricing
+        // Image generation: count images
+        if ("images" in data && Array.isArray(data.images)) {
+            outputUnits = data.images.length;
+        }
+        else if ("image" in data && data.image) {
+            outputUnits = 1;
+        }
+        // Video generation: use duration in seconds
+        if ("video" in data && data.video && "duration" in data) {
+            outputUnits = data.duration || 1;
+        }
+        return {
+            provider: "fal",
+            model,
+            outputUnits,
+            metadata: {
+                requestId: r.requestId || r.request_id,
+            },
+        };
+    },
+};
+/**
+ * ElevenLabs extraction strategy
+ */
+export const elevenlabsStrategy = {
+    provider: "elevenlabs",
+    canHandle(methodPath, result) {
+        // ElevenLabs returns audio buffers, we detect by method path
+        return (methodPath.includes("textToSpeech") ||
+            methodPath.includes("generate") ||
+            methodPath.includes("convert"));
+    },
+    extract(methodPath, result, args) {
+        // For ElevenLabs, we need to extract character count from the input
+        // The result is typically a Buffer/ArrayBuffer
+        let text = "";
+        let model = "eleven_multilingual_v2"; // Default model
+        // Extract from args based on method signature
+        if (args.length >= 2) {
+            // textToSpeech.convert(voiceId, { text, modelId })
+            const options = args[1];
+            if (options?.text) {
+                text = options.text;
+            }
+            if (options?.modelId || options?.model_id) {
+                model = options.modelId || options.model_id || model;
+            }
+        }
+        else if (args.length === 1) {
+            // generate({ text, voice, modelId })
+            const options = args[0];
+            if (options?.text) {
+                text = options.text;
+            }
+            if (options?.modelId || options?.model_id) {
+                model = options.modelId || options.model_id || model;
+            }
+        }
+        return {
+            provider: "elevenlabs",
+            model,
+            inputUnits: text.length, // Character count
+            metadata: {
+                characterCount: text.length,
+            },
+        };
+    },
+};
+/**
+ * Parse Bedrock model ID to extract canonical model name
+ *
+ * @example
+ * "us.anthropic.claude-sonnet-4-20250514-v1:0" -> "anthropic.claude-sonnet-4"
+ * "eu.anthropic.claude-3-5-sonnet-20241022-v2:0" -> "anthropic.claude-3-5-sonnet"
+ */
+function parseBedrockModelId(modelId) {
+    // Remove region prefix (us., eu., ap., etc.)
+    let parsed = modelId.replace(/^[a-z]{2}\./, "");
+    // Remove version suffix (-v1:0, -v2:0, etc.)
+    parsed = parsed.replace(/-v\d+:\d+$/, "");
+    // Remove date suffix (-20250514, -20241022, etc.)
+    parsed = parsed.replace(/-\d{8}$/, "");
+    return parsed;
+}
+/**
+ * AWS Bedrock extraction strategy
+ */
+export const bedrockStrategy = {
+    provider: "bedrock",
+    canHandle(methodPath, result) {
+        if (!result || typeof result !== "object")
+            return false;
+        const r = result;
+        // Bedrock responses have usage with inputTokens/outputTokens
+        // and typically include $metadata or modelId
+        return ("usage" in r &&
+            typeof r.usage === "object" &&
+            r.usage !== null &&
+            "inputTokens" in r.usage &&
+            ("modelId" in r || "$metadata" in r));
+    },
+    extract(methodPath, result, args) {
+        const r = result;
+        if (!r.usage)
+            return null;
+        let model = r.modelId || "unknown";
+        // Try to get model from args if not in result
+        if (!r.modelId && args.length > 0) {
+            const params = args[0];
+            if (params?.modelId) {
+                model = params.modelId;
+            }
+        }
+        // Parse the Bedrock model ID to canonical form
+        const canonicalModel = parseBedrockModelId(model);
+        return {
+            provider: "bedrock",
+            model: canonicalModel,
+            inputUnits: r.usage.inputTokens,
+            outputUnits: r.usage.outputTokens,
+            metadata: {
+                originalModelId: model,
+                requestId: r.$metadata?.requestId,
+            },
+        };
+    },
+};
+/**
+ * Google Vertex AI / Gemini extraction strategy
+ */
+export const vertexAIStrategy = {
+    provider: "google",
+    canHandle(methodPath, result) {
+        if (!result || typeof result !== "object")
+            return false;
+        const r = result;
+        // Vertex AI / Gemini responses have usageMetadata with promptTokenCount/candidatesTokenCount
+        return ("usageMetadata" in r &&
+            typeof r.usageMetadata === "object" &&
+            r.usageMetadata !== null &&
+            "promptTokenCount" in r.usageMetadata);
+    },
+    extract(methodPath, result, args) {
+        const r = result;
+        if (!r.usageMetadata)
+            return null;
+        // Try to extract model from result or args
+        let model = "unknown";
+        if (r.modelVersion) {
+            model = r.modelVersion;
+        }
+        else if (args.length > 0) {
+            const params = args[0];
+            if (params?.model) {
+                model = params.model;
+            }
+        }
+        return {
+            provider: "google",
+            model,
+            inputUnits: r.usageMetadata.promptTokenCount,
+            outputUnits: r.usageMetadata.candidatesTokenCount,
+            cachedInputUnits: r.usageMetadata.cachedContentTokenCount,
+            metadata: {
+                totalTokens: r.usageMetadata.totalTokenCount,
+            },
+        };
+    },
+};
+/**
+ * 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 const bflStrategy = {
+    provider: "bfl",
+    canHandle(methodPath, result) {
+        if (!result || typeof result !== "object")
+            return false;
+        const r = result;
+        // BFL API responses have:
+        // - "id" field (not "requestId" which is fal.ai's format)
+        // - "sample" field with base64 image data
+        // Explicitly exclude fal.ai responses which have "requestId" (camelCase)
+        if ("requestId" in r)
+            return false;
+        return "id" in r && ("sample" in r || "images" in r);
+    },
+    extract(methodPath, result, args) {
+        const r = result;
+        // Extract model from args (BFL uses model in request body or endpoint)
+        let model = "flux-pro"; // Default
+        if (args.length > 0) {
+            const params = args[0];
+            if (params?.model) {
+                model = params.model;
+            }
+            else if (params?.endpoint) {
+                // Extract model from endpoint like "/v1/flux-pro-1.1"
+                const match = params.endpoint.match(/\/(flux-[\w.-]+)/);
+                if (match) {
+                    model = match[1];
+                }
+            }
+        }
+        // Count output images
+        let outputUnits = 1;
+        if (r.images && Array.isArray(r.images)) {
+            outputUnits = r.images.length;
+        }
+        else if (r.sample && Array.isArray(r.sample)) {
+            outputUnits = r.sample.length;
+        }
+        return {
+            provider: "bfl",
+            model,
+            outputUnits,
+            metadata: {
+                requestId: r.id || r.request_id,
+            },
+        };
+    },
+};
+/**
+ * Vercel AI SDK extraction strategy
+ * Handles generateText, streamText, generateObject, streamObject results
+ */
+export const vercelAIStrategy = {
+    provider: "vercel-ai",
+    canHandle(methodPath, result) {
+        if (!result || typeof result !== "object")
+            return false;
+        const r = result;
+        // Vercel AI SDK responses have usage with promptTokens/completionTokens
+        return ("usage" in r &&
+            typeof r.usage === "object" &&
+            r.usage !== null &&
+            "promptTokens" in r.usage);
+    },
+    extract(methodPath, result, args) {
+        const r = result;
+        if (!r.usage)
+            return null;
+        // Try to determine provider and model
+        let provider = "unknown";
+        let model = "unknown";
+        if (r.response?.modelId) {
+            model = r.response.modelId;
+            // Infer provider from model
+            if (model.startsWith("gpt-") ||
+                model.startsWith("o1") ||
+                model.startsWith("o3")) {
+                provider = "openai";
+            }
+            else if (model.startsWith("claude-")) {
+                provider = "anthropic";
+            }
+            else if (model.startsWith("gemini-")) {
+                provider = "google";
+            }
+        }
+        // Try to get model from args
+        if (args.length > 0) {
+            const params = args[0];
+            if (params?.model?.modelId) {
+                model = params.model.modelId;
+            }
+            if (params?.model?.provider) {
+                provider = params.model.provider;
+            }
+        }
+        return {
+            provider,
+            model,
+            inputUnits: r.usage.promptTokens,
+            outputUnits: r.usage.completionTokens,
+        };
+    },
+};
+/**
+ * All registered strategies
+ */
+export const strategies = [
+    openaiStrategy,
+    anthropicStrategy,
+    bedrockStrategy,
+    vertexAIStrategy,
+    falStrategy,
+    bflStrategy,
+    elevenlabsStrategy,
+    vercelAIStrategy,
+];
+/**
+ * Find the appropriate strategy for a given result
+ */
+export function findStrategy(methodPath, result) {
+    for (const strategy of strategies) {
+        if (strategy.canHandle(methodPath, result)) {
+            return strategy;
+        }
+    }
+    return null;
+}
+/**
+ * Extract usage data using the appropriate strategy
+ */
+export function extractUsage(methodPath, result, args, providerHint) {
+    // If provider hint is given, try that strategy first
+    if (providerHint) {
+        const hintedStrategy = strategies.find((s) => s.provider === providerHint);
+        if (hintedStrategy?.canHandle(methodPath, result)) {
+            return hintedStrategy.extract(methodPath, result, args);
+        }
+    }
+    // Otherwise, try all strategies
+    const strategy = findStrategy(methodPath, result);
+    if (strategy) {
+        return strategy.extract(methodPath, result, args);
+    }
+    return null;
+}
+//# sourceMappingURL=index.js.map

package/dist/instrumentation/strategies/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/instrumentation/strategies/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAIH;;GAEG;AACH,MAAM,CAAC,MAAM,cAAc,GAAuB;IAChD,QAAQ,EAAE,QAAQ;IAElB,SAAS,CAAC,UAAoB,EAAE,MAAe;QAC7C,8EAA8E;QAC9E,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ;YAAE,OAAO,KAAK,CAAC;QAExD,MAAM,CAAC,GAAG,MAAiC,CAAC;QAE5C,sCAAsC;QACtC,OAAO,CACL,OAAO,IAAI,CAAC;YACZ,OAAO,CAAC,CAAC,KAAK,KAAK,QAAQ;YAC3B,CAAC,CAAC,KAAK,KAAK,IAAI;YAChB,CAAC,eAAe,IAAK,CAAC,CAAC,KAAgB;gBACrC,cAAc,IAAK,CAAC,CAAC,KAAgB,CAAC,CACzC,CAAC;IACJ,CAAC;IAED,OAAO,CACL,UAAoB,EACpB,MAAe,EACf,IAAe;QAEf,MAAM,CAAC,GAAG,MAQT,CAAC;QAEF,IAAI,CAAC,CAAC,CAAC,KAAK;YAAE,OAAO,IAAI,CAAC;QAE1B,iDAAiD;QACjD,IAAI,KAAK,GAAG,CAAC,CAAC,KAAK,IAAI,SAAS,CAAC;QACjC,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChC,MAAM,MAAM,GAAG,IAAI,CAAC,CAAC,CAAmC,CAAC;YACzD,IAAI,MAAM,EAAE,KAAK,EAAE,CAAC;gBAClB,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;YACvB,CAAC;QACH,CAAC;QAED,OAAO;YACL,QAAQ,EAAE,QAAQ;YAClB,KAAK;YACL,UAAU,EAAE,CAAC,CAAC,KAAK,CAAC,aAAa;YACjC,WAAW,EAAE,CAAC,CAAC,KAAK,CAAC,iBAAiB;YACtC,gBAAgB,EAAE,CAAC,CAAC,KAAK,CAAC,aAAa;YACvC,QAAQ,EAAE;gBACR,WAAW,EAAE,CAAC,CAAC,KAAK,CAAC,YAAY;aAClC;SACF,CAAC;IACJ,CAAC;CACF,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,MAAM,iBAAiB,GAAuB;IACnD,QAAQ,EAAE,WAAW;IAErB,SAAS,CAAC,UAAoB,EAAE,MAAe;QAC7C,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ;YAAE,OAAO,KAAK,CAAC;QAExD,MAAM,CAAC,GAAG,MAAiC,CAAC;QAE5C,iEAAiE;QACjE,OAAO,CACL,OAAO,IAAI,CAAC;YACZ,OAAO,CAAC,CAAC,KAAK,KAAK,QAAQ;YAC3B,CAAC,CAAC,KAAK,KAAK,IAAI;YAChB,cAAc,IAAK,CAAC,CAAC,KAAgB,CACtC,CAAC;IACJ,CAAC;IAED,OAAO,CACL,UAAoB,EACpB,MAAe,EACf,IAAe;QAEf,MAAM,CAAC,GAAG,MAQT,CAAC;QAEF,IAAI,CAAC,CAAC,CAAC,KAAK;YAAE,OAAO,IAAI,CAAC;QAE1B,IAAI,KAAK,GAAG,CAAC,CAAC,KAAK,IAAI,SAAS,CAAC;QACjC,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChC,MAAM,MAAM,GAAG,IAAI,CAAC,CAAC,CAAmC,CAAC;YACzD,IAAI,MAAM,EAAE,KAAK,EAAE,CAAC;gBAClB,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;YACvB,CAAC;QACH,CAAC;QAED,OAAO;YACL,QAAQ,EAAE,WAAW;YACrB,KAAK;YACL,UAAU,EAAE,CAAC,CAAC,KAAK,CAAC,YAAY;YAChC,WAAW,EAAE,CAAC,CAAC,KAAK,CAAC,aAAa;YAClC,gBAAgB,EAAE,CAAC,CAAC,KAAK,CAAC,uBAAuB;YACjD,QAAQ,EAAE;gBACR,mBAAmB,EAAE,CAAC,CAAC,KAAK,CAAC,2BAA2B;aACzD;SACF,CAAC;IACJ,CAAC;CACF,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,MAAM,WAAW,GAAuB;IAC7C,QAAQ,EAAE,KAAK;IAEf,SAAS,CAAC,UAAoB,EAAE,MAAe;QAC7C,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ;YAAE,OAAO,KAAK,CAAC;QAExD,MAAM,CAAC,GAAG,MAAiC,CAAC;QAE5C,qDAAqD;QACrD,OAAO,WAAW,IAAI,CAAC,IAAI,YAAY,IAAI,CAAC,CAAC;IAC/C,CAAC;IAED,OAAO,CACL,UAAoB,EACpB,MAAe,EACf,IAAe;QAEf,MAAM,CAAC,GAAG,MAUT,CAAC;QAEF,gCAAgC;QAChC,IAAI,KAAK,GAAG,SAAS,CAAC;QACtB,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,IAAI,OAAO,IAAI,CAAC,CAAC,CAAC,KAAK,QAAQ,EAAE,CAAC;YACnD,0CAA0C;YAC1C,KAAK,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QACzC,CAAC;QAED,MAAM,IAAI,GAAG,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC;QACzB,IAAI,WAAW,GAAG,CAAC,CAAC,CAAC,yCAAyC;QAE9D,iCAAiC;QACjC,IAAI,QAAQ,IAAI,IAAI,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YACnD,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC;QACnC,CAAC;aAAM,IAAI,OAAO,IAAI,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACzC,WAAW,GAAG,CAAC,CAAC;QAClB,CAAC;QAED,4CAA4C;QAC5C,IAAI,OAAO,IAAI,IAAI,IAAI,IAAI,CAAC,KAAK,IAAI,UAAU,IAAI,IAAI,EAAE,CAAC;YACxD,WAAW,GAAI,IAA8B,CAAC,QAAQ,IAAI,CAAC,CAAC;QAC9D,CAAC;QAED,OAAO;YACL,QAAQ,EAAE,KAAK;YACf,KAAK;YACL,WAAW;YACX,QAAQ,EAAE;gBACR,SAAS,EAAE,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,UAAU;aACvC;SACF,CAAC;IACJ,CAAC;CACF,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,MAAM,kBAAkB,GAAuB;IACpD,QAAQ,EAAE,YAAY;IAEtB,SAAS,CAAC,UAAoB,EAAE,MAAe;QAC7C,6DAA6D;QAC7D,OAAO,CACL,UAAU,CAAC,QAAQ,CAAC,cAAc,CAAC;YACnC,UAAU,CAAC,QAAQ,CAAC,UAAU,CAAC;YAC/B,UAAU,CAAC,QAAQ,CAAC,SAAS,CAAC,CAC/B,CAAC;IACJ,CAAC;IAED,OAAO,CACL,UAAoB,EACpB,MAAe,EACf,IAAe;QAEf,oEAAoE;QACpE,+CAA+C;QAE/C,IAAI,IAAI,GAAG,EAAE,CAAC;QACd,IAAI,KAAK,GAAG,wBAAwB,CAAC,CAAC,gBAAgB;QAEtD,8CAA8C;QAC9C,IAAI,IAAI,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;YACrB,mDAAmD;YACnD,MAAM,OAAO,GAAG,IAAI,CAAC,CAAC,CAET,CAAC;YACd,IAAI,OAAO,EAAE,IAAI,EAAE,CAAC;gBAClB,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;YACtB,CAAC;YACD,IAAI,OAAO,EAAE,OAAO,IAAI,OAAO,EAAE,QAAQ,EAAE,CAAC;gBAC1C,KAAK,GAAG,OAAO,CAAC,OAAO,IAAI,OAAO,CAAC,QAAQ,IAAI,KAAK,CAAC;YACvD,CAAC;QACH,CAAC;aAAM,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC7B,qCAAqC;YACrC,MAAM,OAAO,GAAG,IAAI,CAAC,CAAC,CAET,CAAC;YACd,IAAI,OAAO,EAAE,IAAI,EAAE,CAAC;gBAClB,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;YACtB,CAAC;YACD,IAAI,OAAO,EAAE,OAAO,IAAI,OAAO,EAAE,QAAQ,EAAE,CAAC;gBAC1C,KAAK,GAAG,OAAO,CAAC,OAAO,IAAI,OAAO,CAAC,QAAQ,IAAI,KAAK,CAAC;YACvD,CAAC;QACH,CAAC;QAED,OAAO;YACL,QAAQ,EAAE,YAAY;YACtB,KAAK;YACL,UAAU,EAAE,IAAI,CAAC,MAAM,EAAE,kBAAkB;YAC3C,QAAQ,EAAE;gBACR,cAAc,EAAE,IAAI,CAAC,MAAM;aAC5B;SACF,CAAC;IACJ,CAAC;CACF,CAAC;AAEF;;;;;;GAMG;AACH,SAAS,mBAAmB,CAAC,OAAe;IAC1C,6CAA6C;IAC7C,IAAI,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC;IAEhD,6CAA6C;IAC7C,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;IAE1C,kDAAkD;IAClD,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;IAEvC,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,MAAM,eAAe,GAAuB;IACjD,QAAQ,EAAE,SAAS;IAEnB,SAAS,CAAC,UAAoB,EAAE,MAAe;QAC7C,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ;YAAE,OAAO,KAAK,CAAC;QAExD,MAAM,CAAC,GAAG,MAAiC,CAAC;QAE5C,6DAA6D;QAC7D,6CAA6C;QAC7C,OAAO,CACL,OAAO,IAAI,CAAC;YACZ,OAAO,CAAC,CAAC,KAAK,KAAK,QAAQ;YAC3B,CAAC,CAAC,KAAK,KAAK,IAAI;YAChB,aAAa,IAAK,CAAC,CAAC,KAAgB;YACpC,CAAC,SAAS,IAAI,CAAC,IAAI,WAAW,IAAI,CAAC,CAAC,CACrC,CAAC;IACJ,CAAC;IAED,OAAO,CACL,UAAoB,EACpB,MAAe,EACf,IAAe;QAEf,MAAM,CAAC,GAAG,MAUT,CAAC;QAEF,IAAI,CAAC,CAAC,CAAC,KAAK;YAAE,OAAO,IAAI,CAAC;QAE1B,IAAI,KAAK,GAAG,CAAC,CAAC,OAAO,IAAI,SAAS,CAAC;QAEnC,8CAA8C;QAC9C,IAAI,CAAC,CAAC,CAAC,OAAO,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAClC,MAAM,MAAM,GAAG,IAAI,CAAC,CAAC,CAAqC,CAAC;YAC3D,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;gBACpB,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC;YACzB,CAAC;QACH,CAAC;QAED,+CAA+C;QAC/C,MAAM,cAAc,GAAG,mBAAmB,CAAC,KAAK,CAAC,CAAC;QAElD,OAAO;YACL,QAAQ,EAAE,SAAS;YACnB,KAAK,EAAE,cAAc;YACrB,UAAU,EAAE,CAAC,CAAC,KAAK,CAAC,WAAW;YAC/B,WAAW,EAAE,CAAC,CAAC,KAAK,CAAC,YAAY;YACjC,QAAQ,EAAE;gBACR,eAAe,EAAE,KAAK;gBACtB,SAAS,EAAE,CAAC,CAAC,SAAS,EAAE,SAAS;aAClC;SACF,CAAC;IACJ,CAAC;CACF,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAuB;IAClD,QAAQ,EAAE,QAAQ;IAElB,SAAS,CAAC,UAAoB,EAAE,MAAe;QAC7C,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ;YAAE,OAAO,KAAK,CAAC;QAExD,MAAM,CAAC,GAAG,MAAiC,CAAC;QAE5C,6FAA6F;QAC7F,OAAO,CACL,eAAe,IAAI,CAAC;YACpB,OAAO,CAAC,CAAC,aAAa,KAAK,QAAQ;YACnC,CAAC,CAAC,aAAa,KAAK,IAAI;YACxB,kBAAkB,IAAK,CAAC,CAAC,aAAwB,CAClD,CAAC;IACJ,CAAC;IAED,OAAO,CACL,UAAoB,EACpB,MAAe,EACf,IAAe;QAEf,MAAM,CAAC,GAAG,MAQT,CAAC;QAEF,IAAI,CAAC,CAAC,CAAC,aAAa;YAAE,OAAO,IAAI,CAAC;QAElC,2CAA2C;QAC3C,IAAI,KAAK,GAAG,SAAS,CAAC;QACtB,IAAI,CAAC,CAAC,YAAY,EAAE,CAAC;YACnB,KAAK,GAAG,CAAC,CAAC,YAAY,CAAC;QACzB,CAAC;aAAM,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,IAAI,CAAC,CAAC,CAAmC,CAAC;YACzD,IAAI,MAAM,EAAE,KAAK,EAAE,CAAC;gBAClB,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;YACvB,CAAC;QACH,CAAC;QAED,OAAO;YACL,QAAQ,EAAE,QAAQ;YAClB,KAAK;YACL,UAAU,EAAE,CAAC,CAAC,aAAa,CAAC,gBAAgB;YAC5C,WAAW,EAAE,CAAC,CAAC,aAAa,CAAC,oBAAoB;YACjD,gBAAgB,EAAE,CAAC,CAAC,aAAa,CAAC,uBAAuB;YACzD,QAAQ,EAAE;gBACR,WAAW,EAAE,CAAC,CAAC,aAAa,CAAC,eAAe;aAC7C;SACF,CAAC;IACJ,CAAC;CACF,CAAC;AAEF;;;;;GAKG;AACH,MAAM,CAAC,MAAM,WAAW,GAAuB;IAC7C,QAAQ,EAAE,KAAK;IAEf,SAAS,CAAC,UAAoB,EAAE,MAAe;QAC7C,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ;YAAE,OAAO,KAAK,CAAC;QAExD,MAAM,CAAC,GAAG,MAAiC,CAAC;QAE5C,0BAA0B;QAC1B,0DAA0D;QAC1D,0CAA0C;QAC1C,yEAAyE;QACzE,IAAI,WAAW,IAAI,CAAC;YAAE,OAAO,KAAK,CAAC;QAEnC,OAAO,IAAI,IAAI,CAAC,IAAI,CAAC,QAAQ,IAAI,CAAC,IAAI,QAAQ,IAAI,CAAC,CAAC,CAAC;IACvD,CAAC;IAED,OAAO,CACL,UAAoB,EACpB,MAAe,EACf,IAAe;QAEf,MAAM,CAAC,GAAG,MAMT,CAAC;QAEF,uEAAuE;QACvE,IAAI,KAAK,GAAG,UAAU,CAAC,CAAC,UAAU;QAClC,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpB,MAAM,MAAM,GAAG,IAAI,CAAC,CAAC,CAER,CAAC;YACd,IAAI,MAAM,EAAE,KAAK,EAAE,CAAC;gBAClB,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;YACvB,CAAC;iBAAM,IAAI,MAAM,EAAE,QAAQ,EAAE,CAAC;gBAC5B,sDAAsD;gBACtD,MAAM,KAAK,GAAG,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;gBACxD,IAAI,KAAK,EAAE,CAAC;oBACV,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;gBACnB,CAAC;YACH,CAAC;QACH,CAAC;QAED,sBAAsB;QACtB,IAAI,WAAW,GAAG,CAAC,CAAC;QACpB,IAAI,CAAC,CAAC,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;YACxC,WAAW,GAAG,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC;QAChC,CAAC;aAAM,IAAI,CAAC,CAAC,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;YAC/C,WAAW,GAAG,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC;QAChC,CAAC;QAED,OAAO;YACL,QAAQ,EAAE,KAAK;YACf,KAAK;YACL,WAAW;YACX,QAAQ,EAAE;gBACR,SAAS,EAAE,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,UAAU;aAChC;SACF,CAAC;IACJ,CAAC;CACF,CAAC;AAEF;;;GAGG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAuB;IAClD,QAAQ,EAAE,WAAW;IAErB,SAAS,CAAC,UAAoB,EAAE,MAAe;QAC7C,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ;YAAE,OAAO,KAAK,CAAC;QAExD,MAAM,CAAC,GAAG,MAAiC,CAAC;QAE5C,wEAAwE;QACxE,OAAO,CACL,OAAO,IAAI,CAAC;YACZ,OAAO,CAAC,CAAC,KAAK,KAAK,QAAQ;YAC3B,CAAC,CAAC,KAAK,KAAK,IAAI;YAChB,cAAc,IAAK,CAAC,CAAC,KAAgB,CACtC,CAAC;IACJ,CAAC;IAED,OAAO,CACL,UAAoB,EACpB,MAAe,EACf,IAAe;QAEf,MAAM,CAAC,GAAG,MAQT,CAAC;QAEF,IAAI,CAAC,CAAC,CAAC,KAAK;YAAE,OAAO,IAAI,CAAC;QAE1B,sCAAsC;QACtC,IAAI,QAAQ,GAAG,SAAS,CAAC;QACzB,IAAI,KAAK,GAAG,SAAS,CAAC;QAEtB,IAAI,CAAC,CAAC,QAAQ,EAAE,OAAO,EAAE,CAAC;YACxB,KAAK,GAAG,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC;YAC3B,4BAA4B;YAC5B,IACE,KAAK,CAAC,UAAU,CAAC,MAAM,CAAC;gBACxB,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC;gBACtB,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,EACtB,CAAC;gBACD,QAAQ,GAAG,QAAQ,CAAC;YACtB,CAAC;iBAAM,IAAI,KAAK,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;gBACvC,QAAQ,GAAG,WAAW,CAAC;YACzB,CAAC;iBAAM,IAAI,KAAK,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;gBACvC,QAAQ,GAAG,QAAQ,CAAC;YACtB,CAAC;QACH,CAAC;QAED,6BAA6B;QAC7B,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpB,MAAM,MAAM,GAAG,IAAI,CAAC,CAAC,CAER,CAAC;YACd,IAAI,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;gBAC3B,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC;YAC/B,CAAC;YACD,IAAI,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC;gBAC5B,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC;YACnC,CAAC;QACH,CAAC;QAED,OAAO;YACL,QAAQ;YACR,KAAK;YACL,UAAU,EAAE,CAAC,CAAC,KAAK,CAAC,YAAY;YAChC,WAAW,EAAE,CAAC,CAAC,KAAK,CAAC,gBAAgB;SACtC,CAAC;IACJ,CAAC;CACF,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,MAAM,UAAU,GAAyB;IAC9C,cAAc;IACd,iBAAiB;IACjB,eAAe;IACf,gBAAgB;IAChB,WAAW;IACX,WAAW;IACX,kBAAkB;IAClB,gBAAgB;CACjB,CAAC;AAEF;;GAEG;AACH,MAAM,UAAU,YAAY,CAC1B,UAAoB,EACpB,MAAe;IAEf,KAAK,MAAM,QAAQ,IAAI,UAAU,EAAE,CAAC;QAClC,IAAI,QAAQ,CAAC,SAAS,CAAC,UAAU,EAAE,MAAM,CAAC,EAAE,CAAC;YAC3C,OAAO,QAAQ,CAAC;QAClB,CAAC;IACH,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,YAAY,CAC1B,UAAoB,EACpB,MAAe,EACf,IAAe,EACf,YAAqB;IAErB,qDAAqD;IACrD,IAAI,YAAY,EAAE,CAAC;QACjB,MAAM,cAAc,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,YAAY,CAAC,CAAC;QAC3E,IAAI,cAAc,EAAE,SAAS,CAAC,UAAU,EAAE,MAAM,CAAC,EAAE,CAAC;YAClD,OAAO,cAAc,CAAC,OAAO,CAAC,UAAU,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAED,gCAAgC;IAChC,MAAM,QAAQ,GAAG,YAAY,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;IAClD,IAAI,QAAQ,EAAE,CAAC;QACb,OAAO,QAAQ,CAAC,OAAO,CAAC,UAAU,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;IACpD,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC"}

package/dist/integrations/express/index.d.ts

@@ -0,0 +1,137 @@
+import type { Request, Response, NextFunction, RequestHandler } from "express";
+import type { CostTraceOptions } from "../../types.js";
+/**
+ * Configuration for the Express middleware
+ */
+export interface TokenmeterMiddlewareConfig {
+    /**
+     * Extract the identifier from the request.
+     * This is required and should return a user ID, org ID, or other billing identifier.
+     *
+     * @example
+     * ```typescript
+     * getIdentifier: (req) => req.user?.id || req.headers['x-user-id']
+     * ```
+     */
+    getIdentifier: (req: Request) => string | null | undefined;
+    /**
+     * Optional: Extract a secondary identifier from the request.
+     *
+     * @example
+     * ```typescript
+     * getSecondaryIdentifier: (req) => req.user?.orgId
+     * ```
+     */
+    getSecondaryIdentifier?: (req: Request) => string | null | undefined;
+    /**
+     * Optional: Extract or generate a workflow ID from the request.
+     * Defaults to using the x-request-id header or generating a UUID.
+     */
+    getWorkflowId?: (req: Request) => string | null | undefined;
+    /**
+     * Optional: Extract or specify a workflow type from the request.
+     */
+    getWorkflowType?: (req: Request) => string | null | undefined;
+    /**
+     * Optional: Extract additional metadata from the request.
+     */
+    getMetadata?: (req: Request) => Record<string, unknown> | undefined;
+    /**
+     * Optional: Filter which routes should have cost tracking enabled.
+     * Return true to enable tracking for the request, false to skip.
+     * Defaults to tracking all routes.
+     *
+     * @example
+     * ```typescript
+     * shouldTrack: (req) => req.path.startsWith('/api/')
+     * ```
+     */
+    shouldTrack?: (req: Request) => boolean;
+}
+/**
+ * Create an Express middleware that automatically sets up cost tracking context.
+ *
+ * This middleware wraps incoming requests with `withCostTrace()`, allowing
+ * any AI calls made during request handling to be automatically tracked.
+ *
+ * @param config - Configuration for extracting identifiers and metadata
+ * @returns An Express middleware function
+ *
+ * @example
+ * ```typescript
+ * import express from 'express';
+ * import { init } from 'tokenmeter';
+ * import { tokenmeterMiddleware } from 'tokenmeter/express';
+ *
+ * init({ databaseUrl: process.env.DATABASE_URL });
+ *
+ * const app = express();
+ *
+ * app.use(tokenmeterMiddleware({
+ *   getIdentifier: (req) => req.user?.id || req.headers['x-user-id'] as string,
+ *   shouldTrack: (req) => req.path.startsWith('/api/'),
+ * }));
+ *
+ * app.post('/api/chat', async (req, res) => {
+ *   // AI calls here are automatically tracked
+ *   const result = await openai.chat.completions.create({ ... });
+ *   res.json({ result });
+ * });
+ * ```
+ */
+export declare function tokenmeterMiddleware(config: TokenmeterMiddlewareConfig): RequestHandler;
+/**
+ * Wrap an async Express route handler with cost tracking.
+ *
+ * Use this for individual route handlers when you need more control than middleware.
+ *
+ * @param handler - The async route handler function
+ * @param getOptions - Function to extract trace options from the request
+ * @returns A wrapped route handler
+ *
+ * @example
+ * ```typescript
+ * import { withTokenmeter } from 'tokenmeter/express';
+ *
+ * const chatHandler = withTokenmeter(
+ *   async (req, res) => {
+ *     const result = await openai.chat.completions.create({ ... });
+ *     res.json({ result });
+ *   },
+ *   (req) => ({
+ *     identifier: req.user?.id || 'anonymous',
+ *     workflowType: 'chat',
+ *   })
+ * );
+ *
+ * app.post('/api/chat', chatHandler);
+ * ```
+ */
+export declare function withTokenmeter(handler: (req: Request, res: Response, next: NextFunction) => Promise<void>, getOptions: (req: Request) => CostTraceOptions | Promise<CostTraceOptions>): RequestHandler;
+/**
+ * Helper to extract common identifiers from Express request headers.
+ */
+export declare const headerExtractors: {
+    /**
+     * Get identifier from x-user-id header
+     */
+    userId: (req: Request) => string | undefined;
+    /**
+     * Get identifier from x-org-id header
+     */
+    orgId: (req: Request) => string | undefined;
+    /**
+     * Get identifier from x-api-key header
+     */
+    apiKey: (req: Request) => string | undefined;
+    /**
+     * Get identifier from Authorization header (Bearer token)
+     */
+    bearerToken: (req: Request) => any;
+    /**
+     * Get request ID from common headers
+     */
+    requestId: (req: Request) => string;
+};
+export default tokenmeterMiddleware;
+//# sourceMappingURL=index.d.ts.map

package/dist/integrations/express/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/integrations/express/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,cAAc,EAAE,MAAM,SAAS,CAAC;AAG/E,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAEvD;;GAEG;AACH,MAAM,WAAW,0BAA0B;IACzC;;;;;;;;OAQG;IACH,aAAa,EAAE,CAAC,GAAG,EAAE,OAAO,KAAK,MAAM,GAAG,IAAI,GAAG,SAAS,CAAC;IAE3D;;;;;;;OAOG;IACH,sBAAsB,CAAC,EAAE,CAAC,GAAG,EAAE,OAAO,KAAK,MAAM,GAAG,IAAI,GAAG,SAAS,CAAC;IAErE;;;OAGG;IACH,aAAa,CAAC,EAAE,CAAC,GAAG,EAAE,OAAO,KAAK,MAAM,GAAG,IAAI,GAAG,SAAS,CAAC;IAE5D;;OAEG;IACH,eAAe,CAAC,EAAE,CAAC,GAAG,EAAE,OAAO,KAAK,MAAM,GAAG,IAAI,GAAG,SAAS,CAAC;IAE9D;;OAEG;IACH,WAAW,CAAC,EAAE,CAAC,GAAG,EAAE,OAAO,KAAK,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,CAAC;IAEpE;;;;;;;;;OASG;IACH,WAAW,CAAC,EAAE,CAAC,GAAG,EAAE,OAAO,KAAK,OAAO,CAAC;CACzC;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA8BG;AACH,wBAAgB,oBAAoB,CAClC,MAAM,EAAE,0BAA0B,GACjC,cAAc,CAgFhB;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;GA0BG;AACH,wBAAgB,cAAc,CAC5B,OAAO,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,GAAG,EAAE,QAAQ,EAAE,IAAI,EAAE,YAAY,KAAK,OAAO,CAAC,IAAI,CAAC,EAC3E,UAAU,EAAE,CAAC,GAAG,EAAE,OAAO,KAAK,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,CAAC,GACzE,cAAc,CAuBhB;AAED;;GAEG;AACH,eAAO,MAAM,gBAAgB;IAC3B;;OAEG;kBACW,OAAO,KAAiC,MAAM,GAAG,SAAS;IAExE;;OAEG;iBACU,OAAO,KAAgC,MAAM,GAAG,SAAS;IAEtE;;OAEG;kBACW,OAAO,KAAiC,MAAM,GAAG,SAAS;IAExE;;OAEG;uBACgB,OAAO;IAQ1B;;OAEG;qBACc,OAAO;CAIzB,CAAC;AAEF,eAAe,oBAAoB,CAAC"}