tokenmeter 0.9.0

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

@@ -0,0 +1,170 @@
+import type { CostEvent } from "../../types.js";
+/**
+ * Langfuse client interface - minimal types to avoid hard dependency
+ */
+export interface LangfuseClient {
+    trace: (params: TraceParams) => LangfuseTrace;
+    flushAsync: () => Promise<void>;
+    shutdownAsync: () => Promise<void>;
+}
+export interface TraceParams {
+    id?: string;
+    name?: string;
+    userId?: string;
+    sessionId?: string;
+    metadata?: Record<string, unknown>;
+    input?: unknown;
+    output?: unknown;
+    tags?: string[];
+}
+export interface LangfuseTrace {
+    id: string;
+    generation: (params: GenerationParams) => LangfuseGeneration;
+    span: (params: SpanParams) => LangfuseSpan;
+    update: (params: Partial<TraceParams>) => void;
+}
+export interface GenerationParams {
+    name?: string;
+    model?: string;
+    modelParameters?: Record<string, unknown>;
+    input?: unknown;
+    output?: unknown;
+    metadata?: Record<string, unknown>;
+    startTime?: Date;
+    endTime?: Date;
+    usageDetails?: Record<string, number>;
+    costDetails?: Record<string, number>;
+}
+export interface LangfuseGeneration {
+    id: string;
+    update: (params: Partial<GenerationParams>) => void;
+    end: (params?: Partial<GenerationParams>) => void;
+}
+export interface SpanParams {
+    name?: string;
+    input?: unknown;
+    output?: unknown;
+    metadata?: Record<string, unknown>;
+    startTime?: Date;
+    endTime?: Date;
+}
+export interface LangfuseSpan {
+    id: string;
+    update: (params: Partial<SpanParams>) => void;
+    end: (params?: Partial<SpanParams>) => void;
+    generation: (params: GenerationParams) => LangfuseGeneration;
+}
+/**
+ * Configuration for the Langfuse exporter
+ */
+export interface LangfuseExporterConfig {
+    /**
+     * The Langfuse client instance.
+     * Create one with: `new Langfuse({ publicKey, secretKey })`
+     */
+    client: LangfuseClient;
+    /**
+     * Whether to create a trace per workflow or per event.
+     * - "workflow": One trace per workflowId, with generations as children
+     * - "event": One trace per cost event
+     * Default: "workflow"
+     */
+    traceMode?: "workflow" | "event";
+    /**
+     * Optional: Map tokenmeter identifiers to Langfuse user IDs.
+     * By default, the identifier is used as the userId.
+     */
+    getUserId?: (event: CostEvent) => string | undefined;
+    /**
+     * Optional: Map tokenmeter workflow IDs to Langfuse session IDs.
+     * By default, the workflowId is used as the sessionId.
+     */
+    getSessionId?: (event: CostEvent) => string | undefined;
+    /**
+     * Optional: Add custom tags to traces.
+     */
+    getTags?: (event: CostEvent) => string[];
+    /**
+     * Optional: Transform metadata before sending to Langfuse.
+     */
+    transformMetadata?: (event: CostEvent) => Record<string, unknown> | undefined;
+    /**
+     * Whether to flush after each export. Default: false
+     */
+    flushOnExport?: boolean;
+}
+/**
+ * Export tokenmeter cost events to Langfuse.
+ *
+ * @param events - Array of cost events to export
+ * @param config - Langfuse exporter configuration
+ *
+ * @example
+ * ```typescript
+ * import { Langfuse } from 'langfuse';
+ * import { exportToLangfuse } from 'tokenmeter/langfuse';
+ *
+ * const langfuse = new Langfuse({
+ *   publicKey: process.env.LANGFUSE_PUBLIC_KEY,
+ *   secretKey: process.env.LANGFUSE_SECRET_KEY,
+ * });
+ *
+ * // Export events after a workflow completes
+ * const events = getContextEvents();
+ * await exportToLangfuse(events, { client: langfuse });
+ * ```
+ */
+export declare function exportToLangfuse(events: CostEvent[], config: LangfuseExporterConfig): Promise<void>;
+/**
+ * Create a storage adapter that automatically exports to Langfuse.
+ *
+ * This wraps another storage adapter and exports events to Langfuse
+ * in addition to the primary storage.
+ *
+ * @example
+ * ```typescript
+ * import { Langfuse } from 'langfuse';
+ * import { init, PrismaStorageAdapter } from 'tokenmeter';
+ * import { createLangfuseStorageAdapter } from 'tokenmeter/langfuse';
+ *
+ * const langfuse = new Langfuse({
+ *   publicKey: process.env.LANGFUSE_PUBLIC_KEY,
+ *   secretKey: process.env.LANGFUSE_SECRET_KEY,
+ * });
+ *
+ * const primaryStorage = new PrismaStorageAdapter(process.env.DATABASE_URL);
+ *
+ * init({
+ *   storage: createLangfuseStorageAdapter(primaryStorage, {
+ *     client: langfuse,
+ *     traceMode: 'workflow',
+ *   }),
+ * });
+ * ```
+ */
+export declare function createLangfuseStorageAdapter(primaryAdapter: {
+    writeEvent: (event: CostEvent) => Promise<void>;
+    writeBatch: (events: CostEvent[]) => Promise<void>;
+    query: (params: unknown) => Promise<{
+        totalCost: number;
+        events: CostEvent[];
+    }>;
+    disconnect: () => Promise<void>;
+}, config: LangfuseExporterConfig): {
+    writeEvent(event: CostEvent): Promise<void>;
+    writeBatch(events: CostEvent[]): Promise<void>;
+    query(params: unknown): Promise<{
+        totalCost: number;
+        events: CostEvent[];
+    }>;
+    disconnect(): Promise<void>;
+};
+/**
+ * Clear the workflow traces cache for a specific config.
+ * Call this when switching contexts or at the end of a request.
+ *
+ * @param config - The Langfuse exporter config to clear traces for
+ */
+export declare function clearWorkflowTraces(config: LangfuseExporterConfig): void;
+export default exportToLangfuse;
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/integrations/langfuse/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAEhD;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B,KAAK,EAAE,CAAC,MAAM,EAAE,WAAW,KAAK,aAAa,CAAC;IAC9C,UAAU,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;IAChC,aAAa,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;CACpC;AAED,MAAM,WAAW,WAAW;IAC1B,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACnC,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;CACjB;AAED,MAAM,WAAW,aAAa;IAC5B,EAAE,EAAE,MAAM,CAAC;IACX,UAAU,EAAE,CAAC,MAAM,EAAE,gBAAgB,KAAK,kBAAkB,CAAC;IAC7D,IAAI,EAAE,CAAC,MAAM,EAAE,UAAU,KAAK,YAAY,CAAC;IAC3C,MAAM,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC,WAAW,CAAC,KAAK,IAAI,CAAC;CAChD;AAED,MAAM,WAAW,gBAAgB;IAC/B,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAC1C,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACnC,SAAS,CAAC,EAAE,IAAI,CAAC;IACjB,OAAO,CAAC,EAAE,IAAI,CAAC;IACf,YAAY,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACtC,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CACtC;AAED,MAAM,WAAW,kBAAkB;IACjC,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC,gBAAgB,CAAC,KAAK,IAAI,CAAC;IACpD,GAAG,EAAE,CAAC,MAAM,CAAC,EAAE,OAAO,CAAC,gBAAgB,CAAC,KAAK,IAAI,CAAC;CACnD;AAED,MAAM,WAAW,UAAU;IACzB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACnC,SAAS,CAAC,EAAE,IAAI,CAAC;IACjB,OAAO,CAAC,EAAE,IAAI,CAAC;CAChB;AAED,MAAM,WAAW,YAAY;IAC3B,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC,UAAU,CAAC,KAAK,IAAI,CAAC;IAC9C,GAAG,EAAE,CAAC,MAAM,CAAC,EAAE,OAAO,CAAC,UAAU,CAAC,KAAK,IAAI,CAAC;IAC5C,UAAU,EAAE,CAAC,MAAM,EAAE,gBAAgB,KAAK,kBAAkB,CAAC;CAC9D;AAED;;GAEG;AACH,MAAM,WAAW,sBAAsB;IACrC;;;OAGG;IACH,MAAM,EAAE,cAAc,CAAC;IAEvB;;;;;OAKG;IACH,SAAS,CAAC,EAAE,UAAU,GAAG,OAAO,CAAC;IAEjC;;;OAGG;IACH,SAAS,CAAC,EAAE,CAAC,KAAK,EAAE,SAAS,KAAK,MAAM,GAAG,SAAS,CAAC;IAErD;;;OAGG;IACH,YAAY,CAAC,EAAE,CAAC,KAAK,EAAE,SAAS,KAAK,MAAM,GAAG,SAAS,CAAC;IAExD;;OAEG;IACH,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,SAAS,KAAK,MAAM,EAAE,CAAC;IAEzC;;OAEG;IACH,iBAAiB,CAAC,EAAE,CAAC,KAAK,EAAE,SAAS,KAAK,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,CAAC;IAE9E;;OAEG;IACH,aAAa,CAAC,EAAE,OAAO,CAAC;CACzB;AAyBD;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,wBAAsB,gBAAgB,CACpC,MAAM,EAAE,SAAS,EAAE,EACnB,MAAM,EAAE,sBAAsB,GAC7B,OAAO,CAAC,IAAI,CAAC,CAYf;AA8HD;;;;;;;;;;;;;;;;;;;;;;;;;;GA0BG;AACH,wBAAgB,4BAA4B,CAC1C,cAAc,EAAE;IACd,UAAU,EAAE,CAAC,KAAK,EAAE,SAAS,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAChD,UAAU,EAAE,CAAC,MAAM,EAAE,SAAS,EAAE,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACnD,KAAK,EAAE,CACL,MAAM,EAAE,OAAO,KACZ,OAAO,CAAC;QAAE,SAAS,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,SAAS,EAAE,CAAA;KAAE,CAAC,CAAC;IACzD,UAAU,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;CACjC,EACD,MAAM,EAAE,sBAAsB;sBAGJ,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC;uBAQxB,SAAS,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;kBAS1C,OAAO,GACd,OAAO,CAAC;QAAE,SAAS,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,SAAS,EAAE,CAAA;KAAE,CAAC;kBAKlC,OAAO,CAAC,IAAI,CAAC;EAWpC;AAED;;;;;GAKG;AACH,wBAAgB,mBAAmB,CAAC,MAAM,EAAE,sBAAsB,GAAG,IAAI,CAKxE;AAED,eAAe,gBAAgB,CAAC"}

