intentkit 0.6.7.dev1__tar.gz → 0.6.7.dev2__tar.gz

{intentkit-0.6.7.dev1 → intentkit-0.6.7.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: intentkit
-Version: 0.6.7.dev1
+Version: 0.6.7.dev2
 Summary: Intent-based AI Agent Platform - Core Package
 Project-URL: Homepage, https://github.com/crestal-network/intentkit
 Project-URL: Repository, https://github.com/crestal-network/intentkit

{intentkit-0.6.7.dev1 → intentkit-0.6.7.dev2}/__init__.py

@@ -3,7 +3,7 @@
 A powerful platform for building AI agents with blockchain and cryptocurrency capabilities.
 """
 
-__version__ = "0.6.7-dev1"
+__version__ = "0.6.7-dev2"
 __author__ = "hyacinthus"
 __email__ = "hyacinthus@gmail.com"
 

{intentkit-0.6.7.dev1 → intentkit-0.6.7.dev2}/abstracts/skill.py

@@ -1,7 +1,7 @@
 from abc import ABC, abstractmethod
-from typing import Any, Dict, Optional
+from typing import Any, Dict, List, Optional
 
-from intentkit.models.agent import Agent
+from intentkit.models.agent import Agent, AgentAutonomous
 from intentkit.models.agent_data import AgentData, AgentQuota
 
 
@@ -139,3 +139,43 @@ class SkillStoreABC(ABC):
             data: JSON data to store
         """
         pass
+
+    @staticmethod
+    @abstractmethod
+    async def list_autonomous_tasks(agent_id: str) -> List[AgentAutonomous]:
+        """List all autonomous tasks for an agent.
+
+        Args:
+            agent_id: ID of the agent
+
+        Returns:
+            List[AgentAutonomous]: List of autonomous task configurations
+        """
+        pass
+
+    @staticmethod
+    @abstractmethod
+    async def add_autonomous_task(
+        agent_id: str, task: AgentAutonomous
+    ) -> AgentAutonomous:
+        """Add a new autonomous task to an agent.
+
+        Args:
+            agent_id: ID of the agent
+            task: Autonomous task configuration
+
+        Returns:
+            AgentAutonomous: The created task
+        """
+        pass
+
+    @staticmethod
+    @abstractmethod
+    async def delete_autonomous_task(agent_id: str, task_id: str) -> None:
+        """Delete an autonomous task from an agent.
+
+        Args:
+            agent_id: ID of the agent
+            task_id: ID of the task to delete
+        """
+        pass

{intentkit-0.6.7.dev1 → intentkit-0.6.7.dev2}/core/agent.py

@@ -2,11 +2,11 @@ import logging
 import time
 from datetime import datetime, timedelta, timezone
 from decimal import Decimal
-from typing import Dict
+from typing import Dict, List
 
 from sqlalchemy import func, select, text, update
 
-from intentkit.models.agent import Agent, AgentTable
+from intentkit.models.agent import Agent, AgentAutonomous, AgentTable
 from intentkit.models.agent_data import AgentQuotaTable
 from intentkit.models.credit import CreditEventTable, EventType, UpstreamType
 from intentkit.models.db import get_session
@@ -274,3 +274,119 @@ async def update_agent_action_cost():
     logger.info(
         f"Finished updating action costs for {total_updated} agents in {total_time:.3f}s"
     )
+
+
+async def list_autonomous_tasks(agent_id: str) -> List[AgentAutonomous]:
+    """
+    List all autonomous tasks for an agent.
+
+    Args:
+        agent_id: ID of the agent
+
+    Returns:
+        List[AgentAutonomous]: List of autonomous task configurations
+
+    Raises:
+        IntentKitAPIError: If agent is not found
+    """
+    agent = await Agent.get(agent_id)
+    if not agent:
+        raise IntentKitAPIError(
+            400, "AgentNotFound", f"Agent with ID {agent_id} does not exist."
+        )
+
+    if not agent.autonomous:
+        return []
+
+    return agent.autonomous
+
+
+async def add_autonomous_task(agent_id: str, task: AgentAutonomous) -> AgentAutonomous:
+    """
+    Add a new autonomous task to an agent.
+
+    Args:
+        agent_id: ID of the agent
+        task: Autonomous task configuration (id will be generated if not provided)
+
+    Returns:
+        AgentAutonomous: The created task with generated ID
+
+    Raises:
+        IntentKitAPIError: If agent is not found
+    """
+    agent = await Agent.get(agent_id)
+    if not agent:
+        raise IntentKitAPIError(
+            400, "AgentNotFound", f"Agent with ID {agent_id} does not exist."
+        )
+
+    # Get current autonomous tasks
+    current_tasks = agent.autonomous or []
+    if not isinstance(current_tasks, list):
+        current_tasks = []
+
+    # Add the new task
+    current_tasks.append(task)
+
+    # Update the agent in the database
+    async with get_session() as session:
+        update_stmt = (
+            update(AgentTable)
+            .where(AgentTable.id == agent_id)
+            .values(autonomous=current_tasks)
+        )
+        await session.execute(update_stmt)
+        await session.commit()
+
+    logger.info(f"Added autonomous task {task.id} to agent {agent_id}")
+    return task
+
+
+async def delete_autonomous_task(agent_id: str, task_id: str) -> None:
+    """
+    Delete an autonomous task from an agent.
+
+    Args:
+        agent_id: ID of the agent
+        task_id: ID of the task to delete
+
+    Raises:
+        IntentKitAPIError: If agent is not found or task is not found
+    """
+    agent = await Agent.get(agent_id)
+    if not agent:
+        raise IntentKitAPIError(
+            400, "AgentNotFound", f"Agent with ID {agent_id} does not exist."
+        )
+
+    # Get current autonomous tasks
+    current_tasks = agent.autonomous or []
+    if not isinstance(current_tasks, list):
+        current_tasks = []
+
+    # Find and remove the task
+    task_found = False
+    updated_tasks = []
+    for task_data in current_tasks:
+        if isinstance(task_data, dict) and task_data.get("id") == task_id:
+            task_found = True
+            continue
+        updated_tasks.append(task_data)
+
+    if not task_found:
+        raise IntentKitAPIError(
+            404, "TaskNotFound", f"Autonomous task with ID {task_id} not found."
+        )
+
+    # Update the agent in the database
+    async with get_session() as session:
+        update_stmt = (
+            update(AgentTable)
+            .where(AgentTable.id == agent_id)
+            .values(autonomous=updated_tasks)
+        )
+        await session.execute(update_stmt)
+        await session.commit()
+
+    logger.info(f"Deleted autonomous task {task_id} from agent {agent_id}")

{intentkit-0.6.7.dev1 → intentkit-0.6.7.dev2}/core/engine.py

@@ -1012,15 +1012,3 @@ async def thread_stats(agent_id: str, chat_id: str) -> list[BaseMessage]:
         return snap.values["messages"]
     else:
         return []
-
-
-async def is_payment_required(input: ChatMessageCreate, agent: Agent) -> bool:
-    if not config.payment_enabled:
-        return False
-    payment_settings = await AppSetting.payment()
-    if payment_settings.agent_whitelist_enabled:
-        if agent.id not in payment_settings.agent_whitelist:
-            return False
-    if input.user_id and agent.owner:
-        return True
-    return False

{intentkit-0.6.7.dev1 → intentkit-0.6.7.dev2}/core/skill.py

@@ -1,8 +1,17 @@
-from typing import Any, Dict, Optional
+from typing import Any, Dict, List, Optional
 
 from intentkit.abstracts.skill import SkillStoreABC
 from intentkit.config.config import config
-from intentkit.models.agent import Agent
+from intentkit.core.agent import (
+    add_autonomous_task as _add_autonomous_task,
+)
+from intentkit.core.agent import (
+    delete_autonomous_task as _delete_autonomous_task,
+)
+from intentkit.core.agent import (
+    list_autonomous_tasks as _list_autonomous_tasks,
+)
+from intentkit.models.agent import Agent, AgentAutonomous
 from intentkit.models.agent_data import AgentData, AgentQuota
 from intentkit.models.skill import (
     AgentSkillData,
@@ -131,5 +140,42 @@ class SkillStore(SkillStoreABC):
         )
         await skill_data.save()
 
+    @staticmethod
+    async def list_autonomous_tasks(agent_id: str) -> List[AgentAutonomous]:
+        """List all autonomous tasks for an agent.
+
+        Args:
+            agent_id: ID of the agent
+
+        Returns:
+            List[AgentAutonomous]: List of autonomous task configurations
+        """
+        return await _list_autonomous_tasks(agent_id)
+
+    @staticmethod
+    async def add_autonomous_task(
+        agent_id: str, task: AgentAutonomous
+    ) -> AgentAutonomous:
+        """Add a new autonomous task to an agent.
+
+        Args:
+            agent_id: ID of the agent
+            task: Autonomous task configuration
+
+        Returns:
+            AgentAutonomous: The created task
+        """
+        return await _add_autonomous_task(agent_id, task)
+
+    @staticmethod
+    async def delete_autonomous_task(agent_id: str, task_id: str) -> None:
+        """Delete an autonomous task from an agent.
+
+        Args:
+            agent_id: ID of the agent
+            task_id: ID of the task to delete
+        """
+        await _delete_autonomous_task(agent_id, task_id)
+
 
 skill_store = SkillStore()

{intentkit-0.6.7.dev1 → intentkit-0.6.7.dev2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "intentkit"
-version = "0.6.7-dev1"
+version = "0.6.7-dev2"
 description = "Intent-based AI Agent Platform - Core Package"
 authors = [{ name = "hyacinthus", email = "hyacinthus@gmail.com" }]
 requires-python = ">=3.12"

{intentkit-0.6.7.dev1 → intentkit-0.6.7.dev2}/skills/system/__init__.py

@@ -5,7 +5,10 @@ from typing import TypedDict
 
 from intentkit.abstracts.skill import SkillStoreABC
 from intentkit.skills.base import SkillConfig, SkillOwnerState
+from intentkit.skills.system.add_autonomous_task import AddAutonomousTask
 from intentkit.skills.system.base import SystemBaseTool
+from intentkit.skills.system.delete_autonomous_task import DeleteAutonomousTask
+from intentkit.skills.system.list_autonomous_tasks import ListAutonomousTasks
 from intentkit.skills.system.read_agent_api_key import ReadAgentApiKey
 from intentkit.skills.system.regenerate_agent_api_key import RegenerateAgentApiKey
 
@@ -18,6 +21,9 @@ logger = logging.getLogger(__name__)
 class SkillStates(TypedDict):
     read_agent_api_key: SkillOwnerState
     regenerate_agent_api_key: SkillOwnerState
+    list_autonomous_tasks: SkillOwnerState
+    add_autonomous_task: SkillOwnerState
+    delete_autonomous_task: SkillOwnerState
 
 
 class Config(SkillConfig):
@@ -85,6 +91,24 @@ def get_system_skill(
                 skill_store=store,
             )
         return _cache[name]
+    elif name == "list_autonomous_tasks":
+        if name not in _cache:
+            _cache[name] = ListAutonomousTasks(
+                skill_store=store,
+            )
+        return _cache[name]
+    elif name == "add_autonomous_task":
+        if name not in _cache:
+            _cache[name] = AddAutonomousTask(
+                skill_store=store,
+            )
+        return _cache[name]
+    elif name == "delete_autonomous_task":
+        if name not in _cache:
+            _cache[name] = DeleteAutonomousTask(
+                skill_store=store,
+            )
+        return _cache[name]
     else:
         logger.warning(f"Unknown system skill: {name}")
         return None

intentkit-0.6.7.dev2/skills/system/add_autonomous_task.py

@@ -0,0 +1,91 @@
+from typing import Optional
+
+from langchain_core.runnables import RunnableConfig
+from pydantic import BaseModel, Field
+
+from intentkit.models.agent import AgentAutonomous
+from intentkit.skills.system.base import SystemBaseTool
+
+
+class AddAutonomousTaskInput(BaseModel):
+    """Input model for add_autonomous_task skill."""
+
+    name: Optional[str] = Field(
+        default=None, description="Display name of the autonomous task configuration"
+    )
+    description: Optional[str] = Field(
+        default=None, description="Description of the autonomous task configuration"
+    )
+    minutes: Optional[int] = Field(
+        default=None,
+        description="Interval in minutes between operations, mutually exclusive with cron",
+    )
+    cron: Optional[str] = Field(
+        default=None,
+        description="Cron expression for scheduling operations, mutually exclusive with minutes",
+    )
+    prompt: str = Field(description="Special prompt used during autonomous operation")
+    enabled: Optional[bool] = Field(
+        default=False, description="Whether the autonomous configuration is enabled"
+    )
+
+
+class AddAutonomousTaskOutput(BaseModel):
+    """Output model for add_autonomous_task skill."""
+
+    task: AgentAutonomous = Field(
+        description="The created autonomous task configuration"
+    )
+
+
+class AddAutonomousTask(SystemBaseTool):
+    """Skill to add a new autonomous task to an agent."""
+
+    name: str = "system_add_autonomous_task"
+    description: str = (
+        "Add a new autonomous task configuration to the agent. "
+        "Allows setting up scheduled operations with custom prompts and intervals. "
+        "The minutes and cron fields are mutually exclusive. But you must provide one of them."
+    )
+    args_schema = AddAutonomousTaskInput
+
+    async def _arun(
+        self,
+        name: Optional[str] = None,
+        description: Optional[str] = None,
+        minutes: Optional[int] = None,
+        cron: Optional[str] = None,
+        prompt: str = "",
+        enabled: Optional[bool] = False,
+        config: RunnableConfig = None,
+        **kwargs,
+    ) -> AddAutonomousTaskOutput:
+        """Add an autonomous task to the agent.
+
+        Args:
+            name: Display name of the task
+            description: Description of the task
+            minutes: Interval in minutes (mutually exclusive with cron)
+            cron: Cron expression (mutually exclusive with minutes)
+            prompt: Special prompt for autonomous operation
+            enabled: Whether the task is enabled
+            config: Runtime configuration containing agent context
+
+        Returns:
+            AddAutonomousTaskOutput: The created task
+        """
+        context = self.context_from_config(config)
+        agent_id = context["agent_id"]
+
+        task = AgentAutonomous(
+            name=name,
+            description=description,
+            minutes=minutes,
+            cron=cron,
+            prompt=prompt,
+            enabled=enabled,
+        )
+
+        created_task = await self.skill_store.add_autonomous_task(agent_id, task)
+
+        return AddAutonomousTaskOutput(task=created_task)

intentkit-0.6.7.dev2/skills/system/delete_autonomous_task.py

@@ -0,0 +1,56 @@
+from langchain_core.runnables import RunnableConfig
+from pydantic import BaseModel, Field
+
+from intentkit.skills.system.base import SystemBaseTool
+
+
+class DeleteAutonomousTaskInput(BaseModel):
+    """Input model for delete_autonomous_task skill."""
+
+    task_id: str = Field(
+        description="The unique identifier of the autonomous task to delete"
+    )
+
+
+class DeleteAutonomousTaskOutput(BaseModel):
+    """Output model for delete_autonomous_task skill."""
+
+    success: bool = Field(
+        description="Whether the task was successfully deleted", default=True
+    )
+    message: str = Field(description="Confirmation message about the deletion")
+
+
+class DeleteAutonomousTask(SystemBaseTool):
+    """Skill to delete an autonomous task from an agent."""
+
+    name: str = "system_delete_autonomous_task"
+    description: str = (
+        "Delete an autonomous task configuration from the agent. "
+        "Requires the task ID to identify which task to remove."
+    )
+    args_schema = DeleteAutonomousTaskInput
+
+    async def _arun(
+        self,
+        task_id: str,
+        config: RunnableConfig,
+        **kwargs,
+    ) -> DeleteAutonomousTaskOutput:
+        """Delete an autonomous task from the agent.
+
+        Args:
+            task_id: The ID of the task to delete
+            config: Runtime configuration containing agent context
+
+        Returns:
+            DeleteAutonomousTaskOutput: Confirmation of deletion
+        """
+        context = self.context_from_config(config)
+        agent_id = context["agent_id"]
+
+        await self.skill_store.delete_autonomous_task(agent_id, task_id)
+
+        return DeleteAutonomousTaskOutput(
+            success=True, message=f"Successfully deleted autonomous task {task_id}"
+        )

intentkit-0.6.7.dev2/skills/system/list_autonomous_tasks.py

@@ -0,0 +1,52 @@
+from typing import List
+
+from langchain_core.runnables import RunnableConfig
+from pydantic import BaseModel, Field
+
+from intentkit.models.agent import AgentAutonomous
+from intentkit.skills.system.base import SystemBaseTool
+
+
+class ListAutonomousTasksInput(BaseModel):
+    """Input model for list_autonomous_tasks skill."""
+
+    pass
+
+
+class ListAutonomousTasksOutput(BaseModel):
+    """Output model for list_autonomous_tasks skill."""
+
+    tasks: List[AgentAutonomous] = Field(
+        description="List of autonomous task configurations for the agent"
+    )
+
+
+class ListAutonomousTasks(SystemBaseTool):
+    """Skill to list all autonomous tasks for an agent."""
+
+    name: str = "system_list_autonomous_tasks"
+    description: str = (
+        "List all autonomous task configurations for the agent. "
+        "Returns details about each task including scheduling, prompts, and status."
+    )
+    args_schema = ListAutonomousTasksInput
+
+    async def _arun(
+        self,
+        config: RunnableConfig,
+        **kwargs,
+    ) -> ListAutonomousTasksOutput:
+        """List autonomous tasks for the agent.
+
+        Args:
+            config: Runtime configuration containing agent context
+
+        Returns:
+            ListAutonomousTasksOutput: List of autonomous tasks
+        """
+        context = self.context_from_config(config)
+        agent_id = context["agent_id"]
+
+        tasks = await self.skill_store.list_autonomous_tasks(agent_id)
+
+        return ListAutonomousTasksOutput(tasks=tasks)

{intentkit-0.6.7.dev1 → intentkit-0.6.7.dev2}/skills/system/schema.json

@@ -46,6 +46,48 @@
           ],
           "description": "Generate a new API key for the agent, replacing any existing key.",
           "default": "private"
+        },
+        "list_autonomous_tasks": {
+          "type": "string",
+          "title": "List Autonomous Tasks",
+          "enum": [
+            "disabled",
+            "private"
+          ],
+          "x-enum-title": [
+            "Disabled",
+            "Agent Owner Only"
+          ],
+          "description": "List all autonomous task configurations for the agent.",
+          "default": "private"
+        },
+        "add_autonomous_task": {
+          "type": "string",
+          "title": "Add Autonomous Task",
+          "enum": [
+            "disabled",
+            "private"
+          ],
+          "x-enum-title": [
+            "Disabled",
+            "Agent Owner Only"
+          ],
+          "description": "Add a new autonomous task configuration to the agent.",
+          "default": "private"
+        },
+        "delete_autonomous_task": {
+          "type": "string",
+          "title": "Delete Autonomous Task",
+          "enum": [
+            "disabled",
+            "private"
+          ],
+          "x-enum-title": [
+            "Disabled",
+            "Agent Owner Only"
+          ],
+          "description": "Delete an autonomous task configuration from the agent.",
+          "default": "private"
         }
       }
     }