@routstr/sdk 0.3.10 → 0.3.12

package/dist/storage/bun.mjs

@@ -242,6 +242,149 @@ var SDK_STORAGE_KEYS = {
   PROVIDERS_ON_COOLDOWN: "providers_on_cooldown"
 };
 
+// storage/usageTracking/aggregate.ts
+var pad2 = (n) => String(n).padStart(2, "0");
+var aggregateColumns = "COUNT(*) AS requests, COALESCE(SUM(prompt_tokens), 0) AS promptTokens, COALESCE(SUM(completion_tokens), 0) AS completionTokens, COALESCE(SUM(total_tokens), 0) AS totalTokens, COALESCE(SUM(cost), 0) AS cost, COALESCE(SUM(sats_cost), 0) AS satsCost, COALESCE(SUM(base_msats), 0) AS baseMsats, COALESCE(SUM(input_msats), 0) AS inputMsats, COALESCE(SUM(output_msats), 0) AS outputMsats, COALESCE(SUM(total_msats), 0) AS totalMsats, COALESCE(SUM(total_usd), 0) AS totalUsd, COALESCE(SUM(cache_read_input_tokens), 0) AS cacheReadInputTokens, COALESCE(SUM(cache_creation_input_tokens), 0) AS cacheCreationInputTokens, COALESCE(SUM(cache_read_msats), 0) AS cacheReadMsats, COALESCE(SUM(cache_creation_msats), 0) AS cacheCreationMsats";
+var sqlGroupExpr = (groupBy) => {
+  switch (groupBy) {
+    case "modelId":
+      return { expr: "model_id", usesTz: false };
+    case "baseUrl":
+      return { expr: "base_url", usesTz: false };
+    case "client":
+      return { expr: "client", usesTz: false };
+    case "sessionId":
+      return { expr: "session_id", usesTz: false };
+    case "provider":
+      return { expr: "provider", usesTz: false };
+    case "day":
+      return {
+        expr: "strftime('%Y-%m-%d', (timestamp - ? * 60000) / 1000, 'unixepoch')",
+        usesTz: true
+      };
+    case "hour":
+      return {
+        expr: "strftime('%H', (timestamp - ? * 60000) / 1000, 'unixepoch')",
+        usesTz: true
+      };
+  }
+};
+var buildAggregateSql = (tableName, where, options = {}) => {
+  if (!options.groupBy) {
+    return {
+      sql: `SELECT NULL AS grp, ${aggregateColumns} FROM ${tableName} ${where.sql}`,
+      params: where.params
+    };
+  }
+  const { expr, usesTz } = sqlGroupExpr(options.groupBy);
+  const tzParams = usesTz ? [options.tzOffsetMinutes ?? 0] : [];
+  const orderBy = options.groupBy === "day" || options.groupBy === "hour" ? "ORDER BY grp ASC" : "ORDER BY satsCost DESC";
+  return {
+    sql: `SELECT ${expr} AS grp, ${aggregateColumns} FROM ${tableName} ${where.sql} GROUP BY grp ${orderBy}`,
+    params: [...tzParams, ...where.params]
+  };
+};
+var mapAggregateRow = (row) => ({
+  group: row.grp == null ? null : String(row.grp),
+  requests: Number(row.requests ?? 0),
+  promptTokens: Number(row.promptTokens ?? 0),
+  completionTokens: Number(row.completionTokens ?? 0),
+  totalTokens: Number(row.totalTokens ?? 0),
+  cost: Number(row.cost ?? 0),
+  satsCost: Number(row.satsCost ?? 0),
+  baseMsats: Number(row.baseMsats ?? 0),
+  inputMsats: Number(row.inputMsats ?? 0),
+  outputMsats: Number(row.outputMsats ?? 0),
+  totalMsats: Number(row.totalMsats ?? 0),
+  totalUsd: Number(row.totalUsd ?? 0),
+  cacheReadInputTokens: Number(row.cacheReadInputTokens ?? 0),
+  cacheCreationInputTokens: Number(row.cacheCreationInputTokens ?? 0),
+  cacheReadMsats: Number(row.cacheReadMsats ?? 0),
+  cacheCreationMsats: Number(row.cacheCreationMsats ?? 0)
+});
+var jsGroupKey = (entry, groupBy, tzOffsetMinutes) => {
+  switch (groupBy) {
+    case "modelId":
+      return entry.modelId ?? null;
+    case "baseUrl":
+      return entry.baseUrl ?? null;
+    case "client":
+      return entry.client ?? null;
+    case "sessionId":
+      return entry.sessionId ?? null;
+    case "provider":
+      return entry.provider ?? null;
+    case "day": {
+      const d = new Date(entry.timestamp - tzOffsetMinutes * 6e4);
+      return `${d.getUTCFullYear()}-${pad2(d.getUTCMonth() + 1)}-${pad2(d.getUTCDate())}`;
+    }
+    case "hour": {
+      const d = new Date(entry.timestamp - tzOffsetMinutes * 6e4);
+      return pad2(d.getUTCHours());
+    }
+  }
+};
+var reduceAggregate = (entries, options = {}) => {
+  const emptyRow = (group) => ({
+    group,
+    requests: 0,
+    promptTokens: 0,
+    completionTokens: 0,
+    totalTokens: 0,
+    cost: 0,
+    satsCost: 0,
+    baseMsats: 0,
+    inputMsats: 0,
+    outputMsats: 0,
+    totalMsats: 0,
+    totalUsd: 0,
+    cacheReadInputTokens: 0,
+    cacheCreationInputTokens: 0,
+    cacheReadMsats: 0,
+    cacheCreationMsats: 0
+  });
+  const accumulate = (row, entry) => {
+    row.requests += 1;
+    row.promptTokens += entry.promptTokens;
+    row.completionTokens += entry.completionTokens;
+    row.totalTokens += entry.totalTokens;
+    row.cost += entry.cost;
+    row.satsCost += entry.satsCost;
+    row.baseMsats += entry.baseMsats ?? 0;
+    row.inputMsats += entry.inputMsats ?? 0;
+    row.outputMsats += entry.outputMsats ?? 0;
+    row.totalMsats += entry.totalMsats ?? 0;
+    row.totalUsd += entry.totalUsd ?? 0;
+    row.cacheReadInputTokens += entry.cacheReadInputTokens ?? 0;
+    row.cacheCreationInputTokens += entry.cacheCreationInputTokens ?? 0;
+    row.cacheReadMsats += entry.cacheReadMsats ?? 0;
+    row.cacheCreationMsats += entry.cacheCreationMsats ?? 0;
+  };
+  if (!options.groupBy) {
+    const total = emptyRow(null);
+    for (const entry of entries) accumulate(total, entry);
+    return [total];
+  }
+  const tz = options.tzOffsetMinutes ?? 0;
+  const groups = /* @__PURE__ */ new Map();
+  for (const entry of entries) {
+    const key = jsGroupKey(entry, options.groupBy, tz);
+    let row = groups.get(key);
+    if (!row) {
+      row = emptyRow(key);
+      groups.set(key, row);
+    }
+    accumulate(row, entry);
+  }
+  const rows = [...groups.values()];
+  if (options.groupBy === "day" || options.groupBy === "hour") {
+    rows.sort((a, b) => (a.group ?? "").localeCompare(b.group ?? ""));
+  } else {
+    rows.sort((a, b) => b.satsCost - a.satsCost);
+  }
+  return rows;
+};
+
 // storage/usageTracking/indexedDB.ts
 var DEFAULT_DB_NAME = "routstr-sdk";
 var DEFAULT_STORE_NAME = "usage_tracking";
