zrb 1.14.1__tar.gz → 1.14.3__tar.gz

{zrb-1.14.1 → zrb-1.14.3}/PKG-INFO

@@ -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 → zrb-1.14.3}/pyproject.toml

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

{zrb-1.14.1 → zrb-1.14.3}/src/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-1.14.1 → zrb-1.14.3}/src/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-1.14.1 → zrb-1.14.3}/src/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