zrb 1.9.1__py3-none-any.whl → 1.9.2__py3-none-any.whl

zrb/builtin/llm/chat_session.py

@@ -29,7 +29,7 @@ async def read_user_prompt(ctx: AnyContext) -> str:
 
     _show_info(ctx)
     final_result = ""
-    ctx.print(stylize_faint("💬"), plain=True)
+    ctx.print("💬 >>", plain=True)
     ctx.print(ctx.input.message, plain=True)
     ctx.print("", plain=True)
     result = await _trigger_ask_and_wait_for_result(
@@ -49,12 +49,12 @@ async def read_user_prompt(ctx: AnyContext) -> str:
     while True:
         await asyncio.sleep(0.01)
         if not multiline_mode:
-            ctx.print(stylize_faint("💬"), plain=True)
+            ctx.print("💬 >>", plain=True)
         user_input = await user_input_session.prompt_async()
         if not multiline_mode:
             ctx.print("", plain=True)
         # Handle special input
-        if user_input.strip().lower() in ("/bye", "/quit"):
+        if user_input.strip().lower() in ("/bye", "/quit", "/q", "/exit"):
             user_prompt = "\n".join(user_inputs)
             user_inputs = []
             result = await _trigger_ask_and_wait_for_result(ctx, user_prompt)
@@ -108,12 +108,27 @@ async def _trigger_ask_and_wait_for_result(
         return None
     await _trigger_ask(ctx, user_prompt, previous_session_name, start_new)
     result = await _wait_ask_result(ctx)
-    ctx.print(stylize_faint("\n🤖"), plain=True)
-    ctx.print(result, plain=True)
+    md_result = _render_markdown(result) if result is not None else ""
+    ctx.print("\n🤖 >>", plain=True)
+    ctx.print(md_result, plain=True)
     ctx.print("", plain=True)
     return result
 
 
+def _render_markdown(markdown_text: str) -> str:
+    """
+    Renders Markdown to a string, ensuring link URLs are visible.
+    """
+    from rich.console import Console
+    from rich.markdown import Markdown
+
+    console = Console()
+    markdown = Markdown(markdown_text, hyperlinks=False)
+    with console.capture() as capture:
+        console.print(markdown)
+    return capture.get()
+
+
 def get_llm_ask_input_mapping(callback_ctx: AnyContext):
     """
     Generates the input mapping for the LLM ask task from the callback context.

zrb/task/llm/print_node.py

@@ -18,23 +18,18 @@ async def print_node(print_func: Callable, agent_run: Any, node: Any):
     )
 
     if Agent.is_user_prompt_node(node):
-        # A user prompt node => The user has provided input
-        print_func(stylize_faint(f"    >> UserPromptNode: {node.user_prompt}"))
+        print_func(stylize_faint("🔠 Receiving input..."))
     elif Agent.is_model_request_node(node):
         # A model request node => We can stream tokens from the model's request
-        print_func(
-            stylize_faint("    >> ModelRequestNode: streaming partial request tokens")
-        )
+        print_func(stylize_faint("🧠 Processing..."))
         async with node.stream(agent_run.ctx) as request_stream:
             is_streaming = False
             async for event in request_stream:
-                if isinstance(event, PartStartEvent):
+                if isinstance(event, PartStartEvent) and event.part:
                     if is_streaming:
                         print_func("")
                     print_func(
-                        stylize_faint(
-                            f"    [Request] Starting part {event.index}: {event.part!r}"
-                        ),
+                        stylize_faint(f"    Starting part {event.index}: {event.part}"),
                     )
                     is_streaming = False
                 elif isinstance(event, PartDeltaEvent):
@@ -49,22 +44,18 @@ async def print_node(print_func: Callable, agent_run: Any, node: Any):
                             end="",
                         )
                     is_streaming = True
-                elif isinstance(event, FinalResultEvent):
+                elif isinstance(event, FinalResultEvent) and event.tool_name:
                     if is_streaming:
                         print_func("")
                     print_func(
-                        stylize_faint(f"    [Result] tool_name={event.tool_name}"),
+                        stylize_faint(f"    Result: tool_name={event.tool_name}"),
                     )
                     is_streaming = False
             if is_streaming:
                 print_func("")
     elif Agent.is_call_tools_node(node):
         # A handle-response node => The model returned some data, potentially calls a tool
-        print_func(
-            stylize_faint(
-                "    >> CallToolsNode: streaming partial response & tool usage"
-            )
-        )
+        print_func(stylize_faint("🧰 Calling Tool..."))
         async with node.stream(agent_run.ctx) as handle_stream:
             async for event in handle_stream:
                 if isinstance(event, FunctionToolCallEvent):
@@ -84,16 +75,16 @@ async def print_node(print_func: Callable, agent_run: Any, node: Any):
                         del event.part.args["_dummy"]
                     print_func(
                         stylize_faint(
-                            f"    [Tools] The LLM calls tool={event.part.tool_name!r} with args={event.part.args} (tool_call_id={event.part.tool_call_id!r})"  # noqa
+                            f"    {event.part.tool_call_id} | "
+                            f"Call {event.part.tool_name} {event.part.args}"
                         )
                     )
                 elif isinstance(event, FunctionToolResultEvent):
                     print_func(
                         stylize_faint(
-                            f"    [Tools] Tool call {event.tool_call_id!r} returned => {event.result.content}"  # noqa
+                            f"    {event.tool_call_id} | {event.result.content}"
                         )
                     )
     elif Agent.is_end_node(node):
         # Once an End node is reached, the agent run is complete
-        print_func(stylize_faint("    [End of Response]"))
-        # print_func(stylize_faint(f"{agent_run.result.data}"))
+        print_func(stylize_faint("✅ Completed..."))

zrb/task/llm_task.py

@@ -356,7 +356,7 @@ class LLMTask(BaseTask):
                     ctx.xcom[xcom_usage_key] = Xcom([])
                 usage = agent_run.result.usage()
                 ctx.xcom[xcom_usage_key].push(usage)
-                ctx.print(stylize_faint(f"    Token: {usage}"), plain=True)
+                ctx.print(stylize_faint(f"💸 Token: {usage}"), plain=True)
                 return agent_run.result.output
             else:
                 ctx.log_warning("Agent run did not produce a result.")

{zrb-1.9.1.dist-info → zrb-1.9.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: zrb
-Version: 1.9.1
+Version: 1.9.2
 Summary: Your Automation Powerhouse
 Home-page: https://github.com/state-alchemists/zrb
 License: AGPL-3.0-or-later
@@ -32,6 +32,7 @@ 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.4.0,<4.0.0)
 Requires-Dist: requests (>=2.32.4,<3.0.0)
+Requires-Dist: rich (>=14.0.0,<15.0.0)
 Requires-Dist: tiktoken (>=0.8.0,<0.9.0)
 Requires-Dist: ulid-py (>=1.1.0,<2.0.0)
 Project-URL: Documentation, https://github.com/state-alchemists/zrb

{zrb-1.9.1.dist-info → zrb-1.9.2.dist-info}/RECORD

@@ -9,7 +9,7 @@ zrb/builtin/git_subtree.py,sha256=7BKwOkVTWDrR0DXXQ4iJyHqeR6sV5VYRt8y_rEB0EHg,35
 zrb/builtin/group.py,sha256=t008xLM4_fgbjfZrPoi_fQAnSHIo6MOiQSCHBO4GDYU,2379
 zrb/builtin/http.py,sha256=sLqEczuSxGYXWzyJR6frGOHkPTviu4BeyroUr3-ZuAI,4322
 zrb/builtin/jwt.py,sha256=3M5uaQhJZbKQLjTUft1OwPz_JxtmK-xtkjxWjciOQho,2859
-zrb/builtin/llm/chat_session.py,sha256=iDNHbNX268NWl0uQmjecD3Bn686-mByqqXuMc6IKjOE,7146
+zrb/builtin/llm/chat_session.py,sha256=S7IIY1ZGiNI0jfg1LhFZBKM9edHutiQws_WgwNXzZZ4,7586
 zrb/builtin/llm/history.py,sha256=cnkOyO43uiMQ9cEvmqk-pPoCk1zCAH_fwAqSgBtsjzY,3079
 zrb/builtin/llm/input.py,sha256=Nw-26uTWp2QhUgKJcP_IMHmtk-b542CCSQ_vCOjhvhM,877
 zrb/builtin/llm/llm_ask.py,sha256=A7MmDybKSdlO0T-Wted9t1tEtw6ULjV5aG4SAwpkt-w,4532
@@ -344,11 +344,11 @@ zrb/task/llm/context_enrichment.py,sha256=MBs5lkI1GFZwtW_P7fJfIc6CWcAS0UWQeBO6gg
 zrb/task/llm/error.py,sha256=s5PSK3ibGupMzOM0P81nstHWrMr3205H0FSDwfhWUDA,3662
 zrb/task/llm/history.py,sha256=cpaNqoEzsNAgzZGPPdohA8U5nTrVWmZ3P1Y5RTtXDgc,7986
 zrb/task/llm/history_summarization.py,sha256=TejPLktJmlxvhM-UmC29m_g6U9bC5e4fxRSEwt22GN0,6032
-zrb/task/llm/print_node.py,sha256=VB_ZXD3iQPZcaNWvIMNRxOhmXIJp9__5utRrSbiIRXo,4457
+zrb/task/llm/print_node.py,sha256=zocTKi9gZDxl2I6KNu095TmMc13Yip6SNuWYnswS680,4060
 zrb/task/llm/prompt.py,sha256=obSsLHF4AHlwMIEQ8XTJmtEnkLlD5XTEkk0RJUmIz4Y,3410
 zrb/task/llm/tool_wrapper.py,sha256=8_bL8m_WpRf-pVKSrvQIVqT-m2sUA87a1RBQG13lhp4,6457
 zrb/task/llm/typing.py,sha256=c8VAuPBw_4A3DxfYdydkgedaP-LU61W9_wj3m3CAX1E,58
-zrb/task/llm_task.py,sha256=qNXS078y2s6iF50OoVH3-7UWmV5phgfY2_NiZ9KxGSU,21055
+zrb/task/llm_task.py,sha256=EYU3TYhLrMu3nI2h-hid1vbfqpemYg54uht_Q8H8JBo,21056
 zrb/task/make_task.py,sha256=PD3b_aYazthS8LHeJsLAhwKDEgdurQZpymJDKeN60u0,2265
 zrb/task/rsync_task.py,sha256=GSL9144bmp6F0EckT6m-2a1xG25AzrrWYzH4k3SVUKM,6370
 zrb/task/scaffolder.py,sha256=rME18w1HJUHXgi9eTYXx_T2G4JdqDYzBoNOkdOOo5-o,6806
@@ -391,7 +391,7 @@ zrb/util/todo.py,sha256=r9_KYF2-hLKMNjsp6AFK9zivykMrywd-kJ4bCwfdafI,19323
 zrb/util/todo_model.py,sha256=0SJ8aLYfJAscDOk5JsH7pXP3h1rAG91VMCS20-c2Y6A,1576
 zrb/xcom/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/xcom/xcom.py,sha256=o79rxR9wphnShrcIushA0Qt71d_p3ZTxjNf7x9hJB78,1571
-zrb-1.9.1.dist-info/METADATA,sha256=ohOmILh0PYQYiYzfuNy3dfuqNT7UorikVrHQO5jIE8I,9321
-zrb-1.9.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-zrb-1.9.1.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
-zrb-1.9.1.dist-info/RECORD,,
+zrb-1.9.2.dist-info/METADATA,sha256=Co2iVzYe034DMJ5B1L3Gt6ydfRbsBVxOieHu9akVPnM,9360
+zrb-1.9.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+zrb-1.9.2.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
+zrb-1.9.2.dist-info/RECORD,,