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

zrb/builtin/llm/llm_chat.py

@@ -22,7 +22,6 @@ from zrb.config import (
     LLM_ALLOW_ACCESS_LOCAL_FILE,
     LLM_ALLOW_ACCESS_SHELL,
     LLM_HISTORY_DIR,
-    LLM_SYSTEM_PROMPT,
     SERP_API_KEY,
 )
 from zrb.context.any_shared_context import AnySharedContext
@@ -119,7 +118,7 @@ llm_chat: LLMTask = llm_group.add_task(
                 "system-prompt",
                 description="System prompt",
                 prompt="System prompt",
-                default=LLM_SYSTEM_PROMPT,
+                default="",
                 allow_positional_parsing=False,
                 always_prompt=False,
             ),
@@ -141,17 +140,19 @@ llm_chat: LLMTask = llm_group.add_task(
                 always_prompt=False,
             ),
         ],
-        model=lambda ctx: None if ctx.input.model == "" else ctx.input.model,
+        model=lambda ctx: None if ctx.input.model.strip() == "" else ctx.input.model,
         model_base_url=lambda ctx: (
-            None if ctx.input.base_url == "" else ctx.input.base_url
+            None if ctx.input.base_url.strip() == "" else ctx.input.base_url
         ),
         model_api_key=lambda ctx: (
-            None if ctx.input.api_key == "" else ctx.input.api_key
+            None if ctx.input.api_key.strip() == "" 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=lambda ctx: (
+            None if ctx.input.system_prompt.strip() == "" else ctx.input.system_prompt
+        ),
         message="{ctx.input.message}",
         retries=0,
     ),

zrb/builtin/llm/tool/api.py

@@ -5,7 +5,7 @@ from typing import Annotated, Literal
 def get_current_location() -> (
     Annotated[str, "JSON string representing latitude and longitude"]
 ):  # noqa
-    """Get the user's current location."""
+    """Get the user's current location. This function take no argument."""
     import requests
 
     return json.dumps(requests.get("http://ip-api.com/json?fields=lat,lon").json())

zrb/builtin/llm/tool/file.py

