@diogonzafe/tokenwatch 0.1.17 → 0.2.1

package/README.md

@@ -1,6 +1,6 @@
 # @diogonzafe/tokenwatch
 
-Transparent TypeScript wrapper that intercepts LLM API calls and tracks cost in real-time by session, user and model — without changing anything in your existing code.
+Transparent TypeScript wrapper that intercepts LLM API calls and tracks cost in real-time by session, user, model and feature — without changing anything in your existing code.
 
 Supports **OpenAI**, **Anthropic**, **Google Gemini** and **DeepSeek**.
 
@@ -59,6 +59,7 @@ const res = await openai.chat.completions.create({
   // Optional — removed before sending to the API
   __sessionId: 'session_abc',
   __userId: 'user_123',
+  __feature: 'chat',       // tag calls by product feature → report.byFeature
 })
 // res is identical to the original OpenAI response — zero difference
 ```
@@ -79,6 +80,20 @@ for await (const chunk of stream) {
 // Cost tracked automatically from the final chunk
 ```
 
+Embeddings are also tracked automatically:
+
+```ts
+const embedding = await openai.embeddings.create({
+  model: 'text-embedding-3-small',
+  input: 'The food was delicious',
+  __sessionId: 'session_abc',
+  __feature: 'rag',   // optional
+})
+// inputTokens = usage.total_tokens, outputTokens = 0
+```
+
+> **Note:** `wrapOpenAI` covers `chat.completions` and `embeddings`. Other endpoints (e.g. fine-tuning, images) are not intercepted — use `tracker.track()` manually if needed.
+
 ---
 
 ## Anthropic
@@ -95,6 +110,7 @@ const res = await anthropic.messages.create({
   messages: [{ role: 'user', content: 'Hello' }],
   __sessionId: 'session_abc',
   __userId: 'user_123',
+  __feature: 'summarizer',
 })
 ```
 
@@ -108,7 +124,12 @@ import { wrapGemini } from '@diogonzafe/tokenwatch'
 
 const genAI = wrapGemini(new GoogleGenerativeAI(process.env.GEMINI_API_KEY!), tracker)
 
-const model = genAI.getGenerativeModel({ model: 'gemini-2.5-flash' })
+// __sessionId, __userId, __feature are passed to getGenerativeModel (not per-call)
+const model = genAI.getGenerativeModel({
+  model: 'gemini-2.5-flash',
+  __sessionId: 'session_abc',
+  __feature: 'rag',
+})
 const result = await model.generateContent('Explain quantum computing')
 ```
 
@@ -148,11 +169,13 @@ const report = await tracker.getReport()
 //   totalCostUSD: 0.087,
 //   totalTokens: { input: 24000, output: 6000 },
 //   byModel: {
-//     'gpt-4o': { costUSD: 0.062, calls: 5, tokens: { input: 20000, output: 5000 } },
-//     'claude-sonnet-4-6': { costUSD: 0.025, calls: 2, tokens: { input: 4000, output: 1000 } }
+//     'gpt-4o':          { costUSD: 0.062, calls: 5, tokens: { input: 20000, output: 5000, reasoning: 0 } },
+//     'o3':              { costUSD: 0.041, calls: 1, tokens: { input: 1000,  output: 200,  reasoning: 800 } },
+//     'claude-sonnet-4-6': { costUSD: 0.025, calls: 2, tokens: { input: 4000, output: 1000, reasoning: 0 } }
 //   },
 //   bySession: { 'session_abc': { costUSD: 0.045, calls: 4 } },
 //   byUser:    { 'user_123':    { costUSD: 0.087, calls: 7 } },
+//   byFeature: { 'chat': { costUSD: 0.062, calls: 5 }, 'rag': { costUSD: 0.025, calls: 3 } },
 //   period: { from: '2026-04-16T10:00:00Z', to: '2026-04-16T11:00:00Z' }
 // }
 
@@ -195,7 +218,7 @@ Prices are in **USD per 1 million tokens**.
 }
 ```
 
-Prices are updated every day via a GitHub Action that pulls from the [LiteLLM community model registry](https://github.com/BerriAI/litellm). New models are auto-discovered — no manual updates needed.
+Prices are updated every day via a GitHub Action that pulls from the [LiteLLM community model registry](https://github.com/BerriAI/litellm) and commits the updated `prices.json` to the repo. Users with `syncPrices: true` (the default) always receive fresh prices at runtime — no `npm update` needed.
 
 ---
 
@@ -333,6 +356,10 @@ npx tokenwatch help     # show help
 
   By user:
     user_123                       $0.004231  (11 calls)
+
+  By feature:
+    chat                           $0.002500  (5 calls)
+    rag                            $0.001731  (6 calls)
 ───────────────────────────────────────────────────
 ```
 
