zrb 1.2.1__py3-none-any.whl → 1.3.0__py3-none-any.whl

zrb/builtin/project/add/fastapp/fastapp_template/my_app_name/schema/user.py

@@ -51,7 +51,9 @@ class UserUpdate(SQLModel):
     active: bool | None = None
 
     def with_audit(self, updated_by: str) -> "UserUpdateWithAudit":
-        return UserUpdateWithAudit(**self.model_dump(), updated_by=updated_by)
+        return UserUpdateWithAudit(
+            **self.model_dump(exclude_none=True), updated_by=updated_by
+        )
 
 
 class UserUpdateWithAudit(UserUpdate):
@@ -62,7 +64,9 @@ class UserUpdateWithRoles(UserUpdate):
     role_names: list[str] | None = None
 
     def with_audit(self, updated_by: str) -> "UserUpdateWithRolesAndAudit":
-        return UserUpdateWithRolesAndAudit(**self.model_dump(), updated_by=updated_by)
+        return UserUpdateWithRolesAndAudit(
+            **self.model_dump(exclude_none=True), updated_by=updated_by
+        )
 
 
 class UserUpdateWithRolesAndAudit(UserUpdateWithRoles):
@@ -70,7 +74,9 @@ class UserUpdateWithRolesAndAudit(UserUpdateWithRoles):
 
     def get_user_update_with_audit(self) -> UserUpdateWithAudit:
         data = {
-            key: val for key, val in self.model_dump().items() if key != "role_names"
+            key: val
+            for key, val in self.model_dump(exclude_none=True).items()
+            if key != "role_names"
         }
         return UserUpdateWithAudit(**data)
 

zrb/config.py

