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

intentkit/skills/portfolio/wallet_profitability.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.portfolio.base import PortfolioBaseTool
@@ -50,7 +49,6 @@ class WalletProfitability(PortfolioBaseTool):
         chain: str = DEFAULT_CHAIN,
         days: Optional[str] = "all",
         token_addresses: Optional[List[str]] = None,
-        config: RunnableConfig = None,
         **kwargs,
     ) -> Dict[str, Any]:
         """Fetch detailed wallet profitability from Moralis.
@@ -65,13 +63,13 @@ class WalletProfitability(PortfolioBaseTool):
         Returns:
             Dict containing wallet profitability breakdown data
         """
-        context = self.context_from_config(config)
+        context = self.get_context()
         logger.debug(
             f"wallet_profitability.py: Fetching profitability breakdown with context {context}"
         )
 
         # Get the API key from the agent's configuration
-        api_key = self.get_api_key(context)
+        api_key = self.get_api_key()
         if not api_key:
             return {"error": "No Moralis API key provided in the configuration."}
 

intentkit/skills/portfolio/wallet_profitability_summary.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.portfolio.base import PortfolioBaseTool
@@ -45,7 +44,6 @@ class WalletProfitabilitySummary(PortfolioBaseTool):
         address: str,
         chain: str = DEFAULT_CHAIN,
         days: Optional[str] = "all",
-        config: RunnableConfig = None,
         **kwargs,
     ) -> Dict[str, Any]:
         """Fetch wallet profitability summary from Moralis.
@@ -59,13 +57,13 @@ class WalletProfitabilitySummary(PortfolioBaseTool):
         Returns:
             Dict containing wallet profitability summary data
         """
-        context = self.context_from_config(config)
+        context = self.get_context()
         logger.debug(
             f"wallet_profitability_summary.py: Fetching profitability summary with context {context}"
         )
 
         # Get the API key from the agent's configuration
-        api_key = self.get_api_key(context)
+        api_key = self.get_api_key()
         if not api_key:
             return {"error": "No Moralis API key provided in the configuration."}
 

intentkit/skills/portfolio/wallet_stats.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.portfolio.base import PortfolioBaseTool
@@ -38,7 +37,6 @@ class WalletStats(PortfolioBaseTool):
         self,
         address: str,
         chain: str = DEFAULT_CHAIN,
-        config: RunnableConfig = None,
         **kwargs,
     ) -> Dict[str, Any]:
         """Fetch wallet stats from Moralis.
@@ -51,11 +49,11 @@ class WalletStats(PortfolioBaseTool):
         Returns:
             Dict containing wallet stats data
         """
-        context = self.context_from_config(config)
+        context = self.get_context()
         logger.debug(f"wallet_stats.py: Fetching wallet stats with context {context}")
 
         # Get the API key from the agent's configuration
-        api_key = self.get_api_key(context)
+        api_key = self.get_api_key()
         if not api_key:
             return {"error": "No Moralis API key provided in the configuration."}
 

intentkit/skills/portfolio/wallet_swaps.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.portfolio.base import PortfolioBaseTool
@@ -82,7 +81,6 @@ class WalletSwaps(PortfolioBaseTool):
         to_date: Optional[str] = None,
         order: Optional[str] = DEFAULT_ORDER,
         transaction_types: Optional[List[str]] = None,
-        config: RunnableConfig = None,
         **kwargs,
     ) -> Dict[str, Any]:
         """Fetch wallet swap transactions from Moralis.
@@ -103,11 +101,11 @@ class WalletSwaps(PortfolioBaseTool):
         Returns:
             Dict containing wallet swaps data
         """
-        context = self.context_from_config(config)
+        context = self.get_context()
         logger.debug(f"wallet_swaps.py: Fetching wallet swaps with context {context}")
 
         # Get the API key from the agent's configuration
-        api_key = self.get_api_key(context)
+        api_key = self.get_api_key()
         if not api_key:
             return {"error": "No Moralis API key provided in the configuration."}
 

intentkit/skills/slack/base.py

@@ -1,5 +1,6 @@
 from typing import Optional, Type
 
+from langchain.tools.base import ToolException
 from pydantic import BaseModel, Field
 from slack_sdk import WebClient
 
