avantgate 1.1.0 → 1.1.2

package/dist/index.js

@@ -20,64 +20,236 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // src/index.ts
 var index_exports = {};
 __export(index_exports, {
+  AvantGateBudgetExceededError: () => BudgetExceededError,
+  AvantGateConfigurationError: () => ConfigurationError,
   AvantGateControlLayer: () => AvantGateControlLayer,
+  BudgetExceededError: () => BudgetExceededError,
+  CachedPricingAdapter: () => CachedPricingAdapter,
+  ConfigurationError: () => ConfigurationError,
   DEFAULT_MODEL_PRICES: () => DEFAULT_MODEL_PRICES,
+  HttpProviderClient: () => HttpProviderClient,
+  PricingRegistry: () => PricingRegistry,
   PromptBuilder: () => PromptBuilder,
   PromptRegistry: () => PromptRegistry,
   PromptTemplate: () => PromptTemplate,
+  SEED_MODEL_PRICES: () => SEED_MODEL_PRICES,
   ZenLLMControlLayer: () => AvantGateControlLayer,
   calculateCostUSD: () => calculateCostUSD,
   createAvantGate: () => createAvantGate,
+  createHttpProviderClient: () => createHttpProviderClient,
   createLLMControlLayer: () => createLLMControlLayer,
   extractAndCleanJSON: () => extractAndCleanJSON,
+  resolveModelPrice: () => resolveModelPrice,
+  resolveModelPriceAsync: () => resolveModelPriceAsync,
   sanitizePII: () => sanitizePII,
   validateUserInput: () => validateUserInput,
   validateWithZod: () => validateWithZod
 });
 module.exports = __toCommonJS(index_exports);
 
+// src/types.ts
+var ConfigurationError = class extends Error {
+  constructor(message) {
+    super(message);
+    this.name = "ConfigurationError";
+  }
+};
+var BudgetExceededError = class extends Error {
+  constructor(message) {
+    super(message);
+    this.name = "BudgetExceededError";
+  }
+};
+
 // src/pricing.ts
