@routstr/sdk 0.2.6 → 0.2.7

package/dist/index.js

@@ -468,7 +468,7 @@ var ModelManager = class _ModelManager {
       }
     }
     const DEFAULT_RELAYS = [
-      "wss://relay.primal.net",
+      "wss://relay.damus.io",
       "wss://nos.lol",
       "wss://relay.routstr.com"
     ];
@@ -1083,8 +1083,9 @@ var CashuSpender = class {
     return null;
   }
   /**
-   * Refund all xcashu tokens from storage and increment tryCounts on failure.
-   * Reuses receiveToken from BalanceManager/CashuSpender for receiving refunds.
+   * Refund all xcashu tokens from storage by calling the provider's refund endpoint.
+   * The xcashu token acts as an API key to claim the refund, and the response contains
+   * the actual refunded Cashu token which is then received into the wallet.
    * @param mintUrl - The mint URL for receiving tokens
    * @param excludeBaseUrls - Base URLs to exclude from refund (optional)
    * @returns Results for each xcashu token refund attempt
@@ -1097,7 +1098,20 @@ var CashuSpender = class {
       if (excludedUrls.has(baseUrl)) continue;
       for (const xcashuToken of tokens) {
         try {
-          const receiveResult = await this.receiveToken(xcashuToken.token);
+          if (!this.balanceManager) {
+            throw new Error("BalanceManager not available for xcashu refund");
+          }
+          const fetchResult = await this.balanceManager.fetchRefundToken(
+            baseUrl,
+            xcashuToken.token,
+            true
+          );
+          if (!fetchResult.success || !fetchResult.token) {
+            throw new Error(
+              fetchResult.error || "Failed to fetch refund token from provider"
+            );
+          }
+          const receiveResult = await this.receiveToken(fetchResult.token);
           if (receiveResult.success) {
             this.storageAdapter.removeXcashuToken(baseUrl, xcashuToken.token);
             results.push({
@@ -1112,7 +1126,10 @@ var CashuSpender = class {
           } else {
             const currentTryCount = xcashuToken.tryCount ?? 0;
             const newTryCount = currentTryCount + 1;
-            this.storageAdapter.updateXcashuTokenTryCount(xcashuToken.token, newTryCount);
+            this.storageAdapter.updateXcashuTokenTryCount(
+              xcashuToken.token,
+              newTryCount
+            );
             results.push({
               baseUrl,
               token: xcashuToken.token,
@@ -1121,13 +1138,16 @@ var CashuSpender = class {
             });
             this._log(
               "DEBUG",
-              `[CashuSpender] refundXcashuTokens: Failed to refund xcashu token for ${baseUrl}, incremented tryCount to ${newTryCount}`
+              `[CashuSpender] refundXcashuTokens: Failed to receive refund token for ${baseUrl}, incremented tryCount to ${newTryCount}`
             );
           }
         } catch (error) {
           const currentTryCount = xcashuToken.tryCount ?? 0;
           const newTryCount = currentTryCount + 1;
-          this.storageAdapter.updateXcashuTokenTryCount(xcashuToken.token, newTryCount);
+          this.storageAdapter.updateXcashuTokenTryCount(
+            xcashuToken.token,
+            newTryCount
+          );
           const errorMessage = error instanceof Error ? error.message : String(error);
           results.push({
             baseUrl,
@@ -1164,7 +1184,10 @@ var CashuSpender = class {
         if (refundResult.success) {
           this.storageAdapter.removeApiKey(apiKeyEntry.baseUrl);
         } else {
-          this.storageAdapter.updateApiKeyBalance(apiKeyEntry.baseUrl, apiKeyEntry.amount);
+          this.storageAdapter.updateApiKeyBalance(
+            apiKeyEntry.baseUrl,
+            apiKeyEntry.amount
+          );
         }
         results.push({
           baseUrl: apiKeyEntry.baseUrl,
@@ -1302,7 +1325,7 @@ var BalanceManager = class {
     }
     let fetchResult;
     try {
-      fetchResult = await this._fetchRefundTokenWithApiKey(baseUrl, apiKey);
+      fetchResult = await this.fetchRefundToken(baseUrl, apiKey);
       if (!fetchResult.success) {
         return {
           success: false,
@@ -1338,9 +1361,9 @@ var BalanceManager = class {
     }
   }
   /**
-   * Fetch refund token from provider API using API key authentication
+   * Fetch refund token from provider API using API key (or xcashu token) authentication
    */
