ripperdoc 0.2.2__py3-none-any.whl → 0.2.4__py3-none-any.whl

ripperdoc/__init__.py

@@ -1,3 +1,3 @@
 """Ripperdoc - AI-powered coding agent."""
 
-__version__ = "0.2.2"
+__version__ = "0.2.4"

ripperdoc/cli/cli.py

@@ -31,6 +31,7 @@ from ripperdoc.utils.mcp import (
 )
 from ripperdoc.tools.mcp_tools import load_dynamic_mcp_tools_async, merge_tools_with_dynamic
 from ripperdoc.utils.log import enable_session_file_logging, get_logger
+from ripperdoc.utils.prompt import prompt_secret
 
 from rich.console import Console
 from rich.markdown import Markdown
@@ -147,7 +148,9 @@ async def run_query(
             console.print("\n[yellow]Interrupted by user[/yellow]")
         except Exception as e:
             console.print(f"[red]Error: {escape(str(e))}[/red]")
-            logger.exception("[cli] Unhandled error while running prompt", extra={"session_id": session_id})
+            logger.exception(
+                "[cli] Unhandled error while running prompt", extra={"session_id": session_id}
+            )
             if verbose:
                 import traceback
 
@@ -197,7 +200,11 @@ def check_onboarding() -> bool:
         )
         api_base = click.prompt("API Base URL")
 
-    api_key = click.prompt("Enter your API key", hide_input=True)
+    api_key = ""
+    while not api_key:
+        api_key = prompt_secret("Enter your API key").strip()
+        if not api_key:
+            console.print("[red]API key is required.[/red]")
 
     provider = ProviderType(provider_choice)
 

ripperdoc/cli/commands/agents_cmd.py

@@ -31,14 +31,17 @@ def _handle(ui: Any, trimmed_arg: str) -> bool:
     def print_agents_usage() -> None:
         console.print("[bold]/agents[/bold] — list configured agents")
         console.print(
-            "[bold]/agents create <name> [location] [model][/bold] — create agent (location: user|project, default user)"
+            "[bold]/agents create <name> [location] [model][/bold] — "
+            "create agent (location: user|project, default user)"
         )
         console.print("[bold]/agents edit <name> [location][/bold] — edit an existing agent")
         console.print(
-            "[bold]/agents delete <name> [location][/bold] — delete agent (location: user|project, default user)"
+            "[bold]/agents delete <name> [location][/bold] — "
+            "delete agent (location: user|project, default user)"
         )
         console.print(
-            f"[dim]Agent files live in ~/.ripperdoc/{AGENT_DIR_NAME} or ./.ripperdoc/{AGENT_DIR_NAME}[/dim]"
+            f"[dim]Agent files live in ~/.ripperdoc/{AGENT_DIR_NAME} "
+            f"or ./.ripperdoc/{AGENT_DIR_NAME}[/dim]"
         )
         console.print(
             "[dim]Model can be a profile name or pointer (task/main/etc). Defaults to 'task'.[/dim]"
@@ -92,7 +95,8 @@ def _handle(ui: Any, trimmed_arg: str) -> bool:
             and model_input not in pointer_map
         ):
             console.print(
-                "[yellow]Model not found in profiles or pointers; will fall back to main if unavailable.[/yellow]"
+                "[yellow]Model not found in profiles or pointers; "
+                "will fall back to main if unavailable.[/yellow]"
             )
 
         try:

ripperdoc/cli/commands/context_cmd.py

@@ -9,11 +9,11 @@ from ripperdoc.core.query import QueryContext
 from ripperdoc.core.system_prompt import build_system_prompt
 from ripperdoc.utils.memory import build_memory_instructions
 from ripperdoc.utils.message_compaction import (
-    estimate_tokens_from_text,
     get_remaining_context_tokens,
     resolve_auto_compact_enabled,
     summarize_context_usage,
 )
+from ripperdoc.utils.token_estimation import estimate_tokens
 from ripperdoc.utils.mcp import (
     estimate_mcp_tokens,
     format_mcp_instructions,
@@ -60,7 +60,7 @@ def _handle(ui: Any, _: str) -> bool:
         mcp_instructions=mcp_instructions,
     )
     memory_instructions = build_memory_instructions()
