@cetusprotocol/aggregator-sdk 0.4.0 → 0.4.1

package/README.md

@@ -41,9 +41,22 @@ npm install @cetusprotocol/aggregator-sdk
 
 ## 1. Init client with rpc and package config
 
-```typescript
-const client = new AggregatorClient()
-```
+1. Fast init:
+  ```typescript
+    const client = new AggregatorClient()
+  ```
+
+2. Full init:
+  ```typescript
+    const client = new AggregatorClient({
+      // endpoint, // If you do not have a exclusive aggregator api domain，just use cetus default aggregator endpoints.
+      signer: wallet,
+      client: suiClient,
+      env: Env.Mainnet,
+      pythUrls: ["YOUR_PYTH_URL", "ANOTHER_PYTH_URL"],
+    })
+  ```
+**Notes**: Some providers, such as HaedalHMM and Metastable, rely on Pyth oracle prices when build transactions. Currently, we have a default configuration that connects to Pyth's publicly available node. However, if you frequently build transactions, we recommend setting up a private Pyth node interface as an additional backup. This will ensure uninterrupted access to HaedalHMM and Metastable, even if the public node experiences occasional downtime.
 
 ## 2. Get best router swap result from aggregator service
 
@@ -53,7 +66,7 @@ const from = "0x2::sui::SUI"
 const target =
   "0x06864a6f921804860930db6ddbe2e16acdf8504495ea7481637a1c8b9a8fe54b::cetus::CETUS"
 