@@ -17,6 +18,23 @@ class SlackBaseTool(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":
+            slack_bot_token = skill_config.get("slack_bot_token")
+            if slack_bot_token:
+                return slack_bot_token
+            else:
+                raise ToolException(
+                    "No slack_bot_token found in agent_owner configuration"
+                )
+        else:
+            raise ToolException(
+                f"Invalid API key provider: {api_key_provider}. Only 'agent_owner' is supported for Slack."
+            )
+
     @property
     def category(self) -> str:
         return "slack"

intentkit/skills/slack/get_channel.py

@@ -1,6 +1,5 @@
 from typing import Any, Dict, Optional, Type, Union
 
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.slack.base import SlackBaseTool, SlackChannel
@@ -28,7 +27,6 @@ class SlackGetChannel(SlackBaseTool):
 
     async def _arun(
         self,
-        config: RunnableConfig,
         channel_id: Optional[str] = None,
         channel_name: Optional[str] = None,
         **kwargs,
@@ -46,8 +44,9 @@ class SlackGetChannel(SlackBaseTool):
             ValueError: If neither channel_id nor channel_name is provided
             Exception: If an error occurs getting the channel information
         """
-        context = self.context_from_config(config)
-        client = self.get_client(context.config.get("slack_bot_token"))
+        context = self.get_context()
+        skill_config = context.agent.skill_config(self.category)
+        client = self.get_client(skill_config.get("slack_bot_token"))
 
         try:
             # If no channel specified, return a dict of all channels

intentkit/skills/slack/get_message.py

@@ -1,6 +1,5 @@
 from typing import Any, Dict, Optional, Type
 
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.slack.base import SlackBaseTool, SlackMessage
@@ -35,7 +34,6 @@ class SlackGetMessage(SlackBaseTool):
 
     async def _arun(
         self,
-        config: RunnableConfig,
         channel_id: str,
         ts: Optional[str] = None,
         thread_ts: Optional[str] = None,
@@ -56,8 +54,9 @@ class SlackGetMessage(SlackBaseTool):
         Raises:
             Exception: If an error occurs getting the messages
         """
-        context = self.context_from_config(config)
-        client = self.get_client(context.config.get("slack_bot_token"))
+        context = self.get_context()
+        skill_config = context.agent.skill_config(self.category)
+        client = self.get_client(skill_config.get("slack_bot_token"))
 
         try:
             # Ensure limit is within bounds

intentkit/skills/slack/schedule_message.py

@@ -1,7 +1,6 @@
 from datetime import datetime
 from typing import Any, Dict, Optional, Type
 
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.slack.base import SlackBaseTool
@@ -34,7 +33,6 @@ class SlackScheduleMessage(SlackBaseTool):
 
     async def _arun(
         self,
-        config: RunnableConfig,
         channel_id: str,
         text: str,
         post_at: str,
@@ -55,8 +53,9 @@ class SlackScheduleMessage(SlackBaseTool):
         Raises:
             Exception: If an error occurs scheduling the message
         """
-        context = self.context_from_config(config)
-        client = self.get_client(context.config.get("slack_bot_token"))
+        context = self.get_context()
+        skill_config = context.agent.skill_config(self.category)
+        client = self.get_client(skill_config.get("slack_bot_token"))
 
         try:
             # Convert ISO datetime string to Unix timestamp

intentkit/skills/slack/send_message.py

@@ -1,6 +1,5 @@
 from typing import Optional, Type
 
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.slack.base import SlackBaseTool, SlackMessage
@@ -30,7 +29,6 @@ class SlackSendMessage(SlackBaseTool):
 
     async def _arun(
         self,
-        config: RunnableConfig,
         channel_id: str,
         text: str,
         thread_ts: Optional[str] = None,
@@ -49,8 +47,9 @@ class SlackSendMessage(SlackBaseTool):
         Raises:
             Exception: If an error occurs sending the message
         """
-        context = self.context_from_config(config)
-        client = self.get_client(context.config.get("slack_bot_token"))
+        context = self.get_context()
+        skill_config = context.agent.skill_config(self.category)
+        client = self.get_client(skill_config.get("slack_bot_token"))
 
         try:
             # Prepare message parameters

intentkit/skills/supabase/delete_data.py

@@ -1,7 +1,6 @@
 import logging
 from typing import Any, Dict, Type
 
-from langchain_core.runnables import RunnableConfig
 from langchain_core.tools import ToolException
 from pydantic import BaseModel, Field
 from supabase import Client, create_client
@@ -42,18 +41,16 @@ class SupabaseDeleteData(SupabaseBaseTool):
         table: str,
         filters: Dict[str, Any],
         returning: str = "*",
-        config: RunnableConfig = None,
         **kwargs,
     ):
         try:
-            context = self.context_from_config(config)
+            context = self.get_context()
+            skill_config = context.agent.skill_config(self.category)
 
             # Validate table access for public mode
             self.validate_table_access(table, context)
 
-            supabase_url, supabase_key = self.get_supabase_config(
-                context.config, context
-            )
+            supabase_url, supabase_key = self.get_supabase_config(skill_config, context)
 
             # Create Supabase client
             supabase: Client = create_client(supabase_url, supabase_key)

intentkit/skills/supabase/fetch_data.py

@@ -1,7 +1,6 @@
 import logging
 from typing import Any, Dict, Optional, Type
 
-from langchain_core.runnables import RunnableConfig
 from langchain_core.tools import ToolException
 from pydantic import BaseModel, Field
 from supabase import Client, create_client
@@ -57,14 +56,12 @@ class SupabaseFetchData(SupabaseBaseTool):
         ascending: bool = True,
         limit: Optional[int] = None,
         offset: Optional[int] = None,
-        config: RunnableConfig = None,
         **kwargs,
     ):
         try:
-            context = self.context_from_config(config)
-            supabase_url, supabase_key = self.get_supabase_config(
-                context.config, context
-            )
+            context = self.get_context()
+            skill_config = context.agent.skill_config(self.category)
+            supabase_url, supabase_key = self.get_supabase_config(skill_config, context)
 
             # Create Supabase client
             supabase: Client = create_client(supabase_url, supabase_key)

intentkit/skills/supabase/insert_data.py

@@ -1,7 +1,6 @@
 import logging
 from typing import Any, Dict, List, Type, Union
 
-from langchain_core.runnables import RunnableConfig
 from langchain_core.tools import ToolException
 from pydantic import BaseModel, Field
 from supabase import Client, create_client
@@ -42,18 +41,16 @@ class SupabaseInsertData(SupabaseBaseTool):
         table: str,
         data: Union[Dict[str, Any], List[Dict[str, Any]]],
         returning: str = "*",
-        config: RunnableConfig = None,
         **kwargs,
     ):
         try:
