intentkit 0.6.6__tar.gz → 0.6.7.dev2__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: intentkit
-Version: 0.6.6
+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.6 → 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.6"
+__version__ = "0.6.7-dev2"
 __author__ = "hyacinthus"
 __email__ = "hyacinthus@gmail.com"
 

{intentkit-0.6.6 → 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.6 → intentkit-0.6.7.dev2}/core/agent.py

@@ -2,11 +2,12 @@ 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
+from sqlalchemy import func, select, text, update
 
-from intentkit.models.agent import Agent
+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
 from intentkit.utils.error import IntentKitAPIError
@@ -189,3 +190,203 @@ async def agent_action_cost(agent_id: str) -> Dict[str, Decimal]:
         )
 
         return result
+
+
+async def update_agent_action_cost():
+    """
+    Update action costs for all agents.
+
+    This function processes agents in batches of 100 to avoid memory issues.
+    For each agent, it calculates various action cost metrics:
+    - avg_action_cost: average cost per action
+    - min_action_cost: minimum cost per action
+    - max_action_cost: maximum cost per action
+    - low_action_cost: average cost of the lowest 20% of actions
+    - medium_action_cost: average cost of the middle 60% of actions
+    - high_action_cost: average cost of the highest 20% of actions
+
+    It then updates the corresponding record in the agent_quotas table.
+    """
+    logger.info("Starting update of agent average action costs")
+    start_time = time.time()
+    batch_size = 100
+    last_id = None
+    total_updated = 0
+
+    while True:
+        # Get a batch of agent IDs ordered by ID
+        async with get_session() as session:
+            query = select(AgentTable.id).order_by(AgentTable.id)
+
+            # Apply pagination if we have a last_id from previous batch
+            if last_id:
+                query = query.where(AgentTable.id > last_id)
+
+            query = query.limit(batch_size)
+            result = await session.execute(query)
+            agent_ids = [row[0] for row in result]
+
+            # If no more agents, we're done
+            if not agent_ids:
+                break
+
+            # Update last_id for next batch
+            last_id = agent_ids[-1]
+
+        # Process this batch of agents
+        logger.info(
+            f"Processing batch of {len(agent_ids)} agents starting with ID {agent_ids[0]}"
+        )
+        batch_start_time = time.time()
+
+        for agent_id in agent_ids:
+            try:
+                # Calculate action costs for this agent
+                costs = await agent_action_cost(agent_id)
+
+                # Update the agent's quota record
+                async with get_session() as session:
+                    update_stmt = (
+                        update(AgentQuotaTable)
+                        .where(AgentQuotaTable.id == agent_id)
+                        .values(
+                            avg_action_cost=costs["avg_action_cost"],
+                            min_action_cost=costs["min_action_cost"],
+                            max_action_cost=costs["max_action_cost"],
+                            low_action_cost=costs["low_action_cost"],
+                            medium_action_cost=costs["medium_action_cost"],
+                            high_action_cost=costs["high_action_cost"],
+                        )
+                    )
+                    await session.execute(update_stmt)
+                    await session.commit()
+
+                total_updated += 1
+            except Exception as e:
+                logger.error(
+                    f"Error updating action costs for agent {agent_id}: {str(e)}"
+                )
+
+        batch_time = time.time() - batch_start_time
+        logger.info(f"Completed batch in {batch_time:.3f}s")
+
+    total_time = time.time() - start_time
+    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.6 → 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.6 → 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.6 → intentkit-0.6.7.dev2}/models/agent_data.py

@@ -814,3 +814,33 @@ class AgentQuota(BaseModel):
                 # Update this instance
                 await db.refresh(quota_record)
                 self.model_validate(quota_record)
+
+    @staticmethod
+    async def reset_daily_quotas():
+        """Reset daily quotas for all agents at UTC 00:00.
+        Resets message_count_daily and twitter_count_daily to 0.
+        """
+        from sqlalchemy import update
+
+        async with get_session() as session:
+            stmt = update(AgentQuotaTable).values(
+                message_count_daily=0,
+                twitter_count_daily=0,
+                free_income_daily=0,
+            )
+            await session.execute(stmt)
+            await session.commit()
+
+    @staticmethod
+    async def reset_monthly_quotas():
+        """Reset monthly quotas for all agents at the start of each month.
+        Resets message_count_monthly and autonomous_count_monthly to 0.
+        """
+        from sqlalchemy import update
+
+        async with get_session() as session:
+            stmt = update(AgentQuotaTable).values(
+                message_count_monthly=0, autonomous_count_monthly=0
+            )
+            await session.execute(stmt)
+            await session.commit()

