@dudousxd/nestjs-agent-store-drizzle 0.28.6 → 0.29.0

package/dist/index.cjs

@@ -237,8 +237,10 @@ var agentTokenUsage = (0, import_sqlite_core.sqliteTable)("agent_token_usage", {
   cacheWriteTokens: (0, import_sqlite_core.integer)("cache_write_tokens"),
   /** Subset of `inputTokens` served from the prompt cache this turn; null when not reported. */
   cacheReadTokens: (0, import_sqlite_core.integer)("cache_read_tokens"),
-  /** Provider-reported actual USD cost for the turn; null when only tokens were reported. */
+  /** The turn's USD cost: provider-reported, or estimated from the price row (see `costSource`); null when unpriced. */
   costUsd: (0, import_sqlite_core.real)("cost_usd"),
+  /** Where `cost_usd` came from: `'provider'` (reported) or `'estimate'` (tokens × the price row); null on rows from before it existed (reported). */
+  costSource: (0, import_sqlite_core.text)("cost_source").$type(),
   createdAt: (0, import_sqlite_core.integer)("created_at", {
     mode: "timestamp_ms"
   }).notNull()
@@ -544,6 +546,8 @@ var agentTokenUsage2 = (0, import_mysql_core.mysqlTable)("agent_token_usage", {
   cacheWriteTokens: (0, import_mysql_core.int)("cache_write_tokens"),
   cacheReadTokens: (0, import_mysql_core.int)("cache_read_tokens"),
   costUsd: (0, import_mysql_core.double)("cost_usd"),
+  /** Where `cost_usd` came from: `'provider'` (reported) or `'estimate'` (tokens × the price row); null on rows from before it existed (reported). */
+  costSource: key("cost_source").$type(),
   createdAt: ts("created_at").notNull()
 }, (table) => [
   (0, import_mysql_core.index)("agent_token_usage_actor_created_idx").on(table.actorRef, table.createdAt)
@@ -825,6 +829,8 @@ var agentTokenUsage3 = (0, import_pg_core.pgTable)("agent_token_usage", {
   cacheWriteTokens: (0, import_pg_core.integer)("cache_write_tokens"),
   cacheReadTokens: (0, import_pg_core.integer)("cache_read_tokens"),
   costUsd: (0, import_pg_core.doublePrecision)("cost_usd"),
+  /** Where `cost_usd` came from: `'provider'` (reported) or `'estimate'` (tokens × the price row); null on rows from before it existed (reported). */
+  costSource: (0, import_pg_core.text)("cost_source").$type(),
   createdAt: ts2("created_at").notNull()
 }, (table) => [
   (0, import_pg_core.index)("agent_token_usage_actor_created_idx").on(table.actorRef, table.createdAt)
@@ -2582,6 +2588,7 @@ var DrizzleAgentStore = class {
       cacheWriteTokens: input.usage.cacheWriteTokens ?? null,
       cacheReadTokens: input.usage.cacheReadTokens ?? null,
       costUsd: input.costUsd ?? null,
+      costSource: input.costUsd !== void 0 ? input.costSource ?? "provider" : null,
       createdAt: /* @__PURE__ */ new Date()
     });
   }
@@ -2592,12 +2599,7 @@ var DrizzleAgentStore = class {
     const start = /* @__PURE__ */ new Date(`${fromDay}T00:00:00.000Z`);
     const end = /* @__PURE__ */ new Date(`${toDay}T23:59:59.999Z`);
     const rows = await this.db.select().from(this.t.agentTokenUsage).where((0, import_drizzle_orm4.and)((0, import_drizzle_orm4.eq)(this.t.agentTokenUsage.actorRef, actorRef), (0, import_drizzle_orm4.gte)(this.t.agentTokenUsage.createdAt, start), (0, import_drizzle_orm4.lte)(this.t.agentTokenUsage.createdAt, end)));
-    const usedTokens = rows.reduce((sum2, row) => sum2 + row.inputTokens + row.outputTokens, 0);
-    const costUsd = rows.reduce((sum2, row) => sum2 + (row.costUsd ?? 0), 0);
-    return {
-      usedTokens,
-      costUsd
-    };
+    return (0, import_nestjs_agent_core4.sumUsage)(rows);
   }
   toSummary(thread, lastContent) {
     return {