@diogonzafe/tokenwatch 0.1.6 → 0.1.8

package/dist/adapters.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/adapters/postgres.ts","../src/adapters/mysql.ts","../src/adapters/mongodb.ts"],"sourcesContent":["import type { IStorage, UsageEntry } from '../types/index.js'\n\n/**\n * IStorage adapter for PostgreSQL using the `pg` driver.\n *\n * Install peer dep:  npm install pg\n * Types (optional):  npm install -D @types/pg\n *\n * @example\n * ```ts\n * import { Pool } from 'pg'\n * import { createTracker } from '@diogonzafe/tokenwatch'\n * import { PostgresStorage } from '@diogonzafe/tokenwatch/adapters'\n *\n * const pool = new Pool({ connectionString: process.env.DATABASE_URL })\n * const storage = new PostgresStorage(pool)\n * await storage.migrate()          // create table if it doesn't exist\n *\n * const tracker = createTracker({ storage })\n * ```\n */\n\n// Minimal structural types so the adapter compiles without `pg` installed\ninterface QueryClient {\n  query(sql: string, values?: unknown[]): Promise<{ rows: unknown[] }>\n}\n\nexport class PostgresStorage implements IStorage {\n  constructor(private readonly client: QueryClient) {}\n\n  /** Creates the `tokenwatch_usage` table if it does not already exist. */\n  async migrate(): Promise<void> {\n    await this.client.query(`\n      CREATE TABLE IF NOT EXISTS tokenwatch_usage (\n        id            BIGSERIAL PRIMARY KEY,\n        model         TEXT      NOT NULL,\n        input_tokens  INTEGER   NOT NULL,\n        output_tokens INTEGER   NOT NULL,\n        cost_usd      NUMERIC   NOT NULL,\n        session_id    TEXT,\n        user_id       TEXT,\n        timestamp     TIMESTAMPTZ NOT NULL\n      )\n    `)\n  }\n\n  record(entry: UsageEntry): void {\n    this.client\n      .query(\n        `INSERT INTO tokenwatch_usage\n         (model, input_tokens, output_tokens, cost_usd, session_id, user_id, timestamp)\n         VALUES ($1, $2, $3, $4, $5, $6, $7)`,\n        [\n          entry.model,\n          entry.inputTokens,\n          entry.outputTokens,\n          entry.costUSD,\n          entry.sessionId ?? null,\n          entry.userId ?? null,\n          entry.timestamp,\n        ],\n      )\n      .catch((err: unknown) => {\n        console.warn('[tokenwatch] PostgresStorage.record failed:', err)\n      })\n  }\n\n  async getAll(): Promise<UsageEntry[]> {\n    const result = await this.client.query(\n      'SELECT * FROM tokenwatch_usage ORDER BY timestamp ASC',\n    )\n    return (result.rows as Array<Record<string, unknown>>).map(rowToEntry)\n  }\n\n  async clearAll(): Promise<void> {\n    await this.client.query('DELETE FROM tokenwatch_usage')\n  }\n\n  async clearSession(sessionId: string): Promise<void> {\n    await this.client.query(\n      'DELETE FROM tokenwatch_usage WHERE session_id = $1',\n      [sessionId],\n    )\n  }\n}\n\nfunction rowToEntry(r: Record<string, unknown>): UsageEntry {\n  return {\n    model: r['model'] as string,\n    inputTokens: r['input_tokens'] as number,\n    outputTokens: r['output_tokens'] as number,\n    costUSD: Number(r['cost_usd']),\n    ...(r['session_id'] != null && { sessionId: r['session_id'] as string }),\n    ...(r['user_id'] != null && { userId: r['user_id'] as string }),\n    timestamp:\n      r['timestamp'] instanceof Date\n        ? (r['timestamp'] as Date).toISOString()\n        : (r['timestamp'] as string),\n  }\n}\n","import type { IStorage, UsageEntry } from '../types/index.js'\n\n/**\n * IStorage adapter for MySQL / MariaDB using the `mysql2` driver.\n *\n * Install peer dep:  npm install mysql2\n *\n * @example\n * ```ts\n * import mysql from 'mysql2/promise'\n * import { createTracker } from '@diogonzafe/tokenwatch'\n * import { MySQLStorage } from '@diogonzafe/tokenwatch/adapters'\n *\n * const pool = mysql.createPool({ uri: process.env.MYSQL_URL })\n * const storage = new MySQLStorage(pool)\n * await storage.migrate()          // create table if it doesn't exist\n *\n * const tracker = createTracker({ storage })\n * ```\n */\n\n// Minimal structural type so the adapter compiles without `mysql2` installed\ninterface QueryClient {\n  execute(sql: string, values?: unknown[]): Promise<[unknown]>\n}\n\nexport class MySQLStorage implements IStorage {\n  constructor(private readonly client: QueryClient) {}\n\n  /** Creates the `tokenwatch_usage` table if it does not already exist. */\n  async migrate(): Promise<void> {\n    await this.client.execute(`\n      CREATE TABLE IF NOT EXISTS tokenwatch_usage (\n        id            BIGINT UNSIGNED AUTO_INCREMENT PRIMARY KEY,\n        model         VARCHAR(255) NOT NULL,\n        input_tokens  INT          NOT NULL,\n        output_tokens INT          NOT NULL,\n        cost_usd      DECIMAL(18,8) NOT NULL,\n        session_id    VARCHAR(255),\n        user_id       VARCHAR(255),\n        timestamp     DATETIME(3)  NOT NULL\n      )\n    `)\n  }\n\n  record(entry: UsageEntry): void {\n    this.client\n      .execute(\n        `INSERT INTO tokenwatch_usage\n         (model, input_tokens, output_tokens, cost_usd, session_id, user_id, timestamp)\n         VALUES (?, ?, ?, ?, ?, ?, ?)`,\n        [\n          entry.model,\n          entry.inputTokens,\n          entry.outputTokens,\n          entry.costUSD,\n          entry.sessionId ?? null,\n          entry.userId ?? null,\n          entry.timestamp,\n        ],\n      )\n      .catch((err: unknown) => {\n        console.warn('[tokenwatch] MySQLStorage.record failed:', err)\n      })\n  }\n\n  async getAll(): Promise<UsageEntry[]> {\n    const [rows] = await this.client.execute(\n      'SELECT * FROM tokenwatch_usage ORDER BY timestamp ASC',\n    )\n    return (rows as Array<Record<string, unknown>>).map(rowToEntry)\n  }\n\n  async clearAll(): Promise<void> {\n    await this.client.execute('DELETE FROM tokenwatch_usage')\n  }\n\n  async clearSession(sessionId: string): Promise<void> {\n    await this.client.execute(\n      'DELETE FROM tokenwatch_usage WHERE session_id = ?',\n      [sessionId],\n    )\n  }\n}\n\nfunction rowToEntry(r: Record<string, unknown>): UsageEntry {\n  return {\n    model: r['model'] as string,\n    inputTokens: r['input_tokens'] as number,\n    outputTokens: r['output_tokens'] as number,\n    costUSD: Number(r['cost_usd']),\n    ...(r['session_id'] != null && { sessionId: r['session_id'] as string }),\n    ...(r['user_id'] != null && { userId: r['user_id'] as string }),\n    timestamp:\n      r['timestamp'] instanceof Date\n        ? (r['timestamp'] as Date).toISOString()\n        : (r['timestamp'] as string),\n  }\n}\n","import type { IStorage, UsageEntry } from '../types/index.js'\n\n/**\n * IStorage adapter for MongoDB using the official `mongodb` driver.\n *\n * Install peer dep:  npm install mongodb\n *\n * @example\n * ```ts\n * import { MongoClient } from 'mongodb'\n * import { createTracker } from '@diogonzafe/tokenwatch'\n * import { MongoStorage } from '@diogonzafe/tokenwatch/adapters'\n *\n * const client = new MongoClient(process.env.MONGO_URL!)\n * await client.connect()\n *\n * const storage = new MongoStorage(client.db('myapp'))\n * const tracker = createTracker({ storage })\n * ```\n *\n * Recommended index (run once at startup):\n * ```ts\n * await storage.createIndexes()\n * ```\n */\n\n// Minimal structural types so the adapter compiles without `mongodb` installed\ninterface MongoDocument {\n  _id?: unknown\n  model: string\n  inputTokens: number\n  outputTokens: number\n  costUSD: number\n  sessionId?: string | null\n  userId?: string | null\n  timestamp: string\n}\n\ninterface Collection {\n  insertOne(doc: MongoDocument): Promise<unknown>\n  find(filter: Record<string, unknown>): { toArray(): Promise<MongoDocument[]> }\n  deleteMany(filter: Record<string, unknown>): Promise<unknown>\n  createIndex(index: Record<string, unknown>): Promise<unknown>\n}\n\ninterface Database {\n  collection(name: string): Collection\n}\n\nconst COLLECTION = 'tokenwatch_usage'\n\nexport class MongoStorage implements IStorage {\n  private readonly col: Collection\n\n  constructor(db: Database) {\n    this.col = db.collection(COLLECTION)\n  }\n\n  /** Creates recommended indexes for query performance. Call once at startup. */\n  async createIndexes(): Promise<void> {\n    await this.col.createIndex({ timestamp: 1 })\n    await this.col.createIndex({ sessionId: 1 })\n    await this.col.createIndex({ userId: 1 })\n    await this.col.createIndex({ model: 1 })\n  }\n\n  record(entry: UsageEntry): void {\n    this.col\n      .insertOne({\n        model: entry.model,\n        inputTokens: entry.inputTokens,\n        outputTokens: entry.outputTokens,\n        costUSD: entry.costUSD,\n        sessionId: entry.sessionId ?? null,\n        userId: entry.userId ?? null,\n        timestamp: entry.timestamp,\n      })\n      .catch((err: unknown) => {\n        console.warn('[tokenwatch] MongoStorage.record failed:', err)\n      })\n  }\n\n  async getAll(): Promise<UsageEntry[]> {\n    const docs = await this.col.find({}).toArray()\n    return docs.map(docToEntry)\n  }\n\n  async clearAll(): Promise<void> {\n    await this.col.deleteMany({})\n  }\n\n  async clearSession(sessionId: string): Promise<void> {\n    await this.col.deleteMany({ sessionId })\n  }\n}\n\nfunction docToEntry(doc: MongoDocument): UsageEntry {\n  return {\n    model: doc.model,\n    inputTokens: doc.inputTokens,\n    outputTokens: doc.outputTokens,\n    costUSD: doc.costUSD,\n    ...(doc.sessionId != null && { sessionId: doc.sessionId }),\n    ...(doc.userId != null && { userId: doc.userId }),\n    timestamp: doc.timestamp,\n  }\n}\n"],"mappings":";AA2BO,IAAM,kBAAN,MAA0C;AAAA,EAC/C,YAA6B,QAAqB;AAArB;AAAA,EAAsB;AAAA,EAAtB;AAAA;AAAA,EAG7B,MAAM,UAAyB;AAC7B,UAAM,KAAK,OAAO,MAAM;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,KAWvB;AAAA,EACH;AAAA,EAEA,OAAO,OAAyB;AAC9B,SAAK,OACF;AAAA,MACC;AAAA;AAAA;AAAA,MAGA;AAAA,QACE,MAAM;AAAA,QACN,MAAM;AAAA,QACN,MAAM;AAAA,QACN,MAAM;AAAA,QACN,MAAM,aAAa;AAAA,QACnB,MAAM,UAAU;AAAA,QAChB,MAAM;AAAA,MACR;AAAA,IACF,EACC,MAAM,CAAC,QAAiB;AACvB,cAAQ,KAAK,+CAA+C,GAAG;AAAA,IACjE,CAAC;AAAA,EACL;AAAA,EAEA,MAAM,SAAgC;AACpC,UAAM,SAAS,MAAM,KAAK,OAAO;AAAA,MAC/B;AAAA,IACF;AACA,WAAQ,OAAO,KAAwC,IAAI,UAAU;AAAA,EACvE;AAAA,EAEA,MAAM,WAA0B;AAC9B,UAAM,KAAK,OAAO,MAAM,8BAA8B;AAAA,EACxD;AAAA,EAEA,MAAM,aAAa,WAAkC;AACnD,UAAM,KAAK,OAAO;AAAA,MAChB;AAAA,MACA,CAAC,SAAS;AAAA,IACZ;AAAA,EACF;AACF;AAEA,SAAS,WAAW,GAAwC;AAC1D,SAAO;AAAA,IACL,OAAO,EAAE,OAAO;AAAA,IAChB,aAAa,EAAE,cAAc;AAAA,IAC7B,cAAc,EAAE,eAAe;AAAA,IAC/B,SAAS,OAAO,EAAE,UAAU,CAAC;AAAA,IAC7B,GAAI,EAAE,YAAY,KAAK,QAAQ,EAAE,WAAW,EAAE,YAAY,EAAY;AAAA,IACtE,GAAI,EAAE,SAAS,KAAK,QAAQ,EAAE,QAAQ,EAAE,SAAS,EAAY;AAAA,IAC7D,WACE,EAAE,WAAW,aAAa,OACrB,EAAE,WAAW,EAAW,YAAY,IACpC,EAAE,WAAW;AAAA,EACtB;AACF;;;ACzEO,IAAM,eAAN,MAAuC;AAAA,EAC5C,YAA6B,QAAqB;AAArB;AAAA,EAAsB;AAAA,EAAtB;AAAA;AAAA,EAG7B,MAAM,UAAyB;AAC7B,UAAM,KAAK,OAAO,QAAQ;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,KAWzB;AAAA,EACH;AAAA,EAEA,OAAO,OAAyB;AAC9B,SAAK,OACF;AAAA,MACC;AAAA;AAAA;AAAA,MAGA;AAAA,QACE,MAAM;AAAA,QACN,MAAM;AAAA,QACN,MAAM;AAAA,QACN,MAAM;AAAA,QACN,MAAM,aAAa;AAAA,QACnB,MAAM,UAAU;AAAA,QAChB,MAAM;AAAA,MACR;AAAA,IACF,EACC,MAAM,CAAC,QAAiB;AACvB,cAAQ,KAAK,4CAA4C,GAAG;AAAA,IAC9D,CAAC;AAAA,EACL;AAAA,EAEA,MAAM,SAAgC;AACpC,UAAM,CAAC,IAAI,IAAI,MAAM,KAAK,OAAO;AAAA,MAC/B;AAAA,IACF;AACA,WAAQ,KAAwC,IAAIA,WAAU;AAAA,EAChE;AAAA,EAEA,MAAM,WAA0B;AAC9B,UAAM,KAAK,OAAO,QAAQ,8BAA8B;AAAA,EAC1D;AAAA,EAEA,MAAM,aAAa,WAAkC;AACnD,UAAM,KAAK,OAAO;AAAA,MAChB;AAAA,MACA,CAAC,SAAS;AAAA,IACZ;AAAA,EACF;AACF;AAEA,SAASA,YAAW,GAAwC;AAC1D,SAAO;AAAA,IACL,OAAO,EAAE,OAAO;AAAA,IAChB,aAAa,EAAE,cAAc;AAAA,IAC7B,cAAc,EAAE,eAAe;AAAA,IAC/B,SAAS,OAAO,EAAE,UAAU,CAAC;AAAA,IAC7B,GAAI,EAAE,YAAY,KAAK,QAAQ,EAAE,WAAW,EAAE,YAAY,EAAY;AAAA,IACtE,GAAI,EAAE,SAAS,KAAK,QAAQ,EAAE,QAAQ,EAAE,SAAS,EAAY;AAAA,IAC7D,WACE,EAAE,WAAW,aAAa,OACrB,EAAE,WAAW,EAAW,YAAY,IACpC,EAAE,WAAW;AAAA,EACtB;AACF;;;ACjDA,IAAM,aAAa;AAEZ,IAAM,eAAN,MAAuC;AAAA,EAC3B;AAAA,EAEjB,YAAY,IAAc;AACxB,SAAK,MAAM,GAAG,WAAW,UAAU;AAAA,EACrC;AAAA;AAAA,EAGA,MAAM,gBAA+B;AACnC,UAAM,KAAK,IAAI,YAAY,EAAE,WAAW,EAAE,CAAC;AAC3C,UAAM,KAAK,IAAI,YAAY,EAAE,WAAW,EAAE,CAAC;AAC3C,UAAM,KAAK,IAAI,YAAY,EAAE,QAAQ,EAAE,CAAC;AACxC,UAAM,KAAK,IAAI,YAAY,EAAE,OAAO,EAAE,CAAC;AAAA,EACzC;AAAA,EAEA,OAAO,OAAyB;AAC9B,SAAK,IACF,UAAU;AAAA,MACT,OAAO,MAAM;AAAA,MACb,aAAa,MAAM;AAAA,MACnB,cAAc,MAAM;AAAA,MACpB,SAAS,MAAM;AAAA,MACf,WAAW,MAAM,aAAa;AAAA,MAC9B,QAAQ,MAAM,UAAU;AAAA,MACxB,WAAW,MAAM;AAAA,IACnB,CAAC,EACA,MAAM,CAAC,QAAiB;AACvB,cAAQ,KAAK,4CAA4C,GAAG;AAAA,IAC9D,CAAC;AAAA,EACL;AAAA,EAEA,MAAM,SAAgC;AACpC,UAAM,OAAO,MAAM,KAAK,IAAI,KAAK,CAAC,CAAC,EAAE,QAAQ;AAC7C,WAAO,KAAK,IAAI,UAAU;AAAA,EAC5B;AAAA,EAEA,MAAM,WAA0B;AAC9B,UAAM,KAAK,IAAI,WAAW,CAAC,CAAC;AAAA,EAC9B;AAAA,EAEA,MAAM,aAAa,WAAkC;AACnD,UAAM,KAAK,IAAI,WAAW,EAAE,UAAU,CAAC;AAAA,EACzC;AACF;AAEA,SAAS,WAAW,KAAgC;AAClD,SAAO;AAAA,IACL,OAAO,IAAI;AAAA,IACX,aAAa,IAAI;AAAA,IACjB,cAAc,IAAI;AAAA,IAClB,SAAS,IAAI;AAAA,IACb,GAAI,IAAI,aAAa,QAAQ,EAAE,WAAW,IAAI,UAAU;AAAA,IACxD,GAAI,IAAI,UAAU,QAAQ,EAAE,QAAQ,IAAI,OAAO;AAAA,IAC/C,WAAW,IAAI;AAAA,EACjB;AACF;","names":["rowToEntry"]}

