tokenmeter 0.9.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 Lukas Minnebeck
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,346 @@
+# tokenmeter
+
+[![npm version](https://img.shields.io/npm/v/tokenmeter.svg)](https://www.npmjs.com/package/tokenmeter)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+
+OpenTelemetry-native cost tracking for AI workflows. Track real USD costs per user, workflow, and provider with zero code changes.
+
+## Why tokenmeter?
+
+AI costs are hard to track. Tokens flow through multiple providers, streaming responses don't report usage upfront, and attributing costs to users or workflows requires custom instrumentation everywhere.
+
+tokenmeter solves this by:
+
+- **Wrapping AI clients transparently** - `monitor(client)` returns the same type, no code changes needed
+- **Calculating costs automatically** - Uses up-to-date pricing for OpenAI, Anthropic, Google, fal.ai, ElevenLabs
+- **Propagating context** - `withAttributes()` attaches user/org/workflow IDs to all nested AI calls
+- **Integrating with OTel** - Export to Datadog, Jaeger, Honeycomb, or persist to PostgreSQL
+
+## Installation
+
+```bash
+npm install tokenmeter @opentelemetry/api @opentelemetry/sdk-trace-node
+```
+
+## Quick Start
+
+```typescript
+import OpenAI from 'openai';
+import { NodeTracerProvider } from '@opentelemetry/sdk-trace-node';
+import { SimpleSpanProcessor, ConsoleSpanExporter } from '@opentelemetry/sdk-trace-base';
+import { monitor, withAttributes, TokenMeterProcessor } from 'tokenmeter';
+
+// 1. Set up OpenTelemetry with TokenMeter processor
+const provider = new NodeTracerProvider();
+provider.addSpanProcessor(new TokenMeterProcessor());
+provider.addSpanProcessor(new SimpleSpanProcessor(new ConsoleSpanExporter()));
+provider.register();
+
+// 2. Wrap your AI client
+const openai = monitor(new OpenAI());
+
+// 3. Track costs with context
+await withAttributes({ 'user.id': 'user_123' }, async () => {
+  const response = await openai.chat.completions.create({
+    model: 'gpt-4o',
+    messages: [{ role: 'user', content: 'Hello!' }],
+  });
+  console.log(response.choices[0].message.content);
+});
+
+// Spans now include: tokenmeter.cost_usd, tokenmeter.provider, tokenmeter.model, user.id
+```
+
+## Supported Providers
+
+| Provider | Models | Pricing Unit |
+|----------|--------|--------------|
+| **OpenAI** | GPT-4o, GPT-4-turbo, o1, o3, GPT-3.5, embeddings, DALL-E, Whisper | per 1M tokens |
+| **Anthropic** | Claude 4, Claude 3.5, Claude 3 | per 1M tokens |
+| **Google** | Gemini 2.0, Gemini 1.5 | per 1M tokens |
+| **fal.ai** | 900+ models (Flux, SDXL, Kling, Runway, etc.) | per request/megapixel/second |
+| **ElevenLabs** | All TTS models | per 1K characters |
+
+## Core Concepts
+
+### `monitor(client)`
+
+Wraps any supported AI client with a Proxy that intercepts API calls, extracts usage data, and creates OpenTelemetry spans with cost attributes.
+
+```typescript
+import OpenAI from 'openai';
+import Anthropic from '@anthropic-ai/sdk';
+import { fal } from '@fal-ai/client';
+import { monitor } from 'tokenmeter';
+
+const openai = monitor(new OpenAI());
+const anthropic = monitor(new Anthropic());
+const trackedFal = monitor(fal);
+
+// Types are fully preserved - no changes to your code
+const response = await openai.chat.completions.create({...});
+```
+
+### `withAttributes(attrs, fn)`
+
+Sets context attributes inherited by all AI calls within the callback. Uses OpenTelemetry Baggage for propagation.
+
+```typescript
+import { withAttributes } from 'tokenmeter';
+
+await withAttributes({ 'user.id': 'user_123', 'org.id': 'acme' }, async () => {
+  // All AI calls here are tagged with user.id and org.id
+  await openai.chat.completions.create({...});
+  await anthropic.messages.create({...});
+});
+
+// Nesting merges attributes
+await withAttributes({ 'org.id': 'acme' }, async () => {
+  await withAttributes({ 'user.id': 'user_123' }, async () => {
+    // Has both org.id and user.id
+  });
+});
+```
+
+### `TokenMeterProcessor`
+
+An OpenTelemetry SpanProcessor that calculates costs from span attributes and adds `tokenmeter.cost_usd`.
+
+```typescript
+import { NodeTracerProvider } from '@opentelemetry/sdk-trace-node';
+import { TokenMeterProcessor } from 'tokenmeter';
+
+const provider = new NodeTracerProvider();
+provider.addSpanProcessor(new TokenMeterProcessor());
+provider.register();
+```
+
+## Span Attributes
+
+tokenmeter adds these attributes to spans:
+
+| Attribute | Type | Description |
+|-----------|------|-------------|
+| `tokenmeter.cost_usd` | number | Calculated cost in USD |
+| `tokenmeter.provider` | string | Provider name |
+| `tokenmeter.model` | string | Model identifier |
+| `gen_ai.usage.input_tokens` | number | Input token count |
+| `gen_ai.usage.output_tokens` | number | Output token count |
+
+Plus any attributes set via `withAttributes()` (e.g., `user.id`, `org.id`, `workflow.id`).
+
+## Framework Integrations
+
+### Next.js App Router
+
+```typescript
+import { withTokenmeter } from 'tokenmeter/next';
+
+async function handler(request: Request) {
+  const response = await openai.chat.completions.create({...});
+  return Response.json({ message: response.choices[0].message.content });
+}
+
+export const POST = withTokenmeter(handler, (request) => ({
+  userId: request.headers.get('x-user-id') || undefined,
+}));
+```
+
+### Inngest
+
+```typescript
+import { withInngest, getInngestTraceHeaders } from 'tokenmeter/inngest';
+
+// Send events with trace context
+await inngest.send({
+  name: 'ai/generate',
+  data: { prompt: '...' },
+  ...getInngestTraceHeaders(),
+});
+
+// Restore context in function
+export const generateFn = inngest.createFunction(
+  { id: 'generate' },
+  { event: 'ai/generate' },
+  withInngest(async ({ event }) => {
+    await openai.chat.completions.create({...}); // Linked to original trace
+  })
+);
+```
+
+## PostgreSQL Persistence
+
+Store costs for querying and billing.
+
+```typescript
+import { BatchSpanProcessor } from '@opentelemetry/sdk-trace-base';
+import { PostgresExporter } from 'tokenmeter/exporter';
+import { createQueryClient } from 'tokenmeter/client';
+
+// Export spans to PostgreSQL
+provider.addSpanProcessor(new BatchSpanProcessor(
+  new PostgresExporter({ connectionString: process.env.DATABASE_URL })
+));
+
+// Query costs
+const client = createQueryClient({ connectionString: process.env.DATABASE_URL });
+
+const { totalCost } = await client.getCostByUser('user_123', {
+  from: new Date('2024-01-01'),
+  to: new Date('2024-01-31'),
+});
+
+const byModel = await client.getCosts({ groupBy: ['model'] });
+```
+
+See [DATABASE_SETUP.md](./DATABASE_SETUP.md) for schema and setup instructions.
+
+## Streaming Support
+
+tokenmeter handles streaming responses automatically.
+
+```typescript
+// OpenAI - requires stream_options for usage
+const stream = await openai.chat.completions.create({
+  model: 'gpt-4o',
+  messages: [{...}],
+  stream: true,
+  stream_options: { include_usage: true },
+});
+
+for await (const chunk of stream) {
+  process.stdout.write(chunk.choices[0]?.delta?.content || '');
+}
+// Cost calculated when stream completes
+
+// Anthropic streaming works out of the box
+const stream = anthropic.messages.stream({...});
+for await (const event of stream) {...}
+const finalMessage = await stream.finalMessage();
+```
+
+## Cross-Service Propagation
+
+For distributed systems, propagate trace context across service boundaries.
+
+```typescript
+import { extractTraceHeaders, withExtractedContext } from 'tokenmeter';
+
+// Service A: Extract headers
+const headers = extractTraceHeaders();
+await fetch('https://service-b.example.com', { headers });
+
+// Service B: Restore context
+await withExtractedContext(req.headers, async () => {
+  await openai.chat.completions.create({...}); // Part of Service A's trace
+});
+```
+
+## Pricing Configuration
+
+tokenmeter fetches pricing from a remote manifest with local fallback.
+
+```typescript
+import { configurePricing, loadManifest } from 'tokenmeter';
+
+// Use offline mode (bundled pricing only)
+configurePricing({ offlineMode: true });
+
+// Custom pricing API
+configurePricing({ apiUrl: 'https://your-api.com/pricing' });
+
+// Force refresh
+await loadManifest({ forceRefresh: true });
+```
+
+## Export Destinations
+
+### Datadog
+
+```typescript
+import { OTLPTraceExporter } from '@opentelemetry/exporter-trace-otlp-http';
+
+provider.addSpanProcessor(new BatchSpanProcessor(
+  new OTLPTraceExporter({
+    url: 'https://trace.agent.datadoghq.com/v0.4/traces',
+    headers: { 'DD-API-KEY': process.env.DD_API_KEY },
+  })
+));
+```
+
+### Honeycomb
+
+```typescript
+provider.addSpanProcessor(new BatchSpanProcessor(
+  new OTLPTraceExporter({
+    url: 'https://api.honeycomb.io/v1/traces',
+    headers: { 'x-honeycomb-team': process.env.HONEYCOMB_API_KEY },
+  })
+));
+```
+
+### Jaeger
+
+```typescript
+provider.addSpanProcessor(new BatchSpanProcessor(
+  new OTLPTraceExporter({ url: 'http://localhost:4318/v1/traces' })
+));
+```
+
+## API Reference
+
+### Core
+
+| Export | Description |
+|--------|-------------|
+| `monitor(client, options?)` | Wrap AI client with cost tracking |
+| `withAttributes(attrs, fn)` | Set context attributes for nested calls |
+| `extractTraceHeaders()` | Get W3C trace headers for propagation |
+| `withExtractedContext(headers, fn)` | Restore context from headers |
+
+### Processor & Exporter
+
+| Export | Description |
+|--------|-------------|
+| `TokenMeterProcessor` | OTel SpanProcessor for cost calculation |
+| `PostgresExporter` | OTel SpanExporter for PostgreSQL |
+
+### Query Client (`tokenmeter/client`)
+
+| Method | Description |
+|--------|-------------|
+| `createQueryClient(config)` | Create query client |
+| `client.getCosts(options)` | Query with filters and grouping |
+| `client.getCostByUser(userId, options?)` | Get user costs |
+| `client.getCostByOrg(orgId, options?)` | Get organization costs |
+| `client.getWorkflowCost(workflowId)` | Get workflow costs |
+
+### Integrations
+
+| Export | Description |
+|--------|-------------|
+| `withTokenmeter` (`tokenmeter/next`) | Next.js App Router wrapper |
+| `withInngest` (`tokenmeter/inngest`) | Inngest function wrapper |
+| `getInngestTraceHeaders` (`tokenmeter/inngest`) | Get headers for Inngest events |
+
+## Contributing
+
+Contributions are welcome! Please read our [Contributing Guide](./CONTRIBUTING.md) for details.
+
+```bash
+# Install dependencies
+pnpm install
+
+# Run tests
+pnpm test
+
+# Build
+pnpm build
+
+# Type check
+pnpm check-types
+```
+
+## License
+
+MIT

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

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

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

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

package/dist/__tests__/context.test.js

@@ -0,0 +1,94 @@
+import { describe, it, expect } from "vitest";
+import { withCostTrace, getCurrentContext, getCurrentCost, addCostToContext, getContextEvents, addEventToContext, } from "../context.js";
+describe("Context Module", () => {
+    describe("withCostTrace", () => {
+        it("should create a trace context", async () => {
+            let capturedContext;
+            await withCostTrace({ identifier: "user_123", workflowType: "test" }, async () => {
+                capturedContext = getCurrentContext();
+            });
+            expect(capturedContext).toBeDefined();
+            expect(capturedContext?.identifier).toBe("user_123");
+            expect(capturedContext?.workflowType).toBe("test");
+        });
+        it("should generate workflowId if not provided", async () => {
+            let capturedWorkflowId;
+            await withCostTrace({ identifier: "user_123" }, async () => {
+                capturedWorkflowId = getCurrentContext()?.workflowId;
+            });
+            expect(capturedWorkflowId).toBeDefined();
+            expect(capturedWorkflowId).toMatch(/^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/i);
+        });
+        it("should use provided workflowId", async () => {
+            await withCostTrace({ identifier: "user_123", workflowId: "custom_wf_123" }, async () => {
+                expect(getCurrentContext()?.workflowId).toBe("custom_wf_123");
+            });
+        });
+        it("should return the result of the callback", async () => {
+            const result = await withCostTrace({ identifier: "user_123" }, async () => {
+                return "test_result";
+            });
+            expect(result).toBe("test_result");
+        });
+        it("should clear context after callback completes", async () => {
+            await withCostTrace({ identifier: "user_123" }, async () => {
+                expect(getCurrentContext()).toBeDefined();
+            });
+            expect(getCurrentContext()).toBeUndefined();
+        });
+        it("should handle nested traces (inner overrides outer)", async () => {
+            await withCostTrace({ identifier: "outer_user" }, async () => {
+                expect(getCurrentContext()?.identifier).toBe("outer_user");
+                await withCostTrace({ identifier: "inner_user" }, async () => {
+                    expect(getCurrentContext()?.identifier).toBe("inner_user");
+                });
+                expect(getCurrentContext()?.identifier).toBe("outer_user");
+            });
+        });
+    });
+    describe("getCurrentCost", () => {
+        it("should return 0 when no context", () => {
+            expect(getCurrentCost()).toBe(0);
+        });
+        it("should return accumulated cost within context", async () => {
+            await withCostTrace({ identifier: "user_123" }, async () => {
+                expect(getCurrentCost()).toBe(0);
+                addCostToContext(0.5);
+                expect(getCurrentCost()).toBe(0.5);
+                addCostToContext(0.3);
+                expect(getCurrentCost()).toBe(0.8);
+            });
+        });
+    });
+    describe("getContextEvents", () => {
+        it("should return empty array when no context", () => {
+            expect(getContextEvents()).toEqual([]);
+        });
+        it("should track events within context", async () => {
+            await withCostTrace({ identifier: "user_123" }, async () => {
+                const event = {
+                    identifier: "user_123",
+                    provider: "openai",
+                    model: "gpt-4o",
+                    costUsd: 0.01,
+                };
+                addEventToContext(event);
+                const events = getContextEvents();
+                expect(events).toHaveLength(1);
+                expect(events[0]?.provider).toBe("openai");
+            });
+        });
+        it("should clear events after context ends", async () => {
+            await withCostTrace({ identifier: "user_123" }, async () => {
+                addEventToContext({
+                    identifier: "user_123",
+                    provider: "openai",
+                    model: "gpt-4o",
+                    costUsd: 0.01,
+                });
+            });
+            expect(getContextEvents()).toEqual([]);
+        });
+    });
+});
+//# sourceMappingURL=context.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"context.test.js","sourceRoot":"","sources":["../../src/__tests__/context.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAc,MAAM,QAAQ,CAAC;AAC1D,OAAO,EACL,aAAa,EACb,iBAAiB,EACjB,cAAc,EACd,gBAAgB,EAChB,gBAAgB,EAChB,iBAAiB,GAClB,MAAM,eAAe,CAAC;AAGvB,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;YAC7C,IAAI,eAAe,CAAC;YAEpB,MAAM,aAAa,CACjB,EAAE,UAAU,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,EAAE,EAChD,KAAK,IAAI,EAAE;gBACT,eAAe,GAAG,iBAAiB,EAAE,CAAC;YACxC,CAAC,CACF,CAAC;YAEF,MAAM,CAAC,eAAe,CAAC,CAAC,WAAW,EAAE,CAAC;YACtC,MAAM,CAAC,eAAe,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACrD,MAAM,CAAC,eAAe,EAAE,YAAY,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,IAAI,kBAAkB,CAAC;YAEvB,MAAM,aAAa,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,EAAE,KAAK,IAAI,EAAE;gBACzD,kBAAkB,GAAG,iBAAiB,EAAE,EAAE,UAAU,CAAC;YACvD,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,kBAAkB,CAAC,CAAC,WAAW,EAAE,CAAC;YACzC,MAAM,CAAC,kBAAkB,CAAC,CAAC,OAAO,CAChC,iEAAiE,CAClE,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,MAAM,aAAa,CACjB,EAAE,UAAU,EAAE,UAAU,EAAE,UAAU,EAAE,eAAe,EAAE,EACvD,KAAK,IAAI,EAAE;gBACT,MAAM,CAAC,iBAAiB,EAAE,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YAChE,CAAC,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,EAAE,KAAK,IAAI,EAAE;gBACxE,OAAO,aAAa,CAAC;YACvB,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,aAAa,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,EAAE,KAAK,IAAI,EAAE;gBACzD,MAAM,CAAC,iBAAiB,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;YAC5C,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,iBAAiB,EAAE,CAAC,CAAC,aAAa,EAAE,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,aAAa,CAAC,EAAE,UAAU,EAAE,YAAY,EAAE,EAAE,KAAK,IAAI,EAAE;gBAC3D,MAAM,CAAC,iBAAiB,EAAE,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;gBAE3D,MAAM,aAAa,CAAC,EAAE,UAAU,EAAE,YAAY,EAAE,EAAE,KAAK,IAAI,EAAE;oBAC3D,MAAM,CAAC,iBAAiB,EAAE,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;gBAC7D,CAAC,CAAC,CAAC;gBAEH,MAAM,CAAC,iBAAiB,EAAE,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YAC7D,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,aAAa,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,EAAE,KAAK,IAAI,EAAE;gBACzD,MAAM,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;gBACjC,gBAAgB,CAAC,GAAG,CAAC,CAAC;gBACtB,MAAM,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBACnC,gBAAgB,CAAC,GAAG,CAAC,CAAC;gBACtB,MAAM,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACrC,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,CAAC,gBAAgB,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,aAAa,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,EAAE,KAAK,IAAI,EAAE;gBACzD,MAAM,KAAK,GAAc;oBACvB,UAAU,EAAE,UAAU;oBACtB,QAAQ,EAAE,QAAQ;oBAClB,KAAK,EAAE,QAAQ;oBACf,OAAO,EAAE,IAAI;iBACd,CAAC;gBAEF,iBAAiB,CAAC,KAAK,CAAC,CAAC;gBAEzB,MAAM,MAAM,GAAG,gBAAgB,EAAE,CAAC;gBAClC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;gBAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC7C,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,aAAa,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,EAAE,KAAK,IAAI,EAAE;gBACzD,iBAAiB,CAAC;oBAChB,UAAU,EAAE,UAAU;oBACtB,QAAQ,EAAE,QAAQ;oBAClB,KAAK,EAAE,QAAQ;oBACf,OAAO,EAAE,IAAI;iBACd,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,gBAAgB,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

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

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

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

package/dist/__tests__/elevenlabs.test.js

@@ -0,0 +1,108 @@
+import { describe, it, expect, beforeEach, afterEach, vi } from "vitest";
+import { trackElevenLabs } from "../sdks/elevenlabs.js";
+import { init, shutdown, withCostTrace, flush } from "../index.js";
+import { InMemoryStorageAdapter } from "../storage/interface.js";
+describe("ElevenLabs SDK Wrapper", () => {
+    let storage;
+    beforeEach(() => {
+        storage = new InMemoryStorageAdapter();
+        init({ storage });
+    });
+    afterEach(async () => {
+        await flush();
+        await shutdown();
+    });
+    const createMockClient = () => ({
+        textToSpeech: {
+            convert: vi.fn().mockResolvedValue(new ArrayBuffer(1024)),
+            convertAsStream: vi.fn().mockResolvedValue(new ReadableStream()),
+        },
+        generate: vi.fn().mockResolvedValue(new ArrayBuffer(1024)),
+    });
+    describe("trackElevenLabs", () => {
+        it("should wrap textToSpeech.convert and track costs", async () => {
+            const mockClient = createMockClient();
+            const trackedClient = trackElevenLabs(mockClient);
+            await withCostTrace({ identifier: "user_123" }, async () => {
+                await trackedClient.textToSpeech.convert("voice_123", {
+                    text: "Hello, this is a test message with some text.",
+                    model_id: "eleven_multilingual_v2",
+                });
+            });
+            await flush();
+            const events = storage.getEvents();
+            expect(events).toHaveLength(1);
+            expect(events[0]?.provider).toBe("elevenlabs");
+            expect(events[0]?.model).toBe("eleven_multilingual_v2");
+            expect(events[0]?.operation).toBe("text-to-speech");
+            expect(events[0]?.costUsd).toBeGreaterThan(0);
+            expect((events[0]?.metadata).characterCount).toBe(46);
+        });
+        it("should calculate cost based on character count", async () => {
+            const mockClient = createMockClient();
+            const trackedClient = trackElevenLabs(mockClient);
+            await withCostTrace({ identifier: "user_123" }, async () => {
+                // 1000 characters with eleven_multilingual_v2 ($0.30 per 1K chars)
+                await trackedClient.textToSpeech.convert("voice_123", {
+                    text: "x".repeat(1000),
+                    model_id: "eleven_multilingual_v2",
+                });
+            });
+            await flush();
+            const events = storage.getEvents();
+            expect(events[0]?.costUsd).toBeCloseTo(0.30, 4);
+        });
+        it("should use cheaper pricing for turbo model", async () => {
+            const mockClient = createMockClient();
+            const trackedClient = trackElevenLabs(mockClient);
+            await withCostTrace({ identifier: "user_123" }, async () => {
+                // 1000 characters with eleven_turbo_v2_5 ($0.15 per 1K chars)
+                await trackedClient.textToSpeech.convert("voice_123", {
+                    text: "x".repeat(1000),
+                    model_id: "eleven_turbo_v2_5",
+                });
+            });
+            await flush();
+            const events = storage.getEvents();
+            expect(events[0]?.costUsd).toBeCloseTo(0.15, 4);
+        });
+        it("should track failed calls", async () => {
+            const mockClient = createMockClient();
+            mockClient.textToSpeech.convert.mockRejectedValue(new Error("API Error"));
+            const trackedClient = trackElevenLabs(mockClient);
+            await withCostTrace({ identifier: "user_123" }, async () => {
+                try {
+                    await trackedClient.textToSpeech.convert("voice_123", {
+                        text: "Test",
+                        model_id: "eleven_multilingual_v2",
+                    });
+                }
+                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("API Error");
+        });
+        it("should wrap generate method", async () => {
+            const mockClient = createMockClient();
+            const trackedClient = trackElevenLabs(mockClient);
+            await withCostTrace({ identifier: "user_123" }, async () => {
+                await trackedClient.generate({
+                    text: "Hello world",
+                    voice: "voice_123",
+                    model_id: "eleven_turbo_v2",
+                });
+            });
+            await flush();
+            const events = storage.getEvents();
+            expect(events).toHaveLength(1);
+            expect(events[0]?.provider).toBe("elevenlabs");
+        });
+    });
+});
+//# sourceMappingURL=elevenlabs.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"elevenlabs.test.js","sourceRoot":"","sources":["../../src/__tests__/elevenlabs.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,eAAe,EAAE,MAAM,uBAAuB,CAAC;AACxD,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,aAAa,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AACnE,OAAO,EAAE,sBAAsB,EAAE,MAAM,yBAAyB,CAAC;AAEjE,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;IACtC,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,YAAY,EAAE;YACZ,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,WAAW,CAAC,IAAI,CAAC,CAAC;YACzD,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,cAAc,EAAE,CAAC;SACjE;QACD,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,WAAW,CAAC,IAAI,CAAC,CAAC;KAC3D,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,UAAU,GAAG,gBAAgB,EAAE,CAAC;YACtC,MAAM,aAAa,GAAG,eAAe,CAAC,UAAU,CAAC,CAAC;YAElD,MAAM,aAAa,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,EAAE,KAAK,IAAI,EAAE;gBACzD,MAAM,aAAa,CAAC,YAAY,CAAC,OAAO,CAAC,WAAW,EAAE;oBACpD,IAAI,EAAE,+CAA+C;oBACrD,QAAQ,EAAE,wBAAwB;iBACnC,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,YAAY,CAAC,CAAC;YAC/C,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;YACxD,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YACpD,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAC9C,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,QAAuC,CAAA,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACtF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,UAAU,GAAG,gBAAgB,EAAE,CAAC;YACtC,MAAM,aAAa,GAAG,eAAe,CAAC,UAAU,CAAC,CAAC;YAElD,MAAM,aAAa,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,EAAE,KAAK,IAAI,EAAE;gBACzD,mEAAmE;gBACnE,MAAM,aAAa,CAAC,YAAY,CAAC,OAAO,CAAC,WAAW,EAAE;oBACpD,IAAI,EAAE,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC;oBACtB,QAAQ,EAAE,wBAAwB;iBACnC,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;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,UAAU,GAAG,gBAAgB,EAAE,CAAC;YACtC,MAAM,aAAa,GAAG,eAAe,CAAC,UAAU,CAAC,CAAC;YAElD,MAAM,aAAa,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,EAAE,KAAK,IAAI,EAAE;gBACzD,8DAA8D;gBAC9D,MAAM,aAAa,CAAC,YAAY,CAAC,OAAO,CAAC,WAAW,EAAE;oBACpD,IAAI,EAAE,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC;oBACtB,QAAQ,EAAE,mBAAmB;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,OAAO,CAAC,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;YACzC,MAAM,UAAU,GAAG,gBAAgB,EAAE,CAAC;YACtC,UAAU,CAAC,YAAY,CAAC,OAAO,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC;YAC1E,MAAM,aAAa,GAAG,eAAe,CAAC,UAAU,CAAC,CAAC;YAElD,MAAM,aAAa,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,EAAE,KAAK,IAAI,EAAE;gBACzD,IAAI,CAAC;oBACH,MAAM,aAAa,CAAC,YAAY,CAAC,OAAO,CAAC,WAAW,EAAE;wBACpD,IAAI,EAAE,MAAM;wBACZ,QAAQ,EAAE,wBAAwB;qBACnC,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,WAAW,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;YAC3C,MAAM,UAAU,GAAG,gBAAgB,EAAE,CAAC;YACtC,MAAM,aAAa,GAAG,eAAe,CAAC,UAAU,CAAC,CAAC;YAElD,MAAM,aAAa,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,EAAE,KAAK,IAAI,EAAE;gBACzD,MAAM,aAAa,CAAC,QAAS,CAAC;oBAC5B,IAAI,EAAE,aAAa;oBACnB,KAAK,EAAE,WAAW;oBAClB,QAAQ,EAAE,iBAAiB;iBAC5B,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,YAAY,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

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

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

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