intentkit 0.6.9.dev2__py3-none-any.whl → 0.6.10__py3-none-any.whl

intentkit/skills/tavily/tavily_search.py

@@ -2,7 +2,6 @@ import logging
 from typing import Type
 
 import httpx
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.tavily.base import TavilyBaseTool
@@ -58,7 +57,6 @@ class TavilySearch(TavilyBaseTool):
         max_results: int = 5,
         include_images: bool = False,
         include_raw_content: bool = False,
-        config: RunnableConfig = None,
         **kwargs,
     ) -> str:
         """Implementation of the Tavily search tool.
@@ -68,26 +66,30 @@ class TavilySearch(TavilyBaseTool):
             max_results: Maximum number of search results to return (1-10).
             include_images: Whether to include image URLs in the results.
             include_raw_content: Whether to include raw HTML content in the results.
-            config: The configuration for the tool call.
+
 
         Returns:
             str: Formatted search results with titles, snippets, and URLs.
         """
-        context = self.context_from_config(config)
+        context = self.get_context()
+        skill_config = context.agent.skill_config(self.category)
         logger.debug(f"tavily.py: Running web search with context {context}")
 
-        if context.config.get("api_key_provider") == "agent_owner":
-            if context.config.get("rate_limit_number") and context.config.get(
+        if skill_config.get("api_key_provider") == "agent_owner":
+            if skill_config.get("rate_limit_number") and skill_config.get(
                 "rate_limit_minutes"
             ):
                 await self.user_rate_limit_by_category(
                     context.user_id,
-                    context.config["rate_limit_number"],
-                    context.config["rate_limit_minutes"],
+                    skill_config["rate_limit_number"],
+                    skill_config["rate_limit_minutes"],
                 )
 
         # Get the API key from the agent's configuration
-        api_key = self.get_api_key(context)
+        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")
         if not api_key:
             return "Error: No Tavily API key provided in the configuration."
 

intentkit/skills/token/base.py

@@ -6,7 +6,7 @@ from typing import Any, Dict
 import aiohttp
 
 from intentkit.abstracts.skill import SkillStoreABC
-from intentkit.skills.base import IntentKitSkill, SkillContext
+from intentkit.skills.base import IntentKitSkill
 from intentkit.skills.token.constants import MORALIS_API_BASE_URL
 
 logger = logging.getLogger(__name__)
@@ -27,16 +27,14 @@ class TokenBaseTool(IntentKitSkill):
     def category(self) -> str:
         return "token"
 
-    def get_api_key(self, context: SkillContext) -> str:
+    def get_api_key(self) -> str:
         """Get API key from agent config or system config.
 
-        Args:
-            context: The skill context containing the agent config
-
         Returns:
             The API key to use for API requests
         """
-        skill_config = context.config
+        context = self.get_context()
+        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")

intentkit/skills/token/erc20_transfers.py

@@ -1,7 +1,6 @@
 import logging
 from typing import Any, Dict, List, Optional, Type
 
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.token.base import TokenBaseTool
@@ -80,7 +79,6 @@ class ERC20Transfers(TokenBaseTool):
         limit: Optional[int] = DEFAULT_LIMIT,
         order: Optional[str] = DEFAULT_ORDER,
         cursor: Optional[str] = None,
-        config: RunnableConfig = None,
         **kwargs,
     ) -> Dict[str, Any]:
         """Fetch ERC20 token transfers for a wallet from Moralis.
@@ -101,7 +99,7 @@ class ERC20Transfers(TokenBaseTool):
         Returns:
             Dict containing ERC20 transfer data
         """
-        context = self.context_from_config(config)
+        context = self.get_context()
         if context is None:
             logger.error("Context is None, cannot retrieve API key")
             return {
@@ -109,7 +107,7 @@ class ERC20Transfers(TokenBaseTool):
             }
 
         # Get the API key
-        api_key = self.get_api_key(context)
+        api_key = self.get_api_key()
 
         if not api_key:
             logger.error("No Moralis API key available")

intentkit/skills/token/token_analytics.py

@@ -1,7 +1,6 @@
 import logging
 from typing import Any, Dict, Type
 
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.token.base import TokenBaseTool
@@ -38,7 +37,6 @@ class TokenAnalytics(TokenBaseTool):
         self,
         address: str,
         chain: str = DEFAULT_CHAIN,
