tokenmeter 0.9.0

package/dist/config.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,gBAAgB,EAChB,cAAc,EACd,aAAa,EACb,cAAc,EACf,MAAM,YAAY,CAAC;AAoCpB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAoCG;AACH,wBAAgB,IAAI,CAAC,MAAM,EAAE,gBAAgB,GAAG,IAAI,CA2CnD;AAED;;GAEG;AACH,wBAAgB,UAAU,IAAI,cAAc,CAO3C;AAED;;GAEG;AACH,wBAAgB,UAAU,IAAI,aAAa,CAE1C;AAED;;GAEG;AACH,wBAAgB,eAAe,IAAI,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAExD;AAED;;GAEG;AACH,wBAAgB,WAAW,IAAI,cAAc,CAE5C;AAED;;;;;;;;;;;GAWG;AACH,wBAAgB,aAAa,IAAI,OAAO,CAEvC;AAED;;GAEG;AACH,wBAAgB,YAAY,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,CAElD;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,wBAAsB,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC,CAM9C"}

package/dist/config.js

@@ -0,0 +1,166 @@
+import { PrismaStorageAdapter } from "./storage/prisma.js";
+import { defaultPricing, mergePricing } from "./pricing/index.js";
+import { initRecorder } from "./recorder.js";
+/**
+ * Default batching configuration
+ */
+const DEFAULT_BATCHING = {
+    enabled: true,
+    flushInterval: 1000,
+    maxBatchSize: 100,
+    flushOnExit: true,
+};
+const state = {
+    initialized: false,
+    storage: null,
+    pricing: defaultPricing,
+    modelAliases: {},
+    batching: DEFAULT_BATCHING,
+    identifierType: "userId",
+};
+/**
+ * Initialize tokenmeter with configuration.
+ *
+ * Must be called before using any tokenmeter functions. Typically called
+ * once at application startup.
+ *
+ * @param config - Configuration options
+ * @param config.databaseUrl - PostgreSQL connection URL (required if no storage adapter)
+ * @param config.storage - Custom storage adapter (overrides databaseUrl)
+ * @param config.pricing - Custom pricing overrides for models
+ * @param config.modelAliases - Map of model aliases to canonical names
+ * @param config.batching - Batching configuration (enabled, flushInterval, maxBatchSize, flushOnExit)
+ * @param config.identifierType - Label for the identifier field (documentation only)
+ *
+ * @throws Error if neither databaseUrl nor storage is provided
+ *
+ * @example
+ * ```typescript
+ * // With PostgreSQL
+ * init({
+ *   databaseUrl: process.env.DATABASE_URL,
+ * });
+ *
+ * // With custom storage adapter
+ * init({
+ *   storage: new InMemoryStorageAdapter(),
+ * });
+ *
+ * // With pricing overrides
+ * init({
+ *   databaseUrl: process.env.DATABASE_URL,
+ *   pricing: {
+ *     "custom-model": { input: 0.001, output: 0.002 },
+ *   },
+ * });
+ * ```
+ */
+export function init(config) {
+    if (state.initialized) {
+        console.warn("[tokenmeter] Already initialized. Call shutdown() before reinitializing.");
+        return;
+    }
+    // Set up storage adapter
+    if (config.storage) {
+        state.storage = config.storage;
+    }
+    else if (config.databaseUrl) {
+        state.storage = new PrismaStorageAdapter(config.databaseUrl);
+    }
+    else {
+        throw new Error("[tokenmeter] Either databaseUrl or storage adapter must be provided");
+    }
+    // Merge pricing with defaults
+    if (config.pricing) {
+        state.pricing = mergePricing(defaultPricing, config.pricing);
+    }
+    // Set model aliases
+    if (config.modelAliases) {
+        state.modelAliases = config.modelAliases;
+    }
+    // Merge batching config
+    if (config.batching) {
+        state.batching = { ...DEFAULT_BATCHING, ...config.batching };
+    }
+    // Set identifier type
+    if (config.identifierType) {
+        state.identifierType = config.identifierType;
+    }
+    state.initialized = true;
+    // Initialize recorder (registers shutdown hooks)
+    initRecorder();
+}
+/**
+ * Get the current storage adapter
+ */
+export function getStorage() {
+    if (!state.initialized || !state.storage) {
+        throw new Error("[tokenmeter] Not initialized. Call init() before using tokenmeter.");
+    }
+    return state.storage;
+}
+/**
+ * Get pricing configuration
+ */
+export function getPricing() {
+    return state.pricing;
+}
+/**
+ * Get model aliases
+ */
+export function getModelAliases() {
+    return state.modelAliases;
+}
+/**
+ * Get batching configuration
+ */
+export function getBatching() {
+    return state.batching;
+}
+/**
+ * Check if tokenmeter has been initialized.
+ *
+ * @returns true if init() has been called and shutdown() has not been called
+ *
+ * @example
+ * ```typescript
+ * if (!isInitialized()) {
+ *   init({ databaseUrl: process.env.DATABASE_URL });
+ * }
+ * ```
+ */
+export function isInitialized() {
+    return state.initialized;
+}
+/**
+ * Resolve a model name through aliases
+ */
+export function resolveModel(model) {
+    return state.modelAliases[model] || model;
+}
+/**
+ * Shutdown tokenmeter and clean up resources.
+ *
+ * Disconnects from the database and resets the initialized state.
+ * Call this before reinitializing or when shutting down your application.
+ *
+ * @returns A promise that resolves when shutdown is complete
+ *
+ * @example
+ * ```typescript
+ * // Graceful shutdown
+ * process.on("SIGTERM", async () => {
+ *   await flush();
+ *   await shutdown();
+ *   process.exit(0);
+ * });
+ * ```
+ */
+export async function shutdown() {
+    if (state.storage) {
+        await state.storage.disconnect();
+        state.storage = null;
+    }
+    state.initialized = false;
+}
+//# sourceMappingURL=config.js.map

