intentkit 0.6.0.dev18__tar.gz → 0.6.1.dev1__tar.gz

{intentkit-0.6.0.dev18 → intentkit-0.6.1.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: intentkit
-Version: 0.6.0.dev18
+Version: 0.6.1.dev1
 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.dev18 → intentkit-0.6.1.dev1}/__init__.py

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

{intentkit-0.6.0.dev18 → intentkit-0.6.1.dev1}/clients/cdp.py

@@ -114,6 +114,9 @@ class CdpClient:
                 )
                 address = new_account.address
                 logger.info("Created new wallet: %s", address)
+
+            # close client
+            await cdp_client.close()
             # now it should be created or migrated, store it
             agent_data.evm_wallet_address = address
             await agent_data.save()

{intentkit-0.6.0.dev18 → intentkit-0.6.1.dev1}/core/node.py

@@ -135,9 +135,7 @@ class PreModelNode(RunnableCallable):
                     f"Trimmed messages: {len(messages)} -> {len(trimmed_messages)}"
                 )
                 if len(trimmed_messages) <= 3:
-                    logger.error(
-                        f"Too few messages after trim: {len(trimmed_messages)}"
-                    )
+                    logger.info(f"Too few messages after trim: {len(trimmed_messages)}")
                     return {}
                 return {
                     "messages": [RemoveMessage(REMOVE_ALL_MESSAGES)] + trimmed_messages,

{intentkit-0.6.0.dev18 → intentkit-0.6.1.dev1}/pyproject.toml

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

intentkit-0.6.1.dev1/skills/cdp/get_balance.py

@@ -0,0 +1,110 @@
+from typing import Optional, Type
+
+from pydantic import BaseModel, Field
+
+from intentkit.abstracts.skill import SkillStoreABC
+from intentkit.clients import get_cdp_client
+from intentkit.skills.cdp.base import CDPBaseTool
+
+
+class GetBalanceInput(BaseModel):
+    """Input for GetBalance tool."""
+
+    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.",
+    )
+
+
+class GetBalance(CDPBaseTool):
+    """Tool for getting balance from CDP wallet.
+
+    This tool uses the CDP API to get balance for all addresses in a wallet for a given asset.
+
+    Attributes:
+        name: The name of the tool.
+        description: A description of what the tool does.
+        args_schema: The schema for the tool's input arguments.
+    """
+
+    agent_id: str
+    skill_store: SkillStoreABC
+
+    name: str = "cdp_get_balance"
+    description: str = (
+        "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: Optional[str] = None) -> str:
+        """Async implementation of the tool to get balance.
+
+        Args:
+            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.
+        """
+        # 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()
+        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",
+            "ethereum-mainnet": "ethereum",
+        }
+        api_network = network_mapping.get(network_id, network_id)
+
+        # For native ETH balance, use the account's balance directly
+        if asset_id and asset_id.lower() == "eth":
+            try:
+                # Get native balance using Web3
+                balance_wei = provider.get_balance()
+                balance_eth = balance_wei / (10**18)  # Convert from wei to ETH
+                return f"ETH balance: {balance_eth} ETH"
+            except Exception as e:
+                return f"Error getting ETH balance: {e!s}"
+
+        client = provider.get_client()
+        async with client:
+            account = await client.evm.get_account(provider.get_address())
+            # If no asset_id provided, return all token balances
+            if asset_id is None:
+                # 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
+                )
+
+            # For other tokens, try the list_token_balances API
+            token_balances = await account.list_token_balances(api_network)
+
+            # Find the balance for the specific asset
+            target_balance = None
+            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 {account.address}: {target_balance.amount.decimals} {target_balance.token.name}"
+            else:
+                return f"No balance found for asset {asset_id} in account {account.address}"

{intentkit-0.6.0.dev18 → intentkit-0.6.1.dev1}/skills/firecrawl/schema.json

@@ -16,7 +16,7 @@
       "type": "boolean",
       "title": "Enabled",
       "description": "Whether this skill is enabled",
-      "default": true
+      "default": false
     },
     "states": {
       "type": "object",
@@ -150,4 +150,4 @@
     }
   },
   "additionalProperties": true
-} 
+}

{intentkit-0.6.0.dev18 → intentkit-0.6.1.dev1}/skills/supabase/__init__.py