-        config: RunnableConfig = None,
         **kwargs,
     ) -> Dict[str, Any]:
         """Fetch token analytics from Moralis.
@@ -51,7 +49,7 @@ class TokenAnalytics(TokenBaseTool):
         Returns:
             Dict containing token analytics data
         """
-        context = self.context_from_config(config)
+        context = self.get_context()
         if context is None:
             logger.error("Context is None, cannot retrieve API key")
             return {
@@ -59,7 +57,7 @@ class TokenAnalytics(TokenBaseTool):
             }
 
         # Get the API key
-        api_key = self.get_api_key(context)
+        api_key = self.get_api_key()
 
         if not api_key:
             logger.error("No Moralis API key available")

intentkit/skills/token/token_price.py

@@ -1,7 +1,6 @@
 import logging
 from typing import Any, Dict, Optional, Type
 
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.token.base import TokenBaseTool
@@ -65,7 +64,6 @@ class TokenPrice(TokenBaseTool):
         to_block: Optional[int] = None,
         max_token_inactivity: Optional[int] = None,
         min_pair_side_liquidity_usd: Optional[int] = None,
-        config: RunnableConfig = None,
         **kwargs,
     ) -> Dict[str, Any]:
         """Fetch token price from Moralis.
@@ -84,7 +82,7 @@ class TokenPrice(TokenBaseTool):
             Dict containing token price data
         """
         # Extract context from config
-        context = self.context_from_config(config)
+        context = self.get_context()
 
         if context is None:
             logger.error("Context is None, cannot retrieve API key")
@@ -93,7 +91,7 @@ class TokenPrice(TokenBaseTool):
             }
 
         # Get the API key
-        api_key = self.get_api_key(context)
+        api_key = self.get_api_key()
 
         if not api_key:
             logger.error("No Moralis API key available")

intentkit/skills/token/token_search.py

@@ -1,7 +1,6 @@
 import logging
 from typing import Any, Dict, List, Optional, Type
 
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.token.base import TokenBaseTool
@@ -53,7 +52,6 @@ class TokenSearch(TokenBaseTool):
         chains: Optional[List[str]] = None,
         limit: Optional[int] = None,
         is_verified_contract: Optional[bool] = None,
-        config: RunnableConfig = None,
         **kwargs,
     ) -> Dict[str, Any]:
         """Search for tokens using Moralis.
@@ -69,7 +67,7 @@ class TokenSearch(TokenBaseTool):
             Dict containing token search results
         """
         # Extract context from config
-        context = self.context_from_config(config)
+        context = self.get_context()
         if context is None:
             logger.error("Context is None, cannot retrieve API key")
             return {
@@ -77,7 +75,7 @@ class TokenSearch(TokenBaseTool):
             }
 
         # Get the API key
-        api_key = self.get_api_key(context)
+        api_key = self.get_api_key()
 
         if not api_key:
             logger.error("No Moralis API key available")

intentkit/skills/twitter/base.py

@@ -1,6 +1,7 @@
 from datetime import datetime, timedelta, timezone
 from typing import Type
 
+from langchain.tools.base import ToolException
 from pydantic import BaseModel, Field
 
 from intentkit.abstracts.exception import RateLimitExceeded
@@ -18,6 +19,46 @@ class TwitterBaseTool(IntentKitSkill):
         description="The skill store for persisting data"
     )
 
+    def get_api_key(self) -> dict:
+        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 platform keys (these need to be added to config.py)
+            return {
+                "consumer_key": self.skill_store.get_system_config(
+                    "twitter_consumer_key"
+                ),
+                "consumer_secret": self.skill_store.get_system_config(
+                    "twitter_consumer_secret"
+                ),
+                "access_token": self.skill_store.get_system_config(
+                    "twitter_access_token"
+                ),
+                "access_token_secret": self.skill_store.get_system_config(
+                    "twitter_access_token_secret"
+                ),
+            }
+        # for backward compatibility or agent_owner provider
+        elif api_key_provider == "agent_owner":
+            required_keys = [
+                "consumer_key",
+                "consumer_secret",
+                "access_token",
+                "access_token_secret",
+            ]
+            api_keys = {}
+            for key in required_keys:
+                if skill_config.get(key):
+                    api_keys[key] = skill_config.get(key)
+                else:
+                    raise ToolException(
+                        f"Missing required {key} in agent_owner configuration"
+                    )
+            return api_keys
+        else:
+            raise ToolException(f"Invalid API key provider: {api_key_provider}")
+
     @property
     def category(self) -> str:
         return "twitter"

intentkit/skills/twitter/follow_user.py

