zrb 1.14.8__tar.gz → 1.15.1__tar.gz

{zrb-1.14.8 → zrb-1.15.1}/PKG-INFO

@@ -1,8 +1,7 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: zrb
-Version: 1.14.8
+Version: 1.15.1
 Summary: Your Automation Powerhouse
-Home-page: https://github.com/state-alchemists/zrb
 License: AGPL-3.0-or-later
 Keywords: Automation,Task Runner,Code Generator,Monorepo,Low Code
 Author: Go Frendi Gunawan
@@ -13,6 +12,7 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
 Provides-Extra: all
 Provides-Extra: playwright
 Provides-Extra: rag
@@ -22,11 +22,12 @@ Requires-Dist: chromadb (>=0.6.3,<0.7.0) ; extra == "rag" or extra == "all"
 Requires-Dist: fastapi[standard] (>=0.116.1,<0.117.0)
 Requires-Dist: isort (>=6.0.1,<6.1.0)
 Requires-Dist: libcst (>=1.7.0,<2.0.0)
-Requires-Dist: openai (>=1.92.0,<=1.99.1)
+Requires-Dist: markdownify (>=1.2.0,<2.0.0)
+Requires-Dist: openai (>=1.99.7)
 Requires-Dist: pdfplumber (>=0.11.6,<0.12.0)
 Requires-Dist: playwright (>=1.53.0,<2.0.0) ; extra == "playwright" or extra == "all"
 Requires-Dist: psutil (>=7.0.0,<8.0.0)
-Requires-Dist: pydantic-ai-slim[anthropic,bedrock,cli,cohere,google,groq,huggingface,mcp,mistral,openai,vertexai] (>=0.6.2,<0.7.0)
+Requires-Dist: pydantic-ai-slim[anthropic,bedrock,cli,cohere,google,groq,huggingface,mcp,mistral,openai,vertexai] (>=0.7.0,<0.8.0)
 Requires-Dist: pyjwt (>=2.10.1,<3.0.0)
 Requires-Dist: python-dotenv (>=1.1.1,<2.0.0)
 Requires-Dist: python-jose[cryptography] (>=3.4.0,<4.0.0)
@@ -34,6 +35,7 @@ Requires-Dist: requests (>=2.32.4,<3.0.0)
 Requires-Dist: tiktoken (>=0.8.0,<0.9.0)
 Requires-Dist: ulid-py (>=1.1.0,<2.0.0)
 Project-URL: Documentation, https://github.com/state-alchemists/zrb
+Project-URL: Homepage, https://github.com/state-alchemists/zrb
 Project-URL: Repository, https://github.com/state-alchemists/zrb
 Description-Content-Type: text/markdown
 

{zrb-1.14.8 → zrb-1.15.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "zrb"
-version = "1.14.8"
+version = "1.15.1"
 description = "Your Automation Powerhouse"
 authors = ["Go Frendi Gunawan <gofrendiasgard@gmail.com>"]
 license = "AGPL-3.0-or-later"
@@ -54,14 +54,15 @@ pyjwt = "^2.10.1"
 # NOTE: Look at https://github.com/pydantic/pydantic-ai/blob/main/pydantic_ai_slim/pyproject.toml
 # - We want to include all models, as well as mcp capability.
 # - We also need prompt-toolkit and rich which are included in pydantic ai's cli optional dependency
-pydantic-ai-slim = {extras = ["openai", "cohere", "vertexai", "google", "anthropic", "groq", "mistral", "bedrock", "huggingface", "cli", "mcp"], version = "^0.6.2"} 
+pydantic-ai-slim = {extras = ["openai", "cohere", "vertexai", "google", "anthropic", "groq", "mistral", "bedrock", "huggingface", "cli", "mcp"], version = "^0.7.0"} 
 
-# NOTE: On Pydantic AI 0.6.2, The last working version of OpenAI is 0.99.1
-openai = ">=1.92.0, <=1.99.1"
+# NOTE: On Pydantic AI 0.7.0, OpenAI is 1.99.2 to 1.99.6 seems to be broken
+openai = ">=1.99.7"
 
 tiktoken = "^0.8.0"
 playwright = {version = "^1.53.0", optional = true}
 chromadb = {version = "^0.6.3", optional = true}
+markdownify = "^1.2.0"
 
 [tool.poetry.extras]
 # poetry install -E rag

{zrb-1.14.8 → zrb-1.15.1}/src/zrb/builtin/http.py

@@ -2,6 +2,7 @@ from typing import Any
 
 from zrb.builtin.group import http_group
 from zrb.context.any_context import AnyContext
+from zrb.input.bool_input import BoolInput
 from zrb.input.option_input import OptionInput
 from zrb.input.str_input import StrInput
 from zrb.task.make_task import make_task
@@ -32,10 +33,9 @@ from zrb.task.make_task import make_task
             prompt="Enter body as JSON",
             default="{}",
         ),