@@ -93,6 +93,7 @@ def list_files(
     """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.
+    included_patterns and excluded_patterns already has sane default values.
     """
     all_files: list[str] = []
     for root, dirs, files in os.walk(directory):
@@ -127,12 +128,12 @@ def _should_exclude(full_path: str, excluded_patterns: list[str]) -> bool:
 
 
 def read_text_file(file: str) -> str:
-    """Read a text file"""
+    """Read a text file and return a string containing the file content."""
     return read_file(os.path.abspath(file))
 
 
 def write_text_file(file: str, content: str):
-    """Write a text file"""
+    """Write content to a text file"""
     return write_file(os.path.abspath(file), content)
 
 
@@ -144,6 +145,7 @@ def read_all_files(
     """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.
+    included_patterns and excluded_patterns already has sane default values.
     """
     files = list_files(directory, included_patterns, excluded_patterns)
     for index, file in enumerate(files):

zrb/builtin/llm/tool/rag.py

@@ -9,6 +9,8 @@ import ulid
 
 from zrb.config import (
     RAG_CHUNK_SIZE,
+    RAG_EMBEDDING_API_KEY,
+    RAG_EMBEDDING_BASE_URL,
     RAG_EMBEDDING_MODEL,
     RAG_MAX_RESULT_COUNT,
     RAG_OVERLAP,
@@ -35,24 +37,34 @@ def create_rag_from_directory(
     tool_name: str,
     tool_description: str,
     document_dir_path: str = "./documents",
-    model: str = RAG_EMBEDDING_MODEL,
     vector_db_path: str = "./chroma",
     vector_db_collection: str = "documents",
     chunk_size: int = RAG_CHUNK_SIZE,
     overlap: int = RAG_OVERLAP,
     max_result_count: int = RAG_MAX_RESULT_COUNT,
     file_reader: list[RAGFileReader] = [],
+    openai_api_key: str = RAG_EMBEDDING_API_KEY,
+    openai_base_url: str = RAG_EMBEDDING_BASE_URL,
+    openai_embedding_model: str = RAG_EMBEDDING_MODEL,
 ):
     async def retrieve(query: str) -> str:
         from chromadb import PersistentClient
         from chromadb.config import Settings
-        from fastembed import TextEmbedding
-
-        embedding_model = TextEmbedding(model_name=model)
-        client = PersistentClient(
+        from openai import OpenAI
+
+        # Initialize OpenAI client with custom URL if provided
+        client_args = {}
+        if openai_api_key:
+            client_args["api_key"] = openai_api_key
+        if openai_base_url:
+            client_args["base_url"] = openai_base_url
+        # Initialize OpenAI client for embeddings
+        openai_client = OpenAI(**client_args)
+        # Initialize ChromaDB client
+        chroma_client = PersistentClient(
             path=vector_db_path, settings=Settings(allow_reset=True)
         )
-        collection = client.get_or_create_collection(vector_db_collection)
+        collection = chroma_client.get_or_create_collection(vector_db_collection)
         # Track file changes using a hash-based approach
         hash_file_path = os.path.join(vector_db_path, "file_hashes.json")
         previous_hashes = _load_hashes(hash_file_path)
@@ -89,8 +101,11 @@ def create_rag_from_directory(
                                 ),
                                 file=sys.stderr,
                             )
-                            embedding_result = list(embedding_model.embed([chunk]))
-                            vector = embedding_result[0]
+                            # Get embeddings using OpenAI
+                            embedding_response = openai_client.embeddings.create(
+                                input=chunk, model=openai_embedding_model
+                            )
+                            vector = embedding_response.data[0].embedding
                             collection.upsert(
                                 ids=[chunk_id],
                                 embeddings=[vector],
@@ -113,8 +128,11 @@ def create_rag_from_directory(
             )
         # Vectorize query and get related document chunks
         print(stylize_faint("Vectorizing query"), file=sys.stderr)
-        embedding_result = list(embedding_model.embed([query]))
-        query_vector = embedding_result[0]
+        # Get embeddings using OpenAI
+        embedding_response = openai_client.embeddings.create(
+            input=query, model=openai_embedding_model
+        )
+        query_vector = embedding_response.data[0].embedding
         print(stylize_faint("Searching documents"), file=sys.stderr)
         results = collection.query(
             query_embeddings=query_vector,

zrb/builtin/llm/tool/web.py

@@ -3,21 +3,61 @@ from collections.abc import Callable
 from typing import Annotated
 
 
-def open_web_page(url: str) -> str:
-    """Get content from a web page."""
-    import requests
-
-    response = requests.get(
-        url,
-        headers={
-            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36"  # noqa
-        },
-    )
-    if response.status_code != 200:
-        raise Exception(
-            f"Error: Unable to retrieve search results (status code: {response.status_code})"  # noqa
-        )
-    return json.dumps(parse_html_text(response.text))
+async def open_web_page(url: str) -> str:
+    """Get content from a web page using a headless browser."""
+
+    async def get_page_content(page_url: str):
+        try:
+            from playwright.async_api import async_playwright
+
+            async with async_playwright() as p:
+                browser = await p.chromium.launch(headless=True)
+                page = await browser.new_page()
+                # Set user agent to mimic a regular browser
+                user_agent = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) "
+                user_agent += "AppleWebKit/537.36 (KHTML, like Gecko) "
+                user_agent += "Chrome/91.0.4472.124 Safari/537.36"
+                await page.set_extra_http_headers({"User-Agent": user_agent})
+                try:
+                    # Navigate to the URL with a timeout of 30 seconds
+                    await page.goto(page_url, wait_until="networkidle", timeout=30000)
+                    # Wait for the content to load
+                    await page.wait_for_load_state("domcontentloaded")
+                    # Get the page content
+                    content = await page.content()
+                    # Extract all links from the page
+                    links = await page.eval_on_selector_all(
+                        "a[href]",
+                        """
+                        (elements) => elements.map(el => {
+                            const href = el.getAttribute('href');
+                            if (href && !href.startsWith('#') && !href.startsWith('/')) {
+                                return href;
+                            }
+                            return null;
+                        }).filter(href => href !== null)
+                    """,
+                    )
+                    return {"content": content, "links_on_page": links}
+                finally:
+                    await browser.close()
+        except ImportError:
+            import requests
+
+            response = requests.get(
+                url,
+                headers={
+                    "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36"  # noqa
+                },
+            )
+            if response.status_code != 200:
+                msg = f"Unable to retrieve search results. Status code: {response.status_code}"
+                raise Exception(msg)
+            return {"content": response.text, "links_on_page": []}
+
+    result = await get_page_content(url)
+    # Parse the HTML content
+    return json.dumps(parse_html_text(result["content"]))
 
 
 def create_search_internet_tool(serp_api_key: str) -> Callable[[str, int], str]:

zrb/config.py

@@ -76,15 +76,6 @@ WEB_AUTH_REFRESH_TOKEN_EXPIRE_MINUTES = int(
     os.getenv("ZRB_WEB_REFRESH_TOKEN_EXPIRE_MINUTES", "60")
 )
 
-_DEFAULT_PROMPT = (
-    "You are a helpful AI assistant capable of using various tools to answer user queries. When solving a problem:\n"
-    "1. Carefully analyze the user's request and identify what information is needed to provide a complete answer.\n"
-    "2. Determine which available tools can help you gather the necessary information.\n"
-    "3. Call tools strategically and in a logical sequence to collect required data.\n"
-    "4. If a tool provides incomplete information, intelligently decide which additional tool or approach to use.\n"
-    "5. Always aim to provide the most accurate and helpful response possible."
-)
-LLM_SYSTEM_PROMPT = os.getenv("ZRB_LLM_SYSTEM_PROMPT", _DEFAULT_PROMPT)
 LLM_HISTORY_DIR = os.getenv(
     "ZRB_LLM_HISTORY_DIR", os.path.expanduser(os.path.join("~", ".zrb-llm-history"))
 )
@@ -94,10 +85,10 @@ LLM_HISTORY_FILE = os.getenv(
 LLM_ALLOW_ACCESS_LOCAL_FILE = to_boolean(os.getenv("ZRB_LLM_ACCESS_LOCAL_FILE", "1"))
 LLM_ALLOW_ACCESS_SHELL = to_boolean(os.getenv("ZRB_LLM_ACCESS_SHELL", "1"))
 LLM_ALLOW_ACCESS_INTERNET = to_boolean(os.getenv("ZRB_LLM_ACCESS_INTERNET", "1"))
-# noqa See: https://qdrant.github.io/fastembed/examples/Supported_Models/#supported-text-embedding-models
-RAG_EMBEDDING_MODEL = os.getenv(
-    "ZRB_RAG_EMBEDDING_MODEL", "nomic-ai/nomic-embed-text-v1.5-Q"
-)
+# RAG Configuration
+RAG_EMBEDDING_API_KEY = os.getenv("ZRB_RAG_EMBEDDING_API_KEY", None)
+RAG_EMBEDDING_BASE_URL = os.getenv("ZRB_RAG_EMBEDDING_BASE_URL", None)
+RAG_EMBEDDING_MODEL = os.getenv("ZRB_RAG_EMBEDDING_MODEL", "text-embedding-ada-002")
 RAG_CHUNK_SIZE = int(os.getenv("ZRB_RAG_CHUNK_SIZE", "1024"))
 RAG_OVERLAP = int(os.getenv("ZRB_RAG_OVERLAP", "128"))
 RAG_MAX_RESULT_COUNT = int(os.getenv("ZRB_RAG_MAX_RESULT_COUNT", "5"))

zrb/llm_config.py

@@ -2,35 +2,72 @@ import os
 
 from pydantic_ai.models import Model
 from pydantic_ai.models.openai import OpenAIModel
+from pydantic_ai.providers import Provider
 from pydantic_ai.providers.openai import OpenAIProvider
 
+DEFAULT_SYSTEM_PROMPT = """
+You have access to tools.
+Your goal is to answer user queries accurately.
+Follow these instructions precisely:
+1. ALWAYS use available tools to gather information BEFORE asking the user questions
+2. For tools that require arguments: provide arguments in valid JSON format
+3. For tools with no args: call the tool without args. Do NOT pass "" or {}.
+4. NEVER pass arguments to tools that don't accept parameters
+5. NEVER ask users for information obtainable through tools
+6. Use tools in a logical sequence until you have sufficient information
+7. If a tool call fails, check if you're passing arguments in the correct format
+8. Only after exhausting relevant tools should you request clarification
+""".strip()
+
 
 class LLMConfig:
 
     def __init__(
         self,
-        model_name: str | None = None,
-        base_url: str | None = None,
-        api_key: str | None = None,
+        default_model_name: str | None = None,
+        default_base_url: str | None = None,
+        default_api_key: str | None = None,
+        default_system_prompt: str | None = None,
     ):
         self._model_name = (
-            model_name if model_name is not None else os.getenv("ZRB_LLM_MODEL", None)
+            default_model_name
+            if default_model_name is not None
+            else os.getenv("ZRB_LLM_MODEL", None)
+        )
+        self._model_base_url = (
+            default_base_url
+            if default_base_url is not None
+            else os.getenv("ZRB_LLM_BASE_URL", None)
         )
-        self._base_url = (
-            base_url if base_url is not None else os.getenv("ZRB_LLM_BASE_URL", None)
+        self._model_api_key = (
+            default_api_key
+            if default_api_key is not None
+            else os.getenv("ZRB_LLM_API_KEY", None)
         )
-        self._api_key = (
-            api_key if api_key is not None else os.getenv("ZRB_LLM_API_KEY", None)
+        self._system_prompt = (
+            default_system_prompt
+            if default_system_prompt is not None
+            else os.getenv("ZRB_LLM_SYSTEM_PROMPT", None)
         )
+        self._default_provider = 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:
+    def get_default_model_provider(self) -> Provider | str:
+        if self._default_provider is not None:
+            return self._default_provider
+        if self._model_base_url is None and self._model_api_key is None:
             return "openai"
-        return OpenAIProvider(base_url=self._base_url, api_key=self._api_key)
+        return OpenAIProvider(
+            base_url=self._model_base_url, api_key=self._model_api_key
+        )
+
+    def get_default_system_prompt(self) -> str:
+        if self._system_prompt is not None:
+            return self._system_prompt
+        return DEFAULT_SYSTEM_PROMPT
 
     def get_default_model(self) -> Model | str | None:
         if self._default_model is not None:
@@ -40,9 +77,24 @@ class LLMConfig:
             return None
         return OpenAIModel(
             model_name=model_name,
-            provider=self._get_model_provider(),
+            provider=self.get_default_model_provider(),
         )
 
+    def set_default_system_prompt(self, system_prompt: str):
+        self._system_prompt = system_prompt
+
+    def set_default_model_name(self, model_name: str):
+        self._model_name = model_name
+
+    def set_default_model_api_key(self, model_api_key: str):
+        self._model_api_key = model_api_key
+
+    def set_default_model_base_url(self, model_base_url: str):
+        self._model_base_url = model_base_url
+
+    def set_default_provider(self, provider: Provider | str):
+        self._default_provider = provider
+
     def set_default_model(self, model: Model | str | None):
         self._default_model = model
 

zrb/task/llm_task.py

@@ -1,5 +1,8 @@
+import functools
+import inspect
 import json
 import os
+import traceback
 from collections.abc import Callable
 from typing import Any
 
@@ -18,7 +21,6 @@ from pydantic_ai.models import Model
 from pydantic_ai.settings import ModelSettings
 
 from zrb.attr.type import StrAttr, fstring
-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
@@ -58,7 +60,7 @@ class LLMTask(BaseTask):
             ModelSettings | Callable[[AnySharedContext], ModelSettings] | None
         ) = None,
         agent: Agent | Callable[[AnySharedContext], Agent] | None = None,
-        system_prompt: StrAttr | None = LLM_SYSTEM_PROMPT,
+        system_prompt: StrAttr | None = None,
         render_system_prompt: bool = True,
         message: StrAttr | None = None,
         tools: (
@@ -202,6 +204,9 @@ class LLMTask(BaseTask):
             async with node.stream(agent_run.ctx) as handle_stream:
                 async for event in handle_stream:
                     if isinstance(event, FunctionToolCallEvent):
+                        # Fixing anthrophic claude when call function with empty parameter
+                        if event.part.args == "":
+                            event.part.args = {}
                         ctx.print(
                             stylize_faint(
                                 f"[Tools] The LLM calls tool={event.part.tool_name!r} with args={event.part.args} (tool_call_id={event.part.tool_call_id!r})"  # noqa
@@ -241,7 +246,7 @@ class LLMTask(BaseTask):
         )
         tools_or_callables.extend(self._additional_tools)
         tools = [
-            tool if isinstance(tool, Tool) else Tool(tool, takes_ctx=False)
+            tool if isinstance(tool, Tool) else Tool(_wrap_tool(tool), takes_ctx=False)
             for tool in tools_or_callables
         ]
         return Agent(
@@ -257,21 +262,17 @@ class LLMTask(BaseTask):
         if model is None:
             return default_llm_config.get_default_model()
         if isinstance(model, str):
+            model_base_url = self._get_model_base_url(ctx)
+            model_api_key = self._get_model_api_key(ctx)
             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,
-                ),
+                default_model_name=model,
+                default_base_url=model_base_url,
+                default_api_key=model_api_key,
             )
+            if model_base_url is None and model_api_key is None:
+                default_model_provider = default_llm_config.get_default_model_provider()
+                if default_model_provider is not None:
+                    llm_config.set_default_provider(default_model_provider)
             return llm_config.get_default_model()
         raise ValueError(f"Invalid model: {model}")
 
@@ -289,15 +290,18 @@ class LLMTask(BaseTask):
         )
         if isinstance(api_key, str) or api_key is None:
             return api_key
-        raise ValueError(f"Invalid model base URL: {api_key}")
+        raise ValueError(f"Invalid model API key: {api_key}")
 
     def _get_system_prompt(self, ctx: AnyContext) -> str:
-        return get_str_attr(
+        system_prompt = get_attr(
             ctx,
             self._system_prompt,
-            "You are a helpful assistant",
+            None,
             auto_render=self._render_system_prompt,
         )
+        if system_prompt is not None:
+            return system_prompt
+        return default_llm_config.get_default_system_prompt()
 
     def _get_message(self, ctx: AnyContext) -> str:
         return get_str_attr(ctx, self._message, "How are you?", auto_render=True)
@@ -323,3 +327,16 @@ class LLMTask(BaseTask):
             "",
             auto_render=self._render_history_file,
         )
+
+
+def _wrap_tool(func):
+    @functools.wraps(func)
+    async def wrapper(*args, **kwargs):
+        try:
+            return await run_async(func(*args, **kwargs))
+        except Exception as e:
+            # Optionally, you can include more details from traceback if needed.
+            error_details = traceback.format_exc()
+            return f"Error: {e}\nDetails: {error_details}"
+
+    return wrapper

{zrb-1.4.1.dist-info → zrb-1.4.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: zrb
-Version: 1.4.1
+Version: 1.4.3
 Summary: Your Automation Powerhouse
 Home-page: https://github.com/state-alchemists/zrb
 License: AGPL-3.0-or-later
@@ -13,16 +13,20 @@ 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: all
+Provides-Extra: playwright
 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: chromadb (>=0.5.20,<0.6.0) ; extra == "rag" or extra == "all"
 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: openai (>=1.10.0,<2.0.0) ; extra == "rag" or extra == "all"
+Requires-Dist: pdfplumber (>=0.11.4,<0.12.0) ; extra == "rag" or extra == "all"
+Requires-Dist: playwright (>=1.43.0,<2.0.0) ; extra == "playwright" or extra == "all"
 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)
@@ -44,7 +48,6 @@ Description-Content-Type: text/markdown
 
 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?
 
@@ -54,7 +57,6 @@ Zrb streamlines repetitive tasks, integrates with powerful LLMs, and lets you cr
 - **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
 
@@ -64,11 +66,8 @@ Zrb streamlines repetitive tasks, integrates with powerful LLMs, and lets you cr
 - **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
+# 🛠️ Installation
 
 Install Zrb via pip:
 
@@ -84,40 +83,48 @@ bash -c "$(curl -fsSL https://raw.githubusercontent.com/state-alchemists/zrb/mai
 
 ```
 
-### Your First Task
+# 🍲 Quick Start: Build Your First Automation Workflow
+
+Zrb empowers you to create custom automation tasks using Python. This guide shows you how to define two simple tasks: one to generate a PlantUML script from your source code and another to convert that script into a PNG image.
+
+## 1. Create Your Task Definition File
 
-Create a file at `/home/<your-user-name>/zrb_init.py` with the following content:
+Place a file named `zrb_init.py` in a directory that's accessible from your projects. Zrb will automatically search for this file by starting in your current directory and then moving upward (i.e., checking parent directories) until it finds one. This means if you place your `zrb_init.py` in your home directory (e.g., `/home/<your-user-name>/zrb_init.py`), the tasks defined there will be available for any project.
 
+Add the following content to your zrb_init.py:
 
 ```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
+from zrb import cli, LLMTask, CmdTask, StrInput, Group
+from zrb.builtin.llm.tool.file import (
+    read_text_file, list_files, write_text_file
+)
 
 CURRENT_DIR = os.getcwd()
 
-# Make UML group
+# Create a group for UML-related tasks
 uml_group = cli.add_group(Group(name="uml", description="UML related tasks"))
 
-# Generate UML script
+# Task 1: Generate a PlantUML script from your source code
 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}, "
+            f"Read all necessary files 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,
+            list_files,
+            read_text_file,
             write_text_file,
         ],
     )
 )
 
-# Defining a Cmd Task to transform Plantuml script into a png image.
+# Task 2: Convert the PlantUML script into a PNG image
 make_uml_image = uml_group.add_task(
     CmdTask(
         name="make-image",
@@ -128,62 +135,75 @@ make_uml_image = uml_group.add_task(
     )
 )
 
-# Making sure that make_png has make_uml as its dependency.
+# Set up the dependency: the image task runs after the script is created
 make_uml_script >> make_uml_image
 ```
 
-You have just define two automation tasks.
+**What This Does**
 
-The first one use LLM to read files in your current directory and create a `PlantUML script` on that directory.
+- **Task 1 – make-script**:
 
-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.
+    Uses an LLM to read all files in your current directory and generate a PlantUML script (e.g., `state diagram.uml`).
 
-You can run the tasks by invoking `zrb uml make-script` or `zrb uml make-image` respectively.
+- **Task 2 – make-image**:
 
-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.
+    Executes a command that converts the PlantUML script into a PNG image (e.g., `state diagram.png`). This task will run only after the script has been generated.
 
-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"
-```
+## 2. Run Your Tasks
 
-You can also invoke the task without specifying parameter.
+After setting up your tasks, you can execute them from any project. For example:
 
-```bash
-zrb uml make-image
-```
+- Clone/Create a Project:
 
-Once you do so, Zrb will ask you to provide the diagram type.
+    ```bash
+    git clone git@github.com:jjinux/gotetris.git
+    cd gotetris
+    ```
 
-```
-diagram [state diagram]:
-```
+- Create a state diagram:
+
+    ```bash
+    zrb uml make-image --diagram "state diagram"
+    ```
+
+- Or use the interactive mode:
 
-You can just press enter if you want to use the default value (i.e., in this case `state diagram`).
+    ```bash
+    zrb uml make-image
+    ```
 
-Finally, you can also serve the tasks via a Web UI interface by invoking the following command:
+    Zrb will prompt:
+
+    ```bash
+    diagram [state diagram]:
+    ```
+
+    Press **Enter** to use the default value
+
+![State Diagram](https://raw.githubusercontent.com/state-alchemists/zrb/main/_images/state-diagram.png)
+
+
+## 3. Try Out the Web UI
+
+You can also serve your tasks through a user-friendly web interface:
 
 ```bash
 zrb server start
 ```
 
-You will have a nice web interface running on `http://localhost:12123`
+Then open your browser and visit `http://localhost:21213`
 
 ![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)
+- **Video demo:**
+
+    [![Video Title](https://img.youtube.com/vi/W7dgk96l__o/0.jpg)](https://www.youtube.com/watch?v=W7dgk96l__o)
 
 
 # 🤝 Join the Community

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

@@ -7,13 +7,13 @@ 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=QCfxocM7UQPtpIWLMzr9wKbl9DCPcDZszAnPxszaww0,6071
+zrb/builtin/llm/llm_chat.py,sha256=OwbeXNaskyufYIhbhLmj9JRYB9bw5D8JfntAzOhmrP8,6140
 zrb/builtin/llm/previous-session.js,sha256=xMKZvJoAbrwiyHS0OoPrWuaKxWYLoyR5sguePIoCjTY,816
-zrb/builtin/llm/tool/api.py,sha256=bXFE7jihdhUscxJH8lu5imwlYH735AalbCyUTl28BaQ,826
+zrb/builtin/llm/tool/api.py,sha256=U0_PhVuoDLpq4Jak5S45IHhCF1jKmfS0JC8XAnfnOhA,858
 zrb/builtin/llm/tool/cli.py,sha256=to_IjkfrMGs6eLfG0cpVN9oyADWYsJQCtyluUhUdBww,253
-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/llm/tool/file.py,sha256=v3gaAM442lVZju4LVXqABiIk_H4k5XPiw9JvXYAZbow,4946
+zrb/builtin/llm/tool/rag.py,sha256=pX8N_bYv4axsjhULLvvZtQYW2klZOkeQZ2Tn16083vM,6860
+zrb/builtin/llm/tool/web.py,sha256=LtaKU8BkV5HvKvSwOU59k99_kbbfeBQYS8fuP7l6fJ8,5248
 zrb/builtin/md5.py,sha256=0pNlrfZA0wlZlHvFHLgyqN0JZJWGKQIF5oXxO44_OJk,949
 zrb/builtin/project/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/builtin/project/add/fastapp/fastapp_input.py,sha256=MKlWR_LxWhM_DcULCtLfL_IjTxpDnDBkn9KIqNmajFs,310
@@ -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=X0mlhmpUrYp_l4qI3CnsqOAfvxfLkteCOV9ABGF--Qc,4690
+zrb/config.py,sha256=bYLagRHcReZBrfaQM3y5FaNCflDC9l8Kb6Sw49eD3-o,4023
 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,7 +237,7 @@ 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/llm_config.py,sha256=w_gO_rYUdj8u-lrX3JCZleB7X5PHAC_35ymn-7dYAQo,3601
 zrb/runner/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/runner/cli.py,sha256=0mT0oO_yEhc8N4nYCJNujhgLjVykZ0B-kAOFXyAvAqM,6672
 zrb/runner/common_util.py,sha256=0zhZn1Jdmr194_nsL5_L-Kn9-_NDpMTI2z6_LXUQJ-U,1369
@@ -300,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=Gf_Y8e3-U46wjnH5K36I1XJnFwwU-eTQlG5JL87UobM,13495
+zrb/task/llm_task.py,sha256=oPLruaSCqEF2fX5xtXfXflVYMknT2U5nBqC0CObT-ug,14282
 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
@@ -341,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.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.4.3.dist-info/METADATA,sha256=Ttrj5DxuqudtPbcEULSHZKbU0ithhP1JbMk1FhvCiiE,8557
+zrb-1.4.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+zrb-1.4.3.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
+zrb-1.4.3.dist-info/RECORD,,