fix: get balance issues

node-real · Jan 20, 2025 · e007c61 · e007c61
1 parent c09d91f
commit e007c61
Showing 1 changed file with 6 additions and 9 deletions.
diff --git a/packages/plugin-bnb/src/actions/getBalance.ts b/packages/plugin-bnb/src/actions/getBalance.ts
@@ -65,6 +65,12 @@ export class GetBalanceAction {
                     token as `0x${string}`
                 );
             } else {
+                if (chainId != 56) {
+                    throw new Error(
+                        "Only BSC mainnet is supported for querying balance by token symbol"
+                    );
+                }
+
                 this.walletProvider.configureLiFiSdk(chain);
                 const tokenInfo = await getToken(chainId, token);
                 amount = await this.getERC20TokenBalance(
@@ -120,15 +126,6 @@ export class GetBalanceAction {
                 params.address
             );
         }
-
-        if (params.token.startsWith("0x")) {
-            // if token contract address is not provided, only BSC mainnet is supported
-            if (params.chain != "bsc") {
-                throw new Error(
-                    "Only BSC mainnet is supported for querying balance by token symbol"
-                );
-            }
-        }
     }
 }