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

intentkit/models/agent_data.py

@@ -1,7 +1,9 @@
+from __future__ import annotations
+
 import logging
-from datetime import datetime, timezone
+from datetime import UTC, datetime
 from decimal import Decimal
-from typing import Annotated, Any, Dict, Optional
+from typing import Annotated, Any
 
 from intentkit.models.base import Base
 from intentkit.models.db import get_session
@@ -60,6 +62,9 @@ class AgentDataTable(Base):
     telegram_id = Column(String, nullable=True, comment="Telegram user ID")
     telegram_username = Column(String, nullable=True, comment="Telegram username")
     telegram_name = Column(String, nullable=True, comment="Telegram display name")
+    discord_id = Column(String, nullable=True, comment="Discord user ID")
+    discord_username = Column(String, nullable=True, comment="Discord username")
+    discord_name = Column(String, nullable=True, comment="Discord display name")
     error_message = Column(String, nullable=True, comment="Last error message")
     api_key = Column(
         String, nullable=True, unique=True, comment="API key for the agent"
@@ -77,7 +82,7 @@ class AgentDataTable(Base):
         DateTime(timezone=True),
         nullable=False,
         server_default=func.now(),
-        onupdate=lambda: datetime.now(timezone.utc),
+        onupdate=lambda: datetime.now(UTC),
         comment="Timestamp when the agent data was last updated",
     )
 
@@ -94,70 +99,70 @@ class AgentData(BaseModel):
         ),
     ]
     evm_wallet_address: Annotated[
-        Optional[str],
+        str | None,
         PydanticField(
             default=None,
             description="EVM wallet address",
         ),
     ] = None
     solana_wallet_address: Annotated[
-        Optional[str],
+        str | None,
         PydanticField(
             default=None,
             description="Solana wallet address",
         ),
     ] = None
     cdp_wallet_data: Annotated[
-        Optional[str],
+        str | None,
         PydanticField(
             default=None,
             description="CDP wallet data",
         ),
     ] = None
     twitter_id: Annotated[
-        Optional[str],
+        str | None,
         PydanticField(
             default=None,
             description="Twitter user ID",
         ),
     ] = None
     twitter_username: Annotated[
-        Optional[str],
+        str | None,
         PydanticField(
             default=None,
             description="Twitter username",
         ),
     ] = None
     twitter_name: Annotated[
-        Optional[str],
+        str | None,
         PydanticField(
             default=None,
             description="Twitter display name",
         ),
     ] = None
     twitter_access_token: Annotated[
-        Optional[str],
+        str | None,
         PydanticField(
             default=None,
             description="Twitter access token",
         ),
     ] = None
     twitter_access_token_expires_at: Annotated[
-        Optional[datetime],
+        datetime | None,
         PydanticField(
             default=None,
             description="Twitter access token expiration time",
         ),
     ] = None
     twitter_refresh_token: Annotated[
-        Optional[str],
+        str | None,
         PydanticField(
             default=None,
             description="Twitter refresh token",
         ),
     ] = None
     twitter_self_key_refreshed_at: Annotated[
-        Optional[datetime],
+        datetime | None,
         PydanticField(
             default=None,
             description="Twitter self-key userinfo last refresh time",
@@ -171,42 +176,63 @@ class AgentData(BaseModel):
         ),
     ] = None
     telegram_id: Annotated[
-        Optional[str],
+        str | None,
         PydanticField(
             default=None,
             description="Telegram user ID",
         ),
     ] = None
     telegram_username: Annotated[
-        Optional[str],
+        str | None,
         PydanticField(
             default=None,
             description="Telegram username",
         ),
     ] = None
     telegram_name: Annotated[
-        Optional[str],
+        str | None,
         PydanticField(
             default=None,
             description="Telegram display name",
         ),
     ] = None
+    discord_id: Annotated[
+        str | None,
+        PydanticField(
+            default=None,
+            description="Discord user ID",
+        ),
+    ] = None
+    discord_username: Annotated[
+        str | None,
+        PydanticField(
+            default=None,
+            description="Discord username",
+        ),
+    ] = None
+    discord_name: Annotated[
+        str | None,
+        PydanticField(
+            default=None,
+            description="Discord display name",
+        ),
+    ] = None
     error_message: Annotated[
-        Optional[str],
+        str | None,
         PydanticField(
             default=None,
             description="Last error message",
         ),
     ] = None
     api_key: Annotated[
-        Optional[str],
+        str | None,
         PydanticField(
             default=None,
             description="API key for the agent",
         ),
     ] = None
     api_key_public: Annotated[
-        Optional[str],
+        str | None,
         PydanticField(
             default=None,
             description="Public API key for the agent",
@@ -215,14 +241,14 @@ class AgentData(BaseModel):
     created_at: Annotated[
         datetime,
         PydanticField(
-            default_factory=lambda: datetime.now(timezone.utc),
+            default_factory=lambda: datetime.now(UTC),
             description="Timestamp when the agent data was created",
         ),
     ]
     updated_at: Annotated[
         datetime,
         PydanticField(
-            default_factory=lambda: datetime.now(timezone.utc),
+            default_factory=lambda: datetime.now(UTC),
             description="Timestamp when the agent data was last updated",
         ),
     ]
@@ -247,7 +273,7 @@ class AgentData(BaseModel):
             return cls.model_construct(id=agent_id)
 
     @classmethod
-    async def get_by_api_key(cls, api_key: str) -> Optional["AgentData"]:
+    async def get_by_api_key(cls, api_key: str) -> AgentData | None:
         """Get agent data by API key.
 
         Args:
@@ -356,7 +382,7 @@ class AgentPluginDataTable(Base):
         DateTime(timezone=True),
         nullable=False,
         server_default=func.now(),
-        onupdate=lambda: datetime.now(timezone.utc),
+        onupdate=lambda: datetime.now(UTC),
     )
 
 
@@ -388,28 +414,28 @@ class AgentPluginData(BaseModel):
         PydanticField(description="Key for this specific piece of data"),
     ]
     data: Annotated[
-        Dict[str, Any],
+        dict[str, Any],
         PydanticField(default=None, description="JSON data stored for this key"),
     ]
     created_at: Annotated[
         datetime,
         PydanticField(
             description="Timestamp when this data was created",
-            default_factory=lambda: datetime.now(timezone.utc),
+            default_factory=lambda: datetime.now(UTC),
         ),
     ]
     updated_at: Annotated[
         datetime,
         PydanticField(
             description="Timestamp when this data was last updated",
-            default_factory=lambda: datetime.now(timezone.utc),
+            default_factory=lambda: datetime.now(UTC),
         ),
     ]
 
     @classmethod
     async def get(
         cls, agent_id: str, plugin: str, key: str
-    ) -> Optional["AgentPluginData"]:
+    ) -> "AgentPluginData" | None:
         """Get plugin data for an agent.
 
         Args:
@@ -513,7 +539,7 @@ class AgentQuotaTable(Base):
         DateTime(timezone=True),
         nullable=False,
         server_default=func.now(),
-        onupdate=lambda: datetime.now(timezone.utc),
+        onupdate=lambda: datetime.now(UTC),
     )
 
 
@@ -547,7 +573,7 @@ class AgentQuota(BaseModel):
         int, PydanticField(default=99999999, description="Daily message limit")
     ]
     last_message_time: Annotated[
-        Optional[datetime],
+        datetime | None,
         PydanticField(default=None, description="Last message timestamp"),
     ]
     autonomous_count_total: Annotated[
@@ -578,7 +604,7 @@ class AgentQuota(BaseModel):
         ),
     ]
     last_autonomous_time: Annotated[
-        Optional[datetime],
+        datetime | None,
         PydanticField(default=None, description="Last autonomous operation timestamp"),
     ]
     twitter_count_total: Annotated[
@@ -603,7 +629,7 @@ class AgentQuota(BaseModel):
         PydanticField(default=99999999, description="Daily Twitter operations limit"),
     ]
     last_twitter_time: Annotated[
-        Optional[datetime],
+        datetime | None,
         PydanticField(default=None, description="Last Twitter operation timestamp"),
     ]
     free_income_daily: Annotated[
@@ -638,14 +664,14 @@ class AgentQuota(BaseModel):
         datetime,
         PydanticField(
             description="Timestamp when this quota was created",
-            default_factory=lambda: datetime.now(timezone.utc),
+            default_factory=lambda: datetime.now(UTC),
         ),
     ]
     updated_at: Annotated[
         datetime,
         PydanticField(
             description="Timestamp when this quota was last updated",
-            default_factory=lambda: datetime.now(timezone.utc),
+            default_factory=lambda: datetime.now(UTC),
         ),
     ]
 
@@ -770,7 +796,7 @@ class AgentQuota(BaseModel):
                 quota_record.message_count_total += 1
                 quota_record.message_count_monthly += 1
                 quota_record.message_count_daily += 1
-                quota_record.last_message_time = datetime.now(timezone.utc)
+                quota_record.last_message_time = datetime.now(UTC)
                 db.add(quota_record)
                 await db.commit()
 
@@ -790,7 +816,7 @@ class AgentQuota(BaseModel):
                 # Update record
                 quota_record.autonomous_count_total += 1
                 quota_record.autonomous_count_monthly += 1
-                quota_record.last_autonomous_time = datetime.now(timezone.utc)
+                quota_record.last_autonomous_time = datetime.now(UTC)
                 db.add(quota_record)
                 await db.commit()
 
@@ -811,7 +837,7 @@ class AgentQuota(BaseModel):
                 # Update record
                 quota_record.twitter_count_total += 1
                 quota_record.twitter_count_daily += 1
-                quota_record.last_twitter_time = datetime.now(timezone.utc)
+                quota_record.last_twitter_time = datetime.now(UTC)
                 db.add(quota_record)
                 await db.commit()
 

intentkit/models/app_setting.py

@@ -1,8 +1,8 @@
 import time
-from datetime import datetime, timezone
+from datetime import UTC, datetime
 from decimal import ROUND_HALF_UP, Decimal
 from enum import Enum
-from typing import Annotated, Any, Dict, List
+from typing import Annotated, Any
 
 from intentkit.models.base import Base
 from intentkit.models.db import get_session
@@ -33,7 +33,7 @@ DEFAULT_SYSTEM_MESSAGES = {
 }
 
 # In-memory cache for app settings
-_cache: Dict[str, Dict[str, Any]] = {}
+_cache: dict[str, dict[str, Any]] = {}
 _cache_ttl = 180  # 3 minutes in seconds
 
 
@@ -59,7 +59,7 @@ class AppSettingTable(Base):
         DateTime(timezone=True),
         nullable=False,
         server_default=func.now(),
-        onupdate=lambda: datetime.now(timezone.utc),
+        onupdate=lambda: datetime.now(UTC),
     )
 
 
@@ -117,7 +117,7 @@ class PaymentSettings(BaseModel):
         Field(default=False, description="Whether agent whitelist is enabled"),
     ]
     agent_whitelist: Annotated[
-        List[str],
+        list[str],
         Field(default_factory=list, description="List of whitelisted agent IDs"),
     ]
 
@@ -133,7 +133,7 @@ class PaymentSettings(BaseModel):
         """Round decimal values to 4 decimal places."""
         if isinstance(v, Decimal):
             return v.quantize(Decimal("0.0001"), rounding=ROUND_HALF_UP)
-        elif isinstance(v, (int, float)):
+        elif isinstance(v, int | float):
             return Decimal(str(v)).quantize(Decimal("0.0001"), rounding=ROUND_HALF_UP)
         return v
 

intentkit/models/chat.py

@@ -1,7 +1,9 @@
-from datetime import datetime, timezone
+from __future__ import annotations
+
+from datetime import UTC, datetime
 from decimal import Decimal
 from enum import Enum
-from typing import Annotated, List, NotRequired, Optional, TypedDict
+from typing import Annotated, NotRequired, TypedDict
 
 from epyxid import XID
 from intentkit.models.app_setting import AppSetting, SystemMessageType
@@ -43,6 +45,7 @@ class AuthorType(str, Enum):
     SKILL = "skill"
     TELEGRAM = "telegram"
     TWITTER = "twitter"
+    DISCORD = "discord"
     WEB = "web"
     SYSTEM = "system"
     API = "api"
@@ -65,7 +68,7 @@ class ChatMessageAttachment(TypedDict):
         ),
     ]
     url: Annotated[
-        Optional[str],
+        str | None,
         Field(
             ...,
             description="URL of the attachment",
@@ -73,7 +76,7 @@ class ChatMessageAttachment(TypedDict):
         ),
     ]
     json: Annotated[
-        Optional[dict],
+        dict | None,
         Field(
             None,
             description="JSON data of the attachment",
@@ -114,7 +117,7 @@ class ChatMessageRequest(BaseModel):
         ),
     ]
     app_id: Annotated[
-        Optional[str],
+        str | None,
         Field(
             None,
             description="Optional application identifier",
@@ -141,21 +144,21 @@ class ChatMessageRequest(BaseModel):
         ),
     ]
     search_mode: Annotated[
-        Optional[bool],
+        bool | None,
         Field(
             None,
             description="Optional flag to enable search mode",
         ),
     ]
     super_mode: Annotated[
-        Optional[bool],
+        bool | None,
         Field(
             None,
             description="Optional flag to enable super mode",
         ),
     ]
     attachments: Annotated[
-        Optional[List[ChatMessageAttachment]],
+        list[ChatMessageAttachment] | None,
         Field(
             None,
             description="Optional list of attachments (links, images, or files)",
@@ -309,29 +312,29 @@ class ChatMessageCreate(BaseModel):
     ]
     chat_id: Annotated[str, Field(description="ID of the chat this message belongs to")]
     user_id: Annotated[
-        Optional[str],
+        str | None,
         Field(description="ID of the user this message belongs to or reply to"),
     ]
     author_id: Annotated[str, Field(description="ID of the message author")]
     author_type: Annotated[AuthorType, Field(description="Type of the message author")]
     model: Annotated[
-        Optional[str], Field(None, description="LLM model used if applicable")
+        str | None, Field(None, description="LLM model used if applicable")
     ]
     thread_type: Annotated[
-        Optional[AuthorType],
+        AuthorType | None,
         Field(None, description="Author Type of the message thread start"),
     ]
     reply_to: Annotated[
-        Optional[str],
+        str | None,
         Field(None, description="ID of the message this message is a reply to"),
     ]
     message: Annotated[str, Field(description="Content of the message")]
     attachments: Annotated[
-        Optional[List[ChatMessageAttachment]],
+        list[ChatMessageAttachment] | None,
         Field(None, description="List of attachments in the message"),
     ]
     skill_calls: Annotated[
-        Optional[List[ChatMessageSkillCall]],
+        list[ChatMessageSkillCall] | None,
         Field(None, description="Skill call details"),
     ]
     input_tokens: Annotated[
@@ -344,11 +347,11 @@ class ChatMessageCreate(BaseModel):
         float, Field(0.0, description="Time cost for the message in seconds")
     ]
     credit_event_id: Annotated[
-        Optional[str],
+        str | None,
         Field(None, description="ID of the credit event for this message"),
     ]
     credit_cost: Annotated[
-        Optional[Decimal],
+        Decimal | None,
         Field(None, description="Credit cost for the message in credits"),
     ]
     cold_start_cost: Annotated[
@@ -356,19 +359,19 @@ class ChatMessageCreate(BaseModel):
         Field(0.0, description="Cost for the cold start of the message in seconds"),
     ]
     app_id: Annotated[
-        Optional[str],
+        str | None,
         Field(None, description="Optional application identifier"),
     ]
     search_mode: Annotated[
-        Optional[bool],
+        bool | None,
         Field(None, description="Optional flag to enable search mode"),
     ]
     super_mode: Annotated[
-        Optional[bool],
+        bool | None,
         Field(None, description="Optional flag to enable super mode"),
     ]
     error_type: Annotated[
-        Optional[SystemMessageType],
+        SystemMessageType | None,
         Field(None, description="Optional error type, used when author_type is system"),
     ]
 
@@ -527,7 +530,7 @@ class ChatMessage(ChatMessageCreate):
         return ChatMessage.model_validate(sanitized_data)
 
     @classmethod
-    async def get(cls, message_id: str) -> Optional["ChatMessage"]:
+    async def get(cls, message_id: str) -> ChatMessage | None:
         async with get_session() as db:
             raw = await db.get(ChatMessageTable, message_id)
             if raw:
@@ -570,7 +573,7 @@ class ChatTable(Base):
         DateTime(timezone=True),
         nullable=False,
         server_default=func.now(),
-        onupdate=lambda: datetime.now(timezone.utc),
+        onupdate=lambda: datetime.now(UTC),
     )
 
 