package/dist/config.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.js","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,oBAAoB,EAAE,MAAM,qBAAqB,CAAC;AAC3D,OAAO,EAAE,cAAc,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClE,OAAO,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAE7C;;GAEG;AACH,MAAM,gBAAgB,GAAmB;IACvC,OAAO,EAAE,IAAI;IACb,aAAa,EAAE,IAAI;IACnB,YAAY,EAAE,GAAG;IACjB,WAAW,EAAE,IAAI;CAClB,CAAC;AAcF,MAAM,KAAK,GAAgB;IACzB,WAAW,EAAE,KAAK;IAClB,OAAO,EAAE,IAAI;IACb,OAAO,EAAE,cAAc;IACvB,YAAY,EAAE,EAAE;IAChB,QAAQ,EAAE,gBAAgB;IAC1B,cAAc,EAAE,QAAQ;CACzB,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAoCG;AACH,MAAM,UAAU,IAAI,CAAC,MAAwB;IAC3C,IAAI,KAAK,CAAC,WAAW,EAAE,CAAC;QACtB,OAAO,CAAC,IAAI,CACV,0EAA0E,CAC3E,CAAC;QACF,OAAO;IACT,CAAC;IAED,yBAAyB;IACzB,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,KAAK,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;IACjC,CAAC;SAAM,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC;QAC9B,KAAK,CAAC,OAAO,GAAG,IAAI,oBAAoB,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;IAC/D,CAAC;SAAM,CAAC;QACN,MAAM,IAAI,KAAK,CACb,qEAAqE,CACtE,CAAC;IACJ,CAAC;IAED,8BAA8B;IAC9B,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,KAAK,CAAC,OAAO,GAAG,YAAY,CAAC,cAAc,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;IAC/D,CAAC;IAED,oBAAoB;IACpB,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;QACxB,KAAK,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC;IAC3C,CAAC;IAED,wBAAwB;IACxB,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;QACpB,KAAK,CAAC,QAAQ,GAAG,EAAE,GAAG,gBAAgB,EAAE,GAAG,MAAM,CAAC,QAAQ,EAAE,CAAC;IAC/D,CAAC;IAED,sBAAsB;IACtB,IAAI,MAAM,CAAC,cAAc,EAAE,CAAC;QAC1B,KAAK,CAAC,cAAc,GAAG,MAAM,CAAC,cAAc,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,WAAW,GAAG,IAAI,CAAC;IAEzB,iDAAiD;IACjD,YAAY,EAAE,CAAC;AACjB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,UAAU;IACxB,IAAI,CAAC,KAAK,CAAC,WAAW,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;QACzC,MAAM,IAAI,KAAK,CACb,oEAAoE,CACrE,CAAC;IACJ,CAAC;IACD,OAAO,KAAK,CAAC,OAAO,CAAC;AACvB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,UAAU;IACxB,OAAO,KAAK,CAAC,OAAO,CAAC;AACvB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,eAAe;IAC7B,OAAO,KAAK,CAAC,YAAY,CAAC;AAC5B,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,WAAW;IACzB,OAAO,KAAK,CAAC,QAAQ,CAAC;AACxB,CAAC;AAED;;;;;;;;;;;GAWG;AACH,MAAM,UAAU,aAAa;IAC3B,OAAO,KAAK,CAAC,WAAW,CAAC;AAC3B,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,YAAY,CAAC,KAAa;IACxC,OAAO,KAAK,CAAC,YAAY,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC;AAC5C,CAAC;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,MAAM,CAAC,KAAK,UAAU,QAAQ;IAC5B,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;QAClB,MAAM,KAAK,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;QACjC,KAAK,CAAC,OAAO,GAAG,IAAI,CAAC;IACvB,CAAC;IACD,KAAK,CAAC,WAAW,GAAG,KAAK,CAAC;AAC5B,CAAC"}

