intentkit 0.6.0.dev17__tar.gz → 0.6.0.dev18__tar.gz

{intentkit-0.6.0.dev17 → intentkit-0.6.0.dev18}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: intentkit
-Version: 0.6.0.dev17
+Version: 0.6.0.dev18
 Summary: Intent-based AI Agent Platform - Core Package
 Project-URL: Homepage, https://github.com/crestal-network/intentkit
 Project-URL: Repository, https://github.com/crestal-network/intentkit

{intentkit-0.6.0.dev17 → intentkit-0.6.0.dev18}/__init__.py

@@ -3,7 +3,7 @@
 A powerful platform for building AI agents with blockchain and cryptocurrency capabilities.
 """
 
-__version__ = "0.6.0-dev.17"
+__version__ = "0.6.0-dev.18"
 __author__ = "hyacinthus"
 __email__ = "hyacinthus@gmail.com"
 

{intentkit-0.6.0.dev17 → intentkit-0.6.0.dev18}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "intentkit"
-version = "0.6.0-dev.17"
+version = "0.6.0-dev.18"
 description = "Intent-based AI Agent Platform - Core Package"
 authors = [{ name = "hyacinthus", email = "hyacinthus@gmail.com" }]
 requires-python = ">=3.12"

{intentkit-0.6.0.dev17 → intentkit-0.6.0.dev18}/skills/cdp/__init__.py

@@ -114,10 +114,8 @@ async def get_skills(
     for skill in available_skills:
         if skill == "get_balance":
             # Get the account object for the custom GetBalance skill
-            account = await cdp_client.get_account()
             tools.append(
                 GetBalance(
-                    account=account,
                     agent_id=agent_id,
                     skill_store=store,
                 )

{intentkit-0.6.0.dev17 → intentkit-0.6.0.dev18}/skills/cdp/get_balance.py

@@ -1,4 +1,4 @@
-from typing import Type
+from typing import Optional, Type
 
 from cdp import EvmServerAccount
 from pydantic import BaseModel, Field
@@ -11,8 +11,9 @@ from intentkit.skills.cdp.base import CDPBaseTool
 class GetBalanceInput(BaseModel):
     """Input for GetBalance tool."""
 
-    asset_id: str = Field(
-        description="The asset ID to get the balance for (e.g., 'eth', 'usdc', or a valid contract address)"
+    asset_id: Optional[str] = Field(
+        default=None,
+        description="The asset ID to get the balance for (e.g., 'eth', 'usdc', or a valid contract address). If not provided, returns all token balances.",
     )
 
 
@@ -29,69 +30,92 @@ class GetBalance(CDPBaseTool):
 
     agent_id: str
     skill_store: SkillStoreABC
-    account: EvmServerAccount | None = None
 
     name: str = "cdp_get_balance"
     description: str = (
-        "This tool will get the balance of all the addresses in the wallet for a given asset. It takes the asset ID as input."
+        "This tool will get the balance of all the addresses in the wallet. If asset_id is provided, it returns the balance for that specific asset. "
+        "If no asset_id is provided, it returns all token balances. "
         "Always use 'eth' for the native asset ETH and 'usdc' for USDC. "
         "Other valid asset IDs are: weth,dai,reth,brett,w,cbeth,axl,iotx,prime,aero,rsr,mog,tbtc,npc,yfi"
     )
     args_schema: Type[BaseModel] = GetBalanceInput
 
-    async def _arun(self, asset_id: str) -> str:
+    async def _arun(self, asset_id: Optional[str] = None) -> str:
         """Async implementation of the tool to get balance.
 
         Args:
-            asset_id (str): The asset ID to get the balance for.
+            asset_id (Optional[str]): The asset ID to get the balance for. If None, returns all token balances.
 
         Returns:
             str: A message containing the balance information or error message.
         """
         try:
-            if not self.account:
-                return "Failed to get account."
-
             # Get network information from CDP client
             cdp_client = await get_cdp_client(self.agent_id, self.skill_store)
+            provider = await cdp_client.get_wallet_provider()
+            account: EvmServerAccount = provider._account
             provider_config = await cdp_client.get_provider_config()
             network_id = provider_config.network_id
 
             # Map network_id to the format expected by the API
             network_mapping = {
                 "base-mainnet": "base",
-                "base-sepolia": "base-sepolia",
-                "ethereum": "ethereum",
                 "ethereum-mainnet": "ethereum",
             }
             api_network = network_mapping.get(network_id, network_id)
 
+            # If no asset_id provided, return all token balances
+            if asset_id is None:
+                try:
+                    # Get native ETH balance
+                    balance_wei = provider.get_balance()
+                    balance_eth = balance_wei / (10**18)  # Convert from wei to ETH
+
+                    # Get all token balances
+                    token_balances = await account.list_token_balances(api_network)
+
+                    result = [f"ETH balance: {balance_eth} ETH"]
+
+                    for balance in token_balances.balances:
+                        result.append(
+                            f"{balance.token.symbol} balance: {balance.amount.decimals} {balance.token.name}"
+                        )
+
+                    return f"All balances for account {account.address}:\n" + "\n".join(
+                        result
+                    )
+
+                except Exception as e:
+                    return f"Error getting all balances: {e!s}"
+
             # For native ETH balance, use the account's balance directly
             if asset_id.lower() == "eth":
                 try:
                     # Get native balance using Web3
-                    balance_wei = await self.account.get_balance()
+                    balance_wei = provider.get_balance()
                     balance_eth = balance_wei / (10**18)  # Convert from wei to ETH
-                    return f"ETH balance for account {self.account.address}: {balance_eth} ETH"
+                    return (
+                        f"ETH balance for account {account.address}: {balance_eth} ETH"
+                    )
                 except Exception as e:
                     return f"Error getting ETH balance: {e!s}"
 
             # For other tokens, try the list_token_balances API
             try:
                 # list_token_balances returns all token balances for the account
-                token_balances = await self.account.list_token_balances(api_network)
+                token_balances = await account.list_token_balances(api_network)
 
                 # Find the balance for the specific asset
                 target_balance = None
-                for balance in token_balances:
-                    if balance.asset_id.lower() == asset_id.lower():
+                for balance in token_balances.balances:
+                    if balance.token.symbol.lower() == asset_id.lower():
                         target_balance = balance
                         break
 
                 if target_balance:
-                    return f"Balance for {asset_id} in account {self.account.address}: {target_balance.amount} {target_balance.asset_id}"
+                    return f"Balance for {asset_id} in account {account.address}: {target_balance.amount.decimals} {target_balance.token.name}"
                 else:
-                    return f"No balance found for asset {asset_id} in account {self.account.address}"
+                    return f"No balance found for asset {asset_id} in account {account.address}"
 
             except Exception as e:
                 return f"Error getting balance for account: {e!s}"
@@ -99,7 +123,7 @@ class GetBalance(CDPBaseTool):
         except Exception as e:
             return f"Error getting balance: {str(e)}"
 
-    def _run(self, asset_id: str) -> str:
+    def _run(self, asset_id: Optional[str] = None) -> str:
         """Sync implementation of the tool.
 
         This method is deprecated since we now have native async implementation in _arun.