zrb 1.5.16__py3-none-any.whl → 1.6.0__py3-none-any.whl

zrb/builtin/todo.py

@@ -4,7 +4,7 @@ import os
 from typing import Any
 
 from zrb.builtin.group import todo_group
-from zrb.config import TODO_DIR, TODO_RETENTION, TODO_VISUAL_FILTER
+from zrb.config import CFG
 from zrb.context.any_context import AnyContext
 from zrb.input.str_input import StrInput
 from zrb.input.text_input import TextInput
@@ -25,7 +25,9 @@ from zrb.util.todo import (
 )
 
 
-def _get_filter_input(allow_positional_parsing: bool = False) -> StrInput:
+def _get_filter_input(
+    allow_positional_parsing: bool = False, default: str | None = None
+) -> StrInput:
     return StrInput(
         name="filter",
         description="Visual filter",
@@ -33,7 +35,7 @@ def _get_filter_input(allow_positional_parsing: bool = False) -> StrInput:
         allow_empty=True,
         allow_positional_parsing=allow_positional_parsing,
         always_prompt=False,
-        default=TODO_VISUAL_FILTER,
+        default=default if default is not None else CFG.TODO_VISUAL_FILTER,
     )
 
 
@@ -70,12 +72,12 @@ def _get_filter_input(allow_positional_parsing: bool = False) -> StrInput:
     alias="add",
 )
 def add_todo(ctx: AnyContext):
-    todo_file_path = os.path.join(TODO_DIR, "todo.txt")
+    todo_file_path = os.path.join(CFG.TODO_DIR, "todo.txt")
     todo_list: list[TodoTaskModel] = []
     if os.path.isfile(todo_file_path):
         todo_list = load_todo_list(todo_file_path)
     else:
-        os.makedirs(TODO_DIR, exist_ok=True)
+        os.makedirs(CFG.TODO_DIR, exist_ok=True)
     todo_list.append(
         cascade_todo_task(
             TodoTaskModel(
@@ -106,7 +108,7 @@ def add_todo(ctx: AnyContext):
     alias="list",
 )
 def list_todo(ctx: AnyContext):
-    todo_file_path = os.path.join(TODO_DIR, "todo.txt")
+    todo_file_path = os.path.join(CFG.TODO_DIR, "todo.txt")
     todo_list: list[TodoTaskModel] = []
     if os.path.isfile(todo_file_path):
         todo_list = load_todo_list(todo_file_path)
@@ -121,7 +123,7 @@ def list_todo(ctx: AnyContext):
     alias="show",
 )
 def show_todo(ctx: AnyContext):
-    todo_file_path = os.path.join(TODO_DIR, "todo.txt")
+    todo_file_path = os.path.join(CFG.TODO_DIR, "todo.txt")
     todo_list: list[TodoTaskModel] = []
     todo_list: list[TodoTaskModel] = []
     if os.path.isfile(todo_file_path):
@@ -137,7 +139,7 @@ def show_todo(ctx: AnyContext):
     # Update todo task
     todo_task = cascade_todo_task(todo_task)
     task_id = todo_task.keyval.get("id", "")
-    log_work_path = os.path.join(TODO_DIR, "log-work", f"{task_id}.json")
+    log_work_path = os.path.join(CFG.TODO_DIR, "log-work", f"{task_id}.json")
     log_work_list = []
     if os.path.isfile(log_work_path):
         log_work_list = json.loads(read_file(log_work_path))
@@ -155,7 +157,7 @@ def show_todo(ctx: AnyContext):
     alias="complete",
 )
 def complete_todo(ctx: AnyContext):
-    todo_file_path = os.path.join(TODO_DIR, "todo.txt")
+    todo_file_path = os.path.join(CFG.TODO_DIR, "todo.txt")
     todo_list: list[TodoTaskModel] = []
     if os.path.isfile(todo_file_path):
         todo_list = load_todo_list(todo_file_path)
@@ -185,11 +187,11 @@ def complete_todo(ctx: AnyContext):
     alias="archive",
 )
 def archive_todo(ctx: AnyContext):
-    todo_file_path = os.path.join(TODO_DIR, "todo.txt")
+    todo_file_path = os.path.join(CFG.TODO_DIR, "todo.txt")
     todo_list: list[TodoTaskModel] = []
     if os.path.isfile(todo_file_path):
         todo_list = load_todo_list(todo_file_path)
