caplyr 0.2.1 → 0.2.3

package/dist/index.js

@@ -124,6 +124,9 @@ var Heartbeat = class {
     };
     /** Current protection status */
     this.status = "ACTIVE";
+    /** Local budget limits set via config (not from server) */
+    this.localDailyLimit = null;
+    this.localMonthlyLimit = null;
     this.endpoint = config.endpoint ?? "https://caplyr.com";
     this.apiKey = config.apiKey;
     this.interval = config.heartbeatInterval ?? 6e4;
@@ -137,9 +140,11 @@ var Heartbeat = class {
    */
   applyLocalLimits(budget) {
     if (budget.daily !== void 0) {
+      this.localDailyLimit = budget.daily;
       this.budgetStatus.daily_limit = budget.daily;
     }
     if (budget.monthly !== void 0) {
+      this.localMonthlyLimit = budget.monthly;
       this.budgetStatus.monthly_limit = budget.monthly;
     }
   }
@@ -169,7 +174,21 @@ var Heartbeat = class {
         throw new Error(`Heartbeat failed: ${res.status}`);
       }
       const data = await res.json();
-      this.budgetStatus = data;
+      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(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;
       if (newStatus !== this.status) {
@@ -185,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

@@ -91,6 +91,9 @@ var Heartbeat = class {
     };
     /** Current protection status */
     this.status = "ACTIVE";
+    /** Local budget limits set via config (not from server) */
+    this.localDailyLimit = null;
+    this.localMonthlyLimit = null;
     this.endpoint = config.endpoint ?? "https://caplyr.com";
     this.apiKey = config.apiKey;
     this.interval = config.heartbeatInterval ?? 6e4;
@@ -104,9 +107,11 @@ var Heartbeat = class {
    */
   applyLocalLimits(budget) {
     if (budget.daily !== void 0) {
+      this.localDailyLimit = budget.daily;
       this.budgetStatus.daily_limit = budget.daily;
     }
     if (budget.monthly !== void 0) {
+      this.localMonthlyLimit = budget.monthly;
       this.budgetStatus.monthly_limit = budget.monthly;
     }
   }
@@ -136,7 +141,21 @@ var Heartbeat = class {
         throw new Error(`Heartbeat failed: ${res.status}`);
       }
       const data = await res.json();
-      this.budgetStatus = data;
+      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(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;
       if (newStatus !== this.status) {
@@ -152,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.1",
+  "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",