wrencode 0.3.2__tar.gz → 0.3.3__tar.gz

{wrencode-0.3.2 → wrencode-0.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: wrencode
-Version: 0.3.2
+Version: 0.3.3
 Summary: A minimal agent harness for coding: a readable agent loop and the modules it calls
 Project-URL: Homepage, https://github.com/almostly/wrencode
 Project-URL: Repository, https://github.com/almostly/wrencode
@@ -759,7 +759,9 @@ What a session looks like, and the keys that drive it.
   accent blue on the assistant's dot, the prompt and the banner. The background
   is read from `COLORFGBG` or asked of the terminal itself (most answer); when
   neither says, the first run asks you once and `/theme light|dark|auto`
-  changes the saved answer. `WRENCODE_THEME=light` or `dark` forces one, `ansi` draws with the terminal's
+  changes the saved answer. Colors are 24-bit where the terminal takes them (`COLORTERM`, or iTerm2,
+  WezTerm, Ghostty and VS Code's terminal) and the nearest of 256 elsewhere;
+  `WRENCODE_TRUECOLOR=1|0` overrides. `WRENCODE_THEME=light` or `dark` forces one, `ansi` draws with the terminal's
   own colors instead, and a [Zed](https://zed.dev) theme file
   (`~/.config/zed/themes/mine.json#Mine Light`) replaces the palette with its
   own.

{wrencode-0.3.2 → wrencode-0.3.3}/README.md

@@ -730,7 +730,9 @@ What a session looks like, and the keys that drive it.
   accent blue on the assistant's dot, the prompt and the banner. The background
   is read from `COLORFGBG` or asked of the terminal itself (most answer); when
   neither says, the first run asks you once and `/theme light|dark|auto`
-  changes the saved answer. `WRENCODE_THEME=light` or `dark` forces one, `ansi` draws with the terminal's
+  changes the saved answer. Colors are 24-bit where the terminal takes them (`COLORTERM`, or iTerm2,
+  WezTerm, Ghostty and VS Code's terminal) and the nearest of 256 elsewhere;
+  `WRENCODE_TRUECOLOR=1|0` overrides. `WRENCODE_THEME=light` or `dark` forces one, `ansi` draws with the terminal's
   own colors instead, and a [Zed](https://zed.dev) theme file
   (`~/.config/zed/themes/mine.json#Mine Light`) replaces the palette with its
   own.

{wrencode-0.3.2 → wrencode-0.3.3}/pyproject.toml

@@ -61,7 +61,7 @@ include = ["src/wrencode", "README.md"]
 
 [tool.commitizen]
 name = "cz_conventional_commits"
-version = "0.3.2"
+version = "0.3.3"
 version_scheme = "pep440"
 tag_format = "$version"
 legacy_tag_formats = ["v$version"]

{wrencode-0.3.2 → wrencode-0.3.3}/src/wrencode/__init__.py

@@ -4,4 +4,4 @@
 with `wrencode` (the installed command) or `python -m wrencode`.
 """
 
-__version__ = "0.3.2"
+__version__ = "0.3.3"

{wrencode-0.3.2 → wrencode-0.3.3}/src/wrencode/backends.py

@@ -337,6 +337,27 @@ WEB_SEARCH = os.environ.get("WRENCODE_WEB_SEARCH", "1").lower() not in (
     "no",
 )
 WEB_SEARCH_TOOL: dict[str, Any] = {"type": "web_search_20260209", "name": "web_search"}
+# The models that take the current search tool (with dynamic filtering); the
+# others, Haiku among them, refuse it and take the basic one.
+_WEB_SEARCH_CURRENT = (
+    "opus-5",
+    "opus-4-8",
+    "opus-4-7",
+    "opus-4-6",
+    "sonnet-5",
+    "sonnet-4-6",
+    "fable-5",
+)
+
+
+def web_search_tool(model: str) -> dict[str, Any]:
+    """The server-side web search tool definition `model` accepts."""
+    m = model.lower().replace(".", "-")
+    if any(tag in m for tag in _WEB_SEARCH_CURRENT):
+        return dict(WEB_SEARCH_TOOL)
+    return {"type": "web_search_20250305", "name": "web_search"}
+
+
 WEB_SEARCH_USD = 0.01  # $10 per 1,000 searches
 
 
@@ -1884,7 +1905,7 @@ def get_response(
             defs[-1] = {**defs[-1], "cache_control": {"type": "ephemeral"}}
         if WEB_SEARCH and BACKEND == "anthropic" and tools:
             defs = [
-                dict(WEB_SEARCH_TOOL),
+                web_search_tool(MODEL),
                 *defs,
             ]  # first, so the cache marker stays last
         payload = {

{wrencode-0.3.2 → wrencode-0.3.3}/src/wrencode/ui.py

@@ -15,7 +15,7 @@ import select
 import sys
 import threading
 import time
-from collections.abc import Callable, Generator
+from collections.abc import Callable, Generator, Mapping
 from dataclasses import dataclass
 from typing import Any
 
@@ -186,13 +186,50 @@ def ansi_palette(theme: str) -> Palette:
     )
 
 
+def supports_truecolor(env: Mapping[str, str] | None = None) -> bool:
+    """Whether the terminal takes 24-bit colors: COLORTERM says so, or it is a
+    terminal known to (iTerm2, WezTerm, Ghostty, kitty, VS Code's), or
+    WRENCODE_TRUECOLOR=1 says it does. Terminal.app and a tmux without
+    truecolor configured do not, and get the nearest of 256 colors instead."""
+    env = os.environ if env is None else env
+    if env.get("WRENCODE_TRUECOLOR", "").lower() in ("1", "true", "yes"):
+        return True
+    if env.get("WRENCODE_TRUECOLOR", "").lower() in ("0", "false", "no"):
+        return False
+    if env.get("COLORTERM", "").lower() in ("truecolor", "24bit"):
+        return True
+    return env.get("TERM_PROGRAM", "") in ("iTerm.app", "WezTerm", "ghostty", "vscode")
+
+
+TRUECOLOR = supports_truecolor()
+
+
+def nearest_256(r: int, g: int, b: int) -> int:
+    """The index of the closest color in the xterm 256-color table."""
+    steps = (0, 95, 135, 175, 215, 255)
+
+    def near(v: int) -> int:
+        return min(range(6), key=lambda i: abs(steps[i] - v))
+
+    cr, cg, cb = near(r), near(g), near(b)
+    cube = 16 + 36 * cr + 6 * cg + cb
+    cube_dist = (steps[cr] - r) ** 2 + (steps[cg] - g) ** 2 + (steps[cb] - b) ** 2
+    grey_level = max(0, min(23, round((((r + g + b) / 3) - 8) / 10)))
+    grey = 8 + 10 * grey_level
+    grey_dist = (grey - r) ** 2 + (grey - g) ** 2 + (grey - b) ** 2
+    return 232 + grey_level if grey_dist < cube_dist else cube
+
+
 def truecolor(hex_color: str) -> str:
-    """An SGR for a "#rrggbb" or "#rrggbbaa" color (the alpha is ignored)."""
+    """An SGR for a "#rrggbb" or "#rrggbbaa" color (the alpha is ignored):
+    24-bit where the terminal takes it, else the nearest of 256 colors."""
     h = hex_color.strip().lstrip("#")
     if len(h) not in (6, 8) or any(c not in "0123456789abcdefABCDEF" for c in h):
         raise ValueError(f"not a color: {hex_color!r}")
     r, g, b = (int(h[i : i + 2], 16) for i in (0, 2, 4))
-    return f"\033[38;2;{r};{g};{b}m"
+    if TRUECOLOR:
+        return f"\033[38;2;{r};{g};{b}m"
+    return f"\033[38;5;{nearest_256(r, g, b)}m"
 
 
 def palette_from_zed(style: dict[str, Any]) -> Palette: