aden-ts 0.1.1 → 0.2.0

package/dist/index.d.mts

@@ -140,6 +140,8 @@ interface BudgetRule {
     alerts: BudgetAlert[];
     /** Notification settings */
     notifications: BudgetNotifications;
+    /** Legacy context_id for backwards compatibility */
+    context_id?: string;
 }
 /**
  * Throttle rule - rate limiting
@@ -285,13 +287,13 @@ interface ControlAgentOptions {
     /**
      * Enable hybrid enforcement (local + server-side validation).
      * When enabled, budgets above the threshold are validated with the server.
-     * Default: false
+     * Default: true
      */
     enableHybridEnforcement?: boolean;
     /**
      * Budget usage threshold (percentage) at which to start server validation.
      * Requests below this threshold use local-only enforcement.
-     * Default: 80
+     * Default: 5
      */
     serverValidationThreshold?: number;
     /**
@@ -308,7 +310,7 @@ interface ControlAgentOptions {
     /**
      * Minimum remaining budget (USD) that triggers forced server validation.
      * Only applies when adaptiveThresholdEnabled is true.
-     * Default: 1.0
+     * Default: 5.0
      */
     adaptiveMinRemainingUsd?: number;
     /**
@@ -2354,7 +2356,26 @@ declare class ControlAgent implements IControlAgent {
      */
     private matchesBlockRule;
     /**
-     * Find budgets that apply to the given request based on budget type
+     * Get action priority for finding most restrictive decision.
+     * Higher priority = more restrictive.
+     */
+    private getActionPriority;
+    /**
+     * Evaluate a single budget using local-only enforcement.
+     * Returns a decision if the budget triggers an action, null otherwise.
+     */
+    private evaluateBudgetLocally;
+    /**
+     * Find budgets that apply to the given request based on budget type.
+     *
+     * Matching logic by budget type:
+     * - global: Matches ALL requests
+     * - agent: Matches if request.metadata.agent == budget.name or budget.id
+     * - tenant: Matches if request.metadata.tenant_id == budget.name or budget.id
+     * - customer: Matches if request.metadata.customer_id == budget.name or budget.id
+     * - feature: Matches if request.metadata.feature == budget.name or budget.id
+     * - tag: Matches if any request.metadata.tags intersect with budget.tags
+     * - legacy (context_id): Matches if request.context_id == budget.context_id
      */
     private findApplicableBudgets;
     /**

package/dist/index.d.ts

@@ -140,6 +140,8 @@ interface BudgetRule {
     alerts: BudgetAlert[];
     /** Notification settings */
     notifications: BudgetNotifications;