{intentkit-0.6.6 → intentkit-0.6.7.dev2}/models/app_setting.py

@@ -46,6 +46,8 @@ class PaymentSettings(BaseModel):
                 "credit_per_usdc": 1000,
                 "fee_platform_percentage": 100,
                 "fee_dev_percentage": 20,
+                "free_quota": 480,
+                "refill_amount": 20,
                 "agent_whitelist_enabled": False,
                 "agent_whitelist": [],
             }
@@ -68,6 +70,22 @@ class PaymentSettings(BaseModel):
             default=Decimal("20"), description="Developer fee percentage", ge=0, le=100
         ),
     ]
+    free_quota: Annotated[
+        Decimal,
+        Field(
+            default=Decimal("480"),
+            description="Daily free credit quota for new users",
+            ge=0,
+        ),
+    ]
+    refill_amount: Annotated[
+        Decimal,
+        Field(
+            default=Decimal("20"),
+            description="Hourly refill amount for free credits",
+            ge=0,
+        ),
+    ]
     agent_whitelist_enabled: Annotated[
         bool,
         Field(default=False, description="Whether agent whitelist is enabled"),
@@ -77,7 +95,13 @@ class PaymentSettings(BaseModel):
         Field(default_factory=list, description="List of whitelisted agent IDs"),
     ]
 
-    @field_validator("credit_per_usdc", "fee_platform_percentage", "fee_dev_percentage")
+    @field_validator(
+        "credit_per_usdc",
+        "fee_platform_percentage",
+        "fee_dev_percentage",
+        "free_quota",
+        "refill_amount",
+    )
     @classmethod
     def round_decimal(cls, v: Any) -> Decimal:
         """Round decimal values to 4 decimal places."""

{intentkit-0.6.6 → intentkit-0.6.7.dev2}/models/credit.py

@@ -6,6 +6,7 @@ from typing import Annotated, Any, Dict, List, Optional, Tuple
 
 from epyxid import XID
 from fastapi import HTTPException
+from intentkit.models.app_setting import AppSetting
 from intentkit.models.base import Base
 from intentkit.models.db import get_session
 from pydantic import BaseModel, ConfigDict, Field, field_validator
@@ -437,8 +438,8 @@ class CreditAccount(BaseModel):
         session: AsyncSession,
         owner_type: OwnerType,
         owner_id: str,
-        free_quota: Decimal = Decimal("480.0"),
-        refill_amount: Decimal = Decimal("20.0"),
+        free_quota: Optional[Decimal] = None,
+        refill_amount: Optional[Decimal] = None,
     ) -> "CreditAccount":
         """Get an existing credit account or create a new one if it doesn't exist.
 
@@ -448,15 +449,24 @@ class CreditAccount(BaseModel):
             session: Async session to use for database queries
             owner_type: Type of the owner
             owner_id: ID of the owner
-            free_quota: Daily quota for a new account if created
+            free_quota: Daily quota for a new account if created (if None, reads from payment settings)
+            refill_amount: Hourly refill amount (if None, reads from payment settings)
 
         Returns:
             CreditAccount: The existing or newly created credit account
         """
+        # Get payment settings if values not provided
+        if free_quota is None or refill_amount is None:
+            payment_settings = await AppSetting.payment()
+            if free_quota is None:
+                free_quota = payment_settings.free_quota
+            if refill_amount is None:
+                refill_amount = payment_settings.refill_amount
+
         if owner_type != OwnerType.USER:
             # only users have daily quota
-            free_quota = 0.0
-            refill_amount = 0.0
+            free_quota = Decimal("0.0")
+            refill_amount = Decimal("0.0")
         # Create event_id at the beginning for consistency
         event_id = str(XID())
 

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

@@ -1,6 +1,6 @@
 [project]
 name = "intentkit"
-version = "0.6.6"
+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.6 → 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)