zrb 1.15.14__py3-none-any.whl → 1.15.16__py3-none-any.whl

zrb/config/default_prompt/summarization_prompt.md

@@ -1,17 +1,16 @@
-You are a memory management AI. Your goal is to curate the conversation history by calling the `update_conversation_memory` tool.
+You are a memory management AI. Your only task is to process the provided conversation history and call the `update_conversation_memory` tool **once**.
 
-Follow these steps precisely:
+Follow these instructions carefully:
 
-**Step 1: Consolidate Conversation Information**
+1.  **Summarize:** Create a concise narrative summary that integrates the `Past Conversation Summary` with the `Recent Conversation`. **This summary must not be more than two paragraphs.**
+2.  **Transcript:** Extract ONLY the last 4 (four) turns of the `Recent Conversation` to serve as the new transcript.
+    *   **Do not change or shorten the content of these turns, with one exception:** If a tool call returns a very long output, do not include the full output. Instead, briefly summarize the result of the tool call.
+    *   Ensure the timestamp format is `[YYYY-MM-DD HH:MM:SS UTC+Z] Role: Message/Tool name being called`.
+3.  **Notes:** Review the `Notes` and `Recent Conversation` to identify new or updated facts.
+    *   Update `long_term_note` with global facts about the user.
+    *   Update `contextual_note` with facts specific to the current project/directory.
+    *   **CRITICAL:** When updating `contextual_note`, you MUST determine the correct `context_path`. For example, if a fact was established when the working directory was `/app`, the `context_path` MUST be `/app`.
+    *   **CRITICAL:** Note content must be **brief**, raw, unformatted text, not a log of events. Only update notes if information has changed.
+4.  **Update Memory:** Call the `update_conversation_memory` tool with all the information you consolidated.
 
-1.  Create a concise narrative summary that integrates the `Past Conversation Summary` with the `Recent Conversation`.
-2.  Extract ONLY the last 4 (four) turns of the `Recent Conversation` to serve as the new transcript. Do not change or shorten the content of these turns. Ensure the timestamp format is `[YYYY-MM-DD HH:MM:SS UTC+Z] Role: Message/Tool name being called`.
-3.  Review the `Notes` and the `Recent Conversation` to identify any new or updated facts.
-    *   Update global facts about the user or their preferences for the `long_term_note`.
-    *   Update facts specific to the current project or directory for the `contextual_note`.
-    *   **CRITICAL:** When updating `contextual_note`, you MUST determine the correct `context_path` by analyzing the `Recent Conversation`. For example, if a fact was established when the working directory was `/app`, the `context_path` MUST be `/app`.
-    *   **CRITICAL:** The content for notes must be raw, unformatted text. Do not use Markdown. Notes should be timeless facts, not a log of events. Only update notes if the information has actually changed.
-
-**Step 2: Update Memory**
-
-1.  Call the `update_conversation_memory` tool ONCE, providing all the information you consolidated in Step 1 as arguments.
+After you have called the tool, your task is complete. Your final output **MUST** be the single word `OK`. Do not add any other text, formatting, or tool calls.

zrb/input/base_input.py

@@ -96,11 +96,19 @@ class BaseInput(AnyInput):
         if default_str != "":
             prompt_message = f"{prompt_message} [{default_str}]"
         print(f"{prompt_message}: ", end="")
-        value = input()
+        value = self._read_line(shared_ctx)
         if value.strip() == "":
             value = default_str
         return value
 