-    memory_tokens = estimate_tokens_from_text(memory_instructions) if memory_instructions else 0
+    memory_tokens = estimate_tokens(memory_instructions) if memory_instructions else 0
     mcp_tokens = estimate_mcp_tokens(servers) if mcp_instructions else 0
 
     breakdown = summarize_context_usage(
@@ -98,7 +98,7 @@ def _handle(ui: Any, _: str) -> bool:
                     display = f"{display} ({server})"
                 try:
                     schema = tool.input_schema.model_json_schema()
-                    token_est = estimate_tokens_from_text(json.dumps(schema, sort_keys=True))
+                    token_est = estimate_tokens(json.dumps(schema, sort_keys=True))
                 except Exception:
                     token_est = 0
                 lines.append(f"     └ {display}: {format_tokens(token_est)} tokens")

ripperdoc/cli/commands/cost_cmd.py

@@ -32,6 +32,7 @@ def _handle(ui: Any, _: str) -> bool:
     total_cache_read = usage.total_cache_read_tokens
     total_cache_creation = usage.total_cache_creation_tokens
     total_tokens = total_input + total_output + total_cache_read + total_cache_creation
+    total_cost = usage.total_cost_usd
 
     ui.console.print("\n[bold]Session token usage[/bold]")
     ui.console.print(
@@ -44,6 +45,8 @@ def _handle(ui: Any, _: str) -> bool:
             f"{_fmt_tokens(total_cache_creation)} write"
         )
     ui.console.print(f" Requests: {usage.total_requests}")
+    if total_cost:
+        ui.console.print(f" Cost: ${total_cost:.4f}")
     if usage.total_duration_ms:
         ui.console.print(f" API time: {_format_duration(usage.total_duration_ms)}")
 
@@ -62,6 +65,8 @@ def _handle(ui: Any, _: str) -> bool:
         if stats.duration_ms:
             line += f", {_format_duration(stats.duration_ms)} total"
         line += ")"
+        if stats.cost_usd:
+            line += f", ${stats.cost_usd:.4f}"
         ui.console.print(line)
 
     return True

ripperdoc/cli/commands/doctor_cmd.py

@@ -40,6 +40,7 @@ def _status_row(label: str, status: str, detail: str = "") -> Tuple[str, str, st
 def _api_key_status(provider: ProviderType, profile_key: Optional[str]) -> Tuple[str, str]:
     """Check API key presence and source."""
     import os
+
     for env_var in api_key_env_candidates(provider):
         if os.environ.get(env_var):
             masked = os.environ[env_var]
@@ -59,7 +60,9 @@ def _model_status(project_path: Path) -> List[Tuple[str, str, str]]:
     rows: List[Tuple[str, str, str]] = []
 
     if not profile:
-        rows.append(_status_row("Model profile", "error", "No profile configured for pointer 'main'"))
+        rows.append(
+            _status_row("Model profile", "error", "No profile configured for pointer 'main'")
+        )
         return rows
 
     if pointer not in config.model_profiles:
@@ -147,10 +150,14 @@ def _project_status(project_path: Path) -> Tuple[str, str, str]:
         config = get_project_config(project_path)
         # Access a field to ensure model parsing does not throw.
         _ = len(config.allowed_tools)
-        return _status_row("Project config", "ok", f".ripperdoc/config.json loaded for {project_path}")
+        return _status_row(
+            "Project config", "ok", f".ripperdoc/config.json loaded for {project_path}"
+        )
     except Exception as exc:  # pragma: no cover - defensive
         logger.exception("[doctor] Failed to load project config", exc_info=exc)
-        return _status_row("Project config", "warn", f"Could not read .ripperdoc/config.json: {exc}")
+        return _status_row(
+            "Project config", "warn", f"Could not read .ripperdoc/config.json: {exc}"
+        )
 
 
 def _render_table(console: Any, rows: List[Tuple[str, str, str]]) -> None:
@@ -185,7 +192,8 @@ def _handle(ui: Any, _: str) -> bool:
             ui.console.print(f"  • {escape(err)}")
 
     ui.console.print(
-        "\n[dim]If a check is failing, run `ripperdoc` without flags to rerun onboarding or update ~/.ripperdoc.json[/dim]"
+        "\n[dim]If a check is failing, run `ripperdoc` without flags "
+        "to rerun onboarding or update ~/.ripperdoc.json[/dim]"
     )
     return True
 

ripperdoc/cli/commands/memory_cmd.py

@@ -103,7 +103,8 @@ def _open_in_editor(path: Path, console: Any) -> bool:
     editor_cmd = _determine_editor_command()
     if not editor_cmd:
         console.print(
-            f"[yellow]No editor configured. Set $EDITOR or $VISUAL, or manually edit: {escape(str(path))}[/yellow]"
+            f"[yellow]No editor configured. Set $EDITOR or $VISUAL, "
+            f"or manually edit: {escape(str(path))}[/yellow]"
         )
         return False
 
@@ -152,9 +153,7 @@ def _handle(ui: Any, trimmed_arg: str) -> bool:
             "global": "user",
         }
         if scope not in scope_aliases:
-            ui.console.print(
-                "[red]Unknown scope. Use one of: project, local, user.[/red]"
-            )
+            ui.console.print("[red]Unknown scope. Use one of: project, local, user.[/red]")
             return True
 
         resolved_scope = scope_aliases[scope]
@@ -175,9 +174,7 @@ def _handle(ui: Any, trimmed_arg: str) -> bool:
 
         _open_in_editor(target_path, ui.console)
 
-        messages: List[str] = [
-            f"{heading}: {escape(_shorten_path(target_path, project_path))}"
-        ]
+        messages: List[str] = [f"{heading}: {escape(_shorten_path(target_path, project_path))}"]
         if created:
             messages.append("Created new memory file.")
         if gitignore_added:
@@ -189,12 +186,8 @@ def _handle(ui: Any, trimmed_arg: str) -> bool:
         return True
 
     _render_memory_table(ui.console, project_path)
-    ui.console.print(
-        "[dim]Usage: /memory project | /memory local | /memory user[/dim]"
-    )
-    ui.console.print(
-        "[dim]Project and user memories feed directly into the system prompt.[/dim]"
-    )
+    ui.console.print("[dim]Usage: /memory project | /memory local | /memory user[/dim]")
+    ui.console.print("[dim]Project and user memories feed directly into the system prompt.[/dim]")
     return True
 
 

ripperdoc/cli/commands/models_cmd.py

@@ -1,6 +1,4 @@
-from typing import Any
-from getpass import getpass
-from typing import Optional
+from typing import Any, Optional
 
 from rich.markup import escape
 
@@ -14,6 +12,7 @@ from ripperdoc.core.config import (
     set_model_pointer,
 )
 from ripperdoc.utils.log import get_logger
+from ripperdoc.utils.prompt import prompt_secret
 
 from .base import SlashCommand
 
@@ -110,9 +109,18 @@ def _handle(ui: Any, trimmed_arg: str) -> bool:
             console.print("[red]Model name is required.[/red]")
             return True
 
-        api_key_input = getpass("API key (leave blank to keep unset): ").strip()
+        api_key_input = prompt_secret("API key (leave blank to keep unset)").strip()
         api_key = api_key_input or (existing_profile.api_key if existing_profile else None)
 
+        auth_token = existing_profile.auth_token if existing_profile else None
+        if provider == ProviderType.ANTHROPIC:
+            auth_token_input = prompt_secret(
+                "Auth token (Anthropic only, leave blank to keep unset)"
+            ).strip()
+            auth_token = auth_token_input or auth_token
+        else:
+            auth_token = None
+
         api_base_default = existing_profile.api_base if existing_profile else ""
         api_base = (
             console.input(
@@ -163,6 +171,7 @@ def _handle(ui: Any, trimmed_arg: str) -> bool:
             max_tokens=max_tokens,
             temperature=temperature,
             context_window=context_window,
+            auth_token=auth_token,
         )
 
         try:
@@ -213,8 +222,8 @@ def _handle(ui: Any, trimmed_arg: str) -> bool:
         )
 
         api_key_label = "[set]" if existing_profile.api_key else "[not set]"
-        api_key_prompt = f"API key {api_key_label} (Enter=keep, '-'=clear): "
-        api_key_input = getpass(api_key_prompt).strip()
+        api_key_prompt = f"API key {api_key_label} (Enter=keep, '-'=clear)"
+        api_key_input = prompt_secret(api_key_prompt).strip()
         if api_key_input == "-":
             api_key = None
         elif api_key_input:
@@ -222,6 +231,21 @@ def _handle(ui: Any, trimmed_arg: str) -> bool:
         else:
             api_key = existing_profile.api_key
 
+        auth_token = existing_profile.auth_token
+        if (
+            provider == ProviderType.ANTHROPIC
+            or existing_profile.provider == ProviderType.ANTHROPIC
+        ):
+            auth_label = "[set]" if auth_token else "[not set]"
+            auth_prompt = f"Auth token (Anthropic only) {auth_label} (Enter=keep, '-'=clear)"
+            auth_token_input = prompt_secret(auth_prompt).strip()
+            if auth_token_input == "-":
+                auth_token = None
+            elif auth_token_input:
+                auth_token = auth_token_input
+        else:
+            auth_token = None
+
         api_base = (
             console.input(f"API base (optional) [{existing_profile.api_base or ''}]: ").strip()
             or existing_profile.api_base
@@ -255,6 +279,7 @@ def _handle(ui: Any, trimmed_arg: str) -> bool:
             max_tokens=max_tokens,
             temperature=temperature,
             context_window=context_window,
+            auth_token=auth_token,
         )
 
         try:
@@ -335,6 +360,11 @@ def _handle(ui: Any, trimmed_arg: str) -> bool:
             markup=False,
         )
         console.print(f"      api_key: {'***' if profile.api_key else 'Not set'}", markup=False)
