aden-ts 0.2.1 → 0.2.3

package/dist/index.d.mts

@@ -26,6 +26,8 @@ interface ControlDecision {
     reason?: string;
     /** If action is "degrade", switch to this model */
     degradeToModel?: string;
+    /** If action is "degrade", the provider for the degraded model */
+    degradeToProvider?: string;
     /** If action is "throttle", delay by this many milliseconds */
     throttleDelayMs?: number;
     /** If action is "alert", the severity level */
@@ -63,6 +65,8 @@ interface ControlEvent extends BaseEvent {
     reason?: string;
     /** If degraded, what model was used instead */
     degraded_to?: string;
+    /** If degraded, the provider for the degraded model */
+    degraded_to_provider?: string;
     /** If throttled, how long was the delay in ms */
     throttle_delay_ms?: number;
     /** Estimated cost that triggered the decision */
@@ -136,6 +140,8 @@ interface BudgetRule {
     limitAction: LimitAction;
     /** If limitAction is "degrade", switch to this model */
     degradeToModel?: string;
+    /** If limitAction is "degrade", the provider for the degraded model */
+    degradeToProvider?: string;
     /** Alert thresholds */
     alerts: BudgetAlert[];
     /** Notification settings */
@@ -173,12 +179,14 @@ interface BlockRule {
  * Degrade rule - automatic model downgrade
  */
 interface DegradeRule {
+    /** Provider this rule applies to (required - no cross-vendor degradation) */
+    provider: string;
     /** Model to downgrade from */
     from_model: string;
     /** Model to downgrade to */
     to_model: string;
     /** When to trigger the downgrade */
-    trigger: "budget_threshold" | "rate_limit" | "always";
+    trigger: "budget_threshold" | "rate_limit" | "always" | "budget_exceeded";
     /** For budget_threshold: percentage at which to trigger (0-100) */
     threshold_percent?: number;
     /** Context ID this rule applies to (omit for all) */
@@ -379,6 +387,8 @@ interface BudgetValidationResponse {
     projectedPercent?: number;
     /** Model to degrade to (if action is "degrade") */
     degradeToModel?: string;
+    /** Provider for the degraded model (if action is "degrade") */
+    degradeToProvider?: string;
 }
 /**
  * Control Agent interface - the public API
@@ -596,6 +606,7 @@ type BeforeRequestResult = {
 } | {
     action: "degrade";
     toModel: string;
+    toProvider?: string;
     reason?: string;
     delayMs?: number;
 } | {
@@ -2291,6 +2302,8 @@ declare class ControlAgent implements IControlAgent {
     private requestsSinceLastHeartbeat;
     private errorsSinceLastHeartbeat;
     private requestCounts;
+    private pricingCache;
+    private pricingAliases;
     constructor(options: ControlAgentOptions);
     /**
      * Connect to the control server
@@ -2321,6 +2334,14 @@ declare class ControlAgent implements IControlAgent {
      * Fetch policy via HTTP
      */
     private fetchPolicy;
+    /**
+     * Fetch pricing table from server and cache it
+     */
+    private fetchPricing;
+    /**
+     * Get pricing for a model from cached pricing table
+     */
+    private getModelPricing;
     /**
      * Start heartbeat timer
      */
@@ -2387,8 +2408,8 @@ declare class ControlAgent implements IControlAgent {
      */
     reportMetric(event: MetricEvent): Promise<void>;
     /**
-     * Estimate cost from a metric event
-     * Uses gpt-4o pricing as default: $2.50/1M input, $10/1M output
+     * Estimate cost from a metric event using server pricing table.
+     * Falls back to default pricing if model not found.
      */
     private estimateCost;
     /**

package/dist/index.d.ts

@@ -26,6 +26,8 @@ interface ControlDecision {
     reason?: string;
     /** If action is "degrade", switch to this model */
     degradeToModel?: string;
+    /** If action is "degrade", the provider for the degraded model */
+    degradeToProvider?: string;
     /** If action is "throttle", delay by this many milliseconds */
     throttleDelayMs?: number;
     /** If action is "alert", the severity level */
@@ -63,6 +65,8 @@ interface ControlEvent extends BaseEvent {
     reason?: string;
     /** If degraded, what model was used instead */
     degraded_to?: string;
+    /** If degraded, the provider for the degraded model */
+    degraded_to_provider?: string;
     /** If throttled, how long was the delay in ms */
     throttle_delay_ms?: number;
     /** Estimated cost that triggered the decision */
@@ -136,6 +140,8 @@ interface BudgetRule {
     limitAction: LimitAction;
     /** If limitAction is "degrade", switch to this model */
     degradeToModel?: string;
+    /** If limitAction is "degrade", the provider for the degraded model */
+    degradeToProvider?: string;
     /** Alert thresholds */
     alerts: BudgetAlert[];
     /** Notification settings */
@@ -173,12 +179,14 @@ interface BlockRule {
  * Degrade rule - automatic model downgrade
  */
 interface DegradeRule {
+    /** Provider this rule applies to (required - no cross-vendor degradation) */
+    provider: string;
     /** Model to downgrade from */
     from_model: string;
     /** Model to downgrade to */
     to_model: string;
     /** When to trigger the downgrade */
-    trigger: "budget_threshold" | "rate_limit" | "always";
+    trigger: "budget_threshold" | "rate_limit" | "always" | "budget_exceeded";
     /** For budget_threshold: percentage at which to trigger (0-100) */
     threshold_percent?: number;
     /** Context ID this rule applies to (omit for all) */
@@ -379,6 +387,8 @@ interface BudgetValidationResponse {
     projectedPercent?: number;
     /** Model to degrade to (if action is "degrade") */
     degradeToModel?: string;
+    /** Provider for the degraded model (if action is "degrade") */
+    degradeToProvider?: string;
 }
 /**
  * Control Agent interface - the public API
@@ -596,6 +606,7 @@ type BeforeRequestResult = {
 } | {
     action: "degrade";
     toModel: string;
+    toProvider?: string;
     reason?: string;
     delayMs?: number;
 } | {
@@ -2291,6 +2302,8 @@ declare class ControlAgent implements IControlAgent {
     private requestsSinceLastHeartbeat;
     private errorsSinceLastHeartbeat;
     private requestCounts;
+    private pricingCache;
+    private pricingAliases;
     constructor(options: ControlAgentOptions);
     /**
      * Connect to the control server
@@ -2321,6 +2334,14 @@ declare class ControlAgent implements IControlAgent {
      * Fetch policy via HTTP
      */
     private fetchPolicy;
+    /**
+     * Fetch pricing table from server and cache it
+     */
+    private fetchPricing;
+    /**
+     * Get pricing for a model from cached pricing table
+     */
+    private getModelPricing;
     /**
      * Start heartbeat timer
      */
@@ -2387,8 +2408,8 @@ declare class ControlAgent implements IControlAgent {
      */
     reportMetric(event: MetricEvent): Promise<void>;
     /**
-     * Estimate cost from a metric event
-     * Uses gpt-4o pricing as default: $2.50/1M input, $10/1M output
+     * Estimate cost from a metric event using server pricing table.
+     * Falls back to default pricing if model not found.
      */
     private estimateCost;
     /**

package/dist/index.js

@@ -1970,6 +1970,9 @@ var ControlAgent = class {
     this.errorsSinceLastHeartbeat = 0;
     // Rate limiting tracking
     this.requestCounts = /* @__PURE__ */ new Map();
+    // Pricing cache (fetched from server)
+    this.pricingCache = /* @__PURE__ */ new Map();
+    this.pricingAliases = /* @__PURE__ */ new Map();
     this.options = {
       serverUrl: options.serverUrl.replace(/\/$/, ""),
       apiKey: options.apiKey,
@@ -2003,7 +2006,12 @@ var ControlAgent = class {
       await this.connectWebSocket();
     } else {
       logger.debug("Using HTTP polling mode (no WebSocket URL)");
-      this.startPolling();
+      await this.startPolling();
+    }
+    try {
+      await this.fetchPricing();
+    } catch (error) {
+      logger.warn("Failed to fetch pricing table:", error);
     }
     this.startHeartbeat();
   }
@@ -2137,6 +2145,62 @@ var ControlAgent = class {
       logger.warn("Failed to fetch policy:", error);
     }
   }
+  /**
+   * Fetch pricing table from server and cache it
+   */
+  async fetchPricing() {
+    logger.debug("Fetching pricing table from server...");
+    try {
+      const response = await this.httpRequest("/tsdb/pricing", "GET");
+      if (response.ok) {
+        const data = await response.json();
+        const pricing = data.pricing ?? {};
+        for (const [model, rates] of Object.entries(pricing)) {
+          const modelLower = model.toLowerCase();
+          this.pricingCache.set(modelLower, {
+            input: rates.input ?? 1,
+            output: rates.output ?? 3,
+            cached_input: rates.cached_input ?? (rates.input ?? 1) * 0.25
+          });
+          if (rates.aliases) {
+            for (const alias of rates.aliases) {
+              this.pricingAliases.set(alias.toLowerCase(), modelLower);
+            }
+          }
+        }
+        logger.debug(`Loaded pricing for ${this.pricingCache.size} models`);
+      } else {
+        logger.debug(`Pricing fetch returned status ${response.status}`);
+      }
+    } catch (error) {
+      logger.warn("Failed to fetch pricing:", error);
+    }
+  }
+  /**
+   * Get pricing for a model from cached pricing table
+   */
+  getModelPricing(model) {
+    const fallback = { input: 1, output: 3, cached_input: 0.25 };
+    if (!model) {
+      return fallback;
+    }
+    const modelLower = model.toLowerCase();
+    if (this.pricingCache.has(modelLower)) {
+      return this.pricingCache.get(modelLower);
+    }
+    if (this.pricingAliases.has(modelLower)) {
+      const canonical = this.pricingAliases.get(modelLower);
+      if (this.pricingCache.has(canonical)) {
+        return this.pricingCache.get(canonical);
+      }
+    }
+    for (const [cachedModel, rates] of this.pricingCache) {
+      if (modelLower.startsWith(cachedModel) || cachedModel.startsWith(modelLower)) {
+        return rates;
+      }
+    }
+    return fallback;
+  }
   /**
    * Start heartbeat timer
    */
@@ -2261,13 +2325,14 @@ var ControlAgent = class {
           }
           if (!decision && policy.degradations) {
             for (const degrade of policy.degradations) {
-              if (degrade.from_model === request.model && degrade.trigger === "budget_threshold" && degrade.threshold_percent) {
+              if (degrade.provider === request.provider && degrade.from_model === request.model && degrade.trigger === "budget_threshold" && degrade.threshold_percent) {
                 const usagePercent = budget.spent / budget.limit * 100;
                 if (usagePercent >= degrade.threshold_percent) {
                   decision = {
                     action: "degrade",
                     reason: `Budget "${budget.name}" at ${usagePercent.toFixed(1)}% (threshold: ${degrade.threshold_percent}%)`,
                     degradeToModel: degrade.to_model,
+                    degradeToProvider: degrade.provider,
                     ...throttleInfo && { throttleDelayMs: throttleInfo.delayMs }
                   };
                   break;
@@ -2296,12 +2361,13 @@ var ControlAgent = class {
     }
     if (policy.degradations) {
       for (const degrade of policy.degradations) {
-        if (degrade.from_model === request.model && degrade.trigger === "always") {
+        if (degrade.provider === request.provider && degrade.from_model === request.model && degrade.trigger === "always") {
           if (!degrade.context_id || degrade.context_id === request.context_id) {
             return {
               action: "degrade",
               reason: "Model degradation rule (always)",
               degradeToModel: degrade.to_model,
+              degradeToProvider: degrade.provider,
               ...throttleInfo && { throttleDelayMs: throttleInfo.delayMs }
             };
           }
@@ -2409,6 +2475,7 @@ var ControlAgent = class {
           action: "degrade",
           reason: `Budget "${budget.name}" exceeded: $${projectedSpend.toFixed(4)} > $${budget.limit}`,
           degradeToModel: budget.degradeToModel,
+          degradeToProvider: budget.degradeToProvider,
           ...throttleInfo && { throttleDelayMs: throttleInfo.delayMs }
         };
       }
@@ -2570,13 +2637,20 @@ var ControlAgent = class {
     }
   }
   /**
-   * Estimate cost from a metric event
-   * Uses gpt-4o pricing as default: $2.50/1M input, $10/1M output
+   * Estimate cost from a metric event using server pricing table.
+   * Falls back to default pricing if model not found.
    */
   estimateCost(event) {
-    const inputCost = event.input_tokens * 25e-7;
-    const outputCost = event.output_tokens * 1e-5;
-    return inputCost + outputCost;
+    if (event.total_tokens === 0) {
+      return 0;
+    }
+    const rates = this.getModelPricing(event.model);
+    const cachedTokens = event.cached_tokens ?? 0;
+    const regularInput = Math.max(0, event.input_tokens - cachedTokens);
+    const inputCost = regularInput * rates.input / 1e6;
+    const cachedCost = cachedTokens * rates.cached_input / 1e6;
+    const outputCost = event.output_tokens * rates.output / 1e6;
+    return inputCost + cachedCost + outputCost;
   }
   // ===========================================================================
   // Hybrid Enforcement - Server-Side Budget Validation
@@ -2715,7 +2789,8 @@ var ControlAgent = class {
             updatedSpend: data.updated_spend ?? 0,
             reason: data.reason,
             projectedPercent: data.projected_percent,
-            degradeToModel: data.degrade_to_model
+            degradeToModel: data.degrade_to_model,
+            degradeToProvider: data.degrade_to_provider
           };
         } else {
           logger.warn(`Server validation returned status ${response.status}`);
@@ -2744,7 +2819,8 @@ var ControlAgent = class {
     return {
       action: validation.action,
       reason: validation.reason ?? `Server validation: ${validation.action}`,
-      degradeToModel: validation.degradeToModel
+      degradeToModel: validation.degradeToModel,
+      degradeToProvider: validation.degradeToProvider
     };
   }
   /**
@@ -2793,6 +2869,7 @@ var ControlAgent = class {
           action: "degrade",
           reason: `Budget "${budget.name}" exceeded: $${projectedSpend.toFixed(4)} > $${limit} (${projectedPercent.toFixed(1)}%)`,
           degradeToModel: budget.degradeToModel,
+          degradeToProvider: budget.degradeToProvider,
           ...throttleInfo && { throttleDelayMs: throttleInfo.delayMs }
         };
       }
@@ -3059,6 +3136,7 @@ async function getControlDecision(options, provider, model, spanId) {
     action: decision.action,
     reason: decision.reason,
     degraded_to: decision.degradeToModel,
+    degraded_to_provider: decision.degradeToProvider,
     throttle_delay_ms: decision.throttleDelayMs
   });
   return { decision, originalModel };
@@ -3347,6 +3425,7 @@ function createControlBeforeRequestHook(controlAgent, getContextId, userHook) {
         return {
           action: "degrade",
           toModel: decision.degradeToModel,
+          toProvider: decision.degradeToProvider,
           reason: decision.reason,
           ...decision.throttleDelayMs && { delayMs: decision.throttleDelayMs }
         };

package/dist/index.mjs

@@ -1853,6 +1853,9 @@ var ControlAgent = class {
     this.errorsSinceLastHeartbeat = 0;
     // Rate limiting tracking
     this.requestCounts = /* @__PURE__ */ new Map();
+    // Pricing cache (fetched from server)
+    this.pricingCache = /* @__PURE__ */ new Map();
+    this.pricingAliases = /* @__PURE__ */ new Map();
     this.options = {
       serverUrl: options.serverUrl.replace(/\/$/, ""),
       apiKey: options.apiKey,
@@ -1886,7 +1889,12 @@ var ControlAgent = class {
       await this.connectWebSocket();
     } else {
       logger.debug("Using HTTP polling mode (no WebSocket URL)");
-      this.startPolling();
+      await this.startPolling();
+    }
+    try {
+      await this.fetchPricing();
+    } catch (error) {
+      logger.warn("Failed to fetch pricing table:", error);
     }
     this.startHeartbeat();
   }
@@ -2020,6 +2028,62 @@ var ControlAgent = class {
       logger.warn("Failed to fetch policy:", error);
     }
   }
+  /**
+   * Fetch pricing table from server and cache it
+   */
+  async fetchPricing() {
+    logger.debug("Fetching pricing table from server...");
+    try {
+      const response = await this.httpRequest("/tsdb/pricing", "GET");
+      if (response.ok) {
+        const data = await response.json();
+        const pricing = data.pricing ?? {};
+        for (const [model, rates] of Object.entries(pricing)) {
+          const modelLower = model.toLowerCase();
+          this.pricingCache.set(modelLower, {
+            input: rates.input ?? 1,
+            output: rates.output ?? 3,
+            cached_input: rates.cached_input ?? (rates.input ?? 1) * 0.25
+          });
+          if (rates.aliases) {
+            for (const alias of rates.aliases) {
+              this.pricingAliases.set(alias.toLowerCase(), modelLower);
+            }
+          }
+        }
+        logger.debug(`Loaded pricing for ${this.pricingCache.size} models`);
+      } else {
+        logger.debug(`Pricing fetch returned status ${response.status}`);
+      }
+    } catch (error) {
+      logger.warn("Failed to fetch pricing:", error);
+    }
+  }
+  /**
+   * Get pricing for a model from cached pricing table
+   */
+  getModelPricing(model) {
+    const fallback = { input: 1, output: 3, cached_input: 0.25 };
+    if (!model) {
+      return fallback;
+    }
+    const modelLower = model.toLowerCase();
+    if (this.pricingCache.has(modelLower)) {
+      return this.pricingCache.get(modelLower);
+    }
+    if (this.pricingAliases.has(modelLower)) {
+      const canonical = this.pricingAliases.get(modelLower);
+      if (this.pricingCache.has(canonical)) {
+        return this.pricingCache.get(canonical);
+      }
+    }
+    for (const [cachedModel, rates] of this.pricingCache) {
+      if (modelLower.startsWith(cachedModel) || cachedModel.startsWith(modelLower)) {
+        return rates;
+      }
+    }
+    return fallback;
+  }
   /**
    * Start heartbeat timer
    */
@@ -2144,13 +2208,14 @@ var ControlAgent = class {
           }
           if (!decision && policy.degradations) {
             for (const degrade of policy.degradations) {
-              if (degrade.from_model === request.model && degrade.trigger === "budget_threshold" && degrade.threshold_percent) {
+              if (degrade.provider === request.provider && degrade.from_model === request.model && degrade.trigger === "budget_threshold" && degrade.threshold_percent) {
                 const usagePercent = budget.spent / budget.limit * 100;
                 if (usagePercent >= degrade.threshold_percent) {
                   decision = {
                     action: "degrade",
                     reason: `Budget "${budget.name}" at ${usagePercent.toFixed(1)}% (threshold: ${degrade.threshold_percent}%)`,
                     degradeToModel: degrade.to_model,
+                    degradeToProvider: degrade.provider,
                     ...throttleInfo && { throttleDelayMs: throttleInfo.delayMs }
                   };
                   break;
@@ -2179,12 +2244,13 @@ var ControlAgent = class {
     }
     if (policy.degradations) {
       for (const degrade of policy.degradations) {
-        if (degrade.from_model === request.model && degrade.trigger === "always") {
+        if (degrade.provider === request.provider && degrade.from_model === request.model && degrade.trigger === "always") {
           if (!degrade.context_id || degrade.context_id === request.context_id) {
             return {
               action: "degrade",
               reason: "Model degradation rule (always)",
               degradeToModel: degrade.to_model,
+              degradeToProvider: degrade.provider,
               ...throttleInfo && { throttleDelayMs: throttleInfo.delayMs }
             };
           }
@@ -2292,6 +2358,7 @@ var ControlAgent = class {
           action: "degrade",
           reason: `Budget "${budget.name}" exceeded: $${projectedSpend.toFixed(4)} > $${budget.limit}`,
           degradeToModel: budget.degradeToModel,
+          degradeToProvider: budget.degradeToProvider,
           ...throttleInfo && { throttleDelayMs: throttleInfo.delayMs }
         };
       }
@@ -2453,13 +2520,20 @@ var ControlAgent = class {
     }
   }
   /**
-   * Estimate cost from a metric event
-   * Uses gpt-4o pricing as default: $2.50/1M input, $10/1M output
+   * Estimate cost from a metric event using server pricing table.
+   * Falls back to default pricing if model not found.
    */
   estimateCost(event) {
-    const inputCost = event.input_tokens * 25e-7;
-    const outputCost = event.output_tokens * 1e-5;
-    return inputCost + outputCost;
+    if (event.total_tokens === 0) {
+      return 0;
+    }
+    const rates = this.getModelPricing(event.model);
+    const cachedTokens = event.cached_tokens ?? 0;
+    const regularInput = Math.max(0, event.input_tokens - cachedTokens);
+    const inputCost = regularInput * rates.input / 1e6;
+    const cachedCost = cachedTokens * rates.cached_input / 1e6;
+    const outputCost = event.output_tokens * rates.output / 1e6;
+    return inputCost + cachedCost + outputCost;
   }
   // ===========================================================================
   // Hybrid Enforcement - Server-Side Budget Validation
@@ -2598,7 +2672,8 @@ var ControlAgent = class {
             updatedSpend: data.updated_spend ?? 0,
             reason: data.reason,
             projectedPercent: data.projected_percent,
-            degradeToModel: data.degrade_to_model
+            degradeToModel: data.degrade_to_model,
+            degradeToProvider: data.degrade_to_provider
           };
         } else {
           logger.warn(`Server validation returned status ${response.status}`);
@@ -2627,7 +2702,8 @@ var ControlAgent = class {
     return {
       action: validation.action,
       reason: validation.reason ?? `Server validation: ${validation.action}`,
-      degradeToModel: validation.degradeToModel
+      degradeToModel: validation.degradeToModel,
+      degradeToProvider: validation.degradeToProvider
     };
   }
   /**
@@ -2676,6 +2752,7 @@ var ControlAgent = class {
           action: "degrade",
           reason: `Budget "${budget.name}" exceeded: $${projectedSpend.toFixed(4)} > $${limit} (${projectedPercent.toFixed(1)}%)`,
           degradeToModel: budget.degradeToModel,
+          degradeToProvider: budget.degradeToProvider,
           ...throttleInfo && { throttleDelayMs: throttleInfo.delayMs }
         };
       }
@@ -2942,6 +3019,7 @@ async function getControlDecision(options, provider, model, spanId) {
     action: decision.action,
     reason: decision.reason,
     degraded_to: decision.degradeToModel,
+    degraded_to_provider: decision.degradeToProvider,
     throttle_delay_ms: decision.throttleDelayMs
   });
   return { decision, originalModel };
@@ -3230,6 +3308,7 @@ function createControlBeforeRequestHook(controlAgent, getContextId, userHook) {
         return {
           action: "degrade",
           toModel: decision.degradeToModel,
+          toProvider: decision.degradeToProvider,
           reason: decision.reason,
           ...decision.throttleDelayMs && { delayMs: decision.throttleDelayMs }
         };

package/package.json

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