zrb 1.21.17__py3-none-any.whl → 1.21.33__py3-none-any.whl

zrb/task/llm_task.py

@@ -3,13 +3,14 @@ from collections.abc import Callable
 from typing import TYPE_CHECKING, Any
 
 from zrb.attr.type import BoolAttr, IntAttr, StrAttr, StrListAttr, fstring
-from zrb.config.llm_rate_limitter import LLMRateLimiter
+from zrb.config.llm_rate_limitter import LLMRateLimitter
 from zrb.context.any_context import AnyContext
 from zrb.env.any_env import AnyEnv
 from zrb.input.any_input import AnyInput
 from zrb.task.any_task import AnyTask
 from zrb.task.base_task import BaseTask
-from zrb.task.llm.agent import get_agent, run_agent_iteration
+from zrb.task.llm.agent import get_agent
+from zrb.task.llm.agent_runner import run_agent_iteration
 from zrb.task.llm.config import (
     get_model,
     get_model_settings,
@@ -21,13 +22,17 @@ from zrb.task.llm.conversation_history import (
     write_conversation_history,
 )
 from zrb.task.llm.conversation_history_model import ConversationHistory
-from zrb.task.llm.history_summarization import maybe_summarize_history
+from zrb.task.llm.history_summarization import get_history_summarization_token_threshold
 from zrb.task.llm.prompt import (
     get_attachments,
     get_summarization_system_prompt,
     get_system_and_user_prompt,
     get_user_message,
 )
+from zrb.task.llm.subagent_conversation_history import (
+    extract_subagent_conversation_history_from_ctx,
+    inject_subagent_conversation_history_into_ctx,
+)
 from zrb.task.llm.workflow import load_workflow
 from zrb.util.cli.style import stylize_faint
 from zrb.xcom.xcom import Xcom
@@ -104,11 +109,11 @@ class LLMTask(BaseTask):
         render_history_file: bool = True,
         summarize_history: BoolAttr | None = None,
         render_summarize_history: bool = True,
-        summarization_prompt: StrAttr | None = None,
+        summarization_prompt: "Callable[[AnyContext], str | None] | str | None" = None,
         render_summarization_prompt: bool = False,
         history_summarization_token_threshold: IntAttr | None = None,
         render_history_summarization_token_threshold: bool = True,
-        rate_limitter: LLMRateLimiter | None = None,
+        rate_limitter: LLMRateLimitter | None = None,
         execute_condition: bool | str | Callable[[AnyContext], bool] = True,
         retries: int = 2,
         retry_period: float = 0,
@@ -129,9 +134,6 @@ class LLMTask(BaseTask):
         upstream: list[AnyTask] | AnyTask | None = None,
         fallback: list[AnyTask] | AnyTask | None = None,
         successor: list[AnyTask] | AnyTask | None = None,
-        conversation_context: (
-            dict[str, Any] | Callable[[AnyContext], dict[str, Any]] | None
-        ) = None,
     ):
         super().__init__(
             name=name,
@@ -199,7 +201,6 @@ class LLMTask(BaseTask):
             render_history_summarization_token_threshold
         )
         self._max_call_iteration = max_call_iteration
-        self._conversation_context = conversation_context
         self._yolo_mode = yolo_mode
         self._render_yolo_mode = render_yolo_mode
         self._attachment = attachment
@@ -265,6 +266,7 @@ class LLMTask(BaseTask):
             conversation_history_attr=self._conversation_history,
         )
         inject_conversation_history_notes(conversation_history)
+        inject_subagent_conversation_history_into_ctx(ctx, conversation_history)
         # 2. Get system prompt and user prompt
         system_prompt, user_prompt = get_system_and_user_prompt(
             ctx=ctx,
@@ -279,12 +281,29 @@ class LLMTask(BaseTask):
             render_workflows=self._render_workflows,
             conversation_history=conversation_history,
         )