+        if profile.provider == ProviderType.ANTHROPIC:
+            console.print(
+                f"      auth_token: {'***' if getattr(profile, 'auth_token', None) else 'Not set'}",
+                markup=False,
+            )
         if profile.openai_tool_mode:
             console.print(f"      openai_tool_mode: {profile.openai_tool_mode}", markup=False)
     pointer_labels = ", ".join(f"{p}->{v or '-'}" for p, v in pointer_map.items())

ripperdoc/cli/commands/resume_cmd.py

@@ -30,7 +30,8 @@ def _choose_session(ui: Any, arg: str) -> Optional[SessionSummary]:
             if 0 <= idx < len(sessions):
                 return sessions[idx]
             ui.console.print(
-                f"[red]Invalid session index {escape(str(idx))}. Choose 0-{len(sessions) - 1}.[/red]"
+                f"[red]Invalid session index {escape(str(idx))}. "
+                f"Choose 0-{len(sessions) - 1}.[/red]"
             )
         else:
             # Treat arg as session id if it matches.
@@ -60,7 +61,8 @@ def _choose_session(ui: Any, arg: str) -> Optional[SessionSummary]:
     idx = int(choice_text)
     if idx < 0 or idx >= len(sessions):
         ui.console.print(
-            f"[red]Invalid session index {escape(str(idx))}. Choose 0-{len(sessions) - 1}.[/red]"
+            f"[red]Invalid session index {escape(str(idx))}. "
+            f"Choose 0-{len(sessions) - 1}.[/red]"
         )
         return None
     return sessions[idx]

