tokenmeter 0.9.0

package/dist/integrations/express/index.js

@@ -0,0 +1,186 @@
+import { withCostTrace } from "../../context.js";
+import { isInitialized } from "../../config.js";
+/**
+ * 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 function tokenmeterMiddleware(config) {
+    return (req, res, next) => {
+        // Check if tokenmeter is initialized
+        if (!isInitialized()) {
+            console.warn("[tokenmeter] Not initialized. Call init() before using middleware.");
+            next();
+            return;
+        }
+        // Check if we should track this request
+        if (config.shouldTrack && !config.shouldTrack(req)) {
+            next();
+            return;
+        }
+        // Get identifier
+        const identifier = config.getIdentifier(req);
+        if (!identifier) {
+            // No identifier - skip tracking but continue request
+            next();
+            return;
+        }
+        // Build trace options
+        const traceOptions = {
+            identifier,
+            secondaryIdentifier: config.getSecondaryIdentifier?.(req) ?? undefined,
+            workflowId: config.getWorkflowId?.(req) ??
+                req.headers["x-request-id"] ??
+                crypto.randomUUID(),
+            workflowType: config.getWorkflowType?.(req) ?? undefined,
+            requestId: req.headers["x-request-id"] ?? undefined,
+            metadata: {
+                ...config.getMetadata?.(req),
+                path: req.path,
+                method: req.method,
+            },
+        };
+        // Wrap the request handling in a cost trace context
+        // We need to handle this differently since Express uses callbacks
+        void withCostTrace(traceOptions, async () => {
+            return new Promise((resolve, reject) => {
+                // Override next to resolve our promise when called
+                const wrappedNext = (err) => {
+                    if (err) {
+                        reject(err);
+                    }
+                    else {
+                        resolve();
+                    }
+                };
+                // Call next to continue the middleware chain
+                // The actual next() should not be called until the response is sent
+                // So we need to hook into res.end or similar
+                // Store original end
+                const originalEnd = res.end.bind(res);
+                let ended = false;
+                // Override end to resolve our promise
+                res.end = function (...args) {
+                    if (!ended) {
+                        ended = true;
+                        resolve();
+                    }
+                    return originalEnd(...args);
+                };
+                // Call next to continue
+                next();
+            });
+        }).catch((err) => {
+            // If there's an error, pass it to Express error handling
+            next(err);
+        });
+    };
+}
+/**
+ * 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 function withTokenmeter(handler, getOptions) {
+    return async (req, res, next) => {
+        if (!isInitialized()) {
+            console.warn("[tokenmeter] Not initialized. Call init() before using withTokenmeter.");
+            return handler(req, res, next);
+        }
+        try {
+            const options = await getOptions(req);
+            if (!options.identifier) {
+                return handler(req, res, next);
+            }
+            await withCostTrace(options, async () => {
+                await handler(req, res, next);
+            });
+        }
+        catch (err) {
+            next(err);
+        }
+    };
+}
+/**
+ * Helper to extract common identifiers from Express request headers.
+ */
+export const headerExtractors = {
+    /**
+     * Get identifier from x-user-id header
+     */
+    userId: (req) => req.headers["x-user-id"],
+    /**
+     * Get identifier from x-org-id header
+     */
+    orgId: (req) => req.headers["x-org-id"],
+    /**
+     * Get identifier from x-api-key header
+     */
+    apiKey: (req) => req.headers["x-api-key"],
+    /**
+     * Get identifier from Authorization header (Bearer token)
+     */
+    bearerToken: (req) => {
+        const auth = req.headers.authorization;
+        if (auth?.startsWith("Bearer ")) {
+            return auth.slice(7);
+        }
+        return undefined;
+    },
+    /**
+     * Get request ID from common headers
+     */
+    requestId: (req) => req.headers["x-request-id"] ||
+        req.headers["x-correlation-id"] ||
+        req.headers["x-trace-id"],
+};
+export default tokenmeterMiddleware;
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/integrations/express/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AACjD,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAyDhD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA8BG;AACH,MAAM,UAAU,oBAAoB,CAClC,MAAkC;IAElC,OAAO,CAAC,GAAY,EAAE,GAAa,EAAE,IAAkB,EAAQ,EAAE;QAC/D,qCAAqC;QACrC,IAAI,CAAC,aAAa,EAAE,EAAE,CAAC;YACrB,OAAO,CAAC,IAAI,CACV,oEAAoE,CACrE,CAAC;YACF,IAAI,EAAE,CAAC;YACP,OAAO;QACT,CAAC;QAED,wCAAwC;QACxC,IAAI,MAAM,CAAC,WAAW,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC;YACnD,IAAI,EAAE,CAAC;YACP,OAAO;QACT,CAAC;QAED,iBAAiB;QACjB,MAAM,UAAU,GAAG,MAAM,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;QAC7C,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,qDAAqD;YACrD,IAAI,EAAE,CAAC;YACP,OAAO;QACT,CAAC;QAED,sBAAsB;QACtB,MAAM,YAAY,GAAqB;YACrC,UAAU;YACV,mBAAmB,EAAE,MAAM,CAAC,sBAAsB,EAAE,CAAC,GAAG,CAAC,IAAI,SAAS;YACtE,UAAU,EACR,MAAM,CAAC,aAAa,EAAE,CAAC,GAAG,CAAC;gBAC1B,GAAG,CAAC,OAAO,CAAC,cAAc,CAAY;gBACvC,MAAM,CAAC,UAAU,EAAE;YACrB,YAAY,EAAE,MAAM,CAAC,eAAe,EAAE,CAAC,GAAG,CAAC,IAAI,SAAS;YACxD,SAAS,EAAG,GAAG,CAAC,OAAO,CAAC,cAAc,CAAY,IAAI,SAAS;YAC/D,QAAQ,EAAE;gBACR,GAAG,MAAM,CAAC,WAAW,EAAE,CAAC,GAAG,CAAC;gBAC5B,IAAI,EAAE,GAAG,CAAC,IAAI;gBACd,MAAM,EAAE,GAAG,CAAC,MAAM;aACnB;SACF,CAAC;QAEF,oDAAoD;QACpD,kEAAkE;QAClE,KAAK,aAAa,CAAC,YAAY,EAAE,KAAK,IAAI,EAAE;YAC1C,OAAO,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBAC3C,mDAAmD;gBACnD,MAAM,WAAW,GAAiB,CAAC,GAAa,EAAE,EAAE;oBAClD,IAAI,GAAG,EAAE,CAAC;wBACR,MAAM,CAAC,GAAG,CAAC,CAAC;oBACd,CAAC;yBAAM,CAAC;wBACN,OAAO,EAAE,CAAC;oBACZ,CAAC;gBACH,CAAC,CAAC;gBAEF,6CAA6C;gBAC7C,oEAAoE;gBACpE,6CAA6C;gBAE7C,qBAAqB;gBACrB,MAAM,WAAW,GAAG,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBACtC,IAAI,KAAK,GAAG,KAAK,CAAC;gBAElB,sCAAsC;gBACtC,GAAG,CAAC,GAAG,GAAG,UAAU,GAAG,IAAiC;oBACtD,IAAI,CAAC,KAAK,EAAE,CAAC;wBACX,KAAK,GAAG,IAAI,CAAC;wBACb,OAAO,EAAE,CAAC;oBACZ,CAAC;oBACD,OAAO,WAAW,CAAC,GAAG,IAAI,CAAC,CAAC;gBAC9B,CAAoB,CAAC;gBAErB,wBAAwB;gBACxB,IAAI,EAAE,CAAC;YACT,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;YACf,yDAAyD;YACzD,IAAI,CAAC,GAAG,CAAC,CAAC;QACZ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;GA0BG;AACH,MAAM,UAAU,cAAc,CAC5B,OAA2E,EAC3E,UAA0E;IAE1E,OAAO,KAAK,EAAE,GAAY,EAAE,GAAa,EAAE,IAAkB,EAAE,EAAE;QAC/D,IAAI,CAAC,aAAa,EAAE,EAAE,CAAC;YACrB,OAAO,CAAC,IAAI,CACV,wEAAwE,CACzE,CAAC;YACF,OAAO,OAAO,CAAC,GAAG,EAAE,GAAG,EAAE,IAAI,CAAC,CAAC;QACjC,CAAC;QAED,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,GAAG,CAAC,CAAC;YAEtC,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;gBACxB,OAAO,OAAO,CAAC,GAAG,EAAE,GAAG,EAAE,IAAI,CAAC,CAAC;YACjC,CAAC;YAED,MAAM,aAAa,CAAC,OAAO,EAAE,KAAK,IAAI,EAAE;gBACtC,MAAM,OAAO,CAAC,GAAG,EAAE,GAAG,EAAE,IAAI,CAAC,CAAC;YAChC,CAAC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,GAAG,CAAC,CAAC;QACZ,CAAC;IACH,CAAC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG;IAC9B;;OAEG;IACH,MAAM,EAAE,CAAC,GAAY,EAAE,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,WAAW,CAAuB;IAExE;;OAEG;IACH,KAAK,EAAE,CAAC,GAAY,EAAE,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,UAAU,CAAuB;IAEtE;;OAEG;IACH,MAAM,EAAE,CAAC,GAAY,EAAE,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,WAAW,CAAuB;IAExE;;OAEG;IACH,WAAW,EAAE,CAAC,GAAY,EAAE,EAAE;QAC5B,MAAM,IAAI,GAAG,GAAG,CAAC,OAAO,CAAC,aAAa,CAAC;QACvC,IAAI,IAAI,EAAE,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;YAChC,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACvB,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;OAEG;IACH,SAAS,EAAE,CAAC,GAAY,EAAE,EAAE,CACzB,GAAG,CAAC,OAAO,CAAC,cAAc,CAAY;QACtC,GAAG,CAAC,OAAO,CAAC,kBAAkB,CAAY;QAC1C,GAAG,CAAC,OAAO,CAAC,YAAY,CAAY;CACxC,CAAC;AAEF,eAAe,oBAAoB,CAAC"}

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

