intentkit 0.8.6.dev2__py3-none-any.whl → 0.8.17__py3-none-any.whl

intentkit/skills/aixbt/__init__.py

@@ -1,6 +1,5 @@
 from typing import TypedDict
 
-from intentkit.abstracts.skill import SkillStoreABC
 from intentkit.skills.aixbt.base import AIXBTBaseTool
 from intentkit.skills.aixbt.projects import AIXBTProjects
 from intentkit.skills.base import SkillConfig, SkillState
@@ -27,7 +26,6 @@ class Config(SkillConfig):
 async def get_skills(
     config: "Config",
     is_private: bool,
-    store: SkillStoreABC,
     **_,
 ) -> list[AIXBTBaseTool]:
     """Get all AIXBT API skills."""
@@ -44,26 +42,17 @@ async def get_skills(
             available_skills.append(skill_name)
 
     # Get each skill using the cached getter
-    return [
-        get_aixbt_skill(
-            name=name,
-            store=store,
-        )
-        for name in available_skills
-    ]
+    return [get_aixbt_skill(name) for name in available_skills]
 
 
 def get_aixbt_skill(
     name: str,
-    store: SkillStoreABC,
 ) -> AIXBTBaseTool:
     """Get an AIXBT API skill by name."""
 
     if name == "aixbt_projects":
         if name not in _cache:
-            _cache[name] = AIXBTProjects(
-                skill_store=store,
-            )
+            _cache[name] = AIXBTProjects()
         return _cache[name]
     else:
         raise ValueError(f"Unknown AIXBT skill: {name}")

intentkit/skills/aixbt/base.py

@@ -1,8 +1,5 @@
-from typing import Type
-
 from pydantic import BaseModel, Field
 
-from intentkit.abstracts.skill import SkillStoreABC
 from intentkit.skills.base import IntentKitSkill
 
 
@@ -11,10 +8,7 @@ class AIXBTBaseTool(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]
 
     @property
     def category(self) -> str:

intentkit/skills/aixbt/projects.py

@@ -1,5 +1,5 @@
 import logging
-from typing import Any, Dict, Optional, Type
+from typing import Any
 
 import httpx
 from langchain_core.tools import ToolException
@@ -17,23 +17,23 @@ class ProjectsInput(BaseModel):
         default=10,
         description="Number of projects to return (max 50)",
     )
