zrb 1.14.2__tar.gz → 1.14.4__tar.gz

{zrb-1.14.2 → zrb-1.14.4}/PKG-INFO

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

{zrb-1.14.2 → zrb-1.14.4}/pyproject.toml

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

{zrb-1.14.2 → zrb-1.14.4}/src/zrb/task/llm/tool_wrapper.py

@@ -3,7 +3,7 @@ import inspect
 import traceback
 import typing
 from collections.abc import Callable
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Any
 
 from zrb.context.any_context import AnyContext
 from zrb.task.llm.error import ToolExecutionError
@@ -106,7 +106,7 @@ def _create_wrapper(
             del kwargs["_dummy"]
         try:
             if not is_yolo_mode and not ctx.is_web_mode and ctx.is_tty:
-                approval, reason = await _ask_for_approval(ctx, func, *args, **kwargs)
+                approval, reason = await _ask_for_approval(ctx, func, args, kwargs)
                 if not approval:
                     raise ValueError(f"User disapproving: {reason}")
             return await run_async(func(*args, **kwargs))
@@ -123,14 +123,16 @@ def _create_wrapper(
 
 
 async def _ask_for_approval(
-    ctx: AnyContext, func: Callable, *args, **kwargs
+    ctx: AnyContext, func: Callable, args: list[Any], kwargs: dict[Any]
 ) -> 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(')')}"