zrb 1.9.3__py3-none-any.whl → 1.9.4__py3-none-any.whl

zrb/__init__.py

@@ -1,116 +1,133 @@
-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 import CFG
-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 import llm_config
-from zrb.llm_rate_limitter import llm_rate_limitter
-from zrb.runner.cli import cli
-from zrb.runner.web_auth_config import web_auth_config
-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.llm.history import ConversationHistoryData
-from zrb.task.llm_task import LLMTask
-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
+import importlib
+from typing import TYPE_CHECKING, Any
 
-assert load_file
-assert load_module
-assert fstring
-assert AnyAttr
-assert BoolAttr
-assert IntAttr
-assert FloatAttr
-assert StrAttr
-assert StrDictAttr
-assert AnyCallback
-assert Callback
-assert AnyEnv
-assert Env
-assert EnvFile
-assert EnvMap
-assert AnyInput
-assert BaseInput
-assert BoolInput
-assert Cmd
-assert CmdPath
-assert CmdResult
-assert CmdTask
-assert HttpCheck
-assert TcpCheck
-assert FloatInput
-assert IntInput
-assert OptionInput
-assert PasswordInput
-assert StrInput
-assert TextInput
-assert AnyGroup
-assert Group
-assert AnyTask
-assert BaseTask
-assert BaseTrigger
-assert RsyncTask
-assert Task
-assert LLMTask
-assert ConversationHistoryData
-assert Session
-assert AnyContext
-assert Context
-assert AnySharedContext
-assert SharedContext
-assert make_task
-assert AnyContentTransformer
-assert ContentTransformer
-assert Scaffolder
-assert Scheduler
-assert cli
-assert llm_config
-assert llm_rate_limitter
-assert Xcom
-assert web_auth_config
-assert User
+_LAZY_LOAD = {
+    "AnyAttr": "zrb.attr.type",
+    "BoolAttr": "zrb.attr.type",
+    "FloatAttr": "zrb.attr.type",
+    "IntAttr": "zrb.attr.type",
+    "StrAttr": "zrb.attr.type",
+    "StrDictAttr": "zrb.attr.type",
+    "fstring": "zrb.attr.type",
+    "AnyCallback": "zrb.callback.any_callback",
+    "Callback": "zrb.callback.callback",
+    "CmdResult": "zrb.cmd.cmd_result",
+    "Cmd": "zrb.cmd.cmd_val",
+    "CmdPath": "zrb.cmd.cmd_val",
+    "CFG": "zrb.config",
+    "AnyContentTransformer": "zrb.content_transformer.any_content_transformer",
+    "ContentTransformer": "zrb.content_transformer.content_transformer",
+    "AnyContext": "zrb.context.any_context",
+    "AnySharedContext": "zrb.context.any_shared_context",
+    "Context": "zrb.context.context",
+    "SharedContext": "zrb.context.shared_context",
+    "AnyEnv": "zrb.env.any_env",
+    "Env": "zrb.env.env",
+    "EnvFile": "zrb.env.env_file",
+    "EnvMap": "zrb.env.env_map",
+    "AnyGroup": "zrb.group.any_group",
+    "Group": "zrb.group.group",
+    "AnyInput": "zrb.input.any_input",
+    "BaseInput": "zrb.input.base_input",
+    "BoolInput": "zrb.input.bool_input",
+    "FloatInput": "zrb.input.float_input",
+    "IntInput": "zrb.input.int_input",
+    "OptionInput": "zrb.input.option_input",
+    "PasswordInput": "zrb.input.password_input",
+    "StrInput": "zrb.input.str_input",
+    "TextInput": "zrb.input.text_input",
+    "llm_config": "zrb.llm_config",
+    "llm_rate_limitter": "zrb.llm_rate_limitter",
+    "cli": "zrb.runner.cli",
+    "web_auth_config": "zrb.runner.web_auth_config",
+    "User": "zrb.runner.web_schema.user",
+    "Session": "zrb.session.session",
+    "AnyTask": "zrb.task.any_task",
+    "BaseTask": "zrb.task.base_task",
+    "BaseTrigger": "zrb.task.base_trigger",
+    "CmdTask": "zrb.task.cmd_task",
+    "HttpCheck": "zrb.task.http_check",
+    "ConversationHistoryData": "zrb.task.llm.history",
+    "LLMTask": "zrb.task.llm_task",
+    "make_task": "zrb.task.make_task",
+    "RsyncTask": "zrb.task.rsync_task",
+    "Scaffolder": "zrb.task.scaffolder",
+    "Scheduler": "zrb.task.scheduler",
+    "Task": "zrb.task.task",
+    "TcpCheck": "zrb.task.tcp_check",
+    "load_file": "zrb.util.load",
+    "load_module": "zrb.util.load",
+    "Xcom": "zrb.xcom.xcom",
+}
 
