intentkit 0.6.0.dev13__py3-none-any.whl → 0.6.0.dev14__py3-none-any.whl

intentkit/skills/supabase/update_data.py

@@ -0,0 +1,105 @@
+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
+
+from intentkit.skills.supabase.base import SupabaseBaseTool
+
+NAME = "supabase_update_data"
+PROMPT = "Update existing data in a Supabase table with filtering conditions."
+
+logger = logging.getLogger(__name__)
+
+
+class SupabaseUpdateDataInput(BaseModel):
+    """Input for SupabaseUpdateData tool."""
+
+    table: str = Field(description="The name of the table to update data in")
+    data: Dict[str, Any] = Field(
+        description="The data to update (key-value pairs of columns and new values)"
+    )
+    filters: Dict[str, Any] = Field(
+        description="Dictionary of filters to identify which records to update (e.g., {'id': 123})"
+    )
+    returning: str = Field(
+        default="*", description="Columns to return after update (default: '*' for all)"
+    )
+
+
+class SupabaseUpdateData(SupabaseBaseTool):
+    """Tool for updating data in Supabase tables.
+
+    This tool allows updating records in Supabase tables based on filter conditions.
+    """
+
+    name: str = NAME
+    description: str = PROMPT
+    args_schema: Type[BaseModel] = SupabaseUpdateDataInput
+
+    async def _arun(
+        self,
+        table: str,
+        data: Dict[str, Any],
+        filters: Dict[str, Any],
+        returning: str = "*",
+        config: RunnableConfig = None,
+        **kwargs,
+    ):
+        try:
+            context = self.context_from_config(config)
+
+            # Validate table access for public mode
+            self.validate_table_access(table, context)
+
+            supabase_url, supabase_key = self.get_supabase_config(context.config)
+
+            # Create Supabase client
+            supabase: Client = create_client(supabase_url, supabase_key)
+
+            # Start building the update query
+            query = supabase.table(table).update(data)
+
+            # Apply filters to identify which records to update
+            for column, value in filters.items():
+                if isinstance(value, dict):
+                    # Handle complex filters like {'gte': 18}
+                    for operator, filter_value in value.items():
+                        if operator == "eq":
+                            query = query.eq(column, filter_value)
+                        elif operator == "neq":
+                            query = query.neq(column, filter_value)
+                        elif operator == "gt":
+                            query = query.gt(column, filter_value)
+                        elif operator == "gte":
+                            query = query.gte(column, filter_value)
+                        elif operator == "lt":
+                            query = query.lt(column, filter_value)
+                        elif operator == "lte":
+                            query = query.lte(column, filter_value)
+                        elif operator == "like":
+                            query = query.like(column, filter_value)
+                        elif operator == "ilike":
+                            query = query.ilike(column, filter_value)
+                        elif operator == "in":
+                            query = query.in_(column, filter_value)
+                        else:
+                            logger.warning(f"Unknown filter operator: {operator}")
+                else:
+                    # Simple equality filter
+                    query = query.eq(column, value)
+
+            # Execute the update
+            response = query.execute()
+
+            return {
+                "success": True,
+                "data": response.data,
+                "count": len(response.data) if response.data else 0,
+            }
+
+        except Exception as e:
+            logger.error(f"Error updating data in Supabase: {str(e)}")
+            raise ToolException(f"Failed to update data in table '{table}': {str(e)}")

intentkit/skills/supabase/upsert_data.py

@@ -0,0 +1,77 @@
+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
+
+from intentkit.skills.supabase.base import SupabaseBaseTool
+
+NAME = "supabase_upsert_data"
+PROMPT = (
+    "Upsert (insert or update) data in a Supabase table based on conflict resolution."
+)
+
+logger = logging.getLogger(__name__)
+
+
+class SupabaseUpsertDataInput(BaseModel):
+    """Input for SupabaseUpsertData tool."""
+
+    table: str = Field(description="The name of the table to upsert data into")
+    data: Union[Dict[str, Any], List[Dict[str, Any]]] = Field(
+        description="The data to upsert. Can be a single object or a list of objects"
+    )
+    on_conflict: str = Field(
+        description="The column(s) to use for conflict resolution (e.g., 'id' or 'email,username')"
+    )
+    returning: str = Field(
+        default="*", description="Columns to return after upsert (default: '*' for all)"
+    )
+
+
+class SupabaseUpsertData(SupabaseBaseTool):
+    """Tool for upserting data in Supabase tables.
+
+    This tool allows inserting new records or updating existing ones based on conflict resolution.
+    """
+
+    name: str = NAME
+    description: str = PROMPT
+    args_schema: Type[BaseModel] = SupabaseUpsertDataInput
+
+    async def _arun(
+        self,
+        table: str,
+        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)
+
+            # Validate table access for public mode
+            self.validate_table_access(table, context)
+
+            supabase_url, supabase_key = self.get_supabase_config(context.config)
+
+            # Create Supabase client
+            supabase: Client = create_client(supabase_url, supabase_key)
+
+            # Upsert data
+            response = (
+                supabase.table(table).upsert(data, on_conflict=on_conflict).execute()
+            )
+
+            return {
+                "success": True,
+                "data": response.data,
+                "count": len(response.data) if response.data else 0,
+            }
+
+        except Exception as e:
+            logger.error(f"Error upserting data in Supabase: {str(e)}")
+            raise ToolException(f"Failed to upsert data in table '{table}': {str(e)}")

intentkit/skills/system/read_agent_api_key.py

@@ -40,7 +40,7 @@ class ReadAgentApiKey(SystemBaseTool):
         """Retrieve or generate an API key for the agent."""
         # Get context from runnable config to access agent.id
         context = self.context_from_config(config)
-        agent_id = context.agent.id
+        agent_id = context.agent_id
 
         # Get agent data from skill store
         agent_data = await self.skill_store.get_agent_data(agent_id)

intentkit/skills/system/regenerate_agent_api_key.py

@@ -44,7 +44,7 @@ class RegenerateAgentApiKey(SystemBaseTool):
         """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)
-        agent_id = context.agent.id
+        agent_id = context.agent_id
 
         # Get agent data from skill store
         agent_data = await self.skill_store.get_agent_data(agent_id)

intentkit/skills/token/base.py

@@ -1,10 +1,9 @@
 """Base class for token-related skills."""
 
 import logging
-from typing import Any, Dict, Optional
+from typing import Any, Dict
 
 import aiohttp
-from langchain_core.runnables import RunnableConfig
 
 from intentkit.abstracts.skill import SkillStoreABC
 from intentkit.skills.base import IntentKitSkill, SkillContext
@@ -42,43 +41,6 @@ class TokenBaseTool(IntentKitSkill):
             return skill_config.get("api_key")
         return self.skill_store.get_system_config("moralis_api_key")
 
-    def context_from_config(self, config: Optional[RunnableConfig] = None) -> Any:
-        """Extract context from the runnable config."""
-        if not config:
-            logger.error("No config provided to context_from_config")
-            return None
-
-        if "configurable" not in config:
-            logger.error("'configurable' not in config")
-            return None
-
-        if "agent" not in config["configurable"]:
-            logger.error("'agent' not in config['configurable']")
-            return None
-
-        agent = config["configurable"].get("agent")
-        category_config = None
-
-        if agent.skills:
-            category_config = agent.skills.get(self.category)
-
-        if not category_config:
-            category_config = getattr(agent, self.category + "_config", {})
-
-        if not category_config:
-            category_config = {}
-
-        from intentkit.skills.base import SkillContext
-
-        context = SkillContext(
-            agent=agent,
-            config=category_config,
-            user_id=config["configurable"].get("user_id"),
-            entrypoint=config["configurable"].get("entrypoint"),
-        )
-
-        return context
-
     def _prepare_params(self, params: Dict[str, Any]) -> Dict[str, Any]:
         """Convert boolean values to lowercase strings for API compatibility.
 

intentkit/skills/twitter/follow_user.py

@@ -41,7 +41,7 @@ class TwitterFollowUser(TwitterBaseTool):
         try:
             context = self.context_from_config(config)
             twitter = get_twitter_client(
-                agent_id=context.agent.id,
+                agent_id=context.agent_id,
                 skill_store=self.skill_store,
                 config=context.config,
             )
