zrb 1.6.9__py3-none-any.whl → 1.7.1__py3-none-any.whl

zrb/__main__.py

@@ -6,6 +6,7 @@ from zrb.config import CFG
 from zrb.runner.cli import cli
 from zrb.util.cli.style import stylize_error, stylize_faint, stylize_warning
 from zrb.util.group import NodeNotFoundError
+from zrb.util.init_path import get_init_path_list
 from zrb.util.load import load_file, load_module
 
 
@@ -35,18 +36,27 @@ def serve_cli():
         # load init modules
         for init_module in CFG.INIT_MODULES:
             CFG.LOGGER.info(f"Loading {init_module}")
-            load_module(init_module)
-        zrb_init_path_list = _get_zrb_init_path_list()
+            try:
+                load_module(init_module)
+            except BaseException as e:
+                print(stylize_error(f"{e}"), file=sys.stderr)
+        zrb_init_path_list = get_init_path_list()
         # load init scripts
         for init_script in CFG.INIT_SCRIPTS:
             abs_init_script = os.path.abspath(os.path.expanduser(init_script))
             if abs_init_script not in zrb_init_path_list:
                 CFG.LOGGER.info(f"Loading {abs_init_script}")
-                load_file(abs_init_script, -1)
+                try:
+                    load_file(abs_init_script, -1)
+                except BaseException as e:
+                    print(stylize_error(f"{e}"), file=sys.stderr)
         # load zrb init
         for zrb_init_path in zrb_init_path_list:
             CFG.LOGGER.info(f"Loading {zrb_init_path}")
-            load_file(zrb_init_path)
+            try:
+                load_file(zrb_init_path)
+            except BaseException as e:
+                print(stylize_error(f"{e}"), file=sys.stderr)
         # run the CLI
         cli.run(sys.argv[1:])
     except KeyboardInterrupt:
@@ -59,18 +69,3 @@ def serve_cli():
     except NodeNotFoundError as e:
         print(stylize_error(f"{e}"), file=sys.stderr)
         sys.exit(1)
-
-
-def _get_zrb_init_path_list() -> list[str]:
-    current_path = os.path.abspath(os.getcwd())
-    dir_path_list = [current_path]
-    while current_path != os.path.dirname(current_path):  # Stop at root
-        current_path = os.path.dirname(current_path)
-        dir_path_list.append(current_path)
-    zrb_init_path_list = []
-    for current_path in dir_path_list[::-1]:
-        zrb_init_path = os.path.join(current_path, CFG.INIT_FILE_NAME)
-        CFG.LOGGER.info(f"Finding {zrb_init_path}")
-        if os.path.isfile(zrb_init_path):
-            zrb_init_path_list.append(zrb_init_path)
-    return zrb_init_path_list

zrb/builtin/llm/chat_session.py