-    retention_duration = datetime.timedelta(seconds=parse_duration(TODO_RETENTION))
+    retention_duration = datetime.timedelta(seconds=parse_duration(CFG.TODO_RETENTION))
     threshold_date = datetime.date.today() - retention_duration
     new_archived_todo_list = [
         todo_task
@@ -204,9 +206,9 @@ def archive_todo(ctx: AnyContext):
     if len(new_archived_todo_list) == 0:
         ctx.print("No completed task to archive")
         return get_visual_todo_list(todo_list, filter=ctx.input.filter)
-    archive_file_path = os.path.join(TODO_DIR, "archive.txt")
-    if not os.path.isdir(TODO_DIR):
-        os.make_dirs(TODO_DIR, exist_ok=True)
+    archive_file_path = os.path.join(CFG.TODO_DIR, "archive.txt")
+    if not os.path.isdir(CFG.TODO_DIR):
+        os.make_dirs(CFG.TODO_DIR, exist_ok=True)
     # Get archived todo list
     archived_todo_list = []
     if os.path.isfile(archive_file_path):
@@ -246,7 +248,7 @@ def archive_todo(ctx: AnyContext):
     alias="log",
 )
 def log_todo(ctx: AnyContext):
-    todo_file_path = os.path.join(TODO_DIR, "todo.txt")
+    todo_file_path = os.path.join(CFG.TODO_DIR, "todo.txt")
     todo_list: list[TodoTaskModel] = []
     if os.path.isfile(todo_file_path):
         todo_list = load_todo_list(todo_file_path)
@@ -264,7 +266,7 @@ def log_todo(ctx: AnyContext):
     # Save todo list
     save_todo_list(todo_file_path, todo_list)
     # Add log work
-    log_work_dir = os.path.join(TODO_DIR, "log-work")
+    log_work_dir = os.path.join(CFG.TODO_DIR, "log-work")
     os.makedirs(log_work_dir, exist_ok=True)
     log_work_file_path = os.path.join(
         log_work_dir, f"{todo_task.keyval.get('id')}.json"
@@ -286,7 +288,7 @@ def log_todo(ctx: AnyContext):
     write_file(log_work_file_path, json.dumps(log_work, indent=2))
     # get log work list
     task_id = todo_task.keyval.get("id", "")
-    log_work_path = os.path.join(TODO_DIR, "log-work", f"{task_id}.json")
+    log_work_path = os.path.join(CFG.TODO_DIR, "log-work", f"{task_id}.json")
     log_work_list = []
     if os.path.isfile(log_work_path):
         log_work_list = json.loads(read_file(log_work_path))
@@ -333,14 +335,14 @@ def edit_todo(ctx: AnyContext):
         if line.strip() != ""
     ]
     new_content = "\n".join(todo_task_to_line(todo_task) for todo_task in todo_list)
-    todo_file_path = os.path.join(TODO_DIR, "todo.txt")
+    todo_file_path = os.path.join(CFG.TODO_DIR, "todo.txt")
     write_file(todo_file_path, new_content)
     todo_list = load_todo_list(todo_file_path)
     return get_visual_todo_list(todo_list, filter=ctx.input.filter)
 
 
 def _get_todo_txt_content() -> str:
-    todo_file_path = os.path.join(TODO_DIR, "todo.txt")
+    todo_file_path = os.path.join(CFG.TODO_DIR, "todo.txt")
     if not os.path.isfile(todo_file_path):
         return ""
     return read_file(todo_file_path)

zrb/callback/any_callback.py

@@ -6,5 +6,5 @@ from zrb.session.any_session import AnySession
 
 class AnyCallback(ABC):
     @abstractmethod
-    async def async_run(self, session: AnySession) -> Any:
+    async def async_run(self, parent_session: AnySession, session: AnySession) -> Any:
         pass

zrb/callback/callback.py

@@ -5,23 +5,85 @@ from zrb.callback.any_callback import AnyCallback
 from zrb.session.any_session import AnySession
 from zrb.task.any_task import AnyTask
 from zrb.util.attr import get_str_dict_attr
+from zrb.util.string.conversion import to_snake_case
+from zrb.xcom.xcom import Xcom
 
 
 class Callback(AnyCallback):