@@ -50,7 +50,7 @@ class TwitterFollowUser(TwitterBaseTool):
             # Check rate limit only when not using OAuth
             if not twitter.use_key:
                 await self.check_rate_limit(
-                    context.agent.id, max_requests=5, interval=15
+                    context.agent_id, max_requests=5, interval=15
                 )
 
             # Follow the user using tweepy client
@@ -66,4 +66,4 @@ class TwitterFollowUser(TwitterBaseTool):
 
         except Exception as e:
             logger.error("Error following user: %s", str(e))
-            raise type(e)(f"[agent:{context.agent.id}]: {e}") from e
+            raise type(e)(f"[agent:{context.agent_id}]: {e}") from e

intentkit/skills/twitter/get_mentions.py

@@ -45,7 +45,7 @@ class TwitterGetMentions(TwitterBaseTool):
         try:
             context = self.context_from_config(config)
             twitter = get_twitter_client(
-                agent_id=context.agent.id,
+                agent_id=context.agent_id,
                 skill_store=self.skill_store,
                 config=context.config,
             )
@@ -56,14 +56,14 @@ class TwitterGetMentions(TwitterBaseTool):
             # Check rate limit only when not using OAuth
             if not twitter.use_key:
                 await self.check_rate_limit(
-                    context.agent.id,
+                    context.agent_id,
                     max_requests=1,
                     interval=59,  # TODO: tmp to 59, back to 240 later
                 )
 
             # get since id from store
             last = await self.skill_store.get_agent_skill_data(
-                context.agent.id, self.name, "last"
+                context.agent_id, self.name, "last"
             )
             last = last or {}
             max_results = 10
@@ -114,11 +114,11 @@ class TwitterGetMentions(TwitterBaseTool):
             if mentions.get("meta") and mentions["meta"].get("newest_id"):
                 last["since_id"] = mentions["meta"].get("newest_id")
                 await self.skill_store.save_agent_skill_data(
-                    context.agent.id, self.name, "last", last
+                    context.agent_id, self.name, "last", last
                 )
 
             return mentions
 
         except Exception as e:
-            logger.error(f"[agent:{context.agent.id}]: {e}")
-            raise type(e)(f"[agent:{context.agent.id}]: {e}") from e
+            logger.error(f"[agent:{context.agent_id}]: {e}")
+            raise type(e)(f"[agent:{context.agent_id}]: {e}") from e

intentkit/skills/twitter/get_timeline.py

@@ -44,7 +44,7 @@ class TwitterGetTimeline(TwitterBaseTool):
 
             context = self.context_from_config(config)
             twitter = get_twitter_client(
-                agent_id=context.agent.id,
+                agent_id=context.agent_id,
                 skill_store=self.skill_store,
                 config=context.config,
             )
@@ -53,12 +53,12 @@ class TwitterGetTimeline(TwitterBaseTool):
             # Check rate limit only when not using OAuth
             if not twitter.use_key:
                 await self.check_rate_limit(
-                    context.agent.id, max_requests=3, interval=60 * 24
+                    context.agent_id, max_requests=3, interval=60 * 24
                 )
 
             # get since id from store
             last = await self.skill_store.get_agent_skill_data(
-                context.agent.id, self.name, "last"
+                context.agent_id, self.name, "last"
             )
             last = last or {}
             since_id = last.get("since_id")
@@ -101,11 +101,11 @@ class TwitterGetTimeline(TwitterBaseTool):
             if timeline.get("meta") and timeline["meta"].get("newest_id"):
                 last["since_id"] = timeline["meta"]["newest_id"]
                 await self.skill_store.save_agent_skill_data(
-                    context.agent.id, self.name, "last", last
+                    context.agent_id, self.name, "last", last
                 )
 
             return timeline
 
         except Exception as e:
             logger.error("Error getting timeline: %s", str(e))
-            raise type(e)(f"[agent:{context.agent.id}]: {e}") from e
+            raise type(e)(f"[agent:{context.agent_id}]: {e}") from e

