intentkit 0.8.16.dev1__py3-none-any.whl → 0.8.17.dev2__py3-none-any.whl

intentkit/core/credit.py

@@ -1,7 +1,6 @@
 import logging
 from datetime import datetime
 from decimal import ROUND_HALF_UP, Decimal
-from typing import List, Optional, Tuple
 
 from epyxid import XID
 from pydantic import BaseModel
@@ -50,7 +49,7 @@ FOURPLACES = Decimal("0.0001")
 async def update_credit_event_note(
     session: AsyncSession,
     event_id: str,
-    note: Optional[str] = None,
+    note: str | None = None,
 ) -> CreditEvent:
     """
     Update the note of a credit event.
@@ -89,7 +88,7 @@ async def recharge(
     user_id: str,
     amount: Decimal,
     upstream_tx_id: str,
-    note: Optional[str] = None,
+    note: str | None = None,
 ) -> CreditAccount:
     """
     Recharge credits to a user account.
@@ -220,7 +219,7 @@ async def withdraw(
     agent_id: str,
     amount: Decimal,
     upstream_tx_id: str,
-    note: Optional[str] = None,
+    note: str | None = None,
 ) -> CreditAccount:
     """
     Withdraw credits from an agent account to platform account.
@@ -385,8 +384,8 @@ async def reward(
     user_id: str,
     amount: Decimal,
     upstream_tx_id: str,
-    note: Optional[str] = None,
-    reward_type: Optional[RewardType] = RewardType.REWARD,
+    note: str | None = None,
+    reward_type: RewardType | None = RewardType.REWARD,
 ) -> CreditAccount:
     """
     Reward a user account with reward credits.
