zrb 1.8.12__tar.gz → 1.8.13__tar.gz

{zrb-1.8.12 → zrb-1.8.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: zrb
-Version: 1.8.12
+Version: 1.8.13
 Summary: Your Automation Powerhouse
 Home-page: https://github.com/state-alchemists/zrb
 License: AGPL-3.0-or-later

{zrb-1.8.12 → zrb-1.8.13}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "zrb"
-version = "1.8.12"
+version = "1.8.13"
 description = "Your Automation Powerhouse"
 authors = ["Go Frendi Gunawan <gofrendiasgard@gmail.com>"]
 license = "AGPL-3.0-or-later"

{zrb-1.8.12 → zrb-1.8.13}/src/zrb/builtin/llm/chat_session.py

@@ -8,6 +8,7 @@ conversation flow via XCom.
 import asyncio
 
 from zrb.context.any_context import AnyContext
+from zrb.llm_config import llm_config
 from zrb.util.cli.style import stylize_bold_yellow, stylize_faint
 
 
@@ -113,11 +114,14 @@ def get_llm_ask_input_mapping(callback_ctx: AnyContext):
         A dictionary containing the input mapping for the LLM ask task.
     """
     data = callback_ctx.xcom.ask_trigger.pop()
+    system_prompt = callback_ctx.input.system_prompt
+    if system_prompt is None or system_prompt.strip() == "":
+        system_prompt = llm_config.default_interactive_system_prompt
     return {
         "model": callback_ctx.input.model,
         "base-url": callback_ctx.input.base_url,
         "api-key": callback_ctx.input.api_key,
-        "system-prompt": callback_ctx.input.system_prompt,
+        "system-prompt": system_prompt,
         "start-new": data.get("start_new"),
         "previous-session": data.get("previous_session_name"),
         "message": data.get("message"),

{zrb-1.8.12 → zrb-1.8.13}/src/zrb/config.py

@@ -225,6 +225,10 @@ class Config:
     def LLM_SYSTEM_PROMPT(self) -> str | None:
         return os.getenv("ZRB_LLM_SYSTEM_PROMPT", None)
 
+    @property
+    def LLM_INTERACTIVE_SYSTEM_PROMPT(self) -> str | None:
+        return os.getenv("ZRB_LLM_INTERACTIVE_SYSTEM_PROMPT", None)
+
     @property
     def LLM_PERSONA(self) -> str | None:
         return os.getenv("ZRB_LLM_PERSONA", None)

{zrb-1.8.12 → zrb-1.8.13}/src/zrb/llm_config.py

@@ -17,7 +17,7 @@ DEFAULT_PERSONA = (
     "straight to the point."
 ).strip()
 
-DEFAULT_SYSTEM_PROMPT = (
+DEFAULT_INTERACTIVE_SYSTEM_PROMPT = (
     "You have access to tools and two forms of memory:\n"
     "1.  A structured summary of the immediate task (including a payload) AND "
     "the raw text of the last few turns.\n"
@@ -50,6 +50,13 @@ DEFAULT_SYSTEM_PROMPT = (
     "guess."
 ).strip()
 
+DEFAULT_SYSTEM_PROMPT = (
+    "You are a helpful and precise expert assistant. Your goal is to "
+    "follow instructions carefully to provide an accurate and efficient answer. "
+    "Fulfill the user's request directly. Use your tools if necessary. "
+    "Get straight to the point."
+).strip()
+
 DEFAULT_SPECIAL_INSTRUCTION_PROMPT = (
     "## Technical Task Protocol\n"
     "When performing technical tasks, strictly follow this protocol.\n\n"
@@ -202,6 +209,7 @@ class LLMConfig:
         default_api_key: str | None = None,
         default_persona: str | None = None,
         default_system_prompt: str | None = None,
+        default_interactive_system_prompt: str | None = None,
         default_special_instruction_prompt: str | None = None,
         default_summarization_prompt: str | None = None,
         default_context_enrichment_prompt: str | None = None,
@@ -218,6 +226,7 @@ class LLMConfig:
         self._default_model_api_key = default_api_key
         self._default_persona = default_persona
         self._default_system_prompt = default_system_prompt
+        self._default_interactive_system_prompt = default_interactive_system_prompt
         self._default_special_instruction_prompt = default_special_instruction_prompt
         self._default_summarization_prompt = default_summarization_prompt
         self._default_context_enrichment_prompt = default_context_enrichment_prompt
@@ -282,6 +291,14 @@ class LLMConfig:
             return CFG.LLM_SYSTEM_PROMPT
         return DEFAULT_SYSTEM_PROMPT
 
+    @property
+    def default_interactive_system_prompt(self) -> str:
+        if self._default_interactive_system_prompt is not None:
+            return self._default_interactive_system_prompt
+        if CFG.LLM_INTERACTIVE_SYSTEM_PROMPT is not None:
+            return CFG.LLM_INTERACTIVE_SYSTEM_PROMPT
+        return DEFAULT_INTERACTIVE_SYSTEM_PROMPT
+
     @property
     def default_persona(self) -> str:
         if self._default_persona is not None:
@@ -358,6 +375,9 @@ class LLMConfig:
     def set_default_system_prompt(self, system_prompt: str):
         self._default_system_prompt = system_prompt
 
+    def set_default_interactive_system_prompt(self, interactive_system_prompt: str):
+        self._default_interactive_system_prompt = interactive_system_prompt
+
     def set_default_special_instruction_prompt(self, special_instruction_prompt: str):
         self._default_special_instruction_prompt = special_instruction_prompt