intentkit 0.6.3.dev3__tar.gz → 0.6.5__tar.gz

{intentkit-0.6.3.dev3 → intentkit-0.6.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: intentkit
-Version: 0.6.3.dev3
+Version: 0.6.5
 Summary: Intent-based AI Agent Platform - Core Package
 Project-URL: Homepage, https://github.com/crestal-network/intentkit
 Project-URL: Repository, https://github.com/crestal-network/intentkit

{intentkit-0.6.3.dev3 → intentkit-0.6.5}/__init__.py

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

{intentkit-0.6.3.dev3 → intentkit-0.6.5}/clients/twitter.py

@@ -103,8 +103,6 @@ class TwitterClient(TwitterABC):
         """
         if not self._agent_data:
             self._agent_data = await self._skill_store.get_agent_data(self.agent_id)
-            if not self._agent_data:
-                raise Exception(f"[{self.agent_id}] Agent data not found")
         if not self._client:
             # Check if we have API keys in config
             if self.use_key:
@@ -351,7 +349,7 @@ class TwitterClient(TwitterABC):
         """
         # Get agent data to access the token
         agent_data = await self._skill_store.get_agent_data(agent_id)
-        if not agent_data or not agent_data.twitter_access_token:
+        if not agent_data.twitter_access_token:
             raise ValueError("Only linked X account can post media")
 
         media_ids = []

{intentkit-0.6.3.dev3 → intentkit-0.6.5}/config/config.py

@@ -51,11 +51,11 @@ class Config:
             }
         else:
             self.db = {
-                "username": os.getenv("DB_USERNAME"),
-                "password": os.getenv("DB_PASSWORD"),
-                "host": os.getenv("DB_HOST"),
-                "port": os.getenv("DB_PORT"),
-                "dbname": os.getenv("DB_NAME"),
+                "username": self.load("DB_USERNAME"),
+                "password": self.load("DB_PASSWORD"),
+                "host": self.load("DB_HOST"),
+                "port": self.load("DB_PORT"),
+                "dbname": self.load("DB_NAME"),
             }
         # ==== this part can be load from env or aws secrets manager
         self.db["auto_migrate"] = self.load("DB_AUTO_MIGRATE", "true") == "true"
@@ -66,6 +66,7 @@ class Config:
         # Redis
         self.redis_host = self.load("REDIS_HOST")
         self.redis_port = int(self.load("REDIS_PORT", "6379"))
+        self.redis_db = int(self.load("REDIS_DB", "0"))
         # AWS
         self.aws_s3_bucket = self.load("AWS_S3_BUCKET")
         self.aws_s3_cdn_url = self.load("AWS_S3_CDN_URL")

{intentkit-0.6.3.dev3 → intentkit-0.6.5}/core/engine.py