intentkit/skills/twitter/get_user_by_username.py

@@ -42,7 +42,7 @@ class TwitterGetUserByUsername(TwitterBaseTool):
         try:
             context = self.context_from_config(config)
             twitter = get_twitter_client(
-                agent_id=context.agent.id,
+                agent_id=context.agent_id,
                 skill_store=self.skill_store,
                 config=context.config,
             )
@@ -51,7 +51,7 @@ class TwitterGetUserByUsername(TwitterBaseTool):
             # Check rate limit only when not using OAuth
             if not twitter.use_key:
                 await self.check_rate_limit(
-                    context.agent.id, max_requests=3, interval=60 * 24
+                    context.agent_id, max_requests=3, interval=60 * 24
                 )
 
             user_data = await client.get_user(
@@ -81,4 +81,4 @@ class TwitterGetUserByUsername(TwitterBaseTool):
 
         except Exception as e:
             logger.error(f"Error getting user by username: {str(e)}")
-            raise type(e)(f"[agent:{context.agent.id}]: {e}") from e
+            raise type(e)(f"[agent:{context.agent_id}]: {e}") from e

intentkit/skills/twitter/get_user_tweets.py

@@ -58,7 +58,7 @@ class TwitterGetUserTweets(TwitterBaseTool):
 
             context = self.context_from_config(config)
             twitter = get_twitter_client(
-                agent_id=context.agent.id,
+                agent_id=context.agent_id,
                 skill_store=self.skill_store,
                 config=context.config,
             )
@@ -67,12 +67,12 @@ class TwitterGetUserTweets(TwitterBaseTool):
             # Check rate limit only when not using OAuth
             if not twitter.use_key:
                 await self.check_rate_limit(
-                    context.agent.id, max_requests=3, interval=60 * 24
+                    context.agent_id, max_requests=3, interval=60 * 24
                 )
 
             # get since id from store
             last = await self.skill_store.get_agent_skill_data(
-                context.agent.id, self.name, user_id
+                context.agent_id, self.name, user_id
             )
             last = last or {}
             since_id = last.get("since_id")
@@ -113,11 +113,11 @@ class TwitterGetUserTweets(TwitterBaseTool):
             if tweets.get("meta") and tweets["meta"].get("newest_id"):
                 last["since_id"] = tweets["meta"]["newest_id"]
                 await self.skill_store.save_agent_skill_data(
-                    context.agent.id, self.name, user_id, last
+                    context.agent_id, self.name, user_id, last
                 )
 
             return tweets
 
         except Exception as e:
             logger.error("Error getting user tweets: %s", str(e))
-            raise type(e)(f"[agent:{context.agent.id}]: {e}") from e
+            raise type(e)(f"[agent:{context.agent_id}]: {e}") from e

intentkit/skills/twitter/like_tweet.py

@@ -39,7 +39,7 @@ class TwitterLikeTweet(TwitterBaseTool):
         try:
             context = self.context_from_config(config)
             twitter = get_twitter_client(
-                agent_id=context.agent.id,
+                agent_id=context.agent_id,
                 skill_store=self.skill_store,
                 config=context.config,
             )
@@ -48,7 +48,7 @@ class TwitterLikeTweet(TwitterBaseTool):
             # Check rate limit only when not using OAuth
             if not twitter.use_key:
                 await self.check_rate_limit(
-                    context.agent.id, max_requests=100, interval=1440
+                    context.agent_id, max_requests=100, interval=1440
                 )
 
             # Like the tweet using tweepy client
@@ -62,4 +62,4 @@ class TwitterLikeTweet(TwitterBaseTool):
 
         except Exception as e:
             logger.error(f"Error liking tweet: {str(e)}")
-            raise type(e)(f"[agent:{context.agent.id}]: {e}") from e
+            raise type(e)(f"[agent:{context.agent_id}]: {e}") from e

intentkit/skills/twitter/post_tweet.py

@@ -54,7 +54,7 @@ class TwitterPostTweet(TwitterBaseTool):
         try:
             context = self.context_from_config(config)
             twitter = get_twitter_client(
-                agent_id=context.agent.id,
+                agent_id=context.agent_id,
                 skill_store=self.skill_store,
                 config=context.config,
             )
@@ -63,7 +63,7 @@ class TwitterPostTweet(TwitterBaseTool):
             # Check rate limit only when not using OAuth
             if not twitter.use_key:
                 await self.check_rate_limit(
-                    context.agent.id, max_requests=24, interval=1440
+                    context.agent_id, max_requests=24, interval=1440
                 )
 
             media_ids = []
@@ -71,7 +71,7 @@ class TwitterPostTweet(TwitterBaseTool):
             # Handle image upload if provided
             if image:
                 # Use the TwitterClient method to upload the image
-                media_ids = await twitter.upload_media(context.agent.id, image)
+                media_ids = await twitter.upload_media(context.agent_id, image)
 
             # Post tweet using tweepy client
             tweet_params = {"text": text, "user_auth": twitter.use_key}
@@ -87,4 +87,4 @@ class TwitterPostTweet(TwitterBaseTool):
 
         except Exception as e:
             logger.error(f"Error posting tweet: {str(e)}")
-            raise type(e)(f"[agent:{context.agent.id}]: {e}") from e
+            raise type(e)(f"[agent:{context.agent_id}]: {e}") from e

intentkit/skills/twitter/reply_tweet.py

@@ -56,7 +56,7 @@ class TwitterReplyTweet(TwitterBaseTool):
         try:
             context = self.context_from_config(config)
             twitter = get_twitter_client(
-                agent_id=context.agent.id,
+                agent_id=context.agent_id,
                 skill_store=self.skill_store,
                 config=context.config,
             )
@@ -65,7 +65,7 @@ class TwitterReplyTweet(TwitterBaseTool):
             # Check rate limit only when not using OAuth
             if not twitter.use_key:
                 await self.check_rate_limit(
-                    context.agent.id, max_requests=48, interval=1440
+                    context.agent_id, max_requests=48, interval=1440
                 )
 
             media_ids = []
@@ -73,7 +73,7 @@ class TwitterReplyTweet(TwitterBaseTool):
             # Handle image upload if provided
             if image:
                 # Use the TwitterClient method to upload the image
-                media_ids = await twitter.upload_media(context.agent.id, image)
+                media_ids = await twitter.upload_media(context.agent_id, image)
 
             # Post reply tweet using tweepy client
             tweet_params = {
@@ -95,4 +95,4 @@ class TwitterReplyTweet(TwitterBaseTool):
 
         except Exception as e:
             logger.error(f"Error replying to tweet: {str(e)}")
-            raise type(e)(f"[agent:{context.agent.id}]: {e}") from e
+            raise type(e)(f"[agent:{context.agent_id}]: {e}") from e

intentkit/skills/twitter/retweet.py

@@ -39,7 +39,7 @@ class TwitterRetweet(TwitterBaseTool):
         try:
             context = self.context_from_config(config)
             twitter = get_twitter_client(
-                agent_id=context.agent.id,
+                agent_id=context.agent_id,
                 skill_store=self.skill_store,
                 config=context.config,
             )
@@ -48,7 +48,7 @@ class TwitterRetweet(TwitterBaseTool):
             # Check rate limit only when not using OAuth
             if not twitter.use_key:
                 await self.check_rate_limit(
-                    context.agent.id, max_requests=5, interval=15
+                    context.agent_id, max_requests=5, interval=15
                 )
 
             # Get authenticated user's ID
@@ -73,4 +73,4 @@ class TwitterRetweet(TwitterBaseTool):
 
         except Exception as e:
             logger.error(f"Error retweeting: {str(e)}")
-            raise type(e)(f"[agent:{context.agent.id}]: {e}") from e
+            raise type(e)(f"[agent:{context.agent_id}]: {e}") from e

intentkit/skills/twitter/search_tweets.py

@@ -41,7 +41,7 @@ class TwitterSearchTweets(TwitterBaseTool):
         try:
             context = self.context_from_config(config)
             twitter = get_twitter_client(
-                agent_id=context.agent.id,
+                agent_id=context.agent_id,
                 skill_store=self.skill_store,
                 config=context.config,
             )
@@ -50,12 +50,12 @@ class TwitterSearchTweets(TwitterBaseTool):
             # Check rate limit only when not using OAuth
             if not twitter.use_key:
                 await self.check_rate_limit(
-                    context.agent.id, max_requests=3, interval=60 * 24
+                    context.agent_id, max_requests=3, interval=60 * 24
                 )
 
             # Get since_id from store to avoid duplicate results
             last = await self.skill_store.get_agent_skill_data(
-                context.agent.id, self.name, query
+                context.agent_id, self.name, query
             )
             last = last or {}
             since_id = last.get("since_id")
@@ -105,11 +105,11 @@ class TwitterSearchTweets(TwitterBaseTool):
                 last["since_id"] = tweets["meta"]["newest_id"]
                 last["timestamp"] = datetime.datetime.now().isoformat()
                 await self.skill_store.save_agent_skill_data(
-                    context.agent.id, self.name, query, last
+                    context.agent_id, self.name, query, last
                 )
 
             return tweets
 
         except Exception as e:
             logger.error(f"Error searching tweets: {str(e)}")
-            raise type(e)(f"[agent:{context.agent.id}]: {e}") from e
+            raise type(e)(f"[agent:{context.agent_id}]: {e}") from e

intentkit/skills/unrealspeech/text_to_speech.py

@@ -81,7 +81,7 @@ class TextToSpeech(UnrealSpeechBaseTool):
         # If no API key in config, try to get it from skill store
         if not api_key:
             try:
-                agent_id = context.agent.id if context and context.agent else "default"
+                agent_id = context.agent_id
                 api_key_data = await self.skill_store.get_agent_data(
                     agent_id, "unrealspeech_api_key"
                 )

intentkit/skills/web_scraper/document_indexer.py

@@ -85,10 +85,10 @@ class DocumentIndexer(WebScraperBaseTool):
             raise ValueError("Configuration is required but not provided")
 
         context = self.context_from_config(config)
-        if not context or not context.agent or not context.agent.id:
+        if not context or not context.agent_id:
             raise ValueError("Agent ID is required but not found in configuration")
 
-        agent_id = context.agent.id
+        agent_id = context.agent_id
 
         logger.info(f"[{agent_id}] Starting document indexing for title: '{title}'")
 

intentkit/skills/web_scraper/scrape_and_index.py

@@ -85,10 +85,10 @@ class ScrapeAndIndex(WebScraperBaseTool):
                 raise ValueError("Configuration is required but not provided")
 
             context = self.context_from_config(config)
-            if not context or not context.agent or not context.agent.id:
+            if not context or not context.agent_id:
                 raise ValueError("Agent ID is required but not found in configuration")
 
-            agent_id = context.agent.id
+            agent_id = context.agent_id
 
             logger.info(
                 f"[{agent_id}] Starting scrape and index operation with {len(urls)} URLs"
@@ -149,8 +149,8 @@ class ScrapeAndIndex(WebScraperBaseTool):
             try:
                 if config:
                     context = self.context_from_config(config)
-                    if context and context.agent and context.agent.id:
-                        agent_id = context.agent.id
+                    if context and context.agent_id:
+                        agent_id = context.agent_id
             except Exception:
                 pass
 
@@ -187,10 +187,10 @@ class QueryIndexedContent(WebScraperBaseTool):
                 raise ValueError("Configuration is required but not provided")
 
             context = self.context_from_config(config)
-            if not context or not context.agent or not context.agent.id:
+            if not context or not context.agent_id:
                 raise ValueError("Agent ID is required but not found in configuration")
 
-            agent_id = context.agent.id
+            agent_id = context.agent_id
 
             logger.info(f"[{agent_id}] Starting query operation: '{query}'")
 
@@ -251,8 +251,8 @@ class QueryIndexedContent(WebScraperBaseTool):
             try:
                 if config:
                     context = self.context_from_config(config)
-                    if context and context.agent and context.agent.id:
-                        agent_id = context.agent.id
+                    if context and context.agent_id:
+                        agent_id = context.agent_id
             except Exception:
                 pass