Add getAllEVMSmartWalletsAddressBalances rpc method

src/tools/crypto_evm_wallet_toolkit.py

@@ -12,6 +12,7 @@ class CryptoEVMWalletTools(Toolkit):
         self.register(self.get_evm_smart_wallet_address)
         self.register(self.get_evm_smart_wallet_balances)
         self.register(self.get_all_evm_smart_wallet_address)
+        self.register(self.get_all_evm_smart_wallet_balances)
 
         # Fetch the list of supported EVM chains
         self.chains = self.get_supported_evm_chains()
@@ -83,8 +84,7 @@ class CryptoEVMWalletTools(Toolkit):
 
         Args:
             user_email (str): The email of the user whose wallet addresses are to be fetched.
-            testnet (bool, optional): A flag indicating whether to fetch addresses from the testnet.
-                                    Defaults to True.
+            testnet (bool, optional): A flag indicating whether to fetch addresses from the testnet. Defaults to True.
 
         Returns:
             str: The response from the RPC call containing the wallet addresses.
@@ -133,3 +133,31 @@ class CryptoEVMWalletTools(Toolkit):
         }
         response = asyncio.run(rpc_call(method_name="getEVMSmartWalletAddressBalances", params=params))
         return f"{response}"
+    
+    def get_all_evm_smart_wallet_balances(
+        self,
+        user_email: str,
+        testnet: bool = True
+    ) -> str:
+        """
+        Fetches the balances of all smart EVM wallet addresses for the given user email.
+
+        This method takes a user's email and an optional testnet flag, constructs the
+        necessary parameters, and makes a remote procedure call (RPC) to fetch all EVM
+        smart wallet addresses associated with the user.
+
+        Args:
+            user_email (str): The email of the user whose wallet addresses are to be fetched.
+            testnet (bool, optional): A flag indicating whether to fetch addresses from the testnet. Defaults to True.
+
+        Returns:
+            str: The response from the RPC call containing the wallet addresses.
+        """
+        logger.info(f"Fetching crypto wallet account for {user_email}")
+
+        params = {
+            'testnet': testnet,
+            'userEmail': user_email,
+        }
+        response = asyncio.run(rpc_call(method_name="getAllEVMSmartWalletsAddressBalances", params=params))
+        return f"{response}"