@@ -679,8 +678,8 @@ async def adjustment(
 async def update_daily_quota(
     session: AsyncSession,
     user_id: str,
-    free_quota: Optional[Decimal] = None,
-    refill_amount: Optional[Decimal] = None,
+    free_quota: Decimal | None = None,
+    refill_amount: Decimal | None = None,
     upstream_tx_id: str = "",
     note: str = "",
 ) -> CreditAccount:
@@ -706,11 +705,11 @@ async def update_daily_quota(
 async def list_credit_events_by_user(
     session: AsyncSession,
     user_id: str,
-    direction: Optional[Direction] = None,
-    cursor: Optional[str] = None,
+    direction: Direction | None = None,
+    cursor: str | None = None,
     limit: int = 20,
-    event_type: Optional[EventType] = None,
-) -> Tuple[List[CreditEvent], Optional[str], bool]:
+    event_type: EventType | None = None,
+) -> tuple[list[CreditEvent], str | None, bool]:
     """
     List credit events for a user account with cursor pagination.
 
@@ -771,13 +770,13 @@ async def list_credit_events_by_user(
 
 async def list_credit_events(
     session: AsyncSession,
-    direction: Optional[Direction] = Direction.EXPENSE,
-    cursor: Optional[str] = None,
+    direction: Direction | None = Direction.EXPENSE,
+    cursor: str | None = None,
     limit: int = 20,
-    event_type: Optional[EventType] = None,
-    start_at: Optional[datetime] = None,
-    end_at: Optional[datetime] = None,
-) -> Tuple[List[CreditEvent], Optional[str], bool]:
+    event_type: EventType | None = None,
+    start_at: datetime | None = None,
+    end_at: datetime | None = None,
+) -> tuple[list[CreditEvent], str | None, bool]:
     """
     List all credit events with cursor pagination.
 
@@ -841,9 +840,9 @@ async def list_credit_events(
 async def list_fee_events_by_agent(
     session: AsyncSession,
     agent_id: str,
-    cursor: Optional[str] = None,
+    cursor: str | None = None,
     limit: int = 20,
-) -> Tuple[List[CreditEvent], Optional[str], bool]:
+) -> tuple[list[CreditEvent], str | None, bool]:
     """
     List fee events for an agent with cursor pagination.
     These events represent income for the agent from users' expenses.

intentkit/core/engine.py

@@ -17,7 +17,6 @@ import textwrap
 import time
 import traceback
 from datetime import datetime
-from typing import Optional, Tuple
 
 import sqlalchemy
 from epyxid import XID
@@ -61,7 +60,6 @@ from intentkit.utils.error import IntentKitAPIError
 
 logger = logging.getLogger(__name__)
 
-
 # Global variable to cache all agent executors
 _agents: dict[str, CompiledStateGraph] = {}
 
@@ -229,7 +227,7 @@ async def initialize_agent(aid):
         HTTPException: If agent not found (404) or database error (500)
     """
     # get the agent from the database
-    agent: Optional[Agent] = await Agent.get(aid)
+    agent: Agent | None = await Agent.get(aid)
     if not agent:
         raise IntentKitAPIError(
             status_code=404, key="AgentNotFound", message="Agent not found"
@@ -243,7 +241,7 @@ async def initialize_agent(aid):
     _agents_updated[aid] = agent.deployed_at if agent.deployed_at else agent.updated_at
 
 
-async def agent_executor(agent_id: str) -> Tuple[CompiledStateGraph, float]:
+async def agent_executor(agent_id: str) -> tuple[CompiledStateGraph, float]:
     start = time.perf_counter()
     agent = await Agent.get(agent_id)
     if not agent:

intentkit/core/node.py

@@ -1,5 +1,6 @@
 import logging
-from typing import Any, Sequence
+from collections.abc import Sequence
+from typing import Any
 
 from langchain_core.language_models import LanguageModelLike
 from langchain_core.messages import (

intentkit/core/prompt.py

@@ -1,5 +1,5 @@
 import re
-from typing import Callable, Optional
+from collections.abc import Callable
 
 from eth_utils import is_address
 from langchain_core.messages import BaseMessage
@@ -35,7 +35,6 @@ user confirmation before broadcasting any approval transactions for security rea
 
 """
 
-
 # ============================================================================
 # CORE PROMPT BUILDING FUNCTIONS
 # ============================================================================
@@ -328,7 +327,7 @@ def _build_autonomous_task_prompt(agent: Agent, context: AgentContext) -> str:
     return f"{task_info}. "
 
 
-async def build_entrypoint_prompt(agent: Agent, context: AgentContext) -> Optional[str]:
+async def build_entrypoint_prompt(agent: Agent, context: AgentContext) -> str | None:
     """
     Build entrypoint-specific prompt based on context.
 
@@ -341,7 +340,7 @@ async def build_entrypoint_prompt(agent: Agent, context: AgentContext) -> Option
         context: The agent context containing entrypoint information
 
     Returns:
-        Optional[str]: The entrypoint-specific prompt, or None if no entrypoint
+        str | None: The entrypoint-specific prompt, or None if no entrypoint
     """
     if not context.entrypoint:
         return None

intentkit/core/scheduler.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import Mapping, MutableMapping
+from collections.abc import Mapping, MutableMapping
 
 from apscheduler.jobstores.base import BaseJobStore
 from apscheduler.schedulers.asyncio import AsyncIOScheduler

intentkit/core/statistics.py

@@ -2,9 +2,8 @@
 
 from __future__ import annotations
 
-from datetime import datetime, timedelta, timezone
+from datetime import UTC, datetime, timedelta
 from decimal import Decimal
-from typing import Optional
 
 from pydantic import BaseModel, Field
 from sqlalchemy import func, select
@@ -46,8 +45,8 @@ class AgentStatistics(BaseModel):
 async def get_agent_statistics(
     agent_id: str,
     *,
-    end_time: Optional[datetime] = None,
-    session: Optional[AsyncSession] = None,
+    end_time: datetime | None = None,
+    session: AsyncSession | None = None,
 ) -> AgentStatistics:
     """Calculate statistics for an agent credit account.
 
@@ -64,11 +63,11 @@ async def get_agent_statistics(
 
     managed_session = session is None
     if end_time is None:
-        end_time = datetime.now(timezone.utc)
+        end_time = datetime.now(UTC)
     elif end_time.tzinfo is None:
-        end_time = end_time.replace(tzinfo=timezone.utc)
+        end_time = end_time.replace(tzinfo=UTC)
     else:
-        end_time = end_time.astimezone(timezone.utc)
+        end_time = end_time.astimezone(UTC)
 
     async def _compute(session: AsyncSession) -> AgentStatistics:
         account = await CreditAccount.get_or_create_in_session(