@@ -349,9 +376,34 @@ Requires `storage: 'sqlite'` in your app and `better-sqlite3` installed.
 
 ---
 
+## TypeScript
+
+`__sessionId`, `__userId`, and `__feature` are typed via the `TrackingMeta` interface, which is automatically merged into the `create` params type by the wrapper. In most setups they just work with no cast required.
+
+If you hit a type error (e.g. with stricter SDK versions), use `as Record<string, unknown>`:
+
+```ts
+await openai.chat.completions.create({
+  model: 'gpt-4o',
+  messages: [],
+  __sessionId: 'sess-1',
+  __feature: 'chat',
+} as Record<string, unknown>)
+```
+
+`TrackingMeta` is exported if you need to annotate your own helper types:
+
+```ts
+import type { TrackingMeta } from '@diogonzafe/tokenwatch'
+
+type MyParams = { model: string; messages: Message[] } & TrackingMeta
+```
+
+---
+
 ## Behaviour Guarantees
 
-- `__sessionId` and `__userId` are **stripped before** the request reaches the API
+- `__sessionId`, `__userId`, and `__feature` are **stripped before** the request reaches the API
 - The response object returned is **identical** to the original SDK response
 - `track()` is **synchronous and non-blocking** — zero latency added to API calls
 - If the API call **fails**, no cost is recorded and the original error is re-thrown unchanged

package/dist/adapters.cjs

@@ -32,33 +32,46 @@ var PostgresStorage = class {
     this.client = client;
   }
   client;
