tokenmeter 0.9.0

package/dist/exporter/PostgresExporter.js

@@ -0,0 +1,237 @@
+/**
+ * PostgreSQL Span Exporter
+ *
+ * Exports tokenmeter cost spans to PostgreSQL for persistence and querying.
+ */
+import { ExportResultCode } from "@opentelemetry/core";
+import { TM_ATTRIBUTES } from "../types.js";
+import { logger } from "../logger.js";
+/**
+ * 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 class PostgresExporter {
+    config;
+    pool = null;
+    buffer = [];
+    flushTimer = null;
+    flushPromise = null;
+    isShuttingDown = false;
+    constructor(config) {
+        this.config = {
+            connectionString: config.connectionString,
+            tableName: config.tableName ?? "tokenmeter_events",
+            batchSize: config.batchSize ?? 100,
+            flushIntervalMs: config.flushIntervalMs ?? 5000,
+        };
+        // Start flush interval
+        this.flushTimer = setInterval(() => {
+            this.flushBuffer().catch((err) => {
+                logger.error("Error flushing buffer:", err);
+            });
+        }, this.config.flushIntervalMs);
+    }
+    /**
+     * Lazily initialize the database pool
+     */
+    async getPool() {
+        if (this.pool) {
+            return this.pool;
+        }
+        // Dynamic import to avoid requiring pg at load time
+        const { Pool } = await import("pg");
+        this.pool = new Pool({
+            connectionString: this.config.connectionString,
+        });
+        return this.pool;
+    }
+    /**
+     * Export spans to PostgreSQL
+     */
+    export(spans, resultCallback) {
+        if (this.isShuttingDown) {
+            resultCallback({ code: ExportResultCode.FAILED });
+            return;
+        }
+        // Filter and convert spans to cost records
+        const records = spans
+            .map((span) => this.spanToCostRecord(span))
+            .filter((record) => record !== null);
+        if (records.length === 0) {
+            resultCallback({ code: ExportResultCode.SUCCESS });
+            return;
+        }
+        // Add to buffer
+        this.buffer.push(...records);
+        // Flush if buffer is full
+        if (this.buffer.length >= this.config.batchSize) {
+            this.flushBuffer()
+                .then(() => {
+                resultCallback({ code: ExportResultCode.SUCCESS });
+            })
+                .catch((err) => {
+                logger.error("Export error:", err);
+                resultCallback({ code: ExportResultCode.FAILED });
+            });
+        }
+        else {
+            resultCallback({ code: ExportResultCode.SUCCESS });
+        }
+    }
+    /**
+     * Convert a ReadableSpan to a CostRecord
+     * Returns null if the span doesn't have cost data
+     */
+    spanToCostRecord(span) {
+        const attrs = span.attributes;
+        // Only export spans with tokenmeter cost data
+        const costUsd = attrs[TM_ATTRIBUTES.COST_USD];
+        const provider = attrs[TM_ATTRIBUTES.PROVIDER];
+        const model = attrs[TM_ATTRIBUTES.MODEL];
+        // Skip spans without cost data
+        if (costUsd === undefined || !provider || !model) {
+            return null;
+        }
+        const spanContext = span.spanContext();
+        return {
+            id: crypto.randomUUID(),
+            traceId: spanContext.traceId,
+            spanId: spanContext.spanId,
+            provider,
+            model,
+            organizationId: attrs[TM_ATTRIBUTES.ORG_ID],
+            userId: attrs[TM_ATTRIBUTES.USER_ID],
+            costUsd,
+            inputUnits: attrs[TM_ATTRIBUTES.INPUT_UNITS],
+            outputUnits: attrs[TM_ATTRIBUTES.OUTPUT_UNITS],
+            attributes: this.extractCustomAttributes(attrs),
+            createdAt: this.hrTimeToDate(span.startTime),
+        };
+    }
+    /**
+     * Extract custom attributes (exclude standard tokenmeter attributes)
+     */
+    extractCustomAttributes(attrs) {
+        const standardKeys = new Set(Object.values(TM_ATTRIBUTES));
+        const custom = {};
+        let hasCustom = false;
+        for (const [key, value] of Object.entries(attrs)) {
+            if (!standardKeys.has(key) && !key.startsWith("gen_ai.")) {
+                custom[key] = value;
+                hasCustom = true;
+            }
+        }
+        return hasCustom ? custom : undefined;
+    }
+    /**
+     * Convert HrTime to Date
+     */
+    hrTimeToDate(hrTime) {
+        const milliseconds = hrTime[0] * 1000 + hrTime[1] / 1_000_000;
+        return new Date(milliseconds);
+    }
+    /**
+     * Flush buffered records to the database
+     */
+    async flushBuffer() {
+        // If already flushing, wait for that to complete
+        if (this.flushPromise) {
+            return this.flushPromise;
+        }
+        if (this.buffer.length === 0) {
+            return;
+        }
+        // Take current buffer and clear it
+        const records = this.buffer;
+        this.buffer = [];
+        this.flushPromise = this.insertRecords(records).finally(() => {
+            this.flushPromise = null;
+        });
+        return this.flushPromise;
+    }
+    /**
+     * Insert records into the database
+     */
+    async insertRecords(records) {
+        if (records.length === 0) {
+            return;
+        }
+        const pool = await this.getPool();
+        const tableName = this.config.tableName;
+        // Build bulk insert query
+        const columns = [
+            "id",
+            "trace_id",
+            "span_id",
+            "provider",
+            "model",
+            "organization_id",
+            "user_id",
+            "cost_usd",
+            "input_tokens",
+            "output_tokens",
+            "metadata",
+            "created_at",
+        ];
+        const values = [];
+        const placeholders = [];
+        records.forEach((record, i) => {
+            const offset = i * columns.length;
+            const rowPlaceholders = columns.map((_, j) => `$${offset + j + 1}`);
+            placeholders.push(`(${rowPlaceholders.join(", ")})`);
+            values.push(record.id, record.traceId, record.spanId, record.provider, record.model, record.organizationId ?? null, record.userId ?? null, record.costUsd, record.inputUnits ?? null, record.outputUnits ?? null, record.attributes ? JSON.stringify(record.attributes) : null, record.createdAt);
+        });
+        const query = `
+      INSERT INTO ${tableName} (${columns.join(", ")})
+      VALUES ${placeholders.join(", ")}
+      ON CONFLICT (id) DO NOTHING
+    `;
+        try {
+            await pool.query(query, values);
+        }
+        catch (err) {
+            logger.error("Failed to insert records:", err);
+            throw err;
+        }
+    }
+    /**
+     * Shutdown the exporter
+     */
+    async shutdown() {
+        this.isShuttingDown = true;
+        // Stop flush timer
+        if (this.flushTimer) {
+            clearInterval(this.flushTimer);
+            this.flushTimer = null;
+        }
+        // Flush remaining records
+        await this.flushBuffer();
+        // Close pool
+        if (this.pool) {
+            await this.pool.end();
+            this.pool = null;
+        }
+    }
+    /**
+     * Force flush (for testing or manual control)
+     */
+    async forceFlush() {
+        await this.flushBuffer();
+    }
+}
+export default PostgresExporter;
+//# sourceMappingURL=PostgresExporter.js.map