-var DEFAULT_MODEL_PRICES = {
-  // DeepSeek
-  "deepseek-chat": {
-    promptUSDPerMillion: 0.14,
-    completionUSDPerMillion: 0.28,
-    cacheHitUSDPerMillion: 0.014
-  },
-  "deepseek-reasoner": {
-    promptUSDPerMillion: 0.55,
-    completionUSDPerMillion: 2.19,
-    cacheHitUSDPerMillion: 0.14
-  },
-  // Mistral
-  "mistral-small-latest": {
-    promptUSDPerMillion: 0.2,
-    completionUSDPerMillion: 0.6
-  },
-  "mistral-large-latest": {
-    promptUSDPerMillion: 2,
-    completionUSDPerMillion: 6
+var SEED_MODEL_PRICES = {
+  // DeepSeek Direct
+  "deepseek-chat": { promptUSDPerMillion: 0.14, completionUSDPerMillion: 0.28, cacheHitUSDPerMillion: 0.014 },
+  "deepseek-reasoner": { promptUSDPerMillion: 0.55, completionUSDPerMillion: 2.19, cacheHitUSDPerMillion: 0.14 },
+  "deepseek/deepseek-chat": { promptUSDPerMillion: 0.14, completionUSDPerMillion: 0.28, cacheHitUSDPerMillion: 0.014 },
+  "deepseek/deepseek-reasoner": { promptUSDPerMillion: 0.55, completionUSDPerMillion: 2.19, cacheHitUSDPerMillion: 0.14 },
+  // Mistral AI Direct
+  "mistral-small-latest": { promptUSDPerMillion: 0.2, completionUSDPerMillion: 0.6 },
+  "mistral-large-latest": { promptUSDPerMillion: 2, completionUSDPerMillion: 6 },
+  "mistral/mistral-small-latest": { promptUSDPerMillion: 0.2, completionUSDPerMillion: 0.6 },
+  "mistral/mistral-large-latest": { promptUSDPerMillion: 2, completionUSDPerMillion: 6 },
+  // OpenAI Direct
+  "gpt-4o-mini": { promptUSDPerMillion: 0.15, completionUSDPerMillion: 0.6 },
+  "gpt-4o": { promptUSDPerMillion: 2.5, completionUSDPerMillion: 10 },
+  "openai/gpt-4o-mini": { promptUSDPerMillion: 0.15, completionUSDPerMillion: 0.6 },
+  "openai/gpt-4o": { promptUSDPerMillion: 2.5, completionUSDPerMillion: 10 },
+  // OpenRouter Direct
+  "openrouter/deepseek/deepseek-chat": { promptUSDPerMillion: 0.14, completionUSDPerMillion: 0.28 },
+  "openrouter/deepseek/deepseek-r1": { promptUSDPerMillion: 0.55, completionUSDPerMillion: 2.19 },
+  "openrouter/anthropic/claude-3.5-sonnet": { promptUSDPerMillion: 3, completionUSDPerMillion: 15 },
+  // Local Ollama (Toujours $0)
+  "ollama": { promptUSDPerMillion: 0, completionUSDPerMillion: 0 }
+};
+function normalizeKey(identifier) {
+  return identifier.trim().toLowerCase().replace(":", "/");
+}
+var CachedPricingAdapter = class {
+  cache = /* @__PURE__ */ new Map();
+  ttlMs;
+  delegate;
+  constructor(delegate, ttlMs = 5 * 60 * 1e3) {
+    this.delegate = delegate;
+    this.ttlMs = ttlMs;
+  }
+  async fetchPrice(model, provider) {
+    const key = provider ? normalizeKey(`${provider}/${model}`) : normalizeKey(model);
+    const cached = this.cache.get(key);
+    if (cached && Date.now() < cached.expiresAt) {
+      return cached.price;
+    }
+    try {
+      const freshPrice = await this.delegate.fetchPrice(model, provider);
+      if (freshPrice) {
+        this.cache.set(key, { price: freshPrice, expiresAt: Date.now() + this.ttlMs });
+        return freshPrice;
+      }
+    } catch {
+      if (cached) {
+        return cached.price;
+      }
+    }
+    return void 0;
+  }
+  peek(model, provider) {
+    const key = provider ? normalizeKey(`${provider}/${model}`) : normalizeKey(model);
+    return this.cache.get(key)?.price;
+  }
+  clearCache() {
+    this.cache.clear();
+  }
+  invalidate(model, provider) {
+    if (!model) {
+      this.clearCache();
+      return;
+    }
+    const key = provider ? normalizeKey(`${provider}/${model}`) : normalizeKey(model);
+    this.cache.delete(key);
+  }
+};
+var PricingRegistry = class {
+  static prices = /* @__PURE__ */ new Map();
+  static adapter;
+  static registerPrice(identifier, price) {
+    this.prices.set(normalizeKey(identifier), price);
+  }
+  static registerDistributorPrices(distributor, priceMap) {
+    for (const [model, price] of Object.entries(priceMap)) {
+      this.prices.set(normalizeKey(`${distributor}/${model}`), price);
+    }
+  }
+  static registerPrices(prices) {
+    for (const [key, price] of Object.entries(prices)) {
+      this.prices.set(normalizeKey(key), price);
+    }
+  }
+  static getPrice(model, provider) {
+    if (provider) {
+      const distributorKey = normalizeKey(`${provider}/${model}`);
+      const directDistributorPrice = this.prices.get(distributorKey);
+      if (directDistributorPrice) {
+        return directDistributorPrice;
+      }
+    }
+    return this.prices.get(normalizeKey(model));
+  }
+  static setAdapter(adapter) {
+    this.adapter = adapter;
+  }
+  static getAdapter() {
+    return this.adapter;
+  }
+  static clear() {
+    this.prices.clear();
+    this.adapter = void 0;
+  }
+  /**
+   * Réinitialise le registre en chargeant le catalogue d'exemple SEED_MODEL_PRICES.
+   */
+  static loadSeedPrices() {
+    this.registerPrices(SEED_MODEL_PRICES);
+  }
+  static getAllPrices() {
+    const out = {};
+    for (const [k, v] of this.prices.entries()) {
+      out[k] = v;
+    }
+    return out;
+  }
+};
+var DEFAULT_MODEL_PRICES = new Proxy({}, {
+  get(_target, prop) {
+    return PricingRegistry.getPrice(prop);
   },
-  // OpenAI
-  "gpt-4o-mini": {
-    promptUSDPerMillion: 0.15,
-    completionUSDPerMillion: 0.6
+  has(_target, prop) {
+    return Boolean(PricingRegistry.getPrice(prop));
   },
-  "gpt-4o": {
-    promptUSDPerMillion: 2.5,
-    completionUSDPerMillion: 10
+  ownKeys() {
+    return Object.keys(PricingRegistry.getAllPrices());
   },
-  // Ollama (local)
-  "ollama": {
-    promptUSDPerMillion: 0,
-    completionUSDPerMillion: 0
+  getOwnPropertyDescriptor(_target, prop) {
+    const price = PricingRegistry.getPrice(prop);
+    if (price) {
+      return { value: price, enumerable: true, configurable: true, writable: false };
+    }
+    return void 0;
+  }
+});
+function resolveModelPrice(model, options) {
+  const opts = typeof options === "string" ? { provider: options } : options ?? {};
+  if (opts.providerPricing) {
+    return opts.providerPricing;
+  }
+  if (opts.customPricing) {
+    if (opts.provider) {
+      const customDistributorKey = normalizeKey(`${opts.provider}/${model}`);
+      if (opts.customPricing[customDistributorKey]) {
+        return opts.customPricing[customDistributorKey];
+      }
+    }
+    if (opts.customPricing[normalizeKey(model)] ?? opts.customPricing[model]) {
+      return opts.customPricing[normalizeKey(model)] ?? opts.customPricing[model];
+    }
+  }
+  const activeAdapter = opts.adapter ?? PricingRegistry.getAdapter();
+  if (activeAdapter instanceof CachedPricingAdapter) {
+    const cached = activeAdapter.peek(model, opts.provider);
+    if (cached) {
+      return cached;
+    }
+  }
+  const registeredPrice = PricingRegistry.getPrice(model, opts.provider);
+  if (registeredPrice) {
+    return registeredPrice;
+  }
+  if (opts.provider === "ollama" || model.toLowerCase().includes("ollama")) {
+    return { promptUSDPerMillion: 0, completionUSDPerMillion: 0 };
+  }
+  return void 0;
+}
+async function resolveModelPriceAsync(model, options) {
+  const opts = typeof options === "string" ? { provider: options } : options ?? {};
+  const activeAdapter = opts.adapter ?? PricingRegistry.getAdapter();
+  if (activeAdapter) {
+    try {
+      const adapterPrice = await activeAdapter.fetchPrice(model, opts.provider);
+      if (adapterPrice) {
+        return adapterPrice;
+      }
+    } catch {
+    }
+  }
+  return resolveModelPrice(model, options);
+}
+function calculateCostUSD(model, promptTokens, completionTokens, cacheHitTokens = 0, options) {
+  const pricing = resolveModelPrice(model, options);
+  if (!pricing) {
+    return 0;
   }
-};
-function calculateCostUSD(model, promptTokens, completionTokens, cacheHitTokens = 0) {
-  const pricing = DEFAULT_MODEL_PRICES[model] ?? {
-    promptUSDPerMillion: 0.5,
-    completionUSDPerMillion: 1.5
-  };
   const promptCost = promptTokens / 1e6 * pricing.promptUSDPerMillion;
   const completionCost = completionTokens / 1e6 * pricing.completionUSDPerMillion;
   const cacheDiscount = cacheHitTokens && pricing.cacheHitUSDPerMillion ? cacheHitTokens / 1e6 * (pricing.promptUSDPerMillion - pricing.cacheHitUSDPerMillion) : 0;
@@ -673,11 +845,144 @@ function validateWithZod(rawText, schema, options) {
   return schema.parse(parsed);
 }
 
+// src/providers/http-client.ts
+var DEFAULT_BASE_URLS = {
+  deepseek: "https://api.deepseek.com/v1",
+  mistral: "https://api.mistral.ai/v1",
+  openai: "https://api.openai.com/v1",
+  ollama: "http://localhost:11434/v1",
+  openrouter: "https://openrouter.ai/api/v1"
+};
+var HttpProviderClient = class {
+  name;
+  baseUrl;
+  apiKey;
+  providerType;
+  constructor(config) {
+    this.providerType = config.provider;
+    this.name = `${config.provider}-http`;
+    this.apiKey = config.apiKey;
+    this.baseUrl = config.baseUrl || DEFAULT_BASE_URLS[config.provider] || "https://api.openai.com/v1";
+  }
+  resolveEndpoint() {
+    const trimmed = this.baseUrl.replace(/\/+$/, "");
+    if (trimmed.endsWith("/chat/completions")) {
+      return trimmed;
+    }
+    return `${trimmed}/chat/completions`;
+  }
+  buildHeaders() {
+    const headers = {
+      "Content-Type": "application/json"
+    };
+    if (this.apiKey) {
+      headers["Authorization"] = `Bearer ${this.apiKey}`;
+    }
+    if (this.providerType === "openrouter") {
+      headers["HTTP-Referer"] = "https://avantgate.dev";
+      headers["X-Title"] = "AvantGate";
+    }
+    return headers;
+  }
+  buildPayload(options, model) {
+    return {
+      model,
+      messages: options.messages.map((m) => ({
+        role: m.role,
+        content: m.content
+      })),
+      temperature: options.temperature ?? 0.2,
+      ...options.responseFormat ? { response_format: options.responseFormat } : {}
+    };
+  }
+  extractUsage(usageData) {
+    if (!usageData) {
+      return void 0;
+    }
+    return {
+      promptTokens: usageData.prompt_tokens,
+      completionTokens: usageData.completion_tokens,
+      totalTokens: usageData.total_tokens,
+      promptCacheHitTokens: usageData.prompt_tokens_details?.cached_tokens
+    };
+  }
+  async complete(options) {
+    const endpoint = this.resolveEndpoint();
+    const model = options.model ?? "default";
+    const headers = this.buildHeaders();
+    const body = JSON.stringify(this.buildPayload(options, model));
+    const response = await fetch(endpoint, {
+      method: "POST",
+      headers,
+      body
+    });
+    if (!response.ok) {
+      const errorText = await response.text();
+      throw new Error(
+        `[AvantGate HTTP Provider ${this.providerType}] HTTP ${response.status} ${response.statusText}: ${errorText}`
+      );
+    }
+    const data = await response.json();
+    const text = data.choices?.[0]?.message?.content ?? "";
+    const usage = this.extractUsage(data.usage);
+    return { text, usage };
+  }
+};
+function createHttpProviderClient(config) {
+  return new HttpProviderClient(config);
+}
+
 // src/control-layer.ts
 var AvantGateControlLayer = class {
   config;
+  cachedPricingAdapter;
   constructor(config) {
-    this.config = config;
+    this.config = {
+      ...config,
+      primary: this.resolveProviderConfig(config.primary),
+      fallback: this.resolveProviderConfig(config.fallback),
+      emergencyFallback: this.resolveProviderConfig(config.emergencyFallback)
+    };
+    if (config.pricingAdapter) {
+      this.cachedPricingAdapter = new CachedPricingAdapter(
+        config.pricingAdapter,
+        config.pricingCacheTtlMs ?? 5 * 60 * 1e3
+      );
+    }
+  }
+  resolveProviderConfig(provider) {
+    if (!provider) return void 0;
+    if (provider.client) return provider;
+    if (provider.apiKey || provider.baseUrl || provider.provider === "ollama") {
+      return {
+        ...provider,
+        client: createHttpProviderClient(provider)
+      };
+    }
+    return provider;
+  }
+  findProviderConfig(provider, model) {
+    const list = [this.config.primary, this.config.fallback, this.config.emergencyFallback].filter(
+      (p) => Boolean(p)
+    );
+    if (provider) {
+      const matchProvider = list.find((p) => p.provider === provider);
+      if (matchProvider) return matchProvider;
+    }
+    if (model) {
+      const matchModel = list.find((p) => p.model === model);
+      if (matchModel) return matchModel;
+    }
+    return this.config.primary;
+  }
+  calculateCost(model, promptTokens, completionTokens, cacheHitTokens = 0, provider) {
+    const providerCfg = this.findProviderConfig(provider, model);
+    return calculateCostUSD(model, promptTokens, completionTokens, cacheHitTokens, {
+      provider: provider ?? providerCfg?.provider,
+      providerPricing: providerCfg?.pricing,
+      customPricing: this.config.customPricing,
+      adapter: this.cachedPricingAdapter
+    });
   }
   applySecurityGuards(userQuery) {
     const guard = validateUserInput(userQuery, {
@@ -692,6 +997,46 @@ var AvantGateControlLayer = class {
     }
     return userQuery;
   }
+  checkPreflightBudget(estimatedPromptTokens, targetModel, provider) {
+    if (this.config.maxTokenBudget !== void 0 && estimatedPromptTokens > this.config.maxTokenBudget) {
+      throw new BudgetExceededError(
+        `[AvantGate Budget Guard] Pre-flight token budget exceeded: estimated prompt (${estimatedPromptTokens} tokens) exceeds maxTokenBudget (${this.config.maxTokenBudget}).`
+      );
+    }
+    if (this.config.maxCostUSD !== void 0) {
+      const isLocalFree = provider === "ollama" || targetModel.toLowerCase().includes("ollama");
+      const providerCfg = this.findProviderConfig(provider, targetModel);
+      const price = resolveModelPrice(targetModel, {
+        provider: provider ?? providerCfg?.provider,
+        providerPricing: providerCfg?.pricing,
+        customPricing: this.config.customPricing,
+        adapter: this.cachedPricingAdapter
+      });
+      if (!isLocalFree && !price) {
+        throw new ConfigurationError(
+          `[AvantGate Configuration Error] 'maxCostUSD' was set to $${this.config.maxCostUSD}, but no pricing was configured for model '${targetModel}'. Please define pricing in ProviderConfig, customPricing, or via PricingAdapter.`
+        );
+      }
+      const estimatedPromptCost = this.calculateCost(targetModel, estimatedPromptTokens, 0, 0, provider);
+      if (estimatedPromptCost > this.config.maxCostUSD) {
+        throw new BudgetExceededError(
+          `[AvantGate Budget Guard] Pre-flight cost budget exceeded: estimated prompt cost ($${estimatedPromptCost.toFixed(6)}) exceeds maxCostUSD ($${this.config.maxCostUSD}).`
+        );
+      }
+    }
+  }
+  checkPostExecutionBudget(tokensTotal, costUSD) {
+    if (this.config.maxTokenBudget !== void 0 && tokensTotal > this.config.maxTokenBudget) {
+      throw new BudgetExceededError(
+        `[AvantGate Budget Guard] Execution total tokens (${tokensTotal}) exceeded maxTokenBudget (${this.config.maxTokenBudget}).`
+      );
+    }
+    if (this.config.maxCostUSD !== void 0 && costUSD > this.config.maxCostUSD) {
+      throw new BudgetExceededError(
+        `[AvantGate Budget Guard] Execution cost ($${costUSD.toFixed(6)}) exceeded maxCostUSD ($${this.config.maxCostUSD}).`
+      );
+    }
+  }
   buildMessages(systemPrompt, query) {
     const messages = [];
     if (systemPrompt) {
@@ -704,7 +1049,12 @@ var AvantGateControlLayer = class {
     const promptTokens = rawUsage?.promptTokens ?? Math.ceil(query.length / 4);
     const completionTokens = rawUsage?.completionTokens ?? Math.ceil(text.length / 4);
     const totalTokens = rawUsage?.totalTokens ?? promptTokens + completionTokens;
-    return { promptTokens, completionTokens, totalTokens };
+    return {
+      promptTokens,
+      completionTokens,
+      totalTokens,
+      promptCacheHitTokens: rawUsage?.promptCacheHitTokens
+    };
   }
   getProviderChain() {
     return [
@@ -716,7 +1066,7 @@ var AvantGateControlLayer = class {
   executeSimulation(query, systemPrompt) {
     const promptTokens = Math.ceil(((systemPrompt?.length ?? 0) + query.length) / 4);
     const completionTokens = 50;
-    const cost = calculateCostUSD(this.config.primary.model, promptTokens, completionTokens);
+    const cost = this.calculateCost(this.config.primary.model, promptTokens, completionTokens);
     return {
       text: `[AvantGate In-Process Engine] Response simulation for model: ${this.config.primary.model}`,
       tokens: {
@@ -772,10 +1122,11 @@ var AvantGateControlLayer = class {
     throw lastError;
   }
   assembleResult(output) {
-    const costUSD = calculateCostUSD(
+    const costUSD = this.calculateCost(
       output.modelUsed,
       output.usage.promptTokens,
-      output.usage.completionTokens
+      output.usage.completionTokens,
+      output.usage.promptCacheHitTokens ?? 0
     );
     return {
       text: output.responseText,
@@ -804,18 +1155,27 @@ var AvantGateControlLayer = class {
     });
   }
   /**
-   * Exécute une requête avec garde d'entrée, masquage PII, et calcul des coûts.
+   * Exécute une requête avec garde d'entrée, masquage PII, garde pré-vol et calcul des coûts.
    */
   async execute(options) {
     const sanitizedQuery = this.applySecurityGuards(options.userQuery);
     const messages = this.buildMessages(options.systemPrompt, sanitizedQuery);
+    const promptLength = (options.systemPrompt?.length ?? 0) + sanitizedQuery.length;
+    const estimatedPromptTokens = Math.ceil(promptLength / 4);
+    this.checkPreflightBudget(estimatedPromptTokens, this.config.primary.model, this.config.primary.provider);
     if (!options.providerOverride && this.getProviderChain().length === 0) {
-      const simResult = this.executeSimulation(sanitizedQuery, options.systemPrompt);
-      await this.notifyAuditSink(simResult);
-      return simResult;
+      if (this.config.mockSimulation) {
+        const simResult = this.executeSimulation(sanitizedQuery, options.systemPrompt);
+        await this.notifyAuditSink(simResult);
+        return simResult;
+      }
+      throw new ConfigurationError(
+        "[AvantGate Configuration Error] No active LLM provider configured. Provide a client implementing LLMProviderPort or configure credentials (apiKey / baseUrl)."
+      );
     }
     const output = options.providerOverride ? await this.executeOverrideProvider(options.providerOverride, messages, sanitizedQuery, options.temperature) : await this.executeProviderPipeline(messages, sanitizedQuery, options.temperature);
     const result = this.assembleResult(output);
+    this.checkPostExecutionBudget(result.tokens.total, result.costUSD);
     await this.notifyAuditSink(result);
     return result;
   }
@@ -858,6 +1218,9 @@ var AvantGateControlLayer = class {
       }
       return msg;
     });
+    const promptLength = processedMessages.reduce((sum, msg) => sum + msg.content.length, 0);
+    const estimatedPromptTokens = Math.ceil(promptLength / 4);
+    this.checkPreflightBudget(estimatedPromptTokens, modelToUse);
     let lastError;
     let accumulatedPromptTokens = 0;
     let accumulatedCompletionTokens = 0;
@@ -880,10 +1243,16 @@ var AvantGateControlLayer = class {
           attemptCompletionTokens = usage.completionTokens;
           modelUsed = modelToUse;
         } else if (this.getProviderChain().length === 0) {
-          const sim = this.executeSimulation(JSON.stringify(processedMessages));
-          responseText = sim.text;
-          attemptPromptTokens = sim.tokens.prompt;
-          attemptCompletionTokens = sim.tokens.completion;
+          if (this.config.mockSimulation) {
+            const sim = this.executeSimulation(JSON.stringify(processedMessages));
+            responseText = sim.text;
+            attemptPromptTokens = sim.tokens.prompt;
+            attemptCompletionTokens = sim.tokens.completion;
+          } else {
+            throw new ConfigurationError(
+              "[AvantGate Configuration Error] No active LLM provider configured. Provide a client implementing LLMProviderPort or configure credentials (apiKey / baseUrl)."
+            );
+          }
         } else {
           const output = await this.executeProviderPipeline(
             processedMessages,
@@ -907,7 +1276,8 @@ var AvantGateControlLayer = class {
           jurisdiction: this.config.features?.finance?.jurisdiction
         });
         const totalTokens = accumulatedPromptTokens + accumulatedCompletionTokens;
-        const costUSD = calculateCostUSD(modelUsed, accumulatedPromptTokens, accumulatedCompletionTokens);
+        const costUSD = this.calculateCost(modelUsed, accumulatedPromptTokens, accumulatedCompletionTokens);
+        this.checkPostExecutionBudget(totalTokens, costUSD);
         const result = {
           data: parsedData,
           rawText: responseText,
@@ -1261,16 +1631,27 @@ var PromptRegistry = class {
 };
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  AvantGateBudgetExceededError,
+  AvantGateConfigurationError,
   AvantGateControlLayer,
+  BudgetExceededError,
+  CachedPricingAdapter,
+  ConfigurationError,
   DEFAULT_MODEL_PRICES,
+  HttpProviderClient,
+  PricingRegistry,
   PromptBuilder,
   PromptRegistry,
   PromptTemplate,
+  SEED_MODEL_PRICES,
   ZenLLMControlLayer,
   calculateCostUSD,
   createAvantGate,
+  createHttpProviderClient,
   createLLMControlLayer,
   extractAndCleanJSON,
+  resolveModelPrice,
+  resolveModelPriceAsync,
   sanitizePII,
   validateUserInput,
   validateWithZod