-            context = self.context_from_config(config)
+            context = self.get_context()
+            skill_config = context.agent.skill_config(self.category)
 
             # Validate table access for public mode
             self.validate_table_access(table, context)
 
-            supabase_url, supabase_key = self.get_supabase_config(
-                context.config, context
-            )
+            supabase_url, supabase_key = self.get_supabase_config(skill_config, context)
 
             # Create Supabase client
             supabase: Client = create_client(supabase_url, supabase_key)

intentkit/skills/supabase/invoke_function.py

@@ -1,7 +1,6 @@
 import logging
 from typing import Any, Dict, Optional, Type
 
-from langchain_core.runnables import RunnableConfig
 from langchain_core.tools import ToolException
 from pydantic import BaseModel, Field
 from supabase import Client, create_client
@@ -41,14 +40,12 @@ class SupabaseInvokeFunction(SupabaseBaseTool):
         function_name: str,
         parameters: Optional[Dict[str, Any]] = None,
         headers: Optional[Dict[str, str]] = None,
-        config: RunnableConfig = None,
         **kwargs,
     ):
         try:
-            context = self.context_from_config(config)
-            supabase_url, supabase_key = self.get_supabase_config(
-                context.config, context
-            )
+            context = self.get_context()
+            skill_config = context.agent.skill_config(self.category)
+            supabase_url, supabase_key = self.get_supabase_config(skill_config, context)
 
             # Create Supabase client
             supabase: Client = create_client(supabase_url, supabase_key)

intentkit/skills/supabase/update_data.py

@@ -1,7 +1,6 @@
 import logging
 from typing import Any, Dict, Type
 
