intentkit 0.8.11.dev1__py3-none-any.whl → 0.8.12__py3-none-any.whl

intentkit/skills/dexscreener/get_pair_info.py

@@ -50,10 +50,9 @@ class GetPairInfo(DexScreenerBaseTool):
         """Implementation to get specific pair information."""
 
         # Apply rate limiting
-        await self.user_rate_limit_by_category(
-            user_id=f"{self.category}{self.name}",
+        await self.global_rate_limit_by_skill(
             limit=RATE_LIMITS["pairs"],
-            minutes=1,
+            seconds=60,
         )
 
         logger.info(

intentkit/skills/dexscreener/get_token_pairs.py

@@ -55,10 +55,9 @@ class GetTokenPairs(DexScreenerBaseTool):
         """Implementation to get all pairs for a specific token."""
 
         # Apply rate limiting
-        await self.user_rate_limit_by_category(
-            user_id=f"{self.category}{self.name}",
+        await self.global_rate_limit_by_skill(
             limit=RATE_LIMITS["token_pairs"],
-            minutes=1,
+            seconds=60,
         )
 
         logger.info(

intentkit/skills/dexscreener/get_tokens_info.py

@@ -75,10 +75,9 @@ class GetTokensInfo(DexScreenerBaseTool):
         """Implementation to get information for multiple tokens."""
 
         # Apply rate limiting
-        await self.user_rate_limit_by_category(
-            user_id=f"{self.category}{self.name}",
+        await self.global_rate_limit_by_skill(
             limit=RATE_LIMITS["tokens"],
-            minutes=1,
+            seconds=60,
         )
 
         logger.info(

intentkit/skills/dexscreener/search_token.py

@@ -72,11 +72,9 @@ class SearchToken(DexScreenerBaseTool):
 
         # dexscreener 300 request per minute (across all user) based on dexscreener docs
         # https://docs.dexscreener.com/api/reference#get-latest-dex-search
-        await self.user_rate_limit_by_category(
-            # using hardcoded user_id to make sure it limit across all users
-            user_id=f"{self.category}{self.name}",
+        await self.global_rate_limit_by_skill(
             limit=300,
-            minutes=1,
+            seconds=60,
         )
 
         sort_by = sort_by or SortBy.LIQUIDITY

intentkit/skills/dune_analytics/__init__.py

@@ -6,7 +6,6 @@ Loads and initializes skills for fetching data from Dune Analytics API.
 import logging
 from typing import Dict, List, Optional, TypedDict
 
-from intentkit.abstracts.skill import SkillStoreABC
 from intentkit.skills.base import SkillConfig, SkillState
 from intentkit.skills.dune_analytics.base import DuneBaseTool
 
@@ -33,7 +32,6 @@ class Config(SkillConfig):
 async def get_skills(
     config: Config,
     is_private: bool,
-    store: SkillStoreABC,
     **kwargs,
 ) -> List[DuneBaseTool]:
     """Load Dune Analytics skills based on configuration.
@@ -59,7 +57,7 @@ async def get_skills(
 
     loaded_skills = []
     for name in available_skills:
-        skill = get_dune_skill(name, store)
+        skill = get_dune_skill(name)
         if skill:
             logger.info("Successfully loaded skill: %s", name)
             loaded_skills.append(skill)
@@ -69,7 +67,7 @@ async def get_skills(
     return loaded_skills
 
 
-def get_dune_skill(name: str, store: SkillStoreABC) -> Optional[DuneBaseTool]:
+def get_dune_skill(name: str) -> Optional[DuneBaseTool]:
     """Retrieve a Dune Analytics skill instance by name.
 
     Args:
@@ -87,11 +85,11 @@ def get_dune_skill(name: str, store: SkillStoreABC) -> Optional[DuneBaseTool]:
         if name == "fetch_nation_metrics":
             from .fetch_nation_metrics import FetchNationMetrics
 
-            _skill_cache[name] = FetchNationMetrics(skill_store=store)
+            _skill_cache[name] = FetchNationMetrics()
         elif name == "fetch_kol_buys":
             from .fetch_kol_buys import FetchKOLBuys
 
-            _skill_cache[name] = FetchKOLBuys(skill_store=store)
+            _skill_cache[name] = FetchKOLBuys()
         else:
             logger.warning("Unknown Dune Analytics skill: %s", name)
             return None

intentkit/skills/dune_analytics/base.py

@@ -1,52 +1,50 @@
-"""Base module for Dune Analytics skills.
-
-Provides shared functionality for interacting with the Dune Analytics API.
-"""
-
-from typing import Type
-
-from langchain_core.tools.base import ToolException
-from pydantic import BaseModel, Field
-
-from intentkit.abstracts.skill import SkillStoreABC
-from intentkit.skills.base import IntentKitSkill
-
-
-class DuneBaseTool(IntentKitSkill):
-    """Base class for Dune Analytics skills.
-
-    Offers common functionality like API key retrieval and Dune API interaction.
-    """
-
-    name: str = Field(description="Tool name")
-    description: str = Field(description="Tool description")
-    args_schema: Type[BaseModel]
-    skill_store: SkillStoreABC = Field(description="Skill store for data persistence")
-
-    def get_api_key(self) -> str:
-        """Retrieve the Dune Analytics API key from context.
-
-        Returns:
-            API key string.
-
-        Raises:
-            ToolException: If the API key is not found.
-        """
-        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 == "agent_owner":
-            api_key = skill_config.get("api_key")
-            if api_key:
-                return api_key
-            else:
-                raise ToolException("No api_key found in agent_owner configuration")
-        else:
-            raise ToolException(
-                f"Invalid API key provider: {api_key_provider}. Only 'agent_owner' is supported for Dune Analytics."
-            )
-
-    @property
-    def category(self) -> str:
-        """Category of the skill."""
-        return "dune_analytics"
+"""Base module for Dune Analytics skills.
+
+Provides shared functionality for interacting with the Dune Analytics API.
+"""
+
+from typing import Type
+
+from langchain_core.tools.base import ToolException
+from pydantic import BaseModel, Field
+
+from intentkit.skills.base import IntentKitSkill
+
+
+class DuneBaseTool(IntentKitSkill):
+    """Base class for Dune Analytics skills.
+
+    Offers common functionality like API key retrieval and Dune API interaction.
+    """
+
+    name: str = Field(description="Tool name")
+    description: str = Field(description="Tool description")
+    args_schema: Type[BaseModel]
+
+    def get_api_key(self) -> str:
+        """Retrieve the Dune Analytics API key from context.
+
+        Returns:
+            API key string.
+
+        Raises:
+            ToolException: If the API key is not found.
+        """
+        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 == "agent_owner":
+            api_key = skill_config.get("api_key")
+            if api_key:
+                return api_key
+            else:
+                raise ToolException("No api_key found in agent_owner configuration")
+        else:
+            raise ToolException(
+                f"Invalid API key provider: {api_key_provider}. Only 'agent_owner' is supported for Dune Analytics."
+            )
+
+    @property
+    def category(self) -> str:
+        """Category of the skill."""
+        return "dune_analytics"

intentkit/skills/dune_analytics/fetch_kol_buys.py

@@ -9,7 +9,6 @@ import httpx
 from pydantic import BaseModel, Field
 from tenacity import retry, stop_after_attempt, wait_exponential
 
-from intentkit.abstracts.skill import SkillStoreABC
 from intentkit.skills.dune_analytics.base import DuneBaseTool
 
 BASE_URL = "https://api.dune.com/api/v1/query"
@@ -49,7 +48,6 @@ class FetchKOLBuys(DuneBaseTool):
         "Supports a configurable limit for the number of results. Handles rate limits with retries."
     )
     args_schema: Type[BaseModel] = KOLBuysInput
-    skill_store: SkillStoreABC = Field(description="Skill store for data persistence")
 
     @retry(
         stop=stop_after_attempt(3), wait=wait_exponential(multiplier=5, min=5, max=60)

intentkit/skills/dune_analytics/fetch_nation_metrics.py

@@ -11,7 +11,6 @@ import httpx
 from pydantic import BaseModel, Field
 from tenacity import retry, stop_after_attempt, wait_exponential
 
-from intentkit.abstracts.skill import SkillStoreABC
 from intentkit.skills.dune_analytics.base import DuneBaseTool
 
 SUPPORTED_QUERIES = {
@@ -86,7 +85,6 @@ class FetchNationMetrics(DuneBaseTool):
         "Handles rate limits with retries."
     )
     args_schema: Type[BaseModel] = NationMetricsInput
-    skill_store: SkillStoreABC = Field(description="Skill store for data persistence")
 
     def normalize_metric(self, metric: str) -> str:
         """Normalize a metric string for matching.

intentkit/skills/elfa/__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.elfa.base import ElfaBaseTool
 from intentkit.skills.elfa.mention import (
@@ -36,7 +35,6 @@ class Config(SkillConfig):
 async def get_skills(
     config: "Config",
     is_private: bool,
-    store: SkillStoreABC,
     **_,
 ) -> list[ElfaBaseTool]:
     """Get all Elfa skills.
@@ -44,7 +42,6 @@ async def get_skills(
     Args:
         config: The configuration for Elfa skills.
         is_private: Whether to include private skills.
-        store: The skill store for persisting data.
 
     Returns:
         A list of Elfa skills.
@@ -61,7 +58,7 @@ async def get_skills(
     # Get each skill using the cached getter
     result = []
     for name in available_skills:
-        skill = get_elfa_skill(name, store)
+        skill = get_elfa_skill(name)
         if skill:
             result.append(skill)
     return result
@@ -69,13 +66,11 @@ async def get_skills(
 
 def get_elfa_skill(
     name: str,
-    store: SkillStoreABC,
 ) -> ElfaBaseTool:
     """Get an Elfa skill by name.
 
     Args:
         name: The name of the skill to get
-        store: The skill store for persisting data
 
     Returns:
         The requested Elfa skill
@@ -83,30 +78,22 @@ def get_elfa_skill(
 
     if name == "get_top_mentions":
         if name not in _cache:
-            _cache[name] = ElfaGetTopMentions(
-                skill_store=store,
-            )
+            _cache[name] = ElfaGetTopMentions()
         return _cache[name]
 
     elif name == "search_mentions":
         if name not in _cache:
-            _cache[name] = ElfaSearchMentions(
-                skill_store=store,
-            )
+            _cache[name] = ElfaSearchMentions()
         return _cache[name]
 
     elif name == "get_trending_tokens":
         if name not in _cache:
-            _cache[name] = ElfaGetTrendingTokens(
-                skill_store=store,
-            )
+            _cache[name] = ElfaGetTrendingTokens()
         return _cache[name]
 
     elif name == "get_smart_stats":
         if name not in _cache:
-            _cache[name] = ElfaGetSmartStats(
-                skill_store=store,
-            )
+            _cache[name] = ElfaGetSmartStats()
         return _cache[name]
 
     else:

intentkit/skills/elfa/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
 
 base_url = "https://api.elfa.ai/v2"
@@ -15,23 +15,21 @@ class ElfaBaseTool(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("elfa_api_key")
+            if not config.elfa_api_key:
+                raise ToolException("Elfa API key is not configured")
+            return config.elfa_api_key
         # for backward compatibility, may only have api_key in skill_config
-        elif skill_config.get("api_key"):
+        if skill_config.get("api_key"):
             return skill_config.get("api_key")
-        else:
-            raise ToolException(
-                f"Invalid API key provider: {api_key_provider}, or no api_key in config"
-            )
+        raise ToolException(
+            f"Invalid API key provider: {api_key_provider}, or no api_key in config"
+        )
 
     @property
     def category(self) -> str:

intentkit/skills/enso/__init__.py

@@ -3,7 +3,6 @@
 import logging
 from typing import List, NotRequired, TypedDict
 
-from intentkit.abstracts.skill import SkillStoreABC
 from intentkit.skills.base import SkillConfig, SkillState
 from intentkit.skills.enso.base import EnsoBaseTool
 from intentkit.skills.enso.best_yield import EnsoGetBestYield
@@ -42,7 +41,6 @@ class Config(SkillConfig):
 async def get_skills(
     config: Config,
     is_private: bool,
-    store: SkillStoreABC,
     **_,
 ) -> list[EnsoBaseTool]:
     """Get all Enso skills."""
@@ -58,7 +56,7 @@ async def get_skills(
     # Get each skill using the cached getter
     result = []
     for name in available_skills:
-        skill = get_enso_skill(name, store)
+        skill = get_enso_skill(name)
         if skill:
             result.append(skill)
     return result
@@ -66,49 +64,31 @@ async def get_skills(
 
 def get_enso_skill(
     name: str,
-    skill_store: SkillStoreABC,
 ) -> EnsoBaseTool:
     """Get an Enso skill by name.
 
     Args:
         name: The name of the skill to get
-        skill_store: The skill store for persisting data
 
     Returns:
         The requested Enso skill
     """
     if name == "get_networks":
-        return EnsoGetNetworks(
-            skill_store=skill_store,
-        )
+        return EnsoGetNetworks()
     if name == "get_tokens":
-        return EnsoGetTokens(
-            skill_store=skill_store,
-        )
+        return EnsoGetTokens()
     if name == "get_prices":
-        return EnsoGetPrices(
-            skill_store=skill_store,
-        )
+        return EnsoGetPrices()
     if name == "get_wallet_approvals":
-        return EnsoGetWalletApprovals(
-            skill_store=skill_store,
-        )
+        return EnsoGetWalletApprovals()
     if name == "get_wallet_balances":
-        return EnsoGetWalletBalances(
-            skill_store=skill_store,
-        )
+        return EnsoGetWalletBalances()
     if name == "wallet_approve":
-        return EnsoWalletApprove(
-            skill_store=skill_store,
-        )
+        return EnsoWalletApprove()
     if name == "route_shortcut":
-        return EnsoRouteShortcut(
-            skill_store=skill_store,
-        )
+        return EnsoRouteShortcut()
     if name == "get_best_yield":
-        return EnsoGetBestYield(
-            skill_store=skill_store,
-        )
+        return EnsoGetBestYield()
     else:
         logger.warning(f"Unknown Enso skill: {name}")
         return None

intentkit/skills/enso/base.py

@@ -6,8 +6,8 @@ from langchain_core.tools.base import ToolException
 from pydantic import BaseModel, Field
 
 from intentkit.abstracts.graph import AgentContext
-from intentkit.abstracts.skill import SkillStoreABC
 from intentkit.clients import get_wallet_provider as get_agent_wallet_provider
+from intentkit.config.config import config
 from intentkit.skills.base import IntentKitSkill
 from intentkit.utils.chain import ChainProvider, Network, network_to_id
 
@@ -20,9 +20,6 @@ class EnsoBaseTool(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"
-    )
 
     async def get_wallet_provider(self, context: AgentContext) -> CdpEvmWalletProvider:
         """Get the wallet provider from the CDP client.
@@ -40,7 +37,7 @@ class EnsoBaseTool(IntentKitSkill):
         return provider.get_address()
 
     def get_chain_provider(self, context: AgentContext) -> Optional[ChainProvider]:
-        return self.skill_store.get_system_config("chain_provider")
+        return config.chain_provider
 
     def get_main_tokens(self, context: AgentContext) -> list[str]:
         skill_config = context.agent.skill_config(self.category)
@@ -52,7 +49,7 @@ class EnsoBaseTool(IntentKitSkill):
         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("enso_api_token")
+            return config.enso_api_token
         # for backward compatibility, may only have api_token in skill_config
         elif skill_config.get("api_token"):
             return skill_config.get("api_token")

intentkit/skills/enso/networks.py

@@ -83,12 +83,7 @@ class EnsoGetNetworks(EnsoBaseTool):
                         exclude_none=True
                     )
 
-                await self.skill_store.save_agent_skill_data(
-                    context.agent_id,
-                    "enso_get_networks",
-                    "networks",
-                    networks_memory,
-                )
+                await self.save_agent_skill_data("networks", networks_memory)
 
                 return EnsoGetNetworksOutput(res=networks)
             except httpx.RequestError as req_err:

intentkit/skills/enso/route.py

@@ -181,7 +181,6 @@ class EnsoRouteShortcut(EnsoBaseTool):
         """
 
         context = self.get_context()
-        agent_id = context.agent_id
         resolved_chain_id = self.resolve_chain_id(context, chainId)
         api_token = self.get_api_token(context)
         # Use the wallet provider to send the transaction
@@ -191,8 +190,8 @@ class EnsoRouteShortcut(EnsoBaseTool):
         async with httpx.AsyncClient() as client:
             try:
                 network_name = None
-                networks = await self.skill_store.get_agent_skill_data(
-                    agent_id, "enso_get_networks", "networks"
+                networks = await self.get_agent_skill_data_raw(
+                    "enso_get_networks", "networks"
                 )
 
                 if networks:
@@ -203,9 +202,7 @@ class EnsoRouteShortcut(EnsoBaseTool):
                         else None
                     )
                 if network_name is None:
-                    networks = await EnsoGetNetworks(
-                        skill_store=self.skill_store,
-                    ).arun()
+                    networks = await EnsoGetNetworks().arun()
 
                     for network in networks.res:
                         if network.id == resolved_chain_id:
@@ -221,8 +218,7 @@ class EnsoRouteShortcut(EnsoBaseTool):
                     "Authorization": f"Bearer {api_token}",
                 }
 
-                token_decimals = await self.skill_store.get_agent_skill_data(
-                    agent_id,
+                token_decimals = await self.get_agent_skill_data_raw(
                     "enso_get_tokens",
                     "decimals",
                 )

intentkit/skills/enso/tokens.py

@@ -154,7 +154,6 @@ class EnsoGetTokens(EnsoBaseTool):
         url = f"{base_url}/api/v1/tokens"
 
         context = self.get_context()
-        agent_id = context.agent_id
         resolved_chain_id = self.resolve_chain_id(context, chainId)
         api_token = self.get_api_token(context)
         main_tokens = self.get_main_tokens(context)
@@ -178,11 +177,7 @@ class EnsoGetTokens(EnsoBaseTool):
                 response.raise_for_status()
                 json_dict = response.json()
 
-                token_decimals = await self.skill_store.get_agent_skill_data(
-                    agent_id,
-                    "enso_get_tokens",
-                    "decimals",
-                )
+                token_decimals = await self.get_agent_skill_data("decimals")
                 if not token_decimals:
                     token_decimals = {}
 
@@ -203,12 +198,7 @@ class EnsoGetTokens(EnsoBaseTool):
                                 if u_token.address:
                                     token_decimals[u_token.address] = u_token.decimals
 
-                await self.skill_store.save_agent_skill_data(
-                    agent_id,
-                    "enso_get_tokens",
-                    "decimals",
-                    token_decimals,
-                )
+                await self.save_agent_skill_data("decimals", token_decimals)
 
                 return res
             except httpx.RequestError as req_err:

intentkit/skills/erc20/__init__.py

@@ -4,7 +4,6 @@ from typing import TYPE_CHECKING, Optional, TypedDict
 
 from coinbase_agentkit import erc20_action_provider
 
-from intentkit.abstracts.skill import SkillStoreABC
 from intentkit.skills.base import (
     SkillConfig,
     SkillState,
@@ -31,7 +30,6 @@ class Config(SkillConfig):
 async def get_skills(
     config: "Config",
     is_private: bool,
-    store: SkillStoreABC,
     agent_id: str,
     agent: Optional["Agent"] = None,
     **_,
@@ -45,9 +43,7 @@ async def get_skills(
         if state == "public" or (state == "private" and is_private):
             available_skills.append(skill_name)
 
-    actions = await get_agentkit_actions(
-        agent_id, store, [erc20_action_provider], agent=agent
-    )
+    actions = await get_agentkit_actions(agent_id, [erc20_action_provider], agent=agent)
     tools: list[ERC20BaseTool] = []
     for skill in available_skills:
         for action in actions:

intentkit/skills/erc721/__init__.py

@@ -4,7 +4,6 @@ from typing import TYPE_CHECKING, Optional, TypedDict
 
 from coinbase_agentkit import erc721_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, [erc721_action_provider], agent=agent
+        agent_id, [erc721_action_provider], agent=agent
     )
     tools: list[ERC721BaseTool] = []
     for skill in available_skills: