zrb 1.5.13__py3-none-any.whl → 1.5.15__py3-none-any.whl

zrb/llm_config.py

@@ -1,9 +1,12 @@
 import os
+from typing import TYPE_CHECKING, Any
 
-from pydantic_ai.models import Model
-from pydantic_ai.models.openai import OpenAIModel
-from pydantic_ai.providers import Provider
-from pydantic_ai.providers.openai import OpenAIProvider
+if TYPE_CHECKING:
+    from pydantic_ai.models import Model
+    from pydantic_ai.providers import Provider
+else:
+    Model = Any
+    Provider = Any
 
 from zrb.util.string.conversion import to_boolean
 
@@ -135,6 +138,8 @@ class LLMConfig:
             return self._default_provider
         if self._default_model_base_url is None and self._default_model_api_key is None:
             return "openai"
+        from pydantic_ai.providers.openai import OpenAIProvider
+
         return OpenAIProvider(
             base_url=self._default_model_base_url, api_key=self._default_model_api_key
         )
@@ -178,6 +183,8 @@ class LLMConfig:
         model_name = self._get_model_name()
         if model_name is None:
             return None
+        from pydantic_ai.models.openai import OpenAIModel
+
         return OpenAIModel(
             model_name=model_name,
             provider=self.get_default_model_provider(),

zrb/task/llm/agent.py

@@ -1,12 +1,22 @@
 from collections.abc import Callable
+from typing import TYPE_CHECKING, Any
 
-from openai import APIError
-from pydantic_ai import Agent, Tool
-from pydantic_ai.agent import AgentRun
-from pydantic_ai.mcp import MCPServer
-from pydantic_ai.messages import ModelMessagesTypeAdapter
-from pydantic_ai.models import Model
-from pydantic_ai.settings import ModelSettings
+if TYPE_CHECKING:
+    from openai import APIError
+    from pydantic_ai import Agent, Tool
+    from pydantic_ai.agent import AgentRun
+    from pydantic_ai.mcp import MCPServer
+    from pydantic_ai.models import Model
+    from pydantic_ai.settings import ModelSettings
+else:
+    APIError = Any
+    Agent = Any
+    Tool = Any
+    AgentRun = Any
+    MCPServer = Any
+    ModelMessagesTypeAdapter = Any
+    Model = Any
+    ModelSettings = Any
 
 from zrb.context.any_context import AnyContext
 from zrb.context.any_shared_context import AnySharedContext
@@ -32,6 +42,8 @@ def create_agent_instance(
 ) -> Agent:
     """Creates a new Agent instance with configured tools and servers."""
     # Get tools
+    from pydantic_ai import Agent, Tool
+
     tools_or_callables = list(tools_attr(ctx) if callable(tools_attr) else tools_attr)
     tools_or_callables.extend(additional_tools)
     tools = []
@@ -71,6 +83,8 @@ def get_agent(
     additional_mcp_servers: list[MCPServer],
 ) -> Agent:
     """Retrieves the configured Agent instance or creates one if necessary."""
+    from pydantic_ai import Agent
+
     if isinstance(agent_attr, Agent):
         return agent_attr
     if callable(agent_attr):
@@ -116,6 +130,8 @@ async def run_agent_iteration(
     Raises:
         Exception: If any error occurs during agent execution.
     """
+    from pydantic_ai.messages import ModelMessagesTypeAdapter
+
     async with agent.run_mcp_servers():
         async with agent.iter(
             user_prompt=user_prompt,

zrb/task/llm/config.py

@@ -1,7 +1,11 @@
-from typing import Callable
+from typing import TYPE_CHECKING, Any, Callable
 
-from pydantic_ai.models import Model
-from pydantic_ai.settings import ModelSettings
+if TYPE_CHECKING:
+    from pydantic_ai.models import Model
+    from pydantic_ai.settings import ModelSettings
+else:
+    Model = Any
+    ModelSettings = Any
 
 from zrb.attr.type import StrAttr, fstring
 from zrb.context.any_context import AnyContext
@@ -59,6 +63,8 @@ def get_model(
     render_model_api_key: bool,
 ) -> str | Model | None:
     """Gets the model instance or name, handling defaults and configuration."""
+    from pydantic_ai.models import Model
+
     model = get_attr(ctx, model_attr, None, auto_render=render_model)
     if model is None:
         return default_llm_config.get_default_model()

zrb/task/llm/context_enrichment.py

@@ -1,12 +1,9 @@
 import json
 import traceback
 from textwrap import dedent
-from typing import Any
+from typing import TYPE_CHECKING, Any
 
 from pydantic import BaseModel
-from pydantic_ai import Agent
-from pydantic_ai.models import Model
-from pydantic_ai.settings import ModelSettings
 
 from zrb.attr.type import BoolAttr, IntAttr
 from zrb.context.any_context import AnyContext
@@ -15,6 +12,13 @@ from zrb.task.llm.agent import run_agent_iteration
 from zrb.task.llm.typing import ListOfDict
 from zrb.util.attr import get_bool_attr, get_int_attr
 
+if TYPE_CHECKING:
+    from pydantic_ai.models import Model
+    from pydantic_ai.settings import ModelSettings
+else:
+    Model = Any
+    ModelSettings = Any
+
 
 class EnrichmentConfig(BaseModel):
     model_config = {"arbitrary_types_allowed": True}
@@ -35,6 +39,8 @@ async def enrich_context(
     history_list: ListOfDict,
 ) -> dict[str, Any]:
     """Runs an LLM call to extract key info and merge it into the context."""
+    from pydantic_ai import Agent
+
     ctx.log_info("Attempting to enrich conversation context...")
     # Prepare context and history for the enrichment prompt
     try:
@@ -160,7 +166,6 @@ async def maybe_enrich_context(
         context_enrichment_threshold_attr,
         render_context_enrichment_threshold,
     ):
-        # Use the enrich_context function now defined in this file
         return await enrich_context(
             ctx=ctx,
             config=EnrichmentConfig(

zrb/task/llm/error.py

@@ -1,9 +1,13 @@
 import json
-from typing import Optional
+from typing import TYPE_CHECKING, Any, Optional
 
-from openai import APIError
 from pydantic import BaseModel
 
+if TYPE_CHECKING:
+    from openai import APIError
+else:
+    APIError = Any
+
 
 # Define a structured error model for tool execution failures
 class ToolExecutionError(BaseModel):

zrb/task/llm/history_summarization.py

@@ -1,10 +1,7 @@
 import json
-from typing import Any
+from typing import TYPE_CHECKING, Any
 
 from pydantic import BaseModel
-from pydantic_ai import Agent
-from pydantic_ai.models import Model
-from pydantic_ai.settings import ModelSettings
 
 from zrb.attr.type import BoolAttr, IntAttr
 from zrb.context.any_context import AnyContext
@@ -13,6 +10,13 @@ from zrb.task.llm.agent import run_agent_iteration
 from zrb.task.llm.typing import ListOfDict
 from zrb.util.attr import get_bool_attr, get_int_attr
 
+if TYPE_CHECKING:
+    from pydantic_ai.models import Model
+    from pydantic_ai.settings import ModelSettings
+else:
+    Model = Any
+    ModelSettings = Any
+
 
 def get_history_part_len(history_list: ListOfDict) -> int:
     """Calculates the total number of 'parts' in a history list."""
@@ -90,6 +94,8 @@ async def summarize_history(
     history_list: ListOfDict,
 ) -> dict[str, Any]:
     """Runs an LLM call to summarize history and update the context."""
+    from pydantic_ai import Agent
+
     ctx.log_info("Attempting to summarize conversation history...")
 
     summarization_agent = Agent(

zrb/task/llm/print_node.py

@@ -1,22 +1,22 @@
 from collections.abc import Callable
 from typing import Any
 
-from pydantic_ai import Agent
-from pydantic_ai.messages import (
-    FinalResultEvent,
-    FunctionToolCallEvent,
-    FunctionToolResultEvent,
-    PartDeltaEvent,
-    PartStartEvent,
-    TextPartDelta,
-    ToolCallPartDelta,
-)
-
 from zrb.util.cli.style import stylize_faint
 
 
 async def print_node(print_func: Callable, agent_run: Any, node: Any):
     """Prints the details of an agent execution node using a provided print function."""
+    from pydantic_ai import Agent
+    from pydantic_ai.messages import (
+        FinalResultEvent,
+        FunctionToolCallEvent,
+        FunctionToolResultEvent,
+        PartDeltaEvent,
+        PartStartEvent,
+        TextPartDelta,
+        ToolCallPartDelta,
+    )
+
     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}"))

zrb/task/llm/tool_wrapper.py

@@ -3,16 +3,22 @@ import inspect
 import traceback
 import typing
 from collections.abc import Callable
-
-from pydantic_ai import RunContext, Tool
+from typing import TYPE_CHECKING, Any
 
 from zrb.context.any_context import AnyContext
 from zrb.task.llm.error import ToolExecutionError
 from zrb.util.run import run_async
 
+if TYPE_CHECKING:
+    from pydantic_ai import Tool
+else:
+    Tool = Any
+
 
 def wrap_tool(func: Callable, ctx: AnyContext) -> Tool:
     """Wraps a tool function to handle exceptions and context propagation."""
+    from pydantic_ai import RunContext, Tool
+
     original_sig = inspect.signature(func)
     # Use helper function for clarity
     needs_run_context_for_pydantic = _has_context_parameter(original_sig, RunContext)
@@ -123,6 +129,8 @@ def _adjust_signature(
     # (we inject it). So, the wrapper's signature should be the original signature,
     # minus any parameters annotated with RunContext or AnyContext.
 
+    from pydantic_ai import RunContext
+
     params_for_schema = [
         param
         for param in original_sig.parameters.values()

zrb/task/llm_task.py

@@ -1,11 +1,18 @@
 import json
 from collections.abc import Callable
-from typing import Any
+from typing import TYPE_CHECKING, Any
 
-from pydantic_ai import Agent, Tool
-from pydantic_ai.mcp import MCPServer
-from pydantic_ai.models import Model
-from pydantic_ai.settings import ModelSettings
+if TYPE_CHECKING:
+    from pydantic_ai import Agent, Tool
+    from pydantic_ai.mcp import MCPServer
+    from pydantic_ai.models import Model
+    from pydantic_ai.settings import ModelSettings
+else:
+    Agent = Any
+    Tool = Any
+    MCPServer = Any
+    Model = Any
+    ModelSettings = Any
 
 from zrb.attr.type import BoolAttr, IntAttr, StrAttr, fstring
 from zrb.context.any_context import AnyContext
@@ -39,7 +46,10 @@ from zrb.task.llm.prompt import (
 from zrb.util.cli.style import stylize_faint
 from zrb.xcom.xcom import Xcom
 
-ToolOrCallable = Tool | Callable
+if TYPE_CHECKING:
+    ToolOrCallable = Tool | Callable
+else:
+    ToolOrCallable = Any
 
 
 class LLMTask(BaseTask):
@@ -80,10 +90,11 @@ class LLMTask(BaseTask):
         context_enrichment_threshold: IntAttr | None = None,
         render_context_enrichment_threshold: bool = True,
         tools: (
-            list[ToolOrCallable] | Callable[[AnySharedContext], list[ToolOrCallable]]
+            list["ToolOrCallable"]
+            | Callable[[AnySharedContext], list["ToolOrCallable"]]
         ) = [],
         mcp_servers: (
-            list[MCPServer] | Callable[[AnySharedContext], list[MCPServer]]
+            list["MCPServer"] | Callable[[AnySharedContext], list["MCPServer"]]
         ) = [],
         conversation_history: (
             ConversationHistoryData
@@ -166,9 +177,9 @@ class LLMTask(BaseTask):
         self._context_enrichment_threshold = context_enrichment_threshold
         self._render_context_enrichment_threshold = render_context_enrichment_threshold
         self._tools = tools
-        self._additional_tools: list[ToolOrCallable] = []
+        self._additional_tools: list["ToolOrCallable"] = []
         self._mcp_servers = mcp_servers
-        self._additional_mcp_servers: list[MCPServer] = []
+        self._additional_mcp_servers: list["MCPServer"] = []
         self._conversation_history = conversation_history
         self._conversation_history_reader = conversation_history_reader
         self._conversation_history_writer = conversation_history_writer

{zrb-1.5.13.dist-info → zrb-1.5.15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: zrb
-Version: 1.5.13
+Version: 1.5.15
 Summary: Your Automation Powerhouse
 Home-page: https://github.com/state-alchemists/zrb
 License: AGPL-3.0-or-later

{zrb-1.5.13.dist-info → zrb-1.5.15.dist-info}/RECORD

@@ -241,7 +241,7 @@ zrb/input/option_input.py,sha256=TQB82ko5odgzkULEizBZi0e9TIHEbIgvdP0AR3RhA74,213
 zrb/input/password_input.py,sha256=szBojWxSP9QJecgsgA87OIYwQrY2AQ3USIKdDZY6snU,1465
 zrb/input/str_input.py,sha256=NevZHX9rf1g8eMatPyy-kUX3DglrVAQpzvVpKAzf7bA,81
 zrb/input/text_input.py,sha256=shvVbc2U8Is36h23M5lcW8IEwKc9FR-4uEPZZroj3rU,3377
-zrb/llm_config.py,sha256=2mcPAdulTUyRS0H-RG71jULQxWY2jfX4VML3LsrOY_o,9579
+zrb/llm_config.py,sha256=alAdk23Hnd4JnEQsLcquBjz4PWn-QGaVG5NCrzF5XrI,9702
 zrb/runner/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/runner/cli.py,sha256=0mT0oO_yEhc8N4nYCJNujhgLjVykZ0B-kAOFXyAvAqM,6672
 zrb/runner/common_util.py,sha256=0zhZn1Jdmr194_nsL5_L-Kn9-_NDpMTI2z6_LXUQJ-U,1369
@@ -311,18 +311,18 @@ zrb/task/base_trigger.py,sha256=jC722rDvodaBLeNaFghkTyv1u0QXrK6BLZUUqcmBJ7Q,4581
 zrb/task/cmd_task.py,sha256=xFAdOvLDK9Qaye40T_lG3K6AIKgbPAMHk_3GIAYeJAM,10750
 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=CvnM8y4dxterdM3dpa1G7iu099NJ-CT4NPiAOZp_ymI,4917
-zrb/task/llm/config.py,sha256=fXasnGb_DVTJIeY--HZ8bf1jd7iCUNttfUDJB5PvHRk,3071
+zrb/task/llm/agent.py,sha256=pZAbn0vQOXFkJqnHBo8rfFq3OZD3yePr39jzcRqn43A,5248
+zrb/task/llm/config.py,sha256=R6mkbm4d5ecN4KjjZaXbqNq9-8bXfdUGl_BML8hUWqY,3205
 zrb/task/llm/context.py,sha256=s313jXooOp8ht7j2sc3-d8xyW7oWs9bFv4KEyjicwys,3893
-zrb/task/llm/context_enrichment.py,sha256=NdeXMxGAFUR8UL064mjdiOCGQvK2o2zt5aSV-3rUhqc,5937
-zrb/task/llm/error.py,sha256=YOwnEdFMtqOlaiA83tDHpC6uh2_9r5NeS-inrlb5a8E,3622
+zrb/task/llm/context_enrichment.py,sha256=lLPCtSSdB0jyReB3rk5taNvFPpXMmjDZ_Ro_Fz5GX68,5963
+zrb/task/llm/error.py,sha256=27DQXSG8SH1-XuvXFdZQKzP39wZDWmd_YnSTz6DJKKI,3690
 zrb/task/llm/history.py,sha256=LnrJdXLyo2qz-bNCwLorhoqGmgSiPTUU0bzY63w67-E,9257
-zrb/task/llm/history_summarization.py,sha256=etfBCA9S_7gXeYwlT-HC-o5THgVAT4KR0M8ucJjdhaU,6200
-zrb/task/llm/print_node.py,sha256=Dkb0xFyEXpNRKFRCM4Md0lfg6K3nI0t8yH3Abh20PjE,4430
+zrb/task/llm/history_summarization.py,sha256=0cWChp4OE_OiaNDhHRWi4rwHKTHsqWLYy3pS5IAIHpQ,6293
+zrb/task/llm/print_node.py,sha256=dxwSOZ_DAFdrVSRIAEaxUjgbXWRtju1H87AtC9AJRIM,4470
 zrb/task/llm/prompt.py,sha256=0uN96mPiq5o28Bz_f7wWmOD6-DJYD-nc_N5RovYz3Js,4719
-zrb/task/llm/tool_wrapper.py,sha256=UUO5MnbFPlHuiVRPdke2ylKbaftjPDF06VNVuWcJ5zo,6326
+zrb/task/llm/tool_wrapper.py,sha256=Xygd4VCY3ykjVv63pqlTI16ZG41ySkp683_5VTnL-Zo,6481
 zrb/task/llm/typing.py,sha256=c8VAuPBw_4A3DxfYdydkgedaP-LU61W9_wj3m3CAX1E,58
-zrb/task/llm_task.py,sha256=hMo4wjKFvq8xv4eaxd2I_vXSBJLzL05HvaZ7mdo0pAg,14997
+zrb/task/llm_task.py,sha256=ABIhleg8n1d33FMJq2OqSigeXf8ROnEfUV4_9KmLoNQ,15220
 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
@@ -363,7 +363,7 @@ zrb/util/string/name.py,sha256=SXEfxJ1-tDOzHqmSV8kvepRVyMqs2XdV_vyoh_9XUu0,1584
 zrb/util/todo.py,sha256=VGISej2KQZERpornK-8X7bysp4JydMrMUTnG8B0-liI,20708
 zrb/xcom/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/xcom/xcom.py,sha256=o79rxR9wphnShrcIushA0Qt71d_p3ZTxjNf7x9hJB78,1571
-zrb-1.5.13.dist-info/METADATA,sha256=TQfp35Bm_7FlsY54TUe39XdiGHEmnWzkVeIoLyDZnb0,8347
-zrb-1.5.13.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-zrb-1.5.13.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
-zrb-1.5.13.dist-info/RECORD,,
+zrb-1.5.15.dist-info/METADATA,sha256=G1B30COaisd9TjKv4-r0FpOwQnU-N5UBk5UEexfN2cQ,8347
+zrb-1.5.15.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+zrb-1.5.15.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
+zrb-1.5.15.dist-info/RECORD,,