+if TYPE_CHECKING:
+    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 import CFG
+    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 import llm_config
+    from zrb.llm_rate_limitter import llm_rate_limitter
+    from zrb.runner.cli import cli
+    from zrb.runner.web_auth_config import web_auth_config
+    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.llm.history import ConversationHistoryData
+    from zrb.task.llm_task import LLMTask
+    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
+
+
+def __getattr__(name: str) -> Any:
+    if name in _LAZY_LOAD:
+        module = importlib.import_module(_LAZY_LOAD[name])
+        return getattr(module, name)
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
+
+
+# Eager load CFG
+CFG = __getattr__("CFG")
 if CFG.LOAD_BUILTIN:
     from zrb import builtin
 

zrb/config.py

@@ -167,11 +167,11 @@ class Config:
 
     @property
     def WEB_SUPER_ADMIN_USERNAME(self) -> str:
-        return os.getenv("ZRB_WEB_SUPERADMIN_USERNAME", "admin")
+        return os.getenv("ZRB_WEB_SUPER_ADMIN_USERNAME", "admin")
 
     @property
     def WEB_SUPER_ADMIN_PASSWORD(self) -> str:
-        return os.getenv("ZRB_WEB_SUPERADMIN_PASSWORD", "admin")
+        return os.getenv("ZRB_WEB_SUPER_ADMIN_PASSWORD", "admin")
 
     @property
     def WEB_ACCESS_TOKEN_COOKIE_NAME(self) -> str:

zrb/llm_config.py