@@ -0,0 +1,222 @@
+/**
+ * Inngest Integration
+ *
+ * Provides helpers for propagating trace context to and from Inngest functions.
+ */
+import type { TokenMeterAttributes } from "../../types.js";
+/**
+ * Trace headers that can be passed in Inngest event metadata
+ */
+export interface TraceHeaders {
+    traceparent?: string;
+    tracestate?: string;
+    baggage?: string;
+}
+/**
+ * Inngest event with optional trace metadata
+ */
+export interface InngestEventWithTrace<T = unknown> {
+    name: string;
+    data: T;
+    /** Trace headers for context propagation */
+    trace?: TraceHeaders;
+}
+/**
+ * Options for creating traced Inngest events
+ */
+export interface CreateTracedEventOptions {
+    /** Additional attributes to include in the trace */
+    attributes?: TokenMeterAttributes;
+}
+/**
+ * Extract trace headers to include when sending an Inngest event.
+ *
+ * Call this in your API route before sending an Inngest event to propagate
+ * the trace context to the Inngest function.
+ *
+ * @example
+ * ```typescript
+ * import { inngest } from './inngest';
+ * import { getInngestTraceHeaders } from 'tokenmeter/inngest';
+ *
+ * // In your API route
+ * export async function POST(req: Request) {
+ *   const trace = getInngestTraceHeaders();
+ *
+ *   await inngest.send({
+ *     name: 'document/process',
+ *     data: { documentId: '123' },
+ *     trace, // Pass trace headers
+ *   });
+ *
+ *   return Response.json({ ok: true });
+ * }
+ * ```
+ */
+export declare function getInngestTraceHeaders(): TraceHeaders;
+/**
+ * Create a traced Inngest event with trace headers included.
+ *
+ * This is a convenience wrapper that combines your event data with trace headers.
+ *
+ * @example
+ * ```typescript
+ * import { inngest } from './inngest';
+ * import { createTracedEvent } from 'tokenmeter/inngest';
+ *
+ * await inngest.send(createTracedEvent({
+ *   name: 'document/process',
+ *   data: { documentId: '123' },
+ * }));
+ * ```
+ */
+export declare function createTracedEvent<T>(event: {
+    name: string;
+    data: T;
+}, options?: CreateTracedEventOptions): InngestEventWithTrace<T>;
+/**
+ * Run a function within the trace context from an Inngest event.
+ *
+ * Use this at the start of your Inngest function to resume the trace
+ * from the parent request.
+ *
+ * @example
+ * ```typescript
+ * import { inngest } from './inngest';
+ * import { withInngestTrace } from 'tokenmeter/inngest';
+ * import { monitor } from 'tokenmeter';
+ * import OpenAI from 'openai';
+ *
+ * const openai = monitor(new OpenAI());
+ *
+ * export const processDocument = inngest.createFunction(
+ *   { id: 'process-document' },
+ *   { event: 'document/process' },
+ *   async ({ event }) => {
+ *     return withInngestTrace(event, async () => {
+ *       // All AI calls here are traced back to the original request
+ *       const result = await openai.chat.completions.create({
+ *         model: 'gpt-4o',
+ *         messages: [{ role: 'user', content: 'Summarize this document' }],
+ *       });
+ *
+ *       return { summary: result.choices[0].message.content };
+ *     });
+ *   }
+ * );
+ * ```
+ */
+export declare function withInngestTrace<T>(event: {
+    trace?: TraceHeaders;
+}, fn: () => Promise<T>): Promise<T>;
+/**
+ * Run a function within trace context and with additional attributes.
+ *
+ * Combines trace context restoration with attribute setting for Inngest functions.
+ *
+ * @example
+ * ```typescript
+ * export const processDocument = inngest.createFunction(
+ *   { id: 'process-document' },
+ *   { event: 'document/process' },
+ *   async ({ event }) => {
+ *     return withInngestTraceAndAttributes(
+ *       event,
+ *       { 'workflow.type': 'document-processing' },
+ *       async () => {
+ *         // AI calls have both parent trace and custom attributes
+ *         const result = await openai.chat.completions.create({...});
+ *         return { result };
+ *       }
+ *     );
+ *   }
+ * );
+ * ```
+ */
+export declare function withInngestTraceAndAttributes<T>(event: {
+    trace?: TraceHeaders;
+}, attributes: TokenMeterAttributes, fn: () => Promise<T>): Promise<T>;
+/**
+ * Inngest handler context type
+ */
+export interface InngestHandlerContext<TEvent = unknown> {
+    event: TEvent & {
+        trace?: TraceHeaders;
+    };
+    step: unknown;
+    [key: string]: unknown;
+}
+/**
+ * Inngest handler function type
+ */
+export type InngestHandler<TEvent, TResult> = (ctx: InngestHandlerContext<TEvent>) => Promise<TResult>;
+/**
+ * Wrap an Inngest function handler to automatically restore trace context.
+ *
+ * This is the recommended way to integrate TokenMeter with Inngest. It extracts
+ * trace headers from the event and runs your handler within that context.
+ *
+ * @example
+ * ```typescript
+ * import { inngest } from './inngest';
+ * import { withInngest } from 'tokenmeter/inngest';
+ * import { monitor } from 'tokenmeter';
+ * import OpenAI from 'openai';
+ *
+ * const openai = monitor(new OpenAI());
+ *
+ * export const processDocument = inngest.createFunction(
+ *   { id: 'process-document' },
+ *   { event: 'document/process' },
+ *   withInngest(async ({ event, step }) => {
+ *     // All AI calls here are traced back to the original request
+ *     const result = await openai.chat.completions.create({
+ *       model: 'gpt-4o',
+ *       messages: [{ role: 'user', content: 'Summarize this document' }],
+ *     });
+ *
+ *     return { summary: result.choices[0].message.content };
+ *   })
+ * );
+ * ```
+ */
+export declare function withInngest<TEvent, TResult>(handler: InngestHandler<TEvent, TResult>): InngestHandler<TEvent, TResult>;
+/**
+ * Create Inngest middleware that automatically handles trace context.
+ *
+ * This middleware extracts trace headers from incoming events and sets up
+ * the context for all AI calls within the function.
+ *
+ * @example
+ * ```typescript
+ * import { Inngest } from 'inngest';
+ * import { createInngestMiddleware } from 'tokenmeter/inngest';
+ *
+ * export const inngest = new Inngest({
+ *   id: 'my-app',
+ *   middleware: [createInngestMiddleware()],
+ * });
+ * ```
+ */
+export declare function createInngestMiddleware(): {
+    name: string;
+    init(): {
+        onFunctionRun({ fn, ctx, }: {
+            fn: unknown;
+            ctx: {
+                event: {
+                    trace?: TraceHeaders;
+                };
+            };
+        }): {
+            transformInput({ ctx: inputCtx, steps, }: {
+                ctx: unknown;
+                steps: unknown;
+            }): {
+                ctx: unknown;
+                steps: unknown;
+            };
+        };
+    };
+};
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/integrations/inngest/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAOH,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,gBAAgB,CAAC;AAE3D;;GAEG;AACH,MAAM,WAAW,YAAY;IAC3B,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED;;GAEG;AACH,MAAM,WAAW,qBAAqB,CAAC,CAAC,GAAG,OAAO;IAChD,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,CAAC,CAAC;IACR,4CAA4C;IAC5C,KAAK,CAAC,EAAE,YAAY,CAAC;CACtB;AAED;;GAEG;AACH,MAAM,WAAW,wBAAwB;IACvC,oDAAoD;IACpD,UAAU,CAAC,EAAE,oBAAoB,CAAC;CACnC;AAED;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AACH,wBAAgB,sBAAsB,IAAI,YAAY,CAErD;AAED;;;;;;;;;;;;;;;GAeG;AACH,wBAAgB,iBAAiB,CAAC,CAAC,EACjC,KAAK,EAAE;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,CAAC,CAAA;CAAE,EAChC,OAAO,CAAC,EAAE,wBAAwB,GACjC,qBAAqB,CAAC,CAAC,CAAC,CAK1B;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA+BG;AACH,wBAAsB,gBAAgB,CAAC,CAAC,EACtC,KAAK,EAAE;IAAE,KAAK,CAAC,EAAE,YAAY,CAAA;CAAE,EAC/B,EAAE,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,GACnB,OAAO,CAAC,CAAC,CAAC,CAcZ;AAED;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,wBAAsB,6BAA6B,CAAC,CAAC,EACnD,KAAK,EAAE;IAAE,KAAK,CAAC,EAAE,YAAY,CAAA;CAAE,EAC/B,UAAU,EAAE,oBAAoB,EAChC,EAAE,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,GACnB,OAAO,CAAC,CAAC,CAAC,CAEZ;AAED;;GAEG;AACH,MAAM,WAAW,qBAAqB,CAAC,MAAM,GAAG,OAAO;IACrD,KAAK,EAAE,MAAM,GAAG;QAAE,KAAK,CAAC,EAAE,YAAY,CAAA;KAAE,CAAC;IACzC,IAAI,EAAE,OAAO,CAAC;IACd,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACxB;AAED;;GAEG;AACH,MAAM,MAAM,cAAc,CAAC,MAAM,EAAE,OAAO,IAAI,CAC5C,GAAG,EAAE,qBAAqB,CAAC,MAAM,CAAC,KAC/B,OAAO,CAAC,OAAO,CAAC,CAAC;AAEtB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA6BG;AACH,wBAAgB,WAAW,CAAC,MAAM,EAAE,OAAO,EACzC,OAAO,EAAE,cAAc,CAAC,MAAM,EAAE,OAAO,CAAC,GACvC,cAAc,CAAC,MAAM,EAAE,OAAO,CAAC,CAgBjC;AAED;;;;;;;;;;;;;;;;GAgBG;AACH,wBAAgB,uBAAuB;;;oCAQ5B;YACD,EAAE,EAAE,OAAO,CAAC;YACZ,GAAG,EAAE;gBAAE,KAAK,EAAE;oBAAE,KAAK,CAAC,EAAE,YAAY,CAAA;iBAAE,CAAA;aAAE,CAAC;SAC1C;sDAKM;gBACD,GAAG,EAAE,OAAO,CAAC;gBACb,KAAK,EAAE,OAAO,CAAC;aAChB;;;;;;EA2BZ"}

