zrb 1.15.21__tar.gz → 2.0.0a2__tar.gz

{zrb-1.15.21 → zrb-2.0.0a2}/PKG-INFO

@@ -1,41 +1,58 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: zrb
-Version: 1.15.21
+Version: 2.0.0a2
 Summary: Your Automation Powerhouse
 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
+Requires-Python: >=3.11.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
 Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Provides-Extra: all
+Provides-Extra: anthropic
+Provides-Extra: bedrock
+Provides-Extra: cohere
+Provides-Extra: google
+Provides-Extra: groq
+Provides-Extra: huggingface
+Provides-Extra: mistral
 Provides-Extra: playwright
 Provides-Extra: rag
-Requires-Dist: beautifulsoup4 (>=4.13.4,<5.0.0)
-Requires-Dist: black (>=25.1.0,<26.0.0)
-Requires-Dist: chromadb (>=0.6.3,<0.7.0) ; extra == "rag" or extra == "all"
-Requires-Dist: fastapi[standard] (>=0.116.1,<0.117.0)
-Requires-Dist: isort (>=6.0.1,<7.0.0)
-Requires-Dist: libcst (>=1.8.2,<2.0.0)
-Requires-Dist: markdownify (>=1.2.0,<2.0.0)
-Requires-Dist: mcp (>1.12.3)
-Requires-Dist: openai (>=1.99.9)
+Provides-Extra: vertexai
+Requires-Dist: anthropic (>=0.75.0) ; extra == "anthropic" or extra == "all"
+Requires-Dist: beautifulsoup4 (>=4.14.2,<5.0.0)
+Requires-Dist: black (>=25.11.0,<26.0.0)
+Requires-Dist: boto3 (>=1.42.14) ; extra == "bedrock"
+Requires-Dist: chromadb (>=1.3.5,<2.0.0) ; extra == "rag" or extra == "all"
+Requires-Dist: cohere (>=5.18.0) ; extra == "cohere" or extra == "all"
+Requires-Dist: fastapi[standard] (>=0.123.9,<0.124.0)
+Requires-Dist: google-auth (>=2.36.0) ; extra == "vertexai" or extra == "all"
+Requires-Dist: google-genai (>=1.56.0) ; extra == "google" or extra == "all"
+Requires-Dist: groq (>=0.25.0) ; extra == "groq" or extra == "all"
+Requires-Dist: huggingface-hub[inference] (>=0.33.5,<1.0.0) ; extra == "huggingface"
+Requires-Dist: isort (>=7.0.0,<8.0.0)
+Requires-Dist: libcst (>=1.8.6,<2.0.0)
+Requires-Dist: markdownify (>=1.2.2,<2.0.0)
+Requires-Dist: mcp (>1.18.0)
+Requires-Dist: mistralai (>=1.9.11) ; extra == "mistral"
+Requires-Dist: openai (>=2.11.0)
 Requires-Dist: pdfplumber (>=0.11.7,<0.12.0)
-Requires-Dist: playwright (>=1.54.0,<2.0.0) ; extra == "playwright" or extra == "all"
+Requires-Dist: playwright (>=1.56.0,<2.0.0) ; extra == "playwright" or extra == "all"
 Requires-Dist: prompt-toolkit (>=3)
 Requires-Dist: psutil (>=7.0.0,<8.0.0)
-Requires-Dist: pydantic-ai-slim[anthropic,bedrock,cohere,google,groq,huggingface,mistral,openai,vertexai] (>=0.8.1,<0.9.0)
+Requires-Dist: pydantic-ai-slim (>=1.42.0,<1.43.0)
 Requires-Dist: pyjwt (>=2.10.1,<3.0.0)
 Requires-Dist: python-dotenv (>=1.1.1,<2.0.0)
 Requires-Dist: python-jose[cryptography] (>=3.5.0,<4.0.0)
-Requires-Dist: requests (>=2.32.2,<3.0.0)
+Requires-Dist: pyyaml (>=6.0.3,<7.0.0)
+Requires-Dist: requests (>=2.32.5,<3.0.0)
 Requires-Dist: rich (>=13)
