zrb 1.13.3__py3-none-any.whl → 1.14.1__py3-none-any.whl

zrb/builtin/llm/chat_session.py

@@ -10,7 +10,8 @@ import sys
 
 from zrb.config.llm_config import llm_config
 from zrb.context.any_context import AnyContext
-from zrb.util.cli.style import stylize_bold_yellow, stylize_faint
+from zrb.util.cli.style import stylize_blue, stylize_bold_yellow, stylize_faint
+from zrb.util.string.conversion import to_boolean
 
 
 async def read_user_prompt(ctx: AnyContext) -> str:
@@ -26,6 +27,7 @@ async def read_user_prompt(ctx: AnyContext) -> str:
     reader = await _setup_input_reader(is_tty)
     multiline_mode = False
     current_modes = ctx.input.modes
+    current_yolo_mode = ctx.input.yolo
     user_inputs = []
     while True:
         await asyncio.sleep(0.01)
@@ -40,7 +42,7 @@ async def read_user_prompt(ctx: AnyContext) -> str:
             user_prompt = "\n".join(user_inputs)
             user_inputs = []
             result = await _trigger_ask_and_wait_for_result(
-                ctx, user_prompt, current_modes
+                ctx, user_prompt, current_modes, current_yolo_mode
             )
             if result is not None:
                 final_result = result
@@ -53,7 +55,7 @@ async def read_user_prompt(ctx: AnyContext) -> str:
             user_prompt = "\n".join(user_inputs)
             user_inputs = []
             result = await _trigger_ask_and_wait_for_result(
-                ctx, user_prompt, current_modes
+                ctx, user_prompt, current_modes, current_yolo_mode
             )
             if result is not None:
                 final_result = result
@@ -64,6 +66,13 @@ async def read_user_prompt(ctx: AnyContext) -> str:
             ctx.print(f"Current mode: {current_modes}", plain=True)
             ctx.print("", plain=True)
             continue
+        elif user_input.strip().lower().startswith("/yolo"):
+            yolo_mode_parts = user_input.split(" ", maxsplit=2)
+            if len(yolo_mode_parts) > 1:
+                current_yolo_mode = to_boolean(yolo_mode_parts[1])
+            ctx.print(f"Current_yolo mode: {current_yolo_mode}", plain=True)
+            ctx.print("", plain=True)
+            continue
         elif user_input.strip().lower() in ("/help", "/info"):
             _show_info(ctx)
             continue
@@ -74,7 +83,7 @@ async def read_user_prompt(ctx: AnyContext) -> str:
             user_prompt = "\n".join(user_inputs)
             user_inputs = []
             result = await _trigger_ask_and_wait_for_result(
-                ctx, user_prompt, current_modes
+                ctx, user_prompt, current_modes, current_yolo_mode
             )
             if result is not None:
                 final_result = result
@@ -90,24 +99,32 @@ def _show_info(ctx: AnyContext):
     ctx.print(
         "\n".join(
             [
-                _format_info_line("/bye", "Quit from chat session"),
-                _format_info_line("/multi", "Start multiline input"),
-                _format_info_line("/end", "End multiline input"),
-                _format_info_line("/modes", "Show current modes"),
-                _format_info_line("/modes <mode1,mode2,..>", "Set current modes"),
-                _format_info_line("/help", "Show this message"),
+                _show_command("/bye", "Quit from chat session"),
+                _show_command("/multi", "Start multiline input"),
+                _show_command("/end", "End multiline input"),
+                _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_command("/help", "Show this message"),
             ]
         ),
         plain=True,
     )
 
 
-def _format_info_line(command: str, description: str) -> str:
+def _show_command(command: str, description: str) -> str:
     styled_command = stylize_bold_yellow(command.ljust(25))
     styled_description = stylize_faint(description)
     return f"  {styled_command} {styled_description}"
 
 
+def _show_subcommand(command: str, description: str) -> str:
+    styled_command = stylize_blue(f"    {command}".ljust(25))
+    styled_description = stylize_faint(description)
+    return f"  {styled_command} {styled_description}"
+
+
 async def _handle_initial_message(ctx: AnyContext) -> str:
     """Processes the initial message from the command line."""
     if not ctx.input.message or ctx.input.message.strip() == "":
@@ -119,6 +136,7 @@ async def _handle_initial_message(ctx: AnyContext) -> str:
         ctx,
         user_prompt=ctx.input.message,
         modes=ctx.input.modes,