-        ctx.log_debug(f"SYSTEM PROMPT:\n{system_prompt}")
-        ctx.log_debug(f"USER PROMPT:\n{user_prompt}")
-        # 3. Get the agent instance
+        # 3. Summarization
+        small_model = get_model(
+            ctx=ctx,
+            model_attr=self._small_model,
+            render_model=self._render_small_model,
+            model_base_url_attr=self._small_model_base_url,
+            render_model_base_url=self._render_small_model_base_url,
+            model_api_key_attr=self._small_model_api_key,
+            render_model_api_key=self._render_small_model_api_key,
+        )
+        small_model_settings = get_model_settings(ctx, self._small_model_settings)
+        summarization_token_threshold = get_history_summarization_token_threshold(
+            ctx,
+            self._history_summarization_token_threshold,
+            self._render_history_summarization_token_threshold,
+        )
+        # 4. Get the agent instance
+        ctx.log_info(f"SYSTEM PROMPT:\n{system_prompt}")
+        ctx.log_info(f"USER PROMPT:\n{user_prompt}")
         agent = get_agent(
             ctx=ctx,
             model=model,
+            rate_limitter=self._rate_limitter,
             system_prompt=system_prompt,
             model_settings=model_settings,
             tools_attr=self._tools,
@@ -292,8 +311,14 @@ class LLMTask(BaseTask):
             toolsets_attr=self._toolsets,
             additional_toolsets=self._additional_toolsets,
             yolo_mode=yolo_mode,
+            summarization_model=small_model,
+            summarization_model_settings=small_model_settings,
+            summarization_system_prompt=summarization_prompt,
+            summarization_retries=2,  # TODO: make this a property
+            summarization_token_threshold=summarization_token_threshold,
+            history_processors=[],  # TODO: make this a property
         )
-        # 4. Run the agent iteration and save the results/history
+        # 5. Run the agent iteration and save the results/history
         result = await self._execute_agent(
             ctx=ctx,
             agent=agent,
@@ -301,34 +326,10 @@ class LLMTask(BaseTask):
             attachments=attachments,
             conversation_history=conversation_history,
         )
-        # 5. Summarize
-        small_model = get_model(
-            ctx=ctx,
-            model_attr=self._small_model,
-            render_model=self._render_small_model,
-            model_base_url_attr=self._small_model_base_url,
-            render_model_base_url=self._render_small_model_base_url,
-            model_api_key_attr=self._small_model_api_key,
-            render_model_api_key=self._render_small_model_api_key,
-        )
-        small_model_settings = get_model_settings(ctx, self._small_model_settings)
-        conversation_history = await maybe_summarize_history(
-            ctx=ctx,
-            conversation_history=conversation_history,
-            should_summarize_history_attr=self._should_summarize_history,
-            render_summarize_history=self._render_summarize_history,
-            history_summarization_token_threshold_attr=(
-                self._history_summarization_token_threshold
-            ),
-            render_history_summarization_token_threshold=(
-                self._render_history_summarization_token_threshold
-            ),
-            model=small_model,
-            model_settings=small_model_settings,
-            summarization_prompt=summarization_prompt,
-            rate_limitter=self._rate_limitter,
-        )
         # 6. Write conversation history
