zrb 1.3.1__py3-none-any.whl → 1.4.1__py3-none-any.whl

zrb/__init__.py

@@ -33,6 +33,7 @@ from zrb.input.option_input import OptionInput
 from zrb.input.password_input import PasswordInput
 from zrb.input.str_input import StrInput
 from zrb.input.text_input import TextInput
+from zrb.llm_config import llm_config
 from zrb.runner.cli import cli
 from zrb.runner.web_config.config_factory import web_config
 from zrb.runner.web_schema.user import User
@@ -101,6 +102,7 @@ assert ContentTransformer
 assert Scaffolder
 assert Scheduler
 assert cli
+assert llm_config
 assert Xcom
 assert web_config
 assert User

zrb/__main__.py

@@ -1,3 +1,4 @@
+import os
 import sys
 
 from zrb.config import INIT_MODULES, INIT_SCRIPTS
@@ -9,10 +10,19 @@ from zrb.util.load import load_file, load_module
 
 def serve_cli():
     try:
+        # load init modules
         for init_module in INIT_MODULES:
             load_module(init_module)
+        zrb_init_path_list = _get_zrb_init_path_list()
+        # load init scripts
         for init_script in INIT_SCRIPTS:
-            load_file(init_script, -1)
+            abs_init_script = os.path.abspath(init_script)
+            if abs_init_script not in zrb_init_path_list:
+                load_file(abs_init_script, -1)
+        # load zrb init
+        for zrb_init_path in zrb_init_path_list:
+            load_file(zrb_init_path)
+        # run the CLI
         cli.run(sys.argv[1:])
     except KeyboardInterrupt:
         print(stylize_warning("\nStopped"), file=sys.stderr)
@@ -24,3 +34,17 @@ def serve_cli():
     except NodeNotFoundError as e:
         print(stylize_error(f"{e}"), file=sys.stderr)
         sys.exit(1)
+
+
+def _get_zrb_init_path_list() -> list[str]:
+    current_path = os.path.abspath(os.getcwd())
+    dir_path_list = [current_path]
+    while current_path != os.path.dirname(current_path):  # Stop at root
+        current_path = os.path.dirname(current_path)
+        dir_path_list.append(current_path)
+    zrb_init_path_list = []
+    for current_path in dir_path_list[::-1]:
+        zrb_init_path = os.path.join(current_path, "zrb_init.py")
+        if os.path.isfile(zrb_init_path):
+            zrb_init_path_list.append(zrb_init_path)
+    return zrb_init_path_list

zrb/builtin/llm/llm_chat.py

@@ -2,14 +2,12 @@ import json
 import os
 from typing import Any
 
-from pydantic_ai.models import Model
-
 from zrb.builtin.group import llm_group
 from zrb.builtin.llm.tool.api import get_current_location, get_current_weather
 from zrb.builtin.llm.tool.cli import run_shell_command
 from zrb.builtin.llm.tool.file import (
-    list_file,
-    read_source_code,
+    list_files,
+    read_all_files,
     read_text_file,
     write_text_file,
 )
@@ -24,18 +22,14 @@ from zrb.config import (
     LLM_ALLOW_ACCESS_LOCAL_FILE,
     LLM_ALLOW_ACCESS_SHELL,
     LLM_HISTORY_DIR,
-    LLM_MODEL,
     LLM_SYSTEM_PROMPT,
     SERP_API_KEY,
 )
-from zrb.context.any_context import AnyContext
 from zrb.context.any_shared_context import AnySharedContext
-from zrb.input.any_input import AnyInput
 from zrb.input.bool_input import BoolInput
 from zrb.input.str_input import StrInput
 from zrb.input.text_input import TextInput
 from zrb.task.llm_task import LLMTask
-from zrb.util.attr import get_attr
 from zrb.util.file import read_file, write_file
 from zrb.util.string.conversion import to_pascal_case
 