-    name: Optional[str] = Field(
+    name: str | None = Field(
         default=None,
         description="Filter projects by name (case-insensitive regex match)",
     )
-    ticker: Optional[str] = Field(
+    ticker: str | None = Field(
         default=None,
         description="Filter projects by ticker symbol (case-insensitive match)",
     )
-    xHandle: Optional[str] = Field(
+    xHandle: str | None = Field(
         default=None,
         description="Filter projects by X/Twitter handle",
     )
-    minScore: Optional[float] = Field(
+    minScore: float | None = Field(
         default=None,
         description="Minimum score threshold",
     )
-    chain: Optional[str] = Field(
+    chain: str | None = Field(
         default=None,
         description="Filter projects by blockchain",
     )
@@ -53,18 +53,18 @@ class AIXBTProjects(AIXBTBaseTool):
         "detailed information on recent developments. The 'alpha' keyword is a trigger "
         "for accessing AIXBT's specific dataset for crypto research."
     )
-    args_schema: Type[BaseModel] = ProjectsInput
+    args_schema: type[BaseModel] = ProjectsInput
 
     async def _arun(
         self,
         limit: int = 10,
-        name: Optional[str] = None,
-        ticker: Optional[str] = None,
-        xHandle: Optional[str] = None,
-        minScore: Optional[float] = None,
-        chain: Optional[str] = None,
+        name: str | None = None,
+        ticker: str | None = None,
+        xHandle: str | None = None,
+        minScore: float | None = None,
+        chain: str | None = None,
         **kwargs,
-    ) -> Dict[str, Any]:
+    ) -> dict[str, Any]:
         """
         Search for cryptocurrency projects using AIXBT API.
 
@@ -89,9 +89,8 @@ class AIXBTProjects(AIXBTBaseTool):
             "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/allora/__init__.py

@@ -3,7 +3,6 @@
 import logging
 from typing import NotRequired, TypedDict
 
-from intentkit.abstracts.skill import SkillStoreABC
 from intentkit.skills.allora.base import AlloraBaseTool
 from intentkit.skills.allora.price import AlloraGetPrice
 from intentkit.skills.base import SkillConfig, SkillState
@@ -28,7 +27,6 @@ class Config(SkillConfig):
 async def get_skills(
     config: "Config",
     is_private: bool,
-    store: SkillStoreABC,
     **_,
 ) -> list[AlloraBaseTool]:
     """Get all Allora skills.
@@ -36,7 +34,6 @@ async def get_skills(
     Args:
         config: The configuration for Allora skills.
         is_private: Whether to include private skills.
-        store: The skill store for persisting data.
 
     Returns:
         A list of Allora skills.
@@ -53,7 +50,7 @@ async def get_skills(
     # Get each skill using the cached getter
     result = []
     for name in available_skills:
-        skill = get_allora_skill(name, store)
+        skill = get_allora_skill(name)
         if skill:
             result.append(skill)
     return result
@@ -61,22 +58,18 @@ async def get_skills(
 
 def get_allora_skill(
     name: str,
-    store: SkillStoreABC,
 ) -> AlloraBaseTool:
     """Get an Allora skill by name.
 
     Args:
         name: The name of the skill to get
-        store: The skill store for persisting data
 
     Returns:
         The requested Allora skill
     """
     if name == "get_price_prediction":
         if name not in _cache:
-            _cache[name] = AlloraGetPrice(
-                skill_store=store,
-            )
+            _cache[name] = AlloraGetPrice()
         return _cache[name]
     else:
         logger.warning(f"Unknown Allora skill: {name}")

intentkit/skills/allora/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
 
 base_url = "https://api.upshot.xyz/v2/allora"
@@ -14,17 +12,14 @@ class AlloraBaseTool(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("allora_api_key")
+            return config.allora_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/allora/price.py

@@ -1,7 +1,7 @@
-from typing import Literal, Type
+from typing import Literal
 
 import httpx
-from langchain.tools.base import ToolException
+from langchain_core.tools.base import ToolException
 from pydantic import BaseModel, Field
 
 from intentkit.skills.allora.base import AlloraBaseTool
@@ -57,7 +57,6 @@ class AlloraGetPrice(AlloraBaseTool):
     The Allora Price Prediction Feed tool fetches the price prediction feed from the Allora API.
     Ethereum (ETH) or Bitcoin (BTC) price predictions (5-minute, 8-hour)
 
-
     Attributes:
         name (str): Name of the tool, specifically "get_price_prediction".
         description (str): Comprehensive description of the tool's purpose and functionality.
@@ -69,7 +68,7 @@ class AlloraGetPrice(AlloraBaseTool):
         The Allora Price Prediction Feed tool fetches the price prediction feed from the Allora API.
         Ethereum (ETH) or Bitcoin (BTC) price predictions (5-minute, 8-hour)
         """
-    args_schema: Type[BaseModel] = AlloraGetPriceInput
+    args_schema: type[BaseModel] = AlloraGetPriceInput
 
     def _run(self, question: str) -> AlloraGetPriceOutput:
         """Run the tool to get the token price prediction from Allora API.

intentkit/skills/base.py

@@ -1,15 +1,10 @@
 import logging
-from collections.abc import Sequence
+from collections.abc import Callable, Sequence
 from typing import (
-    TYPE_CHECKING,
     Any,
-    Callable,
-    Dict,
     Literal,
     NotRequired,
-    Optional,
     TypedDict,
-    Union,
 )
 
 from coinbase_agentkit import (
@@ -18,8 +13,7 @@ from coinbase_agentkit import (
     AgentKitConfig,
     CdpEvmWalletProvider,
 )
-from langchain.tools import StructuredTool
-from langchain_core.tools import BaseTool
+from langchain_core.tools import BaseTool, StructuredTool
 from langchain_core.tools.base import ToolException
 from langgraph.runtime import get_runtime
 from pydantic import (
@@ -27,18 +21,19 @@ from pydantic import (
 )
 from pydantic.v1 import ValidationError as ValidationErrorV1
 from redis.exceptions import RedisError
-from web3 import Web3
 
 from intentkit.abstracts.graph import AgentContext
-from intentkit.abstracts.skill import SkillStoreABC
 from intentkit.clients import get_wallet_provider
-from intentkit.clients.web3 import get_web3_client
+from intentkit.models.agent import Agent
 from intentkit.models.redis import get_redis
+from intentkit.models.skill import (
+    AgentSkillData,
+    AgentSkillDataCreate,
+    ChatSkillData,
+    ChatSkillDataCreate,
+)
 from intentkit.utils.error import IntentKitAPIError, RateLimitExceeded
 
-if TYPE_CHECKING:
-    from intentkit.models.agent import Agent
-
 SkillState = Literal["disabled", "public", "private"]
 SkillOwnerState = Literal["disabled", "private"]
 APIKeyProviderValue = Literal["platform", "agent_owner"]
@@ -48,9 +43,9 @@ class SkillConfig(TypedDict):
     """Abstract base class for skill configuration."""
 
     enabled: bool
-    states: Dict[str, SkillState | SkillOwnerState]
+    states: dict[str, SkillState | SkillOwnerState]
     api_key_provider: NotRequired[APIKeyProviderValue]
-    __extra__: NotRequired[Dict[str, Any]]
+    __extra__: NotRequired[dict[str, Any]]
 
 
 class IntentKitSkill(BaseTool):
@@ -58,17 +53,16 @@ class IntentKitSkill(BaseTool):
     Will have predefined abilities.
     """
 
-    skill_store: SkillStoreABC
     # overwrite the value of BaseTool
-    handle_tool_error: Optional[Union[bool, str, Callable[[ToolException], str]]] = (
+    handle_tool_error: bool | str | Callable[[ToolException], str] | None = (
         lambda e: f"tool error: {e}"
     )
     """Handle the content of the ToolException thrown."""
 
     # overwrite the value of BaseTool
-    handle_validation_error: Optional[
-        Union[bool, str, Callable[[Union[ValidationError, ValidationErrorV1]], str]]
-    ] = lambda e: f"validation error: {e}"
+    handle_validation_error: (
+        bool | str | Callable[[ValidationError | ValidationErrorV1], str] | None
+    ) = lambda e: f"validation error: {e}"
     """Handle the content of the ValidationError thrown."""
 
     # Logger for the class
@@ -79,15 +73,12 @@ class IntentKitSkill(BaseTool):
         """Get the category of the skill."""
         raise NotImplementedError
 
-    async def user_rate_limit(
-        self, user_id: str, limit: int, minutes: int, key: str
-    ) -> None:
+    async def user_rate_limit(self, limit: int, seconds: int, key: str) -> None:
         """Check if a user has exceeded the rate limit for this skill.
 
         Args:
-            user_id: The ID of the user to check
             limit: Maximum number of requests allowed
-            minutes: Time window in minutes
+            seconds: Time window in seconds
             key: The key to use for rate limiting (e.g., skill name or category)
 
         Raises:
@@ -96,25 +87,48 @@ class IntentKitSkill(BaseTool):
         Returns:
             None: Always returns None if no exception is raised
         """
-        if not user_id:
-            return None  # No rate limiting for users without ID
+        try:
+            context = self.get_context()
+        except ValueError:
+            self.logger.info(
+                "AgentContext not available, skipping rate limit for %s",
+                key,
+            )
+            return None
+
+        user_identifier = context.user_id or context.agent_id
+        if not user_identifier:
+            return None  # No rate limiting when no identifier is available
+
+        try:
+            max_requests = int(limit)
+            window_seconds = int(seconds)
+        except (TypeError, ValueError):
+            self.logger.info(
+                "Invalid user rate limit parameters for %s: limit=%r, seconds=%r",
+                key,
+                limit,
+                seconds,
+            )
+            return None
+
+        if window_seconds <= 0 or max_requests <= 0:
+            return None
 
         try:
             redis = get_redis()
             # Create a unique key for this rate limit and user
-            rate_limit_key = f"rate_limit:{key}:{user_id}"
+            rate_limit_key = f"rate_limit:{key}:{user_identifier}"
 
             # Get the current count
             count = await redis.incr(rate_limit_key)
 
             # Set expiration if this is the first request
             if count == 1:
-                await redis.expire(
-                    rate_limit_key, minutes * 60
-                )  # Convert minutes to seconds
+                await redis.expire(rate_limit_key, window_seconds)
 
             # Check if user has exceeded the limit
-            if count > limit:
+            if count > max_requests:
                 raise RateLimitExceeded(f"Rate limit exceeded for {key}")
 
             return None
@@ -130,40 +144,97 @@ class IntentKitSkill(BaseTool):
             )
             return None
 
-    async def user_rate_limit_by_skill(
-        self, user_id: str, limit: int, minutes: int
-    ) -> None:
+    async def user_rate_limit_by_skill(self, limit: int, seconds: int) -> None:
         """Check if a user has exceeded the rate limit for this specific skill.
 
         This uses the skill name as the rate limit key.
 
         Args:
-            user_id: The ID of the user to check
             limit: Maximum number of requests allowed
-            minutes: Time window in minutes
+            seconds: Time window in seconds
 
         Raises:
             RateLimitExceeded: If the user has exceeded the rate limit
         """
-        return await self.user_rate_limit(user_id, limit, minutes, self.name)
+        return await self.user_rate_limit(limit, seconds, self.name)
 
-    async def user_rate_limit_by_category(
-        self, user_id: str, limit: int, minutes: int
-    ) -> None:
+    async def user_rate_limit_by_category(self, limit: int, seconds: int) -> None:
         """Check if a user has exceeded the rate limit for this skill category.
 
         This uses the skill category as the rate limit key, which means the limit
         is shared across all skills in the same category.
 
         Args:
-            user_id: The ID of the user to check
             limit: Maximum number of requests allowed
-            minutes: Time window in minutes
+            seconds: Time window in seconds
 
         Raises:
             RateLimitExceeded: If the user has exceeded the rate limit
         """
-        return await self.user_rate_limit(user_id, limit, minutes, self.category)
+        return await self.user_rate_limit(limit, seconds, self.category)
+
+    async def global_rate_limit(self, limit: int, seconds: int, key: str) -> None:
+        """Check if a global rate limit has been exceeded for a given key.
+
+        Args:
+            limit: Maximum number of requests allowed
+            seconds: Time window in seconds
+            key: The key to use for rate limiting (e.g., skill name or category)
+
+        Raises:
+            RateLimitExceeded: If the global limit has been exceeded
+
+        Returns:
+            None: Always returns None if no exception is raised
+        """
+        try:
+            max_requests = int(limit)
+            window_seconds = int(seconds)
+        except (TypeError, ValueError):
+            self.logger.info(
+                "Invalid global rate limit parameters for %s: limit=%r, seconds=%r",
+                key,
+                limit,
+                seconds,
+            )
+            return None
+
+        if window_seconds <= 0 or max_requests <= 0:
+            return None
+
+        try:
+            redis = get_redis()
+            rate_limit_key = f"rate_limit:{key}"
+
+            count = await redis.incr(rate_limit_key)
+
+            if count == 1:
+                await redis.expire(rate_limit_key, window_seconds)
+
+            if count > max_requests:
+                raise RateLimitExceeded(f"Global rate limit exceeded for {key}")
+
+            return None
+
+        except RuntimeError:
+            self.logger.info(
+                "Redis not initialized, skipping global rate limit for %s",
+                key,
+            )
+            return None
+        except RedisError as e:
+            self.logger.info(
+                f"Redis error in global rate limiting: {e}, skipping rate limit for {key}"
+            )
+            return None
+
+    async def global_rate_limit_by_skill(self, limit: int, seconds: int) -> None:
+        """Apply a global rate limit scoped to this specific skill."""
+        return await self.global_rate_limit(limit, seconds, self.name)
+
+    async def global_rate_limit_by_category(self, limit: int, seconds: int) -> None:
+        """Apply a global rate limit scoped to this skill category."""
+        return await self.global_rate_limit(limit, seconds, self.category)
 
     def _run(self, *args: Any, **kwargs: Any) -> Any:
         raise NotImplementedError(
@@ -177,21 +248,73 @@ class IntentKitSkill(BaseTool):
             raise ValueError("No AgentContext found")
         return runtime.context
 
-    def web3_client(self) -> Web3:
-        """Get a Web3 client for the skill."""
+    async def get_agent_skill_data(
+        self,
+        key: str,
+    ) -> dict[str, Any] | None:
+        """Retrieve persisted data for this skill scoped to the active agent."""
+        return await self.get_agent_skill_data_raw(self.name, key)
+
+    async def get_agent_skill_data_raw(
+        self,
+        skill_name: str,
+        key: str,
+    ) -> dict[str, Any] | None:
+        """Retrieve persisted data for a specific skill scoped to the active agent."""
         context = self.get_context()
-        agent = context.agent
-        network_id = agent.network_id
+        return await AgentSkillData.get(context.agent_id, skill_name, key)
+
+    async def save_agent_skill_data(self, key: str, data: dict[str, Any]) -> None:
+        """Persist data for this skill scoped to the active agent."""
+        await self.save_agent_skill_data_raw(self.name, key, data)
+
+    async def save_agent_skill_data_raw(
+        self,
+        skill_name: str,
+        key: str,
+        data: dict[str, Any],
+    ) -> None:
+        """Persist data for a specific skill scoped to the active agent."""
+        context = self.get_context()
+        skill_data = AgentSkillDataCreate(
+            agent_id=context.agent_id,
+            skill=skill_name,
+            key=key,
+            data=data,
+        )
+        await skill_data.save()
+
+    async def delete_agent_skill_data(self, key: str) -> None:
+        """Remove persisted data for this skill scoped to the active agent."""
+        context = self.get_context()
+        await AgentSkillData.delete(context.agent_id, self.name, key)
 
-        return get_web3_client(network_id)
+    async def get_thread_skill_data(
+        self,
+        key: str,
+    ) -> dict[str, Any] | None:
+        """Retrieve persisted data for this skill scoped to the active chat."""
+        context = self.get_context()
+        return await ChatSkillData.get(context.chat_id, self.name, key)
+
+    async def save_thread_skill_data(self, key: str, data: dict[str, Any]) -> None:
+        """Persist data for this skill scoped to the active chat."""
+        context = self.get_context()
+        skill_data = ChatSkillDataCreate(
+            chat_id=context.chat_id,
+            agent_id=context.agent_id,
+            skill=self.name,
+            key=key,
+            data=data,
+        )
+        await skill_data.save()
 
 
 async def get_agentkit_actions(
     agent_id: str,
-    _store: SkillStoreABC,
     provider_factories: Sequence[Callable[[], object]],
     *,
-    agent: Optional["Agent"] = None,
+    agent: Agent | None = None,
 ) -> list[Action]:
     """Build an AgentKit instance and return its actions."""
 

intentkit/skills/basename/__init__.py

@@ -1,10 +1,10 @@
 """Basename AgentKit skills."""
 
-from typing import TYPE_CHECKING, Optional, TypedDict
+from typing import TypedDict
 
 from coinbase_agentkit import basename_action_provider
 
-from intentkit.abstracts.skill import SkillStoreABC
+from intentkit.models.agent import Agent
 from intentkit.skills.base import (
     SkillConfig,
     SkillState,
@@ -13,9 +13,6 @@ from intentkit.skills.base import (
 )
 from intentkit.skills.basename.base import BasenameBaseTool
 
-if TYPE_CHECKING:
-    from intentkit.models.agent import Agent
-
 
 class SkillStates(TypedDict):
     BasenameActionProvider_register_basename: SkillState
@@ -30,9 +27,8 @@ class Config(SkillConfig):
 async def get_skills(
     config: "Config",
     is_private: bool,
-    store: SkillStoreABC,
     agent_id: str,
-    agent: Optional["Agent"] = None,
+    agent: Agent | None = None,
     **_,
 ) -> list[BasenameBaseTool]:
     """Get all Basename skills."""
@@ -45,7 +41,7 @@ async def get_skills(
             available_skills.append(skill_name)
 
     actions = await get_agentkit_actions(
-        agent_id, store, [basename_action_provider], agent=agent
+        agent_id, [basename_action_provider], agent=agent
     )
     tools: list[BasenameBaseTool] = []
     for skill in available_skills: