tokenmeter 0.9.0

package/dist/__tests__/fal.test.js

@@ -0,0 +1,153 @@
+import { describe, it, expect, beforeEach, afterEach, vi } from "vitest";
+import { trackFal } from "../sdks/fal.js";
+import { init, shutdown, withCostTrace, flush } from "../index.js";
+import { InMemoryStorageAdapter } from "../storage/interface.js";
+describe("fal.ai SDK Wrapper", () => {
+    let storage;
+    beforeEach(() => {
+        storage = new InMemoryStorageAdapter();
+        init({ storage });
+    });
+    afterEach(async () => {
+        await flush();
+        await shutdown();
+    });
+    const createMockClient = () => ({
+        subscribe: vi.fn().mockImplementation(async (endpointId, options) => {
+            if (endpointId.includes("flux") || endpointId.includes("sdxl")) {
+                const numImages = options.input.num_images || 1;
+                return {
+                    data: { images: Array(numImages).fill({ url: "https://example.com/image.png" }) },
+                    requestId: "req_123",
+                };
+            }
+            else if (endpointId.includes("video")) {
+                return {
+                    data: { video: { url: "https://example.com/video.mp4" }, duration: options.input.duration || 5 },
+                    requestId: "req_456",
+                };
+            }
+            return { data: {}, requestId: "req_789" };
+        }),
+        run: vi.fn().mockResolvedValue({
+            data: { images: [{ url: "https://example.com/image.png" }] },
+            requestId: "req_run_123",
+        }),
+    });
+    describe("trackFal", () => {
+        it("should wrap subscribe and track image generation costs", async () => {
+            const mockClient = createMockClient();
+            const trackedClient = trackFal(mockClient);
+            await withCostTrace({ identifier: "user_123" }, async () => {
+                await trackedClient.subscribe("fal-ai/flux-pro", {
+                    input: { prompt: "A sunset", num_images: 1 },
+                });
+            });
+            await flush();
+            const events = storage.getEvents();
+            expect(events).toHaveLength(1);
+            expect(events[0]?.provider).toBe("fal");
+            expect(events[0]?.model).toBe("fal/flux-pro");
+            expect(events[0]?.operation).toBe("image");
+            expect(events[0]?.costUsd).toBeCloseTo(0.05, 4); // $0.05 per image
+        });
+        it("should calculate cost for multiple images", async () => {
+            const mockClient = createMockClient();
+            const trackedClient = trackFal(mockClient);
+            await withCostTrace({ identifier: "user_123" }, async () => {
+                await trackedClient.subscribe("fal-ai/flux-pro", {
+                    input: { prompt: "A sunset", num_images: 4 },
+                });
+            });
+            await flush();
+            const events = storage.getEvents();
+            expect(events[0]?.costUsd).toBeCloseTo(0.20, 4); // 4 * $0.05
+            expect((events[0]?.metadata).imageCount).toBe(4);
+        });
+        it("should use cheaper pricing for flux-schnell", async () => {
+            const mockClient = createMockClient();
+            const trackedClient = trackFal(mockClient);
+            await withCostTrace({ identifier: "user_123" }, async () => {
+                await trackedClient.subscribe("fal-ai/flux-schnell", {
+                    input: { prompt: "A sunset", num_images: 1 },
+                });
+            });
+            await flush();
+            const events = storage.getEvents();
+            expect(events[0]?.costUsd).toBeCloseTo(0.003, 4); // $0.003 per image
+        });
+        it("should track video generation costs per second", async () => {
+            const mockClient = createMockClient();
+            const trackedClient = trackFal(mockClient);
+            await withCostTrace({ identifier: "user_123" }, async () => {
+                await trackedClient.subscribe("fal-ai/minimax/video-01", {
+                    input: { prompt: "A product showcase", duration: 10 },
+                });
+            });
+            await flush();
+            const events = storage.getEvents();
+            expect(events[0]?.operation).toBe("video");
+            expect(events[0]?.costUsd).toBeCloseTo(3.0, 4); // 10 seconds * $0.30
+            expect((events[0]?.metadata).videoDuration).toBe(10);
+        });
+        it("should wrap run method", async () => {
+            const mockClient = createMockClient();
+            const trackedClient = trackFal(mockClient);
+            await withCostTrace({ identifier: "user_123" }, async () => {
+                await trackedClient.run("fal-ai/flux-pro", {
+                    input: { prompt: "A sunset" },
+                });
+            });
+            await flush();
+            const events = storage.getEvents();
+            expect(events).toHaveLength(1);
+            expect(events[0]?.provider).toBe("fal");
+        });
+        it("should track failed calls", async () => {
+            const mockClient = createMockClient();
+            mockClient.subscribe.mockRejectedValue(new Error("Rate limit exceeded"));
+            const trackedClient = trackFal(mockClient);
+            await withCostTrace({ identifier: "user_123" }, async () => {
+                try {
+                    await trackedClient.subscribe("fal-ai/flux-pro", {
+                        input: { prompt: "Test" },
+                    });
+                }
+                catch (e) {
+                    // Expected error
+                }
+            });
+            await flush();
+            const events = storage.getEvents();
+            expect(events).toHaveLength(1);
+            expect(events[0]?.status).toBe("failed");
+            expect(events[0]?.billable).toBe(false);
+            expect(events[0]?.errorMessage).toBe("Rate limit exceeded");
+        });
+        it("should normalize fal-ai/ prefix to fal/", async () => {
+            const mockClient = createMockClient();
+            const trackedClient = trackFal(mockClient);
+            await withCostTrace({ identifier: "user_123" }, async () => {
+                await trackedClient.subscribe("fal-ai/flux-pro/v1.1", {
+                    input: { prompt: "A sunset" },
+                });
+            });
+            await flush();
+            const events = storage.getEvents();
+            expect(events[0]?.model).toBe("fal/flux-pro");
+        });
+        it("should include requestId in metadata", async () => {
+            const mockClient = createMockClient();
+            const trackedClient = trackFal(mockClient);
+            await withCostTrace({ identifier: "user_123" }, async () => {
+                await trackedClient.subscribe("fal-ai/flux-pro", {
+                    input: { prompt: "A sunset" },
+                });
+            });
+            await flush();
+            const events = storage.getEvents();
+            expect((events[0]?.metadata).requestId).toBe("req_123");
+        });
+    });
+});
+//# sourceMappingURL=fal.test.js.map

package/dist/__tests__/fal.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"fal.test.js","sourceRoot":"","sources":["../../src/__tests__/fal.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,aAAa,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AACnE,OAAO,EAAE,sBAAsB,EAAE,MAAM,yBAAyB,CAAC;AAEjE,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,IAAI,OAA+B,CAAC;IAEpC,UAAU,CAAC,GAAG,EAAE;QACd,OAAO,GAAG,IAAI,sBAAsB,EAAE,CAAC;QACvC,IAAI,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;IACpB,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,KAAK,EAAE,CAAC;QACd,MAAM,QAAQ,EAAE,CAAC;IACnB,CAAC,CAAC,CAAC;IAEH,MAAM,gBAAgB,GAAG,GAAG,EAAE,CAAC,CAAC;QAC9B,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,KAAK,EAAE,UAAkB,EAAE,OAA2C,EAAE,EAAE;YAC9G,IAAI,UAAU,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,UAAU,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC/D,MAAM,SAAS,GAAI,OAAO,CAAC,KAAK,CAAC,UAAqB,IAAI,CAAC,CAAC;gBAC5D,OAAO;oBACL,IAAI,EAAE,EAAE,MAAM,EAAE,KAAK,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,+BAA+B,EAAE,CAAC,EAAE;oBACjF,SAAS,EAAE,SAAS;iBACrB,CAAC;YACJ,CAAC;iBAAM,IAAI,UAAU,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;gBACxC,OAAO;oBACL,IAAI,EAAE,EAAE,KAAK,EAAE,EAAE,GAAG,EAAE,+BAA+B,EAAE,EAAE,QAAQ,EAAG,OAAO,CAAC,KAAK,CAAC,QAAmB,IAAI,CAAC,EAAE;oBAC5G,SAAS,EAAE,SAAS;iBACrB,CAAC;YACJ,CAAC;YACD,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;QAC5C,CAAC,CAAC;QACF,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;YAC7B,IAAI,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE,GAAG,EAAE,+BAA+B,EAAE,CAAC,EAAE;YAC5D,SAAS,EAAE,aAAa;SACzB,CAAC;KACH,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACtE,MAAM,UAAU,GAAG,gBAAgB,EAAE,CAAC;YACtC,MAAM,aAAa,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC;YAE3C,MAAM,aAAa,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,EAAE,KAAK,IAAI,EAAE;gBACzD,MAAM,aAAa,CAAC,SAAU,CAAC,iBAAiB,EAAE;oBAChD,KAAK,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC,EAAE;iBAC7C,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,MAAM,KAAK,EAAE,CAAC;YAEd,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;YACnC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACxC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC3C,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,kBAAkB;QACrE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,UAAU,GAAG,gBAAgB,EAAE,CAAC;YACtC,MAAM,aAAa,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC;YAE3C,MAAM,aAAa,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,EAAE,KAAK,IAAI,EAAE;gBACzD,MAAM,aAAa,CAAC,SAAU,CAAC,iBAAiB,EAAE;oBAChD,KAAK,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC,EAAE;iBAC7C,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,MAAM,KAAK,EAAE,CAAC;YAEd,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;YACnC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,YAAY;YAC7D,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,QAAmC,CAAA,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC7E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,UAAU,GAAG,gBAAgB,EAAE,CAAC;YACtC,MAAM,aAAa,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC;YAE3C,MAAM,aAAa,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,EAAE,KAAK,IAAI,EAAE;gBACzD,MAAM,aAAa,CAAC,SAAU,CAAC,qBAAqB,EAAE;oBACpD,KAAK,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC,EAAE;iBAC7C,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,MAAM,KAAK,EAAE,CAAC;YAEd,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;YACnC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,mBAAmB;QACvE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,UAAU,GAAG,gBAAgB,EAAE,CAAC;YACtC,MAAM,aAAa,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC;YAE3C,MAAM,aAAa,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,EAAE,KAAK,IAAI,EAAE;gBACzD,MAAM,aAAa,CAAC,SAAU,CAAC,yBAAyB,EAAE;oBACxD,KAAK,EAAE,EAAE,MAAM,EAAE,oBAAoB,EAAE,QAAQ,EAAE,EAAE,EAAE;iBACtD,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,MAAM,KAAK,EAAE,CAAC;YAEd,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;YACnC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC3C,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,WAAW,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC,qBAAqB;YACrE,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,QAAsC,CAAA,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACpF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;YACtC,MAAM,UAAU,GAAG,gBAAgB,EAAE,CAAC;YACtC,MAAM,aAAa,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC;YAE3C,MAAM,aAAa,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,EAAE,KAAK,IAAI,EAAE;gBACzD,MAAM,aAAa,CAAC,GAAI,CAAC,iBAAiB,EAAE;oBAC1C,KAAK,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE;iBAC9B,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,MAAM,KAAK,EAAE,CAAC;YAEd,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;YACnC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;YACzC,MAAM,UAAU,GAAG,gBAAgB,EAAE,CAAC;YACtC,UAAU,CAAC,SAAS,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC,CAAC;YACzE,MAAM,aAAa,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC;YAE3C,MAAM,aAAa,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,EAAE,KAAK,IAAI,EAAE;gBACzD,IAAI,CAAC;oBACH,MAAM,aAAa,CAAC,SAAU,CAAC,iBAAiB,EAAE;wBAChD,KAAK,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE;qBAC1B,CAAC,CAAC;gBACL,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACX,iBAAiB;gBACnB,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,MAAM,KAAK,EAAE,CAAC;YAEd,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;YACnC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACxC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,YAAY,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,UAAU,GAAG,gBAAgB,EAAE,CAAC;YACtC,MAAM,aAAa,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC;YAE3C,MAAM,aAAa,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,EAAE,KAAK,IAAI,EAAE;gBACzD,MAAM,aAAa,CAAC,SAAU,CAAC,sBAAsB,EAAE;oBACrD,KAAK,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE;iBAC9B,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,MAAM,KAAK,EAAE,CAAC;YAEd,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;YACnC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,UAAU,GAAG,gBAAgB,EAAE,CAAC;YACtC,MAAM,aAAa,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC;YAE3C,MAAM,aAAa,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,EAAE,KAAK,IAAI,EAAE;gBACzD,MAAM,aAAa,CAAC,SAAU,CAAC,iBAAiB,EAAE;oBAChD,KAAK,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE;iBAC9B,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,MAAM,KAAK,EAAE,CAAC;YAEd,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;YACnC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,QAAkC,CAAA,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACnF,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/__tests__/pricing.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=pricing.test.d.ts.map

package/dist/__tests__/pricing.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pricing.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/pricing.test.ts"],"names":[],"mappings":""}

package/dist/__tests__/pricing.test.js

@@ -0,0 +1,76 @@
+import { describe, it, expect } from "vitest";
+import { calculateCost, defaultPricing, mergePricing } from "../pricing/index.js";
+describe("Pricing Module", () => {
+    describe("calculateCost", () => {
+        it("should calculate cost for OpenAI GPT-4o", () => {
+            const cost = calculateCost("gpt-4o", 1000, 500, defaultPricing);
+            // Input: 1000 tokens * $0.0025/1K = $0.0025
+            // Output: 500 tokens * $0.01/1K = $0.005
+            // Total: $0.0075
+            expect(cost).toBeCloseTo(0.0075, 6);
+        });
+        it("should calculate cost for OpenAI GPT-4o-mini", () => {
+            const cost = calculateCost("gpt-4o-mini", 10000, 5000, defaultPricing);
+            // Input: 10000 tokens * $0.00015/1K = $0.0015
+            // Output: 5000 tokens * $0.0006/1K = $0.003
+            // Total: $0.0045
+            expect(cost).toBeCloseTo(0.0045, 6);
+        });
+        it("should calculate cost with provider prefix", () => {
+            const cost = calculateCost("openai/gpt-4o", 1000, 500, defaultPricing);
+            expect(cost).toBeCloseTo(0.0075, 6);
+        });
+        it("should return 0 for unknown model and warn", () => {
+            const consoleSpy = vi.spyOn(console, "warn").mockImplementation(() => { });
+            const cost = calculateCost("unknown-model", 1000, 500, defaultPricing);
+            expect(cost).toBe(0);
+            expect(consoleSpy).toHaveBeenCalledWith("[tokenmeter] No pricing found for model: unknown-model");
+            consoleSpy.mockRestore();
+        });
+        it("should handle zero tokens", () => {
+            const cost = calculateCost("gpt-4o", 0, 0, defaultPricing);
+            expect(cost).toBe(0);
+        });
+        it("should handle cached input tokens", () => {
+            const cost = calculateCost("gpt-4o", 1000, 500, defaultPricing, 500);
+            // Regular input: 1000 * $0.0025/1K = $0.0025
+            // Cached input: 500 * $0.00125/1K = $0.000625
+            // Output: 500 * $0.01/1K = $0.005
+            // Total: $0.008125
+            expect(cost).toBeCloseTo(0.008125, 6);
+        });
+    });
+    describe("mergePricing", () => {
+        it("should merge custom pricing with defaults", () => {
+            const custom = {
+                "custom-model": { input: 0.01, output: 0.02 },
+            };
+            const merged = mergePricing(defaultPricing, custom);
+            expect(merged["custom-model"]).toEqual({ input: 0.01, output: 0.02 });
+            expect(merged["gpt-4o"]).toEqual(defaultPricing["gpt-4o"]);
+        });
+        it("should override default pricing with custom", () => {
+            const custom = {
+                "gpt-4o": { input: 0.005, output: 0.02 },
+            };
+            const merged = mergePricing(defaultPricing, custom);
+            expect(merged["gpt-4o"]).toEqual({ input: 0.005, output: 0.02 });
+        });
+    });
+    describe("defaultPricing", () => {
+        it("should have ElevenLabs models", () => {
+            expect(defaultPricing["elevenlabs/eleven_multilingual_v2"]).toBeDefined();
+            expect(defaultPricing["eleven_turbo_v2_5"]).toBeDefined();
+        });
+        it("should have fal.ai models", () => {
+            expect(defaultPricing["fal/flux-pro"]).toBeDefined();
+            expect(defaultPricing["fal/flux-schnell"]).toBeDefined();
+            expect(defaultPricing["fal/minimax/video-01"]).toBeDefined();
+        });
+        it("should have Anthropic models", () => {
+            expect(defaultPricing["anthropic/claude-3-5-sonnet-20241022"]).toBeDefined();
+            expect(defaultPricing["claude-3-5-sonnet-20241022"]).toBeDefined();
+        });
+    });
+});
+//# sourceMappingURL=pricing.test.js.map

package/dist/__tests__/pricing.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pricing.test.js","sourceRoot":"","sources":["../../src/__tests__/pricing.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,aAAa,EAAE,cAAc,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAElF,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,IAAI,GAAG,aAAa,CAAC,QAAQ,EAAE,IAAI,EAAE,GAAG,EAAE,cAAc,CAAC,CAAC;YAChE,4CAA4C;YAC5C,yCAAyC;YACzC,iBAAiB;YACjB,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,IAAI,GAAG,aAAa,CAAC,aAAa,EAAE,KAAK,EAAE,IAAI,EAAE,cAAc,CAAC,CAAC;YACvE,8CAA8C;YAC9C,4CAA4C;YAC5C,iBAAiB;YACjB,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,MAAM,IAAI,GAAG,aAAa,CAAC,eAAe,EAAE,IAAI,EAAE,GAAG,EAAE,cAAc,CAAC,CAAC;YACvE,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,MAAM,UAAU,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAC1E,MAAM,IAAI,GAAG,aAAa,CAAC,eAAe,EAAE,IAAI,EAAE,GAAG,EAAE,cAAc,CAAC,CAAC;YACvE,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACrB,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CACrC,wDAAwD,CACzD,CAAC;YACF,UAAU,CAAC,WAAW,EAAE,CAAC;QAC3B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;YACnC,MAAM,IAAI,GAAG,aAAa,CAAC,QAAQ,EAAE,CAAC,EAAE,CAAC,EAAE,cAAc,CAAC,CAAC;YAC3D,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACvB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;YAC3C,MAAM,IAAI,GAAG,aAAa,CAAC,QAAQ,EAAE,IAAI,EAAE,GAAG,EAAE,cAAc,EAAE,GAAG,CAAC,CAAC;YACrE,6CAA6C;YAC7C,8CAA8C;YAC9C,kCAAkC;YAClC,mBAAmB;YACnB,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,MAAM,GAAG;gBACb,cAAc,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE;aAC9C,CAAC;YACF,MAAM,MAAM,GAAG,YAAY,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;YAEpD,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;YACtE,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,MAAM,MAAM,GAAG;gBACb,QAAQ,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE;aACzC,CAAC;YACF,MAAM,MAAM,GAAG,YAAY,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;YAEpD,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;YACvC,MAAM,CAAC,cAAc,CAAC,mCAAmC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;YAC1E,MAAM,CAAC,cAAc,CAAC,mBAAmB,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;YACnC,MAAM,CAAC,cAAc,CAAC,cAAc,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;YACrD,MAAM,CAAC,cAAc,CAAC,kBAAkB,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;YACzD,MAAM,CAAC,cAAc,CAAC,sBAAsB,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,MAAM,CAAC,cAAc,CAAC,sCAAsC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;YAC7E,MAAM,CAAC,cAAc,CAAC,4BAA4B,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;QACrE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/__tests__/recorder.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=recorder.test.d.ts.map

package/dist/__tests__/recorder.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"recorder.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/recorder.test.ts"],"names":[],"mappings":""}

