add evm smart wallet toolkit

chatxbt-assistant.py

@@ -11,7 +11,7 @@ from phi.tools.yfinance import YFinanceTools
 from src.databases.postgres import sqlalchemy_engine
 from src.tools.crypto_swap_toolkit import CryptoSwapTools
 from src.tools.crypto_data_toolkit import CryptoDataTools
-from src.tools.crypto_wallet_toolkit import CryptoWalletTools
+from src.tools.crypto_evm_wallet_toolkit import CryptoEVMWalletTools
 from phi.storage.assistant.postgres import PgAssistantStorage
 from src.knowledge_bases.combined import knowledge_base
 
@@ -69,7 +69,7 @@ async def start():
             DuckDuckGo(),
             CryptoDataTools(),
             CryptoSwapTools(),
-            CryptoWalletTools(),
+            CryptoEVMWalletTools(),
             YFinanceTools(stock_price=True)
         ],
         show_tool_calls= is_dev_mode,

src/tools/crypto_evm_wallet_toolkit.py

@@ -0,0 +1,97 @@
+import asyncio
+from enum import Enum
+from phi.tools import Toolkit
+from phi.utils.log import logger
+from src.libs.rpc_client import rpc_call
+
+class CryptoEVMWalletTools(Toolkit):
+    def __init__(self):
+        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_supported_evm_chains)
+        self.register(self.get_evm_smart_wallet_address)
+        self.chains = self.get_supported_evm_chains()
+
+    def get_supported_evm_chains(self) -> list[str]:
+        """
+        Fetches the list of supported EVM chains.
+
+        Returns:
+        str: A string representation of the response from the RPC call containing the list of supported EVM chains.
+
+        Raises:
+        None
+        """
+        logger.info("Fetching supported EVM chains")
+        
+        params = {}
+        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:
+        """
+        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 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 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:
+        - 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"Fetching crypto wallet account for {user_email}")
+
+        params = {
+            'chain': chain,
+            'gasless': gasless,
+            'testnet': testnet,
+            'userEmail': user_email,
+        }
+        response = asyncio.run(rpc_call(method_name="getEVMSmartWallet", params=params))
+        return f"{response}"

src/tools/crypto_wallet_toolkit.py

@@ -1,62 +0,0 @@
-import asyncio
-from phi.tools import Toolkit
-from phi.utils.log import logger
-from src.libs.rpc_client import rpc_call
-
-class CryptoWalletTools(Toolkit):
-    def __init__(self):
-        super().__init__(name="crypto_wallet_tools")
-
-        # Registering methods to make them accessible via the toolkit
-        self.register(self.get_crypto_wallet_address)
-        self.register(self.create_crypto_wallet)
-
-    def create_crypto_wallet(self, user_email: str) -> str:
-        """
-        Creates a crypto wallet account for the given user email.
-
-        Parameters:
-        user_email (str): The email of the user for whom the wallet is being created.
-
-        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 = {
-            'userEmail': user_email,
-        }
-        response = asyncio.run(rpc_call(method_name="createLocalWallet", params=params))
-        return f"{response}"
-    
-    def get_crypto_wallet_address(self, user_email: str) -> str:
-        """
-        Fetches the crypto wallet account addresse for the given user email.
-
-        Parameters:
-        user_email (str): The email of the user for whom the wallet is being fetched.
-
-        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"Fetching crypto wallet account for {user_email}")
-        
-        params = {
-            'userEmail': user_email,
-        }
-        response = asyncio.run(rpc_call(method_name="getLocalWalletAddress", params=params))
-        return f"{response}"
-        
-