aden-ts 0.2.0 → 0.2.1

package/dist/index.js

@@ -1856,6 +1856,101 @@ function isAnthropicInstrumented() {
 // src/control-agent.ts
 var import_crypto6 = require("crypto");
 var import_ws = __toESM(require("ws"));
+
+// src/logging.ts
+var LOG_LEVEL_PRIORITY = {
+  debug: 0,
+  info: 1,
+  warn: 2,
+  error: 3,
+  silent: 4
+};
+var currentConfig = {
+  level: parseLogLevel(process.env.ADEN_LOG_LEVEL) ?? "info",
+  metricsLevel: parseLogLevel(process.env.ADEN_METRICS_LOG_LEVEL) ?? parseLogLevel(process.env.ADEN_LOG_LEVEL) ?? "info",
+  handler: console
+};
+function parseLogLevel(level) {
+  if (!level) return void 0;
+  const normalized = level.toLowerCase();
+  if (normalized in LOG_LEVEL_PRIORITY) {
+    return normalized;
+  }
+  return void 0;
+}
+function shouldLog(level, configLevel) {
+  return LOG_LEVEL_PRIORITY[level] >= LOG_LEVEL_PRIORITY[configLevel];
+}
+function configureLogging(config) {
+  if (config.level !== void 0) {
+    currentConfig.level = config.level;
+    if (config.metricsLevel === void 0) {
+      currentConfig.metricsLevel = config.level;
+    }
+  }
+  if (config.metricsLevel !== void 0) {
+    currentConfig.metricsLevel = config.metricsLevel;
+  }
+  if (config.handler !== void 0) {
+    currentConfig.handler = config.handler;
+  }
+}
+function getLoggingConfig() {
+  return { ...currentConfig };
+}
+function resetLoggingConfig() {
+  currentConfig = {
+    level: parseLogLevel(process.env.ADEN_LOG_LEVEL) ?? "info",
+    metricsLevel: parseLogLevel(process.env.ADEN_METRICS_LOG_LEVEL) ?? parseLogLevel(process.env.ADEN_LOG_LEVEL) ?? "info",
+    handler: console
+  };
+}
+var logger = {
+  debug(message, ...args) {
+    if (shouldLog("debug", currentConfig.level)) {
+      currentConfig.handler.debug(`[aden] ${message}`, ...args);
+    }
+  },
+  info(message, ...args) {
+    if (shouldLog("info", currentConfig.level)) {
+      currentConfig.handler.info(`[aden] ${message}`, ...args);
+    }
+  },
+  warn(message, ...args) {
+    if (shouldLog("warn", currentConfig.level)) {
+      currentConfig.handler.warn(`[aden] ${message}`, ...args);
+    }
+  },
+  error(message, ...args) {
+    if (shouldLog("error", currentConfig.level)) {
+      currentConfig.handler.error(`[aden] ${message}`, ...args);
+    }
+  }
+};
+var metricsLogger = {
+  debug(message, ...args) {
+    if (shouldLog("debug", currentConfig.metricsLevel)) {
+      currentConfig.handler.debug(`[aden.metrics] ${message}`, ...args);
+    }
+  },
+  info(message, ...args) {
+    if (shouldLog("info", currentConfig.metricsLevel)) {
+      currentConfig.handler.info(`[aden.metrics] ${message}`, ...args);
+    }
+  },
+  warn(message, ...args) {
+    if (shouldLog("warn", currentConfig.metricsLevel)) {
+      currentConfig.handler.warn(`[aden.metrics] ${message}`, ...args);
+    }
+  },
+  error(message, ...args) {
+    if (shouldLog("error", currentConfig.metricsLevel)) {
+      currentConfig.handler.error(`[aden.metrics] ${message}`, ...args);
+    }
+  }
+};
+
+// src/control-agent.ts
 var SDK_VERSION = "0.1.0";
 var ControlAgent = class {
   constructor(options) {
@@ -1903,9 +1998,11 @@ var ControlAgent = class {
    */
   async connect() {
     const url = this.options.serverUrl;
+    logger.debug(`Connecting to control server: ${url}`);
     if (url.startsWith("wss://") || url.startsWith("ws://")) {
       await this.connectWebSocket();
     } else {
+      logger.debug("Using HTTP polling mode (no WebSocket URL)");
       this.startPolling();
     }
     this.startHeartbeat();
@@ -1921,6 +2018,7 @@ var ControlAgent = class {
       };
       try {
         const wsUrl = `${this.options.serverUrl}/v1/control/ws`;
+        logger.debug(`Attempting WebSocket connection to: ${wsUrl}`);
         this.ws = new import_ws.default(wsUrl, {
           headers: {
             Authorization: `Bearer ${this.options.apiKey}`,
@@ -1930,7 +2028,7 @@ var ControlAgent = class {
         this.ws.on("open", () => {
           this.connected = true;
           this.reconnectAttempts = 0;
-          console.log("[aden] WebSocket connected to control server");
+          logger.info("WebSocket connected to control server");
           this.flushEventQueue();
           resolve();
         });
@@ -1939,12 +2037,12 @@ var ControlAgent = class {
         });
         this.ws.on("close", () => {
           this.connected = false;
-          console.log("[aden] WebSocket disconnected, falling back to polling");
+          logger.info("WebSocket disconnected, falling back to polling");
           this.scheduleReconnect();
           this.startPolling();
         });
         this.ws.on("error", (error) => {
-          console.warn("[aden] WebSocket error:", error.message);
+          logger.warn("WebSocket error:", error.message);
           this.errorsSinceLastHeartbeat++;
           if (!this.connected) {
             fallbackToPolling();
@@ -1952,12 +2050,12 @@ var ControlAgent = class {
         });
         setTimeout(() => {
           if (!this.connected) {
-            console.warn("[aden] WebSocket connection timeout, using polling");
+            logger.warn("WebSocket connection timeout, using polling");
             fallbackToPolling();
           }
         }, this.options.timeoutMs);
       } catch (error) {
-        console.warn("[aden] WebSocket setup failed:", error);
+        logger.warn("WebSocket setup failed:", error);
         fallbackToPolling();
       }
     });
@@ -1968,7 +2066,7 @@ var ControlAgent = class {
   scheduleReconnect() {
     if (this.reconnectTimer) return;
     if (this.reconnectAttempts >= this.maxReconnectAttempts) {
-      console.warn("[aden] Max reconnect attempts reached, using polling only");
+      logger.warn("Max reconnect attempts reached, using polling only");
       return;
     }
     const delay = Math.min(1e3 * Math.pow(2, this.reconnectAttempts), 3e4);
@@ -1989,12 +2087,12 @@ var ControlAgent = class {
       if (message.type === "policy") {
         this.cachedPolicy = message.policy;
         this.lastPolicyFetch = Date.now();
-        console.log("[aden] Policy updated:", this.cachedPolicy.version);
+        logger.info("Policy updated:", this.cachedPolicy.version);
       } else if (message.type === "command") {
-        console.log("[aden] Command received:", message);
+        logger.info("Command received:", message);
       }
     } catch (error) {
-      console.warn("[aden] Failed to parse message:", error);
+      logger.warn("Failed to parse message:", error);
     }
   }
   /**
@@ -2003,6 +2101,7 @@ var ControlAgent = class {
    */
   async startPolling() {
     if (this.pollingTimer) return;
+    logger.debug(`Starting HTTP polling (interval: ${this.options.pollingIntervalMs}ms)`);
     await this.fetchPolicy();
     this.pollingTimer = setInterval(() => {
       if (!this.connected) {
@@ -2023,15 +2122,19 @@ var ControlAgent = class {
    * Fetch policy via HTTP
    */
   async fetchPolicy() {
+    logger.debug("Fetching policy from server...");
     try {
       const response = await this.httpRequest("/v1/control/policy", "GET");
       if (response.ok) {
         const policy = await response.json();
         this.cachedPolicy = policy;
         this.lastPolicyFetch = Date.now();
+        logger.debug(`Policy fetched successfully (version: ${policy.version}, budgets: ${policy.budgets?.length ?? 0})`);
+      } else {
+        logger.debug(`Policy fetch returned status ${response.status}`);
       }
     } catch (error) {
-      console.warn("[aden] Failed to fetch policy:", error);
+      logger.warn("Failed to fetch policy:", error);
     }
   }
   /**
@@ -2039,6 +2142,7 @@ var ControlAgent = class {
    */
   startHeartbeat() {
     if (this.heartbeatTimer) return;
+    logger.debug(`Starting heartbeat (interval: ${this.options.heartbeatIntervalMs}ms)`);
     this.heartbeatTimer = setInterval(() => {
       this.sendHeartbeat();
     }, this.options.heartbeatIntervalMs);
@@ -2075,6 +2179,7 @@ var ControlAgent = class {
    * Disconnect from the control server
    */
   async disconnect() {
+    logger.debug("Disconnecting from control server...");
     this.stopPolling();
     this.stopHeartbeat();
     if (this.reconnectTimer) {
@@ -2086,6 +2191,7 @@ var ControlAgent = class {
       this.ws = null;
     }
     this.connected = false;
+    logger.debug("Disconnected from control server");
   }
   /**
    * Get a control decision for a request
@@ -2215,7 +2321,7 @@ var ControlAgent = class {
             timestamp: /* @__PURE__ */ new Date()
           };
           Promise.resolve(this.options.onAlert(alertEvent)).catch((err) => {
-            console.warn("[aden] Alert callback error:", err);
+            logger.warn("Alert callback error:", err);
           });
           return {
             action: "alert",
@@ -2522,8 +2628,8 @@ var ControlAgent = class {
     }
     if (this.options.adaptiveThresholdEnabled) {
       if (remainingBudgetUsd <= this.options.adaptiveMinRemainingUsd) {
-        console.debug(
-          `[aden] Remaining budget $${remainingBudgetUsd.toFixed(4)} <= $${this.options.adaptiveMinRemainingUsd.toFixed(2)}, forcing validation`
+        logger.debug(
+          `Remaining budget $${remainingBudgetUsd.toFixed(4)} <= $${this.options.adaptiveMinRemainingUsd.toFixed(2)}, forcing validation`
         );
         return true;
       }
@@ -2532,13 +2638,13 @@ var ControlAgent = class {
       const samplingRate = this.calculateSamplingRate(budgetUsagePercent);
       const shouldSample = Math.random() < samplingRate;
       if (!shouldSample) {
-        console.debug(
-          `[aden] Skipping validation (sampling rate: ${(samplingRate * 100).toFixed(1)}%, usage: ${budgetUsagePercent.toFixed(1)}%)`
+        logger.debug(
+          `Skipping validation (sampling rate: ${(samplingRate * 100).toFixed(1)}%, usage: ${budgetUsagePercent.toFixed(1)}%)`
         );
         return false;
       }
-      console.debug(
-        `[aden] Sampled for validation (rate: ${(samplingRate * 100).toFixed(1)}%, usage: ${budgetUsagePercent.toFixed(1)}%)`
+      logger.debug(
+        `Sampled for validation (rate: ${(samplingRate * 100).toFixed(1)}%, usage: ${budgetUsagePercent.toFixed(1)}%)`
       );
       return true;
     }
@@ -2596,8 +2702,8 @@ var ControlAgent = class {
         });
         if (response.ok) {
           const data = await response.json();
-          console.debug(
-            `[aden] Server validation response: allowed=${data.allowed}, action=${data.action}, reason=${data.reason}`
+          logger.debug(
+            `Server validation response: allowed=${data.allowed}, action=${data.action}, reason=${data.reason}`
           );
           return {
             allowed: data.allowed ?? true,
@@ -2612,14 +2718,14 @@ var ControlAgent = class {
             degradeToModel: data.degrade_to_model
           };
         } else {
-          console.warn(`[aden] Server validation returned status ${response.status}`);
+          logger.warn(`Server validation returned status ${response.status}`);
           return null;
         }
       } finally {
         clearTimeout(timeoutId);
       }
     } catch (error) {
-      console.warn(`[aden] Server validation failed: ${error}`);
+      logger.warn(`Server validation failed: ${error}`);
       return null;
     }
   }
@@ -2659,8 +2765,8 @@ var ControlAgent = class {
       }
     }
     if (this.shouldValidateWithServer(usagePercent, remaining, limit)) {
-      console.debug(
-        `[aden] Budget '${budget.name}' at ${usagePercent.toFixed(1)}% ($${currentSpend.toFixed(6)}/$${limit.toFixed(6)}), validating with server`
+      logger.debug(
+        `Budget '${budget.name}' at ${usagePercent.toFixed(1)}% ($${currentSpend.toFixed(6)}/$${limit.toFixed(6)}), validating with server`
       );
       const validation = await this.validateBudgetWithServer(
         budget.id,
@@ -2670,8 +2776,8 @@ var ControlAgent = class {
       if (validation) {
         return this.applyServerValidationResult(validation, budget.id);
       } else {
-        console.warn(
-          `[aden] Server validation failed for budget '${budget.id}', using local enforcement`
+        logger.warn(
+          `Server validation failed for budget '${budget.id}', using local enforcement`
         );
         if (!this.options.failOpen) {
           return {
@@ -2733,12 +2839,14 @@ var ControlAgent = class {
     if (this.connected && this.ws?.readyState === import_ws.default.OPEN) {
       try {
         this.ws.send(JSON.stringify(event));
+        logger.debug(`Event sent via WebSocket: ${event.event_type}`);
         return;
       } catch (error) {
-        console.warn("[aden] WebSocket send failed, queuing event");
+        logger.warn("WebSocket send failed, queuing event");
       }
     }
     this.queueEvent(event);
+    logger.debug(`Event queued: ${event.event_type} (queue size: ${this.eventQueue.length})`);
     if (!this.connected) {
       await this.flushEventQueue();
     }
@@ -2757,7 +2865,9 @@ var ControlAgent = class {
    */
   async flushEventQueue() {
     if (this.eventQueue.length === 0) return;
+    const eventCount = this.eventQueue.length;
     if (this.connected && this.ws?.readyState === import_ws.default.OPEN) {
+      logger.debug(`Flushing ${eventCount} events via WebSocket`);
       const events = [...this.eventQueue];
       this.eventQueue = [];
       for (const event of events) {
@@ -2769,12 +2879,14 @@ var ControlAgent = class {
       }
       return;
     }
+    logger.debug(`Flushing ${eventCount} events via HTTP`);
     try {
       const events = [...this.eventQueue];
       this.eventQueue = [];
       await this.httpRequest("/v1/control/events", "POST", { events });
+      logger.debug(`Successfully sent ${eventCount} events via HTTP`);
     } catch (error) {
-      console.warn("[aden] Failed to flush event queue:", error);
+      logger.warn("Failed to flush event queue:", error);
     }
   }
   /**
@@ -6646,99 +6758,6 @@ function createReportEmitter(builder) {
     builder.recordEvent(event);
   };
 }
-
-// src/logging.ts
-var LOG_LEVEL_PRIORITY = {
-  debug: 0,
-  info: 1,
-  warn: 2,
-  error: 3,
-  silent: 4
-};
-var currentConfig = {
-  level: parseLogLevel(process.env.ADEN_LOG_LEVEL) ?? "info",
-  metricsLevel: parseLogLevel(process.env.ADEN_METRICS_LOG_LEVEL) ?? parseLogLevel(process.env.ADEN_LOG_LEVEL) ?? "info",
-  handler: console
-};
-function parseLogLevel(level) {
-  if (!level) return void 0;
-  const normalized = level.toLowerCase();
-  if (normalized in LOG_LEVEL_PRIORITY) {
-    return normalized;
-  }
-  return void 0;
-}
-function shouldLog(level, configLevel) {
-  return LOG_LEVEL_PRIORITY[level] >= LOG_LEVEL_PRIORITY[configLevel];
-}
-function configureLogging(config) {
-  if (config.level !== void 0) {
-    currentConfig.level = config.level;
-    if (config.metricsLevel === void 0) {
-      currentConfig.metricsLevel = config.level;
-    }
-  }
-  if (config.metricsLevel !== void 0) {
-    currentConfig.metricsLevel = config.metricsLevel;
-  }
-  if (config.handler !== void 0) {
-    currentConfig.handler = config.handler;
-  }
-}
-function getLoggingConfig() {
-  return { ...currentConfig };
-}
-function resetLoggingConfig() {
-  currentConfig = {
-    level: parseLogLevel(process.env.ADEN_LOG_LEVEL) ?? "info",
-    metricsLevel: parseLogLevel(process.env.ADEN_METRICS_LOG_LEVEL) ?? parseLogLevel(process.env.ADEN_LOG_LEVEL) ?? "info",
-    handler: console
-  };
-}
-var logger = {
-  debug(message, ...args) {
-    if (shouldLog("debug", currentConfig.level)) {
-      currentConfig.handler.debug(`[aden] ${message}`, ...args);
-    }
-  },
-  info(message, ...args) {
-    if (shouldLog("info", currentConfig.level)) {
-      currentConfig.handler.info(`[aden] ${message}`, ...args);
-    }
-  },
-  warn(message, ...args) {
-    if (shouldLog("warn", currentConfig.level)) {
-      currentConfig.handler.warn(`[aden] ${message}`, ...args);
-    }
-  },
-  error(message, ...args) {
-    if (shouldLog("error", currentConfig.level)) {
-      currentConfig.handler.error(`[aden] ${message}`, ...args);
-    }
-  }
-};
-var metricsLogger = {
-  debug(message, ...args) {
-    if (shouldLog("debug", currentConfig.metricsLevel)) {
-      currentConfig.handler.debug(`[aden.metrics] ${message}`, ...args);
-    }
-  },
-  info(message, ...args) {
-    if (shouldLog("info", currentConfig.metricsLevel)) {
-      currentConfig.handler.info(`[aden.metrics] ${message}`, ...args);
-    }
-  },
-  warn(message, ...args) {
-    if (shouldLog("warn", currentConfig.metricsLevel)) {
-      currentConfig.handler.warn(`[aden.metrics] ${message}`, ...args);
-    }
-  },
-  error(message, ...args) {
-    if (shouldLog("error", currentConfig.metricsLevel)) {
-      currentConfig.handler.error(`[aden.metrics] ${message}`, ...args);
-    }
-  }
-};
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   AnalyticsEngine,

package/dist/index.mjs

@@ -1739,6 +1739,101 @@ function isAnthropicInstrumented() {
 // src/control-agent.ts
 import { randomUUID as randomUUID6 } from "crypto";
 import WebSocket from "ws";
+
+// src/logging.ts
+var LOG_LEVEL_PRIORITY = {
+  debug: 0,
+  info: 1,
+  warn: 2,
+  error: 3,
+  silent: 4
+};
+var currentConfig = {
+  level: parseLogLevel(process.env.ADEN_LOG_LEVEL) ?? "info",
+  metricsLevel: parseLogLevel(process.env.ADEN_METRICS_LOG_LEVEL) ?? parseLogLevel(process.env.ADEN_LOG_LEVEL) ?? "info",
+  handler: console
+};
+function parseLogLevel(level) {
+  if (!level) return void 0;
+  const normalized = level.toLowerCase();
+  if (normalized in LOG_LEVEL_PRIORITY) {
+    return normalized;
+  }
+  return void 0;
+}
+function shouldLog(level, configLevel) {
+  return LOG_LEVEL_PRIORITY[level] >= LOG_LEVEL_PRIORITY[configLevel];
+}
+function configureLogging(config) {
+  if (config.level !== void 0) {
+    currentConfig.level = config.level;
+    if (config.metricsLevel === void 0) {
+      currentConfig.metricsLevel = config.level;
+    }
+  }
+  if (config.metricsLevel !== void 0) {
+    currentConfig.metricsLevel = config.metricsLevel;
+  }
+  if (config.handler !== void 0) {
+    currentConfig.handler = config.handler;
+  }
+}
+function getLoggingConfig() {
+  return { ...currentConfig };
+}
+function resetLoggingConfig() {
+  currentConfig = {
+    level: parseLogLevel(process.env.ADEN_LOG_LEVEL) ?? "info",
+    metricsLevel: parseLogLevel(process.env.ADEN_METRICS_LOG_LEVEL) ?? parseLogLevel(process.env.ADEN_LOG_LEVEL) ?? "info",
+    handler: console
+  };
+}
+var logger = {
+  debug(message, ...args) {
+    if (shouldLog("debug", currentConfig.level)) {
+      currentConfig.handler.debug(`[aden] ${message}`, ...args);
+    }
+  },
+  info(message, ...args) {
+    if (shouldLog("info", currentConfig.level)) {
+      currentConfig.handler.info(`[aden] ${message}`, ...args);
+    }
+  },
+  warn(message, ...args) {
+    if (shouldLog("warn", currentConfig.level)) {
+      currentConfig.handler.warn(`[aden] ${message}`, ...args);
+    }
+  },
+  error(message, ...args) {
+    if (shouldLog("error", currentConfig.level)) {
+      currentConfig.handler.error(`[aden] ${message}`, ...args);
+    }
+  }
+};
+var metricsLogger = {
+  debug(message, ...args) {
+    if (shouldLog("debug", currentConfig.metricsLevel)) {
+      currentConfig.handler.debug(`[aden.metrics] ${message}`, ...args);
+    }
+  },
+  info(message, ...args) {
+    if (shouldLog("info", currentConfig.metricsLevel)) {
+      currentConfig.handler.info(`[aden.metrics] ${message}`, ...args);
+    }
+  },
+  warn(message, ...args) {
+    if (shouldLog("warn", currentConfig.metricsLevel)) {
+      currentConfig.handler.warn(`[aden.metrics] ${message}`, ...args);
+    }
+  },
+  error(message, ...args) {
+    if (shouldLog("error", currentConfig.metricsLevel)) {
+      currentConfig.handler.error(`[aden.metrics] ${message}`, ...args);
+    }
+  }
+};
+
+// src/control-agent.ts
 var SDK_VERSION = "0.1.0";
 var ControlAgent = class {
   constructor(options) {
@@ -1786,9 +1881,11 @@ var ControlAgent = class {
    */
   async connect() {
     const url = this.options.serverUrl;
+    logger.debug(`Connecting to control server: ${url}`);
     if (url.startsWith("wss://") || url.startsWith("ws://")) {
       await this.connectWebSocket();
     } else {
+      logger.debug("Using HTTP polling mode (no WebSocket URL)");
       this.startPolling();
     }
     this.startHeartbeat();
@@ -1804,6 +1901,7 @@ var ControlAgent = class {
       };
       try {
         const wsUrl = `${this.options.serverUrl}/v1/control/ws`;
+        logger.debug(`Attempting WebSocket connection to: ${wsUrl}`);
         this.ws = new WebSocket(wsUrl, {
           headers: {
             Authorization: `Bearer ${this.options.apiKey}`,
@@ -1813,7 +1911,7 @@ var ControlAgent = class {
         this.ws.on("open", () => {
           this.connected = true;
           this.reconnectAttempts = 0;
-          console.log("[aden] WebSocket connected to control server");
+          logger.info("WebSocket connected to control server");
           this.flushEventQueue();
           resolve();
         });
@@ -1822,12 +1920,12 @@ var ControlAgent = class {
         });
         this.ws.on("close", () => {
           this.connected = false;
-          console.log("[aden] WebSocket disconnected, falling back to polling");
+          logger.info("WebSocket disconnected, falling back to polling");
           this.scheduleReconnect();
           this.startPolling();
         });
         this.ws.on("error", (error) => {
-          console.warn("[aden] WebSocket error:", error.message);
+          logger.warn("WebSocket error:", error.message);
           this.errorsSinceLastHeartbeat++;
           if (!this.connected) {
             fallbackToPolling();
@@ -1835,12 +1933,12 @@ var ControlAgent = class {
         });
         setTimeout(() => {
           if (!this.connected) {
-            console.warn("[aden] WebSocket connection timeout, using polling");
+            logger.warn("WebSocket connection timeout, using polling");
             fallbackToPolling();
           }
         }, this.options.timeoutMs);
       } catch (error) {
-        console.warn("[aden] WebSocket setup failed:", error);
+        logger.warn("WebSocket setup failed:", error);
         fallbackToPolling();
       }
     });
@@ -1851,7 +1949,7 @@ var ControlAgent = class {
   scheduleReconnect() {
     if (this.reconnectTimer) return;
     if (this.reconnectAttempts >= this.maxReconnectAttempts) {
-      console.warn("[aden] Max reconnect attempts reached, using polling only");
+      logger.warn("Max reconnect attempts reached, using polling only");
       return;
     }
     const delay = Math.min(1e3 * Math.pow(2, this.reconnectAttempts), 3e4);
@@ -1872,12 +1970,12 @@ var ControlAgent = class {
       if (message.type === "policy") {
         this.cachedPolicy = message.policy;
         this.lastPolicyFetch = Date.now();
-        console.log("[aden] Policy updated:", this.cachedPolicy.version);
+        logger.info("Policy updated:", this.cachedPolicy.version);
       } else if (message.type === "command") {
-        console.log("[aden] Command received:", message);
+        logger.info("Command received:", message);
       }
     } catch (error) {
-      console.warn("[aden] Failed to parse message:", error);
+      logger.warn("Failed to parse message:", error);
     }
   }
   /**
@@ -1886,6 +1984,7 @@ var ControlAgent = class {
    */
   async startPolling() {
     if (this.pollingTimer) return;
+    logger.debug(`Starting HTTP polling (interval: ${this.options.pollingIntervalMs}ms)`);
     await this.fetchPolicy();
     this.pollingTimer = setInterval(() => {
       if (!this.connected) {
@@ -1906,15 +2005,19 @@ var ControlAgent = class {
    * Fetch policy via HTTP
    */
   async fetchPolicy() {
+    logger.debug("Fetching policy from server...");
     try {
       const response = await this.httpRequest("/v1/control/policy", "GET");
       if (response.ok) {
         const policy = await response.json();
         this.cachedPolicy = policy;
         this.lastPolicyFetch = Date.now();
+        logger.debug(`Policy fetched successfully (version: ${policy.version}, budgets: ${policy.budgets?.length ?? 0})`);
+      } else {
+        logger.debug(`Policy fetch returned status ${response.status}`);
       }
     } catch (error) {
-      console.warn("[aden] Failed to fetch policy:", error);
+      logger.warn("Failed to fetch policy:", error);
     }
   }
   /**
@@ -1922,6 +2025,7 @@ var ControlAgent = class {
    */
   startHeartbeat() {
     if (this.heartbeatTimer) return;
+    logger.debug(`Starting heartbeat (interval: ${this.options.heartbeatIntervalMs}ms)`);
     this.heartbeatTimer = setInterval(() => {
       this.sendHeartbeat();
     }, this.options.heartbeatIntervalMs);
@@ -1958,6 +2062,7 @@ var ControlAgent = class {
    * Disconnect from the control server
    */
   async disconnect() {
+    logger.debug("Disconnecting from control server...");
     this.stopPolling();
     this.stopHeartbeat();
     if (this.reconnectTimer) {
@@ -1969,6 +2074,7 @@ var ControlAgent = class {
       this.ws = null;
     }
     this.connected = false;
+    logger.debug("Disconnected from control server");
   }
   /**
    * Get a control decision for a request
@@ -2098,7 +2204,7 @@ var ControlAgent = class {
             timestamp: /* @__PURE__ */ new Date()
           };
           Promise.resolve(this.options.onAlert(alertEvent)).catch((err) => {
-            console.warn("[aden] Alert callback error:", err);
+            logger.warn("Alert callback error:", err);
           });
           return {
             action: "alert",
@@ -2405,8 +2511,8 @@ var ControlAgent = class {
     }
     if (this.options.adaptiveThresholdEnabled) {
       if (remainingBudgetUsd <= this.options.adaptiveMinRemainingUsd) {
-        console.debug(
-          `[aden] Remaining budget $${remainingBudgetUsd.toFixed(4)} <= $${this.options.adaptiveMinRemainingUsd.toFixed(2)}, forcing validation`
+        logger.debug(
+          `Remaining budget $${remainingBudgetUsd.toFixed(4)} <= $${this.options.adaptiveMinRemainingUsd.toFixed(2)}, forcing validation`
         );
         return true;
       }
@@ -2415,13 +2521,13 @@ var ControlAgent = class {
       const samplingRate = this.calculateSamplingRate(budgetUsagePercent);
       const shouldSample = Math.random() < samplingRate;
       if (!shouldSample) {
-        console.debug(
-          `[aden] Skipping validation (sampling rate: ${(samplingRate * 100).toFixed(1)}%, usage: ${budgetUsagePercent.toFixed(1)}%)`
+        logger.debug(
+          `Skipping validation (sampling rate: ${(samplingRate * 100).toFixed(1)}%, usage: ${budgetUsagePercent.toFixed(1)}%)`
         );
         return false;
       }
-      console.debug(
-        `[aden] Sampled for validation (rate: ${(samplingRate * 100).toFixed(1)}%, usage: ${budgetUsagePercent.toFixed(1)}%)`
+      logger.debug(
+        `Sampled for validation (rate: ${(samplingRate * 100).toFixed(1)}%, usage: ${budgetUsagePercent.toFixed(1)}%)`
       );
       return true;
     }
@@ -2479,8 +2585,8 @@ var ControlAgent = class {
         });
         if (response.ok) {
           const data = await response.json();
-          console.debug(
-            `[aden] Server validation response: allowed=${data.allowed}, action=${data.action}, reason=${data.reason}`
+          logger.debug(
+            `Server validation response: allowed=${data.allowed}, action=${data.action}, reason=${data.reason}`
           );
           return {
             allowed: data.allowed ?? true,
@@ -2495,14 +2601,14 @@ var ControlAgent = class {
             degradeToModel: data.degrade_to_model
           };
         } else {
-          console.warn(`[aden] Server validation returned status ${response.status}`);
+          logger.warn(`Server validation returned status ${response.status}`);
           return null;
         }
       } finally {
         clearTimeout(timeoutId);
       }
     } catch (error) {
-      console.warn(`[aden] Server validation failed: ${error}`);
+      logger.warn(`Server validation failed: ${error}`);
       return null;
     }
   }
@@ -2542,8 +2648,8 @@ var ControlAgent = class {
       }
     }
     if (this.shouldValidateWithServer(usagePercent, remaining, limit)) {
-      console.debug(
-        `[aden] Budget '${budget.name}' at ${usagePercent.toFixed(1)}% ($${currentSpend.toFixed(6)}/$${limit.toFixed(6)}), validating with server`
+      logger.debug(
+        `Budget '${budget.name}' at ${usagePercent.toFixed(1)}% ($${currentSpend.toFixed(6)}/$${limit.toFixed(6)}), validating with server`
       );
       const validation = await this.validateBudgetWithServer(
         budget.id,
@@ -2553,8 +2659,8 @@ var ControlAgent = class {
       if (validation) {
         return this.applyServerValidationResult(validation, budget.id);
       } else {
-        console.warn(
-          `[aden] Server validation failed for budget '${budget.id}', using local enforcement`
+        logger.warn(
+          `Server validation failed for budget '${budget.id}', using local enforcement`
         );
         if (!this.options.failOpen) {
           return {
@@ -2616,12 +2722,14 @@ var ControlAgent = class {
     if (this.connected && this.ws?.readyState === WebSocket.OPEN) {
       try {
         this.ws.send(JSON.stringify(event));
+        logger.debug(`Event sent via WebSocket: ${event.event_type}`);
         return;
       } catch (error) {
-        console.warn("[aden] WebSocket send failed, queuing event");
+        logger.warn("WebSocket send failed, queuing event");
       }
     }
     this.queueEvent(event);
+    logger.debug(`Event queued: ${event.event_type} (queue size: ${this.eventQueue.length})`);
     if (!this.connected) {
       await this.flushEventQueue();
     }
@@ -2640,7 +2748,9 @@ var ControlAgent = class {
    */
   async flushEventQueue() {
     if (this.eventQueue.length === 0) return;
+    const eventCount = this.eventQueue.length;
     if (this.connected && this.ws?.readyState === WebSocket.OPEN) {
+      logger.debug(`Flushing ${eventCount} events via WebSocket`);
       const events = [...this.eventQueue];
       this.eventQueue = [];
       for (const event of events) {
@@ -2652,12 +2762,14 @@ var ControlAgent = class {
       }
       return;
     }
+    logger.debug(`Flushing ${eventCount} events via HTTP`);
     try {
       const events = [...this.eventQueue];
       this.eventQueue = [];
       await this.httpRequest("/v1/control/events", "POST", { events });
+      logger.debug(`Successfully sent ${eventCount} events via HTTP`);
     } catch (error) {
-      console.warn("[aden] Failed to flush event queue:", error);
+      logger.warn("Failed to flush event queue:", error);
     }
   }
   /**
@@ -6529,99 +6641,6 @@ function createReportEmitter(builder) {
     builder.recordEvent(event);
   };
 }
-
-// src/logging.ts
-var LOG_LEVEL_PRIORITY = {
-  debug: 0,
-  info: 1,
-  warn: 2,
-  error: 3,
-  silent: 4
-};
-var currentConfig = {
-  level: parseLogLevel(process.env.ADEN_LOG_LEVEL) ?? "info",
-  metricsLevel: parseLogLevel(process.env.ADEN_METRICS_LOG_LEVEL) ?? parseLogLevel(process.env.ADEN_LOG_LEVEL) ?? "info",
-  handler: console
-};
-function parseLogLevel(level) {
-  if (!level) return void 0;
-  const normalized = level.toLowerCase();
-  if (normalized in LOG_LEVEL_PRIORITY) {
-    return normalized;
-  }
-  return void 0;
-}
-function shouldLog(level, configLevel) {
-  return LOG_LEVEL_PRIORITY[level] >= LOG_LEVEL_PRIORITY[configLevel];
-}
-function configureLogging(config) {
-  if (config.level !== void 0) {
-    currentConfig.level = config.level;
-    if (config.metricsLevel === void 0) {
-      currentConfig.metricsLevel = config.level;
-    }
-  }
-  if (config.metricsLevel !== void 0) {
-    currentConfig.metricsLevel = config.metricsLevel;
-  }
-  if (config.handler !== void 0) {
-    currentConfig.handler = config.handler;
-  }
-}
-function getLoggingConfig() {
-  return { ...currentConfig };
-}
-function resetLoggingConfig() {
-  currentConfig = {
-    level: parseLogLevel(process.env.ADEN_LOG_LEVEL) ?? "info",
-    metricsLevel: parseLogLevel(process.env.ADEN_METRICS_LOG_LEVEL) ?? parseLogLevel(process.env.ADEN_LOG_LEVEL) ?? "info",
-    handler: console
-  };
-}
-var logger = {
-  debug(message, ...args) {
-    if (shouldLog("debug", currentConfig.level)) {
-      currentConfig.handler.debug(`[aden] ${message}`, ...args);
-    }
-  },
-  info(message, ...args) {
-    if (shouldLog("info", currentConfig.level)) {
-      currentConfig.handler.info(`[aden] ${message}`, ...args);
-    }
-  },
-  warn(message, ...args) {
-    if (shouldLog("warn", currentConfig.level)) {
-      currentConfig.handler.warn(`[aden] ${message}`, ...args);
-    }
-  },
-  error(message, ...args) {
-    if (shouldLog("error", currentConfig.level)) {
-      currentConfig.handler.error(`[aden] ${message}`, ...args);
-    }
-  }
-};
-var metricsLogger = {
-  debug(message, ...args) {
-    if (shouldLog("debug", currentConfig.metricsLevel)) {
-      currentConfig.handler.debug(`[aden.metrics] ${message}`, ...args);
-    }
-  },
-  info(message, ...args) {
-    if (shouldLog("info", currentConfig.metricsLevel)) {
-      currentConfig.handler.info(`[aden.metrics] ${message}`, ...args);
-    }
-  },
-  warn(message, ...args) {
-    if (shouldLog("warn", currentConfig.metricsLevel)) {
-      currentConfig.handler.warn(`[aden.metrics] ${message}`, ...args);
-    }
-  },
-  error(message, ...args) {
-    if (shouldLog("error", currentConfig.metricsLevel)) {
-      currentConfig.handler.error(`[aden.metrics] ${message}`, ...args);
-    }
-  }
-};
 export {
   AnalyticsEngine,
   BudgetExceededError,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "aden-ts",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "description": "LLM Observability & Cost Control SDK - Real-time usage tracking, budget enforcement, and cost control for OpenAI, Anthropic, and Gemini",
   "main": "dist/index.js",
   "module": "dist/index.mjs",