intentkit 0.7.5.dev3__py3-none-any.whl → 0.8.34.dev7__py3-none-any.whl

intentkit/skills/system/list_autonomous_tasks.py

@@ -1,5 +1,3 @@
-from typing import List
-
 from pydantic import BaseModel, Field
 
 from intentkit.models.agent import AgentAutonomous
@@ -15,7 +13,7 @@ class ListAutonomousTasksInput(BaseModel):
 class ListAutonomousTasksOutput(BaseModel):
     """Output model for list_autonomous_tasks skill."""
 
-    tasks: List[AgentAutonomous] = Field(
+    tasks: list[AgentAutonomous] = Field(
         description="List of autonomous task configurations for the agent"
     )
 
@@ -43,8 +41,8 @@ class ListAutonomousTasks(SystemBaseTool):
             ListAutonomousTasksOutput: List of autonomous tasks
         """
         context = self.get_context()
-        agent_id = context.agent_id
+        agent = context.agent
 
-        tasks = await self.skill_store.list_autonomous_tasks(agent_id)
+        tasks = await agent.list_autonomous_tasks()
 
         return ListAutonomousTasksOutput(tasks=tasks)

intentkit/skills/system/read_agent_api_key.py

@@ -1,5 +1,7 @@
 from pydantic import BaseModel, Field
 
+from intentkit.config.config import config
+from intentkit.models.agent_data import AgentData
 from intentkit.skills.system.base import SystemBaseTool
 
 
@@ -41,11 +43,11 @@ class ReadAgentApiKey(SystemBaseTool):
         context = self.get_context()
         agent_id = context.agent_id
 
-        # Get agent data from skill store
-        agent_data = await self.skill_store.get_agent_data(agent_id)
+        # Get agent data from the database
+        agent_data = await AgentData.get(agent_id)
 
         # Get API base URL from system config
-        open_api_base_url = self.skill_store.get_system_config("open_api_base_url")
+        open_api_base_url = config.open_api_base_url
         api_endpoint = f"{open_api_base_url}/v1/chat/completions"
 
         # Check if API keys exist
@@ -72,7 +74,7 @@ class ReadAgentApiKey(SystemBaseTool):
             update_data["api_key_public"] = new_public_api_key
 
         if update_data:
-            await self.skill_store.set_agent_data(agent_id, update_data)
+            await AgentData.patch(agent_id, update_data)
 
         return ReadAgentApiKeyOutput(
             api_key=new_api_key,

intentkit/skills/system/regenerate_agent_api_key.py

@@ -1,5 +1,7 @@
 from pydantic import BaseModel, Field
 
+from intentkit.config.config import config
+from intentkit.models.agent_data import AgentData
 from intentkit.skills.system.base import SystemBaseTool
 
 
@@ -43,11 +45,11 @@ class RegenerateAgentApiKey(SystemBaseTool):
         context = self.get_context()
         agent_id = context.agent_id
 
-        # Get agent data from skill store
-        agent_data = await self.skill_store.get_agent_data(agent_id)
+        # Get agent data directly from the model
+        agent_data = await AgentData.get(agent_id)
 
         # Get API base URL from system config
-        open_api_base_url = self.skill_store.get_system_config("open_api_base_url")
+        open_api_base_url = config.open_api_base_url
         api_endpoint = f"{open_api_base_url}/v1/chat/completions"
 
         # Check if previous API keys existed
@@ -58,7 +60,7 @@ class RegenerateAgentApiKey(SystemBaseTool):
         new_public_api_key = self._generate_public_api_key()
 
         # Save the new API keys to agent data (overwrites existing)
-        await self.skill_store.set_agent_data(
+        await AgentData.patch(
             agent_id, {"api_key": new_api_key, "api_key_public": new_public_api_key}
         )
 

intentkit/skills/system/schema.json

@@ -5,16 +5,14 @@
   "description": "System management and configuration skills for agent operations including API key management",
   "x-icon": "https://ai.service.crestal.dev/skills/system/system.svg",
   "x-tags": [
-    "System",
-    "Management",
-    "Configuration"
+    "Infrastructure"
   ],
   "properties": {
     "enabled": {
       "type": "boolean",
       "title": "Enabled",
       "description": "Whether this skill is enabled",
-      "default": false
+      "default": true
     },
     "states": {
       "type": "object",
@@ -59,7 +57,7 @@
             "Agent Owner Only"
           ],
           "description": "List all autonomous task configurations for the agent.",
-          "default": "disabled"
+          "default": "private"
         },
         "add_autonomous_task": {
           "type": "string",
@@ -73,7 +71,7 @@
             "Agent Owner Only"
           ],
           "description": "Add a new autonomous task configuration to the agent.",
-          "default": "disabled"
+          "default": "private"
         },
         "delete_autonomous_task": {
           "type": "string",
@@ -87,7 +85,7 @@
             "Agent Owner Only"
           ],
           "description": "Delete an autonomous task configuration from the agent.",
-          "default": "disabled"
+          "default": "private"
         },
         "edit_autonomous_task": {
           "type": "string",
@@ -101,7 +99,7 @@
             "Agent Owner Only"
           ],
           "description": "Edit an existing autonomous task configuration for the agent.",
-          "default": "disabled"
+          "default": "private"
         }
       }
     }

intentkit/skills/tavily/__init__.py

@@ -3,7 +3,6 @@
 import logging
 from typing import TypedDict
 
-from intentkit.abstracts.skill import SkillStoreABC
 from intentkit.skills.base import SkillConfig, SkillState
 from intentkit.skills.tavily.base import TavilyBaseTool
 from intentkit.skills.tavily.tavily_extract import TavilyExtract
@@ -30,7 +29,6 @@ class Config(SkillConfig):
 async def get_skills(
     config: "Config",
     is_private: bool,
-    store: SkillStoreABC,
     **_,
 ) -> list[TavilyBaseTool]:
     """Get all Tavily search skills.
@@ -38,7 +36,6 @@ async def get_skills(
     Args:
         config: The configuration for Tavily search skills.
         is_private: Whether to include private skills.
-        store: The skill store for persisting data.
 
     Returns:
         A list of Tavily search skills.
@@ -55,7 +52,7 @@ async def get_skills(
     # Get each skill using the cached getter
     result = []
     for name in available_skills:
-        skill = get_tavily_skill(name, store)
+        skill = get_tavily_skill(name)
         if skill:
             result.append(skill)
     return result
@@ -63,28 +60,22 @@ async def get_skills(
 
 def get_tavily_skill(
     name: str,
-    store: SkillStoreABC,
 ) -> TavilyBaseTool:
     """Get a Tavily search skill by name.
 
     Args:
         name: The name of the skill to get
-        store: The skill store for persisting data
 
     Returns:
         The requested Tavily search skill
     """
     if name == "tavily_search":
         if name not in _cache:
-            _cache[name] = TavilySearch(
-                skill_store=store,
-            )
+            _cache[name] = TavilySearch()
         return _cache[name]
     elif name == "tavily_extract":
         if name not in _cache:
-            _cache[name] = TavilyExtract(
-                skill_store=store,
-            )
+            _cache[name] = TavilyExtract()
         return _cache[name]
     else:
         logger.warning(f"Unknown Tavily skill: {name}")

intentkit/skills/tavily/base.py

@@ -1,9 +1,7 @@
-from typing import Type
-
-from langchain.tools.base import ToolException
+from langchain_core.tools.base import ToolException
 from pydantic import BaseModel, Field
 
-from intentkit.abstracts.skill import SkillStoreABC
+from intentkit.config.config import config
 from intentkit.skills.base import IntentKitSkill
 
 
@@ -12,17 +10,14 @@ class TavilyBaseTool(IntentKitSkill):
 
     name: str = Field(description="The name of the tool")
     description: str = Field(description="A description of what the tool does")
-    args_schema: Type[BaseModel]
-    skill_store: SkillStoreABC = Field(
-        description="The skill store for persisting data"
-    )
+    args_schema: type[BaseModel]
 
     def get_api_key(self) -> str:
         context = self.get_context()
         skill_config = context.agent.skill_config(self.category)
         api_key_provider = skill_config.get("api_key_provider")
         if api_key_provider == "platform":
-            return self.skill_store.get_system_config("tavily_api_key")
+            return config.tavily_api_key
         # for backward compatibility, may only have api_key in skill_config
         elif skill_config.get("api_key"):
             return skill_config.get("api_key")

intentkit/skills/tavily/schema.json

@@ -5,10 +5,8 @@
   "description": "Web search and content extraction capabilities using Tavily",
   "x-icon": "https://ai.service.crestal.dev/skills/tavily/tavily.jpg",
   "x-tags": [
-    "Internet",
-    "Search",
-    "Information",
-    "Content Extraction"
+    "Knowledge Base",
+    "Search"
   ],
   "x-nft-requirement": 1,
   "properties": {
@@ -116,4 +114,4 @@
     }
   },
   "additionalProperties": true
-}
+}

intentkit/skills/tavily/tavily_extract.py

@@ -1,5 +1,4 @@
 import logging
-from typing import Type
 
 import httpx
 from pydantic import BaseModel, Field
@@ -42,7 +41,7 @@ class TavilyExtract(TavilyBaseTool):
         "This tool is useful when you need to get the full text content from a webpage. "
         "You must call this tool whenever the user asks to extract or scrape content from a specific URL."
     )
