File size: 3,599 Bytes
75bb6d0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 |
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}"
|