zrb 1.15.22__py3-none-any.whl → 1.15.24__py3-none-any.whl

zrb/config/default_prompt/summarization_prompt.md

@@ -1,4 +1,4 @@
-You are a memory management AI. Your only task is to process the provided conversation history and call the `update_conversation_memory` tool **once**.
+You are a memory management AI. Your only task is to process the provided conversation history and call the `final_result` tool **once**.
 
 Follow these instructions carefully:
 
@@ -11,6 +11,6 @@ Follow these instructions carefully:
     *   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.
+4.  **Update Memory:** Call the `final_result` tool with all the information you consolidated.
 
-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.
+After you have called the tool, your task is complete.

zrb/task/llm/agent.py

@@ -16,6 +16,7 @@ if TYPE_CHECKING:
     from pydantic_ai.agent import AgentRun
     from pydantic_ai.messages import UserContent
     from pydantic_ai.models import Model
+    from pydantic_ai.output import OutputDataT, OutputSpec
     from pydantic_ai.settings import ModelSettings
     from pydantic_ai.toolsets import AbstractToolset
 
@@ -25,13 +26,14 @@ if TYPE_CHECKING:
 def create_agent_instance(
     ctx: AnyContext,
     model: "str | Model",
+    output_type: "OutputSpec[OutputDataT]" = str,
     system_prompt: str = "",
     model_settings: "ModelSettings | None" = None,
     tools: "list[ToolOrCallable]" = [],
     toolsets: list["AbstractToolset[Agent]"] = [],
     retries: int = 3,
     yolo_mode: bool | list[str] | None = None,
-) -> "Agent":
+) -> "Agent[None, Any]":
     """Creates a new Agent instance with configured tools and servers."""
     from pydantic_ai import Agent, Tool
     from pydantic_ai.tools import GenerateToolJsonSchema