@@ -126,7 +126,7 @@ async def create_agent(
     Returns:
         CompiledStateGraph: Initialized LangChain agent
     """
-    agent_data: Optional[AgentData] = await AgentData.get(agent.id)
+    agent_data = await AgentData.get(agent.id)
 
     # ==== Initialize LLM using the LLM abstraction.
     from intentkit.models.llm import create_llm_model

{intentkit-0.6.3.dev3 → intentkit-0.6.5}/core/skill.py

@@ -31,15 +31,15 @@ class SkillStore(SkillStoreABC):
         return await Agent.get(agent_id)
 
     @staticmethod
-    async def get_agent_data(agent_id: str) -> Optional[AgentData]:
+    async def get_agent_data(agent_id: str) -> AgentData:
         return await AgentData.get(agent_id)
 
     @staticmethod
-    async def set_agent_data(agent_id: str, data: Dict) -> None:
+    async def set_agent_data(agent_id: str, data: Dict) -> AgentData:
         return await AgentData.patch(agent_id, data)
 
     @staticmethod
-    async def get_agent_quota(agent_id: str) -> Optional[AgentQuota]:
+    async def get_agent_quota(agent_id: str) -> AgentQuota:
         return await AgentQuota.get(agent_id)
 
     @staticmethod

{intentkit-0.6.3.dev3 → intentkit-0.6.5}/models/agent.py

@@ -18,12 +18,10 @@ from pydantic import BaseModel, ConfigDict, field_validator, model_validator
 from pydantic import Field as PydanticField
 from pydantic.json_schema import SkipJsonSchema
 from sqlalchemy import (
-    BigInteger,
     Boolean,
     Column,
     DateTime,
     Float,
-    Identity,
     Numeric,
     String,
     func,
@@ -199,12 +197,6 @@ class AgentTable(Base):
         primary_key=True,
         comment="Unique identifier for the agent. Must be URL-safe, containing only lowercase letters, numbers, and hyphens",
     )
-    number = Column(
-        BigInteger(),
-        Identity(start=1, increment=1),
-        nullable=True,
-        comment="Auto-incrementing number assigned by the system for easy reference",
-    )
     name = Column(
         String,
         nullable=True,
@@ -1114,13 +1106,6 @@ class Agent(AgentCreate):
 
     model_config = ConfigDict(from_attributes=True)
 
-    # auto increment number by db
-    number: Annotated[
-        Optional[int],
-        PydanticField(
-            description="Auto-incrementing number assigned by the system for easy reference",
-        ),
-    ]
     # auto timestamp
     created_at: Annotated[
         datetime,
@@ -1317,13 +1302,6 @@ class AgentResponse(BaseModel):
             description="Unique identifier for the agent. Must be URL-safe, containing only lowercase letters, numbers, and hyphens",
         ),
     ]
-    # auto increment number by db
-    number: Annotated[
-        Optional[int],
-        PydanticField(
-            description="Auto-incrementing number assigned by the system for easy reference",
-        ),
-    ]
     # auto timestamp
     created_at: Annotated[
         datetime,
@@ -1608,7 +1586,9 @@ class AgentResponse(BaseModel):
                     if skill_config.get("enabled") is True:
                         filtered_config = {"enabled": True}
                         # Only keep states with public or private values
-                        if "states" in skill_config:
+                        if "states" in skill_config and isinstance(
+                            skill_config["states"], dict
+                        ):
                             filtered_states = {}
                             for state_key, state_value in skill_config[
                                 "states"

{intentkit-0.6.3.dev3 → intentkit-0.6.5}/models/agent_data.py

@@ -34,11 +34,6 @@ class AgentDataTable(Base):
         String, nullable=True, comment="Solana wallet address"
     )
     cdp_wallet_data = Column(String, nullable=True, comment="CDP wallet data")
-    crossmint_wallet_data = Column(
-        JSON().with_variant(JSONB(), "postgresql"),
-        nullable=True,
-        comment="Crossmint wallet information",
-    )
     twitter_id = Column(String, nullable=True, comment="Twitter user ID")
     twitter_username = Column(String, nullable=True, comment="Twitter username")
     twitter_name = Column(String, nullable=True, comment="Twitter display name")
@@ -104,126 +99,119 @@ class AgentData(BaseModel):
             default=None,
             description="EVM wallet address",
         ),
-    ]
+    ] = None
     solana_wallet_address: Annotated[
         Optional[str],
         PydanticField(
             default=None,
             description="Solana wallet address",
         ),
-    ]
+    ] = None
     cdp_wallet_data: Annotated[
         Optional[str],
         PydanticField(
             default=None,
             description="CDP wallet data",
         ),
-    ]
-    crossmint_wallet_data: Annotated[
-        Optional[dict],
-        PydanticField(
-            default=None,
-            description="Crossmint wallet information",
-        ),
-    ]
+    ] = None
     twitter_id: Annotated[
         Optional[str],
         PydanticField(
             default=None,
             description="Twitter user ID",
         ),
-    ]
+    ] = None
     twitter_username: Annotated[
         Optional[str],
         PydanticField(
             default=None,
             description="Twitter username",
         ),
-    ]
+    ] = None
     twitter_name: Annotated[
         Optional[str],
         PydanticField(
             default=None,
             description="Twitter display name",
         ),
-    ]
+    ] = None
     twitter_access_token: Annotated[
         Optional[str],
         PydanticField(
             default=None,
             description="Twitter access token",
         ),
-    ]
+    ] = None
     twitter_access_token_expires_at: Annotated[
         Optional[datetime],
         PydanticField(
             default=None,
             description="Twitter access token expiration time",
         ),
-    ]
+    ] = None
     twitter_refresh_token: Annotated[
         Optional[str],
         PydanticField(
             default=None,
             description="Twitter refresh token",
         ),
-    ]
+    ] = None
     twitter_self_key_refreshed_at: Annotated[
         Optional[datetime],
         PydanticField(
             default=None,
             description="Twitter self-key userinfo last refresh time",
         ),
-    ]
+    ] = None
     twitter_is_verified: Annotated[
         bool,
         PydanticField(
             default=False,
             description="Whether the Twitter account is verified",
         ),
-    ]
+    ] = None
     telegram_id: Annotated[
         Optional[str],
         PydanticField(
             default=None,
             description="Telegram user ID",
         ),
-    ]
+    ] = None
     telegram_username: Annotated[
         Optional[str],
         PydanticField(
             default=None,
             description="Telegram username",
         ),
-    ]
+    ] = None
     telegram_name: Annotated[
         Optional[str],
         PydanticField(
             default=None,
             description="Telegram display name",
         ),
-    ]
+    ] = None
     error_message: Annotated[
         Optional[str],
         PydanticField(
             default=None,
             description="Last error message",
         ),
-    ]
+    ] = None
     api_key: Annotated[
         Optional[str],
         PydanticField(
             default=None,
             description="API key for the agent",
         ),
-    ]
+    ] = None
     api_key_public: Annotated[
         Optional[str],
         PydanticField(
             default=None,
             description="Public API key for the agent",
         ),
-    ]
+    ] = None
     created_at: Annotated[
         datetime,
         PydanticField(
@@ -240,7 +228,7 @@ class AgentData(BaseModel):
     ]
 
     @classmethod
-    async def get(cls, agent_id: str) -> Optional["AgentData"]:
+    async def get(cls, agent_id: str) -> "AgentData":
         """Get agent data by ID.
 
         Args:

{intentkit-0.6.3.dev3 → intentkit-0.6.5}/models/agent_schema.json

@@ -56,11 +56,7 @@
     }
   ],
   "required": [
-    "name",
-    "model",
-    "purpose",
-    "personality",
-    "principles"
+    "name"
   ],
   "properties": {
     "name": {

{intentkit-0.6.3.dev3 → intentkit-0.6.5}/models/chat.py

@@ -447,6 +447,32 @@ class ChatMessage(ChatMessageCreate):
 
         return resp
 
+    def sanitize_privacy(self) -> "ChatMessage":
+        """Remove sensitive information from the chat message.
+
+        This method clears the message content and removes parameters and response
+        from skill calls while preserving the structure and metadata.
+
+        Returns:
+            ChatMessage: A new ChatMessage instance with sensitive data removed
+        """
+        # Create a copy of the current message
+        sanitized_data = self.model_dump()
+
+        # Clear the message content
+        sanitized_data["message"] = ""
+
+        # Clear sensitive data from skill calls
+        if sanitized_data.get("skill_calls"):
+            for skill_call in sanitized_data["skill_calls"]:
+                # Clear parameters and response while keeping structure
+                skill_call["parameters"] = {}
+                if "response" in skill_call:
+                    skill_call["response"] = ""
+
+        # Return a new ChatMessage instance with sanitized data
+        return ChatMessage.model_validate(sanitized_data)
+
     @classmethod
     async def get(cls, message_id: str) -> Optional["ChatMessage"]:
         async with get_session() as db:

{intentkit-0.6.3.dev3 → intentkit-0.6.5}/models/db.py

@@ -38,8 +38,11 @@ async def init_db(
     # Initialize psycopg pool and AsyncPostgresSaver if not already initialized
     if _langgraph_checkpointer is None:
         if host:
+            conn_string = (
+                f"postgresql://{username}:{quote_plus(password)}@{host}:{port}/{dbname}"
+            )
             pool = AsyncConnectionPool(
-                conninfo=f"postgresql://{username}:{quote_plus(password)}@{host}:{port}/{dbname}",
+                conninfo=conn_string,
                 min_size=3,
                 max_size=20,
                 timeout=60,
@@ -47,7 +50,9 @@ async def init_db(
             )
             _langgraph_checkpointer = AsyncPostgresSaver(pool)
             if auto_migrate:
-                await _langgraph_checkpointer.setup()
+                # Migrate can not use pool, so we start from scratch
+                async with AsyncPostgresSaver.from_conn_string(conn_string) as saver:
+                    await saver.setup()
         else:
             _langgraph_checkpointer = InMemorySaver()
     # Initialize SQLAlchemy engine with pool settings

{intentkit-0.6.3.dev3 → intentkit-0.6.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "intentkit"
-version = "0.6.3-dev3"
+version = "0.6.5"
 description = "Intent-based AI Agent Platform - Core Package"
 authors = [{ name = "hyacinthus", email = "hyacinthus@gmail.com" }]
 requires-python = ">=3.12"

{intentkit-0.6.3.dev3 → intentkit-0.6.5}/skills/system/read_agent_api_key.py

@@ -45,9 +45,6 @@ class ReadAgentApiKey(SystemBaseTool):
         # Get agent data from skill store
         agent_data = await self.skill_store.get_agent_data(agent_id)
 
-        if not agent_data:
-            raise ValueError(f"Agent data not found for agent_id: {agent_id}")
-
         # Get API base URL from system config
         open_api_base_url = self.skill_store.get_system_config("open_api_base_url")
         api_endpoint = f"{open_api_base_url}/v1/chat/completions"

{intentkit-0.6.3.dev3 → intentkit-0.6.5}/skills/system/regenerate_agent_api_key.py

@@ -49,9 +49,6 @@ class RegenerateAgentApiKey(SystemBaseTool):
         # Get agent data from skill store
         agent_data = await self.skill_store.get_agent_data(agent_id)
 
-        if not agent_data:
-            raise ValueError(f"Agent data not found for agent_id: {agent_id}")
-
         # Get API base URL from system config
         open_api_base_url = self.skill_store.get_system_config("open_api_base_url")
         api_endpoint = f"{open_api_base_url}/v1/chat/completions"

{intentkit-0.6.3.dev3 → intentkit-0.6.5}/skills/twitter/reply_tweet.py

@@ -10,8 +10,8 @@ from intentkit.skills.twitter.base import TwitterBaseTool
 
 NAME = "twitter_reply_tweet"
 PROMPT = (
-    "Reply to an existing tweet on Twitter. If you want to post image, "
-    "you must provide image url in parameters, do not add image link in text."
+    "Reply to an existing tweet on Twitter. Do not reply to your own tweet. "
+    "If you want to post image, you must provide image url in parameters, do not add image link in text."
 )
 
 logger = logging.getLogger(__name__)

{intentkit-0.6.3.dev3 → intentkit-0.6.5}/skills/unrealspeech/text_to_speech.py

@@ -78,27 +78,6 @@ class TextToSpeech(UnrealSpeechBaseTool):
             context.config.get("api_key", None) if context and context.config else None
         )
 
-        # If no API key in config, try to get it from skill store
-        if not api_key:
-            try:
-                agent_id = context.agent_id
-                api_key_data = await self.skill_store.get_agent_data(
-                    agent_id, "unrealspeech_api_key"
-                )
-                api_key = api_key_data.get("api_key") if api_key_data else None
-            except Exception as e:
-                logger.warning(f"Failed to get API key from skill store: {e}")
-
-        # If still no API key, check environment variables (handled by UnrealSpeech client)
-        if not api_key:
-            env_key = self.get_env_var("UNREALSPEECH_API_KEY")
-            if not env_key:
-                return {
-                    "success": False,
-                    "error": "No UnrealSpeech API key found. Please set the UNREALSPEECH_API_KEY environment variable or provide it in the agent configuration.",
-                }
-            api_key = env_key
-
         # Clean up and validate input
         if not text:
             return {"success": False, "error": "Text cannot be empty."}