caplyr 0.1.1 → 0.1.3

package/dist/index.mjs

@@ -1,4 +1,5 @@
 // src/logger.ts
+var DEFAULT_MAX_BUFFER = 500;
 var LogShipper = class {
   constructor(config) {
     this.buffer = [];
@@ -7,6 +8,7 @@ var LogShipper = class {
     this.apiKey = config.apiKey;
     this.batchSize = config.batchSize ?? 10;
     this.flushInterval = config.flushInterval ?? 3e4;
+    this.maxBufferSize = config.maxBufferSize ?? DEFAULT_MAX_BUFFER;
     this.onError = config.onError;
     this.timer = setInterval(() => this.flush(), this.flushInterval);
     if (typeof process !== "undefined" && process.on) {
@@ -19,20 +21,12 @@ var LogShipper = class {
       process.on("SIGINT", flushAndExit);
     }
   }
-  /**
-   * Add a log entry to the buffer.
-   * Auto-flushes when batch size is reached.
-   */
   push(log) {
     this.buffer.push(log);
     if (this.buffer.length >= this.batchSize) {
       this.flush();
     }
   }
-  /**
-   * Flush all buffered logs to the backend.
-   * Non-blocking — errors are swallowed and reported via onError.
-   */
   async flush() {
     if (this.buffer.length === 0) return;
     const batch = this.buffer.splice(0);
@@ -47,7 +41,7 @@ var LogShipper = class {
         signal: AbortSignal.timeout(1e4)
       });
       if (!res.ok) {
-        this.buffer.unshift(...batch);
+        this.requeueFailed(batch);
         throw new Error(`Ingest failed: ${res.status} ${res.statusText}`);
       }
     } catch (err) {
@@ -55,9 +49,14 @@ var LogShipper = class {
     }
   }
   /**
-   * Stop the periodic flush timer.
-   * Call this when tearing down the SDK.
+   * Re-queue failed logs, dropping oldest if buffer would exceed max size.
    */
+  requeueFailed(batch) {
+    const available = this.maxBufferSize - this.buffer.length;
+    if (available <= 0) return;
+    const toKeep = batch.slice(-available);
+    this.buffer.unshift(...toKeep);
+  }
   destroy() {
     if (this.timer) {
       clearInterval(this.timer);
@@ -80,7 +79,6 @@ var Heartbeat = class {
     this.timer = null;
     this.consecutiveFailures = 0;
     this.maxFailuresBeforeDegraded = 3;
-    /** Current budget status from the backend */
     this.budgetStatus = {
       daily_used: 0,
       daily_limit: null,
@@ -89,7 +87,6 @@ var Heartbeat = class {
       status: "ACTIVE",
       kill_switch_active: false
     };
-    /** Current protection status */
     this.status = "ACTIVE";
     this.endpoint = config.endpoint ?? "https://api.caplyr.com";
     this.apiKey = config.apiKey;
@@ -97,17 +94,10 @@ var Heartbeat = class {
     this.onStatusChange = config.onStatusChange;
     this.onError = config.onError;
   }
-  /**
-   * Start the heartbeat loop.
-   * Immediately sends first heartbeat, then repeats on interval.
-   */
   start() {
     this.beat();
     this.timer = setInterval(() => this.beat(), this.interval);
   }
-  /**
-   * Send a single heartbeat and update local state.
-   */
   async beat() {
     try {
       const res = await fetch(`${this.endpoint}/api/heartbeat`, {
@@ -119,9 +109,7 @@ var Heartbeat = class {
         body: JSON.stringify({ timestamp: Date.now() }),
         signal: AbortSignal.timeout(5e3)
       });
-      if (!res.ok) {
-        throw new Error(`Heartbeat failed: ${res.status}`);
-      }
+      if (!res.ok) throw new Error(`Heartbeat failed: ${res.status}`);
       const data = await res.json();
       this.budgetStatus = data;
       this.consecutiveFailures = 0;
@@ -139,32 +127,18 @@ var Heartbeat = class {
       }
     }
   }
-  /**
-   * Update local budget tracking (called after each request).
-   * This provides real-time budget awareness between heartbeats.
-   */
   trackSpend(cost) {
     this.budgetStatus.daily_used += cost;
     this.budgetStatus.monthly_used += cost;
   }
-  /**
-   * Check if the monthly budget is exceeded.
-   */
   isMonthlyBudgetExceeded() {
     if (this.budgetStatus.monthly_limit === null) return false;
     return this.budgetStatus.monthly_used >= this.budgetStatus.monthly_limit;
   }
-  /**
-   * Check if the daily budget is exceeded.
-   */
   isDailyBudgetExceeded() {
     if (this.budgetStatus.daily_limit === null) return false;
     return this.budgetStatus.daily_used >= this.budgetStatus.daily_limit;
   }
-  /**
-   * Check if the downgrade threshold is reached.
-   * Returns true if usage exceeds the given threshold (0-1) of any budget.
-   */
   isDowngradeThresholdReached(threshold) {
     if (this.budgetStatus.monthly_limit !== null && this.budgetStatus.monthly_used >= this.budgetStatus.monthly_limit * threshold) {
       return true;
@@ -174,15 +148,9 @@ var Heartbeat = class {
     }
     return false;
   }
-  /**
-   * Check if the kill switch is active.
-   */
   isKillSwitchActive() {
     return this.budgetStatus.kill_switch_active;
   }
-  /**
-   * Stop the heartbeat loop.
-   */
   destroy() {
     if (this.timer) {
       clearInterval(this.timer);
@@ -197,7 +165,6 @@ var MODEL_PRICING = {
   "claude-opus-4-20250514": { input: 15, output: 75 },
   "claude-sonnet-4-20250514": { input: 3, output: 15 },
   "claude-haiku-4-5-20251001": { input: 0.8, output: 4 },
-  // Aliases
   "claude-opus-4": { input: 15, output: 75 },
   "claude-sonnet-4": { input: 3, output: 15 },
   "claude-3-5-sonnet-20241022": { input: 3, output: 15 },
@@ -215,13 +182,11 @@ var MODEL_PRICING = {
   "o3-mini": { input: 1.1, output: 4.4 }
 };
 var DEFAULT_FALLBACKS = {
-  // Anthropic downgrades
   "claude-opus-4-20250514": "claude-sonnet-4-20250514",
   "claude-opus-4": "claude-sonnet-4",
   "claude-sonnet-4-20250514": "claude-haiku-4-5-20251001",
   "claude-sonnet-4": "claude-haiku-4-5-20251001",
   "claude-3-5-sonnet-20241022": "claude-haiku-4-5-20251001",
-  // OpenAI downgrades
   "gpt-4o": "gpt-4o-mini",
   "gpt-4o-2024-11-20": "gpt-4o-mini",
   "gpt-4-turbo": "gpt-4o-mini",
@@ -241,9 +206,7 @@ function getDefaultFallback(model) {
 }
 function registerModel(model, pricing, fallback) {
   MODEL_PRICING[model] = pricing;
-  if (fallback) {
-    DEFAULT_FALLBACKS[model] = fallback;
-  }
+  if (fallback) DEFAULT_FALLBACKS[model] = fallback;
 }
 function isKnownModel(model) {
   return model in MODEL_PRICING;
@@ -252,160 +215,184 @@ function getModelPricing(model) {
   return MODEL_PRICING[model] ?? null;
 }
 
-// src/interceptors/anthropic.ts
+// src/utils.ts
 var idCounter = 0;
 function generateId() {
-  return `caplyr_${Date.now()}_${++idCounter}`;
+  return `caplyr_${Date.now()}_${process.pid ?? 0}_${++idCounter}`;
 }
-function wrapAnthropic(client, config, shipper, heartbeat) {
-  const downgradeThreshold = config.downgradeThreshold ?? 0.8;
+function getNextResetTime(reason) {
+  const now = /* @__PURE__ */ new Date();
+  if (reason.includes("daily")) {
+    const tomorrow = new Date(now);
+    tomorrow.setDate(tomorrow.getDate() + 1);
+    tomorrow.setHours(0, 0, 0, 0);
+    return tomorrow.toISOString();
+  }
+  const nextMonth = new Date(now.getFullYear(), now.getMonth() + 1, 1);
+  return nextMonth.toISOString();
+}
+
+// src/interceptors/enforce.ts
+function enforcePreCall(model, config, heartbeat, shipper, provider, startTime) {
   const dashboardUrl = `${config.endpoint ?? "https://app.caplyr.com"}/dashboard`;
+  const downgradeThreshold = config.downgradeThreshold ?? 0.8;
+  if (heartbeat.isKillSwitchActive()) {
+    const reason = "kill_switch_active";
+    const blockError = {
+      code: "KILL_SWITCH_ACTIVE",
+      message: "Caplyr kill switch is active. All AI API calls are halted.",
+      budget_used: heartbeat.budgetStatus.monthly_used,
+      budget_limit: heartbeat.budgetStatus.monthly_limit ?? 0,
+      dashboard_url: dashboardUrl
+    };
+    pushBlockedLog(shipper, provider, model, startTime, reason, config.endpoint_tag);
+    if (config.mode === "alert_only") {
+      config.onEnforcement?.({
+        type: "kill_switch",
+        timestamp: Date.now(),
+        reason: blockError.message,
+        budget_used: blockError.budget_used,
+        budget_limit: blockError.budget_limit
+      });
+      return { proceed: true, model, downgraded: false, enforcementReason: reason };
+    }
+    throw Object.assign(new Error(blockError.message), { caplyr: blockError });
+  }
+  if (config.mode === "cost_protect") {
+    if (heartbeat.isMonthlyBudgetExceeded() || heartbeat.isDailyBudgetExceeded()) {
+      const reason = heartbeat.isDailyBudgetExceeded() ? "daily_budget_exceeded" : "monthly_budget_exceeded";
+      const blockError = {
+        code: "BUDGET_EXCEEDED",
+        message: `AI budget exceeded. ${reason.replace(/_/g, " ")}.`,
+        budget_used: heartbeat.budgetStatus.monthly_used,
+        budget_limit: heartbeat.budgetStatus.monthly_limit ?? 0,
+        retry_after: getNextResetTime(reason),
+        dashboard_url: dashboardUrl
+      };
+      pushBlockedLog(shipper, provider, model, startTime, reason, config.endpoint_tag);
+      throw Object.assign(new Error(blockError.message), { caplyr: blockError });
+    }
+    if (heartbeat.isDowngradeThresholdReached(downgradeThreshold)) {
+      const fallback = config.fallback ?? getDefaultFallback(model);
+      if (fallback && fallback !== model) {
+        config.onEnforcement?.({
+          type: "downgrade",
+          timestamp: Date.now(),
+          reason: `Budget at ${Math.round(downgradeThreshold * 100)}% \u2014 downgraded ${model} \u2192 ${fallback}`,
+          original_model: model,
+          fallback_model: fallback,
+          budget_used: heartbeat.budgetStatus.monthly_used,
+          budget_limit: heartbeat.budgetStatus.monthly_limit ?? 0,
+          estimated_savings: 0
+        });
+        return {
+          proceed: true,
+          model: fallback,
+          downgraded: true,
+          originalModel: model,
+          enforcementReason: "auto_downgrade_threshold"
+        };
+      }
+    }
+  }
+  return { proceed: true, model, downgraded: false };
+}
+function logSuccess(shipper, provider, model, startTime, inputTokens, outputTokens, enforcement, heartbeat, endpointTag) {
+  const cost = calculateCost(model, inputTokens, outputTokens);
+  heartbeat.trackSpend(cost);
+  shipper.push({
+    id: generateId(),
+    timestamp: startTime,
+    provider,
+    model,
+    input_tokens: inputTokens,
+    output_tokens: outputTokens,
+    cost,
+    latency_ms: Date.now() - startTime,
+    endpoint_tag: endpointTag,
+    downgraded: enforcement.downgraded,
+    original_model: enforcement.originalModel,
+    blocked: false,
+    enforcement_reason: enforcement.enforcementReason
+  });
+  return cost;
+}
+function logProviderError(shipper, provider, model, startTime, enforcement, endpointTag) {
+  shipper.push({
+    id: generateId(),
+    timestamp: startTime,
+    provider,
+    model,
+    input_tokens: 0,
+    output_tokens: 0,
+    cost: 0,
+    latency_ms: Date.now() - startTime,
+    endpoint_tag: endpointTag,
+    downgraded: enforcement.downgraded,
+    original_model: enforcement.originalModel,
+    blocked: false,
+    enforcement_reason: "provider_error"
+  });
+}
+function pushBlockedLog(shipper, provider, model, startTime, reason, endpointTag) {
+  shipper.push({
+    id: generateId(),
+    timestamp: startTime,
+    provider,
+    model,
+    input_tokens: 0,
+    output_tokens: 0,
+    cost: 0,
+    latency_ms: Date.now() - startTime,
+    endpoint_tag: endpointTag,
+    downgraded: false,
+    blocked: true,
+    enforcement_reason: reason
+  });
+}
+
+// src/interceptors/anthropic.ts
+function wrapAnthropic(client, config, shipper, heartbeat) {
   const messagesProxy = new Proxy(client.messages, {
     get(target, prop, receiver) {
       if (prop === "create") {
         return async function caplyrInterceptedCreate(params, options) {
           const startTime = Date.now();
-          let model = params.model;
-          let downgraded = false;
-          let originalModel;
-          let blocked = false;
-          let enforcementReason;
-          if (heartbeat.isKillSwitchActive()) {
-            blocked = true;
-            enforcementReason = "kill_switch_active";
-            const blockError = {
-              code: "KILL_SWITCH_ACTIVE",
-              message: "Caplyr kill switch is active. All AI API calls are halted.",
-              budget_used: heartbeat.budgetStatus.monthly_used,
-              budget_limit: heartbeat.budgetStatus.monthly_limit ?? 0,
-              dashboard_url: dashboardUrl
-            };
-            shipper.push({
-              id: generateId(),
-              timestamp: startTime,
-              provider: "anthropic",
-              model,
-              input_tokens: 0,
-              output_tokens: 0,
-              cost: 0,
-              latency_ms: Date.now() - startTime,
-              endpoint_tag: config.endpoint_tag,
-              downgraded: false,
-              blocked: true,
-              enforcement_reason: enforcementReason
-            });
-            if (config.mode === "alert_only") {
-            } else {
-              throw Object.assign(new Error(blockError.message), {
-                caplyr: blockError
-              });
-            }
-          }
-          if (config.mode === "cost_protect") {
-            if (heartbeat.isMonthlyBudgetExceeded() || heartbeat.isDailyBudgetExceeded()) {
-              blocked = true;
-              enforcementReason = heartbeat.isDailyBudgetExceeded() ? "daily_budget_exceeded" : "monthly_budget_exceeded";
-              const blockError = {
-                code: "BUDGET_EXCEEDED",
-                message: `AI budget exceeded. ${enforcementReason.replace(/_/g, " ")}.`,
-                budget_used: heartbeat.budgetStatus.monthly_used,
-                budget_limit: heartbeat.budgetStatus.monthly_limit ?? 0,
-                retry_after: getNextResetTime(enforcementReason),
-                dashboard_url: dashboardUrl
-              };
-              shipper.push({
-                id: generateId(),
-                timestamp: startTime,
-                provider: "anthropic",
-                model,
-                input_tokens: 0,
-                output_tokens: 0,
-                cost: 0,
-                latency_ms: Date.now() - startTime,
-                endpoint_tag: config.endpoint_tag,
-                downgraded: false,
-                blocked: true,
-                enforcement_reason: enforcementReason
-              });
-              throw Object.assign(new Error(blockError.message), {
-                caplyr: blockError
-              });
-            }
-            if (heartbeat.isDowngradeThresholdReached(downgradeThreshold)) {
-              const fallback = config.fallback ?? getDefaultFallback(model);
-              if (fallback && fallback !== model) {
-                originalModel = model;
-                model = fallback;
-                downgraded = true;
-                enforcementReason = "auto_downgrade_threshold";
-                config.onEnforcement?.({
-                  type: "downgrade",
-                  timestamp: Date.now(),
-                  reason: `Budget at ${Math.round(downgradeThreshold * 100)}% \u2014 downgraded ${originalModel} \u2192 ${model}`,
-                  original_model: originalModel,
-                  fallback_model: model,
-                  budget_used: heartbeat.budgetStatus.monthly_used,
-                  budget_limit: heartbeat.budgetStatus.monthly_limit ?? 0,
-                  estimated_savings: 0
-                  // Calculated after response
-                });
-              }
-            }
-          }
-          const requestParams = downgraded ? { ...params, model } : params;
+          const enforcement = enforcePreCall(
+            params.model,
+            config,
+            heartbeat,
+            shipper,
+            "anthropic",
+            startTime
+          );
+          const requestParams = enforcement.downgraded ? { ...params, model: enforcement.model } : params;
           try {
-            const response = await target.create.call(
-              target,
-              requestParams,
-              options
-            );
-            const latency = Date.now() - startTime;
+            const response = await target.create.call(target, requestParams, options);
             const inputTokens = response?.usage?.input_tokens ?? 0;
             const outputTokens = response?.usage?.output_tokens ?? 0;
-            const cost = calculateCost(model, inputTokens, outputTokens);
-            heartbeat.trackSpend(cost);
-            let estimatedSavings = 0;
-            if (downgraded && originalModel) {
-              const originalCost = calculateCost(
-                originalModel,
-                inputTokens,
-                outputTokens
-              );
-              estimatedSavings = originalCost - cost;
-            }
-            shipper.push({
-              id: generateId(),
-              timestamp: startTime,
-              provider: "anthropic",
-              model,
-              input_tokens: inputTokens,
-              output_tokens: outputTokens,
-              cost,
-              latency_ms: latency,
-              endpoint_tag: config.endpoint_tag,
-              downgraded,
-              original_model: originalModel,
-              blocked: false,
-              enforcement_reason: enforcementReason
-            });
+            logSuccess(
+              shipper,
+              "anthropic",
+              enforcement.model,
+              startTime,
+              inputTokens,
+              outputTokens,
+              enforcement,
+              heartbeat,
+              config.endpoint_tag
+            );
             return response;
           } catch (err) {
             if (err?.caplyr) throw err;
-            shipper.push({
-              id: generateId(),
-              timestamp: startTime,
-              provider: "anthropic",
-              model,
-              input_tokens: 0,
-              output_tokens: 0,
-              cost: 0,
-              latency_ms: Date.now() - startTime,
-              endpoint_tag: config.endpoint_tag,
-              downgraded,
-              original_model: originalModel,
-              blocked: false,
-              enforcement_reason: "provider_error"
-            });
+            logProviderError(
+              shipper,
+              "anthropic",
+              enforcement.model,
+              startTime,
+              enforcement,
+              config.endpoint_tag
+            );
             throw err;
           }
         };
@@ -415,170 +402,55 @@ function wrapAnthropic(client, config, shipper, heartbeat) {
   });
   return new Proxy(client, {
     get(target, prop, receiver) {
-      if (prop === "messages") {
-        return messagesProxy;
-      }
+      if (prop === "messages") return messagesProxy;
       return Reflect.get(target, prop, receiver);
     }
   });
 }
-function getNextResetTime(reason) {
-  const now = /* @__PURE__ */ new Date();
-  if (reason.includes("daily")) {
-    const tomorrow = new Date(now);
-    tomorrow.setDate(tomorrow.getDate() + 1);
-    tomorrow.setHours(0, 0, 0, 0);
-    return tomorrow.toISOString();
-  }
-  const nextMonth = new Date(now.getFullYear(), now.getMonth() + 1, 1);
-  return nextMonth.toISOString();
-}
 
 // src/interceptors/openai.ts
-var idCounter2 = 0;
-function generateId2() {
-  return `caplyr_${Date.now()}_${++idCounter2}`;
-}
 function wrapOpenAI(client, config, shipper, heartbeat) {
-  const downgradeThreshold = config.downgradeThreshold ?? 0.8;
-  const dashboardUrl = `${config.endpoint ?? "https://app.caplyr.com"}/dashboard`;
   const completionsProxy = new Proxy(client.chat.completions, {
     get(target, prop, receiver) {
       if (prop === "create") {
         return async function caplyrInterceptedCreate(params, options) {
           const startTime = Date.now();
-          let model = params.model;
-          let downgraded = false;
-          let originalModel;
-          let blocked = false;
-          let enforcementReason;
-          if (heartbeat.isKillSwitchActive()) {
-            blocked = true;
-            enforcementReason = "kill_switch_active";
-            const blockError = {
-              code: "KILL_SWITCH_ACTIVE",
-              message: "Caplyr kill switch is active. All AI API calls are halted.",
-              budget_used: heartbeat.budgetStatus.monthly_used,
-              budget_limit: heartbeat.budgetStatus.monthly_limit ?? 0,
-              dashboard_url: dashboardUrl
-            };
-            shipper.push({
-              id: generateId2(),
-              timestamp: startTime,
-              provider: "openai",
-              model,
-              input_tokens: 0,
-              output_tokens: 0,
-              cost: 0,
-              latency_ms: Date.now() - startTime,
-              endpoint_tag: config.endpoint_tag,
-              downgraded: false,
-              blocked: true,
-              enforcement_reason: enforcementReason
-            });
-            if (config.mode === "alert_only") {
-            } else {
-              throw Object.assign(new Error(blockError.message), {
-                caplyr: blockError
-              });
-            }
-          }
-          if (config.mode === "cost_protect") {
-            if (heartbeat.isMonthlyBudgetExceeded() || heartbeat.isDailyBudgetExceeded()) {
-              blocked = true;
-              enforcementReason = heartbeat.isDailyBudgetExceeded() ? "daily_budget_exceeded" : "monthly_budget_exceeded";
-              const blockError = {
-                code: "BUDGET_EXCEEDED",
-                message: `AI budget exceeded. ${enforcementReason.replace(/_/g, " ")}.`,
-                budget_used: heartbeat.budgetStatus.monthly_used,
-                budget_limit: heartbeat.budgetStatus.monthly_limit ?? 0,
-                retry_after: getNextResetTime2(enforcementReason),
-                dashboard_url: dashboardUrl
-              };
-              shipper.push({
-                id: generateId2(),
-                timestamp: startTime,
-                provider: "openai",
-                model,
-                input_tokens: 0,
-                output_tokens: 0,
-                cost: 0,
-                latency_ms: Date.now() - startTime,
-                endpoint_tag: config.endpoint_tag,
-                downgraded: false,
-                blocked: true,
-                enforcement_reason: enforcementReason
-              });
-              throw Object.assign(new Error(blockError.message), {
-                caplyr: blockError
-              });
-            }
-            if (heartbeat.isDowngradeThresholdReached(downgradeThreshold)) {
-              const fallback = config.fallback ?? getDefaultFallback(model);
-              if (fallback && fallback !== model) {
-                originalModel = model;
-                model = fallback;
-                downgraded = true;
-                enforcementReason = "auto_downgrade_threshold";
-                config.onEnforcement?.({
-                  type: "downgrade",
-                  timestamp: Date.now(),
-                  reason: `Budget at ${Math.round(downgradeThreshold * 100)}% \u2014 downgraded ${originalModel} \u2192 ${model}`,
-                  original_model: originalModel,
-                  fallback_model: model,
-                  budget_used: heartbeat.budgetStatus.monthly_used,
-                  budget_limit: heartbeat.budgetStatus.monthly_limit ?? 0,
-                  estimated_savings: 0
-                });
-              }
-            }
-          }
-          const requestParams = downgraded ? { ...params, model } : params;
+          const enforcement = enforcePreCall(
+            params.model,
+            config,
+            heartbeat,
+            shipper,
+            "openai",
+            startTime
+          );
+          const requestParams = enforcement.downgraded ? { ...params, model: enforcement.model } : params;
           try {
-            const response = await target.create.call(
-              target,
-              requestParams,
-              options
-            );
-            const latency = Date.now() - startTime;
+            const response = await target.create.call(target, requestParams, options);
             const usage = response?.usage;
             const inputTokens = usage?.prompt_tokens ?? 0;
             const outputTokens = usage?.completion_tokens ?? 0;
-            const cost = calculateCost(model, inputTokens, outputTokens);
-            heartbeat.trackSpend(cost);
-            shipper.push({
-              id: generateId2(),
-              timestamp: startTime,
-              provider: "openai",
-              model,
-              input_tokens: inputTokens,
-              output_tokens: outputTokens,
-              cost,
-              latency_ms: latency,
-              endpoint_tag: config.endpoint_tag,
-              downgraded,
-              original_model: originalModel,
-              blocked: false,
-              enforcement_reason: enforcementReason
-            });
+            logSuccess(
+              shipper,
+              "openai",
+              enforcement.model,
+              startTime,
+              inputTokens,
+              outputTokens,
+              enforcement,
+              heartbeat,
+              config.endpoint_tag
+            );
             return response;
           } catch (err) {
             if (err?.caplyr) throw err;
-            shipper.push({
-              id: generateId2(),
-              timestamp: startTime,
-              provider: "openai",
-              model,
-              input_tokens: 0,
-              output_tokens: 0,
-              cost: 0,
-              latency_ms: Date.now() - startTime,
-              endpoint_tag: config.endpoint_tag,
-              downgraded,
-              original_model: originalModel,
-              blocked: false,
-              enforcement_reason: "provider_error"
-            });
+            logProviderError(
+              shipper,
+              "openai",
+              enforcement.model,
+              startTime,
+              enforcement,
+              config.endpoint_tag
+            );
             throw err;
           }
         };
@@ -588,32 +460,17 @@ function wrapOpenAI(client, config, shipper, heartbeat) {
   });
   const chatProxy = new Proxy(client.chat, {
     get(target, prop, receiver) {
-      if (prop === "completions") {
-        return completionsProxy;
-      }
+      if (prop === "completions") return completionsProxy;
       return Reflect.get(target, prop, receiver);
     }
   });
   return new Proxy(client, {
     get(target, prop, receiver) {
-      if (prop === "chat") {
-        return chatProxy;
-      }
+      if (prop === "chat") return chatProxy;
       return Reflect.get(target, prop, receiver);
     }
   });
 }
-function getNextResetTime2(reason) {
-  const now = /* @__PURE__ */ new Date();
-  if (reason.includes("daily")) {
-    const tomorrow = new Date(now);
-    tomorrow.setDate(tomorrow.getDate() + 1);
-    tomorrow.setHours(0, 0, 0, 0);
-    return tomorrow.toISOString();
-  }
-  const nextMonth = new Date(now.getFullYear(), now.getMonth() + 1, 1);
-  return nextMonth.toISOString();
-}
 
 // src/protect.ts
 function detectProvider(client) {
@@ -628,27 +485,20 @@ function detectProvider(client) {
 var instances = /* @__PURE__ */ new Map();
 function protect(client, config) {
   if (!config.apiKey) {
-    throw new Error(
-      "Caplyr: apiKey is required. Get yours at https://app.caplyr.com"
-    );
+    throw new Error("Caplyr: apiKey is required. Get yours at https://app.caplyr.com");
   }
+  const mode = config.mode ?? (config.budget ? "cost_protect" : "alert_only");
   const resolvedConfig = {
-    mode: "alert_only",
-    downgradeThreshold: 0.8,
-    batchSize: 10,
-    flushInterval: 3e4,
-    heartbeatInterval: 6e4,
-    ...config
+    ...config,
+    mode,
+    downgradeThreshold: config.downgradeThreshold ?? 0.8
   };
-  if (resolvedConfig.budget && !resolvedConfig.mode) {
-    resolvedConfig.mode = "cost_protect";
-  }
   let shared = instances.get(resolvedConfig.apiKey);
   if (!shared) {
     const shipper2 = new LogShipper(resolvedConfig);
     const heartbeat2 = new Heartbeat(resolvedConfig);
     heartbeat2.start();
-    shared = { shipper: shipper2, heartbeat: heartbeat2 };
+    shared = { shipper: shipper2, heartbeat: heartbeat2, mode, requestCount: 0 };
     instances.set(resolvedConfig.apiKey, shared);
   }
   const { shipper, heartbeat } = shared;
@@ -660,28 +510,25 @@ function protect(client, config) {
       return wrapOpenAI(client, resolvedConfig, shipper, heartbeat);
     default:
       throw new Error(
-        `Caplyr: Unrecognized AI client. Supported providers: Anthropic, OpenAI. Make sure you're passing a client instance (e.g., new Anthropic() or new OpenAI()).`
+        "Caplyr: Unrecognized AI client. Supported providers: Anthropic, OpenAI. Make sure you're passing a client instance (e.g., new Anthropic() or new OpenAI())."
       );
   }
 }
 function getStatus(apiKey) {
-  const shared = instances.get(apiKey);
-  return shared?.heartbeat.status ?? "OFF";
+  return instances.get(apiKey)?.heartbeat.status ?? "OFF";
 }
 function getState(apiKey) {
   const shared = instances.get(apiKey);
   if (!shared) return null;
-  const { heartbeat } = shared;
+  const { heartbeat, mode, requestCount } = shared;
   return {
     status: heartbeat.status,
-    mode: "alert_only",
-    // Will be stored properly in next iteration
+    mode,
     budget_daily_used: heartbeat.budgetStatus.daily_used,
     budget_monthly_used: heartbeat.budgetStatus.monthly_used,
     kill_switch_active: heartbeat.budgetStatus.kill_switch_active,
     last_heartbeat: Date.now(),
-    request_count: 0,
-    // Will be tracked in next iteration
+    request_count: requestCount,
     total_cost: heartbeat.budgetStatus.monthly_used,
     total_savings: 0
   };
@@ -695,7 +542,7 @@ async function shutdown(apiKey) {
       instances.delete(apiKey);
     }
   } else {
-    for (const [key, shared] of instances) {
+    for (const [, shared] of instances) {
       shared.heartbeat.destroy();
       shared.shipper.destroy();
     }