@@ -625,7 +628,7 @@ class Chat(ChatCreate):
     ]
 
     @classmethod
-    async def get(cls, id: str) -> Optional["Chat"]:
+    async def get(cls, id: str) -> "Chat" | None:
         """Get a chat by its ID.
 
         Args:
@@ -689,7 +692,7 @@ class Chat(ChatCreate):
             return self
 
     @classmethod
-    async def get_by_agent_user(cls, agent_id: str, user_id: str) -> List["Chat"]:
+    async def get_by_agent_user(cls, agent_id: str, user_id: str) -> list["Chat"]:
         """Get all chats for a specific agent and user.
 
         Args:

intentkit/models/conversation.py

@@ -5,7 +5,7 @@ related to agent generation sessions.
 """
 
 from datetime import datetime
-from typing import Annotated, List, Optional
+from typing import Annotated
 
 from epyxid import XID
 from intentkit.models.base import Base
@@ -103,7 +103,7 @@ class ConversationProjectCreate(BaseModel):
         ),
     ]
     user_id: Annotated[
-        Optional[str],
+        str | None,
         Field(None, description="User ID associated with this project"),
     ]
 
@@ -144,7 +144,7 @@ class ConversationProject(ConversationProjectCreate):
     ]
 
     @classmethod
-    async def get(cls, project_id: str) -> Optional["ConversationProject"]:
+    async def get(cls, project_id: str) -> "ConversationProject" | None:
         """Get a conversation project by ID."""
         async with get_session() as db:
             result = await db.execute(
@@ -179,8 +179,8 @@ class ConversationProject(ConversationProjectCreate):
 
     @classmethod
     async def get_by_user(
-        cls, user_id: Optional[str] = None, limit: int = 50
-    ) -> List["ConversationProject"]:
+        cls, user_id: str | None = None, limit: int = 50
+    ) -> list["ConversationProject"]:
         """Get conversation projects by user ID."""
         async with get_session() as db:
             query = select(ConversationProjectTable).order_by(
@@ -213,7 +213,7 @@ class ConversationMessageCreate(BaseModel):
     role: Annotated[str, Field(description="Role of the message sender")]
     content: Annotated[str, Field(description="Content of the message")]
     message_metadata: Annotated[
-        Optional[dict],
+        dict | None,
         Field(None, description="Additional metadata for the message"),
     ]
 
@@ -255,8 +255,8 @@ class ConversationMessage(ConversationMessageCreate):
 
     @classmethod
     async def get_by_project(
-        cls, project_id: str, user_id: Optional[str] = None
-    ) -> List["ConversationMessage"]:
+        cls, project_id: str, user_id: str | None = None
+    ) -> list["ConversationMessage"]:
         """Get conversation messages for a project."""
         async with get_session() as db:
             # First check if project exists and user has access