-        OptionInput(
+        BoolInput(
             name="verify_ssl",
-            default="true",
-            options=["true", "false"],
+            default=True,
             description="Verify SSL certificate",
         ),
     ],
@@ -55,7 +55,7 @@ def http_request(ctx: AnyContext) -> Any:
             body = json.loads(ctx.input.body)
 
         # Make request
-        verify = ctx.input.verify_ssl.lower() == "true"
+        verify = ctx.input.verify_ssl
         response = requests.request(
             method=ctx.input.method,
             url=ctx.input.url,
@@ -107,10 +107,9 @@ def http_request(ctx: AnyContext) -> Any:
             prompt="Enter body as JSON",
             default="{}",
         ),
-        OptionInput(
+        BoolInput(
             name="verify_ssl",
-            default="true",
-            options=["true", "false"],
+            default=True,
             description="Verify SSL certificate",
         ),
     ],
@@ -137,7 +136,7 @@ def generate_curl(ctx: AnyContext) -> str:
             parts.extend(["--data-raw", shlex.quote(ctx.input.body)])
 
         # Add SSL verification
-        if ctx.input.verify_ssl.lower() == "false":
+        if not ctx.input.verify_ssl:
             parts.append("--insecure")
 
         # Add URL

{zrb-1.14.8 → zrb-1.15.1}/src/zrb/builtin/llm/llm_ask.py

@@ -148,10 +148,14 @@ llm_group.add_task(
     alias="chat",
 )
 