-  async _fetchRefundTokenWithApiKey(baseUrl, apiKey) {
+  async fetchRefundToken(baseUrl, apiKeyOrToken, xCashu = false) {
     if (!baseUrl) {
       return {
         success: false,
@@ -1354,12 +1377,17 @@ var BalanceManager = class {
       controller.abort();
     }, 6e4);
     try {
+      const headers = {
+        "Content-Type": "application/json"
+      };
+      if (xCashu) {
+        headers["X-Cashu"] = apiKeyOrToken;
+      } else {
+        headers["Authorization"] = `Bearer ${apiKeyOrToken}`;
+      }
       const response = await fetch(url, {
         method: "POST",
-        headers: {
-          Authorization: `Bearer ${apiKey}`,
-          "Content-Type": "application/json"
-        },
+        headers,
         signal: controller.signal
       });
       clearTimeout(timeoutId);
@@ -1380,10 +1408,7 @@ var BalanceManager = class {
       };
     } catch (error) {
       clearTimeout(timeoutId);
-      console.error(
-        "[BalanceManager._fetchRefundTokenWithApiKey] Fetch error",
-        error
-      );
+      console.error("[BalanceManager.fetchRefundToken] Fetch error", error);
       if (error instanceof Error) {
         if (error.name === "AbortError") {
           return {
@@ -1430,11 +1455,7 @@ var BalanceManager = class {
         };
       }
       cashuToken = tokenResult.token;
-      const topUpResult = await this._postTopUp(
-        baseUrl,
-        apiKey,
-        cashuToken
-      );
+      const topUpResult = await this._postTopUp(baseUrl, apiKey, cashuToken);
       requestId = topUpResult.requestId;
       console.log(topUpResult);
       if (!topUpResult.success) {
@@ -1800,7 +1821,7 @@ var BalanceManager = class {
         console.log(response.status);
         const data = await response.json();
         console.log("FAILED ", data);
-        const isInvalidApiKey = response.status === 401 && data?.code === "invalid_api_key" && data?.message?.includes("proofs already spent");
+        const isInvalidApiKey = response.status === 401 && data?.detail?.error?.code === "invalid_api_key" && data?.detail?.error?.message?.includes("proofs already spent");
         return {
           amount: -1,
           reserved: data.reserved ?? 0,
@@ -2298,8 +2319,13 @@ function isInsecureHttpUrl(url) {
   return url.startsWith("http://");
 }
 var ProviderManager = class _ProviderManager {
-  constructor(providerRegistry) {
+  constructor(providerRegistry, store) {
     this.providerRegistry = providerRegistry;
+    this.instanceId = `pm_${Date.now().toString(36)}_${Math.random().toString(36).slice(2, 8)}`;
+    if (store) {
+      this.store = store;
+      this.hydrateFromStore();
+    }
   }
   failedProviders = /* @__PURE__ */ new Set();
   /** Track when each provider last failed (provider URL -> timestamp) */
@@ -2308,14 +2334,57 @@ var ProviderManager = class _ProviderManager {
   providersOnCoolDown = [];
   /** Cooldown duration in milliseconds (5 minutes) */
   static COOLDOWN_DURATION_MS = 5 * 60 * 1e3;
+  /** Optional persistent store for failure tracking */
+  store = null;
+  /** Instance ID for debugging */
+  instanceId;
+  /**
+   * Hydrate in-memory state from persistent store
+   */
+  hydrateFromStore() {
+    if (!this.store) return;
+    const state = this.store.getState();
+    this.failedProviders = new Set(state.failedProviders);
+    this.lastFailed = new Map(Object.entries(state.lastFailed));
+    const now = Date.now();
+    this.providersOnCoolDown = state.providersOnCooldown.filter(
+      (entry) => now - entry.timestamp < _ProviderManager.COOLDOWN_DURATION_MS
+    ).map((entry) => [entry.baseUrl, entry.timestamp]);
+    console.log(`[ProviderManager:${this.instanceId}] Hydrated from store:`);
+    console.log(`  failedProviders: ${this.failedProviders.size}`);
+    console.log(`  lastFailed: ${this.lastFailed.size}`);
+    console.log(`  providersOnCooldown: ${this.providersOnCoolDown.length}`);
+  }
+  /**
+   * Get instance ID for debugging
+   */
+  getInstanceId() {
+    return this.instanceId;
+  }
   /**
    * Clean up expired cooldown entries
    */
   cleanupExpiredCooldowns() {
     const now = Date.now();
+    const before = this.providersOnCoolDown.length;
     this.providersOnCoolDown = this.providersOnCoolDown.filter(
-      ([, timestamp]) => now - timestamp < _ProviderManager.COOLDOWN_DURATION_MS
+      ([url, timestamp]) => {
+        const age = now - timestamp;
+        const isExpired = age >= _ProviderManager.COOLDOWN_DURATION_MS;
+        if (isExpired) {
+          console.log(
+            `[cleanupExpiredCooldowns:${this.instanceId}] Removing expired cooldown for ${url} (age: ${age}ms, cooldown: ${_ProviderManager.COOLDOWN_DURATION_MS}ms)`
+          );
+        }
+        return !isExpired;
+      }
     );
+    const after = this.providersOnCoolDown.length;
+    if (before !== after) {
+      console.log(
+        `[cleanupExpiredCooldowns:${this.instanceId}] Cleaned up ${before - after} expired cooldown(s), ${after} remaining`
+      );
+    }
   }
   /**
    * Get the cooldown duration in milliseconds
@@ -2328,7 +2397,8 @@ var ProviderManager = class _ProviderManager {
    */
   isOnCooldown(baseUrl) {
     this.cleanupExpiredCooldowns();
-    return this.providersOnCoolDown.some(([url]) => url === baseUrl);
+    const result = this.providersOnCoolDown.some(([url]) => url === baseUrl);
+    return result;
   }
   /**
    * Get all providers currently on cooldown
@@ -2342,6 +2412,9 @@ var ProviderManager = class _ProviderManager {
    */
   resetFailedProviders() {
     this.failedProviders.clear();
+    if (this.store) {
+      this.store.getState().setFailedProviders([]);
+    }
   }
   /**
    * Get the last failed timestamp for a provider
@@ -2362,13 +2435,62 @@ var ProviderManager = class _ProviderManager {
   markFailed(baseUrl) {
     const now = Date.now();
     const lastFailure = this.lastFailed.get(baseUrl);
+    console.log(`[markFailed:${this.instanceId}] baseUrl: ${baseUrl}`);
+    console.log(
+      `[markFailed:${this.instanceId}] lastFailure from map: ${lastFailure}`
+    );
+    console.log(
+      `[markFailed:${this.instanceId}] current timestamp (now): ${now}`
+    );
+    console.log(
+      `[markFailed:${this.instanceId}] COOLDOWN_DURATION_MS: ${_ProviderManager.COOLDOWN_DURATION_MS}`
+    );
+    if (lastFailure !== void 0) {
+      const timeSinceLastFailure = now - lastFailure;
+      console.log(
+        `[markFailed:${this.instanceId}] timeSinceLastFailure: ${timeSinceLastFailure}ms`
+      );
+      console.log(
+        `[markFailed:${this.instanceId}] isWithinCooldownWindow: ${timeSinceLastFailure < _ProviderManager.COOLDOWN_DURATION_MS}`
+      );
+    }
     this.lastFailed.set(baseUrl, now);
     this.failedProviders.add(baseUrl);
+    if (this.store) {
+      this.store.getState().setLastFailedTimestamp(baseUrl, now);
+      this.store.getState().addFailedProvider(baseUrl);
+    }
+    console.log(
+      `[markFailed:${this.instanceId}] Updated lastFailed map for ${baseUrl} to ${now}`
+    );
+    console.log(
+      `[markFailed:${this.instanceId}] failedProviders set size: ${this.failedProviders.size}`
+    );
     if (lastFailure !== void 0 && now - lastFailure < _ProviderManager.COOLDOWN_DURATION_MS) {
+      console.log(
+        `[markFailed:${this.instanceId}] Second failure detected within cooldown window for ${baseUrl}`
+      );
       if (!this.isOnCooldown(baseUrl)) {
         this.providersOnCoolDown.push([baseUrl, now]);
+        if (this.store) {
+          this.store.getState().addProviderOnCooldown(baseUrl, now);
+        }
         console.log(
-          `Provider ${baseUrl} added to cooldown after second failure within 5 minutes`
+          `[markFailed:${this.instanceId}] Provider ${baseUrl} added to cooldown after second failure within 5 minutes`
+        );
+      } else {
+        console.log(
+          `[markFailed:${this.instanceId}] Provider ${baseUrl} is already on cooldown`
+        );
+      }
+    } else {
+      if (lastFailure === void 0) {
+        console.log(
+          `[markFailed:${this.instanceId}] First failure for ${baseUrl} - not adding to cooldown yet`
+        );
+      } else {
+        console.log(
+          `[markFailed:${this.instanceId}] Failure outside cooldown window for ${baseUrl} (timeSinceLastFailure: ${now - lastFailure}ms)`
         );
       }
     }
@@ -2380,18 +2502,27 @@ var ProviderManager = class _ProviderManager {
     this.providersOnCoolDown = this.providersOnCoolDown.filter(
       ([url]) => url !== baseUrl
     );
+    if (this.store) {
+      this.store.getState().removeProviderFromCooldown(baseUrl);
+    }
   }
   /**
    * Clear all cooldown tracking
    */
   clearCooldowns() {
     this.providersOnCoolDown = [];
+    if (this.store) {
+      this.store.getState().clearProvidersOnCooldown();
+    }
   }
   /**
    * Clear all failure tracking (lastFailed timestamps)
    */
   clearFailureHistory() {
     this.lastFailed.clear();
+    if (this.store) {
+      this.store.getState().setLastFailed({});
+    }
   }
   /**
    * Check if a provider has failed
@@ -2960,7 +3091,10 @@ var SDK_STORAGE_KEYS = {
   LAST_ROUTSTR21_MODELS_UPDATE: "lastRoutstr21ModelsUpdate",
   CACHED_RECEIVE_TOKENS: "cached_receive_tokens",
   USAGE_TRACKING: "usage_tracking",
-  CLIENT_IDS: "client_ids"
+  CLIENT_IDS: "client_ids",
+  FAILED_PROVIDERS: "failed_providers",
+  LAST_FAILED: "last_failed",
+  PROVIDERS_ON_COOLDOWN: "providers_on_cooldown"
 };
 
 // storage/usageTracking/indexedDB.ts
@@ -3607,6 +3741,9 @@ var createEmptyStore = (driver) => vanilla.createStore((set, get) => ({
   lastRoutstr21ModelsUpdate: null,
   cachedReceiveTokens: [],
   clientIds: [],
+  failedProviders: [],
+  lastFailed: {},
+  providersOnCooldown: [],
   setModelsFromAllProviders: (value) => {
     const normalized = {};
     for (const [baseUrl, models] of Object.entries(value)) {
@@ -3746,6 +3883,71 @@ var createEmptyStore = (driver) => vanilla.createStore((set, get) => ({
       void driver.setItem(SDK_STORAGE_KEYS.CLIENT_IDS, normalized);
       return { clientIds: normalized };
     });
+  },
+  // ========== Failure Tracking ==========
+  setFailedProviders: (value) => {
+    const normalized = value.map((url) => normalizeBaseUrl5(url));
+    void driver.setItem(SDK_STORAGE_KEYS.FAILED_PROVIDERS, normalized);
+    set({ failedProviders: normalized });
+  },
+  addFailedProvider: (baseUrl) => {
+    const normalized = normalizeBaseUrl5(baseUrl);
+    const current = get().failedProviders;
+    if (!current.includes(normalized)) {
+      const updated = [...current, normalized];
+      void driver.setItem(SDK_STORAGE_KEYS.FAILED_PROVIDERS, updated);
+      set({ failedProviders: updated });
+    }
+  },
+  removeFailedProvider: (baseUrl) => {
+    const normalized = normalizeBaseUrl5(baseUrl);
+    const current = get().failedProviders;
+    const updated = current.filter((url) => url !== normalized);
+    void driver.setItem(SDK_STORAGE_KEYS.FAILED_PROVIDERS, updated);
+    set({ failedProviders: updated });
+  },
+  setLastFailed: (value) => {
+    const normalized = {};
+    for (const [baseUrl, timestamp] of Object.entries(value)) {
+      normalized[normalizeBaseUrl5(baseUrl)] = timestamp;
+    }
+    void driver.setItem(SDK_STORAGE_KEYS.LAST_FAILED, normalized);
+    set({ lastFailed: normalized });
+  },
+  setLastFailedTimestamp: (baseUrl, timestamp) => {
+    const normalized = normalizeBaseUrl5(baseUrl);
+    const current = get().lastFailed;
+    const updated = { ...current, [normalized]: timestamp };
+    void driver.setItem(SDK_STORAGE_KEYS.LAST_FAILED, updated);
+    set({ lastFailed: updated });
+  },
+  setProvidersOnCooldown: (value) => {
+    const normalized = value.map((entry) => ({
+      baseUrl: normalizeBaseUrl5(entry.baseUrl),
+      timestamp: entry.timestamp
+    }));
+    void driver.setItem(SDK_STORAGE_KEYS.PROVIDERS_ON_COOLDOWN, normalized);
+    set({ providersOnCooldown: normalized });
+  },
+  addProviderOnCooldown: (baseUrl, timestamp) => {
+    const normalized = normalizeBaseUrl5(baseUrl);
+    const current = get().providersOnCooldown;
+    if (!current.some((entry) => entry.baseUrl === normalized)) {
+      const updated = [...current, { baseUrl: normalized, timestamp }];
+      void driver.setItem(SDK_STORAGE_KEYS.PROVIDERS_ON_COOLDOWN, updated);
+      set({ providersOnCooldown: updated });
+    }
+  },
+  removeProviderFromCooldown: (baseUrl) => {
+    const normalized = normalizeBaseUrl5(baseUrl);
+    const current = get().providersOnCooldown;
+    const updated = current.filter((entry) => entry.baseUrl !== normalized);
+    void driver.setItem(SDK_STORAGE_KEYS.PROVIDERS_ON_COOLDOWN, updated);
+    set({ providersOnCooldown: updated });
+  },
+  clearProvidersOnCooldown: () => {
+    void driver.setItem(SDK_STORAGE_KEYS.PROVIDERS_ON_COOLDOWN, []);
+    set({ providersOnCooldown: [] });
   }
 }));
 var hydrateStoreFromDriver = async (store, driver) => {
@@ -3764,7 +3966,10 @@ var hydrateStoreFromDriver = async (store, driver) => {
     rawRoutstr21Models,
     rawLastRoutstr21ModelsUpdate,
     rawCachedReceiveTokens,
-    rawClientIds
+    rawClientIds,
+    rawFailedProviders,
+    rawLastFailed,
+    rawProvidersOnCooldown
   ] = await Promise.all([
     driver.getItem(
       SDK_STORAGE_KEYS.MODELS_FROM_ALL_PROVIDERS,
@@ -3795,7 +4000,10 @@ var hydrateStoreFromDriver = async (store, driver) => {
       null
     ),
     driver.getItem(SDK_STORAGE_KEYS.CACHED_RECEIVE_TOKENS, []),
-    driver.getItem(SDK_STORAGE_KEYS.CLIENT_IDS, [])
+    driver.getItem(SDK_STORAGE_KEYS.CLIENT_IDS, []),
+    driver.getItem(SDK_STORAGE_KEYS.FAILED_PROVIDERS, []),
+    driver.getItem(SDK_STORAGE_KEYS.LAST_FAILED, {}),
+    driver.getItem(SDK_STORAGE_KEYS.PROVIDERS_ON_COOLDOWN, [])
   ]);
   const modelsFromAllProviders = Object.fromEntries(
     Object.entries(rawModels).map(([baseUrl, models]) => [
@@ -3863,6 +4071,17 @@ var hydrateStoreFromDriver = async (store, driver) => {
     createdAt: entry.createdAt ?? Date.now(),
     lastUsed: entry.lastUsed ?? null
   }));
+  const failedProviders = rawFailedProviders.map((url) => normalizeBaseUrl5(url));
+  const lastFailed = Object.fromEntries(
+    Object.entries(rawLastFailed).map(([baseUrl, timestamp]) => [
+      normalizeBaseUrl5(baseUrl),
+      timestamp
+    ])
+  );
+  const providersOnCooldown = rawProvidersOnCooldown.map((entry) => ({
+    baseUrl: normalizeBaseUrl5(entry.baseUrl),
+    timestamp: entry.timestamp
+  }));
   store.setState({
     modelsFromAllProviders,
     lastUsedModel,
@@ -3878,7 +4097,10 @@ var hydrateStoreFromDriver = async (store, driver) => {
     routstr21Models,
     lastRoutstr21ModelsUpdate,
     cachedReceiveTokens,
-    clientIds
+    clientIds,
+    failedProviders,
+    lastFailed,
+    providersOnCooldown
   });
 };
 var createSdkStore = ({
@@ -4294,11 +4516,11 @@ var RoutstrClient = class {
       this.balanceManager
     );
     this.streamProcessor = new StreamProcessor();
-    this.providerManager = new ProviderManager(providerRegistry);
     this.alertLevel = alertLevel;
     this.mode = mode;
     this.usageTrackingDriver = options.usageTrackingDriver;
     this.sdkStore = options.sdkStore;
+    this.providerManager = options.providerManager ?? new ProviderManager(providerRegistry, this.sdkStore);
   }
   cashuSpender;
   balanceManager;
@@ -4828,6 +5050,10 @@ var RoutstrClient = class {
         const currentBalance = currentBalanceInfo.unit === "msat" ? currentBalanceInfo.amount / 1e3 : currentBalanceInfo.amount;
         const shortfall = Math.max(0, params.requiredSats - currentBalance);
         topupAmount = shortfall > 0 ? shortfall : params.requiredSats;
+        this._log(
+          "DEBUG",
+          `The shortfall is: ${shortfall}. requiredSats: ${params.requiredSats}. Current Balance: ${currentBalance} `
+        );
       } catch (e) {
         this._log(
           "WARN",
@@ -4957,6 +5183,20 @@ var RoutstrClient = class {
         tryNextProvider = true;
       }
     }
+    if (status === 401 && this.mode === "apikeys") {
+      this._log(
+        "DEBUG",
+        `[RoutstrClient] _handleErrorResponse: Checking balance for ${baseUrl}, key preview=${token}`
+      );
+      const latestBalanceInfo = await this.balanceManager.getTokenBalance(
+        token,
+        baseUrl
+      );
+      if (latestBalanceInfo.isInvalidApiKey) {
+        this.storageAdapter.removeApiKey(baseUrl);
+        tryNextProvider = true;
+      }
+    }
     if ((status === 401 || status === 403 || status === 413 || status === 400 || status === 500 || status === 502 || status === 503 || status === 504 || status === 521) && !tryNextProvider) {
       this._log(
         "DEBUG",
@@ -4967,13 +5207,13 @@ var RoutstrClient = class {
           "DEBUG",
           `[RoutstrClient] _handleErrorResponse: Attempting API key refund for ${baseUrl}, key preview=${token}`
         );
-        const initialBalance = await this.balanceManager.getTokenBalance(
+        const latestBalanceInfo = await this.balanceManager.getTokenBalance(
           token,
           baseUrl
         );
         this._log(
           "DEBUG",
-          `[RoutstrClient] _handleErrorResponse: Initial API key balance: ${initialBalance.amount}`
+          `[RoutstrClient] _handleErrorResponse: Initial API key balance: ${latestBalanceInfo.amount}`
         );
         const refundResult = await this.balanceManager.refundApiKey({
           mintUrl,
@@ -4985,7 +5225,7 @@ var RoutstrClient = class {
           "DEBUG",
           `[RoutstrClient] _handleErrorResponse: API key refund result: success=${refundResult.success}, message=${refundResult.message}`
         );
-        if (!refundResult.success && initialBalance.amount > 0) {
+        if (!refundResult.success && latestBalanceInfo.amount > 0) {
           throw new ProviderError(
             baseUrl,
             status,
@@ -5125,7 +5365,6 @@ var RoutstrClient = class {
       try {
         const xcashuResults = await this.cashuSpender.refundXcashuTokens(mintUrl);
         this._log("DEBUG", "Refund xcashu tokens results:", xcashuResults);
-        const results = await this.cashuSpender.refundProviders(mintUrl);
       } catch (error) {
         this._log("ERROR", "Failed to refund providers:", error);
       }
@@ -5466,7 +5705,8 @@ async function resolveRouteRequestContext(options) {
     debugLevel,
     mode = "apikeys",
     usageTrackingDriver,
-    sdkStore
+    sdkStore,
+    providerManager: providedProviderManager
   } = options;
   let modelManager;
   let providers;
@@ -5486,7 +5726,7 @@ async function resolveRouteRequestContext(options) {
     }
     await modelManager.fetchModels(providers, forceRefresh);
   }
-  const providerManager = new ProviderManager(providerRegistry);
+  const providerManager = providedProviderManager ?? new ProviderManager(providerRegistry, sdkStore);
   let baseUrl;
   let selectedModel;
   if (forcedProvider) {
@@ -5531,7 +5771,7 @@ async function resolveRouteRequestContext(options) {
     providerRegistry,
     "min",
     mode,
-    { usageTrackingDriver, sdkStore }
+    { usageTrackingDriver, sdkStore, providerManager }
   );
   if (debugLevel) {
     client.setDebugLevel(debugLevel);