-from langchain_core.runnables import RunnableConfig
 from langchain_core.tools import ToolException
 from pydantic import BaseModel, Field
 from supabase import Client, create_client
@@ -45,18 +44,16 @@ class SupabaseUpdateData(SupabaseBaseTool):
         data: Dict[str, Any],
         filters: Dict[str, Any],
         returning: str = "*",
-        config: RunnableConfig = None,
         **kwargs,
     ):
         try:
-            context = self.context_from_config(config)
+            context = self.get_context()
+            skill_config = context.agent.skill_config(self.category)
 
             # Validate table access for public mode
             self.validate_table_access(table, context)
 
-            supabase_url, supabase_key = self.get_supabase_config(
-                context.config, context
-            )
+            supabase_url, supabase_key = self.get_supabase_config(skill_config, context)
 
             # Create Supabase client
             supabase: Client = create_client(supabase_url, supabase_key)

intentkit/skills/supabase/upsert_data.py

@@ -1,7 +1,6 @@
 import logging
 from typing import Any, Dict, List, Type, Union
 
-from langchain_core.runnables import RunnableConfig
 from langchain_core.tools import ToolException
 from pydantic import BaseModel, Field
 from supabase import Client, create_client
@@ -47,18 +46,16 @@ class SupabaseUpsertData(SupabaseBaseTool):
         data: Union[Dict[str, Any], List[Dict[str, Any]]],
         on_conflict: str,
         returning: str = "*",
-        config: RunnableConfig = None,
         **kwargs,
     ):
         try:
-            context = self.context_from_config(config)
+            context = self.get_context()
+            skill_config = context.agent.skill_config(self.category)
 
             # Validate table access for public mode
             self.validate_table_access(table, context)
 
-            supabase_url, supabase_key = self.get_supabase_config(
-                context.config, context
-            )
+            supabase_url, supabase_key = self.get_supabase_config(skill_config, context)
 
             # Create Supabase client
             supabase: Client = create_client(supabase_url, supabase_key)

intentkit/skills/system/add_autonomous_task.py

@@ -1,6 +1,5 @@
 from typing import Optional
 
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.models.agent import AgentAutonomous
@@ -60,7 +59,6 @@ class AddAutonomousTask(SystemBaseTool):
         minutes: Optional[int] = None,
         cron: Optional[str] = None,
         prompt: str = "",
-        config: RunnableConfig = None,
         **kwargs,
     ) -> AddAutonomousTaskOutput:
         """Add an autonomous task to the agent.
@@ -76,7 +74,7 @@ class AddAutonomousTask(SystemBaseTool):
         Returns:
             AddAutonomousTaskOutput: The created task
         """
-        context = self.context_from_config(config)
+        context = self.get_context()
         agent_id = context.agent_id
 
         task = AgentAutonomous(

intentkit/skills/system/delete_autonomous_task.py

@@ -1,4 +1,3 @@
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.system.base import SystemBaseTool
@@ -34,7 +33,6 @@ class DeleteAutonomousTask(SystemBaseTool):
     async def _arun(
         self,
         task_id: str,
-        config: RunnableConfig,
         **kwargs,
     ) -> DeleteAutonomousTaskOutput:
         """Delete an autonomous task from the agent.
@@ -46,7 +44,7 @@ class DeleteAutonomousTask(SystemBaseTool):
         Returns:
             DeleteAutonomousTaskOutput: Confirmation of deletion
         """
-        context = self.context_from_config(config)
+        context = self.get_context()
         agent_id = context.agent_id
 
         await self.skill_store.delete_autonomous_task(agent_id, task_id)

intentkit/skills/system/edit_autonomous_task.py

@@ -1,6 +1,5 @@
 from typing import Optional
 
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.models.agent import AgentAutonomous
@@ -68,7 +67,6 @@ class EditAutonomousTask(SystemBaseTool):
         cron: Optional[str] = None,
         prompt: Optional[str] = None,
         enabled: Optional[bool] = None,
-        config: RunnableConfig = None,
         **kwargs,
     ) -> EditAutonomousTaskOutput:
         """Edit an autonomous task for the agent.
@@ -86,7 +84,7 @@ class EditAutonomousTask(SystemBaseTool):
         Returns:
             EditAutonomousTaskOutput: The updated task
         """