package/dist/context.d.ts

@@ -0,0 +1,80 @@
+/**
+ * Context Management
+ *
+ * Provides a simple API for setting attributes that propagate to all spans
+ * created within a scope. Abstracts OTel Baggage/Context complexity.
+ */
+import { type Context } from "@opentelemetry/api";
+import type { TokenMeterAttributes } from "./types.js";
+/**
+ * Run a function with custom attributes that will be added to all spans created within.
+ *
+ * Uses OTel Baggage for propagation across async boundaries and to child spans.
+ *
+ * @param attributes - Key-value pairs to add to all spans in this scope
+ * @param fn - The async function to execute
+ * @returns The result of the function
+ *
+ * @example
+ * ```typescript
+ * import { withAttributes } from 'tokenmeter';
+ *
+ * await withAttributes({
+ *   'org.id': 'org_123',
+ *   'user.id': 'user_456',
+ *   'workflow.id': 'video-gen-99'
+ * }, async () => {
+ *   // All spans created here will have these attributes
+ *   await openai.chat.completions.create({...});
+ *   await fal.subscribe({...});
+ * });
+ * ```
+ */
+export declare function withAttributes<T>(attributes: TokenMeterAttributes, fn: () => Promise<T>): Promise<T>;
+/**
+ * Synchronous version of withAttributes for cases where async isn't needed
+ *
+ * @param attributes - Key-value pairs to add to all spans in this scope
+ * @param fn - The function to execute
+ * @returns The result of the function
+ */
+export declare function withAttributesSync<T>(attributes: TokenMeterAttributes, fn: () => T): T;
+/**
+ * Get the current attributes from the active context's baggage
+ *
+ * @returns Record of current baggage entries
+ */
+export declare function getCurrentAttributes(): Record<string, string>;
+/**
+ * Get a specific attribute from the current context
+ *
+ * @param key - The attribute key to retrieve
+ * @returns The attribute value, or undefined if not set
+ */
+export declare function getAttribute(key: string): string | undefined;
+/**
+ * Extract traceparent header for cross-service propagation
+ *
+ * Useful for propagating context to external services (e.g., Inngest jobs)
+ *
+ * @returns Object with traceparent and tracestate headers
+ */
+export declare function extractTraceHeaders(): Record<string, string>;
+/**
+ * Create a context from trace headers (for receiving propagated context)
+ *
+ * @param headers - Object containing traceparent/tracestate headers
+ * @returns OTel Context with extracted trace info
+ */
+export declare function contextFromHeaders(headers: Record<string, string>): Context;
+/**
+ * Run a function within a context extracted from headers
+ *
+ * Useful for continuing a trace from an external source (e.g., incoming webhook)
+ *
+ * @param headers - Object containing traceparent/tracestate headers
+ * @param fn - The async function to execute
+ * @returns The result of the function
+ */
+export declare function withExtractedContext<T>(headers: Record<string, string>, fn: () => Promise<T>): Promise<T>;
+//# sourceMappingURL=context.d.ts.map