-const routerRes = await client.findRouters({
+const routers = await client.findRouters({
   from,
   target,
   amount,
@@ -68,8 +81,7 @@ const routerTx = new Transaction()
 
 if (routerRes != null) {
   await client.fastRouterSwap({
-    routers: routerRes.routes,
-    byAmountIn,
+    routers,
     txb: routerTx,
     slippage: 0.01,
     isMergeTragetCoin: true,
@@ -94,8 +106,7 @@ const byAmountIn = true;
 
 if (routerRes != null) {
   const targetCoin = await client.routerSwap({
-    routers: routerRes.routes,
-    byAmountIn,
+    routers,
     txb: routerTx,
     inputCoin,
     slippage: 0.01,

package/dist/index.d.mts

@@ -4,6 +4,7 @@ import { TransactionObjectArgument, Transaction } from '@mysten/sui/transactions
 import { Signer } from '@mysten/sui/cryptography';
 import BN from 'bn.js';
 import Decimal from 'decimal.js';
+import { SuiPriceServiceConnection } from '@pythnetwork/pyth-sui-js';
 
 /**
  * Represents a SUI address, which is a string.
@@ -128,16 +129,26 @@ interface SwapInPoolsResult {
     isExceed: boolean;
     routeData?: RouterData;
 }
+type AggregatorClientParams = {
+    endpoint?: string;
+    signer?: string;
+    client?: SuiClient;
+    env?: Env;
+    pythUrls?: string[];
+    apiKey?: string;
+};
 declare class AggregatorClient {
     endpoint: string;
     signer: string;
     client: SuiClient;
     env: Env;
+    apiKey: string;
     private allCoins;
     private pythConnections;
     private pythClient;
     private static readonly CONFIG;
-    constructor(endpoint?: string, signer?: string, client?: SuiClient, env?: Env);
+    constructor(params: AggregatorClientParams);
+    newPythClients(pythUrls: string[]): SuiPriceServiceConnection[];
     getCoins(coinType: string, refresh?: boolean): Promise<CoinAsset[]>;
     findRouters(params: FindRouterParams): Promise<RouterData | null>;
     executeFlexibleInputSwap(txb: Transaction, inputCoin: TransactionObjectArgument, routers: Router[], amountOutLimit: BN, pythPriceIDs: Map<string, string>, partner?: string, deepbookv3DeepFee?: TransactionObjectArgument, packages?: Map<string, string>): Promise<TransactionObjectArgument>;
@@ -296,7 +307,7 @@ type AggregatorResponse = {
     msg: string;
     data: RouterData;
 };
-declare function getRouterResult(endpoint: string, params: FindRouterParams): Promise<RouterData | null>;
+declare function getRouterResult(endpoint: string, apiKey: string, params: FindRouterParams): Promise<RouterData | null>;
 type DeepbookV3Config = {
     id: string;
     is_alternative_payment: boolean;
@@ -321,4 +332,4 @@ declare enum Env {
     Testnet = 1
 }
 
-export { AFSUI, AFTERMATH, AGGREGATOR_V2, AGGREGATOR_V2_EXTEND, ALPHAFI, AggregatorClient, type AggregatorResponse, BLUEFIN, BLUEMOVE, type BuildCoinResult, type BuildFastRouterSwapParams, type BuildFastRouterSwapParamsV2, type BuildRouterSwapParams, type BuildRouterSwapParamsV2, CETUS, CLOCK_ADDRESS, DEEPBOOKV2, DEEPBOOKV3, DEFAULT_ENDPOINT, type DeepbookV3Config, type DeepbookV3ConfigResponse, type Dex, Env, type ExtendedDetails, FLOWXV2, FLOWXV3, type FindRouterParams, HAEDAL, HAEDALPMM, KRIYA, KRIYAV3, METASTABLE, OBRIC, ONE, type Path, type PreSwapLpChangeParams, type Router, type RouterData, type RouterError, SCALLOP, SPRINGSUI, STEAMM, SUILEND, type SwapInPoolsParams, type SwapInPoolsResult, TEN_POW_NINE, TURBOS, TWO, U128, U64_MAX, U64_MAX_BN, VOLO, ZERO, buildInputCoin, checkInvalidSuiAddress, compareCoins, completionCoin, composeType, createTarget, dealWithFastRouterSwapParamsForMsafe, extractAddressFromType, extractStructTagFromType, findPythPriceIDs, fixSuiObjectId, getAggregatorV2ExtendPublishedAt, getAggregatorV2PublishedAt, getDeepbookV3Config, getRouterResult, isSortedSymbols, mintZeroCoin, normalizeCoinType, parseRouterResponse, patchFixSuiObjectId, printTransaction, processEndpoint, restituteMsafeFastRouterSwapParams };
+export { AFSUI, AFTERMATH, AGGREGATOR_V2, AGGREGATOR_V2_EXTEND, ALPHAFI, AggregatorClient, type AggregatorClientParams, type AggregatorResponse, BLUEFIN, BLUEMOVE, type BuildCoinResult, type BuildFastRouterSwapParams, type BuildFastRouterSwapParamsV2, type BuildRouterSwapParams, type BuildRouterSwapParamsV2, CETUS, CLOCK_ADDRESS, DEEPBOOKV2, DEEPBOOKV3, DEFAULT_ENDPOINT, type DeepbookV3Config, type DeepbookV3ConfigResponse, type Dex, Env, type ExtendedDetails, FLOWXV2, FLOWXV3, type FindRouterParams, HAEDAL, HAEDALPMM, KRIYA, KRIYAV3, METASTABLE, OBRIC, ONE, type Path, type PreSwapLpChangeParams, type Router, type RouterData, type RouterError, SCALLOP, SPRINGSUI, STEAMM, SUILEND, type SwapInPoolsParams, type SwapInPoolsResult, TEN_POW_NINE, TURBOS, TWO, U128, U64_MAX, U64_MAX_BN, VOLO, ZERO, buildInputCoin, checkInvalidSuiAddress, compareCoins, completionCoin, composeType, createTarget, dealWithFastRouterSwapParamsForMsafe, extractAddressFromType, extractStructTagFromType, findPythPriceIDs, fixSuiObjectId, getAggregatorV2ExtendPublishedAt, getAggregatorV2PublishedAt, getDeepbookV3Config, getRouterResult, isSortedSymbols, mintZeroCoin, normalizeCoinType, parseRouterResponse, patchFixSuiObjectId, printTransaction, processEndpoint, restituteMsafeFastRouterSwapParams };

package/dist/index.d.ts

@@ -4,6 +4,7 @@ import { TransactionObjectArgument, Transaction } from '@mysten/sui/transactions
 import { Signer } from '@mysten/sui/cryptography';
 import BN from 'bn.js';
 import Decimal from 'decimal.js';
+import { SuiPriceServiceConnection } from '@pythnetwork/pyth-sui-js';
 
 /**
  * Represents a SUI address, which is a string.
@@ -128,16 +129,26 @@ interface SwapInPoolsResult {
     isExceed: boolean;
     routeData?: RouterData;
 }
+type AggregatorClientParams = {
+    endpoint?: string;
+    signer?: string;
+    client?: SuiClient;
+    env?: Env;
+    pythUrls?: string[];
+    apiKey?: string;
+};
 declare class AggregatorClient {
     endpoint: string;
     signer: string;
     client: SuiClient;
     env: Env;
+    apiKey: string;
     private allCoins;
     private pythConnections;
     private pythClient;
     private static readonly CONFIG;
-    constructor(endpoint?: string, signer?: string, client?: SuiClient, env?: Env);
+    constructor(params: AggregatorClientParams);
+    newPythClients(pythUrls: string[]): SuiPriceServiceConnection[];
     getCoins(coinType: string, refresh?: boolean): Promise<CoinAsset[]>;
     findRouters(params: FindRouterParams): Promise<RouterData | null>;
     executeFlexibleInputSwap(txb: Transaction, inputCoin: TransactionObjectArgument, routers: Router[], amountOutLimit: BN, pythPriceIDs: Map<string, string>, partner?: string, deepbookv3DeepFee?: TransactionObjectArgument, packages?: Map<string, string>): Promise<TransactionObjectArgument>;
@@ -296,7 +307,7 @@ type AggregatorResponse = {
     msg: string;
     data: RouterData;
 };
-declare function getRouterResult(endpoint: string, params: FindRouterParams): Promise<RouterData | null>;
+declare function getRouterResult(endpoint: string, apiKey: string, params: FindRouterParams): Promise<RouterData | null>;
 type DeepbookV3Config = {
     id: string;
     is_alternative_payment: boolean;
@@ -321,4 +332,4 @@ declare enum Env {
     Testnet = 1
 }
 
-export { AFSUI, AFTERMATH, AGGREGATOR_V2, AGGREGATOR_V2_EXTEND, ALPHAFI, AggregatorClient, type AggregatorResponse, BLUEFIN, BLUEMOVE, type BuildCoinResult, type BuildFastRouterSwapParams, type BuildFastRouterSwapParamsV2, type BuildRouterSwapParams, type BuildRouterSwapParamsV2, CETUS, CLOCK_ADDRESS, DEEPBOOKV2, DEEPBOOKV3, DEFAULT_ENDPOINT, type DeepbookV3Config, type DeepbookV3ConfigResponse, type Dex, Env, type ExtendedDetails, FLOWXV2, FLOWXV3, type FindRouterParams, HAEDAL, HAEDALPMM, KRIYA, KRIYAV3, METASTABLE, OBRIC, ONE, type Path, type PreSwapLpChangeParams, type Router, type RouterData, type RouterError, SCALLOP, SPRINGSUI, STEAMM, SUILEND, type SwapInPoolsParams, type SwapInPoolsResult, TEN_POW_NINE, TURBOS, TWO, U128, U64_MAX, U64_MAX_BN, VOLO, ZERO, buildInputCoin, checkInvalidSuiAddress, compareCoins, completionCoin, composeType, createTarget, dealWithFastRouterSwapParamsForMsafe, extractAddressFromType, extractStructTagFromType, findPythPriceIDs, fixSuiObjectId, getAggregatorV2ExtendPublishedAt, getAggregatorV2PublishedAt, getDeepbookV3Config, getRouterResult, isSortedSymbols, mintZeroCoin, normalizeCoinType, parseRouterResponse, patchFixSuiObjectId, printTransaction, processEndpoint, restituteMsafeFastRouterSwapParams };
+export { AFSUI, AFTERMATH, AGGREGATOR_V2, AGGREGATOR_V2_EXTEND, ALPHAFI, AggregatorClient, type AggregatorClientParams, type AggregatorResponse, BLUEFIN, BLUEMOVE, type BuildCoinResult, type BuildFastRouterSwapParams, type BuildFastRouterSwapParamsV2, type BuildRouterSwapParams, type BuildRouterSwapParamsV2, CETUS, CLOCK_ADDRESS, DEEPBOOKV2, DEEPBOOKV3, DEFAULT_ENDPOINT, type DeepbookV3Config, type DeepbookV3ConfigResponse, type Dex, Env, type ExtendedDetails, FLOWXV2, FLOWXV3, type FindRouterParams, HAEDAL, HAEDALPMM, KRIYA, KRIYAV3, METASTABLE, OBRIC, ONE, type Path, type PreSwapLpChangeParams, type Router, type RouterData, type RouterError, SCALLOP, SPRINGSUI, STEAMM, SUILEND, type SwapInPoolsParams, type SwapInPoolsResult, TEN_POW_NINE, TURBOS, TWO, U128, U64_MAX, U64_MAX_BN, VOLO, ZERO, buildInputCoin, checkInvalidSuiAddress, compareCoins, completionCoin, composeType, createTarget, dealWithFastRouterSwapParamsForMsafe, extractAddressFromType, extractStructTagFromType, findPythPriceIDs, fixSuiObjectId, getAggregatorV2ExtendPublishedAt, getAggregatorV2PublishedAt, getDeepbookV3Config, getRouterResult, isSortedSymbols, mintZeroCoin, normalizeCoinType, parseRouterResponse, patchFixSuiObjectId, printTransaction, processEndpoint, restituteMsafeFastRouterSwapParams };

package/dist/index.js

@@ -6514,19 +6514,28 @@ function isBuilderFastRouterSwapParams(params) {
   return Array.isArray(params.routers);
 }
 var _AggregatorClient = class _AggregatorClient {
-  constructor(endpoint, signer, client$1, env) {
-    this.endpoint = endpoint ? processEndpoint(endpoint) : DEFAULT_ENDPOINT;
-    this.client = client$1 || new client.SuiClient({ url: client.getFullnodeUrl("mainnet") });
-    this.signer = signer || "";
-    this.env = env || 0 /* Mainnet */;
+  constructor(params) {
+    var _a;
+    this.endpoint = params.endpoint ? processEndpoint(params.endpoint) : DEFAULT_ENDPOINT;
+    this.client = params.client || new client.SuiClient({ url: client.getFullnodeUrl("mainnet") });
+    this.signer = params.signer || "";
+    this.env = params.env || 0 /* Mainnet */;
     this.allCoins = /* @__PURE__ */ new Map();
     const config2 = _AggregatorClient.CONFIG[this.env];
-    this.pythConnections = config2.connections;
+    this.pythConnections = this.newPythClients((_a = params.pythUrls) != null ? _a : []);
     this.pythClient = new pythSuiJs.SuiPythClient(
       this.client,
       config2.pythStateId,
       config2.wormholeStateId
     );
+    this.apiKey = params.apiKey || "";
+  }
+  newPythClients(pythUrls) {
+    if (!pythUrls.includes("https://hermes.pyth.network")) {
+      pythUrls.push("https://hermes.pyth.network");
+    }
+    const connections = pythUrls.map((url) => new pythSuiJs.SuiPriceServiceConnection(url, { timeout: 3e3 }));
+    return connections;
   }
   getCoins(coinType, refresh = true) {
     return __async(this, null, function* () {
@@ -6572,7 +6581,7 @@ var _AggregatorClient = class _AggregatorClient {
   }
   findRouters(params) {
     return __async(this, null, function* () {
-      return getRouterResult(this.endpoint, params);
+      return getRouterResult(this.endpoint, this.apiKey, params);
     });
   }
   executeFlexibleInputSwap(txb, inputCoin, routers, amountOutLimit, pythPriceIDs, partner, deepbookv3DeepFee, packages) {
@@ -7104,7 +7113,7 @@ var _AggregatorClient = class _AggregatorClient {
       }
       if (priceUpdateData == null) {
         throw new Error(
-          `No pyth price seeds update data found: ${lastError == null ? void 0 : lastError.message}`
+          `All Pyth price nodes are unavailable. Cannot fetch price data. Please switch to or add new available Pyth nodes. Detailed error: ${lastError == null ? void 0 : lastError.message}`
         );
       }
       let priceInfoObjectIds = [];
@@ -7115,7 +7124,7 @@ var _AggregatorClient = class _AggregatorClient {
           priceIDs
         );
       } catch (e) {
-        throw new Error(`Failed to update price feeds: ${e}`);
+        throw new Error(`All Pyth price nodes are unavailable. Cannot fetch price data. Please switch to or add new available Pyth nodes in the pythUrls parameter when initializing AggregatorClient, for example: new AggregatorClient({ pythUrls: ["https://your-pyth-node-url"] }). Detailed error: ${e}`);
       }
       let priceInfoObjectIdsMap = /* @__PURE__ */ new Map();
       for (let i = 0; i < priceIDs.length; i++) {
@@ -7127,24 +7136,10 @@ var _AggregatorClient = class _AggregatorClient {
 };
 _AggregatorClient.CONFIG = {
   [1 /* Testnet */]: {
-    connections: [
-      new pythSuiJs.SuiPriceServiceConnection("https://hermes-beta.pyth.network")
-    ],
     wormholeStateId: "0x31358d198147da50db32eda2562951d53973a0c0ad5ed738e9b17d88b213d790",
     pythStateId: "0x243759059f4c3111179da5878c12f68d612c21a8d54d85edc86164bb18be1c7c"
   },
   [0 /* Mainnet */]: {
-    connections: [
-      new pythSuiJs.SuiPriceServiceConnection(
-        "https://cetus-pythnet-a648.mainnet.pythnet.rpcpool.com/219cf7a8-6d75-432d-a648-d487a6dd5dc3/hermes",
-        {
-          timeout: 3e3
-        }
-      ),
-      new pythSuiJs.SuiPriceServiceConnection("https://hermes.pyth.network", {
-        timeout: 3e3
-      })
-    ],
     wormholeStateId: "0xaeab97f96cf9877fee2883315d459552b2b921edc16d7ceac6eab944dd88919c",
     pythStateId: "0x1f9310238ee9298fb703c3419030b35b22bb1cc37113e3bb5007c99aec79e5b8"
   }
@@ -7324,14 +7319,14 @@ function processEndpoint(endpoint) {
 }
 
 // src/api.ts
-var SDK_VERSION = 1000400;
-function getRouterResult(endpoint, params) {
+var SDK_VERSION = 1000401;
+function getRouterResult(endpoint, apiKey, params) {
   return __async(this, null, function* () {
     let response;
     if (params.liquidityChanges && params.liquidityChanges.length > 0) {
       response = yield postRouterWithLiquidityChanges(endpoint, params);
     } else {
-      response = yield getRouter(endpoint, params);
+      response = yield getRouter(endpoint, apiKey, params);
     }
     if (!response) {
       return null;
@@ -7389,7 +7384,7 @@ function getRouterResult(endpoint, params) {
     };
   });
 }
-function getRouter(endpoint, params) {
+function getRouter(endpoint, apiKey, params) {
   return __async(this, null, function* () {
     try {
       const {
@@ -7423,6 +7418,9 @@ function getRouter(endpoint, params) {
           url += `&providers=${providers.join(",")}`;
         }
       }
+      if (apiKey.length > 0) {
+        url += `&apiKey=${apiKey}`;
+      }
       url += `&v=${SDK_VERSION}`;
       const response = yield fetch(url);
       return response;

package/dist/index.mjs

@@ -1,7 +1,7 @@
 import { SuiClient, getFullnodeUrl } from '@mysten/sui/client';
 import { normalizeSuiObjectId, SUI_CLOCK_OBJECT_ID, SUI_FRAMEWORK_ADDRESS } from '@mysten/sui/utils';
 import { Transaction } from '@mysten/sui/transactions';
-import { SuiPriceServiceConnection, SuiPythClient } from '@pythnetwork/pyth-sui-js';
+import { SuiPythClient, SuiPriceServiceConnection } from '@pythnetwork/pyth-sui-js';
 
 var __create = Object.create;
 var __defProp = Object.defineProperty;
@@ -6512,19 +6512,28 @@ function isBuilderFastRouterSwapParams(params) {
   return Array.isArray(params.routers);
 }
 var _AggregatorClient = class _AggregatorClient {
-  constructor(endpoint, signer, client, env) {
-    this.endpoint = endpoint ? processEndpoint(endpoint) : DEFAULT_ENDPOINT;
-    this.client = client || new SuiClient({ url: getFullnodeUrl("mainnet") });
-    this.signer = signer || "";
-    this.env = env || 0 /* Mainnet */;
+  constructor(params) {
+    var _a;
+    this.endpoint = params.endpoint ? processEndpoint(params.endpoint) : DEFAULT_ENDPOINT;
+    this.client = params.client || new SuiClient({ url: getFullnodeUrl("mainnet") });
+    this.signer = params.signer || "";
+    this.env = params.env || 0 /* Mainnet */;
     this.allCoins = /* @__PURE__ */ new Map();
     const config2 = _AggregatorClient.CONFIG[this.env];
-    this.pythConnections = config2.connections;
+    this.pythConnections = this.newPythClients((_a = params.pythUrls) != null ? _a : []);
     this.pythClient = new SuiPythClient(
       this.client,
       config2.pythStateId,
       config2.wormholeStateId
     );
+    this.apiKey = params.apiKey || "";
+  }
+  newPythClients(pythUrls) {
+    if (!pythUrls.includes("https://hermes.pyth.network")) {
+      pythUrls.push("https://hermes.pyth.network");
+    }
+    const connections = pythUrls.map((url) => new SuiPriceServiceConnection(url, { timeout: 3e3 }));
+    return connections;
   }
   getCoins(coinType, refresh = true) {
     return __async(this, null, function* () {
@@ -6570,7 +6579,7 @@ var _AggregatorClient = class _AggregatorClient {
   }
   findRouters(params) {
     return __async(this, null, function* () {
-      return getRouterResult(this.endpoint, params);
+      return getRouterResult(this.endpoint, this.apiKey, params);
     });
   }
   executeFlexibleInputSwap(txb, inputCoin, routers, amountOutLimit, pythPriceIDs, partner, deepbookv3DeepFee, packages) {
@@ -7102,7 +7111,7 @@ var _AggregatorClient = class _AggregatorClient {
       }
       if (priceUpdateData == null) {
         throw new Error(
-          `No pyth price seeds update data found: ${lastError == null ? void 0 : lastError.message}`
+          `All Pyth price nodes are unavailable. Cannot fetch price data. Please switch to or add new available Pyth nodes. Detailed error: ${lastError == null ? void 0 : lastError.message}`
         );
       }
       let priceInfoObjectIds = [];
@@ -7113,7 +7122,7 @@ var _AggregatorClient = class _AggregatorClient {
           priceIDs
         );
       } catch (e) {
-        throw new Error(`Failed to update price feeds: ${e}`);
+        throw new Error(`All Pyth price nodes are unavailable. Cannot fetch price data. Please switch to or add new available Pyth nodes in the pythUrls parameter when initializing AggregatorClient, for example: new AggregatorClient({ pythUrls: ["https://your-pyth-node-url"] }). Detailed error: ${e}`);
       }
       let priceInfoObjectIdsMap = /* @__PURE__ */ new Map();
       for (let i = 0; i < priceIDs.length; i++) {
@@ -7125,24 +7134,10 @@ var _AggregatorClient = class _AggregatorClient {
 };
 _AggregatorClient.CONFIG = {
   [1 /* Testnet */]: {
-    connections: [
-      new SuiPriceServiceConnection("https://hermes-beta.pyth.network")
-    ],
     wormholeStateId: "0x31358d198147da50db32eda2562951d53973a0c0ad5ed738e9b17d88b213d790",
     pythStateId: "0x243759059f4c3111179da5878c12f68d612c21a8d54d85edc86164bb18be1c7c"
   },
   [0 /* Mainnet */]: {
-    connections: [
-      new SuiPriceServiceConnection(
-        "https://cetus-pythnet-a648.mainnet.pythnet.rpcpool.com/219cf7a8-6d75-432d-a648-d487a6dd5dc3/hermes",
-        {
-          timeout: 3e3
-        }
-      ),
-      new SuiPriceServiceConnection("https://hermes.pyth.network", {
-        timeout: 3e3
-      })
-    ],
     wormholeStateId: "0xaeab97f96cf9877fee2883315d459552b2b921edc16d7ceac6eab944dd88919c",
     pythStateId: "0x1f9310238ee9298fb703c3419030b35b22bb1cc37113e3bb5007c99aec79e5b8"
   }
@@ -7322,14 +7317,14 @@ function processEndpoint(endpoint) {
 }
 
 // src/api.ts
-var SDK_VERSION = 1000400;
-function getRouterResult(endpoint, params) {
+var SDK_VERSION = 1000401;
+function getRouterResult(endpoint, apiKey, params) {
   return __async(this, null, function* () {
     let response;
     if (params.liquidityChanges && params.liquidityChanges.length > 0) {
       response = yield postRouterWithLiquidityChanges(endpoint, params);
     } else {
-      response = yield getRouter(endpoint, params);
+      response = yield getRouter(endpoint, apiKey, params);
     }
     if (!response) {
       return null;
@@ -7387,7 +7382,7 @@ function getRouterResult(endpoint, params) {
     };
   });
 }
-function getRouter(endpoint, params) {
+function getRouter(endpoint, apiKey, params) {
   return __async(this, null, function* () {
     try {
       const {
@@ -7421,6 +7416,9 @@ function getRouter(endpoint, params) {
           url += `&providers=${providers.join(",")}`;
         }
       }
+      if (apiKey.length > 0) {
+        url += `&apiKey=${apiKey}`;
+      }
       url += `&v=${SDK_VERSION}`;
       const response = yield fetch(url);
       return response;

package/dist/src/api.d.ts

@@ -83,7 +83,7 @@ export type AggregatorResponse = {
     msg: string;
     data: RouterData;
 };
-export declare function getRouterResult(endpoint: string, params: FindRouterParams): Promise<RouterData | null>;
+export declare function getRouterResult(endpoint: string, apiKey: string, params: FindRouterParams): Promise<RouterData | null>;
 export type DeepbookV3Config = {
     id: string;
     is_alternative_payment: boolean;

package/dist/src/client.d.ts

@@ -4,6 +4,7 @@ import { Signer } from "@mysten/sui/cryptography";
 import BN from "bn.js";
 import { Dex, Env, FindRouterParams, Router, RouterData, DeepbookV3Config } from ".";
 import { CoinAsset } from "./types/sui";
+import { SuiPriceServiceConnection } from "@pythnetwork/pyth-sui-js";
 export declare const CETUS = "CETUS";
 export declare const DEEPBOOKV2 = "DEEPBOOK";
 export declare const KRIYA = "KRIYA";
@@ -72,16 +73,26 @@ export interface SwapInPoolsResult {
     isExceed: boolean;
     routeData?: RouterData;
 }
+export type AggregatorClientParams = {
+    endpoint?: string;
+    signer?: string;
+    client?: SuiClient;
+    env?: Env;
+    pythUrls?: string[];
+    apiKey?: string;
+};
 export declare class AggregatorClient {
     endpoint: string;
     signer: string;
     client: SuiClient;
     env: Env;
+    apiKey: string;
     private allCoins;
     private pythConnections;
     private pythClient;
     private static readonly CONFIG;
-    constructor(endpoint?: string, signer?: string, client?: SuiClient, env?: Env);
+    constructor(params: AggregatorClientParams);
+    newPythClients(pythUrls: string[]): SuiPriceServiceConnection[];
     getCoins(coinType: string, refresh?: boolean): Promise<CoinAsset[]>;
     findRouters(params: FindRouterParams): Promise<RouterData | null>;
     executeFlexibleInputSwap(txb: Transaction, inputCoin: TransactionObjectArgument, routers: Router[], amountOutLimit: BN, pythPriceIDs: Map<string, string>, partner?: string, deepbookv3DeepFee?: TransactionObjectArgument, packages?: Map<string, string>): Promise<TransactionObjectArgument>;

package/dist/tests/router/alphafi.test.d.ts

@@ -0,0 +1,2 @@
+import { Ed25519Keypair } from "@mysten/sui/keypairs/ed25519";
+export declare function buildTestAccount(): Ed25519Keypair;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cetusprotocol/aggregator-sdk",
-  "version": "0.4.0",
+  "version": "0.4.1",
   "sideEffects": false,
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

package/src/api.ts

@@ -8,7 +8,7 @@ import {
 } from "./errors"
 import { parseRouterResponse } from "./client"
 
-const SDK_VERSION = 1000400
+const SDK_VERSION = 1000401
 
 export interface FindRouterParams {
   from: string
@@ -110,13 +110,14 @@ export type AggregatorResponse = {
 
 export async function getRouterResult(
   endpoint: string,
+  apiKey: string,
   params: FindRouterParams
 ): Promise<RouterData | null> {
   let response
   if (params.liquidityChanges && params.liquidityChanges.length > 0) {
     response = await postRouterWithLiquidityChanges(endpoint, params)
   } else {
-    response = await getRouter(endpoint, params)
+    response = await getRouter(endpoint, apiKey, params)
   }
 
   if (!response) {
@@ -179,7 +180,7 @@ export async function getRouterResult(
   }
 }
 
-async function getRouter(endpoint: string, params: FindRouterParams) {
+async function getRouter(endpoint: string, apiKey: string, params: FindRouterParams) {
   try {
     const {
       from,
@@ -219,6 +220,10 @@ async function getRouter(endpoint: string, params: FindRouterParams) {
       }
     }
 
+    if (apiKey.length > 0) {
+      url += `&apiKey=${apiKey}`
+    }
+
     // set newest sdk version
     url += `&v=${SDK_VERSION}`
 

package/src/client.ts

@@ -126,7 +126,6 @@ export interface SwapInPoolsParams {
 }
 
 interface PythConfig {
-  connections: SuiPriceServiceConnection[]
   wormholeStateId: string
   pythStateId: string
 }
@@ -148,11 +147,21 @@ function isBuilderFastRouterSwapParams(
   return Array.isArray((params as BuildFastRouterSwapParams).routers)
 }
 
+export type AggregatorClientParams = {
+  endpoint?: string
+  signer?: string
+  client?: SuiClient
+  env?: Env
+  pythUrls?: string[]
+  apiKey?: string
+}
+
 export class AggregatorClient {
   public endpoint: string
   public signer: string
   public client: SuiClient
   public env: Env
+  public apiKey: string
   private allCoins: Map<string, CoinAsset[]>
 
   private pythConnections: SuiPriceServiceConnection[]
@@ -160,26 +169,12 @@ export class AggregatorClient {
 
   private static readonly CONFIG: Record<Env, PythConfig> = {
     [Env.Testnet]: {
-      connections: [
-        new SuiPriceServiceConnection("https://hermes-beta.pyth.network"),
-      ],
       wormholeStateId:
         "0x31358d198147da50db32eda2562951d53973a0c0ad5ed738e9b17d88b213d790",
       pythStateId:
         "0x243759059f4c3111179da5878c12f68d612c21a8d54d85edc86164bb18be1c7c",
     },
     [Env.Mainnet]: {
-      connections: [
-        new SuiPriceServiceConnection(
-          "https://cetus-pythnet-a648.mainnet.pythnet.rpcpool.com/219cf7a8-6d75-432d-a648-d487a6dd5dc3/hermes",
-          {
-            timeout: 3000,
-          }
-        ),
-        new SuiPriceServiceConnection("https://hermes.pyth.network", {
-          timeout: 3000,
-        }),
-      ],
       wormholeStateId:
         "0xaeab97f96cf9877fee2883315d459552b2b921edc16d7ceac6eab944dd88919c",
       pythStateId:
@@ -187,25 +182,30 @@ export class AggregatorClient {
     },
   }
 
-  constructor(
-    endpoint?: string,
-    signer?: string,
-    client?: SuiClient,
-    env?: Env
-  ) {
-    this.endpoint = endpoint ? processEndpoint(endpoint) : DEFAULT_ENDPOINT
-    this.client = client || new SuiClient({ url: getFullnodeUrl("mainnet") })
-    this.signer = signer || ""
-    this.env = env || Env.Mainnet
+  constructor(params: AggregatorClientParams) {
+    this.endpoint = params.endpoint ? processEndpoint(params.endpoint) : DEFAULT_ENDPOINT
+    this.client = params.client || new SuiClient({ url: getFullnodeUrl("mainnet") })
+    this.signer = params.signer || ""
+    this.env = params.env || Env.Mainnet
     this.allCoins = new Map<string, CoinAsset[]>()
 
     const config = AggregatorClient.CONFIG[this.env]
-    this.pythConnections = config.connections
+    this.pythConnections = this.newPythClients(params.pythUrls ?? [])
     this.pythClient = new SuiPythClient(
       this.client,
       config.pythStateId,
       config.wormholeStateId
     )
+    this.apiKey = params.apiKey || ""
+  }
+
+  newPythClients(pythUrls: string[]) {
+    if (!pythUrls.includes("https://hermes.pyth.network")) {
+      pythUrls.push("https://hermes.pyth.network")
+    }
+
+    const connections = pythUrls.map(url => new SuiPriceServiceConnection(url, { timeout: 3000 }))
+    return connections
   }
 
   async getCoins(
@@ -257,7 +257,7 @@ export class AggregatorClient {
   }
 
   async findRouters(params: FindRouterParams): Promise<RouterData | null> {
-    return getRouterResult(this.endpoint, params)
+    return getRouterResult(this.endpoint, this.apiKey, params)
   }
 
   async executeFlexibleInputSwap(
@@ -916,7 +916,7 @@ export class AggregatorClient {
 
     if (priceUpdateData == null) {
       throw new Error(
-        `No pyth price seeds update data found: ${lastError?.message}`
+        `All Pyth price nodes are unavailable. Cannot fetch price data. Please switch to or add new available Pyth nodes. Detailed error: ${lastError?.message}`
       )
     }
 
@@ -928,7 +928,7 @@ export class AggregatorClient {
         priceIDs
       )
     } catch (e) {
-      throw new Error(`Failed to update price feeds: ${e}`)
+      throw new Error(`All Pyth price nodes are unavailable. Cannot fetch price data. Please switch to or add new available Pyth nodes in the pythUrls parameter when initializing AggregatorClient, for example: new AggregatorClient({ pythUrls: ["https://your-pyth-node-url"] }). Detailed error: ${e}`)
     }
 
     let priceInfoObjectIdsMap = new Map<string, string>()

package/tests/router/alphafi.test.ts

@@ -0,0 +1,132 @@
+import { describe, test } from "@jest/globals"
+import dotenv from "dotenv"
+import { AggregatorClient } from "~/client"
+import { Ed25519Keypair } from "@mysten/sui/keypairs/ed25519"
+import { printTransaction } from "~/utils/transaction"
+import BN from "bn.js"
+import { fromB64 } from "@mysten/sui/utils"
+import { SuiClient } from "@mysten/sui/client"
+import { Env } from "~/index"
+import { Transaction } from "@mysten/sui/transactions"
+
+dotenv.config()
+
+export function buildTestAccount(): Ed25519Keypair {
+  const mnemonics = process.env.SUI_WALLET_MNEMONICS || ""
+  const testAccountObject = Ed25519Keypair.deriveKeypair(mnemonics)
+  return testAccountObject
+}
+
+describe("Test scallop provider", () => {
+  let client: AggregatorClient
+  let keypair: Ed25519Keypair
+
+  const T_SUI = "0x0000000000000000000000000000000000000000000000000000000000000002::sui::SUI"
+  const LSDT_SUI  = "0xd1b72982e40348d069bb1ff701e634c117bb5f741f44dff91e472d3b01461e55::stsui::STSUI"
+  const T_USDC = "0xdba34672e30cb065b1f93e3ab55318768fd6fef66c15942c9f7cb846e2f900e7::usdc::USDC"
+
+  beforeAll(() => {
+    const fullNodeURL = process.env.SUI_RPC!
+    const aggregatorURL = process.env.CETUS_AGGREGATOR!
+    const secret = process.env.SUI_WALLET_SECRET!
+
+    if (secret) {
+      keypair = Ed25519Keypair.fromSecretKey(fromB64(secret).slice(1, 33))
+    } else {
+      keypair = buildTestAccount()
+    }
+
+    const wallet =
+      "0x4dde66fc52ec16d5e6c0fbd0968580cdf0d962cbb970591ec1e47617b9265617"
+
+    console.log("wallet: ", wallet)
+
+    const endpoint = aggregatorURL
+
+    const suiClient = new SuiClient({
+      url: fullNodeURL,
+    })
+
+    client = new AggregatorClient({
+      endpoint,
+      signer: wallet,
+      client: suiClient,
+      env: Env.Mainnet,
+      pythUrls: ["https://cetus-pythnet-a648.mainnet.pythnet.rpcpool.com/219cf7a8-6d75-432d-a648-d487a6dd5dc3/hermes"],
+      apiKey: "8MJDUzLDPJxCgbc7I0bHXSg994mVfh8NRMqV6hcQ",
+    })
+  })
+
+  test("Find Routers", async () => {
+    const amounts = ["1000", "1000000", "100000000", "5000000000", "1000000000000000000000000000"]
+    
+    while (true) {
+      const res = await client.findRouters({
+        from: LSDT_SUI,
+        target: T_USDC,
+        amount: new BN("1000000000000"),
+        byAmountIn: true,
+        depth: 3,
+        splitCount: 1,
+        providers: ["ALPHAFI"],
+      })
+
+      if (res != null) {
+        console.log(JSON.stringify(res, null, 2))
+      }
+      console.log("amount in", res?.amountIn.toString())
+      console.log("amount out", res?.amountOut.toString())
+    }
+  }, 6000000)
+
+  test("Build Router TX", async () => {
+    const amount = "100000"
+
+    const res = await client.findRouters({
+      from: LSDT_SUI,
+      target: T_SUI,
+      amount: new BN(amount),
+      byAmountIn: true,
+      depth: 3,
+      splitCount: 2,
+      providers: ["ALPHAFI", "CETUS"],
+    })
+
+    console.log("amount in", res?.amountIn.toString())
+    console.log("amount out", res?.amountOut.toString())
+
+    const txb = new Transaction()
+
+    if (res != null) {
+      console.log(JSON.stringify(res, null, 2))
+      await client.fastRouterSwap({
+        routers: res,
+        txb,
+        slippage: 0.01,
+        refreshAllCoins: true,
+        payDeepFeeAmount: 0,
+      })
+
+      txb.setSender(client.signer)
+      const buildTxb = await txb.build({ client: client.client })
+      // const buildTxb = await txb.getData()
+      
+      console.log("buildTxb", buildTxb)
+
+      printTransaction(txb)
+
+      let result = await client.devInspectTransactionBlock(txb)
+      console.log("🚀 ~ file: router.test.ts:180 ~ test ~ result:", result)
+      for (const event of result.events) {
+        console.log("event", JSON.stringify(event, null, 2))
+      }
+
+      if (result.effects.status.status === "success") {
+        const result = await client.signAndExecuteTransaction(txb, keypair)
+        console.log("result", result)
+      } else {
+        console.log("result", result)
+      }
+    }
+  }, 600000)
+})

package/tests/router/metastable.test.ts

@@ -57,7 +57,13 @@ describe("Test metastable provider", () => {
     const suiClient = new SuiClient({
       url: fullNodeURL,
     })
-    client = new AggregatorClient(endpoint, wallet, suiClient, Env.Mainnet)
+    client = new AggregatorClient({
+      endpoint,
+      signer: wallet,
+      client: suiClient,
+      env: Env.Mainnet,
+      pythUrls: ["https://cetus-pythnet-a648.mainnet.pythnet.rpcpool.com/219cf7a8-6d75-432d-a648-d487a6dd5dc3/hermes"],
+    })
   })
 
   test("Find Routers --> SUI -> SUPER_SUI", async () => {

package/tests/router/obric.test.ts

@@ -46,7 +46,13 @@ describe("Test obric provider", () => {
     const suiClient = new SuiClient({
       url: fullNodeURL,
     })
-    client = new AggregatorClient(endpoint, wallet, suiClient, Env.Mainnet)
+    client = new AggregatorClient({
+      endpoint,
+      signer: wallet,
+      client: suiClient,
+      env: Env.Mainnet,
+      pythUrls: ["https://cetus-pythnet-a648.mainnet.pythnet.rpcpool.com/219cf7a8-6d75-432d-a648-d487a6dd5dc3/hermes"],
+    })
   })
 
   test("Find Routers --> SUI -> USDC, locked", async () => {

package/tests/router/scallop.test.ts

@@ -50,7 +50,13 @@ describe("Test scallop provider", () => {
       url: fullNodeURL,
     })
 
-    client = new AggregatorClient(endpoint, wallet, suiClient, Env.Mainnet)
+    client = new AggregatorClient({
+      endpoint,
+      signer: wallet,
+      client: suiClient,
+      env: Env.Mainnet,
+      pythUrls: ["https://cetus-pythnet-a648.mainnet.pythnet.rpcpool.com/219cf7a8-6d75-432d-a648-d487a6dd5dc3/hermes"],
+    })
   })
 
   test("Find Routers", async () => {

package/tests/router/steamm.test.ts

@@ -47,7 +47,13 @@ describe("Test steammfe module", () => {
       url: fullNodeURL,
     })
 
-    client = new AggregatorClient(endpoint, wallet, suiClient, Env.Mainnet)
+    client = new AggregatorClient({
+      endpoint,
+      signer: wallet,
+      client: suiClient,
+      env: Env.Mainnet,
+      pythUrls: ["https://cetus-pythnet-a648.mainnet.pythnet.rpcpool.com/219cf7a8-6d75-432d-a648-d487a6dd5dc3/hermes"],
+    })
   })
 
   test("Find Routers", async () => {

package/tests/router.test.ts

@@ -43,7 +43,13 @@ describe("router module", () => {
       url: fullNodeURL,
     })
 
-    client = new AggregatorClient(endpoint, wallet, suiClient, Env.Mainnet)
+    client = new AggregatorClient({
+      endpoint,
+      signer: wallet,
+      client: suiClient,
+      env: Env.Mainnet,
+      pythUrls: ["https://cetus-pythnet-a648.mainnet.pythnet.rpcpool.com/219cf7a8-6d75-432d-a648-d487a6dd5dc3/hermes"],
+    })
   })
 
   test("Get coins", () => {