intentkit 0.6.0.dev17__py3-none-any.whl → 0.6.0.dev18__py3-none-any.whl

intentkit/__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/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/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.

{intentkit-0.6.0.dev17.dist-info → intentkit-0.6.0.dev18.dist-info}/METADATA

@@ -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.dist-info → intentkit-0.6.0.dev18.dist-info}/RECORD

@@ -1,4 +1,4 @@
-intentkit/__init__.py,sha256=oMztmHNzrAOrgpnkdinwmAhTrI5tdiQ98QZm6WfZWGA,385
+intentkit/__init__.py,sha256=IyGegFUuu3eVl1hNwwS8XT0f62Iz4DMDQmvX6OjejFM,385
 intentkit/abstracts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 intentkit/abstracts/agent.py,sha256=108gb5W8Q1Sy4G55F2_ZFv2-_CnY76qrBtpIr0Oxxqk,1489
 intentkit/abstracts/api.py,sha256=ZUc24vaQvQVbbjznx7bV0lbbQxdQPfEV8ZxM2R6wZWo,166
@@ -63,10 +63,10 @@ intentkit/skills/carv/fetch_news.py,sha256=-xCWPvdVQxI0tEynR8mnjNEFpDkE4k4vNo1A-
 intentkit/skills/carv/onchain_query.py,sha256=EYV7N2Y-CisyhNVyUm2NbJhdikeFg0Y_a6lToJ5iqDM,6371
 intentkit/skills/carv/schema.json,sha256=oSReeG50ZJTOxSRniQBJW6KfilscZfp66PiBVOjphS4,5021
 intentkit/skills/carv/token_info_and_price.py,sha256=R7M5-nkjrauvxaNkORYtLcP9yOYj8PvYu1xRkOBT6lU,4275
-intentkit/skills/cdp/__init__.py,sha256=VTBCyycdxsrrUaM8OZIbJDkEXeLDVun5Lo5a8oBW6mg,4489
+intentkit/skills/cdp/__init__.py,sha256=sBuRJdpCrqDQUY576GCph_VS2vnDTGeBnriSfSv7S3A,4399
 intentkit/skills/cdp/base.py,sha256=BcleKSXm0oNcHYky4uUPCJ3roXxMeTJs2OUS-8MkfMI,581
 intentkit/skills/cdp/cdp.png,sha256=dxPF6jPbKVfxJNMvbGTmBhXM-rSDvweF06txoX1cIM4,4425
-intentkit/skills/cdp/get_balance.py,sha256=IAsU7mQAPSQqCbYQsokkklsqESEVl82VxMt8dEDJ6cM,4211
+intentkit/skills/cdp/get_balance.py,sha256=pNIOvS8oxl7g01H48bM5IOFwwPaE6htEGSskkXfvV_A,5408
 intentkit/skills/cdp/schema.json,sha256=3ULPR95TW_Fp2GzzHwmwhAqm8vlxs2tN7fG6WoPY87c,11774
 intentkit/skills/chainlist/README.md,sha256=FEVQObs2W_oL2kyV-vZpHNwZCCgqm5Kt0sv0GD_Xo1M,958
 intentkit/skills/chainlist/__init__.py,sha256=2aRC-jXW3WYiR_-RM-_Ls1lby_54ZemZ6qtggntGa-U,1502
@@ -390,7 +390,7 @@ intentkit/utils/random.py,sha256=DymMxu9g0kuQLgJUqalvgksnIeLdS-v0aRk5nQU0mLI,452
 intentkit/utils/s3.py,sha256=9trQNkKQ5VgxWsewVsV8Y0q_pXzGRvsCYP8xauyUYkg,8549
 intentkit/utils/slack_alert.py,sha256=s7UpRgyzLW7Pbmt8cKzTJgMA9bm4EP-1rQ5KXayHu6E,2264
 intentkit/utils/tx.py,sha256=2yLLGuhvfBEY5n_GJ8wmIWLCzn0FsYKv5kRNzw_sLUI,1454
-intentkit-0.6.0.dev17.dist-info/METADATA,sha256=ml-xJtWAndVheeiJIfeEz1QKhlerzjoBejx9CXcrK2w,7286
-intentkit-0.6.0.dev17.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-intentkit-0.6.0.dev17.dist-info/licenses/LICENSE,sha256=Bln6DhK-LtcO4aXy-PBcdZv2f24MlJFm_qn222biJtE,1071
-intentkit-0.6.0.dev17.dist-info/RECORD,,
+intentkit-0.6.0.dev18.dist-info/METADATA,sha256=dD_D3EAa1oly7R5V91-hAl22oQqbkwJT-3zT7cht_bU,7286
+intentkit-0.6.0.dev18.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+intentkit-0.6.0.dev18.dist-info/licenses/LICENSE,sha256=Bln6DhK-LtcO4aXy-PBcdZv2f24MlJFm_qn222biJtE,1071
+intentkit-0.6.0.dev18.dist-info/RECORD,,