intentkit 0.6.7.dev6__tar.gz → 0.6.7.dev7__tar.gz

{intentkit-0.6.7.dev6 → intentkit-0.6.7.dev7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: intentkit
-Version: 0.6.7.dev6
+Version: 0.6.7.dev7
 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.dev6 → intentkit-0.6.7.dev7}/__init__.py

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

{intentkit-0.6.7.dev6 → intentkit-0.6.7.dev7}/abstracts/skill.py

@@ -179,3 +179,20 @@ class SkillStoreABC(ABC):
             task_id: ID of the task to delete
         """
         pass
+
+    @staticmethod
+    @abstractmethod
+    async def update_autonomous_task(
+        agent_id: str, task_id: str, task_updates: dict
+    ) -> AgentAutonomous:
+        """Update an autonomous task for an agent.
+
+        Args:
+            agent_id: ID of the agent
+            task_id: ID of the task to update
+            task_updates: Dictionary containing fields to update
+
+        Returns:
+            AgentAutonomous: The updated task
+        """
+        pass

{intentkit-0.6.7.dev6 → intentkit-0.6.7.dev7}/core/agent.py

@@ -396,3 +396,69 @@ async def delete_autonomous_task(agent_id: str, task_id: str) -> None:
         await session.commit()
 
     logger.info(f"Deleted autonomous task {task_id} from agent {agent_id}")
+
+
+async def update_autonomous_task(
+    agent_id: str, task_id: str, task_updates: dict
+) -> AgentAutonomous:
+    """
+    Update an autonomous task for an agent.
+
+    Args:
+        agent_id: ID of the agent
+        task_id: ID of the task to update
+        task_updates: Dictionary containing fields to update
+
+    Returns:
+        AgentAutonomous: The updated task
+
+    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: List[AgentAutonomous] = agent.autonomous or []
+
+    # Find and update the task
+    task_found = False
+    updated_tasks: List[AgentAutonomous] = []
+    updated_task = None
+
+    for task_data in current_tasks:
+        if task_data.id == task_id:
+            task_found = True
+            # Create a dictionary with current task data
+            task_dict = task_data.model_dump()
+            # Update with provided fields
+            task_dict.update(task_updates)
+            # Create new AgentAutonomous instance
+            updated_task = AgentAutonomous.model_validate(task_dict)
+            updated_tasks.append(updated_task)
+        else:
+            updated_tasks.append(task_data)
+
+    if not task_found:
+        raise IntentKitAPIError(
+            404, "TaskNotFound", f"Autonomous task with ID {task_id} not found."
+        )
+
+    # Convert all AgentAutonomous objects to dictionaries for JSON serialization
+    serializable_tasks = [task_item.model_dump() for task_item in updated_tasks]
+
+    # Update the agent in the database
+    async with get_session() as session:
+        update_stmt = (
+            update(AgentTable)
+            .where(AgentTable.id == agent_id)
+            .values(autonomous=serializable_tasks)
+        )
+        await session.execute(update_stmt)
+        await session.commit()
+
+    logger.info(f"Updated autonomous task {task_id} for agent {agent_id}")
+    return updated_task

{intentkit-0.6.7.dev6 → intentkit-0.6.7.dev7}/core/skill.py

