zrb 1.15.6__tar.gz → 1.15.7__tar.gz

{zrb-1.15.6 → zrb-1.15.7}/PKG-INFO

@@ -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 → zrb-1.15.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "zrb"
-version = "1.15.6"
+version = "1.15.7"
 description = "Your Automation Powerhouse"
 authors = ["Go Frendi Gunawan <gofrendiasgard@gmail.com>"]
 license = "AGPL-3.0-or-later"

{zrb-1.15.6 → zrb-1.15.7}/src/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-1.15.6 → zrb-1.15.7}/src/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()