zrb 1.15.13__py3-none-any.whl → 1.15.15__py3-none-any.whl

zrb/builtin/llm/tool/web.py

@@ -98,7 +98,13 @@ def search_wikipedia(query: str) -> str:
     import requests
 
     params = {"action": "query", "list": "search", "srsearch": query, "format": "json"}
-    response = requests.get("https://en.wikipedia.org/w/api.php", params=params)
+    response = requests.get(
+        "https://en.wikipedia.org/w/api.php",
+        headers={
+            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36"
+        },
+        params=params,
+    )
     return response.json()
 
 
@@ -123,7 +129,13 @@ def search_arxiv(query: str, num_results: int = 10) -> str:
     import requests
 
     params = {"search_query": f"all:{query}", "start": 0, "max_results": num_results}
-    response = requests.get("http://export.arxiv.org/api/query", params=params)
+    response = requests.get(
+        "http://export.arxiv.org/api/query",
+        headers={
+            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36"
+        },
+        params=params,
+    )
     return response.content
 
 

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/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.13.dist-info → zrb-1.15.15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: zrb
-Version: 1.15.13
+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.13.dist-info → zrb-1.15.15.dist-info}/RECORD

@@ -21,7 +21,7 @@ zrb/builtin/llm/tool/code.py,sha256=fr9FbmtfwizQTyTztvuvwAb9MD_auRZhPZfoJVBlKT4,
 zrb/builtin/llm/tool/file.py,sha256=eXFGGFxxpdpWGVw0svyQNQc03I5M7wotSsA_HjkXw7c,23670
 zrb/builtin/llm/tool/rag.py,sha256=Ab8_ZljnG_zfkwxPezImvorshuz3Fi4CmSzNOtU1a-g,9770
 zrb/builtin/llm/tool/sub_agent.py,sha256=dPqdFCXxJ-3hZnPjheZr-bPeKNbfSBSDKcZZR5vkWFM,5075
-zrb/builtin/llm/tool/web.py,sha256=zNQWDNOz_MyyzhncBeV5E_9XijQxTFVp6BSzz_e5tBI,7261
+zrb/builtin/llm/tool/web.py,sha256=Hc9ikgBWZMgLB2O5lX-P9k8jvNSZeyUufqHXa_n_v4I,7653
 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,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
@@ -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.13.dist-info/METADATA,sha256=lDKm2YdMNhKPRFayict53Twk3Bd_nnQExw7tZFqWe5c,9775
-zrb-1.15.13.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-zrb-1.15.13.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
-zrb-1.15.13.dist-info/RECORD,,
+zrb-1.15.15.dist-info/METADATA,sha256=VWZ52YsfHRjEU0_qygR3DMqufPQWd_rWaNqFabU7EDo,9775
+zrb-1.15.15.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+zrb-1.15.15.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
+zrb-1.15.15.dist-info/RECORD,,