package/dist/__tests__/recorder.test.js

@@ -0,0 +1,133 @@
+import { describe, it, expect, beforeEach, afterEach, vi } from "vitest";
+import { init, shutdown, withCostTrace, trackCost, flush } from "../index.js";
+import { InMemoryStorageAdapter } from "../storage/interface.js";
+describe("Recorder Module", () => {
+    let storage;
+    beforeEach(() => {
+        storage = new InMemoryStorageAdapter();
+        init({
+            storage,
+            batching: {
+                enabled: true,
+                flushInterval: 10000, // Long interval to prevent auto-flush
+                maxBatchSize: 100,
+                flushOnExit: false,
+            },
+        });
+    });
+    afterEach(async () => {
+        await flush();
+        await shutdown();
+    });
+    describe("trackCost", () => {
+        it("should record manual cost events", async () => {
+            await withCostTrace({ identifier: "user_123" }, async () => {
+                trackCost({
+                    provider: "custom-provider",
+                    model: "custom-model",
+                    operation: "inference",
+                    costUsd: 0.05,
+                });
+            });
+            await flush();
+            const events = storage.getEvents();
+            expect(events).toHaveLength(1);
+            expect(events[0]?.provider).toBe("custom-provider");
+            expect(events[0]?.model).toBe("custom-model");
+            expect(events[0]?.costUsd).toBe(0.05);
+        });
+        it("should inherit context from withCostTrace", async () => {
+            await withCostTrace({
+                identifier: "org_456",
+                workflowId: "wf_custom",
+                workflowType: "custom-workflow",
+                metadata: { source: "api" },
+            }, async () => {
+                trackCost({
+                    provider: "runway",
+                    model: "gen3",
+                    costUsd: 0.50,
+                });
+            });
+            await flush();
+            const events = storage.getEvents();
+            expect(events[0]?.identifier).toBe("org_456");
+            expect(events[0]?.workflowId).toBe("wf_custom");
+            expect(events[0]?.workflowType).toBe("custom-workflow");
+            expect((events[0]?.metadata).source).toBe("api");
+        });
+        it("should track failed costs with error message", async () => {
+            await withCostTrace({ identifier: "user_123" }, async () => {
+                trackCost({
+                    provider: "veo",
+                    model: "veo-2",
+                    operation: "video",
+                    costUsd: 0,
+                    status: "failed",
+                    billable: false,
+                    errorMessage: "Insufficient credits",
+                });
+            });
+            await flush();
+            const events = storage.getEvents();
+            expect(events[0]?.status).toBe("failed");
+            expect(events[0]?.billable).toBe(false);
+            expect(events[0]?.errorMessage).toBe("Insufficient credits");
+        });
+        it("should warn when called outside of context", async () => {
+            const consoleSpy = vi.spyOn(console, "warn").mockImplementation(() => { });
+            trackCost({
+                provider: "test",
+                model: "test",
+                costUsd: 0.01,
+            });
+            expect(consoleSpy).toHaveBeenCalledWith("[tokenmeter] No trace context. Wrap your code in withCostTrace().");
+            consoleSpy.mockRestore();
+        });
+    });
+    describe("flush", () => {
+        it("should write batched events to storage", async () => {
+            await withCostTrace({ identifier: "user_123" }, async () => {
+                trackCost({ provider: "a", model: "a", costUsd: 0.01 });
+                trackCost({ provider: "b", model: "b", costUsd: 0.02 });
+                trackCost({ provider: "c", model: "c", costUsd: 0.03 });
+            });
+            // Events should be batched, not yet in storage
+            expect(storage.getEvents()).toHaveLength(0);
+            await flush();
+            expect(storage.getEvents()).toHaveLength(3);
+        });
+        it("should handle empty queue gracefully", async () => {
+            await flush();
+            expect(storage.getEvents()).toHaveLength(0);
+        });
+    });
+    describe("batching", () => {
+        it("should flush when batch size is reached", async () => {
+            // Reinit with small batch size
+            await shutdown();
+            storage = new InMemoryStorageAdapter();
+            init({
+                storage,
+                batching: {
+                    enabled: true,
+                    flushInterval: 10000,
+                    maxBatchSize: 3,
+                    flushOnExit: false,
+                },
+            });
+            await withCostTrace({ identifier: "user_123" }, async () => {
+                trackCost({ provider: "a", model: "a", costUsd: 0.01 });
+                trackCost({ provider: "b", model: "b", costUsd: 0.02 });
+                // Not yet flushed
+                expect(storage.getEvents()).toHaveLength(0);
+                trackCost({ provider: "c", model: "c", costUsd: 0.03 });
+                // Give time for async flush
+                await new Promise((r) => setTimeout(r, 10));
+            });
+            // Should have auto-flushed at batch size 3
+            expect(storage.getEvents()).toHaveLength(3);
+        });
+    });
+});
+//# sourceMappingURL=recorder.test.js.map

