@tollgateai/sdk 0.1.0

package/README.md

@@ -0,0 +1,103 @@
+# @tollgateai/sdk
+
+Track **real** LLM model usage and compute live gross margin with
+[Tollgate](https://tollgateapp.vercel.app). The SDK reads the actual `usage`
+object off each provider response — you never hand-count tokens.
+
+```bash
+npm install @tollgateai/sdk
+# or: pnpm add @tollgateai/sdk / yarn add @tollgateai/sdk
+```
+
+Create an API key in **Tollgate → Integrations**, then set:
+
+```bash
+TOLLGATE_API_KEY=tg_live_xxx
+# optional, defaults to the hosted app:
+TOLLGATE_BASE_URL=https://tollgateapp.vercel.app
+```
+
+## Auto-instrumentation (recommended)
+
+Wrap your provider client once; every call reports real usage in the background.
+
+### Anthropic
+
+```ts
+import Anthropic from '@anthropic-ai/sdk';
+import { createTollgateClient, wrapAnthropic } from '@tollgateai/sdk';
+
+const tollgate = createTollgateClient(); // reads TOLLGATE_API_KEY
+const anthropic = wrapAnthropic(new Anthropic(), tollgate, {
+  customerId: 'cust_A',     // your end customer
+  revenueUnitCents: 50,     // what you charge for this unit ($0.50)
+});
+
+// Use the client normally — usage is tracked automatically.
+await anthropic.messages.create({
+  model: 'claude-sonnet-4-6',
+  max_tokens: 512,
+  messages: [{ role: 'user', content: 'Summarize this ticket…' }],
+});
+```
+
+### OpenAI
+
+```ts
+import OpenAI from 'openai';
+import { createTollgateClient, wrapOpenAI } from '@tollgateai/sdk';
+
+const tollgate = createTollgateClient();
+const openai = wrapOpenAI(new OpenAI(), tollgate, { customerId: 'cust_A' });
+
+await openai.chat.completions.create({
+  model: 'gpt-4o',
+  messages: [{ role: 'user', content: 'Hello' }],
+});
+```
+
+`revenueUnitCents` may also be a function of the response, e.g.
+`revenueUnitCents: (res) => res.someField ? 50 : 0`.
+
+## Manual tracking
+
+For providers without a wrapper (Bedrock, custom gateways) or full control:
+
+```ts
+import { createTollgateClient } from '@tollgateai/sdk';
+
+const tollgate = createTollgateClient({ apiKey: process.env.TOLLGATE_API_KEY });
+
+await tollgate.track({
+  customerId: 'cust_A',
+  runId: 'run_12345',
+  provider: 'anthropic',
+  model: 'claude-sonnet-4-6',
+  tokensIn: 1200,
+  tokensOut: 450,
+  reasoningTokens: 0,
+  cachedTokens: 0,
+  revenueUnitCents: 50,
+  idempotencyKey: 'run_12345#step_1', // exactly-once: safe to retry
+});
+```
+
+## Notes
+
+- **Idempotent.** Events are deduplicated on `idempotencyKey` (auto-set to the
+  provider response id by the wrappers), so retries never double-count.
+- **No prompt content is ever sent** — only token counts and metadata.
+- **Streaming** responses are not auto-tracked yet (the wrappers only report when
+  a non-streaming `usage` object is present). Track those manually for now.
+- **Non-blocking.** Auto-instrumented tracking runs in the background; failures
+  are passed to `onError` (default `console.warn`) and never break your call.
+
+## API
+
+- `createTollgateClient(options?)` → `{ track(event) }`
+- `wrapAnthropic(client, tollgate, options)` → instrumented Anthropic client
+- `wrapOpenAI(client, tollgate, options)` → instrumented OpenAI client
+- `anthropicEventFrom(msg, options)` / `openAIEventFrom(completion, options)` →
+  build a track payload manually from a provider response
+
+Licensed for use with Tollgate. Not open source.

package/dist/index.cjs

@@ -0,0 +1,172 @@
+'use strict';
+
+// src/client.ts
+var TollgateError = class extends Error {
+  constructor(message, status, body) {
+    super(message);
+    this.status = status;
+    this.body = body;
+    this.name = "TollgateError";
+  }
+};
+var DEFAULT_BASE_URL = "https://tollgateapp.vercel.app";
+function getEnv(key) {
+  return typeof process !== "undefined" ? process.env?.[key] : void 0;
+}
+var sleep = (ms) => new Promise((r) => setTimeout(r, ms));
+function createTollgateClient(opts = {}) {
+  const apiKey = opts.apiKey ?? getEnv("TOLLGATE_API_KEY");
+  const baseUrl = (opts.baseUrl ?? getEnv("TOLLGATE_BASE_URL") ?? DEFAULT_BASE_URL).replace(/\/$/, "");
+  const timeoutMs = opts.timeoutMs ?? 1e4;
+  const maxRetries = opts.maxRetries ?? 2;
+  const doFetch = opts.fetch ?? globalThis.fetch;
+  if (typeof doFetch !== "function") {
+    throw new TollgateError("No fetch implementation available \u2014 pass `fetch` in options.");
+  }
+  async function track(event) {
+    if (!apiKey) {
+      throw new TollgateError("Missing API key \u2014 set opts.apiKey or TOLLGATE_API_KEY.");
+    }
+    let lastErr;
+    for (let attempt = 0; attempt <= maxRetries; attempt++) {
+      const controller = new AbortController();
+      const timer = setTimeout(() => controller.abort(), timeoutMs);
+      try {
+        const res = await doFetch(`${baseUrl}/api/track`, {
+          method: "POST",
+          headers: {
+            "Content-Type": "application/json",
+            Authorization: `Bearer ${apiKey}`
+          },
+          body: JSON.stringify(event),
+          signal: controller.signal
+        });
+        if (res.ok) {
+          return await res.json();
+        }
+        if (res.status >= 500 || res.status === 429) {
+          lastErr = new TollgateError(`Tollgate track failed (${res.status})`, res.status);
+        } else {
+          const body = await res.json().catch(() => ({}));
+          throw new TollgateError(`Tollgate track failed (${res.status})`, res.status, body);
+        }
+      } catch (err) {
+        if (err instanceof TollgateError && err.status && err.status < 500 && err.status !== 429) {
+          throw err;
+        }
+        lastErr = err;
+      } finally {
+        clearTimeout(timer);
+      }
+      if (attempt < maxRetries) {
+        await sleep(2 ** attempt * 200);
+      }
+    }
+    throw lastErr instanceof Error ? lastErr : new TollgateError("Tollgate track failed after retries");
+  }
+  return { track };
+}
+
+// src/instrument.ts
+function randomId() {
+  const c = globalThis.crypto;
+  if (c?.randomUUID) return c.randomUUID();
+  return `${Date.now()}-${Math.random().toString(36).slice(2)}`;
+}
+function resolveRunId(opts, responseId) {
+  if (typeof opts.runId === "function") return opts.runId();
+  return opts.runId ?? responseId ?? randomId();
+}
+function resolveRevenue(opts, response) {
+  return typeof opts.revenueUnitCents === "function" ? opts.revenueUnitCents(response) : opts.revenueUnitCents;
+}
+function fireAndForget(p, onError) {
+  p.catch((err) => (onError ?? ((e) => console.warn("[tollgate] track failed:", e)))(err));
+}
+function anthropicEventFrom(msg, opts) {
+  const usage = msg?.usage;
+  if (!usage) return null;
+  const runId = resolveRunId(opts, msg.id);
+  return {
+    customerId: opts.customerId,
+    agentId: opts.agentId,
+    runId,
+    provider: "anthropic",
+    model: msg.model ?? "unknown",
+    tokensIn: usage.input_tokens ?? 0,
+    tokensOut: usage.output_tokens ?? 0,
+    cachedTokens: usage.cache_read_input_tokens ?? 0,
+    revenueUnitCents: resolveRevenue(opts, msg),
+    idempotencyKey: msg.id ?? `${runId}#${randomId()}`
+  };
+}
+function wrapAnthropic(client, tollgate, opts) {
+  const messages = client.messages;
+  const original = messages.create.bind(messages);
+  const create = async (...args) => {
+    const result = await original(...args);
+    const event = anthropicEventFrom(result, opts);
+    if (event) fireAndForget(tollgate.track(event), opts.onError);
+    return result;
+  };
+  return new Proxy(client, {
+    get(target, prop, recv) {
+      if (prop === "messages") {
+        return new Proxy(messages, {
+          get: (m, p, r) => p === "create" ? create : Reflect.get(m, p, r)
+        });
+      }
+      return Reflect.get(target, prop, recv);
+    }
+  });
+}
+function openAIEventFrom(completion, opts) {
+  const usage = completion?.usage;
+  if (!usage) return null;
+  const runId = resolveRunId(opts, completion.id);
+  return {
+    customerId: opts.customerId,
+    agentId: opts.agentId,
+    runId,
+    provider: "openai",
+    model: completion.model ?? "unknown",
+    tokensIn: usage.prompt_tokens ?? 0,
+    tokensOut: usage.completion_tokens ?? 0,
+    reasoningTokens: usage.completion_tokens_details?.reasoning_tokens ?? 0,
+    cachedTokens: usage.prompt_tokens_details?.cached_tokens ?? 0,
+    revenueUnitCents: resolveRevenue(opts, completion),
+    idempotencyKey: completion.id ?? `${runId}#${randomId()}`
+  };
+}
+function wrapOpenAI(client, tollgate, opts) {
+  const completions = client.chat.completions;
+  const original = completions.create.bind(completions);
+  const create = async (...args) => {
+    const result = await original(...args);
+    const event = openAIEventFrom(result, opts);
+    if (event) fireAndForget(tollgate.track(event), opts.onError);
+    return result;
+  };
+  return new Proxy(client, {
+    get(target, prop, recv) {
+      if (prop === "chat") {
+        const chat = Reflect.get(target, prop, recv);
+        return new Proxy(chat, {
+          get: (c, p, r) => p === "completions" ? new Proxy(completions, {
+            get: (co, pp, rr) => pp === "create" ? create : Reflect.get(co, pp, rr)
+          }) : Reflect.get(c, p, r)
+        });
+      }
+      return Reflect.get(target, prop, recv);
+    }
+  });
+}
+
+exports.TollgateError = TollgateError;
+exports.anthropicEventFrom = anthropicEventFrom;
+exports.createTollgateClient = createTollgateClient;
+exports.openAIEventFrom = openAIEventFrom;
+exports.wrapAnthropic = wrapAnthropic;
+exports.wrapOpenAI = wrapOpenAI;
+//# sourceMappingURL=index.cjs.map
+//# sourceMappingURL=index.cjs.map

package/dist/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/client.ts","../src/instrument.ts"],"names":[],"mappings":";;;AAeO,IAAM,aAAA,GAAN,cAA4B,KAAA,CAAM;AAAA,EACvC,WAAA,CACE,OAAA,EACS,MAAA,EACA,IAAA,EACT;AACA,IAAA,KAAA,CAAM,OAAO,CAAA;AAHJ,IAAA,IAAA,CAAA,MAAA,GAAA,MAAA;AACA,IAAA,IAAA,CAAA,IAAA,GAAA,IAAA;AAGT,IAAA,IAAA,CAAK,IAAA,GAAO,eAAA;AAAA,EACd;AACF;AAEA,IAAM,gBAAA,GAAmB,gCAAA;AAOzB,SAAS,OAAO,GAAA,EAAiC;AAE/C,EAAA,OAAO,OAAO,OAAA,KAAY,WAAA,GAAc,OAAA,CAAQ,GAAA,GAAM,GAAG,CAAA,GAAI,MAAA;AAC/D;AAEA,IAAM,KAAA,GAAQ,CAAC,EAAA,KAAe,IAAI,OAAA,CAAQ,CAAC,CAAA,KAAM,UAAA,CAAW,CAAA,EAAG,EAAE,CAAC,CAAA;AAE3D,SAAS,oBAAA,CAAqB,IAAA,GAA8B,EAAC,EAAmB;AACrF,EAAA,MAAM,MAAA,GAAS,IAAA,CAAK,MAAA,IAAU,MAAA,CAAO,kBAAkB,CAAA;AACvD,EAAA,MAAM,OAAA,GAAA,CAAW,KAAK,OAAA,IAAW,MAAA,CAAO,mBAAmB,CAAA,IAAK,gBAAA,EAAkB,OAAA,CAAQ,KAAA,EAAO,EAAE,CAAA;AACnG,EAAA,MAAM,SAAA,GAAY,KAAK,SAAA,IAAa,GAAA;AACpC,EAAA,MAAM,UAAA,GAAa,KAAK,UAAA,IAAc,CAAA;AACtC,EAAA,MAAM,OAAA,GAAU,IAAA,CAAK,KAAA,IAAS,UAAA,CAAW,KAAA;AAEzC,EAAA,IAAI,OAAO,YAAY,UAAA,EAAY;AACjC,IAAA,MAAM,IAAI,cAAc,mEAA8D,CAAA;AAAA,EACxF;AAEA,EAAA,eAAe,MAAM,KAAA,EAA8C;AACjE,IAAA,IAAI,CAAC,MAAA,EAAQ;AACX,MAAA,MAAM,IAAI,cAAc,6DAAwD,CAAA;AAAA,IAClF;AAEA,IAAA,IAAI,OAAA;AACJ,IAAA,KAAA,IAAS,OAAA,GAAU,CAAA,EAAG,OAAA,IAAW,UAAA,EAAY,OAAA,EAAA,EAAW;AACtD,MAAA,MAAM,UAAA,GAAa,IAAI,eAAA,EAAgB;AACvC,MAAA,MAAM,QAAQ,UAAA,CAAW,MAAM,UAAA,CAAW,KAAA,IAAS,SAAS,CAAA;AAC5D,MAAA,IAAI;AACF,QAAA,MAAM,GAAA,GAAM,MAAM,OAAA,CAAQ,CAAA,EAAG,OAAO,CAAA,UAAA,CAAA,EAAc;AAAA,UAChD,MAAA,EAAQ,MAAA;AAAA,UACR,OAAA,EAAS;AAAA,YACP,cAAA,EAAgB,kBAAA;AAAA,YAChB,aAAA,EAAe,UAAU,MAAM,CAAA;AAAA,WACjC;AAAA,UACA,IAAA,EAAM,IAAA,CAAK,SAAA,CAAU,KAAK,CAAA;AAAA,UAC1B,QAAQ,UAAA,CAAW;AAAA,SACpB,CAAA;AAGD,QAAA,IAAI,IAAI,EAAA,EAAI;AACV,UAAA,OAAQ,MAAM,IAAI,IAAA,EAAK;AAAA,QACzB;AAGA,QAAA,IAAI,GAAA,CAAI,MAAA,IAAU,GAAA,IAAO,GAAA,CAAI,WAAW,GAAA,EAAK;AAC3C,UAAA,OAAA,GAAU,IAAI,aAAA,CAAc,CAAA,uBAAA,EAA0B,IAAI,MAAM,CAAA,CAAA,CAAA,EAAK,IAAI,MAAM,CAAA;AAAA,QACjF,CAAA,MAAO;AACL,UAAA,MAAM,IAAA,GAAO,MAAM,GAAA,CAAI,IAAA,GAAO,KAAA,CAAM,OAAO,EAAC,CAAE,CAAA;AAC9C,UAAA,MAAM,IAAI,cAAc,CAAA,uBAAA,EAA0B,GAAA,CAAI,MAAM,CAAA,CAAA,CAAA,EAAK,GAAA,CAAI,QAAQ,IAAI,CAAA;AAAA,QACnF;AAAA,MACF,SAAS,GAAA,EAAK;AAEZ,QAAA,IAAI,GAAA,YAAe,iBAAiB,GAAA,CAAI,MAAA,IAAU,IAAI,MAAA,GAAS,GAAA,IAAO,GAAA,CAAI,MAAA,KAAW,GAAA,EAAK;AACxF,UAAA,MAAM,GAAA;AAAA,QACR;AACA,QAAA,OAAA,GAAU,GAAA;AAAA,MACZ,CAAA,SAAE;AACA,QAAA,YAAA,CAAa,KAAK,CAAA;AAAA,MACpB;AAEA,MAAA,IAAI,UAAU,UAAA,EAAY;AACxB,QAAA,MAAM,KAAA,CAAM,CAAA,IAAK,OAAA,GAAU,GAAG,CAAA;AAAA,MAChC;AAAA,IACF;AAEA,IAAA,MAAM,OAAA,YAAmB,KAAA,GACrB,OAAA,GACA,IAAI,cAAc,qCAAqC,CAAA;AAAA,EAC7D;AAEA,EAAA,OAAO,EAAE,KAAA,EAAM;AACjB;;;ACpFA,SAAS,QAAA,GAAmB;AAC1B,EAAA,MAAM,IAAK,UAAA,CAAmC,MAAA;AAC9C,EAAA,IAAI,CAAA,EAAG,UAAA,EAAY,OAAO,CAAA,CAAE,UAAA,EAAW;AACvC,EAAA,OAAO,CAAA,EAAG,IAAA,CAAK,GAAA,EAAK,CAAA,CAAA,EAAI,IAAA,CAAK,MAAA,EAAO,CAAE,QAAA,CAAS,EAAE,CAAA,CAAE,KAAA,CAAM,CAAC,CAAC,CAAA,CAAA;AAC7D;AAEA,SAAS,YAAA,CAAa,MAAyB,UAAA,EAA6B;AAC1E,EAAA,IAAI,OAAO,IAAA,CAAK,KAAA,KAAU,UAAA,EAAY,OAAO,KAAK,KAAA,EAAM;AACxD,EAAA,OAAO,IAAA,CAAK,KAAA,IAAS,UAAA,IAAc,QAAA,EAAS;AAC9C;AAEA,SAAS,cAAA,CAAe,MAAyB,QAAA,EAAuC;AACtF,EAAA,OAAO,OAAO,KAAK,gBAAA,KAAqB,UAAA,GACpC,KAAK,gBAAA,CAAiB,QAAQ,IAC9B,IAAA,CAAK,gBAAA;AACX;AAEA,SAAS,aAAA,CAAc,GAAqB,OAAA,EAA8C;AACxF,EAAA,CAAA,CAAE,KAAA,CAAM,CAAC,GAAA,KAAA,CAAS,OAAA,KAAY,CAAC,CAAA,KAAM,OAAA,CAAQ,IAAA,CAAK,0BAAA,EAA4B,CAAC,CAAA,CAAA,EAAI,GAAG,CAAC,CAAA;AACzF;AAgBO,SAAS,kBAAA,CACd,KACA,IAAA,EACwB;AACxB,EAAA,MAAM,QAAQ,GAAA,EAAK,KAAA;AACnB,EAAA,IAAI,CAAC,OAAO,OAAO,IAAA;AACnB,EAAA,MAAM,KAAA,GAAQ,YAAA,CAAa,IAAA,EAAM,GAAA,CAAI,EAAE,CAAA;AACvC,EAAA,OAAO;AAAA,IACL,YAAY,IAAA,CAAK,UAAA;AAAA,IACjB,SAAS,IAAA,CAAK,OAAA;AAAA,IACd,KAAA;AAAA,IACA,QAAA,EAAU,WAAA;AAAA,IACV,KAAA,EAAO,IAAI,KAAA,IAAS,SAAA;AAAA,IACpB,QAAA,EAAU,MAAM,YAAA,IAAgB,CAAA;AAAA,IAChC,SAAA,EAAW,MAAM,aAAA,IAAiB,CAAA;AAAA,IAClC,YAAA,EAAc,MAAM,uBAAA,IAA2B,CAAA;AAAA,IAC/C,gBAAA,EAAkB,cAAA,CAAe,IAAA,EAAM,GAAG,CAAA;AAAA,IAC1C,gBAAgB,GAAA,CAAI,EAAA,IAAM,GAAG,KAAK,CAAA,CAAA,EAAI,UAAU,CAAA;AAAA,GAClD;AACF;AAOO,SAAS,aAAA,CACd,MAAA,EACA,QAAA,EACA,IAAA,EACG;AACH,EAAA,MAAM,WAAW,MAAA,CAAO,QAAA;AACxB,EAAA,MAAM,QAAA,GAAW,QAAA,CAAS,MAAA,CAAO,IAAA,CAAK,QAAQ,CAAA;AAE9C,EAAA,MAAM,MAAA,GAAS,UAAU,IAAA,KAAoC;AAC3D,IAAA,MAAM,MAAA,GAAS,MAAM,QAAA,CAAS,GAAG,IAAI,CAAA;AACrC,IAAA,MAAM,KAAA,GAAQ,kBAAA,CAAmB,MAAA,EAA4B,IAAI,CAAA;AACjE,IAAA,IAAI,OAAO,aAAA,CAAc,QAAA,CAAS,MAAM,KAAK,CAAA,EAAG,KAAK,OAAO,CAAA;AAC5D,IAAA,OAAO,MAAA;AAAA,EACT,CAAA;AAEA,EAAA,OAAO,IAAI,MAAM,MAAA,EAAQ;AAAA,IACvB,GAAA,CAAI,MAAA,EAAQ,IAAA,EAAM,IAAA,EAAM;AACtB,MAAA,IAAI,SAAS,UAAA,EAAY;AACvB,QAAA,OAAO,IAAI,MAAM,QAAA,EAAU;AAAA,UACzB,GAAA,EAAK,CAAC,CAAA,EAAG,CAAA,EAAG,CAAA,KAAO,CAAA,KAAM,QAAA,GAAW,MAAA,GAAS,OAAA,CAAQ,GAAA,CAAI,CAAA,EAAG,CAAA,EAAG,CAAC;AAAA,SACjE,CAAA;AAAA,MACH;AACA,MAAA,OAAO,OAAA,CAAQ,GAAA,CAAI,MAAA,EAAQ,IAAA,EAAM,IAAI,CAAA;AAAA,IACvC;AAAA,GACD,CAAA;AACH;AAiBO,SAAS,eAAA,CACd,YACA,IAAA,EACwB;AACxB,EAAA,MAAM,QAAQ,UAAA,EAAY,KAAA;AAC1B,EAAA,IAAI,CAAC,OAAO,OAAO,IAAA;AACnB,EAAA,MAAM,KAAA,GAAQ,YAAA,CAAa,IAAA,EAAM,UAAA,CAAW,EAAE,CAAA;AAC9C,EAAA,OAAO;AAAA,IACL,YAAY,IAAA,CAAK,UAAA;AAAA,IACjB,SAAS,IAAA,CAAK,OAAA;AAAA,IACd,KAAA;AAAA,IACA,QAAA,EAAU,QAAA;AAAA,IACV,KAAA,EAAO,WAAW,KAAA,IAAS,SAAA;AAAA,IAC3B,QAAA,EAAU,MAAM,aAAA,IAAiB,CAAA;AAAA,IACjC,SAAA,EAAW,MAAM,iBAAA,IAAqB,CAAA;AAAA,IACtC,eAAA,EAAiB,KAAA,CAAM,yBAAA,EAA2B,gBAAA,IAAoB,CAAA;AAAA,IACtE,YAAA,EAAc,KAAA,CAAM,qBAAA,EAAuB,aAAA,IAAiB,CAAA;AAAA,IAC5D,gBAAA,EAAkB,cAAA,CAAe,IAAA,EAAM,UAAU,CAAA;AAAA,IACjD,gBAAgB,UAAA,CAAW,EAAA,IAAM,GAAG,KAAK,CAAA,CAAA,EAAI,UAAU,CAAA;AAAA,GACzD;AACF;AAOO,SAAS,UAAA,CACd,MAAA,EACA,QAAA,EACA,IAAA,EACG;AACH,EAAA,MAAM,WAAA,GAAc,OAAO,IAAA,CAAK,WAAA;AAChC,EAAA,MAAM,QAAA,GAAW,WAAA,CAAY,MAAA,CAAO,IAAA,CAAK,WAAW,CAAA;AAEpD,EAAA,MAAM,MAAA,GAAS,UAAU,IAAA,KAAoC;AAC3D,IAAA,MAAM,MAAA,GAAS,MAAM,QAAA,CAAS,GAAG,IAAI,CAAA;AACrC,IAAA,MAAM,KAAA,GAAQ,eAAA,CAAgB,MAAA,EAA4B,IAAI,CAAA;AAC9D,IAAA,IAAI,OAAO,aAAA,CAAc,QAAA,CAAS,MAAM,KAAK,CAAA,EAAG,KAAK,OAAO,CAAA;AAC5D,IAAA,OAAO,MAAA;AAAA,EACT,CAAA;AAEA,EAAA,OAAO,IAAI,MAAM,MAAA,EAAQ;AAAA,IACvB,GAAA,CAAI,MAAA,EAAQ,IAAA,EAAM,IAAA,EAAM;AACtB,MAAA,IAAI,SAAS,MAAA,EAAQ;AACnB,QAAA,MAAM,IAAA,GAAO,OAAA,CAAQ,GAAA,CAAI,MAAA,EAAQ,MAAM,IAAI,CAAA;AAC3C,QAAA,OAAO,IAAI,MAAM,IAAA,EAAM;AAAA,UACrB,GAAA,EAAK,CAAC,CAAA,EAAG,CAAA,EAAG,MACV,CAAA,KAAM,aAAA,GACF,IAAI,KAAA,CAAM,WAAA,EAAa;AAAA,YACrB,GAAA,EAAK,CAAC,EAAA,EAAI,EAAA,EAAI,EAAA,KAAQ,EAAA,KAAO,QAAA,GAAW,MAAA,GAAS,OAAA,CAAQ,GAAA,CAAI,EAAA,EAAI,EAAA,EAAI,EAAE;AAAA,WACxE,CAAA,GACD,OAAA,CAAQ,GAAA,CAAI,CAAA,EAAG,GAAG,CAAC;AAAA,SAC1B,CAAA;AAAA,MACH;AACA,MAAA,OAAO,OAAA,CAAQ,GAAA,CAAI,MAAA,EAAQ,IAAA,EAAM,IAAI,CAAA;AAAA,IACvC;AAAA,GACD,CAAA;AACH","file":"index.cjs","sourcesContent":["import type { TrackEventInput, TrackResult } from './types';\n\nexport interface TollgateClientOptions {\n  /** Account API key (`tg_live_…`). Falls back to `process.env.TOLLGATE_API_KEY`. */\n  apiKey?: string;\n  /** Base URL of your Tollgate deployment. Defaults to `TOLLGATE_BASE_URL` or production. */\n  baseUrl?: string;\n  /** Per-request timeout in ms. Default 10_000. */\n  timeoutMs?: number;\n  /** Retry attempts on network error / 5xx / 429. Default 2. */\n  maxRetries?: number;\n  /** Custom fetch (for testing or non-standard runtimes). Defaults to global fetch. */\n  fetch?: typeof fetch;\n}\n\nexport class TollgateError extends Error {\n  constructor(\n    message: string,\n    readonly status?: number,\n    readonly body?: unknown,\n  ) {\n    super(message);\n    this.name = 'TollgateError';\n  }\n}\n\nconst DEFAULT_BASE_URL = 'https://tollgateapp.vercel.app';\n\nexport interface TollgateClient {\n  /** Report a single usage event. Idempotent on `idempotencyKey`. */\n  track(event: TrackEventInput): Promise<TrackResult>;\n}\n\nfunction getEnv(key: string): string | undefined {\n  // Guarded so the SDK works in browsers/edge runtimes without `process`.\n  return typeof process !== 'undefined' ? process.env?.[key] : undefined;\n}\n\nconst sleep = (ms: number) => new Promise((r) => setTimeout(r, ms));\n\nexport function createTollgateClient(opts: TollgateClientOptions = {}): TollgateClient {\n  const apiKey = opts.apiKey ?? getEnv('TOLLGATE_API_KEY');\n  const baseUrl = (opts.baseUrl ?? getEnv('TOLLGATE_BASE_URL') ?? DEFAULT_BASE_URL).replace(/\\/$/, '');\n  const timeoutMs = opts.timeoutMs ?? 10_000;\n  const maxRetries = opts.maxRetries ?? 2;\n  const doFetch = opts.fetch ?? globalThis.fetch;\n\n  if (typeof doFetch !== 'function') {\n    throw new TollgateError('No fetch implementation available — pass `fetch` in options.');\n  }\n\n  async function track(event: TrackEventInput): Promise<TrackResult> {\n    if (!apiKey) {\n      throw new TollgateError('Missing API key — set opts.apiKey or TOLLGATE_API_KEY.');\n    }\n\n    let lastErr: unknown;\n    for (let attempt = 0; attempt <= maxRetries; attempt++) {\n      const controller = new AbortController();\n      const timer = setTimeout(() => controller.abort(), timeoutMs);\n      try {\n        const res = await doFetch(`${baseUrl}/api/track`, {\n          method: 'POST',\n          headers: {\n            'Content-Type': 'application/json',\n            Authorization: `Bearer ${apiKey}`,\n          },\n          body: JSON.stringify(event),\n          signal: controller.signal,\n        });\n\n        // 200 = idempotent duplicate, 201 = created. Both are success.\n        if (res.ok) {\n          return (await res.json()) as TrackResult;\n        }\n\n        // Retry transient failures; fail fast on 4xx (except 429).\n        if (res.status >= 500 || res.status === 429) {\n          lastErr = new TollgateError(`Tollgate track failed (${res.status})`, res.status);\n        } else {\n          const body = await res.json().catch(() => ({}));\n          throw new TollgateError(`Tollgate track failed (${res.status})`, res.status, body);\n        }\n      } catch (err) {\n        // Don't retry deterministic client errors.\n        if (err instanceof TollgateError && err.status && err.status < 500 && err.status !== 429) {\n          throw err;\n        }\n        lastErr = err;\n      } finally {\n        clearTimeout(timer);\n      }\n\n      if (attempt < maxRetries) {\n        await sleep(2 ** attempt * 200); // 200ms, 400ms, …\n      }\n    }\n\n    throw lastErr instanceof Error\n      ? lastErr\n      : new TollgateError('Tollgate track failed after retries');\n  }\n\n  return { track };\n}\n","// Auto-instrumentation: wrap a provider client so every completion reports its\n// REAL usage to Tollgate — no manual token counting. Wrappers are structurally\n// typed, so this package never has to depend on the provider SDKs.\n\nimport type { TollgateClient } from './client';\nimport type { TrackEventInput } from './types';\n\nexport interface InstrumentOptions {\n  /** Your end customer's stable id. Required for margin attribution. */\n  customerId: string;\n  /** Optional agent/workflow id. */\n  agentId?: string;\n  /** Revenue per call in cents (or a function of the response). */\n  revenueUnitCents?: number | ((response: unknown) => number | undefined);\n  /** Override the run id; defaults to the provider response id. */\n  runId?: string | (() => string);\n  /** Called if a background track() fails. Defaults to console.warn. */\n  onError?: (err: unknown) => void;\n}\n\nfunction randomId(): string {\n  const c = (globalThis as { crypto?: Crypto }).crypto;\n  if (c?.randomUUID) return c.randomUUID();\n  return `${Date.now()}-${Math.random().toString(36).slice(2)}`;\n}\n\nfunction resolveRunId(opts: InstrumentOptions, responseId?: string): string {\n  if (typeof opts.runId === 'function') return opts.runId();\n  return opts.runId ?? responseId ?? randomId();\n}\n\nfunction resolveRevenue(opts: InstrumentOptions, response: unknown): number | undefined {\n  return typeof opts.revenueUnitCents === 'function'\n    ? opts.revenueUnitCents(response)\n    : opts.revenueUnitCents;\n}\n\nfunction fireAndForget(p: Promise<unknown>, onError?: InstrumentOptions['onError']): void {\n  p.catch((err) => (onError ?? ((e) => console.warn('[tollgate] track failed:', e)))(err));\n}\n\n// --- Anthropic ------------------------------------------------------------\n\ninterface AnthropicUsage {\n  input_tokens?: number;\n  output_tokens?: number;\n  cache_read_input_tokens?: number;\n}\ninterface AnthropicMessage {\n  id?: string;\n  model?: string;\n  usage?: AnthropicUsage;\n}\n\n/** Map a non-streaming Anthropic message to a track payload (or null if no usage). */\nexport function anthropicEventFrom(\n  msg: AnthropicMessage,\n  opts: InstrumentOptions,\n): TrackEventInput | null {\n  const usage = msg?.usage;\n  if (!usage) return null;\n  const runId = resolveRunId(opts, msg.id);\n  return {\n    customerId: opts.customerId,\n    agentId: opts.agentId,\n    runId,\n    provider: 'anthropic',\n    model: msg.model ?? 'unknown',\n    tokensIn: usage.input_tokens ?? 0,\n    tokensOut: usage.output_tokens ?? 0,\n    cachedTokens: usage.cache_read_input_tokens ?? 0,\n    revenueUnitCents: resolveRevenue(opts, msg),\n    idempotencyKey: msg.id ?? `${runId}#${randomId()}`,\n  };\n}\n\ninterface AnthropicLike {\n  messages: { create: (...args: never[]) => Promise<unknown> };\n}\n\n/** Wrap an Anthropic client so `messages.create` auto-reports usage. */\nexport function wrapAnthropic<T extends AnthropicLike>(\n  client: T,\n  tollgate: TollgateClient,\n  opts: InstrumentOptions,\n): T {\n  const messages = client.messages;\n  const original = messages.create.bind(messages) as (...a: never[]) => Promise<unknown>;\n\n  const create = async (...args: never[]): Promise<unknown> => {\n    const result = await original(...args);\n    const event = anthropicEventFrom(result as AnthropicMessage, opts);\n    if (event) fireAndForget(tollgate.track(event), opts.onError);\n    return result;\n  };\n\n  return new Proxy(client, {\n    get(target, prop, recv) {\n      if (prop === 'messages') {\n        return new Proxy(messages, {\n          get: (m, p, r) => (p === 'create' ? create : Reflect.get(m, p, r)),\n        });\n      }\n      return Reflect.get(target, prop, recv);\n    },\n  });\n}\n\n// --- OpenAI ---------------------------------------------------------------\n\ninterface OpenAIUsage {\n  prompt_tokens?: number;\n  completion_tokens?: number;\n  completion_tokens_details?: { reasoning_tokens?: number };\n  prompt_tokens_details?: { cached_tokens?: number };\n}\ninterface OpenAICompletion {\n  id?: string;\n  model?: string;\n  usage?: OpenAIUsage;\n}\n\n/** Map a non-streaming OpenAI chat completion to a track payload (or null). */\nexport function openAIEventFrom(\n  completion: OpenAICompletion,\n  opts: InstrumentOptions,\n): TrackEventInput | null {\n  const usage = completion?.usage;\n  if (!usage) return null;\n  const runId = resolveRunId(opts, completion.id);\n  return {\n    customerId: opts.customerId,\n    agentId: opts.agentId,\n    runId,\n    provider: 'openai',\n    model: completion.model ?? 'unknown',\n    tokensIn: usage.prompt_tokens ?? 0,\n    tokensOut: usage.completion_tokens ?? 0,\n    reasoningTokens: usage.completion_tokens_details?.reasoning_tokens ?? 0,\n    cachedTokens: usage.prompt_tokens_details?.cached_tokens ?? 0,\n    revenueUnitCents: resolveRevenue(opts, completion),\n    idempotencyKey: completion.id ?? `${runId}#${randomId()}`,\n  };\n}\n\ninterface OpenAILike {\n  chat: { completions: { create: (...args: never[]) => Promise<unknown> } };\n}\n\n/** Wrap an OpenAI client so `chat.completions.create` auto-reports usage. */\nexport function wrapOpenAI<T extends OpenAILike>(\n  client: T,\n  tollgate: TollgateClient,\n  opts: InstrumentOptions,\n): T {\n  const completions = client.chat.completions;\n  const original = completions.create.bind(completions) as (...a: never[]) => Promise<unknown>;\n\n  const create = async (...args: never[]): Promise<unknown> => {\n    const result = await original(...args);\n    const event = openAIEventFrom(result as OpenAICompletion, opts);\n    if (event) fireAndForget(tollgate.track(event), opts.onError);\n    return result;\n  };\n\n  return new Proxy(client, {\n    get(target, prop, recv) {\n      if (prop === 'chat') {\n        const chat = Reflect.get(target, prop, recv) as OpenAILike['chat'];\n        return new Proxy(chat, {\n          get: (c, p, r) =>\n            p === 'completions'\n              ? new Proxy(completions, {\n                  get: (co, pp, rr) => (pp === 'create' ? create : Reflect.get(co, pp, rr)),\n                })\n              : Reflect.get(c, p, r),\n        });\n      }\n      return Reflect.get(target, prop, recv);\n    },\n  });\n}\n"]}

package/dist/index.d.cts

@@ -0,0 +1,116 @@
+type Provider = 'anthropic' | 'openai' | 'bedrock';
+type EventType = 'llm' | 'tool' | 'retrieval';
+/** The payload accepted by `POST /api/track`. Money is in integer cents. */
+interface TrackEventInput {
+    /** Your end customer's stable id (e.g. "cust_A"). Used for margin attribution. */
+    customerId: string;
+    /** Logical run/request id (e.g. "run_12345"). */
+    runId: string;
+    /** Optional agent/workflow id for per-agent margin. */
+    agentId?: string;
+    /** Defaults to "llm". */
+    type?: EventType;
+    provider: Provider;
+    model: string;
+    tokensIn?: number;
+    tokensOut?: number;
+    /** Reasoning/thinking tokens — billed at output rates. */
+    reasoningTokens?: number;
+    /** Cached / cache-read input tokens. */
+    cachedTokens?: number;
+    toolCalls?: number;
+    toolName?: string;
+    /** Per-event revenue contribution in cents (e.g. 50 for a $0.50 unit). */
+    revenueUnitCents?: number;
+    /** Required for exactly-once ingestion (e.g. "run_12345#step_1"). */
+    idempotencyKey: string;
+    /** ISO timestamp; defaults to server receive time. */
+    ts?: string;
+}
+interface TrackResult {
+    status: 'created' | 'duplicate' | string;
+    eventId: string;
+}
+
+interface TollgateClientOptions {
+    /** Account API key (`tg_live_…`). Falls back to `process.env.TOLLGATE_API_KEY`. */
+    apiKey?: string;
+    /** Base URL of your Tollgate deployment. Defaults to `TOLLGATE_BASE_URL` or production. */
+    baseUrl?: string;
+    /** Per-request timeout in ms. Default 10_000. */
+    timeoutMs?: number;
+    /** Retry attempts on network error / 5xx / 429. Default 2. */
+    maxRetries?: number;
+    /** Custom fetch (for testing or non-standard runtimes). Defaults to global fetch. */
+    fetch?: typeof fetch;
+}
+declare class TollgateError extends Error {
+    readonly status?: number | undefined;
+    readonly body?: unknown | undefined;
+    constructor(message: string, status?: number | undefined, body?: unknown | undefined);
+}
+interface TollgateClient {
+    /** Report a single usage event. Idempotent on `idempotencyKey`. */
+    track(event: TrackEventInput): Promise<TrackResult>;
+}
+declare function createTollgateClient(opts?: TollgateClientOptions): TollgateClient;
+
+interface InstrumentOptions {
+    /** Your end customer's stable id. Required for margin attribution. */
+    customerId: string;
+    /** Optional agent/workflow id. */
+    agentId?: string;
+    /** Revenue per call in cents (or a function of the response). */
+    revenueUnitCents?: number | ((response: unknown) => number | undefined);
+    /** Override the run id; defaults to the provider response id. */
+    runId?: string | (() => string);
+    /** Called if a background track() fails. Defaults to console.warn. */
+    onError?: (err: unknown) => void;
+}
+interface AnthropicUsage {
+    input_tokens?: number;
+    output_tokens?: number;
+    cache_read_input_tokens?: number;
+}
+interface AnthropicMessage {
+    id?: string;
+    model?: string;
+    usage?: AnthropicUsage;
+}
+/** Map a non-streaming Anthropic message to a track payload (or null if no usage). */
+declare function anthropicEventFrom(msg: AnthropicMessage, opts: InstrumentOptions): TrackEventInput | null;
+interface AnthropicLike {
+    messages: {
+        create: (...args: never[]) => Promise<unknown>;
+    };
+}
+/** Wrap an Anthropic client so `messages.create` auto-reports usage. */
+declare function wrapAnthropic<T extends AnthropicLike>(client: T, tollgate: TollgateClient, opts: InstrumentOptions): T;
+interface OpenAIUsage {
+    prompt_tokens?: number;
+    completion_tokens?: number;
+    completion_tokens_details?: {
+        reasoning_tokens?: number;
+    };
+    prompt_tokens_details?: {
+        cached_tokens?: number;
+    };
+}
+interface OpenAICompletion {
+    id?: string;
+    model?: string;
+    usage?: OpenAIUsage;
+}
+/** Map a non-streaming OpenAI chat completion to a track payload (or null). */
+declare function openAIEventFrom(completion: OpenAICompletion, opts: InstrumentOptions): TrackEventInput | null;
+interface OpenAILike {
+    chat: {
+        completions: {
+            create: (...args: never[]) => Promise<unknown>;
+        };
+    };
+}
+/** Wrap an OpenAI client so `chat.completions.create` auto-reports usage. */
+declare function wrapOpenAI<T extends OpenAILike>(client: T, tollgate: TollgateClient, opts: InstrumentOptions): T;
+
+export { type EventType, type InstrumentOptions, type Provider, type TollgateClient, type TollgateClientOptions, TollgateError, type TrackEventInput, type TrackResult, anthropicEventFrom, createTollgateClient, openAIEventFrom, wrapAnthropic, wrapOpenAI };

package/dist/index.d.ts

@@ -0,0 +1,116 @@
+type Provider = 'anthropic' | 'openai' | 'bedrock';
+type EventType = 'llm' | 'tool' | 'retrieval';
+/** The payload accepted by `POST /api/track`. Money is in integer cents. */
+interface TrackEventInput {
+    /** Your end customer's stable id (e.g. "cust_A"). Used for margin attribution. */
+    customerId: string;
+    /** Logical run/request id (e.g. "run_12345"). */
+    runId: string;
+    /** Optional agent/workflow id for per-agent margin. */
+    agentId?: string;
+    /** Defaults to "llm". */
+    type?: EventType;
+    provider: Provider;
+    model: string;
+    tokensIn?: number;
+    tokensOut?: number;
+    /** Reasoning/thinking tokens — billed at output rates. */
+    reasoningTokens?: number;
+    /** Cached / cache-read input tokens. */
+    cachedTokens?: number;
+    toolCalls?: number;
+    toolName?: string;
+    /** Per-event revenue contribution in cents (e.g. 50 for a $0.50 unit). */
+    revenueUnitCents?: number;
+    /** Required for exactly-once ingestion (e.g. "run_12345#step_1"). */
+    idempotencyKey: string;
+    /** ISO timestamp; defaults to server receive time. */
+    ts?: string;
+}
+interface TrackResult {
+    status: 'created' | 'duplicate' | string;
+    eventId: string;
+}
+
+interface TollgateClientOptions {
+    /** Account API key (`tg_live_…`). Falls back to `process.env.TOLLGATE_API_KEY`. */
+    apiKey?: string;
+    /** Base URL of your Tollgate deployment. Defaults to `TOLLGATE_BASE_URL` or production. */
+    baseUrl?: string;
+    /** Per-request timeout in ms. Default 10_000. */
+    timeoutMs?: number;
+    /** Retry attempts on network error / 5xx / 429. Default 2. */
+    maxRetries?: number;
+    /** Custom fetch (for testing or non-standard runtimes). Defaults to global fetch. */
+    fetch?: typeof fetch;
+}
+declare class TollgateError extends Error {
+    readonly status?: number | undefined;
+    readonly body?: unknown | undefined;
+    constructor(message: string, status?: number | undefined, body?: unknown | undefined);
+}
+interface TollgateClient {
+    /** Report a single usage event. Idempotent on `idempotencyKey`. */
+    track(event: TrackEventInput): Promise<TrackResult>;
+}
+declare function createTollgateClient(opts?: TollgateClientOptions): TollgateClient;
+
+interface InstrumentOptions {
+    /** Your end customer's stable id. Required for margin attribution. */
+    customerId: string;
+    /** Optional agent/workflow id. */
+    agentId?: string;
+    /** Revenue per call in cents (or a function of the response). */
+    revenueUnitCents?: number | ((response: unknown) => number | undefined);
+    /** Override the run id; defaults to the provider response id. */
+    runId?: string | (() => string);
+    /** Called if a background track() fails. Defaults to console.warn. */
+    onError?: (err: unknown) => void;
+}
+interface AnthropicUsage {
+    input_tokens?: number;
+    output_tokens?: number;
+    cache_read_input_tokens?: number;
+}
+interface AnthropicMessage {
+    id?: string;
+    model?: string;
+    usage?: AnthropicUsage;
+}
+/** Map a non-streaming Anthropic message to a track payload (or null if no usage). */
+declare function anthropicEventFrom(msg: AnthropicMessage, opts: InstrumentOptions): TrackEventInput | null;
+interface AnthropicLike {
+    messages: {
+        create: (...args: never[]) => Promise<unknown>;
+    };
+}
+/** Wrap an Anthropic client so `messages.create` auto-reports usage. */
+declare function wrapAnthropic<T extends AnthropicLike>(client: T, tollgate: TollgateClient, opts: InstrumentOptions): T;
+interface OpenAIUsage {
+    prompt_tokens?: number;
+    completion_tokens?: number;
+    completion_tokens_details?: {
+        reasoning_tokens?: number;
+    };
+    prompt_tokens_details?: {
+        cached_tokens?: number;
+    };
+}
+interface OpenAICompletion {
+    id?: string;
+    model?: string;
+    usage?: OpenAIUsage;
+}
+/** Map a non-streaming OpenAI chat completion to a track payload (or null). */
+declare function openAIEventFrom(completion: OpenAICompletion, opts: InstrumentOptions): TrackEventInput | null;
+interface OpenAILike {
+    chat: {
+        completions: {
+            create: (...args: never[]) => Promise<unknown>;
+        };
+    };
+}
+/** Wrap an OpenAI client so `chat.completions.create` auto-reports usage. */
+declare function wrapOpenAI<T extends OpenAILike>(client: T, tollgate: TollgateClient, opts: InstrumentOptions): T;
+
+export { type EventType, type InstrumentOptions, type Provider, type TollgateClient, type TollgateClientOptions, TollgateError, type TrackEventInput, type TrackResult, anthropicEventFrom, createTollgateClient, openAIEventFrom, wrapAnthropic, wrapOpenAI };

package/dist/index.js

@@ -0,0 +1,165 @@
+// src/client.ts
+var TollgateError = class extends Error {
+  constructor(message, status, body) {
+    super(message);
+    this.status = status;
+    this.body = body;
+    this.name = "TollgateError";
+  }
+};
+var DEFAULT_BASE_URL = "https://tollgateapp.vercel.app";
+function getEnv(key) {
+  return typeof process !== "undefined" ? process.env?.[key] : void 0;
+}
+var sleep = (ms) => new Promise((r) => setTimeout(r, ms));
+function createTollgateClient(opts = {}) {
+  const apiKey = opts.apiKey ?? getEnv("TOLLGATE_API_KEY");
+  const baseUrl = (opts.baseUrl ?? getEnv("TOLLGATE_BASE_URL") ?? DEFAULT_BASE_URL).replace(/\/$/, "");
+  const timeoutMs = opts.timeoutMs ?? 1e4;
+  const maxRetries = opts.maxRetries ?? 2;
+  const doFetch = opts.fetch ?? globalThis.fetch;
+  if (typeof doFetch !== "function") {
+    throw new TollgateError("No fetch implementation available \u2014 pass `fetch` in options.");
+  }
+  async function track(event) {
+    if (!apiKey) {
+      throw new TollgateError("Missing API key \u2014 set opts.apiKey or TOLLGATE_API_KEY.");
+    }
+    let lastErr;
+    for (let attempt = 0; attempt <= maxRetries; attempt++) {
+      const controller = new AbortController();
+      const timer = setTimeout(() => controller.abort(), timeoutMs);
+      try {
+        const res = await doFetch(`${baseUrl}/api/track`, {
+          method: "POST",
+          headers: {
+            "Content-Type": "application/json",
+            Authorization: `Bearer ${apiKey}`
+          },
+          body: JSON.stringify(event),
+          signal: controller.signal
+        });
+        if (res.ok) {
+          return await res.json();
+        }
+        if (res.status >= 500 || res.status === 429) {
+          lastErr = new TollgateError(`Tollgate track failed (${res.status})`, res.status);
+        } else {
+          const body = await res.json().catch(() => ({}));
+          throw new TollgateError(`Tollgate track failed (${res.status})`, res.status, body);
+        }
+      } catch (err) {
+        if (err instanceof TollgateError && err.status && err.status < 500 && err.status !== 429) {
+          throw err;
+        }
+        lastErr = err;
+      } finally {
+        clearTimeout(timer);
+      }
+      if (attempt < maxRetries) {
+        await sleep(2 ** attempt * 200);
+      }
+    }
+    throw lastErr instanceof Error ? lastErr : new TollgateError("Tollgate track failed after retries");
+  }
+  return { track };
+}
+
+// src/instrument.ts
+function randomId() {
+  const c = globalThis.crypto;
+  if (c?.randomUUID) return c.randomUUID();
+  return `${Date.now()}-${Math.random().toString(36).slice(2)}`;
+}
+function resolveRunId(opts, responseId) {
+  if (typeof opts.runId === "function") return opts.runId();
+  return opts.runId ?? responseId ?? randomId();
+}
+function resolveRevenue(opts, response) {
+  return typeof opts.revenueUnitCents === "function" ? opts.revenueUnitCents(response) : opts.revenueUnitCents;
+}
+function fireAndForget(p, onError) {
+  p.catch((err) => (onError ?? ((e) => console.warn("[tollgate] track failed:", e)))(err));
+}
+function anthropicEventFrom(msg, opts) {
+  const usage = msg?.usage;
+  if (!usage) return null;
+  const runId = resolveRunId(opts, msg.id);
+  return {
+    customerId: opts.customerId,
+    agentId: opts.agentId,
+    runId,
+    provider: "anthropic",
+    model: msg.model ?? "unknown",
+    tokensIn: usage.input_tokens ?? 0,
+    tokensOut: usage.output_tokens ?? 0,
+    cachedTokens: usage.cache_read_input_tokens ?? 0,
+    revenueUnitCents: resolveRevenue(opts, msg),
+    idempotencyKey: msg.id ?? `${runId}#${randomId()}`
+  };
+}
+function wrapAnthropic(client, tollgate, opts) {
+  const messages = client.messages;
+  const original = messages.create.bind(messages);
+  const create = async (...args) => {
+    const result = await original(...args);
+    const event = anthropicEventFrom(result, opts);
+    if (event) fireAndForget(tollgate.track(event), opts.onError);
+    return result;
+  };
+  return new Proxy(client, {
+    get(target, prop, recv) {
+      if (prop === "messages") {
+        return new Proxy(messages, {
+          get: (m, p, r) => p === "create" ? create : Reflect.get(m, p, r)
+        });
+      }
+      return Reflect.get(target, prop, recv);
+    }
+  });
+}
+function openAIEventFrom(completion, opts) {
+  const usage = completion?.usage;
+  if (!usage) return null;
+  const runId = resolveRunId(opts, completion.id);
+  return {
+    customerId: opts.customerId,
+    agentId: opts.agentId,
+    runId,
+    provider: "openai",
+    model: completion.model ?? "unknown",
+    tokensIn: usage.prompt_tokens ?? 0,
+    tokensOut: usage.completion_tokens ?? 0,
+    reasoningTokens: usage.completion_tokens_details?.reasoning_tokens ?? 0,
+    cachedTokens: usage.prompt_tokens_details?.cached_tokens ?? 0,
+    revenueUnitCents: resolveRevenue(opts, completion),
+    idempotencyKey: completion.id ?? `${runId}#${randomId()}`
+  };
+}
+function wrapOpenAI(client, tollgate, opts) {
+  const completions = client.chat.completions;
+  const original = completions.create.bind(completions);
+  const create = async (...args) => {
+    const result = await original(...args);
+    const event = openAIEventFrom(result, opts);
+    if (event) fireAndForget(tollgate.track(event), opts.onError);
+    return result;
+  };
+  return new Proxy(client, {
+    get(target, prop, recv) {
+      if (prop === "chat") {
+        const chat = Reflect.get(target, prop, recv);
+        return new Proxy(chat, {
+          get: (c, p, r) => p === "completions" ? new Proxy(completions, {
+            get: (co, pp, rr) => pp === "create" ? create : Reflect.get(co, pp, rr)
+          }) : Reflect.get(c, p, r)
+        });
+      }
+      return Reflect.get(target, prop, recv);
+    }
+  });
+}
+
+export { TollgateError, anthropicEventFrom, createTollgateClient, openAIEventFrom, wrapAnthropic, wrapOpenAI };
+//# sourceMappingURL=index.js.map
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/client.ts","../src/instrument.ts"],"names":[],"mappings":";AAeO,IAAM,aAAA,GAAN,cAA4B,KAAA,CAAM;AAAA,EACvC,WAAA,CACE,OAAA,EACS,MAAA,EACA,IAAA,EACT;AACA,IAAA,KAAA,CAAM,OAAO,CAAA;AAHJ,IAAA,IAAA,CAAA,MAAA,GAAA,MAAA;AACA,IAAA,IAAA,CAAA,IAAA,GAAA,IAAA;AAGT,IAAA,IAAA,CAAK,IAAA,GAAO,eAAA;AAAA,EACd;AACF;AAEA,IAAM,gBAAA,GAAmB,gCAAA;AAOzB,SAAS,OAAO,GAAA,EAAiC;AAE/C,EAAA,OAAO,OAAO,OAAA,KAAY,WAAA,GAAc,OAAA,CAAQ,GAAA,GAAM,GAAG,CAAA,GAAI,MAAA;AAC/D;AAEA,IAAM,KAAA,GAAQ,CAAC,EAAA,KAAe,IAAI,OAAA,CAAQ,CAAC,CAAA,KAAM,UAAA,CAAW,CAAA,EAAG,EAAE,CAAC,CAAA;AAE3D,SAAS,oBAAA,CAAqB,IAAA,GAA8B,EAAC,EAAmB;AACrF,EAAA,MAAM,MAAA,GAAS,IAAA,CAAK,MAAA,IAAU,MAAA,CAAO,kBAAkB,CAAA;AACvD,EAAA,MAAM,OAAA,GAAA,CAAW,KAAK,OAAA,IAAW,MAAA,CAAO,mBAAmB,CAAA,IAAK,gBAAA,EAAkB,OAAA,CAAQ,KAAA,EAAO,EAAE,CAAA;AACnG,EAAA,MAAM,SAAA,GAAY,KAAK,SAAA,IAAa,GAAA;AACpC,EAAA,MAAM,UAAA,GAAa,KAAK,UAAA,IAAc,CAAA;AACtC,EAAA,MAAM,OAAA,GAAU,IAAA,CAAK,KAAA,IAAS,UAAA,CAAW,KAAA;AAEzC,EAAA,IAAI,OAAO,YAAY,UAAA,EAAY;AACjC,IAAA,MAAM,IAAI,cAAc,mEAA8D,CAAA;AAAA,EACxF;AAEA,EAAA,eAAe,MAAM,KAAA,EAA8C;AACjE,IAAA,IAAI,CAAC,MAAA,EAAQ;AACX,MAAA,MAAM,IAAI,cAAc,6DAAwD,CAAA;AAAA,IAClF;AAEA,IAAA,IAAI,OAAA;AACJ,IAAA,KAAA,IAAS,OAAA,GAAU,CAAA,EAAG,OAAA,IAAW,UAAA,EAAY,OAAA,EAAA,EAAW;AACtD,MAAA,MAAM,UAAA,GAAa,IAAI,eAAA,EAAgB;AACvC,MAAA,MAAM,QAAQ,UAAA,CAAW,MAAM,UAAA,CAAW,KAAA,IAAS,SAAS,CAAA;AAC5D,MAAA,IAAI;AACF,QAAA,MAAM,GAAA,GAAM,MAAM,OAAA,CAAQ,CAAA,EAAG,OAAO,CAAA,UAAA,CAAA,EAAc;AAAA,UAChD,MAAA,EAAQ,MAAA;AAAA,UACR,OAAA,EAAS;AAAA,YACP,cAAA,EAAgB,kBAAA;AAAA,YAChB,aAAA,EAAe,UAAU,MAAM,CAAA;AAAA,WACjC;AAAA,UACA,IAAA,EAAM,IAAA,CAAK,SAAA,CAAU,KAAK,CAAA;AAAA,UAC1B,QAAQ,UAAA,CAAW;AAAA,SACpB,CAAA;AAGD,QAAA,IAAI,IAAI,EAAA,EAAI;AACV,UAAA,OAAQ,MAAM,IAAI,IAAA,EAAK;AAAA,QACzB;AAGA,QAAA,IAAI,GAAA,CAAI,MAAA,IAAU,GAAA,IAAO,GAAA,CAAI,WAAW,GAAA,EAAK;AAC3C,UAAA,OAAA,GAAU,IAAI,aAAA,CAAc,CAAA,uBAAA,EAA0B,IAAI,MAAM,CAAA,CAAA,CAAA,EAAK,IAAI,MAAM,CAAA;AAAA,QACjF,CAAA,MAAO;AACL,UAAA,MAAM,IAAA,GAAO,MAAM,GAAA,CAAI,IAAA,GAAO,KAAA,CAAM,OAAO,EAAC,CAAE,CAAA;AAC9C,UAAA,MAAM,IAAI,cAAc,CAAA,uBAAA,EAA0B,GAAA,CAAI,MAAM,CAAA,CAAA,CAAA,EAAK,GAAA,CAAI,QAAQ,IAAI,CAAA;AAAA,QACnF;AAAA,MACF,SAAS,GAAA,EAAK;AAEZ,QAAA,IAAI,GAAA,YAAe,iBAAiB,GAAA,CAAI,MAAA,IAAU,IAAI,MAAA,GAAS,GAAA,IAAO,GAAA,CAAI,MAAA,KAAW,GAAA,EAAK;AACxF,UAAA,MAAM,GAAA;AAAA,QACR;AACA,QAAA,OAAA,GAAU,GAAA;AAAA,MACZ,CAAA,SAAE;AACA,QAAA,YAAA,CAAa,KAAK,CAAA;AAAA,MACpB;AAEA,MAAA,IAAI,UAAU,UAAA,EAAY;AACxB,QAAA,MAAM,KAAA,CAAM,CAAA,IAAK,OAAA,GAAU,GAAG,CAAA;AAAA,MAChC;AAAA,IACF;AAEA,IAAA,MAAM,OAAA,YAAmB,KAAA,GACrB,OAAA,GACA,IAAI,cAAc,qCAAqC,CAAA;AAAA,EAC7D;AAEA,EAAA,OAAO,EAAE,KAAA,EAAM;AACjB;;;ACpFA,SAAS,QAAA,GAAmB;AAC1B,EAAA,MAAM,IAAK,UAAA,CAAmC,MAAA;AAC9C,EAAA,IAAI,CAAA,EAAG,UAAA,EAAY,OAAO,CAAA,CAAE,UAAA,EAAW;AACvC,EAAA,OAAO,CAAA,EAAG,IAAA,CAAK,GAAA,EAAK,CAAA,CAAA,EAAI,IAAA,CAAK,MAAA,EAAO,CAAE,QAAA,CAAS,EAAE,CAAA,CAAE,KAAA,CAAM,CAAC,CAAC,CAAA,CAAA;AAC7D;AAEA,SAAS,YAAA,CAAa,MAAyB,UAAA,EAA6B;AAC1E,EAAA,IAAI,OAAO,IAAA,CAAK,KAAA,KAAU,UAAA,EAAY,OAAO,KAAK,KAAA,EAAM;AACxD,EAAA,OAAO,IAAA,CAAK,KAAA,IAAS,UAAA,IAAc,QAAA,EAAS;AAC9C;AAEA,SAAS,cAAA,CAAe,MAAyB,QAAA,EAAuC;AACtF,EAAA,OAAO,OAAO,KAAK,gBAAA,KAAqB,UAAA,GACpC,KAAK,gBAAA,CAAiB,QAAQ,IAC9B,IAAA,CAAK,gBAAA;AACX;AAEA,SAAS,aAAA,CAAc,GAAqB,OAAA,EAA8C;AACxF,EAAA,CAAA,CAAE,KAAA,CAAM,CAAC,GAAA,KAAA,CAAS,OAAA,KAAY,CAAC,CAAA,KAAM,OAAA,CAAQ,IAAA,CAAK,0BAAA,EAA4B,CAAC,CAAA,CAAA,EAAI,GAAG,CAAC,CAAA;AACzF;AAgBO,SAAS,kBAAA,CACd,KACA,IAAA,EACwB;AACxB,EAAA,MAAM,QAAQ,GAAA,EAAK,KAAA;AACnB,EAAA,IAAI,CAAC,OAAO,OAAO,IAAA;AACnB,EAAA,MAAM,KAAA,GAAQ,YAAA,CAAa,IAAA,EAAM,GAAA,CAAI,EAAE,CAAA;AACvC,EAAA,OAAO;AAAA,IACL,YAAY,IAAA,CAAK,UAAA;AAAA,IACjB,SAAS,IAAA,CAAK,OAAA;AAAA,IACd,KAAA;AAAA,IACA,QAAA,EAAU,WAAA;AAAA,IACV,KAAA,EAAO,IAAI,KAAA,IAAS,SAAA;AAAA,IACpB,QAAA,EAAU,MAAM,YAAA,IAAgB,CAAA;AAAA,IAChC,SAAA,EAAW,MAAM,aAAA,IAAiB,CAAA;AAAA,IAClC,YAAA,EAAc,MAAM,uBAAA,IAA2B,CAAA;AAAA,IAC/C,gBAAA,EAAkB,cAAA,CAAe,IAAA,EAAM,GAAG,CAAA;AAAA,IAC1C,gBAAgB,GAAA,CAAI,EAAA,IAAM,GAAG,KAAK,CAAA,CAAA,EAAI,UAAU,CAAA;AAAA,GAClD;AACF;AAOO,SAAS,aAAA,CACd,MAAA,EACA,QAAA,EACA,IAAA,EACG;AACH,EAAA,MAAM,WAAW,MAAA,CAAO,QAAA;AACxB,EAAA,MAAM,QAAA,GAAW,QAAA,CAAS,MAAA,CAAO,IAAA,CAAK,QAAQ,CAAA;AAE9C,EAAA,MAAM,MAAA,GAAS,UAAU,IAAA,KAAoC;AAC3D,IAAA,MAAM,MAAA,GAAS,MAAM,QAAA,CAAS,GAAG,IAAI,CAAA;AACrC,IAAA,MAAM,KAAA,GAAQ,kBAAA,CAAmB,MAAA,EAA4B,IAAI,CAAA;AACjE,IAAA,IAAI,OAAO,aAAA,CAAc,QAAA,CAAS,MAAM,KAAK,CAAA,EAAG,KAAK,OAAO,CAAA;AAC5D,IAAA,OAAO,MAAA;AAAA,EACT,CAAA;AAEA,EAAA,OAAO,IAAI,MAAM,MAAA,EAAQ;AAAA,IACvB,GAAA,CAAI,MAAA,EAAQ,IAAA,EAAM,IAAA,EAAM;AACtB,MAAA,IAAI,SAAS,UAAA,EAAY;AACvB,QAAA,OAAO,IAAI,MAAM,QAAA,EAAU;AAAA,UACzB,GAAA,EAAK,CAAC,CAAA,EAAG,CAAA,EAAG,CAAA,KAAO,CAAA,KAAM,QAAA,GAAW,MAAA,GAAS,OAAA,CAAQ,GAAA,CAAI,CAAA,EAAG,CAAA,EAAG,CAAC;AAAA,SACjE,CAAA;AAAA,MACH;AACA,MAAA,OAAO,OAAA,CAAQ,GAAA,CAAI,MAAA,EAAQ,IAAA,EAAM,IAAI,CAAA;AAAA,IACvC;AAAA,GACD,CAAA;AACH;AAiBO,SAAS,eAAA,CACd,YACA,IAAA,EACwB;AACxB,EAAA,MAAM,QAAQ,UAAA,EAAY,KAAA;AAC1B,EAAA,IAAI,CAAC,OAAO,OAAO,IAAA;AACnB,EAAA,MAAM,KAAA,GAAQ,YAAA,CAAa,IAAA,EAAM,UAAA,CAAW,EAAE,CAAA;AAC9C,EAAA,OAAO;AAAA,IACL,YAAY,IAAA,CAAK,UAAA;AAAA,IACjB,SAAS,IAAA,CAAK,OAAA;AAAA,IACd,KAAA;AAAA,IACA,QAAA,EAAU,QAAA;AAAA,IACV,KAAA,EAAO,WAAW,KAAA,IAAS,SAAA;AAAA,IAC3B,QAAA,EAAU,MAAM,aAAA,IAAiB,CAAA;AAAA,IACjC,SAAA,EAAW,MAAM,iBAAA,IAAqB,CAAA;AAAA,IACtC,eAAA,EAAiB,KAAA,CAAM,yBAAA,EAA2B,gBAAA,IAAoB,CAAA;AAAA,IACtE,YAAA,EAAc,KAAA,CAAM,qBAAA,EAAuB,aAAA,IAAiB,CAAA;AAAA,IAC5D,gBAAA,EAAkB,cAAA,CAAe,IAAA,EAAM,UAAU,CAAA;AAAA,IACjD,gBAAgB,UAAA,CAAW,EAAA,IAAM,GAAG,KAAK,CAAA,CAAA,EAAI,UAAU,CAAA;AAAA,GACzD;AACF;AAOO,SAAS,UAAA,CACd,MAAA,EACA,QAAA,EACA,IAAA,EACG;AACH,EAAA,MAAM,WAAA,GAAc,OAAO,IAAA,CAAK,WAAA;AAChC,EAAA,MAAM,QAAA,GAAW,WAAA,CAAY,MAAA,CAAO,IAAA,CAAK,WAAW,CAAA;AAEpD,EAAA,MAAM,MAAA,GAAS,UAAU,IAAA,KAAoC;AAC3D,IAAA,MAAM,MAAA,GAAS,MAAM,QAAA,CAAS,GAAG,IAAI,CAAA;AACrC,IAAA,MAAM,KAAA,GAAQ,eAAA,CAAgB,MAAA,EAA4B,IAAI,CAAA;AAC9D,IAAA,IAAI,OAAO,aAAA,CAAc,QAAA,CAAS,MAAM,KAAK,CAAA,EAAG,KAAK,OAAO,CAAA;AAC5D,IAAA,OAAO,MAAA;AAAA,EACT,CAAA;AAEA,EAAA,OAAO,IAAI,MAAM,MAAA,EAAQ;AAAA,IACvB,GAAA,CAAI,MAAA,EAAQ,IAAA,EAAM,IAAA,EAAM;AACtB,MAAA,IAAI,SAAS,MAAA,EAAQ;AACnB,QAAA,MAAM,IAAA,GAAO,OAAA,CAAQ,GAAA,CAAI,MAAA,EAAQ,MAAM,IAAI,CAAA;AAC3C,QAAA,OAAO,IAAI,MAAM,IAAA,EAAM;AAAA,UACrB,GAAA,EAAK,CAAC,CAAA,EAAG,CAAA,EAAG,MACV,CAAA,KAAM,aAAA,GACF,IAAI,KAAA,CAAM,WAAA,EAAa;AAAA,YACrB,GAAA,EAAK,CAAC,EAAA,EAAI,EAAA,EAAI,EAAA,KAAQ,EAAA,KAAO,QAAA,GAAW,MAAA,GAAS,OAAA,CAAQ,GAAA,CAAI,EAAA,EAAI,EAAA,EAAI,EAAE;AAAA,WACxE,CAAA,GACD,OAAA,CAAQ,GAAA,CAAI,CAAA,EAAG,GAAG,CAAC;AAAA,SAC1B,CAAA;AAAA,MACH;AACA,MAAA,OAAO,OAAA,CAAQ,GAAA,CAAI,MAAA,EAAQ,IAAA,EAAM,IAAI,CAAA;AAAA,IACvC;AAAA,GACD,CAAA;AACH","file":"index.js","sourcesContent":["import type { TrackEventInput, TrackResult } from './types';\n\nexport interface TollgateClientOptions {\n  /** Account API key (`tg_live_…`). Falls back to `process.env.TOLLGATE_API_KEY`. */\n  apiKey?: string;\n  /** Base URL of your Tollgate deployment. Defaults to `TOLLGATE_BASE_URL` or production. */\n  baseUrl?: string;\n  /** Per-request timeout in ms. Default 10_000. */\n  timeoutMs?: number;\n  /** Retry attempts on network error / 5xx / 429. Default 2. */\n  maxRetries?: number;\n  /** Custom fetch (for testing or non-standard runtimes). Defaults to global fetch. */\n  fetch?: typeof fetch;\n}\n\nexport class TollgateError extends Error {\n  constructor(\n    message: string,\n    readonly status?: number,\n    readonly body?: unknown,\n  ) {\n    super(message);\n    this.name = 'TollgateError';\n  }\n}\n\nconst DEFAULT_BASE_URL = 'https://tollgateapp.vercel.app';\n\nexport interface TollgateClient {\n  /** Report a single usage event. Idempotent on `idempotencyKey`. */\n  track(event: TrackEventInput): Promise<TrackResult>;\n}\n\nfunction getEnv(key: string): string | undefined {\n  // Guarded so the SDK works in browsers/edge runtimes without `process`.\n  return typeof process !== 'undefined' ? process.env?.[key] : undefined;\n}\n\nconst sleep = (ms: number) => new Promise((r) => setTimeout(r, ms));\n\nexport function createTollgateClient(opts: TollgateClientOptions = {}): TollgateClient {\n  const apiKey = opts.apiKey ?? getEnv('TOLLGATE_API_KEY');\n  const baseUrl = (opts.baseUrl ?? getEnv('TOLLGATE_BASE_URL') ?? DEFAULT_BASE_URL).replace(/\\/$/, '');\n  const timeoutMs = opts.timeoutMs ?? 10_000;\n  const maxRetries = opts.maxRetries ?? 2;\n  const doFetch = opts.fetch ?? globalThis.fetch;\n\n  if (typeof doFetch !== 'function') {\n    throw new TollgateError('No fetch implementation available — pass `fetch` in options.');\n  }\n\n  async function track(event: TrackEventInput): Promise<TrackResult> {\n    if (!apiKey) {\n      throw new TollgateError('Missing API key — set opts.apiKey or TOLLGATE_API_KEY.');\n    }\n\n    let lastErr: unknown;\n    for (let attempt = 0; attempt <= maxRetries; attempt++) {\n      const controller = new AbortController();\n      const timer = setTimeout(() => controller.abort(), timeoutMs);\n      try {\n        const res = await doFetch(`${baseUrl}/api/track`, {\n          method: 'POST',\n          headers: {\n            'Content-Type': 'application/json',\n            Authorization: `Bearer ${apiKey}`,\n          },\n          body: JSON.stringify(event),\n          signal: controller.signal,\n        });\n\n        // 200 = idempotent duplicate, 201 = created. Both are success.\n        if (res.ok) {\n          return (await res.json()) as TrackResult;\n        }\n\n        // Retry transient failures; fail fast on 4xx (except 429).\n        if (res.status >= 500 || res.status === 429) {\n          lastErr = new TollgateError(`Tollgate track failed (${res.status})`, res.status);\n        } else {\n          const body = await res.json().catch(() => ({}));\n          throw new TollgateError(`Tollgate track failed (${res.status})`, res.status, body);\n        }\n      } catch (err) {\n        // Don't retry deterministic client errors.\n        if (err instanceof TollgateError && err.status && err.status < 500 && err.status !== 429) {\n          throw err;\n        }\n        lastErr = err;\n      } finally {\n        clearTimeout(timer);\n      }\n\n      if (attempt < maxRetries) {\n        await sleep(2 ** attempt * 200); // 200ms, 400ms, …\n      }\n    }\n\n    throw lastErr instanceof Error\n      ? lastErr\n      : new TollgateError('Tollgate track failed after retries');\n  }\n\n  return { track };\n}\n","// Auto-instrumentation: wrap a provider client so every completion reports its\n// REAL usage to Tollgate — no manual token counting. Wrappers are structurally\n// typed, so this package never has to depend on the provider SDKs.\n\nimport type { TollgateClient } from './client';\nimport type { TrackEventInput } from './types';\n\nexport interface InstrumentOptions {\n  /** Your end customer's stable id. Required for margin attribution. */\n  customerId: string;\n  /** Optional agent/workflow id. */\n  agentId?: string;\n  /** Revenue per call in cents (or a function of the response). */\n  revenueUnitCents?: number | ((response: unknown) => number | undefined);\n  /** Override the run id; defaults to the provider response id. */\n  runId?: string | (() => string);\n  /** Called if a background track() fails. Defaults to console.warn. */\n  onError?: (err: unknown) => void;\n}\n\nfunction randomId(): string {\n  const c = (globalThis as { crypto?: Crypto }).crypto;\n  if (c?.randomUUID) return c.randomUUID();\n  return `${Date.now()}-${Math.random().toString(36).slice(2)}`;\n}\n\nfunction resolveRunId(opts: InstrumentOptions, responseId?: string): string {\n  if (typeof opts.runId === 'function') return opts.runId();\n  return opts.runId ?? responseId ?? randomId();\n}\n\nfunction resolveRevenue(opts: InstrumentOptions, response: unknown): number | undefined {\n  return typeof opts.revenueUnitCents === 'function'\n    ? opts.revenueUnitCents(response)\n    : opts.revenueUnitCents;\n}\n\nfunction fireAndForget(p: Promise<unknown>, onError?: InstrumentOptions['onError']): void {\n  p.catch((err) => (onError ?? ((e) => console.warn('[tollgate] track failed:', e)))(err));\n}\n\n// --- Anthropic ------------------------------------------------------------\n\ninterface AnthropicUsage {\n  input_tokens?: number;\n  output_tokens?: number;\n  cache_read_input_tokens?: number;\n}\ninterface AnthropicMessage {\n  id?: string;\n  model?: string;\n  usage?: AnthropicUsage;\n}\n\n/** Map a non-streaming Anthropic message to a track payload (or null if no usage). */\nexport function anthropicEventFrom(\n  msg: AnthropicMessage,\n  opts: InstrumentOptions,\n): TrackEventInput | null {\n  const usage = msg?.usage;\n  if (!usage) return null;\n  const runId = resolveRunId(opts, msg.id);\n  return {\n    customerId: opts.customerId,\n    agentId: opts.agentId,\n    runId,\n    provider: 'anthropic',\n    model: msg.model ?? 'unknown',\n    tokensIn: usage.input_tokens ?? 0,\n    tokensOut: usage.output_tokens ?? 0,\n    cachedTokens: usage.cache_read_input_tokens ?? 0,\n    revenueUnitCents: resolveRevenue(opts, msg),\n    idempotencyKey: msg.id ?? `${runId}#${randomId()}`,\n  };\n}\n\ninterface AnthropicLike {\n  messages: { create: (...args: never[]) => Promise<unknown> };\n}\n\n/** Wrap an Anthropic client so `messages.create` auto-reports usage. */\nexport function wrapAnthropic<T extends AnthropicLike>(\n  client: T,\n  tollgate: TollgateClient,\n  opts: InstrumentOptions,\n): T {\n  const messages = client.messages;\n  const original = messages.create.bind(messages) as (...a: never[]) => Promise<unknown>;\n\n  const create = async (...args: never[]): Promise<unknown> => {\n    const result = await original(...args);\n    const event = anthropicEventFrom(result as AnthropicMessage, opts);\n    if (event) fireAndForget(tollgate.track(event), opts.onError);\n    return result;\n  };\n\n  return new Proxy(client, {\n    get(target, prop, recv) {\n      if (prop === 'messages') {\n        return new Proxy(messages, {\n          get: (m, p, r) => (p === 'create' ? create : Reflect.get(m, p, r)),\n        });\n      }\n      return Reflect.get(target, prop, recv);\n    },\n  });\n}\n\n// --- OpenAI ---------------------------------------------------------------\n\ninterface OpenAIUsage {\n  prompt_tokens?: number;\n  completion_tokens?: number;\n  completion_tokens_details?: { reasoning_tokens?: number };\n  prompt_tokens_details?: { cached_tokens?: number };\n}\ninterface OpenAICompletion {\n  id?: string;\n  model?: string;\n  usage?: OpenAIUsage;\n}\n\n/** Map a non-streaming OpenAI chat completion to a track payload (or null). */\nexport function openAIEventFrom(\n  completion: OpenAICompletion,\n  opts: InstrumentOptions,\n): TrackEventInput | null {\n  const usage = completion?.usage;\n  if (!usage) return null;\n  const runId = resolveRunId(opts, completion.id);\n  return {\n    customerId: opts.customerId,\n    agentId: opts.agentId,\n    runId,\n    provider: 'openai',\n    model: completion.model ?? 'unknown',\n    tokensIn: usage.prompt_tokens ?? 0,\n    tokensOut: usage.completion_tokens ?? 0,\n    reasoningTokens: usage.completion_tokens_details?.reasoning_tokens ?? 0,\n    cachedTokens: usage.prompt_tokens_details?.cached_tokens ?? 0,\n    revenueUnitCents: resolveRevenue(opts, completion),\n    idempotencyKey: completion.id ?? `${runId}#${randomId()}`,\n  };\n}\n\ninterface OpenAILike {\n  chat: { completions: { create: (...args: never[]) => Promise<unknown> } };\n}\n\n/** Wrap an OpenAI client so `chat.completions.create` auto-reports usage. */\nexport function wrapOpenAI<T extends OpenAILike>(\n  client: T,\n  tollgate: TollgateClient,\n  opts: InstrumentOptions,\n): T {\n  const completions = client.chat.completions;\n  const original = completions.create.bind(completions) as (...a: never[]) => Promise<unknown>;\n\n  const create = async (...args: never[]): Promise<unknown> => {\n    const result = await original(...args);\n    const event = openAIEventFrom(result as OpenAICompletion, opts);\n    if (event) fireAndForget(tollgate.track(event), opts.onError);\n    return result;\n  };\n\n  return new Proxy(client, {\n    get(target, prop, recv) {\n      if (prop === 'chat') {\n        const chat = Reflect.get(target, prop, recv) as OpenAILike['chat'];\n        return new Proxy(chat, {\n          get: (c, p, r) =>\n            p === 'completions'\n              ? new Proxy(completions, {\n                  get: (co, pp, rr) => (pp === 'create' ? create : Reflect.get(co, pp, rr)),\n                })\n              : Reflect.get(c, p, r),\n        });\n      }\n      return Reflect.get(target, prop, recv);\n    },\n  });\n}\n"]}

package/package.json

@@ -0,0 +1,44 @@
+{
+  "name": "@tollgateai/sdk",
+  "version": "0.1.0",
+  "description": "Track real LLM model usage and compute live gross margin with Tollgate.",
+  "license": "UNLICENSED",
+  "type": "module",
+  "main": "./dist/index.cjs",
+  "module": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js",
+      "require": "./dist/index.cjs"
+    }
+  },
+  "files": [
+    "dist",
+    "README.md"
+  ],
+  "sideEffects": false,
+  "scripts": {
+    "build": "tsup",
+    "typecheck": "tsc --noEmit",
+    "prepublishOnly": "npm run build"
+  },
+  "keywords": [
+    "llm",
+    "tokens",
+    "cost",
+    "margin",
+    "observability",
+    "anthropic",
+    "openai",
+    "tollgate"
+  ],
+  "engines": {
+    "node": ">=18"
+  },
+  "devDependencies": {
+    "tsup": "^8.3.5",
+    "typescript": "^5.7.3"
+  }
+}