package/dist/context.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"context.d.ts","sourceRoot":"","sources":["../src/context.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAIL,KAAK,OAAO,EAGb,MAAM,oBAAoB,CAAC;AAC5B,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,YAAY,CAAC;AAEvD;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,wBAAsB,cAAc,CAAC,CAAC,EACpC,UAAU,EAAE,oBAAoB,EAChC,EAAE,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,GACnB,OAAO,CAAC,CAAC,CAAC,CAkBZ;AAED;;;;;;GAMG;AACH,wBAAgB,kBAAkB,CAAC,CAAC,EAClC,UAAU,EAAE,oBAAoB,EAChC,EAAE,EAAE,MAAM,CAAC,GACV,CAAC,CAaH;AAED;;;;GAIG;AACH,wBAAgB,oBAAoB,IAAI,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAc7D;AAED;;;;;GAKG;AACH,wBAAgB,YAAY,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CAU5D;AAED;;;;;;GAMG;AACH,wBAAgB,mBAAmB,IAAI,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAI5D;AAED;;;;;GAKG;AACH,wBAAgB,kBAAkB,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,OAAO,CAE3E;AAED;;;;;;;;GAQG;AACH,wBAAsB,oBAAoB,CAAC,CAAC,EAC1C,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,EAC/B,EAAE,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,GACnB,OAAO,CAAC,CAAC,CAAC,CAGZ"}

package/dist/context.js

@@ -0,0 +1,131 @@
+/**
+ * Context Management
+ *
+ * Provides a simple API for setting attributes that propagate to all spans
+ * created within a scope. Abstracts OTel Baggage/Context complexity.
+ */
+import { context, propagation, ROOT_CONTEXT, } from "@opentelemetry/api";
+/**
+ * Run a function with custom attributes that will be added to all spans created within.
+ *
+ * Uses OTel Baggage for propagation across async boundaries and to child spans.
+ *
+ * @param attributes - Key-value pairs to add to all spans in this scope
+ * @param fn - The async function to execute
+ * @returns The result of the function
+ *
+ * @example
+ * ```typescript
+ * import { withAttributes } from 'tokenmeter';
+ *
+ * await withAttributes({
+ *   'org.id': 'org_123',
+ *   'user.id': 'user_456',
+ *   'workflow.id': 'video-gen-99'
+ * }, async () => {
+ *   // All spans created here will have these attributes
+ *   await openai.chat.completions.create({...});
+ *   await fal.subscribe({...});
+ * });
+ * ```
+ */
+export async function withAttributes(attributes, fn) {
+    // Get current baggage or create new one
+    const currentContext = context.active();
+    let baggage = propagation.getBaggage(currentContext) || propagation.createBaggage();
+    // Add attributes to baggage
+    for (const [key, value] of Object.entries(attributes)) {
+        baggage = baggage.setEntry(key, {
+            value: String(value),
+        });
+    }
+    // Create new context with updated baggage
+    const newContext = propagation.setBaggage(currentContext, baggage);
+    // Run function within the new context
+    return context.with(newContext, fn);
+}
+/**
+ * Synchronous version of withAttributes for cases where async isn't needed
+ *
+ * @param attributes - Key-value pairs to add to all spans in this scope
+ * @param fn - The function to execute
+ * @returns The result of the function
+ */
+export function withAttributesSync(attributes, fn) {
+    const currentContext = context.active();
+    let baggage = propagation.getBaggage(currentContext) || propagation.createBaggage();
+    for (const [key, value] of Object.entries(attributes)) {
+        baggage = baggage.setEntry(key, {
+            value: String(value),
+        });
+    }
+    const newContext = propagation.setBaggage(currentContext, baggage);
+    return context.with(newContext, fn);
+}
+/**
+ * Get the current attributes from the active context's baggage
+ *
+ * @returns Record of current baggage entries
+ */
+export function getCurrentAttributes() {
+    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;
+}
+/**
+ * Get a specific attribute from the current context
+ *
+ * @param key - The attribute key to retrieve
+ * @returns The attribute value, or undefined if not set
+ */
+export function getAttribute(key) {
+    const currentContext = context.active();
+    const baggage = propagation.getBaggage(currentContext);
+    if (!baggage) {
+        return undefined;
+    }
+    const entry = baggage.getEntry(key);
+    return entry?.value;
+}
+/**
+ * Extract traceparent header for cross-service propagation
+ *
+ * Useful for propagating context to external services (e.g., Inngest jobs)
+ *
+ * @returns Object with traceparent and tracestate headers
+ */
+export function extractTraceHeaders() {
+    const headers = {};
+    propagation.inject(context.active(), headers);
+    return headers;
+}
+/**
+ * Create a context from trace headers (for receiving propagated context)
+ *
+ * @param headers - Object containing traceparent/tracestate headers
+ * @returns OTel Context with extracted trace info
+ */
+export function contextFromHeaders(headers) {
+    return propagation.extract(ROOT_CONTEXT, headers);
+}
+/**
+ * Run a function within a context extracted from headers
+ *
+ * Useful for continuing a trace from an external source (e.g., incoming webhook)
+ *
+ * @param headers - Object containing traceparent/tracestate headers
+ * @param fn - The async function to execute
+ * @returns The result of the function
+ */
+export async function withExtractedContext(headers, fn) {
+    const extractedContext = contextFromHeaders(headers);
+    return context.with(extractedContext, fn);
+}
+//# sourceMappingURL=context.js.map

