tau-coding-agent 0.5.7__tar.gz → 0.6.0__tar.gz

{tau_coding_agent-0.5.7 → tau_coding_agent-0.6.0}/MANIFEST.in

@@ -2,3 +2,5 @@ recursive-include tau/builtins *
 recursive-include docs *
 include README.md
 include LICENSE
+global-exclude *.py[cod]
+global-exclude __pycache__

{tau_coding_agent-0.5.7/tau_coding_agent.egg-info → tau_coding_agent-0.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tau-coding-agent
-Version: 0.5.7
+Version: 0.6.0
 Summary: A self-extensible agent CLI with terminal UI, multi-provider LLM support, session management with branching, and a plugin system for tools and commands
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown
@@ -25,9 +25,11 @@ Requires-Dist: pyyaml==6.0.2
 Requires-Dist: pygments==2.20.0
 Requires-Dist: pathspec==1.1.1
 Requires-Dist: posthog==7.21.3
+Requires-Dist: pylatexenc==2.10
 Requires-Dist: gitpython==3.1.50
 Requires-Dist: grapheme==0.6.0
 Requires-Dist: arrow==1.4.0
+Requires-Dist: kiwisolver==1.5.0
 Provides-Extra: tools
 Requires-Dist: ripgrep>=14.0.0; extra == "tools"
 Requires-Dist: ast-grep-cli>=0.44.0; extra == "tools"
@@ -50,7 +52,7 @@ Start a conversation with the agent in the terminal.
 ### Installation
 
 ```bash
-uv pip install tau-coding-agent
+pip install tau-coding-agent
 ```
 
 Or install from source:

{tau_coding_agent-0.5.7 → tau_coding_agent-0.6.0}/README.md

@@ -15,7 +15,7 @@ Start a conversation with the agent in the terminal.
 ### Installation
 
 ```bash
-uv pip install tau-coding-agent
+pip install tau-coding-agent
 ```
 
 Or install from source:

{tau_coding_agent-0.5.7 → tau_coding_agent-0.6.0}/docs/inference-providers.md

@@ -60,7 +60,9 @@ tau --model openai/gpt-4o -p "Say hello"
 
 ## Google Gemini
 
-Google provides Gemini models with multimodal capabilities.
+Google AI Studio provides API-key access to Gemini models through the Google
+Gen AI SDK. Tau's `google` provider uses the Gemini Developer API; it is
+separate from the `google-vertex` provider, which uses Google Cloud.
 
 ### Setup
 
@@ -75,9 +77,16 @@ export GOOGLE_API_KEY=...
 ### Verify
 
 ```bash
-tau --model google/gemini-2.0-flash -p "Say hello"
+tau --model google/gemini-2.5-flash -p "Say hello"
 ```
 
+Current Google AI Studio model IDs include:
+
+- `gemini-3.5-flash` (stable)
+- `gemini-3.1-flash-lite` (stable)
+- `gemini-3.1-pro-preview` (preview)
+- `gemini-2.5-pro`, `gemini-2.5-flash`, and `gemini-2.5-flash-lite`
+
 ## Mistral AI
 
 Mistral offers efficient open-source-based models.
@@ -152,6 +161,17 @@ tau --model ollama/mistral -p "Say hello"
 
 ## Switching Providers
 
+## Speech-to-Text Timestamps
+
+Pass `TimestampGranularity.Word` or `TimestampGranularity.Segment` in
+`STTContext.timestamp_granularities` when using the Python API.
+
+- OpenAI `whisper-1` and Groq Whisper models support word and segment timestamps.
+- OpenAI GPT-4o transcription models return plain JSON and do not expose detailed
+  timestamps through this API.
+- ElevenLabs Scribe and Sarvam Saaras support word timestamps. Sarvam receives
+  `with_timestamps=true` when word timestamps are requested.
+
 ### Command Line
 
 Use a specific provider for a single session:

{tau_coding_agent-0.5.7 → tau_coding_agent-0.6.0}/docs/python-api.md

@@ -325,6 +325,9 @@ await runtime.execute_terminal("git status")
 | `runtime.settings_manager` | `SettingsManager` | Settings access |
 | `runtime.extension_runtime` | `ExtensionRuntime \| None` | Loaded extensions |
 
+`ExtensionRuntime.get_extensions()` returns the loaded extensions in
+registration order as a read-only tuple.
+
 ## Listening to Events
 
 Subscribe to the hook bus directly to observe what the agent does:

{tau_coding_agent-0.5.7 → tau_coding_agent-0.6.0}/docs/sessions.md

@@ -63,6 +63,9 @@ Or the short form:
 tau -r
 ```
 
+After resuming, `/new` always starts an empty session; the startup resume
+selection is not reused for later session changes.
+
 ### Browse Sessions
 
 Choose from a list of previous sessions interactively:

{tau_coding_agent-0.5.7 → tau_coding_agent-0.6.0}/docs/themes.md

@@ -39,6 +39,8 @@ Set your default theme in `~/.tau/settings.json` or `.tau/settings.json`:
 ```
 
 Opens an interactive picker. Use ↑↓ to preview, Enter to apply.
+The theme submenu in `/settings` uses the same ↑/↓ live preview. Press Enter to
+apply or Escape to restore the previous theme.
 
 ---
 

{tau_coding_agent-0.5.7 → tau_coding_agent-0.6.0}/docs/tools.md

@@ -64,9 +64,11 @@ Commands run in the agent's current working directory.
 
 Output is streamed through tool-update events while the command runs. Updates
 are throttled to at most once every 100 milliseconds, with guaranteed initial
-and final updates. Timeout and cancellation terminate the command's complete
-process tree. Programs may still buffer their own output; use an unbuffered mode
-such as `python -u` when immediate output is required.
+and final updates. The interactive TUI refreshes the existing result block for
+each update; when a long result is collapsed, its newest lines remain visible
+while the command is running. Timeout and cancellation terminate the command's
+complete process tree. Programs may still buffer their own output; use an
+unbuffered mode such as `python -u` when immediate output is required.
 
 ### glob
 

{tau_coding_agent-0.5.7 → tau_coding_agent-0.6.0}/docs/tui.md

@@ -32,6 +32,14 @@ from tau.tui import (
 | Overlays | `OverlayOptions`, `OverlayHandle` |
 | Styling | Theme dataclasses, color helpers, `render_markdown` |
 
+### Markdown math
+
+`render_markdown` converts inline (`$...$`) and display (`$$...$$`) LaTeX math
+to terminal-readable Unicode text through `pylatexenc`. Display math is placed
+on separate lines. Code spans and fenced code blocks retain their original
+LaTeX source. This is a readable plain-text representation, not fully typeset
+mathematical layout.
+
 ## Minimal Application
 
 ```python
@@ -78,15 +86,17 @@ automatically during shutdown.
 Subclass `Component` to create a reusable view:
 
 ```python
-from tau.tui import Component, KeyEvent
+from tau.tui import Buffer, Component, KeyEvent, Line, Rect, Span
 
 
 class Counter(Component):
     def __init__(self) -> None:
         self.value = 0
 
-    def render(self, width: int) -> list[str]:
-        return [f"Count: {self.value}"[:width]]
+    def render_cells(self, area: Rect, buf: Buffer) -> int:
+        buf.grow_to(area.y + 1)
+        buf.set_line(area.x, area.y, Line([Span(f"Count: {self.value}")]), area.width)
+        return 1
 
     def handle_input(self, event: object) -> bool:
         if isinstance(event, KeyEvent) and event.matches("up"):
@@ -95,8 +105,9 @@ class Counter(Component):
         return False
 ```
 
-`render()` returns one string per terminal row. Components must respect the
-provided width and must not include newline characters inside a returned line.
+`render_cells()` writes styled cells into the provided `Buffer` and returns the
+number of rows written. Components must grow dynamic buffers before writing and
+must respect the provided `Rect`.
 
 ## Layout
 
@@ -106,6 +117,8 @@ provided width and must not include newline characters inside a returned line.
 - `Columns` creates multi-line fixed and flexible columns.
 - `Rows` distributes a fixed terminal height across children.
 - `Constrained` places a child at an absolute or percentage width.
+- `Box` wraps a child `Component` with horizontal/vertical padding and an
+  optional background `Style`.
 
 ## Focus and Input
 
@@ -166,7 +179,8 @@ Runtime-aware behavior belongs in `tau.modes.interactive`.
 
 This boundary is enforced by `tests/test_tui_public_api.py`.
 
-The renderer retains only the current transcript frame and its current
-line-wrapping cache. Unchanged lines reuse width calculations across frames so
-streaming updates do not repeatedly parse ANSI styling across the complete
-session.
+The renderer retains only the current transcript frame. Styled content that
+exceeds its available width wraps directly into Buffer rows without losing
+content, and reflows when the terminal is resized. Finalized message rows are
+cached as cells so streaming updates do not repeatedly parse ANSI styling
+across the complete session.

{tau_coding_agent-0.5.7 → tau_coding_agent-0.6.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "tau-coding-agent"
-version = "0.5.7"
+version = "0.6.0"
 description = "A self-extensible agent CLI with terminal UI, multi-provider LLM support, session management with branching, and a plugin system for tools and commands"
 readme = "README.md"
 requires-python = ">=3.12"
@@ -29,9 +29,11 @@ dependencies = [
     "pygments==2.20.0",
     "pathspec==1.1.1",
     "posthog==7.21.3",
+    "pylatexenc==2.10",
     "gitpython==3.1.50",
     "grapheme==0.6.0",
     "arrow==1.4.0",
+    "kiwisolver==1.5.0",
 ]
 
 [project.scripts]

{tau_coding_agent-0.5.7 → tau_coding_agent-0.6.0}/tau/builtins/extensions/ask_user/__init__.py

@@ -6,7 +6,7 @@ from pathlib import Path
 # Add this directory to sys.path to allow easy resolution of schema, component, and tool.
 sys.path.insert(0, str(Path(__file__).parent))
 
-from tool import AskUserTool
+from tool import AskUserTool  # type: ignore[import-not-found]
 
 
 def register(tau) -> None:

{tau_coding_agent-0.5.7 → tau_coding_agent-0.6.0}/tau/builtins/extensions/ask_user/component.py

@@ -1,12 +1,16 @@
 from __future__ import annotations
 
-from typing import Any
+from typing import TYPE_CHECKING, Any
 
-from schema import AskUserOption
+from schema import AskUserOption  # type: ignore[import-not-found]
 
 from tau.tui.component import Component
 from tau.tui.input import InputEvent, KeyEvent
 
+if TYPE_CHECKING:
+    from tau.tui.buffer import Buffer
+    from tau.tui.geometry import Rect
+
 FREEFORM_LABEL = "Type something…"
 
 
@@ -81,8 +85,20 @@ class _AskUserComponent(Component):
 
     # ── Render ────────────────────────────────────────────────────────────
 
-    def render(self, width: int) -> list[str]:
-        from tau.modes.interactive.components.overlays import _box
+    def render_cells(self, area: Rect, buf: Buffer) -> int:
+        from tau.modes.interactive.components.overlays import _box_cells
+        from tau.tui.ansi_bridge import parse_ansi_wrapped_into
+        from tau.tui.buffer import Buffer as _Buffer
+        from tau.tui.geometry import Rect as _Rect
+
+        inner_w = max(1, area.width - 4)
+
+        def _boxed(lines: list[str]) -> int:
+            inner_buf = _Buffer.empty(_Rect(0, 0, inner_w, 0))
+            row = 0
+            for line in lines:
+                row += parse_ansi_wrapped_into(inner_buf, 0, row, line, inner_w)
+            return _box_cells(buf, area, inner_buf, row, "", None)
 
         inner: list[str] = []
         if self._context:
@@ -120,14 +136,14 @@ class _AskUserComponent(Component):
                 "  \x1b[2mEnter to submit  ·  \\+Enter or Shift+Enter for newline  ·  "
                 f"{back}\x1b[0m"
             )
-            return _box(inner, "", width, bg="")
+            return _boxed(inner)
 
         if self._mode == "freeform":
             inner.append(f"  {self._freeform_value}█")
             inner.append("")
             back = "Esc to cancel" if not self._options else "Esc to go back"
             inner.append(f"  \x1b[2mEnter to submit  ·  {back}\x1b[0m")
-            return _box(inner, "", width, bg="")
+            return _boxed(inner)
 
         for i in range(self._row_count):
             is_freeform_row = i == self._freeform_index
@@ -164,7 +180,7 @@ class _AskUserComponent(Component):
         if self._allow_multiple:
             hints.insert(1, "Space toggle")
         inner.append("  \x1b[2m" + "  ·  ".join(hints) + "\x1b[0m")
-        return _box(inner, "", width)
+        return _boxed(inner)
 
     # ── Input ─────────────────────────────────────────────────────────────
 

{tau_coding_agent-0.5.7 → tau_coding_agent-0.6.0}/tau/builtins/extensions/ask_user/tool.py

@@ -3,8 +3,8 @@ from __future__ import annotations
 import asyncio
 from typing import Any
 
-from component import _AskUserComponent
-from schema import AskUserParams, normalize_options
+from component import _AskUserComponent  # type: ignore[import-not-found]
+from schema import AskUserParams, normalize_options  # type: ignore[import-not-found]
 
 from tau.tool.render import call_line
 from tau.tool.types import (

{tau_coding_agent-0.5.7 → tau_coding_agent-0.6.0}/tau/builtins/extensions/btw/__init__.py

@@ -21,6 +21,8 @@ if TYPE_CHECKING:
     from tau.extensions.api import ExtensionAPI
     from tau.extensions.context import ExtensionContext
     from tau.inference.api.text.service import TextLLM
+    from tau.tui.buffer import Buffer
+    from tau.tui.geometry import Rect
 
 
 # ── Widget component ───────────────────────────────────────────────────────────
@@ -35,11 +37,20 @@ class BtwWidget(Component):
     def set_lines(self, lines: list[str]) -> None:
         self._lines = list(lines)
 
-    def render(self, width: int) -> list[str]:
-        from tau.modes.interactive.components.overlays import _box
+    def render_cells(self, area: Rect, buf: Buffer) -> int:
+        from tau.modes.interactive.components.overlays import _box_cells
+        from tau.tui.ansi_bridge import parse_ansi_wrapped_into
+        from tau.tui.buffer import Buffer as _Buffer
+        from tau.tui.geometry import Rect as _Rect
 
-        inner = list(self._lines) or ["  (empty)"]
-        return _box(inner, "btw", width)
+        inner_w = max(1, area.width - 4)
+        inner = _Buffer.empty(_Rect(0, 0, inner_w, 0))
+        lines = list(self._lines) or ["  (empty)"]
+        row = 0
+        for line in lines:
+            row += parse_ansi_wrapped_into(inner, 0, row, line, inner_w)
+
+        return _box_cells(buf, area, inner, row, "btw", None)
 
     def handle_input(self, event: Any) -> bool:
         return False  # never captures input — editor stays focused

{tau_coding_agent-0.5.7 → tau_coding_agent-0.6.0}/tau/builtins/extensions/footer/git.py

@@ -2,10 +2,20 @@
 
 from __future__ import annotations
 
+from typing import TYPE_CHECKING
+
+from tau.tui.component import Component
+from tau.tui.style import Style
+from tau.tui.text import Line, Span
+
 from .utils import read_branch, shorten_home
 
+if TYPE_CHECKING:
+    from tau.tui.buffer import Buffer
+    from tau.tui.geometry import Rect
 
-class GitBadge:
+
+class GitBadge(Component):
     """Renders ``~/path (branch)`` for the footer Row left slot."""
 
     def __init__(self) -> None:
@@ -16,10 +26,10 @@ class GitBadge:
         display = shorten_home(cwd)
         self._text = f"{display} ({branch})" if branch else display
 
-    def render(self, width: int) -> list[str]:  # noqa: ARG002
-        from tau.tui.utils import DIM, RESET
-
-        return [DIM + self._text + RESET]
+    def render_cells(self, area: Rect, buf: Buffer) -> int:
+        buf.grow_to(area.y + 1)
+        buf.set_line(area.x, area.y, Line([Span(self._text, Style().dim())]), area.width)
+        return 1
 
     def handle_input(self, event: object) -> bool:  # noqa: ARG002
         return False

{tau_coding_agent-0.5.7 → tau_coding_agent-0.6.0}/tau/builtins/extensions/footer/model.py

@@ -2,8 +2,18 @@
 
 from __future__ import annotations
 
+from typing import TYPE_CHECKING
 
-class ModelBadge:
+from tau.tui.component import Component
+from tau.tui.style import Style
+from tau.tui.text import Line, Span
+
+if TYPE_CHECKING:
+    from tau.tui.buffer import Buffer
+    from tau.tui.geometry import Rect
+
+
+class ModelBadge(Component):
     """Renders ``(provider) model ∙ Level|context%`` for the footer Row right slot.
 
     The ``∙ Level`` segment only appears when the active model supports
@@ -73,21 +83,25 @@ class ModelBadge:
         else:
             self.update_context_from_ctx(ctx)
 
-    def render(self, width: int) -> list[str]:  # noqa: ARG002
-        from tau.tui.utils import DIM, RESET
-
+    def _text(self) -> str:
         if not self._provider and not self._model:
-            return []
-
+            return ""
         left = f"({self._provider}) {self._model}" if self._provider else self._model
         if self._thinking and self._thinking_level and self._thinking_level != "off":
             left += f" ∙ {self._thinking_level.title()}"
-
         if self._context_window > 0 and self._tokens > 0:
             pct = self._tokens / self._context_window * 100
             label = f"{pct:.1f}%" if pct < 1 else f"{int(round(pct))}%"
-            return [DIM + f"{left}|{label}" + RESET]
-        return [DIM + left + RESET]
+            return f"{left}|{label}"
+        return left
+
+    def render_cells(self, area: Rect, buf: Buffer) -> int:
+        text = self._text()
+        if not text:
+            return 0
+        buf.grow_to(area.y + 1)
+        buf.set_line(area.x, area.y, Line([Span(text, Style().dim())]), area.width)
+        return 1
 
     def handle_input(self, event: object) -> bool:  # noqa: ARG002
         return False

{tau_coding_agent-0.5.7 → tau_coding_agent-0.6.0}/tau/builtins/extensions/web/__init__.py

@@ -40,9 +40,9 @@ from pathlib import Path
 
 sys.path.insert(0, str(Path(__file__).parent))
 
-from engines import build_engine
-from tools.fetch import WebFetchTool
-from tools.search import WebSearchTool
+from engines import build_engine  # type: ignore[import-not-found]
+from tools.fetch import WebFetchTool  # type: ignore[import-not-found]
+from tools.search import WebSearchTool  # type: ignore[import-not-found]
 
 
 def register(tau) -> None:

{tau_coding_agent-0.5.7 → tau_coding_agent-0.6.0}/tau/builtins/extensions/web/engines/ddgs_engine.py

@@ -27,7 +27,7 @@ class DDGSearchEngine(BaseSearchEngine):
         region, safe = self._region, self._safesearch
 
         if mode is SearchMode.books:
-            from ddgs import DDGS
+            from ddgs import DDGS  # type: ignore[import-not-found]
 
             raw = await asyncio.to_thread(
                 lambda: DDGS().books(query, max_results=max_results) or []
@@ -43,7 +43,7 @@ class DDGSearchEngine(BaseSearchEngine):
                 for r in raw
             ]
 
-        from asyncddgs import aDDGS
+        from asyncddgs import aDDGS  # type: ignore[import-not-found]
 
         async with aDDGS() as d:
             match mode:
@@ -106,7 +106,7 @@ class DDGSearchEngine(BaseSearchEngine):
                     ]
 
     async def fetch(self, url: str, timeout: int) -> str:
-        from ddgs import DDGS
+        from ddgs import DDGS  # type: ignore[import-not-found]
 
         def _fetch() -> str:
             ddgs = DDGS(timeout=timeout)

{tau_coding_agent-0.5.7 → tau_coding_agent-0.6.0}/tau/builtins/extensions/web/engines/exa_engine.py

@@ -20,7 +20,7 @@ class ExaSearchEngine(BaseSearchEngine):
 
     def _get_client(self):
         if self._client is None:
-            from exa_py import Exa
+            from exa_py import Exa  # type: ignore[import-not-found]
 
             self._client = Exa(self._api_key)
         return self._client

{tau_coding_agent-0.5.7 → tau_coding_agent-0.6.0}/tau/builtins/extensions/web/engines/tavily_engine.py

@@ -1,4 +1,6 @@
-"""Tavily engine (LLM-oriented search, https://tavily.com). Requires an API key + ``tavily-python``."""
+"""Tavily engine (LLM-oriented search, https://tavily.com).
+Requires an API key + ``tavily-python``.
+"""
 
 from __future__ import annotations
 
@@ -20,7 +22,7 @@ class TavilySearchEngine(BaseSearchEngine):
 
     def _get_client(self):
         if self._client is None:
-            from tavily import TavilyClient
+            from tavily import TavilyClient  # type: ignore[import-not-found]
 
             self._client = TavilyClient(api_key=self._api_key)
         return self._client

{tau_coding_agent-0.5.7 → tau_coding_agent-0.6.0}/tau/builtins/extensions/web/tools/fetch.py

@@ -71,8 +71,12 @@ def _human_size(n: int) -> str:
 def _render_web_fetch(content: str, opts: Any) -> list[str]:
     # Style via the theme passed on the render options — the stable surface for
     # extensions — rather than importing ANSI codes from Tau internals.
-    _id = lambda s: s  # noqa: E731 — fallback when no theme (e.g. outside the TUI)
-    error = getattr(opts.theme, "error", _id)
+    from tau.tui.style import Style, apply_style
+
+    error_style = getattr(opts.theme, "error", Style())
+
+    def error(text: str) -> str:
+        return apply_style(error_style, text)
 
     if opts.is_error:
         return [error(content.strip())]

{tau_coding_agent-0.5.7 → tau_coding_agent-0.6.0}/tau/builtins/extensions/web/tools/search.py

@@ -2,8 +2,8 @@ from __future__ import annotations
 
 from typing import Any
 
-from engines import BaseSearchEngine
-from engines import SearchMode as _SearchMode
+from engines import BaseSearchEngine  # type: ignore[import-not-found]
+from engines import SearchMode as _SearchMode  # type: ignore[import-not-found]
 from pydantic import BaseModel, Field
 
 from tau.tool.render import call_line
@@ -42,7 +42,7 @@ class _WebSearchSchema(BaseModel):
     )
     max_results: int = Field(
         default=10,
-        description="Number of results to return (default 10). Increase to 20+ for broader coverage.",
+        description=("Results to return (default 10). Increase to 20+ for broader coverage."),
         examples=[10, 20],
     )
 
@@ -70,14 +70,18 @@ def _result_lines(r: dict, mode: _SearchMode) -> tuple[str, str]:
 def _render_web_search(content: str, opts: Any) -> list[str]:
     # Style via the theme on the render options (stable extension surface)
     # instead of importing ANSI codes from Tau internals.
-    _id = lambda s: s  # noqa: E731 — fallback when no theme (e.g. outside the TUI)
-    muted = getattr(opts.theme, "muted", _id)
+    from tau.tui.style import Style, apply_style
+
+    muted_style = getattr(opts.theme, "muted", Style())
+
+    def muted(text: str) -> str:
+        return apply_style(muted_style, text)
 
     metadata = opts.metadata or {}
-    query = metadata.get("query", "")
     mode = metadata.get("mode", "text")
     result_count = metadata.get("result_count", 0)
     results = metadata.get("results", [])
+    results = metadata.get("results", [])
 
     mode_tag = f"  {muted(mode)}" if mode != "text" else ""
     result_word = "result" if result_count == 1 else "results"
@@ -132,7 +136,10 @@ class WebSearchTool(Tool):
             render_result=_render_web_search,
             render_call=_render_web_search_call,
             render_shell="default",
-            prompt_guidelines="Use for current information not in the codebase. Follow up with web_fetch to read a full page from the results.",
+            prompt_guidelines=(
+                "Use for current information not in the codebase. "
+                "Follow up with web_fetch to read a full page from the results."
+            ),
         )
 
     async def execute(
@@ -150,10 +157,11 @@ class WebSearchTool(Tool):
         max_results = invocation.params.get("max_results", 10)
 
         if not self._engine.supports(mode):
+            modes_list = ", ".join(sorted(m.value for m in self._engine.supported_modes))
             return ToolResult.error(
                 invocation.id,
                 f"The '{self._engine.name}' engine does not support '{mode}' mode. "
-                f"Supported modes: {', '.join(sorted(m.value for m in self._engine.supported_modes))}.",
+                f"Supported modes: {modes_list}.",
             )
 
         try:
@@ -171,8 +179,14 @@ class WebSearchTool(Tool):
         }
 
         if not results:
-            return ToolResult.ok(invocation.id, f"No results found for: {query}", metadata=metadata)
+            return ToolResult.ok(
+                invocation.id,
+                f"No results found for: {query}",
+                metadata=metadata,
+            )
 
         return ToolResult.ok(
-            invocation.id, _format_results(mode, query, results), metadata=metadata
+            invocation.id,
+            _format_results(mode, query, results),
+            metadata=metadata,
         )