+    def _read_line(self, shared_ctx: AnySharedContext) -> str:
+        if not shared_ctx.is_tty:
+            return input()
+        from prompt_toolkit import PromptSession
+
+        reader = PromptSession()
+        return reader.prompt()
+
     def get_default_str(self, shared_ctx: AnySharedContext) -> str:
         """Get default value as str"""
         default_value = get_attr(

zrb/task/llm/agent.py

@@ -185,7 +185,7 @@ async def _run_single_agent_iteration(
     agent_payload = _estimate_request_payload(
         agent, user_prompt, attachments, history_list
     )
-    callback = lambda: _print_throttle_notif(ctx)
+    callback = _create_print_throttle_notif(ctx)
     if rate_limitter:
         await rate_limitter.throttle(agent_payload, callback)
     else:
@@ -216,8 +216,11 @@ async def _run_single_agent_iteration(
             return agent_run
 
 
-def _print_throttle_notif(ctx: AnyContext):
-    ctx.print(stylize_faint("  ⌛>> Request Throttled"), plain=True)
+def _create_print_throttle_notif(ctx: AnyContext) -> Callable[[], None]:
+    def _print_throttle_notif(ctx: AnyContext):
+        ctx.print(stylize_faint("  ⌛>> Request Throttled"), plain=True)
+
+    return _print_throttle_notif
 
 
 def _estimate_request_payload(

zrb/task/llm/print_node.py

@@ -48,6 +48,12 @@ async def print_node(
                                 _format_stream_content(content_delta, log_indent_level),
                                 end="",
                             )
+                        elif isinstance(event.delta, ThinkingPartDelta):
+                            content_delta = event.delta.content_delta
+                            print_func(
+                                _format_stream_content(content_delta, log_indent_level),
+                                end="",
+                            )
                         elif isinstance(event.delta, ToolCallPartDelta):
                             args_delta = event.delta.args_delta
                             print_func(
@@ -71,7 +77,10 @@ async def print_node(
             print_func("")  # ensure newline consistency
             print_func(
                 _format_content(
-                    f"⚠️ Unexpected Model Behavior: {e}. node_id={meta}",
+                    (
+                        f"⚠️ Unexpected Model Behavior: {e}. "
+                        f"Cause: {e.__cause__}. Node.Id: {meta}"
+                    ),
                     log_indent_level,
                 )
             )
@@ -102,7 +111,10 @@ async def print_node(
             print_func("")  # ensure newline consistency
             print_func(
                 _format_content(
-                    f"⚠️ Unexpected Model Behavior: {e}. node_id={meta}",
+                    (
+                        f"⚠️ Unexpected Model Behavior: {e}. "
+                        f"Cause: {e.__cause__}. Node.Id: {meta}"
+                    ),
                     log_indent_level,
                 )
             )

{zrb-1.15.14.dist-info → zrb-1.15.16.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: zrb
-Version: 1.15.14
+Version: 1.15.16
 Summary: Your Automation Powerhouse
 License: AGPL-3.0-or-later
 Keywords: Automation,Task Runner,Code Generator,Monorepo,Low Code
@@ -26,7 +26,7 @@ Requires-Dist: markdownify (>=1.2.0,<2.0.0)
 Requires-Dist: pdfplumber (>=0.11.7,<0.12.0)
 Requires-Dist: playwright (>=1.54.0,<2.0.0) ; extra == "playwright" or extra == "all"
 Requires-Dist: psutil (>=7.0.0,<8.0.0)
-Requires-Dist: pydantic-ai-slim[anthropic,bedrock,cli,cohere,google,groq,huggingface,mcp,mistral,openai,vertexai] (>=0.7.5,<0.8.0)
+Requires-Dist: pydantic-ai-slim[anthropic,bedrock,cli,cohere,google,groq,huggingface,mcp,mistral,openai,vertexai] (>=0.8.0,<0.9.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.15.14.dist-info → zrb-1.15.16.dist-info}/RECORD

@@ -223,7 +223,7 @@ zrb/config/default_prompt/interactive_system_prompt.md,sha256=XvXI51dMpQmuuYah_L
 zrb/config/default_prompt/persona.md,sha256=GfUJ4-Mlf_Bm1YTzxFNkPkdVbAi06ZDVYh-iIma3NOs,253
 zrb/config/default_prompt/repo_extractor_system_prompt.md,sha256=EGZ-zj78RlMEg2jduRBs8WzO4VJTkXHR96IpBepZMsY,3881
 zrb/config/default_prompt/repo_summarizer_system_prompt.md,sha256=RNy37Wg7ibXj3DlsFKaYvgMpMS-lyXlM1LZlc59_4ic,2009
-zrb/config/default_prompt/summarization_prompt.md,sha256=GMqbw7mmDWQf7rhnJcJjXQFtGu7nVYIZfK6sStJ4coc,1472
+zrb/config/default_prompt/summarization_prompt.md,sha256=vbxzJGeH7uCdgnfbVt3GYOjY-BygmeRfJ6tnzXmFdoY,1698
 zrb/config/default_prompt/system_prompt.md,sha256=gEb6N-cFg6VvOV-7ZffNwVt39DavAGesMqn9u0epbRc,2282
 zrb/config/llm_config.py,sha256=xt-Xf8ZuNoUT_GKCSFz5yy0BhbeHzxP-jrezB06WeiY,8857
 zrb/config/llm_context/config.py,sha256=TPQX_kU772r0AHmVFeo1WGLDAidacT-qDyuMWxY_avg,5878
@@ -250,7 +250,7 @@ zrb/group/any_group.py,sha256=sdUk82YRuJBMwDhD_b-vJ3v-Yxz8K7LFW52TM19k3HI,1242
 zrb/group/group.py,sha256=t5JnXlrLfthFfqmkVdRLKCIyms4JqsthcTODVncSmPk,4312
 zrb/input/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/input/any_input.py,sha256=3EQgg2Qk_2t2Ege_4pHV2G95tV3XAIThLLuFZ6uV6oM,1051
-zrb/input/base_input.py,sha256=Rb2m9xLJgGDph2D8L9HnLFPtQZqmPAPKrSP2K4pK97A,3848
+zrb/input/base_input.py,sha256=oK8lm2UsPe859I8TzmvIhN8a4GZR78BMZuhhVt_GqGI,4107
 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
@@ -346,7 +346,7 @@ zrb/task/base_trigger.py,sha256=WSGcmBcGAZw8EzUXfmCjqJQkz8GEmi1RzogpF6A1V4s,6902
 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=jfC90N8oER2z-fTmPEuWH3ASPgw5t8cZm8rk98HBBlw,8876
+zrb/task/llm/agent.py,sha256=-DVECvW3nLQNmppkD9DStG6L_VUaUUAJb2iMmd-JB90,8990
 zrb/task/llm/config.py,sha256=n1SPmwab09K2i1sL_OCwrEOWHI0Owx_hvWelg3Dreus,3781
 zrb/task/llm/conversation_history.py,sha256=oMdKUV2__mBZ4znnA-prl-gfyoleKC8Nj5KNpmLQJ4o,6764
 zrb/task/llm/conversation_history_model.py,sha256=kk-7niTl29Rm2EUIhTHzPXgZ5tp4IThMnIB3dS-1OdU,3062
@@ -356,7 +356,7 @@ zrb/task/llm/default_workflow/researching.md,sha256=KD-aYHFHir6Ti-4FsBBtGwiI0seS
 zrb/task/llm/error.py,sha256=QR-nIohS6pBpC_16cWR-fw7Mevo1sNYAiXMBsh_CJDE,4157
 zrb/task/llm/history_summarization.py,sha256=UIT8bpdT3hy1xn559waDLFWZlNtIqdIpIvRGcZEpHm0,8057
 zrb/task/llm/history_summarization_tool.py,sha256=Wazi4WMr3k1WJ1v7QgjAPbuY1JdBpHUsTWGt3DSTsLc,1706
-zrb/task/llm/print_node.py,sha256=fNTL0LtoZBQQPHYCAUaOFP97nNNfobPXvY2RlxewQCE,7556
+zrb/task/llm/print_node.py,sha256=si619OvCUZJYTUSUpoOqVAgLSu1BGw-dlBq02MSD7FE,8096
 zrb/task/llm/prompt.py,sha256=FGXWYHecWtrNNkPnjg-uhnkqp7fYt8V91-AjFM_5fpA,11550
 zrb/task/llm/tool_wrapper.py,sha256=IavXwW9C0ojI3ivGDY8j7XbMJE_NTJeI86TVbHNRwJ0,9684
 zrb/task/llm/typing.py,sha256=c8VAuPBw_4A3DxfYdydkgedaP-LU61W9_wj3m3CAX1E,58
@@ -409,7 +409,7 @@ zrb/util/todo_model.py,sha256=hhzAX-uFl5rsg7iVX1ULlJOfBtblwQ_ieNUxBWfc-Os,1670
 zrb/util/truncate.py,sha256=eSzmjBpc1Qod3lM3M73snNbDOcARHukW_tq36dWdPvc,921
 zrb/xcom/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/xcom/xcom.py,sha256=o79rxR9wphnShrcIushA0Qt71d_p3ZTxjNf7x9hJB78,1571
-zrb-1.15.14.dist-info/METADATA,sha256=UzPUFFhPCo01R3K87PDAMkPGMi_0mGYliiKr0zPFDSs,9775
-zrb-1.15.14.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-zrb-1.15.14.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
-zrb-1.15.14.dist-info/RECORD,,
+zrb-1.15.16.dist-info/METADATA,sha256=7XfExXjX7pcW0SzAEoDwJQ1WXa02JfM6JyIddL9n96k,9775
+zrb-1.15.16.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+zrb-1.15.16.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
+zrb-1.15.16.dist-info/RECORD,,