+        yolo_mode=ctx.input.yolo,
         previous_session_name=ctx.input.previous_session,
         start_new=ctx.input.start_new,
     )
@@ -157,6 +175,7 @@ async def _trigger_ask_and_wait_for_result(
     ctx: AnyContext,
     user_prompt: str,
     modes: str,
+    yolo_mode: bool,
     previous_session_name: str | None = None,
     start_new: bool = False,
 ) -> str | None:
@@ -174,7 +193,9 @@ async def _trigger_ask_and_wait_for_result(
     """
     if user_prompt.strip() == "":
         return None
-    await _trigger_ask(ctx, user_prompt, modes, previous_session_name, start_new)
+    await _trigger_ask(
+        ctx, user_prompt, modes, yolo_mode, previous_session_name, start_new
+    )
     result = await _wait_ask_result(ctx)
     md_result = _render_markdown(result) if result is not None else ""
     ctx.print("\n🤖 >>", plain=True)
@@ -220,6 +241,7 @@ def get_llm_ask_input_mapping(callback_ctx: AnyContext):
         "previous-session": data.get("previous_session_name"),
         "message": data.get("message"),
         "modes": data.get("modes"),
+        "yolo": data.get("yolo"),
     }
 
 
@@ -227,6 +249,7 @@ async def _trigger_ask(
     ctx: AnyContext,
     user_prompt: str,
     modes: str,
+    yolo_mode: bool,
     previous_session_name: str | None = None,
     start_new: bool = False,
 ):
@@ -247,6 +270,7 @@ async def _trigger_ask(
             "start_new": start_new,
             "message": user_prompt,
             "modes": modes,
+            "yolo": yolo_mode,
         }
     )
 

zrb/builtin/llm/llm_ask.py

@@ -23,6 +23,7 @@ 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.input.bool_input import BoolInput
 from zrb.input.str_input import StrInput
 from zrb.input.text_input import TextInput
@@ -75,12 +76,20 @@ _llm_ask_inputs = [
     ),
     BoolInput(
         "start-new",
-        description="Start new conversation (LLM will forget everything)",
-        prompt="Start new conversation (LLM will forget everything)",
+        description="Start new session (LLM Agent will forget past conversation)",
+        prompt="Start new session (LLM Agent will forget past conversation)",
         default=False,
         allow_positional_parsing=False,
         always_prompt=False,
     ),
+    BoolInput(
+        "yolo",
+        description="YOLO mode (LLM Agent will start in YOLO Mode)",
+        prompt="YOLO mode (LLM Agent will start in YOLO Mode)",
+        default=lambda ctx: llm_config.default_yolo_mode,
+        allow_positional_parsing=False,
+        always_prompt=False,
+    ),
     TextInput("message", description="User message", prompt="Your message"),
     PreviousSessionInput(
         "previous-session",
@@ -154,8 +163,20 @@ if CFG.LLM_ALLOW_ACCESS_LOCAL_FILE:
 if CFG.LLM_ALLOW_ACCESS_SHELL:
     llm_ask.append_tool(run_shell_command)
 
-if CFG.LLM_ALLOW_ACCESS_INTERNET:
-    llm_ask.append_tool(open_web_page, search_wikipedia, search_arxiv)
-    if CFG.SERPAPI_KEY != "":
-        llm_ask.append_tool(create_search_internet_tool(CFG.SERPAPI_KEY))
-    llm_ask.append_tool(get_current_location, get_current_weather)
+if CFG.LLM_ALLOW_OPEN_WEB_PAGE:
+    llm_ask.append_tool(open_web_page)
+
+if CFG.LLM_ALLOW_SEARCH_WIKIPEDIA:
+    llm_ask.append_tool(search_wikipedia)
+
+if CFG.LLM_ALLOW_SEARCH_ARXIV:
+    llm_ask.append_tool(search_arxiv)
+
+if CFG.LLM_ALLOW_GET_CURRENT_LOCATION:
+    llm_ask.append_tool(get_current_location)
+
+if CFG.LLM_ALLOW_GET_CURRENT_WEATHER:
+    llm_ask.append_tool(get_current_weather)
+
+if CFG.SERPAPI_KEY != "" and CFG.LLM_ALLOW_SEARCH_INTERNET:
+    llm_ask.append_tool(create_search_internet_tool(CFG.SERPAPI_KEY))

zrb/builtin/llm/tool/file.py

@@ -2,7 +2,8 @@ import fnmatch
 import json
 import os
 import re
-from typing import Any, Dict, List, Optional, Tuple
+import sys
+from typing import Any, Optional
 
 from zrb.builtin.llm.tool.sub_agent import create_sub_agent_tool
 from zrb.config.config import CFG
@@ -10,7 +11,17 @@ from zrb.config.llm_rate_limitter import llm_rate_limitter
 from zrb.context.any_context import AnyContext
 from zrb.util.file import read_file, read_file_with_line_numbers, write_file
 
-_EXTRACT_INFO_FROM_FILE_SYSTEM_PROMPT = CFG.LLM_FILE_EXTRACTOR_SYSTEM_PROMPT
+if sys.version_info >= (3, 12):
+    from typing import TypedDict
+else:
+    from typing_extensions import TypedDict
+
+
+class FileToWrite(TypedDict):
+    """Represents a file to be written, with a 'path' and 'content'."""
+
+    path: str
+    content: str
 
 
 DEFAULT_EXCLUDED_PATTERNS = [
@@ -481,14 +492,14 @@ async def analyze_file(
     return await _analyze_file(ctx, clipped_payload)
 
 
-def read_many_files(paths: List[str]) -> str:
+def read_many_files(paths: list[str]) -> str:
     """
     Reads and returns the full content of multiple files at once.
 
     This tool is highly efficient for gathering context from several files simultaneously. Use it when you need to understand how different files in a project relate to each other, or when you need to inspect a set of related configuration or source code files.
 
     Args:
-        paths (List[str]): A list of paths to the files you want to read. It is crucial to provide accurate paths. Use the `list_files` tool first if you are unsure about the exact file locations.
+        paths (list[str]): A list of paths to the files you want to read. It is crucial to provide accurate paths. Use the `list_files` tool first if you are unsure about the exact file locations.
 
     Returns:
         str: A JSON object where keys are the file paths and values are their corresponding contents, prefixed with line numbers. If a file cannot be read, its value will be an error message.
@@ -507,7 +518,7 @@ def read_many_files(paths: List[str]) -> str:
     return json.dumps({"results": results})
 
 
-def write_many_files(files: List[Tuple[str, str]]) -> str:
+def write_many_files(files: list[FileToWrite]) -> str:
     """
     Writes content to multiple files in a single, atomic operation.
 
@@ -516,7 +527,7 @@ def write_many_files(files: List[Tuple[str, str]]) -> str:
     Each file's content is completely replaced. If a file does not exist, it will be created. If it exists, its current content will be entirely overwritten. Therefore, you must provide the full, intended content for each file.
 
     Args:
-        files (List[Tuple[str, str]]): A list of tuples, where each tuple contains the file path and the complete content to be written to that file.
+        files: A list of file objects, where each object is a dictionary containing a 'path' and the complete 'content'.
 
     Returns:
         str: A JSON object summarizing the operation, listing successfully written files and any files that failed, along with corresponding error messages.
@@ -524,8 +535,13 @@ def write_many_files(files: List[Tuple[str, str]]) -> str:
     """
     success = []
     errors = {}
-    for path, content in files:
+    # 4. Access the data using dictionary key-lookup syntax.
+    for file in files:
         try:
+            # Use file['path'] and file['content'] instead of file.path
+            path = file["path"]
+            content = file["content"]
+
             abs_path = os.path.abspath(os.path.expanduser(path))
             directory = os.path.dirname(abs_path)
             if directory and not os.path.exists(directory):
@@ -535,6 +551,3 @@ def write_many_files(files: List[Tuple[str, str]]) -> str:
         except Exception as e:
             errors[path] = f"Error writing file: {e}"
     return json.dumps({"success": success, "errors": errors})
-
-
-apply_diff = replace_in_file

zrb/builtin/llm/tool/sub_agent.py

@@ -27,6 +27,7 @@ def create_sub_agent_tool(
     model_settings: "ModelSettings | None" = None,
     tools: list[ToolOrCallable] = [],
     toolsets: list["AbstractToolset[Agent]"] = [],
+    is_yolo_mode: bool | None = None,
 ) -> Callable[[AnyContext, str], Coroutine[Any, Any, str]]:
     """
     Creates a "tool that is another AI agent," capable of handling complex, multi-step sub-tasks.
@@ -86,6 +87,7 @@ def create_sub_agent_tool(
             model_settings=resolved_model_settings,
             tools=tools,
             toolsets=toolsets,
+            is_yolo_mode=is_yolo_mode,
         )
 
         sub_agent_run = None

zrb/config/config.py

@@ -350,8 +350,28 @@ class Config:
         return to_boolean(os.getenv("ZRB_LLM_ACCESS_SHELL", "1"))
 
     @property
-    def LLM_ALLOW_ACCESS_INTERNET(self) -> bool:
-        return to_boolean(os.getenv("ZRB_LLM_ACCESS_INTERNET", "1"))
+    def LLM_ALLOW_OPEN_WEB_PAGE(self) -> bool:
+        return to_boolean(os.getenv("ZRB_LLM_ALLOW_OPEN_WEB_PAGE", "1"))
+
+    @property
+    def LLM_ALLOW_SEARCH_INTERNET(self) -> bool:
+        return to_boolean(os.getenv("ZRB_LLM_ALLOW_SEARCH_INTERNET", "1"))
+
+    @property
+    def LLM_ALLOW_SEARCH_ARXIV(self) -> bool:
+        return to_boolean(os.getenv("ZRB_LLM_ALLOW_SEARCH_ARXIV", "1"))
+
+    @property
+    def LLM_ALLOW_SEARCH_WIKIPEDIA(self) -> bool:
+        return to_boolean(os.getenv("ZRB_LLM_ALLOW_SEARCH_WIKIPEDIA", "1"))
+
+    @property
+    def LLM_ALLOW_GET_CURRENT_LOCATION(self) -> bool:
+        return to_boolean(os.getenv("ZRB_LLM_ALLOW_GET_CURRENT_LOCATION", "1"))
+
+    @property
+    def LLM_ALLOW_GET_CURRENT_WEATHER(self) -> bool:
+        return to_boolean(os.getenv("ZRB_LLM_ALLOW_GET_CURRENT_WEATHER", "1"))
 
     @property
     def RAG_EMBEDDING_API_KEY(self) -> str:

zrb/config/llm_config.py

@@ -2,8 +2,6 @@ import os
 from typing import TYPE_CHECKING, Any, Callable
 
 from zrb.config.config import CFG
-from zrb.config.llm_context.config import llm_context_config
-from zrb.util.llm.prompt import make_prompt_section
 
 if TYPE_CHECKING:
     from pydantic_ai.models import Model
@@ -28,6 +26,7 @@ class LLMConfig:
         default_model: "Model | None" = None,
         default_model_settings: "ModelSettings | None" = None,
         default_model_provider: "Provider | None" = None,
+        default_yolo_mode: bool | None = None,
     ):
         self.__internal_default_prompt: dict[str, str] = {}
         self._default_model_name = default_model_name
@@ -46,6 +45,7 @@ class LLMConfig:
         self._default_model = default_model
         self._default_model_settings = default_model_settings
         self._default_model_provider = default_model_provider
+        self._default_yolo_mode = default_yolo_mode
 
     def _get_internal_default_prompt(self, name: str) -> str:
         if name not in self.__internal_default_prompt:
@@ -174,6 +174,12 @@ class LLMConfig:
             lambda: 1000,
         )
 
+    @property
+    def default_yolo_mode(self) -> bool:
+        return self._get_property(
+            self._default_yolo_mode, CFG.LLM_YOLO_MODE, lambda: False
+        )
+
     def set_default_persona(self, persona: str):
         self._default_persona = persona
 
@@ -230,5 +236,8 @@ class LLMConfig:
     def set_default_model_settings(self, model_settings: "ModelSettings"):
         self._default_model_settings = model_settings
 
+    def set_default_yolo_mode(self, yolo_mode: bool):
+        self._default_yolo_mode = yolo_mode
+
 
 llm_config = LLMConfig()

zrb/context/shared_context.py

@@ -27,6 +27,7 @@ class SharedContext(AnySharedContext):
         env: dict[str, str] = {},
         xcom: dict[str, Xcom] = {},
         logging_level: int | None = None,
+        is_web_mode: bool = False,
     ):
         self.__logging_level = logging_level
         self._input = DotDict(input)
@@ -35,6 +36,7 @@ class SharedContext(AnySharedContext):
         self._xcom = DotDict(xcom)
         self._session: AnySession | None = None
         self._log = []
+        self._is_web_mode = is_web_mode
 
     def __repr__(self):
         class_name = self.__class__.__name__
@@ -46,7 +48,7 @@ class SharedContext(AnySharedContext):
 
     @property
     def is_web_mode(self) -> bool:
-        return self.env.get("_ZRB_IS_WEB_MODE", "0") == "1"
+        return self._is_web_mode
 
     @property
     def is_tty(self) -> bool:

zrb/runner/web_route/task_session_api_route.py

@@ -1,5 +1,4 @@
 import asyncio
-import os
 from datetime import datetime, timedelta
 from typing import TYPE_CHECKING, Any
 
@@ -57,9 +56,7 @@ def serve_task_session_api(
                 return JSONResponse(content={"detail": "Forbidden"}, status_code=403)
             session_name = residual_args[0] if residual_args else None
             if not session_name:
-                shared_ctx = SharedContext(
-                    env={**dict(os.environ), "_ZRB_IS_WEB_MODE": "1"}
-                )
+                shared_ctx = SharedContext(is_web_mode=True)
                 session = Session(shared_ctx=shared_ctx, root_group=root_group)
                 coro = asyncio.create_task(task.async_run(session, str_kwargs=inputs))
                 coroutines.append(coro)

zrb/task/llm/agent.py

@@ -30,11 +30,14 @@ def create_agent_instance(
     tools: list[ToolOrCallable] = [],
     toolsets: list["AbstractToolset[Agent]"] = [],
     retries: int = 3,
+    is_yolo_mode: bool | 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
     # Normalize tools
     tool_list = []
     for tool_or_callable in tools:
@@ -44,7 +47,7 @@ def create_agent_instance(
             tool = tool_or_callable
             tool_list.append(
                 Tool(
-                    function=wrap_func(tool.function),
+                    function=wrap_func(tool.function, ctx, is_yolo_mode),
                     takes_ctx=tool.takes_ctx,
                     max_retries=tool.max_retries,
                     name=tool.name,
@@ -58,7 +61,7 @@ def create_agent_instance(
             )
         else:
             # Turn function into tool
-            tool_list.append(wrap_tool(tool_or_callable, ctx))
+            tool_list.append(wrap_tool(tool_or_callable, ctx, is_yolo_mode))
     # Return Agent
     return Agent(
         model=model,
@@ -83,6 +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,
 ) -> "Agent":
     """Retrieves the configured Agent instance or creates one if necessary."""
     from pydantic_ai import Agent
@@ -114,6 +118,7 @@ def get_agent(
         toolsets=tool_sets,
         model_settings=model_settings,
         retries=retries,
+        is_yolo_mode=is_yolo_mode,
     )
 
 

zrb/task/llm/config.py

@@ -1,16 +1,27 @@
-from typing import TYPE_CHECKING, Any, Callable
+from typing import TYPE_CHECKING, Callable
 
 if TYPE_CHECKING:
     from pydantic_ai.models import Model
     from pydantic_ai.settings import ModelSettings
 
-from zrb.attr.type import StrAttr, fstring
+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
 
 
+def get_is_yolo_mode(
+    ctx: AnyContext,
+    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
+
+
 def get_model_settings(
     ctx: AnyContext,
     model_settings_attr: (

zrb/task/llm/tool_wrapper.py

@@ -5,10 +5,15 @@ import typing
 from collections.abc import Callable
 from typing import TYPE_CHECKING
 
-from zrb.config.config import CFG
 from zrb.context.any_context import AnyContext
 from zrb.task.llm.error import ToolExecutionError
 from zrb.util.callable import get_callable_name
+from zrb.util.cli.style import (
+    stylize_blue,
+    stylize_error,
+    stylize_green,
+    stylize_yellow,
+)
 from zrb.util.run import run_async
 from zrb.util.string.conversion import to_boolean
 
@@ -16,22 +21,24 @@ if TYPE_CHECKING:
     from pydantic_ai import Tool
 
 
-def wrap_tool(func: Callable, ctx: AnyContext) -> "Tool":
+def wrap_tool(func: Callable, ctx: AnyContext, is_yolo_mode: bool) -> "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)
+    wrapper = wrap_func(func, ctx, is_yolo_mode)
     return Tool(wrapper, takes_ctx=needs_run_context_for_pydantic)
 
 
-def wrap_func(func: Callable, ctx: AnyContext) -> Callable:
+def wrap_func(func: Callable, ctx: AnyContext, is_yolo_mode: bool) -> 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)
+    wrapper = _create_wrapper(
+        func, original_sig, ctx, needs_any_context_for_injection, is_yolo_mode
+    )
     _adjust_signature(wrapper, original_sig, takes_no_args)
     return wrapper
 
@@ -68,8 +75,9 @@ def _is_annotated_with_context(param_annotation, context_type):
 def _create_wrapper(
     func: Callable,
     original_sig: inspect.Signature,
-    ctx: AnyContext,  # Accept ctx
+    ctx: AnyContext,
     needs_any_context_for_injection: bool,
+    is_yolo_mode: bool,
 ) -> Callable:
     """Creates the core wrapper function."""
 
@@ -97,19 +105,10 @@ def _create_wrapper(
         if "_dummy" in kwargs and "_dummy" not in original_sig.parameters:
             del kwargs["_dummy"]
         try:
-            if not CFG.LLM_YOLO_MODE and not ctx.is_web_mode and ctx.is_tty:
-                func_name = get_callable_name(func)
-                ctx.print(f"✅ >> Allow to run tool: {func_name} (Y/n)", plain=True)
-                user_confirmation_str = await _read_line()
-                try:
-                    user_confirmation = to_boolean(user_confirmation_str)
-                except Exception:
-                    user_confirmation = False
-                if not user_confirmation:
-                    ctx.print(f"❌ >> Rejecting {func_name} call. Why?", plain=True)
-                    reason = await _read_line()
-                    ctx.print("", plain=True)
-                    raise ValueError(f"User disapproval: {reason}")
+            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 ValueError(f"User disapproving: {reason}")
             return await run_async(func(*args, **kwargs))
         except Exception as e:
             error_model = ToolExecutionError(
@@ -123,6 +122,55 @@ def _create_wrapper(
     return wrapper
 
 
+async def _ask_for_approval(
+    ctx: AnyContext, func: Callable, *args, **kwargs
+) -> 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()
+    ]
+    func_param_str = ",".join(normalized_args + normalized_kwargs)
+    func_call_str = (
+        f"{stylize_blue(func_name + '(')}{func_param_str}{stylize_blue(')')}"
+    )
+    while True:
+        ctx.print(
+            f"✅ >> Allow to run tool: {func_call_str} (Yes | No, <reason>)", plain=True
+        )
+        user_input = await _read_line()
+        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:
+            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(
+                    f"Invalid approval value for {func_call_str}: {approval_str}"
+                ),
+                plain=True,
+            )
+            continue
+
+
+def _truncate_arg(arg: str, length: int = 19) -> str:
+    if len(arg) > length:
+        return f"{arg[:length-4]} ..."
+    return arg
+
+
 async def _read_line():
     from prompt_toolkit import PromptSession
 

zrb/task/llm_task.py

@@ -12,6 +12,7 @@ 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,
 )
@@ -104,6 +105,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,
+        render_yolo_mode: bool = True,
         readiness_check: list[AnyTask] | AnyTask | None = None,
         readiness_check_delay: float = 0.5,
         readiness_check_period: float = 5,
@@ -179,6 +182,8 @@ class LLMTask(BaseTask):
         )
         self._max_call_iteration = max_call_iteration
         self._conversation_context = conversation_context
+        self._is_yolo_mode = is_yolo_mode
+        self._render_yolo_mode = render_yolo_mode
 
     def add_tool(self, *tool: ToolOrCallable):
         self.append_tool(*tool)
@@ -214,6 +219,11 @@ 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(
+            ctx=ctx,
+            is_yolo_mode_attr=self._is_yolo_mode,
+            render_yolo_mode=self._render_yolo_mode,
+        )
         summarization_prompt = get_summarization_system_prompt(
             ctx=ctx,
             summarization_prompt_attr=self._summarization_prompt,
@@ -254,6 +264,7 @@ class LLMTask(BaseTask):
             additional_tools=self._additional_tools,
             toolsets_attr=self._toolsets,
             additional_toolsets=self._additional_toolsets,
+            is_yolo_mode=is_yolo_mode,
         )
         # 4. Run the agent iteration and save the results/history
         result = await self._execute_agent(

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

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

{zrb-1.13.3.dist-info → zrb-1.14.1.dist-info}/RECORD

@@ -9,18 +9,18 @@ zrb/builtin/git_subtree.py,sha256=7BKwOkVTWDrR0DXXQ4iJyHqeR6sV5VYRt8y_rEB0EHg,35
 zrb/builtin/group.py,sha256=t008xLM4_fgbjfZrPoi_fQAnSHIo6MOiQSCHBO4GDYU,2379
 zrb/builtin/http.py,sha256=sLqEczuSxGYXWzyJR6frGOHkPTviu4BeyroUr3-ZuAI,4322
 zrb/builtin/jwt.py,sha256=3M5uaQhJZbKQLjTUft1OwPz_JxtmK-xtkjxWjciOQho,2859
-zrb/builtin/llm/chat_session.py,sha256=u8bW67uKCq22hVv4ZkOsKIZxBeOdKtJh4Bjyy552RM4,9424
+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=18XAxyPWF7daE0TZkRkRt8opmqLUjhpM3oMVdOP-qWY,4857
+zrb/builtin/llm/llm_ask.py,sha256=ZwuECO1BQaPEzIYvlA0ptQiHVBCKA7_a8DPHj2aXfLM,5460
 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
 zrb/builtin/llm/tool/cli.py,sha256=dUWZrW2X5J_lONuzR__6-SbewSdi28E3RRuksjd4mWo,1234
 zrb/builtin/llm/tool/code.py,sha256=GRP_IZAkeL6RIlUm407BQRF992ES57pdzPaQdC5UsJU,8218
-zrb/builtin/llm/tool/file.py,sha256=OLg8RfWZOypTxTtK7YzyoZp6O2ITszGTaqos-7c0V-c,22288
+zrb/builtin/llm/tool/file.py,sha256=hKqT-r4S9FTMv3CmNWuIL5y6q13hp9Ggo0vwUGg3HxI,22585
 zrb/builtin/llm/tool/rag.py,sha256=wB74JV7bxs0ec77b_09Z2lPjoR1WzPUvZbuXOdb9Q9g,9675
-zrb/builtin/llm/tool/sub_agent.py,sha256=9Su64FpNTVeE6O2qgNzo-eo4pcmv8qi_sd_QWLQBXYw,4870
+zrb/builtin/llm/tool/sub_agent.py,sha256=yYqTIQWfG5mLL1nItc4jox-QK8vPOb_ZGvQ6sH0BcKw,4947
 zrb/builtin/llm/tool/web.py,sha256=gQlUsmYCJOFJtNjwpjK-xk13LMvrMSpSaFHXUTnIayQ,7090
 zrb/builtin/md5.py,sha256=690RV2LbW7wQeTFxY-lmmqTSVEEZv3XZbjEUW1Q3XpE,1480
 zrb/builtin/project/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -217,7 +217,7 @@ zrb/callback/callback.py,sha256=PFhCqzfxdk6IAthmXcZ13DokT62xtBzJr_ciLw6I8Zg,4030
 zrb/cmd/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/cmd/cmd_result.py,sha256=L8bQJzWCpcYexIxHBNsXj2pT3BtLmWex0iJSMkvimOA,597
 zrb/cmd/cmd_val.py,sha256=7Doowyg6BK3ISSGBLt-PmlhzaEkBjWWm51cED6fAUOQ,1014
-zrb/config/config.py,sha256=d_F-hdPLADjeVRHtnpOxtOkfUBu5huSLclyD53uxO4U,12306
+zrb/config/config.py,sha256=9yl4LtPpROvjYMqdDy3hXjlD-LA8zQXFrcsO5XWXmSw,13017
 zrb/config/default_prompt/file_extractor_system_prompt.md,sha256=tmeZMPzF9MGExsZZw7M2PZN6V0oFVRp1nIjiqUPvQ9M,1013
 zrb/config/default_prompt/interactive_system_prompt.md,sha256=ZFPeDEV2vlcksHiVG2o-TCehmqkFolDjtH0_Fzo1gGI,3566
 zrb/config/default_prompt/persona.md,sha256=WU4JKp-p7qJePDA6NZ_CYdBggo2B3PEq8IEnNVblIHU,41
@@ -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=fpG5B416OK3oE41bWPrh1M6pdH5SSadCPte_NJ_79z0,858
 zrb/config/default_prompt/summarization_prompt.md,sha256=hRXH5E78TugSze_Hgp-KTbIhCeyrMcJg-pSXvXH3C9E,1629
 zrb/config/default_prompt/system_prompt.md,sha256=Jkne5n9HJcBCgfeENwxvqH-kbDO2CaiUzqR4VoWMRHY,3054
-zrb/config/llm_config.py,sha256=bNLxorctwtVW1F9hA-hEYpDBe7FLSZHC25Nx8NlR4-M,8597
+zrb/config/llm_config.py,sha256=WfbxP5wXolP-BzGUqidnbacMd55uuSdU2yO7EtL5F_8,8853
 zrb/config/llm_context/config.py,sha256=zeqSVOKK5yyApvqTbcO3ayGxtyoag22qlWWaXp1nINs,4950
 zrb/config/llm_context/config_parser.py,sha256=h95FbOjvVobhrsfGtG_BY3hxS-OLzQj-9F5vGZuehkY,1473
 zrb/config/llm_rate_limitter.py,sha256=P4vR7qxwiGwjlKx2kHcfdIxwGbJB98vdN-UQEH-Q2WU,4894
@@ -237,7 +237,7 @@ zrb/context/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/context/any_context.py,sha256=2hgVKbbDwmwrEl1h1L1FaTUjuUYaDd_b7YRGkaorW6Q,6362
 zrb/context/any_shared_context.py,sha256=wJawL1jGgApcKPRcpw3js7W4-MhJRA3GMbR5zTsJmt0,1929
 zrb/context/context.py,sha256=ErGhXJgjgNaAqi6iPMejWxFZ3YvWnysC6mHEU-wodKk,6884
-zrb/context/shared_context.py,sha256=Jaa7AYCeCksOiEAwOnY3xD6Y2Yy2wJAkpehAkbKQ-Wc,3076
+zrb/context/shared_context.py,sha256=c87VujBGKbkEC6R8Bby0-Gp0h2UvX8ZsH_3S0ElEKSs,3124
 zrb/dot_dict/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/dot_dict/dot_dict.py,sha256=ubw_x8I7AOJ59xxtFVJ00VGmq_IYdZP3mUhNlO4nEK0,556
 zrb/env/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -316,7 +316,7 @@ zrb/runner/web_route/static/resources/session/event.js,sha256=X5OlSHefK0SDB9VkFC
 zrb/runner/web_route/static/resources/session/past-session.js,sha256=RwGJYKSp75K8NZ-iZP58XppWgdzkiKFaiC5wgcMLxDo,5470
 zrb/runner/web_route/static/static_route.py,sha256=QPs5XW4O_8CuzG0Wy4sHh5wRcLbU63CLDI4YNqkUxHA,1555
 zrb/runner/web_route/task_input_api_route.py,sha256=6JIehRjXPhzclq9qGMYkztaKB0TzWsBBbim0m47-YmA,1767
-zrb/runner/web_route/task_session_api_route.py,sha256=N4kg7uNfxiiuF-YEpk6khuorkyv_H5aDm_l3pwxNozo,6262
+zrb/runner/web_route/task_session_api_route.py,sha256=U9fPOh_nmAzsmRnS5xe713KFUU15n0IkbbzU_Eg8YGI,6181
 zrb/runner/web_schema/session.py,sha256=NwbuS2Sv-CXO52nU-EZv8OMlD4vgCQWNeLC_dT0FK7I,92
 zrb/runner/web_schema/token.py,sha256=Y7XCPS4WzrxslTDtHeLcPTTUpmWhPOkRcl4b99zrC7c,185
 zrb/runner/web_schema/user.py,sha256=Kp10amg4i-f8Y-4czogv1YN7rwy0HdbePFiuovYu1ts,1018
@@ -346,8 +346,8 @@ 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=aZvtcL5HmZQvD3c79R9sDIOMawO0rUMcRiq2wZ1FNas,7457
-zrb/task/llm/config.py,sha256=TlyH925_fboIlK2Ixf34tynmenqs9s9rfsnPs4jff78,3490
+zrb/task/llm/agent.py,sha256=bdrsmozPxSrUQXhv4FUs_BhIeSppQcVn2dOm40rySW4,7659
+zrb/task/llm/config.py,sha256=GDL9RzH7LSzIaQ_YIrxG-4SqsUadI8i224oSu0pvcLo,3801
 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,9 +357,9 @@ 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=jfKMAtTzm--HnF6TppOrbkDVsuTOIFRpowQqgwqd-7s,6756
+zrb/task/llm/tool_wrapper.py,sha256=-BfyZzPPPR8mSMBCHqltds6FTDprutUonipdzzlyCbk,8153
 zrb/task/llm/typing.py,sha256=c8VAuPBw_4A3DxfYdydkgedaP-LU61W9_wj3m3CAX1E,58
-zrb/task/llm_task.py,sha256=jVuVeN2ylcPkycUiaTpavkkd1tBLbvcVsjMNN5FMHnk,13536
+zrb/task/llm_task.py,sha256=88yc0SbU2ocWK6VBGoqBy-Lo49ebBTI_KKjsLxDp3sM,13945
 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
@@ -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.13.3.dist-info/METADATA,sha256=WhAGmkemQMrm0fxn88gq6bSWUv4KLgruOZTrC_cSmnM,9777
-zrb-1.13.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-zrb-1.13.3.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
-zrb-1.13.3.dist-info/RECORD,,
+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,,