zrb 1.15.14__tar.gz → 1.15.15__tar.gz

{zrb-1.15.14 → zrb-1.15.15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: zrb
-Version: 1.15.14
+Version: 1.15.15
 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 → zrb-1.15.15}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "zrb"
-version = "1.15.14"
+version = "1.15.15"
 description = "Your Automation Powerhouse"
 authors = ["Go Frendi Gunawan <gofrendiasgard@gmail.com>"]
 license = "AGPL-3.0-or-later"
@@ -53,7 +53,7 @@ psutil = "^7.0.0"
 # NOTE: Look at https://github.com/pydantic/pydantic-ai/blob/main/pydantic_ai_slim/pyproject.toml
 # - We want to include all models, as well as mcp capability.
 # - We also need prompt-toolkit and rich which are included in pydantic ai's cli optional dependency
-pydantic-ai-slim = {extras = ["openai", "cohere", "vertexai", "google", "anthropic", "groq", "mistral", "bedrock", "huggingface", "cli", "mcp"], version = "~0.7.5"} 
+pydantic-ai-slim = {extras = ["openai", "cohere", "vertexai", "google", "anthropic", "groq", "mistral", "bedrock", "huggingface", "cli", "mcp"], version = "~0.8.0"} 
 
 tiktoken = "^0.8.0"
 playwright = {version = "^1.54.0", optional = true}

zrb-1.15.15/src/zrb/config/default_prompt/summarization_prompt.md

@@ -0,0 +1,16 @@
+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 instructions carefully:
+
+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.
+
+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-1.15.14 → zrb-1.15.15}/src/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-1.15.14 → zrb-1.15.15}/src/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/src/zrb/config/default_prompt/summarization_prompt.md

@@ -1,17 +0,0 @@
-You are a memory management AI. Your goal is to curate the conversation history by calling the `update_conversation_memory` tool.
-
-Follow these steps precisely:
-
-**Step 1: Consolidate Conversation Information**
-
-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.