intentkit 0.8.12rc0__py3-none-any.whl → 0.8.13.dev1__py3-none-any.whl

intentkit/skills/slack/__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.slack.base import SlackBaseTool
 from intentkit.skills.slack.get_channel import SlackGetChannel
@@ -34,7 +33,6 @@ class Config(SkillConfig):
 async def get_skills(
     config: "Config",
     is_private: bool,
-    store: SkillStoreABC,
     **_,
 ) -> list[SlackBaseTool]:
     """Get all Slack skills."""
@@ -50,7 +48,7 @@ async def get_skills(
     # Get each skill using the cached getter
     result = []
     for name in available_skills:
-        skill = get_slack_skill(name, store)
+        skill = get_slack_skill(name)
         if skill:
             result.append(skill)
     return result
@@ -58,40 +56,30 @@ async def get_skills(
 
 def get_slack_skill(
     name: str,
-    store: SkillStoreABC,
 ) -> SlackBaseTool:
     """Get a Slack skill by name.
 
     Args:
         name: The name of the skill to get
-        store: The skill store for persisting data
 
     Returns:
         The requested Slack skill
     """
     if name == "get_channel":
         if name not in _cache:
-            _cache[name] = SlackGetChannel(
-                skill_store=store,
-            )
+            _cache[name] = SlackGetChannel()
         return _cache[name]
     elif name == "get_message":
         if name not in _cache:
-            _cache[name] = SlackGetMessage(
-                skill_store=store,
-            )
+            _cache[name] = SlackGetMessage()
         return _cache[name]
     elif name == "schedule_message":
         if name not in _cache:
-            _cache[name] = SlackScheduleMessage(
-                skill_store=store,
-            )
+            _cache[name] = SlackScheduleMessage()
         return _cache[name]
     elif name == "send_message":
         if name not in _cache:
-            _cache[name] = SlackSendMessage(
-                skill_store=store,
-            )
+            _cache[name] = SlackSendMessage()
         return _cache[name]
     else:
         logger.warning(f"Unknown Slack skill: {name}")

intentkit/skills/slack/base.py

@@ -4,7 +4,6 @@ from langchain_core.tools.base import ToolException
 from pydantic import BaseModel, Field
 from slack_sdk import WebClient
 
-from intentkit.abstracts.skill import SkillStoreABC
 from intentkit.skills.base import IntentKitSkill
 
 
@@ -14,9 +13,6 @@ class SlackBaseTool(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"
-    )
 
     def get_api_key(self) -> str:
         context = self.get_context()

intentkit/skills/supabase/__init__.py

@@ -3,7 +3,6 @@
 import logging
 from typing import NotRequired, TypedDict
 
-from intentkit.abstracts.skill import SkillStoreABC
 from intentkit.skills.base import SkillConfig, SkillState
 from intentkit.skills.supabase.base import SupabaseBaseTool
 from intentkit.skills.supabase.delete_data import SupabaseDeleteData
@@ -41,7 +40,6 @@ class Config(SkillConfig):
 async def get_skills(
     config: "Config",
     is_private: bool,
-    store: SkillStoreABC,
     **_,
 ) -> list[SupabaseBaseTool]:
     """Get all Supabase skills."""
@@ -57,7 +55,7 @@ async def get_skills(
     # Get each skill using the cached getter
     result = []
     for name in available_skills:
-        skill = get_supabase_skill(name, store)
+        skill = get_supabase_skill(name)
         if skill:
             result.append(skill)
     return result
@@ -65,52 +63,38 @@ async def get_skills(
 
 def get_supabase_skill(
     name: str,
-    store: SkillStoreABC,
 ) -> SupabaseBaseTool:
     """Get a Supabase skill by name.
 
     Args:
         name: The name of the skill to get
-        store: The skill store for persisting data
 
     Returns:
         The requested Supabase skill
     """
     if name == "fetch_data":
         if name not in _cache:
-            _cache[name] = SupabaseFetchData(
-                skill_store=store,
-            )
+            _cache[name] = SupabaseFetchData()
         return _cache[name]
     elif name == "insert_data":
         if name not in _cache:
-            _cache[name] = SupabaseInsertData(
-                skill_store=store,
-            )
+            _cache[name] = SupabaseInsertData()
         return _cache[name]
     elif name == "update_data":
         if name not in _cache:
-            _cache[name] = SupabaseUpdateData(
-                skill_store=store,
-            )
+            _cache[name] = SupabaseUpdateData()
         return _cache[name]
     elif name == "upsert_data":
         if name not in _cache:
-            _cache[name] = SupabaseUpsertData(
-                skill_store=store,
-            )
+            _cache[name] = SupabaseUpsertData()
         return _cache[name]
     elif name == "delete_data":
         if name not in _cache:
-            _cache[name] = SupabaseDeleteData(
-                skill_store=store,
-            )
+            _cache[name] = SupabaseDeleteData()
         return _cache[name]
     elif name == "invoke_function":
         if name not in _cache:
-            _cache[name] = SupabaseInvokeFunction(
-                skill_store=store,
-            )
+            _cache[name] = SupabaseInvokeFunction()
         return _cache[name]
     else:
         logger.warning(f"Unknown Supabase skill: {name}")

intentkit/skills/supabase/base.py

@@ -4,7 +4,6 @@ from langchain_core.tools import ToolException
 from pydantic import BaseModel, Field
 
 from intentkit.abstracts.graph import AgentContext
-from intentkit.abstracts.skill import SkillStoreABC
 from intentkit.skills.base import IntentKitSkill
 
 
@@ -14,9 +13,6 @@ class SupabaseBaseTool(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"
-    )
 
     @property
     def category(self) -> str:

intentkit/skills/superfluid/__init__.py

@@ -4,7 +4,6 @@ from typing import TYPE_CHECKING, Optional, TypedDict
 
 from coinbase_agentkit import superfluid_action_provider
 
-from intentkit.abstracts.skill import SkillStoreABC
 from intentkit.skills.base import (
     SkillConfig,
     SkillState,
@@ -32,7 +31,6 @@ class Config(SkillConfig):
 async def get_skills(
     config: "Config",
     is_private: bool,
-    store: SkillStoreABC,
     agent_id: str,
     agent: Optional["Agent"] = None,
     **_,
@@ -47,7 +45,7 @@ async def get_skills(
             available_skills.append(skill_name)
 
     actions = await get_agentkit_actions(
-        agent_id, store, [superfluid_action_provider], agent=agent
+        agent_id, [superfluid_action_provider], agent=agent
     )
     tools: list[SuperfluidBaseTool] = []
     for skill in available_skills:

intentkit/skills/system/__init__.py

@@ -3,7 +3,6 @@
 import logging
 from typing import TypedDict
 
-from intentkit.abstracts.skill import SkillStoreABC
 from intentkit.skills.base import SkillConfig, SkillOwnerState
 from intentkit.skills.system.add_autonomous_task import AddAutonomousTask
 from intentkit.skills.system.base import SystemBaseTool
@@ -37,7 +36,6 @@ class Config(SkillConfig):
 async def get_skills(
     config: "Config",
     is_private: bool,
-    store: SkillStoreABC,
     **_,
 ) -> list[SystemBaseTool]:
     """Get all system skills.
@@ -45,7 +43,6 @@ async def get_skills(
     Args:
         config: The configuration for system skills.
         is_private: Whether to include private skills.
-        store: The skill store for persisting data.
 
     Returns:
         A list of system skills.
@@ -62,7 +59,7 @@ async def get_skills(
     # Get each skill using the cached getter
     result = []
     for name in available_skills:
-        skill = get_system_skill(name, store)
+        skill = get_system_skill(name)
         if skill:
             result.append(skill)
     return result
@@ -70,52 +67,38 @@ async def get_skills(
 
 def get_system_skill(
     name: str,
-    store: SkillStoreABC,
 ) -> SystemBaseTool:
     """Get a system skill by name.
 
     Args:
         name: The name of the skill to get
-        store: The skill store for persisting data
 
     Returns:
         The requested system skill
     """
     if name == "read_agent_api_key":
         if name not in _cache:
-            _cache[name] = ReadAgentApiKey(
-                skill_store=store,
-            )
+            _cache[name] = ReadAgentApiKey()
         return _cache[name]
     elif name == "regenerate_agent_api_key":
         if name not in _cache:
-            _cache[name] = RegenerateAgentApiKey(
-                skill_store=store,
-            )
+            _cache[name] = RegenerateAgentApiKey()
         return _cache[name]
     elif name == "list_autonomous_tasks":
         if name not in _cache:
-            _cache[name] = ListAutonomousTasks(
-                skill_store=store,
-            )
+            _cache[name] = ListAutonomousTasks()
         return _cache[name]
     elif name == "add_autonomous_task":
         if name not in _cache:
-            _cache[name] = AddAutonomousTask(
-                skill_store=store,
-            )
+            _cache[name] = AddAutonomousTask()
         return _cache[name]
     elif name == "delete_autonomous_task":
         if name not in _cache:
-            _cache[name] = DeleteAutonomousTask(
-                skill_store=store,
-            )
+            _cache[name] = DeleteAutonomousTask()
         return _cache[name]
     elif name == "edit_autonomous_task":
         if name not in _cache:
-            _cache[name] = EditAutonomousTask(
-                skill_store=store,
-            )
+            _cache[name] = EditAutonomousTask()
         return _cache[name]
     else:
         logger.warning(f"Unknown system skill: {name}")

intentkit/skills/system/add_autonomous_task.py

@@ -75,7 +75,7 @@ class AddAutonomousTask(SystemBaseTool):
             AddAutonomousTaskOutput: The created task
         """
         context = self.get_context()
-        agent_id = context.agent_id
+        agent = context.agent
 
         task = AgentAutonomous(
             name=name,
@@ -86,6 +86,6 @@ class AddAutonomousTask(SystemBaseTool):
             enabled=True,
         )
 
-        created_task = await self.skill_store.add_autonomous_task(agent_id, task)
+        created_task = await agent.add_autonomous_task(task)
 
         return AddAutonomousTaskOutput(task=created_task)

intentkit/skills/system/delete_autonomous_task.py

@@ -45,9 +45,9 @@ class DeleteAutonomousTask(SystemBaseTool):
             DeleteAutonomousTaskOutput: Confirmation of deletion
         """
         context = self.get_context()
-        agent_id = context.agent_id
+        agent = context.agent
 
-        await self.skill_store.delete_autonomous_task(agent_id, task_id)
+        await agent.delete_autonomous_task(task_id)
 
         return DeleteAutonomousTaskOutput(
             success=True, message=f"Successfully deleted autonomous task {task_id}"

intentkit/skills/system/edit_autonomous_task.py

@@ -85,7 +85,7 @@ class EditAutonomousTask(SystemBaseTool):
             EditAutonomousTaskOutput: The updated task
         """
         context = self.get_context()
-        agent_id = context.agent_id
+        agent = context.agent
 
         if minutes is not None and cron is not None:
             raise ValueError("minutes and cron are mutually exclusive")
@@ -107,8 +107,6 @@ class EditAutonomousTask(SystemBaseTool):
         if enabled is not None:
             task_updates["enabled"] = enabled
 
-        updated_task = await self.skill_store.update_autonomous_task(
-            agent_id, task_id, task_updates
-        )
+        updated_task = await agent.update_autonomous_task(task_id, task_updates)
 
         return EditAutonomousTaskOutput(task=updated_task)

intentkit/skills/system/list_autonomous_tasks.py

@@ -43,8 +43,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/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

@@ -3,7 +3,7 @@ from typing import Type
 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
 
 
@@ -13,16 +13,13 @@ 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"
-    )
 
     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/tavily_extract.py

@@ -73,9 +73,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

@@ -4,6 +4,7 @@ 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__)
@@ -80,16 +81,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

@@ -5,7 +5,7 @@ from typing import Any, Dict
 
 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,7 +33,7 @@ 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]:
         """Convert boolean values to lowercase strings for API compatibility.