zrb 1.0.0b4__py3-none-any.whl → 1.0.0b6__py3-none-any.whl

zrb/builtin/llm/llm_chat.py

@@ -92,8 +92,8 @@ llm_chat: LLMTask = llm_group.add_task(
             ),
             BoolInput(
                 "start-new",
-                description="Start new conversation session",
-                prompt="Forget everything and start new conversation session",
+                description="Start new conversation (LLM will forget everything)",
+                prompt="Start new conversation (LLM will forget everything)",
                 default_str="false",
                 allow_positional_parsing=False,
             ),

zrb/builtin/llm/previous-session.js

@@ -1,7 +1,6 @@
-let hasUpdateCurrentPascalInputName = false;
-document.getElementById("submit-task-form").addEventListener("change", async function(event) {
+async function updatePreviousSession(event) {
     const currentInput = event.target;
-    if (hasUpdateCurrentPascalInputName || currentInput.name === "CURRENT_INPUT_NAME") {
+    if (currentInput.name === "CURRENT_INPUT_NAME") {
         return
     }
     const previousSessionInput = submitTaskForm.querySelector('[name="CURRENT_INPUT_NAME"]');
@@ -9,5 +8,14 @@ document.getElementById("submit-task-form").addEventListener("change", async fun
         const currentSessionName = cfg.SESSION_NAME
         previousSessionInput.value = currentSessionName;
     }
-    hasUpdateCurrentPascalInputName = true;
+}
+
+document.getElementById("submit-task-form").querySelectorAll("input[name], textarea[name]").forEach((element) => {
+    element.addEventListener("input", updatePreviousSession);
+    element.addEventListener("keyup", updatePreviousSession);
 });
+
+document.getElementById("submit-task-form").querySelectorAll("select[name]").forEach((element) => {
+    element.addEventListener("change", updatePreviousSession);
+});
+

zrb/builtin/llm/tool/rag.py

@@ -3,7 +3,6 @@ import json
 import os
 import sys
 
-import litellm
 import ulid
 
 from zrb.config import (
@@ -30,7 +29,9 @@ def create_rag_from_directory(
     async def retrieve(query: str) -> str:
         from chromadb import PersistentClient
         from chromadb.config import Settings
+        from fastembed import TextEmbedding
 
+        embedding_model = TextEmbedding(model_name=model)
         client = PersistentClient(
             path=vector_db_path, settings=Settings(allow_reset=True)
         )
@@ -75,10 +76,8 @@ def create_rag_from_directory(
                                 ),
                                 file=sys.stderr,
                             )
-                            response = await litellm.aembedding(
-                                model=model, input=[chunk]
-                            )
-                            vector = response["data"][0]["embedding"]
+                            embedding_result = list(embedding_model.embed([chunk]))
+                            vector = embedding_result[0]
                             collection.upsert(
                                 ids=[chunk_id],
                                 embeddings=[vector],
@@ -102,9 +101,8 @@ def create_rag_from_directory(
             )
 
         print(stylize_faint("Vectorizing query"), file=sys.stderr)
-        query_response = await litellm.aembedding(model=model, input=[query])
-        query_vector = query_response["data"][0]["embedding"]
-
+        embedding_result = list(embedding_model.embed([query]))
+        query_vector = embedding_result[0]
         print(stylize_faint("Searching documents"), file=sys.stderr)
         results = collection.query(
             query_embeddings=query_vector,

zrb/config.py

@@ -78,22 +78,14 @@ WEB_AUTH_REFRESH_TOKEN_EXPIRE_MINUTES = int(
 )
 LLM_MODEL = os.getenv("ZRB_LLM_MODEL", "ollama_chat/llama3.1")
 
-_DEFAULT_PROMPT = """
-You are a helpful assistant and you have access to several tools.
-Your goal is to provide a final answer by executing a series of planning, actions, reasoning, and evaluations.
-
-Breakdown user request into several actionable tasks. For example, when user ask about current weather on current location, you should get the current location first.
-
-DO NOT TRY TO SIMULATE TOOL OUTPUT.
-
-ERROR HANDLING
-1. If you receive an error, read the error message carefully and identify the specific issue.
-2. Adjust your response accordingly and perform the review process again before resubmitting.
-
-REMINDER:
-- ALWAYS double-check your response format and function arguments before submitting.
-- DON'T make up answers.
-""".strip()
+_DEFAULT_PROMPT = (
+    "You are a helpful AI assistant capable of using various tools to answer user queries. When solving a problem:\n"
+    "1. Carefully analyze the user's request and identify what information is needed to provide a complete answer.\n"
+    "2. Determine which available tools can help you gather the necessary information.\n"
+    "3. Call tools strategically and in a logical sequence to collect required data.\n"
+    "4. If a tool provides incomplete information, intelligently decide which additional tool or approach to use.\n"
+    "5. Always aim to provide the most accurate and helpful response possible."
+)
 LLM_SYSTEM_PROMPT = os.getenv("ZRB_LLM_SYSTEM_PROMPT", _DEFAULT_PROMPT)
 LLM_HISTORY_DIR = os.getenv(
     "ZRB_LLM_HISTORY_DIR", os.path.expanduser(os.path.join("~", ".zrb-llm-history"))
@@ -103,7 +95,10 @@ LLM_HISTORY_FILE = os.getenv(
 )
 LLM_ALLOW_ACCESS_SHELL = to_boolean(os.getenv("ZRB_LLM_ACCESS_FILE", "1"))
 LLM_ALLOW_ACCESS_INTERNET = to_boolean(os.getenv("ZRB_LLM_ACCESS_INTERNET", "1"))
-RAG_EMBEDDING_MODEL = os.getenv("ZRB_RAG_EMBEDDING_MODEL", "ollama/nomic-embed-text")
+# noqa See: https://qdrant.github.io/fastembed/examples/Supported_Models/#supported-text-embedding-models
+RAG_EMBEDDING_MODEL = os.getenv(
+    "ZRB_RAG_EMBEDDING_MODEL", "nomic-ai/nomic-embed-text-v1.5-Q"
+)
 RAG_CHUNK_SIZE = int(os.getenv("ZRB_RAG_CHUNK_SIZE", "1024"))
 RAG_OVERLAP = int(os.getenv("ZRB_RAG_OVERLAP", "128"))
 RAG_MAX_RESULT_COUNT = int(os.getenv("ZRB_RAG_MAX_RESULT_COUNT", "5"))

zrb/runner/web_route/node_page/task/view.html

@@ -63,7 +63,7 @@
             <article>
                 <form id="submit-task-form" onsubmit="submitNewSessionForm(event)">
                     {task_inputs}
-                    <button>🚀 Run</button>
+                    <button>🚀 Run New Session</button>
                 </form>
             </article>
             <article>

zrb/runner/web_route/static/resources/session/current-session.js

@@ -4,18 +4,22 @@ const CURRENT_SESSION = {
         const logTextarea = document.getElementById("log-textarea");
         const submitTaskForm = document.getElementById("submit-task-form");
         let isFinished = false;
+        let isInputUpdated = false;
         let errorCount = 0;
         while (!isFinished) {
             try {
                 const data = await this.getCurrentSession();
                 // update inputs
-                const dataInputs = data.input;
-                for (const inputName in dataInputs) {
-                    const inputValue = dataInputs[inputName];
-                    const input = submitTaskForm.querySelector(`[name="${inputName}"]`);
-                    if (input) {
-                        input.value = inputValue;
+                if (!isInputUpdated) {
+                    const dataInputs = data.input;
+                    for (const inputName in dataInputs) {
+                        const inputValue = dataInputs[inputName];
+                        const input = submitTaskForm.querySelector(`[name="${inputName}"]`);
+                        if (input) {
+                            input.value = inputValue;
+                        }
                     }
+                    isInputUpdated = true;
                 }
                 resultLineCount = data.final_result.split("\n").length;
                 resultTextarea.rows = resultLineCount <= 5 ? resultLineCount : 5;

zrb/runner/web_route/static/resources/session/event.js

@@ -20,7 +20,7 @@ window.addEventListener("load", async function () {
 
 
 const submitTaskForm = document.getElementById("submit-task-form");
-submitTaskForm.addEventListener("change", async function(event) {
+async function handleInputUpdate(event) {
     const currentInput = event.target;
     const inputs = Array.from(submitTaskForm.querySelectorAll("input[name], textarea[name], select[name]"));
     const inputMap = {};
@@ -59,6 +59,7 @@ submitTaskForm.addEventListener("change", async function(event) {
                 if (value === "") {
                     return;
                 }
+                console.log(input, data);
                 input.value = value;
             });
         } else {
@@ -67,6 +68,14 @@ submitTaskForm.addEventListener("change", async function(event) {
     } catch (error) {
         console.error("Error during fetch:", error);
     }
+}
+
+submitTaskForm.querySelectorAll("input[name], textarea[name]").forEach((element) => {
+    element.addEventListener("input", handleInputUpdate);
+    element.addEventListener("keyup", handleInputUpdate);
+});
+submitTaskForm.querySelectorAll("select[name]").forEach((element) => {
+    element.addEventListener("change", handleInputUpdate);
 });
 
 

zrb/task/llm_task.py

@@ -3,7 +3,9 @@ import os
 from collections.abc import Callable
 from typing import Any
 
-from pydantic import BaseModel
+from pydantic_ai import Agent, Tool
+from pydantic_ai.messages import ModelMessagesTypeAdapter
+from pydantic_ai.settings import ModelSettings
 
 from zrb.attr.type import StrAttr
 from zrb.config import LLM_MODEL, LLM_SYSTEM_PROMPT
@@ -16,25 +18,10 @@ from zrb.task.base_task import BaseTask
 from zrb.util.attr import get_str_attr
 from zrb.util.cli.style import stylize_faint
 from zrb.util.file import read_file, write_file
-from zrb.util.llm.tool import callable_to_tool_schema
 from zrb.util.run import run_async
 
 ListOfDict = list[dict[str, Any]]
-
-
-class AdditionalTool(BaseModel):
-    fn: Callable
-    name: str | None
-
-
-def scratchpad(thought: str) -> str:
-    """Write your thought, analysis, reasoning, and evaluation here."""
-    return thought
-
-
-def end_conversation(final_answer: str) -> str:
-    """End conversation with a final answer containing all necessary information"""
-    return final_answer
+ToolOrCallable = Tool | Callable
 
 
 class LLMTask(BaseTask):
@@ -48,11 +35,17 @@ class LLMTask(BaseTask):
         input: list[AnyInput | None] | AnyInput | None = None,
         env: list[AnyEnv | None] | AnyEnv | None = None,
         model: StrAttr | None = LLM_MODEL,
+        model_settings: (
+            ModelSettings | Callable[[AnySharedContext], ModelSettings] | None
+        ) = None,
         render_model: bool = True,
+        agent: Agent | Callable[[AnySharedContext], Agent] | None = None,
         system_prompt: StrAttr | None = LLM_SYSTEM_PROMPT,
         render_system_prompt: bool = True,
         message: StrAttr | None = None,
-        tools: list[Callable] | Callable[[AnySharedContext], list[Callable]] = [],
+        tools: (
+            list[ToolOrCallable] | Callable[[AnySharedContext], list[ToolOrCallable]]
+        ) = [],
         conversation_history: (
             ListOfDict | Callable[[AnySharedContext], ListOfDict]
         ) = [],
@@ -64,9 +57,6 @@ class LLMTask(BaseTask):
         ) = None,
         conversation_history_file: StrAttr | None = None,
         render_history_file: bool = True,
-        model_kwargs: (
-            dict[str, Any] | Callable[[AnySharedContext], dict[str, Any]]
-        ) = {},
         execute_condition: bool | str | Callable[[AnySharedContext], bool] = True,
         retries: int = 2,
         retry_period: float = 0,
@@ -76,6 +66,7 @@ class LLMTask(BaseTask):
         readiness_failure_threshold: int = 1,
         readiness_timeout: int = 60,
         monitor_readiness: bool = False,
+        max_call_iteration: int = 20,
         upstream: list[AnyTask] | AnyTask | None = None,
         fallback: list[AnyTask] | AnyTask | None = None,
         successor: list[AnyTask] | AnyTask | None = None,
@@ -102,97 +93,37 @@ class LLMTask(BaseTask):
             successor=successor,
         )
         self._model = model
+        self._model_settings = (model_settings,)
+        self._agent = agent
         self._render_model = render_model
-        self._model_kwargs = model_kwargs
         self._system_prompt = system_prompt
         self._render_system_prompt = render_system_prompt
         self._message = message
         self._tools = tools
+        self._additional_tools: list[ToolOrCallable] = []
         self._conversation_history = conversation_history
         self._conversation_history_reader = conversation_history_reader
         self._conversation_history_writer = conversation_history_writer
         self._conversation_history_file = conversation_history_file
         self._render_history_file = render_history_file
+        self._max_call_iteration = max_call_iteration
 
-    def add_tool(self, tool: Callable):
-        self._tools.append(tool)
+    def add_tool(self, tool: ToolOrCallable):
+        self._additional_tools.append(tool)
 
     async def _exec_action(self, ctx: AnyContext) -> Any:
-        import litellm
-        from litellm.utils import supports_function_calling
-
-        user_message = {"role": "user", "content": self._get_message(ctx)}
-        ctx.print(stylize_faint(f"{user_message}"))
-        model = self._get_model(ctx)
-        try:
-            is_function_call_supported = supports_function_calling(model=model)
-        except Exception:
-            is_function_call_supported = False
-            litellm.add_function_to_prompt = True
-        if not is_function_call_supported:
-            ctx.log_warning(f"Model {model} doesn't support function call")
-        available_tools = self._get_available_tools(
-            ctx, include_end_conversation=not is_function_call_supported
-        )
-        model_kwargs = self._get_model_kwargs(ctx, available_tools)
-        ctx.log_debug("MODEL KWARGS", model_kwargs)
-        system_prompt = self._get_system_prompt(ctx)
-        ctx.log_debug("SYSTEM PROMPT", system_prompt)
         history = await self._read_conversation_history(ctx)
-        ctx.log_debug("HISTORY PROMPT", history)
-        conversations = history + [user_message]
-        while True:
-            llm_response = await self._get_llm_response(
-                model, system_prompt, conversations, model_kwargs
-            )
-            llm_response_dict = llm_response.to_dict()
-            ctx.print(stylize_faint(f"{llm_response_dict}"))
-            conversations.append(llm_response_dict)
-            ctx.log_debug("RESPONSE MESSAGE", llm_response)
-            if is_function_call_supported:
-                if not llm_response.tool_calls:
-                    # No tool call, end conversation
-                    await self._write_conversation_history(ctx, conversations)
-                    return llm_response.content
-                await self._handle_tool_calls(
-                    ctx, available_tools, conversations, llm_response
-                )
-            if not is_function_call_supported:
-                try:
-                    json_payload = json.loads(llm_response.content)
-                    function_name = _get_fallback_function_name(json_payload)
-                    function_kwargs = _get_fallback_function_kwargs(json_payload)
-                    tool_execution_message = (
-                        await self._create_fallback_tool_exec_message(
-                            available_tools, function_name, function_kwargs
-                        )
-                    )
-                    ctx.print(stylize_faint(f"{tool_execution_message}"))
-                    conversations.append(tool_execution_message)
-                    if function_name == "end_conversation":
-                        await self._write_conversation_history(ctx, conversations)
-                        return function_kwargs.get("final_answer", "")
-                except Exception as e:
-                    ctx.log_error(e)
-                    tool_execution_message = self._create_exec_scratchpad_message(
-                        f"{e}"
-                    )
-                    conversations.append(tool_execution_message)
-
-    async def _handle_tool_calls(
-        self,
-        ctx: AnyContext,
-        available_tools: dict[str, Callable],
-        conversations: list[dict[str, Any]],
-        llm_response: Any,
-    ):
-        # noqa Reference: https://docs.litellm.ai/docs/completion/function_call#full-code---parallel-function-calling-with-gpt-35-turbo-1106
-        for tool_call in llm_response.tool_calls:
-            tool_execution_message = await self._create_tool_exec_message(
-                available_tools, tool_call
-            )
-            ctx.print(stylize_faint(f"{tool_execution_message}"))
-            conversations.append(tool_execution_message)
+        user_prompt = self._get_message(ctx)
+        agent = self._get_agent(ctx)
+        result = await agent.run(
+            user_prompt=user_prompt,
+            message_history=ModelMessagesTypeAdapter.validate_python(history),
+        )
+        new_history = json.loads(result.all_messages_json())
+        for history in new_history:
+            ctx.print(stylize_faint(json.dumps(history)))
+        await self._write_conversation_history(ctx, new_history)
+        return result.data
 
     async def _write_conversation_history(
         self, ctx: AnyContext, conversations: list[Any]
@@ -203,71 +134,32 @@ class LLMTask(BaseTask):
         if history_file != "":
             write_file(history_file, json.dumps(conversations, indent=2))
 
-    async def _get_llm_response(
-        self,
-        model: str,
-        system_prompt: str,
-        conversations: list[Any],
-        model_kwargs: dict[str, Any],
-    ) -> Any:
-        from litellm import acompletion
-
-        llm_response = await acompletion(
-            model=model,
-            messages=[{"role": "system", "content": system_prompt}] + conversations,
-            **model_kwargs,
+    def _get_model_settings(self, ctx: AnyContext) -> ModelSettings | None:
+        if isinstance(self._model_settings, ModelSettings):
+            return self._model_settings
+        if callable(self._model_settings):
+            return self._model_settings(ctx)
+        return None
+
+    def _get_agent(self, ctx: AnyContext) -> Agent:
+        if isinstance(self._agent, Agent):
+            return self._agent
+        if callable(self._agent):
+            return self._agent(ctx)
+        tools_or_callables = list(
+            self._tools(ctx) if callable(self._tools) else self._tools
         )
-        return llm_response.choices[0].message
-
-    async def _create_tool_exec_message(
-        self, available_tools: dict[str, Callable], tool_call: Any
-    ) -> dict[str, Any]:
-        function_name = tool_call.function.name
-        function_kwargs = json.loads(tool_call.function.arguments)
-        return {
-            "tool_call_id": tool_call.id,
-            "role": "tool",
-            "name": function_name,
-            "content": await self._get_exec_tool_result(
-                available_tools, function_name, function_kwargs
-            ),
-        }
-
-    async def _create_fallback_tool_exec_message(
-        self,
-        available_tools: dict[str, Callable],
-        function_name: str,
-        function_kwargs: dict[str, Any],
-    ) -> dict[str, Any]:
-        result = await self._get_exec_tool_result(
-            available_tools, function_name, function_kwargs
-        )
-        return self._create_exec_scratchpad_message(
-            f"Result of {function_name} call: {result}"
+        tools_or_callables.extend(self._additional_tools)
+        tools = [
+            tool if isinstance(tool, Tool) else Tool(tool, takes_ctx=False)
+            for tool in tools_or_callables
+        ]
+        return Agent(
+            self._get_model(ctx),
+            system_prompt=self._get_system_prompt(ctx),
+            tools=tools,
         )
 
-    def _create_exec_scratchpad_message(self, message: str) -> dict[str, Any]:
-        return {
-            "role": "assistant",
-            "content": json.dumps(
-                {"name": "scratchpad", "arguments": {"thought": message}}
-            ),
-        }
-
-    async def _get_exec_tool_result(
-        self,
-        available_tools: dict[str, Callable],
-        function_name: str,
-        function_kwargs: dict[str, Any],
-    ) -> str:
-        if function_name not in available_tools:
-            return f"[ERROR] Invalid tool: {function_name}"
-        function_to_call = available_tools[function_name]
-        try:
-            return await run_async(function_to_call(**function_kwargs))
-        except Exception as e:
-            return f"[ERROR] {e}"
-
     def _get_model(self, ctx: AnyContext) -> str:
         return get_str_attr(
             ctx, self._model, "ollama_chat/llama3.1", auto_render=self._render_model
@@ -284,30 +176,6 @@ class LLMTask(BaseTask):
     def _get_message(self, ctx: AnyContext) -> str:
         return get_str_attr(ctx, self._message, "How are you?", auto_render=True)
 
-    def _get_model_kwargs(
-        self, ctx: AnyContext, available_tools: dict[str, Callable]
-    ) -> dict[str, Any]:
-        model_kwargs = {}
-        if callable(self._model_kwargs):
-            model_kwargs = self._model_kwargs(ctx)
-        else:
-            model_kwargs = self._model_kwargs
-        model_kwargs["tools"] = [
-            callable_to_tool_schema(tool) for tool in available_tools.values()
-        ]
-        return model_kwargs
-
-    def _get_available_tools(
-        self, ctx: AnyContext, include_end_conversation: bool
-    ) -> dict[str, Callable]:
-        tools = {"scratchpad": scratchpad}
-        if include_end_conversation:
-            tools["end_conversation"] = end_conversation
-        tool_list = self._tools(ctx) if callable(self._tools) else self._tools
-        for tool in tool_list:
-            tools[tool.__name__] = tool
-        return tools
-
     async def _read_conversation_history(self, ctx: AnyContext) -> ListOfDict:
         if self._conversation_history_reader is not None:
             return await run_async(self._conversation_history_reader(ctx))
@@ -329,22 +197,3 @@ class LLMTask(BaseTask):
             "",
             auto_render=self._render_history_file,
         )
-
-
-def _get_fallback_function_name(json_payload: dict[str, Any]) -> str:
-    for key in ("name",):
-        if key in json_payload:
-            return json_payload[key]
-    raise ValueError("Function name not provided")
-
-
-def _get_fallback_function_kwargs(json_payload: dict[str, Any]) -> str:
-    for key in (
-        "arguments",
-        "args",
-        "parameters",
-        "params",
-    ):
-        if key in json_payload:
-            return json_payload[key]
-    raise ValueError("Function arguments not provided")

zrb/util/llm/tool.py

@@ -1,6 +1,6 @@
 import inspect
 from collections.abc import Callable
-from typing import Any, get_type_hints
+from typing import Annotated, Any, Literal, get_type_hints
 
 
 def callable_to_tool_schema(callable_obj: Callable) -> dict[str, Any]:
@@ -21,10 +21,14 @@ def callable_to_tool_schema(callable_obj: Callable) -> dict[str, Any]:
     # Build parameter schema
     param_schema = {"type": "object", "properties": {}, "required": []}
     for param_name, param in sig.parameters.items():
-        param_type = hints.get(param_name, str)  # Default type is string
-        param_schema["properties"][param_name] = {
-            "type": _python_type_to_json_type(param_type)
-        }
+        # Get the type hint or default to str
+        param_type = hints.get(param_name, str)
+
+        # Handle annotated types (e.g., Annotated[str, "description"])
+        json_type, param_metadata = _process_type_annotation(param_type)
+        param_schema["properties"][param_name] = param_metadata
+
+        # Mark required parameters
         if param.default is inspect.Parameter.empty:
             param_schema["required"].append(param_name)
     return {
@@ -37,6 +41,30 @@ def callable_to_tool_schema(callable_obj: Callable) -> dict[str, Any]:
     }
 
 
+def _process_type_annotation(py_type: Any) -> tuple[str, dict]:
+    """
+    Process type annotations and return the JSON Schema type and metadata.
+
+    :param py_type: The type annotation.
+    :return: A tuple of (JSON type, parameter metadata).
+    """
+    if hasattr(py_type, "__origin__") and py_type.__origin__ is Literal:
+        # Handle Literal (enum)
+        enum_values = list(py_type.__args__)
+        return "string", {"type": "string", "enum": enum_values}
+
+    if hasattr(py_type, "__origin__") and py_type.__origin__ is Annotated:
+        # Handle Annotated types
+        base_type = py_type.__args__[0]
+        description = py_type.__args__[1]
+        json_type = _python_type_to_json_type(base_type)
+        return json_type, {"type": json_type, "description": description}
+
+    # Fallback to basic type conversion
+    json_type = _python_type_to_json_type(py_type)
+    return json_type, {"type": json_type}
+
+
 def _python_type_to_json_type(py_type):
     """
     Map Python types to JSON Schema types.

{zrb-1.0.0b4.dist-info → zrb-1.0.0b6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: zrb
-Version: 1.0.0b4
+Version: 1.0.0b6
 Summary: Your Automation Powerhouse
 Home-page: https://github.com/state-alchemists/zrb
 License: AGPL-3.0-or-later
@@ -19,10 +19,11 @@ Requires-Dist: beautifulsoup4 (>=4.12.3,<5.0.0)
 Requires-Dist: black (>=24.10.0,<24.11.0)
 Requires-Dist: chromadb (>=0.5.20,<0.6.0) ; extra == "rag"
 Requires-Dist: fastapi[standard] (>=0.115.6,<0.116.0)
+Requires-Dist: fastembed (>=0.5.1,<0.6.0)
 Requires-Dist: isort (>=5.13.2,<5.14.0)
 Requires-Dist: libcst (>=1.5.0,<2.0.0)
-Requires-Dist: litellm (>=1.52.12,<2.0.0)
 Requires-Dist: pdfplumber (>=0.11.4,<0.12.0) ; extra == "rag"
+Requires-Dist: pydantic-ai (>=0.0.19,<0.0.20)
 Requires-Dist: python-dotenv (>=1.0.1,<2.0.0)
 Requires-Dist: python-jose[cryptography] (>=3.3.0,<4.0.0)
 Requires-Dist: requests (>=2.32.3,<3.0.0)

{zrb-1.0.0b4.dist-info → zrb-1.0.0b6.dist-info}/RECORD

@@ -7,11 +7,11 @@ zrb/builtin/base64.py,sha256=1YnSwASp7OEAvQcsnHZGpJEvYoI1Z2zTIJ1bCDHfcPQ,921
 zrb/builtin/git.py,sha256=xHzg0srhp1uOSXWvwA--Fo8idkt0G9010iJ8uIndzg4,5463
 zrb/builtin/git_subtree.py,sha256=GwI8befmvXEoX1xyZ4jkeG8nsyCkuRG1lzPiGss3yqw,3493
 zrb/builtin/group.py,sha256=-phJfVpTX3_gUwS1u8-RbZUHe-X41kxDBSmrVh4rq8E,1682
-zrb/builtin/llm/llm_chat.py,sha256=zrUzVjHBTOJS81--npePIJQo7zzmeERUVRMon91yI3c,4488
-zrb/builtin/llm/previous-session.js,sha256=lZcQIdxr_5Qygm9U2tkMlOg_-gz7G-r6YEdQqz0_RnE,578
+zrb/builtin/llm/llm_chat.py,sha256=vUUchYKJuHr-N_HarpKRFsV0EdQDAFZzAfbK4C0vx88,4508
+zrb/builtin/llm/previous-session.js,sha256=xMKZvJoAbrwiyHS0OoPrWuaKxWYLoyR5sguePIoCjTY,816
 zrb/builtin/llm/tool/api.py,sha256=yQ3XV8O7Fx7hHssLSOcmiHDnevPhz9ktWi44HK7zTls,801
 zrb/builtin/llm/tool/cli.py,sha256=to_IjkfrMGs6eLfG0cpVN9oyADWYsJQCtyluUhUdBww,253
-zrb/builtin/llm/tool/rag.py,sha256=jJRLERW6824JeEzEQ_OqLMaaa3mjuNqsRcRWoL1wVx0,5192
+zrb/builtin/llm/tool/rag.py,sha256=PawaLZL-ThctxtBtsQuP3XsgTxQKyCGFqrudCANPJKk,5162
 zrb/builtin/llm/tool/web.py,sha256=N2HYuXbKPUpjVAq_UnQMbUrTIE8u0Ut3TeQadZ7_NJc,2217
 zrb/builtin/md5.py,sha256=0pNlrfZA0wlZlHvFHLgyqN0JZJWGKQIF5oXxO44_OJk,949
 zrb/builtin/project/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -171,7 +171,7 @@ zrb/callback/callback.py,sha256=hKefB_Jd1XGjPSLQdMKDsGLHPzEGO2dqrIArLl_EmD0,848
 zrb/cmd/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/cmd/cmd_result.py,sha256=L8bQJzWCpcYexIxHBNsXj2pT3BtLmWex0iJSMkvimOA,597
 zrb/cmd/cmd_val.py,sha256=7Doowyg6BK3ISSGBLt-PmlhzaEkBjWWm51cED6fAUOQ,1014
-zrb/config.py,sha256=tFPjLqgcb6cH33trooDgJRsoJBSneB03ccnzx90iH9M,4664
+zrb/config.py,sha256=RB0ikM82RnCDZaarJOEBfD7GeYPfcq6jLb6F359il1s,4677
 zrb/content_transformer/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/content_transformer/any_content_transformer.py,sha256=v8ZUbcix1GGeDQwB6OKX_1TjpY__ksxWVeqibwa_iZA,850
 zrb/content_transformer/content_transformer.py,sha256=YU6Xr3G_IaCWKQGsf9z9YlCclbiwcJ7ytQv3wKpPIiI,2125
@@ -225,7 +225,7 @@ zrb/runner/web_route/node_page/group/view.html,sha256=wISun627ciFZcvGpxANG0pr1zg
 zrb/runner/web_route/node_page/node_page_route.py,sha256=LYi60eZ5ZGgykTIeSQk5Hn9OYjh3ocYgBIAue7Bznvw,2268
 zrb/runner/web_route/node_page/task/partial/input.html,sha256=X2jy0q7TLQGP853exZMed0lqPezL3gzn6mnhB5QKfkc,178
 zrb/runner/web_route/node_page/task/show_task_page.py,sha256=0HIFEuy5DLOKqff4Wib5WaMe5Om0B4C7BH63pPIA-OU,2639
-zrb/runner/web_route/node_page/task/view.html,sha256=CwrIiPJAwHEEIVNp_wkTzW7kQJjTh41FSDamCswI3S8,3593
+zrb/runner/web_route/node_page/task/view.html,sha256=T6kXNYKGhHsY5A74vQt_tj68WrULmPRaSuMqI1ygD1o,3605
 zrb/runner/web_route/refresh_token_api_route.py,sha256=JOuzhQUtRA62w3l27mq-jXgpaV7Rbj20jzxpQacssio,1478
 zrb/runner/web_route/static/refresh-token.template.js,sha256=v_nF7nU1AXp-KtsHNNzamhciEi7NCSTPEDT5hCxn29g,735
 zrb/runner/web_route/static/resources/common.css,sha256=u5rGLsPx2943z324iQ2X81krM3z-kc-8e1SkBdYAvKU,157
@@ -234,8 +234,8 @@ zrb/runner/web_route/static/resources/login/event.js,sha256=1-NxaUwU-X7Tu2RAwVkz
 zrb/runner/web_route/static/resources/logout/event.js,sha256=MfZxrTa2yL49Lbh7cCZDdqsIcf9e1q3W8-WjmZXV5pA,692
 zrb/runner/web_route/static/resources/pico.min.css,sha256=_Esfkjs_U_igYn-tXBUaK3AEKb7d4l9DlmaOiw9bXfI,82214
 zrb/runner/web_route/static/resources/session/common-util.js,sha256=t7_s5DXgMyZlT8L8LYZTkzOT6vWVeZvmCKjt-bflQY0,2117
-zrb/runner/web_route/static/resources/session/current-session.js,sha256=Stewc8SOpv14ARJMlWJW4hegGMFCKwcnnNQZ2wKKW6I,6571
-zrb/runner/web_route/static/resources/session/event.js,sha256=1WujNriQupSX-zE-uqXlrzgZn6jchDKzWTATvDuZjhA,4359
+zrb/runner/web_route/static/resources/session/current-session.js,sha256=GlRBLwItCwITqVR_hUQFr6W1myD9WRl8R_TTbrzCovw,6739
+zrb/runner/web_route/static/resources/session/event.js,sha256=X5OlSHefK0SDB9VkFCRyBKE_Pb7mqM319mW9jRGoDOk,4716
 zrb/runner/web_route/static/resources/session/past-session.js,sha256=RwGJYKSp75K8NZ-iZP58XppWgdzkiKFaiC5wgcMLxDo,5470
 zrb/runner/web_route/static/static_route.py,sha256=7x069VfACZLkLykY0vLL5t13jIQPgkeEJtkpbfNQfLg,1540
 zrb/runner/web_route/task_input_api_route.py,sha256=xkZ36vmHXMPj0ekp6ocUysO0QUgl1PLaSHt3YL_OfK8,1749
@@ -262,7 +262,7 @@ zrb/task/base_task.py,sha256=ImA0ReyB6neVUfY4nKLnL0h2EMGIJ9wvvNvIAN92-RE,21194
 zrb/task/base_trigger.py,sha256=jC722rDvodaBLeNaFghkTyv1u0QXrK6BLZUUqcmBJ7Q,4581
 zrb/task/cmd_task.py,sha256=JpClYoEmJTqKSxhuCErXd2kHLS3Hk2zXeYnl7entNeU,10378
 zrb/task/http_check.py,sha256=Gf5rOB2Se2EdizuN9rp65HpGmfZkGc-clIAlHmPVehs,2565
-zrb/task/llm_task.py,sha256=Hxu2kaAMHPquCE96bBt5MFh9A2EF_-bYeAMf0fOqSz8,13466
+zrb/task/llm_task.py,sha256=ptXC3x9Dwn7-4JrGQyEtzOXZ4dNQATDgCeowkvwAu9U,7723
 zrb/task/make_task.py,sha256=PD3b_aYazthS8LHeJsLAhwKDEgdurQZpymJDKeN60u0,2265
 zrb/task/rsync_task.py,sha256=pVVslZ46qgcpU_EKhyTQEQie8kUOMuTsVQdbQG2L-yk,6318
 zrb/task/scaffolder.py,sha256=rME18w1HJUHXgi9eTYXx_T2G4JdqDYzBoNOkdOOo5-o,6806
@@ -293,7 +293,7 @@ zrb/util/file.py,sha256=cBPkIonfcWytoqtG3ScJd6FFK7HVYeCIuLmfAFO1HIQ,791
 zrb/util/git.py,sha256=o_kLF1fySv5uQdLlhY-ztc-z0zLOdcDf0IpuPAl2ciA,4733
 zrb/util/git_subtree.py,sha256=US8oCHUOKgt14Ie6SaEelwTs__jLGLPsDQZvI-1P4KY,2640
 zrb/util/group.py,sha256=Bg7HrSycoK110U5s_Tca6-uUQuZ5CMgb8wxZSrvDQ98,2790
-zrb/util/llm/tool.py,sha256=Ux58wYSymVCXxgIWDUObsluZzYzmfyB0uwgKagL0bVg,1898
+zrb/util/llm/tool.py,sha256=NkENrUlGxcqqU7jzHAH7DBXNcm_ndEo2dFnJ5nhvWmk,2991
 zrb/util/load.py,sha256=i8_83ApWJXlZlbFMNfEptrOzfXdvtaIhAErsd6tU9y8,1649
 zrb/util/run.py,sha256=DGHUP9x1Q8V8UF3FbpmjLGuhVVCCLfjTH2teT8qXlNI,207
 zrb/util/string/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -303,7 +303,7 @@ zrb/util/string/name.py,sha256=8picJfUBXNpdh64GNaHv3om23QHhUZux7DguFLrXHp8,1163
 zrb/util/todo.py,sha256=1nDdwPc22oFoK_1ZTXyf3638Bg6sqE2yp_U4_-frHoc,16015
 zrb/xcom/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/xcom/xcom.py,sha256=o79rxR9wphnShrcIushA0Qt71d_p3ZTxjNf7x9hJB78,1571
-zrb-1.0.0b4.dist-info/METADATA,sha256=G-QIlz6Gls_B72tCDApTqI-7B0minjgSYXZJDBAqp8Y,4224
-zrb-1.0.0b4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-zrb-1.0.0b4.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
-zrb-1.0.0b4.dist-info/RECORD,,
+zrb-1.0.0b6.dist-info/METADATA,sha256=ZGOBMXjA82fN6lsJ3yNmJTGqwoZWJcNEKCxXtTouk68,4270
+zrb-1.0.0b6.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+zrb-1.0.0b6.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
+zrb-1.0.0b6.dist-info/RECORD,,