@@ -63,8 +65,9 @@ def create_agent_instance(
             # Turn function into tool
             tool_list.append(wrap_tool(tool_or_callable, ctx, yolo_mode))
     # Return Agent
-    return Agent(
+    return Agent[None, Any](
         model=model,
+        output_type=output_type,
         system_prompt=system_prompt,
         tools=tool_list,
         toolsets=toolsets,
@@ -77,14 +80,15 @@ def get_agent(
     ctx: AnyContext,
     agent_attr: "Agent | Callable[[AnySharedContext], Agent] | None",
     model: "str | Model",
-    system_prompt: str,
-    model_settings: "ModelSettings | None",
+    output_type: "OutputSpec[OutputDataT]" = str,
+    system_prompt: str = "",
+    model_settings: "ModelSettings | None" = None,
     tools_attr: (
         "list[ToolOrCallable] | Callable[[AnySharedContext], list[ToolOrCallable]]"
-    ),
-    additional_tools: "list[ToolOrCallable]",
-    toolsets_attr: "list[AbstractToolset[Agent]] | Callable[[AnySharedContext], list[AbstractToolset[Agent]]]",  # noqa
-    additional_toolsets: "list[AbstractToolset[Agent]]",
+    ) = [],
+    additional_tools: "list[ToolOrCallable]" = [],
+    toolsets_attr: "list[AbstractToolset[Agent]] | Callable[[AnySharedContext], list[AbstractToolset[Agent]]]" = [],  # noqa
+    additional_toolsets: "list[AbstractToolset[Agent]]" = [],
     retries: int = 3,
     yolo_mode: bool | list[str] | None = None,
 ) -> "Agent":
@@ -113,6 +117,7 @@ def get_agent(
     return create_agent_instance(
         ctx=ctx,
         model=model,
+        output_type=output_type,
         system_prompt=system_prompt,
         tools=tools,
         toolsets=tool_sets,
@@ -124,7 +129,7 @@ def get_agent(
 
 async def run_agent_iteration(
     ctx: AnyContext,
-    agent: "Agent",
+    agent: "Agent[None, Any]",
     user_prompt: str,
     attachments: "list[UserContent] | None" = None,
     history_list: ListOfDict | None = None,
@@ -199,7 +204,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 +221,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/task/llm/history_summarization.py

@@ -143,12 +143,13 @@ async def summarize_history(
             ),
         ]
     )
-    summarization_agent = Agent(
+    summarize = create_history_summarization_tool(conversation_history)
+    summarization_agent = Agent[None, str](
         model=model,
+        output_type=summarize,
         system_prompt=system_prompt,
         model_settings=settings,
         retries=retries,
-        tools=[create_history_summarization_tool(conversation_history)],
     )
     try:
         ctx.print(stylize_faint("  📝 Rollup Conversation"), plain=True)

zrb/task/llm/history_summarization_tool.py

@@ -7,14 +7,14 @@ from zrb.task.llm.conversation_history_model import ConversationHistory
 
 def create_history_summarization_tool(
     conversation_history: ConversationHistory,
-) -> Callable:
+) -> Callable[[str, str, str | None, str | None, str | None], str]:
     def update_conversation_memory(
         past_conversation_summary: str,
         past_conversation_transcript: str,
         long_term_note: str | None = None,
         contextual_note: str | None = None,
         context_path: str | None = None,
-    ):
+    ) -> str:
         """
         Update the conversation memory including summary, transcript, and notes.
         - past_conversation_summary: A concise narrative that integrates the
@@ -33,6 +33,6 @@ def create_history_summarization_tool(
             if context_path is None:
                 context_path = conversation_history.project_path
             llm_context_config.write_context(contextual_note, context_path=context_path)
-        return json.dumps({"success": True})
+        return "Conversation memory updated"
 
     return update_conversation_memory

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/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()
 
 

{zrb-1.15.22.dist-info → zrb-1.15.24.dist-info}/METADATA

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

@@ -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=vbxzJGeH7uCdgnfbVt3GYOjY-BygmeRfJ6tnzXmFdoY,1698
+zrb/config/default_prompt/summarization_prompt.md,sha256=qGSZJHQ0KVaSoeDVeAmvxaOHUGeLPaq813jepGqnkbk,1564
 zrb/config/default_prompt/system_prompt.md,sha256=gEb6N-cFg6VvOV-7ZffNwVt39DavAGesMqn9u0epbRc,2282
 zrb/config/llm_config.py,sha256=9I_msAibvTY-ADG6lXCBXQ0EshaA-GQzJym9EZKsetw,8901
 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=uI0X_injDrxq-IhYuZsAMdoq9AXy-gnipozGuZ72QIs,8990
+zrb/task/llm/agent.py,sha256=gi7JlUpNBHQ0WhlZcTgTaJQSNlkVYb999NUAWYttqt8,9194
 zrb/task/llm/config.py,sha256=zOPf4NpdWPuBc_R8d-kljYcOKfUAypDxiSjRDrxV66M,4059
 zrb/task/llm/conversation_history.py,sha256=oMdKUV2__mBZ4znnA-prl-gfyoleKC8Nj5KNpmLQJ4o,6764
 zrb/task/llm/conversation_history_model.py,sha256=kk-7niTl29Rm2EUIhTHzPXgZ5tp4IThMnIB3dS-1OdU,3062
@@ -354,11 +354,11 @@ zrb/task/llm/default_workflow/coding.md,sha256=2uythvPsnBpYfIhiIH1cCinQXX0i0yUqs
 zrb/task/llm/default_workflow/copywriting.md,sha256=xSO7GeDolwGxiuz6kXsK2GKGpwp8UgtG0yRqTmill_s,1999
 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=UIT8bpdT3hy1xn559waDLFWZlNtIqdIpIvRGcZEpHm0,8057
-zrb/task/llm/history_summarization_tool.py,sha256=Wazi4WMr3k1WJ1v7QgjAPbuY1JdBpHUsTWGt3DSTsLc,1706
-zrb/task/llm/print_node.py,sha256=TG8i3MrAkIj3cLkU9_fSX-u49jlTdU8t9FpHGI_VtoM,8077
+zrb/task/llm/history_summarization.py,sha256=blTKfaSpgaqvORWzGL3BKrRWAsfNdZB03oJTEOS-j5s,8098
+zrb/task/llm/history_summarization_tool.py,sha256=RmYxWAxmBzhgwz9Cyu5aZ9so8r3zeAVhekEhqfhuTrY,1766
+zrb/task/llm/print_node.py,sha256=Nnf4F6eDJR4PFcOqQ1jLWBTFnzNGl1Stux2DZ3SMhsY,8062
 zrb/task/llm/prompt.py,sha256=FGXWYHecWtrNNkPnjg-uhnkqp7fYt8V91-AjFM_5fpA,11550
-zrb/task/llm/tool_wrapper.py,sha256=v3y4FO14xStpq9K0lA3GIVv6-3dbq85I7xZqdtG-j9U,10243
+zrb/task/llm/tool_wrapper.py,sha256=ifOP-smxHcDNpZvZMZm0XkKH49BicbyGTtpHNdcIA6Q,10241
 zrb/task/llm/typing.py,sha256=c8VAuPBw_4A3DxfYdydkgedaP-LU61W9_wj3m3CAX1E,58
 zrb/task/llm_task.py,sha256=OxJ9QpqjEyeOI1_zqzNZHtQlRHi0ANOvL9FYaWLzO3Y,14913
 zrb/task/make_task.py,sha256=PD3b_aYazthS8LHeJsLAhwKDEgdurQZpymJDKeN60u0,2265
@@ -410,7 +410,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.22.dist-info/METADATA,sha256=OHahYYqF0_2Z_Ht40qggq5Z538ITZNVtaC-UagYam6o,9892
-zrb-1.15.22.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-zrb-1.15.22.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
-zrb-1.15.22.dist-info/RECORD,,
+zrb-1.15.24.dist-info/METADATA,sha256=1AenVGPfxpku7IJv9cOpRvSLGpct09bH79HTU7hAkrI,9892
+zrb-1.15.24.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+zrb-1.15.24.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
+zrb-1.15.24.dist-info/RECORD,,