+    """
+    Represents a callback that runs a task after a trigger or scheduler event.
+
+    It handles input mapping and can publish results and session names
+    back to the parent session via XCom.
+    """
+
     def __init__(
         self,
-        input_mapping: StrDictAttr,
         task: AnyTask,
-        auto_render: bool = True,
+        input_mapping: StrDictAttr,
+        render_input_mapping: bool = True,
+        result_queue: str | None = None,
+        session_name_queue: str | None = None,
     ):
-        self._input_mapping = input_mapping
+        """
+        Initializes a new instance of the Callback class.
+
+        Args:
+            task: The task to be executed by the callback.
+            input_mapping: A dictionary or attribute mapping to prepare inputs for the task.
+            render_input_mapping: Whether to render the input mapping using
+                f-string like syntax.
+            result_queue: The name of the XCom queue in the parent session
+                to publish the task result.
+            session_name_queue: The name of the XCom queue in the parent
+                session to publish the session name.
+        """
         self._task = task
-        self._auto_render = auto_render
+        self._input_mapping = input_mapping
+        self._render_input_mapping = render_input_mapping
+        self._result_queue = result_queue
+        self._session_name_queue = session_name_queue
 
-    async def async_run(self, session: AnySession) -> Any:
+    async def async_run(self, parent_session: AnySession, session: AnySession) -> Any:
+        self._maybe_publish_session_name_to_parent_session(
+            parent_session=parent_session, session=session
+        )
+        # prepare input
         inputs = get_str_dict_attr(
-            session.shared_ctx, self._input_mapping, auto_render=self._auto_render
+            session.shared_ctx,
+            self._input_mapping,
+            auto_render=self._render_input_mapping,
         )
         for name, value in inputs.items():
             session.shared_ctx.input[name] = value
-        return await self._task.async_run(session)
+            session.shared_ctx.input[to_snake_case(name)] = value
+        # run task and get result
+        result = await self._task.async_run(session)
+        self._maybe_publish_result_to_parent_session(parent_session, result)
+        return result
+
+    def _maybe_publish_session_name_to_parent_session(
+        self, parent_session: AnySession, session: AnySession
+    ):
+        self._maybe_publish_to_parent_session(
+            parent_session=parent_session,
+            xcom_name=self._session_name_queue,
+            value=session.name,
+        )
+
+    def _maybe_publish_result_to_parent_session(
+        self, parent_session: AnySession, result: Any
+    ):
+        self._maybe_publish_to_parent_session(
+            parent_session=parent_session, xcom_name=self._result_queue, value=result
+        )
+
+    def _maybe_publish_to_parent_session(
+        self, parent_session: AnySession, xcom_name: str | None, value: Any
+    ):
+        if xcom_name is None:
+            return
+        parent_xcom = parent_session.shared_ctx.xcom
+        if xcom_name not in parent_xcom:
+            parent_xcom[xcom_name] = Xcom([])
+        parent_xcom[xcom_name].push(value)

zrb/config.py

@@ -2,108 +2,278 @@ import importlib.metadata as metadata
 import logging
 import os
 import platform
+from textwrap import dedent
 
 from zrb.util.string.conversion import to_boolean
+from zrb.util.string.format import fstring_format
 
