zrb 1.8.12__py3-none-any.whl → 1.8.13__py3-none-any.whl

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/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/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
 

{zrb-1.8.12.dist-info → zrb-1.8.13.dist-info}/METADATA

@@ -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.dist-info → zrb-1.8.13.dist-info}/RECORD

@@ -9,7 +9,7 @@ zrb/builtin/git_subtree.py,sha256=7BKwOkVTWDrR0DXXQ4iJyHqeR6sV5VYRt8y_rEB0EHg,35
 zrb/builtin/group.py,sha256=t008xLM4_fgbjfZrPoi_fQAnSHIo6MOiQSCHBO4GDYU,2379
 zrb/builtin/http.py,sha256=sLqEczuSxGYXWzyJR6frGOHkPTviu4BeyroUr3-ZuAI,4322
 zrb/builtin/jwt.py,sha256=3M5uaQhJZbKQLjTUft1OwPz_JxtmK-xtkjxWjciOQho,2859
-zrb/builtin/llm/chat_session.py,sha256=ot2ss6yA4qIINg0nl3KJYnLag8H0eB9ggAgRGEUkZdE,6639
+zrb/builtin/llm/chat_session.py,sha256=Nfn_HXAKjKeIKWQt4nfS9k0VWmrg0JhexlJKVw58WLQ,6841
 zrb/builtin/llm/history.py,sha256=cnkOyO43uiMQ9cEvmqk-pPoCk1zCAH_fwAqSgBtsjzY,3079
 zrb/builtin/llm/input.py,sha256=Nw-26uTWp2QhUgKJcP_IMHmtk-b542CCSQ_vCOjhvhM,877
 zrb/builtin/llm/llm_ask.py,sha256=QUV29gOAFKiMfJlAKbY9YfGPoxYv-4RPv6p7cWogK4U,4438
@@ -217,7 +217,7 @@ zrb/callback/callback.py,sha256=PFhCqzfxdk6IAthmXcZ13DokT62xtBzJr_ciLw6I8Zg,4030
 zrb/cmd/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/cmd/cmd_result.py,sha256=L8bQJzWCpcYexIxHBNsXj2pT3BtLmWex0iJSMkvimOA,597
 zrb/cmd/cmd_val.py,sha256=7Doowyg6BK3ISSGBLt-PmlhzaEkBjWWm51cED6fAUOQ,1014
-zrb/config.py,sha256=qFtVVme30fMyi5x_mgvvULczNbORqK8ZEN8agXokXO4,10222
+zrb/config.py,sha256=L4nrdO8uK1MNOJiTgywkjlYn5kX1dOsq48b5mPZ2004,10364
 zrb/content_transformer/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/content_transformer/any_content_transformer.py,sha256=v8ZUbcix1GGeDQwB6OKX_1TjpY__ksxWVeqibwa_iZA,850
 zrb/content_transformer/content_transformer.py,sha256=STl77wW-I69QaGzCXjvkppngYFLufow8ybPLSyAvlHs,2404
@@ -246,7 +246,7 @@ zrb/input/option_input.py,sha256=TQB82ko5odgzkULEizBZi0e9TIHEbIgvdP0AR3RhA74,213
 zrb/input/password_input.py,sha256=szBojWxSP9QJecgsgA87OIYwQrY2AQ3USIKdDZY6snU,1465
 zrb/input/str_input.py,sha256=NevZHX9rf1g8eMatPyy-kUX3DglrVAQpzvVpKAzf7bA,81
 zrb/input/text_input.py,sha256=6T3MngWdUs0u0ZVs5Dl11w5KS7nN1RkgrIR_zKumzPM,3695
-zrb/llm_config.py,sha256=pXWHp-7WcAF3s5Gec8IBqoC_l3aRvU3aGt2Yr96SxGo,16607
+zrb/llm_config.py,sha256=PaOsd9i5rwq_Ry68T6tpgQ2sWdjq-SeCuw7XMaHxw7Q,17577
 zrb/llm_rate_limitter.py,sha256=uM9zmSgV10fQq1dlaDGLDrv72uLj6ldBxMoGjO2Az14,4429
 zrb/runner/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/runner/cli.py,sha256=AbLTNqFy5FuyGQOWOjHZGaBC8e2yuE_Dx1sBdnisR18,6984
@@ -390,7 +390,7 @@ zrb/util/string/name.py,sha256=SXEfxJ1-tDOzHqmSV8kvepRVyMqs2XdV_vyoh_9XUu0,1584
 zrb/util/todo.py,sha256=VGISej2KQZERpornK-8X7bysp4JydMrMUTnG8B0-liI,20708
 zrb/xcom/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/xcom/xcom.py,sha256=o79rxR9wphnShrcIushA0Qt71d_p3ZTxjNf7x9hJB78,1571
-zrb-1.8.12.dist-info/METADATA,sha256=W42D3Cq0i9W2m7C2i0c0eY0JFXViJP3FNhdAt2gsrNw,9274
-zrb-1.8.12.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-zrb-1.8.12.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
-zrb-1.8.12.dist-info/RECORD,,
+zrb-1.8.13.dist-info/METADATA,sha256=oqzBj3hnl1XlXUSSib6obfP_75DGHvZnJCIjZIx7tJ4,9274
+zrb-1.8.13.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+zrb-1.8.13.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
+zrb-1.8.13.dist-info/RECORD,,