+if CFG.LLM_ALLOW_ANALYZE_REPO:
+    llm_ask.append_tool(analyze_repo)
+
+if CFG.LLM_ALLOW_ANALYZE_FILE:
+    llm_ask.append_tool(analyze_file)
+
 if CFG.LLM_ALLOW_ACCESS_LOCAL_FILE:
     llm_ask.append_tool(
-        analyze_repo,
-        analyze_file,
         search_files,
         list_files,
         read_from_file,

{zrb-1.14.8 → zrb-1.15.1}/src/zrb/builtin/llm/tool/api.py

@@ -4,15 +4,18 @@ from typing import Literal
 
 def get_current_location() -> str:
     """
-    Fetches the user's current geographical location (latitude and longitude) based on their IP address.
+    Fetches the user's current geographical location based on their IP address.
 
-    Use this tool when the user asks "Where am I?", "What is my current location?", or has a query that requires knowing their location to be answered.
+    Use this tool when the user asks "Where am I?", "What is my current
+    location?", or has a query that requires knowing their location to be
+    answered.
 
     Returns:
-        str: A JSON string containing the 'lat' and 'lon' of the current location.
-             Example: '{"lat": 48.8584, "lon": 2.2945}'
+        str: A JSON string containing the 'lat' and 'lon' of the current
+            location. Example: '{"lat": 48.8584, "lon": 2.2945}'
     Raises:
-        requests.RequestException: If the API request to the location service fails.
+        requests.RequestException: If the API request to the location service
+            fails.
     """
     import requests
 
@@ -32,17 +35,22 @@ def get_current_weather(
     """
     Retrieves the current weather conditions for a given geographical location.
 
-    Use this tool when the user asks about the weather. If the user does not provide a location, first use the `get_current_location` tool to determine their location.
+    Use this tool when the user asks about the weather. If the user does not
+    provide a location, first use the `get_current_location` tool to
+    determine their location.
 
     Args:
         latitude (float): The latitude of the location.
         longitude (float): The longitude of the location.
-        temperature_unit (Literal["celsius", "fahrenheit"]): The desired unit for the temperature reading.
+        temperature_unit (Literal["celsius", "fahrenheit"]): The desired unit
+            for the temperature reading.
 
     Returns:
-        str: A JSON string containing detailed weather data, including temperature, wind speed, and weather code.
+        str: A JSON string containing detailed weather data, including
+            temperature, wind speed, and weather code.
     Raises:
-        requests.RequestException: If the API request to the weather service fails.
+        requests.RequestException: If the API request to the weather service
+            fails.
     """
     import requests
 

{zrb-1.14.8 → zrb-1.15.1}/src/zrb/builtin/llm/tool/cli.py

@@ -6,9 +6,14 @@ def run_shell_command(command: str) -> str:
     """
     Executes a shell command on the user's local machine and returns the output.
 
-    This tool is powerful and should be used for tasks that require interacting with the command line, such as running scripts, managing system processes, or using command-line tools.
+    This tool is powerful and should be used for tasks that require interacting
+    with the command line, such as running scripts, managing system processes,
+    or using command-line tools.
 
-    **Security Warning:** This tool executes commands with the same permissions as the user running the assistant. Before executing any command that could modify files or system state (e.g., `git`, `npm`, `pip`, `docker`), you MUST explain what the command does and ask the user for confirmation.
+    **Security Warning:** This tool executes commands with the same permissions
+    as the user running the assistant. Before executing any command that could
+    modify files or system state (e.g., `git`, `npm`, `pip`, `docker`), you
+    MUST explain what the command does and ask the user for confirmation.
 
     Args:
         command (str): The exact shell command to execute.

{zrb-1.14.8 → zrb-1.15.1}/src/zrb/builtin/llm/tool/code.py

@@ -58,25 +58,49 @@ async def analyze_repo(
     """
     Performs a deep, goal-oriented analysis of a code repository or directory.
 
-    This powerful tool recursively reads all relevant files in a directory, extracts key information, and then summarizes that information in relation to a specific goal. It uses intelligent sub-agents for extraction and summarization, making it ideal for complex tasks that require a holistic understanding of a codebase.
+    This powerful tool recursively reads all relevant files in a directory,
+    extracts key information, and then summarizes that information in relation
+    to a specific goal. It uses intelligent sub-agents for extraction and
+    summarization, making it ideal for complex tasks that require a holistic
+    understanding of a codebase.
+
+    To ensure a focused and effective analysis, it is crucial to provide a
+    clear and specific goal. Vague goals will result in a vague analysis and
+    may cause the tool to run for a long time.
 
     Use this tool for:
     - Understanding a large or unfamiliar codebase.
     - Generating high-level summaries of a project's architecture.
     - Performing a preliminary code review.
-    - Creating documentation or diagrams (e.g., "Generate a Mermaid C4 diagram for this service").
-    - Answering broad questions like "How does the authentication in this project work?".
+    - Creating documentation or diagrams (e.g., "Generate a Mermaid C4 diagram
+      for this service").
+    - Answering broad questions like "How does the authentication in this
+      project work?".
 
     Args:
         path (str): The path to the directory or repository to analyze.
-        goal (str): A clear and specific description of what you want to achieve. A good goal is critical for getting a useful result. For example: "Understand the database schema by analyzing all the .sql files" or "Create a summary of all the API endpoints defined in the 'api' directory".
-        extensions (list[str], optional): A list of file extensions to include in the analysis. Defaults to a comprehensive list of common code and configuration files.
-        exclude_patterns (list[str], optional): A list of glob patterns for files and directories to exclude from the analysis. Defaults to common patterns like '.git', 'node_modules', and '.venv'.
-        extraction_token_threshold (int, optional): The maximum token threshold for the extraction sub-agent.
-        summarization_token_threshold (int, optional): The maximum token threshold for the summarization sub-agent.
+        goal (str): A clear and specific description of what you want to
+            achieve. A good goal is critical for getting a useful result.
+            - Good goal: "Understand the database schema by analyzing all the
+              .sql files"
+            - Good goal: "Create a summary of all the API endpoints defined in
+              the 'api' directory"
+            - Bad goal: "Analyze the repo"
+            - Bad goal: "Tell me about the code"
+        extensions (list[str], optional): A list of file extensions to include
+            in the analysis. Defaults to a comprehensive list of common code
+            and configuration files.
+        exclude_patterns (list[str], optional): A list of glob patterns for
+            files and directories to exclude from the analysis. Defaults to
+            common patterns like '.git', 'node_modules', and '.venv'.
+        extraction_token_threshold (int, optional): The maximum token
+            threshold for the extraction sub-agent.
+        summarization_token_threshold (int, optional): The maximum token
+            threshold for the summarization sub-agent.
 
     Returns:
-        str: A detailed, markdown-formatted analysis and summary of the repository, tailored to the specified goal.
+        str: A detailed, markdown-formatted analysis and summary of the
+            repository, tailored to the specified goal.
     Raises:
         Exception: If an error occurs during the analysis.
     """

{zrb-1.14.8 → zrb-1.15.1}/src/zrb/builtin/llm/tool/file.py

@@ -102,17 +102,27 @@ def list_files(
     """
     Lists the files and directories within a specified path.
 
-    This is a fundamental tool for exploring the file system. Use it to discover the structure of a directory, find specific files, or get a general overview of the project layout before performing other operations.
+    This is a fundamental tool for exploring the file system. Use it to
+    discover the structure of a directory, find specific files, or get a
+    general overview of the project layout before performing other operations.
 
     Args:
-        path (str, optional): The directory path to list. Defaults to the current directory (".").
-        recursive (bool, optional): If True, lists files and directories recursively. If False, lists only the top-level contents. Defaults to True.
-        include_hidden (bool, optional): If True, includes hidden files and directories (those starting with a dot). Defaults to False.
-        excluded_patterns (list[str], optional): A list of glob patterns to exclude from the listing. This is useful for ignoring irrelevant files like build artifacts or virtual environments. Defaults to a standard list of common exclusion patterns.
+        path (str, optional): The directory path to list. Defaults to the
+            current directory (".").
+        recursive (bool, optional): If True, lists files and directories
+            recursively. If False, lists only the top-level contents.
+            Defaults to True.
+        include_hidden (bool, optional): If True, includes hidden files and
+            directories (those starting with a dot). Defaults to False.
+        excluded_patterns (list[str], optional): A list of glob patterns to
+            exclude from the listing. This is useful for ignoring irrelevant
+            files like build artifacts or virtual environments. Defaults to a
+            standard list of common exclusion patterns.
 
     Returns:
-        str: A JSON string containing a list of file and directory paths relative to the input path.
-             Example: '{"files": ["src/main.py", "README.md"]}'
+        str: A JSON string containing a list of file and directory paths
+            relative to the input path.
+            Example: '{"files": ["src/main.py", "README.md"]}'
     Raises:
         FileNotFoundError: If the specified path does not exist.
     """
@@ -213,26 +223,37 @@ def read_from_file(
     end_line: Optional[int] = None,
 ) -> str:
     """
-    Reads the content of a file, optionally from a specific start line to an end line.
-
-    This tool is essential for inspecting file contents. It can read both text and PDF files. The returned content is prefixed with line numbers, which is crucial for providing context when you need to modify the file later with the `apply_diff` tool.
-
-    Use this tool to:
-    - Examine the source code of a file.
-    - Read configuration files.
-    - Check the contents of a document.
-
-    Args:
-        path (str): The path to the file to read.
-        start_line (int, optional): The 1-based line number to start reading from. If omitted, reading starts from the beginning of the file.
-        end_line (int, optional): The 1-based line number to stop reading at (inclusive). If omitted, reads to the end of the file.
-
-    Returns:
-        str: A JSON object containing the file path, the requested content with line numbers, the start and end lines, and the total number of lines in the file.
-             Example: '{"path": "src/main.py", "content": "1| import os\n2| \n3| print(\"Hello, World!\")", "start_line": 1, "end_line": 3, "total_lines": 3}'
-    Raises:
-        FileNotFoundError: If the specified file does not exist.
+        Reads the content of a file, optionally from a specific start line to an
+        end line.
+
+        This tool is essential for inspecting file contents. It can read both text
+        and PDF files. The returned content is prefixed with line numbers, which is
+        crucial for providing context when you need to modify the file later with
+        the `apply_diff` tool.
+
+        Use this tool to:
+        - Examine the source code of a file.
+        - Read configuration files.
+        - Check the contents of a document.
+
+        Args:
+            path (str): The path to the file to read.
+            start_line (int, optional): The 1-based line number to start reading
+                from. If omitted, reading starts from the beginning of the file.
+            end_line (int, optional): The 1-based line number to stop reading at
+                (inclusive). If omitted, reads to the end of the file.
+
+        Returns:
+            str: A JSON object containing the file path, the requested content
+                with line numbers, the start and end lines, and the total number
+                of lines in the file.
+                Example: '{"path": "src/main.py", "content": "1| import os\n2|
+    3| print(\"Hello, World!\")", "start_line": 1, "end_line": 3,
+                "total_lines": 3}'
+        Raises:
+            FileNotFoundError: If the specified file does not exist.
     """
+
     abs_path = os.path.abspath(os.path.expanduser(path))
     # Check if file exists
     if not os.path.exists(abs_path):
@@ -274,13 +295,18 @@ def write_to_file(
     content: str,
 ) -> str:
     """
-    Writes content to a file, completely overwriting it if it exists or creating it if it doesn't.
+    Writes content to a file, completely overwriting it if it exists or
+    creating it if it doesn't.
 
-    Use this tool to create new files or to replace the entire content of existing files. This is a destructive operation, so be certain of your actions. Always read the file first to understand its contents before overwriting it, unless you are creating a new file.
+    Use this tool to create new files or to replace the entire content of
+    existing files. This is a destructive operation, so be certain of your
+    actions. Always read the file first to understand its contents before
+    overwriting it, unless you are creating a new file.
 
     Args:
         path (str): The path to the file to write to.
-        content (str): The full, complete content to be written to the file. Do not use partial content or omit any lines.
+        content (str): The full, complete content to be written to the file.
+            Do not use partial content or omit any lines.
 
     Returns:
         str: A JSON object indicating success or failure.
@@ -308,18 +334,28 @@ def search_files(
     include_hidden: bool = True,
 ) -> str:
     """
-    Searches for a regular expression (regex) pattern within files in a specified directory.
+    Searches for a regular expression (regex) pattern within files in a
+    specified directory.
 
-    This tool is invaluable for finding specific code, configuration, or text across multiple files. Use it to locate function definitions, variable assignments, error messages, or any other text pattern.
+    This tool is invaluable for finding specific code, configuration, or text
+    across multiple files. Use it to locate function definitions, variable
+    assignments, error messages, or any other text pattern.
 
     Args:
         path (str): The directory path to start the search from.
-        regex (str): The Python-compatible regular expression pattern to search for.
-        file_pattern (str, optional): A glob pattern to filter which files get searched (e.g., "*.py", "*.md"). If omitted, all files are searched.
-        include_hidden (bool, optional): If True, the search will include hidden files and directories. Defaults to True.
+        regex (str): The Python-compatible regular expression pattern to search
+            for.
+        file_pattern (str, optional): A glob pattern to filter which files get
+            searched (e.g., "*.py", "*.md"). If omitted, all files are
+            searched.
+        include_hidden (bool, optional): If True, the search will include
+            hidden files and directories. Defaults to True.
 
     Returns:
-        str: A JSON object containing a summary of the search and a list of results. Each result includes the file path and a list of matches, with each match showing the line number, line content, and a few lines of context from before and after the match.
+        str: A JSON object containing a summary of the search and a list of
+            results. Each result includes the file path and a list of matches,
+            with each match showing the line number, line content, and a few
+            lines of context from before and after the match.
     Raises:
         ValueError: If the provided `regex` pattern is invalid.
     """
@@ -416,13 +452,18 @@ def replace_in_file(
     """
     Replaces the first occurrence of a string in a file.
 
-    This tool is for making targeted modifications to a file. It is a single-step operation that is generally safer and more ergonomic than `write_to_file` for small changes.
+    This tool is for making targeted modifications to a file. It is a
+    single-step operation that is generally safer and more ergonomic than
+    `write_to_file` for small changes.
 
-    To ensure the replacement is applied correctly and to avoid ambiguity, the `old_string` parameter should be a unique, multi-line string that includes context from before and after the code you want to change.
+    To ensure the replacement is applied correctly and to avoid ambiguity, the
+    `old_string` parameter should be a unique, multi-line string that includes
+    context from before and after the code you want to change.
 
     Args:
         path (str): The path of the file to modify.
-        old_string (str): The exact, verbatim string to search for and replace. This should be a unique, multi-line block of text.
+        old_string (str): The exact, verbatim string to search for and replace.
+            This should be a unique, multi-line block of text.
         new_string (str): The new string that will replace the `old_string`.
 
     Returns:
@@ -455,21 +496,37 @@ async def analyze_file(
     """
     Performs a deep, goal-oriented analysis of a single file using a sub-agent.
 
-    This tool is ideal for complex questions about a single file that go beyond simple reading or searching. It uses a specialized sub-agent to analyze the file's content in relation to a specific query.
+    This tool is ideal for complex questions about a single file that go beyond
+    simple reading or searching. It uses a specialized sub-agent to analyze the
+    file's content in relation to a specific query.
+
+    To ensure a focused and effective analysis, it is crucial to provide a
+    clear and specific query. Vague queries will result in a vague analysis
+    and may cause the tool to run for a long time.
 
     Use this tool to:
     - Summarize the purpose and functionality of a script or configuration file.
-    - Extract the structure of a file (e.g., "List all the function names in this Python file").
+    - Extract the structure of a file (e.g., "List all the function names in
+      this Python file").
     - Perform a detailed code review of a specific file.
-    - Answer complex questions like, "How is the 'User' class used in this file?".
+    - Answer complex questions like, "How is the 'User' class used in this
+      file?".
 
     Args:
         path (str): The path to the file to be analyzed.
-        query (str): A clear and specific question or instruction about what to analyze in the file.
-        token_limit (int, optional): The maximum token length of the file content to be passed to the analysis sub-agent.
+        query (str): A clear and specific question or instruction about what to
+            analyze in the file.
+            - Good query: "What is the purpose of the 'User' class in this
+              file?"
+            - Good query: "List all the function names in this Python file."
+            - Bad query: "Analyze this file."
+            - Bad query: "Tell me about this code."
+        token_limit (int, optional): The maximum token length of the file
+            content to be passed to the analysis sub-agent.
 
     Returns:
-        str: A detailed, markdown-formatted analysis of the file, tailored to the specified query.
+        str: A detailed, markdown-formatted analysis of the file, tailored to
+            the specified query.
     Raises:
         FileNotFoundError: If the specified file does not exist.
     """
@@ -496,14 +553,22 @@ 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.
+    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.
-             Example: '{"results": {"src/api.py": "1| import ...", "config.yaml": "1| key: value"}}'
+        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.
+            Example: '{"results": {"src/api.py": "1| import ...",
+            "config.yaml": "1| key: value"}}'
     """
     results = {}
     for path in paths:
@@ -522,16 +587,24 @@ def write_many_files(files: list[FileToWrite]) -> str:
     """
     Writes content to multiple files in a single, atomic operation.
 
-    This tool is for applying widespread changes to a project, such as creating a set of new files from a template, updating multiple configuration files, or performing a large-scale refactoring.
+    This tool is for applying widespread changes to a project, such as
+    creating a set of new files from a template, updating multiple
+    configuration files, or performing a large-scale refactoring.
 
-    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.
+    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: A list of file objects, where each object is a dictionary containing a 'path' and the complete 'content'.
+        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.
-             Example: '{"success": ["file1.py", "file2.txt"], "errors": {}}'
+        str: A JSON object summarizing the operation, listing successfully
+            written files and any files that failed, along with corresponding
+            error messages.
+            Example: '{"success": ["file1.py", "file2.txt"], "errors": {}}'
     """
     success = []
     errors = {}

{zrb-1.14.8 → zrb-1.15.1}/src/zrb/builtin/llm/tool/rag.py

@@ -44,26 +44,40 @@ def create_rag_from_directory(
     openai_embedding_model: str | None = None,
 ):
     """
-    Creates a powerful Retrieval-Augmented Generation (RAG) tool for querying a local knowledge base.
+    Creates a powerful RAG tool for querying a local knowledge base.
 
-    This factory function generates a tool that can perform semantic searches over a directory of documents. It automatically indexes the documents into a vector database, keeping it updated as files change. The generated tool is ideal for answering questions based on a specific set of documents, such as project documentation, research papers, or internal wikis.
+    This factory function generates a tool that can perform semantic searches
+    over a directory of documents. It automatically indexes the documents into
+    a vector database, keeping it updated as files change. The generated tool
+    is ideal for answering questions based on a specific set of documents,
+    such as project documentation, research papers, or internal wikis.
 
     The created tool will:
     1.  Monitor a specified directory for file changes.
-    2.  Automatically update a vector database (ChromaDB) with the latest content.
-    3.  Accept a user query, embed it, and perform a similarity search against the document vectors.
+    2.  Automatically update a vector database (ChromaDB) with the latest
+        content.
+    3.  Accept a user query, embed it, and perform a similarity search against
+        the document vectors.
     4.  Return the most relevant document chunks that match the query.
 
     Args:
-        tool_name (str): The name for the generated RAG tool (e.g., "search_project_docs").
-        tool_description (str): A clear description of what the generated tool does and when to use it (e.g., "Searches the project's technical documentation to answer questions.").
-        document_dir_path (str, optional): The path to the directory containing the documents to be indexed.
-        vector_db_path (str, optional): The path to store the ChromaDB vector database.
-        vector_db_collection (str, optional): The name of the collection within the vector database.
+        tool_name (str): The name for the generated RAG tool (e.g.,
+            "search_project_docs").
+        tool_description (str): A clear description of what the generated tool
+            does and when to use it (e.g., "Searches the project's technical
+            documentation to answer questions.").
+        document_dir_path (str, optional): The path to the directory
+            containing the documents to be indexed.
+        vector_db_path (str, optional): The path to store the ChromaDB vector
+            database.
+        vector_db_collection (str, optional): The name of the collection within
+            the vector database.
         chunk_size (int, optional): The size of text chunks for embedding.
         overlap (int, optional): The overlap between text chunks.
-        max_result_count (int, optional): The maximum number of search results to return.
-        file_reader (list[RAGFileReader], optional): Custom file readers for specific file types.
+        max_result_count (int, optional): The maximum number of search results
+            to return.
+        file_reader (list[RAGFileReader], optional): Custom file readers for
+            specific file types.
         openai_api_key (str, optional): OpenAI API key for embeddings.
         openai_base_url (str, optional): OpenAI base URL for embeddings.
         openai_embedding_model (str, optional): The embedding model to use.