package/dist/context.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"context.js","sourceRoot":"","sources":["../src/context.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EACL,OAAO,EACP,WAAW,EACX,YAAY,GAIb,MAAM,oBAAoB,CAAC;AAG5B;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,UAAgC,EAChC,EAAoB;IAEpB,wCAAwC;IACxC,MAAM,cAAc,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IACxC,IAAI,OAAO,GACT,WAAW,CAAC,UAAU,CAAC,cAAc,CAAC,IAAI,WAAW,CAAC,aAAa,EAAE,CAAC;IAExE,4BAA4B;IAC5B,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;QACtD,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE;YAC9B,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC;SACrB,CAAC,CAAC;IACL,CAAC;IAED,0CAA0C;IAC1C,MAAM,UAAU,GAAG,WAAW,CAAC,UAAU,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;IAEnE,sCAAsC;IACtC,OAAO,OAAO,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;AACtC,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,kBAAkB,CAChC,UAAgC,EAChC,EAAW;IAEX,MAAM,cAAc,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IACxC,IAAI,OAAO,GACT,WAAW,CAAC,UAAU,CAAC,cAAc,CAAC,IAAI,WAAW,CAAC,aAAa,EAAE,CAAC;IAExE,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;QACtD,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE;YAC9B,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC;SACrB,CAAC,CAAC;IACL,CAAC;IAED,MAAM,UAAU,GAAG,WAAW,CAAC,UAAU,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;IACnE,OAAO,OAAO,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;AACtC,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,oBAAoB;IAClC,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;;;;;GAKG;AACH,MAAM,UAAU,YAAY,CAAC,GAAW;IACtC,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,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;IACpC,OAAO,KAAK,EAAE,KAAK,CAAC;AACtB,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,mBAAmB;IACjC,MAAM,OAAO,GAA2B,EAAE,CAAC;IAC3C,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,OAAO,CAAC,CAAC;IAC9C,OAAO,OAAO,CAAC;AACjB,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,kBAAkB,CAAC,OAA+B;IAChE,OAAO,WAAW,CAAC,OAAO,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;AACpD,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,CAAC,KAAK,UAAU,oBAAoB,CACxC,OAA+B,EAC/B,EAAoB;IAEpB,MAAM,gBAAgB,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAC;IACrD,OAAO,OAAO,CAAC,IAAI,CAAC,gBAAgB,EAAE,EAAE,CAAC,CAAC;AAC5C,CAAC"}

package/dist/exporter/PostgresExporter.d.ts