-    args_schema: Type[BaseModel] = TavilyExtractInput
+    args_schema: type[BaseModel] = TavilyExtractInput
 
     async def _arun(
         self,
@@ -73,9 +72,8 @@ class TavilyExtract(TavilyBaseTool):
                 "rate_limit_minutes"
             ):
                 await self.user_rate_limit_by_category(
-                    context.user_id,
                     skill_config["rate_limit_number"],
-                    skill_config["rate_limit_minutes"],
+                    skill_config["rate_limit_minutes"] * 60,
                 )
 
         # Get the API key from the agent's configuration

intentkit/skills/tavily/tavily_search.py

@@ -1,9 +1,9 @@
 import logging
-from typing import Type
 
 import httpx
 from pydantic import BaseModel, Field
 
+from intentkit.config.config import config
 from intentkit.skills.tavily.base import TavilyBaseTool
 
 logger = logging.getLogger(__name__)
@@ -49,7 +49,7 @@ class TavilySearch(TavilyBaseTool):
         "You must call this tool whenever the user asks for information that may not be in your training data,"
         " requires current data, or when you're unsure about facts."
     )
-    args_schema: Type[BaseModel] = TavilySearchInput
+    args_schema: type[BaseModel] = TavilySearchInput
 
     async def _arun(
         self,
@@ -67,7 +67,6 @@ class TavilySearch(TavilyBaseTool):
             include_images: Whether to include image URLs in the results.
             include_raw_content: Whether to include raw HTML content in the results.
 
-
         Returns:
             str: Formatted search results with titles, snippets, and URLs.
         """
@@ -80,16 +79,15 @@ class TavilySearch(TavilyBaseTool):
                 "rate_limit_minutes"
             ):
                 await self.user_rate_limit_by_category(
-                    context.user_id,
                     skill_config["rate_limit_number"],
-                    skill_config["rate_limit_minutes"],
+                    skill_config["rate_limit_minutes"] * 60,
                 )
 
         # Get the API key from the agent's configuration
         if skill_config.get("api_key_provider") == "agent_owner":
             api_key = skill_config.get("api_key")
         else:
-            api_key = self.skill_store.get_system_config("tavily_api_key")
+            api_key = config.tavily_api_key
         if not api_key:
             return "Error: No Tavily API key provided in the configuration."
 

intentkit/skills/token/__init__.py

@@ -3,7 +3,6 @@
 import logging
 from typing import TypedDict
 
-from intentkit.abstracts.skill import SkillStoreABC
 from intentkit.skills.base import SkillConfig, SkillState
 from intentkit.skills.token.base import TokenBaseTool
 from intentkit.skills.token.erc20_transfers import ERC20Transfers
@@ -36,7 +35,6 @@ class Config(SkillConfig):
 async def get_skills(
     config: "Config",
     is_private: bool,
-    store: SkillStoreABC,
     **_,
 ) -> list[TokenBaseTool]:
     """Get all Token blockchain analysis skills.
@@ -44,7 +42,6 @@ async def get_skills(
     Args:
         config: The configuration for Token skills.
         is_private: Whether to include private skills.
-        store: The skill store for persisting data.
 
     Returns:
         A list of Token blockchain analysis skills.
@@ -65,7 +62,7 @@ async def get_skills(
     # Get each skill using the cached getter
     result = []
     for name in available_skills:
-        skill = get_token_skill(name, store)
+        skill = get_token_skill(name)
         if skill:
             result.append(skill)
 
@@ -74,13 +71,11 @@ async def get_skills(
 
 def get_token_skill(
     name: str,
-    store: SkillStoreABC,
 ) -> TokenBaseTool:
     """Get a Token blockchain analysis skill by name.
 
     Args:
         name: The name of the skill to get
-        store: The skill store for persisting data
 
     Returns:
         The requested Token blockchain analysis skill
@@ -90,13 +85,13 @@ def get_token_skill(
 
     skill = None
     if name == "token_price":
-        skill = TokenPrice(skill_store=store)
+        skill = TokenPrice()
     elif name == "token_erc20_transfers":
-        skill = ERC20Transfers(skill_store=store)
+        skill = ERC20Transfers()
     elif name == "token_search":
-        skill = TokenSearch(skill_store=store)
+        skill = TokenSearch()
     elif name == "token_analytics":
-        skill = TokenAnalytics(skill_store=store)
+        skill = TokenAnalytics()
     else:
         logger.warning(f"Unknown Token skill: {name}")
         return None

intentkit/skills/token/base.py

@@ -1,11 +1,11 @@
 """Base class for token-related skills."""
 
 import logging
-from typing import Any, Dict
+from typing import Any
 
 import aiohttp
 
-from intentkit.abstracts.skill import SkillStoreABC
+from intentkit.config.config import config
 from intentkit.skills.base import IntentKitSkill
 from intentkit.skills.token.constants import MORALIS_API_BASE_URL
 
@@ -19,10 +19,6 @@ class TokenBaseTool(IntentKitSkill):
     including making HTTP requests to the Moralis API.
     """
 
-    def __init__(self, skill_store: SkillStoreABC = None):
-        """Initialize the token tool with a skill store."""
-        super().__init__(skill_store=skill_store)
-
     @property
     def category(self) -> str:
         return "token"
@@ -37,9 +33,9 @@ class TokenBaseTool(IntentKitSkill):
         skill_config = context.agent.skill_config(self.category)
         if skill_config.get("api_key_provider") == "agent_owner":
             return skill_config.get("api_key")
-        return self.skill_store.get_system_config("moralis_api_key")
+        return config.moralis_api_key
 
-    def _prepare_params(self, params: Dict[str, Any]) -> Dict[str, Any]:
+    def _prepare_params(self, params: dict[str, Any]) -> dict[str, Any]:
         """Convert boolean values to lowercase strings for API compatibility.
 
         Args:
@@ -64,9 +60,9 @@ class TokenBaseTool(IntentKitSkill):
         method: str,
         endpoint: str,
         api_key: str,
-        params: Dict[str, Any] = None,
-        data: Dict[str, Any] = None,
-    ) -> Dict[str, Any]:
+        params: dict[str, Any] = None,
+        data: dict[str, Any] = None,
+    ) -> dict[str, Any]:
         """Make a request to the Moralis API.
 
         Args:

intentkit/skills/token/erc20_transfers.py

@@ -1,5 +1,5 @@
 import logging
-from typing import Any, Dict, List, Optional, Type
+from typing import Any
 
 from pydantic import BaseModel, Field
 
@@ -19,35 +19,35 @@ class ERC20TransfersInput(BaseModel):
         description="The chain to query (e.g., 'eth', 'bsc', 'polygon').",
         default=DEFAULT_CHAIN,
     )
-    contract_addresses: Optional[List[str]] = Field(
+    contract_addresses: list[str] | None = Field(
         description="List of contract addresses of transfers to filter by.",
         default=None,
     )
-    from_block: Optional[int] = Field(
+    from_block: int | None = Field(
         description="The minimum block number from which to get the transactions.",
         default=None,
     )
-    to_block: Optional[int] = Field(
+    to_block: int | None = Field(
         description="The maximum block number from which to get the transactions.",
         default=None,
     )
-    from_date: Optional[str] = Field(
+    from_date: str | None = Field(
         description="The start date from which to get the transactions (any format accepted by momentjs).",
         default=None,
     )
-    to_date: Optional[str] = Field(
+    to_date: str | None = Field(
         description="Get the transactions up to this date (any format accepted by momentjs).",
         default=None,
     )
-    limit: Optional[int] = Field(
+    limit: int | None = Field(
         description="The desired page size of the result.",
         default=DEFAULT_LIMIT,
     )
-    order: Optional[str] = Field(
+    order: str | None = Field(
         description="The order of the result, in ascending (ASC) or descending (DESC).",
         default=DEFAULT_ORDER,
     )
-    cursor: Optional[str] = Field(
+    cursor: str | None = Field(
         description="The cursor returned in the previous response (for pagination).",
         default=None,
     )
@@ -65,22 +65,22 @@ class ERC20Transfers(TokenBaseTool):
         "Get ERC20 token transactions for a wallet address, ordered by block number. "
         "Returns transaction details, token information, and wallet interactions."
     )
-    args_schema: Type[BaseModel] = ERC20TransfersInput
+    args_schema: type[BaseModel] = ERC20TransfersInput
 
     async def _arun(
         self,
         address: str,
         chain: str = DEFAULT_CHAIN,
-        contract_addresses: Optional[List[str]] = None,
-        from_block: Optional[int] = None,
-        to_block: Optional[int] = None,
-        from_date: Optional[str] = None,
-        to_date: Optional[str] = None,
-        limit: Optional[int] = DEFAULT_LIMIT,
-        order: Optional[str] = DEFAULT_ORDER,
-        cursor: Optional[str] = None,
+        contract_addresses: list[str] | None = None,
+        from_block: int | None = None,
+        to_block: int | None = None,
+        from_date: str | None = None,
+        to_date: str | None = None,
+        limit: int | None = DEFAULT_LIMIT,
+        order: str | None = DEFAULT_ORDER,
+        cursor: str | None = None,
         **kwargs,
-    ) -> Dict[str, Any]:
+    ) -> dict[str, Any]:
         """Fetch ERC20 token transfers for a wallet from Moralis.
 
         Args:

intentkit/skills/token/schema.json

@@ -5,11 +5,8 @@
   "type": "object",
   "x-icon": "https://ai.service.crestal.dev/skills/portfolio/moralis.png",
   "x-tags": [
-    "Blockchain",
-    "Web3",
-    "Crypto",
-    "Token",
-    "DeFi"
+    "Analytics",
+    "Crypto"
   ],
   "properties": {
     "enabled": {
@@ -138,4 +135,4 @@
     }
   },
   "additionalProperties": true
-}
+}

intentkit/skills/token/token_analytics.py

@@ -1,5 +1,5 @@
 import logging
-from typing import Any, Dict, Type
+from typing import Any
 
 from pydantic import BaseModel, Field
 
@@ -31,14 +31,14 @@ class TokenAnalytics(TokenBaseTool):
         "Get analytics for a token by token address. "
         "Returns trading volumes, number of buyers/sellers, and liquidity information over various time periods."
     )
-    args_schema: Type[BaseModel] = TokenAnalyticsInput
+    args_schema: type[BaseModel] = TokenAnalyticsInput
 
     async def _arun(
         self,
         address: str,
         chain: str = DEFAULT_CHAIN,
         **kwargs,
-    ) -> Dict[str, Any]:
+    ) -> dict[str, Any]:
         """Fetch token analytics from Moralis.
 
         Args:

intentkit/skills/token/token_price.py

@@ -1,5 +1,5 @@
 import logging
-from typing import Any, Dict, Optional, Type
+from typing import Any
 
 from pydantic import BaseModel, Field
 
@@ -19,23 +19,23 @@ class TokenPriceInput(BaseModel):
         description="The chain to query (e.g., 'eth', 'bsc', 'polygon').",
         default=DEFAULT_CHAIN,
     )
-    include: Optional[str] = Field(
+    include: str | None = Field(
         description="If the result should contain the 24hr percent change (use 'percent_change').",
         default=None,
     )
-    exchange: Optional[str] = Field(
+    exchange: str | None = Field(
         description="The factory name or address of the token exchange.",
         default=None,
     )
-    to_block: Optional[int] = Field(
+    to_block: int | None = Field(
         description="The block number from which the token price should be checked.",
         default=None,
     )
-    max_token_inactivity: Optional[int] = Field(
+    max_token_inactivity: int | None = Field(
         description="Exclude tokens inactive for more than the given amount of days.",
         default=None,
     )
-    min_pair_side_liquidity_usd: Optional[int] = Field(
+    min_pair_side_liquidity_usd: int | None = Field(
         description="Exclude tokens with liquidity less than the specified amount in USD.",
         default=None,
     )
@@ -53,19 +53,19 @@ class TokenPrice(TokenBaseTool):
         "Get the token price denominated in the blockchain's native token and USD for a given token contract address. "
         "Returns price, token information and exchange data."
     )
-    args_schema: Type[BaseModel] = TokenPriceInput
+    args_schema: type[BaseModel] = TokenPriceInput
 
     async def _arun(
         self,
         address: str,
         chain: str = DEFAULT_CHAIN,
-        include: Optional[str] = None,
-        exchange: Optional[str] = None,
-        to_block: Optional[int] = None,
-        max_token_inactivity: Optional[int] = None,
-        min_pair_side_liquidity_usd: Optional[int] = None,
+        include: str | None = None,
+        exchange: str | None = None,
+        to_block: int | None = None,
+        max_token_inactivity: int | None = None,
+        min_pair_side_liquidity_usd: int | None = None,
         **kwargs,
-    ) -> Dict[str, Any]:
+    ) -> dict[str, Any]:
         """Fetch token price from Moralis.
 
         Args: