intentkit 0.6.6__tar.gz → 0.6.7.dev1__tar.gz

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

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

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

@@ -4,9 +4,10 @@ from datetime import datetime, timedelta, timezone
 from decimal import Decimal
 from typing import Dict
 
-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, 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,87 @@ 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"
+    )

{intentkit-0.6.6 → intentkit-0.6.7.dev1}/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.dev1}/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.dev1}/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.dev1}/pyproject.toml

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