@@ -1,7 +1,6 @@
 import logging
 from typing import Type
 
-from langchain_core.runnables import RunnableConfig
 from langchain_core.tools import ToolException
 from pydantic import BaseModel, Field
 
@@ -37,13 +36,14 @@ class TwitterFollowUser(TwitterBaseTool):
     description: str = PROMPT
     args_schema: Type[BaseModel] = TwitterFollowUserInput
 
-    async def _arun(self, user_id: str, config: RunnableConfig, **kwargs) -> bool:
+    async def _arun(self, user_id: str, **kwargs) -> bool:
         try:
-            context = self.context_from_config(config)
+            context = self.get_context()
+            skill_config = context.agent.skill_config(self.category)
             twitter = get_twitter_client(
                 agent_id=context.agent_id,
                 skill_store=self.skill_store,
-                config=context.config,
+                config=skill_config,
             )
             client = await twitter.get_client()
 

intentkit/skills/twitter/get_mentions.py

@@ -2,7 +2,6 @@ import logging
 from datetime import datetime, timedelta, timezone
 from typing import Type
 
-from langchain_core.runnables import RunnableConfig
 from langchain_core.tools import ToolException
 from pydantic import BaseModel
 
@@ -41,13 +40,14 @@ class TwitterGetMentions(TwitterBaseTool):
     description: str = PROMPT
     args_schema: Type[BaseModel] = TwitterGetMentionsInput
 
-    async def _arun(self, config: RunnableConfig, **kwargs) -> list[Tweet]:
+    async def _arun(self, **kwargs) -> list[Tweet]:
         try:
-            context = self.context_from_config(config)
+            context = self.get_context()
+            skill_config = context.agent.skill_config(self.category)
             twitter = get_twitter_client(
                 agent_id=context.agent_id,
                 skill_store=self.skill_store,
-                config=context.config,
+                config=skill_config,
             )
             client = await twitter.get_client()
 

intentkit/skills/twitter/get_timeline.py

@@ -1,7 +1,6 @@
 import logging
 from typing import Type
 
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel
 
 from intentkit.clients import get_twitter_client
@@ -37,16 +36,17 @@ class TwitterGetTimeline(TwitterBaseTool):
     description: str = PROMPT
     args_schema: Type[BaseModel] = TwitterGetTimelineInput
 
-    async def _arun(self, config: RunnableConfig, **kwargs):
+    async def _arun(self, **kwargs):
         try:
             # Ensure max_results is an integer
             max_results = 10
 
-            context = self.context_from_config(config)
+            context = self.get_context()
+            skill_config = context.agent.skill_config(self.category)
             twitter = get_twitter_client(
                 agent_id=context.agent_id,
                 skill_store=self.skill_store,
-                config=context.config,
+                config=skill_config,
             )
             client = await twitter.get_client()
 

intentkit/skills/twitter/get_user_by_username.py

@@ -1,7 +1,6 @@
 import logging
 from typing import Type
 
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.clients import get_twitter_client
@@ -38,13 +37,14 @@ class TwitterGetUserByUsername(TwitterBaseTool):
     description: str = PROMPT
     args_schema: Type[BaseModel] = TwitterGetUserByUsernameInput
 
-    async def _arun(self, username: str, config: RunnableConfig, **kwargs):
+    async def _arun(self, username: str, **kwargs):
         try:
-            context = self.context_from_config(config)
+            context = self.get_context()
+            skill_config = context.agent.skill_config(self.category)
             twitter = get_twitter_client(
                 agent_id=context.agent_id,
                 skill_store=self.skill_store,
-                config=context.config,
+                config=skill_config,
             )
             client = await twitter.get_client()
 

intentkit/skills/twitter/get_user_tweets.py

@@ -1,7 +1,6 @@
 import logging
 from typing import List, Optional, Type
 
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.clients import get_twitter_client
@@ -44,7 +43,7 @@ class TwitterGetUserTweets(TwitterBaseTool):
     description: str = PROMPT
     args_schema: Type[BaseModel] = TwitterGetUserTweetsInput
 
-    async def _arun(self, config: RunnableConfig, **kwargs):
+    async def _arun(self, **kwargs):
         try:
             user_id = kwargs.get("user_id")
             if not user_id:
@@ -56,11 +55,12 @@ class TwitterGetUserTweets(TwitterBaseTool):
             # Get exclude parameter with default
             exclude = kwargs.get("exclude", ["replies", "retweets"])
 
