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

zrb/builtin/llm/chat_session.py

@@ -29,7 +29,9 @@ async def read_user_prompt(ctx: AnyContext) -> str:
 
     _show_info(ctx)
     final_result = ""
-    ctx.print(stylize_faint("🧑 >> ") + f"{ctx.input.message}", plain=True)
+    ctx.print(stylize_faint("💬"), plain=True)
+    ctx.print(ctx.input.message, plain=True)
+    ctx.print("", plain=True)
     result = await _trigger_ask_and_wait_for_result(
         ctx,
         user_prompt=ctx.input.message,
@@ -46,9 +48,11 @@ async def read_user_prompt(ctx: AnyContext) -> str:
     user_input_session = PromptSession()
     while True:
         await asyncio.sleep(0.01)
-        ctx.print(stylize_faint("🧑 >> "), end="", plain=True)
+        if not multiline_mode:
+            ctx.print(stylize_faint("💬"), plain=True)
         user_input = await user_input_session.prompt_async()
-        # user_input = input()
+        if not multiline_mode:
+            ctx.print("", plain=True)
         # Handle special input
         if user_input.strip().lower() in ("/bye", "/quit"):
             user_prompt = "\n".join(user_inputs)
@@ -60,6 +64,7 @@ async def read_user_prompt(ctx: AnyContext) -> str:
         elif user_input.strip().lower() in ("/multi",):
             multiline_mode = True
         elif user_input.strip().lower() in ("/end",):
+            ctx.print("", plain=True)
             multiline_mode = False
             user_prompt = "\n".join(user_inputs)
             user_inputs = []
@@ -103,7 +108,9 @@ 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(f"{stylize_faint('🤖 >>')} {result}", plain=True)
+    ctx.print(stylize_faint("\n🤖"), plain=True)
+    ctx.print(result, plain=True)
+    ctx.print("", plain=True)
     return result
 
 
@@ -199,13 +206,11 @@ def _show_info(ctx: AnyContext):
         ctx: The context object for the task.
     """
     ctx.print(
-        "\n".join(
-            [
-                f"{stylize_bold_yellow('/bye')}   {stylize_faint('Quit from chat session')}",
-                f"{stylize_bold_yellow('/multi')} {stylize_faint('Start multiline input')}",
-                f"{stylize_bold_yellow('/end')}   {stylize_faint('End multiline input')}",
-                f"{stylize_bold_yellow('/help')}  {stylize_faint('Show this message')}",
-            ]
+        (
+            f"  {stylize_bold_yellow('/bye')}   {stylize_faint('Quit from chat session')}\n"
+            f"  {stylize_bold_yellow('/multi')} {stylize_faint('Start multiline input')}\n"
+            f"  {stylize_bold_yellow('/end')}   {stylize_faint('End multiline input')}\n"
+            f"  {stylize_bold_yellow('/help')}  {stylize_faint('Show this message')}\n"
         ),
         plain=True,
     )

zrb/builtin/llm/llm_ask.py

@@ -10,7 +10,9 @@ from zrb.builtin.llm.tool.file import (
     apply_diff,
     list_files,
     read_from_file,
+    read_many_files,
     search_files,
+    write_many_files,
     write_to_file,
 )
 from zrb.builtin.llm.tool.web import (
@@ -132,8 +134,10 @@ if CFG.LLM_ALLOW_ACCESS_LOCAL_FILE:
         search_files,
         list_files,
         read_from_file,
+        read_many_files,
         apply_diff,
         write_to_file,
+        write_many_files,
     )
 
 if CFG.LLM_ALLOW_ACCESS_SHELL:

zrb/builtin/llm/tool/code.py

@@ -82,8 +82,8 @@ async def analyze_repo(
     goal: str,
     extensions: list[str] = _DEFAULT_EXTENSIONS,
     exclude_patterns: list[str] = DEFAULT_EXCLUDED_PATTERNS,
-    extraction_token_limit: int = 30000,
-    summarization_token_limit: int = 30000,
+    extraction_token_limit: int = 40000,
+    summarization_token_limit: int = 40000,
 ) -> str:
     """
     Extract and summarize information from any directory.

zrb/builtin/llm/tool/file.py

@@ -2,7 +2,7 @@ import fnmatch
 import json
 import os
 import re
-from typing import Any, Optional
+from typing import Any, Dict, List, Optional
 
 from zrb.builtin.llm.tool.sub_agent import create_sub_agent_tool
 from zrb.context.any_context import AnyContext
@@ -215,6 +215,7 @@ def read_from_file(
     end_line: Optional[int] = None,
 ) -> str:
     """Read file content (or specific lines) at a path, including line numbers.
+    This tool can read both, text and pdf file.
     Args:
         path (str): Path to read. Pass exactly as provided, including '~'.
         start_line (Optional[int]): Starting line number (1-based).
@@ -466,7 +467,7 @@ def apply_diff(
 
 
 async def analyze_file(
-    ctx: AnyContext, path: str, query: str, token_limit: int = 30000
+    ctx: AnyContext, path: str, query: str, token_limit: int = 40000
 ) -> str:
     """Analyze file using LLM capability to reduce context usage.
     Use this tool for:
@@ -498,3 +499,105 @@ async def analyze_file(
     )
     clipped_payload = llm_rate_limitter.clip_prompt(payload, token_limit)
     return await _analyze_file(ctx, clipped_payload)
+
+
+def read_many_files(paths: List[str]) -> str:
+    """
+    Read and return the content of multiple files.
+
+    This function is ideal for when you need to inspect the contents of
+    several files at once. For each file path provided in the input list,
+    it reads the entire file content. The result is a JSON string
+    containing a dictionary where keys are the file paths and values are
+    the corresponding file contents.
+
+    Use this tool when you need a comprehensive view of multiple files,
+    for example, to understand how different parts of a module interact,
+    to check configurations across various files, or to gather context
+    before making widespread changes.
+
+    Args:
+        paths (List[str]): A list of absolute or relative paths to the
+                           files you want to read. It is crucial to
+                           provide accurate paths. Use the `list_files`
+                           tool if you are unsure about the exact file
+                           locations.
+
+    Returns:
+        str: A JSON string representing a dictionary where each key is a
+             file path and the corresponding value is the content of that
+             file. If a file cannot be read, its entry in the dictionary
+             will contain an error message.
+             Example:
+             {
+                 "results": {
+                     "path/to/file1.py": "...",
+                     "path/to/file2.txt": "..."
+                 }
+             }
+    """
+    results = {}
+    for path in paths:
+        try:
+            abs_path = os.path.abspath(os.path.expanduser(path))
+            if not os.path.exists(abs_path):
+                raise FileNotFoundError(f"File not found: {path}")
+            content = read_file_with_line_numbers(abs_path)
+            results[path] = content
+        except Exception as e:
+            results[path] = f"Error reading file: {e}"
+    return json.dumps({"results": results})
+
+
+def write_many_files(files: Dict[str, str]) -> str:
+    """
+    Write content to multiple files simultaneously.
+
+    This function allows you to create, overwrite, or update multiple
+    files in a single operation. You provide a dictionary where each
+    key is a file path and the corresponding value is the content to be
+    written to that file. This is particularly useful for applying
+    changes across a project, such as refactoring code, updating
+    configuration files, or creating a set of new files from a template.
+
+    Each file is handled as a complete replacement of its content. If the
+    file does not exist, it will be created. If it already exists, its
+
+    entire content will be overwritten with the new content you provide.
+    Therefore, it is essential to provide the full, intended content for
+    each file.
+
+    Args:
+        files (Dict[str, str]): A dictionary where keys are the file paths
+                                (absolute or relative) and values are the
+                                complete contents to be written to those
+                                files.
+
+    Returns:
+        str: A JSON string summarizing the operation. It includes a list
+             of successfully written files and a list of files that
+             failed, along with the corresponding error messages.
+             Example:
+             {
+                 "success": [
+                     "path/to/file1.py",
+                     "path/to/file2.txt"
+                 ],
+                 "errors": {
+                     "path/to/problematic/file.py": "Error message"
+                 }
+             }
+    """
+    success = []
+    errors = {}
+    for path, content in files.items():
+        try:
+            abs_path = os.path.abspath(os.path.expanduser(path))
+            directory = os.path.dirname(abs_path)
+            if directory and not os.path.exists(directory):
+                os.makedirs(directory, exist_ok=True)
+            write_file(abs_path, content)
+            success.append(path)
+        except Exception as e:
+            errors[path] = f"Error writing file: {e}"
+    return json.dumps({"success": success, "errors": errors})

zrb/builtin/llm/tool/web.py

@@ -58,7 +58,7 @@ async def open_web_page(url: str) -> str:
 
 def create_search_internet_tool(serp_api_key: str) -> Callable[[str, int], str]:
     def search_internet(query: str, num_results: int = 10) -> str:
-        """Search the internet using SerpApi (Google Search) and return parsed results.
+        """Search the internet Google Search and return parsed results.
         Args:
             query (str): Search query.
             num_results (int): Search result count. Defaults to 10.

zrb/config.py

@@ -243,13 +243,19 @@ class Config:
 
     @property
     def LLM_MAX_REQUESTS_PER_MINUTE(self) -> int:
-        """Maximum number of LLM requests allowed per minute."""
-        return int(os.getenv("LLM_MAX_REQUESTS_PER_MINUTE", "60"))
+        """
+        Maximum number of LLM requests allowed per minute.
+        Default is conservative to accommodate free-tier LLM providers.
+        """
+        return int(os.getenv("LLM_MAX_REQUESTS_PER_MINUTE", "15"))
 
     @property
     def LLM_MAX_TOKENS_PER_MINUTE(self) -> int:
-        """Maximum number of LLM tokens allowed per minute."""
-        return int(os.getenv("ZRB_LLM_MAX_TOKENS_PER_MINUTE", "200000"))
+        """
+        Maximum number of LLM tokens allowed per minute.
+        Default is conservative to accommodate free-tier LLM providers.
+        """
+        return int(os.getenv("ZRB_LLM_MAX_TOKENS_PER_MINUTE", "100000"))
 
     @property
     def LLM_MAX_TOKENS_PER_REQUEST(self) -> int:
@@ -270,16 +276,16 @@ class Config:
         return to_boolean(os.getenv("ZRB_LLM_SUMMARIZE_HISTORY", "true"))
 
     @property
-    def LLM_HISTORY_SUMMARIZATION_THRESHOLD(self) -> int:
-        return int(os.getenv("ZRB_LLM_HISTORY_SUMMARIZATION_THRESHOLD", "5"))
+    def LLM_HISTORY_SUMMARIZATION_TOKEN_THRESHOLD(self) -> int:
+        return int(os.getenv("ZRB_LLM_HISTORY_SUMMARIZATION_TOKEN_THRESHOLD", "3000"))
 
     @property
     def LLM_ENRICH_CONTEXT(self) -> bool:
         return to_boolean(os.getenv("ZRB_LLM_ENRICH_CONTEXT", "true"))
 
     @property
-    def LLM_CONTEXT_ENRICHMENT_THRESHOLD(self) -> int:
-        return int(os.getenv("ZRB_LLM_CONTEXT_ENRICHMENT_THRESHOLD", "5"))
+    def LLM_CONTEXT_ENRICHMENT_TOKEN_THRESHOLD(self) -> int:
+        return int(os.getenv("ZRB_LLM_CONTEXT_ENRICHMENT_TOKEN_THRESHOLD", "3000"))
 
     @property
     def LLM_HISTORY_DIR(self) -> str: