intentkit 0.6.0.dev4__tar.gz → 0.6.0.dev5__tar.gz

{intentkit-0.6.0.dev4 → intentkit-0.6.0.dev5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: intentkit
-Version: 0.6.0.dev4
+Version: 0.6.0.dev5
 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.0.dev4 → intentkit-0.6.0.dev5}/__init__.py

@@ -3,7 +3,7 @@
 A powerful platform for building AI agents with blockchain and cryptocurrency capabilities.
 """
 
-__version__ = "0.6.0-dev.4"
+__version__ = "0.6.0-dev.5"
 __author__ = "hyacinthus"
 __email__ = "hyacinthus@gmail.com"
 

{intentkit-0.6.0.dev4 → intentkit-0.6.0.dev5}/core/engine.py

@@ -187,8 +187,7 @@ async def create_agent(
     if config.admin_llm_skill_control:
         escaped_prompt = await explain_prompt(escaped_prompt)
     prompt_array = [
-        ("system", escaped_prompt),
-        ("placeholder", "{entrypoint_prompt}"),
+        ("placeholder", "{system_prompt}"),
         ("placeholder", "{messages}"),
     ]
     if agent.prompt_append:
@@ -201,18 +200,43 @@ async def create_agent(
 
     prompt_temp = ChatPromptTemplate.from_messages(prompt_array)
 
-    def formatted_prompt(
+    async def formatted_prompt(
         state: AgentState, config: RunnableConfig
     ) -> list[BaseMessage]:
-        entrypoint_prompt = []
-        if config.get("configurable") and config["configurable"].get(
-            "entrypoint_prompt"
-        ):
-            entrypoint_prompt = [
-                ("system", config["configurable"]["entrypoint_prompt"])
-            ]
+        final_system_prompt = escaped_prompt
+        if config.get("configurable") and config["configurable"].get("entrypoint"):
+            entrypoint = config["configurable"]["entrypoint"]
+            entrypoint_prompt = None
+            if (
+                agent.twitter_entrypoint_enabled
+                and agent.twitter_entrypoint_prompt
+                and entrypoint == AuthorType.TWITTER.value
+            ):
+                entrypoint_prompt = agent.twitter_entrypoint_prompt
+                logger.debug("twitter entrypoint prompt added")
+            elif (
+                agent.telegram_entrypoint_enabled
+                and agent.telegram_entrypoint_prompt
+                and entrypoint == AuthorType.TELEGRAM.value
+            ):
+                entrypoint_prompt = agent.telegram_entrypoint_prompt
+                logger.debug("telegram entrypoint prompt added")
+            if entrypoint_prompt:
+                entrypoint_prompt = await explain_prompt(entrypoint_prompt)
+                final_system_prompt = f"{final_system_prompt}## Entrypoint rules\n\n{entrypoint_prompt}\n\n"
+        if config.get("configurable"):
+            final_system_prompt = f"{final_system_prompt}## Internal Info\n\n"
+            "These are for your internal use. You can use them when querying or storing data, "
+            "but please do not directly share this information with users.\n\n"
+            chat_id = config["configurable"].get("chat_id")
+            if chat_id:
+                final_system_prompt = f"{final_system_prompt}chat_id: {chat_id}\n\n"
+            user_id = config["configurable"].get("user_id")
+            if user_id:
+                final_system_prompt = f"{final_system_prompt}user_id: {user_id}\n\n"
+        system_prompt = [("system", final_system_prompt)]
         return prompt_temp.invoke(
-            {"messages": state["messages"], "entrypoint_prompt": entrypoint_prompt},
+            {"messages": state["messages"], "system_prompt": system_prompt},
             config,
         )
 
@@ -518,33 +542,16 @@ async def stream_agent(message: ChatMessageCreate):
         HumanMessage(content=content),
     ]
 
-    entrypoint_prompt = None
-    if (
-        agent.twitter_entrypoint_enabled
-        and agent.twitter_entrypoint_prompt
-        and input.author_type == AuthorType.TWITTER
-    ):
-        entrypoint_prompt = agent.twitter_entrypoint_prompt
-        logger.debug("twitter entrypoint prompt added")
-    elif (
-        agent.telegram_entrypoint_enabled
-        and agent.telegram_entrypoint_prompt
-        and input.author_type == AuthorType.TELEGRAM
-    ):
-        entrypoint_prompt = agent.telegram_entrypoint_prompt
-        logger.debug("telegram entrypoint prompt added")
-    if entrypoint_prompt and config.admin_llm_skill_control:
-        entrypoint_prompt = await explain_prompt(entrypoint_prompt)
-
     # stream config
     thread_id = f"{input.agent_id}-{input.chat_id}"
     stream_config = {
         "configurable": {
             "agent": agent,
             "thread_id": thread_id,
+            "chat_id": input.chat_id,
             "user_id": input.user_id,
+            "app_id": input.app_id,
             "entrypoint": input.author_type,
-            "entrypoint_prompt": entrypoint_prompt,
             "payer": payer if payment_enabled else None,
         },
         "recursion_limit": recursion_limit,

{intentkit-0.6.0.dev4 → intentkit-0.6.0.dev5}/core/node.py

@@ -138,9 +138,11 @@ class PreModelNode(RunnableCallable):
                         f"Too few messages after trim: {len(trimmed_messages)}"
                     )
                     return {}
-            return {
-                "messages": [RemoveMessage(REMOVE_ALL_MESSAGES)] + trimmed_messages,
-            }
+                return {
+                    "messages": [RemoveMessage(REMOVE_ALL_MESSAGES)] + trimmed_messages,
+                }
+            else:
+                return {}
         if self.short_term_memory_strategy == "summarize":
             # if last message is not human message, skip summarize
             if not isinstance(messages[-1], HumanMessage):

{intentkit-0.6.0.dev4 → intentkit-0.6.0.dev5}/models/chat.py

@@ -8,6 +8,7 @@ from intentkit.models.base import Base
 from intentkit.models.db import get_session
 from pydantic import BaseModel, ConfigDict, Field
 from sqlalchemy import (
+    Boolean,
     Column,
     DateTime,
     Float,
@@ -101,6 +102,14 @@ class ChatMessageRequest(BaseModel):
             min_length=1,
         ),
     ]
+    app_id: Annotated[
+        Optional[str],
+        Field(
+            None,
+            description="Optional application identifier",
+            examples=["app-789"],
+        ),
+    ]
     user_id: Annotated[
         str,
         Field(
@@ -120,6 +129,20 @@ class ChatMessageRequest(BaseModel):
             max_length=65535,
         ),
     ]
+    search_mode: Annotated[
+        Optional[bool],
+        Field(
+            None,
+            description="Optional flag to enable search mode",
+        ),
+    ]
+    super_mode: Annotated[
+        Optional[bool],
+        Field(
+            None,
+            description="Optional flag to enable super mode",
+        ),
+    ]
     attachments: Annotated[
         Optional[List[ChatMessageAttachment]],
         Field(
@@ -134,8 +157,11 @@ class ChatMessageRequest(BaseModel):
         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,
+                "super_mode": False,
                 "attachments": [
                     {
                         "type": "link",
@@ -229,6 +255,18 @@ class ChatMessageTable(Base):
         Float,
         default=0,
     )
+    app_id = Column(
+        String,
+        nullable=True,
+    )
+    search_mode = Column(
+        Boolean,
+        nullable=True,
+    )
+    super_mode = Column(
+        Boolean,
+        nullable=True,
+    )
     created_at = Column(
         DateTime(timezone=True),
         nullable=False,
@@ -302,6 +340,18 @@ class ChatMessageCreate(BaseModel):
         float,
         Field(0.0, description="Cost for the cold start of the message in seconds"),
     ]
+    app_id: Annotated[
+        Optional[str],
+        Field(None, description="Optional application identifier"),
+    ]
+    search_mode: Annotated[
+        Optional[bool],
+        Field(None, description="Optional flag to enable search mode"),
+    ]
+    super_mode: Annotated[
+        Optional[bool],
+        Field(None, description="Optional flag to enable super mode"),
+    ]
 
     async def save_in_session(self, db: AsyncSession) -> "ChatMessage":
         """Save the chat message to the database.

{intentkit-0.6.0.dev4 → intentkit-0.6.0.dev5}/pyproject.toml

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

{intentkit-0.6.0.dev4 → intentkit-0.6.0.dev5}/utils/logging.py

@@ -50,8 +50,8 @@ def setup_logging(env: str, debug: bool = False):
             format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
             handlers=[logging.StreamHandler()],
         )
-        logging.getLogger("openai._base_client").setLevel(logging.INFO)
-        logging.getLogger("httpcore.http11").setLevel(logging.INFO)
+        # logging.getLogger("openai._base_client").setLevel(logging.INFO)
+        # logging.getLogger("httpcore.http11").setLevel(logging.INFO)
         # logging.getLogger("sqlalchemy.engine").setLevel(logging.DEBUG)
     else:
         # For non-local environments, use JSON format