-
-def _get_current_shell() -> str:
-    if platform.system() == "Windows":
-        return "PowerShell"
-    current_shell = os.getenv("SHELL", "")
-    if current_shell.endswith("zsh"):
-        return "zsh"
-    return "bash"
-
-
-def _get_log_level(level: str) -> int:
-    level = level.upper()
-    log_levels = {
-        "CRITICAL": logging.CRITICAL,  # 50
-        "ERROR": logging.ERROR,  # 40
-        "WARN": logging.WARNING,  # 30
-        "WARNING": logging.WARNING,  # 30
-        "INFO": logging.INFO,  # 20
-        "DEBUG": logging.DEBUG,  # 10
-        "NOTSET": logging.NOTSET,  # 0
-    }
-    if level in log_levels:
-        return log_levels[level]
-    return logging.WARNING
-
-
-LOGGER = logging.getLogger()
-DEFAULT_SHELL = os.getenv("ZRB_SHELL", _get_current_shell())
-DEFAULT_EDITOR = os.getenv("ZRB_EDITOR", "nano")
-INIT_MODULES_STR = os.getenv("ZRB_INIT_MODULES", "")
-INIT_MODULES = (
-    [module.strip() for module in INIT_MODULES_STR.split(":") if module.strip() != ""]
-    if INIT_MODULES_STR != ""
-    else []
-)
-INIT_SCRIPTS_STR = os.getenv("ZRB_INIT_SCRIPTS", "")
-INIT_SCRIPTS = (
-    [script.strip() for script in INIT_SCRIPTS_STR.split(":") if script.strip() != ""]
-    if INIT_SCRIPTS_STR != ""
-    else []
-)
-LOGGING_LEVEL = _get_log_level(os.getenv("ZRB_LOGGING_LEVEL", "WARNING"))
-LOAD_BUILTIN = to_boolean(os.getenv("ZRB_LOAD_BUILTIN", "1"))
-WARN_UNRECOMMENDED_COMMAND = to_boolean(
-    os.getenv("ZRB_WARN_UNRECOMMENDED_COMMAND", "1")
-)
-SESSION_LOG_DIR = os.getenv(
-    "ZRB_SESSION_LOG_DIR", os.path.expanduser(os.path.join("~", ".zrb-session"))
-)
-TODO_DIR = os.getenv("ZRB_TODO_DIR", os.path.expanduser(os.path.join("~", "todo")))
-TODO_VISUAL_FILTER = os.getenv("ZRB_TODO_FILTER", "")
-TODO_RETENTION = os.getenv("ZRB_TODO_RETENTION", "2w")
-VERSION = metadata.version("zrb")
-WEB_HTTP_PORT = int(os.getenv("ZRB_WEB_HTTP_PORT", "21213"))
-WEB_GUEST_USERNAME = os.getenv("ZRB_WEB_GUEST_USERNAME", "user")
-WEB_SUPER_ADMIN_USERNAME = os.getenv("ZRB_WEB_SUPERADMIN_USERNAME", "admin")
-WEB_SUPER_ADMIN_PASSWORD = os.getenv("ZRB_WEB_SUPERADMIN_PASSWORD", "admin")
-WEB_ACCESS_TOKEN_COOKIE_NAME = os.getenv(
-    "ZRB_WEB_ACCESS_TOKEN_COOKIE_NAME", "access_token"
-)
-WEB_REFRESH_TOKEN_COOKIE_NAME = os.getenv(
-    "ZRB_WEB_REFRESH_TOKEN_COOKIE_NAME", "refresh_token"
-)
-WEB_SECRET_KEY = os.getenv("ZRB_WEB_SECRET", "zrb")
-WEB_ENABLE_AUTH = to_boolean(os.getenv("ZRB_WEB_ENABLE_AUTH", "0"))
-WEB_AUTH_ACCESS_TOKEN_EXPIRE_MINUTES = int(
-    os.getenv("ZRB_WEB_ACCESS_TOKEN_EXPIRE_MINUTES", "30")
-)
-WEB_AUTH_REFRESH_TOKEN_EXPIRE_MINUTES = int(
-    os.getenv("ZRB_WEB_REFRESH_TOKEN_EXPIRE_MINUTES", "60")
-)
-
-LLM_HISTORY_DIR = os.getenv(
-    "ZRB_LLM_HISTORY_DIR", os.path.expanduser(os.path.join("~", ".zrb-llm-history"))
-)
-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"))
-# 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"))
-SERP_API_KEY = os.getenv("SERP_API_KEY", "")
-
-
-BANNER = f"""
+_DEFAULT_BANNER = """
                 bb
    zzzzz rr rr  bb
      zz  rrr  r bbbbbb
     zz   rr     bb   bb
    zzzzz rr     bbbbbb   {VERSION} Janggala
    _ _ . .  . _ .  _ . . .
-
 Your Automation Powerhouse
