@cetusprotocol/aggregator-sdk 0.3.21 → 0.3.22

package/src/client.ts

@@ -43,7 +43,10 @@ import { Bluefin } from "./transaction/bluefin"
 import { HaedalPmm } from "./transaction/haedal_pmm"
 import { Alphafi } from "./transaction/alphafi"
 import { CoinUtils } from "./types/CoinAssist"
-
+import {
+  SuiPriceServiceConnection,
+  SuiPythClient,
+} from "@pythnetwork/pyth-sui-js"
 
 export const CETUS = "CETUS"
 export const DEEPBOOKV2 = "DEEPBOOK"
@@ -116,6 +119,12 @@ export interface SwapInPoolsParams {
   pools: string[]
 }
 
+interface PythConfig {
+  connections: SuiPriceServiceConnection[]
+  wormholeStateId: string
+  pythStateId: string
+}
+
 export interface SwapInPoolsResult {
   isExceed: boolean
   routeData?: RouterData
@@ -124,13 +133,13 @@ export interface SwapInPoolsResult {
 function isBuilderRouterSwapParams(
   params: BuildRouterSwapParams | BuildRouterSwapParamsV2
 ): params is BuildRouterSwapParams {
-  return Array.isArray((params as BuildRouterSwapParams).routers);
+  return Array.isArray((params as BuildRouterSwapParams).routers)
 }
 
 function isBuilderFastRouterSwapParams(
   params: BuildFastRouterSwapParams | BuildFastRouterSwapParamsV2
 ): params is BuildFastRouterSwapParams {
-  return Array.isArray((params as BuildFastRouterSwapParams).routers);
+  return Array.isArray((params as BuildFastRouterSwapParams).routers)
 }
 
 export class AggregatorClient {
@@ -140,6 +149,38 @@ export class AggregatorClient {
   public env: Env
   private allCoins: Map<string, CoinAsset[]>
 
+  private pythConnections: SuiPriceServiceConnection[]
+  private pythClient: SuiPythClient
+
+  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.liquify.com/api=X9LTVPQD7Y3R5N2A/hermes",
+          {
+            timeout: 3000,
+          }
+        ),
+        new SuiPriceServiceConnection("https://hermes.pyth.network", {
+          timeout: 3000,
+        }),
+      ],
+      wormholeStateId:
+        "0xaeab97f96cf9877fee2883315d459552b2b921edc16d7ceac6eab944dd88919c",
+      pythStateId:
+        "0x1f9310238ee9298fb703c3419030b35b22bb1cc37113e3bb5007c99aec79e5b8",
+    },
+  }
+
   constructor(
     endpoint?: string,
     signer?: string,
@@ -151,6 +192,14 @@ export class AggregatorClient {
     this.signer = signer || ""
     this.env = env || Env.Mainnet
     this.allCoins = new Map<string, CoinAsset[]>()
+
+    const config = AggregatorClient.CONFIG[this.env]
+    this.pythConnections = config.connections
+    this.pythClient = new SuiPythClient(
+      this.client,
+      config.pythStateId,
+      config.wormholeStateId
+    )
   }
 
   async getCoins(
@@ -205,6 +254,7 @@ export class AggregatorClient {
     inputCoin: TransactionObjectArgument,
     routers: Router[],
     amountOutLimit: BN,
+    pythPriceIDs: Map<string, string>,
     partner?: string,
     deepbookv3DeepFee?: TransactionObjectArgument,
     packages?: Map<string, string>
@@ -223,16 +273,31 @@ export class AggregatorClient {
       }
       let nextCoin = inputCoins[i] as TransactionObjectArgument
       for (const path of routers[i].path) {
-        const dex = this.newDex(path.provider, partner)
-        nextCoin = await dex.swap(this, txb, path, nextCoin, packages, deepbookv3DeepFee)
+        const dex = this.newDex(path.provider, pythPriceIDs, partner)
+        nextCoin = await dex.swap(
+          this,
+          txb,
+          path,
+          nextCoin,
+          packages,
+          deepbookv3DeepFee
+        )
       }
 
       outputCoins.push(nextCoin)
     }
 
-    const aggregatorV2PublishedAt = getAggregatorV2PublishedAt(this.publishedAtV2(), packages)
+    const aggregatorV2PublishedAt = getAggregatorV2PublishedAt(
+      this.publishedAtV2(),
+      packages
+    )
 
-    this.transferOrDestoryCoin(txb, inputCoin, inputCoinType, this.publishedAtV2())
+    this.transferOrDestoryCoin(
+      txb,
+      inputCoin,
+      inputCoinType,
+      this.publishedAtV2()
+    )
     const mergedTargetCointhis = this.checkCoinThresholdAndMergeCoin(
       txb,
       outputCoins,
@@ -255,7 +320,10 @@ export class AggregatorClient {
     const targetCoins = []
     const dex = new Cetus(this.env, partner)
 
-    const aggregatorV2PublishedAt = getAggregatorV2PublishedAt(this.publishedAtV2(), packages)
+    const aggregatorV2PublishedAt = getAggregatorV2PublishedAt(
+      this.publishedAtV2(),
+      packages
+    )
 
     for (let i = 0; i < routers.length; i++) {
       const router = routers[i]
@@ -280,7 +348,12 @@ export class AggregatorClient {
         if (j === 0) {
           inputCoin = repayResult
         } else {
-          this.transferOrDestoryCoin(txb, repayResult, path.from, aggregatorV2PublishedAt)
+          this.transferOrDestoryCoin(
+            txb,
+            repayResult,
+            path.from,
+            aggregatorV2PublishedAt
+          )
         }
         if (j === router.path.length - 1) {
           targetCoins.push(nextRepayCoin)
@@ -288,7 +361,12 @@ export class AggregatorClient {
       }
     }
     const inputCoinType = routers[0].path[0].from
-    this.transferOrDestoryCoin(txb, inputCoin, inputCoinType, aggregatorV2PublishedAt)
+    this.transferOrDestoryCoin(
+      txb,
+      inputCoin,
+      inputCoinType,
+      aggregatorV2PublishedAt
+    )
     if (targetCoins.length > 1) {
       const vec = txb.makeMoveVec({ elements: targetCoins.slice(1) })
       txb.moveCall({
@@ -317,19 +395,13 @@ export class AggregatorClient {
   async routerSwap(
     params: BuildRouterSwapParams | BuildRouterSwapParamsV2
   ): Promise<TransactionObjectArgument> {
-    const {
-      routers,
-      inputCoin,
-      slippage,
-      txb,
-      partner,
-      deepbookv3DeepFee,
-    } = params
+    const { routers, inputCoin, slippage, txb, partner, deepbookv3DeepFee } =
+      params
 
     const routerData = Array.isArray(routers) ? routers : routers.routes
-    const byAmountIn = isBuilderRouterSwapParams(params) 
-    ? params.byAmountIn 
-    : params.routers.byAmountIn
+    const byAmountIn = isBuilderRouterSwapParams(params)
+      ? params.byAmountIn
+      : params.routers.byAmountIn
 
     const amountIn = routerData.reduce(
       (acc, router) => acc.add(router.amountIn),
@@ -346,11 +418,23 @@ export class AggregatorClient {
       slippage
     )
 
-    const packages = isBuilderRouterSwapParams(params) ? undefined : params.routers.packages
+    const packages = isBuilderRouterSwapParams(params)
+      ? undefined
+      : params.routers.packages
 
     console.log("packages11", packages)
 
-    const aggregatorV2PublishedAt = getAggregatorV2PublishedAt(this.publishedAtV2(), packages)
+    const aggregatorV2PublishedAt = getAggregatorV2PublishedAt(
+      this.publishedAtV2(),
+      packages
+    )
+
+    const priceIDs = findPythPriceIDs(routerData)
+
+    const priceInfoObjectIds =
+      priceIDs.length > 0
+        ? await this.updatePythPriceIDs(priceIDs, txb)
+        : new Map<string, string>()
 
     if (byAmountIn) {
       const targetCoin = await this.expectInputSwap(
@@ -358,9 +442,10 @@ export class AggregatorClient {
         inputCoin,
         routerData,
         amountLimit,
+        priceInfoObjectIds,
         partner,
         deepbookv3DeepFee,
-        packages,
+        packages
       )
       return targetCoin
     }
@@ -369,7 +454,12 @@ export class AggregatorClient {
     const splitedInputCoins = txb.splitCoins(inputCoin, [
       amountLimit.toString(),
     ])
-    this.transferOrDestoryCoin(txb, inputCoin, routerData[0].path[0].from, aggregatorV2PublishedAt)
+    this.transferOrDestoryCoin(
+      txb,
+      inputCoin,
+      routerData[0].path[0].from,
+      aggregatorV2PublishedAt
+    )
     const targetCoin = await this.expectOutputSwap(
       txb,
       splitedInputCoins[0],
@@ -381,7 +471,9 @@ export class AggregatorClient {
 
   // auto build input coin
   // auto merge, transfer or destory target coin.
-  async fastRouterSwap(params: BuildFastRouterSwapParams | BuildFastRouterSwapParamsV2) {
+  async fastRouterSwap(
+    params: BuildFastRouterSwapParams | BuildFastRouterSwapParamsV2
+  ) {
     const {
       routers,
       slippage,
@@ -395,7 +487,8 @@ export class AggregatorClient {
     const fromCoinType = routerData[0].path[0].from
     let fromCoins = await this.getCoins(fromCoinType, refreshAllCoins)
 
-    const targetCoinType = routerData[0].path[routerData[0].path.length - 1].target
+    const targetCoinType =
+      routerData[0].path[routerData[0].path.length - 1].target
     const amountIn = routerData.reduce(
       (acc, router) => acc.add(router.amountIn),
       new BN(0)
@@ -405,7 +498,9 @@ export class AggregatorClient {
       new BN(0)
     )
 
-    const byAmountIn = isBuilderFastRouterSwapParams(params) ? params.byAmountIn : params.routers.byAmountIn
+    const byAmountIn = isBuilderFastRouterSwapParams(params)
+      ? params.byAmountIn
+      : params.routers.byAmountIn
     const amountLimit = CalculateAmountLimit(
       byAmountIn ? amountOut : amountIn,
       byAmountIn,
@@ -430,23 +525,25 @@ export class AggregatorClient {
       ).targetCoin
     }
 
-    const routerSwapParams = isBuilderFastRouterSwapParams(params) ? {
-      routers: routerData,
-      inputCoin: buildFromCoinRes.targetCoin,
-      slippage,
-      byAmountIn,
-      txb,
-      partner,
-      deepbookv3DeepFee: deepCoin,
-    } : {
-      routers: params.routers,
-      inputCoin: buildFromCoinRes.targetCoin,
-      slippage,
-      byAmountIn,
-      txb,
-      partner,
-      deepbookv3DeepFee: deepCoin,
-    }
+    const routerSwapParams = isBuilderFastRouterSwapParams(params)
+      ? {
+          routers: routerData,
+          inputCoin: buildFromCoinRes.targetCoin,
+          slippage,
+          byAmountIn,
+          txb,
+          partner,
+          deepbookv3DeepFee: deepCoin,
+        }
+      : {
+          routers: params.routers,
+          inputCoin: buildFromCoinRes.targetCoin,
+          slippage,
+          byAmountIn,
+          txb,
+          partner,
+          deepbookv3DeepFee: deepCoin,
+        }
 
     const targetCoin = await this.routerSwap(routerSwapParams)
 
@@ -460,10 +557,15 @@ export class AggregatorClient {
         BigInt(0),
         targetCoinType
       )
-  
-      const packages = isBuilderFastRouterSwapParams(params) ? undefined : params.routers.packages
-      const aggregatorV2PublishedAt = getAggregatorV2PublishedAt(this.publishedAtV2(), packages)
-  
+
+      const packages = isBuilderFastRouterSwapParams(params)
+        ? undefined
+        : params.routers.packages
+      const aggregatorV2PublishedAt = getAggregatorV2PublishedAt(
+        this.publishedAtV2(),
+        packages
+      )
+
       txb.mergeCoins(targetCoinRes.targetCoin, [targetCoin])
       if (targetCoinRes.isMintZeroCoin) {
         this.transferOrDestoryCoin(
@@ -498,7 +600,7 @@ export class AggregatorClient {
       // return "0x3b6d71bdeb8ce5b06febfd3cfc29ecd60d50da729477c8b8038ecdae34541b91" // version 5, add bluefin
       // return "0x81ade554cb24a7564ca43a4bfb7381b08d9e5c5f375162c95215b696ab903502" // version 6, force upgrade scallop
       // return "0x347dd58bbd11cd82c8b386b344729717c04a998da73386e82a239cc196d5706b" // version 7
-      return "0xf2fcea41dc217385019828375764fa06d9bd25e8e4726ba1962680849fb8d613"    // version 8
+      return "0xf2fcea41dc217385019828375764fa06d9bd25e8e4726ba1962680849fb8d613" // version 8
     } else {
       return "0xabb6a81c8a216828e317719e06125de5bb2cb0fe8f9916ff8c023ca5be224c78"
     }
@@ -551,7 +653,11 @@ export class AggregatorClient {
     return targetCoin
   }
 
-  newDex(provider: string, partner?: string): Dex {
+  newDex(
+    provider: string,
+    pythPriceIDs: Map<string, string>,
+    partner?: string
+  ): Dex {
     switch (provider) {
       case CETUS:
         return new Cetus(this.env, partner)
@@ -588,7 +694,7 @@ export class AggregatorClient {
       case BLUEFIN:
         return new Bluefin(this.env)
       case HAEDALPMM:
-        return new HaedalPmm(this.env, this.client)
+        return new HaedalPmm(this.env, pythPriceIDs)
       case ALPHAFI:
         return new Alphafi(this.env)
       default:
@@ -634,9 +740,74 @@ export class AggregatorClient {
     }
     return null
   }
+
+  async updatePythPriceIDs(
+    priceIDs: string[],
+    txb: Transaction
+  ): Promise<Map<string, string>> {
+    let priceUpdateData: Buffer[] | null = null
+    let lastError: Error | null = null
+
+    for (const connection of this.pythConnections) {
+      try {
+        priceUpdateData = await connection.getPriceFeedsUpdateData(priceIDs)
+        break
+      } catch (e) {
+        lastError = e as Error
+        console.log("Error: ", e)
+        continue
+      }
+    }
+
+    if (priceUpdateData == null) {
+      throw new Error(
+        `No pyth price seeds update data found: ${lastError?.message}`
+      )
+    }
+
+    let priceInfoObjectIds = []
+    try {
+      priceInfoObjectIds = await this.pythClient.updatePriceFeeds(
+        txb,
+        priceUpdateData,
+        priceIDs
+      )
+    } catch (e) {
+      throw new Error(`Failed to update price feeds: ${e}`)
+    }
+
+    let priceInfoObjectIdsMap = new Map<string, string>()
+    for (let i = 0; i < priceIDs.length; i++) {
+      priceInfoObjectIdsMap.set(priceIDs[i], priceInfoObjectIds[i])
+    }
+    return priceInfoObjectIdsMap
+  }
+}
+
+export function findPythPriceIDs(routes: Router[]): string[] {
+  const priceIDs = new Set<string>()
+
+  for (const route of routes) {
+    for (const path of route.path) {
+      if (path.provider === HAEDALPMM) {
+        if (
+          path.extendedDetails &&
+          path.extendedDetails.haedalPmmBasePriceSeed &&
+          path.extendedDetails.haedalPmmQuotePriceSeed
+        ) {
+          priceIDs.add(path.extendedDetails.haedalPmmBasePriceSeed)
+          priceIDs.add(path.extendedDetails.haedalPmmQuotePriceSeed)
+        }
+      }
+    }
+  }
+  return Array.from(priceIDs)
 }
 
-export function parseRouterResponse(data: any, byAmountIn: boolean): RouterData {
+export function parseRouterResponse(
+  data: any,
+  byAmountIn: boolean
+): RouterData {
   let totalDeepFee = 0
   for (const route of data.routes) {
     for (const path of route.path) {
@@ -674,7 +845,7 @@ export function parseRouterResponse(data: any, byAmountIn: boolean): RouterData
             path.provider === AFTERMATH ||
             path.provider === CETUS ||
             path.provider === DEEPBOOKV3 ||
-            path.provider === SCALLOP || 
+            path.provider === SCALLOP ||
             path.provider === HAEDALPMM
           ) {
             extendedDetails = {
@@ -685,8 +856,10 @@ export function parseRouterResponse(data: any, byAmountIn: boolean): RouterData
               deepbookv3DeepFee: path.extended_details?.deepbookv3_deep_fee,
               scallopScoinTreasury:
                 path.extended_details?.scallop_scoin_treasury,
-              haedalPmmBasePriceSeed: path.extended_details?.haedal_pmm_base_price_seed,
-              haedalPmmQuotePriceSeed: path.extended_details?.haedal_pmm_quote_price_seed,
+              haedalPmmBasePriceSeed:
+                path.extended_details?.haedal_pmm_base_price_seed,
+              haedalPmmQuotePriceSeed:
+                path.extended_details?.haedal_pmm_quote_price_seed,
             }
           }
 

package/src/transaction/haedal_pmm.ts

@@ -3,26 +3,23 @@ import {
   TransactionArgument,
   TransactionObjectArgument,
 } from "@mysten/sui/transactions"
-import { AggregatorClient, CLOCK_ADDRESS, Dex, Env, getAggregatorV2ExtendPublishedAt, Path } from ".."
-import { SuiPriceServiceConnection, SuiPythClient } from "@pythnetwork/pyth-sui-js"
-import { SuiClient } from "@mysten/sui/client"
+import {
+  AggregatorClient,
+  CLOCK_ADDRESS,
+  Dex,
+  Env,
+  getAggregatorV2ExtendPublishedAt,
+  Path,
+} from ".."
 
 export class HaedalPmm implements Dex {
-  private connection: SuiPriceServiceConnection
-  private pythClient: SuiPythClient
-  
-  constructor(env: Env, suiClient: SuiClient) {
+  private pythPriceIDs: Map<string, string>
+
+  constructor(env: Env, pythPriceIDs: Map<string, string>) {
     if (env === Env.Testnet) {
-      this.connection = new SuiPriceServiceConnection("https://hermes-beta.pyth.network")
-      const wormholeStateId = "0x31358d198147da50db32eda2562951d53973a0c0ad5ed738e9b17d88b213d790";
-      const pythStateId = "0x243759059f4c3111179da5878c12f68d612c21a8d54d85edc86164bb18be1c7c";
-      this.pythClient = new SuiPythClient(suiClient, pythStateId, wormholeStateId)
-    } else {
-      this.connection = new SuiPriceServiceConnection("https://hermes.pyth.network")
-      const wormholeStateId = "0xaeab97f96cf9877fee2883315d459552b2b921edc16d7ceac6eab944dd88919c";
-      const pythStateId = "0x1f9310238ee9298fb703c3419030b35b22bb1cc37113e3bb5007c99aec79e5b8";
-      this.pythClient = new SuiPythClient(suiClient, pythStateId, wormholeStateId)
+      throw new Error("HaedalPmm is not supported on testnet")
     }
+    this.pythPriceIDs = pythPriceIDs
   }
 
   async swap(
@@ -36,31 +33,43 @@ export class HaedalPmm implements Dex {
     const [func, coinAType, coinBType] = direction
       ? ["swap_a2b", from, target]
       : ["swap_b2a", target, from]
-    
+
     let basePriceSeed: string
     let quotePriceSeed: string
 
     if (path.extendedDetails == null) {
       throw new Error("Extended details not supported haedal pmm")
     } else {
-      if (!path.extendedDetails.haedalPmmBasePriceSeed || !path.extendedDetails.haedalPmmQuotePriceSeed) {
+      if (
+        !path.extendedDetails.haedalPmmBasePriceSeed ||
+        !path.extendedDetails.haedalPmmQuotePriceSeed
+      ) {
         throw new Error("Base price seed or quote price seed not supported")
       }
       basePriceSeed = path.extendedDetails.haedalPmmBasePriceSeed
       quotePriceSeed = path.extendedDetails.haedalPmmQuotePriceSeed
     }
 
-    const priceIDs = [basePriceSeed, quotePriceSeed]
-    const priceUpdateData = await this.connection.getPriceFeedsUpdateData(priceIDs);
-    const priceInfoObjectIds = await this.pythClient.updatePriceFeeds(txb, priceUpdateData, priceIDs);
+    const basePriceInfoObjectId = this.pythPriceIDs.get(basePriceSeed)
+    const quotePriceInfoObjectId = this.pythPriceIDs.get(quotePriceSeed)
+
+    if (!basePriceInfoObjectId || !quotePriceInfoObjectId) {
+      throw new Error(
+        "Base price info object id or quote price info object id not found"
+      )
+    }
+
     const args = [
       txb.object(path.id),
-      txb.object(priceInfoObjectIds[0]),
-      txb.object(priceInfoObjectIds[1]),
+      txb.object(basePriceInfoObjectId),
+      txb.object(quotePriceInfoObjectId),
       inputCoin,
       txb.object(CLOCK_ADDRESS),
     ]
-    const publishedAt = getAggregatorV2ExtendPublishedAt(client.publishedAtV2Extend(), packages)
+    const publishedAt = getAggregatorV2ExtendPublishedAt(
+      client.publishedAtV2Extend(),
+      packages
+    )
     const res = txb.moveCall({
       target: `${publishedAt}::haedalpmm::${func}`,
       typeArguments: [coinAType, coinBType],

package/src/transaction/index.ts

@@ -10,14 +10,20 @@ export const CLOCK_ADDRESS =
 export const AGGREGATOR_V2 = "aggregator_v2"
 export const AGGREGATOR_V2_EXTEND = "aggregator_v2_extend"
 
-export function getAggregatorV2PublishedAt(aggregatorV2PublishedAt: string, packages?: Map<string, string> | Record<string, string>) {
+export function getAggregatorV2PublishedAt(
+  aggregatorV2PublishedAt: string,
+  packages?: Map<string, string> | Record<string, string>
+) {
   if (packages instanceof Map) {
     return packages.get(AGGREGATOR_V2) ?? aggregatorV2PublishedAt
   }
   return aggregatorV2PublishedAt
 }
 
-export function getAggregatorV2ExtendPublishedAt(aggregatorV2ExtendPublishedAt: string, packages?: Map<string, string> | Record<string, string>) {
+export function getAggregatorV2ExtendPublishedAt(
+  aggregatorV2ExtendPublishedAt: string,
+  packages?: Map<string, string> | Record<string, string>
+) {
   if (packages instanceof Map) {
     return packages.get(AGGREGATOR_V2_EXTEND) ?? aggregatorV2ExtendPublishedAt
   }
@@ -31,7 +37,6 @@ export interface Dex {
     path: Path,
     inputCoin: TransactionObjectArgument,
     packages?: Map<string, string>,
-    deepbookv3DeepFee?: TransactionObjectArgument,
+    deepbookv3DeepFee?: TransactionObjectArgument
   ): Promise<TransactionObjectArgument>
 }
-

package/tests/router.test.ts

@@ -33,7 +33,10 @@ describe("router module", () => {
       keypair = buildTestAccount()
     }
 
-    const wallet = keypair.getPublicKey().toSuiAddress()
+    // const wallet = keypair.getPublicKey().toSuiAddress()
+    const wallet =
+      "0x073dfdcbb5d6552efd954273f7ab9e0b918d7bf30103e3f4e632502289cd92ee"
+
     console.log("wallet: ", wallet)
 
     const endpoint = aggregatorURL
@@ -61,7 +64,7 @@ describe("router module", () => {
       amount: new BN(amount),
       byAmountIn,
       pools: [
-        '0xb8d7d9e66a60c239e7a60110efcf8de6c705580ed924d0dde141f4a0e2c90105'
+        "0xb8d7d9e66a60c239e7a60110efcf8de6c705580ed924d0dde141f4a0e2c90105",
       ],
     })
 
@@ -108,8 +111,9 @@ describe("router module", () => {
 
   test("Build router tx", async () => {
     const byAmountIn = true
-    const amount = "1000000000"
-    const from = "0xbde4ba4c2e274a60ce15c1cfff9e5c42e41654ac8b6d906a57efa4bd3c29f47d::hasui::HASUI"
+    const amount = "100000"
+    const from =
+      "0xdba34672e30cb065b1f93e3ab55318768fd6fef66c15942c9f7cb846e2f900e7::usdc::USDC"
     const target = "0x2::sui::SUI"
 
     const res = await client.findRouters({
@@ -118,8 +122,8 @@ describe("router module", () => {
       amount: new BN(amount),
       byAmountIn,
       depth: 3,
-      providers: ["HAEDAL"],
-      splitCount:  1,
+      providers: ["CETUS"],
+      splitCount: 1,
     })
 
     if (res != null) {
@@ -205,8 +209,18 @@ describe("router module", () => {
 
   test("Test Multi Input", async () => {
     const amounts = [1000000000, 2000000000, 10000000000000]
-    const froms = [testData.M_USDC, testData.M_SUI, testData.M_CETUS, testData.M_NAVI]
-    const tos = [testData.M_SUI, testData.M_USDC, testData.M_USDC, testData.M_SUI]
+    const froms = [
+      testData.M_USDC,
+      testData.M_SUI,
+      testData.M_CETUS,
+      testData.M_NAVI,
+    ]
+    const tos = [
+      testData.M_SUI,
+      testData.M_USDC,
+      testData.M_USDC,
+      testData.M_SUI,
+    ]
 
     for (let i = 0; i < froms.length; i++) {
       const from = froms[i]
@@ -339,12 +353,12 @@ describe("router module", () => {
     console.log("config", config)
   }, 60000)
 
-
   test("Find router", async () => {
     const amount = "4239267610000000000"
     const res = await client.findRouters({
-      from: '0x2::sui::SUI',
-      target: '0x188c6239bda71ba5751990cc0271e91b6362c67bccf7aa381e018db7fe39c6d7::BUILD::BUILD',
+      from: "0x2::sui::SUI",
+      target:
+        "0x188c6239bda71ba5751990cc0271e91b6362c67bccf7aa381e018db7fe39c6d7::BUILD::BUILD",
       amount: new BN(amount),
       byAmountIn: true,
       depth: 3,