package/dist/integrations/langfuse/index.js

@@ -0,0 +1,225 @@
+/**
+ * Active workflow traces cache - stored per-config to avoid cross-request contamination
+ * Uses WeakMap with config as key so cache is garbage collected with the config
+ */
+const workflowTracesCache = new WeakMap();
+/**
+ * Get or create the workflow traces map for a given config
+ */
+function getWorkflowTracesMap(config) {
+    let traces = workflowTracesCache.get(config);
+    if (!traces) {
+        traces = new Map();
+        workflowTracesCache.set(config, traces);
+    }
+    return traces;
+}
+/**
+ * Export tokenmeter cost events to Langfuse.
+ *
+ * @param events - Array of cost events to export
+ * @param config - Langfuse exporter configuration
+ *
+ * @example
+ * ```typescript
+ * import { Langfuse } from 'langfuse';
+ * import { exportToLangfuse } from 'tokenmeter/langfuse';
+ *
+ * const langfuse = new Langfuse({
+ *   publicKey: process.env.LANGFUSE_PUBLIC_KEY,
+ *   secretKey: process.env.LANGFUSE_SECRET_KEY,
+ * });
+ *
+ * // Export events after a workflow completes
+ * const events = getContextEvents();
+ * await exportToLangfuse(events, { client: langfuse });
+ * ```
+ */
+export async function exportToLangfuse(events, config) {
+    const { client, traceMode = "workflow" } = config;
+    if (traceMode === "workflow") {
+        await exportAsWorkflows(events, config);
+    }
+    else {
+        await exportAsEvents(events, config);
+    }
+    if (config.flushOnExport) {
+        await client.flushAsync();
+    }
+}
+/**
+ * Export events grouped by workflow
+ */
+async function exportAsWorkflows(events, config) {
+    const { client } = config;
+    const workflowTraces = getWorkflowTracesMap(config);
+    // Group events by workflowId
+    const byWorkflow = new Map();
+    for (const event of events) {
+        const workflowId = event.workflowId || "default";
+        const existing = byWorkflow.get(workflowId) || [];
+        existing.push(event);
+        byWorkflow.set(workflowId, existing);
+    }
+    // Create traces and generations for each workflow
+    for (const [workflowId, workflowEvents] of byWorkflow) {
+        // Get or create trace for this workflow
+        let trace = workflowTraces.get(workflowId);
+        if (!trace) {
+            const firstEvent = workflowEvents[0];
+            if (!firstEvent)
+                continue;
+            trace = client.trace({
+                id: workflowId,
+                name: firstEvent.workflowType || "tokenmeter-workflow",
+                userId: config.getUserId?.(firstEvent) ?? firstEvent.identifier,
+                sessionId: config.getSessionId?.(firstEvent) ?? workflowId,
+                metadata: {
+                    source: "tokenmeter",
+                    identifier: firstEvent.identifier,
+                    secondaryIdentifier: firstEvent.secondaryIdentifier,
+                },
+                tags: config.getTags?.(firstEvent) ?? ["tokenmeter"],
+            });
+            workflowTraces.set(workflowId, trace);
+        }
+        // Add generations for each event
+        for (const event of workflowEvents) {
+            const metadata = config.transformMetadata?.(event) ?? event.metadata;
+            trace.generation({
+                name: `${event.provider}/${event.model}`,
+                model: event.model,
+                startTime: event.createdAt,
+                endTime: event.createdAt,
+                metadata: {
+                    ...metadata,
+                    provider: event.provider,
+                    operation: event.operation,
+                    status: event.status,
+                    billable: event.billable,
+                    errorMessage: event.errorMessage,
+                    requestId: event.requestId,
+                },
+                usageDetails: {
+                    input: event.inputTokens ?? 0,
+                    output: event.outputTokens ?? 0,
+                },
+                costDetails: {
+                    total: event.costUsd,
+                },
+            });
+        }
+    }
+}
+/**
+ * Export each event as a separate trace
+ */
+async function exportAsEvents(events, config) {
+    const { client } = config;
+    for (const event of events) {
+        const metadata = config.transformMetadata?.(event) ?? event.metadata;
+        const trace = client.trace({
+            id: event.id || crypto.randomUUID(),
+            name: `${event.provider}/${event.model}`,
+            userId: config.getUserId?.(event) ?? event.identifier,
+            sessionId: config.getSessionId?.(event) ?? event.workflowId,
+            metadata: {
+                source: "tokenmeter",
+                identifier: event.identifier,
+                secondaryIdentifier: event.secondaryIdentifier,
+                workflowType: event.workflowType,
+            },
+            tags: config.getTags?.(event) ?? ["tokenmeter"],
+        });
+        trace.generation({
+            name: event.operation || "generation",
+            model: event.model,
+            startTime: event.createdAt,
+            endTime: event.createdAt,
+            metadata: {
+                ...metadata,
+                provider: event.provider,
+                status: event.status,
+                billable: event.billable,
+                errorMessage: event.errorMessage,
+                requestId: event.requestId,
+            },
+            usageDetails: {
+                input: event.inputTokens ?? 0,
+                output: event.outputTokens ?? 0,
+            },
+            costDetails: {
+                total: event.costUsd,
+            },
+        });
+    }
+}
+/**
+ * Create a storage adapter that automatically exports to Langfuse.
+ *
+ * This wraps another storage adapter and exports events to Langfuse
+ * in addition to the primary storage.
+ *
+ * @example
+ * ```typescript
+ * import { Langfuse } from 'langfuse';
+ * import { init, PrismaStorageAdapter } from 'tokenmeter';
+ * import { createLangfuseStorageAdapter } from 'tokenmeter/langfuse';
+ *
+ * const langfuse = new Langfuse({
+ *   publicKey: process.env.LANGFUSE_PUBLIC_KEY,
+ *   secretKey: process.env.LANGFUSE_SECRET_KEY,
+ * });
+ *
+ * const primaryStorage = new PrismaStorageAdapter(process.env.DATABASE_URL);
+ *
+ * init({
+ *   storage: createLangfuseStorageAdapter(primaryStorage, {
+ *     client: langfuse,
+ *     traceMode: 'workflow',
+ *   }),
+ * });
+ * ```
+ */
+export function createLangfuseStorageAdapter(primaryAdapter, config) {
+    return {
+        async writeEvent(event) {
+            // Write to primary storage
+            await primaryAdapter.writeEvent(event);
+            // Export to Langfuse
+            await exportToLangfuse([event], config);
+        },
+        async writeBatch(events) {
+            // Write to primary storage
+            await primaryAdapter.writeBatch(events);
+            // Export to Langfuse
+            await exportToLangfuse(events, config);
+        },
+        async query(params) {
+            // Query from primary storage only
+            return primaryAdapter.query(params);
+        },
+        async disconnect() {
+            // Flush Langfuse before disconnecting
+            await config.client.flushAsync();
+            // Clear workflow traces cache for this config
+            clearWorkflowTraces(config);
+            // Disconnect primary storage
+            await primaryAdapter.disconnect();
+        },
+    };
+}
+/**
+ * Clear the workflow traces cache for a specific config.
+ * Call this when switching contexts or at the end of a request.
+ *
+ * @param config - The Langfuse exporter config to clear traces for
+ */
+export function clearWorkflowTraces(config) {
+    const traces = workflowTracesCache.get(config);
+    if (traces) {
+        traces.clear();
+    }
+}
+export default exportToLangfuse;
+//# sourceMappingURL=index.js.map