package/dist/index-Cy_sl3FI.d.cts

@@ -0,0 +1,89 @@
+interface ModelPrice {
+    /** USD per 1 million input tokens */
+    input: number;
+    /** USD per 1 million output tokens */
+    output: number;
+    /** Maximum context window (input tokens) for this model */
+    maxInputTokens?: number;
+}
+type PriceMap = Record<string, ModelPrice>;
+interface PricesFile {
+    updated_at: string;
+    source: string;
+    models: PriceMap;
+}
+interface TrackerConfig {
+    /** 'memory' (default), 'sqlite', or a custom IStorage instance (e.g. PostgresStorage, MySQLStorage, MongoStorage) */
+    storage?: 'memory' | 'sqlite' | IStorage;
+    /** USD threshold — fires webhookUrl when totalCostUSD exceeds this */
+    alertThreshold?: number;
+    /** Discord / Slack / generic webhook URL */
+    webhookUrl?: string;
+    /** Fetch fresh prices from remote GitHub source (default: true) */
+    syncPrices?: boolean;
+    /** Per-model price overrides — highest priority */
+    customPrices?: PriceMap;
+}
+interface UsageEntry {
+    model: string;
+    inputTokens: number;
+    outputTokens: number;
+    costUSD: number;
+    sessionId?: string;
+    userId?: string;
+    timestamp: string;
+}
+interface ModelStats {
+    costUSD: number;
+    calls: number;
+    tokens: {
+        input: number;
+        output: number;
+    };
+}
+interface SessionStats {
+    costUSD: number;
+    calls: number;
+}
+interface UserStats {
+    costUSD: number;
+    calls: number;
+}
+interface Report {
+    totalCostUSD: number;
+    totalTokens: {
+        input: number;
+        output: number;
+    };
+    byModel: Record<string, ModelStats>;
+    bySession: Record<string, SessionStats>;
+    byUser: Record<string, UserStats>;
+    period: {
+        from: string;
+        to: string;
+    };
+}
+interface IStorage {
+    /** Fire-and-forget — implementations may write async and swallow errors internally */
+    record(entry: UsageEntry): void | Promise<void>;
+    getAll(): UsageEntry[] | Promise<UsageEntry[]>;
+    clearAll(): void | Promise<void>;
+    clearSession(sessionId: string): void | Promise<void>;
+}
+interface Tracker {
+    /** Accumulate a usage entry (called by providers) */
+    track(entry: Omit<UsageEntry, 'costUSD' | 'timestamp'>): void;
+    getReport(): Promise<Report>;
+    reset(): Promise<void>;
+    resetSession(sessionId: string): Promise<void>;
+    exportJSON(): Promise<string>;
+    exportCSV(): Promise<string>;
+    /** Returns price and context window info for a model, or null if unknown */
+    getModelInfo(model: string): ModelPrice | null;
+}
+interface TrackingMeta {
+    __sessionId?: string;
+    __userId?: string;
+}
+
+export type { IStorage as I, ModelPrice as M, PriceMap as P, Report as R, SessionStats as S, TrackerConfig as T, UsageEntry as U, Tracker as a, TrackingMeta as b, ModelStats as c, PricesFile as d, UserStats as e };