-            context = self.context_from_config(config)
+            context = self.get_context()
+            skill_config = context.agent.skill_config(self.category)
             twitter = get_twitter_client(
                 agent_id=context.agent_id,
                 skill_store=self.skill_store,
-                config=context.config,
+                config=skill_config,
             )
             client = await twitter.get_client()
 

intentkit/skills/twitter/like_tweet.py

@@ -1,7 +1,6 @@
 import logging
 from typing import Type
 
-from langchain_core.runnables import RunnableConfig
 from langchain_core.tools import ToolException
 from pydantic import BaseModel, Field
 
@@ -35,13 +34,14 @@ class TwitterLikeTweet(TwitterBaseTool):
     description: str = PROMPT
     args_schema: Type[BaseModel] = TwitterLikeTweetInput
 
-    async def _arun(self, tweet_id: str, config: RunnableConfig, **kwargs):
+    async def _arun(self, tweet_id: str, **kwargs):
         try:
-            context = self.context_from_config(config)
+            context = self.get_context()
+            skill_config = context.agent.skill_config(self.category)
             twitter = get_twitter_client(
                 agent_id=context.agent_id,
                 skill_store=self.skill_store,
-                config=context.config,
+                config=skill_config,
             )
             client = await twitter.get_client()
 

intentkit/skills/twitter/post_tweet.py

@@ -1,7 +1,6 @@
 import logging
 from typing import Optional, Type
 
-from langchain_core.runnables import RunnableConfig
 from langchain_core.tools import ToolException
 from pydantic import BaseModel, Field
 
@@ -48,15 +47,15 @@ class TwitterPostTweet(TwitterBaseTool):
         self,
         text: str,
         image: Optional[str] = None,
-        config: RunnableConfig = None,
         **kwargs,
     ):
         try:
-            context = self.context_from_config(config)
+            context = self.get_context()
+            skill_config = context.agent.skill_config(self.category)
             twitter = get_twitter_client(
                 agent_id=context.agent_id,
                 skill_store=self.skill_store,
-                config=context.config,
+                config=skill_config,
             )
             client = await twitter.get_client()
 

intentkit/skills/twitter/reply_tweet.py

@@ -1,7 +1,6 @@
 import logging
 from typing import Optional, Type
 
-from langchain_core.runnables import RunnableConfig
 from langchain_core.tools import ToolException
 from pydantic import BaseModel, Field
 
@@ -50,15 +49,15 @@ class TwitterReplyTweet(TwitterBaseTool):
         tweet_id: str,
         text: str,
         image: Optional[str] = None,
-        config: RunnableConfig = None,
         **kwargs,
     ):
         try:
-            context = self.context_from_config(config)
+            context = self.get_context()
+            skill_config = context.agent.skill_config(self.category)
             twitter = get_twitter_client(
                 agent_id=context.agent_id,
                 skill_store=self.skill_store,
-                config=context.config,
+                config=skill_config,
             )
             client = await twitter.get_client()
 

intentkit/skills/twitter/retweet.py

@@ -1,7 +1,6 @@
 import logging
 from typing import Type
 
-from langchain_core.runnables import RunnableConfig
 from langchain_core.tools import ToolException
 from pydantic import BaseModel, Field
 
@@ -35,13 +34,14 @@ class TwitterRetweet(TwitterBaseTool):
     description: str = PROMPT
     args_schema: Type[BaseModel] = TwitterRetweetInput
 
-    async def _arun(self, tweet_id: str, config: RunnableConfig, **kwargs):
+    async def _arun(self, tweet_id: str, **kwargs):
         try:
-            context = self.context_from_config(config)
+            context = self.get_context()
+            skill_config = context.agent.skill_config(self.category)
             twitter = get_twitter_client(
                 agent_id=context.agent_id,
                 skill_store=self.skill_store,
-                config=context.config,
+                config=skill_config,
             )
             client = await twitter.get_client()
 

intentkit/skills/twitter/search_tweets.py

@@ -2,7 +2,6 @@ import datetime
 import logging
 from typing import Type
 
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.clients import get_twitter_client
@@ -36,14 +35,15 @@ class TwitterSearchTweets(TwitterBaseTool):
     description: str = PROMPT
     args_schema: Type[BaseModel] = TwitterSearchTweetsInput
 
-    async def _arun(self, query: str, config: RunnableConfig, **kwargs):
+    async def _arun(self, query: str, **kwargs):
         max_results = 10
         try:
