intentkit 0.8.22.dev4__tar.gz → 0.8.22.dev6__tar.gz

{intentkit-0.8.22.dev4 → intentkit-0.8.22.dev6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: intentkit
-Version: 0.8.22.dev4
+Version: 0.8.22.dev6
 Summary: Intent-based AI Agent Platform - Core Package
 Project-URL: Homepage, https://github.com/crestalnetwork/intentkit
 Project-URL: Repository, https://github.com/crestalnetwork/intentkit

{intentkit-0.8.22.dev4 → intentkit-0.8.22.dev6}/intentkit/__init__.py

@@ -3,7 +3,7 @@
 A powerful platform for building AI agents with blockchain and cryptocurrency capabilities.
 """
 
-__version__ = "0.8.22-dev4"
+__version__ = "0.8.22-dev6"
 __author__ = "Ruihua"
 __email__ = "ruihua@crestal.network"
 

{intentkit-0.8.22.dev4 → intentkit-0.8.22.dev6}/intentkit/config/config.py

@@ -151,9 +151,6 @@ class Config:
             )
         if hasattr(self, "chain_provider"):
             self.chain_provider.init_chain_configs()
-        self.rpc_networks = self.load(
-            "RPC_NETWORKS", "base-mainnet,base-sepolia,ethereum-sepolia,solana-mainnet"
-        )
 
         # Nation
         self.nation_api_key = self.load("NATION_API_KEY")

{intentkit-0.8.22.dev4 → intentkit-0.8.22.dev6}/intentkit/models/agent.py

@@ -38,11 +38,18 @@ from intentkit.models.credit import CreditAccount
 from intentkit.models.db import get_session
 from intentkit.models.llm import LLMModelInfo, LLMProvider
 from intentkit.models.skill import Skill
+from intentkit.utils.ens import resolve_ens_to_address
 from intentkit.utils.error import IntentKitAPIError
 
 logger = logging.getLogger(__name__)
 
 
+ENS_NAME_PATTERN = re.compile(
+    r"^(?:[a-z0-9](?:[a-z0-9-]{0,61}[a-z0-9])?\.)+(?:eth|base\.eth)$",
+    re.IGNORECASE,
+)
+
+
 class AgentAutonomous(BaseModel):
     """Autonomous agent configuration."""
 
@@ -1337,16 +1344,20 @@ class Agent(AgentCreate, AgentPublicInfo):
         Returns:
             Agent if found, None otherwise
         """
+        query_id = agent_id
+        if ENS_NAME_PATTERN.fullmatch(agent_id):
+            query_id = await resolve_ens_to_address(agent_id)
+
         async with get_session() as db:
             agent = None
 
             # Try to get by ID if length <= 20
-            if len(agent_id) <= 20:
-                agent = await Agent.get(agent_id)
+            if len(query_id) <= 20 or query_id.startswith("0x"):
+                agent = await Agent.get(query_id)
 
             # If not found, try to get by slug
             if agent is None:
-                slug_stmt = select(AgentTable).where(AgentTable.slug == agent_id)
+                slug_stmt = select(AgentTable).where(AgentTable.slug == query_id)
                 agent_row = await db.scalar(slug_stmt)
                 if agent_row is not None:
                     agent = Agent.model_validate(agent_row)

{intentkit-0.8.22.dev4 → intentkit-0.8.22.dev6}/intentkit/models/chat.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 from datetime import UTC, datetime
 from decimal import Decimal
 from enum import Enum
-from typing import Annotated, NotRequired, TypedDict
+from typing import Annotated, ClassVar, NotRequired, TypedDict, final
 
 from epyxid import XID
 from pydantic import BaseModel, ConfigDict, Field
@@ -23,6 +23,7 @@ from sqlalchemy import (
 )
 from sqlalchemy.dialects.postgresql import JSON, JSONB
 from sqlalchemy.ext.asyncio import AsyncSession
+from sqlalchemy.orm import Mapped, mapped_column
 
 from intentkit.models.app_setting import AppSetting, SystemMessageType
 from intentkit.models.base import Base
@@ -77,7 +78,7 @@ class ChatMessageAttachment(TypedDict):
         ),
     ]
     json: Annotated[
-        dict | None,
+        dict[str, object] | None,
         Field(
             None,
             description="JSON data of the attachment",
@@ -90,7 +91,7 @@ class ChatMessageSkillCall(TypedDict):
 
     id: NotRequired[str]
     name: str
-    parameters: dict
+    parameters: dict[str, object]
     success: bool
     response: NotRequired[
         str
@@ -167,126 +168,67 @@ class ChatMessageRequest(BaseModel):
         ),
     ]
 
-    model_config = ConfigDict(
+    model_config: ClassVar[ConfigDict] = ConfigDict(
         use_enum_values=True,
         json_schema_extra={
             "example": {
                 "chat_id": "chat-123",
                 "app_id": "app-789",
                 "user_id": "user-456",
-                "message": "Hello, how can you help me today?",
-                "search_mode": True,
+                "message": "Hello, what can you do?",
+                "search_mode": False,
                 "super_mode": False,
-                "attachments": [
-                    {
-                        "type": "link",
-                        "url": "https://example.com",
-                    }
-                ],
             }
         },
     )
 
 
+@final
 class ChatMessageTable(Base):
     """Chat message database table model."""
 
     __tablename__ = "chat_messages"
-    __table_args__ = (
+    __table_args__ = (  # pyright: ignore[reportAny]
         Index("ix_chat_messages_chat_id", "chat_id"),
         Index("ix_chat_messages_agent_id_author_type", "agent_id", "author_type"),
         Index("ix_chat_messages_agent_id_chat_id", "agent_id", "chat_id"),
     )
 
-    id = Column(
-        String,
-        primary_key=True,
-    )
-    agent_id = Column(
-        String,
-        nullable=False,
-    )
-    chat_id = Column(
-        String,
-        nullable=False,
-    )
-    user_id = Column(
-        String,
-        nullable=True,
-    )
-    author_id = Column(
-        String,
-        nullable=False,
-    )
-    author_type = Column(
-        String,
-        nullable=False,
-    )
-    model = Column(
-        String,
-        nullable=True,
-    )
-    thread_type = Column(
-        String,
-        nullable=True,
-    )
-    reply_to = Column(
-        String,
-        nullable=True,
-    )
-    message = Column(
-        String,
-        nullable=False,
-    )
-    attachments = Column(
+    id: Mapped[str] = mapped_column(String, primary_key=True)
+    agent_id: Mapped[str] = mapped_column(String, nullable=False)
+    chat_id: Mapped[str] = mapped_column(String, nullable=False)
+    user_id: Mapped[str | None] = mapped_column(String, nullable=True)
+    author_id: Mapped[str] = mapped_column(String, nullable=False)
+    author_type: Mapped[str] = mapped_column(String, nullable=False)
+    model: Mapped[str | None] = mapped_column(String, nullable=True)
+    thread_type: Mapped[AuthorType | None] = mapped_column(String, nullable=True)
+    reply_to: Mapped[str | None] = mapped_column(String, nullable=True)
+    message: Mapped[str] = mapped_column(String, nullable=False)
+    attachments: Mapped[list[ChatMessageAttachment] | None] = mapped_column(
         JSON().with_variant(JSONB(), "postgresql"),
         nullable=True,
     )
-    skill_calls = Column(
+    skill_calls: Mapped[list[ChatMessageSkillCall] | None] = mapped_column(
         JSON().with_variant(JSONB(), "postgresql"),
         nullable=True,
     )
-    input_tokens = Column(
-        Integer,
-        default=0,
-    )
-    output_tokens = Column(
-        Integer,
-        default=0,
-    )
-    time_cost = Column(
-        Float,
-        default=0,
-    )
-    credit_event_id = Column(
-        String,
-        nullable=True,
-    )
-    credit_cost = Column(
+    input_tokens: Mapped[int] = mapped_column(Integer, default=0)
+    output_tokens: Mapped[int] = mapped_column(Integer, default=0)
+    time_cost: Mapped[float] = mapped_column(Float, default=0)
+    credit_event_id: Mapped[str | None] = mapped_column(String, nullable=True)
+    credit_cost: Mapped[Decimal | None] = mapped_column(
         Numeric(22, 4),
         nullable=True,
     )
-    cold_start_cost = Column(
-        Float,
-        default=0,
-    )
-    app_id = Column(
-        String,
-        nullable=True,
-    )
-    search_mode = Column(
-        Boolean,
-        nullable=True,
-    )
-    super_mode = Column(
-        Boolean,
-        nullable=True,
-    )
-    error_type = Column(
+    cold_start_cost: Mapped[float] = mapped_column(Float, default=0)
+    app_id: Mapped[str | None] = mapped_column(String, nullable=True)
+    search_mode: Mapped[bool | None] = mapped_column(Boolean, nullable=True)
+    super_mode: Mapped[bool | None] = mapped_column(Boolean, nullable=True)
+    error_type: Mapped[SystemMessageType | None] = mapped_column(
         String,
         nullable=True,
     )
-    created_at = Column(
+    created_at: Mapped[datetime] = mapped_column(
         DateTime(timezone=True),
         nullable=False,
         server_default=func.now(),
@@ -296,7 +238,7 @@ class ChatMessageTable(Base):
 class ChatMessageCreate(BaseModel):
     """Base model for creating chat messages with fields needed for creation."""
 
-    model_config = ConfigDict(
+    model_config: ClassVar[ConfigDict] = ConfigDict(
         use_enum_values=True,
         from_attributes=True,
     )
@@ -320,61 +262,61 @@ class ChatMessageCreate(BaseModel):
     author_type: Annotated[AuthorType, Field(description="Type of the message author")]
     model: Annotated[
         str | None, Field(None, description="LLM model used if applicable")
-    ]
+    ] = None
     thread_type: Annotated[
         AuthorType | None,
         Field(None, description="Author Type of the message thread start"),
-    ]
+    ] = None
     reply_to: Annotated[
         str | None,
         Field(None, description="ID of the message this message is a reply to"),
-    ]
+    ] = None
     message: Annotated[str, Field(description="Content of the message")]
     attachments: Annotated[
         list[ChatMessageAttachment] | None,
         Field(None, description="List of attachments in the message"),
-    ]
+    ] = None
     skill_calls: Annotated[
         list[ChatMessageSkillCall] | None,
         Field(None, description="Skill call details"),
-    ]
+    ] = None
     input_tokens: Annotated[
         int, Field(0, description="Number of tokens in the input message")
-    ]
+    ] = 0
     output_tokens: Annotated[
         int, Field(0, description="Number of tokens in the output message")
-    ]
+    ] = 0
     time_cost: Annotated[
         float, Field(0.0, description="Time cost for the message in seconds")
-    ]
+    ] = 0.0
     credit_event_id: Annotated[
         str | None,
         Field(None, description="ID of the credit event for this message"),
-    ]
+    ] = None
     credit_cost: Annotated[
         Decimal | None,
         Field(None, description="Credit cost for the message in credits"),
-    ]
+    ] = None
     cold_start_cost: Annotated[
         float,
         Field(0.0, description="Cost for the cold start of the message in seconds"),
-    ]
+    ] = 0.0
     app_id: Annotated[
         str | None,
         Field(None, description="Optional application identifier"),
-    ]
+    ] = None
     search_mode: Annotated[
         bool | None,
         Field(None, description="Optional flag to enable search mode"),
-    ]
+    ] = None
     super_mode: Annotated[
         bool | None,
         Field(None, description="Optional flag to enable super mode"),
-    ]
+    ] = None
     error_type: Annotated[
         SystemMessageType | None,
         Field(None, description="Optional error type, used when author_type is system"),
-    ]
+    ] = None
 
     async def save_in_session(self, db: AsyncSession) -> "ChatMessage":
         """Save the chat message to the database.
@@ -438,7 +380,7 @@ class ChatMessageCreate(BaseModel):
 class ChatMessage(ChatMessageCreate):
     """Chat message model with all fields including server-generated ones."""
 
-    model_config = ConfigDict(
+    model_config: ClassVar[ConfigDict] = ConfigDict(
         use_enum_values=True,
         json_encoders={
             datetime: lambda v: v.isoformat(timespec="milliseconds"),

{intentkit-0.8.22.dev4 → intentkit-0.8.22.dev6}/intentkit/skills/carv/token_info_and_price.py

@@ -26,7 +26,7 @@ class TokenInfoAndPriceTool(CarvBaseTool):
     Fetches detailed information and the current USD price of a cryptocurrency token from the CARV API,
     given its ticker symbol (e.g., 'eth', 'btc', 'aave').
     Returns metadata including the token's name, symbol, platform, category tags, and contract addresses
-    Useful for understanding a token's identity, ecosystem, and market valu
+    Useful for understanding a token's identity, ecosystem, and market value
     Use this tool when you need comprehensive token data and live pricing from CARV
     """
 

{intentkit-0.8.22.dev4 → intentkit-0.8.22.dev6}/intentkit/skills/enso/base.py

@@ -8,7 +8,11 @@ from intentkit.abstracts.graph import AgentContext
 from intentkit.clients import get_wallet_provider as get_agent_wallet_provider
 from intentkit.config.config import config
 from intentkit.skills.base import IntentKitSkill
-from intentkit.utils.chain import ChainProvider, Network, network_to_id
+from intentkit.utils.chain import (
+    ChainProvider,
+    network_to_id,
+    resolve_quicknode_network,
+)
 
 base_url = "https://api.enso.finance"
 
@@ -65,7 +69,7 @@ class EnsoBaseTool(IntentKitSkill):
 
         agent = context.agent
         try:
-            network = Network(agent.network_id)
+            network = resolve_quicknode_network(agent.network_id)
         except ValueError as exc:  # pragma: no cover - defensive
             raise ToolException(
                 f"Unsupported network configured for agent: {agent.network_id}"

intentkit-0.8.22.dev6/intentkit/utils/__init__.py

@@ -0,0 +1,5 @@
+"""Utility exports for IntentKit."""
+
+from intentkit.utils.ens import resolve_ens_to_address
+
+__all__ = ["resolve_ens_to_address"]