@@ -92,7 +92,8 @@ LLM_HISTORY_DIR = os.getenv(
 LLM_HISTORY_FILE = os.getenv(
     "ZRB_LLM_HISTORY_FILE", os.path.join(LLM_HISTORY_DIR, "history.json")
 )
-LLM_ALLOW_ACCESS_SHELL = to_boolean(os.getenv("ZRB_LLM_ACCESS_FILE", "1"))
+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(
@@ -101,6 +102,7 @@ RAG_EMBEDDING_MODEL = os.getenv(
 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"))
+SERP_API_KEY = os.getenv("SERP_API_KEY", "")
 
 
 BANNER = f"""

zrb/content_transformer/content_transformer.py

@@ -1,4 +1,5 @@
 import fnmatch
+import os
 import re
 from collections.abc import Callable
 
@@ -40,7 +41,12 @@ class ContentTransformer(AnyContentTransformer):
                     return True
             except re.error:
                 pass
-            return fnmatch.fnmatch(file_path, pattern)
+            if os.sep not in pattern and (
+                os.altsep is None or os.altsep not in pattern
+            ):
+                # Pattern like "*.txt" – match only the basename.
+                return fnmatch.fnmatch(file_path, os.path.basename(file_path))
+            return fnmatch.fnmatch(file_path, file_path)
 
     def transform_file(self, ctx: AnyContext, file_path: str):
         if callable(self._transform_file):

zrb/context/context.py

@@ -88,7 +88,7 @@ class Context(AnyContext):
             return template
         return int(self.render(template))
 
-    def render_float(self, template: str) -> float:
+    def render_float(self, template: str | float) -> float:
         if isinstance(template, float):
             return template
         return float(self.render(template))
@@ -102,9 +102,10 @@ class Context(AnyContext):
         flush: bool = True,
         plain: bool = False,
     ):
+        sep = " " if sep is None else sep
         message = sep.join([f"{value}" for value in values])
         if plain:
-            self.append_to_shared_log(remove_style(message))
+            # self.append_to_shared_log(remove_style(message))
             print(message, sep=sep, end=end, file=file, flush=flush)
             return
         color = self._color
@@ -132,6 +133,7 @@ class Context(AnyContext):
         flush: bool = True,
     ):
         if self._shared_ctx.get_logging_level() <= logging.DEBUG:
+            sep = " " if sep is None else sep
             message = sep.join([f"{value}" for value in values])
             stylized_message = stylize_log(f"[DEBUG] {message}")
             self.print(stylized_message, sep=sep, end=end, file=file, flush=flush)
@@ -145,6 +147,7 @@ class Context(AnyContext):
         flush: bool = True,
     ):
         if self._shared_ctx.get_logging_level() <= logging.INFO:
+            sep = " " if sep is None else sep
             message = sep.join([f"{value}" for value in values])
             stylized_message = stylize_log(f"[INFO] {message}")
             self.print(stylized_message, sep=sep, end=end, file=file, flush=flush)
@@ -158,6 +161,7 @@ class Context(AnyContext):
         flush: bool = True,
     ):
         if self._shared_ctx.get_logging_level() <= logging.INFO:
+            sep = " " if sep is None else sep
             message = sep.join([f"{value}" for value in values])
             stylized_message = stylize_warning(f"[WARNING] {message}")
             self.print(stylized_message, sep=sep, end=end, file=file, flush=flush)
@@ -171,6 +175,7 @@ class Context(AnyContext):
         flush: bool = True,
     ):
         if self._shared_ctx.get_logging_level() <= logging.ERROR:
+            sep = " " if sep is None else sep
             message = sep.join([f"{value}" for value in values])
             stylized_message = stylize_error(f"[ERROR] {message}")
             self.print(stylized_message, sep=sep, end=end, file=file, flush=flush)
@@ -184,6 +189,7 @@ class Context(AnyContext):
         flush: bool = True,
     ):
         if self._shared_ctx.get_logging_level() <= logging.CRITICAL:
+            sep = " " if sep is None else sep
             message = sep.join([f"{value}" for value in values])
             stylized_message = stylize_error(f"[CRITICAL] {message}")
             self.print(stylized_message, sep=sep, end=end, file=file, flush=flush)

zrb/input/any_input.py

@@ -14,6 +14,11 @@ class AnyInput(ABC):
     def description(self) -> str:
         pass
 
+    @property
+    @abstractmethod
+    def always_prompt(self) -> bool:
+        pass
+
     @property
     @abstractmethod
     def prompt_message(self) -> str:

zrb/input/base_input.py

@@ -17,6 +17,7 @@ class BaseInput(AnyInput):
         auto_render: bool = True,
         allow_empty: bool = False,
         allow_positional_parsing: bool = True,
+        always_prompt: bool = True,
     ):
         self._name = name
         self._description = description
@@ -25,6 +26,7 @@ class BaseInput(AnyInput):
         self._auto_render = auto_render
         self._allow_empty = allow_empty
         self._allow_positional_parsing = allow_positional_parsing
+        self._always_prompt = always_prompt
 
     def __repr__(self):
         return f"<{self.__class__.__name__} name={self._name}>"
@@ -37,6 +39,10 @@ class BaseInput(AnyInput):
     def description(self) -> str:
         return self._description if self._description is not None else self.name
 
+    @property
+    def always_prompt(self) -> bool:
+        return self._always_prompt
+
     @property
     def prompt_message(self) -> str:
         return self._prompt if self._prompt is not None else self.name

zrb/input/bool_input.py

@@ -15,6 +15,7 @@ class BoolInput(BaseInput):
         auto_render: bool = True,
         allow_empty: bool = False,
         allow_positional_parsing: bool = True,
+        always_prompt: bool = True,
     ):
         super().__init__(
             name=name,
@@ -24,6 +25,7 @@ class BoolInput(BaseInput):
             auto_render=auto_render,
             allow_empty=allow_empty,
             allow_positional_parsing=allow_positional_parsing,
+            always_prompt=always_prompt,
         )
 
     def to_html(self, shared_ctx: AnySharedContext) -> str:

zrb/input/float_input.py

@@ -14,6 +14,7 @@ class FloatInput(BaseInput):
         auto_render: bool = True,
         allow_empty: bool = False,
         allow_positional_parsing: bool = True,
+        always_prompt: bool = True,
     ):
         super().__init__(
             name=name,
@@ -23,6 +24,7 @@ class FloatInput(BaseInput):
             auto_render=auto_render,
             allow_empty=allow_empty,
             allow_positional_parsing=allow_positional_parsing,
+            always_prompt=always_prompt,
         )
 
     def to_html(self, shared_ctx: AnySharedContext) -> str:

zrb/input/int_input.py

@@ -14,6 +14,7 @@ class IntInput(BaseInput):
         auto_render: bool = True,
         allow_empty: bool = False,
         allow_positional_parsing: bool = True,
+        always_prompt: bool = True,
     ):
         super().__init__(
             name=name,
@@ -23,6 +24,7 @@ class IntInput(BaseInput):
             auto_render=auto_render,
             allow_empty=allow_empty,
             allow_positional_parsing=allow_positional_parsing,
+            always_prompt=always_prompt,
         )
 
     def to_html(self, shared_ctx: AnySharedContext) -> str:

zrb/input/option_input.py

@@ -15,6 +15,7 @@ class OptionInput(BaseInput):
         auto_render: bool = True,
         allow_empty: bool = False,
         allow_positional_parsing: bool = True,
+        always_prompt: bool = True,
     ):
         super().__init__(
             name=name,
@@ -24,6 +25,7 @@ class OptionInput(BaseInput):
             auto_render=auto_render,
             allow_empty=allow_empty,
             allow_positional_parsing=allow_positional_parsing,
+            always_prompt=always_prompt,
         )
         self._options = options
 

zrb/input/password_input.py

@@ -15,6 +15,7 @@ class PasswordInput(BaseInput):
         auto_render: bool = True,
         allow_empty: bool = False,
         allow_positional_parsing: bool = True,
+        always_prompt: bool = True,
     ):
         super().__init__(
             name=name,
@@ -24,6 +25,7 @@ class PasswordInput(BaseInput):
             auto_render=auto_render,
             allow_empty=allow_empty,
             allow_positional_parsing=allow_positional_parsing,
+            always_prompt=always_prompt,
         )
         self._is_secret = True
 

zrb/input/text_input.py

@@ -19,6 +19,7 @@ class TextInput(BaseInput):
         auto_render: bool = True,
         allow_empty: bool = False,
         allow_positional_parsing: bool = True,
+        always_prompt: bool = True,
         editor: str = DEFAULT_EDITOR,
         extension: str = ".txt",
         comment_start: str | None = None,
@@ -32,6 +33,7 @@ class TextInput(BaseInput):
             auto_render=auto_render,
             allow_empty=allow_empty,
             allow_positional_parsing=allow_positional_parsing,
+            always_prompt=always_prompt,
         )
         self._editor = editor
         self._extension = extension
@@ -69,16 +71,17 @@ class TextInput(BaseInput):
         )
 
     def _prompt_cli_str(self, shared_ctx: AnySharedContext) -> str:
-        prompt_message = (
-            f"{self.comment_start}{super().prompt_message}{self.comment_end}"
+        prompt_message = super().prompt_message
+        comment_prompt_message = (
+            f"{self.comment_start}{prompt_message}{self.comment_end}"
         )
-        prompt_message_eol = f"{prompt_message}\n"
+        comment_prompt_message_eol = f"{comment_prompt_message}\n"
         default_value = self.get_default_str(shared_ctx)
         with tempfile.NamedTemporaryFile(
             delete=False, suffix=self._extension
         ) as temp_file:
             temp_file_name = temp_file.name
-            temp_file.write(prompt_message_eol.encode())
+            temp_file.write(comment_prompt_message_eol.encode())
             # Pre-fill with default content
             if default_value:
                 temp_file.write(default_value.encode())
@@ -87,7 +90,10 @@ class TextInput(BaseInput):
         subprocess.call([self._editor, temp_file_name])
         # Read the edited content
         edited_content = read_file(temp_file_name)
-        parts = [text.strip() for text in edited_content.split(prompt_message, 1)]
+        parts = [
+            text.strip() for text in edited_content.split(comment_prompt_message, 1)
+        ]
         edited_content = "\n".join(parts).lstrip()
         os.remove(temp_file_name)
+        print(f"{prompt_message}: {edited_content}")
         return edited_content

zrb/runner/cli.py

@@ -33,7 +33,7 @@ class Cli(Group):
         if "h" in kwargs or "help" in kwargs:
             self._show_task_info(node)
             return
-        run_kwargs = get_run_kwargs(task=node, args=args, kwargs=kwargs, prompt=True)
+        run_kwargs = get_run_kwargs(task=node, args=args, kwargs=kwargs, cli_mode=True)
         try:
             result = self._run_task(node, args, run_kwargs)
             if result is not None:

zrb/runner/common_util.py

@@ -5,10 +5,10 @@ from zrb.task.any_task import AnyTask
 
 
 def get_run_kwargs(
-    task: AnyTask, args: list[str], kwargs: dict[str, str], prompt: bool = True
+    task: AnyTask, args: list[str], kwargs: dict[str, str], cli_mode: bool
 ) -> tuple[Any]:
     arg_index = 0
-    str_kwargs = {key: val for key, val in kwargs.items()}
+    str_kwargs = {key: f"{val}" for key, val in kwargs.items()}
     run_kwargs = {**str_kwargs}
     shared_ctx = SharedContext(args=args)
     for task_input in task.inputs:
@@ -21,7 +21,7 @@ def get_run_kwargs(
             task_input.update_shared_context(shared_ctx, run_kwargs[task_input.name])
             arg_index += 1
         else:
-            if prompt:
+            if cli_mode and task_input.always_prompt:
                 str_value = task_input.prompt_cli_str(shared_ctx)
             else:
                 str_value = task_input.get_default_str(shared_ctx)

zrb/runner/web_route/task_input_api_route.py

@@ -41,7 +41,7 @@ def serve_task_input_api(
                 return JSONResponse(content={"detail": "Forbidden"}, status_code=403)
             query_dict = json.loads(query)
             run_kwargs = get_run_kwargs(
-                task=task, args=[], kwargs=query_dict, prompt=False
+                task=task, args=[], kwargs=query_dict, cli_mode=False
             )
             return run_kwargs
         return JSONResponse(content={"detail": "Not found"}, status_code=404)

zrb/task/llm_task.py

@@ -4,10 +4,20 @@ from collections.abc import Callable
 from typing import Any
 
 from pydantic_ai import Agent, Tool
-from pydantic_ai.messages import ModelMessagesTypeAdapter
+from pydantic_ai.messages import (
+    FinalResultEvent,
+    FunctionToolCallEvent,
+    FunctionToolResultEvent,
+    ModelMessagesTypeAdapter,
+    PartDeltaEvent,
+    PartStartEvent,
+    TextPartDelta,
+    ToolCallPartDelta,
+)
+from pydantic_ai.models import Model
 from pydantic_ai.settings import ModelSettings
 
-from zrb.attr.type import StrAttr
+from zrb.attr.type import StrAttr, fstring
 from zrb.config import LLM_MODEL, LLM_SYSTEM_PROMPT
 from zrb.context.any_context import AnyContext
 from zrb.context.any_shared_context import AnySharedContext
@@ -15,7 +25,7 @@ from zrb.env.any_env import AnyEnv
 from zrb.input.any_input import AnyInput
 from zrb.task.any_task import AnyTask
 from zrb.task.base_task import BaseTask
-from zrb.util.attr import get_str_attr
+from zrb.util.attr import get_attr, get_str_attr
 from zrb.util.cli.style import stylize_faint
 from zrb.util.file import read_file, write_file
 from zrb.util.run import run_async
@@ -34,7 +44,9 @@ class LLMTask(BaseTask):
         cli_only: bool = False,
         input: list[AnyInput | None] | AnyInput | None = None,
         env: list[AnyEnv | None] | AnyEnv | None = None,
-        model: StrAttr | None = LLM_MODEL,
+        model: (
+            Callable[[AnySharedContext], Model | str | fstring] | Model | None
+        ) = LLM_MODEL,
         model_settings: (
             ModelSettings | Callable[[AnySharedContext], ModelSettings] | None
         ) = None,
@@ -93,7 +105,7 @@ class LLMTask(BaseTask):
             successor=successor,
         )
         self._model = model
-        self._model_settings = (model_settings,)
+        self._model_settings = model_settings
         self._agent = agent
         self._render_model = render_model
         self._system_prompt = system_prompt
@@ -108,6 +120,9 @@ 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)
 
@@ -115,15 +130,85 @@ class LLMTask(BaseTask):
         history = await self._read_conversation_history(ctx)
         user_prompt = self._get_message(ctx)
         agent = self._get_agent(ctx)
-        result = await agent.run(
+        async with agent.iter(
             user_prompt=user_prompt,
             message_history=ModelMessagesTypeAdapter.validate_python(history),
-        )
-        new_history = json.loads(result.all_messages_json())
-        for history in new_history:
-            ctx.print(stylize_faint(json.dumps(history)))
+        ) as agent_run:
+            async for node in agent_run:
+                # Each node represents a step in the agent's execution
+                await self._print_node(ctx, agent_run, node)
+        new_history = json.loads(agent_run.result.all_messages_json())
         await self._write_conversation_history(ctx, new_history)
-        return result.data
+        return agent_run.result.data
+
+    async def _print_node(self, ctx: AnyContext, agent_run: Any, node: Any):
+        if Agent.is_user_prompt_node(node):
+            # A user prompt node => The user has provided input
+            ctx.print(stylize_faint(f">> UserPromptNode: {node.user_prompt}"))
+        elif Agent.is_model_request_node(node):
+            # A model request node => We can stream tokens from the model"s request
+            ctx.print(
+                stylize_faint(">> ModelRequestNode: streaming partial request tokens")
+            )
+            async with node.stream(agent_run.ctx) as request_stream:
+                is_streaming = False
+                async for event in request_stream:
+                    if isinstance(event, PartStartEvent):
+                        if is_streaming:
+                            ctx.print("", plain=True)
+                        ctx.print(
+                            stylize_faint(
+                                f"[Request] Starting part {event.index}: {event.part!r}"
+                            ),
+                        )
+                        is_streaming = False
+                    elif isinstance(event, PartDeltaEvent):
+                        if isinstance(event.delta, TextPartDelta):
+                            ctx.print(
+                                stylize_faint(f"{event.delta.content_delta}"),
+                                end="",
+                                plain=is_streaming,
+                            )
+                        elif isinstance(event.delta, ToolCallPartDelta):
+                            ctx.print(
+                                stylize_faint(f"{event.delta.args_delta}"),
+                                end="",
+                                plain=is_streaming,
+                            )
+                        is_streaming = True
+                    elif isinstance(event, FinalResultEvent):
+                        if is_streaming:
+                            ctx.print("", plain=True)
+                        ctx.print(
+                            stylize_faint(f"[Result] tool_name={event.tool_name}"),
+                        )
+                        is_streaming = False
+                if is_streaming:
+                    ctx.print("", plain=True)
+        elif Agent.is_call_tools_node(node):
+            # A handle-response node => The model returned some data, potentially calls a tool
+            ctx.print(
+                stylize_faint(
+                    ">> CallToolsNode: streaming partial response & tool usage"
+                )
+            )
+            async with node.stream(agent_run.ctx) as handle_stream:
+                async for event in handle_stream:
+                    if isinstance(event, FunctionToolCallEvent):
+                        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
+                            )
+                        )
+                    elif isinstance(event, FunctionToolResultEvent):
+                        ctx.print(
+                            stylize_faint(
+                                f"[Tools] Tool call {event.tool_call_id!r} returned => {event.result.content}"  # noqa
+                            )
+                        )
+        elif Agent.is_end_node(node):
+            # Once an End node is reached, the agent run is complete
+            ctx.print(stylize_faint(f"{agent_run.result.data}"))
 
     async def _write_conversation_history(
         self, ctx: AnyContext, conversations: list[Any]
@@ -135,11 +220,9 @@ class LLMTask(BaseTask):
             write_file(history_file, json.dumps(conversations, indent=2))
 
     def _get_model_settings(self, ctx: AnyContext) -> ModelSettings | None:
-        if isinstance(self._model_settings, ModelSettings):
-            return self._model_settings
         if callable(self._model_settings):
             return self._model_settings(ctx)
-        return None
+        return self._model_settings
 
     def _get_agent(self, ctx: AnyContext) -> Agent:
         if isinstance(self._agent, Agent):
@@ -158,12 +241,16 @@ class LLMTask(BaseTask):
             self._get_model(ctx),
             system_prompt=self._get_system_prompt(ctx),
             tools=tools,
+            model_settings=self._get_model_settings(ctx),
         )
 
-    def _get_model(self, ctx: AnyContext) -> str:
-        return get_str_attr(
+    def _get_model(self, ctx: AnyContext) -> str | Model | None:
+        model = get_attr(
             ctx, self._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")
 
     def _get_system_prompt(self, ctx: AnyContext) -> str:
         return get_str_attr(

{zrb-1.2.1.dist-info → zrb-1.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: zrb
-Version: 1.2.1
+Version: 1.3.0
 Summary: Your Automation Powerhouse
 Home-page: https://github.com/state-alchemists/zrb
 License: AGPL-3.0-or-later
@@ -24,7 +24,7 @@ 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.19,<0.0.20)
+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)
@@ -39,29 +39,96 @@ Description-Content-Type: text/markdown
 
 # 🤖 Zrb: Your Automation Powerhouse
 
-Zrb allows you to write your automation tasks in Python and declaratively:
+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
-# Filename: zrb_init.py
-from zrb import cli, Task, Group, IntInput
-
-math = cli.add_group(Group("math", description="Math tools"))
-math.add_task(Task(
-    name="add",
-    input=[
-        IntInput("a", description="First number"),
-        IntInput("b", description="Second number")
-    ],
-    action=lambda ctx: ctx.input.a + ctx.input.b
-))
+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, you will be able to access your automation tasks from the CLI, Web Interface, or via HTTP API.
+Once defined, your automation tasks are immediately accessible from the CLI. You can then invoke the tasks by invoking.
 
-For more complex scenario, you can also defined Task dependencies (upstreams) and retry mechanisms. You can also make a scheduled tasks, just like in Apache Airflow.
+```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`.
 
-Furthermore, Zrb has some builtin tasks to manage monorepo, generate FastAPI application, or play around with LLM.
+![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: