caplyr 0.2.2 → 0.2.3

package/dist/index.js

@@ -176,14 +176,18 @@ var Heartbeat = class {
       const data = await res.json();
       const localDailyUsed = this.budgetStatus.daily_used;
       const localMonthlyUsed = this.budgetStatus.monthly_used;
+      const serverDailyUsed = Number(data.daily_used) || 0;
+      const serverMonthlyUsed = Number(data.monthly_used) || 0;
+      const serverDailyLimit = data.daily_limit != null ? Number(data.daily_limit) : null;
+      const serverMonthlyLimit = data.monthly_limit != null ? Number(data.monthly_limit) : null;
       this.budgetStatus = {
         ...data,
         // Use whichever spend is higher — server or local tracking
-        daily_used: Math.max(data.daily_used ?? 0, localDailyUsed),
-        monthly_used: Math.max(data.monthly_used ?? 0, localMonthlyUsed),
-        // Preserve local limits if server doesn't provide them
-        daily_limit: data.daily_limit ?? this.localDailyLimit,
-        monthly_limit: data.monthly_limit ?? this.localMonthlyLimit
+        daily_used: Math.max(serverDailyUsed, localDailyUsed),
+        monthly_used: Math.max(serverMonthlyUsed, localMonthlyUsed),
+        // Use the stricter (lower) limit — local config takes priority if lower
+        daily_limit: this.pickStricterLimit(serverDailyLimit, this.localDailyLimit),
+        monthly_limit: this.pickStricterLimit(serverMonthlyLimit, this.localMonthlyLimit)
       };
       this.consecutiveFailures = 0;
       const newStatus = data.kill_switch_active ? "OFF" : data.status;
@@ -200,6 +204,22 @@ var Heartbeat = class {
       }
     }
   }
+  /**
+   * Pick the stricter (lower) of two limits.
+   * If one is null, use the other.
+   */
+  pickStricterLimit(a, b) {
+    if (a === null) return b;
+    if (b === null) return a;
+    return Math.min(a, b);
+  }
+  /**
+   * Force an immediate heartbeat poll (useful for kill switch checks).
+   * Returns a promise that resolves when the poll completes.
+   */
+  async forcePoll() {
+    await this.beat();
+  }
   /**
    * Update local budget tracking (called after each request).
    * This provides real-time budget awareness between heartbeats.

package/dist/index.mjs

@@ -143,14 +143,18 @@ var Heartbeat = class {
       const data = await res.json();
       const localDailyUsed = this.budgetStatus.daily_used;
       const localMonthlyUsed = this.budgetStatus.monthly_used;
+      const serverDailyUsed = Number(data.daily_used) || 0;
+      const serverMonthlyUsed = Number(data.monthly_used) || 0;
+      const serverDailyLimit = data.daily_limit != null ? Number(data.daily_limit) : null;
+      const serverMonthlyLimit = data.monthly_limit != null ? Number(data.monthly_limit) : null;
       this.budgetStatus = {
         ...data,
         // Use whichever spend is higher — server or local tracking
-        daily_used: Math.max(data.daily_used ?? 0, localDailyUsed),
-        monthly_used: Math.max(data.monthly_used ?? 0, localMonthlyUsed),
-        // Preserve local limits if server doesn't provide them
-        daily_limit: data.daily_limit ?? this.localDailyLimit,
-        monthly_limit: data.monthly_limit ?? this.localMonthlyLimit
+        daily_used: Math.max(serverDailyUsed, localDailyUsed),
+        monthly_used: Math.max(serverMonthlyUsed, localMonthlyUsed),
+        // Use the stricter (lower) limit — local config takes priority if lower
+        daily_limit: this.pickStricterLimit(serverDailyLimit, this.localDailyLimit),
+        monthly_limit: this.pickStricterLimit(serverMonthlyLimit, this.localMonthlyLimit)
       };
       this.consecutiveFailures = 0;
       const newStatus = data.kill_switch_active ? "OFF" : data.status;
@@ -167,6 +171,22 @@ var Heartbeat = class {
       }
     }
   }
+  /**
+   * Pick the stricter (lower) of two limits.
+   * If one is null, use the other.
+   */
+  pickStricterLimit(a, b) {
+    if (a === null) return b;
+    if (b === null) return a;
+    return Math.min(a, b);
+  }
+  /**
+   * Force an immediate heartbeat poll (useful for kill switch checks).
+   * Returns a promise that resolves when the poll completes.
+   */
+  async forcePoll() {
+    await this.beat();
+  }
   /**
    * Update local budget tracking (called after each request).
    * This provides real-time budget awareness between heartbeats.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "caplyr",
-  "version": "0.2.2",
+  "version": "0.2.3",
   "description": "AI Cost Control Plane — budget guardrails, auto-downgrade, and kill switch for AI API calls",
   "main": "dist/index.js",
   "module": "dist/index.mjs",