package/dist/integrations/inngest/index.js

@@ -0,0 +1,223 @@
+/**
+ * Inngest Integration
+ *
+ * Provides helpers for propagating trace context to and from Inngest functions.
+ */
+import { extractTraceHeaders, withExtractedContext, withAttributes, } from "../../context.js";
+/**
+ * Extract trace headers to include when sending an Inngest event.
+ *
+ * Call this in your API route before sending an Inngest event to propagate
+ * the trace context to the Inngest function.
+ *
+ * @example
+ * ```typescript
+ * import { inngest } from './inngest';
+ * import { getInngestTraceHeaders } from 'tokenmeter/inngest';
+ *
+ * // In your API route
+ * export async function POST(req: Request) {
+ *   const trace = getInngestTraceHeaders();
+ *
+ *   await inngest.send({
+ *     name: 'document/process',
+ *     data: { documentId: '123' },
+ *     trace, // Pass trace headers
+ *   });
+ *
+ *   return Response.json({ ok: true });
+ * }
+ * ```
+ */
+export function getInngestTraceHeaders() {
+    return extractTraceHeaders();
+}
+/**
+ * Create a traced Inngest event with trace headers included.
+ *
+ * This is a convenience wrapper that combines your event data with trace headers.
+ *
+ * @example
+ * ```typescript
+ * import { inngest } from './inngest';
+ * import { createTracedEvent } from 'tokenmeter/inngest';
+ *
+ * await inngest.send(createTracedEvent({
+ *   name: 'document/process',
+ *   data: { documentId: '123' },
+ * }));
+ * ```
+ */
+export function createTracedEvent(event, options) {
+    return {
+        ...event,
+        trace: getInngestTraceHeaders(),
+    };
+}
+/**
+ * Run a function within the trace context from an Inngest event.
+ *
+ * Use this at the start of your Inngest function to resume the trace
+ * from the parent request.
+ *
+ * @example
+ * ```typescript
+ * import { inngest } from './inngest';
+ * import { withInngestTrace } from 'tokenmeter/inngest';
+ * import { monitor } from 'tokenmeter';
+ * import OpenAI from 'openai';
+ *
+ * const openai = monitor(new OpenAI());
+ *
+ * export const processDocument = inngest.createFunction(
+ *   { id: 'process-document' },
+ *   { event: 'document/process' },
+ *   async ({ event }) => {
+ *     return withInngestTrace(event, async () => {
+ *       // All AI calls here are traced back to the original request
+ *       const result = await openai.chat.completions.create({
+ *         model: 'gpt-4o',
+ *         messages: [{ role: 'user', content: 'Summarize this document' }],
+ *       });
+ *
+ *       return { summary: result.choices[0].message.content };
+ *     });
+ *   }
+ * );
+ * ```
+ */
+export async function withInngestTrace(event, fn) {
+    const headers = {};
+    if (event.trace?.traceparent) {
+        headers.traceparent = event.trace.traceparent;
+    }
+    if (event.trace?.tracestate) {
+        headers.tracestate = event.trace.tracestate;
+    }
+    if (event.trace?.baggage) {
+        headers.baggage = event.trace.baggage;
+    }
+    return withExtractedContext(headers, fn);
+}
+/**
+ * Run a function within trace context and with additional attributes.
+ *
+ * Combines trace context restoration with attribute setting for Inngest functions.
+ *
+ * @example
+ * ```typescript
+ * export const processDocument = inngest.createFunction(
+ *   { id: 'process-document' },
+ *   { event: 'document/process' },
+ *   async ({ event }) => {
+ *     return withInngestTraceAndAttributes(
+ *       event,
+ *       { 'workflow.type': 'document-processing' },
+ *       async () => {
+ *         // AI calls have both parent trace and custom attributes
+ *         const result = await openai.chat.completions.create({...});
+ *         return { result };
+ *       }
+ *     );
+ *   }
+ * );
+ * ```
+ */
+export async function withInngestTraceAndAttributes(event, attributes, fn) {
+    return withInngestTrace(event, () => withAttributes(attributes, fn));
+}
+/**
+ * Wrap an Inngest function handler to automatically restore trace context.
+ *
+ * This is the recommended way to integrate TokenMeter with Inngest. It extracts
+ * trace headers from the event and runs your handler within that context.
+ *
+ * @example
+ * ```typescript
+ * import { inngest } from './inngest';
+ * import { withInngest } from 'tokenmeter/inngest';
+ * import { monitor } from 'tokenmeter';
+ * import OpenAI from 'openai';
+ *
+ * const openai = monitor(new OpenAI());
+ *
+ * export const processDocument = inngest.createFunction(
+ *   { id: 'process-document' },
+ *   { event: 'document/process' },
+ *   withInngest(async ({ event, step }) => {
+ *     // All AI calls here are traced back to the original request
+ *     const result = await openai.chat.completions.create({
+ *       model: 'gpt-4o',
+ *       messages: [{ role: 'user', content: 'Summarize this document' }],
+ *     });
+ *
+ *     return { summary: result.choices[0].message.content };
+ *   })
+ * );
+ * ```
+ */
+export function withInngest(handler) {
+    return async (ctx) => {
+        const headers = {};
+        if (ctx.event?.trace?.traceparent) {
+            headers.traceparent = ctx.event.trace.traceparent;
+        }
+        if (ctx.event?.trace?.tracestate) {
+            headers.tracestate = ctx.event.trace.tracestate;
+        }
+        if (ctx.event?.trace?.baggage) {
+            headers.baggage = ctx.event.trace.baggage;
+        }
+        return withExtractedContext(headers, () => handler(ctx));
+    };
+}
+/**
+ * Create Inngest middleware that automatically handles trace context.
+ *
+ * This middleware extracts trace headers from incoming events and sets up
+ * the context for all AI calls within the function.
+ *
+ * @example
+ * ```typescript
+ * import { Inngest } from 'inngest';
+ * import { createInngestMiddleware } from 'tokenmeter/inngest';
+ *
+ * export const inngest = new Inngest({
+ *   id: 'my-app',
+ *   middleware: [createInngestMiddleware()],
+ * });
+ * ```
+ */
+export function createInngestMiddleware() {
+    return {
+        name: "tokenmeter",
+        init() {
+            return {
+                onFunctionRun({ fn, ctx, }) {
+                    return {
+                        transformInput({ ctx: inputCtx, steps, }) {
+                            // Extract trace headers from event
+                            const event = inputCtx
+                                ?.event;
+                            const headers = {};
+                            if (event?.trace?.traceparent) {
+                                headers.traceparent = event.trace.traceparent;
+                            }
+                            if (event?.trace?.tracestate) {
+                                headers.tracestate = event.trace.tracestate;
+                            }
+                            if (event?.trace?.baggage) {
+                                headers.baggage = event.trace.baggage;
+                            }
+                            // Note: Middleware can't wrap async execution directly
+                            // Users should still use withInngest() in their function
+                            // This middleware is for future compatibility
+                            return { ctx: inputCtx, steps };
+                        },
+                    };
+                },
+            };
+        },
+    };
+}
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/integrations/inngest/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,mBAAmB,EACnB,oBAAoB,EACpB,cAAc,GACf,MAAM,kBAAkB,CAAC;AA8B1B;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AACH,MAAM,UAAU,sBAAsB;IACpC,OAAO,mBAAmB,EAAkB,CAAC;AAC/C,CAAC;AAED;;;;;;;;;;;;;;;GAeG;AACH,MAAM,UAAU,iBAAiB,CAC/B,KAAgC,EAChC,OAAkC;IAElC,OAAO;QACL,GAAG,KAAK;QACR,KAAK,EAAE,sBAAsB,EAAE;KAChC,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA+BG;AACH,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,KAA+B,EAC/B,EAAoB;IAEpB,MAAM,OAAO,GAA2B,EAAE,CAAC;IAE3C,IAAI,KAAK,CAAC,KAAK,EAAE,WAAW,EAAE,CAAC;QAC7B,OAAO,CAAC,WAAW,GAAG,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC;IAChD,CAAC;IACD,IAAI,KAAK,CAAC,KAAK,EAAE,UAAU,EAAE,CAAC;QAC5B,OAAO,CAAC,UAAU,GAAG,KAAK,CAAC,KAAK,CAAC,UAAU,CAAC;IAC9C,CAAC;IACD,IAAI,KAAK,CAAC,KAAK,EAAE,OAAO,EAAE,CAAC;QACzB,OAAO,CAAC,OAAO,GAAG,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC;IACxC,CAAC;IAED,OAAO,oBAAoB,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;AAC3C,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,MAAM,CAAC,KAAK,UAAU,6BAA6B,CACjD,KAA+B,EAC/B,UAAgC,EAChC,EAAoB;IAEpB,OAAO,gBAAgB,CAAC,KAAK,EAAE,GAAG,EAAE,CAAC,cAAc,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,CAAC;AACvE,CAAC;AAkBD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA6BG;AACH,MAAM,UAAU,WAAW,CACzB,OAAwC;IAExC,OAAO,KAAK,EAAE,GAAkC,EAAoB,EAAE;QACpE,MAAM,OAAO,GAA2B,EAAE,CAAC;QAE3C,IAAI,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC;YAClC,OAAO,CAAC,WAAW,GAAG,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC;QACpD,CAAC;QACD,IAAI,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC;YACjC,OAAO,CAAC,UAAU,GAAG,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,UAAU,CAAC;QAClD,CAAC;QACD,IAAI,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;YAC9B,OAAO,CAAC,OAAO,GAAG,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC;QAC5C,CAAC;QAED,OAAO,oBAAoB,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC;IAC3D,CAAC,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;;;;;;GAgBG;AACH,MAAM,UAAU,uBAAuB;IACrC,OAAO;QACL,IAAI,EAAE,YAAY;QAClB,IAAI;YACF,OAAO;gBACL,aAAa,CAAC,EACZ,EAAE,EACF,GAAG,GAIJ;oBACC,OAAO;wBACL,cAAc,CAAC,EACb,GAAG,EAAE,QAAQ,EACb,KAAK,GAIN;4BACC,mCAAmC;4BACnC,MAAM,KAAK,GAAI,QAAiD;gCAC9D,EAAE,KAAK,CAAC;4BACV,MAAM,OAAO,GAA2B,EAAE,CAAC;4BAE3C,IAAI,KAAK,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC;gCAC9B,OAAO,CAAC,WAAW,GAAG,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC;4BAChD,CAAC;4BACD,IAAI,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC;gCAC7B,OAAO,CAAC,UAAU,GAAG,KAAK,CAAC,KAAK,CAAC,UAAU,CAAC;4BAC9C,CAAC;4BACD,IAAI,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;gCAC1B,OAAO,CAAC,OAAO,GAAG,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC;4BACxC,CAAC;4BAED,uDAAuD;4BACvD,yDAAyD;4BACzD,8CAA8C;4BAE9C,OAAO,EAAE,GAAG,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;wBAClC,CAAC;qBACF,CAAC;gBACJ,CAAC;aACF,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC"}