zrb 1.14.0__tar.gz → 1.14.1__tar.gz

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

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

{zrb-1.14.0 → zrb-1.14.1}/pyproject.toml

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

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