caplyr 0.2.0 → 0.2.2

package/dist/index.js

@@ -124,12 +124,30 @@ var Heartbeat = class {
     };
     /** Current protection status */
     this.status = "ACTIVE";
-    this.endpoint = config.endpoint ?? "https://api.caplyr.com";
+    /** 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;
     this.onStatusChange = config.onStatusChange;
     this.onError = config.onError;
   }
+  /**
+   * Apply local budget limits from config.
+   * These act as client-side enforcement even if the server
+   * doesn't return limits (e.g. during heartbeat failures).
+   */
+  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;
+    }
+  }
   /**
    * Start the heartbeat loop.
    * Immediately sends first heartbeat, then repeats on interval.
@@ -156,7 +174,17 @@ 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;
+      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
+      };
       this.consecutiveFailures = 0;
       const newStatus = data.kill_switch_active ? "OFF" : data.status;
       if (newStatus !== this.status) {
@@ -665,8 +693,9 @@ function protect(client, config) {
       "Caplyr: apiKey is required. Get yours at https://app.caplyr.com"
     );
   }
+  const userWantsProtect = config.budget && !config.mode;
   const resolvedConfig = {
-    mode: config.budget && !config.mode ? "cost_protect" : "alert_only",
+    mode: userWantsProtect ? "cost_protect" : "alert_only",
     downgradeThreshold: 0.8,
     batchSize: 10,
     flushInterval: 3e4,
@@ -682,6 +711,10 @@ function protect(client, config) {
     instances.set(resolvedConfig.apiKey, shared);
   }
   const { shipper, heartbeat } = shared;
+  if (resolvedConfig.budget) {
+    const budgetConfig = typeof resolvedConfig.budget === "number" ? { monthly: resolvedConfig.budget } : resolvedConfig.budget;
+    heartbeat.applyLocalLimits(budgetConfig);
+  }
   const provider = detectProvider(client);
   switch (provider) {
     case "anthropic":

package/dist/index.mjs

@@ -91,12 +91,30 @@ var Heartbeat = class {
     };
     /** Current protection status */
     this.status = "ACTIVE";
-    this.endpoint = config.endpoint ?? "https://api.caplyr.com";
+    /** 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;
     this.onStatusChange = config.onStatusChange;
     this.onError = config.onError;
   }
+  /**
+   * Apply local budget limits from config.
+   * These act as client-side enforcement even if the server
+   * doesn't return limits (e.g. during heartbeat failures).
+   */
+  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;
+    }
+  }
   /**
    * Start the heartbeat loop.
    * Immediately sends first heartbeat, then repeats on interval.
@@ -123,7 +141,17 @@ 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;
+      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
+      };
       this.consecutiveFailures = 0;
       const newStatus = data.kill_switch_active ? "OFF" : data.status;
       if (newStatus !== this.status) {
@@ -632,8 +660,9 @@ function protect(client, config) {
       "Caplyr: apiKey is required. Get yours at https://app.caplyr.com"
     );
   }
+  const userWantsProtect = config.budget && !config.mode;
   const resolvedConfig = {
-    mode: config.budget && !config.mode ? "cost_protect" : "alert_only",
+    mode: userWantsProtect ? "cost_protect" : "alert_only",
     downgradeThreshold: 0.8,
     batchSize: 10,
     flushInterval: 3e4,
@@ -649,6 +678,10 @@ function protect(client, config) {
     instances.set(resolvedConfig.apiKey, shared);
   }
   const { shipper, heartbeat } = shared;
+  if (resolvedConfig.budget) {
+    const budgetConfig = typeof resolvedConfig.budget === "number" ? { monthly: resolvedConfig.budget } : resolvedConfig.budget;
+    heartbeat.applyLocalLimits(budgetConfig);
+  }
   const provider = detectProvider(client);
   switch (provider) {
     case "anthropic":

package/package.json

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