-        context = self.context_from_config(config)
+        context = self.get_context()
         agent_id = context.agent_id
 
         if minutes is not None and cron is not None:

intentkit/skills/system/list_autonomous_tasks.py

@@ -1,6 +1,5 @@
 from typing import List
 
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.models.agent import AgentAutonomous
@@ -33,7 +32,6 @@ class ListAutonomousTasks(SystemBaseTool):
 
     async def _arun(
         self,
-        config: RunnableConfig,
         **kwargs,
     ) -> ListAutonomousTasksOutput:
         """List autonomous tasks for the agent.
@@ -44,7 +42,7 @@ class ListAutonomousTasks(SystemBaseTool):
         Returns:
             ListAutonomousTasksOutput: List of autonomous tasks
         """
-        context = self.context_from_config(config)
+        context = self.get_context()
         agent_id = context.agent_id
 
         tasks = await self.skill_store.list_autonomous_tasks(agent_id)

intentkit/skills/system/read_agent_api_key.py

@@ -1,4 +1,3 @@
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.system.base import SystemBaseTool
@@ -36,10 +35,10 @@ class ReadAgentApiKey(SystemBaseTool):
     )
     args_schema = ReadAgentApiKeyInput
 
-    async def _arun(self, config: RunnableConfig, **kwargs) -> ReadAgentApiKeyOutput:
+    async def _arun(self, **kwargs) -> ReadAgentApiKeyOutput:
         """Retrieve or generate an API key for the agent."""
         # Get context from runnable config to access agent.id
-        context = self.context_from_config(config)
+        context = self.get_context()
         agent_id = context.agent_id
 
         # Get agent data from skill store

intentkit/skills/system/regenerate_agent_api_key.py

@@ -1,4 +1,3 @@
-from langchain_core.runnables import RunnableConfig
 from pydantic import BaseModel, Field
 
 from intentkit.skills.system.base import SystemBaseTool
@@ -38,12 +37,10 @@ class RegenerateAgentApiKey(SystemBaseTool):
     )
     args_schema = RegenerateAgentApiKeyInput
 
-    async def _arun(
-        self, config: RunnableConfig, **kwargs
-    ) -> RegenerateAgentApiKeyOutput:
+    async def _arun(self, **kwargs) -> RegenerateAgentApiKeyOutput:
         """Generate and set a new API key for the agent."""
         # Get context from runnable config to access agent.id
-        context = self.context_from_config(config)
+        context = self.get_context()
         agent_id = context.agent_id
 
         # Get agent data from skill store

intentkit/skills/tavily/base.py

@@ -1,9 +1,10 @@
 from typing import 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
+from intentkit.skills.base import IntentKitSkill
 
 
 class TavilyBaseTool(IntentKitSkill):
@@ -16,11 +17,19 @@ class TavilyBaseTool(IntentKitSkill):
         description="The skill store for persisting data"
     )
 
-    def get_api_key(self, context: SkillContext) -> str:
-        skill_config = context.config
-        if skill_config.get("api_key_provider") == "agent_owner":
+    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")
+        # for backward compatibility, may only have api_key in skill_config
+        elif skill_config.get("api_key"):
             return skill_config.get("api_key")
-        return self.skill_store.get_system_config("tavily_api_key")
+        else:
+            raise ToolException(
+                f"Invalid API key provider: {api_key_provider}, or no api_key in config"
+            )
 
     @property
     def category(self) -> str:

intentkit/skills/tavily/tavily_extract.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
@@ -50,7 +49,6 @@ class TavilyExtract(TavilyBaseTool):
         urls: str,
         include_images: bool = False,
         extract_depth: str = "basic",
-        config: RunnableConfig = None,
         **kwargs,
     ) -> str:
         """Implementation of the Tavily extract tool.
@@ -64,23 +62,24 @@ class TavilyExtract(TavilyBaseTool):
         Returns:
             str: Formatted extraction results with content from the URL.
         """
-        context = self.context_from_config(config)
+        context = self.get_context()
+        skill_config = context.agent.skill_config(self.category)
         logger.debug(
             f"tavily_extract.py: Running web extraction 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)
+        api_key = self.get_api_key()
         if not api_key:
             return "Error: No Tavily API key provided in the configuration."