-
 ☕ Donate at: https://stalchmst.com/donation
 🐙 Submit issues/PR at: https://github.com/state-alchemists/zrb
 🐤 Follow us at: https://twitter.com/zarubastalchmst
 """
+
+
+class Config:
+    @property
+    def LOGGER(self) -> logging.Logger:
+        return logging.getLogger()
+
+    @property
+    def DEFAULT_SHELL(self) -> str:
+        return os.getenv("ZRB_SHELL", self._get_current_shell())
+
+    def _get_current_shell(self) -> str:
+        if platform.system() == "Windows":
+            return "PowerShell"
+        current_shell = os.getenv("SHELL", "")
+        if current_shell.endswith("zsh"):
+            return "zsh"
+        return "bash"
+
+    @property
+    def DEFAULT_EDITOR(self) -> str:
+        return os.getenv("ZRB_EDITOR", "nano")
+
+    @property
+    def INIT_MODULES(self) -> list[str]:
+        init_modules_str = os.getenv("ZRB_INIT_MODULES", "")
+        return (
+            [
+                module.strip()
+                for module in init_modules_str.split(":")
+                if module.strip() != ""
+            ]
+            if init_modules_str != ""
+            else []
+        )
+
+    @property
+    def INIT_SCRIPTS(self) -> list[str]:
+        init_scripts_str = os.getenv("ZRB_INIT_SCRIPTS", "")
+        return (
+            [
+                script.strip()
+                for script in init_scripts_str.split(":")
+                if script.strip() != ""
+            ]
+            if init_scripts_str != ""
+            else []
+        )
+
+    @property
+    def INIT_FILE_NAME(self) -> str:
+        return os.getenv("ZRB_INIT_FILE_NAME", "zrb_init.py")
+
+    @property
+    def LOGGING_LEVEL(self) -> int:
+        return self._get_log_level(os.getenv("ZRB_LOGGING_LEVEL", "WARNING"))
+
+    def _get_log_level(self, level: str) -> int:
+        level = level.upper()
+        log_levels = {
+            "CRITICAL": logging.CRITICAL,  # 50
+            "ERROR": logging.ERROR,  # 40
+            "WARN": logging.WARNING,  # 30
+            "WARNING": logging.WARNING,  # 30
+            "INFO": logging.INFO,  # 20
+            "DEBUG": logging.DEBUG,  # 10
+            "NOTSET": logging.NOTSET,  # 0
+        }
+        if level in log_levels:
+            return log_levels[level]
+        return logging.WARNING
+
+    @property
+    def LOAD_BUILTIN(self) -> bool:
+        return to_boolean(os.getenv("ZRB_LOAD_BUILTIN", "1"))
+
+    @property
+    def WARN_UNRECOMMENDED_COMMAND(self) -> bool:
+        return to_boolean(os.getenv("ZRB_WARN_UNRECOMMENDED_COMMAND", "1"))
+
+    @property
+    def SESSION_LOG_DIR(self) -> str:
+        return os.getenv(
+            "ZRB_SESSION_LOG_DIR", os.path.expanduser(os.path.join("~", ".zrb-session"))
+        )
+
+    @property
+    def TODO_DIR(self) -> str:
+        return os.getenv("ZRB_TODO_DIR", os.path.expanduser(os.path.join("~", "todo")))
+
+    @property
+    def TODO_VISUAL_FILTER(self) -> str:
+        return os.getenv("ZRB_TODO_FILTER", "")
+
+    @property
+    def TODO_RETENTION(self) -> str:
+        return os.getenv("ZRB_TODO_RETENTION", "2w")
+
+    @property
+    def VERSION(self) -> str:
+        return metadata.version("zrb")
+
+    @property
+    def WEB_HTTP_PORT(self) -> int:
+        return int(os.getenv("ZRB_WEB_HTTP_PORT", "21213"))
+
+    @property
+    def WEB_GUEST_USERNAME(self) -> str:
+        return os.getenv("ZRB_WEB_GUEST_USERNAME", "user")
+
+    @property
+    def WEB_SUPER_ADMIN_USERNAME(self) -> str:
+        return os.getenv("ZRB_WEB_SUPERADMIN_USERNAME", "admin")
+
+    @property
+    def WEB_SUPER_ADMIN_PASSWORD(self) -> str:
+        return os.getenv("ZRB_WEB_SUPERADMIN_PASSWORD", "admin")
+
+    @property
+    def WEB_ACCESS_TOKEN_COOKIE_NAME(self) -> str:
+        return os.getenv("ZRB_WEB_ACCESS_TOKEN_COOKIE_NAME", "access_token")
+
+    @property
+    def WEB_REFRESH_TOKEN_COOKIE_NAME(self) -> str:
+        return os.getenv("ZRB_WEB_REFRESH_TOKEN_COOKIE_NAME", "refresh_token")
+
+    @property
+    def WEB_SECRET_KEY(self) -> str:
+        return os.getenv("ZRB_WEB_SECRET", "zrb")
+
+    @property
+    def WEB_ENABLE_AUTH(self) -> bool:
+        return to_boolean(os.getenv("ZRB_WEB_ENABLE_AUTH", "0"))
+
+    @property
+    def WEB_AUTH_ACCESS_TOKEN_EXPIRE_MINUTES(self) -> int:
+        return int(os.getenv("ZRB_WEB_ACCESS_TOKEN_EXPIRE_MINUTES", "30"))
+
+    @property
+    def WEB_AUTH_REFRESH_TOKEN_EXPIRE_MINUTES(self) -> int:
+        return int(os.getenv("ZRB_WEB_REFRESH_TOKEN_EXPIRE_MINUTES", "60"))
+
+    @property
+    def WEB_TITLE(self) -> str:
+        return os.getenv("ZRB_WEB_TITLE", "Zrb")
+
+    @property
+    def WEB_JARGON(self) -> str:
+        return os.getenv("ZRB_WEB_JARGON", "Your Automation PowerHouse")
+
+    @property
+    def WEB_HOMEPAGE_INTRO(self) -> str:
+        return os.getenv("ZRB_WEB_HOMEPAGE_INTRO", "Welcome to Zrb Web Interface")
+
+    @property
+    def LLM_MODEL(self) -> str | None:
+        return os.getenv("ZRB_LLM_MODEL", None)
+
+    @property
+    def LLM_BASE_URL(self) -> str | None:
+        return os.getenv("ZRB_LLM_BASE_URL", None)
+
+    @property
+    def LLM_API_KEY(self) -> str | None:
+        return os.getenv("ZRB_LLM_API_KEY", None)
+
+    @property
+    def LLM_SYSTEM_PROMPT(self) -> str | None:
+        return os.getenv("ZRB_LLM_SYSTEM_PROMPT", None)
+
+    @property
+    def LLM_PERSONA(self) -> str | None:
+        return os.getenv("ZRB_LLM_PERSONA", None)
+
+    @property
+    def LLM_SPECIAL_INSTRUCTION_PROMPT(self) -> str | None:
+        return os.getenv("ZRB_LLM_SPECIAL_INSTRUCTION_PROMPT", None)
+
+    @property
+    def LLM_SUMMARIZATION_PROMPT(self) -> str | None:
+        return os.getenv("ZRB_LLM_SUMMARIZATION_PROMPT", None)
+
+    @property
+    def LLM_CONTEXT_ENRICHMENT_PROMPT(self) -> str | None:
+        return os.getenv("ZRB_LLM_CONTEXT_ENRICHMENT_PROMPT", None)
+
+    @property
+    def LLM_SUMMARIZE_HISTORY(self) -> bool:
+        return to_boolean(os.getenv("ZRB_LLM_SUMMARIZE_HISTORY", "true"))
+
+    @property
+    def LLM_HISTORY_SUMMARIZATION_THRESHOLD(self) -> int:
+        return int(os.getenv("ZRB_LLM_HISTORY_SUMMARIZATION_THRESHOLD", "5"))
+
+    @property
+    def LLM_ENRICH_CONTEXT(self) -> bool:
+        return to_boolean(os.getenv("ZRB_LLM_ENRICH_CONTEXT", "true"))
+
+    @property
+    def LLM_CONTEXT_ENRICHMENT_THRESHOLD(self) -> int:
+        return int(os.getenv("ZRB_LLM_CONTEXT_ENRICHMENT_THRESHOLD", "5"))
+
+    @property
+    def LLM_HISTORY_DIR(self) -> str:
+        return os.getenv(
+            "ZRB_LLM_HISTORY_DIR",
+            os.path.expanduser(os.path.join("~", ".zrb-llm-history")),
+        )
+
+    @property
+    def LLM_ALLOW_ACCESS_LOCAL_FILE(self) -> bool:
+        return to_boolean(os.getenv("ZRB_LLM_ACCESS_LOCAL_FILE", "1"))
+
+    @property
+    def LLM_ALLOW_ACCESS_SHELL(self) -> bool:
+        return to_boolean(os.getenv("ZRB_LLM_ACCESS_SHELL", "1"))
+
+    @property
+    def LLM_ALLOW_ACCESS_INTERNET(self) -> bool:
+        return to_boolean(os.getenv("ZRB_LLM_ACCESS_INTERNET", "1"))
+
+    @property
+    def RAG_EMBEDDING_API_KEY(self) -> str:
+        return os.getenv("ZRB_RAG_EMBEDDING_API_KEY", None)
+
+    @property
+    def RAG_EMBEDDING_BASE_URL(self) -> str:
+        return os.getenv("ZRB_RAG_EMBEDDING_BASE_URL", None)
+
+    @property
+    def RAG_EMBEDDING_MODEL(self) -> str:
+        return os.getenv("ZRB_RAG_EMBEDDING_MODEL", "text-embedding-ada-002")
+
+    @property
+    def RAG_CHUNK_SIZE(self) -> int:
+        return int(os.getenv("ZRB_RAG_CHUNK_SIZE", "1024"))
+
+    @property
+    def RAG_OVERLAP(self) -> int:
+        return int(os.getenv("ZRB_RAG_OVERLAP", "128"))
+
+    @property
+    def RAG_MAX_RESULT_COUNT(self) -> int:
+        return int(os.getenv("ZRB_RAG_MAX_RESULT_COUNT", "5"))
+
+    @property
+    def SERP_API_KEY(self) -> str:
+        return os.getenv("SERP_API_KEY", "")
+
+    @property
+    def BANNER(self) -> str:
+        return fstring_format(
+            os.getenv("ZRB_BANNER", _DEFAULT_BANNER),
+            {"VERSION": self.VERSION},
+        )
+
+
+CFG = Config()

zrb/context/shared_context.py

@@ -1,7 +1,7 @@
 import datetime
 from typing import Any
 
-from zrb.config import LOGGING_LEVEL
+from zrb.config import CFG
 from zrb.context.any_shared_context import AnySharedContext
 from zrb.dot_dict.dot_dict import DotDict
 from zrb.session.any_session import AnySession
@@ -25,7 +25,7 @@ class SharedContext(AnySharedContext):
         args: list[Any] = [],
         env: dict[str, str] = {},
         xcom: dict[str, Xcom] = {},
-        logging_level: int = LOGGING_LEVEL,
+        logging_level: int | None = None,
     ):
         self.__logging_level = logging_level
         self._input = DotDict(input)
@@ -79,6 +79,8 @@ class SharedContext(AnySharedContext):
         self._session = session
 
     def get_logging_level(self) -> int:
+        if self.__logging_level is None:
+            return CFG.LOGGING_LEVEL
         return self.__logging_level
 
     def render(self, template: str) -> str:

zrb/input/text_input.py

@@ -3,7 +3,7 @@ import subprocess
 import tempfile
 from collections.abc import Callable
 
-from zrb.config import DEFAULT_EDITOR
+from zrb.config import CFG
 from zrb.context.any_shared_context import AnySharedContext
 from zrb.input.base_input import BaseInput
 from zrb.util.file import read_file
@@ -20,7 +20,7 @@ class TextInput(BaseInput):
         allow_empty: bool = False,
         allow_positional_parsing: bool = True,
         always_prompt: bool = True,
-        editor: str = DEFAULT_EDITOR,
+        editor: str | None = None,
         extension: str = ".txt",
         comment_start: str | None = None,
         comment_end: str | None = None,
@@ -86,10 +86,13 @@ class TextInput(BaseInput):
             if default_value:
                 temp_file.write(default_value.encode())
             temp_file.flush()
-        # Open the editor
-        subprocess.call([self._editor, temp_file_name])
-        # Read the edited content
-        edited_content = read_file(temp_file_name)
+            # Open the editor
+            editor_cmd = (
+                self._editor if self._editor is not None else CFG.DEFAULT_EDITOR
+            )
+            subprocess.call([editor_cmd, temp_file_name])
+            # Read the edited content
+            edited_content = read_file(temp_file_name)
         parts = [
             text.strip() for text in edited_content.split(comment_prompt_message, 1)
         ]