-            context = self.context_from_config(config)
+            context = self.get_context()
+            skill_config = context.agent.skill_config(self.category)
             twitter = get_twitter_client(
                 agent_id=context.agent_id,
                 skill_store=self.skill_store,
-                config=context.config,
+                config=skill_config,
             )
             client = await twitter.get_client()
 

intentkit/skills/unrealspeech/base.py

@@ -1,5 +1,6 @@
 from typing import Type
 
+from langchain.tools.base import ToolException
 from pydantic import BaseModel, Field
 
 from intentkit.abstracts.skill import SkillStoreABC
@@ -16,6 +17,21 @@ class UnrealSpeechBaseTool(IntentKitSkill):
         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 == "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 UnrealSpeech."
+            )
+
     @property
     def category(self) -> str:
         return "unrealspeech"

intentkit/skills/unrealspeech/text_to_speech.py

@@ -4,7 +4,6 @@ from typing import Any, Dict, Literal, Optional, Type
 
 import httpx
 from langchain_core.callbacks.manager import CallbackManagerForToolRun
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.unrealspeech.base import UnrealSpeechBaseTool
@@ -66,16 +65,17 @@ class TextToSpeech(UnrealSpeechBaseTool):
         bitrate: str = "192k",
         speed: float = 0.0,
         timestamp_type: Optional[Literal["word", "sentence"]] = "word",
-        config: Optional[RunnableConfig] = None,
+        config: Optional[Any] = None,
         run_manager: Optional[CallbackManagerForToolRun] = None,
         **kwargs,
     ) -> Dict[str, Any]:
         """Run the tool to convert text to speech."""
 
         # Get the API key from context config if available
-        context = self.context_from_config(config) if config else None
+        context = self.get_context()
+        skill_config = context.agent.skill_config(self.category) if config else None
         api_key = (
-            context.config.get("api_key", None) if context and context.config else None
+            skill_config.get("api_key", None) if context and skill_config else None
         )
 
         # Clean up and validate input

intentkit/skills/venice_audio/base.py

@@ -1,10 +1,11 @@
 import logging
 from typing import Dict, List, Optional, Tuple, Type
 
+from langchain.tools.base import ToolException
 from pydantic import BaseModel, Field
 
 from intentkit.abstracts.skill import SkillStoreABC
-from intentkit.skills.base import IntentKitSkill, SkillContext, ToolException
+from intentkit.skills.base import IntentKitSkill
 
 logger = logging.getLogger(__name__)
 
@@ -24,7 +25,7 @@ class VeniceAudioBaseTool(IntentKitSkill):
         return "venice_audio"
 
     def validate_voice_model(
-        self, context: SkillContext, voice_model: str
+        self, context, voice_model: str
     ) -> Tuple[bool, Optional[Dict[str, object]]]:
         config = context.config
 
@@ -47,7 +48,7 @@ class VeniceAudioBaseTool(IntentKitSkill):
 
         return True, None
 
-    def get_api_key(self, context: SkillContext) -> str:
+    def get_api_key(self) -> str:
         """
         Retrieves the Venice AI API key based on the api_key_provider setting.
 
@@ -58,10 +59,11 @@ class VeniceAudioBaseTool(IntentKitSkill):
             ToolException: If the API key is not found or provider is invalid.
         """
         try:
-            skillConfig = context.config
-            api_key_provider = skillConfig.get("api_key_provider")
+            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":
-                agent_api_key = context.config.get("api_key")
+                agent_api_key = skill_config.get("api_key")
                 if agent_api_key:
                     logger.debug(
                         f"Using agent-specific Venice API key for skill {self.name} in category {self.category}"
@@ -90,15 +92,15 @@ class VeniceAudioBaseTool(IntentKitSkill):
         except Exception as e:
             raise ToolException(f"Failed to retrieve Venice API key: {str(e)}") from e
 
-    async def apply_rate_limit(self, context: SkillContext) -> None:
+    async def apply_rate_limit(self, context) -> None:
         """
         Applies rate limiting ONLY if specified in the agent's config ('skill_config').
         Checks for 'rate_limit_number' and 'rate_limit_minutes'.
         If not configured, NO rate limiting is applied.
         Raises ConnectionAbortedError if the configured limit is exceeded.
         """
-        skill_config = context.config
-        user_id = context.user_id
+        skill_config = context.agent.skill_config(self.category)
+        user_id = context.agent.id
 
         # Get agent-specific limits safely
         limit_num = skill_config.get("rate_limit_number")