@@ -48,37 +48,52 @@ DEFAULT_SYSTEM_PROMPT = (
 ).strip()
 
 DEFAULT_SPECIAL_INSTRUCTION_PROMPT = (
-    "If the user's request falls into a specialized category below, follow "
-    "the associated protocol.\n\n"
+    "If the user's request falls into a specialized category below, you MUST "
+    "follow the associated protocol.\n\n"
     "---\n"
     "## Software Engineering Protocol\n"
-    "This protocol applies to any request involving coding, debugging, or "
-    "other software development tasks.\n\n"
-    "### 1. Guiding Mandates\n"
-    "- **Safety First:** Never perform destructive actions without explicit "
-    "user confirmation. Explain critical commands before executing them.\n"
-    "- **Adhere to Conventions:** Your changes must blend in seamlessly with "
-    "the existing codebase. Match the formatting, naming, and architectural "
-    "patterns. *Never assume* a project's conventions; use your tools to "
-    "discover them.\n"
+    "This protocol applies to any request involving coding, debugging, "
+    "performance analysis, or other software development tasks.\n\n"
+    "### 1. Core Mandates\n"
+    "- **Safety First:** Never perform destructive actions (e.g., `rm -rf`) "
+    "without explicit user confirmation. Explain all critical commands before "
+    "executing them.\n"
+    "- **Adhere to Conventions:** Your changes MUST blend in seamlessly. "
+    "Match existing formatting, naming, and architectural patterns. *Never "
+    "assume* conventions; use your tools to discover them by reading "
+    "configuration files (e.g., `.gitignore`, `.flake8`), analyzing "
+    "neighboring code, and observing test patterns.\n"
     "- **No Assumptions on Dependencies:** Never assume a library or "
     "framework is available. Verify its presence in `package.json`, "
-    "`requirements.txt`, or similar files first.\n\n"
-    "### 2. Core Workflow\n"
-    "1.  **Understand:** Use your tools to analyze the relevant files and "
-    "codebase. Announce what you are inspecting (e.g., 'Okay, I will read "
-    "`main.py` and `utils.py` to understand the context.').\n"
-    "2.  **Plan:** Announce your step-by-step plan (e.g., 'Here is my plan: "
-    "1. Add the function to `utils.py`. 2. Import and call it in `main.py`. "
-    "3. Run the tests.').\n"
-    "3.  **Implement:** Execute the plan using your tools, following the "
-    "mandates above.\n"
-    "4.  **Verify:** After making changes, run the relevant verification "
-    "commands (e.g., tests, linters). Announce the command you will run "
-    "(e.g., 'Now, I will run `npm test` to verify the changes.').\n"
-    "5.  **Conclude:** Report the results of the verification step. If "
-    "successful, ask for the next step (e.g., 'All tests passed. Should I "
-    "commit these changes?')."
+    "`requirements.txt`, `pyproject.toml`, or similar files first.\n\n"
+    "### 2. Diagnostic & Implementation Workflow\n"
+    "1.  **Understand & Investigate (Tool-First):**\n"
+    "    - **Initial Assessment:** Start by using tools to understand the "
+    "task and environment. Do not begin by simply reading files. Formulate "
+    "an investigation plan.\n"
+    "    - **Information Gathering:** Use available tools to find relevant "
+    "information (e.g., project structure, framework, etc.) to complete the task.\n"
+    "    - **For Performance/Bugs:** If the task is to fix a bug or analyze "
+    "performance, your first step is to try and reproduce it. Then, use "
+    "diagnostic tools. For performance, use a profiler (e.g., `cProfile`, etc.). "
+    "For bugs, run the tests and analyze logs.\n"
+    "2.  **Plan:**\n"
+    "    - Announce your step-by-step plan based on your investigation "
+    "(e.g., 'Here is my plan: 1. Add a profiling decorator to the slow "
+    "function. 2. Run the code to generate a profile. 3. Analyze the "
+    "results to identify the bottleneck.').\n"
+    "3.  **Implement:**\n"
+    "    - Execute the plan using your tools, strictly following the Core "
+    "Mandates.\n"
+    "4.  **Verify:**\n"
+    "    - After making changes, ALWAYS run the relevant verification "
+    "commands (e.g., tests, linters, build scripts). Announce the command "
+    "you will run (e.g., 'Now, I will run `pytest` to verify the "
+    "changes.').\n"
+    "5.  **Conclude:**\n"
+    "    - Report the results of the verification. If successful, the task "
+    "is complete. If it fails, return to step 1 to diagnose the new "
+    "problem."
 ).strip()
 
 DEFAULT_SUMMARIZATION_PROMPT = (

zrb/task/llm/context_enrichment.py

@@ -75,7 +75,7 @@ async def enrich_context(
             return new_long_term_context
         else:
             ctx.log_warning("Context enrichment returned no data.")
-    except Exception as e:
+    except BaseException as e:
         ctx.log_warning(f"Error during context enrichment LLM call: {e}")
         traceback.print_exc()
 

zrb/task/llm/history_summarization.py

@@ -122,7 +122,7 @@ async def summarize_history(
             return new_summary
         else:
             ctx.log_warning("History summarization failed or returned no data.")
-    except Exception as e:
+    except BaseException as e:
         ctx.log_warning(f"Error during history summarization: {e}")
         traceback.print_exc()
 

{zrb-1.9.3.dist-info → zrb-1.9.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: zrb
-Version: 1.9.3
+Version: 1.9.4
 Summary: Your Automation Powerhouse
 Home-page: https://github.com/state-alchemists/zrb
 License: AGPL-3.0-or-later

{zrb-1.9.3.dist-info → zrb-1.9.4.dist-info}/RECORD

@@ -1,4 +1,4 @@
-zrb/__init__.py,sha256=6q6sd4KnIwerCPOjv0ouyoXYlmnCfzT87KApckucp9M,3180
+zrb/__init__.py,sha256=v0BgFP9Q22QpaJBAKqhzsWKOIsNflOcDxAA7H1xqDfE,5059
 zrb/__main__.py,sha256=aeIpBjlLef8bfdp0CYumnn5jVkHDPS5bwAxfuCJVUNI,2650
 zrb/attr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/attr/type.py,sha256=4TV5gPYMMrKh5V-yB6iRYKCbsXAH_AvGXMsjxKLHcUs,568
@@ -217,7 +217,7 @@ zrb/callback/callback.py,sha256=PFhCqzfxdk6IAthmXcZ13DokT62xtBzJr_ciLw6I8Zg,4030
 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=TRVQgUgX9yIso0Z6GioX8HASX_tITv5m_AP0Z4UIWb4,10574
+zrb/config.py,sha256=PbtmVWIdCTCFdeAIF5Y-J7wa99ZhLWIe_GMYqpZzhXk,10576
 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
@@ -246,7 +246,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=6T3MngWdUs0u0ZVs5Dl11w5KS7nN1RkgrIR_zKumzPM,3695
-zrb/llm_config.py,sha256=Cjfce8Idov3D9gFeU44ycpWjgd4uEeOjw7Zmn6IGTxk,18069
+zrb/llm_config.py,sha256=XjicvJDfygFW-pwnRCIyBc4kd6xps-Jo3NJvmQ1XHsQ,18902
 zrb/llm_rate_limitter.py,sha256=uM9zmSgV10fQq1dlaDGLDrv72uLj6ldBxMoGjO2Az14,4429
 zrb/runner/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/runner/cli.py,sha256=y_n7O4kQKXpiLAsBsf1WnHNfWgKlrxE9TU0MU1ICUMg,6989
@@ -340,10 +340,10 @@ zrb/task/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/task/llm/agent.py,sha256=IUGhU7vSfGVpVbd0rGLJ9kxIpt04Fm6UBjBWCVBVaZs,6695
 zrb/task/llm/config.py,sha256=WsO_kLXi0JXB6SPNL_dWm1Izu04KRrzmuX6DSrRYeBE,3483
 zrb/task/llm/context.py,sha256=LGGQ_mb1dWorfshHjGgXEW_pRweGj-6MZcIUFq3AHy4,2213
-zrb/task/llm/context_enrichment.py,sha256=MBs5lkI1GFZwtW_P7fJfIc6CWcAS0UWQeBO6ggSBSk0,6033
+zrb/task/llm/context_enrichment.py,sha256=YJPUKmNc4SI3a_y4I0xfBB24uJNE1mVTQ_LxFAEETyE,6037
 zrb/task/llm/error.py,sha256=s5PSK3ibGupMzOM0P81nstHWrMr3205H0FSDwfhWUDA,3662
 zrb/task/llm/history.py,sha256=cpaNqoEzsNAgzZGPPdohA8U5nTrVWmZ3P1Y5RTtXDgc,7986
-zrb/task/llm/history_summarization.py,sha256=TejPLktJmlxvhM-UmC29m_g6U9bC5e4fxRSEwt22GN0,6032
+zrb/task/llm/history_summarization.py,sha256=AhoBtzuGf4srANjxZqKC_0fKwyS-BxX6KXT54vo9E3w,6036
 zrb/task/llm/print_node.py,sha256=zocTKi9gZDxl2I6KNu095TmMc13Yip6SNuWYnswS680,4060
 zrb/task/llm/prompt.py,sha256=obSsLHF4AHlwMIEQ8XTJmtEnkLlD5XTEkk0RJUmIz4Y,3410
 zrb/task/llm/tool_wrapper.py,sha256=8_bL8m_WpRf-pVKSrvQIVqT-m2sUA87a1RBQG13lhp4,6457
@@ -391,7 +391,7 @@ zrb/util/todo.py,sha256=r9_KYF2-hLKMNjsp6AFK9zivykMrywd-kJ4bCwfdafI,19323
 zrb/util/todo_model.py,sha256=0SJ8aLYfJAscDOk5JsH7pXP3h1rAG91VMCS20-c2Y6A,1576
 zrb/xcom/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/xcom/xcom.py,sha256=o79rxR9wphnShrcIushA0Qt71d_p3ZTxjNf7x9hJB78,1571
-zrb-1.9.3.dist-info/METADATA,sha256=kFpbPlQwRBKBcKMQgRCnMuqvj2tamyF7j_qJIRYaOUo,9597
-zrb-1.9.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-zrb-1.9.3.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
-zrb-1.9.3.dist-info/RECORD,,
+zrb-1.9.4.dist-info/METADATA,sha256=tuGFlPCJqzMFk3D3UpOWxycH2am8H7s2blBVU9dwLbc,9597
+zrb-1.9.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+zrb-1.9.4.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
+zrb-1.9.4.dist-info/RECORD,,