cost-limiter 0.1.0

package/dist/index.cjs

@@ -0,0 +1,384 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/index.ts
+var src_exports = {};
+__export(src_exports, {
+  CostLimitError: () => CostLimitError,
+  CostLimiter: () => CostLimiter,
+  DEFAULT_PRICING: () => DEFAULT_PRICING,
+  MemoryCostStorage: () => MemoryCostStorage,
+  RedisCostStorage: () => RedisCostStorage,
+  WINDOWS: () => WINDOWS,
+  priceCall: () => priceCall,
+  windowBucket: () => windowBucket,
+  windowResetAt: () => windowResetAt
+});
+module.exports = __toCommonJS(src_exports);
+
+// src/limiter.ts
+var import_node_events = require("events");
+
+// src/errors.ts
+var CostLimitError = class _CostLimitError extends Error {
+  limit;
+  used;
+  remaining;
+  resetAt;
+  window;
+  dimension;
+  dimensionKey;
+  constructor(input) {
+    super(
+      `cost-limiter: budget exhausted for ${input.dimension}=${input.dimensionKey} (${input.window}: $${input.used.toFixed(4)} / $${input.limit.toFixed(4)})`
+    );
+    this.name = "CostLimitError";
+    this.limit = input.limit;
+    this.used = input.used;
+    this.remaining = Math.max(0, input.limit - input.used);
+    this.resetAt = input.resetAt;
+    this.window = input.window;
+    this.dimension = input.dimension;
+    this.dimensionKey = input.dimensionKey;
+    Object.setPrototypeOf(this, _CostLimitError.prototype);
+  }
+};
+
+// src/pricing.ts
+var DEFAULT_PRICING = {
+  // OpenAI
+  "gpt-4o": { inputPerMTokens: 2.5, outputPerMTokens: 10 },
+  "gpt-4o-mini": { inputPerMTokens: 0.15, outputPerMTokens: 0.6 },
+  "gpt-4-turbo": { inputPerMTokens: 10, outputPerMTokens: 30 },
+  "gpt-3.5-turbo": { inputPerMTokens: 0.5, outputPerMTokens: 1.5 },
+  "o1": { inputPerMTokens: 15, outputPerMTokens: 60 },
+  "o1-mini": { inputPerMTokens: 3, outputPerMTokens: 12 },
+  "o3-mini": { inputPerMTokens: 1.1, outputPerMTokens: 4.4 },
+  // Anthropic
+  "claude-opus-4": { inputPerMTokens: 15, outputPerMTokens: 75 },
+  "claude-sonnet-4": { inputPerMTokens: 3, outputPerMTokens: 15 },
+  "claude-haiku-4": { inputPerMTokens: 1, outputPerMTokens: 5 },
+  // Gemini
+  "gemini-2.0-flash": { inputPerMTokens: 0.1, outputPerMTokens: 0.4 },
+  "gemini-1.5-pro": { inputPerMTokens: 1.25, outputPerMTokens: 5 },
+  "gemini-1.5-flash": { inputPerMTokens: 0.075, outputPerMTokens: 0.3 },
+  // Groq
+  "llama-3.3-70b": { inputPerMTokens: 0.59, outputPerMTokens: 0.79 },
+  "mixtral-8x7b": { inputPerMTokens: 0.24, outputPerMTokens: 0.24 }
+};
+function priceCall(model, inputTokens, outputTokens, custom = {}) {
+  const p = custom[model] ?? DEFAULT_PRICING[model];
+  if (!p) return 0;
+  return (inputTokens * p.inputPerMTokens + outputTokens * p.outputPerMTokens) / 1e6;
+}
+
+// src/storage.ts
+var WINDOWS = ["minute", "hour", "day", "month"];
+function windowResetAt(now, window) {
+  const d = new Date(now);
+  switch (window) {
+    case "minute":
+      d.setUTCSeconds(0, 0);
+      d.setUTCMinutes(d.getUTCMinutes() + 1);
+      return d;
+    case "hour":
+      d.setUTCMinutes(0, 0, 0);
+      d.setUTCHours(d.getUTCHours() + 1);
+      return d;
+    case "day":
+      d.setUTCHours(0, 0, 0, 0);
+      d.setUTCDate(d.getUTCDate() + 1);
+      return d;
+    case "month":
+      d.setUTCHours(0, 0, 0, 0);
+      d.setUTCDate(1);
+      d.setUTCMonth(d.getUTCMonth() + 1);
+      return d;
+  }
+}
+function windowBucket(now, window) {
+  const y = now.getUTCFullYear();
+  const m = String(now.getUTCMonth() + 1).padStart(2, "0");
+  const d = String(now.getUTCDate()).padStart(2, "0");
+  const h = String(now.getUTCHours()).padStart(2, "0");
+  const min = String(now.getUTCMinutes()).padStart(2, "0");
+  switch (window) {
+    case "minute":
+      return `${y}${m}${d}${h}${min}`;
+    case "hour":
+      return `${y}${m}${d}${h}`;
+    case "day":
+      return `${y}${m}${d}`;
+    case "month":
+      return `${y}${m}`;
+  }
+}
+var MemoryCostStorage = class {
+  store = /* @__PURE__ */ new Map();
+  async increment(key, amount, window, resetAt) {
+    const fullKey = `${key}:${window}`;
+    const entry = this.store.get(fullKey);
+    if (entry && entry.resetAt <= Date.now()) {
+      this.store.delete(fullKey);
+    }
+    const cur = this.store.get(fullKey) ?? { value: 0, resetAt: resetAt.getTime() };
+    cur.value = +(cur.value + amount).toFixed(6);
+    this.store.set(fullKey, cur);
+    return cur.value;
+  }
+  async get(key, window) {
+    const entry = this.store.get(`${key}:${window}`);
+    if (!entry) return 0;
+    if (entry.resetAt <= Date.now()) {
+      this.store.delete(`${key}:${window}`);
+      return 0;
+    }
+    return entry.value;
+  }
+  async reset(key) {
+    for (const w of WINDOWS) this.store.delete(`${key}:${w}`);
+  }
+};
+var RedisCostStorage = class {
+  constructor(redis, prefix = "cost-limiter") {
+    this.redis = redis;
+    this.prefix = prefix;
+  }
+  redis;
+  prefix;
+  key(k, w) {
+    return `${this.prefix}:${k}:${w}`;
+  }
+  async increment(key, amount, window, resetAt) {
+    const k = this.key(key, window);
+    const newVal = await this.redis.incrbyfloat(k, amount);
+    const ttl = Math.max(1, resetAt.getTime() - Date.now());
+    await this.redis.pexpire(k, ttl);
+    return Number(newVal);
+  }
+  async get(key, window) {
+    const v = await this.redis.get(this.key(key, window));
+    return v ? Number(v) : 0;
+  }
+  async reset(key) {
+    await this.redis.del(...WINDOWS.map((w) => this.key(key, w)));
+  }
+};
+
+// src/limiter.ts
+var DIMENSION_PREFIX = {
+  user: "u",
+  team: "t",
+  apiKey: "k",
+  global: "g"
+};
+var CostLimiter = class extends import_node_events.EventEmitter {
+  storage;
+  budgets;
+  pricing;
+  warnThreshold;
+  constructor(opts = {}) {
+    super();
+    this.storage = opts.storage ?? new MemoryCostStorage();
+    this.budgets = opts.budgets ?? {};
+    this.pricing = opts.pricing === "auto" || !opts.pricing ? DEFAULT_PRICING : opts.pricing;
+    this.warnThreshold = opts.warnThreshold ?? 0.8;
+  }
+  dimensionKey(dim, key) {
+    return `${DIMENSION_PREFIX[dim]}:${key}`;
+  }
+  checks(input) {
+    const out = [];
+    if (input.userId && this.budgets.perUser) out.push({ dim: "user", key: input.userId, budget: this.budgets.perUser });
+    if (input.teamId && this.budgets.perTeam) out.push({ dim: "team", key: input.teamId, budget: this.budgets.perTeam });
+    if (input.apiKeyId && this.budgets.perApiKey) out.push({ dim: "apiKey", key: input.apiKeyId, budget: this.budgets.perApiKey });
+    if (this.budgets.global) out.push({ dim: "global", key: "global", budget: this.budgets.global });
+    return out;
+  }
+  /** Throws CostLimitError when any *remaining* budget can not absorb `costUsd`. */
+  async check(input) {
+    const cost = input.estimatedCostUsd ?? 0;
+    const now = /* @__PURE__ */ new Date();
+    for (const c of this.checks({ ...input, outputTokens: 0 })) {
+      for (const w of WINDOWS) {
+        const limit = c.budget[w];
+        if (limit === void 0) continue;
+        const fullKey = `${this.dimensionKey(c.dim, c.key)}:${windowBucket(now, w)}`;
+        const used = await this.storage.get(fullKey, w);
+        if (used + cost > limit) {
+          throw new CostLimitError({
+            limit,
+            used,
+            resetAt: windowResetAt(now, w),
+            window: w,
+            dimension: c.dim,
+            dimensionKey: c.key
+          });
+        }
+      }
+    }
+  }
+  async charge(input) {
+    const costUsd = input.customCostUsd ?? priceCall(input.model, input.inputTokens, input.outputTokens, this.pricing);
+    const now = /* @__PURE__ */ new Date();
+    for (const c of this.checks(input)) {
+      for (const w of WINDOWS) {
+        const limit = c.budget[w];
+        if (limit === void 0) continue;
+        const bucket = windowBucket(now, w);
+        const fullKey = `${this.dimensionKey(c.dim, c.key)}:${bucket}`;
+        const newTotal = await this.storage.increment(fullKey, costUsd, w, windowResetAt(now, w));
+        if (newTotal > limit) {
+          throw new CostLimitError({
+            limit,
+            used: newTotal,
+            resetAt: windowResetAt(now, w),
+            window: w,
+            dimension: c.dim,
+            dimensionKey: c.key
+          });
+        }
+        if (newTotal / limit >= this.warnThreshold) {
+          const ev = {
+            dimension: c.dim,
+            key: c.key,
+            window: w,
+            used: newTotal,
+            limit,
+            percent: newTotal / limit
+          };
+          this.emit("BudgetWarning", ev);
+        }
+      }
+    }
+    return { costUsd };
+  }
+  async getUsage(userId) {
+    const now = /* @__PURE__ */ new Date();
+    const spend = { minute: 0, hour: 0, day: 0, month: 0 };
+    const limit = { minute: void 0, hour: void 0, day: void 0, month: void 0 };
+    for (const w of WINDOWS) {
+      const key = `${this.dimensionKey("user", userId)}:${windowBucket(now, w)}`;
+      spend[w] = await this.storage.get(key, w);
+      limit[w] = this.budgets.perUser?.[w];
+    }
+    return { dimension: "user", key: userId, spend, limit };
+  }
+  async resetUsage(userId) {
+    const now = /* @__PURE__ */ new Date();
+    for (const w of WINDOWS) {
+      const key = `${this.dimensionKey("user", userId)}:${windowBucket(now, w)}`;
+      await this.storage.reset(key);
+    }
+  }
+  /** Wrap an OpenAI v4 client so usage is charged automatically. */
+  wrap(client, ctx) {
+    return new Proxy(client, this.makeHandler("openai"));
+  }
+  /** Wrap an Anthropic SDK client. */
+  wrapAnthropic(client) {
+    return new Proxy(client, this.makeHandler("anthropic"));
+  }
+  /** Express/Hono/Fastify-style middleware: parses `userId`/`teamId` off `req` before allowing the request. */
+  middleware(getDims) {
+    return async (req, _res, next) => {
+      try {
+        const { estimatedCostUsd, ...dims } = getDims(req);
+        await this.check({
+          ...dims,
+          provider: "openai",
+          model: "gpt-4o-mini",
+          inputTokens: 0,
+          ...estimatedCostUsd !== void 0 ? { estimatedCostUsd } : {}
+        });
+        next();
+      } catch (err) {
+        next(err);
+      }
+    };
+  }
+  makeHandler(provider) {
+    const limiter = this;
+    const handler = {
+      get(target, prop, receiver) {
+        const value = Reflect.get(target, prop, receiver);
+        if (typeof value === "object" && value !== null) {
+          return new Proxy(value, handler);
+        }
+        if (typeof value === "function") {
+          return async (req) => {
+            const result = await value.call(target, stripDims(req));
+            try {
+              const usage = extractUsage(result, provider);
+              if (usage) {
+                await limiter.charge({
+                  ...req.userId !== void 0 ? { userId: req.userId } : {},
+                  ...req.teamId !== void 0 ? { teamId: req.teamId } : {},
+                  ...req.apiKeyId !== void 0 ? { apiKeyId: req.apiKeyId } : {},
+                  provider,
+                  model: String(req.model ?? usage.model ?? "unknown"),
+                  inputTokens: usage.input,
+                  outputTokens: usage.output
+                });
+              }
+            } catch (err) {
+              if (err instanceof CostLimitError) throw err;
+            }
+            return result;
+          };
+        }
+        return value;
+      }
+    };
+    return handler;
+  }
+};
+function stripDims(req) {
+  const { userId: _u, teamId: _t, apiKeyId: _k, ...rest } = req;
+  return rest;
+}
+function extractUsage(result, provider) {
+  const r = result;
+  if (!r?.usage) return void 0;
+  if (provider === "openai") {
+    return {
+      input: r.usage.prompt_tokens ?? 0,
+      output: r.usage.completion_tokens ?? 0,
+      ...r.model ? { model: r.model } : {}
+    };
+  }
+  return {
+    input: r.usage.input_tokens ?? 0,
+    output: r.usage.output_tokens ?? 0,
+    ...r.model ? { model: r.model } : {}
+  };
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  CostLimitError,
+  CostLimiter,
+  DEFAULT_PRICING,
+  MemoryCostStorage,
+  RedisCostStorage,
+  WINDOWS,
+  priceCall,
+  windowBucket,
+  windowResetAt
+});
+//# sourceMappingURL=index.cjs.map