package/dist/index-Cy_sl3FI.d.ts

@@ -0,0 +1,89 @@
+interface ModelPrice {
+    /** USD per 1 million input tokens */
+    input: number;
+    /** USD per 1 million output tokens */
+    output: number;
+    /** Maximum context window (input tokens) for this model */
+    maxInputTokens?: number;
+}
+type PriceMap = Record<string, ModelPrice>;
+interface PricesFile {
+    updated_at: string;
+    source: string;
+    models: PriceMap;
+}
+interface TrackerConfig {
+    /** 'memory' (default), 'sqlite', or a custom IStorage instance (e.g. PostgresStorage, MySQLStorage, MongoStorage) */
+    storage?: 'memory' | 'sqlite' | IStorage;
+    /** USD threshold — fires webhookUrl when totalCostUSD exceeds this */
+    alertThreshold?: number;
+    /** Discord / Slack / generic webhook URL */
+    webhookUrl?: string;
+    /** Fetch fresh prices from remote GitHub source (default: true) */
+    syncPrices?: boolean;
+    /** Per-model price overrides — highest priority */
+    customPrices?: PriceMap;
+}
+interface UsageEntry {
+    model: string;
+    inputTokens: number;
+    outputTokens: number;
+    costUSD: number;
+    sessionId?: string;
+    userId?: string;
+    timestamp: string;
+}
+interface ModelStats {
+    costUSD: number;
+    calls: number;
+    tokens: {
+        input: number;
+        output: number;
+    };
+}
+interface SessionStats {
+    costUSD: number;
+    calls: number;
+}
+interface UserStats {
+    costUSD: number;
+    calls: number;
+}
+interface Report {
+    totalCostUSD: number;
+    totalTokens: {
+        input: number;
+        output: number;
+    };
+    byModel: Record<string, ModelStats>;
+    bySession: Record<string, SessionStats>;
+    byUser: Record<string, UserStats>;
+    period: {
+        from: string;
+        to: string;
+    };
+}
+interface IStorage {
+    /** Fire-and-forget — implementations may write async and swallow errors internally */
+    record(entry: UsageEntry): void | Promise<void>;
+    getAll(): UsageEntry[] | Promise<UsageEntry[]>;
+    clearAll(): void | Promise<void>;
+    clearSession(sessionId: string): void | Promise<void>;
+}
+interface Tracker {
+    /** Accumulate a usage entry (called by providers) */
+    track(entry: Omit<UsageEntry, 'costUSD' | 'timestamp'>): void;
+    getReport(): Promise<Report>;
+    reset(): Promise<void>;
+    resetSession(sessionId: string): Promise<void>;
+    exportJSON(): Promise<string>;
+    exportCSV(): Promise<string>;
+    /** Returns price and context window info for a model, or null if unknown */
+    getModelInfo(model: string): ModelPrice | null;
+}
+interface TrackingMeta {
+    __sessionId?: string;
+    __userId?: string;
+}
+
+export type { IStorage as I, ModelPrice as M, PriceMap as P, Report as R, SessionStats as S, TrackerConfig as T, UsageEntry as U, Tracker as a, TrackingMeta as b, ModelStats as c, PricesFile as d, UserStats as e };

