add smart wallet balances rpc method

src/tools/crypto_evm_wallet_toolkit.py

@@ -9,9 +9,12 @@ class CryptoEVMWalletTools(Toolkit):
         super().__init__(name="crypto_evm_wallet_tools")
 
         # Registering methods to make them accessible via the toolkit
-        self.register(self.get_evm_wallet_address)
+        # self.register(self.get_evm_wallet_address)
         self.register(self.get_supported_evm_chains)
         self.register(self.get_evm_smart_wallet_address)
+        self.register(self.get_evm_smart_wallet_balances)
+
+        # Fetching the list of supported EVM chains
         self.chains = self.get_supported_evm_chains()
 
     def get_supported_evm_chains(self) -> list[str]:
@@ -30,14 +33,50 @@ class CryptoEVMWalletTools(Toolkit):
         response = asyncio.run(rpc_call(method_name="getEVMSupportedChains", params=params))
         return f"{response}"
 
-    def get_evm_wallet_address(self, user_email: str, chain: str, testnet: bool = True) -> str:
+    # def get_evm_wallet_address(self, user_email: str, chain: str, testnet: bool = True) -> str:
+    #     """
+    #     Fetches an EVM wallet address for the given user email and supported chain.
+    #     Creates and returns an EVM wallet address for the given user email and supported chain.
+
+    #     Parameters:
+    #     - user_email (str): The email of the user for whom the wallet is being created.
+    #     - chain (ethereum | binance | base | polygon): The EVM chain for which the wallet is being fetched.
+    #     - testnet (bool, optional): A flag indicating whether the wallet should be on the testnet. Defaults to `True`.
+
+    #     Returns:
+    #     - str: A string representation of the response from the RPC call.
+
+    #     Raises:
+    #     None
+
+    #     Note:
+    #     This method uses asyncio.run() to run the asynchronous RPC call.
+    #     """
+    #     logger.info(f"Creating crypto wallet account for {user_email}")
+
+    #     params = {
+    #         'chain': chain,
+    #         'testnet': testnet,
+    #         'userEmail': user_email,
+    #     }
+    #     response = asyncio.run(rpc_call(method_name="getEVMWallet", params=params))
+    #     return f"{response}"
+    
+    def get_evm_smart_wallet_address(
+        self,
+        user_email: str,
+        chain: str,
+        gasless: bool = True,
+        testnet: bool = True
+    ) -> str:
         """
-        Fetches an EVM wallet address for the given user email and supported chain.
-        Creates and returns an EVM wallet address for the given user email and supported chain.
+        Fetches a smart EVM wallet address for the given user email and supported chain.
+        Creates and returns a smart EVM wallet address for the given user email and supported chain.
 
         Parameters:
-        - user_email (str): The email of the user for whom the wallet is being created.
+        - user_email (str): The email of the user for whom the wallet is being fetched.
         - chain (ethereum | binance | base | polygon): The EVM chain for which the wallet is being fetched.
+        - gasless (bool, optional): A flag indicating whether the wallet should be gasless. Defaults to `True`.
         - testnet (bool, optional): A flag indicating whether the wallet should be on the testnet. Defaults to `True`.
 
         Returns:
@@ -47,37 +86,37 @@ class CryptoEVMWalletTools(Toolkit):
         None
 
         Note:
-        This method uses asyncio.run() to run the asynchronous RPC call.
+        This method uses `asyncio.run()` to run the asynchronous RPC call.
         """
-        logger.info(f"Creating crypto wallet account for {user_email}")
+        logger.info(f"Fetching crypto wallet account for {user_email}")
 
         params = {
             'chain': chain,
+            'gasless': gasless,
             'testnet': testnet,
             'userEmail': user_email,
         }
-        response = asyncio.run(rpc_call(method_name="getEVMWallet", params=params))
+        response = asyncio.run(rpc_call(method_name="getEVMSmartWallet", params=params))
         return f"{response}"
     
-    def get_evm_smart_wallet_address(
+    def get_evm_smart_wallet_balances(
         self,
+        address: str,
         user_email: str,
         chain: str,
-        gasless: bool = True,
         testnet: bool = True
     ) -> str:
         """
-        Fetches a smart EVM wallet address for the given user email and supported chain.
-        Creates and returns a smart EVM wallet address for the given user email and supported chain.
+        Fetches the balances of a smart EVM wallet address for the given user email and supported chain.
 
         Parameters:
+        - address (str): The address of the smart EVM wallet for which the balances are being fetched.
         - user_email (str): The email of the user for whom the wallet is being fetched.
-        - chain (ethereum | binance | base | polygon): The EVM chain for which the wallet is being fetched.
-        - gasless (bool, optional): A flag indicating whether the wallet should be gasless. Defaults to `True`.
+        - chain (str): The EVM chain for which the wallet is being fetched.
         - testnet (bool, optional): A flag indicating whether the wallet should be on the testnet. Defaults to `True`.
 
         Returns:
-        - str: A string representation of the response from the RPC call.
+        - str: A string representation of the response from the RPC call containing the balances of the smart EVM wallet address.
 
         Raises:
         None
@@ -89,9 +128,9 @@ class CryptoEVMWalletTools(Toolkit):
 
         params = {
             'chain': chain,
-            'gasless': gasless,
+            'address': address,
             'testnet': testnet,
             'userEmail': user_email,
         }
-        response = asyncio.run(rpc_call(method_name="getEVMSmartWallet", params=params))
+        response = asyncio.run(rpc_call(method_name="getEVMSmartWalletAddressBalances", params=params))
         return f"{response}"