zrb 1.14.1__py3-none-any.whl → 1.14.3__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

@@ -127,10 +127,12 @@ async def _ask_for_approval(
 ) -> tuple[bool, str]:
     func_name = get_callable_name(func)
     normalized_args = [stylize_green(_truncate_arg(arg)) for arg in args]
-    normalized_kwargs = [
-        f"{stylize_yellow(key)}={stylize_green(_truncate_arg(val))}"
-        for key, val in kwargs.items()
-    ]
+    normalized_kwargs = []
+    for key, val in kwargs.items():
+        truncated_val = _truncate_arg(f"{val}")
+        normalized_kwargs.append(
+            f"{stylize_yellow(key)}={stylize_green(truncated_val)}"
+        )
     func_param_str = ",".join(normalized_args + normalized_kwargs)
     func_call_str = (
         f"{stylize_blue(func_name + '(')}{func_param_str}{stylize_blue(')')}"
@@ -149,8 +151,8 @@ async def _ask_for_approval(
             if not approved and reason == "":
                 ctx.print(
                     stylize_error(
-                        f"You must specify rejection reason (i.e., No, <why>) for {func_call_str}"
-                    ),  # noqa
+                        f"You must specify rejection reason (i.e., No, <why>) for {func_call_str}"  # noqa
+                    ),
                     plain=True,
                 )
                 continue

{zrb-1.14.1.dist-info → zrb-1.14.3.dist-info}/METADATA

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

{zrb-1.14.1.dist-info → zrb-1.14.3.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=-BfyZzPPPR8mSMBCHqltds6FTDprutUonipdzzlyCbk,8153
+zrb/task/llm/tool_wrapper.py,sha256=7fBZU4uDmmwaeeEnR6fsbiELxXuYY8iHliW1HZ_xIv0,8236
 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.1.dist-info/METADATA,sha256=7QS2abopkUJEBHHFUG3kwHNQy0L6cr44wHyzaTJcK14,9777
-zrb-1.14.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-zrb-1.14.1.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
-zrb-1.14.1.dist-info/RECORD,,
+zrb-1.14.3.dist-info/METADATA,sha256=hgYWuKB29Cc1DcbaKxejldxDZwPWvnSCqQgugfa8OHs,9777
+zrb-1.14.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+zrb-1.14.3.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
+zrb-1.14.3.dist-info/RECORD,,