package/dist/__tests__/recorder.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"recorder.test.js","sourceRoot":"","sources":["../../src/__tests__/recorder.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,aAAa,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AAC9E,OAAO,EAAE,sBAAsB,EAAE,MAAM,yBAAyB,CAAC;AAEjE,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,IAAI,OAA+B,CAAC;IAEpC,UAAU,CAAC,GAAG,EAAE;QACd,OAAO,GAAG,IAAI,sBAAsB,EAAE,CAAC;QACvC,IAAI,CAAC;YACH,OAAO;YACP,QAAQ,EAAE;gBACR,OAAO,EAAE,IAAI;gBACb,aAAa,EAAE,KAAK,EAAE,sCAAsC;gBAC5D,YAAY,EAAE,GAAG;gBACjB,WAAW,EAAE,KAAK;aACnB;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,KAAK,EAAE,CAAC;QACd,MAAM,QAAQ,EAAE,CAAC;IACnB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;YAChD,MAAM,aAAa,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,EAAE,KAAK,IAAI,EAAE;gBACzD,SAAS,CAAC;oBACR,QAAQ,EAAE,iBAAiB;oBAC3B,KAAK,EAAE,cAAc;oBACrB,SAAS,EAAE,WAAW;oBACtB,OAAO,EAAE,IAAI;iBACd,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,MAAM,KAAK,EAAE,CAAC;YAEd,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;YACnC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YACpD,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,aAAa,CACjB;gBACE,UAAU,EAAE,SAAS;gBACrB,UAAU,EAAE,WAAW;gBACvB,YAAY,EAAE,iBAAiB;gBAC/B,QAAQ,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE;aAC5B,EACD,KAAK,IAAI,EAAE;gBACT,SAAS,CAAC;oBACR,QAAQ,EAAE,QAAQ;oBAClB,KAAK,EAAE,MAAM;oBACb,OAAO,EAAE,IAAI;iBACd,CAAC,CAAC;YACL,CAAC,CACF,CAAC;YAEF,MAAM,KAAK,EAAE,CAAC;YAEd,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;YACnC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,YAAY,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YACxD,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,QAA+B,CAAA,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACzE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,aAAa,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,EAAE,KAAK,IAAI,EAAE;gBACzD,SAAS,CAAC;oBACR,QAAQ,EAAE,KAAK;oBACf,KAAK,EAAE,OAAO;oBACd,SAAS,EAAE,OAAO;oBAClB,OAAO,EAAE,CAAC;oBACV,MAAM,EAAE,QAAQ;oBAChB,QAAQ,EAAE,KAAK;oBACf,YAAY,EAAE,sBAAsB;iBACrC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,MAAM,KAAK,EAAE,CAAC;YAEd,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;YACnC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACxC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,YAAY,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,UAAU,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAE1E,SAAS,CAAC;gBACR,QAAQ,EAAE,MAAM;gBAChB,KAAK,EAAE,MAAM;gBACb,OAAO,EAAE,IAAI;aACd,CAAC,CAAC;YAEH,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CACrC,mEAAmE,CACpE,CAAC;YAEF,UAAU,CAAC,WAAW,EAAE,CAAC;QAC3B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE;QACrB,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,aAAa,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,EAAE,KAAK,IAAI,EAAE;gBACzD,SAAS,CAAC,EAAE,QAAQ,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;gBACxD,SAAS,CAAC,EAAE,QAAQ,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;gBACxD,SAAS,CAAC,EAAE,QAAQ,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;YAC1D,CAAC,CAAC,CAAC;YAEH,+CAA+C;YAC/C,MAAM,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAE5C,MAAM,KAAK,EAAE,CAAC;YAEd,MAAM,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,KAAK,EAAE,CAAC;YACd,MAAM,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,+BAA+B;YAC/B,MAAM,QAAQ,EAAE,CAAC;YACjB,OAAO,GAAG,IAAI,sBAAsB,EAAE,CAAC;YACvC,IAAI,CAAC;gBACH,OAAO;gBACP,QAAQ,EAAE;oBACR,OAAO,EAAE,IAAI;oBACb,aAAa,EAAE,KAAK;oBACpB,YAAY,EAAE,CAAC;oBACf,WAAW,EAAE,KAAK;iBACnB;aACF,CAAC,CAAC;YAEH,MAAM,aAAa,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,EAAE,KAAK,IAAI,EAAE;gBACzD,SAAS,CAAC,EAAE,QAAQ,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;gBACxD,SAAS,CAAC,EAAE,QAAQ,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;gBAExD,kBAAkB;gBAClB,MAAM,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;gBAE5C,SAAS,CAAC,EAAE,QAAQ,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;gBAExD,4BAA4B;gBAC5B,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;YAC9C,CAAC,CAAC,CAAC;YAEH,2CAA2C;YAC3C,MAAM,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/__tests__/storage.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=storage.test.d.ts.map

package/dist/__tests__/storage.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"storage.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/storage.test.ts"],"names":[],"mappings":""}

package/dist/__tests__/storage.test.js

@@ -0,0 +1,106 @@
+import { describe, it, expect, beforeEach } from "vitest";
+import { InMemoryStorageAdapter } from "../storage/interface.js";
+describe("InMemoryStorageAdapter", () => {
+    let storage;
+    beforeEach(() => {
+        storage = new InMemoryStorageAdapter();
+    });
+    const createEvent = (overrides = {}) => ({
+        identifier: "user_123",
+        provider: "openai",
+        model: "gpt-4o",
+        costUsd: 0.01,
+        createdAt: new Date(),
+        ...overrides,
+    });
+    describe("writeEvent", () => {
+        it("should store a single event", async () => {
+            const event = createEvent();
+            await storage.writeEvent(event);
+            const events = storage.getEvents();
+            expect(events).toHaveLength(1);
+            expect(events[0]).toEqual(event);
+        });
+    });
+    describe("writeBatch", () => {
+        it("should store multiple events", async () => {
+            const events = [
+                createEvent({ costUsd: 0.01 }),
+                createEvent({ costUsd: 0.02 }),
+                createEvent({ costUsd: 0.03 }),
+            ];
+            await storage.writeBatch(events);
+            expect(storage.getEvents()).toHaveLength(3);
+        });
+    });
+    describe("query", () => {
+        beforeEach(async () => {
+            await storage.writeBatch([
+                createEvent({
+                    identifier: "user_1",
+                    provider: "openai",
+                    workflowId: "wf_1",
+                    costUsd: 0.01,
+                    createdAt: new Date("2024-01-15"),
+                }),
+                createEvent({
+                    identifier: "user_1",
+                    provider: "anthropic",
+                    workflowId: "wf_1",
+                    costUsd: 0.02,
+                    createdAt: new Date("2024-01-16"),
+                }),
+                createEvent({
+                    identifier: "user_2",
+                    provider: "openai",
+                    workflowId: "wf_2",
+                    costUsd: 0.03,
+                    createdAt: new Date("2024-01-17"),
+                }),
+            ]);
+        });
+        it("should query by identifier", async () => {
+            const result = await storage.query({ identifier: "user_1" });
+            expect(result.events).toHaveLength(2);
+            expect(result.totalCost).toBeCloseTo(0.03, 6);
+        });
+        it("should query by workflowId", async () => {
+            const result = await storage.query({ workflowId: "wf_1" });
+            expect(result.events).toHaveLength(2);
+            expect(result.totalCost).toBeCloseTo(0.03, 6);
+        });
+        it("should query by provider", async () => {
+            const result = await storage.query({ provider: "openai" });
+            expect(result.events).toHaveLength(2);
+            expect(result.totalCost).toBeCloseTo(0.04, 6);
+        });
+        it("should query by date range", async () => {
+            const result = await storage.query({
+                from: new Date("2024-01-15"),
+                to: new Date("2024-01-16"),
+            });
+            expect(result.events).toHaveLength(2);
+        });
+        it("should group costs by provider", async () => {
+            const result = await storage.query({ identifier: "user_1" });
+            expect(result.byProvider).toEqual({
+                openai: 0.01,
+                anthropic: 0.02,
+            });
+        });
+        it("should return all events when no filters", async () => {
+            const result = await storage.query({});
+            expect(result.events).toHaveLength(3);
+            expect(result.totalCost).toBeCloseTo(0.06, 6);
+        });
+    });
+    describe("clear", () => {
+        it("should clear all events", async () => {
+            await storage.writeEvent(createEvent());
+            expect(storage.getEvents()).toHaveLength(1);
+            storage.clear();
+            expect(storage.getEvents()).toHaveLength(0);
+        });
+    });
+});
+//# sourceMappingURL=storage.test.js.map

package/dist/__tests__/storage.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"storage.test.js","sourceRoot":"","sources":["../../src/__tests__/storage.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC1D,OAAO,EAAE,sBAAsB,EAAE,MAAM,yBAAyB,CAAC;AAGjE,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;IACtC,IAAI,OAA+B,CAAC;IAEpC,UAAU,CAAC,GAAG,EAAE;QACd,OAAO,GAAG,IAAI,sBAAsB,EAAE,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,MAAM,WAAW,GAAG,CAAC,YAAgC,EAAE,EAAa,EAAE,CAAC,CAAC;QACtE,UAAU,EAAE,UAAU;QACtB,QAAQ,EAAE,QAAQ;QAClB,KAAK,EAAE,QAAQ;QACf,OAAO,EAAE,IAAI;QACb,SAAS,EAAE,IAAI,IAAI,EAAE;QACrB,GAAG,SAAS;KACb,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;YAC3C,MAAM,KAAK,GAAG,WAAW,EAAE,CAAC;YAC5B,MAAM,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YAEhC,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;YACnC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;YAC5C,MAAM,MAAM,GAAG;gBACb,WAAW,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;gBAC9B,WAAW,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;gBAC9B,WAAW,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;aAC/B,CAAC;YAEF,MAAM,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;YAEjC,MAAM,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE;QACrB,UAAU,CAAC,KAAK,IAAI,EAAE;YACpB,MAAM,OAAO,CAAC,UAAU,CAAC;gBACvB,WAAW,CAAC;oBACV,UAAU,EAAE,QAAQ;oBACpB,QAAQ,EAAE,QAAQ;oBAClB,UAAU,EAAE,MAAM;oBAClB,OAAO,EAAE,IAAI;oBACb,SAAS,EAAE,IAAI,IAAI,CAAC,YAAY,CAAC;iBAClC,CAAC;gBACF,WAAW,CAAC;oBACV,UAAU,EAAE,QAAQ;oBACpB,QAAQ,EAAE,WAAW;oBACrB,UAAU,EAAE,MAAM;oBAClB,OAAO,EAAE,IAAI;oBACb,SAAS,EAAE,IAAI,IAAI,CAAC,YAAY,CAAC;iBAClC,CAAC;gBACF,WAAW,CAAC;oBACV,UAAU,EAAE,QAAQ;oBACpB,QAAQ,EAAE,QAAQ;oBAClB,UAAU,EAAE,MAAM;oBAClB,OAAO,EAAE,IAAI;oBACb,SAAS,EAAE,IAAI,IAAI,CAAC,YAAY,CAAC;iBAClC,CAAC;aACH,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;YAC1C,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC,CAAC;YAE7D,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACtC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;YAC1C,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC,CAAC;YAE3D,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACtC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;YACxC,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC;YAE3D,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACtC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;YAC1C,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC;gBACjC,IAAI,EAAE,IAAI,IAAI,CAAC,YAAY,CAAC;gBAC5B,EAAE,EAAE,IAAI,IAAI,CAAC,YAAY,CAAC;aAC3B,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC,CAAC;YAE7D,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC;gBAChC,MAAM,EAAE,IAAI;gBACZ,SAAS,EAAE,IAAI;aAChB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YAEvC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACtC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE;QACrB,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;YACvC,MAAM,OAAO,CAAC,UAAU,CAAC,WAAW,EAAE,CAAC,CAAC;YACxC,MAAM,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAE5C,OAAO,CAAC,KAAK,EAAE,CAAC;YAChB,MAAM,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/client/index.d.ts

@@ -0,0 +1,8 @@
+/**
+ * Client module
+ *
+ * Re-exports the query client for the ./client export path.
+ */
+export { createQueryClient } from "../query/client.js";
+export type { QueryClient, QueryClientConfig } from "../query/client.js";
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/client/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AACvD,YAAY,EAAE,WAAW,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC"}

package/dist/client/index.js

@@ -0,0 +1,7 @@
+/**
+ * Client module
+ *
+ * Re-exports the query client for the ./client export path.
+ */
+export { createQueryClient } from "../query/client.js";
+//# sourceMappingURL=index.js.map

package/dist/client/index.js.map

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

package/dist/config.d.ts

@@ -0,0 +1,92 @@
+import type { TokenmeterConfig, StorageAdapter, PricingConfig, BatchingConfig } from "./types.js";
+/**
+ * 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 declare function init(config: TokenmeterConfig): void;
+/**
+ * Get the current storage adapter
+ */
+export declare function getStorage(): StorageAdapter;
+/**
+ * Get pricing configuration
+ */
+export declare function getPricing(): PricingConfig;
+/**
+ * Get model aliases
+ */
+export declare function getModelAliases(): Record<string, string>;
+/**
+ * Get batching configuration
+ */
+export declare function getBatching(): BatchingConfig;
+/**
+ * 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 declare function isInitialized(): boolean;
+/**
+ * Resolve a model name through aliases
+ */
+export declare function resolveModel(model: string): string;
+/**
+ * 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 declare function shutdown(): Promise<void>;
+//# sourceMappingURL=config.d.ts.map