+        conversation_history.subagent_history = (
+            extract_subagent_conversation_history_from_ctx(ctx)
+        )
         await write_conversation_history(
             ctx=ctx,
             history_data=conversation_history,

zrb/util/attr.py

@@ -10,11 +10,14 @@ from zrb.attr.type import (
     StrListAttr,
 )
 from zrb.context.any_context import AnyContext
+from zrb.context.any_shared_context import AnySharedContext
 from zrb.util.string.conversion import to_boolean
 
 
 def get_str_list_attr(
-    ctx: AnyContext, attr: StrListAttr | None, auto_render: bool = True
+    ctx: AnyContext | AnySharedContext,
+    attr: StrListAttr | None,
+    auto_render: bool = True,
 ) -> list[str]:
     """
     Retrieve a list of strings from shared context attributes.
@@ -35,7 +38,9 @@ def get_str_list_attr(
 
 
 def get_str_dict_attr(
-    ctx: AnyContext, attr: StrDictAttr | None, auto_render: bool = True
+    ctx: AnyContext | AnySharedContext,
+    attr: StrDictAttr | None,
+    auto_render: bool = True,
 ) -> dict[str, Any]:
     """
     Retrieve a dictionary of strings from shared context attributes.
@@ -56,7 +61,7 @@ def get_str_dict_attr(
 
 
 def get_str_attr(
-    ctx: AnyContext,
+    ctx: AnyContext | AnySharedContext,
     attr: StrAttr | None,
     default: StrAttr = "",
     auto_render: bool = True,
@@ -82,7 +87,7 @@ def get_str_attr(
 
 
 def get_bool_attr(
-    ctx: AnyContext,
+    ctx: AnyContext | AnySharedContext,
     attr: BoolAttr | None,
     default: BoolAttr = False,
     auto_render: bool = True,
@@ -108,7 +113,7 @@ def get_bool_attr(
 
 
 def get_int_attr(
-    ctx: AnyContext,
+    ctx: AnyContext | AnySharedContext,
     attr: IntAttr | None,
     default: IntAttr = 0,
     auto_render: bool = True,
@@ -134,7 +139,7 @@ def get_int_attr(
 
 
 def get_float_attr(
-    ctx: AnyContext,
+    ctx: AnyContext | AnySharedContext,
     attr: FloatAttr | None,
     default: FloatAttr = 0.0,
     auto_render: bool = True,
@@ -160,7 +165,7 @@ def get_float_attr(
 
 
 def get_attr(
-    ctx: AnyContext,
+    ctx: AnyContext | AnySharedContext,
     attr: AnyAttr,
     default: AnyAttr,
     auto_render: bool = True,

zrb/util/git.py

@@ -131,7 +131,7 @@ async def get_branches(
         Exception: If the git command returns a non-zero exit code.
     """
     cmd_result, exit_code = await run_command(
-        cmd=["git", "rev-parse", "--abbrev-ref", "HEAD"],
+        cmd=["git", "branch"],
         cwd=repo_dir,
         print_method=print_method,
     )
@@ -160,7 +160,7 @@ async def delete_branch(
         Exception: If the git command returns a non-zero exit code.
     """
     cmd_result, exit_code = await run_command(
-        cmd=["git", "branch", "-D", branch_name],
+        cmd=["git", "branch", "-d", branch_name],
         cwd=repo_dir,
         print_method=print_method,
     )

zrb/xcom/xcom.py

@@ -34,6 +34,16 @@ class Xcom(deque):
         else:
             raise IndexError("Xcom is empty")
 
+    def get(self, default_value: Any = None) -> Any:
+        if len(self) > 0:
+            return self[0]
+        return default_value
+
+    def set(self, new_value: Any):
+        self.push(new_value)
+        while len(self) > 1:
+            self.pop()
+
     def add_push_callback(self, callback: Callable[[], Any]):
         if not hasattr(self, "push_callbacks"):
             self.push_callbacks: list[Callable[[], Any]] = []

{zrb-1.21.17.dist-info → zrb-1.21.33.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: zrb
-Version: 1.21.17
+Version: 1.21.33
 Summary: Your Automation Powerhouse
 License: AGPL-3.0-or-later
 Keywords: Automation,Task Runner,Code Generator,Monorepo,Low Code
@@ -40,12 +40,12 @@ Requires-Dist: libcst (>=1.8.6,<2.0.0)
 Requires-Dist: markdownify (>=1.2.2,<2.0.0)
 Requires-Dist: mcp (>1.18.0)
 Requires-Dist: mistralai (>=1.9.10) ; extra == "mistral"
-Requires-Dist: openai (>=2.8.0)
+Requires-Dist: openai (>=2.11.0)
 Requires-Dist: pdfplumber (>=0.11.7,<0.12.0)
 Requires-Dist: playwright (>=1.56.0,<2.0.0) ; extra == "playwright" or extra == "all"
 Requires-Dist: prompt-toolkit (>=3)
 Requires-Dist: psutil (>=7.0.0,<8.0.0)
-Requires-Dist: pydantic-ai-slim (>=1.27.0,<1.28.0)
+Requires-Dist: pydantic-ai-slim (>=1.32.0,<1.33.0)
 Requires-Dist: pyjwt (>=2.10.1,<3.0.0)
 Requires-Dist: python-dotenv (>=1.1.1,<2.0.0)
 Requires-Dist: python-jose[cryptography] (>=3.5.0,<4.0.0)

{zrb-1.21.17.dist-info → zrb-1.21.33.dist-info}/RECORD

@@ -1,31 +1,36 @@
 zrb/__init__.py,sha256=qkCV2EnAGIgvsawBHYvKgPAp0zzPcikYSmbQXATLzg4,5060
 zrb/__main__.py,sha256=9SXH9MK4PVyU9lkEyHxiIUABbcsV2wseP94HmlqTR4M,2657
 zrb/attr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-zrb/attr/type.py,sha256=n8elNKAbcZo_AR2apFYWNLdpfEibF5P4yZxHTH41R3Q,541
+zrb/attr/type.py,sha256=eAsYk4aW8Pyi82KU5hfNpQuy6-FLsqcacSj0hJOGNnk,742
 zrb/builtin/__init__.py,sha256=X_uDUEeu86MrYLehHEFiLFdbqrRXMvHECEeIJJkKWh4,1724
 zrb/builtin/base64.py,sha256=UjaFttE2oRx0T7_RpKtKfgMtWfiQXfJBAJmA16ek8Ic,1507
-zrb/builtin/git.py,sha256=8_qVE_2lVQEVXQ9vhiw8Tn4Prj1VZB78ZjEJJS5Ab3M,5461
+zrb/builtin/git.py,sha256=HWxCHW78HJ9SCiph4BMHFfrIpM4V8-bR2MmHLDaWEQM,5810
 zrb/builtin/git_subtree.py,sha256=7BKwOkVTWDrR0DXXQ4iJyHqeR6sV5VYRt8y_rEB0EHg,3505
 zrb/builtin/group.py,sha256=W_IA_4414c8Wi8QazqcT6Gq6UftrKwy95yMF114EJtI,2677
 zrb/builtin/http.py,sha256=L6RE73c65wWwG5iHFN-tpOhyh56KsrgVskDd3c3YXtk,4246
 zrb/builtin/jwt.py,sha256=3M5uaQhJZbKQLjTUft1OwPz_JxtmK-xtkjxWjciOQho,2859
 zrb/builtin/llm/attachment.py,sha256=UwhOGtZY9Px9aebsKfTTw2_ZIpbsqx0XgoAonk6Ewkc,1344
+zrb/builtin/llm/chat_completion.py,sha256=cWVju5ApC2OPGM3ZWGaXwSgLwgT9leKS_TbV4z610zc,12252
 zrb/builtin/llm/chat_session.py,sha256=TmXL7IkVT22blgWqG625qpr2_BnDMIm7ERYbR61BL4Q,9154
-zrb/builtin/llm/chat_session_cmd.py,sha256=xKkOsj0b_ScFfflWYMQJXsC0iUDRACobFkCjRqHnwmE,8306
-zrb/builtin/llm/chat_trigger.py,sha256=xaJmzrvBGz6LFPOpYnG9bMeT1dY6XqZPXamtr9e72-w,2427
+zrb/builtin/llm/chat_session_cmd.py,sha256=lGiN-YNtv957wYuuF_mXV0opbRN61U9pyxVVq-_Gwv8,10842
+zrb/builtin/llm/chat_trigger.py,sha256=i6_lej2xYfY8k5-CMIUsdYQdUvBV9ma_DM6ZmgLaQIU,2673
 zrb/builtin/llm/history.py,sha256=qNRYq6SlJH5iuzdra_FOJxevWXCDh5YMvN2qPNLjI8I,3099
 zrb/builtin/llm/input.py,sha256=Nw-26uTWp2QhUgKJcP_IMHmtk-b542CCSQ_vCOjhvhM,877
 zrb/builtin/llm/llm_ask.py,sha256=6fUbCFjgjFiTVf-88WJZ1RD5wgOzmsVEpD3dR16Jl_8,8598
 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=ItGJLr_1vhmLcitfIKeVDrnJEWdsyqucDnQ7enIC8fQ,2394
-zrb/builtin/llm/tool/cli.py,sha256=iwQtiyPgd5RibUHEtO38c4KS6D4_oCdLqv_bZZqM160,1257
-zrb/builtin/llm/tool/code.py,sha256=gViF2YA6RdCTA1SLZq0v66D0LZv2fdE12qyhDpAYVDg,7575
-zrb/builtin/llm/tool/file.py,sha256=FK_u-gd2OipdGJTU9Vzm_v2BF8YybeBx9W56tJt6Vkc,19131
-zrb/builtin/llm/tool/note.py,sha256=Pg4WgAdZNsIGwP1B7xON4lGiccjl2P0cNpBxTsVh6jI,2584
+zrb/builtin/llm/tool/cli.py,sha256=9BCbB6S5gBladOWO9dJhPgMq8dqpB2LQEfC6druZPDs,2091
+zrb/builtin/llm/tool/code.py,sha256=1sErE5tsticXGNhiE0bTjI21Zq6zHMgS2hVM6dsAcww,7916
+zrb/builtin/llm/tool/file.py,sha256=pzaIWB64aaEuW2I3TyuZwpNQa_nMv-zqJzSfTfjT9GQ,20183
+zrb/builtin/llm/tool/note.py,sha256=USDQe8F1ddQp5dhv2u35rUN9EfJQl5BaWJIs6Xg9ZZY,2597
 zrb/builtin/llm/tool/rag.py,sha256=lmoA42IWHcBOWTsWzYGklLs1YoiEldvu_Bi5bp6gbts,9439
-zrb/builtin/llm/tool/sub_agent.py,sha256=9_DgiXIeieLVe-ccP0T5pA3uY8VIVVh5pfyDRV0JMlk,5005
-zrb/builtin/llm/tool/web.py,sha256=x7iHaGRr71yU3qjfKsoyHvht5SB_S3kWPXX2qliyy6k,6278
+zrb/builtin/llm/tool/search/__init__.py,sha256=wiMaie_IYt00fak-LrioC1qEZDs0SwYz4UWwidHfpFU,49
+zrb/builtin/llm/tool/search/brave.py,sha256=ifuq3qX1rWZjJWw1afGD8VNkqED9Vwv5GJaJYwXmYdM,2271
+zrb/builtin/llm/tool/search/searxng.py,sha256=kSp3HhhHu43eR8RvuzgF5f9EwSy0-uQQyWBYaYfZ82I,2092
+zrb/builtin/llm/tool/search/serpapi.py,sha256=t5AM5PJ_j8VQQWHxFN61-WkDmcXVuvC1ZrFQ1tXZw1w,2101
+zrb/builtin/llm/tool/sub_agent.py,sha256=UNgsvZvSQ1b5ybB0nRzTiFCJpZG3g2RLj_nzjRZzXMY,5808
+zrb/builtin/llm/tool/web.py,sha256=nBpCN9sZ66Ly6nzpnNnUunXN_OJqClmShGI-6saEWFk,4364
 zrb/builtin/md5.py,sha256=690RV2LbW7wQeTFxY-lmmqTSVEEZv3XZbjEUW1Q3XpE,1480
 zrb/builtin/project/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/builtin/project/add/fastapp/fastapp_input.py,sha256=MKlWR_LxWhM_DcULCtLfL_IjTxpDnDBkn9KIqNmajFs,310
@@ -223,19 +228,19 @@ 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/config.py,sha256=_jhYGeZevx8za2MSwZ0HtHyJzFoPmxVtZMKyfw_1trs,18728
+zrb/config/config.py,sha256=bqC_B9OpofNWJkWA11HvtWD-Ut0VeKseIso5OACU6v4,19889
 zrb/config/default_prompt/file_extractor_system_prompt.md,sha256=cVqhI016-fKa8PKz2kIxQAC0hPExgMT2ks-rVlz4K60,3789
-zrb/config/default_prompt/interactive_system_prompt.md,sha256=0-SqyWqrrhpJ84qjSSpi7p4sFV31qnTgLxG0vZ2wKsE,2304
+zrb/config/default_prompt/interactive_system_prompt.md,sha256=LewDZo_zXMP6U_42c5egAY0Hb4KYp4Tbf3rUd9bPK8k,2659
 zrb/config/default_prompt/persona.md,sha256=GfUJ4-Mlf_Bm1YTzxFNkPkdVbAi06ZDVYh-iIma3NOs,253
 zrb/config/default_prompt/repo_extractor_system_prompt.md,sha256=lmfMMc9BLLATHAVMmSEeXr7xTeiF4FWv_AWYedgcbDE,3786
 zrb/config/default_prompt/repo_summarizer_system_prompt.md,sha256=-T_baTblam2XlW99e72jfgsrIBTjpABMBXRFI_7L6RA,1987
-zrb/config/default_prompt/summarization_prompt.md,sha256=7Syk3uhf3AUuVP07QpEbIAiQa6ac32DlfNvWaa3OFog,947
-zrb/config/default_prompt/system_prompt.md,sha256=zGG1ngG4bFURDsNNJn91CC4LIJnRBbwdJPDxAnBEoHw,2670
+zrb/config/default_prompt/summarization_prompt.md,sha256=0FI7GO9wkOOP7I_CrJPECtrqDHE_SMNr9aKKHC1GtPI,3633
+zrb/config/default_prompt/system_prompt.md,sha256=ipPaECOM4KQj5u2bzQj1f1w0J78vuZuLnxF5orZWElo,2857
 zrb/config/llm_config.py,sha256=Nt7P0Q6bRpLbGi3IZ1WPeOxUsUUL438_8ufpML1WN28,12798
 zrb/config/llm_context/config.py,sha256=WdqoFxc-PSwNh9aeTWhHMNiQZotbdQ8WaVr3YcezGNo,6487
 zrb/config/llm_context/config_parser.py,sha256=PLfLfMgxPBj7Ryy-37gCZIHrweo1XiFQGnUqjvKXU_Y,1465
 zrb/config/llm_context/workflow.py,sha256=cHkUYBk2Mc35RBuma7Vn3QCNaHd95Ajqr1N4L43FFZs,2681
-zrb/config/llm_rate_limitter.py,sha256=gge6CvwZVSZ1ainL7fubvNpsFFmMEW5sPgJXmyRfFGE,6489
+zrb/config/llm_rate_limitter.py,sha256=sNi_fttwRKjMTCn1pBoqx-p2uI4wHSl8CAAP8poe94g,7012
 zrb/config/web_auth_config.py,sha256=_PXatQTYh2mX9H3HSYSQKp13zm1RlLyVIoeIr6KYMQ8,6279
 zrb/content_transformer/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/content_transformer/any_content_transformer.py,sha256=v8ZUbcix1GGeDQwB6OKX_1TjpY__ksxWVeqibwa_iZA,850
@@ -261,7 +266,7 @@ zrb/input/base_input.py,sha256=HCTxBZwnK-gP-F7XQGDbyyZHvwYR0fwNICzqVUw5Cg8,4169
 zrb/input/bool_input.py,sha256=9ir8aTm8zaufrJ84_EerC5vOjyn4hHZ3BiLzFqPgPFM,1839
 zrb/input/float_input.py,sha256=8G9lJKLlb_Upk9m5pBF9JwqlAAiYJLrbIItLnyzPxpg,1475
 zrb/input/int_input.py,sha256=UhxCFYlZdJcgUSGGEkz301zOgRVpK0KDG_IxxWpQfMU,1457
-zrb/input/option_input.py,sha256=TQB82ko5odgzkULEizBZi0e9TIHEbIgvdP0AR3RhA74,2135
+zrb/input/option_input.py,sha256=pG-7R5QboXRLyb4E63PXIUnhs5fHu2RCkGLBue46mmU,2687
 zrb/input/password_input.py,sha256=szBojWxSP9QJecgsgA87OIYwQrY2AQ3USIKdDZY6snU,1465
 zrb/input/str_input.py,sha256=NevZHX9rf1g8eMatPyy-kUX3DglrVAQpzvVpKAzf7bA,81
 zrb/input/text_input.py,sha256=9vpXJt7ASa_Gq2zFbVWT1p-PBjYf6DpyXfuhIH-rJgs,2969
@@ -353,10 +358,11 @@ zrb/task/base_trigger.py,sha256=nbhFY3QCLNZkXxRorjyC4k2sEJumzR8YCNjOhFBSjJ8,7165
 zrb/task/cmd_task.py,sha256=myM8WZm6NrUD-Wv0Vb5sTOrutrAVZLt5LVsSBKwX6SM,10860
 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=sbwRhN6La-ugng6JVpNHxk81tZoe6LYnktPj-hNaiKg,11102
+zrb/task/llm/agent.py,sha256=FrwqAt2SKm78C1fXAPJrXOZfd8iu-13n_R3tVetJlfI,8333
+zrb/task/llm/agent_runner.py,sha256=7VmLnhtF1oWfuV6xMCIwd4xPewpDXgfIPEJRoX9boiA,5202
 zrb/task/llm/config.py,sha256=wCcwxcaW2dgjxsjrXnG1inyKpQWs37kXJ-L_CO-KJ5s,4486
-zrb/task/llm/conversation_history.py,sha256=gxnRd0-X5olQmhJ9aWq4v0KVxYN7jDpPZHJSgkCeDvk,6682
-zrb/task/llm/conversation_history_model.py,sha256=kk-7niTl29Rm2EUIhTHzPXgZ5tp4IThMnIB3dS-1OdU,3062
+zrb/task/llm/conversation_history.py,sha256=Fye5oryjdDN_JGBkxoPGUsjuU0FTHis1Ue6nTovo1Zo,7112
+zrb/task/llm/conversation_history_model.py,sha256=2ScAuu5ne6X6F0bXm-21ckEoqcDlWDeuD5L4DPPdz-Q,2629
 zrb/task/llm/default_workflow/coding/workflow.md,sha256=tGiZuR5SYtLvvILJ_AkHaWdvXn5wvLd3RqNyjnHq9NE,5474
 zrb/task/llm/default_workflow/copywriting/workflow.md,sha256=CL9083lrMZHa-J6GlbyGxgRxa8CqJJQFMw_I7WFK13w,3182
 zrb/task/llm/default_workflow/git/workflow.md,sha256=BBkIChqLkex-irkaWH9lQlQkVHed1SM28nfgx8K3yZA,4263
@@ -371,14 +377,16 @@ zrb/task/llm/default_workflow/shell/workflow.md,sha256=_5aW5DEjOauiUANNWgbfEFmH5
 zrb/task/llm/error.py,sha256=QR-nIohS6pBpC_16cWR-fw7Mevo1sNYAiXMBsh_CJDE,4157
 zrb/task/llm/file_replacement.py,sha256=cUgtcfH9OlHzPwdZhSzWVNs7ZO0YYh0ajxqMwfIbvD0,7193
 zrb/task/llm/file_tool_model.py,sha256=bNH-nrHamXtc_sUiPZ1TYI6CqefPIXj3tP4FbURiU1Y,1721
-zrb/task/llm/history_summarization.py,sha256=V8YA8Q9cIcH7L0YfEJf87Gys5lGMhDrVV3WWqHz-4R0,7501
-zrb/task/llm/history_summarization_tool.py,sha256=3R7-7iNLFQbKTgc1EE_Q_RfP8MvJovPukx_E3tD6apE,949
-zrb/task/llm/print_node.py,sha256=jJ-IaQ9PYNeFr_NWnW9BidTVrETOQN1_5aSYX7L8di8,8253
-zrb/task/llm/prompt.py,sha256=4nutDWNQHcJbvKRXrgEnicY8NjMSCBH2aIx4QwPdiiU,11931
-zrb/task/llm/tool_wrapper.py,sha256=fUXGHTaGPG3NTQW0rFhGPF0wrhQJoSCjPppfoWIdHEU,12077
+zrb/task/llm/history_processor.py,sha256=QQHoE2RNz8w75li-RC4PFgCelCLhVv4ObOiSQQ8JxDM,7684
+zrb/task/llm/history_summarization.py,sha256=YqLm9Q5xzFVTL5XwnWu3XwaMgSxKskDFvoi2MIThRB0,927
+zrb/task/llm/print_node.py,sha256=d7OMOJ2kj6iXZS2cC5cyuDF1QDFsA9FxDGX-xfirvxs,8674
+zrb/task/llm/prompt.py,sha256=jlFNWa1IMCYNQh9Ccq9DB1dq9964IOQJ5jNSC9rMpkw,11366
+zrb/task/llm/subagent_conversation_history.py,sha256=y8UbKF2GpjB-r1bc2xCpzepriVEq0wfdyFuHOwfpPfk,1647
+zrb/task/llm/tool_confirmation_completer.py,sha256=rOYZdq5No9DfLUG889cr99cUX6XmoO29Yh9FUcdMMio,1534
+zrb/task/llm/tool_wrapper.py,sha256=FfW2JcZ8vUVie0pekpwpQABQ8DhIv8EG7_kPSYGwsVw,12440
 zrb/task/llm/typing.py,sha256=c8VAuPBw_4A3DxfYdydkgedaP-LU61W9_wj3m3CAX1E,58
 zrb/task/llm/workflow.py,sha256=eGpUXoIKgbPFPZJ7hHy_HkGYnGWWx9-B9FNjjNm3_8I,2968
-zrb/task/llm_task.py,sha256=HTkE6vjgs4AJ4RiJ58xZgoYhTrqZJHBZbALBjEcZNOc,16203
+zrb/task/llm_task.py,sha256=ZH5XN064tJy2dTEnjf43Tk32-cL3-0Ey9JISz1hooTw,16388
 zrb/task/make_task.py,sha256=rWay-Wz2nAOr4mhMcJZuuehuWULfSBmZm5XcDN4ZN3c,2198
 zrb/task/rsync_task.py,sha256=0M5g2E_b28uCNLXSq2vTsDP2TdGpHfhNn1sDv4MhzNw,7089
 zrb/task/scaffolder.py,sha256=rME18w1HJUHXgi9eTYXx_T2G4JdqDYzBoNOkdOOo5-o,6806
@@ -388,7 +396,7 @@ zrb/task/tcp_check.py,sha256=P_QgGqwd5dXDaud3oQRxe_WuxyxG4s7CTY2wDk9Qcu0,2511
 zrb/task_status/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/task_status/task_status.py,sha256=blZ8dxg9g_8MuViq-t7yJRLoE7yGUf5srgHf-PCsXNc,3069
 zrb/util/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-zrb/util/attr.py,sha256=ctS4wh2skAGsG1E3_6yPLFPRU3--Vhf7ivOzHn3rJak,5312
+zrb/util/attr.py,sha256=3-arwThz8pOnLf4PRA_D-5Pue9xKVEmLtESsIXHCBSE,5523
 zrb/util/callable.py,sha256=b6OFXbCXp2twow3wh2E_h5hNHLs2pXaLfGQz4iVyiQc,771
 zrb/util/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/util/cli/markdown.py,sha256=Uhuw8XR-jAG9AG3oNK8VHJpYOdU40Q_8yVN74uu0RJ8,384
@@ -410,7 +418,7 @@ zrb/util/codemod/modify_method.py,sha256=5fioXjqNQmrf4CV2qlTZHpViF9PMnNer4FvuKam
 zrb/util/codemod/modify_module.py,sha256=2mzi_NxJ-kNFo5G0U_Rqb3JoYQl1s6Izt7b_wAl10F0,715
 zrb/util/cron.py,sha256=UWqqhhM7DDTPx6wjCIdBndnVh3NRu-sdKazp8aZkXh8,3833
 zrb/util/file.py,sha256=BUWGfJwlS5FoaFgZ2CBG9LVcm-5cdqtFjaYda0VVs2s,3262
-zrb/util/git.py,sha256=GS08OqE-gs6mN-_VqACN82DbI81nzG3FiXeim_-jVAU,8193
+zrb/util/git.py,sha256=Zy8F22IDJy-PIHH7wNy6KpADetrm4us2l0s9JeMS_DM,8166
 zrb/util/git_diff_model.py,sha256=Tg2q6oxQ5chryThzjs0cLcKFGM03CnqvusTo_Ug_oX4,369
 zrb/util/git_subtree.py,sha256=AyQWCWEi2EIzEpYXRnYN55157KMUql0WHj70QNw5PHU,4612
 zrb/util/git_subtree_model.py,sha256=P_gJ0zhOAc3gFM6sYcjc0Ack9dFBt75TI5fXdE0q320,871
@@ -428,8 +436,8 @@ zrb/util/todo_model.py,sha256=hhzAX-uFl5rsg7iVX1ULlJOfBtblwQ_ieNUxBWfc-Os,1670
 zrb/util/truncate.py,sha256=eSzmjBpc1Qod3lM3M73snNbDOcARHukW_tq36dWdPvc,921
 zrb/util/yaml.py,sha256=MXcqQTOYzx16onFxxg6HNdavxDK4RQjvKUi2NMKd19Y,6510
 zrb/xcom/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-zrb/xcom/xcom.py,sha256=o79rxR9wphnShrcIushA0Qt71d_p3ZTxjNf7x9hJB78,1571
-zrb-1.21.17.dist-info/METADATA,sha256=i7CFPGUQ2XMW7HmXlIbuq-b9GkvL5dtxsEXeS9SmW8g,10621
-zrb-1.21.17.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-zrb-1.21.17.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
-zrb-1.21.17.dist-info/RECORD,,
+zrb/xcom/xcom.py,sha256=W5BOF9w5fVX2ZBlbOTMsXHbY0yU8Advm-_oWJ3DJvDM,1824
+zrb-1.21.33.dist-info/METADATA,sha256=8_GuNfrNEPw5ZvdP-LqhFH7HAP4CfApFxhli-t3TI3k,10622
+zrb-1.21.33.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+zrb-1.21.33.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
+zrb-1.21.33.dist-info/RECORD,,

zrb/task/llm/history_summarization_tool.py

@@ -1,24 +0,0 @@
-from typing import Callable
-
-from zrb.task.llm.conversation_history_model import ConversationHistory
-
-
-def create_history_summarization_tool(
-    conversation_history: ConversationHistory,
-) -> Callable[[str, str], str]:
-    def update_conversation_memory(
-        past_conversation_summary: str,
-        past_conversation_transcript: str,
-    ) -> str:
-        """
-        Update the conversation memory including summary and transcript.
-        - past_conversation_summary: A concise narrative that integrates the
-          previous summary with the recent conversation.
-        - past_conversation_transcript: MUST be ONLY the last 4 (four) turns
-          of the conversation.
-        """
-        conversation_history.past_conversation_summary = past_conversation_summary
-        conversation_history.past_conversation_transcript = past_conversation_transcript
-        return "Conversation memory updated"
-
-    return update_conversation_memory