package/dist/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/index.ts","../src/limiter.ts","../src/errors.ts","../src/pricing.ts","../src/storage.ts"],"sourcesContent":["export { CostLimiter } from \"./limiter.js\";\nexport type { CostLimiterOptions, ChargeInput } from \"./limiter.js\";\nexport { CostLimitError } from \"./errors.js\";\nexport {\n  MemoryCostStorage,\n  RedisCostStorage,\n  WINDOWS,\n  windowBucket,\n  windowResetAt,\n} from \"./storage.js\";\nexport type { StorageAdapter } from \"./storage.js\";\nexport {\n  DEFAULT_PRICING,\n  priceCall,\n} from \"./pricing.js\";\nexport type { ModelPrice, PricingConfig } from \"./pricing.js\";\nexport type {\n  BudgetConfig,\n  WindowBudget,\n  Window,\n  Dimension,\n  UsageReport,\n  BudgetWarningEvent,\n} from \"./types.js\";\n","import { EventEmitter } from \"node:events\";\nimport { CostLimitError } from \"./errors.js\";\nimport { DEFAULT_PRICING, priceCall, type ModelPrice } from \"./pricing.js\";\nimport {\n  MemoryCostStorage,\n  type StorageAdapter,\n  WINDOWS,\n  windowBucket,\n  windowResetAt,\n} from \"./storage.js\";\nimport type {\n  BudgetConfig,\n  BudgetWarningEvent,\n  Dimension,\n  UsageReport,\n  Window,\n  WindowBudget,\n} from \"./types.js\";\n\nexport interface CostLimiterOptions {\n  budgets?: BudgetConfig;\n  storage?: StorageAdapter;\n  pricing?: Record<string, ModelPrice> | \"auto\";\n  /** Soft-warn threshold (0-1). Defaults to 0.8 (80%). */\n  warnThreshold?: number;\n}\n\nexport interface ChargeInput {\n  userId?: string;\n  teamId?: string;\n  apiKeyId?: string;\n  provider: \"openai\" | \"anthropic\" | string;\n  model: string;\n  inputTokens: number;\n  outputTokens: number;\n  customCostUsd?: number;\n}\n\nconst DIMENSION_PREFIX: Record<\"user\" | \"team\" | \"apiKey\" | \"global\", string> = {\n  user: \"u\",\n  team: \"t\",\n  apiKey: \"k\",\n  global: \"g\",\n};\n\nexport class CostLimiter extends EventEmitter {\n  readonly storage: StorageAdapter;\n  readonly budgets: BudgetConfig;\n  readonly pricing: Record<string, ModelPrice>;\n  readonly warnThreshold: number;\n\n  constructor(opts: CostLimiterOptions = {}) {\n    super();\n    this.storage = opts.storage ?? new MemoryCostStorage();\n    this.budgets = opts.budgets ?? {};\n    this.pricing = opts.pricing === \"auto\" || !opts.pricing ? DEFAULT_PRICING : opts.pricing;\n    this.warnThreshold = opts.warnThreshold ?? 0.8;\n  }\n\n  private dimensionKey(dim: keyof typeof DIMENSION_PREFIX, key: string): string {\n    return `${DIMENSION_PREFIX[dim]}:${key}`;\n  }\n\n  private checks(input: ChargeInput): { dim: keyof typeof DIMENSION_PREFIX; key: string; budget: WindowBudget }[] {\n    const out: { dim: keyof typeof DIMENSION_PREFIX; key: string; budget: WindowBudget }[] = [];\n    if (input.userId && this.budgets.perUser) out.push({ dim: \"user\", key: input.userId, budget: this.budgets.perUser });\n    if (input.teamId && this.budgets.perTeam) out.push({ dim: \"team\", key: input.teamId, budget: this.budgets.perTeam });\n    if (input.apiKeyId && this.budgets.perApiKey) out.push({ dim: \"apiKey\", key: input.apiKeyId, budget: this.budgets.perApiKey });\n    if (this.budgets.global) out.push({ dim: \"global\", key: \"global\", budget: this.budgets.global });\n    return out;\n  }\n\n  /** Throws CostLimitError when any *remaining* budget can not absorb `costUsd`. */\n  async check(input: Omit<ChargeInput, \"outputTokens\"> & { estimatedCostUsd?: number }): Promise<void> {\n    const cost = input.estimatedCostUsd ?? 0;\n    const now = new Date();\n    for (const c of this.checks({ ...input, outputTokens: 0 })) {\n      for (const w of WINDOWS) {\n        const limit = c.budget[w];\n        if (limit === undefined) continue;\n        const fullKey = `${this.dimensionKey(c.dim, c.key)}:${windowBucket(now, w)}`;\n        const used = await this.storage.get(fullKey, w);\n        if (used + cost > limit) {\n          throw new CostLimitError({\n            limit,\n            used,\n            resetAt: windowResetAt(now, w),\n            window: w,\n            dimension: c.dim,\n            dimensionKey: c.key,\n          });\n        }\n      }\n    }\n  }\n\n  async charge(input: ChargeInput): Promise<{ costUsd: number }> {\n    const costUsd = input.customCostUsd ?? priceCall(input.model, input.inputTokens, input.outputTokens, this.pricing);\n    const now = new Date();\n    for (const c of this.checks(input)) {\n      for (const w of WINDOWS) {\n        const limit = c.budget[w];\n        if (limit === undefined) continue;\n        const bucket = windowBucket(now, w);\n        const fullKey = `${this.dimensionKey(c.dim, c.key)}:${bucket}`;\n        const newTotal = await this.storage.increment(fullKey, costUsd, w, windowResetAt(now, w));\n        if (newTotal > limit) {\n          throw new CostLimitError({\n            limit,\n            used: newTotal,\n            resetAt: windowResetAt(now, w),\n            window: w,\n            dimension: c.dim,\n            dimensionKey: c.key,\n          });\n        }\n        if (newTotal / limit >= this.warnThreshold) {\n          const ev: BudgetWarningEvent = {\n            dimension: c.dim,\n            key: c.key,\n            window: w,\n            used: newTotal,\n            limit,\n            percent: newTotal / limit,\n          };\n          this.emit(\"BudgetWarning\", ev);\n        }\n      }\n    }\n    return { costUsd };\n  }\n\n  async getUsage(userId: string): Promise<UsageReport> {\n    const now = new Date();\n    const spend: Record<Window, number> = { minute: 0, hour: 0, day: 0, month: 0 };\n    const limit: Record<Window, number | undefined> = { minute: undefined, hour: undefined, day: undefined, month: undefined };\n    for (const w of WINDOWS) {\n      const key = `${this.dimensionKey(\"user\", userId)}:${windowBucket(now, w)}`;\n      spend[w] = await this.storage.get(key, w);\n      limit[w] = this.budgets.perUser?.[w];\n    }\n    return { dimension: \"user\", key: userId, spend, limit };\n  }\n\n  async resetUsage(userId: string): Promise<void> {\n    const now = new Date();\n    for (const w of WINDOWS) {\n      const key = `${this.dimensionKey(\"user\", userId)}:${windowBucket(now, w)}`;\n      await this.storage.reset(key);\n    }\n  }\n\n  /** Wrap an OpenAI v4 client so usage is charged automatically. */\n  wrap<T extends OpenAILike>(client: T, ctx?: { provider?: \"openai\" }): T {\n    return new Proxy(client, this.makeHandler(\"openai\")) as T;\n  }\n\n  /** Wrap an Anthropic SDK client. */\n  wrapAnthropic<T extends AnthropicLike>(client: T): T {\n    return new Proxy(client, this.makeHandler(\"anthropic\")) as T;\n  }\n\n  /** Express/Hono/Fastify-style middleware: parses `userId`/`teamId` off `req` before allowing the request. */\n  middleware(getDims: (req: unknown) => { userId?: string; teamId?: string; apiKeyId?: string; estimatedCostUsd?: number }) {\n    return async (req: unknown, _res: unknown, next: (err?: unknown) => void) => {\n      try {\n        const { estimatedCostUsd, ...dims } = getDims(req);\n        await this.check({\n          ...dims,\n          provider: \"openai\",\n          model: \"gpt-4o-mini\",\n          inputTokens: 0,\n          ...(estimatedCostUsd !== undefined ? { estimatedCostUsd } : {}),\n        });\n        next();\n      } catch (err) {\n        next(err);\n      }\n    };\n  }\n\n  private makeHandler(provider: \"openai\" | \"anthropic\"): ProxyHandler<object> {\n    const limiter = this;\n    const handler: ProxyHandler<object> = {\n      get(target, prop, receiver) {\n        const value = Reflect.get(target, prop, receiver);\n        if (typeof value === \"object\" && value !== null) {\n          return new Proxy(value, handler);\n        }\n        if (typeof value === \"function\") {\n          return async (req: Record<string, unknown> & { userId?: string; teamId?: string; apiKeyId?: string; model?: string }) => {\n            const result = await (value as Function).call(target, stripDims(req));\n            try {\n              const usage = extractUsage(result, provider);\n              if (usage) {\n                await limiter.charge({\n                  ...(req.userId !== undefined ? { userId: req.userId } : {}),\n                  ...(req.teamId !== undefined ? { teamId: req.teamId } : {}),\n                  ...(req.apiKeyId !== undefined ? { apiKeyId: req.apiKeyId } : {}),\n                  provider,\n                  model: String(req.model ?? usage.model ?? \"unknown\"),\n                  inputTokens: usage.input,\n                  outputTokens: usage.output,\n                });\n              }\n            } catch (err) {\n              if (err instanceof CostLimitError) throw err;\n            }\n            return result;\n          };\n        }\n        return value;\n      },\n    };\n    return handler;\n  }\n}\n\ninterface OpenAILike { chat?: unknown }\ninterface AnthropicLike { messages?: unknown }\n\nfunction stripDims<T extends Record<string, unknown>>(req: T): Omit<T, \"userId\" | \"teamId\" | \"apiKeyId\"> {\n  const { userId: _u, teamId: _t, apiKeyId: _k, ...rest } = req;\n  return rest as Omit<T, \"userId\" | \"teamId\" | \"apiKeyId\">;\n}\n\nfunction extractUsage(result: unknown, provider: \"openai\" | \"anthropic\"): { input: number; output: number; model?: string } | undefined {\n  const r = result as { usage?: { prompt_tokens?: number; completion_tokens?: number; input_tokens?: number; output_tokens?: number }; model?: string } | null;\n  if (!r?.usage) return undefined;\n  if (provider === \"openai\") {\n    return {\n      input: r.usage.prompt_tokens ?? 0,\n      output: r.usage.completion_tokens ?? 0,\n      ...(r.model ? { model: r.model } : {}),\n    };\n  }\n  return {\n    input: r.usage.input_tokens ?? 0,\n    output: r.usage.output_tokens ?? 0,\n    ...(r.model ? { model: r.model } : {}),\n  };\n}\n","import type { Window, Dimension } from \"./types.js\";\n\nexport class CostLimitError extends Error {\n  readonly limit: number;\n  readonly used: number;\n  readonly remaining: number;\n  readonly resetAt: Date;\n  readonly window: Window;\n  readonly dimension: Dimension;\n  readonly dimensionKey: string;\n\n  constructor(input: {\n    limit: number;\n    used: number;\n    resetAt: Date;\n    window: Window;\n    dimension: Dimension;\n    dimensionKey: string;\n  }) {\n    super(\n      `cost-limiter: budget exhausted for ${input.dimension}=${input.dimensionKey} ` +\n        `(${input.window}: $${input.used.toFixed(4)} / $${input.limit.toFixed(4)})`,\n    );\n    this.name = \"CostLimitError\";\n    this.limit = input.limit;\n    this.used = input.used;\n    this.remaining = Math.max(0, input.limit - input.used);\n    this.resetAt = input.resetAt;\n    this.window = input.window;\n    this.dimension = input.dimension;\n    this.dimensionKey = input.dimensionKey;\n    Object.setPrototypeOf(this, CostLimitError.prototype);\n  }\n}\n","export interface ModelPrice {\n  /** Dollar cost per 1,000,000 input tokens. */\n  inputPerMTokens: number;\n  /** Dollar cost per 1,000,000 output tokens. */\n  outputPerMTokens: number;\n}\n\nexport const DEFAULT_PRICING: Record<string, ModelPrice> = {\n  // OpenAI\n  \"gpt-4o\": { inputPerMTokens: 2.50, outputPerMTokens: 10.00 },\n  \"gpt-4o-mini\": { inputPerMTokens: 0.15, outputPerMTokens: 0.60 },\n  \"gpt-4-turbo\": { inputPerMTokens: 10.00, outputPerMTokens: 30.00 },\n  \"gpt-3.5-turbo\": { inputPerMTokens: 0.50, outputPerMTokens: 1.50 },\n  \"o1\": { inputPerMTokens: 15.00, outputPerMTokens: 60.00 },\n  \"o1-mini\": { inputPerMTokens: 3.00, outputPerMTokens: 12.00 },\n  \"o3-mini\": { inputPerMTokens: 1.10, outputPerMTokens: 4.40 },\n  // Anthropic\n  \"claude-opus-4\": { inputPerMTokens: 15.00, outputPerMTokens: 75.00 },\n  \"claude-sonnet-4\": { inputPerMTokens: 3.00, outputPerMTokens: 15.00 },\n  \"claude-haiku-4\": { inputPerMTokens: 1.00, outputPerMTokens: 5.00 },\n  // Gemini\n  \"gemini-2.0-flash\": { inputPerMTokens: 0.10, outputPerMTokens: 0.40 },\n  \"gemini-1.5-pro\": { inputPerMTokens: 1.25, outputPerMTokens: 5.00 },\n  \"gemini-1.5-flash\": { inputPerMTokens: 0.075, outputPerMTokens: 0.30 },\n  // Groq\n  \"llama-3.3-70b\": { inputPerMTokens: 0.59, outputPerMTokens: 0.79 },\n  \"mixtral-8x7b\": { inputPerMTokens: 0.24, outputPerMTokens: 0.24 },\n};\n\nexport interface PricingConfig {\n  pricing?: Record<string, ModelPrice>;\n}\n\nexport function priceCall(model: string, inputTokens: number, outputTokens: number, custom: Record<string, ModelPrice> = {}): number {\n  const p = custom[model] ?? DEFAULT_PRICING[model];\n  if (!p) return 0;\n  return (inputTokens * p.inputPerMTokens + outputTokens * p.outputPerMTokens) / 1_000_000;\n}\n","import type { Window } from \"./types.js\";\n\nexport interface StorageAdapter {\n  /** Increment the spend counter for the given key/window and return the new total. */\n  increment(key: string, amount: number, window: Window, resetAt: Date): Promise<number>;\n  /** Read the current spend for the given key/window. */\n  get(key: string, window: Window): Promise<number>;\n  /** Reset (clear) all windows for the given key. */\n  reset(key: string): Promise<void>;\n}\n\nexport const WINDOWS: Window[] = [\"minute\", \"hour\", \"day\", \"month\"];\n\nexport function windowResetAt(now: Date, window: Window): Date {\n  const d = new Date(now);\n  switch (window) {\n    case \"minute\":\n      d.setUTCSeconds(0, 0);\n      d.setUTCMinutes(d.getUTCMinutes() + 1);\n      return d;\n    case \"hour\":\n      d.setUTCMinutes(0, 0, 0);\n      d.setUTCHours(d.getUTCHours() + 1);\n      return d;\n    case \"day\":\n      d.setUTCHours(0, 0, 0, 0);\n      d.setUTCDate(d.getUTCDate() + 1);\n      return d;\n    case \"month\":\n      d.setUTCHours(0, 0, 0, 0);\n      d.setUTCDate(1);\n      d.setUTCMonth(d.getUTCMonth() + 1);\n      return d;\n  }\n}\n\nexport function windowBucket(now: Date, window: Window): string {\n  const y = now.getUTCFullYear();\n  const m = String(now.getUTCMonth() + 1).padStart(2, \"0\");\n  const d = String(now.getUTCDate()).padStart(2, \"0\");\n  const h = String(now.getUTCHours()).padStart(2, \"0\");\n  const min = String(now.getUTCMinutes()).padStart(2, \"0\");\n  switch (window) {\n    case \"minute\": return `${y}${m}${d}${h}${min}`;\n    case \"hour\":   return `${y}${m}${d}${h}`;\n    case \"day\":    return `${y}${m}${d}`;\n    case \"month\":  return `${y}${m}`;\n  }\n}\n\nexport class MemoryCostStorage implements StorageAdapter {\n  private store = new Map<string, { value: number; resetAt: number }>();\n\n  async increment(key: string, amount: number, window: Window, resetAt: Date) {\n    const fullKey = `${key}:${window}`;\n    const entry = this.store.get(fullKey);\n    if (entry && entry.resetAt <= Date.now()) {\n      this.store.delete(fullKey);\n    }\n    const cur = this.store.get(fullKey) ?? { value: 0, resetAt: resetAt.getTime() };\n    cur.value = +(cur.value + amount).toFixed(6);\n    this.store.set(fullKey, cur);\n    return cur.value;\n  }\n\n  async get(key: string, window: Window): Promise<number> {\n    const entry = this.store.get(`${key}:${window}`);\n    if (!entry) return 0;\n    if (entry.resetAt <= Date.now()) {\n      this.store.delete(`${key}:${window}`);\n      return 0;\n    }\n    return entry.value;\n  }\n\n  async reset(key: string): Promise<void> {\n    for (const w of WINDOWS) this.store.delete(`${key}:${w}`);\n  }\n}\n\ninterface RedisLike {\n  incrbyfloat(key: string, amount: number): Promise<string>;\n  pexpire(key: string, ms: number): Promise<number>;\n  get(key: string): Promise<string | null>;\n  del(...keys: string[]): Promise<number>;\n}\n\nexport class RedisCostStorage implements StorageAdapter {\n  constructor(private redis: RedisLike, private prefix = \"cost-limiter\") {}\n\n  private key(k: string, w: Window) { return `${this.prefix}:${k}:${w}`; }\n\n  async increment(key: string, amount: number, window: Window, resetAt: Date) {\n    const k = this.key(key, window);\n    const newVal = await this.redis.incrbyfloat(k, amount);\n    const ttl = Math.max(1, resetAt.getTime() - Date.now());\n    await this.redis.pexpire(k, ttl);\n    return Number(newVal);\n  }\n\n  async get(key: string, window: Window) {\n    const v = await this.redis.get(this.key(key, window));\n    return v ? Number(v) : 0;\n  }\n\n  async reset(key: string) {\n    await this.redis.del(...WINDOWS.map((w) => this.key(key, w)));\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,yBAA6B;;;ACEtB,IAAM,iBAAN,MAAM,wBAAuB,MAAM;AAAA,EAC/B;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAET,YAAY,OAOT;AACD;AAAA,MACE,sCAAsC,MAAM,SAAS,IAAI,MAAM,YAAY,KACrE,MAAM,MAAM,MAAM,MAAM,KAAK,QAAQ,CAAC,CAAC,OAAO,MAAM,MAAM,QAAQ,CAAC,CAAC;AAAA,IAC5E;AACA,SAAK,OAAO;AACZ,SAAK,QAAQ,MAAM;AACnB,SAAK,OAAO,MAAM;AAClB,SAAK,YAAY,KAAK,IAAI,GAAG,MAAM,QAAQ,MAAM,IAAI;AACrD,SAAK,UAAU,MAAM;AACrB,SAAK,SAAS,MAAM;AACpB,SAAK,YAAY,MAAM;AACvB,SAAK,eAAe,MAAM;AAC1B,WAAO,eAAe,MAAM,gBAAe,SAAS;AAAA,EACtD;AACF;;;AC1BO,IAAM,kBAA8C;AAAA;AAAA,EAEzD,UAAU,EAAE,iBAAiB,KAAM,kBAAkB,GAAM;AAAA,EAC3D,eAAe,EAAE,iBAAiB,MAAM,kBAAkB,IAAK;AAAA,EAC/D,eAAe,EAAE,iBAAiB,IAAO,kBAAkB,GAAM;AAAA,EACjE,iBAAiB,EAAE,iBAAiB,KAAM,kBAAkB,IAAK;AAAA,EACjE,MAAM,EAAE,iBAAiB,IAAO,kBAAkB,GAAM;AAAA,EACxD,WAAW,EAAE,iBAAiB,GAAM,kBAAkB,GAAM;AAAA,EAC5D,WAAW,EAAE,iBAAiB,KAAM,kBAAkB,IAAK;AAAA;AAAA,EAE3D,iBAAiB,EAAE,iBAAiB,IAAO,kBAAkB,GAAM;AAAA,EACnE,mBAAmB,EAAE,iBAAiB,GAAM,kBAAkB,GAAM;AAAA,EACpE,kBAAkB,EAAE,iBAAiB,GAAM,kBAAkB,EAAK;AAAA;AAAA,EAElE,oBAAoB,EAAE,iBAAiB,KAAM,kBAAkB,IAAK;AAAA,EACpE,kBAAkB,EAAE,iBAAiB,MAAM,kBAAkB,EAAK;AAAA,EAClE,oBAAoB,EAAE,iBAAiB,OAAO,kBAAkB,IAAK;AAAA;AAAA,EAErE,iBAAiB,EAAE,iBAAiB,MAAM,kBAAkB,KAAK;AAAA,EACjE,gBAAgB,EAAE,iBAAiB,MAAM,kBAAkB,KAAK;AAClE;AAMO,SAAS,UAAU,OAAe,aAAqB,cAAsB,SAAqC,CAAC,GAAW;AACnI,QAAM,IAAI,OAAO,KAAK,KAAK,gBAAgB,KAAK;AAChD,MAAI,CAAC,EAAG,QAAO;AACf,UAAQ,cAAc,EAAE,kBAAkB,eAAe,EAAE,oBAAoB;AACjF;;;AC1BO,IAAM,UAAoB,CAAC,UAAU,QAAQ,OAAO,OAAO;AAE3D,SAAS,cAAc,KAAW,QAAsB;AAC7D,QAAM,IAAI,IAAI,KAAK,GAAG;AACtB,UAAQ,QAAQ;AAAA,IACd,KAAK;AACH,QAAE,cAAc,GAAG,CAAC;AACpB,QAAE,cAAc,EAAE,cAAc,IAAI,CAAC;AACrC,aAAO;AAAA,IACT,KAAK;AACH,QAAE,cAAc,GAAG,GAAG,CAAC;AACvB,QAAE,YAAY,EAAE,YAAY,IAAI,CAAC;AACjC,aAAO;AAAA,IACT,KAAK;AACH,QAAE,YAAY,GAAG,GAAG,GAAG,CAAC;AACxB,QAAE,WAAW,EAAE,WAAW,IAAI,CAAC;AAC/B,aAAO;AAAA,IACT,KAAK;AACH,QAAE,YAAY,GAAG,GAAG,GAAG,CAAC;AACxB,QAAE,WAAW,CAAC;AACd,QAAE,YAAY,EAAE,YAAY,IAAI,CAAC;AACjC,aAAO;AAAA,EACX;AACF;AAEO,SAAS,aAAa,KAAW,QAAwB;AAC9D,QAAM,IAAI,IAAI,eAAe;AAC7B,QAAM,IAAI,OAAO,IAAI,YAAY,IAAI,CAAC,EAAE,SAAS,GAAG,GAAG;AACvD,QAAM,IAAI,OAAO,IAAI,WAAW,CAAC,EAAE,SAAS,GAAG,GAAG;AAClD,QAAM,IAAI,OAAO,IAAI,YAAY,CAAC,EAAE,SAAS,GAAG,GAAG;AACnD,QAAM,MAAM,OAAO,IAAI,cAAc,CAAC,EAAE,SAAS,GAAG,GAAG;AACvD,UAAQ,QAAQ;AAAA,IACd,KAAK;AAAU,aAAO,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,GAAG;AAAA,IAC5C,KAAK;AAAU,aAAO,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC;AAAA,IACtC,KAAK;AAAU,aAAO,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC;AAAA,IAClC,KAAK;AAAU,aAAO,GAAG,CAAC,GAAG,CAAC;AAAA,EAChC;AACF;AAEO,IAAM,oBAAN,MAAkD;AAAA,EAC/C,QAAQ,oBAAI,IAAgD;AAAA,EAEpE,MAAM,UAAU,KAAa,QAAgB,QAAgB,SAAe;AAC1E,UAAM,UAAU,GAAG,GAAG,IAAI,MAAM;AAChC,UAAM,QAAQ,KAAK,MAAM,IAAI,OAAO;AACpC,QAAI,SAAS,MAAM,WAAW,KAAK,IAAI,GAAG;AACxC,WAAK,MAAM,OAAO,OAAO;AAAA,IAC3B;AACA,UAAM,MAAM,KAAK,MAAM,IAAI,OAAO,KAAK,EAAE,OAAO,GAAG,SAAS,QAAQ,QAAQ,EAAE;AAC9E,QAAI,QAAQ,EAAE,IAAI,QAAQ,QAAQ,QAAQ,CAAC;AAC3C,SAAK,MAAM,IAAI,SAAS,GAAG;AAC3B,WAAO,IAAI;AAAA,EACb;AAAA,EAEA,MAAM,IAAI,KAAa,QAAiC;AACtD,UAAM,QAAQ,KAAK,MAAM,IAAI,GAAG,GAAG,IAAI,MAAM,EAAE;AAC/C,QAAI,CAAC,MAAO,QAAO;AACnB,QAAI,MAAM,WAAW,KAAK,IAAI,GAAG;AAC/B,WAAK,MAAM,OAAO,GAAG,GAAG,IAAI,MAAM,EAAE;AACpC,aAAO;AAAA,IACT;AACA,WAAO,MAAM;AAAA,EACf;AAAA,EAEA,MAAM,MAAM,KAA4B;AACtC,eAAW,KAAK,QAAS,MAAK,MAAM,OAAO,GAAG,GAAG,IAAI,CAAC,EAAE;AAAA,EAC1D;AACF;AASO,IAAM,mBAAN,MAAiD;AAAA,EACtD,YAAoB,OAA0B,SAAS,gBAAgB;AAAnD;AAA0B;AAAA,EAA0B;AAAA,EAApD;AAAA,EAA0B;AAAA,EAEtC,IAAI,GAAW,GAAW;AAAE,WAAO,GAAG,KAAK,MAAM,IAAI,CAAC,IAAI,CAAC;AAAA,EAAI;AAAA,EAEvE,MAAM,UAAU,KAAa,QAAgB,QAAgB,SAAe;AAC1E,UAAM,IAAI,KAAK,IAAI,KAAK,MAAM;AAC9B,UAAM,SAAS,MAAM,KAAK,MAAM,YAAY,GAAG,MAAM;AACrD,UAAM,MAAM,KAAK,IAAI,GAAG,QAAQ,QAAQ,IAAI,KAAK,IAAI,CAAC;AACtD,UAAM,KAAK,MAAM,QAAQ,GAAG,GAAG;AAC/B,WAAO,OAAO,MAAM;AAAA,EACtB;AAAA,EAEA,MAAM,IAAI,KAAa,QAAgB;AACrC,UAAM,IAAI,MAAM,KAAK,MAAM,IAAI,KAAK,IAAI,KAAK,MAAM,CAAC;AACpD,WAAO,IAAI,OAAO,CAAC,IAAI;AAAA,EACzB;AAAA,EAEA,MAAM,MAAM,KAAa;AACvB,UAAM,KAAK,MAAM,IAAI,GAAG,QAAQ,IAAI,CAAC,MAAM,KAAK,IAAI,KAAK,CAAC,CAAC,CAAC;AAAA,EAC9D;AACF;;;AHtEA,IAAM,mBAA0E;AAAA,EAC9E,MAAM;AAAA,EACN,MAAM;AAAA,EACN,QAAQ;AAAA,EACR,QAAQ;AACV;AAEO,IAAM,cAAN,cAA0B,gCAAa;AAAA,EACnC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAET,YAAY,OAA2B,CAAC,GAAG;AACzC,UAAM;AACN,SAAK,UAAU,KAAK,WAAW,IAAI,kBAAkB;AACrD,SAAK,UAAU,KAAK,WAAW,CAAC;AAChC,SAAK,UAAU,KAAK,YAAY,UAAU,CAAC,KAAK,UAAU,kBAAkB,KAAK;AACjF,SAAK,gBAAgB,KAAK,iBAAiB;AAAA,EAC7C;AAAA,EAEQ,aAAa,KAAoC,KAAqB;AAC5E,WAAO,GAAG,iBAAiB,GAAG,CAAC,IAAI,GAAG;AAAA,EACxC;AAAA,EAEQ,OAAO,OAAiG;AAC9G,UAAM,MAAmF,CAAC;AAC1F,QAAI,MAAM,UAAU,KAAK,QAAQ,QAAS,KAAI,KAAK,EAAE,KAAK,QAAQ,KAAK,MAAM,QAAQ,QAAQ,KAAK,QAAQ,QAAQ,CAAC;AACnH,QAAI,MAAM,UAAU,KAAK,QAAQ,QAAS,KAAI,KAAK,EAAE,KAAK,QAAQ,KAAK,MAAM,QAAQ,QAAQ,KAAK,QAAQ,QAAQ,CAAC;AACnH,QAAI,MAAM,YAAY,KAAK,QAAQ,UAAW,KAAI,KAAK,EAAE,KAAK,UAAU,KAAK,MAAM,UAAU,QAAQ,KAAK,QAAQ,UAAU,CAAC;AAC7H,QAAI,KAAK,QAAQ,OAAQ,KAAI,KAAK,EAAE,KAAK,UAAU,KAAK,UAAU,QAAQ,KAAK,QAAQ,OAAO,CAAC;AAC/F,WAAO;AAAA,EACT;AAAA;AAAA,EAGA,MAAM,MAAM,OAAyF;AACnG,UAAM,OAAO,MAAM,oBAAoB;AACvC,UAAM,MAAM,oBAAI,KAAK;AACrB,eAAW,KAAK,KAAK,OAAO,EAAE,GAAG,OAAO,cAAc,EAAE,CAAC,GAAG;AAC1D,iBAAW,KAAK,SAAS;AACvB,cAAM,QAAQ,EAAE,OAAO,CAAC;AACxB,YAAI,UAAU,OAAW;AACzB,cAAM,UAAU,GAAG,KAAK,aAAa,EAAE,KAAK,EAAE,GAAG,CAAC,IAAI,aAAa,KAAK,CAAC,CAAC;AAC1E,cAAM,OAAO,MAAM,KAAK,QAAQ,IAAI,SAAS,CAAC;AAC9C,YAAI,OAAO,OAAO,OAAO;AACvB,gBAAM,IAAI,eAAe;AAAA,YACvB;AAAA,YACA;AAAA,YACA,SAAS,cAAc,KAAK,CAAC;AAAA,YAC7B,QAAQ;AAAA,YACR,WAAW,EAAE;AAAA,YACb,cAAc,EAAE;AAAA,UAClB,CAAC;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,OAAO,OAAkD;AAC7D,UAAM,UAAU,MAAM,iBAAiB,UAAU,MAAM,OAAO,MAAM,aAAa,MAAM,cAAc,KAAK,OAAO;AACjH,UAAM,MAAM,oBAAI,KAAK;AACrB,eAAW,KAAK,KAAK,OAAO,KAAK,GAAG;AAClC,iBAAW,KAAK,SAAS;AACvB,cAAM,QAAQ,EAAE,OAAO,CAAC;AACxB,YAAI,UAAU,OAAW;AACzB,cAAM,SAAS,aAAa,KAAK,CAAC;AAClC,cAAM,UAAU,GAAG,KAAK,aAAa,EAAE,KAAK,EAAE,GAAG,CAAC,IAAI,MAAM;AAC5D,cAAM,WAAW,MAAM,KAAK,QAAQ,UAAU,SAAS,SAAS,GAAG,cAAc,KAAK,CAAC,CAAC;AACxF,YAAI,WAAW,OAAO;AACpB,gBAAM,IAAI,eAAe;AAAA,YACvB;AAAA,YACA,MAAM;AAAA,YACN,SAAS,cAAc,KAAK,CAAC;AAAA,YAC7B,QAAQ;AAAA,YACR,WAAW,EAAE;AAAA,YACb,cAAc,EAAE;AAAA,UAClB,CAAC;AAAA,QACH;AACA,YAAI,WAAW,SAAS,KAAK,eAAe;AAC1C,gBAAM,KAAyB;AAAA,YAC7B,WAAW,EAAE;AAAA,YACb,KAAK,EAAE;AAAA,YACP,QAAQ;AAAA,YACR,MAAM;AAAA,YACN;AAAA,YACA,SAAS,WAAW;AAAA,UACtB;AACA,eAAK,KAAK,iBAAiB,EAAE;AAAA,QAC/B;AAAA,MACF;AAAA,IACF;AACA,WAAO,EAAE,QAAQ;AAAA,EACnB;AAAA,EAEA,MAAM,SAAS,QAAsC;AACnD,UAAM,MAAM,oBAAI,KAAK;AACrB,UAAM,QAAgC,EAAE,QAAQ,GAAG,MAAM,GAAG,KAAK,GAAG,OAAO,EAAE;AAC7E,UAAM,QAA4C,EAAE,QAAQ,QAAW,MAAM,QAAW,KAAK,QAAW,OAAO,OAAU;AACzH,eAAW,KAAK,SAAS;AACvB,YAAM,MAAM,GAAG,KAAK,aAAa,QAAQ,MAAM,CAAC,IAAI,aAAa,KAAK,CAAC,CAAC;AACxE,YAAM,CAAC,IAAI,MAAM,KAAK,QAAQ,IAAI,KAAK,CAAC;AACxC,YAAM,CAAC,IAAI,KAAK,QAAQ,UAAU,CAAC;AAAA,IACrC;AACA,WAAO,EAAE,WAAW,QAAQ,KAAK,QAAQ,OAAO,MAAM;AAAA,EACxD;AAAA,EAEA,MAAM,WAAW,QAA+B;AAC9C,UAAM,MAAM,oBAAI,KAAK;AACrB,eAAW,KAAK,SAAS;AACvB,YAAM,MAAM,GAAG,KAAK,aAAa,QAAQ,MAAM,CAAC,IAAI,aAAa,KAAK,CAAC,CAAC;AACxE,YAAM,KAAK,QAAQ,MAAM,GAAG;AAAA,IAC9B;AAAA,EACF;AAAA;AAAA,EAGA,KAA2B,QAAW,KAAkC;AACtE,WAAO,IAAI,MAAM,QAAQ,KAAK,YAAY,QAAQ,CAAC;AAAA,EACrD;AAAA;AAAA,EAGA,cAAuC,QAAc;AACnD,WAAO,IAAI,MAAM,QAAQ,KAAK,YAAY,WAAW,CAAC;AAAA,EACxD;AAAA;AAAA,EAGA,WAAW,SAA+G;AACxH,WAAO,OAAO,KAAc,MAAe,SAAkC;AAC3E,UAAI;AACF,cAAM,EAAE,kBAAkB,GAAG,KAAK,IAAI,QAAQ,GAAG;AACjD,cAAM,KAAK,MAAM;AAAA,UACf,GAAG;AAAA,UACH,UAAU;AAAA,UACV,OAAO;AAAA,UACP,aAAa;AAAA,UACb,GAAI,qBAAqB,SAAY,EAAE,iBAAiB,IAAI,CAAC;AAAA,QAC/D,CAAC;AACD,aAAK;AAAA,MACP,SAAS,KAAK;AACZ,aAAK,GAAG;AAAA,MACV;AAAA,IACF;AAAA,EACF;AAAA,EAEQ,YAAY,UAAwD;AAC1E,UAAM,UAAU;AAChB,UAAM,UAAgC;AAAA,MACpC,IAAI,QAAQ,MAAM,UAAU;AAC1B,cAAM,QAAQ,QAAQ,IAAI,QAAQ,MAAM,QAAQ;AAChD,YAAI,OAAO,UAAU,YAAY,UAAU,MAAM;AAC/C,iBAAO,IAAI,MAAM,OAAO,OAAO;AAAA,QACjC;AACA,YAAI,OAAO,UAAU,YAAY;AAC/B,iBAAO,OAAO,QAA2G;AACvH,kBAAM,SAAS,MAAO,MAAmB,KAAK,QAAQ,UAAU,GAAG,CAAC;AACpE,gBAAI;AACF,oBAAM,QAAQ,aAAa,QAAQ,QAAQ;AAC3C,kBAAI,OAAO;AACT,sBAAM,QAAQ,OAAO;AAAA,kBACnB,GAAI,IAAI,WAAW,SAAY,EAAE,QAAQ,IAAI,OAAO,IAAI,CAAC;AAAA,kBACzD,GAAI,IAAI,WAAW,SAAY,EAAE,QAAQ,IAAI,OAAO,IAAI,CAAC;AAAA,kBACzD,GAAI,IAAI,aAAa,SAAY,EAAE,UAAU,IAAI,SAAS,IAAI,CAAC;AAAA,kBAC/D;AAAA,kBACA,OAAO,OAAO,IAAI,SAAS,MAAM,SAAS,SAAS;AAAA,kBACnD,aAAa,MAAM;AAAA,kBACnB,cAAc,MAAM;AAAA,gBACtB,CAAC;AAAA,cACH;AAAA,YACF,SAAS,KAAK;AACZ,kBAAI,eAAe,eAAgB,OAAM;AAAA,YAC3C;AACA,mBAAO;AAAA,UACT;AAAA,QACF;AACA,eAAO;AAAA,MACT;AAAA,IACF;AACA,WAAO;AAAA,EACT;AACF;AAKA,SAAS,UAA6C,KAAmD;AACvG,QAAM,EAAE,QAAQ,IAAI,QAAQ,IAAI,UAAU,IAAI,GAAG,KAAK,IAAI;AAC1D,SAAO;AACT;AAEA,SAAS,aAAa,QAAiB,UAAiG;AACtI,QAAM,IAAI;AACV,MAAI,CAAC,GAAG,MAAO,QAAO;AACtB,MAAI,aAAa,UAAU;AACzB,WAAO;AAAA,MACL,OAAO,EAAE,MAAM,iBAAiB;AAAA,MAChC,QAAQ,EAAE,MAAM,qBAAqB;AAAA,MACrC,GAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,IAAI,CAAC;AAAA,IACtC;AAAA,EACF;AACA,SAAO;AAAA,IACL,OAAO,EAAE,MAAM,gBAAgB;AAAA,IAC/B,QAAQ,EAAE,MAAM,iBAAiB;AAAA,IACjC,GAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,IAAI,CAAC;AAAA,EACtC;AACF;","names":[]}