@@ -304,6 +447,9 @@ var matchesFilters = (entry, options = {}) => {
   if (options.client && entry.client !== options.client) {
     return false;
   }
+  if (options.clients && options.clients.length > 0 && (entry.client == null || !options.clients.includes(entry.client))) {
+    return false;
+  }
   if (options.provider && entry.provider !== options.provider) {
     return false;
   }
@@ -402,6 +548,10 @@ var createIndexedDBUsageTrackingDriver = (options = {}) => {
       const results = await this.list(options2);
       return results.length;
     },
+    async aggregate(options2 = {}) {
+      const entries = await this.list(options2);
+      return reduceAggregate(entries, options2);
+    },
     async deleteOlderThan(timestamp) {
       await ensureMigrated();
       const db = await getDb();
@@ -459,6 +609,9 @@ var matchesFilters2 = (entry, options = {}) => {
   if (options.client && entry.client !== options.client) {
     return false;
   }
+  if (options.clients && options.clients.length > 0 && (entry.client == null || !options.clients.includes(entry.client))) {
+    return false;
+  }
   if (options.provider && entry.provider !== options.provider) {
     return false;
   }
@@ -491,6 +644,10 @@ var createMemoryUsageTrackingDriver = (seed = []) => {
     async count(options = {}) {
       return (await this.list(options)).length;
     },
+    async aggregate(options = {}) {
+      const entries = [...store.values()].filter((entry) => matchesFilters2(entry, options));
+      return reduceAggregate(entries, options);
+    },
     async deleteOlderThan(timestamp) {
       let deleted = 0;
       for (const [id, entry] of store.entries()) {
@@ -1488,6 +1645,8 @@ async function createBunSqliteDriver(dbPath, options) {
   const logger = (options?.logger ?? consoleLogger).child("BunSqliteDriver");
   const SQLite = (await import('bun:sqlite')).default;
   const db = new SQLite(dbPath);
+  db.run("PRAGMA journal_mode = WAL");
+  db.run("PRAGMA busy_timeout = 5000");
   db.run(`
     CREATE TABLE IF NOT EXISTS sdk_storage (
       key TEXT PRIMARY KEY,
@@ -1574,6 +1733,11 @@ var buildWhereClause = (options = {}) => {
     clauses.push("client = ?");
     params.push(options.client);
   }
+  if (options.clients && options.clients.length > 0) {
+    const placeholders = options.clients.map(() => "?").join(", ");
+    clauses.push(`client IN (${placeholders})`);
+    params.push(...options.clients);
+  }
   if (options.provider) {
     clauses.push("provider = ?");
     params.push(options.provider);
@@ -1595,6 +1759,8 @@ var createBunSqliteUsageTrackingDriver = (options = {}) => {
     );
   }
   const db = new SQLiteDatabase(dbPath);
+  db.run("PRAGMA journal_mode = WAL");
+  db.run("PRAGMA busy_timeout = 5000");
   db.run(`
     CREATE TABLE IF NOT EXISTS ${tableName} (
       id TEXT PRIMARY KEY,
@@ -1747,6 +1913,13 @@ var createBunSqliteUsageTrackingDriver = (options = {}) => {
       const row = db.query(query).get(...params);
       return Number(row?.count ?? 0);
     },
+    async aggregate(options2 = {}) {
+      await ensureMigrated();
+      const where = buildWhereClause(options2);
+      const { sql, params } = buildAggregateSql(tableName, where, options2);
+      const rows = db.query(sql).all(...params);
+      return rows.map(mapAggregateRow);
+    },
     async deleteOlderThan(timestamp) {
       await ensureMigrated();
       const before = timestamp;