package/dist/integrations/langfuse/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/integrations/langfuse/index.ts"],"names":[],"mappings":"AA8GA;;;GAGG;AACH,MAAM,mBAAmB,GAAG,IAAI,OAAO,EAGpC,CAAC;AAEJ;;GAEG;AACH,SAAS,oBAAoB,CAC3B,MAA8B;IAE9B,IAAI,MAAM,GAAG,mBAAmB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAC7C,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,GAAG,IAAI,GAAG,EAAE,CAAC;QACnB,mBAAmB,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC1C,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,MAAmB,EACnB,MAA8B;IAE9B,MAAM,EAAE,MAAM,EAAE,SAAS,GAAG,UAAU,EAAE,GAAG,MAAM,CAAC;IAElD,IAAI,SAAS,KAAK,UAAU,EAAE,CAAC;QAC7B,MAAM,iBAAiB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC1C,CAAC;SAAM,CAAC;QACN,MAAM,cAAc,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACvC,CAAC;IAED,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;QACzB,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC;IAC5B,CAAC;AACH,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,iBAAiB,CAC9B,MAAmB,EACnB,MAA8B;IAE9B,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,CAAC;IAC1B,MAAM,cAAc,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC;IAEpD,6BAA6B;IAC7B,MAAM,UAAU,GAAG,IAAI,GAAG,EAAuB,CAAC;IAClD,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,MAAM,UAAU,GAAG,KAAK,CAAC,UAAU,IAAI,SAAS,CAAC;QACjD,MAAM,QAAQ,GAAG,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;QAClD,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACrB,UAAU,CAAC,GAAG,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;IACvC,CAAC;IAED,kDAAkD;IAClD,KAAK,MAAM,CAAC,UAAU,EAAE,cAAc,CAAC,IAAI,UAAU,EAAE,CAAC;QACtD,wCAAwC;QACxC,IAAI,KAAK,GAAG,cAAc,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAE3C,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,UAAU,GAAG,cAAc,CAAC,CAAC,CAAC,CAAC;YACrC,IAAI,CAAC,UAAU;gBAAE,SAAS;YAE1B,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;gBACnB,EAAE,EAAE,UAAU;gBACd,IAAI,EAAE,UAAU,CAAC,YAAY,IAAI,qBAAqB;gBACtD,MAAM,EAAE,MAAM,CAAC,SAAS,EAAE,CAAC,UAAU,CAAC,IAAI,UAAU,CAAC,UAAU;gBAC/D,SAAS,EAAE,MAAM,CAAC,YAAY,EAAE,CAAC,UAAU,CAAC,IAAI,UAAU;gBAC1D,QAAQ,EAAE;oBACR,MAAM,EAAE,YAAY;oBACpB,UAAU,EAAE,UAAU,CAAC,UAAU;oBACjC,mBAAmB,EAAE,UAAU,CAAC,mBAAmB;iBACpD;gBACD,IAAI,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,YAAY,CAAC;aACrD,CAAC,CAAC;YAEH,cAAc,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QACxC,CAAC;QAED,iCAAiC;QACjC,KAAK,MAAM,KAAK,IAAI,cAAc,EAAE,CAAC;YACnC,MAAM,QAAQ,GAAG,MAAM,CAAC,iBAAiB,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC;YAErE,KAAK,CAAC,UAAU,CAAC;gBACf,IAAI,EAAE,GAAG,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,KAAK,EAAE;gBACxC,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,SAAS,EAAE,KAAK,CAAC,SAAS;gBAC1B,OAAO,EAAE,KAAK,CAAC,SAAS;gBACxB,QAAQ,EAAE;oBACR,GAAG,QAAQ;oBACX,QAAQ,EAAE,KAAK,CAAC,QAAQ;oBACxB,SAAS,EAAE,KAAK,CAAC,SAAS;oBAC1B,MAAM,EAAE,KAAK,CAAC,MAAM;oBACpB,QAAQ,EAAE,KAAK,CAAC,QAAQ;oBACxB,YAAY,EAAE,KAAK,CAAC,YAAY;oBAChC,SAAS,EAAE,KAAK,CAAC,SAAS;iBAC3B;gBACD,YAAY,EAAE;oBACZ,KAAK,EAAE,KAAK,CAAC,WAAW,IAAI,CAAC;oBAC7B,MAAM,EAAE,KAAK,CAAC,YAAY,IAAI,CAAC;iBAChC;gBACD,WAAW,EAAE;oBACX,KAAK,EAAE,KAAK,CAAC,OAAO;iBACrB;aACF,CAAC,CAAC;QACL,CAAC;IACH,CAAC;AACH,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,cAAc,CAC3B,MAAmB,EACnB,MAA8B;IAE9B,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,CAAC;IAE1B,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,MAAM,QAAQ,GAAG,MAAM,CAAC,iBAAiB,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC;QAErE,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;YACzB,EAAE,EAAE,KAAK,CAAC,EAAE,IAAI,MAAM,CAAC,UAAU,EAAE;YACnC,IAAI,EAAE,GAAG,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,KAAK,EAAE;YACxC,MAAM,EAAE,MAAM,CAAC,SAAS,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,UAAU;YACrD,SAAS,EAAE,MAAM,CAAC,YAAY,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,UAAU;YAC3D,QAAQ,EAAE;gBACR,MAAM,EAAE,YAAY;gBACpB,UAAU,EAAE,KAAK,CAAC,UAAU;gBAC5B,mBAAmB,EAAE,KAAK,CAAC,mBAAmB;gBAC9C,YAAY,EAAE,KAAK,CAAC,YAAY;aACjC;YACD,IAAI,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC;SAChD,CAAC,CAAC;QAEH,KAAK,CAAC,UAAU,CAAC;YACf,IAAI,EAAE,KAAK,CAAC,SAAS,IAAI,YAAY;YACrC,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,OAAO,EAAE,KAAK,CAAC,SAAS;YACxB,QAAQ,EAAE;gBACR,GAAG,QAAQ;gBACX,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,MAAM,EAAE,KAAK,CAAC,MAAM;gBACpB,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,YAAY,EAAE,KAAK,CAAC,YAAY;gBAChC,SAAS,EAAE,KAAK,CAAC,SAAS;aAC3B;YACD,YAAY,EAAE;gBACZ,KAAK,EAAE,KAAK,CAAC,WAAW,IAAI,CAAC;gBAC7B,MAAM,EAAE,KAAK,CAAC,YAAY,IAAI,CAAC;aAChC;YACD,WAAW,EAAE;gBACX,KAAK,EAAE,KAAK,CAAC,OAAO;aACrB;SACF,CAAC,CAAC;IACL,CAAC;AACH,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;GA0BG;AACH,MAAM,UAAU,4BAA4B,CAC1C,cAOC,EACD,MAA8B;IAE9B,OAAO;QACL,KAAK,CAAC,UAAU,CAAC,KAAgB;YAC/B,2BAA2B;YAC3B,MAAM,cAAc,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YAEvC,qBAAqB;YACrB,MAAM,gBAAgB,CAAC,CAAC,KAAK,CAAC,EAAE,MAAM,CAAC,CAAC;QAC1C,CAAC;QAED,KAAK,CAAC,UAAU,CAAC,MAAmB;YAClC,2BAA2B;YAC3B,MAAM,cAAc,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;YAExC,qBAAqB;YACrB,MAAM,gBAAgB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACzC,CAAC;QAED,KAAK,CAAC,KAAK,CACT,MAAe;YAEf,kCAAkC;YAClC,OAAO,cAAc,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACtC,CAAC;QAED,KAAK,CAAC,UAAU;YACd,sCAAsC;YACtC,MAAM,MAAM,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;YAEjC,8CAA8C;YAC9C,mBAAmB,CAAC,MAAM,CAAC,CAAC;YAE5B,6BAA6B;YAC7B,MAAM,cAAc,CAAC,UAAU,EAAE,CAAC;QACpC,CAAC;KACF,CAAC;AACJ,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,mBAAmB,CAAC,MAA8B;IAChE,MAAM,MAAM,GAAG,mBAAmB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAC/C,IAAI,MAAM,EAAE,CAAC;QACX,MAAM,CAAC,KAAK,EAAE,CAAC;IACjB,CAAC;AACH,CAAC;AAED,eAAe,gBAAgB,CAAC"}

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

@@ -0,0 +1,138 @@
+/**
+ * Next.js Integration
+ *
+ * Provides helpers for App Router route handlers to automatically
+ * set up cost tracking context.
+ */
+/**
+ * Context for tokenmeter tracking
+ */
+export interface TokenMeterContext {
+    /** User ID for cost attribution */
+    userId?: string;
+    /** Organization ID for cost attribution */
+    orgId?: string;
+    /** Workflow ID for grouping related calls */
+    workflowId?: string;
+    /** Additional metadata */
+    metadata?: Record<string, string>;
+}
+/**
+ * Next.js App Router route handler type
+ */
+export type RouteHandler = (request: Request, context?: {
+    params?: Promise<Record<string, string>>;
+}) => Response | Promise<Response>;
+/**
+ * Function to extract context from a request
+ */
+export type ContextExtractor = (request: Request) => TokenMeterContext | Promise<TokenMeterContext>;
+/**
+ * Options for withTokenmeter wrapper
+ */
+export interface WithTokenmeterOptions {
+    /** Custom context extractor function */
+    getContext?: ContextExtractor;
+    /** Header name for user ID (default: x-user-id) */
+    userIdHeader?: string;
+    /** Header name for org ID (default: x-org-id) */
+    orgIdHeader?: string;
+    /** Header name for workflow ID (default: x-request-id) */
+    workflowIdHeader?: string;
+}
+/**
+ * Wrap a Next.js App Router route handler with tokenmeter context.
+ *
+ * All AI calls made with `monitor()`-wrapped clients within the handler
+ * will automatically be attributed to the user/org extracted from the request.
+ *
+ * @example
+ * ```typescript
+ * // app/api/chat/route.ts
+ * import { withTokenmeter } from 'tokenmeter/next';
+ * import { monitor } from 'tokenmeter';
+ * import OpenAI from 'openai';
+ *
+ * const openai = monitor(new OpenAI());
+ *
+ * export const POST = withTokenmeter(async (req) => {
+ *   const { message } = await req.json();
+ *
+ *   // This call is automatically attributed to the user
+ *   const result = await openai.chat.completions.create({
+ *     model: 'gpt-4o',
+ *     messages: [{ role: 'user', content: message }],
+ *   });
+ *
+ *   return Response.json({ reply: result.choices[0].message.content });
+ * });
+ * ```
+ *
+ * @example
+ * ```typescript
+ * // With custom context extraction (e.g., from auth)
+ * import { auth } from '@clerk/nextjs/server';
+ *
+ * export const POST = withTokenmeter(
+ *   async (req) => {
+ *     // Handler code
+ *   },
+ *   {
+ *     getContext: async (req) => {
+ *       const { userId, orgId } = await auth();
+ *       return { userId, orgId };
+ *     },
+ *   }
+ * );
+ * ```
+ */
+export declare function withTokenmeter(handler: RouteHandler, options?: WithTokenmeterOptions): RouteHandler;
+/**
+ * Create a reusable tokenmeter wrapper with preset options.
+ *
+ * Useful when you have common context extraction logic across multiple routes.
+ *
+ * @example
+ * ```typescript
+ * // lib/tokenmeter.ts
+ * import { createTokenmeterWrapper } from 'tokenmeter/next';
+ * import { auth } from '@clerk/nextjs/server';
+ *
+ * export const withCostTracking = createTokenmeterWrapper({
+ *   getContext: async (req) => {
+ *     const { userId, orgId } = await auth();
+ *     return { userId, orgId };
+ *   },
+ * });
+ *
+ * // app/api/chat/route.ts
+ * import { withCostTracking } from '@/lib/tokenmeter';
+ *
+ * export const POST = withCostTracking(async (req) => {
+ *   // ...
+ * });
+ * ```
+ */
+export declare function createTokenmeterWrapper(defaultOptions: WithTokenmeterOptions): (handler: RouteHandler, options?: WithTokenmeterOptions) => RouteHandler;
+/**
+ * Helper to extract common headers from a request
+ */
+export declare const headerExtractors: {
+    /**
+     * Extract user ID from x-user-id header
+     */
+    userId: (request: Request) => string | undefined;
+    /**
+     * Extract org ID from x-org-id header
+     */
+    orgId: (request: Request) => string | undefined;
+    /**
+     * Extract API key from Authorization header (Bearer token)
+     */
+    apiKey: (request: Request) => string | undefined;
+    /**
+     * Extract request ID from x-request-id header
+     */
+    requestId: (request: Request) => string | undefined;
+};
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/integrations/next/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAKH;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC,mCAAmC;IACnC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,2CAA2C;IAC3C,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,6CAA6C;IAC7C,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,0BAA0B;IAC1B,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CACnC;AAED;;GAEG;AACH,MAAM,MAAM,YAAY,GAAG,CACzB,OAAO,EAAE,OAAO,EAChB,OAAO,CAAC,EAAE;IAAE,MAAM,CAAC,EAAE,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAA;CAAE,KACnD,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;AAElC;;GAEG;AACH,MAAM,MAAM,gBAAgB,GAAG,CAC7B,OAAO,EAAE,OAAO,KACb,iBAAiB,GAAG,OAAO,CAAC,iBAAiB,CAAC,CAAC;AAEpD;;GAEG;AACH,MAAM,WAAW,qBAAqB;IACpC,wCAAwC;IACxC,UAAU,CAAC,EAAE,gBAAgB,CAAC;IAC9B,mDAAmD;IACnD,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,iDAAiD;IACjD,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,0DAA0D;IAC1D,gBAAgB,CAAC,EAAE,MAAM,CAAC;CAC3B;AA6CD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA6CG;AACH,wBAAgB,cAAc,CAC5B,OAAO,EAAE,YAAY,EACrB,OAAO,GAAE,qBAA0B,GAClC,YAAY,CAkBd;AAED;;;;;;;;;;;;;;;;;;;;;;;;;GAyBG;AACH,wBAAgB,uBAAuB,CACrC,cAAc,EAAE,qBAAqB,GACpC,CAAC,OAAO,EAAE,YAAY,EAAE,OAAO,CAAC,EAAE,qBAAqB,KAAK,YAAY,CAI1E;AAED;;GAEG;AACH,eAAO,MAAM,gBAAgB;IAC3B;;OAEG;sBACe,OAAO,KAAG,MAAM,GAAG,SAAS;IAI9C;;OAEG;qBACc,OAAO,KAAG,MAAM,GAAG,SAAS;IAI7C;;OAEG;sBACe,OAAO,KAAG,MAAM,GAAG,SAAS;IAQ9C;;OAEG;yBACkB,OAAO,KAAG,MAAM,GAAG,SAAS;CAGlD,CAAC"}