package/dist/exporter/PostgresExporter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"PostgresExporter.js","sourceRoot":"","sources":["../../src/exporter/PostgresExporter.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAIH,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAEvD,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAgCtC;;;;;;;;;;;;;;;;;GAiBG;AACH,MAAM,OAAO,gBAAgB;IACnB,MAAM,CAAiB;IACvB,IAAI,GAAgB,IAAI,CAAC;IACzB,MAAM,GAAiB,EAAE,CAAC;IAC1B,UAAU,GAA0C,IAAI,CAAC;IACzD,YAAY,GAAyB,IAAI,CAAC;IAC1C,cAAc,GAAG,KAAK,CAAC;IAE/B,YAAY,MAA8B;QACxC,IAAI,CAAC,MAAM,GAAG;YACZ,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;YACzC,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,mBAAmB;YAClD,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,GAAG;YAClC,eAAe,EAAE,MAAM,CAAC,eAAe,IAAI,IAAI;SAChD,CAAC;QAEF,uBAAuB;QACvB,IAAI,CAAC,UAAU,GAAG,WAAW,CAAC,GAAG,EAAE;YACjC,IAAI,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;gBAC/B,MAAM,CAAC,KAAK,CAAC,wBAAwB,EAAE,GAAG,CAAC,CAAC;YAC9C,CAAC,CAAC,CAAC;QACL,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;IAClC,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,OAAO;QACnB,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YACd,OAAO,IAAI,CAAC,IAAI,CAAC;QACnB,CAAC;QAED,oDAAoD;QACpD,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC;QACpC,IAAI,CAAC,IAAI,GAAG,IAAI,IAAI,CAAC;YACnB,gBAAgB,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB;SAC/C,CAAC,CAAC;QAEH,OAAO,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC;IAED;;OAEG;IACH,MAAM,CACJ,KAAqB,EACrB,cAA8C;QAE9C,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,cAAc,CAAC,EAAE,IAAI,EAAE,gBAAgB,CAAC,MAAM,EAAE,CAAC,CAAC;YAClD,OAAO;QACT,CAAC;QAED,2CAA2C;QAC3C,MAAM,OAAO,GAAG,KAAK;aAClB,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;aAC1C,MAAM,CAAC,CAAC,MAAM,EAAwB,EAAE,CAAC,MAAM,KAAK,IAAI,CAAC,CAAC;QAE7D,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzB,cAAc,CAAC,EAAE,IAAI,EAAE,gBAAgB,CAAC,OAAO,EAAE,CAAC,CAAC;YACnD,OAAO;QACT,CAAC;QAED,gBAAgB;QAChB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,CAAC;QAE7B,0BAA0B;QAC1B,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;YAChD,IAAI,CAAC,WAAW,EAAE;iBACf,IAAI,CAAC,GAAG,EAAE;gBACT,cAAc,CAAC,EAAE,IAAI,EAAE,gBAAgB,CAAC,OAAO,EAAE,CAAC,CAAC;YACrD,CAAC,CAAC;iBACD,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;gBACb,MAAM,CAAC,KAAK,CAAC,eAAe,EAAE,GAAG,CAAC,CAAC;gBACnC,cAAc,CAAC,EAAE,IAAI,EAAE,gBAAgB,CAAC,MAAM,EAAE,CAAC,CAAC;YACpD,CAAC,CAAC,CAAC;QACP,CAAC;aAAM,CAAC;YACN,cAAc,CAAC,EAAE,IAAI,EAAE,gBAAgB,CAAC,OAAO,EAAE,CAAC,CAAC;QACrD,CAAC;IACH,CAAC;IAED;;;OAGG;IACK,gBAAgB,CAAC,IAAkB;QACzC,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC;QAE9B,8CAA8C;QAC9C,MAAM,OAAO,GAAG,KAAK,CAAC,aAAa,CAAC,QAAQ,CAAuB,CAAC;QACpE,MAAM,QAAQ,GAAG,KAAK,CAAC,aAAa,CAAC,QAAQ,CAAuB,CAAC;QACrE,MAAM,KAAK,GAAG,KAAK,CAAC,aAAa,CAAC,KAAK,CAAuB,CAAC;QAE/D,+BAA+B;QAC/B,IAAI,OAAO,KAAK,SAAS,IAAI,CAAC,QAAQ,IAAI,CAAC,KAAK,EAAE,CAAC;YACjD,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;QAEvC,OAAO;YACL,EAAE,EAAE,MAAM,CAAC,UAAU,EAAE;YACvB,OAAO,EAAE,WAAW,CAAC,OAAO;YAC5B,MAAM,EAAE,WAAW,CAAC,MAAM;YAC1B,QAAQ;YACR,KAAK;YACL,cAAc,EAAE,KAAK,CAAC,aAAa,CAAC,MAAM,CAAuB;YACjE,MAAM,EAAE,KAAK,CAAC,aAAa,CAAC,OAAO,CAAuB;YAC1D,OAAO;YACP,UAAU,EAAE,KAAK,CAAC,aAAa,CAAC,WAAW,CAAuB;YAClE,WAAW,EAAE,KAAK,CAAC,aAAa,CAAC,YAAY,CAAuB;YACpE,UAAU,EAAE,IAAI,CAAC,uBAAuB,CAAC,KAAK,CAAC;YAC/C,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC;SAC7C,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,uBAAuB,CAC7B,KAA8B;QAE9B,MAAM,YAAY,GAAG,IAAI,GAAG,CAAS,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC;QACnE,MAAM,MAAM,GAA4B,EAAE,CAAC;QAC3C,IAAI,SAAS,GAAG,KAAK,CAAC;QAEtB,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YACjD,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;gBACzD,MAAM,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;gBACpB,SAAS,GAAG,IAAI,CAAC;YACnB,CAAC;QACH,CAAC;QAED,OAAO,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;IACxC,CAAC;IAED;;OAEG;IACK,YAAY,CAAC,MAAwB;QAC3C,MAAM,YAAY,GAAG,MAAM,CAAC,CAAC,CAAC,GAAG,IAAI,GAAG,MAAM,CAAC,CAAC,CAAC,GAAG,SAAS,CAAC;QAC9D,OAAO,IAAI,IAAI,CAAC,YAAY,CAAC,CAAC;IAChC,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,WAAW;QACvB,iDAAiD;QACjD,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,OAAO,IAAI,CAAC,YAAY,CAAC;QAC3B,CAAC;QAED,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC7B,OAAO;QACT,CAAC;QAED,mCAAmC;QACnC,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC;QAEjB,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE;YAC3D,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QAC3B,CAAC,CAAC,CAAC;QAEH,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,aAAa,CAAC,OAAqB;QAC/C,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzB,OAAO;QACT,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QAClC,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC;QAExC,0BAA0B;QAC1B,MAAM,OAAO,GAAG;YACd,IAAI;YACJ,UAAU;YACV,SAAS;YACT,UAAU;YACV,OAAO;YACP,iBAAiB;YACjB,SAAS;YACT,UAAU;YACV,cAAc;YACd,eAAe;YACf,UAAU;YACV,YAAY;SACb,CAAC;QAEF,MAAM,MAAM,GAAc,EAAE,CAAC;QAC7B,MAAM,YAAY,GAAa,EAAE,CAAC;QAElC,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;YAC5B,MAAM,MAAM,GAAG,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;YAClC,MAAM,eAAe,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,IAAI,MAAM,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACpE,YAAY,CAAC,IAAI,CAAC,IAAI,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAErD,MAAM,CAAC,IAAI,CACT,MAAM,CAAC,EAAE,EACT,MAAM,CAAC,OAAO,EACd,MAAM,CAAC,MAAM,EACb,MAAM,CAAC,QAAQ,EACf,MAAM,CAAC,KAAK,EACZ,MAAM,CAAC,cAAc,IAAI,IAAI,EAC7B,MAAM,CAAC,MAAM,IAAI,IAAI,EACrB,MAAM,CAAC,OAAO,EACd,MAAM,CAAC,UAAU,IAAI,IAAI,EACzB,MAAM,CAAC,WAAW,IAAI,IAAI,EAC1B,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,IAAI,EAC5D,MAAM,CAAC,SAAS,CACjB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG;oBACE,SAAS,KAAK,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC;eACrC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC;;KAEjC,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAClC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,CAAC,KAAK,CAAC,2BAA2B,EAAE,GAAG,CAAC,CAAC;YAC/C,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ;QACZ,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAE3B,mBAAmB;QACnB,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,aAAa,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAC/B,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QACzB,CAAC;QAED,0BAA0B;QAC1B,MAAM,IAAI,CAAC,WAAW,EAAE,CAAC;QAEzB,aAAa;QACb,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YACd,MAAM,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;YACtB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACnB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU;QACd,MAAM,IAAI,CAAC,WAAW,EAAE,CAAC;IAC3B,CAAC;CACF;AAED,eAAe,gBAAgB,CAAC"}

