zrb 1.14.0__py3-none-any.whl → 1.14.2__py3-none-any.whl

zrb/builtin/llm/llm_ask.py

@@ -122,6 +122,7 @@ llm_ask: LLMTask = llm_group.add_task(
             None if ctx.input.modes.strip() == "" else ctx.input.modes.split(",")
         ),
         message="{ctx.input.message}",
+        is_yolo_mode="{ctx.input.yolo}",
         retries=0,
     ),
     alias="ask",

zrb/task/llm/config.py

@@ -8,7 +8,7 @@ from zrb.attr.type import BoolAttr, StrAttr, fstring
 from zrb.config.llm_config import LLMConfig, llm_config
 from zrb.context.any_context import AnyContext
 from zrb.context.any_shared_context import AnySharedContext
-from zrb.util.attr import get_attr
+from zrb.util.attr import get_attr, get_bool_attr
 
 
 def get_is_yolo_mode(
@@ -16,10 +16,12 @@ def get_is_yolo_mode(
     is_yolo_mode_attr: BoolAttr | None = None,
     render_yolo_mode: bool = True,
 ):
-    yolo_mode = get_attr(ctx, is_yolo_mode_attr, None, auto_render=render_yolo_mode)
-    if yolo_mode is None:
-        return llm_config.default_yolo_mode
-    return yolo_mode
+    return get_bool_attr(
+        ctx,
+        is_yolo_mode_attr,
+        llm_config.default_yolo_mode,
+        auto_render=render_yolo_mode,
+    )
 
 
 def get_model_settings(

zrb/task/llm/tool_wrapper.py

@@ -140,18 +140,21 @@ async def _ask_for_approval(
             f"✅ >> Allow to run tool: {func_call_str} (Yes | No, <reason>)", plain=True
         )
         user_input = await _read_line()
-        approval_str, reason = [
-            val.strip() for val in user_input.split(",", maxsplit=2)
-        ]
-        if reason == "":
-            ctx.print(
-                stylize_error(f"You must specify rejection reason for {func_call_str}"),
-                plain=True,
-            )
-            continue
+        user_responses = [val.strip() for val in user_input.split(",", maxsplit=2)]
+        while len(user_responses) < 2:
+            user_responses.append("")
+        approval_str, reason = user_responses
         try:
-            approval = to_boolean(approval_str)
-            return approval, reason
+            approved = to_boolean(approval_str)
+            if not approved and reason == "":
+                ctx.print(
+                    stylize_error(
+                        f"You must specify rejection reason (i.e., No, <why>) for {func_call_str}"  # noqa
+                    ),
+                    plain=True,
+                )
+                continue
+            return approved, reason
         except Exception:
             ctx.print(
                 stylize_error(

{zrb-1.14.0.dist-info → zrb-1.14.2.dist-info}/METADATA

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

{zrb-1.14.0.dist-info → zrb-1.14.2.dist-info}/RECORD

@@ -12,7 +12,7 @@ zrb/builtin/jwt.py,sha256=3M5uaQhJZbKQLjTUft1OwPz_JxtmK-xtkjxWjciOQho,2859
 zrb/builtin/llm/chat_session.py,sha256=syMOQzAGcRmxs7ctrBkfvAZ7-IhxnywuxuyeoBMteEs,10464
 zrb/builtin/llm/history.py,sha256=LDOrL0p7r_AHLa5L8Dp7bHNsOALugmJd7OguXRWGnm4,3087
 zrb/builtin/llm/input.py,sha256=Nw-26uTWp2QhUgKJcP_IMHmtk-b542CCSQ_vCOjhvhM,877
-zrb/builtin/llm/llm_ask.py,sha256=ZwuECO1BQaPEzIYvlA0ptQiHVBCKA7_a8DPHj2aXfLM,5460
+zrb/builtin/llm/llm_ask.py,sha256=CNFLKgmLutP3ymAhl-KdVbC-Ahnm1tBEPVinO9o2IrU,5501
 zrb/builtin/llm/previous-session.js,sha256=xMKZvJoAbrwiyHS0OoPrWuaKxWYLoyR5sguePIoCjTY,816
 zrb/builtin/llm/tool/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/builtin/llm/tool/api.py,sha256=OhmfLc2TwWKQYIMweGelqb5s4JF4nB-YynbSO4yb_Jk,2342
@@ -347,7 +347,7 @@ zrb/task/cmd_task.py,sha256=myM8WZm6NrUD-Wv0Vb5sTOrutrAVZLt5LVsSBKwX6SM,10860
 zrb/task/http_check.py,sha256=Gf5rOB2Se2EdizuN9rp65HpGmfZkGc-clIAlHmPVehs,2565
 zrb/task/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/task/llm/agent.py,sha256=bdrsmozPxSrUQXhv4FUs_BhIeSppQcVn2dOm40rySW4,7659
-zrb/task/llm/config.py,sha256=GDL9RzH7LSzIaQ_YIrxG-4SqsUadI8i224oSu0pvcLo,3801
+zrb/task/llm/config.py,sha256=UwTsC3YqQPaNfl75T8wMRZWhBjGQce-xaselRqs37Lc,3788
 zrb/task/llm/conversation_history.py,sha256=B_PDWYL_q66s0xwWBzMSomqPN6u3gkXlIeXBD5A0Apg,4416
 zrb/task/llm/conversation_history_model.py,sha256=DJ0KDBB0BriQuE5ugC_q0aSHhjNIBcfjUk1f0S_3I9U,9245
 zrb/task/llm/default_workflow/coding.md,sha256=2uythvPsnBpYfIhiIH1cCinQXX0i0yUqsL474Zpemw0,2484
@@ -357,7 +357,7 @@ zrb/task/llm/error.py,sha256=QR-nIohS6pBpC_16cWR-fw7Mevo1sNYAiXMBsh_CJDE,4157
 zrb/task/llm/history_summarization.py,sha256=_0RmzIeJdJA3KvtdTdKnd2Ga7_7x8C1J2PM0oSn-IYw,8000
 zrb/task/llm/print_node.py,sha256=mwdqsO2IVf5rDz-jdH9HXz6MFGCWrZ4Pv2xbUBtoNgc,4179
 zrb/task/llm/prompt.py,sha256=pHYuo4cLzuH9VYk2-PAyxng_l1ItXVZl0llZDQcbnWA,9748
-zrb/task/llm/tool_wrapper.py,sha256=guXdGqzsoiDT6iLB0FEKm7umB5iHoDWpTPmFDO5qU30,7945
+zrb/task/llm/tool_wrapper.py,sha256=FtoYWn0tz9LEzaxYS3_75uSWOspe32GC0gnFn2zlLUI,8153
 zrb/task/llm/typing.py,sha256=c8VAuPBw_4A3DxfYdydkgedaP-LU61W9_wj3m3CAX1E,58
 zrb/task/llm_task.py,sha256=88yc0SbU2ocWK6VBGoqBy-Lo49ebBTI_KKjsLxDp3sM,13945
 zrb/task/make_task.py,sha256=PD3b_aYazthS8LHeJsLAhwKDEgdurQZpymJDKeN60u0,2265
@@ -406,7 +406,7 @@ zrb/util/todo.py,sha256=r9_KYF2-hLKMNjsp6AFK9zivykMrywd-kJ4bCwfdafI,19323
 zrb/util/todo_model.py,sha256=hhzAX-uFl5rsg7iVX1ULlJOfBtblwQ_ieNUxBWfc-Os,1670
 zrb/xcom/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/xcom/xcom.py,sha256=o79rxR9wphnShrcIushA0Qt71d_p3ZTxjNf7x9hJB78,1571
-zrb-1.14.0.dist-info/METADATA,sha256=oEBL7QK3hoSF2u41M1gVfXl8oB48OPU5ioYTAkQi8GY,9777
-zrb-1.14.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-zrb-1.14.0.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
-zrb-1.14.0.dist-info/RECORD,,
+zrb-1.14.2.dist-info/METADATA,sha256=Xc8xnQ69oEi7Xn3LKMnmkz17Pig9QmYSDhIvabsFIB8,9777
+zrb-1.14.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+zrb-1.14.2.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
+zrb-1.14.2.dist-info/RECORD,,