intentkit 0.6.13.dev2__py3-none-any.whl → 0.8.17__py3-none-any.whl

intentkit/models/agent_data.py

@@ -1,11 +1,13 @@
+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 fastapi import HTTPException
 from intentkit.models.base import Base
 from intentkit.models.db import get_session
+from intentkit.utils.error import IntentKitAPIError
 from pydantic import BaseModel, ConfigDict
 from pydantic import Field as PydanticField
 from sqlalchemy import (
@@ -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),
         ),
     ]
 
@@ -754,7 +780,11 @@ class AgentQuota(BaseModel):
                 await session.execute(stmt)
         except Exception as e:
             logger.error(f"Error adding free income: {str(e)}")
-            raise HTTPException(status_code=500, detail=f"Database error: {str(e)}")
+            raise IntentKitAPIError(
+                status_code=500,
+                key="DatabaseError",
+                message=f"Database error: {str(e)}",
+            )
 
     async def add_message(self) -> None:
         """Add a message to the agent's message count."""
@@ -766,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()
 
@@ -786,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()
 
@@ -807,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/agent_public.json

@@ -0,0 +1,98 @@
+{
+  "$schema": "http://json-schema.org/draft-07/schema#",
+  "type": "object",
+  "title": "AgentPublicInfo",
+  "description": "Public information of the agent",
+  "properties": {
+    "description": {
+      "title": "Description",
+      "type": "string",
+      "description": "Description of the agent, for public view, not contained in prompt",
+      "x-group": "basic",
+      "x-placeholder": "Introduce your agent"
+    },
+    "external_website": {
+      "title": "External Website",
+      "type": "string",
+      "description": "Link of external website of the agent, if you have one",
+      "format": "uri",
+      "x-group": "basic",
+      "x-placeholder": "Enter agent external website url"
+    },
+    "ticker": {
+      "title": "Ticker",
+      "type": "string",
+      "description": "Ticker symbol of the agent",
+      "maxLength": 10,
+      "minLength": 1,
+      "x-group": "basic",
+      "x-placeholder": "If one day, your agent has it's own token, what will it be?"
+    },
+    "token_address": {
+      "title": "Token Address",
+      "type": "string",
+      "description": "Token address of the agent",
+      "maxLength": 42,
+      "x-group": "internal",
+      "readOnly": true
+    },
+    "token_pool": {
+      "title": "Token Pool",
+      "type": "string",
+      "description": "Pool of the agent token",
+      "maxLength": 42,
+      "x-group": "internal",
+      "readOnly": true
+    },
+    "fee_percentage": {
+      "title": "Fee Percentage",
+      "type": "number",
+      "description": "Fee percentage of the agent",
+      "minimum": 0,
+      "x-group": "basic"
+    },
+    "example_intro": {
+      "title": "Example Introduction",
+      "type": "string",
+      "description": "Introduction of the example",
+      "maxLength": 2000,
+      "x-group": "examples"
+    },
+    "examples": {
+      "title": "Examples",
+      "type": "array",
+      "description": "List of example prompts for the agent",
+      "maxItems": 6,
+      "items": {
+        "type": "object",
+        "properties": {
+          "name": {
+            "title": "Name",
+            "type": "string",
+            "description": "Name of the example",
+            "maxLength": 50,
+            "x-group": "examples"
+          },
+          "description": {
+            "title": "Description",
+            "type": "string",
+            "description": "Description of the example",
+            "maxLength": 200,
+            "x-group": "examples"
+          },
+          "prompt": {
+            "title": "Prompt",
+            "type": "string",
+            "description": "Example prompt",
+            "maxLength": 2000,
+            "x-group": "examples"
+          }
+        },
+        "required": ["name", "description", "prompt"]
+      },
+      "x-group": "examples",
+      "x-inline": true
+    }
+  },
+  "additionalProperties": true
+}