@@ -0,0 +1,82 @@
+/**
+ * PostgreSQL Span Exporter
+ *
+ * Exports tokenmeter cost spans to PostgreSQL for persistence and querying.
+ */
+import type { ReadableSpan } from "@opentelemetry/sdk-trace-base";
+import type { ExportResult } from "@opentelemetry/core";
+import type { PostgresExporterConfig } from "../types.js";
+/**
+ * SpanExporter interface (subset we need)
+ * We define this to avoid requiring @opentelemetry/sdk-trace-base at runtime
+ */
+export interface SpanExporter {
+    export(spans: ReadableSpan[], resultCallback: (result: ExportResult) => void): void;
+    shutdown(): Promise<void>;
+}
+/**
+ * PostgreSQL exporter for tokenmeter cost spans
+ *
+ * @example
+ * ```typescript
+ * import { NodeTracerProvider } from '@opentelemetry/sdk-trace-node';
+ * import { BatchSpanProcessor } from '@opentelemetry/sdk-trace-base';
+ * import { PostgresExporter } from 'tokenmeter/exporter';
+ *
+ * const exporter = new PostgresExporter({
+ *   connectionString: process.env.DATABASE_URL,
+ * });
+ *
+ * const provider = new NodeTracerProvider();
+ * provider.addSpanProcessor(new BatchSpanProcessor(exporter));
+ * provider.register();
+ * ```
+ */
+export declare class PostgresExporter implements SpanExporter {
+    private config;
+    private pool;
+    private buffer;
+    private flushTimer;
+    private flushPromise;
+    private isShuttingDown;
+    constructor(config: PostgresExporterConfig);
+    /**
+     * Lazily initialize the database pool
+     */
+    private getPool;
+    /**
+     * Export spans to PostgreSQL
+     */
+    export(spans: ReadableSpan[], resultCallback: (result: ExportResult) => void): void;
+    /**
+     * Convert a ReadableSpan to a CostRecord
+     * Returns null if the span doesn't have cost data
+     */
+    private spanToCostRecord;
+    /**
+     * Extract custom attributes (exclude standard tokenmeter attributes)
+     */
+    private extractCustomAttributes;
+    /**
+     * Convert HrTime to Date
+     */
+    private hrTimeToDate;
+    /**
+     * Flush buffered records to the database
+     */
+    private flushBuffer;
+    /**
+     * Insert records into the database
+     */
+    private insertRecords;
+    /**
+     * Shutdown the exporter
+     */
+    shutdown(): Promise<void>;
+    /**
+     * Force flush (for testing or manual control)
+     */
+    forceFlush(): Promise<void>;
+}
+export default PostgresExporter;
+//# sourceMappingURL=PostgresExporter.d.ts.map

package/dist/exporter/PostgresExporter.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"PostgresExporter.d.ts","sourceRoot":"","sources":["../../src/exporter/PostgresExporter.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAC;AAClE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAExD,OAAO,KAAK,EAAE,sBAAsB,EAAc,MAAM,aAAa,CAAC;AAItE;;;GAGG;AACH,MAAM,WAAW,YAAY;IAC3B,MAAM,CACJ,KAAK,EAAE,YAAY,EAAE,EACrB,cAAc,EAAE,CAAC,MAAM,EAAE,YAAY,KAAK,IAAI,GAC7C,IAAI,CAAC;IACR,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;CAC3B;AAoBD;;;;;;;;;;;;;;;;;GAiBG;AACH,qBAAa,gBAAiB,YAAW,YAAY;IACnD,OAAO,CAAC,MAAM,CAAiB;IAC/B,OAAO,CAAC,IAAI,CAAqB;IACjC,OAAO,CAAC,MAAM,CAAoB;IAClC,OAAO,CAAC,UAAU,CAA+C;IACjE,OAAO,CAAC,YAAY,CAA8B;IAClD,OAAO,CAAC,cAAc,CAAS;gBAEnB,MAAM,EAAE,sBAAsB;IAgB1C;;OAEG;YACW,OAAO;IAcrB;;OAEG;IACH,MAAM,CACJ,KAAK,EAAE,YAAY,EAAE,EACrB,cAAc,EAAE,CAAC,MAAM,EAAE,YAAY,KAAK,IAAI,GAC7C,IAAI;IAkCP;;;OAGG;IACH,OAAO,CAAC,gBAAgB;IA+BxB;;OAEG;IACH,OAAO,CAAC,uBAAuB;IAiB/B;;OAEG;IACH,OAAO,CAAC,YAAY;IAKpB;;OAEG;YACW,WAAW;IAqBzB;;OAEG;YACW,aAAa;IA8D3B;;OAEG;IACG,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IAmB/B;;OAEG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;CAGlC;AAED,eAAe,gBAAgB,CAAC"}