package/dist/index.cjs

@@ -1369,7 +1369,9 @@ var ModelPriceSchema = import_zod.z.object({
   maxInputTokens: import_zod.z.number().positive().optional()
 });
 var TrackerConfigSchema = import_zod.z.object({
-  storage: import_zod.z.enum(["memory", "sqlite"]).optional().default("memory"),
+  storage: import_zod.z.union([import_zod.z.enum(["memory", "sqlite"]), import_zod.z.custom((v) => {
+    return v !== null && typeof v === "object" && typeof v.record === "function" && typeof v.getAll === "function" && typeof v.clearAll === "function" && typeof v.clearSession === "function";
+  })]).optional().default("memory"),
   alertThreshold: import_zod.z.number().positive().optional(),
   webhookUrl: import_zod.z.string().url().optional(),
   syncPrices: import_zod.z.boolean().optional().default(true),
@@ -1383,13 +1385,13 @@ function createTracker(config = {}) {
 ${issues}`);
   }
   const {
-    storage: storageType,
+    storage: storageOption,
     alertThreshold,
     webhookUrl,
     syncPrices,
     customPrices
   } = parsed.data;
-  const storage = createStorage(storageType);
+  const storage = typeof storageOption === "object" ? storageOption : createStorage(storageOption);
   let remotePrices;
   if (syncPrices) {
     getRemotePrices().then((result) => {
@@ -1419,9 +1421,13 @@ ${issues}`);
   }
   function maybeFireAlert() {
     if (!alertThreshold || !webhookUrl || alertFired) return;
-    const total = computeTotal(storage.getAll());
-    if (total >= alertThreshold) {
-      alertFired = true;
+    alertFired = true;
+    Promise.resolve(storage.getAll()).then((entries) => {
+      const total = computeTotal(entries);
+      if (total < alertThreshold) {
+        alertFired = false;
+        return;
+      }
       const payload = {
         text: `[tokenwatch] Alert: total cost reached $${total.toFixed(4)} USD (threshold: $${alertThreshold})`
       };
@@ -1431,10 +1437,12 @@ ${issues}`);
         body: JSON.stringify(payload)
       }).catch(() => {
       });
-    }
+    }).catch(() => {
+      alertFired = false;
+    });
   }
-  function getReport() {
-    const entries = storage.getAll();
+  async function getReport() {
+    const entries = await Promise.resolve(storage.getAll());
     const byModel = {};
     const bySession = {};
     const byUser = {};
@@ -1472,18 +1480,18 @@ ${issues}`);
       period: { from: startedAt, to: lastTimestamp }
     };
   }
-  function reset() {
-    storage.clearAll();
+  async function reset() {
+    await Promise.resolve(storage.clearAll());
     alertFired = false;
   }
-  function resetSession(sessionId) {
-    storage.clearSession(sessionId);
+  async function resetSession(sessionId) {
+    await Promise.resolve(storage.clearSession(sessionId));
   }
-  function exportJSON() {
-    return JSON.stringify(getReport(), null, 2);
+  async function exportJSON() {
+    return JSON.stringify(await getReport(), null, 2);
   }
-  function exportCSV() {
-    const entries = storage.getAll();
+  async function exportCSV() {
+    const entries = await Promise.resolve(storage.getAll());
     const header = "timestamp,model,inputTokens,outputTokens,costUSD,sessionId,userId";
     const rows = entries.map(
       (e) => [
@@ -1558,12 +1566,7 @@ function wrapOpenAI(client, tracker) {
       return async function(params) {
         const { cleaned, sessionId, userId } = extractMeta(params);
         const model = typeof cleaned["model"] === "string" ? cleaned["model"] : "unknown";
-        let result;
-        try {
-          result = await target.create(cleaned);
-        } catch (err) {
-          throw err;
-        }
+        const result = await target.create(cleaned);
         if (result && typeof result === "object" && Symbol.asyncIterator in result) {
           return wrapStream(
             result,
@@ -1648,12 +1651,7 @@ function wrapAnthropic(client, tracker) {
       return async function(params) {
         const { cleaned, sessionId, userId } = extractMeta2(params);
         const model = typeof cleaned["model"] === "string" ? cleaned["model"] : "unknown";
-        let result;
-        try {
-          result = await target.create(cleaned);
-        } catch (err) {
-          throw err;
-        }
+        const result = await target.create(cleaned);
         if (result && typeof result === "object" && Symbol.asyncIterator in result) {
           return wrapStream2(
             result,
@@ -1698,12 +1696,7 @@ function wrapGemini(client, tracker) {
           get(mTarget, mProp) {
             if (mProp === "generateContent") {
               return async function(params) {
-                let result;
-                try {
-                  result = await mTarget.generateContent(params);
-                } catch (err) {
-                  throw err;
-                }
+                const result = await mTarget.generateContent(params);
                 const meta = result.response.usageMetadata;
                 tracker.track({
                   model: modelId,
@@ -1715,12 +1708,7 @@ function wrapGemini(client, tracker) {
             }
             if (mProp === "generateContentStream") {
               return async function(params) {
-                let streamResult;
-                try {
-                  streamResult = await mTarget.generateContentStream(params);
-                } catch (err) {
-                  throw err;
-                }
+                const streamResult = await mTarget.generateContentStream(params);
                 streamResult.response.then((res) => {
                   const meta = res.usageMetadata;
                   tracker.track({