package/dist/exporter/index.d.ts

@@ -0,0 +1,8 @@
+/**
+ * Exporter module
+ *
+ * Provides the PostgresExporter for persisting cost data.
+ */
+export { PostgresExporter } from "./PostgresExporter.js";
+export type { SpanExporter } from "./PostgresExporter.js";
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/exporter/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AACzD,YAAY,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC"}

package/dist/exporter/index.js

@@ -0,0 +1,7 @@
+/**
+ * Exporter module
+ *
+ * Provides the PostgresExporter for persisting cost data.
+ */
+export { PostgresExporter } from "./PostgresExporter.js";
+//# sourceMappingURL=index.js.map

package/dist/exporter/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/exporter/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC"}

package/dist/index.d.ts

@@ -0,0 +1,31 @@
+/**
+ * TokenMeter v5
+ *
+ * OpenTelemetry-native cost tracking for AI workflows.
+ *
+ * @example
+ * ```typescript
+ * import OpenAI from 'openai';
+ * import { monitor, withAttributes } from 'tokenmeter';
+ *
+ * // 1. Wrap your client
+ * const openai = monitor(new OpenAI({ apiKey: '...' }));
+ *
+ * // 2. Track with attributes
+ * await withAttributes({ 'org.id': 'org_123' }, async () => {
+ *   await openai.chat.completions.create({
+ *     model: 'gpt-4o',
+ *     messages: [{ role: 'user', content: 'Hello!' }]
+ *   });
+ * });
+ * ```
+ */
+export { monitor } from "./instrumentation/proxy.js";
+export { withAttributes, withAttributesSync, getCurrentAttributes, getAttribute, extractTraceHeaders, contextFromHeaders, withExtractedContext, } from "./context.js";
+export { TokenMeterProcessor } from "./processor/TokenMeterProcessor.js";
+export { loadManifest, getModelPricing, calculateCost, getCachedManifest, clearManifestCache, configurePricing, getPricingConfig, setModelAliases, clearModelAliases, getModelAliases, type PricingConfig, type ModelAlias, } from "./pricing/manifest.js";
+export type { PricingUnit, ModelPricing, ProviderPricing, PricingManifest, MonitorOptions, UsageData, ExtractionStrategy, TokenMeterAttributes, TokenMeterProcessorConfig, PostgresExporterConfig, CostRecord, CostQueryOptions, CostResult, } from "./types.js";
+export { TM_ATTRIBUTES, GEN_AI_ATTRIBUTES } from "./types.js";
+export { configureLogger, getLoggerConfig, resetLogger, type LogLevel, type LoggerConfig, } from "./logger.js";
+export { strategies, findStrategy, extractUsage, openaiStrategy, anthropicStrategy, bedrockStrategy, vertexAIStrategy, falStrategy, bflStrategy, elevenlabsStrategy, vercelAIStrategy, } from "./instrumentation/strategies/index.js";
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;GAqBG;AAGH,OAAO,EAAE,OAAO,EAAE,MAAM,4BAA4B,CAAC;AAGrD,OAAO,EACL,cAAc,EACd,kBAAkB,EAClB,oBAAoB,EACpB,YAAY,EACZ,mBAAmB,EACnB,kBAAkB,EAClB,oBAAoB,GACrB,MAAM,cAAc,CAAC;AAGtB,OAAO,EAAE,mBAAmB,EAAE,MAAM,oCAAoC,CAAC;AAGzE,OAAO,EACL,YAAY,EACZ,eAAe,EACf,aAAa,EACb,iBAAiB,EACjB,kBAAkB,EAClB,gBAAgB,EAChB,gBAAgB,EAChB,eAAe,EACf,iBAAiB,EACjB,eAAe,EACf,KAAK,aAAa,EAClB,KAAK,UAAU,GAChB,MAAM,uBAAuB,CAAC;AAG/B,YAAY,EAEV,WAAW,EACX,YAAY,EACZ,eAAe,EACf,eAAe,EAEf,cAAc,EACd,SAAS,EACT,kBAAkB,EAElB,oBAAoB,EAEpB,yBAAyB,EAEzB,sBAAsB,EACtB,UAAU,EAEV,gBAAgB,EAChB,UAAU,GACX,MAAM,YAAY,CAAC;AAGpB,OAAO,EAAE,aAAa,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAG9D,OAAO,EACL,eAAe,EACf,eAAe,EACf,WAAW,EACX,KAAK,QAAQ,EACb,KAAK,YAAY,GAClB,MAAM,aAAa,CAAC;AAGrB,OAAO,EACL,UAAU,EACV,YAAY,EACZ,YAAY,EACZ,cAAc,EACd,iBAAiB,EACjB,eAAe,EACf,gBAAgB,EAChB,WAAW,EACX,WAAW,EACX,kBAAkB,EAClB,gBAAgB,GACjB,MAAM,uCAAuC,CAAC"}

