intentkit 0.5.2__py3-none-any.whl → 0.6.0.dev5__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.5.1"
+__version__ = "0.6.0-dev.5"
 __author__ = "hyacinthus"
 __email__ = "hyacinthus@gmail.com"
 

intentkit/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/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/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/utils/logging.py

@@ -43,7 +43,7 @@ def setup_logging(env: str, debug: bool = False):
         debug: Debug mode flag
     """
 
-    if env == "local" or debug:
+    if debug:
         # Set up logging configuration for local/debug
         logging.basicConfig(
             level=logging.DEBUG,

{intentkit-0.5.2.dist-info → intentkit-0.6.0.dev5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: intentkit
-Version: 0.5.2
+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.5.2.dist-info → intentkit-0.6.0.dev5.dist-info}/RECORD

@@ -1,4 +1,4 @@
-intentkit/__init__.py,sha256=OEwQR6p5VXkEANCAdLRcg2bOK1kVrn6SoZeQpP3FUtE,378
+intentkit/__init__.py,sha256=cS9nwSRx1rI1GJ3SyNpnwTmqo_ziSVO0Mpi9ecGnYeI,384
 intentkit/abstracts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 intentkit/abstracts/agent.py,sha256=108gb5W8Q1Sy4G55F2_ZFv2-_CnY76qrBtpIr0Oxxqk,1489
 intentkit/abstracts/api.py,sha256=ZUc24vaQvQVbbjznx7bV0lbbQxdQPfEV8ZxM2R6wZWo,166
@@ -17,8 +17,8 @@ intentkit/core/agent.py,sha256=PTHsFV_EbsfPUcTI0ErkebWurjxVykP9ISbNUsqV44Q,7764
 intentkit/core/api.py,sha256=3GIMJpwduLUSbVPNW6mQVxZncYHH3OlLwdiqFtYtEAE,1208
 intentkit/core/client.py,sha256=rIwtJVVm-7piXtFNDbeykt9vWdNTecgjW0aA3N-lHnM,1495
 intentkit/core/credit.py,sha256=fpRe8BRNGGaumAvJFgce1T2zkclExSFWQdoUVnl6k5g,60789
-intentkit/core/engine.py,sha256=cNzHJLlObDRrOtLUmNPgjaHRaR17JBbMwd2wzv4Eqi4,40344
-intentkit/core/node.py,sha256=36lKDi3Lil_arYMelry8Hb0tu3yG_2VVjc8YdcRZHRY,9010
+intentkit/core/engine.py,sha256=elCKe-NzllJDkE9M8uyzb5w3mozEY7TL4tIslF0HjUA,41148
+intentkit/core/node.py,sha256=ULIyJvVFu9gHuBfSLXe_qV6BMgGOFyAKPx0m8oScC8E,9066
 intentkit/core/prompt.py,sha256=9jxRYUUqSdBj8bdmCUAa-5yTbiQFVInOJsDqbAuUcfo,3512
 intentkit/core/skill.py,sha256=rE37qwDmpnHnIG0MwKxuonVO_lOq47gy-tvXMOz9VZs,3498
 intentkit/models/agent.py,sha256=2HlJLlJShEjhjDR290ndFaaVqv94FXzqdIKPBTuwpn4,57477
@@ -26,7 +26,7 @@ intentkit/models/agent_data.py,sha256=h__b3658ZOclV1Pwpp3UCCu0Nt49CKYfc2JZKG1dKe
 intentkit/models/agent_schema.json,sha256=kacuFT9gPuj16vhZg8Yg1kBJ-3La4uCzBa__uTnQmfY,21368
 intentkit/models/app_setting.py,sha256=WgW-9t0EbiVemRLrVaC6evdfRU5QFSDK0elsnUU5nIo,5008
 intentkit/models/base.py,sha256=o-zRjVrak-f5Jokdvj8BjLm8gcC3yYiYMCTLegwT2lA,185
-intentkit/models/chat.py,sha256=59S7pYGrQGi3a7_4EWBQT0JNkpTcHz0wKI8aX0ZOqkA,16785
+intentkit/models/chat.py,sha256=H4fKBgrseOaFIp83sYYiuyYpYufQAvnoca6V4TVbibE,18013
 intentkit/models/conversation.py,sha256=nrbDIw-3GK5BYi_xkI15FLdx4a6SNrFK8wfAGLCsrqk,9032
 intentkit/models/credit.py,sha256=tHydPpGk8c8p9TLyIb7kvII4vo2WiXvI_i6xzv9PwaQ,42091
 intentkit/models/db.py,sha256=2OpdTjQWUM9FkDP8Ni0mGLeVJ5q9ah3bGlGe9-IzDp0,3999
@@ -353,13 +353,13 @@ intentkit/skills/web_scraper/scrape_and_index.py,sha256=9_U67U8bu36zmHo4C0gyoqLi
 intentkit/utils/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 intentkit/utils/chain.py,sha256=3GBHuAbXxQr_HlOvkbB2kruYSkweucfxI5u-swXzY40,15135
 intentkit/utils/error.py,sha256=JxnzDdKjwZX6Pa-bt_qaibcrWAeF6QAoHsu2uYPmqu4,4483
-intentkit/utils/logging.py,sha256=xqPi9WSCWhyHlDVqsx5DoL5RgRhnMgviDzF4IhNiF8k,2407
+intentkit/utils/logging.py,sha256=bhwZi5vscjBTd9kaNp_L6ijrfv9Sl3lsr4ARaUB4Iec,2389
 intentkit/utils/middleware.py,sha256=bh6rrbm2oT7d0A0ypUwp-I3jsr_98rtYIUlS5foootE,1842
 intentkit/utils/random.py,sha256=DymMxu9g0kuQLgJUqalvgksnIeLdS-v0aRk5nQU0mLI,452
 intentkit/utils/s3.py,sha256=9trQNkKQ5VgxWsewVsV8Y0q_pXzGRvsCYP8xauyUYkg,8549
 intentkit/utils/slack_alert.py,sha256=s7UpRgyzLW7Pbmt8cKzTJgMA9bm4EP-1rQ5KXayHu6E,2264
 intentkit/utils/tx.py,sha256=2yLLGuhvfBEY5n_GJ8wmIWLCzn0FsYKv5kRNzw_sLUI,1454
-intentkit-0.5.2.dist-info/METADATA,sha256=0PYXP_QLsNLIoQlTjj6XEvEbXK52fkIoGgEf5Mli5Tc,7280
-intentkit-0.5.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-intentkit-0.5.2.dist-info/licenses/LICENSE,sha256=Bln6DhK-LtcO4aXy-PBcdZv2f24MlJFm_qn222biJtE,1071
-intentkit-0.5.2.dist-info/RECORD,,
+intentkit-0.6.0.dev5.dist-info/METADATA,sha256=FyJOQ3Cuy_6lmt0SRb_GuaRCrqDc2GGN0DcNcLPQ-B4,7285
+intentkit-0.6.0.dev5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+intentkit-0.6.0.dev5.dist-info/licenses/LICENSE,sha256=Bln6DhK-LtcO4aXy-PBcdZv2f24MlJFm_qn222biJtE,1071
+intentkit-0.6.0.dev5.dist-info/RECORD,,