zrb 1.15.17__py3-none-any.whl → 1.15.18__py3-none-any.whl

zrb/builtin/llm/chat_session.py

@@ -106,7 +106,7 @@ def _show_info(ctx: AnyContext):
                 _show_command("/modes", "Show current modes"),
                 _show_subcommand("<mode1,mode2,..>", "Set current modes"),
                 _show_command("/yolo", "Get current YOLO mode"),
-                _show_subcommand("<true|false>", "Set YOLO mode to true/false"),
+                _show_subcommand("<true|false|list-of-tools>", "Set YOLO mode"),
                 _show_command("/help", "Show this message"),
             ]
         ),
@@ -179,7 +179,7 @@ async def _trigger_ask_and_wait_for_result(
     ctx: AnyContext,
     user_prompt: str,
     modes: str,
-    yolo_mode: bool,
+    yolo_mode: bool | str,
     previous_session_name: str | None = None,
     start_new: bool = False,
 ) -> str | None:
@@ -239,7 +239,7 @@ async def _trigger_ask(
     ctx: AnyContext,
     user_prompt: str,
     modes: str,
-    yolo_mode: bool,
+    yolo_mode: bool | str,
     previous_session_name: str | None = None,
     start_new: bool = False,
 ):

zrb/builtin/llm/llm_ask.py