@@ -90,46 +84,37 @@ def _write_chat_conversation(
     write_file(last_session_file_path, current_session_name)
 
 
-class _LLMChat(LLMTask):
-
-    _default_model: Model | str | None = None
-
-    def set_default_model(self, model: Model | str):
-        self._default_model = model
-
-    @property
-    def inputs(self) -> list[AnyInput]:
-        task_inputs = super().inputs
-        model_input_default = LLM_MODEL if self._default_model is None else "default"
-        return [
+llm_chat: LLMTask = llm_group.add_task(
+    LLMTask(
+        name="llm-chat",
+        input=[
             StrInput(
                 "model",
                 description="LLM Model",
                 prompt="LLM Model",
-                default=model_input_default,
+                default="",
                 allow_positional_parsing=False,
                 always_prompt=False,
+                allow_empty=True,
+            ),
+            StrInput(
+                "base-url",
+                description="LLM API Base URL",
+                prompt="LLM API Base URL",
+                default="",
+                allow_positional_parsing=False,
+                always_prompt=False,
+                allow_empty=True,
+            ),
+            StrInput(
+                "api-key",
+                description="LLM API Key",
+                prompt="LLM API Key",
+                default="",
+                allow_positional_parsing=False,
+                always_prompt=False,
+                allow_empty=True,
             ),
-            *task_inputs,
-        ]
-
-    def _get_model(self, ctx: AnyContext) -> str | Model | None:
-        if ctx.input.model == "default":
-            if self._default_model is not None:
-                return self._default_model
-            return super()._get_model(ctx)
-        model = get_attr(
-            ctx, ctx.input.model, "ollama_chat/llama3.1", auto_render=self._render_model
-        )
-        if isinstance(model, (Model, str)) or model is None:
-            return model
-        raise ValueError("Invalid model")
-
-
-llm_chat: LLMTask = llm_group.add_task(
-    _LLMChat(
-        name="llm-chat",
-        input=[
             TextInput(
                 "system-prompt",
                 description="System prompt",
@@ -156,10 +141,17 @@ llm_chat: LLMTask = llm_group.add_task(
                 always_prompt=False,
             ),
         ],
+        model=lambda ctx: None if ctx.input.model == "" else ctx.input.model,
+        model_base_url=lambda ctx: (
+            None if ctx.input.base_url == "" else ctx.input.base_url
+        ),
+        model_api_key=lambda ctx: (
+            None if ctx.input.api_key == "" else ctx.input.api_key
+        ),
         conversation_history_reader=_read_chat_conversation,
         conversation_history_writer=_write_chat_conversation,
         description="Chat with LLM",
-        system_prompt="{ctx.input['system-prompt']}",
+        system_prompt="{ctx.input.system_prompt}",
         message="{ctx.input.message}",
         retries=0,
     ),
@@ -168,8 +160,8 @@ llm_chat: LLMTask = llm_group.add_task(
 
 
 if LLM_ALLOW_ACCESS_LOCAL_FILE:
-    llm_chat.add_tool(read_source_code)
-    llm_chat.add_tool(list_file)
+    llm_chat.add_tool(read_all_files)
+    llm_chat.add_tool(list_files)
     llm_chat.add_tool(read_text_file)
     llm_chat.add_tool(write_text_file)
 

zrb/builtin/llm/tool/file.py

@@ -1,22 +1,131 @@
+import fnmatch
 import os
 
 from zrb.util.file import read_file, write_file
 
+_INCLUDED_PATTERNS: list[str] = [
+    "*.py",  # Python
+    "*.go",  # Go
+    "*.rs",  # Rust
+    "*.js",  # JavaScript
+    "*.ts",  # TypeScript
+    "*.java",  # Java
+    "*.c",  # C
+    "*.cpp",  # C++
+    "*.cc",  # Alternative C++ extension
+    "*.cxx",  # Alternative C++ extension
+    "*.rb",  # Ruby
+    "*.swift",  # Swift
+    "*.kt",  # Kotlin
+    "*.php",  # PHP
+    "*.pl",  # Perl / Prolog
+    "*.pm",  # Perl module
+    "*.sh",  # Shell
+    "*.bat",  # Batch
+    "*.ps1",  # PowerShell
+    "*.R",  # R (capital)
+    "*.r",  # R (lowercase)
+    "*.scala",  # Scala
+    "*.hs",  # Haskell
+    "*.cs",  # C#
+    "*.fs",  # F#
+    "*.ex",  # Elixir
+    "*.exs",  # Elixir script
+    "*.erl",  # Erlang
+    "*.hrl",  # Erlang header
+    "*.dart",  # Dart
+    "*.m",  # Objective-C / Matlab (note: conflicts may arise)
+    "*.mm",  # Objective-C++
+    "*.lua",  # Lua
+    "*.jl",  # Julia
+    "*.groovy",  # Groovy
+    "*.clj",  # Clojure
+    "*.cljs",  # ClojureScript
+    "*.cljc",  # Clojure common
+    "*.vb",  # Visual Basic
+    "*.f90",  # Fortran
+    "*.f95",  # Fortran
+    "*.adb",  # Ada
+    "*.ads",  # Ada specification
+    "*.pas",  # Pascal
+    "*.pp",  # Pascal
+    "*.ml",  # OCaml
+    "*.mli",  # OCaml interface
+    "*.nim",  # Nim
+    "*.rkt",  # Racket
+    "*.d",  # D
+    "*.lisp",  # Common Lisp
+    "*.lsp",  # Lisp variant
+    "*.cl",  # Common Lisp
+    "*.scm",  # Scheme
+    "*.st",  # Smalltalk
+    "*.vhd",  # VHDL
+    "*.vhdl",  # VHDL
+    "*.v",  # Verilog
+    "*.asm",  # Assembly
+    "*.s",  # Assembly (alternative)
+    "*.sql",  # SQL (if desired)
+]
 
-def list_file(
+# Extended list of directories and patterns to exclude.
+_EXCLUDED_PATTERNS: list[str] = [
+    "venv",  # Python virtual environments
+    ".venv",
+    "node_modules",  # Node.js dependencies
+    ".git",  # Git repositories
+    "__pycache__",  # Python cache directories
+    "build",  # Build directories
+    "dist",  # Distribution directories
+    "target",  # Build output directories (Java, Rust, etc.)
+    "bin",  # Binary directories
+    "obj",  # Object files directories
+    ".idea",  # JetBrains IDEs
+    ".vscode",  # VS Code settings
+    ".eggs",  # Python eggs
+]
+
+
+def list_files(
     directory: str = ".",
-    extensions: list[str] = [".py", ".go", ".js", ".ts", ".java", ".c", ".cpp"],
+    included_patterns: list[str] = _INCLUDED_PATTERNS,
+    excluded_patterns: list[str] = _EXCLUDED_PATTERNS,
 ) -> list[str]:
-    """List all files in a directory"""
+    """List all files in a directory that match any of the included glob patterns
+    and do not reside in any directory matching an excluded pattern.
+    Patterns are evaluated using glob-style matching.
+    """
     all_files: list[str] = []
-    for root, _, files in os.walk(directory):
+    for root, dirs, files in os.walk(directory):
         for filename in files:
-            for extension in extensions:
-                if filename.lower().endswith(extension):
-                    all_files.append(os.path.join(root, filename))
+            if any(fnmatch.fnmatch(filename, pat) for pat in included_patterns):
+                full_path = os.path.join(root, filename)
+                if _should_exclude(full_path, excluded_patterns):
+                    continue
+                all_files.append(full_path)
     return all_files
 
 
+def _should_exclude(full_path: str, excluded_patterns: list[str]) -> bool:
+    """
+    Return True if the file at full_path should be excluded based on
+    the list of excluded_patterns. Patterns that include a path separator
+    are applied to the full normalized path; otherwise they are matched
+    against each individual component of the path.
+    """
+    norm_path = os.path.normpath(full_path)
+    path_parts = norm_path.split(os.sep)
+    for pat in excluded_patterns:
+        # If the pattern seems intended for full path matching (contains a separator)
+        if os.sep in pat or "/" in pat:
+            if fnmatch.fnmatch(norm_path, pat):
+                return True
+        else:
+            # Otherwise check each part of the path
+            if any(fnmatch.fnmatch(part, pat) for part in path_parts):
+                return True
+    return False
+
+
 def read_text_file(file: str) -> str:
     """Read a text file"""
     return read_file(os.path.abspath(file))
@@ -27,12 +136,16 @@ def write_text_file(file: str, content: str):
     return write_file(os.path.abspath(file), content)
 
 
-def read_source_code(
+def read_all_files(
     directory: str = ".",
-    extensions: list[str] = [".py", ".go", ".js", ".ts", ".java", ".c", ".cpp"],
+    included_patterns: list[str] = _INCLUDED_PATTERNS,
+    excluded_patterns: list[str] = _EXCLUDED_PATTERNS,
 ) -> list[str]:
-    """Read source code in a directory"""
-    files = list_file(directory, extensions)
+    """Read all files in a directory that match any of the included glob patterns
+    and do not match any of the excluded glob patterns.
+    Patterns are evaluated using glob-style matching.
+    """
+    files = list_files(directory, included_patterns, excluded_patterns)
     for index, file in enumerate(files):
         content = read_text_file(file)
         files[index] = f"# {file}\n```\n{content}\n```"

zrb/config.py

@@ -75,7 +75,6 @@ WEB_AUTH_ACCESS_TOKEN_EXPIRE_MINUTES = int(
 WEB_AUTH_REFRESH_TOKEN_EXPIRE_MINUTES = int(
     os.getenv("ZRB_WEB_REFRESH_TOKEN_EXPIRE_MINUTES", "60")
 )
-LLM_MODEL = os.getenv("ZRB_LLM_MODEL", "ollama_chat/llama3.1")
 
 _DEFAULT_PROMPT = (
     "You are a helpful AI assistant capable of using various tools to answer user queries. When solving a problem:\n"

zrb/llm_config.py

@@ -0,0 +1,50 @@
+import os
+
+from pydantic_ai.models import Model
+from pydantic_ai.models.openai import OpenAIModel
+from pydantic_ai.providers.openai import OpenAIProvider
+
+
+class LLMConfig:
+
+    def __init__(
+        self,
+        model_name: str | None = None,
+        base_url: str | None = None,
+        api_key: str | None = None,
+    ):
+        self._model_name = (
+            model_name if model_name is not None else os.getenv("ZRB_LLM_MODEL", None)
+        )
+        self._base_url = (
+            base_url if base_url is not None else os.getenv("ZRB_LLM_BASE_URL", None)
+        )
+        self._api_key = (
+            api_key if api_key is not None else os.getenv("ZRB_LLM_API_KEY", None)
+        )
+        self._default_model = None
+
+    def _get_model_name(self) -> str | None:
+        return self._model_name if self._model_name is not None else None
+
+    def _get_model_provider(self) -> OpenAIProvider:
+        if self._base_url is None and self._api_key is None:
+            return "openai"
+        return OpenAIProvider(base_url=self._base_url, api_key=self._api_key)
+
+    def get_default_model(self) -> Model | str | None:
+        if self._default_model is not None:
+            return self._default_model
+        model_name = self._get_model_name()
+        if model_name is None:
+            return None
+        return OpenAIModel(
+            model_name=model_name,
+            provider=self._get_model_provider(),
+        )
+
+    def set_default_model(self, model: Model | str | None):
+        self._default_model = model
+
+
+llm_config = LLMConfig()

zrb/runner/cli.py

@@ -18,13 +18,11 @@ from zrb.util.cli.style import (
     stylize_section_header,
 )
 from zrb.util.group import extract_node_from_args, get_non_empty_subgroups, get_subtasks
-from zrb.util.load import load_zrb_init
 from zrb.util.string.conversion import double_quote
 
 
 class Cli(Group):
     def run(self, args: list[str] = []):
-        load_zrb_init()
         kwargs, args = self._extract_kwargs_from_args(args)
         node, node_path, args = extract_node_from_args(self, args)
         if isinstance(node, Group):

zrb/task/llm_task.py

@@ -18,11 +18,13 @@ from pydantic_ai.models import Model
 from pydantic_ai.settings import ModelSettings
 
 from zrb.attr.type import StrAttr, fstring
-from zrb.config import LLM_MODEL, LLM_SYSTEM_PROMPT
+from zrb.config import LLM_SYSTEM_PROMPT
 from zrb.context.any_context import AnyContext
 from zrb.context.any_shared_context import AnySharedContext
 from zrb.env.any_env import AnyEnv
 from zrb.input.any_input import AnyInput
+from zrb.llm_config import LLMConfig
+from zrb.llm_config import llm_config as default_llm_config
 from zrb.task.any_task import AnyTask
 from zrb.task.base_task import BaseTask
 from zrb.util.attr import get_attr, get_str_attr
@@ -46,11 +48,15 @@ class LLMTask(BaseTask):
         env: list[AnyEnv | None] | AnyEnv | None = None,
         model: (
             Callable[[AnySharedContext], Model | str | fstring] | Model | None
-        ) = LLM_MODEL,
+        ) = None,
+        render_model: bool = True,
+        model_base_url: StrAttr = None,
+        render_model_base_url: bool = True,
+        model_api_key: StrAttr = None,
+        render_model_api_key: bool = True,
         model_settings: (
             ModelSettings | Callable[[AnySharedContext], ModelSettings] | None
         ) = None,
-        render_model: bool = True,
         agent: Agent | Callable[[AnySharedContext], Agent] | None = None,
         system_prompt: StrAttr | None = LLM_SYSTEM_PROMPT,
         render_system_prompt: bool = True,
@@ -105,9 +111,13 @@ class LLMTask(BaseTask):
             successor=successor,
         )
         self._model = model
+        self._render_model = render_model
+        self._model_base_url = model_base_url
+        self._render_model_base_url = render_model_base_url
+        self._model_api_key = model_api_key
+        self._render_model_api_key = render_model_api_key
         self._model_settings = model_settings
         self._agent = agent
-        self._render_model = render_model
         self._system_prompt = system_prompt
         self._render_system_prompt = render_system_prompt
         self._message = message
@@ -120,9 +130,6 @@ class LLMTask(BaseTask):
         self._render_history_file = render_history_file
         self._max_call_iteration = max_call_iteration
 
-    def set_model(self, model: Model | str):
-        self._model = model
-
     def add_tool(self, tool: ToolOrCallable):
         self._additional_tools.append(tool)
 
@@ -242,15 +249,47 @@ class LLMTask(BaseTask):
             system_prompt=self._get_system_prompt(ctx),
             tools=tools,
             model_settings=self._get_model_settings(ctx),
+            retries=3,
         )
 
     def _get_model(self, ctx: AnyContext) -> str | Model | None:
-        model = get_attr(
-            ctx, self._model, "ollama_chat/llama3.1", auto_render=self._render_model
+        model = get_attr(ctx, self._model, None, auto_render=self._render_model)
+        if model is None:
+            return default_llm_config.get_default_model()
+        if isinstance(model, str):
+            llm_config = LLMConfig(
+                model_name=model,
+                base_url=get_attr(
+                    ctx,
+                    self._get_model_base_url(ctx),
+                    None,
+                    auto_render=self._render_model_base_url,
+                ),
+                api_key=get_attr(
+                    ctx,
+                    self._get_model_api_key(ctx),
+                    None,
+                    auto_render=self._render_model_api_key,
+                ),
+            )
+            return llm_config.get_default_model()
+        raise ValueError(f"Invalid model: {model}")
+
+    def _get_model_base_url(self, ctx: AnyContext) -> str | None:
+        base_url = get_attr(
+            ctx, self._model_base_url, None, auto_render=self._render_model_base_url
+        )
+        if isinstance(base_url, str) or base_url is None:
+            return base_url
+        raise ValueError(f"Invalid model base URL: {base_url}")
+
+    def _get_model_api_key(self, ctx: AnyContext) -> str | None:
+        api_key = get_attr(
+            ctx, self._model_api_key, None, auto_render=self._render_model_api_key
         )
-        if isinstance(model, (Model, str)) or model is None:
-            return model
-        raise ValueError("Invalid model")
+        if isinstance(api_key, str) or api_key is None:
+            return api_key
+        raise ValueError(f"Invalid model base URL: {api_key}")
 
     def _get_system_prompt(self, ctx: AnyContext) -> str:
         return get_str_attr(

zrb/util/load.py

@@ -9,22 +9,6 @@ from typing import Any
 pattern = re.compile("[^a-zA-Z0-9]")
 
 
-def load_zrb_init(dir_path: str | None = None) -> Any | None:
-    if dir_path is None:
-        dir_path = os.getcwd()
-    # get path list from current path to the absolute root
-    current_path = os.path.abspath(dir_path)
-    path_list = [current_path]
-    while current_path != os.path.dirname(current_path):  # Stop at root
-        current_path = os.path.dirname(current_path)
-        path_list.append(current_path)
-    # loop from root to current path to load zrb_init
-    for current_path in path_list[::-1]:
-        script_path = os.path.join(current_path, "zrb_init.py")
-        if os.path.isfile(script_path):
-            load_file(script_path)
-
-
 @lru_cache
 def load_file(script_path: str, sys_path_index: int = 0) -> Any | None:
     if not os.path.isfile(script_path):

zrb-1.4.1.dist-info/METADATA

@@ -0,0 +1,212 @@
+Metadata-Version: 2.1
+Name: zrb
+Version: 1.4.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
+Author-email: gofrendiasgard@gmail.com
+Requires-Python: >=3.10.0,<4.0.0
+Classifier: License :: OSI Approved :: GNU Affero General Public License v3 or later (AGPLv3+)
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Provides-Extra: rag
+Requires-Dist: autopep8 (>=2.0.4,<3.0.0)
+Requires-Dist: beautifulsoup4 (>=4.12.3,<5.0.0)
+Requires-Dist: black (>=24.10.0,<24.11.0)
+Requires-Dist: chromadb (>=0.5.20,<0.6.0) ; extra == "rag"
+Requires-Dist: fastapi[standard] (>=0.115.6,<0.116.0)
+Requires-Dist: fastembed (>=0.5.1,<0.6.0)
+Requires-Dist: isort (>=5.13.2,<5.14.0)
+Requires-Dist: libcst (>=1.5.0,<2.0.0)
+Requires-Dist: pdfplumber (>=0.11.4,<0.12.0) ; extra == "rag"
+Requires-Dist: psutil (>=6.1.1,<7.0.0)
+Requires-Dist: pydantic-ai (>=0.0.42,<0.0.43)
+Requires-Dist: python-dotenv (>=1.0.1,<2.0.0)
+Requires-Dist: python-jose[cryptography] (>=3.4.0,<4.0.0)
+Requires-Dist: requests (>=2.32.3,<3.0.0)
+Requires-Dist: ulid-py (>=1.1.0,<2.0.0)
+Project-URL: Documentation, https://github.com/state-alchemists/zrb
+Project-URL: Repository, https://github.com/state-alchemists/zrb
+Description-Content-Type: text/markdown
+
+![](https://raw.githubusercontent.com/state-alchemists/zrb/main/_images/zrb/android-chrome-192x192.png)
+
+[Documentation](https://github.com/state-alchemists/zrb/blob/main/docs/README.md)
+
+# 🤖 Zrb: Your Automation Powerhouse
+
+
+**Unlock the full potential of automation in your projects!**
+
+Zrb streamlines repetitive tasks, integrates with powerful LLMs, and lets you create custom automation workflows effortlessly. Whether you’re building CI/CD pipelines, code generators, or unique automation scripts, Zrb is designed to simplify and supercharge your workflow.
+
+---
+
+## 🚀 Why Zrb?
+
+- **Easy Automation with Python:** Write your tasks in Python and let Zrb handle the rest.
+- **Seamless Integration:** Utilize built-in support for LLM tasks, command execution, and more.
+- **Custom Workflows:** Chain tasks, set dependencies, and build robust automation pipelines.
+- **Developer-Friendly:** Quick to install and get started, with clear documentation and examples.
+- **Web Interface:** Run Zrb as a server to make tasks accessible even to non-technical team members.
+
+---
+
+## 🔥 Key Features
+
+- **LLM Integration:** Leverage state-of-the-art language models to generate code, diagrams, and documentation.
+- **Task Chaining:** Easily define dependencies between tasks to create complex workflows.
+- **CLI & Server Mode:** Run tasks directly from the command line or through a user-friendly web UI.
+- **Flexible Input Handling:** Defaults, prompts, and command-line parameters to suit any workflow.
+- **Extensible & Open Source:** Contribute, customize, or extend Zrb to fit your unique needs.
+
+---
+
+## 🛠️ Getting Started
+
+### Quick Installation
+
+Install Zrb via pip:
+
+```bash
+pip install zrb
+
+```
+
+Or run our installation script to set up Zrb along with all prerequisites:
+
+```bash
+bash -c "$(curl -fsSL https://raw.githubusercontent.com/state-alchemists/zrb/main/install.sh)"
+
+```
+
+### Your First Task
+
+Create a file at `/home/<your-user-name>/zrb_init.py` with the following content:
+
+
+```python
+import os
+from zrb import cli, llm_config, LLMTask, CmdTask, StrInput, Group
+from zrb.builtin.llm.tool.file import read_all_files, write_text_file
+
+CURRENT_DIR = os.getcwd()
+
+# Make UML group
+uml_group = cli.add_group(Group(name="uml", description="UML related tasks"))
+
+# Generate UML script
+make_uml_script = uml_group.add_task(
+    LLMTask(
+        name="make-script",
+        description="Creating plantuml diagram based on source code in current directory",
+        input=StrInput(name="diagram", default="state diagram"),
+        message=(
+            f"Read source code in {CURRENT_DIR}, "
+            "make a {ctx.input.diagram} in plantuml format. "
+            f"Write the script into {CURRENT_DIR}/{{ctx.input.diagram}}.uml"
+        ),
+        tools=[
+            read_all_files,
+            write_text_file,
+        ],
+    )
+)
+
+# Defining a Cmd Task to transform Plantuml script into a png image.
+make_uml_image = uml_group.add_task(
+    CmdTask(
+        name="make-image",
+        description="Creating png based on source code in current directory",
+        input=StrInput(name="diagram", default="state diagram"),
+        cmd="plantuml -tpng '{ctx.input.diagram}.uml'",
+        cwd=CURRENT_DIR,
+    )
+)
+
+# Making sure that make_png has make_uml as its dependency.
+make_uml_script >> make_uml_image
+```
+
+You have just define two automation tasks.
+
+The first one use LLM to read files in your current directory and create a `PlantUML script` on that directory.
+
+The second task turn the PlantUML script into a `*.png` file. The second task depends on the first task and both of them are located under the same group.
+
+You can run the tasks by invoking `zrb uml make-script` or `zrb uml make-image` respectively.
+
+When you run zrb, it automatically searches for a file named `zrb_init.py` starting from your current directory and moving upward through its parent directories. This design lets you set up common automation tasks in a central location—like placing a `zrb_init.py` in your home directory (`/home/<your-user>/zrb_init.py`)—so that your tasks are available across all your projects.
+
+Now, go to your project and create a state diagram:
+
+```bash
+git clone git@github.com:jjinux/gotetris.git
+cd gotetris
+zrb uml make-image --diagram "state diagram"
+```
+
+You can also invoke the task without specifying parameter.
+
+```bash
+zrb uml make-image
+```
+
+Once you do so, Zrb will ask you to provide the diagram type.
+
+```
+diagram [state diagram]:
+```
+
+You can just press enter if you want to use the default value (i.e., in this case `state diagram`).
+
+Finally, you can also serve the tasks via a Web UI interface by invoking the following command:
+
+```bash
+zrb server start
+```
+
+You will have a nice web interface running on `http://localhost:12123`
+
+![Zrb Web UI](https://raw.githubusercontent.com/state-alchemists/zrb/main/_images/zrb-web-ui.png)
+
+Now, let's see how things work in detail. First, Zrb generates a `state diagram.uml` in your current directory, it then transform the UML script into a PNG image `state diagram.png`.
+
+![State Diagram](https://raw.githubusercontent.com/state-alchemists/zrb/main/_images/state-diagram.png)
+
+
+# 🎥 Demo & Documentation
+
+- **Step by step guide:** [Getting started with Zrb](https://github.com/state-alchemists/zrb/blob/main/docs/recipes/getting-started/README.md).
+- **Full documentation:** [Zrb Documentation](https://github.com/state-alchemists/zrb/blob/main/docs/README.md)
+- **Video demo:** [![Video Title](https://img.youtube.com/vi/W7dgk96l__o/0.jpg)](https://www.youtube.com/watch?v=W7dgk96l__o)
+
+
+# 🤝 Join the Community
+
+- **Bug Reports & Feature Requests:** Create an [issue](https://github.com/state-alchemists/zrb/issues) on Zrb's GitHub Repositories and include:
+    - Your Zrb version (i.e., `zrb version`).
+    - Steps you’ve taken and what you expected versus what happened
+- **Contributions:** We welcome pull requests! Check out our [contribution guidelines](https://github.com/state-alchemists/zrb/pulls).
+
+
+# ☕ Support The Project
+
+If you find Zrb valuable, please consider donating:
+
+[![](https://raw.githubusercontent.com/state-alchemists/zrb/main/_images/donator.png)](https://stalchmst.com/donation)
+
+# 🎉 Fun Fact
+
+Did you know?
+
+Zrb is named after `Zaruba`, a powerful support tool from the Garo universe!
+
+> Madou Ring Zaruba (魔導輪ザルバ, Madōrin Zaruba) is a Madougu which supports bearers of the Garo Armor. [(Garo Wiki | Fandom)](https://garo.fandom.com/wiki/Zaruba)
+
+![Madou Ring Zaruba on Kouga's Hand](https://raw.githubusercontent.com/state-alchemists/zrb/main/_images/madou-ring-zaruba.jpg)
+

{zrb-1.3.1.dist-info → zrb-1.4.1.dist-info}/RECORD

@@ -1,5 +1,5 @@
-zrb/__init__.py,sha256=JYLyBeSv-FP2iVKgsXJH8Ae-Cmjp5nmmIiwqayhCOEE,2964
-zrb/__main__.py,sha256=QcMnHfAFbDUFw9p9tgfFS4U0Ra9nE-TLU5YoMBiAriE,808
+zrb/__init__.py,sha256=1waPjZcA3IHUEvIuVQso0YfNfW9i7SCJgEfzhiNTaCk,3020
+zrb/__main__.py,sha256=MvAGzoM3ElJZOPMKNqaTdnrT9PXi9Saq8CPa11RiLQk,1748
 zrb/attr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/attr/type.py,sha256=4TV5gPYMMrKh5V-yB6iRYKCbsXAH_AvGXMsjxKLHcUs,568
 zrb/builtin/__init__.py,sha256=oXG4Zm_rIp3G81Y7hiSe38jeS2sGZAnADoP_yxxhYEc,1926
@@ -7,11 +7,11 @@ zrb/builtin/base64.py,sha256=1YnSwASp7OEAvQcsnHZGpJEvYoI1Z2zTIJ1bCDHfcPQ,921
 zrb/builtin/git.py,sha256=8_qVE_2lVQEVXQ9vhiw8Tn4Prj1VZB78ZjEJJS5Ab3M,5461
 zrb/builtin/git_subtree.py,sha256=7BKwOkVTWDrR0DXXQ4iJyHqeR6sV5VYRt8y_rEB0EHg,3505
 zrb/builtin/group.py,sha256=-phJfVpTX3_gUwS1u8-RbZUHe-X41kxDBSmrVh4rq8E,1682
-zrb/builtin/llm/llm_chat.py,sha256=yiL16XRy4oh5jKMzH0W7RBxfEwGKflvvrsYKleJFKWc,6221
+zrb/builtin/llm/llm_chat.py,sha256=QCfxocM7UQPtpIWLMzr9wKbl9DCPcDZszAnPxszaww0,6071
 zrb/builtin/llm/previous-session.js,sha256=xMKZvJoAbrwiyHS0OoPrWuaKxWYLoyR5sguePIoCjTY,816
 zrb/builtin/llm/tool/api.py,sha256=bXFE7jihdhUscxJH8lu5imwlYH735AalbCyUTl28BaQ,826
 zrb/builtin/llm/tool/cli.py,sha256=to_IjkfrMGs6eLfG0cpVN9oyADWYsJQCtyluUhUdBww,253
-zrb/builtin/llm/tool/file.py,sha256=ibvh0zrsnponwyZvw6bWMUbpwSv5S5WUWCDfQ6BjVwk,1160
+zrb/builtin/llm/tool/file.py,sha256=8H_qXt4imsy28DD21wBIo4ud5zfnkFjLLm8Simo84Q8,4732
 zrb/builtin/llm/tool/rag.py,sha256=vEIThEy0JGwXEiNRLOEJAHAE0l1Qie2qvU3ryioeYMk,6066
 zrb/builtin/llm/tool/web.py,sha256=SDnCtYHZ0Q4DtLbIhc11a0UyyKbTTeW60UfeIKzK35k,3204
 zrb/builtin/md5.py,sha256=0pNlrfZA0wlZlHvFHLgyqN0JZJWGKQIF5oXxO44_OJk,949
@@ -208,7 +208,7 @@ zrb/callback/callback.py,sha256=hKefB_Jd1XGjPSLQdMKDsGLHPzEGO2dqrIArLl_EmD0,848
 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.py,sha256=MfHwcQ4OhCmCw6jXpFI8483Ase6YrqNGBvqYzwnwopw,4753
+zrb/config.py,sha256=X0mlhmpUrYp_l4qI3CnsqOAfvxfLkteCOV9ABGF--Qc,4690
 zrb/content_transformer/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/content_transformer/any_content_transformer.py,sha256=v8ZUbcix1GGeDQwB6OKX_1TjpY__ksxWVeqibwa_iZA,850
 zrb/content_transformer/content_transformer.py,sha256=STl77wW-I69QaGzCXjvkppngYFLufow8ybPLSyAvlHs,2404
@@ -237,8 +237,9 @@ zrb/input/option_input.py,sha256=TQB82ko5odgzkULEizBZi0e9TIHEbIgvdP0AR3RhA74,213
 zrb/input/password_input.py,sha256=szBojWxSP9QJecgsgA87OIYwQrY2AQ3USIKdDZY6snU,1465
 zrb/input/str_input.py,sha256=NevZHX9rf1g8eMatPyy-kUX3DglrVAQpzvVpKAzf7bA,81
 zrb/input/text_input.py,sha256=shvVbc2U8Is36h23M5lcW8IEwKc9FR-4uEPZZroj3rU,3377
+zrb/llm_config.py,sha256=SXSkDpmXxGLJaoUrT09oNdOGwHXc82TwIGssVeo6S7U,1553
 zrb/runner/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-zrb/runner/cli.py,sha256=HtfJQecFg2keIMK-7bVATBlTeZC_RvT8DsbUC58TMKU,6736
+zrb/runner/cli.py,sha256=0mT0oO_yEhc8N4nYCJNujhgLjVykZ0B-kAOFXyAvAqM,6672
 zrb/runner/common_util.py,sha256=0zhZn1Jdmr194_nsL5_L-Kn9-_NDpMTI2z6_LXUQJ-U,1369
 zrb/runner/web_app.py,sha256=Ji2AWeFpJu5guXmur7mAAbjMToyjgmPDdfYu8047FFI,2616
 zrb/runner/web_config/config.py,sha256=0wR58KreAmawGGfamm0GLZY344HaXs7qfDgHLavBDwo,3125
@@ -299,7 +300,7 @@ zrb/task/base_task.py,sha256=SQRf37bylS586KwyW0eYDe9JZ5Hl18FP8kScHae6y3A,21251
 zrb/task/base_trigger.py,sha256=jC722rDvodaBLeNaFghkTyv1u0QXrK6BLZUUqcmBJ7Q,4581
 zrb/task/cmd_task.py,sha256=pUKRSR4DZKjbmluB6vi7cxqyhxOLfJ2czSpYeQbiDvo,10705
 zrb/task/http_check.py,sha256=Gf5rOB2Se2EdizuN9rp65HpGmfZkGc-clIAlHmPVehs,2565
-zrb/task/llm_task.py,sha256=B4qhza-4fk7odI7-rv2rLYvBLt1dmZMNgKu8OK7rajM,11849
+zrb/task/llm_task.py,sha256=Gf_Y8e3-U46wjnH5K36I1XJnFwwU-eTQlG5JL87UobM,13495
 zrb/task/make_task.py,sha256=PD3b_aYazthS8LHeJsLAhwKDEgdurQZpymJDKeN60u0,2265
 zrb/task/rsync_task.py,sha256=GSL9144bmp6F0EckT6m-2a1xG25AzrrWYzH4k3SVUKM,6370
 zrb/task/scaffolder.py,sha256=rME18w1HJUHXgi9eTYXx_T2G4JdqDYzBoNOkdOOo5-o,6806
@@ -331,7 +332,7 @@ zrb/util/file.py,sha256=mgNobIKCr0eIQUlg6W2Yg1fvg943VyuOUF8WMFpJA5A,859
 zrb/util/git.py,sha256=TShnMxPAk20Tglp25d_XPVZX-q0mvKeqdprVMeXQ5f0,4787
 zrb/util/git_subtree.py,sha256=zyWl0aUEZJyUJKjfw1uglozB4R1kF9pWtfKjhu8DN44,2658
 zrb/util/group.py,sha256=Bg7HrSycoK110U5s_Tca6-uUQuZ5CMgb8wxZSrvDQ98,2790
-zrb/util/load.py,sha256=m0e5DVLV7_RON6AZdjkquPym6BBrvYBbDMrllV44Y_k,1993
+zrb/util/load.py,sha256=Aeyh1EWtp-oJGVAhcjZn-VSB9innoOe8ZkUawha_ddk,1339
 zrb/util/run.py,sha256=DGHUP9x1Q8V8UF3FbpmjLGuhVVCCLfjTH2teT8qXlNI,207
 zrb/util/string/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/util/string/conversion.py,sha256=636sfF1a3_bpzXNw5bdSzbJzwakyAoo70KT3_ItgZEo,4333
@@ -340,7 +341,7 @@ zrb/util/string/name.py,sha256=8picJfUBXNpdh64GNaHv3om23QHhUZux7DguFLrXHp8,1163
 zrb/util/todo.py,sha256=1nDdwPc22oFoK_1ZTXyf3638Bg6sqE2yp_U4_-frHoc,16015
 zrb/xcom/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/xcom/xcom.py,sha256=o79rxR9wphnShrcIushA0Qt71d_p3ZTxjNf7x9hJB78,1571
-zrb-1.3.1.dist-info/METADATA,sha256=xYoLV2fDDSGBjJdgYOjXNy1rfAifx7VsGxIAGHCpBns,6303
-zrb-1.3.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-zrb-1.3.1.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
-zrb-1.3.1.dist-info/RECORD,,
+zrb-1.4.1.dist-info/METADATA,sha256=qKftyv7rxRUk1eiIo5hUOZJY80melIT4RpgAPXVa1Ew,8135
+zrb-1.4.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+zrb-1.4.1.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
+zrb-1.4.1.dist-info/RECORD,,

zrb-1.3.1.dist-info/METADATA

@@ -1,175 +0,0 @@
-Metadata-Version: 2.1
-Name: zrb
-Version: 1.3.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
-Author-email: gofrendiasgard@gmail.com
-Requires-Python: >=3.10.0,<4.0.0
-Classifier: License :: OSI Approved :: GNU Affero General Public License v3 or later (AGPLv3+)
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Provides-Extra: rag
-Requires-Dist: autopep8 (>=2.0.4,<3.0.0)
-Requires-Dist: beautifulsoup4 (>=4.12.3,<5.0.0)
-Requires-Dist: black (>=24.10.0,<24.11.0)
-Requires-Dist: chromadb (>=0.5.20,<0.6.0) ; extra == "rag"
-Requires-Dist: fastapi[standard] (>=0.115.6,<0.116.0)
-Requires-Dist: fastembed (>=0.5.1,<0.6.0)
-Requires-Dist: isort (>=5.13.2,<5.14.0)
-Requires-Dist: libcst (>=1.5.0,<2.0.0)
-Requires-Dist: pdfplumber (>=0.11.4,<0.12.0) ; extra == "rag"
-Requires-Dist: psutil (>=6.1.1,<7.0.0)
-Requires-Dist: pydantic-ai (>=0.0.31,<0.0.32)
-Requires-Dist: python-dotenv (>=1.0.1,<2.0.0)
-Requires-Dist: python-jose[cryptography] (>=3.4.0,<4.0.0)
-Requires-Dist: requests (>=2.32.3,<3.0.0)
-Requires-Dist: ulid-py (>=1.1.0,<2.0.0)
-Project-URL: Documentation, https://github.com/state-alchemists/zrb
-Project-URL: Repository, https://github.com/state-alchemists/zrb
-Description-Content-Type: text/markdown
-
-![](https://raw.githubusercontent.com/state-alchemists/zrb/main/_images/zrb/android-chrome-192x192.png)
-
-[Documentation](https://github.com/state-alchemists/zrb/blob/main/docs/README.md)
-
-# 🤖 Zrb: Your Automation Powerhouse
-
-Zrb allows you to write your automation tasks in Python. For example, you can define the following script in your home directory (`/home/<your-user-name>/zrb_init.py`).
-
-
-```python
-import os
-from zrb import cli, LLMTask, CmdTask, StrInput
-from zrb.builtin.llm.tool.file import read_source_code, write_text_file
-from pydantic_ai.models.openai import OpenAIModel
-
-
-CURRENT_DIR = os.getcwd()
-OPENROUTER_BASE_URL = "https://openrouter.ai/api/v1"
-OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_KEY", "")
-OPENROUTER_MODEL_NAME = os.getenv(
-    "AGENT_MODEL_NAME", "anthropic/claude-3.7-sonnet"
-)
-
-
-# Defining a LLM Task to create a Plantuml script based on source code in current directory.
-# User can choose the diagram type. By default it is "state diagram"
-make_uml = cli.add_task(
-    LLMTask(
-        name="make-uml",
-        description="Creating plantuml diagram based on source code in current directory",
-        input=StrInput(name="diagram", default="state diagram"),
-        model=OpenAIModel(
-            OPENROUTER_MODEL_NAME,
-            base_url=OPENROUTER_BASE_URL,
-            api_key=OPENROUTER_API_KEY,
-        ),
-        message=(
-            f"Read source code in {CURRENT_DIR}, "
-            "make a {ctx.input.diagram} in plantuml format. "
-            f"Write the script into {CURRENT_DIR}/{{ctx.input.diagram}}.uml"
-        ),
-        tools=[
-            read_source_code,
-            write_text_file,
-        ],
-    )
-)
-
-# Defining a Cmd Task to transform Plantuml script into a png image.
-make_png = cli.add_task(
-    CmdTask(
-        name="make-png",
-        description="Creating png based on source code in current directory",
-        input=StrInput(name="diagram", default="state diagram"),
-        cmd="plantuml -tpng '{ctx.input.diagram}.uml'",
-        cwd=CURRENT_DIR,
-    )
-)
-
-# Making sure that make_png has make_uml as its dependency.
-make_uml >> make_png
-```
-
-Once defined, your automation tasks are immediately accessible from the CLI. You can then invoke the tasks by invoking.
-
-```bash
-zrb make-png --diagram "state diagram"
-```
-
-Or you can invoke the tasks without parameter.
-
-```bash
-zrb make-png
-```
-
-At this point, Zrb will politely ask you to provide the diagram type.
-
-```
-diagram [state diagram]:
-```
-
-You can just press enter if you want to use the default value.
-
-Finally, you can run Zrb as a server and make your tasks available for non technical users by invoking the following command.
-
-```bash
-zrb server start
-```
-
-You will have a nice web interface running on `http://localhost:12123`
-
-![Zrb Web UI](https://raw.githubusercontent.com/state-alchemists/zrb/main/_images/zrb-web-ui.png)
-
-Now, let's see how Zrb generate the state diagram. Based on the source code in your current directory, Zrb will generate a `state diagram.uml` and transform it into `state diagram.png`.
-
-![State Diagram](https://raw.githubusercontent.com/state-alchemists/zrb/main/_images/state-diagram.png)
-
-See the [getting started guide](https://github.com/state-alchemists/zrb/blob/main/docs/recipes/getting-started/README.md) for more information. Or just watch the demo:
-
-[![Video Title](https://img.youtube.com/vi/W7dgk96l__o/0.jpg)](https://www.youtube.com/watch?v=W7dgk96l__o)
-
-
-# 🫰 Installing Zrb
-
-You can install Zrb as a pip package by invoking the following command:
-
-```bash
-pip install --pre zrb
-```
-
-Alternatively, you can also use our installation script to install Zrb along with some prerequisites:
-
-```bash
-bash -c "$(curl -fsSL https://raw.githubusercontent.com/state-alchemists/zrb/main/install.sh)"
-```
-
-# 🐞 Bug Report + Feature Request
-
-You can submit bug reports and feature requests by creating a new [issue](https://github.com/state-alchemists/zrb/issues) on Zrb's GitHub Repositories. When reporting a bug or requesting a feature, please be sure to:
-
-- Include the version of Zrb you are using (i.e., `zrb version`)
-- Tell us what you have tried
-- Tell us what you expect
-- Tell us what you get
-
-We will also welcome your [pull requests and contributions](https://github.com/state-alchemists/zrb/pulls).
-
-
-# ☕ Donation
-
-Help Red Skull to click the donation button:
-
-[![](https://raw.githubusercontent.com/state-alchemists/zrb/main/_images/donator.png)](https://stalchmst.com/donation)
-
-# 🎉 Fun Fact
-
-> Madou Ring Zaruba (魔導輪ザルバ, Madōrin Zaruba) is a Madougu which supports bearers of the Garo Armor. [(Garo Wiki | Fandom)](https://garo.fandom.com/wiki/Zaruba)
-
-![Madou Ring Zaruba on Kouga's Hand](https://raw.githubusercontent.com/state-alchemists/zrb/main/_images/madou-ring-zaruba.jpg)
-