zrb 1.15.22__tar.gz → 1.15.23__tar.gz

{zrb-1.15.22 → zrb-1.15.23}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: zrb
-Version: 1.15.22
+Version: 1.15.23
 Summary: Your Automation Powerhouse
 License: AGPL-3.0-or-later
 Keywords: Automation,Task Runner,Code Generator,Monorepo,Low Code
@@ -29,7 +29,7 @@ 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: prompt-toolkit (>=3)
 Requires-Dist: psutil (>=7.0.0,<8.0.0)
-Requires-Dist: pydantic-ai-slim[anthropic,bedrock,cohere,google,groq,huggingface,mistral,openai,vertexai] (>=0.8.1,<0.9.0)
+Requires-Dist: pydantic-ai-slim[anthropic,bedrock,cohere,google,groq,huggingface,mistral,openai,vertexai] (>=1.0.1,<1.1.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.22 → zrb-1.15.23}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "zrb"
-version = "1.15.22"
+version = "1.15.23"
 description = "Your Automation Powerhouse"
 authors = ["Go Frendi Gunawan <gofrendiasgard@gmail.com>"]
 license = "AGPL-3.0-or-later"
@@ -50,7 +50,7 @@ ulid-py = "^1.1.0"
 psutil = "^7.0.0"
 
 # NOTE: Look at https://github.com/pydantic/pydantic-ai/blob/main/pydantic_ai_slim/pyproject.toml
-pydantic-ai-slim = {extras = ["openai", "cohere", "vertexai", "google", "anthropic", "groq", "mistral", "bedrock", "huggingface"], version = "~0.8.1"} 
+pydantic-ai-slim = {extras = ["openai", "cohere", "vertexai", "google", "anthropic", "groq", "mistral", "bedrock", "huggingface"], version = "~1.0.1"} 
 
 # NOTE: Make sure the following dependencies are not conflicting with pydantic-ai-slim
 openai = ">=1.99.9"

{zrb-1.15.22 → zrb-1.15.23}/src/zrb/task/llm/agent.py

@@ -199,7 +199,6 @@ async def _run_single_agent_iteration(
         ) as agent_run:
             async for node in agent_run:
                 # Each node represents a step in the agent's execution
-                # Reference: https://ai.pydantic.dev/agents/#streaming
                 try:
                     await print_node(
                         _get_plain_printer(ctx), agent_run, node, log_indent_level
@@ -217,7 +216,7 @@ async def _run_single_agent_iteration(
 
 
 def _create_print_throttle_notif(ctx: AnyContext) -> Callable[[], None]:
-    def _print_throttle_notif(ctx: AnyContext):
+    def _print_throttle_notif():
         ctx.print(stylize_faint("  ⌛>> Request Throttled"), plain=True)
 
     return _print_throttle_notif

{zrb-1.15.22 → zrb-1.15.23}/src/zrb/task/llm/print_node.py

@@ -28,7 +28,7 @@ async def print_node(
     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
+        # Reference: https://ai.pydantic.dev/agents/#streaming-all-events-and-output
         try:
             async with node.stream(agent_run.ctx) as request_stream:
                 is_streaming = False
@@ -40,9 +40,7 @@ async def print_node(
                         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
-                        ):
+                        if isinstance(event.delta, TextPartDelta):
                             content_delta = event.delta.content_delta
                             print_func(
                                 _format_stream_content(content_delta, log_indent_level),
@@ -78,7 +76,7 @@ async def print_node(
             print_func(
                 _format_content(
                     (
-                        f"⚠️ Unexpected Model Behavior: {e}. "
+                        f"🟡 Unexpected Model Behavior: {e}. "
                         f"Cause: {e.__cause__}. Node.Id: {meta}"
                     ),
                     log_indent_level,
@@ -112,7 +110,7 @@ async def print_node(
             print_func(
                 _format_content(
                     (
-                        f"⚠️ Unexpected Model Behavior: {e}. "
+                        f"🟡 Unexpected Model Behavior: {e}. "
                         f"Cause: {e.__cause__}. Node.Id: {meta}"
                     ),
                     log_indent_level,
@@ -123,7 +121,7 @@ async def print_node(
         print_func(_format_header("✅ Completed...", log_indent_level))
 
 
-def _format_header(text: str, log_indent_level: int = 0) -> str:
+def _format_header(text: str | None, log_indent_level: int = 0) -> str:
     return _format(
         text,
         base_indent=2,
@@ -133,7 +131,7 @@ def _format_header(text: str, log_indent_level: int = 0) -> str:
     )
 
 
-def _format_content(text: str, log_indent_level: int = 0) -> str:
+def _format_content(text: str | None, log_indent_level: int = 0) -> str:
     return _format(
         text,
         base_indent=2,
@@ -143,7 +141,7 @@ def _format_content(text: str, log_indent_level: int = 0) -> str:
     )
 
 
-def _format_stream_content(text: str, log_indent_level: int = 0) -> str:
+def _format_stream_content(text: str | None, log_indent_level: int = 0) -> str:
     return _format(
         text,
         base_indent=2,
@@ -154,13 +152,15 @@ def _format_stream_content(text: str, log_indent_level: int = 0) -> str:
 
 
 def _format(
-    text: str,
+    text: str | None,
     base_indent: int = 0,
     first_indent: int = 0,
     indent: int = 0,
     log_indent_level: int = 0,
     is_stream: bool = False,
 ) -> str:
+    if text is None:
+        text = ""
     line_prefix = (base_indent * (log_indent_level + 1) + indent) * " "
     processed_text = text.replace("\n", f"\n{line_prefix}")
     if is_stream:

{zrb-1.15.22 → zrb-1.15.23}/src/zrb/task/llm/tool_wrapper.py

@@ -220,7 +220,7 @@ def _get_user_approval_and_reason(
 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>` | ✏️ `Edit <param> <value>`)"
+        f"Allow to run `{func_name}`? (✅ `Yes` | ⛔ `No, <reason>` | 📝 `Edit <param> <value>`)"
     ).strip()