zrb 1.15.6__py3-none-any.whl → 1.15.7__py3-none-any.whl

zrb/task/llm/print_node.py

@@ -10,6 +10,7 @@ async def print_node(
 ):
     """Prints the details of an agent execution node using a provided print function."""
     from pydantic_ai import Agent
+    from pydantic_ai.exceptions import UnexpectedModelBehavior
     from pydantic_ai.messages import (
         FinalResultEvent,
         FunctionToolCallEvent,
@@ -21,70 +22,90 @@ async def print_node(
         ToolCallPartDelta,
     )
 
+    meta = getattr(node, "id", None) or getattr(node, "request_id", None)
     if Agent.is_user_prompt_node(node):
         print_func(_format_header("🔠 Receiving input...", log_indent_level))
     elif Agent.is_model_request_node(node):
         # A model request node => We can stream tokens from the model's request
         print_func(_format_header("🧠 Processing...", log_indent_level))
         # Reference: https://ai.pydantic.dev/agents/#streaming
-        async with node.stream(agent_run.ctx) as request_stream:
-            is_streaming = False
-            async for event in request_stream:
-                if isinstance(event, PartStartEvent) and event.part:
-                    if is_streaming:
-                        print_func("")
-                    content = _get_event_part_content(event)
-                    print_func(_format_content(content, log_indent_level), end="")
-                    is_streaming = True
-                elif isinstance(event, PartDeltaEvent):
-                    if isinstance(event.delta, TextPartDelta) or 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
+        try:
+            async with node.stream(agent_run.ctx) as request_stream:
+                is_streaming = False
+                async for event in request_stream:
+                    if isinstance(event, PartStartEvent) and event.part:
+                        if is_streaming:
+                            print_func("")
+                        content = _get_event_part_content(event)
+                        print_func(_format_content(content, log_indent_level), end="")
+                        is_streaming = True
+                    elif isinstance(event, PartDeltaEvent):
+                        if isinstance(event.delta, TextPartDelta) or 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(
+                                _format_stream_content(args_delta, log_indent_level),
+                                end="",
+                            )
+                        is_streaming = True
+                    elif isinstance(event, FinalResultEvent) and event.tool_name:
+                        if is_streaming:
+                            print_func("")
+                        tool_name = event.tool_name
                         print_func(
-                            _format_stream_content(args_delta, log_indent_level), end=""
-                        )
-                    is_streaming = True
-                elif isinstance(event, FinalResultEvent) and event.tool_name:
-                    if is_streaming:
-                        print_func("")
-                    tool_name = event.tool_name
-                    print_func(
-                        _format_content(
-                            f"Result: tool_name={tool_name}", log_indent_level
+                            _format_content(
+                                f"Result: tool_name={tool_name}", log_indent_level
+                            )
                         )
-                    )
-                    is_streaming = False
-            if is_streaming:
-                print_func("")
+                        is_streaming = False
+                if is_streaming:
+                    print_func("")
+        except UnexpectedModelBehavior as e:
+            print_func("")  # ensure newline consistency
+            print_func(
+                _format_content(
+                    f"⚠️ Unexpected Model Behavior: {e}. node_id={meta}",
+                    log_indent_level,
+                )
+            )
     elif Agent.is_call_tools_node(node):
         # A handle-response node => The model returned some data, potentially calls a tool
         print_func(_format_header("🧰 Calling Tool...", log_indent_level))
-        async with node.stream(agent_run.ctx) as handle_stream:
-            async for event in handle_stream:
-                if isinstance(event, FunctionToolCallEvent):
-                    args = _get_event_part_args(event)
-                    call_id = event.part.tool_call_id
-                    tool_name = event.part.tool_name
-                    print_func(
-                        _format_content(
-                            f"{call_id} | Call {tool_name} {args}", log_indent_level
+        try:
+            async with node.stream(agent_run.ctx) as handle_stream:
+                async for event in handle_stream:
+                    if isinstance(event, FunctionToolCallEvent):
+                        args = _get_event_part_args(event)
+                        call_id = event.part.tool_call_id
+                        tool_name = event.part.tool_name
+                        print_func(
+                            _format_content(
+                                f"{call_id} | Call {tool_name} {args}", log_indent_level
+                            )
                         )
-                    )
-                elif isinstance(event, FunctionToolResultEvent):
-                    call_id = event.tool_call_id
-                    result_content = event.result.content
-                    print_func(
-                        _format_content(
-                            f"{call_id} | {result_content}", log_indent_level
+                    elif isinstance(event, FunctionToolResultEvent):
+                        call_id = event.tool_call_id
+                        result_content = event.result.content
+                        print_func(
+                            _format_content(
+                                f"{call_id} | {result_content}", log_indent_level
+                            )
                         )
-                    )
+        except UnexpectedModelBehavior as e:
+            print_func("")  # ensure newline consistency
+            print_func(
+                _format_content(
+                    f"⚠️ Unexpected Model Behavior: {e}. node_id={meta}",
+                    log_indent_level,
+                )
+            )
     elif Agent.is_end_node(node):
         # Once an End node is reached, the agent run is complete
         print_func(_format_header("✅ Completed...", log_indent_level))

zrb/task/llm/tool_wrapper.py

@@ -133,7 +133,7 @@ async def _ask_for_approval(
         [
             f"\n🎰 >> {func_call_str}",
             _get_detail_func_param(args, kwargs),
-            f"✅ >> {_get_run_func_confirmation(func)}",
+            f"🎰  >> {_get_run_func_confirmation(func)}",
         ]
     )
     while True:
@@ -168,7 +168,7 @@ async def _ask_for_approval(
 def _get_run_func_confirmation(func: Callable) -> str:
     func_name = get_callable_name(func)
     return render_markdown(
-        f"Allow to run `{func_name}`? (`Yes` | `No, <reason>`)"
+        f"Allow to run `{func_name}`? (✅ `Yes` | ⛔ `No, <reason>`)"
     ).strip()
 
 

{zrb-1.15.6.dist-info → zrb-1.15.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: zrb
-Version: 1.15.6
+Version: 1.15.7
 Summary: Your Automation Powerhouse
 License: AGPL-3.0-or-later
 Keywords: Automation,Task Runner,Code Generator,Monorepo,Low Code

{zrb-1.15.6.dist-info → zrb-1.15.7.dist-info}/RECORD

@@ -355,9 +355,9 @@ zrb/task/llm/default_workflow/copywriting.md,sha256=xSO7GeDolwGxiuz6kXsK2GKGpwp8
 zrb/task/llm/default_workflow/researching.md,sha256=KD-aYHFHir6Ti-4FsBBtGwiI0seSVgleYbKJZi_POXA,2139
 zrb/task/llm/error.py,sha256=QR-nIohS6pBpC_16cWR-fw7Mevo1sNYAiXMBsh_CJDE,4157
 zrb/task/llm/history_summarization.py,sha256=_eWFP4cAn3tTfyt7YPjnPwm_PgMhMChPg9g2q2Gsf0c,8172
-zrb/task/llm/print_node.py,sha256=Sd8ovTO6KKI9hllZf5TxRXog7QGCne1A1E83jnrH_kg,6526
+zrb/task/llm/print_node.py,sha256=fNTL0LtoZBQQPHYCAUaOFP97nNNfobPXvY2RlxewQCE,7556
 zrb/task/llm/prompt.py,sha256=FGXWYHecWtrNNkPnjg-uhnkqp7fYt8V91-AjFM_5fpA,11550
-zrb/task/llm/tool_wrapper.py,sha256=2cbaSNvOMlIb3l5RXEFbH7cnox01TAr4b4Vtg0wzuYs,9609
+zrb/task/llm/tool_wrapper.py,sha256=SfDBDikT7cIqBThhLqrx4CFiNOmti8FAoBakRoEySqU,9619
 zrb/task/llm/typing.py,sha256=c8VAuPBw_4A3DxfYdydkgedaP-LU61W9_wj3m3CAX1E,58
 zrb/task/llm_task.py,sha256=0c9F_AjeMLxQX8VlAJHfZAkC52PhRaWbcKoqVOggcMg,14451
 zrb/task/make_task.py,sha256=PD3b_aYazthS8LHeJsLAhwKDEgdurQZpymJDKeN60u0,2265
@@ -408,7 +408,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.6.dist-info/METADATA,sha256=octlKaqi1aMWs5qYYb5d02at13BGvqvMdOEjYrKbp4E,9774
-zrb-1.15.6.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-zrb-1.15.6.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
-zrb-1.15.6.dist-info/RECORD,,
+zrb-1.15.7.dist-info/METADATA,sha256=N5AhEHwPxLzMKB9rL3NjqE4S-xyqWW4y3M-ATcz3CaQ,9774
+zrb-1.15.7.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+zrb-1.15.7.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
+zrb-1.15.7.dist-info/RECORD,,