@@ -98,7 +98,7 @@ async def _trigger_ask_and_wait_for_result(
         return None
     await _trigger_ask(ctx, user_prompt, previous_session_name, start_new)
     result = await _wait_ask_result(ctx)
-    ctx.print(stylize_faint("🤖 >> ") + result, plain=True)
+    ctx.print(f"{stylize_faint('🤖 >>')} {result}", plain=True)
     return result
 
 
@@ -150,7 +150,7 @@ async def _trigger_ask(
     )
 
 
-async def _wait_ask_result(ctx: AnyContext) -> str:
+async def _wait_ask_result(ctx: AnyContext) -> str | None:
     """
     Waits for and retrieves the LLM task result from the 'ask_result' XCom queue.
 
@@ -162,6 +162,9 @@ async def _wait_ask_result(ctx: AnyContext) -> str:
     """
     while "ask_result" not in ctx.xcom or len(ctx.xcom.ask_result) == 0:
         await asyncio.sleep(0.1)
+        if "ask_error" in ctx.xcom and len(ctx.xcom.ask_error) > 0:
+            ctx.xcom.ask_error.pop()
+            return None
     return ctx.xcom.ask_result.pop()
 
 

zrb/builtin/llm/llm_ask.py

@@ -11,6 +11,7 @@ from zrb.builtin.llm.tool.file import (
     search_files,
     write_to_file,
 )
+from zrb.builtin.llm.tool.sub_agent import create_sub_agent_tool
 from zrb.builtin.llm.tool.web import (
     create_search_internet_tool,
     open_web_page,
@@ -114,6 +115,7 @@ llm_group.add_task(
             task=llm_ask,
             input_mapping=get_llm_ask_input_mapping,
             result_queue="ask_result",
+            error_queue="ask_error",
             session_name_queue="ask_session_name",
         ),
         retries=0,
@@ -123,20 +125,41 @@ llm_group.add_task(
 )
 
 if CFG.LLM_ALLOW_ACCESS_LOCAL_FILE:
-    llm_ask.add_tool(list_files)
-    llm_ask.add_tool(read_from_file)
-    llm_ask.add_tool(write_to_file)
-    llm_ask.add_tool(search_files)
-    llm_ask.add_tool(apply_diff)
+    llm_ask.append_tool(
+        list_files,
+        read_from_file,
+        write_to_file,
+        search_files,
+        apply_diff,
+        create_sub_agent_tool(
+            tool_name="analyze_file",
+            tool_description="\n".join(
+                [
+                    "Analyze file using LLM capability.",
+                    "This tool can do:",
+                    "- summarization",
+                    "- outline/structure extraction",
+                    "- code review",
+                    "- other tasks requiring deep understanding.",
+                    "Always use this tool to get deep understanding of file content",
+                ]
+            ),
+            sub_agent_system_prompt="\n".join(
+                [
+                    "You are a file analyzer assistant",
+                    "Your goal is to help the main asisstant by reading file",
+                    "and perform necessary indepth analysis required by the main assistant",
+                ]
+            ),
+            sub_agent_tools=[read_from_file, search_files],
+        ),
+    )
 
 if CFG.LLM_ALLOW_ACCESS_SHELL:
-    llm_ask.add_tool(run_shell_command)
+    llm_ask.append_tool(run_shell_command)
 
 if CFG.LLM_ALLOW_ACCESS_INTERNET:
-    llm_ask.add_tool(open_web_page)
-    llm_ask.add_tool(search_wikipedia)
-    llm_ask.add_tool(search_arxiv)
+    llm_ask.append_tool(open_web_page, search_wikipedia, search_arxiv)
     if CFG.SERP_API_KEY != "":
-        llm_ask.add_tool(create_search_internet_tool(CFG.SERP_API_KEY))
-    llm_ask.add_tool(get_current_location)
-    llm_ask.add_tool(get_current_weather)
+        llm_ask.append_tool(create_search_internet_tool(CFG.SERP_API_KEY))
+    llm_ask.append_tool(get_current_location, get_current_weather)

zrb/callback/callback.py

@@ -23,6 +23,7 @@ class Callback(AnyCallback):
         input_mapping: StrDictAttr,
         render_input_mapping: bool = True,
         result_queue: str | None = None,
+        error_queue: str | None = None,
         session_name_queue: str | None = None,
     ):
         """
@@ -35,6 +36,8 @@ class Callback(AnyCallback):
                 f-string like syntax.
             result_queue: The name of the XCom queue in the parent session
                 to publish the task result.
+            result_queue: The name of the Xcom queue in the parent session
+                to publish the task error.
             session_name_queue: The name of the XCom queue in the parent
                 session to publish the session name.
         """
@@ -42,6 +45,7 @@ class Callback(AnyCallback):
         self._input_mapping = input_mapping
         self._render_input_mapping = render_input_mapping
         self._result_queue = result_queue
+        self._error_queue = error_queue
         self._session_name_queue = session_name_queue
 
     async def async_run(self, parent_session: AnySession, session: AnySession) -> Any:
@@ -58,9 +62,12 @@ class Callback(AnyCallback):
             session.shared_ctx.input[name] = value
             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
+        try:
+            result = await self._task.async_run(session)
+            self._maybe_publish_result_to_parent_session(parent_session, result)
+            return result
+        except BaseException as e:
+            self._maybe_publish_error_to_parent_session(parent_session, e)
 
     def _maybe_publish_session_name_to_parent_session(
         self, parent_session: AnySession, session: AnySession
@@ -71,6 +78,13 @@ class Callback(AnyCallback):
             value=session.name,
         )
 
+    def _maybe_publish_error_to_parent_session(
+        self, parent_session: AnySession, error: Any
+    ):
+        self._maybe_publish_to_parent_session(
+            parent_session=parent_session, xcom_name=self._error_queue, value=error
+        )
+
     def _maybe_publish_result_to_parent_session(
         self, parent_session: AnySession, result: Any
     ):

zrb/config.py

@@ -199,6 +199,10 @@ class Config:
     def LLM_PERSONA(self) -> str | None:
         return os.getenv("ZRB_LLM_PERSONA", None)
 
+    @property
+    def LLM_CODE_REVIEW_INSTRUCTION_PROMPT(self) -> str | None:
+        return os.getenv("ZRB_LLM_CODE_REVIEW_INSTRUCTION_PROMPT", None)
+
     @property
     def LLM_SPECIAL_INSTRUCTION_PROMPT(self) -> str | None:
         return os.getenv("ZRB_LLM_SPECIAL_INSTRUCTION_PROMPT", None)

zrb/llm_config.py

@@ -20,6 +20,8 @@ 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.
+Make sure to always include all necessary information in your final answer.
+Remember that you have limited set of context.
 """.strip()
 
 DEFAULT_PERSONA = """
@@ -41,15 +43,35 @@ Output *only* the updated summary text.
 DEFAULT_CONTEXT_ENRICHMENT_PROMPT = """
 You are an information extraction assistant.
 Your goal is to help main assistant to continue the conversation by extracting
-important informations.
+important long-term/short-term informations.
 Analyze the conversation history and current context to extract key facts like
-user_name, user_roles, preferences, goals, etc.
+user_name, user_roles, user_address, etc.
 Return only a JSON object containing a single key "response", whose value is
 another JSON object with these details (i.e., {"response": {"context_name": "value"}}).
 If no context can be extracted, return {"response": {}}.
 """.strip()
 
 DEFAULT_SPECIAL_INSTRUCTION_PROMPT = ""  # Default to empty
+DEFAULT_CODE_REVIEW_INSTRUCTION_PROMPT = """
+Your goal is to review code provided by the user for correctness, readability, 
+performance, security, and maintainability.
+Follow these principles:
+1. **Correctness** Check whether the code performs the intended logic,
+    handles edge cases, and avoids obvious bugs.
+2. **Readability** Evaluate naming conventions, code structure, and clarity.
+    Suggest improvements where code could be more understandable.
+3. **Performance** Identify inefficient patterns or unnecessary operations. 
+    Recommend optimizations only when they provide meaningful benefit.
+4. **Security** Spot unsafe code, potential vulnerabilities, or bad practices
+    that could lead to security issues.
+5. **Consistency** Ensure the code adheres to common language idioms,
+    style guides, and project conventions.
+Provide clear, concise, and actionable feedback.
+Use inline code examples when helpful.
+Do not restate the code unnecessarily.
+Focus on meaningful insights that help the user improve the code quality.
+Avoid excessive nitpicking unless requested.
+""".strip()
 
 
 class LLMConfig:
@@ -62,6 +84,7 @@ class LLMConfig:
         default_persona: str | None = None,
         default_system_prompt: str | None = None,
         default_special_instruction_prompt: str | None = None,
+        default_code_review_instruction_prompt: str | None = None,
         default_summarization_prompt: str | None = None,
         default_context_enrichment_prompt: str | None = None,
         default_summarize_history: bool | None = None,
@@ -78,6 +101,9 @@ class LLMConfig:
         self._default_persona = default_persona
         self._default_system_prompt = default_system_prompt
         self._default_special_instruction_prompt = default_special_instruction_prompt
+        self._default_code_review_instruction_prompt = (
+            default_code_review_instruction_prompt
+        )
         self._default_summarization_prompt = default_summarization_prompt
         self._default_context_enrichment_prompt = default_context_enrichment_prompt
         self._default_summarize_history = default_summarize_history
@@ -149,6 +175,14 @@ class LLMConfig:
             return CFG.LLM_PERSONA
         return DEFAULT_PERSONA
 
+    @property
+    def default_code_review_instruction_prompt(self) -> str:
+        if self._default_code_review_instruction_prompt is not None:
+            return self._default_code_review_instruction_prompt
+        if CFG.LLM_CODE_REVIEW_INSTRUCTION_PROMPT is not None:
+            return CFG.LLM_CODE_REVIEW_INSTRUCTION_PROMPT
+        return DEFAULT_CODE_REVIEW_INSTRUCTION_PROMPT
+
     @property
     def default_special_instruction_prompt(self) -> str:
         if self._default_special_instruction_prompt is not None:

zrb/runner/cli.py

@@ -19,6 +19,7 @@ from zrb.util.cli.style import (
     stylize_section_header,
 )
 from zrb.util.group import extract_node_from_args, get_non_empty_subgroups, get_subtasks
+from zrb.util.init_path import get_init_path_list
 from zrb.util.string.conversion import double_quote
 
 
@@ -189,6 +190,11 @@ async def start_server(_: AnyContext):
 
     app = create_web_app(cli, web_config, session_state_logger)
     server = Server(
-        Config(app=app, host="0.0.0.0", port=CFG.WEB_HTTP_PORT, loop="asyncio")
+        Config(
+            app=app,
+            host="0.0.0.0",
+            port=CFG.WEB_HTTP_PORT,
+            loop="asyncio",
+        )
     )
     await server.serve()

zrb/runner/web_app.py

@@ -47,7 +47,7 @@ def create_web_app(
         asyncio.gather(*_COROS)
 
     app = FastAPI(
-        title="Zrb",
+        title=CFG.WEB_TITLE,
         version=CFG.VERSION,
         summary="Your Automation Powerhouse",
         lifespan=lifespan,

zrb/task/llm/context_enrichment.py

@@ -53,7 +53,10 @@ async def enrich_context(
         # The user prompt will now contain the dynamic data
         user_prompt_data = "\n".join(
             [
-                "Extract context from the following conversation info",
+                "Extract context from the following conversation info.",
+                "Maintain the relevant context and remove the irrelevant ones.",
+                "Restructure the context in a helpful way",
+                "Keep the context small",
                 f"Existing Context: {context_json}",
                 f"Conversation History: {history_json}",
             ]
@@ -86,7 +89,7 @@ async def enrich_context(
             usage = enrichment_run.result.usage()
             ctx.print(stylize_faint(f"[Token Usage] {usage}"), plain=True)
             if response:
-                conversation_context.update(response)
+                conversation_context = response
                 ctx.log_info("Context enriched based on history.")
                 ctx.log_info(
                     f"Updated conversation context: {json.dumps(conversation_context)}"

zrb/task/llm/prompt.py

@@ -52,7 +52,19 @@ def get_special_instruction_prompt(
     )
     if special_instruction is not None:
         return special_instruction
-    return llm_config.default_special_instruction_prompt or ""
+    aggregated_special_instruction = ""
+    if llm_config.default_code_review_instruction_prompt:
+        aggregated_special_instruction += "\n".join(
+            [
+                "# Code review instruction",
+                llm_config.default_code_review_instruction_prompt,
+            ]
+        )
+    if llm_config.default_special_instruction_prompt:
+        aggregated_special_instruction += "\n" + "\n".join(
+            ["# Code review instruction", llm_config.default_special_instruction_prompt]
+        )
+    return aggregated_special_instruction or ""
 
 
 def get_combined_system_prompt(

zrb/task/llm_task.py

@@ -192,17 +192,19 @@ class LLMTask(BaseTask):
         self._max_call_iteration = max_call_iteration
         self._conversation_context = conversation_context
 
-    def add_tool(self, tool: ToolOrCallable):
-        self.append_tool(tool)
+    def add_tool(self, *tool: ToolOrCallable):
+        self.append_tool(*tool)
 
-    def append_tool(self, tool: ToolOrCallable):
-        self._additional_tools.append(tool)
+    def append_tool(self, *tool: ToolOrCallable):
+        for single_tool in tool:
+            self._additional_tools.append(single_tool)
 
-    def add_mcp_server(self, mcp_server: MCPServer):
-        self.append_mcp_server(mcp_server)
+    def add_mcp_server(self, *mcp_server: MCPServer):
+        self.append_mcp_server(*mcp_server)
 
-    def append_mcp_server(self, mcp_server: MCPServer):
-        self._additional_mcp_servers.append(mcp_server)
+    def append_mcp_server(self, *mcp_server: MCPServer):
+        for single_mcp_server in mcp_server:
+            self._additional_mcp_servers.append(single_mcp_server)
 
     def set_should_enrich_context(self, enrich_context: bool):
         self._should_enrich_context = enrich_context

zrb/util/init_path.py

@@ -0,0 +1,18 @@
+import os
+
+from zrb.config import CFG
+
+
+def get_init_path_list() -> list[str]:
+    current_path = os.path.abspath(os.getcwd())
+    dir_path_list = [current_path]
+    while current_path != os.path.dirname(current_path):  # Stop at root
+        current_path = os.path.dirname(current_path)
+        dir_path_list.append(current_path)
+    zrb_init_path_list = []
+    for current_path in dir_path_list[::-1]:
+        zrb_init_path = os.path.join(current_path, CFG.INIT_FILE_NAME)
+        CFG.LOGGER.info(f"Finding {zrb_init_path}")
+        if os.path.isfile(zrb_init_path):
+            zrb_init_path_list.append(zrb_init_path)
+    return zrb_init_path_list

{zrb-1.6.9.dist-info → zrb-1.7.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: zrb
-Version: 1.6.9
+Version: 1.7.1
 Summary: Your Automation Powerhouse
 Home-page: https://github.com/state-alchemists/zrb
 License: AGPL-3.0-or-later
@@ -26,7 +26,7 @@ Requires-Dist: openai (>=1.76.0,<2.0.0) ; extra == "rag" or extra == "all"
 Requires-Dist: pdfplumber (>=0.11.6,<0.12.0) ; extra == "rag" or extra == "all"
 Requires-Dist: playwright (>=1.51.0,<2.0.0) ; extra == "playwright" or extra == "all"
 Requires-Dist: psutil (>=7.0.0,<8.0.0)
-Requires-Dist: pydantic-ai (>=0.1.6,<0.2.0)
+Requires-Dist: pydantic-ai (>=0.2.7,<0.3.0)
 Requires-Dist: pyjwt (>=2.10.1,<3.0.0)
 Requires-Dist: python-dotenv (>=1.1.0,<2.0.0)
 Requires-Dist: python-jose[cryptography] (>=3.4.0,<4.0.0)

{zrb-1.6.9.dist-info → zrb-1.7.1.dist-info}/RECORD

@@ -1,5 +1,5 @@
 zrb/__init__.py,sha256=JgYNwsKYVzYzpoFXRn_LwuNFreMffPEvoSehTZisp4M,3103
-zrb/__main__.py,sha256=8LMzIvdKtOe7sIBfEhXYW8SM7dyTcd5JIySHFa6shbQ,2799
+zrb/__main__.py,sha256=Kr_AkgmwBEycQcPVkZmrNh4hp2g62G-7ZXeOhJg0Qis,2603
 zrb/attr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/attr/type.py,sha256=4TV5gPYMMrKh5V-yB6iRYKCbsXAH_AvGXMsjxKLHcUs,568
 zrb/builtin/__init__.py,sha256=N-h-BoXWv0jYOldixXwgk6ekiWtrGZsGv57iqonsYdc,2657
@@ -9,10 +9,10 @@ zrb/builtin/git_subtree.py,sha256=7BKwOkVTWDrR0DXXQ4iJyHqeR6sV5VYRt8y_rEB0EHg,35
 zrb/builtin/group.py,sha256=t008xLM4_fgbjfZrPoi_fQAnSHIo6MOiQSCHBO4GDYU,2379
 zrb/builtin/http.py,sha256=sLqEczuSxGYXWzyJR6frGOHkPTviu4BeyroUr3-ZuAI,4322
 zrb/builtin/jwt.py,sha256=kjCf8qt7tkW9BpBDRAVTMJaEPQGzCbO1wo9xt5JoM8A,2836
-zrb/builtin/llm/chat_session.py,sha256=PSFfXPASMT-TLb-BQXI-nXbVDOnjmVJcUW6InxrpXKU,6491
+zrb/builtin/llm/chat_session.py,sha256=HqFwrE1DiSlJrR-S3LRYWQBHkVsD-sfAV8_IIbnmtqY,6631
 zrb/builtin/llm/history.py,sha256=cnkOyO43uiMQ9cEvmqk-pPoCk1zCAH_fwAqSgBtsjzY,3079
 zrb/builtin/llm/input.py,sha256=Nw-26uTWp2QhUgKJcP_IMHmtk-b542CCSQ_vCOjhvhM,877
-zrb/builtin/llm/llm_ask.py,sha256=bC4OtlYChhOlUIwstzPM4_LOfL_8NNhNpN60xpjzeNg,4375
+zrb/builtin/llm/llm_ask.py,sha256=7Dx5XWf0QAbPwscGohloKOp_iqoI-JuaVojTaCJL3Jc,5332
 zrb/builtin/llm/previous-session.js,sha256=xMKZvJoAbrwiyHS0OoPrWuaKxWYLoyR5sguePIoCjTY,816
 zrb/builtin/llm/tool/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/builtin/llm/tool/api.py,sha256=yR9I0ZsI96OeQl9pgwORMASVuXsAL0a89D_iPS4C8Dc,1699
@@ -212,11 +212,11 @@ zrb/builtin/todo.py,sha256=pDbDKp94VHy-JsOr1sFtY8K4nIpNr1v6siqs5ptypsg,11568
 zrb/builtin/uuid.py,sha256=lIdhSGzPQ1rixRzMXxQDcgFgV7W-gUduHIudZXlzZzg,5393
 zrb/callback/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/callback/any_callback.py,sha256=PqEJYX_RigXEmoPniSeZusZBZSLWEoVIHvHk8MZ0Mvg,253
-zrb/callback/callback.py,sha256=yPVsbMqRkFH5kKIcFVg-U0VEKTP2VXVArhgQYwD-Jv8,3293
+zrb/callback/callback.py,sha256=mk_RIHuWi-oP5b81jfhzU6fruhsIjhRtKpwh2yYmsiM,3876
 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=Ipb1RhOJqM0O_CGF0-UPMiF1XFnYLu7pdWVk8BASO50,8556
+zrb/config.py,sha256=ZQKSNZmMfCyxTq7_4eCVzn_XsbMd7SlErjxAdB3wwoA,8708
 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
@@ -245,11 +245,11 @@ 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=Twe7-32OxvzCSPFOY-cV3SQGPSqCcjO1ZmiYvSjNpCk,10443
+zrb/llm_config.py,sha256=h3LyjoOunc-2FxOOyCwM6Xcll3iXDykvvbV6xob-iB8,12244
 zrb/runner/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-zrb/runner/cli.py,sha256=2xy4cPGqDZOm4EBGcP4WtKBXWjk74O2whDS3XarX0XQ,6925
+zrb/runner/cli.py,sha256=knWItZ0fm25gy4hjSgjliCtMSP2i3m2N8rNpS8B9csQ,7034
 zrb/runner/common_util.py,sha256=JDMcwvQ8cxnv9kQrAoKVLA40Q1omfv-u5_d5MvvwHeE,1373
-zrb/runner/web_app.py,sha256=huY9SCvMC9akhEQDW6mUC0osr0b0H3EosD7GFClVt-A,2612
+zrb/runner/web_app.py,sha256=y7lA9urDg1ImZHFzhzogjVMVa9K8E-uSbMK_aNXYstM,2620
 zrb/runner/web_config/config.py,sha256=0wR58KreAmawGGfamm0GLZY344HaXs7qfDgHLavBDwo,3125
 zrb/runner/web_config/config_factory.py,sha256=GIvAwQKY_jkAb_IVt179KvphqHPY7gfBiQrZmELVQdQ,636
 zrb/runner/web_route/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -320,15 +320,15 @@ zrb/task/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/task/llm/agent.py,sha256=6wGSsw03GdY_fj12CsJh7wxB6BnE13N8RYXaWfbiUsk,5451
 zrb/task/llm/config.py,sha256=oGxHYMdIvhASnKwNuMPwcdeJiFfS0tNskzGHakpfpQU,3458
 zrb/task/llm/context.py,sha256=U9a8lxa2ikz6my0Sd5vpO763legHrMHyvBjbrqNmv0Y,3838
-zrb/task/llm/context_enrichment.py,sha256=BKiWTorUrwDCMxghFXua3d-CHA3faUaCc3am7I3YGCg,6446
+zrb/task/llm/context_enrichment.py,sha256=kGFEwma06edLygtEvJ4331i39rmEwdDTSrXm-G1LolM,6624
 zrb/task/llm/error.py,sha256=27DQXSG8SH1-XuvXFdZQKzP39wZDWmd_YnSTz6DJKKI,3690
 zrb/task/llm/history.py,sha256=3WMXoi7RquxosXQf3iv2_BCeF8iKtY1f407pR71xERs,7745
 zrb/task/llm/history_summarization.py,sha256=n3GbgwXlDIkgpJppMGfpqF_8Wpi9yAoZYh46O1pFQeU,6432
 zrb/task/llm/print_node.py,sha256=bpISOUxSH_JBLR-4Nq6-iLrzNWFagrKFX6u8ogYYMw8,4395
-zrb/task/llm/prompt.py,sha256=2JG9eey3wbysz4IjcmiNNYRIeffSJkbj_vr0Laxa6wE,3833
+zrb/task/llm/prompt.py,sha256=AUuyxjFgdK72yFNmNqsUDe55wjjxjK37vL7q7XPcgt8,4332
 zrb/task/llm/tool_wrapper.py,sha256=Xygd4VCY3ykjVv63pqlTI16ZG41ySkp683_5VTnL-Zo,6481
 zrb/task/llm/typing.py,sha256=c8VAuPBw_4A3DxfYdydkgedaP-LU61W9_wj3m3CAX1E,58
-zrb/task/llm_task.py,sha256=OQ-Zp00tldmtB3pH7qugTh85guBB94yh2_o236EpfTg,15537
+zrb/task/llm_task.py,sha256=Yav1pmV26Eh4h9xTh16dN-DbTvhfYINI0EDp_ptJHLg,15643
 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
@@ -360,6 +360,7 @@ zrb/util/file.py,sha256=Y5pRJxkpshyELxsv8qw_gp1qM5OXgeDwYXqt7Y6Lqu8,2242
 zrb/util/git.py,sha256=gS_Y9sQgJbY0PfgSQiowLvV3Nf0y9C8nT3j6z6oEsG8,8186
 zrb/util/git_subtree.py,sha256=E_UB5OIgm8WkHL9beifRxpZ25_BB9p1H578OhLZTgRU,4611
 zrb/util/group.py,sha256=T82yr3qg9I5k10VPXkMyrIRIqyfzadSH813bqzwKEPI,4718
+zrb/util/init_path.py,sha256=n4BgLGeq3mPLS1la8VEqZpqJHx0vJRe2WRwTtbw-FjE,652
 zrb/util/load.py,sha256=DK0KYSlu48HCoGPqnW1IxnE3pHrZSPCstfz8Fjyqqv8,2140
 zrb/util/run.py,sha256=FPRCCvl5g6GuDvHTkaV95CFDlqxQ-5FZb2-F-Jz1fnI,485
 zrb/util/string/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -369,7 +370,7 @@ zrb/util/string/name.py,sha256=SXEfxJ1-tDOzHqmSV8kvepRVyMqs2XdV_vyoh_9XUu0,1584
 zrb/util/todo.py,sha256=VGISej2KQZERpornK-8X7bysp4JydMrMUTnG8B0-liI,20708
 zrb/xcom/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/xcom/xcom.py,sha256=o79rxR9wphnShrcIushA0Qt71d_p3ZTxjNf7x9hJB78,1571
-zrb-1.6.9.dist-info/METADATA,sha256=Df_JAjjx5sOjoHEgD5xSvYBD_H2HL3VljWQI-VFzk-4,8385
-zrb-1.6.9.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-zrb-1.6.9.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
-zrb-1.6.9.dist-info/RECORD,,
+zrb-1.7.1.dist-info/METADATA,sha256=N_pEPK58I8nQ5O7J-1yyl0Q6gxH1PPpLN82ldGB7rMo,8385
+zrb-1.7.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+zrb-1.7.1.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
+zrb-1.7.1.dist-info/RECORD,,