@@ -24,11 +24,13 @@ from zrb.builtin.llm.tool.web import (
 from zrb.callback.callback import Callback
 from zrb.config.config import CFG
 from zrb.config.llm_config import llm_config
+from zrb.context.any_context import AnyContext
 from zrb.input.bool_input import BoolInput
 from zrb.input.str_input import StrInput
 from zrb.input.text_input import TextInput
 from zrb.task.base_trigger import BaseTrigger
 from zrb.task.llm_task import LLMTask
+from zrb.util.string.conversion import to_boolean
 
 _llm_ask_inputs = [
     StrInput(
@@ -82,7 +84,7 @@ _llm_ask_inputs = [
         allow_positional_parsing=False,
         always_prompt=False,
     ),
-    BoolInput(
+    StrInput(
         "yolo",
         description="YOLO mode (LLM Agent will start in YOLO Mode)",
         prompt="YOLO mode (LLM Agent will start in YOLO Mode)",
@@ -101,6 +103,19 @@ _llm_ask_inputs = [
     ),
 ]
 
+
+def _render_yolo_mode_input(ctx: AnyContext) -> list[str] | bool | None:
+    if ctx.input.yolo.strip() == "":
+        return None
+    elements = ctx.input.yolo.split(",")
+    if len(elements) == 0:
+        try:
+            return to_boolean(elements[0])
+        except Exception:
+            pass
+    return elements
+
+
 llm_ask: LLMTask = llm_group.add_task(
     LLMTask(
         name="llm-ask",
@@ -122,7 +137,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}",
+        yolo_mode=_render_yolo_mode_input,
         retries=0,
     ),
     alias="ask",

zrb/builtin/llm/tool/sub_agent.py

@@ -25,7 +25,7 @@ def create_sub_agent_tool(
     model_settings: "ModelSettings | None" = None,
     tools: "list[ToolOrCallable]" = [],
     toolsets: list["AbstractToolset[Agent]"] = [],
-    is_yolo_mode: bool | None = None,
+    yolo_mode: bool | list[str] | None = None,
     log_indent_level: int = 2,
 ) -> Callable[[AnyContext, str], Coroutine[Any, Any, str]]:
     """
@@ -99,7 +99,7 @@ def create_sub_agent_tool(
             model_settings=resolved_model_settings,
             tools=tools,
             toolsets=toolsets,
-            is_yolo_mode=is_yolo_mode,
+            yolo_mode=yolo_mode,
         )
 
         sub_agent_run = None

zrb/config/llm_config.py

@@ -153,9 +153,9 @@ class LLMConfig:
         model_name = self.default_model_name
         if model_name is None:
             return "openai:gpt-4o"
-        from pydantic_ai.models.openai import OpenAIModel
+        from pydantic_ai.models.openai import OpenAIChatModel
 
-        return OpenAIModel(
+        return OpenAIChatModel(
             model_name=model_name,
             provider=self.default_model_provider,
         )

zrb/task/any_task.py

@@ -1,7 +1,7 @@
 from __future__ import annotations  # Enables forward references
 
 from abc import ABC, abstractmethod
-from typing import TYPE_CHECKING, Any
+from typing import TYPE_CHECKING, Any, Callable
 
 from zrb.env.any_env import AnyEnv
 from zrb.input.any_input import AnyInput
@@ -211,3 +211,8 @@ class AnyTask(ABC):
             session (AnySession): The shared session.
         """
         pass
+
+    @abstractmethod
+    def to_function(self) -> Callable[..., Any]:
+        """Turn a task into a function"""
+        pass

zrb/task/base/context.py

@@ -1,5 +1,5 @@
 import os
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Any
 
 from zrb.context.any_context import AnyContext
 from zrb.context.any_shared_context import AnySharedContext
@@ -29,7 +29,7 @@ def fill_shared_context_inputs(
     shared_ctx: AnySharedContext,
     task: AnyTask,
     str_kwargs: dict[str, str] | None = None,
-    kwargs: dict[str, str] | None = None,
+    kwargs: dict[str, Any] | None = None,
 ):
     """
     Populates the shared context with input values provided via str_kwargs.

zrb/task/base_task.py

@@ -21,6 +21,7 @@ from zrb.task.base.execution import (
 )
 from zrb.task.base.lifecycle import execute_root_tasks, run_and_cleanup, run_task_async
 from zrb.task.base.operators import handle_lshift, handle_rshift
+from zrb.util.string.conversion import to_snake_case
 
 
 class BaseTask(AnyTask):
@@ -219,7 +220,7 @@ class BaseTask(AnyTask):
         self,
         session: AnySession | None = None,
         str_kwargs: dict[str, str] | None = None,
-        kwargs: dict[str, str] | None = None,
+        kwargs: dict[str, Any] | None = None,
     ) -> Any:
         """
         Synchronously runs the task and its dependencies, handling async setup and cleanup.
@@ -286,47 +287,60 @@ class BaseTask(AnyTask):
             # Add definition location to the error
             if hasattr(e, "add_note"):
                 e.add_note(additional_error_note)
-            else:
+            elif hasattr(e, "__notes__"):
                 # fallback: use the __notes__ attribute directly
                 e.__notes__ = getattr(e, "__notes__", []) + [additional_error_note]
             raise e
 
-    def to_fn(self):
-        # NOTE: Non documented feature, untested
-        from inspect import Parameter, Signature
-
+    def to_function(self) -> Callable[..., Any]:
         from zrb.context.shared_context import SharedContext
         from zrb.session.session import Session
 
-        def task_runner_fn(**kwargs):
-            str_kwargs = {k: str(v) for k, v in kwargs.items()}
+        def task_runner_fn(**kwargs) -> Any:
+            task_kwargs = self._get_func_kwargs(kwargs)
             shared_ctx = SharedContext()
             session = Session(shared_ctx=shared_ctx)
-            return self.run(session=session, str_kwargs=str_kwargs)
+            return self.run(session=session, kwargs=task_kwargs)
+
+        task_runner_fn.__doc__ = self._create_fn_docstring()
+        task_runner_fn.__signature__ = self._create_fn_signature()
+        task_runner_fn.__name__ = self.name
+        return task_runner_fn
+
+    def _get_func_kwargs(self, kwargs: dict[str, Any]) -> dict[str, Any]:
+        fn_kwargs = {}
+        for inp in self.inputs:
+            snake_input_name = to_snake_case(inp.name)
+            if snake_input_name in kwargs:
+                fn_kwargs[inp.name] = kwargs[snake_input_name]
+        return fn_kwargs
 
+    def _create_fn_docstring(self) -> str:
+        from zrb.context.shared_context import SharedContext
+
+        stub_shared_ctx = SharedContext()
+        str_input_default_values = {}
+        for inp in self.inputs:
+            str_input_default_values[inp.name] = inp.get_default_str(stub_shared_ctx)
         # Create docstring
         doc = f"{self.description}\n\n"
         if len(self.inputs) > 0:
             doc += "Args:\n"
             for inp in self.inputs:
-                doc += f"    {inp.name}: {inp.description}"
-                if inp.default is not None:
-                    doc += f" (default: {inp.default})"
+                str_input_default = str_input_default_values.get(inp.name, "")
+                doc += (
+                    f"    {inp.name}: {inp.description} (default: {str_input_default})"
+                )
                 doc += "\n"
-        task_runner_fn.__doc__ = doc
+        return doc
 
-        # Create signature
+    def _create_fn_signature(self) -> inspect.Signature:
         params = []
         for inp in self.inputs:
             params.append(
-                Parameter(
-                    name=inp.name,
-                    kind=Parameter.POSITIONAL_OR_KEYWORD,
-                    default=inp.default if inp.default is not None else Parameter.empty,
+                inspect.Parameter(
+                    name=to_snake_case(inp.name),
+                    kind=inspect.Parameter.POSITIONAL_OR_KEYWORD,
                 )
             )
-        sig = Signature(params)
-        task_runner_fn.__signature__ = sig
-        task_runner_fn.__name__ = self.name
-
-        return task_runner_fn
+        return inspect.Signature(params)

zrb/task/llm/agent.py

@@ -30,14 +30,14 @@ def create_agent_instance(
     tools: "list[ToolOrCallable]" = [],
     toolsets: list["AbstractToolset[Agent]"] = [],
     retries: int = 3,
-    is_yolo_mode: bool | None = None,
+    yolo_mode: bool | list[str] | None = None,
 ) -> "Agent":
     """Creates a new Agent instance with configured tools and servers."""
     from pydantic_ai import Agent, Tool
     from pydantic_ai.tools import GenerateToolJsonSchema
 
-    if is_yolo_mode is None:
-        is_yolo_mode = False
+    if yolo_mode is None:
+        yolo_mode = False
     # Normalize tools
     tool_list = []
     for tool_or_callable in tools:
@@ -47,7 +47,7 @@ def create_agent_instance(
             tool = tool_or_callable
             tool_list.append(
                 Tool(
-                    function=wrap_func(tool.function, ctx, is_yolo_mode),
+                    function=wrap_func(tool.function, ctx, yolo_mode),
                     takes_ctx=tool.takes_ctx,
                     max_retries=tool.max_retries,
                     name=tool.name,
@@ -61,7 +61,7 @@ def create_agent_instance(
             )
         else:
             # Turn function into tool
-            tool_list.append(wrap_tool(tool_or_callable, ctx, is_yolo_mode))
+            tool_list.append(wrap_tool(tool_or_callable, ctx, yolo_mode))
     # Return Agent
     return Agent(
         model=model,
@@ -86,7 +86,7 @@ def get_agent(
     toolsets_attr: "list[AbstractToolset[Agent]] | Callable[[AnySharedContext], list[AbstractToolset[Agent]]]",  # noqa
     additional_toolsets: "list[AbstractToolset[Agent]]",
     retries: int = 3,
-    is_yolo_mode: bool | None = None,
+    yolo_mode: bool | list[str] | None = None,
 ) -> "Agent":
     """Retrieves the configured Agent instance or creates one if necessary."""
     from pydantic_ai import Agent
@@ -118,7 +118,7 @@ def get_agent(
         toolsets=tool_sets,
         model_settings=model_settings,
         retries=retries,
-        is_yolo_mode=is_yolo_mode,
+        yolo_mode=yolo_mode,
     )
 
 

zrb/task/llm/config.py

@@ -8,20 +8,27 @@ 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, get_bool_attr
+from zrb.util.attr import get_attr, get_bool_attr, get_str_list_attr
 
 
-def get_is_yolo_mode(
+def get_yolo_mode(
     ctx: AnyContext,
-    is_yolo_mode_attr: BoolAttr | None = None,
+    yolo_mode_attr: BoolAttr | StrAttr | None = None,
     render_yolo_mode: bool = True,
-):
-    return get_bool_attr(
-        ctx,
-        is_yolo_mode_attr,
-        llm_config.default_yolo_mode,
-        auto_render=render_yolo_mode,
-    )
+) -> bool | list[str]:
+    try:
+        return get_bool_attr(
+            ctx,
+            yolo_mode_attr,
+            llm_config.default_yolo_mode,
+            auto_render=render_yolo_mode,
+        )
+    except Exception:
+        return get_str_list_attr(
+            ctx,
+            yolo_mode_attr,
+            auto_render=render_yolo_mode,
+        )
 
 
 def get_model_settings(

zrb/task/llm/tool_wrapper.py

@@ -26,23 +26,27 @@ class ToolExecutionCancelled(ValueError):
     pass
 
 
-def wrap_tool(func: Callable, ctx: AnyContext, is_yolo_mode: bool) -> "Tool":
+def wrap_tool(func: Callable, ctx: AnyContext, yolo_mode: bool | list[str]) -> "Tool":
     """Wraps a tool function to handle exceptions and context propagation."""
     from pydantic_ai import RunContext, Tool
 
     original_sig = inspect.signature(func)
     needs_run_context_for_pydantic = _has_context_parameter(original_sig, RunContext)
-    wrapper = wrap_func(func, ctx, is_yolo_mode)
+    wrapper = wrap_func(func, ctx, yolo_mode)
     return Tool(wrapper, takes_ctx=needs_run_context_for_pydantic)
 
 
-def wrap_func(func: Callable, ctx: AnyContext, is_yolo_mode: bool) -> Callable:
+def wrap_func(func: Callable, ctx: AnyContext, yolo_mode: bool | list[str]) -> Callable:
     original_sig = inspect.signature(func)
     needs_any_context_for_injection = _has_context_parameter(original_sig, AnyContext)
     takes_no_args = len(original_sig.parameters) == 0
     # Pass individual flags to the wrapper creator
     wrapper = _create_wrapper(
-        func, original_sig, ctx, needs_any_context_for_injection, is_yolo_mode
+        func=func,
+        original_sig=original_sig,
+        ctx=ctx,
+        needs_any_context_for_injection=needs_any_context_for_injection,
+        yolo_mode=yolo_mode,
     )
     _adjust_signature(wrapper, original_sig, takes_no_args)
     return wrapper
@@ -82,7 +86,7 @@ def _create_wrapper(
     original_sig: inspect.Signature,
     ctx: AnyContext,
     needs_any_context_for_injection: bool,
-    is_yolo_mode: bool,
+    yolo_mode: bool | list[str],
 ) -> Callable:
     """Creates the core wrapper function."""
 
@@ -110,10 +114,13 @@ def _create_wrapper(
         if "_dummy" in kwargs and "_dummy" not in original_sig.parameters:
             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)
-                if not approval:
-                    raise ToolExecutionCancelled(f"User disapproving: {reason}")
+            if not ctx.is_web_mode and ctx.is_tty:
+                if (
+                    isinstance(yolo_mode, list) and func.__name__ not in yolo_mode
+                ) or not yolo_mode:
+                    approval, reason = await _ask_for_approval(ctx, func, args, kwargs)
+                    if not approval:
+                        raise ToolExecutionCancelled(f"User disapproving: {reason}")
             return await run_async(func(*args, **kwargs))
         except KeyboardInterrupt as e:
             raise e

zrb/task/llm_task.py

@@ -12,9 +12,9 @@ from zrb.task.any_task import AnyTask
 from zrb.task.base_task import BaseTask
 from zrb.task.llm.agent import get_agent, run_agent_iteration
 from zrb.task.llm.config import (
-    get_is_yolo_mode,
     get_model,
     get_model_settings,
+    get_yolo_mode,
 )
 from zrb.task.llm.conversation_history import (
     inject_conversation_history_notes,
@@ -107,7 +107,8 @@ class LLMTask(BaseTask):
         execute_condition: bool | str | Callable[[AnySharedContext], bool] = True,
         retries: int = 2,
         retry_period: float = 0,
-        is_yolo_mode: bool | None = None,
+        yolo_mode: StrListAttr | BoolAttr | None = None,
+        is_yolo_mode: BoolAttr | None = None,
         render_yolo_mode: bool = True,
         readiness_check: list[AnyTask] | AnyTask | None = None,
         readiness_check_delay: float = 0.5,
@@ -184,7 +185,11 @@ class LLMTask(BaseTask):
         )
         self._max_call_iteration = max_call_iteration
         self._conversation_context = conversation_context
-        self._is_yolo_mode = is_yolo_mode
+        self._yolo_mode = yolo_mode
+        if is_yolo_mode is not None:
+            print("[DEPRECATED] use `yolo_mode` instead of `is_yolo_mode`")
+            if self._yolo_mode is None:
+                self._yolo_mode = is_yolo_mode
         self._render_yolo_mode = render_yolo_mode
         self._attachment = attachment
 
@@ -210,6 +215,12 @@ class LLMTask(BaseTask):
     ):
         self._history_summarization_token_threshold = summarization_token_threshold
 
+    def set_modes(self, modes: StrListAttr):
+        self._modes = modes
+
+    def set_yolo_mode(self, yolo_mode: StrListAttr | BoolAttr):
+        self._yolo_mode = yolo_mode
+
     async def _exec_action(self, ctx: AnyContext) -> Any:
         # Get dependent configurations first
         model_settings = get_model_settings(ctx, self._model_settings)
@@ -222,9 +233,9 @@ class LLMTask(BaseTask):
             model_api_key_attr=self._model_api_key,
             render_model_api_key=self._render_model_api_key,
         )
-        is_yolo_mode = get_is_yolo_mode(
+        yolo_mode = get_yolo_mode(
             ctx=ctx,
-            is_yolo_mode_attr=self._is_yolo_mode,
+            yolo_mode_attr=self._yolo_mode,
             render_yolo_mode=self._render_yolo_mode,
         )
         summarization_prompt = get_summarization_system_prompt(
@@ -269,7 +280,7 @@ class LLMTask(BaseTask):
             additional_tools=self._additional_tools,
             toolsets_attr=self._toolsets,
             additional_toolsets=self._additional_toolsets,
-            is_yolo_mode=is_yolo_mode,
+            yolo_mode=yolo_mode,
         )
         # 4. Run the agent iteration and save the results/history
         result = await self._execute_agent(

zrb/util/attr.py

@@ -96,9 +96,9 @@ def get_bool_attr(
         bool: The boolean attribute value.
     """
     val = get_attr(shared_ctx, attr, default, auto_render)
-    if isinstance(val, str):
-        return to_boolean(val)
-    return val
+    if isinstance(val, bool):
+        return val
+    return to_boolean(val)
 
 
 def get_int_attr(
@@ -120,9 +120,9 @@ def get_int_attr(
         int: The integer attribute value.
     """
     val = get_attr(shared_ctx, attr, default, auto_render)
-    if isinstance(val, str):
-        return int(val)
-    return val
+    if isinstance(val, int):
+        return val
+    return int(val)
 
 
 def get_float_attr(
@@ -144,9 +144,9 @@ def get_float_attr(
         float | None: The float attribute value.
     """
     val = get_attr(shared_ctx, attr, default, auto_render)
-    if isinstance(val, str):
-        return float(val)
-    return val
+    if isinstance(val, (int, float)):
+        return val
+    return float(val)
 
 
 def get_attr(

{zrb-1.15.17.dist-info → zrb-1.15.18.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: zrb
-Version: 1.15.17
+Version: 1.15.18
 Summary: Your Automation Powerhouse
 License: AGPL-3.0-or-later
 Keywords: Automation,Task Runner,Code Generator,Monorepo,Low Code

{zrb-1.15.17.dist-info → zrb-1.15.18.dist-info}/RECORD

@@ -9,10 +9,10 @@ zrb/builtin/git_subtree.py,sha256=7BKwOkVTWDrR0DXXQ4iJyHqeR6sV5VYRt8y_rEB0EHg,35
 zrb/builtin/group.py,sha256=t008xLM4_fgbjfZrPoi_fQAnSHIo6MOiQSCHBO4GDYU,2379
 zrb/builtin/http.py,sha256=L6RE73c65wWwG5iHFN-tpOhyh56KsrgVskDd3c3YXtk,4246
 zrb/builtin/jwt.py,sha256=3M5uaQhJZbKQLjTUft1OwPz_JxtmK-xtkjxWjciOQho,2859
-zrb/builtin/llm/chat_session.py,sha256=FnZAhrAHQ8tUgB1ejZby4p_v53c8HwsaMoVvqJWKELw,10191
+zrb/builtin/llm/chat_session.py,sha256=p0giSVYuQMQ5H1hbQzl7cMb49XZqWG0SF7X-ixB5EBU,10203
 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=03d1-2jh7NS0VP3eNl9wCdrUzud98i6zFUXW5NNu3_o,5597
+zrb/builtin/llm/llm_ask.py,sha256=BuZdQKw4oXCdZyFYr-pKKMKGMG3XDQKa9aaCGOoSsHM,6015
 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=vMEiZhhTZ3o2jRBxWcJ62b0M85wd_w4W0X4Hx23NXto,2380
@@ -20,7 +20,7 @@ zrb/builtin/llm/tool/cli.py,sha256=8rugrKaNPEatHjr7nN4OIRLRT2TcF-oylEZGbLI9Brs,1
 zrb/builtin/llm/tool/code.py,sha256=fr9FbmtfwizQTyTztvuvwAb9MD_auRZhPZfoJVBlKT4,8777
 zrb/builtin/llm/tool/file.py,sha256=eXFGGFxxpdpWGVw0svyQNQc03I5M7wotSsA_HjkXw7c,23670
 zrb/builtin/llm/tool/rag.py,sha256=Ab8_ZljnG_zfkwxPezImvorshuz3Fi4CmSzNOtU1a-g,9770
-zrb/builtin/llm/tool/sub_agent.py,sha256=dPqdFCXxJ-3hZnPjheZr-bPeKNbfSBSDKcZZR5vkWFM,5075
+zrb/builtin/llm/tool/sub_agent.py,sha256=qJTJ2GSH-2Cma2QyHEJm8l_VuDHMHwhAWGls217YA6A,5078
 zrb/builtin/llm/tool/web.py,sha256=Hc9ikgBWZMgLB2O5lX-P9k8jvNSZeyUufqHXa_n_v4I,7653
 zrb/builtin/md5.py,sha256=690RV2LbW7wQeTFxY-lmmqTSVEEZv3XZbjEUW1Q3XpE,1480
 zrb/builtin/project/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -225,7 +225,7 @@ zrb/config/default_prompt/repo_extractor_system_prompt.md,sha256=EGZ-zj78RlMEg2j
 zrb/config/default_prompt/repo_summarizer_system_prompt.md,sha256=RNy37Wg7ibXj3DlsFKaYvgMpMS-lyXlM1LZlc59_4ic,2009
 zrb/config/default_prompt/summarization_prompt.md,sha256=vbxzJGeH7uCdgnfbVt3GYOjY-BygmeRfJ6tnzXmFdoY,1698
 zrb/config/default_prompt/system_prompt.md,sha256=gEb6N-cFg6VvOV-7ZffNwVt39DavAGesMqn9u0epbRc,2282
-zrb/config/llm_config.py,sha256=xt-Xf8ZuNoUT_GKCSFz5yy0BhbeHzxP-jrezB06WeiY,8857
+zrb/config/llm_config.py,sha256=-XmSCkFk858NlKHSbAWo4e2bMzE9-mMriIoR5_RaXtE,8865
 zrb/config/llm_context/config.py,sha256=TPQX_kU772r0AHmVFeo1WGLDAidacT-qDyuMWxY_avg,5878
 zrb/config/llm_context/config_parser.py,sha256=h95FbOjvVobhrsfGtG_BY3hxS-OLzQj-9F5vGZuehkY,1473
 zrb/config/llm_rate_limitter.py,sha256=_iQRv3d6kUPeRvmUYZX_iwCE7iDSEK1oKa4bQ9GROho,5261
@@ -334,20 +334,20 @@ zrb/session_state_logger/any_session_state_logger.py,sha256=T_-aJv63HwoTeiDOmXKH
 zrb/session_state_logger/file_session_state_logger.py,sha256=NMyj_g2ImQc6ZRM9f35EpA-CM1UO-ZgoDnPkN1DSi9U,3915
 zrb/session_state_logger/session_state_logger_factory.py,sha256=_oT1UIqm25nMwKhsUvXehnCdDvNqxCyuN8GLdv2D_U4,188
 zrb/task/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-zrb/task/any_task.py,sha256=SAPX1uuq4c0mxgGEQ6lyioR4eRmFovhH-Wr2xCtrHnA,6078
+zrb/task/any_task.py,sha256=JZOEYaSZ9XL2dQvNhrstcxF9D_jf95Bz6d0JcVzE8kE,6213
 zrb/task/base/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-zrb/task/base/context.py,sha256=uj0fHPtbjiRWE4gRB8DKNJhLmEN1ZWH6FCjItlE45q4,3918
+zrb/task/base/context.py,sha256=ptz9-Am1s4yq6EHZUpnSrB0Ps3k7BgyakVbE-ZpiBzc,3923
 zrb/task/base/execution.py,sha256=scDLfNYBe8Bc8Ct1LCIKmFtjpPxm7FjqZ2bJXIQAzv8,11042
 zrb/task/base/lifecycle.py,sha256=c2v977pUm7S4EqrTMcUJKhnYOWugACVwU3qORDKiLXQ,7639
 zrb/task/base/monitoring.py,sha256=UAOEcPiYNtZR4FFxzWCosuOEFE_P3c4GT5vAhQmohqI,5663
 zrb/task/base/operators.py,sha256=uAMFqpZJsPnCrojgOl1FUDXTS15mtOa_IqiAXltyYRU,1576
-zrb/task/base_task.py,sha256=qPtEIP3TEX7YbDadQtsUXLJ4_7Khzk_JxXDh_mRrluU,12602
+zrb/task/base_task.py,sha256=upwuqAfwNDXTYM-uRDJhgZqqqARI03T6ksUbFHHLEH0,13321
 zrb/task/base_trigger.py,sha256=WSGcmBcGAZw8EzUXfmCjqJQkz8GEmi1RzogpF6A1V4s,6902
 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=-DVECvW3nLQNmppkD9DStG6L_VUaUUAJb2iMmd-JB90,8990
-zrb/task/llm/config.py,sha256=n1SPmwab09K2i1sL_OCwrEOWHI0Owx_hvWelg3Dreus,3781
+zrb/task/llm/agent.py,sha256=uI0X_injDrxq-IhYuZsAMdoq9AXy-gnipozGuZ72QIs,8990
+zrb/task/llm/config.py,sha256=UrRdjgCpTllJaH6LIptHOj69h7upnYd0q9iKHhszErk,4007
 zrb/task/llm/conversation_history.py,sha256=oMdKUV2__mBZ4znnA-prl-gfyoleKC8Nj5KNpmLQJ4o,6764
 zrb/task/llm/conversation_history_model.py,sha256=kk-7niTl29Rm2EUIhTHzPXgZ5tp4IThMnIB3dS-1OdU,3062
 zrb/task/llm/default_workflow/coding.md,sha256=2uythvPsnBpYfIhiIH1cCinQXX0i0yUqsL474Zpemw0,2484
@@ -358,9 +358,9 @@ zrb/task/llm/history_summarization.py,sha256=UIT8bpdT3hy1xn559waDLFWZlNtIqdIpIvR
 zrb/task/llm/history_summarization_tool.py,sha256=Wazi4WMr3k1WJ1v7QgjAPbuY1JdBpHUsTWGt3DSTsLc,1706
 zrb/task/llm/print_node.py,sha256=si619OvCUZJYTUSUpoOqVAgLSu1BGw-dlBq02MSD7FE,8096
 zrb/task/llm/prompt.py,sha256=FGXWYHecWtrNNkPnjg-uhnkqp7fYt8V91-AjFM_5fpA,11550
-zrb/task/llm/tool_wrapper.py,sha256=IavXwW9C0ojI3ivGDY8j7XbMJE_NTJeI86TVbHNRwJ0,9684
+zrb/task/llm/tool_wrapper.py,sha256=AsRzVAZtuAMsutLHGetpsPPncu6pcd4LrwxYRZxsp7c,9933
 zrb/task/llm/typing.py,sha256=c8VAuPBw_4A3DxfYdydkgedaP-LU61W9_wj3m3CAX1E,58
-zrb/task/llm_task.py,sha256=ctEp-5R7ZnjA3V6pSkthKN_o7xjASre8MgMcQVHtsj8,14504
+zrb/task/llm_task.py,sha256=OxJ9QpqjEyeOI1_zqzNZHtQlRHi0ANOvL9FYaWLzO3Y,14913
 zrb/task/make_task.py,sha256=PD3b_aYazthS8LHeJsLAhwKDEgdurQZpymJDKeN60u0,2265
 zrb/task/rsync_task.py,sha256=WfqNSaicJgYWpunNU34eYxXDqHDHOftuDHyWJKjqwg0,6365
 zrb/task/scaffolder.py,sha256=rME18w1HJUHXgi9eTYXx_T2G4JdqDYzBoNOkdOOo5-o,6806
@@ -370,7 +370,7 @@ zrb/task/tcp_check.py,sha256=P_QgGqwd5dXDaud3oQRxe_WuxyxG4s7CTY2wDk9Qcu0,2511
 zrb/task_status/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/task_status/task_status.py,sha256=blZ8dxg9g_8MuViq-t7yJRLoE7yGUf5srgHf-PCsXNc,3069
 zrb/util/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-zrb/util/attr.py,sha256=5GlYSmVAzbcSFjNDXiqqHqNMR6NWjJ6bUHZXdE35mj8,5359
+zrb/util/attr.py,sha256=PdN5Jt0fGLQkTpfiziVMYXq22f56nwnaMHxLO063qCI,5369
 zrb/util/callable.py,sha256=b6OFXbCXp2twow3wh2E_h5hNHLs2pXaLfGQz4iVyiQc,771
 zrb/util/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/util/cli/markdown.py,sha256=Uhuw8XR-jAG9AG3oNK8VHJpYOdU40Q_8yVN74uu0RJ8,384
@@ -409,7 +409,7 @@ zrb/util/todo_model.py,sha256=hhzAX-uFl5rsg7iVX1ULlJOfBtblwQ_ieNUxBWfc-Os,1670
 zrb/util/truncate.py,sha256=eSzmjBpc1Qod3lM3M73snNbDOcARHukW_tq36dWdPvc,921
 zrb/xcom/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/xcom/xcom.py,sha256=o79rxR9wphnShrcIushA0Qt71d_p3ZTxjNf7x9hJB78,1571
-zrb-1.15.17.dist-info/METADATA,sha256=BU4mNVepzuJpdZZhgOGUbXmXqyUIoqK8IJgCAxF92_Q,9892
-zrb-1.15.17.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-zrb-1.15.17.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
-zrb-1.15.17.dist-info/RECORD,,
+zrb-1.15.18.dist-info/METADATA,sha256=cGOrmv5M8MXtOP4PyxO1Wfxln5G8EVsL6R4gAaN5ANo,9892
+zrb-1.15.18.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+zrb-1.15.18.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
+zrb-1.15.18.dist-info/RECORD,,