@dudousxd/nestjs-agent 1.26.0 → 1.27.1

package/dist/index.cjs

@@ -5190,7 +5190,7 @@ var LedgerQuotaProvider = class {
     if (this.store.usageBetween !== void 0) {
       const range = (0, import_nestjs_agent_core28.quotaPeriodRange)("month", now);
       const used = await this.store.usageBetween(actorRef, range.fromDay, range.toDay);
-      windows.push(this.window("month", used.usedTokens, used.costUsd, range.resetsAt));
+      windows.push(this.window("month", used.usedTokens, this.usd(used), range.resetsAt));
     }
     const blocked = (0, import_nestjs_agent_core28.exhaustedWindow)(windows);
     const warning = (0, import_nestjs_agent_core28.quotaWarning)(windows);
@@ -5206,8 +5206,15 @@ var LedgerQuotaProvider = class {
   }
   async dayWindow(actorRef, now) {
     const range = (0, import_nestjs_agent_core28.quotaPeriodRange)("day", now);
-    const { usedTokens, costUsd } = await this.store.quotaToday(actorRef, range.fromDay);
-    return this.window("day", usedTokens, costUsd, range.resetsAt);
+    const used = await this.store.quotaToday(actorRef, range.fromDay);
+    return this.window("day", used.usedTokens, this.usd(used), range.resetsAt);
+  }
+  /** The window's USD: every priced row, or (with `countEstimatedCost: false`) reported cost only. */
+  usd(used) {
+    if (this.options.countEstimatedCost === false) {
+      return Math.max(0, used.costUsd - (used.estimatedCostUsd ?? 0));
+    }
+    return used.costUsd;
   }
   window(period, usedTokens, usedUsd, resetsAt) {
     const limits = this.limits[period] ?? {};
@@ -6312,9 +6319,14 @@ function sharedProviders(durable, engine) {
       // `quota` is either a provider of the host's own or ceilings for the built-in ledger one;
       // omitted, the ledger provider still reports usage, with no ceiling to block on.
       provide: import_nestjs_agent_core34.AGENT_QUOTA_PROVIDER,
-      useFactory: /* @__PURE__ */ __name((o, store) => isQuotaProvider(o.quota) ? o.quota : new LedgerQuotaProvider(store, o.quota?.limits, o.quota?.warnAt !== void 0 ? {
-        warnAt: o.quota.warnAt
-      } : {}), "useFactory"),
+      useFactory: /* @__PURE__ */ __name((o, store) => isQuotaProvider(o.quota) ? o.quota : new LedgerQuotaProvider(store, o.quota?.limits, {
+        ...o.quota?.warnAt !== void 0 ? {
+          warnAt: o.quota.warnAt
+        } : {},
+        ...o.quota?.countEstimatedCost !== void 0 ? {
+          countEstimatedCost: o.quota.countEstimatedCost
+        } : {}
+      }), "useFactory"),
       inject: [
         import_nestjs_agent_core34.AGENT_OPTIONS,
         import_nestjs_agent_core34.AGENT_STORE