"""CLI presentation -- spinner, kawaii faces, tool preview formatting. Pure display functions with no AIAgent dependency; used for CLI feedback. """ import logging import os import re import sys import threading import time from dataclasses import dataclass, field from difflib import unified_diff from pathlib import Path from typing import Any from urllib.parse import urlsplit from utils import safe_json_loads from agent.redact import redact_sensitive_text from agent.tool_result_classification import file_mutation_result_landed logger = logging.getLogger(__name__) _ANSI_RESET = "\033[0m" _MAX_INLINE_DIFF_FILES = 6 _MAX_INLINE_DIFF_LINES = 80 def _display_url(value: Any) -> str: """Extract a display-only URL without assuming model argument types.""" if isinstance(value, dict): value = value.get("url") or value.get("href") return value.strip() if isinstance(value, str) else "" def _hex_rgb(h: str) -> tuple[int, int, int]: return int(h[1:3], 16), int(h[3:5], 16), int(h[5:7], 16) # Diff colors resolve lazily from the skin engine (light/dark aware) and are # cached after the first resolution. _diff_colors_cached: dict[str, str] | None = None def _diff_ansi() -> dict[str, str]: """Return ANSI escapes for diff display, resolved from the active skin.""" global _diff_colors_cached if _diff_colors_cached is not None: return _diff_colors_cached # Defaults that work on dark terminals dim = "\033[38;2;150;150;150m" file_c = "\033[38;2;180;160;255m" hunk = "\033[38;2;120;120;140m" minus = "\033[38;2;255;255;255;48;2;120;20;20m" plus = "\033[38;2;255;255;255;48;2;20;90;20m" try: from hermes_cli.skin_engine import get_active_skin skin = get_active_skin() def _hex_fg(key: str, fallback_rgb: tuple[int, int, int]) -> str: h = skin.get_color(key, "") r, g, b = _hex_rgb(h) if h and len(h) == 7 and h[0] == "#" else fallback_rgb return f"\033[38;2;{r};{g};{b}m" dim = _hex_fg("banner_dim", (150, 150, 150)) file_c = _hex_fg("session_label", (180, 160, 255)) hunk = _hex_fg("session_border", (120, 120, 140)) # minus/plus use dark-tinted backgrounds derived from ui_error/ui_ok err_h = skin.get_color("ui_error", "#ef5350") ok_h = skin.get_color("ui_ok", "#4caf50") if err_h and len(err_h) == 7: er, eg, eb = _hex_rgb(err_h) minus = f"\033[38;2;255;255;255;48;2;{max(er//2,20)};{max(eg//4,10)};{max(eb//4,10)}m" if ok_h and len(ok_h) == 7: or_, og, ob = _hex_rgb(ok_h) plus = f"\033[38;2;255;255;255;48;2;{max(or_//4,10)};{max(og//2,20)};{max(ob//4,10)}m" except Exception: pass _diff_colors_cached = {"dim": dim, "file": file_c, "hunk": hunk, "minus": minus, "plus": plus} return _diff_colors_cached @dataclass class LocalEditSnapshot: """Pre-tool filesystem snapshot used to render diffs locally after writes.""" paths: list[Path] = field(default_factory=list) before: dict[str, str | None] = field(default_factory=dict) # Configurable tool preview length; set once at startup from display.tool_preview_length. _tool_preview_max_len: int = 0 # 0 = unlimited def set_tool_preview_max_len(n: int) -> None: """Set the global max length for tool call previews. 0 = no limit.""" global _tool_preview_max_len _tool_preview_max_len = max(int(n), 0) if n else 0 def get_tool_preview_max_len() -> int: """Return the configured max preview length (0 = unlimited).""" return _tool_preview_max_len def _get_skin(): """Get the active skin config, or None if not available (lazy import avoids cycles).""" try: from hermes_cli.skin_engine import get_active_skin return get_active_skin() except Exception: return None def get_skin_tool_prefix() -> str: """Get tool output prefix character from active skin.""" skin = _get_skin() return skin.tool_prefix if skin else "┊" def get_tool_emoji(tool_name: str, default: str = "⚡") -> str: """Display emoji for a tool: skin ``tool_emojis`` override, then registry, then *default*.""" skin = _get_skin() if skin and skin.tool_emojis: override = skin.tool_emojis.get(tool_name) if override: return override try: from tools.registry import registry emoji = registry.get_emoji(tool_name, default="") if emoji: return emoji except Exception: pass return default # ========================================================================= # Tool preview (one-line summary of a tool call's primary argument) # ========================================================================= def _oneline(text: str) -> str: """Collapse whitespace (including newlines) to single spaces.""" return " ".join(text.split()) def _truncate_preview(text: str, max_len: int | None) -> str: if max_len and max_len > 0 and len(text) > max_len: if max_len <= 3: return "." * max_len return text[:max_len - 3] + "..." return text @dataclass(frozen=True) class ToolPreview: """A compact tool preview plus presentation facts lost to truncation.""" text: str truncated: bool = False url: str | None = None _SHELL_SILENT_HEADS = {"cd", "pushd", "popd", "export", "set", "unset", "source", ".", "true", "false", ":"} _SHELL_PIPE_TAIL_HEADS = {"head", "tail", "wc", "sort", "uniq"} def _shell_basename(head: str) -> str: return head.rsplit("/", 1)[-1] if head else "" def _split_shell_words(segment: str) -> list[str]: words: list[str] = [] buf: list[str] = [] quote: str | None = None for i, ch in enumerate(segment): if quote: buf.append(ch) if ch == quote and (i == 0 or segment[i - 1] != "\\"): quote = None continue if ch in {"'", '"'}: quote = ch buf.append(ch) continue if ch.isspace(): if buf: words.append("".join(buf)) buf = [] continue buf.append(ch) if buf: words.append("".join(buf)) return words def _strip_shell_pipe_tail(segment: str) -> str: words = _split_shell_words(segment) out: list[str] = [] for i, word in enumerate(words): if word == "|" and _shell_basename(words[i + 1] if i + 1 < len(words) else "") in _SHELL_PIPE_TAIL_HEADS: break out.append(word) return " ".join(out).strip() def _split_shell_compound(command: str) -> list[str]: segments: list[str] = [] buf: list[str] = [] quote: str | None = None i = 0 def _flush() -> None: segment = _strip_shell_pipe_tail("".join(buf).strip()) if segment: segments.append(segment) while i < len(command): ch = command[i] if quote: buf.append(ch) if ch == quote and (i == 0 or command[i - 1] != "\\"): quote = None i += 1 continue if ch in {"'", '"'}: quote = ch buf.append(ch) i += 1 continue op_len = 2 if command.startswith("&&", i) or command.startswith("||", i) else 1 if ch in {";", "\n"} else 0 if op_len: _flush() buf = [] i += op_len continue buf.append(ch) i += 1 _flush() return segments def _shell_head_word(segment: str) -> str: words = _split_shell_words(segment) index = 0 while index < len(words) and re.match(r"^[A-Za-z_]\w*=", words[index]): index += 1 return _shell_basename(words[index] if index < len(words) else "") def _clean_shell_segment(segment: str) -> str: words = _split_shell_words(segment) out: list[str] = [] i = 0 while i < len(words): word = words[i] if re.match(r"^\d*(?:>>?|<)$", word): i += 2 continue if re.match(r"^\d*(?:>&|<&)\d+$", word) or re.match(r"^\d*>&\d+$", word): i += 1 continue out.append(word) i += 1 return " ".join(out).strip() def _is_shell_boundary_echo(segment: str) -> bool: words = _split_shell_words(segment) if _shell_basename(words[0] if words else "") != "echo": return False rest = " ".join(words[1:]) return bool(re.search(r"-{2,}|_exit=|(?:^|\s|=)\$[?{]|PIPESTATUS", rest)) def summarize_shell_command(command: str) -> str: """Compact shell wrapper/plumbing for display while preserving raw command elsewhere.""" original = _oneline(command) if not original: return "" segments = _split_shell_compound(original) if len(segments) <= 1: return _clean_shell_segment(segments[0] if segments else original) or original core: list[str] = [] for segment in segments: cleaned = _clean_shell_segment(segment) head = _shell_head_word(cleaned) if cleaned and head not in _SHELL_SILENT_HEADS and not _is_shell_boundary_echo(cleaned): core.append(cleaned) if not core: return original if len(core) == 1: return core[0] count = len(core) - 1 return f"{core[0]} + {count} {'command' if count == 1 else 'commands'}" def _read_file_line_label(args: dict) -> str: offset = args.get("offset") limit = args.get("limit") if not isinstance(offset, int) or offset <= 0: return "" if not isinstance(limit, int) or limit <= 1: return f"L{offset}" return f"L{offset}-{offset + limit - 1}" def redact_browser_typed_text_for_display(value: Any, typed_text: Any) -> Any: """Replace every occurrence of a secret-looking browser_type value with its redacted form. Backends echo the attempted input in error strings/fallback metadata, so the raw value is swapped for its redacted form before the result reaches logs, callbacks, the model, or chat history. Normal typed text matches no secret pattern and passes through unchanged. Redaction is forced regardless of ``security.redact_secrets``: a typed credential leaking into chat history is a security boundary, not log hygiene. """ if typed_text is None: return value needle = str(typed_text) if needle == "": return value redacted = redact_sensitive_text(needle, force=True) if redacted == needle: return value if isinstance(value, str): return value.replace(needle, redacted) if isinstance(value, dict): return {key: redact_browser_typed_text_for_display(item, typed_text) for key, item in value.items()} if isinstance(value, list): return [redact_browser_typed_text_for_display(item, typed_text) for item in value] if isinstance(value, tuple): return tuple(redact_browser_typed_text_for_display(item, typed_text) for item in value) return value def redact_tool_args_for_display(tool_name: str, args: dict | None) -> dict | None: """Return a copy of tool args safe for logs/progress UI (masks ``browser_type`` secrets).""" if not isinstance(args, dict): return args if tool_name == "browser_type" and isinstance(args.get("text"), str): safe_args = dict(args) safe_args["text"] = redact_sensitive_text(args["text"], force=True) return safe_args return args def _delegate_task_goal_parts(tasks: Any, *, per_goal_len: int) -> tuple[int, list[str]]: if not isinstance(tasks, list): return 0, [] goals: list[str] = [] for task in tasks: if not isinstance(task, dict): continue raw_goal = task.get("goal") goal = "?" if raw_goal is None else _oneline(str(raw_goal)) goals.append(_truncate_preview(goal or "?", per_goal_len)) return len(goals), goals def _browser_exec_step_label(args: dict, max_chars: int = 80) -> str | None: """User-friendly step label from browser_exec code's leading comment.""" code = str(args.get("code", "") or "").strip() if not code: return None first = code.split("\n", 1)[0].strip() if not first.startswith("#"): return None label = first.lstrip("#").strip() if not label: return None if len(label) > max_chars: label = label[: max_chars - 1] + "…" return label _PRIMARY_ARGS = { "terminal": "command", "web_search": "query", "web_extract": "urls", "read_file": "path", "write_file": "path", "patch": "path", "search_files": "pattern", "browser_navigate": "url", "browser_click": "ref", "browser_type": "text", "image_generate": "prompt", "text_to_speech": "text", "vision_analyze": "question", "skill_view": "name", "skills_list": "category", "cronjob_manage": "action", "execute_code": "code", "browser_exec": "code", "delegate_task": "goal", "clarify": "question", "skill_manage": "name", } _FALLBACK_PREVIEW_KEYS = ("query", "text", "command", "path", "name", "prompt", "code", "goal") def _delegate_action_preview(args: dict) -> str | None: """Shared ``list/steer/stop `` preview for delegate_task, or None for spawn calls.""" action = str(args.get("action") or "").strip().lower() if action in ("list", "steer", "stop"): return f"{action} {str(args.get('subagent_id') or '').strip()}".strip() return None def _preview_browser_exec(args: dict, max_len: int) -> str | None: label = _browser_exec_step_label(args) if label is not None: return _truncate_preview(label, max_len) return _truncate_preview(_oneline(str(args.get("code", "") or "")), max_len) or None def _preview_delegate_task(args: dict, max_len: int) -> str | None: action_preview = _delegate_action_preview(args) if action_preview is not None: return _truncate_preview(action_preview, max_len) tasks = args.get("tasks") if tasks and isinstance(tasks, list): task_count, goals = _delegate_task_goal_parts(tasks, per_goal_len=40) preview = f"{task_count} tasks: " + " | ".join(goals) if goals else f"{len(tasks)} parallel tasks" return _truncate_preview(preview, max_len) goal = args.get("goal", "") if goal is None: return None return _truncate_preview(_oneline(str(goal)), max_len) or None def _preview_process_manage(args: dict, _max_len: int) -> str | None: action = args.get("action", "") sid = args.get("session_id", "") data = args.get("data", "") timeout_val = args.get("timeout") parts = [str(action) if action else ""] if sid: parts.append(str(sid)[:16]) if data: parts.append(f'"{_oneline(str(data)[:20])}"') if timeout_val and action == "wait": parts.append(f"{timeout_val}s") parts = [p for p in parts if p] return " ".join(parts) if parts else None def _preview_todo_list(args: dict, _max_len: int) -> str: todos_arg = args.get("todos") if todos_arg is None: return "reading task list" if args.get("merge", False): return f"updating {len(todos_arg)} task(s)" return f"planning {len(todos_arg)} task(s)" def _preview_shell(key: str): def _build(args: dict, max_len: int) -> str | None: command = args.get(key) if command is None: return None return _truncate_preview(summarize_shell_command(str(command)), max_len) or None return _build def _preview_read_file(args: dict, max_len: int) -> str | None: path = args.get("path") or args.get("file") or args.get("filepath") if path is None: return None label = Path(str(path).replace("\\", "/")).name or str(path) return _truncate_preview(f"{label} {_read_file_line_label(args)}".strip(), max_len) or None def _preview_session_search(args: dict, _max_len: int) -> str: query = _oneline(args.get("query", "")) return f"recall: \"{query[:25]}{'...' if len(query) > 25 else ''}\"" def _preview_memory(args: dict, _max_len: int) -> str: action = args.get("action", "") target = args.get("target", "") if action == "add": content = _oneline(args.get("content", "")) return f"+{target}: \"{content[:25]}{'...' if len(content) > 25 else ''}\"" if action in ("replace", "remove"): old = _oneline(args.get("old_text") or "") or "" return f"{'~' if action == 'replace' else '-'}{target}: \"{old[:20]}\"" return action def _preview_send_message(args: dict, _max_len: int) -> str: target = args.get("target", "?") msg = _oneline(args.get("message", "")) if len(msg) > 20: msg = msg[:17] + "..." return f"to {target}: \"{msg}\"" def _preview_skill_view(args: dict, max_len: int) -> str | None: name = _oneline(str(args.get("name") or "")) file_path = args.get("file_path") if file_path: file_path = _oneline(str(file_path)) return _truncate_preview(f"{name} → {file_path}" if name else file_path, max_len) or None return _truncate_preview(name, max_len) or None # Tool-specific preview builders: f(args, max_len) -> preview. Tools not listed # fall through to the primary-argument lookup in build_tool_preview. _PREVIEW_BUILDERS = { "browser_exec": _preview_browser_exec, "delegate_task": _preview_delegate_task, "process_manage": _preview_process_manage, "todo_list": _preview_todo_list, "terminal": _preview_shell("command"), "execute_code": _preview_shell("code"), "read_file": _preview_read_file, "session_search": _preview_session_search, "memory": _preview_memory, "send_message": _preview_send_message, "skill_view": _preview_skill_view, } def build_tool_preview(tool_name: str, args: dict, max_len: int | None = None) -> str | None: """Build a short preview of a tool call's primary argument for display. *max_len* ``None`` defers to the global ``_tool_preview_max_len``; ``0`` means unlimited. """ if max_len is None: max_len = _tool_preview_max_len if not args: return None args = redact_tool_args_for_display(tool_name, args) or args builder = _PREVIEW_BUILDERS.get(tool_name) if builder is not None: return builder(args, max_len) key = _PRIMARY_ARGS.get(tool_name) or next((k for k in _FALLBACK_PREVIEW_KEYS if k in args), None) if not key or key not in args: return None value = args[key] if isinstance(value, list): value = value[0] if value else "" preview = _oneline(str(value)) if not preview: return None if max_len > 0 and len(preview) > max_len: preview = preview[:max_len - 3] + "..." return preview def prepare_tool_preview( tool_name: str, args: dict | None, *, fallback: str, max_len: int, ) -> ToolPreview: """Build one canonical compact preview plus explicit truncation/URL metadata. The uncapped preview is rebuilt from the arguments so an upstream display cap cannot discard its link target; platforms get truncation/URL facts explicitly instead of inferring them from the rendered text. """ full_text = build_tool_preview(tool_name, args, max_len=0) or fallback text = _truncate_preview(full_text, max_len) truncated = text != full_text url = None if truncated: candidate = _display_url(full_text) try: parsed = urlsplit(candidate) except ValueError: parsed = None if parsed and parsed.scheme.lower() in {"http", "https"} and parsed.netloc: url = candidate return ToolPreview(text=text, truncated=truncated, url=url) # ========================================================================= # Friendly tool labels: "web_search " -> "Searching the web for ". # Curated built-ins only — we know each core tool's semantics so the verb is fixed, # not computed; custom/plugin/MCP tools have no entry and fall back to the raw preview. # ========================================================================= _TOOL_VERBS: dict[str, str] = { "web_search": "Searching the web", "web_extract": "Reading", "browser_navigate": "Browsing", "browser_click": "Clicking", "browser_type": "Typing", "read_file": "Reading", "write_file": "Writing", "patch": "Editing", "search_files": "Searching files", "terminal": "Running", "execute_code": "Running code", "image_generate": "Generating image", "video_generate": "Generating video", "text_to_speech": "Generating speech", "vision_analyze": "Looking at the image", "session_search": "Searching past sessions", "skill_view": "Reading skill", "skills_list": "Listing skills", "skill_manage": "Updating skill", "delegate_task": "Delegating", "cronjob_manage": "Scheduling", "clarify": "Asking", "memory": "Updating memory", "todo_list": "Updating tasks", } # Verbs that read better without the argument preview appended. _TOOL_VERBS_NO_PREVIEW: frozenset[str] = frozenset({"skills_list", "session_search"}) # Verbs joined to the preview with " for " (search-style phrasing). _TOOL_VERBS_FOR_CONNECTOR: frozenset[str] = frozenset({"web_search", "search_files"}) _friendly_tool_labels: bool = True def set_friendly_tool_labels(enabled: bool) -> None: """Toggle friendly human-phrased tool labels (display.friendly_tool_labels).""" global _friendly_tool_labels _friendly_tool_labels = bool(enabled) def get_tool_verb(tool_name: str) -> str | None: """Friendly verb for a built-in tool, or None (labels disabled / no curated verb). Callers holding a computed preview compose ``f"{verb}{connector}{preview}"`` themselves via :func:`tool_verb_connector`. """ if not _friendly_tool_labels: return None return _TOOL_VERBS.get(tool_name) def tool_verb_connector(tool_name: str) -> str: """Return the connector between a verb and its preview (" for " or " ").""" return " for " if tool_name in _TOOL_VERBS_FOR_CONNECTOR else " " def verb_drops_preview(tool_name: str) -> bool: """Whether the verb should render alone, without the argument preview.""" return tool_name in _TOOL_VERBS_NO_PREVIEW def build_status_phrase(tool_name: str, args: dict | None, max_len: int = 49) -> str | None: """Lowercase "is …" phrase for platform status lines (e.g. Slack setStatus). Phrased to follow the bot's display name ("Hermes is running …"). ``args=None`` gives a verb-only phrase (``display.live_status: verb`` keeps argument previews out of shared channels). Returns None for ``_thinking`` or when friendly labels are disabled so callers fall back to their static default. Default ``max_len`` stays under Slack's ~50-char status truncation. """ if not tool_name or tool_name == "_thinking" or not _friendly_tool_labels: return None verb = _TOOL_VERBS.get(tool_name) head = f"is {verb[0].lower()}{verb[1:]}" if verb else f"is using {tool_name}" phrase = head if args and verb and tool_name not in _TOOL_VERBS_NO_PREVIEW: preview = build_tool_preview(tool_name, args, max_len=None) if preview: # Previews can contain newlines (terminal commands); keep the first line. preview = preview.splitlines()[0].strip() phrase = f"{head}{tool_verb_connector(tool_name)}{preview}" if len(phrase) > max_len - 1: return phrase[: max_len - 2].rstrip() + "…" return phrase + "…" def build_tool_label(tool_name: str, args: dict, max_len: int | None = None) -> str | None: """Human-phrased label ("Searching the web for ...") for curated built-ins. Custom/plugin/MCP tools (or labels disabled) get the raw preview, so this is a drop-in replacement for :func:`build_tool_preview`. """ verb = _TOOL_VERBS.get(tool_name) if _friendly_tool_labels else None if not verb: return build_tool_preview(tool_name, args, max_len=max_len) if tool_name in _TOOL_VERBS_NO_PREVIEW: return verb preview = build_tool_preview(tool_name, args, max_len=max_len) if not preview: return verb return f"{verb}{tool_verb_connector(tool_name)}{preview}" # ========================================================================= # Inline diff previews for write actions # ========================================================================= def _resolved_path(path: str) -> Path: """Resolve a possibly-relative filesystem path against the current cwd.""" candidate = Path(os.path.expanduser(path)) return candidate if candidate.is_absolute() else Path.cwd() / candidate def _snapshot_text(path: Path) -> str | None: """Return UTF-8 file content, or None for missing/unreadable files.""" try: return path.read_text(encoding="utf-8") except (FileNotFoundError, IsADirectoryError, UnicodeDecodeError, OSError): return None def _display_diff_path(path: Path) -> str: """Prefer cwd-relative paths in diffs when available.""" try: return str(path.resolve().relative_to(Path.cwd().resolve())) except Exception: return str(path) def _resolve_skill_manage_paths(args: dict) -> list[Path]: """Resolve skill_manage write targets to filesystem paths.""" action = args.get("action") name = args.get("name") if not action or not name: return [] from tools.skill_manager_tool import _find_skill, _resolve_skill_dir if action == "create": return [_resolve_skill_dir(name, args.get("category")) / "SKILL.md"] existing = _find_skill(name) if not existing: return [] skill_dir = Path(existing["path"]) file_path = args.get("file_path") if action in {"edit", "patch"}: return [skill_dir / file_path] if file_path else [skill_dir / "SKILL.md"] if action in {"write_file", "remove_file"}: return [skill_dir / file_path] if file_path else [] if action == "delete": return [path for path in sorted(skill_dir.rglob("*")) if path.is_file()] return [] def _resolve_local_edit_paths(tool_name: str, function_args: dict | None) -> list[Path]: """Resolve local filesystem targets for write-capable tools.""" if not isinstance(function_args, dict): return [] if tool_name in {"write_file", "patch"}: path = function_args.get("path") return [_resolved_path(path)] if path else [] if tool_name == "skill_manage": return _resolve_skill_manage_paths(function_args) return [] def capture_local_edit_snapshot(tool_name: str, function_args: dict | None) -> LocalEditSnapshot | None: """Capture before-state for local write previews.""" paths = _resolve_local_edit_paths(tool_name, function_args) if not paths: return None return LocalEditSnapshot(paths=paths, before={str(path): _snapshot_text(path) for path in paths}) def _result_succeeded(result: str | None) -> bool: """Conservatively detect whether a tool result represents success.""" if not result: return False data = safe_json_loads(result) if not isinstance(data, dict) or data.get("error"): return False if "success" in data: return bool(data.get("success")) return True def _diff_from_snapshot(snapshot: LocalEditSnapshot | None) -> str | None: """Generate unified diff text from a stored before-state and current files.""" if not snapshot: return None chunks: list[str] = [] for path in snapshot.paths: before = snapshot.before.get(str(path)) after = _snapshot_text(path) if before == after: continue display_path = _display_diff_path(path) diff = "".join( unified_diff( [] if before is None else before.splitlines(keepends=True), [] if after is None else after.splitlines(keepends=True), fromfile=f"a/{display_path}", tofile=f"b/{display_path}", ) ) if diff: chunks.append(diff) if not chunks: return None return "".join(chunk if chunk.endswith("\n") else chunk + "\n" for chunk in chunks) def extract_edit_diff( tool_name: str, result: str | None, *, function_args: dict | None = None, snapshot: LocalEditSnapshot | None = None, ) -> str | None: """Extract a unified diff from a file-edit tool result.""" if tool_name == "patch" and result: data = safe_json_loads(result) if isinstance(data, dict): diff = data.get("diff") if isinstance(diff, str) and diff.strip(): return diff if tool_name not in {"write_file", "patch", "skill_manage"} or not _result_succeeded(result): return None return _diff_from_snapshot(snapshot) def _emit_inline_diff(diff_text: str, print_fn) -> bool: """Emit rendered diff text through the CLI's prompt_toolkit-safe printer.""" if print_fn is None or not diff_text: return False try: print_fn(" ┊ review diff") for line in diff_text.rstrip("\n").splitlines(): print_fn(line) return True except Exception: return False # Unified-diff line prefix -> diff color key (checked in order; "--- "/"+++ " handled first). _DIFF_LINE_COLORS = (("@@", "hunk"), ("-", "minus"), ("+", "plus"), (" ", "dim")) def _render_inline_unified_diff(diff: str) -> list[str]: """Render unified diff lines in Hermes' inline transcript style.""" rendered: list[str] = [] from_file = None to_file = None for raw_line in diff.splitlines(): if raw_line.startswith("--- "): from_file = raw_line[4:].strip() continue if raw_line.startswith("+++ "): to_file = raw_line[4:].strip() if from_file or to_file: rendered.append(f"{_diff_ansi()['file']}{from_file or 'a/?'} → {to_file or 'b/?'}{_ANSI_RESET}") continue for prefix, color in _DIFF_LINE_COLORS: if raw_line.startswith(prefix): rendered.append(f"{_diff_ansi()[color]}{raw_line}{_ANSI_RESET}") break else: if raw_line: rendered.append(raw_line) return rendered def _split_unified_diff_sections(diff: str) -> list[str]: """Split a unified diff into per-file sections.""" sections: list[list[str]] = [] current: list[str] = [] for line in diff.splitlines(): if line.startswith("--- ") and current: sections.append(current) current = [line] continue current.append(line) if current: sections.append(current) return ["\n".join(section) for section in sections if section] def _summarize_rendered_diff_sections( diff: str, *, max_files: int = _MAX_INLINE_DIFF_FILES, max_lines: int = _MAX_INLINE_DIFF_LINES, ) -> list[str]: """Render diff sections while capping file count and total line count.""" sections = _split_unified_diff_sections(diff) rendered: list[str] = [] omitted_files = 0 omitted_lines = 0 for idx, section in enumerate(sections): section_lines = _render_inline_unified_diff(section) remaining_budget = max_lines - len(rendered) if idx >= max_files or remaining_budget <= 0: omitted_files += 1 omitted_lines += len(section_lines) continue if len(section_lines) <= remaining_budget: rendered.extend(section_lines) continue rendered.extend(section_lines[:remaining_budget]) omitted_lines += len(section_lines) - remaining_budget omitted_files += 1 + max(0, len(sections) - idx - 1) for leftover in sections[idx + 1:]: omitted_lines += len(_render_inline_unified_diff(leftover)) break if omitted_files or omitted_lines: summary = f"… omitted {omitted_lines} diff line(s)" if omitted_files: summary += f" across {omitted_files} additional file(s)/section(s)" rendered.append(f"{_diff_ansi()['hunk']}{summary}{_ANSI_RESET}") return rendered def render_edit_diff_with_delta( tool_name: str, result: str | None, *, function_args: dict | None = None, snapshot: LocalEditSnapshot | None = None, print_fn=None, ) -> bool: """Render an edit diff inline without taking over the terminal UI.""" diff = extract_edit_diff(tool_name, result, function_args=function_args, snapshot=snapshot) if not diff: return False try: rendered_lines = _summarize_rendered_diff_sections(diff) except Exception as exc: logger.debug("Could not render inline diff: %s", exc) return False return _emit_inline_diff("\n".join(rendered_lines), print_fn) # ========================================================================= # KawaiiSpinner # ========================================================================= class KawaiiSpinner: """Animated spinner with kawaii faces for CLI feedback during tool execution.""" SPINNERS = { 'dots': ['⠋', '⠙', '⠹', '⠸', '⠼', '⠴', '⠦', '⠧', '⠇', '⠏'], 'bounce': ['⠁', '⠂', '⠄', '⡀', '⢀', '⠠', '⠐', '⠈'], 'grow': ['▁', '▂', '▃', '▄', '▅', '▆', '▇', '█', '▇', '▆', '▅', '▄', '▃', '▂'], 'arrows': ['←', '↖', '↑', '↗', '→', '↘', '↓', '↙'], 'star': ['✶', '✷', '✸', '✹', '✺', '✹', '✸', '✷'], 'moon': ['🌑', '🌒', '🌓', '🌔', '🌕', '🌖', '🌗', '🌘'], 'pulse': ['◜', '◠', '◝', '◞', '◡', '◟'], 'brain': ['🧠', '💭', '💡', '✨', '💫', '🌟', '💡', '💭'], 'sparkle': ['⁺', '˚', '*', '✧', '✦', '✧', '*', '˚'], } KAWAII_WAITING = [ "(。◕‿◕。)", "(◕‿◕✿)", "٩(◕‿◕。)۶", "(✿◠‿◠)", "( ˘▽˘)っ", "♪(´ε` )", "(◕ᴗ◕✿)", "ヾ(^∇^)", "(≧◡≦)", "(★ω★)", ] KAWAII_THINKING = [ "(。•́︿•̀。)", "(◔_◔)", "(¬‿¬)", "( •_•)>⌐■-■", "(⌐■_■)", "(´・_・`)", "◉_◉", "(°ロ°)", "( ˘⌣˘)♡", "ヽ(>∀<☆)☆", "٩(๑❛ᴗ❛๑)۶", "(⊙_⊙)", "(¬_¬)", "( ͡° ͜ʖ ͡°)", "ಠ_ಠ", ] THINKING_VERBS = [ "pondering", "contemplating", "musing", "cogitating", "ruminating", "deliberating", "mulling", "reflecting", "processing", "reasoning", "analyzing", "computing", "synthesizing", "formulating", "brainstorming", ] @staticmethod def _skin_spinner_list(key: str, fallback: list) -> list: """Return the active skin's ``spinner[key]`` list, or *fallback* when absent/empty.""" try: skin = _get_skin() if skin: values = skin.spinner.get(key, []) if values: return values except Exception: pass return fallback @classmethod def get_waiting_faces(cls) -> list: return cls._skin_spinner_list("waiting_faces", cls.KAWAII_WAITING) @classmethod def get_thinking_faces(cls) -> list: return cls._skin_spinner_list("thinking_faces", cls.KAWAII_THINKING) @classmethod def get_thinking_verbs(cls) -> list: return cls._skin_spinner_list("thinking_verbs", cls.THINKING_VERBS) def __init__(self, message: str = "", spinner_type: str = 'dots', print_fn=None): self.message = message self.spinner_frames = self.SPINNERS.get(spinner_type, self.SPINNERS['dots']) self.running = False self.thread = None self.frame_idx = 0 self.start_time = None self.last_line_len = 0 # When set, all output bypasses self._out so silenced agents stay silent. self._print_fn = print_fn # Capture stdout NOW, before any child redirect_stdout(devnull) replaces it. self._out = sys.stdout def _write(self, text: str, end: str = '\n', flush: bool = False): """Write via print_fn when supplied, else to the stdout captured at creation.""" if self._print_fn is not None: try: self._print_fn(text) except Exception: pass return try: self._out.write(text + end) if flush: self._out.flush() except (ValueError, OSError): pass @property def _is_tty(self) -> bool: """Check if output is a real terminal, safe against closed streams.""" try: return hasattr(self._out, 'isatty') and self._out.isatty() except (ValueError, OSError): return False def _is_patch_stdout_proxy(self) -> bool: """True when stdout is prompt_toolkit's StdoutProxy. StdoutProxy queues writes and injects newlines around each flush, so the \\r overwrite never lands: each frame would land on its own line. The CLI drives its own TUI spinner widget in that mode, so we stay silent. """ try: from prompt_toolkit.patch_stdout import StdoutProxy return isinstance(self._out, StdoutProxy) except ImportError: return False def _animate(self): # Non-TTY (Docker, systemd, pipe): log once instead of spamming frames. if not self._is_tty: self._write(f" [tool] {self.message}", flush=True) while self.running: time.sleep(0.5) return # Under patch_stdout the \r animation would overdraw the TUI status bar. if self._is_patch_stdout_proxy(): while self.running: time.sleep(0.1) return skin = _get_skin() wings = skin.get_spinner_wings() if skin else [] while self.running: if os.getenv("HERMES_SPINNER_PAUSE"): time.sleep(0.1) continue frame = self.spinner_frames[self.frame_idx % len(self.spinner_frames)] elapsed = time.time() - self.start_time if wings: left, right = wings[self.frame_idx % len(wings)] line = f" {left} {frame} {self.message} {right} ({elapsed:.1f}s)" else: line = f" {frame} {self.message} ({elapsed:.1f}s)" pad = max(self.last_line_len - len(line), 0) self._write(f"\r{line}{' ' * pad}", end='', flush=True) self.last_line_len = len(line) self.frame_idx += 1 time.sleep(0.12) def start(self): if self.running: return self.running = True self.start_time = time.time() self.thread = threading.Thread(target=self._animate, daemon=True) self.thread.start() def update_text(self, new_message: str): self.message = new_message def _clear_line_blanks(self) -> str: # Clear with spaces (not \033[K) to avoid garbled escapes under patch_stdout. return ' ' * max(self.last_line_len + 5, 40) def print_above(self, text: str): """Print a line above the spinner; the next tick redraws the spinner below it. Works inside redirect_stdout(devnull) because _write targets the stdout captured at spinner creation, not the current sys.stdout. """ if not self.running: self._write(f" {text}", flush=True) return self._write(f"\r{self._clear_line_blanks()}\r {text}", flush=True) def stop(self, final_message: str = None): self.running = False if self.thread: self.thread.join(timeout=0.5) is_tty = self._is_tty if is_tty: self._write(f"\r{self._clear_line_blanks()}\r", end='', flush=True) if final_message: elapsed = f" ({time.time() - self.start_time:.1f}s)" if self.start_time else "" if is_tty: self._write(f" {final_message}", flush=True) else: self._write(f" [done] {final_message}{elapsed}", flush=True) def __enter__(self): self.start() return self def __exit__(self, *exc): self.stop() return False # ========================================================================= # Cute tool message (completion line that replaces the spinner) # ========================================================================= _ERROR_SUFFIX_MAX_LEN = 48 def _trim_error(msg: str) -> str: """Shrink an error message for inline display (long 'File not found' paths -> filename).""" msg = msg.strip() if "File not found:" in msg: _, _, tail = msg.partition("File not found:") tail = tail.strip() if "/" in tail: msg = f"File not found: {tail.rsplit('/', 1)[-1]}" if len(msg) > _ERROR_SUFFIX_MAX_LEN: msg = msg[: _ERROR_SUFFIX_MAX_LEN - 3] + "..." return msg def _detect_tool_failure(tool_name: str, result: str | None) -> tuple[bool, str]: """Return ``(is_failure, suffix)`` for a tool result, e.g. ``(True, " [exit 1]")``.""" if result is None or file_mutation_result_landed(tool_name, result): return False, "" data = safe_json_loads(result) # Terminal: non-zero exit code is the canonical failure signal. if tool_name == "terminal": if isinstance(data, dict): exit_code = data.get("exit_code") if exit_code is not None and exit_code != 0: err_msg = data.get("error") if err_msg: return True, f" [{_trim_error(str(err_msg))}]" return True, f" [exit {exit_code}]" return False, "" if isinstance(data, dict): # Memory: distinguish "store full" from real errors. if ( tool_name == "memory" and data.get("success") is False and "exceed the limit" in data.get("error", "") ): return True, " [full]" err = data.get("error") or data.get("message") if err and (data.get("success") is False or "error" in data): return True, f" [{_trim_error(str(err))}]" # Multimodal results (dicts) are successes; failures arrive as JSON-encoded strings. if not isinstance(result, str): return False, "" lower = result[:500].lower() if '"error"' in lower or '"failed"' in lower or result.startswith("Error"): return True, " [error]" return False, "" def _domain(url: str) -> str: return url.replace("https://", "").replace("http://", "").split("/")[0] def _cute_trunc(s) -> str: """Tail-truncate to the configured preview cap (0 = unlimited).""" s = str(s) limit = _tool_preview_max_len if limit == 0: return s return (s[:limit-3] + "...") if len(s) > limit else s def _cute_path(p) -> str: """Head-truncate a path to the configured preview cap, keeping the filename end.""" p = str(p) limit = _tool_preview_max_len if limit == 0: return p return ("..." + p[-(limit-3):]) if len(p) > limit else p def _cute_web_extract(a: dict, _r) -> str: urls = a.get("urls", []) if urls: url = _display_url(urls[0] if isinstance(urls, list) else urls) if url: extra = f" +{len(urls)-1}" if isinstance(urls, list) and len(urls) > 1 else "" return f"┊ 📄 fetch {_cute_trunc(_domain(url))}{extra}" return "┊ 📄 fetch pages" def _cute_todo_list(a: dict, result) -> str: todos_arg = a.get("todos") total = done = 0 if result: try: data = safe_json_loads(result) if data: s = data.get("summary", {}) total = s.get("total", 0) done = s.get("completed", 0) except Exception: pass if todos_arg is None: detail = f"{done}/{total} task(s)" if total > 0 else "reading tasks" elif a.get("merge", False): detail = f"update {done}/{total} ✓" if total > 0 and done > 0 else f"update {len(todos_arg)} task(s)" else: detail = f"{done}/{total} task(s)" if total > 0 and done > 0 else f"{len(todos_arg)} task(s)" return f"┊ 📋 plan {detail}" def _cute_memory(a: dict, _r) -> str: action = a.get("action", "?") target = a.get("target", "") if action == "add": return f"┊ 🧠 memory +{target}: \"{_cute_trunc(a.get('content', ''))}\"" if action in ("replace", "remove"): old = a.get("old_text") or "" return f"┊ 🧠 memory {'~' if action == 'replace' else '-'}{target}: \"{_cute_trunc(old)}\"" return f"┊ 🧠 memory {action}" def _cute_skill_view(a: dict, _r) -> str: label = a.get("name", "") file_path = a.get("file_path") if file_path: label = f"{label} → {file_path}" if label else str(file_path) return f"┊ 📚 skill {_cute_trunc(label)}" def _cute_cronjob(a: dict, _r) -> str: action = a.get("action", "?") if action == "create": skills = a.get("skills") or ([] if not a.get("skill") else [a.get("skill")]) label = a.get("name") or (skills[0] if skills else None) or a.get("prompt", "task") return f"┊ ⏰ cron create {_cute_trunc(label)}" if action == "list": return "┊ ⏰ cron listing" return f"┊ ⏰ cron {action} {a.get('job_id', '')}" def _cute_execute_code(a: dict, _r) -> str: code = a.get("code", "") first_line = code.strip().split("\n")[0] if code.strip() else "" return f"┊ 🐍 exec {_cute_trunc(first_line)}" def _cute_browser_exec(a: dict, _r) -> str: # Leading `# …` comment becomes the step label; code stays collapsed behind the preview cap. label = _browser_exec_step_label(a) if label is not None: return f"┊ 🌐 browser {label}" return f"┊ 🌐 browser {_cute_trunc(' '.join(str(a.get('code', '') or '').split()))}" def _cute_delegate(a: dict, _r) -> str: action_preview = _delegate_action_preview(a) if action_preview is not None: return f"┊ 🔀 delegate {_cute_trunc(action_preview)}" tasks = a.get("tasks") if tasks and isinstance(tasks, list): task_count, goals = _delegate_task_goal_parts(tasks, per_goal_len=30) detail = " | ".join(goals) if goals else "parallel" return f"┊ 🔀 delegate {task_count or len(tasks)}x: {_cute_trunc(detail)}" return f"┊ 🔀 delegate {_cute_trunc(a.get('goal', ''))}" def _cute_process_manage(a: dict, _r) -> str: action = a.get("action", "?") sid = a.get("session_id", "")[:12] return f"┊ ⚙️ proc {'ls processes' if action == 'list' else f'{action} {sid}'}" _SCROLL_ARROWS = {"down": "↓", "up": "↑", "right": "→", "left": "←"} # Completion-line renderers: tool -> f(args, result) -> "┊ {emoji} {verb:9} {detail}" (duration appended by caller). _CUTE_LINES = { "web_search": lambda a, r: f"┊ 🔍 search {_cute_trunc(a.get('query', ''))}", "web_extract": _cute_web_extract, "terminal": lambda a, r: f"┊ 💻 $ {_cute_trunc(build_tool_preview('terminal', a) or a.get('command', ''))}", "process_manage": _cute_process_manage, "read_file": lambda a, r: f"┊ 📖 read {_cute_trunc(build_tool_preview('read_file', a) or a.get('path', ''))}", "write_file": lambda a, r: f"┊ ✍️ write {_cute_path(a.get('path', ''))}", "patch": lambda a, r: f"┊ 🔧 patch {_cute_path(a.get('path', ''))}", "search_files": lambda a, r: ( f"┊ 🔎 {'find' if a.get('target', 'content') == 'files' else 'grep':9} {_cute_trunc(a.get('pattern', ''))}" ), "browser_navigate": lambda a, r: f"┊ 🌐 navigate {_cute_trunc(_domain(a.get('url', '')))}", "browser_snapshot": lambda a, r: f"┊ 📸 snapshot {'full' if a.get('full') else 'compact'}", "browser_click": lambda a, r: f"┊ 👆 click {a.get('ref', '?')}", "browser_type": lambda a, r: f"┊ ⌨️ type \"{_cute_trunc(a.get('text', ''))}\"", "browser_scroll": lambda a, r: ( f"┊ {_SCROLL_ARROWS.get(a.get('direction', 'down'), '↓')} scroll {a.get('direction', 'down')}" ), "browser_back": lambda a, r: "┊ ◀️ back ", "browser_press": lambda a, r: f"┊ ⌨️ press {a.get('key', '?')}", "browser_get_images": lambda a, r: "┊ 🖼️ images extracting", "browser_vision": lambda a, r: "┊ 👁️ vision analyzing page", "todo_list": _cute_todo_list, "session_search": lambda a, r: f"┊ 🔍 recall \"{_cute_trunc(a.get('query', ''))}\"", "memory": _cute_memory, "skills_list": lambda a, r: f"┊ 📚 skills list {a.get('category', 'all')}", "skill_view": _cute_skill_view, "image_generate": lambda a, r: f"┊ 🎨 create {_cute_trunc(a.get('prompt', ''))}", "text_to_speech": lambda a, r: f"┊ 🔊 speak {_cute_trunc(a.get('text', ''))}", "vision_analyze": lambda a, r: f"┊ 👁️ vision {_cute_trunc(a.get('question', ''))}", "send_message": lambda a, r: f"┊ 📨 send {a.get('target', '?')}: \"{_cute_trunc(a.get('message', ''))}\"", "cronjob_manage": _cute_cronjob, "execute_code": _cute_execute_code, "browser_exec": _cute_browser_exec, "delegate_task": _cute_delegate, } def _get_cute_tool_message( tool_name: str, args: dict, duration: float, result: str | None = None, ) -> str: """Formatted tool completion line for CLI quiet mode: ``| {emoji} {verb:9} {detail} {duration}``. Failed tool calls get an informational suffix from :func:`_detect_tool_failure`; the leading ``┊`` is swapped for the active skin's tool prefix. """ args = redact_tool_args_for_display(tool_name, args) or args is_failure, failure_suffix = _detect_tool_failure(tool_name, result) skin_prefix = get_skin_tool_prefix() render = _CUTE_LINES.get(tool_name) if render is not None: body = render(args, result) else: body = f"┊ ⚡ {tool_name[:9]:9} {_cute_trunc(build_tool_preview(tool_name, args) or '')}" line = f"{body} {duration:.1f}s" if skin_prefix != "┊": line = line.replace("┊", skin_prefix, 1) return line if not is_failure else f"{line}{failure_suffix}" def get_cute_tool_message( tool_name: str, args: dict, duration: float, result: str | None = None, ) -> str: """Render a completion label without letting cosmetic failures escape.""" try: return _get_cute_tool_message(tool_name, args, duration, result=result) except Exception as exc: # noqa: BLE001 — display must never abort a turn logger.debug("Tool completion label failed for %s: %s", tool_name, exc) safe_name = tool_name[:9] if isinstance(tool_name, str) and tool_name else "tool" safe_duration = f"{duration:.1f}s" if isinstance(duration, (int, float)) else "done" return f"┊ ⚡ {safe_name:9} completed {safe_duration}"