ripperdoc/cli/commands/status_cmd.py

@@ -34,7 +34,7 @@ def _auth_token_display(profile: Optional[ModelProfile]) -> Tuple[str, Optional[
     env_var = next((name for name in env_candidates if os.environ.get(name)), None)
     if env_var:
         return (f"{env_var} (env)", env_var)
-    if profile.api_key:
+    if profile.api_key or getattr(profile, "auth_token", None):
         return ("Configured in profile", None)
     return ("Missing", None)
 

ripperdoc/cli/ui/rich_ui.py

@@ -34,6 +34,7 @@ from ripperdoc.cli.commands import (
 from ripperdoc.cli.ui.helpers import get_profile_for_pointer
 from ripperdoc.core.permissions import make_permission_checker
 from ripperdoc.cli.ui.spinner import Spinner
+from ripperdoc.cli.ui.thinking_spinner import ThinkingSpinner
 from ripperdoc.cli.ui.context_display import context_usage_lines
 from ripperdoc.utils.message_compaction import (
     compact_messages,
@@ -43,6 +44,7 @@ from ripperdoc.utils.message_compaction import (
     get_remaining_context_tokens,
     resolve_auto_compact_enabled,
 )
+from ripperdoc.utils.token_estimation import estimate_tokens
 from ripperdoc.utils.mcp import (
     format_mcp_instructions,
     load_mcp_servers_async,
@@ -64,6 +66,95 @@ from ripperdoc.utils.log import enable_session_file_logging, get_logger
 # Type alias for conversation messages
 ConversationMessage = Union[UserMessage, AssistantMessage, ProgressMessage]
 
+THINKING_WORDS: list[str] = [
+    "Accomplishing",
+    "Actioning",
+    "Actualizing",
+    "Baking",
+    "Booping",
+    "Brewing",
+    "Calculating",
+    "Cerebrating",
+    "Channelling",
+    "Churning",
+    "Clauding",
+    "Coalescing",
+    "Cogitating",
+    "Computing",
+    "Combobulating",
+    "Concocting",
+    "Conjuring",
+    "Considering",
+    "Contemplating",
+    "Cooking",
+    "Crafting",
+    "Creating",
+    "Crunching",
+    "Deciphering",
+    "Deliberating",
+    "Determining",
+    "Discombobulating",
+    "Divining",
+    "Doing",
+    "Effecting",
+    "Elucidating",
+    "Enchanting",
+    "Envisioning",
+    "Finagling",
+    "Flibbertigibbeting",
+    "Forging",
+    "Forming",
+    "Frolicking",
+    "Generating",
+    "Germinating",
+    "Hatching",
+    "Herding",
+    "Honking",
+    "Ideating",
+    "Imagining",
+    "Incubating",
+    "Inferring",
+    "Manifesting",
+    "Marinating",
+    "Meandering",
+    "Moseying",
+    "Mulling",
+    "Mustering",
+    "Musing",
+    "Noodling",
+    "Percolating",
+    "Perusing",
+    "Philosophising",
+    "Pontificating",
+    "Pondering",
+    "Processing",
+    "Puttering",
+    "Puzzling",
+    "Reticulating",
+    "Ruminating",
+    "Scheming",
+    "Schlepping",
+    "Shimmying",
+    "Simmering",
+    "Smooshing",
+    "Spelunking",
+    "Spinning",
+    "Stewing",
+    "Sussing",
+    "Synthesizing",
+    "Thinking",
+    "Tinkering",
+    "Transmuting",
+    "Unfurling",
+    "Unravelling",
+    "Vibing",
+    "Wandering",
+    "Whirring",
+    "Wibbling",
+    "Wizarding",
+    "Working",
+    "Wrangling",
+]
 
 console = Console()
 logger = get_logger()
@@ -357,6 +448,13 @@ class RichUI:
             else:
                 success = getattr(tool_data, "success", None)
                 failed = failed or (success is False)
+            failed = failed or bool(self._get_tool_field(tool_data, "is_error"))
+
+        warning_text = None
+        token_estimate = None
+        if tool_data is not None:
+            warning_text = self._get_tool_field(tool_data, "warning")
+            token_estimate = self._get_tool_field(tool_data, "token_estimate")
 
         if failed:
             if content:
@@ -365,6 +463,17 @@ class RichUI:
                 self.console.print(f"  ⎿  [red]{escape(sender)} failed[/red]")
             return
 
+        if warning_text:
+            self.console.print(f"  ⎿  [yellow]{escape(str(warning_text))}[/yellow]")
+            if token_estimate:
+                self.console.print(
+                    f"      [dim]Estimated tokens: {escape(str(token_estimate))}[/dim]"
+                )
+        elif token_estimate and self.verbose:
+            self.console.print(
+                f"  ⎿  [dim]Estimated tokens: {escape(str(token_estimate))}[/dim]"
+            )
+
         if not content:
             self.console.print("  ⎿  [dim]Tool completed[/]")
             return
@@ -739,7 +848,23 @@ class RichUI:
                         },
                     )
 
-            spinner = Spinner(console, "Thinking...", spinner="dots")
+            prompt_tokens_est = estimate_conversation_tokens(messages, protocol=protocol)
+            spinner = ThinkingSpinner(console, prompt_tokens_est)
+
+            # Define pause/resume callbacks for tools that need user interaction
+            def pause_ui() -> None:
+                if spinner:
+                    spinner.stop()
+
+            def resume_ui() -> None:
+                if spinner:
+                    spinner.start()
+                    spinner.update("Thinking...")
+
+            # Set the UI callbacks on the query context
+            self.query_context.pause_ui = pause_ui
+            self.query_context.resume_ui = resume_ui
+
             # Wrap permission checker to pause the spinner while waiting for user input.
             base_permission_checker = self._permission_checker
 
@@ -759,6 +884,7 @@ class RichUI:
             # Track tool uses by ID so results align even when multiple tools fire.
             tool_registry: Dict[str, Dict[str, Any]] = {}
             last_tool_name = None
+            output_token_est = 0
 
             try:
                 spinner.start()
@@ -853,7 +979,14 @@ class RichUI:
                                 )
                             elif message.content.startswith("Subagent"):
                                 self.display_message("Subagent", message.content, is_tool=True)
-                        spinner.update(f"Working... {message.content}")
+                        if message.tool_use_id == "stream":
+                            delta_tokens = estimate_tokens(message.content)
+                            output_token_est += delta_tokens
+                            spinner.update_tokens(output_token_est)
+                        else:
+                            spinner.update_tokens(
+                                output_token_est, suffix=f"Working... {message.content}"
+                            )
 
                     # Add message to history
                     self._log_message(message)

ripperdoc/cli/ui/thinking_spinner.py

@@ -0,0 +1,128 @@
+"""Specialized spinner that shows token progress with playful verbs."""
+
+from __future__ import annotations
+
+import random
+import time
+from typing import Optional
+
+from rich.console import Console
+
+from ripperdoc.cli.ui.spinner import Spinner
+
+
+THINKING_WORDS: list[str] = [
+    "Accomplishing",
+    "Actioning",
+    "Actualizing",
+    "Baking",
+    "Booping",
+    "Brewing",
+    "Calculating",
+    "Cerebrating",
+    "Channelling",
+    "Churning",
+    "Clauding",
+    "Coalescing",
+    "Cogitating",
+    "Computing",
+    "Combobulating",
+    "Concocting",
+    "Conjuring",
+    "Considering",
+    "Contemplating",
+    "Cooking",
+    "Crafting",
+    "Creating",
+    "Crunching",
+    "Deciphering",
+    "Deliberating",
+    "Determining",
+    "Discombobulating",
+    "Divining",
+    "Doing",
+    "Effecting",
+    "Elucidating",
+    "Enchanting",
+    "Envisioning",
+    "Finagling",
+    "Flibbertigibbeting",
+    "Forging",
+    "Forming",
+    "Frolicking",
+    "Generating",
+    "Germinating",
+    "Hatching",
+    "Herding",
+    "Honking",
+    "Ideating",
+    "Imagining",
+    "Incubating",
+    "Inferring",
+    "Manifesting",
+    "Marinating",
+    "Meandering",
+    "Moseying",
+    "Mulling",
+    "Mustering",
+    "Musing",
+    "Noodling",
+    "Percolating",
+    "Perusing",
+    "Philosophising",
+    "Pontificating",
+    "Pondering",
+    "Processing",
+    "Puttering",
+    "Puzzling",
+    "Reticulating",
+    "Ruminating",
+    "Scheming",
+    "Schlepping",
+    "Shimmying",
+    "Simmering",
+    "Smooshing",
+    "Spelunking",
+    "Spinning",
+    "Stewing",
+    "Sussing",
+    "Synthesizing",
+    "Thinking",
+    "Tinkering",
+    "Transmuting",
+    "Unfurling",
+    "Unravelling",
+    "Vibing",
+    "Wandering",
+    "Whirring",
+    "Wibbling",
+    "Wizarding",
+    "Working",
+    "Wrangling",
+]
+
+
+class ThinkingSpinner(Spinner):
+    """Spinner that shows elapsed time and token progress."""
+
+    def __init__(self, console: Console, prompt_tokens: int) -> None:
+        self.prompt_tokens = prompt_tokens
+        self.start_time = time.monotonic()
+        self.out_tokens = 0
+        self.thinking_word = random.choice(THINKING_WORDS)
+        super().__init__(console, self._format_text(), spinner="dots")
+
+    def _format_text(self, suffix: Optional[str] = None) -> str:
+        elapsed = int(time.monotonic() - self.start_time)
+        base = f"✽ {self.thinking_word}… (esc to interrupt · {elapsed}s"
+        if self.out_tokens > 0:
+            base += f" · ↓ {self.out_tokens} tokens"
+        else:
+            base += f" · ↑ {self.prompt_tokens} tokens"
+        if suffix:
+            base += f" · {suffix}"
+        return base + ")"
+
+    def update_tokens(self, out_tokens: int, suffix: Optional[str] = None) -> None:
+        self.out_tokens = max(0, out_tokens)
+        self.update(self._format_text(suffix))