@@ -1,7 +1,7 @@
 """Supabase skills."""
 
 import logging
-from typing import TypedDict
+from typing import NotRequired, TypedDict
 
 from intentkit.abstracts.skill import SkillStoreABC
 from intentkit.skills.base import SkillConfig, SkillState
@@ -34,7 +34,8 @@ class Config(SkillConfig):
     states: SkillStates
     supabase_url: str
     supabase_key: str
-    public_write_tables: str = ""
+    public_write_tables: NotRequired[str]
+    public_key: NotRequired[str]
 
 
 async def get_skills(

{intentkit-0.6.0.dev18 → intentkit-0.6.1.dev1}/skills/supabase/base.py

@@ -21,11 +21,14 @@ class SupabaseBaseTool(IntentKitSkill):
     def category(self) -> str:
         return "supabase"
 
-    def get_supabase_config(self, config: dict) -> tuple[str, str]:
+    def get_supabase_config(
+        self, config: dict, context: SkillContext
+    ) -> tuple[str, str]:
         """Get Supabase URL and key from config.
 
         Args:
             config: The agent configuration
+            context: The skill context containing configuration and mode info
 
         Returns:
             Tuple of (supabase_url, supabase_key)
@@ -34,7 +37,13 @@ class SupabaseBaseTool(IntentKitSkill):
             ValueError: If required config is missing
         """
         supabase_url = config.get("supabase_url")
-        supabase_key = config.get("supabase_key")
+
+        # Use public_key for public operations if available, otherwise fall back to supabase_key
+        if context.is_private:
+            supabase_key = config.get("supabase_key")
+        else:
+            # Try public_key first, fall back to supabase_key if public_key doesn't exist
+            supabase_key = config.get("public_key") or config.get("supabase_key")
 
         if not supabase_url:
             raise ValueError("supabase_url is required in config")

{intentkit-0.6.0.dev18 → intentkit-0.6.1.dev1}/skills/supabase/delete_data.py

@@ -51,7 +51,9 @@ class SupabaseDeleteData(SupabaseBaseTool):
             # Validate table access for public mode
             self.validate_table_access(table, context)
 
-            supabase_url, supabase_key = self.get_supabase_config(context.config)
+            supabase_url, supabase_key = self.get_supabase_config(
+                context.config, context
+            )
 
             # Create Supabase client
             supabase: Client = create_client(supabase_url, supabase_key)

{intentkit-0.6.0.dev18 → intentkit-0.6.1.dev1}/skills/supabase/fetch_data.py

@@ -62,7 +62,9 @@ class SupabaseFetchData(SupabaseBaseTool):
     ):
         try:
             context = self.context_from_config(config)
-            supabase_url, supabase_key = self.get_supabase_config(context.config)
+            supabase_url, supabase_key = self.get_supabase_config(
+                context.config, context
+            )
 
             # Create Supabase client
             supabase: Client = create_client(supabase_url, supabase_key)

{intentkit-0.6.0.dev18 → intentkit-0.6.1.dev1}/skills/supabase/insert_data.py

@@ -51,7 +51,9 @@ class SupabaseInsertData(SupabaseBaseTool):
             # Validate table access for public mode
             self.validate_table_access(table, context)
 
-            supabase_url, supabase_key = self.get_supabase_config(context.config)
+            supabase_url, supabase_key = self.get_supabase_config(
+                context.config, context
+            )
 
             # Create Supabase client
             supabase: Client = create_client(supabase_url, supabase_key)

{intentkit-0.6.0.dev18 → intentkit-0.6.1.dev1}/skills/supabase/invoke_function.py

@@ -46,7 +46,9 @@ class SupabaseInvokeFunction(SupabaseBaseTool):
     ):
         try:
             context = self.context_from_config(config)
-            supabase_url, supabase_key = self.get_supabase_config(context.config)
+            supabase_url, supabase_key = self.get_supabase_config(
+                context.config, context
+            )
 
             # Create Supabase client
             supabase: Client = create_client(supabase_url, supabase_key)

{intentkit-0.6.0.dev18 → intentkit-0.6.1.dev1}/skills/supabase/schema.json

@@ -143,11 +143,17 @@
       "x-sensitive": true,
       "format": "password"
     },