package/dist/index.d.cts

@@ -0,0 +1,151 @@
+import { EventEmitter } from 'node:events';
+
+interface ModelPrice {
+    /** Dollar cost per 1,000,000 input tokens. */
+    inputPerMTokens: number;
+    /** Dollar cost per 1,000,000 output tokens. */
+    outputPerMTokens: number;
+}
+declare const DEFAULT_PRICING: Record<string, ModelPrice>;
+interface PricingConfig {
+    pricing?: Record<string, ModelPrice>;
+}
+declare function priceCall(model: string, inputTokens: number, outputTokens: number, custom?: Record<string, ModelPrice>): number;
+
+type Dimension = "user" | "team" | "apiKey" | "global" | string;
+type Window = "minute" | "hour" | "day" | "month";
+interface WindowBudget {
+    minute?: number;
+    hour?: number;
+    day?: number;
+    month?: number;
+}
+interface BudgetConfig {
+    perUser?: WindowBudget;
+    perTeam?: WindowBudget;
+    perApiKey?: WindowBudget;
+    global?: WindowBudget;
+}
+interface UsageReport {
+    dimension: Dimension;
+    key: string;
+    spend: Record<Window, number>;
+    limit: Record<Window, number | undefined>;
+}
+interface BudgetWarningEvent {
+    dimension: Dimension;
+    key: string;
+    window: Window;
+    used: number;
+    limit: number;
+    percent: number;
+}
+
+interface StorageAdapter {
+    /** Increment the spend counter for the given key/window and return the new total. */
+    increment(key: string, amount: number, window: Window, resetAt: Date): Promise<number>;
+    /** Read the current spend for the given key/window. */
+    get(key: string, window: Window): Promise<number>;
+    /** Reset (clear) all windows for the given key. */
+    reset(key: string): Promise<void>;
+}
+declare const WINDOWS: Window[];
+declare function windowResetAt(now: Date, window: Window): Date;
+declare function windowBucket(now: Date, window: Window): string;
+declare class MemoryCostStorage implements StorageAdapter {
+    private store;
+    increment(key: string, amount: number, window: Window, resetAt: Date): Promise<number>;
+    get(key: string, window: Window): Promise<number>;
+    reset(key: string): Promise<void>;
+}
+interface RedisLike {
+    incrbyfloat(key: string, amount: number): Promise<string>;
+    pexpire(key: string, ms: number): Promise<number>;
+    get(key: string): Promise<string | null>;
+    del(...keys: string[]): Promise<number>;
+}
+declare class RedisCostStorage implements StorageAdapter {
+    private redis;
+    private prefix;
+    constructor(redis: RedisLike, prefix?: string);
+    private key;
+    increment(key: string, amount: number, window: Window, resetAt: Date): Promise<number>;
+    get(key: string, window: Window): Promise<number>;
+    reset(key: string): Promise<void>;
+}
+
+interface CostLimiterOptions {
+    budgets?: BudgetConfig;
+    storage?: StorageAdapter;
+    pricing?: Record<string, ModelPrice> | "auto";
+    /** Soft-warn threshold (0-1). Defaults to 0.8 (80%). */
+    warnThreshold?: number;
+}
+interface ChargeInput {
+    userId?: string;
+    teamId?: string;
+    apiKeyId?: string;
+    provider: "openai" | "anthropic" | string;
+    model: string;
+    inputTokens: number;
+    outputTokens: number;
+    customCostUsd?: number;
+}
+declare class CostLimiter extends EventEmitter {
+    readonly storage: StorageAdapter;
+    readonly budgets: BudgetConfig;
+    readonly pricing: Record<string, ModelPrice>;
+    readonly warnThreshold: number;
+    constructor(opts?: CostLimiterOptions);
+    private dimensionKey;
+    private checks;
+    /** Throws CostLimitError when any *remaining* budget can not absorb `costUsd`. */
+    check(input: Omit<ChargeInput, "outputTokens"> & {
+        estimatedCostUsd?: number;
+    }): Promise<void>;
+    charge(input: ChargeInput): Promise<{
+        costUsd: number;
+    }>;
+    getUsage(userId: string): Promise<UsageReport>;
+    resetUsage(userId: string): Promise<void>;
+    /** Wrap an OpenAI v4 client so usage is charged automatically. */
+    wrap<T extends OpenAILike>(client: T, ctx?: {
+        provider?: "openai";
+    }): T;
+    /** Wrap an Anthropic SDK client. */
+    wrapAnthropic<T extends AnthropicLike>(client: T): T;
+    /** Express/Hono/Fastify-style middleware: parses `userId`/`teamId` off `req` before allowing the request. */
+    middleware(getDims: (req: unknown) => {
+        userId?: string;
+        teamId?: string;
+        apiKeyId?: string;
+        estimatedCostUsd?: number;
+    }): (req: unknown, _res: unknown, next: (err?: unknown) => void) => Promise<void>;
+    private makeHandler;
+}
+interface OpenAILike {
+    chat?: unknown;
+}
+interface AnthropicLike {
+    messages?: unknown;
+}
+
+declare class CostLimitError extends Error {
+    readonly limit: number;
+    readonly used: number;
+    readonly remaining: number;
+    readonly resetAt: Date;
+    readonly window: Window;
+    readonly dimension: Dimension;
+    readonly dimensionKey: string;
+    constructor(input: {
+        limit: number;
+        used: number;
+        resetAt: Date;
+        window: Window;
+        dimension: Dimension;
+        dimensionKey: string;
+    });
+}
+
+export { type BudgetConfig, type BudgetWarningEvent, type ChargeInput, CostLimitError, CostLimiter, type CostLimiterOptions, DEFAULT_PRICING, type Dimension, MemoryCostStorage, type ModelPrice, type PricingConfig, RedisCostStorage, type StorageAdapter, type UsageReport, WINDOWS, type Window, type WindowBudget, priceCall, windowBucket, windowResetAt };