@dudousxd/nestjs-agent-store-drizzle 0.16.0 → 0.17.0

package/dist/index.d.cts

@@ -4036,6 +4036,10 @@ declare class DrizzleAgentStore implements AgentStore, ThreadTurnReader {
         usedTokens: number;
         costUsd: number;
     }>;
+    usageBetween(actorRef: string, fromDay: string, toDay: string): Promise<{
+        usedTokens: number;
+        costUsd: number;
+    }>;
     private toSummary;
     private toStoredMessage;
 }

package/dist/index.d.ts

@@ -4036,6 +4036,10 @@ declare class DrizzleAgentStore implements AgentStore, ThreadTurnReader {
         usedTokens: number;
         costUsd: number;
     }>;
+    usageBetween(actorRef: string, fromDay: string, toDay: string): Promise<{
+        usedTokens: number;
+        costUsd: number;
+    }>;
     private toSummary;
     private toStoredMessage;
 }

package/dist/index.js

@@ -978,8 +978,11 @@ var DrizzleAgentStore = class {
     });
   }
   async quotaToday(actorRef, day) {
-    const start = /* @__PURE__ */ new Date(`${day}T00:00:00.000Z`);
-    const end = /* @__PURE__ */ new Date(`${day}T23:59:59.999Z`);
+    return this.usageBetween(actorRef, day, day);
+  }
+  async usageBetween(actorRef, fromDay, toDay) {
+    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(agentTokenUsage).where(and(eq(agentTokenUsage.actorRef, actorRef), gte(agentTokenUsage.createdAt, start), lte(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);