+    /** Legacy context_id for backwards compatibility */
+    context_id?: string;
 }
 /**
  * Throttle rule - rate limiting
@@ -285,13 +287,13 @@ interface ControlAgentOptions {
     /**
      * Enable hybrid enforcement (local + server-side validation).
      * When enabled, budgets above the threshold are validated with the server.
-     * Default: false
+     * Default: true
      */
     enableHybridEnforcement?: boolean;
     /**
      * Budget usage threshold (percentage) at which to start server validation.
      * Requests below this threshold use local-only enforcement.
-     * Default: 80
+     * Default: 5
      */
     serverValidationThreshold?: number;
     /**
@@ -308,7 +310,7 @@ interface ControlAgentOptions {
     /**
      * Minimum remaining budget (USD) that triggers forced server validation.
      * Only applies when adaptiveThresholdEnabled is true.
-     * Default: 1.0
+     * Default: 5.0
      */
     adaptiveMinRemainingUsd?: number;
     /**
@@ -2354,7 +2356,26 @@ declare class ControlAgent implements IControlAgent {
      */
     private matchesBlockRule;
     /**
-     * Find budgets that apply to the given request based on budget type
+     * Get action priority for finding most restrictive decision.
+     * Higher priority = more restrictive.
+     */
+    private getActionPriority;
+    /**
+     * Evaluate a single budget using local-only enforcement.
+     * Returns a decision if the budget triggers an action, null otherwise.
+     */
+    private evaluateBudgetLocally;
+    /**
+     * Find budgets that apply to the given request based on budget type.
+     *
+     * Matching logic by budget type:
+     * - global: Matches ALL requests
+     * - agent: Matches if request.metadata.agent == budget.name or budget.id
+     * - tenant: Matches if request.metadata.tenant_id == budget.name or budget.id
+     * - customer: Matches if request.metadata.customer_id == budget.name or budget.id
+     * - feature: Matches if request.metadata.feature == budget.name or budget.id
+     * - tag: Matches if any request.metadata.tags intersect with budget.tags
+     * - legacy (context_id): Matches if request.context_id == budget.context_id
      */
     private findApplicableBudgets;
     /**

package/dist/index.js

@@ -1886,12 +1886,12 @@ var ControlAgent = class {
       instanceId: options.instanceId ?? (0, import_crypto6.randomUUID)(),
       onAlert: options.onAlert ?? (() => {
       }),
-      // Hybrid enforcement options
-      enableHybridEnforcement: options.enableHybridEnforcement ?? false,
-      serverValidationThreshold: options.serverValidationThreshold ?? 80,
+      // Hybrid enforcement options (defaults match Python SDK)
+      enableHybridEnforcement: options.enableHybridEnforcement ?? true,
+      serverValidationThreshold: options.serverValidationThreshold ?? 5,
       serverValidationTimeoutMs: options.serverValidationTimeoutMs ?? 2e3,
       adaptiveThresholdEnabled: options.adaptiveThresholdEnabled ?? true,
-      adaptiveMinRemainingUsd: options.adaptiveMinRemainingUsd ?? 1,
+      adaptiveMinRemainingUsd: options.adaptiveMinRemainingUsd ?? 5,
       samplingEnabled: options.samplingEnabled ?? true,
       samplingBaseRate: options.samplingBaseRate ?? 0.1,
       samplingFullValidationPercent: options.samplingFullValidationPercent ?? 95,
@@ -2139,66 +2139,53 @@ var ControlAgent = class {
     }
     if (policy.budgets) {
       const applicableBudgets = this.findApplicableBudgets(policy.budgets, request);
-      if (this.options.enableHybridEnforcement) {
+      if (applicableBudgets.length > 0) {
+        let mostRestrictiveDecision = null;
+        let mostRestrictivePriority = -1;
         for (const budget of applicableBudgets) {
-          const decision = await this.evaluateBudgetWithHybridEnforcement(
-            request,
-            budget,
-            throttleInfo
-          );
-          if (decision) {
-            return decision;
+          let decision = null;
+          if (this.options.enableHybridEnforcement) {
+            decision = await this.evaluateBudgetWithHybridEnforcement(
+              request,
+              budget,
+              throttleInfo
+            );
+          } else {
+            decision = this.evaluateBudgetLocally(request, budget, throttleInfo);
           }
-          if (policy.degradations) {
+          if (!decision && policy.degradations) {
             for (const degrade of policy.degradations) {
               if (degrade.from_model === request.model && degrade.trigger === "budget_threshold" && degrade.threshold_percent) {
                 const usagePercent = budget.spent / budget.limit * 100;
                 if (usagePercent >= degrade.threshold_percent) {
-                  return {
+                  decision = {
                     action: "degrade",
                     reason: `Budget "${budget.name}" at ${usagePercent.toFixed(1)}% (threshold: ${degrade.threshold_percent}%)`,
                     degradeToModel: degrade.to_model,
                     ...throttleInfo && { throttleDelayMs: throttleInfo.delayMs }
                   };
+                  break;
                 }
               }
             }
           }
-        }
-      } else {
-        for (const budget of applicableBudgets) {
-          const projectedSpend = budget.spent + (request.estimated_cost ?? 0);
-          if (projectedSpend > budget.limit) {
-            if (budget.limitAction === "degrade" && budget.degradeToModel) {
-              return {
-                action: "degrade",
-                reason: `Budget "${budget.name}" exceeded: $${projectedSpend.toFixed(4)} > $${budget.limit}`,
-                degradeToModel: budget.degradeToModel,
-                ...throttleInfo && { throttleDelayMs: throttleInfo.delayMs }
-              };
+          if (decision) {
+            const priority = this.getActionPriority(decision.action);
+            if (priority > mostRestrictivePriority) {
+              mostRestrictiveDecision = decision;
+              mostRestrictivePriority = priority;
             }
-            const action = budget.limitAction === "kill" ? "block" : budget.limitAction;
-            return {
-              action,
-              reason: `Budget "${budget.name}" exceeded: $${projectedSpend.toFixed(4)} > $${budget.limit}`
-            };
-          }
-          if (policy.degradations) {
-            for (const degrade of policy.degradations) {
-              if (degrade.from_model === request.model && degrade.trigger === "budget_threshold" && degrade.threshold_percent) {
-                const usagePercent = budget.spent / budget.limit * 100;
-                if (usagePercent >= degrade.threshold_percent) {
-                  return {
-                    action: "degrade",
-                    reason: `Budget "${budget.name}" at ${usagePercent.toFixed(1)}% (threshold: ${degrade.threshold_percent}%)`,
-                    degradeToModel: degrade.to_model,
-                    ...throttleInfo && { throttleDelayMs: throttleInfo.delayMs }
-                  };
-                }
-              }
+            if (decision.action === "block") {
+              break;
             }
           }
         }
+        if (mostRestrictiveDecision) {
+          if (throttleInfo && mostRestrictiveDecision.action !== "block" && !mostRestrictiveDecision.throttleDelayMs) {
+            mostRestrictiveDecision.throttleDelayMs = throttleInfo.delayMs;
+          }
+          return mostRestrictiveDecision;
+        }
       }
     }
     if (policy.degradations) {
@@ -2291,36 +2278,96 @@ var ControlAgent = class {
     return true;
   }
   /**
-   * Find budgets that apply to the given request based on budget type
+   * Get action priority for finding most restrictive decision.
+   * Higher priority = more restrictive.
+   */
+  getActionPriority(action) {
+    const priority = {
+      allow: 0,
+      alert: 1,
+      throttle: 2,
+      degrade: 3,
+      block: 4
+    };
+    return priority[action] ?? 0;
+  }
+  /**
+   * Evaluate a single budget using local-only enforcement.
+   * Returns a decision if the budget triggers an action, null otherwise.
+   */
+  evaluateBudgetLocally(request, budget, throttleInfo) {
+    const projectedSpend = budget.spent + (request.estimated_cost ?? 0);
+    if (projectedSpend > budget.limit) {
+      if (budget.limitAction === "degrade" && budget.degradeToModel) {
+        return {
+          action: "degrade",
+          reason: `Budget "${budget.name}" exceeded: $${projectedSpend.toFixed(4)} > $${budget.limit}`,
+          degradeToModel: budget.degradeToModel,
+          ...throttleInfo && { throttleDelayMs: throttleInfo.delayMs }
+        };
+      }
+      const action = budget.limitAction === "kill" ? "block" : budget.limitAction;
+      return {
+        action,
+        reason: `Budget "${budget.name}" exceeded: $${projectedSpend.toFixed(4)} > $${budget.limit}`
+      };
+    }
+    return null;
+  }
+  /**
+   * Find budgets that apply to the given request based on budget type.
+   *
+   * Matching logic by budget type:
+   * - global: Matches ALL requests
+   * - agent: Matches if request.metadata.agent == budget.name or budget.id
+   * - tenant: Matches if request.metadata.tenant_id == budget.name or budget.id
+   * - customer: Matches if request.metadata.customer_id == budget.name or budget.id
+   * - feature: Matches if request.metadata.feature == budget.name or budget.id
+   * - tag: Matches if any request.metadata.tags intersect with budget.tags
+   * - legacy (context_id): Matches if request.context_id == budget.context_id
    */
   findApplicableBudgets(budgets, request) {
     const result = [];
     const metadata = request.metadata || {};
     for (const budget of budgets) {
+      if (budget.context_id && request.context_id) {
+        if (budget.context_id === request.context_id) {
+          result.push(budget);
+          continue;
+        }
+      }
       switch (budget.type) {
         case "global":
           result.push(budget);
           break;
-        case "agent":
-          if (metadata.agent_id && budget.id.includes(String(metadata.agent_id))) {
+        case "agent": {
+          const agent = metadata.agent;
+          if (agent && (agent === budget.name || agent === budget.id)) {
             result.push(budget);
           }
           break;
-        case "tenant":
-          if (metadata.tenant_id && budget.id.includes(String(metadata.tenant_id))) {
+        }
+        case "tenant": {
+          const tenantId = metadata.tenant_id;
+          if (tenantId && (tenantId === budget.name || tenantId === budget.id)) {
             result.push(budget);
           }
           break;
-        case "customer":
-          if (metadata.customer_id && budget.id.includes(String(metadata.customer_id)) || request.context_id && budget.id.includes(request.context_id)) {
+        }
+        case "customer": {
+          const customerId = metadata.customer_id;
+          if (customerId && (customerId === budget.name || customerId === budget.id) || request.context_id && (request.context_id === budget.name || request.context_id === budget.id)) {
             result.push(budget);
           }
           break;
-        case "feature":
-          if (metadata.feature && budget.id.includes(String(metadata.feature))) {
+        }
+        case "feature": {
+          const feature = metadata.feature;
+          if (feature && (feature === budget.name || feature === budget.id)) {
             result.push(budget);
           }
           break;
+        }
         case "tag":
           if (budget.tags && Array.isArray(metadata.tags)) {
             const requestTags = metadata.tags;
@@ -2370,13 +2417,48 @@ var ControlAgent = class {
     if (this.cachedPolicy?.budgets && event.total_tokens > 0) {
       const estimatedCost = this.estimateCost(event);
       const contextId2 = this.options.getContextId?.();
+      const metadata = event.metadata || {};
       for (const budget of this.cachedPolicy.budgets) {
-        if (budget.type === "global") {
-          budget.spent += estimatedCost;
-        } else if (budget.type === "customer" && contextId2) {
-          if (budget.id === contextId2 || budget.id.includes(contextId2)) {
-            budget.spent += estimatedCost;
+        let shouldUpdate = false;
+        switch (budget.type) {
+          case "global":
+            shouldUpdate = true;
+            break;
+          case "agent": {
+            const agent = metadata.agent;
+            shouldUpdate = Boolean(agent && (agent === budget.name || agent === budget.id));
+            break;
+          }
+          case "tenant": {
+            const tenantId = metadata.tenant_id;
+            shouldUpdate = Boolean(tenantId && (tenantId === budget.name || tenantId === budget.id));
+            break;
           }
+          case "customer": {
+            const customerId = metadata.customer_id;
+            shouldUpdate = Boolean(
+              customerId && (customerId === budget.name || customerId === budget.id) || contextId2 && (contextId2 === budget.name || contextId2 === budget.id)
+            );
+            break;
+          }
+          case "feature": {
+            const feature = metadata.feature;
+            shouldUpdate = Boolean(feature && (feature === budget.name || feature === budget.id));
+            break;
+          }
+          case "tag": {
+            if (budget.tags && Array.isArray(metadata.tags)) {
+              const requestTags = metadata.tags;
+              shouldUpdate = budget.tags.some((tag) => requestTags.includes(tag));
+            }
+            break;
+          }
+        }
+        if (!shouldUpdate && budget.context_id && contextId2) {
+          shouldUpdate = budget.context_id === contextId2;
+        }
+        if (shouldUpdate) {
+          budget.spent += estimatedCost;
         }
       }
     }
@@ -2441,7 +2523,7 @@ var ControlAgent = class {
     if (this.options.adaptiveThresholdEnabled) {
       if (remainingBudgetUsd <= this.options.adaptiveMinRemainingUsd) {
         console.debug(
-          `[aden] Remaining budget $${remainingBudgetUsd.toFixed(2)} <= $${this.options.adaptiveMinRemainingUsd.toFixed(2)}, forcing validation`
+          `[aden] Remaining budget $${remainingBudgetUsd.toFixed(4)} <= $${this.options.adaptiveMinRemainingUsd.toFixed(2)}, forcing validation`
         );
         return true;
       }
@@ -2578,7 +2660,7 @@ var ControlAgent = class {
     }
     if (this.shouldValidateWithServer(usagePercent, remaining, limit)) {
       console.debug(
-        `[aden] Budget '${budget.id}' at ${usagePercent.toFixed(1)}% ($${currentSpend.toFixed(2)}/$${limit.toFixed(2)}), validating with server`
+        `[aden] Budget '${budget.name}' at ${usagePercent.toFixed(1)}% ($${currentSpend.toFixed(6)}/$${limit.toFixed(6)}), validating with server`
       );
       const validation = await this.validateBudgetWithServer(
         budget.id,

package/dist/index.mjs

@@ -1769,12 +1769,12 @@ var ControlAgent = class {
       instanceId: options.instanceId ?? randomUUID6(),
       onAlert: options.onAlert ?? (() => {
       }),
-      // Hybrid enforcement options
-      enableHybridEnforcement: options.enableHybridEnforcement ?? false,
-      serverValidationThreshold: options.serverValidationThreshold ?? 80,
+      // Hybrid enforcement options (defaults match Python SDK)
+      enableHybridEnforcement: options.enableHybridEnforcement ?? true,
+      serverValidationThreshold: options.serverValidationThreshold ?? 5,
       serverValidationTimeoutMs: options.serverValidationTimeoutMs ?? 2e3,
       adaptiveThresholdEnabled: options.adaptiveThresholdEnabled ?? true,
-      adaptiveMinRemainingUsd: options.adaptiveMinRemainingUsd ?? 1,
+      adaptiveMinRemainingUsd: options.adaptiveMinRemainingUsd ?? 5,
       samplingEnabled: options.samplingEnabled ?? true,
       samplingBaseRate: options.samplingBaseRate ?? 0.1,
       samplingFullValidationPercent: options.samplingFullValidationPercent ?? 95,
@@ -2022,66 +2022,53 @@ var ControlAgent = class {
     }
     if (policy.budgets) {
       const applicableBudgets = this.findApplicableBudgets(policy.budgets, request);
-      if (this.options.enableHybridEnforcement) {
+      if (applicableBudgets.length > 0) {
+        let mostRestrictiveDecision = null;
+        let mostRestrictivePriority = -1;
         for (const budget of applicableBudgets) {
-          const decision = await this.evaluateBudgetWithHybridEnforcement(
-            request,
-            budget,
-            throttleInfo
-          );
-          if (decision) {
-            return decision;
+          let decision = null;
+          if (this.options.enableHybridEnforcement) {
+            decision = await this.evaluateBudgetWithHybridEnforcement(
+              request,
+              budget,
+              throttleInfo
+            );
+          } else {
+            decision = this.evaluateBudgetLocally(request, budget, throttleInfo);
           }
-          if (policy.degradations) {
+          if (!decision && policy.degradations) {
             for (const degrade of policy.degradations) {
               if (degrade.from_model === request.model && degrade.trigger === "budget_threshold" && degrade.threshold_percent) {
                 const usagePercent = budget.spent / budget.limit * 100;
                 if (usagePercent >= degrade.threshold_percent) {
-                  return {
+                  decision = {
                     action: "degrade",
                     reason: `Budget "${budget.name}" at ${usagePercent.toFixed(1)}% (threshold: ${degrade.threshold_percent}%)`,
                     degradeToModel: degrade.to_model,
                     ...throttleInfo && { throttleDelayMs: throttleInfo.delayMs }
                   };
+                  break;
                 }
               }
             }
           }
-        }
-      } else {
-        for (const budget of applicableBudgets) {
-          const projectedSpend = budget.spent + (request.estimated_cost ?? 0);
-          if (projectedSpend > budget.limit) {
-            if (budget.limitAction === "degrade" && budget.degradeToModel) {
-              return {
-                action: "degrade",
-                reason: `Budget "${budget.name}" exceeded: $${projectedSpend.toFixed(4)} > $${budget.limit}`,
-                degradeToModel: budget.degradeToModel,
-                ...throttleInfo && { throttleDelayMs: throttleInfo.delayMs }
-              };
+          if (decision) {
+            const priority = this.getActionPriority(decision.action);
+            if (priority > mostRestrictivePriority) {
+              mostRestrictiveDecision = decision;
+              mostRestrictivePriority = priority;
             }
-            const action = budget.limitAction === "kill" ? "block" : budget.limitAction;
-            return {
-              action,
-              reason: `Budget "${budget.name}" exceeded: $${projectedSpend.toFixed(4)} > $${budget.limit}`
-            };
-          }
-          if (policy.degradations) {
-            for (const degrade of policy.degradations) {
-              if (degrade.from_model === request.model && degrade.trigger === "budget_threshold" && degrade.threshold_percent) {
-                const usagePercent = budget.spent / budget.limit * 100;
-                if (usagePercent >= degrade.threshold_percent) {
-                  return {
-                    action: "degrade",
-                    reason: `Budget "${budget.name}" at ${usagePercent.toFixed(1)}% (threshold: ${degrade.threshold_percent}%)`,
-                    degradeToModel: degrade.to_model,
-                    ...throttleInfo && { throttleDelayMs: throttleInfo.delayMs }
-                  };
-                }
-              }
+            if (decision.action === "block") {
+              break;
             }
           }
         }
+        if (mostRestrictiveDecision) {
+          if (throttleInfo && mostRestrictiveDecision.action !== "block" && !mostRestrictiveDecision.throttleDelayMs) {
+            mostRestrictiveDecision.throttleDelayMs = throttleInfo.delayMs;
+          }
+          return mostRestrictiveDecision;
+        }
       }
     }
     if (policy.degradations) {
@@ -2174,36 +2161,96 @@ var ControlAgent = class {
     return true;
   }
   /**
-   * Find budgets that apply to the given request based on budget type
+   * Get action priority for finding most restrictive decision.
+   * Higher priority = more restrictive.
+   */
+  getActionPriority(action) {
+    const priority = {
+      allow: 0,
+      alert: 1,
+      throttle: 2,
+      degrade: 3,
+      block: 4
+    };
+    return priority[action] ?? 0;
+  }
+  /**
+   * Evaluate a single budget using local-only enforcement.
+   * Returns a decision if the budget triggers an action, null otherwise.
+   */
+  evaluateBudgetLocally(request, budget, throttleInfo) {
+    const projectedSpend = budget.spent + (request.estimated_cost ?? 0);
+    if (projectedSpend > budget.limit) {
+      if (budget.limitAction === "degrade" && budget.degradeToModel) {
+        return {
+          action: "degrade",
+          reason: `Budget "${budget.name}" exceeded: $${projectedSpend.toFixed(4)} > $${budget.limit}`,
+          degradeToModel: budget.degradeToModel,
+          ...throttleInfo && { throttleDelayMs: throttleInfo.delayMs }
+        };
+      }
+      const action = budget.limitAction === "kill" ? "block" : budget.limitAction;
+      return {
+        action,
+        reason: `Budget "${budget.name}" exceeded: $${projectedSpend.toFixed(4)} > $${budget.limit}`
+      };
+    }
+    return null;
+  }
+  /**
+   * Find budgets that apply to the given request based on budget type.
+   *
+   * Matching logic by budget type:
+   * - global: Matches ALL requests
+   * - agent: Matches if request.metadata.agent == budget.name or budget.id
+   * - tenant: Matches if request.metadata.tenant_id == budget.name or budget.id
+   * - customer: Matches if request.metadata.customer_id == budget.name or budget.id
+   * - feature: Matches if request.metadata.feature == budget.name or budget.id
+   * - tag: Matches if any request.metadata.tags intersect with budget.tags
+   * - legacy (context_id): Matches if request.context_id == budget.context_id
    */
   findApplicableBudgets(budgets, request) {
     const result = [];
     const metadata = request.metadata || {};
     for (const budget of budgets) {
+      if (budget.context_id && request.context_id) {
+        if (budget.context_id === request.context_id) {
+          result.push(budget);
+          continue;
+        }
+      }
       switch (budget.type) {
         case "global":
           result.push(budget);
           break;
-        case "agent":
-          if (metadata.agent_id && budget.id.includes(String(metadata.agent_id))) {
+        case "agent": {
+          const agent = metadata.agent;
+          if (agent && (agent === budget.name || agent === budget.id)) {
             result.push(budget);
           }
           break;
-        case "tenant":
-          if (metadata.tenant_id && budget.id.includes(String(metadata.tenant_id))) {
+        }
+        case "tenant": {
+          const tenantId = metadata.tenant_id;
+          if (tenantId && (tenantId === budget.name || tenantId === budget.id)) {
             result.push(budget);
           }
           break;
-        case "customer":
-          if (metadata.customer_id && budget.id.includes(String(metadata.customer_id)) || request.context_id && budget.id.includes(request.context_id)) {
+        }
+        case "customer": {
+          const customerId = metadata.customer_id;
+          if (customerId && (customerId === budget.name || customerId === budget.id) || request.context_id && (request.context_id === budget.name || request.context_id === budget.id)) {
             result.push(budget);
           }
           break;
-        case "feature":
-          if (metadata.feature && budget.id.includes(String(metadata.feature))) {
+        }
+        case "feature": {
+          const feature = metadata.feature;
+          if (feature && (feature === budget.name || feature === budget.id)) {
             result.push(budget);
           }
           break;
+        }
         case "tag":
           if (budget.tags && Array.isArray(metadata.tags)) {
             const requestTags = metadata.tags;
@@ -2253,13 +2300,48 @@ var ControlAgent = class {
     if (this.cachedPolicy?.budgets && event.total_tokens > 0) {
       const estimatedCost = this.estimateCost(event);
       const contextId2 = this.options.getContextId?.();
+      const metadata = event.metadata || {};
       for (const budget of this.cachedPolicy.budgets) {
-        if (budget.type === "global") {
-          budget.spent += estimatedCost;
-        } else if (budget.type === "customer" && contextId2) {
-          if (budget.id === contextId2 || budget.id.includes(contextId2)) {
-            budget.spent += estimatedCost;
+        let shouldUpdate = false;
+        switch (budget.type) {
+          case "global":
+            shouldUpdate = true;
+            break;
+          case "agent": {
+            const agent = metadata.agent;
+            shouldUpdate = Boolean(agent && (agent === budget.name || agent === budget.id));
+            break;
+          }
+          case "tenant": {
+            const tenantId = metadata.tenant_id;
+            shouldUpdate = Boolean(tenantId && (tenantId === budget.name || tenantId === budget.id));
+            break;
           }
+          case "customer": {
+            const customerId = metadata.customer_id;
+            shouldUpdate = Boolean(
+              customerId && (customerId === budget.name || customerId === budget.id) || contextId2 && (contextId2 === budget.name || contextId2 === budget.id)
+            );
+            break;
+          }
+          case "feature": {
+            const feature = metadata.feature;
+            shouldUpdate = Boolean(feature && (feature === budget.name || feature === budget.id));
+            break;
+          }
+          case "tag": {
+            if (budget.tags && Array.isArray(metadata.tags)) {
+              const requestTags = metadata.tags;
+              shouldUpdate = budget.tags.some((tag) => requestTags.includes(tag));
+            }
+            break;
+          }
+        }
+        if (!shouldUpdate && budget.context_id && contextId2) {
+          shouldUpdate = budget.context_id === contextId2;
+        }
+        if (shouldUpdate) {
+          budget.spent += estimatedCost;
         }
       }
     }
@@ -2324,7 +2406,7 @@ var ControlAgent = class {
     if (this.options.adaptiveThresholdEnabled) {
       if (remainingBudgetUsd <= this.options.adaptiveMinRemainingUsd) {
         console.debug(
-          `[aden] Remaining budget $${remainingBudgetUsd.toFixed(2)} <= $${this.options.adaptiveMinRemainingUsd.toFixed(2)}, forcing validation`
+          `[aden] Remaining budget $${remainingBudgetUsd.toFixed(4)} <= $${this.options.adaptiveMinRemainingUsd.toFixed(2)}, forcing validation`
         );
         return true;
       }
@@ -2461,7 +2543,7 @@ var ControlAgent = class {
     }
     if (this.shouldValidateWithServer(usagePercent, remaining, limit)) {
       console.debug(
-        `[aden] Budget '${budget.id}' at ${usagePercent.toFixed(1)}% ($${currentSpend.toFixed(2)}/$${limit.toFixed(2)}), validating with server`
+        `[aden] Budget '${budget.name}' at ${usagePercent.toFixed(1)}% ($${currentSpend.toFixed(6)}/$${limit.toFixed(6)}), validating with server`
       );
       const validation = await this.validateBudgetWithServer(
         budget.id,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "aden-ts",
-  "version": "0.1.1",
+  "version": "0.2.0",
   "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",