package/dist/index.js

@@ -0,0 +1,37 @@
+/**
+ * TokenMeter v5
+ *
+ * OpenTelemetry-native cost tracking for AI workflows.
+ *
+ * @example
+ * ```typescript
+ * import OpenAI from 'openai';
+ * import { monitor, withAttributes } from 'tokenmeter';
+ *
+ * // 1. Wrap your client
+ * const openai = monitor(new OpenAI({ apiKey: '...' }));
+ *
+ * // 2. Track with attributes
+ * await withAttributes({ 'org.id': 'org_123' }, async () => {
+ *   await openai.chat.completions.create({
+ *     model: 'gpt-4o',
+ *     messages: [{ role: 'user', content: 'Hello!' }]
+ *   });
+ * });
+ * ```
+ */
+// Core instrumentation
+export { monitor } from "./instrumentation/proxy.js";
+// Context management
+export { withAttributes, withAttributesSync, getCurrentAttributes, getAttribute, extractTraceHeaders, contextFromHeaders, withExtractedContext, } from "./context.js";
+// Processor
+export { TokenMeterProcessor } from "./processor/TokenMeterProcessor.js";
+// Pricing utilities
+export { loadManifest, getModelPricing, calculateCost, getCachedManifest, clearManifestCache, configurePricing, getPricingConfig, setModelAliases, clearModelAliases, getModelAliases, } from "./pricing/manifest.js";
+// Semantic conventions
+export { TM_ATTRIBUTES, GEN_AI_ATTRIBUTES } from "./types.js";
+// Logger configuration
+export { configureLogger, getLoggerConfig, resetLogger, } from "./logger.js";
+// Extraction strategies (for custom implementations)
+export { strategies, findStrategy, extractUsage, openaiStrategy, anthropicStrategy, bedrockStrategy, vertexAIStrategy, falStrategy, bflStrategy, elevenlabsStrategy, vercelAIStrategy, } from "./instrumentation/strategies/index.js";
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;GAqBG;AAEH,uBAAuB;AACvB,OAAO,EAAE,OAAO,EAAE,MAAM,4BAA4B,CAAC;AAErD,qBAAqB;AACrB,OAAO,EACL,cAAc,EACd,kBAAkB,EAClB,oBAAoB,EACpB,YAAY,EACZ,mBAAmB,EACnB,kBAAkB,EAClB,oBAAoB,GACrB,MAAM,cAAc,CAAC;AAEtB,YAAY;AACZ,OAAO,EAAE,mBAAmB,EAAE,MAAM,oCAAoC,CAAC;AAEzE,oBAAoB;AACpB,OAAO,EACL,YAAY,EACZ,eAAe,EACf,aAAa,EACb,iBAAiB,EACjB,kBAAkB,EAClB,gBAAgB,EAChB,gBAAgB,EAChB,eAAe,EACf,iBAAiB,EACjB,eAAe,GAGhB,MAAM,uBAAuB,CAAC;AAyB/B,uBAAuB;AACvB,OAAO,EAAE,aAAa,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAE9D,uBAAuB;AACvB,OAAO,EACL,eAAe,EACf,eAAe,EACf,WAAW,GAGZ,MAAM,aAAa,CAAC;AAErB,qDAAqD;AACrD,OAAO,EACL,UAAU,EACV,YAAY,EACZ,YAAY,EACZ,cAAc,EACd,iBAAiB,EACjB,eAAe,EACf,gBAAgB,EAChB,WAAW,EACX,WAAW,EACX,kBAAkB,EAClB,gBAAgB,GACjB,MAAM,uCAAuC,CAAC"}