@@ -11,6 +11,9 @@ from intentkit.core.agent import (
 from intentkit.core.agent import (
     list_autonomous_tasks as _list_autonomous_tasks,
 )
+from intentkit.core.agent import (
+    update_autonomous_task as _update_autonomous_task,
+)
 from intentkit.models.agent import Agent, AgentAutonomous
 from intentkit.models.agent_data import AgentData, AgentQuota
 from intentkit.models.skill import (
@@ -177,5 +180,21 @@ class SkillStore(SkillStoreABC):
         """
         await _delete_autonomous_task(agent_id, task_id)
 
+    @staticmethod
+    async def update_autonomous_task(
+        agent_id: str, task_id: str, task_updates: dict
+    ) -> AgentAutonomous:
+        """Update an autonomous task for an agent.
+
+        Args:
+            agent_id: ID of the agent
+            task_id: ID of the task to update
+            task_updates: Dictionary containing fields to update
+
+        Returns:
+            AgentAutonomous: The updated task
+        """
+        return await _update_autonomous_task(agent_id, task_id, task_updates)
+
 
 skill_store = SkillStore()

{intentkit-0.6.7.dev6 → intentkit-0.6.7.dev7}/pyproject.toml

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

{intentkit-0.6.7.dev6 → intentkit-0.6.7.dev7}/skills/system/__init__.py

@@ -8,6 +8,7 @@ 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.edit_autonomous_task import EditAutonomousTask
 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
@@ -24,6 +25,7 @@ class SkillStates(TypedDict):
     list_autonomous_tasks: SkillOwnerState
     add_autonomous_task: SkillOwnerState
     delete_autonomous_task: SkillOwnerState
+    edit_autonomous_task: SkillOwnerState
 
 
 class Config(SkillConfig):
@@ -109,6 +111,12 @@ def get_system_skill(
                 skill_store=store,
             )
         return _cache[name]
+    elif name == "edit_autonomous_task":
+        if name not in _cache:
+            _cache[name] = EditAutonomousTask(
+                skill_store=store,
+            )
+        return _cache[name]
     else:
         logger.warning(f"Unknown system skill: {name}")
         return None

{intentkit-0.6.7.dev6 → intentkit-0.6.7.dev7}/skills/system/add_autonomous_task.py

@@ -11,10 +11,14 @@ class AddAutonomousTaskInput(BaseModel):
     """Input model for add_autonomous_task skill."""
 
     name: Optional[str] = Field(
-        default=None, description="Display name of the autonomous task configuration"
+        default=None,
+        description="Display name of the autonomous task configuration",
+        max_length=50,
     )
     description: Optional[str] = Field(
-        default=None, description="Description of the autonomous task configuration"
+        default=None,
+        description="Description of the autonomous task configuration",
+        max_length=200,
     )
     minutes: Optional[int] = Field(
         default=None,

intentkit-0.6.7.dev7/skills/system/edit_autonomous_task.py

@@ -0,0 +1,116 @@
+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 EditAutonomousTaskInput(BaseModel):
+    """Input model for edit_autonomous_task skill."""
+
+    task_id: str = Field(
+        description="The unique identifier of the autonomous task to edit"
+    )
+    name: Optional[str] = Field(
+        default=None,
+        description="Display name of the autonomous task configuration",
+        max_length=50,
+    )
+    description: Optional[str] = Field(
+        default=None,
+        description="Description of the autonomous task configuration",
+        max_length=200,
+    )
+    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: Optional[str] = Field(
+        default=None, description="Special prompt used during autonomous operation"
+    )
+    enabled: Optional[bool] = Field(
+        default=None, description="Whether the autonomous task is enabled"
+    )
+
+
+class EditAutonomousTaskOutput(BaseModel):
+    """Output model for edit_autonomous_task skill."""
+
+    task: AgentAutonomous = Field(
+        description="The updated autonomous task configuration"
+    )
+
+
+class EditAutonomousTask(SystemBaseTool):
+    """Skill to edit an existing autonomous task for an agent."""
+
+    name: str = "system_edit_autonomous_task"
+    description: str = (
+        "Edit an existing autonomous task configuration for the agent. "
+        "Allows updating the name, description, schedule (minutes or cron), prompt, and enabled status. "
+        "Only provided fields will be updated; omitted fields will keep their current values. "
+        "The minutes and cron fields are mutually exclusive. Do not provide both of them. "
+    )
+    args_schema = EditAutonomousTaskInput
+
+    async def _arun(
+        self,
+        task_id: str,
+        name: Optional[str] = None,
+        description: Optional[str] = None,
+        minutes: Optional[int] = None,
+        cron: Optional[str] = None,
+        prompt: Optional[str] = None,
+        enabled: Optional[bool] = None,
+        config: RunnableConfig = None,
+        **kwargs,
+    ) -> EditAutonomousTaskOutput:
+        """Edit an autonomous task for the agent.
+
+        Args:
+            task_id: ID of the task to edit
+            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:
+            EditAutonomousTaskOutput: The updated task
+        """
+        context = self.context_from_config(config)
+        agent_id = context.agent_id
+
+        if minutes is not None and cron is not None:
+            raise ValueError("minutes and cron are mutually exclusive")
+
+        # Build the updates dictionary with only provided fields
+        task_updates = {}
+        if name is not None:
+            task_updates["name"] = name
+        if description is not None:
+            task_updates["description"] = description
+        if minutes is not None:
+            task_updates["minutes"] = minutes
+            task_updates["cron"] = None
+        if cron is not None:
+            task_updates["cron"] = cron
+            task_updates["minutes"] = None
+        if prompt is not None:
+            task_updates["prompt"] = prompt
+        if enabled is not None:
+            task_updates["enabled"] = enabled
+
+        updated_task = await self.skill_store.update_autonomous_task(
+            agent_id, task_id, task_updates
+        )
+
+        return EditAutonomousTaskOutput(task=updated_task)

{intentkit-0.6.7.dev6 → intentkit-0.6.7.dev7}/skills/system/schema.json

@@ -88,6 +88,20 @@
           ],
           "description": "Delete an autonomous task configuration from the agent.",
           "default": "disabled"
+        },
+        "edit_autonomous_task": {
+          "type": "string",
+          "title": "Edit Autonomous Task",
+          "enum": [
+            "disabled",
+            "private"
+          ],
+          "x-enum-title": [
+            "Disabled",
+            "Agent Owner Only"
+          ],
+          "description": "Edit an existing autonomous task configuration for the agent.",
+          "default": "disabled"
         }
       }
     }