intentkit 0.7.5.dev23__py3-none-any.whl → 0.7.5.dev25__py3-none-any.whl

intentkit/__init__.py

@@ -3,7 +3,7 @@
 A powerful platform for building AI agents with blockchain and cryptocurrency capabilities.
 """
 
-__version__ = "0.7.5-dev23"
+__version__ = "0.7.5-dev25"
 __author__ = "hyacinthus"
 __email__ = "hyacinthus@gmail.com"
 

intentkit/abstracts/graph.py

@@ -34,6 +34,8 @@ class AgentContext(BaseModel):
     app_id: Optional[str] = None
     entrypoint: AuthorType
     is_private: bool
+    search: bool = False
+    thinking: bool = False
     payer: Optional[str] = None
 
     @property

intentkit/clients/__init__.py

@@ -1,4 +1,4 @@
-from intentkit.clients.cdp import CdpClient, get_cdp_client
+from intentkit.clients.cdp import get_origin_cdp_client, get_wallet_provider
 from intentkit.clients.twitter import (
     TwitterClient,
     TwitterClientConfig,
@@ -10,7 +10,7 @@ __all__ = [
     "TwitterClient",
     "TwitterClientConfig",
     "get_twitter_client",
-    "CdpClient",
-    "get_cdp_client",
+    "get_origin_cdp_client",
+    "get_wallet_provider",
     "get_web3_client",
 ]

intentkit/clients/cdp.py

@@ -1,11 +1,10 @@
 import asyncio
 import json
 import logging
-from typing import Dict, Optional
+from typing import Dict, Optional, Tuple
 
 from bip32 import BIP32
 from cdp import CdpClient as OriginCdpClient  # noqa: E402
-from cdp import EvmServerAccount  # noqa: E402
 from coinbase_agentkit import (  # noqa: E402
     CdpEvmWalletProvider,
     CdpEvmWalletProviderConfig,
@@ -13,11 +12,12 @@ from coinbase_agentkit import (  # noqa: E402
 from eth_keys.datatypes import PrivateKey
 from eth_utils import to_checksum_address
 
-from intentkit.abstracts.skill import SkillStoreABC  # noqa: E402
+from intentkit.config.config import config
 from intentkit.models.agent import Agent  # noqa: E402
-from intentkit.models.agent_data import AgentData  # noqa: E402
+from intentkit.models.agent_data import AgentData
+from intentkit.utils.error import IntentKitAPIError  # noqa: E402
 
-_clients: Dict[str, "CdpClient"] = {}
+_wallet_providers: Dict[str, Tuple[str, str, CdpEvmWalletProvider]] = {}
 _origin_cdp_client: Optional[OriginCdpClient] = None
 
 logger = logging.getLogger(__name__)
@@ -58,15 +58,15 @@ def bip39_seed_to_eth_keys(seed_hex: str) -> Dict[str, str]:
     }
 
 
-def get_origin_cdp_client(skill_store: SkillStoreABC) -> OriginCdpClient:
+def get_origin_cdp_client() -> OriginCdpClient:
     global _origin_cdp_client
     if _origin_cdp_client:
         return _origin_cdp_client
 
-    # Get credentials from skill store system config
-    api_key_id = skill_store.get_system_config("cdp_api_key_id")
-    api_key_secret = skill_store.get_system_config("cdp_api_key_secret")
-    wallet_secret = skill_store.get_system_config("cdp_wallet_secret")
+    # Get credentials from global configuration
+    api_key_id = config.cdp_api_key_id
+    api_key_secret = config.cdp_api_key_secret
+    wallet_secret = config.cdp_wallet_secret
 
     _origin_cdp_client = OriginCdpClient(
         api_key_id=api_key_id,
@@ -76,96 +76,84 @@ def get_origin_cdp_client(skill_store: SkillStoreABC) -> OriginCdpClient:
     return _origin_cdp_client
 
 
-class CdpClient:
-    def __init__(self, agent_id: str, skill_store: SkillStoreABC) -> None:
-        self._agent_id = agent_id
-        self._skill_store = skill_store
-        self._wallet_provider: Optional[CdpEvmWalletProvider] = None
-        self._wallet_provider_config: Optional[CdpEvmWalletProviderConfig] = None
-
-    async def get_wallet_provider(self) -> CdpEvmWalletProvider:
-        if self._wallet_provider:
-            return self._wallet_provider
-        agent: Agent = await self._skill_store.get_agent_config(self._agent_id)
-        agent_data: AgentData = await self._skill_store.get_agent_data(self._agent_id)
-        network_id = agent.network_id
+async def get_wallet_provider(agent: Agent) -> CdpEvmWalletProvider:
+    if agent.wallet_provider != "cdp":
+        raise IntentKitAPIError(
+            400,
+            "BadWalletProvider",
+            "Your agent wallet provider is not cdp but you selected a skill that requires a cdp wallet.",
+        )
 
-        # Get credentials from skill store system config
-        api_key_id = self._skill_store.get_system_config("cdp_api_key_id")
-        api_key_secret = self._skill_store.get_system_config("cdp_api_key_secret")
-        wallet_secret = self._skill_store.get_system_config("cdp_wallet_secret")
+    if not agent.network_id:
+        raise IntentKitAPIError(
+            400,
+            "BadNetworkID",
+            "Your agent network ID is not set. Please set it in the agent config.",
+        )
 
-        # already have address
-        address = agent_data.evm_wallet_address
+    agent_data = await AgentData.get(agent.id)
+    address = agent_data.evm_wallet_address
 
-        # new agent or address not migrated yet
-        if not address:
-            # create cdp client for later use
-            cdp_client = get_origin_cdp_client(self._skill_store)
-            # try migrating from v1 cdp_wallet_data
-            if agent_data.cdp_wallet_data:
-                wallet_data = json.loads(agent_data.cdp_wallet_data)
-                if not isinstance(wallet_data, dict):
-                    raise ValueError("Invalid wallet data format")
-                if wallet_data.get("default_address_id") and wallet_data.get("seed"):
-                    # verify seed and convert to pk
-                    keys = bip39_seed_to_eth_keys(wallet_data["seed"])
-                    if keys["address"] != wallet_data["default_address_id"]:
-                        raise ValueError(
-                            "Bad wallet data, seed does not match default_address_id"
-                        )
-                    # try to import wallet to v2
-                    logger.info("Migrating wallet data to v2...")
-                    await cdp_client.evm.import_account(
-                        name=agent.id,
-                        private_key=keys["private_key"],
-                    )
-                    address = keys["address"]
-                    logger.info("Migrated wallet data to v2 successfully: %s", address)
-            # still not address
+    cache_entry = _wallet_providers.get(agent.id)
+    if cache_entry:
+        cached_network_id, cached_address, provider = cache_entry
+        if cached_network_id == agent.network_id:
             if not address:
-                logger.info("Creating new wallet...")
-                new_account = await cdp_client.evm.create_account(
+                address = cached_address or provider.get_address()
+            if cached_address == address:
+                return provider
+
+    # Get credentials from global config
+    api_key_id = config.cdp_api_key_id
+    api_key_secret = config.cdp_api_key_secret
+    wallet_secret = config.cdp_wallet_secret
+
+    network_id = agent.network_id
+
+    # new agent or address not migrated yet
+    if not address:
+        cdp_client = get_origin_cdp_client()
+        # try migrating from v1 cdp_wallet_data
+        if agent_data.cdp_wallet_data:
+            wallet_data = json.loads(agent_data.cdp_wallet_data)
+            if not isinstance(wallet_data, dict):
+                raise ValueError("Invalid wallet data format")
+            if wallet_data.get("default_address_id") and wallet_data.get("seed"):
+                # verify seed and convert to pk
+                keys = bip39_seed_to_eth_keys(wallet_data["seed"])
+                if keys["address"] != wallet_data["default_address_id"]:
+                    raise ValueError(
+                        "Bad wallet data, seed does not match default_address_id"
+                    )
+                # try to import wallet to v2
+                logger.info("Migrating wallet data to v2...")
+                await cdp_client.evm.import_account(
                     name=agent.id,
+                    private_key=keys["private_key"],
                 )
-                address = new_account.address
-                logger.info("Created new wallet: %s", address)
-
-            # do not close cached global client
-            # now it should be created or migrated, store it
-            agent_data.evm_wallet_address = address
-            await agent_data.save()
-
-        # it must have v2 account now, load agentkit wallet provider
-        self._wallet_provider_config = CdpEvmWalletProviderConfig(
-            api_key_id=api_key_id,
-            api_key_secret=api_key_secret,
-            network_id=network_id,
-            address=address,
-            wallet_secret=wallet_secret,
-        )
-        self._wallet_provider = await asyncio.to_thread(
-            CdpEvmWalletProvider, self._wallet_provider_config
-        )
-        return self._wallet_provider
-
-    async def get_account(self) -> EvmServerAccount:
-        """Get the account object from the wallet provider.
-
-        Returns:
-            EvmServerAccount: The account object that can be used for balance checks, transfers, etc.
-        """
-        wallet_provider = await self.get_wallet_provider()
-        # Access the internal account object
-        return wallet_provider._account
-
-    async def get_provider_config(self) -> CdpEvmWalletProviderConfig:
-        if not self._wallet_provider_config:
-            await self.get_wallet_provider()
-        return self._wallet_provider_config
+                address = keys["address"]
+                logger.info("Migrated wallet data to v2 successfully: %s", address)
+        # still not address
+        if not address:
+            logger.info("Creating new wallet...")
+            new_account = await cdp_client.evm.create_account(
+                name=agent.id,
+            )
+            address = new_account.address
+            logger.info("Created new wallet: %s", address)
 
+        agent_data.evm_wallet_address = address
+        await agent_data.save()
 
-async def get_cdp_client(agent_id: str, skill_store: SkillStoreABC) -> "CdpClient":
-    if agent_id not in _clients:
-        _clients[agent_id] = CdpClient(agent_id, skill_store)
-    return _clients[agent_id]
+    wallet_provider_config = CdpEvmWalletProviderConfig(
+        api_key_id=api_key_id,
+        api_key_secret=api_key_secret,
+        network_id=network_id,
+        address=address,
+        wallet_secret=wallet_secret,
+    )
+    wallet_provider = await asyncio.to_thread(
+        CdpEvmWalletProvider, wallet_provider_config
+    )
+    _wallet_providers[agent.id] = (network_id, address, wallet_provider)
+    return wallet_provider

intentkit/core/agent.py

@@ -7,7 +7,7 @@ from typing import Any, Dict, List, Optional, Tuple
 from sqlalchemy import func, select, text, update
 
 from intentkit.abstracts.skill import SkillStoreABC
-from intentkit.clients.cdp import get_cdp_client
+from intentkit.clients.cdp import get_wallet_provider
 from intentkit.config.config import config
 from intentkit.models.agent import (
     Agent,
@@ -73,8 +73,7 @@ async def process_agent_wallet(
 
     # 4. Initialize wallet based on provider type
     if config.cdp_api_key_id and current_wallet_provider == "cdp":
-        cdp_client = await get_cdp_client(agent.id, agent_store)
-        await cdp_client.get_wallet_provider()
+        await get_wallet_provider(agent)
         agent_data = await AgentData.get(agent.id)
     elif current_wallet_provider == "readonly":
         agent_data = await AgentData.patch(
@@ -158,7 +157,7 @@ def send_agent_notification(agent: Agent, agent_data: AgentData, message: str) -
                     {
                         "title": "Network",
                         "short": True,
-                        "value": agent.network_id or "Default",
+                        "value": agent.network_id or "Not Set",
                     },
                     {
                         "title": "X Username",

intentkit/core/engine.py

@@ -96,12 +96,6 @@ async def build_agent(agent: Agent, agent_data: AgentData) -> CompiledStateGraph
         presence_penalty=agent.presence_penalty,
     )
 
-    # Get the LLM instance
-    llm = await llm_model.create_instance(config)
-
-    # Get the token limit from the model info
-    input_token_limit = min(config.input_token_limit, llm_model.info.context_length)
-
     # ==== Store buffered conversation history in memory.
     memory = get_langgraph_checkpointer()
 
@@ -118,13 +112,13 @@ async def build_agent(agent: Agent, agent_data: AgentData) -> CompiledStateGraph
                 if hasattr(skill_module, "get_skills"):
                     # all
                     skill_tools = await skill_module.get_skills(
-                        v, False, agent_store, agent_id=agent.id
+                        v, False, agent_store, agent_id=agent.id, agent=agent
                     )
                     if skill_tools and len(skill_tools) > 0:
                         tools.extend(skill_tools)
                     # private
                     skill_private_tools = await skill_module.get_skills(
-                        v, True, agent_store, agent_id=agent.id
+                        v, True, agent_store, agent_id=agent.id, agent=agent
                     )
                     if skill_private_tools and len(skill_private_tools) > 0:
                         private_tools.extend(skill_private_tools)
@@ -137,24 +131,27 @@ async def build_agent(agent: Agent, agent_data: AgentData) -> CompiledStateGraph
     tools = list({tool.name: tool for tool in tools}.values())
     private_tools = list({tool.name: tool for tool in private_tools}.values())
 
-    # Add search tools if requested
-    if (
-        llm_model.info.provider == LLMProvider.OPENAI and llm_model.info.supports_search
-        # and not agent.model.startswith(
-        #     "gpt-5"
-        # )  # tmp disable gpt-5 search since package bugs
-    ):
-        tools.append({"type": "web_search"})
-        private_tools.append({"type": "web_search"})
-
     # Create the formatted_prompt function using the refactored prompt module
     formatted_prompt = create_formatted_prompt_function(agent, agent_data)
 
     # bind tools to llm
-    def select_model(
+    async def select_model(
         state: AgentState, runtime: Runtime[AgentContext]
     ) -> BaseChatModel:
+        llm_params = {}
         context = runtime.context
+        if context.search:
+            if llm_model.info.supports_search:
+                if llm_model.info.provider == LLMProvider.OPENAI:
+                    tools.append({"type": "web_search"})
+                    private_tools.append({"type": "web_search"})
+                    if agent.model.startswith("gpt-5-"):
+                        llm_params["reasoning_effort"] = "low"
+                if llm_model.info.provider == LLMProvider.XAI:
+                    llm_params["search_parameters"] = {"mode": "auto"}
+            # TODO: else use a search skill
+        # build llm now
+        llm = await llm_model.create_instance(llm_params)
         if context.is_private:
             return llm.bind_tools(private_tools)
         return llm.bind_tools(tools)
@@ -165,10 +162,12 @@ async def build_agent(agent: Agent, agent_data: AgentData) -> CompiledStateGraph
         )
 
     # Pre model hook
+    summarize_llm = await create_llm_model(model_name="gpt-5-mini")
+    summarize_model = await summarize_llm.create_instance()
     pre_model_hook = PreModelNode(
-        model=llm,
+        model=summarize_model,
         short_term_memory_strategy=agent.short_term_memory_strategy,
-        max_tokens=input_token_limit // 2,
+        max_tokens=llm_model.info.context_length // 2,
         max_summary_tokens=2048,
     )
 
@@ -403,9 +402,11 @@ async def stream_agent_raw(
         input_message = re.sub(r"\b@super\b", "", input_message).strip()
 
     # llm native search
+    search = False
     if re.search(r"\b@search\b", input_message) or re.search(
         r"\b@web\b", input_message
     ):
+        search = True
         if model.supports_search:
             input_message = re.sub(
                 r"\b@search\b",
@@ -467,6 +468,7 @@ async def stream_agent_raw(
         app_id=user_message.app_id,
         entrypoint=user_message.author_type,
         is_private=is_private,
+        search=search,
         payer=payer if payment_enabled else None,
     )
 

intentkit/core/node.py

@@ -66,7 +66,7 @@ class PreModelNode(RunnableCallable):
         model: LanguageModelLike,
         short_term_memory_strategy: str,
         max_tokens: int,
-        max_summary_tokens: int = 1024,
+        max_summary_tokens: int = 2048,
     ) -> None:
         super().__init__(self._func, self._afunc, name="pre_model_node", trace=False)
         self.model = model

intentkit/models/agent.py

@@ -581,13 +581,13 @@ class AgentCore(BaseModel):
     wallet_provider: Annotated[
         Optional[Literal["cdp", "readonly"]],
         PydanticField(
-            default="cdp",
+            default=None,
             description="Provider of the agent's wallet",
             json_schema_extra={
                 "x-group": "onchain",
             },
         ),
-    ] = "cdp"
+    ]
     readonly_wallet_address: Annotated[
         Optional[str],
         PydanticField(
@@ -598,12 +598,12 @@ class AgentCore(BaseModel):
     network_id: Annotated[
         Optional[
             Literal[
+                "base-mainnet",
+                "base-sepolia",
                 "ethereum-mainnet",
                 "ethereum-sepolia",
                 "polygon-mainnet",
                 "polygon-mumbai",
-                "base-mainnet",
-                "base-sepolia",
                 "arbitrum-mainnet",
                 "arbitrum-sepolia",
                 "optimism-mainnet",

intentkit/models/agent_schema.json

@@ -79,80 +79,11 @@
       "type": "string",
       "description": "Select the LLM for your agent. Note that each LLM has its specific advantages, behaviour and cost.",
       "default": "gpt-5-mini",
-      "enum": [
-        "gpt-5-nano",
-        "gpt-5-mini",
-        "gpt-5",
-        "gpt-4.1-nano",
-        "gpt-4.1-mini",
-        "gpt-4.1",
-        "gpt-4o",
-        "gpt-4o-mini",
-        "o3",
-        "o4-mini",
-        "deepseek-chat",
-        "deepseek-reasoner",
-        "grok-2",
-        "grok-3",
-        "grok-3-mini",
-        "eternalai",
-        "reigent",
-        "venice-uncensored",
-        "venice-llama-4-maverick-17b"
-      ],
+      "enum": [],
       "x-component": "category-select",
-      "x-enum-title": [
-        "GPT-5 nano",
-        "GPT-5 mini",
-        "GPT-5",
-        "GPT-4.1 nano",
-        "GPT-4.1 mini",
-        "GPT-4.1",
-        "GPT-4o",
-        "GPT-4o mini",
-        "OpenAI o3",
-        "OpenAI o4-mini",
-        "Deepseek V3.1",
-        "Deepseek V3.1 Thinking",
-        "Grok 2",
-        "Grok 3",
-        "Grok 3 Mini"
-      ],
-      "x-enum-category": [
-        "OpenAI",
-        "OpenAI",
-        "OpenAI",
-        "OpenAI",
-        "OpenAI",
-        "OpenAI",
-        "OpenAI",
-        "OpenAI",
-        "OpenAI",
-        "OpenAI",
-        "Deepseek",
-        "Deepseek",
-        "Grok",
-        "Grok",
-        "Grok"
-      ],
-      "x-support-skill": [
-        true,
-        true,
-        true,
-        true,
-        true,
-        true,
-        true,
-        true,
-        true,
-        true,
-        true,
-        true,
-        true,
-        true,
-        true,
-        true
-      ],
+      "x-enum-title": [],
+      "x-enum-category": [],
+      "x-support-skill": [],
       "x-group": "llm"
     },
     "purpose": {
@@ -218,6 +149,17 @@
       "x-str-type": "prompt",
       "x-group": "llm"
     },
+    "short_term_memory_strategy": {
+      "title": "Short Term Memory Strategy",
+      "type": "string",
+      "description": "Strategy for managing short-term memory when context limit is reached. 'trim' removes oldest messages, 'summarize' creates summaries.",
+      "default": "trim",
+      "enum": [
+        "trim",
+        "summarize"
+      ],
+      "x-group": "llm"
+    },
     "temperature": {
       "title": "Temperature",
       "type": "number",
@@ -236,7 +178,8 @@
       "minimum": -2.0,
       "maximum": 2.0,
       "x-group": "llm",
-      "x-step": 0.1
+      "x-step": 0.1,
+      "x-advanced": true
     },
     "presence_penalty": {
       "title": "Presence Penalty",
@@ -246,18 +189,8 @@
       "minimum": -2.0,
       "maximum": 2.0,
       "x-group": "llm",
-      "x-step": 0.1
-    },
-    "short_term_memory_strategy": {
-      "title": "Short Term Memory Strategy",
-      "type": "string",
-      "description": "Strategy for managing short-term memory when context limit is reached. 'trim' removes oldest messages, 'summarize' creates summaries.",
-      "default": "trim",
-      "enum": [
-        "trim",
-        "summarize"
-      ],
-      "x-group": "llm"
+      "x-step": 0.1,
+      "x-advanced": true
     },
     "telegram_entrypoint_enabled": {
       "title": "Enable Telegram Communication",
@@ -402,7 +335,6 @@
         "Coinbase Server Wallet V2",
         "Readonly Wallet"
       ],
-      "default": "cdp",
       "x-group": "onchain"
     },
     "network_id": {