package/dist/instrumentation/proxy.d.ts

@@ -0,0 +1,26 @@
+/**
+ * Proxy Engine
+ *
+ * Creates a recursive Proxy that intercepts method calls and creates OTel spans.
+ * Calculates costs and adds them to spans before export.
+ */
+import type { MonitorOptions } from "../types.js";
+/**
+ * Create a monitored proxy for a client instance
+ *
+ * @param client - The SDK client instance to wrap
+ * @param options - Configuration options
+ * @returns A proxied version of the client that creates spans for method calls
+ *
+ * @example
+ * ```typescript
+ * import OpenAI from 'openai';
+ * import { monitor } from 'tokenmeter';
+ *
+ * const openai = monitor(new OpenAI({ apiKey: '...' }));
+ * await openai.chat.completions.create({ model: 'gpt-4o', messages: [...] });
+ * ```
+ */
+export declare function monitor<T extends object>(client: T, options?: MonitorOptions): T;
+export default monitor;
+//# sourceMappingURL=proxy.d.ts.map

package/dist/instrumentation/proxy.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"proxy.d.ts","sourceRoot":"","sources":["../../src/instrumentation/proxy.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAmBH,OAAO,KAAK,EAAE,cAAc,EAA8B,MAAM,aAAa,CAAC;AAsQ9E;;;;;;;;;;;;;;;GAeG;AACH,wBAAgB,OAAO,CAAC,CAAC,SAAS,MAAM,EACtC,MAAM,EAAE,CAAC,EACT,OAAO,GAAE,cAAmB,GAC3B,CAAC,CA4IH;AAED,eAAe,OAAO,CAAC"}