+    "public_key": {
+      "type": "string",
+      "title": "Public API Key",
+      "description": "You can add a key with more restrictive permissions for public skills. If not provided, supabase_key will be used.",
+      "x-sensitive": true,
+      "format": "password"
+    },
     "public_write_tables": {
       "type": "string",
       "title": "Public Write Tables",
-      "description": "Add tables separated by commas. When insert, update, upsert, or delete operations are enabled for public use, only tables from this list can be used. This list does not restrict the skills executed by the owner or in autonomous chat.",
-      "default": ""
+      "description": "Add tables separated by commas. When insert, update, upsert, or delete operations are enabled for public use, only tables from this list can be used. This list does not restrict the skills executed by the owner or in autonomous chat."
     }
   },
   "required": [

{intentkit-0.6.0.dev18 → intentkit-0.6.1.dev1}/skills/supabase/update_data.py

@@ -54,7 +54,9 @@ class SupabaseUpdateData(SupabaseBaseTool):
             # Validate table access for public mode
             self.validate_table_access(table, context)
 
-            supabase_url, supabase_key = self.get_supabase_config(context.config)
+            supabase_url, supabase_key = self.get_supabase_config(
+                context.config, context
+            )
 
             # Create Supabase client
             supabase: Client = create_client(supabase_url, supabase_key)

{intentkit-0.6.0.dev18 → intentkit-0.6.1.dev1}/skills/supabase/upsert_data.py

@@ -56,7 +56,9 @@ class SupabaseUpsertData(SupabaseBaseTool):
             # Validate table access for public mode
             self.validate_table_access(table, context)
 
-            supabase_url, supabase_key = self.get_supabase_config(context.config)
+            supabase_url, supabase_key = self.get_supabase_config(
+                context.config, context
+            )
 
             # Create Supabase client
             supabase: Client = create_client(supabase_url, supabase_key)

{intentkit-0.6.0.dev18 → intentkit-0.6.1.dev1}/skills/system/schema.json

@@ -14,7 +14,7 @@
       "type": "boolean",
       "title": "Enabled",
       "description": "Whether this skill is enabled",
-      "default": true
+      "default": false
     },
     "states": {
       "type": "object",

{intentkit-0.6.0.dev18 → intentkit-0.6.1.dev1}/skills/tavily/schema.json

@@ -16,7 +16,7 @@
       "type": "boolean",
       "title": "Enabled",
       "description": "Whether this skill is enabled",
-      "default": true
+      "default": false
     },
     "states": {
       "type": "object",

intentkit-0.6.0.dev18/skills/cdp/get_balance.py

@@ -1,133 +0,0 @@
-from typing import Optional, Type
-
-from cdp import EvmServerAccount
-from pydantic import BaseModel, Field
-
-from intentkit.abstracts.skill import SkillStoreABC
-from intentkit.clients import get_cdp_client
-from intentkit.skills.cdp.base import CDPBaseTool
-
-
-class GetBalanceInput(BaseModel):
-    """Input for GetBalance tool."""
-
-    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.",
-    )
-
-
-class GetBalance(CDPBaseTool):
-    """Tool for getting balance from CDP wallet.
-
-    This tool uses the CDP API to get balance for all addresses in a wallet for a given asset.
-
-    Attributes:
-        name: The name of the tool.
-        description: A description of what the tool does.
-        args_schema: The schema for the tool's input arguments.
-    """
-
-    agent_id: str
-    skill_store: SkillStoreABC
-
-    name: str = "cdp_get_balance"
-    description: str = (
-        "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: Optional[str] = None) -> str:
-        """Async implementation of the tool to get balance.
-
-        Args:
-            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:
-            # 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",
-                "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 = provider.get_balance()
-                    balance_eth = balance_wei / (10**18)  # Convert from wei to 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 account.list_token_balances(api_network)
-
-                # Find the balance for the specific asset
-                target_balance = None
-                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 {account.address}: {target_balance.amount.decimals} {target_balance.token.name}"
-                else:
-                    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}"
-
-        except Exception as e:
-            return f"Error getting balance: {str(e)}"
-
-    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.
-        """
-        raise NotImplementedError(
-            "Use _arun instead, which is the async implementation"
-        )