zrb 1.5.9__py3-none-any.whl → 1.5.11__py3-none-any.whl

zrb/builtin/llm/tool/file.py

@@ -92,6 +92,10 @@ def list_files(
     """
     all_files: list[str] = []
     abs_path = os.path.abspath(os.path.expanduser(path))
+    # Explicitly check if path exists before proceeding
+    if not os.path.exists(abs_path):
+        # Raise FileNotFoundError, which is a subclass of OSError
+        raise FileNotFoundError(f"Path does not exist: {path}")
     # Determine effective exclusion patterns
     patterns_to_exclude = (
         excluded_patterns
@@ -132,9 +136,7 @@ def list_files(
                     all_files.append(full_path)
         # Return paths relative to the original path requested
         try:
-            rel_files = [
-                os.path.relpath(f, os.path.dirname(abs_path)) for f in all_files
-            ]
+            rel_files = [os.path.relpath(f, abs_path) for f in all_files]
             return json.dumps({"files": sorted(rel_files)})
         except (
             ValueError
@@ -152,14 +154,17 @@ def list_files(
 
 def _is_hidden(path: str) -> bool:
     """
-    Check if path is hidden (starts with '.').
+    Check if path is hidden (starts with '.') but ignore '.' and '..'.
     Args:
         path: File or directory path to check
     Returns:
         True if the path is hidden, False otherwise
     """
-    # Extract just the basename to check if it starts with a dot
-    return os.path.basename(path).startswith(".")
+    basename = os.path.basename(path)
+    # Ignore '.' and '..' as they are not typically considered hidden in listings
+    if basename == "." or basename == "..":
+        return False
+    return basename.startswith(".")
 
 
 def _is_excluded(name: str, patterns: list[str]) -> bool:
@@ -375,41 +380,27 @@ def _get_file_matches(
 
 def apply_diff(
     path: str,
-    diff: str,
-    search_marker: str = "<<<<<< SEARCH",
-    meta_marker: str = "------",
-    separator: str = "======",
-    replace_marker: str = ">>>>>> REPLACE",
+    start_line: int,
+    end_line: int,
+    search_content: str,
+    replace_content: str,
 ) -> str:
-    """Apply a precise search/replace diff to a file.
+    """Apply a precise search/replace to a file based on line numbers and content.
     Args:
         path (str): Path to modify. Pass exactly as provided, including '~'.
-        diff (str): Search/replace block defining changes (see format example below).
-        search_marker (str): Marker for start of search block. Defaults to "<<<<<< SEARCH".
-        meta_marker (str): Marker for start of content to search for. Defaults to "------".
-        separator (str): Marker separating search/replace content. Defaults to "======".
-        replace_marker (str): Marker for end of replacement block.
-            Defaults to ">>>>>> REPLACE".
-    SEARCH block must exactly match file content including whitespace/indentation.
-    SEARCH block should NOT contains line numbers
-    Format example:
-        [Search Marker, e.g., <<<<<< SEARCH]
-        :start_line:10
-        :end_line:15
-        [Meta Marker, e.g., ------]
-        [exact content to find including whitespace]
-        [Separator, e.g., ======]
-        [new content to replace with]
-        [Replace Marker, e.g., >>>>>> REPLACE]
+        start_line (int): The 1-based starting line number of the content to replace.
+        end_line (int): The 1-based ending line number (inclusive) of the content to replace.
+        search_content (str): The exact content expected to be found in the specified
+            line range. Must exactly match file content including whitespace/indentation,
+            excluding line numbers.
+        replace_content (str): The new content to replace the search_content with.
+            Excluding line numbers.
     Returns:
         str: JSON: {"success": true, "path": "f.py"} or {"success": false, "error": "..."}
     Raises:
         Exception: If an error occurs.
     """
     try:
-        start_line, end_line, search_content, replace_content = _parse_diff(
-            diff, search_marker, meta_marker, separator, replace_marker
-        )
         abs_path = os.path.abspath(os.path.expanduser(path))
         if not os.path.exists(abs_path):
             return json.dumps(
@@ -453,62 +444,3 @@ def apply_diff(
         raise OSError(f"Error applying diff to {path}: {e}")
     except Exception as e:
         raise RuntimeError(f"Unexpected error applying diff to {path}: {e}")
-
-
-def _parse_diff(
-    diff: str,
-    search_marker: str,
-    meta_marker: str,
-    separator: str,
-    replace_marker: str,
-) -> tuple[int, int, str, str]:
-    """
-    Parse diff content into components.
-    Args:
-        diff: The diff content to parse
-        search_marker: Marker indicating the start of the search block
-        meta_marker: Marker indicating the start of the content to search for
-        separator: Marker separating search content from replacement content
-        replace_marker: Marker indicating the end of the replacement block
-    Returns:
-        Tuple of (start_line, end_line, search_content, replace_content)
-    Raises:
-        ValueError: If diff format is invalid or missing required markers
-        ValueError: If start_line or end_line cannot be parsed
-    """
-    # Find all marker positions
-    search_start_idx = diff.find(search_marker)
-    meta_start_idx = diff.find(meta_marker)
-    separator_idx = diff.find(separator)
-    replace_end_idx = diff.find(replace_marker)
-    # Validate all markers are present
-    missing_markers = []
-    if search_start_idx == -1:
-        missing_markers.append("search marker")
-    if meta_start_idx == -1:
-        missing_markers.append("meta marker")
-    if separator_idx == -1:
-        missing_markers.append("separator")
-    if replace_end_idx == -1:
-        missing_markers.append("replace marker")
-    if missing_markers:
-        raise ValueError(f"Invalid diff format - missing: {', '.join(missing_markers)}")
-    # Extract metadata
-    meta_content = diff[search_start_idx + len(search_marker) : meta_start_idx].strip()
-    # Parse line numbers
-    start_line_match = re.search(r":start_line:(\d+)", meta_content)
-    end_line_match = re.search(r":end_line:(\d+)", meta_content)
-    if not start_line_match:
-        raise ValueError("Missing start_line in diff metadata")
-    if not end_line_match:
-        raise ValueError("Missing end_line in diff metadata")
-    start_line = int(start_line_match.group(1))
-    end_line = int(end_line_match.group(1))
-    # Extract content sections
-    search_content = diff[meta_start_idx + len(meta_marker) : separator_idx].strip(
-        "\r\n"
-    )
-    replace_content = diff[separator_idx + len(separator) : replace_end_idx].strip(
-        "\r\n"
-    )
-    return start_line, end_line, search_content, replace_content

zrb/llm_config.py

@@ -5,62 +5,45 @@ from pydantic_ai.models.openai import OpenAIModel
 from pydantic_ai.providers import Provider
 from pydantic_ai.providers.openai import OpenAIProvider
 
+from zrb.util.string.conversion import to_boolean
+
 DEFAULT_SYSTEM_PROMPT = """
-You are a highly capable AI assistant with access to tools. Your primary goal is to
-provide accurate, reliable, and helpful responses.
-
-Key Instructions:
-1.  **Prioritize Tool Use:** Always attempt to use available tools to find
-    information or perform actions before asking the user.
-2.  **Correct Tool Invocation:** Use tools precisely as defined. Provide arguments
-    in valid JSON where required. Do not pass arguments to tools that don't accept
-    them. Handle tool errors gracefully and retry or adapt your strategy if necessary.
-3.  **Accuracy is Paramount:** Ensure all information, code, or outputs provided are
-    correct and reliable. Verify facts and generate executable code when requested.
-4.  **Clarity and Conciseness:** Respond clearly and directly to the user's query
-    after gathering the necessary information. Avoid unnecessary conversation.
-5.  **Context Awareness:** Understand the user's request fully to provide the most
-    relevant and effective assistance.
+You have access to tools.
+Your goal is to complete the user's task efficiently.
+Analyze the request and use the available tools proactively to achieve the goal.
+Infer parameters and actions from the context.
+Do not ask for confirmation unless strictly necessary due to ambiguity or
+missing critical information.
+Apply relevant domain knowledge and best practices.
+Respond directly and concisely upon task completion or when clarification is essential.
 """.strip()
 
 DEFAULT_PERSONA = """
 You are an expert in various fields including technology, science, history, and more.
 """.strip()
 
+# Concise summarization focused on preserving critical context for continuity.
 DEFAULT_SUMMARIZATION_PROMPT = """
-You are a summarization assistant. Your task is to synthesize the provided 
-conversation history and the existing context (which might contain a
-previous 'history_summary') into a comprehensive, updated summary
-Carefully review the '# Current Context' which includes any previous summary
-('history_summary').
-Then, review the '# Conversation History to Summarize'.
-Combine the information from both the existing context/summary and the new
-history. Create a single, coherent summary that captures ALL essential
-information, including:
-- Key decisions made
-- Actions taken (including tool usage and their results)
-- Important facts or data points mentioned
-- Outcomes of discussions or actions
-- Any unresolved questions or pending items
-The goal is to provide a complete yet concise background so that the main
-assistant can seamlessly continue the conversation without losing critical
-context from the summarized parts.
-Output *only* the updated summary text."
+You are a summarization assistant. Create an updated, concise summary integrating
+the previous summary (if any) with the new conversation history.
+Your primary goal is to preserve ALL critical context needed for the main assistant
+to continue the task effectively. This includes key facts, decisions, tool usage
+results, and essential background. Do not omit details that would force the main
+assistant to re-gather information.
+Output *only* the updated summary text.
 """.strip()
 
 DEFAULT_CONTEXT_ENRICHMENT_PROMPT = """
 You are an information extraction assistant.
-Analyze the conversation history and current context to extract key facts such as:
-  - user_name
-  - user_roles
-  - preferences
-  - goals
-  - etc
-Return only a JSON object containing a single key "response",
-whose value is another JSON object with these details.
+Analyze the conversation history and current context to extract key facts like
+user_name, user_roles, preferences, goals, etc.
+Return only a JSON object containing a single key "response", whose value is
+another JSON object with these details.
 If nothing is found, output {"response": {}}.
 """.strip()
 
+DEFAULT_SPECIAL_INSTRUCTION_PROMPT = ""  # Default to empty
+
 
 class LLMConfig:
 
@@ -71,6 +54,7 @@ class LLMConfig:
         default_api_key: str | None = None,
         default_persona: str | None = None,
         default_system_prompt: str | None = None,
+        default_special_instruction_prompt: str | None = None,
         default_summarization_prompt: str | None = None,
         default_context_enrichment_prompt: str | None = None,
         default_summarize_history: bool | None = None,
@@ -102,6 +86,11 @@ class LLMConfig:
             if default_persona is not None
             else os.getenv("ZRB_LLM_PERSONA", None)
         )
+        self._default_special_instruction_prompt = (
+            default_special_instruction_prompt
+            if default_special_instruction_prompt is not None
+            else os.getenv("ZRB_LLM_SPECIAL_INSTRUCTION_PROMPT", None)
+        )
         self._default_summarization_prompt = (
             default_summarization_prompt
             if default_summarization_prompt is not None
@@ -125,7 +114,7 @@ class LLMConfig:
         self._default_enrich_context = (
             default_enrich_context
             if default_enrich_context is not None
-            else os.getenv("ZRB_LLM_ENRICH_CONTEXT", "false").lower() == "true"
+            else to_boolean(os.getenv("ZRB_LLM_ENRICH_CONTEXT", "0"))
         )
         self._default_provider = None
         self._default_model = None
@@ -145,17 +134,23 @@ class LLMConfig:
         )
 
     def get_default_system_prompt(self) -> str:
-        system_prompt = (
+        return (
             DEFAULT_SYSTEM_PROMPT
             if self._default_system_prompt is None
             else self._default_system_prompt
         )
-        persona = (
+
+    def get_default_persona(self) -> str:
+        return (
             DEFAULT_PERSONA if self._default_persona is None else self._default_persona
         )
-        if persona is not None:
-            return f"{persona}\n{system_prompt}"
-        return system_prompt
+
+    def get_default_special_instruction_prompt(self) -> str:
+        return (
+            DEFAULT_SPECIAL_INSTRUCTION_PROMPT
+            if self._default_special_instruction_prompt is None
+            else self._default_special_instruction_prompt
+        )
 
     def get_default_summarization_prompt(self) -> str:
         return (
@@ -197,6 +192,9 @@ class LLMConfig:
     def set_default_system_prompt(self, system_prompt: str):
         self._default_system_prompt = system_prompt
 
+    def set_default_special_instruction_prompt(self, special_instruction_prompt: str):
+        self._default_special_instruction_prompt = special_instruction_prompt
+
     def set_default_summarization_prompt(self, summarization_prompt: str):
         self._default_summarization_prompt = summarization_prompt
 

zrb/task/base/lifecycle.py

@@ -77,7 +77,7 @@ async def run_task_async(
     fill_shared_context_envs(session.shared_ctx)  # Inject OS env vars
 
     # Start the execution chain from the root tasks
-    result = await run_async(execute_root_tasks(task, session))
+    result = await task.exec_root_tasks(session)
     return result
 
 

zrb/task/base_task.py

@@ -194,6 +194,22 @@ class BaseTask(AnyTask):
     def run(
         self, session: AnySession | None = None, str_kwargs: dict[str, str] = {}
     ) -> Any:
+        """
+        Synchronously runs the task and its dependencies, handling async setup and cleanup.
+
+        Uses `asyncio.run()` internally, which creates a new event loop.
+        WARNING: Do not call this method from within an already running asyncio
+        event loop, as it will raise a RuntimeError. Use `async_run` instead
+        if you are in an async context.
+
+        Args:
+            session (AnySession | None): The session to use. If None, a new one
+                might be created implicitly.
+            str_kwargs (dict[str, str]): String-based key-value arguments for inputs.
+
+        Returns:
+            Any: The final result of the main task execution.
+        """
         # Use asyncio.run() to execute the async cleanup wrapper
         return asyncio.run(run_and_cleanup(self, session, str_kwargs))
 

zrb/task/cmd_task.py

@@ -255,14 +255,14 @@ class CmdTask(BaseTask):
     def __render_single_cmd_val(
         self, ctx: AnyContext, single_cmd_val: SingleCmdVal
     ) -> str | None:
+        if isinstance(single_cmd_val, AnyCmdVal):
+            return single_cmd_val.to_str(ctx)
         if callable(single_cmd_val):
             return single_cmd_val(ctx)
         if isinstance(single_cmd_val, str):
             if self._render_cmd:
                 return ctx.render(single_cmd_val)
             return single_cmd_val
-        if isinstance(single_cmd_val, AnyCmdVal):
-            return single_cmd_val.to_str(ctx)
         return None
 
     def __get_multiline_repr(self, text: str) -> str:

zrb/task/llm/prompt.py

@@ -9,12 +9,29 @@ from zrb.task.llm.context import get_default_context  # Updated import
 from zrb.util.attr import get_attr, get_str_attr
 
 
-def get_system_prompt(
+def get_persona(
+    ctx: AnyContext,
+    persona_attr: StrAttr | None,
+    render_persona: bool,
+) -> str:
+    """Gets the persona, prioritizing task-specific, then default."""
+    persona = get_attr(
+        ctx,
+        persona_attr,
+        None,
+        auto_render=render_persona,
+    )
+    if persona is not None:
+        return persona
+    return default_llm_config.get_default_persona() or ""
+
+
+def get_base_system_prompt(
     ctx: AnyContext,
     system_prompt_attr: StrAttr | None,
     render_system_prompt: bool,
 ) -> str:
-    """Gets the system prompt, rendering if configured and handling defaults."""
+    """Gets the base system prompt, prioritizing task-specific, then default."""
     system_prompt = get_attr(
         ctx,
         system_prompt_attr,
@@ -23,7 +40,51 @@ def get_system_prompt(
     )
     if system_prompt is not None:
         return system_prompt
-    return default_llm_config.get_default_system_prompt()
+    return default_llm_config.get_default_system_prompt() or ""
+
+
+def get_special_instruction_prompt(
+    ctx: AnyContext,
+    special_instruction_prompt_attr: StrAttr | None,
+    render_special_instruction_prompt: bool,
+) -> str:
+    """Gets the special instruction prompt, prioritizing task-specific, then default."""
+    special_instruction = get_attr(
+        ctx,
+        special_instruction_prompt_attr,
+        None,
+        auto_render=render_special_instruction_prompt,
+    )
+    if special_instruction is not None:
+        return special_instruction
+    return default_llm_config.get_default_special_instruction_prompt() or ""
+
+
+def get_combined_system_prompt(
+    ctx: AnyContext,
+    persona_attr: StrAttr | None,
+    render_persona: bool,
+    system_prompt_attr: StrAttr | None,
+    render_system_prompt: bool,
+    special_instruction_prompt_attr: StrAttr | None,
+    render_special_instruction_prompt: bool,
+) -> str:
+    """Combines persona, base system prompt, and special instructions."""
+    persona = get_persona(ctx, persona_attr, render_persona)
+    base_system_prompt = get_base_system_prompt(
+        ctx, system_prompt_attr, render_system_prompt
+    )
+    special_instruction = get_special_instruction_prompt(
+        ctx, special_instruction_prompt_attr, render_special_instruction_prompt
+    )
+    parts = []
+    if persona:
+        parts.append(persona)
+    if base_system_prompt:
+        parts.append(base_system_prompt)
+    if special_instruction:
+        parts.append(special_instruction)
+    return "\n\n".join(parts).strip()
 
 
 def get_user_message(

zrb/task/llm_task.py

@@ -32,9 +32,9 @@ from zrb.task.llm.history import (
 from zrb.task.llm.history_summarization import maybe_summarize_history
 from zrb.task.llm.prompt import (
     build_user_prompt,
+    get_combined_system_prompt,
     get_context_enrichment_prompt,
     get_summarization_prompt,
-    get_system_prompt,
 )
 from zrb.util.cli.style import stylize_faint
 from zrb.xcom.xcom import Xcom
@@ -64,12 +64,16 @@ class LLMTask(BaseTask):
             ModelSettings | Callable[[AnySharedContext], ModelSettings] | None
         ) = None,
         agent: Agent | Callable[[AnySharedContext], Agent] | None = None,
+        persona: StrAttr | None = None,
+        render_persona: bool = True,
         system_prompt: StrAttr | None = None,
         render_system_prompt: bool = True,
+        special_instruction_prompt: StrAttr | None = None,
+        render_special_instruction_prompt: bool = True,
         message: StrAttr | None = None,
         summarization_prompt: StrAttr | None = None,
         render_summarization_prompt: bool = True,
-        enrich_context: BoolAttr | None = None,  # Default to None
+        enrich_context: BoolAttr | None = None,
         render_enrich_context: bool = True,
         context_enrichment_prompt: StrAttr | None = None,
         render_context_enrichment_prompt: bool = True,
@@ -80,28 +84,23 @@ class LLMTask(BaseTask):
             list[MCPServer] | Callable[[AnySharedContext], list[MCPServer]]
         ) = [],
         conversation_history: (
-            ConversationHistoryData  # Use the new BaseModel
-            | Callable[
-                [AnySharedContext], ConversationHistoryData | dict | list
-            ]  # Allow returning raw dict/list too
-            | dict  # Allow raw dict
-            | list  # Allow raw list (old format)
-        ) = ConversationHistoryData(),  # Default to an empty model instance
+            ConversationHistoryData
+            | Callable[[AnySharedContext], ConversationHistoryData | dict | list]
+            | dict
+            | list
+        ) = ConversationHistoryData(),
         conversation_history_reader: (
             Callable[[AnySharedContext], ConversationHistoryData | dict | list | None]
             | None
-            # Allow returning raw dict/list or None
         ) = None,
         conversation_history_writer: (
-            Callable[[AnySharedContext, ConversationHistoryData], None]
-            | None
-            # Writer expects the model instance
+            Callable[[AnySharedContext, ConversationHistoryData], None] | None
         ) = None,
         conversation_history_file: StrAttr | None = None,
         render_history_file: bool = True,
-        summarize_history: BoolAttr | None = None,  # Default to None
+        summarize_history: BoolAttr | None = None,
         render_summarize_history: bool = True,
-        history_summarization_threshold: IntAttr | None = None,  # Default to None
+        history_summarization_threshold: IntAttr | None = None,
         render_history_summarization_threshold: bool = True,
         execute_condition: bool | str | Callable[[AnySharedContext], bool] = True,
         retries: int = 2,
@@ -149,8 +148,12 @@ class LLMTask(BaseTask):
         self._render_model_api_key = render_model_api_key
         self._model_settings = model_settings
         self._agent = agent
+        self._persona = persona
+        self._render_persona = render_persona
         self._system_prompt = system_prompt
         self._render_system_prompt = render_system_prompt
+        self._special_instruction_prompt = special_instruction_prompt
+        self._render_special_instruction_prompt = render_special_instruction_prompt
         self._message = message
         self._summarization_prompt = summarization_prompt
         self._render_summarization_prompt = render_summarization_prompt
@@ -213,10 +216,15 @@ class LLMTask(BaseTask):
             summarization_prompt_attr=self._summarization_prompt,
             render_summarization_prompt=self._render_summarization_prompt,
         )
-        system_prompt = get_system_prompt(
+        # Get the combined system prompt using the new getter
+        system_prompt = get_combined_system_prompt(
             ctx=ctx,
+            persona_attr=self._persona,
+            render_persona=self._render_persona,
             system_prompt_attr=self._system_prompt,
             render_system_prompt=self._render_system_prompt,
+            special_instruction_prompt_attr=self._special_instruction_prompt,
+            render_special_instruction_prompt=self._render_special_instruction_prompt,
         )
         # 1. Prepare initial state (read history, get initial context)
         history_list, conversation_context = await prepare_initial_state(

zrb/task/scheduler.py

@@ -12,7 +12,6 @@ from zrb.task.any_task import AnyTask
 from zrb.task.base_trigger import BaseTrigger
 from zrb.util.attr import get_str_attr
 from zrb.util.cron import match_cron
-from zrb.xcom.xcom import Xcom
 
 
 class Scheduler(BaseTrigger):
@@ -77,6 +76,6 @@ class Scheduler(BaseTrigger):
             ctx.print(f"Current time: {now}")
             if match_cron(cron_pattern, now):
                 ctx.print(f"Matching {now} with pattern: {cron_pattern}")
-                xcom: Xcom = ctx.xcom[self.queue_name]
+                xcom = self.get_exchange_xcom(ctx.session)
                 xcom.push(now)
             await asyncio.sleep(60)

{zrb-1.5.9.dist-info → zrb-1.5.11.dist-info}/METADATA

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

{zrb-1.5.9.dist-info → zrb-1.5.11.dist-info}/RECORD

@@ -13,7 +13,7 @@ zrb/builtin/llm/llm_chat.py,sha256=XL5HK_o1EejkYS0fJIBI39vApuqYYsGHFSigvXfS7CI,4
 zrb/builtin/llm/previous-session.js,sha256=xMKZvJoAbrwiyHS0OoPrWuaKxWYLoyR5sguePIoCjTY,816
 zrb/builtin/llm/tool/api.py,sha256=yR9I0ZsI96OeQl9pgwORMASVuXsAL0a89D_iPS4C8Dc,1699
 zrb/builtin/llm/tool/cli.py,sha256=_CNEmEc6K2Z0i9ppYeM7jGpqaEdT3uxaWQatmxP3jKE,858
-zrb/builtin/llm/tool/file.py,sha256=ecFBmbMaqCIfqNGTmvd-Yt3vLtKuQ15KbjXlnIMWdho,19269
+zrb/builtin/llm/tool/file.py,sha256=ig4tZGYnGjE96U9KgOpbANmyAgFmTcQzym1wVAsZYRM,16620
 zrb/builtin/llm/tool/rag.py,sha256=45t0o88l7F62oq2P61NnC1hsZJ4h72dZsVQfcsOIUc8,7521
 zrb/builtin/llm/tool/web.py,sha256=4qzom9xX-JxztIaTWneNfyTRlgweHIxzC1uSEAxJ00A,5507
 zrb/builtin/md5.py,sha256=0pNlrfZA0wlZlHvFHLgyqN0JZJWGKQIF5oXxO44_OJk,949
@@ -239,7 +239,7 @@ zrb/input/option_input.py,sha256=TQB82ko5odgzkULEizBZi0e9TIHEbIgvdP0AR3RhA74,213
 zrb/input/password_input.py,sha256=szBojWxSP9QJecgsgA87OIYwQrY2AQ3USIKdDZY6snU,1465
 zrb/input/str_input.py,sha256=NevZHX9rf1g8eMatPyy-kUX3DglrVAQpzvVpKAzf7bA,81
 zrb/input/text_input.py,sha256=shvVbc2U8Is36h23M5lcW8IEwKc9FR-4uEPZZroj3rU,3377
-zrb/llm_config.py,sha256=Zfmv1sv1KbfUxB_jOBRYKbbsajenEwQ1jsB3-Fk5DUM,9173
+zrb/llm_config.py,sha256=sf2cMjD-eSTvOywT1qKVeSftRmxLQ92K3mzho6yP2S8,8952
 zrb/runner/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/runner/cli.py,sha256=0mT0oO_yEhc8N4nYCJNujhgLjVykZ0B-kAOFXyAvAqM,6672
 zrb/runner/common_util.py,sha256=0zhZn1Jdmr194_nsL5_L-Kn9-_NDpMTI2z6_LXUQJ-U,1369
@@ -300,14 +300,13 @@ zrb/task/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/task/any_task.py,sha256=zklUjkLRQ62TEvfnOUUYfXChj8Zk4igee3w8V3_rN08,5846
 zrb/task/base/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/task/base/context.py,sha256=73k3fKwup0AJwTTLpay0f_-axJextaxgbTem4w4Bmas,3670
-zrb/task/base/dependencies.py,sha256=Kcxhn7z4OU9Fc_y-cD1Sc96wgNXs0VDoi8r5cJMu0oY,1952
 zrb/task/base/execution.py,sha256=lB6cfivk-EM6sZSaPjYs_ufb7jb-A2jLJNhBupwBFgI,11101
-zrb/task/base/lifecycle.py,sha256=3p3mDxE97oizmh9PnF54ud9eoI_PmCXI_VsikIs-VQ8,7293
+zrb/task/base/lifecycle.py,sha256=YIugyqRmEKMnc9MTCDEZr9jVhie3Kwt7LTMtAVAN9Ks,7278
 zrb/task/base/monitoring.py,sha256=UAOEcPiYNtZR4FFxzWCosuOEFE_P3c4GT5vAhQmohqI,5663
 zrb/task/base/operators.py,sha256=uAMFqpZJsPnCrojgOl1FUDXTS15mtOa_IqiAXltyYRU,1576
-zrb/task/base_task.py,sha256=jgo9WKMMppSWvXiWCaPnm42KH6NgbFZKYfJfzRCZKeA,8222
+zrb/task/base_task.py,sha256=3MLPYDzNgjIJiQTxSyKLmvG_FspOl61htuRZG7l34QA,8915
 zrb/task/base_trigger.py,sha256=jC722rDvodaBLeNaFghkTyv1u0QXrK6BLZUUqcmBJ7Q,4581
-zrb/task/cmd_task.py,sha256=f1OWajOBmdtx2QcXBr_8s6o82Fp4UTLqCXJqp2gxwzU,10750
+zrb/task/cmd_task.py,sha256=xFAdOvLDK9Qaye40T_lG3K6AIKgbPAMHk_3GIAYeJAM,10750
 zrb/task/http_check.py,sha256=Gf5rOB2Se2EdizuN9rp65HpGmfZkGc-clIAlHmPVehs,2565
 zrb/task/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/task/llm/agent.py,sha256=2u1zlX41oBzMKozXWXD3gDEOzOsHNsFpedRcJXbUNHI,5105
@@ -318,14 +317,14 @@ zrb/task/llm/error.py,sha256=YOwnEdFMtqOlaiA83tDHpC6uh2_9r5NeS-inrlb5a8E,3622
 zrb/task/llm/history.py,sha256=LnrJdXLyo2qz-bNCwLorhoqGmgSiPTUU0bzY63w67-E,9257
 zrb/task/llm/history_summarization.py,sha256=UaeepcIVMTxJTwqy3V22rpeBXXN04KLvEzOsFtWmyDM,6259
 zrb/task/llm/print_node.py,sha256=Dkb0xFyEXpNRKFRCM4Md0lfg6K3nI0t8yH3Abh20PjE,4430
-zrb/task/llm/prompt.py,sha256=PmzeSeUBoqZqqHox1Kq8FkikvmB2lNA3hus-pybXAQg,2719
+zrb/task/llm/prompt.py,sha256=Xql45nZfkB9BXbVVxfKiawtnAVCao2QhaowVdL7qTLg,4627
 zrb/task/llm/tool_wrapper.py,sha256=gZgoxcuOCgAVDPnLqfJ3ps57ZCVQi7q68z_KnS5Mx1U,3350
 zrb/task/llm/typing.py,sha256=c8VAuPBw_4A3DxfYdydkgedaP-LU61W9_wj3m3CAX1E,58
-zrb/task/llm_task.py,sha256=ZfIRryhpS3NsczIUVZBl-f2XWzJkCcvBFmU95v0q4t0,13808
+zrb/task/llm_task.py,sha256=uLptMfoRawCCPo8_DSneKd8Uag9WDCXkU0oOvO5Lh0M,14218
 zrb/task/make_task.py,sha256=PD3b_aYazthS8LHeJsLAhwKDEgdurQZpymJDKeN60u0,2265
 zrb/task/rsync_task.py,sha256=GSL9144bmp6F0EckT6m-2a1xG25AzrrWYzH4k3SVUKM,6370
 zrb/task/scaffolder.py,sha256=rME18w1HJUHXgi9eTYXx_T2G4JdqDYzBoNOkdOOo5-o,6806
-zrb/task/scheduler.py,sha256=aSDF_WtDbmwrYXaVo3BesCG8w21UNV8XLtn-5x9GWzY,3137
+zrb/task/scheduler.py,sha256=5xoet0bMjVq0JKwSwwivD1Jo84wwwA2PrL5WaKwJAGA,3110
 zrb/task/task.py,sha256=KCrCaWYOQQvv1RJsYtHDeo9RBFmlXQ28oKyEFU4Q7pA,73
 zrb/task/tcp_check.py,sha256=P_QgGqwd5dXDaud3oQRxe_WuxyxG4s7CTY2wDk9Qcu0,2511
 zrb/task_status/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -362,7 +361,7 @@ zrb/util/string/name.py,sha256=8picJfUBXNpdh64GNaHv3om23QHhUZux7DguFLrXHp8,1163
 zrb/util/todo.py,sha256=1nDdwPc22oFoK_1ZTXyf3638Bg6sqE2yp_U4_-frHoc,16015
 zrb/xcom/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/xcom/xcom.py,sha256=o79rxR9wphnShrcIushA0Qt71d_p3ZTxjNf7x9hJB78,1571
-zrb-1.5.9.dist-info/METADATA,sha256=jAqyRUhFM5nG5wTG_YW1QaL6nIlo1g2N95EoVACbWaU,8468
-zrb-1.5.9.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-zrb-1.5.9.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
-zrb-1.5.9.dist-info/RECORD,,
+zrb-1.5.11.dist-info/METADATA,sha256=TAOvXdXnUwI1_y0VSDOLxYElSCv441WHSRogvF5EUYU,8469
+zrb-1.5.11.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+zrb-1.5.11.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
+zrb-1.5.11.dist-info/RECORD,,

zrb/task/base/dependencies.py

@@ -1,57 +0,0 @@
-from typing import TypeVar
-
-from zrb.task.any_task import AnyTask
-
-T = TypeVar("T", bound="AnyTask")
-
-
-def get_dependency_list(task: AnyTask, attr_name: str) -> list[AnyTask]:
-    """
-    Safely retrieves a list of dependencies (upstreams, fallbacks, etc.)
-    from a task attribute, handling None or single-task values.
-    """
-    dependencies = getattr(task, attr_name, None)
-    if dependencies is None:
-        return []
-    elif isinstance(dependencies, list):
-        # Ensure all elements are AnyTask (or compatible) if needed,
-        # but for now, assume the list contains tasks.
-        return dependencies
-    # Assume it's a single AnyTask instance
-    return [dependencies]
-
-
-def append_dependency(
-    task: T, attr_name: str, dependencies_to_add: AnyTask | list[AnyTask]
-) -> None:
-    """
-    Appends one or more dependencies to the specified attribute list of a task,
-    ensuring the attribute becomes a list and avoiding duplicates.
-
-    Modifies the attribute on the task object directly.
-    """
-    # Retrieve the current list, handling None or single item cases
-    current_deps_list = getattr(task, attr_name, None)
-    if current_deps_list is None:
-        current_deps_list = []
-    elif not isinstance(current_deps_list, list):
-        current_deps_list = [current_deps_list]
-    else:
-        # Create a copy to avoid modifying the original list if no changes occur
-        current_deps_list = list(current_deps_list)
-
-    if isinstance(dependencies_to_add, list):
-        new_deps = dependencies_to_add
-    else:
-        new_deps = [dependencies_to_add]  # Ensure it's always a list
-
-    added = False
-    for dep in new_deps:
-        # Check for existence before appending
-        if dep not in current_deps_list:
-            current_deps_list.append(dep)
-            added = True
-
-    # Update the attribute on the task object only if changes were made
-    if added:
-        setattr(task, attr_name, current_deps_list)