-  /** Creates the `tokenwatch_usage` table if it does not already exist. */
+  /** Creates the `tokenwatch_usage` table if it does not already exist.
+   *  Also adds new columns for databases created before v0.2.0. */
   async migrate() {
     await this.client.query(`
       CREATE TABLE IF NOT EXISTS tokenwatch_usage (
-        id            BIGSERIAL PRIMARY KEY,
-        model         TEXT      NOT NULL,
-        input_tokens  INTEGER   NOT NULL,
-        output_tokens INTEGER   NOT NULL,
-        cost_usd      NUMERIC   NOT NULL,
-        session_id    TEXT,
-        user_id       TEXT,
-        timestamp     TIMESTAMPTZ NOT NULL
+        id               BIGSERIAL PRIMARY KEY,
+        model            TEXT        NOT NULL,
+        input_tokens     INTEGER     NOT NULL,
+        output_tokens    INTEGER     NOT NULL,
+        reasoning_tokens INTEGER     NOT NULL DEFAULT 0,
+        cost_usd         NUMERIC     NOT NULL,
+        session_id       TEXT,
+        user_id          TEXT,
+        feature          TEXT,
+        timestamp        TIMESTAMPTZ NOT NULL
       )
     `);
+    await this.client.query(`
+      ALTER TABLE tokenwatch_usage ADD COLUMN IF NOT EXISTS reasoning_tokens INTEGER NOT NULL DEFAULT 0
+    `).catch(() => {
+    });
+    await this.client.query(`
+      ALTER TABLE tokenwatch_usage ADD COLUMN IF NOT EXISTS feature TEXT
+    `).catch(() => {
+    });
   }
   record(entry) {
     this.client.query(
       `INSERT INTO tokenwatch_usage
-         (model, input_tokens, output_tokens, cost_usd, session_id, user_id, timestamp)
-         VALUES ($1, $2, $3, $4, $5, $6, $7)`,
+         (model, input_tokens, output_tokens, reasoning_tokens, cost_usd, session_id, user_id, feature, timestamp)
+         VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9)`,
       [
         entry.model,
         entry.inputTokens,
         entry.outputTokens,
+        entry.reasoningTokens ?? 0,
         entry.costUSD,
         entry.sessionId ?? null,
         entry.userId ?? null,
+        entry.feature ?? null,
         entry.timestamp
       ]
     ).catch((err) => {
@@ -82,13 +95,16 @@ var PostgresStorage = class {
   }
 };
 function rowToEntry(r) {
+  const reasoningTokens = r["reasoning_tokens"] ?? 0;
   return {
     model: r["model"],
     inputTokens: r["input_tokens"],
     outputTokens: r["output_tokens"],
+    ...reasoningTokens > 0 && { reasoningTokens },
     costUSD: Number(r["cost_usd"]),
     ...r["session_id"] != null && { sessionId: r["session_id"] },
     ...r["user_id"] != null && { userId: r["user_id"] },
+    ...r["feature"] != null && { feature: r["feature"] },
     timestamp: r["timestamp"] instanceof Date ? r["timestamp"].toISOString() : r["timestamp"]
   };
 }
@@ -99,33 +115,44 @@ var MySQLStorage = class {
     this.client = client;
   }
   client;
-  /** Creates the `tokenwatch_usage` table if it does not already exist. */
+  /** Creates the `tokenwatch_usage` table if it does not already exist.
+   *  Also adds new columns for databases created before v0.2.0. */
   async migrate() {
     await this.client.execute(`
       CREATE TABLE IF NOT EXISTS tokenwatch_usage (
-        id            BIGINT UNSIGNED AUTO_INCREMENT PRIMARY KEY,
-        model         VARCHAR(255) NOT NULL,
-        input_tokens  INT          NOT NULL,
-        output_tokens INT          NOT NULL,
-        cost_usd      DECIMAL(18,8) NOT NULL,
-        session_id    VARCHAR(255),
-        user_id       VARCHAR(255),
-        timestamp     DATETIME(3)  NOT NULL
+        id               BIGINT UNSIGNED AUTO_INCREMENT PRIMARY KEY,
+        model            VARCHAR(255)  NOT NULL,
+        input_tokens     INT           NOT NULL,
+        output_tokens    INT           NOT NULL,
+        reasoning_tokens INT           NOT NULL DEFAULT 0,
+        cost_usd         DECIMAL(18,8) NOT NULL,
+        session_id       VARCHAR(255),
+        user_id          VARCHAR(255),
+        feature          VARCHAR(255),
+        timestamp        DATETIME(3)   NOT NULL
       )
     `);
+    await this.client.execute(`
+      ALTER TABLE tokenwatch_usage
+        ADD COLUMN IF NOT EXISTS reasoning_tokens INT NOT NULL DEFAULT 0,
+        ADD COLUMN IF NOT EXISTS feature VARCHAR(255)
+    `).catch(() => {
+    });
   }
   record(entry) {
     this.client.execute(
       `INSERT INTO tokenwatch_usage
-         (model, input_tokens, output_tokens, cost_usd, session_id, user_id, timestamp)
-         VALUES (?, ?, ?, ?, ?, ?, ?)`,
+         (model, input_tokens, output_tokens, reasoning_tokens, cost_usd, session_id, user_id, feature, timestamp)
+         VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)`,
       [
         entry.model,
         entry.inputTokens,
         entry.outputTokens,
+        entry.reasoningTokens ?? 0,
         entry.costUSD,
         entry.sessionId ?? null,
         entry.userId ?? null,
+        entry.feature ?? null,
         entry.timestamp
       ]
     ).catch((err) => {
@@ -149,13 +176,16 @@ var MySQLStorage = class {
   }
 };
 function rowToEntry2(r) {
+  const reasoningTokens = r["reasoning_tokens"] ?? 0;
   return {
     model: r["model"],
     inputTokens: r["input_tokens"],
     outputTokens: r["output_tokens"],
+    ...reasoningTokens > 0 && { reasoningTokens },
     costUSD: Number(r["cost_usd"]),
     ...r["session_id"] != null && { sessionId: r["session_id"] },
     ...r["user_id"] != null && { userId: r["user_id"] },
+    ...r["feature"] != null && { feature: r["feature"] },
     timestamp: r["timestamp"] instanceof Date ? r["timestamp"].toISOString() : r["timestamp"]
   };
 }
@@ -179,9 +209,11 @@ var MongoStorage = class {
       model: entry.model,
       inputTokens: entry.inputTokens,
       outputTokens: entry.outputTokens,
+      ...entry.reasoningTokens !== void 0 && { reasoningTokens: entry.reasoningTokens },
       costUSD: entry.costUSD,
       sessionId: entry.sessionId ?? null,
       userId: entry.userId ?? null,
+      ...entry.feature !== void 0 && { feature: entry.feature },
       timestamp: entry.timestamp
     }).catch((err) => {
       console.warn("[tokenwatch] MongoStorage.record failed:", err);
@@ -203,9 +235,11 @@ function docToEntry(doc) {
     model: doc.model,
     inputTokens: doc.inputTokens,
     outputTokens: doc.outputTokens,
+    ...doc.reasoningTokens != null && doc.reasoningTokens > 0 && { reasoningTokens: doc.reasoningTokens },
     costUSD: doc.costUSD,
     ...doc.sessionId != null && { sessionId: doc.sessionId },
     ...doc.userId != null && { userId: doc.userId },
+    ...doc.feature != null && { feature: doc.feature },
     timestamp: doc.timestamp
   };
 }

package/dist/adapters.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/adapters/index.ts","../src/adapters/postgres.ts","../src/adapters/mysql.ts","../src/adapters/mongodb.ts"],"sourcesContent":["export { PostgresStorage } from './postgres.js'\nexport { MySQLStorage } from './mysql.js'\nexport { MongoStorage } from './mongodb.js'\n","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 MongoCursor {\n  sort(sort: Record<string, unknown>): MongoCursor\n  toArray(): Promise<MongoDocument[]>\n}\n\ninterface Collection {\n  insertOne(doc: MongoDocument): Promise<unknown>\n  find(filter: Record<string, unknown>): MongoCursor\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({}).sort({ timestamp: 1 }).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":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;AC2BO,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;;;AC5CA,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,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE,QAAQ;AACpE,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"]}
+{"version":3,"sources":["../src/adapters/index.ts","../src/adapters/postgres.ts","../src/adapters/mysql.ts","../src/adapters/mongodb.ts"],"sourcesContent":["export { PostgresStorage } from './postgres.js'\nexport { MySQLStorage } from './mysql.js'\nexport { MongoStorage } from './mongodb.js'\n","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   *  Also adds new columns for databases created before v0.2.0. */\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        reasoning_tokens INTEGER     NOT NULL DEFAULT 0,\n        cost_usd         NUMERIC     NOT NULL,\n        session_id       TEXT,\n        user_id          TEXT,\n        feature          TEXT,\n        timestamp        TIMESTAMPTZ NOT NULL\n      )\n    `)\n    // Incremental migrations for databases created before v0.2.0\n    await this.client.query(`\n      ALTER TABLE tokenwatch_usage ADD COLUMN IF NOT EXISTS reasoning_tokens INTEGER NOT NULL DEFAULT 0\n    `).catch(() => { /* column already exists in older Postgres versions that don't support IF NOT EXISTS */ })\n    await this.client.query(`\n      ALTER TABLE tokenwatch_usage ADD COLUMN IF NOT EXISTS feature TEXT\n    `).catch(() => { /* same */ })\n  }\n\n  record(entry: UsageEntry): void {\n    this.client\n      .query(\n        `INSERT INTO tokenwatch_usage\n         (model, input_tokens, output_tokens, reasoning_tokens, cost_usd, session_id, user_id, feature, timestamp)\n         VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9)`,\n        [\n          entry.model,\n          entry.inputTokens,\n          entry.outputTokens,\n          entry.reasoningTokens ?? 0,\n          entry.costUSD,\n          entry.sessionId ?? null,\n          entry.userId ?? null,\n          entry.feature ?? 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  const reasoningTokens = (r['reasoning_tokens'] as number | null) ?? 0\n  return {\n    model: r['model'] as string,\n    inputTokens: r['input_tokens'] as number,\n    outputTokens: r['output_tokens'] as number,\n    ...(reasoningTokens > 0 && { reasoningTokens }),\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    ...(r['feature'] != null && { feature: r['feature'] 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   *  Also adds new columns for databases created before v0.2.0. */\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        reasoning_tokens INT           NOT NULL DEFAULT 0,\n        cost_usd         DECIMAL(18,8) NOT NULL,\n        session_id       VARCHAR(255),\n        user_id          VARCHAR(255),\n        feature          VARCHAR(255),\n        timestamp        DATETIME(3)   NOT NULL\n      )\n    `)\n    // Incremental migrations for databases created before v0.2.0\n    await this.client.execute(`\n      ALTER TABLE tokenwatch_usage\n        ADD COLUMN IF NOT EXISTS reasoning_tokens INT NOT NULL DEFAULT 0,\n        ADD COLUMN IF NOT EXISTS feature VARCHAR(255)\n    `).catch(() => { /* MySQL < 8.0 may not support IF NOT EXISTS — ignore if columns already exist */ })\n  }\n\n  record(entry: UsageEntry): void {\n    this.client\n      .execute(\n        `INSERT INTO tokenwatch_usage\n         (model, input_tokens, output_tokens, reasoning_tokens, cost_usd, session_id, user_id, feature, timestamp)\n         VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)`,\n        [\n          entry.model,\n          entry.inputTokens,\n          entry.outputTokens,\n          entry.reasoningTokens ?? 0,\n          entry.costUSD,\n          entry.sessionId ?? null,\n          entry.userId ?? null,\n          entry.feature ?? 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  const reasoningTokens = (r['reasoning_tokens'] as number | null) ?? 0\n  return {\n    model: r['model'] as string,\n    inputTokens: r['input_tokens'] as number,\n    outputTokens: r['output_tokens'] as number,\n    ...(reasoningTokens > 0 && { reasoningTokens }),\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    ...(r['feature'] != null && { feature: r['feature'] 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  reasoningTokens?: number\n  costUSD: number\n  sessionId?: string | null\n  userId?: string | null\n  feature?: string | null\n  timestamp: string\n}\n\ninterface MongoCursor {\n  sort(sort: Record<string, unknown>): MongoCursor\n  toArray(): Promise<MongoDocument[]>\n}\n\ninterface Collection {\n  insertOne(doc: MongoDocument): Promise<unknown>\n  find(filter: Record<string, unknown>): MongoCursor\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        ...(entry.reasoningTokens !== undefined && { reasoningTokens: entry.reasoningTokens }),\n        costUSD: entry.costUSD,\n        sessionId: entry.sessionId ?? null,\n        userId: entry.userId ?? null,\n        ...(entry.feature !== undefined && { feature: entry.feature }),\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({}).sort({ timestamp: 1 }).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    ...(doc.reasoningTokens != null && doc.reasoningTokens > 0 && { reasoningTokens: doc.reasoningTokens }),\n    costUSD: doc.costUSD,\n    ...(doc.sessionId != null && { sessionId: doc.sessionId }),\n    ...(doc.userId != null && { userId: doc.userId }),\n    ...(doc.feature != null && { feature: doc.feature }),\n    timestamp: doc.timestamp,\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;AC2BO,IAAM,kBAAN,MAA0C;AAAA,EAC/C,YAA6B,QAAqB;AAArB;AAAA,EAAsB;AAAA,EAAtB;AAAA;AAAA;AAAA,EAI7B,MAAM,UAAyB;AAC7B,UAAM,KAAK,OAAO,MAAM;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,KAavB;AAED,UAAM,KAAK,OAAO,MAAM;AAAA;AAAA,KAEvB,EAAE,MAAM,MAAM;AAAA,IAA0F,CAAC;AAC1G,UAAM,KAAK,OAAO,MAAM;AAAA;AAAA,KAEvB,EAAE,MAAM,MAAM;AAAA,IAAa,CAAC;AAAA,EAC/B;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,mBAAmB;AAAA,QACzB,MAAM;AAAA,QACN,MAAM,aAAa;AAAA,QACnB,MAAM,UAAU;AAAA,QAChB,MAAM,WAAW;AAAA,QACjB,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,QAAM,kBAAmB,EAAE,kBAAkB,KAAuB;AACpE,SAAO;AAAA,IACL,OAAO,EAAE,OAAO;AAAA,IAChB,aAAa,EAAE,cAAc;AAAA,IAC7B,cAAc,EAAE,eAAe;AAAA,IAC/B,GAAI,kBAAkB,KAAK,EAAE,gBAAgB;AAAA,IAC7C,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,GAAI,EAAE,SAAS,KAAK,QAAQ,EAAE,SAAS,EAAE,SAAS,EAAY;AAAA,IAC9D,WACE,EAAE,WAAW,aAAa,OACrB,EAAE,WAAW,EAAW,YAAY,IACpC,EAAE,WAAW;AAAA,EACtB;AACF;;;ACxFO,IAAM,eAAN,MAAuC;AAAA,EAC5C,YAA6B,QAAqB;AAArB;AAAA,EAAsB;AAAA,EAAtB;AAAA;AAAA;AAAA,EAI7B,MAAM,UAAyB;AAC7B,UAAM,KAAK,OAAO,QAAQ;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,KAazB;AAED,UAAM,KAAK,OAAO,QAAQ;AAAA;AAAA;AAAA;AAAA,KAIzB,EAAE,MAAM,MAAM;AAAA,IAAoF,CAAC;AAAA,EACtG;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,mBAAmB;AAAA,QACzB,MAAM;AAAA,QACN,MAAM,aAAa;AAAA,QACnB,MAAM,UAAU;AAAA,QAChB,MAAM,WAAW;AAAA,QACjB,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,QAAM,kBAAmB,EAAE,kBAAkB,KAAuB;AACpE,SAAO;AAAA,IACL,OAAO,EAAE,OAAO;AAAA,IAChB,aAAa,EAAE,cAAc;AAAA,IAC7B,cAAc,EAAE,eAAe;AAAA,IAC/B,GAAI,kBAAkB,KAAK,EAAE,gBAAgB;AAAA,IAC7C,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,GAAI,EAAE,SAAS,KAAK,QAAQ,EAAE,SAAS,EAAE,SAAS,EAAY;AAAA,IAC9D,WACE,EAAE,WAAW,aAAa,OACrB,EAAE,WAAW,EAAW,YAAY,IACpC,EAAE,WAAW;AAAA,EACtB;AACF;;;ACxDA,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,GAAI,MAAM,oBAAoB,UAAa,EAAE,iBAAiB,MAAM,gBAAgB;AAAA,MACpF,SAAS,MAAM;AAAA,MACf,WAAW,MAAM,aAAa;AAAA,MAC9B,QAAQ,MAAM,UAAU;AAAA,MACxB,GAAI,MAAM,YAAY,UAAa,EAAE,SAAS,MAAM,QAAQ;AAAA,MAC5D,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,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE,QAAQ;AACpE,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,GAAI,IAAI,mBAAmB,QAAQ,IAAI,kBAAkB,KAAK,EAAE,iBAAiB,IAAI,gBAAgB;AAAA,IACrG,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,GAAI,IAAI,WAAW,QAAQ,EAAE,SAAS,IAAI,QAAQ;AAAA,IAClD,WAAW,IAAI;AAAA,EACjB;AACF;","names":["rowToEntry"]}

package/dist/adapters.d.cts

@@ -1,4 +1,4 @@
-import { I as IStorage, U as UsageEntry } from './index-Cy_sl3FI.cjs';
+import { I as IStorage, U as UsageEntry } from './index-B_EmA3K7.cjs';
 
 /**
  * IStorage adapter for PostgreSQL using the `pg` driver.
@@ -27,7 +27,8 @@ interface QueryClient$1 {
 declare class PostgresStorage implements IStorage {
     private readonly client;
     constructor(client: QueryClient$1);
-    /** Creates the `tokenwatch_usage` table if it does not already exist. */
+    /** Creates the `tokenwatch_usage` table if it does not already exist.
+     *  Also adds new columns for databases created before v0.2.0. */
     migrate(): Promise<void>;
     record(entry: UsageEntry): void;
     getAll(): Promise<UsageEntry[]>;
@@ -59,7 +60,8 @@ interface QueryClient {
 declare class MySQLStorage implements IStorage {
     private readonly client;
     constructor(client: QueryClient);
-    /** Creates the `tokenwatch_usage` table if it does not already exist. */
+    /** Creates the `tokenwatch_usage` table if it does not already exist.
+     *  Also adds new columns for databases created before v0.2.0. */
     migrate(): Promise<void>;
     record(entry: UsageEntry): void;
     getAll(): Promise<UsageEntry[]>;
@@ -95,9 +97,11 @@ interface MongoDocument {
     model: string;
     inputTokens: number;
     outputTokens: number;
+    reasoningTokens?: number;
     costUSD: number;
     sessionId?: string | null;
     userId?: string | null;
+    feature?: string | null;
     timestamp: string;
 }
 interface MongoCursor {

package/dist/adapters.d.ts

@@ -1,4 +1,4 @@
-import { I as IStorage, U as UsageEntry } from './index-Cy_sl3FI.js';
+import { I as IStorage, U as UsageEntry } from './index-B_EmA3K7.js';
 
 /**
  * IStorage adapter for PostgreSQL using the `pg` driver.
@@ -27,7 +27,8 @@ interface QueryClient$1 {
 declare class PostgresStorage implements IStorage {
     private readonly client;
     constructor(client: QueryClient$1);
-    /** Creates the `tokenwatch_usage` table if it does not already exist. */
+    /** Creates the `tokenwatch_usage` table if it does not already exist.
+     *  Also adds new columns for databases created before v0.2.0. */
     migrate(): Promise<void>;
     record(entry: UsageEntry): void;
     getAll(): Promise<UsageEntry[]>;
@@ -59,7 +60,8 @@ interface QueryClient {
 declare class MySQLStorage implements IStorage {
     private readonly client;
     constructor(client: QueryClient);
-    /** Creates the `tokenwatch_usage` table if it does not already exist. */
+    /** Creates the `tokenwatch_usage` table if it does not already exist.
+     *  Also adds new columns for databases created before v0.2.0. */
     migrate(): Promise<void>;
     record(entry: UsageEntry): void;
     getAll(): Promise<UsageEntry[]>;
@@ -95,9 +97,11 @@ interface MongoDocument {
     model: string;
     inputTokens: number;
     outputTokens: number;
+    reasoningTokens?: number;
     costUSD: number;
     sessionId?: string | null;
     userId?: string | null;
+    feature?: string | null;
     timestamp: string;
 }
 interface MongoCursor {

package/dist/adapters.js

@@ -4,33 +4,46 @@ var PostgresStorage = class {
     this.client = client;
   }
   client;
-  /** Creates the `tokenwatch_usage` table if it does not already exist. */
+  /** Creates the `tokenwatch_usage` table if it does not already exist.
+   *  Also adds new columns for databases created before v0.2.0. */
   async migrate() {
     await this.client.query(`
       CREATE TABLE IF NOT EXISTS tokenwatch_usage (
-        id            BIGSERIAL PRIMARY KEY,
-        model         TEXT      NOT NULL,
-        input_tokens  INTEGER   NOT NULL,
-        output_tokens INTEGER   NOT NULL,
-        cost_usd      NUMERIC   NOT NULL,
-        session_id    TEXT,
-        user_id       TEXT,
-        timestamp     TIMESTAMPTZ NOT NULL
+        id               BIGSERIAL PRIMARY KEY,
+        model            TEXT        NOT NULL,
+        input_tokens     INTEGER     NOT NULL,
+        output_tokens    INTEGER     NOT NULL,
+        reasoning_tokens INTEGER     NOT NULL DEFAULT 0,
+        cost_usd         NUMERIC     NOT NULL,
+        session_id       TEXT,
+        user_id          TEXT,
+        feature          TEXT,
+        timestamp        TIMESTAMPTZ NOT NULL
       )
     `);
+    await this.client.query(`
+      ALTER TABLE tokenwatch_usage ADD COLUMN IF NOT EXISTS reasoning_tokens INTEGER NOT NULL DEFAULT 0
+    `).catch(() => {
+    });
+    await this.client.query(`
+      ALTER TABLE tokenwatch_usage ADD COLUMN IF NOT EXISTS feature TEXT
+    `).catch(() => {
+    });
   }
   record(entry) {
     this.client.query(
       `INSERT INTO tokenwatch_usage
-         (model, input_tokens, output_tokens, cost_usd, session_id, user_id, timestamp)
-         VALUES ($1, $2, $3, $4, $5, $6, $7)`,
+         (model, input_tokens, output_tokens, reasoning_tokens, cost_usd, session_id, user_id, feature, timestamp)
+         VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9)`,
       [
         entry.model,
         entry.inputTokens,
         entry.outputTokens,
+        entry.reasoningTokens ?? 0,
         entry.costUSD,
         entry.sessionId ?? null,
         entry.userId ?? null,
+        entry.feature ?? null,
         entry.timestamp
       ]
     ).catch((err) => {
@@ -54,13 +67,16 @@ var PostgresStorage = class {
   }
 };
 function rowToEntry(r) {
+  const reasoningTokens = r["reasoning_tokens"] ?? 0;
   return {
     model: r["model"],
     inputTokens: r["input_tokens"],
     outputTokens: r["output_tokens"],
+    ...reasoningTokens > 0 && { reasoningTokens },
     costUSD: Number(r["cost_usd"]),
     ...r["session_id"] != null && { sessionId: r["session_id"] },
     ...r["user_id"] != null && { userId: r["user_id"] },
+    ...r["feature"] != null && { feature: r["feature"] },
     timestamp: r["timestamp"] instanceof Date ? r["timestamp"].toISOString() : r["timestamp"]
   };
 }
@@ -71,33 +87,44 @@ var MySQLStorage = class {
     this.client = client;
   }
   client;
-  /** Creates the `tokenwatch_usage` table if it does not already exist. */
+  /** Creates the `tokenwatch_usage` table if it does not already exist.
+   *  Also adds new columns for databases created before v0.2.0. */
   async migrate() {
     await this.client.execute(`
       CREATE TABLE IF NOT EXISTS tokenwatch_usage (
-        id            BIGINT UNSIGNED AUTO_INCREMENT PRIMARY KEY,
-        model         VARCHAR(255) NOT NULL,
-        input_tokens  INT          NOT NULL,
-        output_tokens INT          NOT NULL,
-        cost_usd      DECIMAL(18,8) NOT NULL,
-        session_id    VARCHAR(255),
-        user_id       VARCHAR(255),
-        timestamp     DATETIME(3)  NOT NULL
+        id               BIGINT UNSIGNED AUTO_INCREMENT PRIMARY KEY,
+        model            VARCHAR(255)  NOT NULL,
+        input_tokens     INT           NOT NULL,
+        output_tokens    INT           NOT NULL,
+        reasoning_tokens INT           NOT NULL DEFAULT 0,
+        cost_usd         DECIMAL(18,8) NOT NULL,
+        session_id       VARCHAR(255),
+        user_id          VARCHAR(255),
+        feature          VARCHAR(255),
+        timestamp        DATETIME(3)   NOT NULL
       )
     `);
+    await this.client.execute(`
+      ALTER TABLE tokenwatch_usage
+        ADD COLUMN IF NOT EXISTS reasoning_tokens INT NOT NULL DEFAULT 0,
+        ADD COLUMN IF NOT EXISTS feature VARCHAR(255)
+    `).catch(() => {
+    });
   }
   record(entry) {
     this.client.execute(
       `INSERT INTO tokenwatch_usage
-         (model, input_tokens, output_tokens, cost_usd, session_id, user_id, timestamp)
-         VALUES (?, ?, ?, ?, ?, ?, ?)`,
+         (model, input_tokens, output_tokens, reasoning_tokens, cost_usd, session_id, user_id, feature, timestamp)
+         VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)`,
       [
         entry.model,
         entry.inputTokens,
         entry.outputTokens,
+        entry.reasoningTokens ?? 0,
         entry.costUSD,
         entry.sessionId ?? null,
         entry.userId ?? null,
+        entry.feature ?? null,
         entry.timestamp
       ]
     ).catch((err) => {
@@ -121,13 +148,16 @@ var MySQLStorage = class {
   }
 };
 function rowToEntry2(r) {
+  const reasoningTokens = r["reasoning_tokens"] ?? 0;
   return {
     model: r["model"],
     inputTokens: r["input_tokens"],
     outputTokens: r["output_tokens"],
+    ...reasoningTokens > 0 && { reasoningTokens },
     costUSD: Number(r["cost_usd"]),
     ...r["session_id"] != null && { sessionId: r["session_id"] },
     ...r["user_id"] != null && { userId: r["user_id"] },
+    ...r["feature"] != null && { feature: r["feature"] },
     timestamp: r["timestamp"] instanceof Date ? r["timestamp"].toISOString() : r["timestamp"]
   };
 }
@@ -151,9 +181,11 @@ var MongoStorage = class {
       model: entry.model,
       inputTokens: entry.inputTokens,
       outputTokens: entry.outputTokens,
+      ...entry.reasoningTokens !== void 0 && { reasoningTokens: entry.reasoningTokens },
       costUSD: entry.costUSD,
       sessionId: entry.sessionId ?? null,
       userId: entry.userId ?? null,
+      ...entry.feature !== void 0 && { feature: entry.feature },
       timestamp: entry.timestamp
     }).catch((err) => {
       console.warn("[tokenwatch] MongoStorage.record failed:", err);
@@ -175,9 +207,11 @@ function docToEntry(doc) {
     model: doc.model,
     inputTokens: doc.inputTokens,
     outputTokens: doc.outputTokens,
+    ...doc.reasoningTokens != null && doc.reasoningTokens > 0 && { reasoningTokens: doc.reasoningTokens },
     costUSD: doc.costUSD,
     ...doc.sessionId != null && { sessionId: doc.sessionId },
     ...doc.userId != null && { userId: doc.userId },
+    ...doc.feature != null && { feature: doc.feature },
     timestamp: doc.timestamp
   };
 }