-Requires-Dist: tiktoken (>=0.8.0,<0.9.0)
+Requires-Dist: tiktoken (>=0.12.0,<0.13.0)
 Requires-Dist: ulid-py (>=1.1.0,<2.0.0)
 Project-URL: Documentation, https://github.com/state-alchemists/zrb
 Project-URL: Homepage, https://github.com/state-alchemists/zrb
@@ -114,8 +131,8 @@ Add the following Python code to your `zrb_init.py`:
 
 ```python
 from zrb import cli, LLMTask, CmdTask, StrInput, Group
-from zrb.builtin.llm.tool.code import analyze_repo
-from zrb.builtin.llm.tool.file import write_to_file
+from zrb.llm.tool.code import analyze_code
+from zrb.llm.tool.file import write_file
 
 
 # Create a group for Mermaid-related tasks
@@ -139,7 +156,7 @@ make_mermaid_script = mermaid_group.add_task(
             "Write the script into `{ctx.input.dir}/{ctx.input.diagram}.mmd`"
         ),
         tools=[
-            analyze_repo, write_to_file
+            analyze_code, write_file
         ],
     )
 )
@@ -215,13 +232,7 @@ Start a chat session with an LLM to ask questions, brainstorm ideas, or get codi
 zrb llm chat
 ```
 
-### Quick Questions
 
-For a single question, use the `ask` command for a fast response.
-
-```bash
-zrb llm ask "What is the capital of Indonesia?"
-```
 
 ---
 

{zrb-1.15.21 → zrb-2.0.0a2}/README.md

@@ -70,8 +70,8 @@ Add the following Python code to your `zrb_init.py`:
 
 ```python
 from zrb import cli, LLMTask, CmdTask, StrInput, Group
-from zrb.builtin.llm.tool.code import analyze_repo
-from zrb.builtin.llm.tool.file import write_to_file
+from zrb.llm.tool.code import analyze_code
+from zrb.llm.tool.file import write_file
 
 
 # Create a group for Mermaid-related tasks
@@ -95,7 +95,7 @@ make_mermaid_script = mermaid_group.add_task(
             "Write the script into `{ctx.input.dir}/{ctx.input.diagram}.mmd`"
         ),
         tools=[
-            analyze_repo, write_to_file
+            analyze_code, write_file
         ],
     )
 )
@@ -171,13 +171,7 @@ Start a chat session with an LLM to ask questions, brainstorm ideas, or get codi
 zrb llm chat
 ```
 
-### Quick Questions
 
-For a single question, use the `ask` command for a fast response.
-
-```bash
-zrb llm ask "What is the capital of Indonesia?"
-```
 
 ---
 

{zrb-1.15.21 → zrb-2.0.0a2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "zrb"
-version = "1.15.21"
+version = "2.0.0a2"
 description = "Your Automation Powerhouse"
 authors = ["Go Frendi Gunawan <gofrendiasgard@gmail.com>"]
 license = "AGPL-3.0-or-later"
@@ -37,39 +37,66 @@ exclude = [
 ]
 
 [tool.poetry.dependencies]
-python = ">=3.10.0,<4.0.0"
+python = ">=3.11.0,<4.0.0"
 python-dotenv = "^1.1.1"
-black = "^25.1.0"
-isort = "^6.0.1"
-libcst = "^1.8.2"
-beautifulsoup4 = "^4.13.4"
+black = "^25.11.0"
+isort = "^7.0.0"
+libcst = "^1.8.6"
+beautifulsoup4 = "^4.14.2"
 pdfplumber = "^0.11.7"
-fastapi = {extras = ["standard"], version = "^0.116.1"}
+fastapi = {extras = ["standard"], version = "^0.123.9"}
 python-jose = {extras = ["cryptography"], version = "^3.5.0"}
 ulid-py = "^1.1.0"
 psutil = "^7.0.0"
 
 # NOTE: Look at https://github.com/pydantic/pydantic-ai/blob/main/pydantic_ai_slim/pyproject.toml
-pydantic-ai-slim = {extras = ["openai", "cohere", "vertexai", "google", "anthropic", "groq", "mistral", "bedrock", "huggingface"], version = "~0.8.1"} 
+pydantic-ai-slim = "~1.42.0"
 
 # NOTE: Make sure the following dependencies are not conflicting with pydantic-ai-slim
-openai = ">=1.99.9"
+openai = ">=2.11.0"
 prompt-toolkit = ">=3"
 rich = ">=13"
-mcp = ">1.12.3"
-requests = "^2.32.2"
+mcp = ">1.18.0"
+requests = "^2.32.5"
 
-tiktoken = "^0.8.0"
-playwright = {version = "^1.54.0", optional = true}
-chromadb = {version = "^0.6.3", optional = true}
-markdownify = "^1.2.0"
+tiktoken = "^0.12.0"
+markdownify = "^1.2.2"
 pyjwt = "^2.10.1"
+pyyaml = "^6.0.3"
+
+# Optional packages
+playwright = {version = "^1.56.0", optional = true}
+chromadb = {version = "^1.3.5", optional = true}
+
+# NOTE: Make sure the following dependencies are not conflicting with pydantic-ai-slim
+cohere = {version = ">=5.18.0", optional = true}
+google-auth = {version = ">=2.36.0", optional = true}
+google-genai = {version = ">=1.56.0", optional = true}
+anthropic = {version = ">=0.75.0", optional = true}
+groq = {version = ">=0.25.0", optional = true}
+mistralai = {version = ">=1.9.11", optional = true}
+boto3 = {version = ">=1.42.14", optional = true}
+huggingface-hub = {version = ">=0.33.5,<1.0.0", extras = ["inference"], optional = true}
 
 [tool.poetry.extras]
-# poetry install -E rag
+# Install all extras:
+# - poetry install --all-extras
+# - pip install zrb[all]
+# Install specific extra:
+# - poetry install -E rag
+# - pip install zrb[rag]
+
 rag = ["chromadb"]
 playwright = ["playwright"]
-all = ["chromadb", "playwright"]
+cohere = ["cohere"]
+vertexai = ["google-auth"]
+google = ["google-genai"]
+anthropic = ["anthropic"]
+groq = ["groq"]
+mistral = ["mistralai"]
+bedrock = ["boto3"]
+huggingface = ["huggingface-hub"]
+all = ["chromadb", "playwright", "cohere", "google-auth", "google-genai", "anthropic", "groq", "mistral", "bedrock", "huggingface"]
 
 [tool.poetry.group.dev.dependencies]
 alembic = "^1.16.4" # FastApp dependencies

zrb-2.0.0a2/src/zrb/__init__.py

@@ -0,0 +1,119 @@
+from zrb import builtin
+from zrb.attr.type import (
+    AnyAttr,
+    BoolAttr,
+    FloatAttr,
+    IntAttr,
+    StrAttr,
+    StrDictAttr,
+    fstring,
+)
+from zrb.callback.any_callback import AnyCallback
+from zrb.callback.callback import Callback
+from zrb.cmd.cmd_result import CmdResult
+from zrb.cmd.cmd_val import Cmd, CmdPath
+from zrb.config.config import CFG
+from zrb.config.web_auth_config import web_auth_config
+from zrb.content_transformer.any_content_transformer import AnyContentTransformer
+from zrb.content_transformer.content_transformer import ContentTransformer
+from zrb.context.any_context import AnyContext
+from zrb.context.any_shared_context import AnySharedContext
+from zrb.context.context import Context
+from zrb.context.shared_context import SharedContext
+from zrb.env.any_env import AnyEnv
+from zrb.env.env import Env
+from zrb.env.env_file import EnvFile
+from zrb.env.env_map import EnvMap
+from zrb.group.any_group import AnyGroup
+from zrb.group.group import Group
+from zrb.input.any_input import AnyInput
+from zrb.input.base_input import BaseInput
+from zrb.input.bool_input import BoolInput
+from zrb.input.float_input import FloatInput
+from zrb.input.int_input import IntInput
+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.config import LLMConfig, llm_config
+from zrb.llm.config.limiter import LLMLimiter, llm_limiter
+from zrb.llm.task.llm_chat_task import LLMChatTask
+from zrb.llm.task.llm_task import LLMTask
+from zrb.runner.cli import cli
+from zrb.runner.web_schema.user import User
+from zrb.session.session import Session
+from zrb.task.any_task import AnyTask
+from zrb.task.base_task import BaseTask
+from zrb.task.base_trigger import BaseTrigger
+from zrb.task.cmd_task import CmdTask
+from zrb.task.http_check import HttpCheck
+from zrb.task.make_task import make_task
+from zrb.task.rsync_task import RsyncTask
+from zrb.task.scaffolder import Scaffolder
+from zrb.task.scheduler import Scheduler
+from zrb.task.task import Task
+from zrb.task.tcp_check import TcpCheck
+from zrb.util.load import load_file, load_module
+from zrb.xcom.xcom import Xcom
+
+__all__ = [
+    "builtin",
+    "AnyAttr",
+    "BoolAttr",
+    "FloatAttr",
+    "IntAttr",
+    "StrAttr",
+    "StrDictAttr",
+    "fstring",
+    "AnyCallback",
+    "Callback",
+    "CmdResult",
+    "Cmd",
+    "CmdPath",
+    "CFG",
+    "web_auth_config",
+    "AnyContentTransformer",
+    "ContentTransformer",
+    "AnyContext",
+    "AnySharedContext",
+    "Context",
+    "SharedContext",
+    "AnyEnv",
+    "Env",
+    "EnvFile",
+    "EnvMap",
+    "AnyGroup",
+    "Group",
+    "AnyInput",
+    "BaseInput",
+    "BoolInput",
+    "FloatInput",
+    "IntInput",
+    "OptionInput",
+    "PasswordInput",
+    "StrInput",
+    "TextInput",
+    "cli",
+    "User",
+    "Session",
+    "AnyTask",
+    "BaseTask",
+    "BaseTrigger",
+    "CmdTask",
+    "HttpCheck",
+    "make_task",
+    "RsyncTask",
+    "Scaffolder",
+    "Scheduler",
+    "Task",
+    "TcpCheck",
+    "load_file",
+    "load_module",
+    "Xcom",
+    "LLMTask",
+    "LLMChatTask",
+    "LLMConfig",
+    "llm_config",
+    "LLMLimiter",
+    "llm_limiter",
+]

zrb-2.0.0a2/src/zrb/attr/type.py

@@ -0,0 +1,15 @@
+from typing import Any, Callable
+
+from zrb.context.any_context import AnyContext
+from zrb.context.any_shared_context import AnySharedContext
+
+fstring = str
+AnyAttr = Any | fstring | Callable[[AnyContext | AnySharedContext], Any]
+StrAttr = str | fstring | Callable[[AnyContext | AnySharedContext], str | None]
+BoolAttr = bool | fstring | Callable[[AnyContext | AnySharedContext], bool | None]
+IntAttr = int | fstring | Callable[[AnyContext | AnySharedContext], int | None]
+FloatAttr = float | fstring | Callable[[AnyContext | AnySharedContext], float | None]
+StrDictAttr = (
+    dict[str, StrAttr] | Callable[[AnyContext | AnySharedContext], dict[str, Any]]
+)
+StrListAttr = list[StrAttr] | Callable[[AnyContext | AnySharedContext], list[str]]

{zrb-1.15.21 → zrb-2.0.0a2}/src/zrb/builtin/__init__.py

@@ -9,12 +9,13 @@ from zrb.builtin.git import (
 from zrb.builtin.git_subtree import git_add_subtree, git_pull_subtree, git_push_subtree
 from zrb.builtin.http import generate_curl, http_request
 from zrb.builtin.jwt import decode_jwt, encode_jwt, validate_jwt
-from zrb.builtin.llm.llm_ask import llm_ask
+from zrb.builtin.llm.chat import llm_chat
 from zrb.builtin.md5 import hash_md5, sum_md5, validate_md5
 from zrb.builtin.project.add.fastapp.fastapp_task import add_fastapp_to_project
 from zrb.builtin.project.create.project_task import create_project
 from zrb.builtin.python import format_python_code
 from zrb.builtin.random import shuffle_values, throw_dice
+from zrb.builtin.searxng.start import start_searxng
 from zrb.builtin.setup.asdf.asdf import setup_asdf
 from zrb.builtin.setup.latex.ubuntu import setup_latex_on_ubuntu
 from zrb.builtin.setup.tmux.tmux import setup_tmux
@@ -43,3 +44,56 @@ from zrb.builtin.uuid import (
     validate_uuid_v4,
     validate_uuid_v5,
 )
+
+__all__ = [
+    "decode_base64",
+    "encode_base64",
+    "validate_base64",
+    "get_git_diff",
+    "git_commit",
+    "git_pull",
+    "git_push",
+    "prune_local_branches",
+    "git_add_subtree",
+    "git_pull_subtree",
+    "git_push_subtree",
+    "generate_curl",
+    "http_request",
+    "decode_jwt",
+    "encode_jwt",
+    "validate_jwt",
+    "llm_chat",
+    "hash_md5",
+    "sum_md5",
+    "validate_md5",
+    "add_fastapp_to_project",
+    "create_project",
+    "format_python_code",
+    "shuffle_values",
+    "throw_dice",
+    "start_searxng",
+    "setup_asdf",
+    "setup_latex_on_ubuntu",
+    "setup_tmux",
+    "setup_ubuntu",
+    "setup_zsh",
+    "make_bash_autocomplete",
+    "get_shell_subcommands",
+    "make_zsh_autocomplete",
+    "add_todo",
+    "archive_todo",
+    "complete_todo",
+    "edit_todo",
+    "list_todo",
+    "log_todo",
+    "show_todo",
+    "generate_uuid_v1",
+    "generate_uuid_v3",
+    "generate_uuid_v4",
+    "generate_uuid_v5",
+    "validate_uuid",
+    "validate_uuid_v1",
+    "validate_uuid_v3",
+    "validate_uuid_v4",
+    "validate_uuid_v5",
+]

{zrb-1.15.21 → zrb-2.0.0a2}/src/zrb/builtin/git.py

@@ -82,6 +82,12 @@ async def get_git_diff(ctx: AnyContext):
 
 @make_task(
     name="prune-local-git-branches",
+    input=StrInput(
+        name="preserved-branch",
+        description="Branches to be preserved",
+        prompt="Branches to be preserved, comma separated",
+        default="master,main,dev,develop",
+    ),
     description="🧹 Prune local branches",
     group=git_branch_group,
     alias="prune",
@@ -93,8 +99,13 @@ async def prune_local_branches(ctx: AnyContext):
     branches = await get_branches(repo_dir, print_method=ctx.print)
     ctx.print(stylize_faint("Get current branch"))
     current_branch = await get_current_branch(repo_dir, print_method=ctx.print)
+    preserved_branches = [
+        branch.strip()
+        for branch in ctx.input.preserved_branch.split(",")
+        if branch.strip() != ""
+    ]
     for branch in branches:
-        if branch == current_branch or branch == "main" or branch == "master":
+        if branch == current_branch or branch in preserved_branches:
             continue
         ctx.print(stylize_faint(f"Removing local branch: {branch}"))
         try:

zrb-2.0.0a2/src/zrb/builtin/group.py

@@ -0,0 +1,65 @@
+from zrb.config.config import CFG
+from zrb.group.group import Group
+from zrb.runner.cli import cli
+
+
+def _maybe_add_group(group: Group):
+    if CFG.LOAD_BUILTIN:
+        cli.add_group(group)
+    return group
+
+
+base64_group = _maybe_add_group(
+    Group(name="base64", description="📄 Base64 operations")
+)
+uuid_group = _maybe_add_group(Group(name="uuid", description="🆔 UUID operations"))
+uuid_v1_group = uuid_group.add_group(Group(name="v1", description="UUID V1 operations"))
+uuid_v3_group = uuid_group.add_group(Group(name="v3", description="UUID V3 operations"))
+uuid_v4_group = uuid_group.add_group(Group(name="v4", description="UUID V4 operations"))
+uuid_v5_group = uuid_group.add_group(Group(name="v5", description="UUID V5 operations"))
+ulid_group = _maybe_add_group(Group(name="ulid", description="🔢 ULID operations"))
+jwt_group = _maybe_add_group(Group(name="jwt", description="🔒 JWT encode/decode"))
+http_group = _maybe_add_group(
+    Group(name="http", description="🌐 HTTP request operations")
+)
+
+random_group = _maybe_add_group(Group(name="random", description="🔀 Random operation"))
+git_group = _maybe_add_group(Group(name="git", description="🌱 Git related commands"))
+git_branch_group = git_group.add_group(
+    Group(name="branch", description="🌿 Git branch related commands")
+)
+git_subtree_group = git_group.add_group(
+    Group(name="subtree", description="🌳 Git subtree related commands")
+)
+llm_group = _maybe_add_group(Group(name="llm", description="🤖 LLM operations"))
+md5_group = _maybe_add_group(Group(name="md5", description="🔢 Md5 operations"))
+python_group = _maybe_add_group(
+    Group(name="python", description="🐍 Python related commands")
+)
+todo_group = _maybe_add_group(Group(name="todo", description="✅ Todo management"))
+
+shell_group = _maybe_add_group(
+    Group(name="shell", description="💬 Shell related commands")
+)
+shell_autocomplete_group = shell_group.add_group(
+    Group(name="autocomplete", description="⌨️ Shell autocomplete related commands")
+)
+
+project_group = _maybe_add_group(
+    Group(name="project", description="📁 Project related commands")
+)
+add_to_project_group = project_group.add_group(
+    Group(name="add", description="➕ Add things to project")
+)
+add_fastapp_to_project_group = add_to_project_group.add_group(
+    Group(name="fastapp", description="🚀 Add Fastapp resources")
+)
+
+setup_group = _maybe_add_group(Group(name="setup", description="🔧 Setup"))
+setup_latex_group = setup_group.add_group(
+    Group(name="latex", description="✍️ Setup LaTeX")
+)
+
+searxng_group = _maybe_add_group(
+    Group(name="searxng", description="🔎 Searxng related command")
+)

zrb-2.0.0a2/src/zrb/builtin/llm/chat.py

@@ -0,0 +1,147 @@
+from zrb.builtin.group import llm_group
+from zrb.config.config import CFG
+from zrb.input.bool_input import BoolInput
+from zrb.input.str_input import StrInput
+from zrb.llm.app.confirmation.replace_confirmation import replace_confirmation
+from zrb.llm.history_processor.summarizer import create_summarizer_history_processor
+from zrb.llm.note.manager import NoteManager
+from zrb.llm.prompt.claude_compatibility import (
+    create_claude_compatibility_prompt,
+)
+from zrb.llm.prompt.compose import PromptManager, new_prompt
+from zrb.llm.prompt.default import (
+    get_mandate_prompt,
+    get_persona_prompt,
+)
+from zrb.llm.prompt.note import create_note_prompt
+from zrb.llm.prompt.system_context import system_context
+from zrb.llm.prompt.zrb import create_zrb_prompt
+from zrb.llm.skill.manager import SkillManager
+from zrb.llm.task.llm_chat_task import LLMChatTask
+from zrb.llm.tool.bash import run_shell_command
+from zrb.llm.tool.code import analyze_code
+from zrb.llm.tool.file import (
+    analyze_file,
+    list_files,
+    read_file,
+    read_files,
+    replace_in_file,
+    search_files,
+    write_file,
+    write_files,
+)
+from zrb.llm.tool.note import create_note_tools
+from zrb.llm.tool.skill import create_activate_skill_tool
+from zrb.llm.tool.sub_agent import create_sub_agent_tool
+from zrb.llm.tool.web import open_web_page, search_internet
+from zrb.llm.tool.zrb_task import create_list_zrb_task_tool, create_run_zrb_task_tool
+from zrb.runner.cli import cli
+
+skill_manager = SkillManager()
+note_manager = NoteManager()
+
+llm_chat = LLMChatTask(
+    name="chat",
+    description="🤖 Chat with your AI Assistant",
+    input=[
+        StrInput("message", "Message", allow_empty=True, always_prompt=False),
+        StrInput(
+            "session", "Conversation Session", allow_empty=True, always_prompt=False
+        ),
+        BoolInput(
+            "yolo", "YOLO Mode", default=False, allow_empty=True, always_prompt=False
+        ),
+        StrInput("attach", "Attachments", allow_empty=True, always_prompt=False),
+        BoolInput(
+            "interactive",
+            "Interactive Mode",
+            default=True,
+            allow_empty=True,
+            always_prompt=False,
+        ),
+    ],
+    yolo="{ctx.input.yolo}",
+    message="{ctx.input.message}",
+    conversation_name="{ctx.input.session}",
+    interactive="{ctx.input.interactive}",
+    history_processors=[
+        create_summarizer_history_processor(
+            token_threshold=CFG.LLM_HISTORY_SUMMARIZATION_TOKEN_THRESHOLD,
+            summary_window=CFG.LLM_HISTORY_SUMMARIZATION_WINDOW,
+        )
+    ],
+    prompt_manager=PromptManager(),
+    ui_ascii_art=lambda ctx: CFG.LLM_ASSISTANT_ASCII_ART,
+    ui_assistant_name=lambda ctx: CFG.LLM_ASSISTANT_NAME,
+    ui_greeting=lambda ctx: f"{CFG.LLM_ASSISTANT_NAME}\n{CFG.LLM_ASSISTANT_JARGON}",
+    ui_jargon=lambda ctx: CFG.LLM_ASSISTANT_JARGON,
+    ui_summarize_commands=["/compress", "/compact"],
+    ui_attach_commands=["/attach"],
+    ui_exit_commands=["/q", "/bye", "/quit", "/exit"],
+    ui_info_commands=["/info", "/help"],
+    ui_save_commands=["/save"],
+    ui_load_commands=["/load"],
+    ui_yolo_toggle_commands=["/yolo"],
+    ui_redirect_output_commands=[">", "/redirect"],
+    ui_exec_commands=["!", "/exec"],
+)
+llm_group.add_task(llm_chat)
+cli.add_task(llm_chat)
+
+
+async def roll_dice() -> str:
+    """Roll a six-sided die and return the result."""
+    import asyncio
+    import random
+
+    await asyncio.sleep(3)
+    return str(random.randint(1, 6))
+
+
+async def get_current_time() -> str:
+    """Get the current time."""
+    from datetime import datetime
+
+    return datetime.now().strftime("%H:%M:%S")
+
+
+joke_agent = create_sub_agent_tool(
+    name="joke_agent",
+    description="Generates jokes about the current directory content.",
+    system_prompt=(
+        "You are a comedian. Use the 'run_shell_command' tool to list files "
+        "(ls -la) and make a funny joke about the project structure."
+    ),
+    tools=[run_shell_command],
+)
+
+llm_chat.prompt_manager.add_middleware(
+    new_prompt(get_persona_prompt(CFG.LLM_ASSISTANT_NAME)),
+    new_prompt(get_mandate_prompt()),
+    system_context,
+    create_note_prompt(note_manager),
+    create_claude_compatibility_prompt(skill_manager),
+    create_zrb_prompt(),
+)
+llm_chat.add_confirmation_middleware(replace_confirmation)
+llm_chat.add_tool(
+    roll_dice,
+    joke_agent,
+    run_shell_command,
+    list_files,
+    read_file,
+    read_files,
+    write_file,
+    write_files,
+    replace_in_file,
+    search_files,
+    analyze_file,
+    analyze_code,
+    search_internet,
+    open_web_page,
+    create_list_zrb_task_tool(),
+    create_run_zrb_task_tool(),
+    create_activate_skill_tool(skill_manager),
+    *create_note_tools(note_manager),
+)
+llm_chat.add_tool(get_current_time)