@dudousxd/nestjs-agent-store-drizzle 0.35.0 → 0.36.0

package/dist/index.cjs

@@ -271,6 +271,10 @@ var agentTokenUsage = (0, import_sqlite_core.sqliteTable)("agent_token_usage", {
   metadata: (0, import_sqlite_core.text)("metadata", {
     mode: "json"
   }).$type(),
+  /** `false` keeps the row out of the agent's quota windows (`recordUsage({ countsTowardsQuota: false })`). */
+  countsTowardsQuota: (0, import_sqlite_core.integer)("counts_towards_quota", {
+    mode: "boolean"
+  }).notNull().default(true),
   createdAt: (0, import_sqlite_core.integer)("created_at", {
     mode: "timestamp_ms"
   }).notNull()
@@ -744,6 +748,8 @@ var agentTokenUsage2 = (0, import_mysql_core.mysqlTable)("agent_token_usage", {
   requests: (0, import_mysql_core.int)("requests"),
   /** The caller's own JSON, never read by the library. */
   metadata: (0, import_mysql_core.json)("metadata").$type(),
+  /** `false` keeps the row out of the agent's quota windows (`recordUsage({ countsTowardsQuota: false })`). */
+  countsTowardsQuota: (0, import_mysql_core.boolean)("counts_towards_quota").notNull().default(true),
   createdAt: ts("created_at").notNull()
 }, (table) => [
   (0, import_mysql_core.index)("agent_token_usage_actor_created_idx").on(table.actorRef, table.createdAt),
@@ -1229,6 +1235,8 @@ var agentTokenUsage3 = (0, import_pg_core.pgTable)("agent_token_usage", {
   requests: (0, import_pg_core.integer)("requests"),
   /** The caller's own JSON, never read by the library. */
   metadata: (0, import_pg_core.jsonb)("metadata").$type(),
+  /** `false` keeps the row out of the agent's quota windows (`recordUsage({ countsTowardsQuota: false })`). */
+  countsTowardsQuota: (0, import_pg_core.boolean)("counts_towards_quota").notNull().default(true),
   createdAt: ts2("created_at").notNull()
 }, (table) => [
   (0, import_pg_core.index)("agent_token_usage_actor_created_idx").on(table.actorRef, table.createdAt),
@@ -3299,6 +3307,7 @@ var DrizzleAgentStore = class {
       characters: input.characters ?? null,
       requests: input.requests ?? null,
       metadata: input.metadata ?? null,
+      countsTowardsQuota: input.countsTowardsQuota !== false,
       modelId: input.modelId,
       purpose: input.purpose,
       inputTokens: input.usage.inputTokens,
@@ -3383,7 +3392,7 @@ var DrizzleAgentStore = class {
     });
     const from = new Date(Math.min(...bounds.map(({ start }) => start.getTime())));
     const to = new Date(Math.max(...bounds.map(({ end }) => end.getTime())));
-    const [row] = await this.db.select(columns).from(usage).where((0, import_drizzle_orm4.and)((0, import_drizzle_orm4.eq)(usage.actorRef, actorRef), within(from, to)));
+    const [row] = await this.db.select(columns).from(usage).where((0, import_drizzle_orm4.and)((0, import_drizzle_orm4.eq)(usage.actorRef, actorRef), within(from, to), (0, import_drizzle_orm4.eq)(usage.countsTowardsQuota, true)));
     return bounds.map((_, index4) => ({
       usedTokens: Number(row?.[`tokens${index4}`] ?? 0),
       costUsd: Number(row?.[`cost${index4}`] ?? 0),
@@ -3710,7 +3719,7 @@ var DrizzleGovernanceQueries = class {
   * per-feature reads that group by them.
   */
   async usageInRange(range, withFeature = false) {
-    const { start, end } = (0, import_nestjs_agent_core5.dayBoundsUtc)(range);
+    const { start, end } = (0, import_nestjs_agent_core5.usageRangeBounds)(range);
     const usage = this.t.agentTokenUsage;
     const rows = await this.db.select({
       modelId: usage.modelId,
@@ -3730,7 +3739,7 @@ var DrizzleGovernanceQueries = class {
         audioSeconds: usage.audioSeconds,
         characters: usage.characters
       } : {}
-    }).from(usage).where((0, import_drizzle_orm5.and)((0, import_drizzle_orm5.gte)(this.t.agentTokenUsage.createdAt, start), (0, import_drizzle_orm5.lte)(this.t.agentTokenUsage.createdAt, end)));
+    }).from(usage).where((0, import_drizzle_orm5.and)((0, import_drizzle_orm5.gte)(this.t.agentTokenUsage.createdAt, start), (0, import_drizzle_orm5.lt)(this.t.agentTokenUsage.createdAt, end)));
     return rows.map(toUsageInput);
   }
   async spendByModel(range) {