klaude-code 1.2.16__py3-none-any.whl → 1.2.18__py3-none-any.whl

klaude_code/protocol/events.py

@@ -133,6 +133,13 @@ class TodoChangeEvent(BaseModel):
     todos: list[model.TodoItem]
 
 
+class ContextUsageEvent(BaseModel):
+    """Real-time context usage update during task execution."""
+
+    session_id: str
+    context_percent: float  # Context usage percentage (0-100)
+
+
 HistoryItemEvent = (
     ThinkingEvent
     | TaskStartEvent
@@ -178,4 +185,5 @@ Event = (
     | TurnStartEvent
     | TurnEndEvent
     | TurnToolCallStartEvent
+    | ContextUsageEvent
 )

klaude_code/protocol/sub_agent/__init__.py

@@ -114,4 +114,4 @@ def sub_agent_tool_names(enabled_only: bool = False, model_name: str | None = No
 from klaude_code.protocol.sub_agent import explore as explore  # noqa: E402
 from klaude_code.protocol.sub_agent import oracle as oracle  # noqa: E402
 from klaude_code.protocol.sub_agent import task as task  # noqa: E402
-from klaude_code.protocol.sub_agent import web_fetch as web_fetch  # noqa: E402
+from klaude_code.protocol.sub_agent import web as web  # noqa: E402

klaude_code/protocol/sub_agent/explore.py

@@ -37,7 +37,7 @@ EXPLORE_PARAMETERS = {
             "description": "Optional JSON Schema for sub-agent structured output",
         },
     },
-    "required": ["description", "prompt", "output_format"],
+    "required": ["description", "prompt"],
     "additionalProperties": False,
 }
 

klaude_code/protocol/sub_agent/web.py

@@ -0,0 +1,79 @@
+from __future__ import annotations
+
+from typing import Any
+
+from klaude_code.protocol import tools
+from klaude_code.protocol.sub_agent import SubAgentProfile, register_sub_agent
+
+WEB_AGENT_DESCRIPTION = """\
+Launch a sub-agent to search the web, fetch pages, and analyze content. Use this for:
+- Accessing up-to-date information beyond your knowledge cutoff (current events, recent releases, latest docs)
+- Researching topics, news, APIs, or technical references
+- Fetching and analyzing specific URLs
+- Gathering comprehensive information from multiple web sources
+
+Capabilities:
+- Search the web to find relevant pages (no URL required)
+- Fetch and parse web pages (HTML-to-Markdown conversion)
+- Follow links across multiple pages autonomously
+- Aggregate findings from multiple sources
+
+How to use:
+- Write a clear prompt describing what information you need - the agent will search and fetch as needed
+- Account for "Today's date" in <env>. For example, if <env> says "Today's date: 2025-07-01", and the user wants the latest docs, do not use 2024 in the search query. Use 2025.
+- Optionally provide a `url` if you already know the target page
+- Use `output_format` (JSON Schema) to get structured data back from the agent
+
+What you receive:
+- The agent returns a text response summarizing its findings
+- With `output_format`, you receive structured JSON matching your schema
+- The response is the agent's analysis, not raw web content
+- Web content is saved to local files (paths included in Sources) - read them directly if you need full content\
+"""
+
+WEB_AGENT_PARAMETERS = {
+    "type": "object",
+    "properties": {
+        "description": {
+            "type": "string",
+            "description": "A short (3-5 word) description of the task",
+        },
+        "url": {
+            "type": "string",
+            "description": "The URL to fetch and analyze. If not provided, the agent will search the web first",
+        },
+        "prompt": {
+            "type": "string",
+            "description": "Instructions for searching, analyzing, or extracting content from the web page",
+        },
+        "output_format": {
+            "type": "object",
+            "description": "Optional JSON Schema for sub-agent structured output",
+        },
+    },
+    "required": ["description", "prompt"],
+    "additionalProperties": False,
+}
+
+
+def _web_agent_prompt_builder(args: dict[str, Any]) -> str:
+    """Build the WebAgent prompt from tool arguments."""
+    url = args.get("url", "")
+    prompt = args.get("prompt", "")
+    if url:
+        return f"URL to fetch: {url}\nTask: {prompt}"
+    return prompt
+
+
+register_sub_agent(
+    SubAgentProfile(
+        name="WebAgent",
+        description=WEB_AGENT_DESCRIPTION,
+        parameters=WEB_AGENT_PARAMETERS,
+        prompt_file="prompts/prompt-sub-agent-web.md",
+        tool_set=(tools.BASH, tools.READ, tools.WEB_FETCH, tools.WEB_SEARCH),
+        prompt_builder=_web_agent_prompt_builder,
+        active_form="Surfing",
+        output_schema_arg="output_format",
+    )
+)

klaude_code/protocol/tools.py

@@ -10,6 +10,7 @@ SKILL = "Skill"
 MERMAID = "Mermaid"
 MEMORY = "Memory"
 WEB_FETCH = "WebFetch"
+WEB_SEARCH = "WebSearch"
 REPORT_BACK = "report_back"
 
 # SubAgentType is just a string alias now; agent types are defined via SubAgentProfile

klaude_code/session/session.py

@@ -36,7 +36,7 @@ class Session(BaseModel):
 
     @property
     def messages_count(self) -> int:
-        """Count of user and assistant messages in conversation history.
+        """Count of user, assistant messages, and tool calls in conversation history.
 
         This is a cached property that is invalidated when append_history is called.
         """
@@ -44,7 +44,7 @@ class Session(BaseModel):
             self._messages_count_cache = sum(
                 1
                 for it in self.conversation_history
-                if isinstance(it, (model.UserMessageItem, model.AssistantMessageItem))
+                if isinstance(it, (model.UserMessageItem, model.AssistantMessageItem, model.ToolCallItem))
             )
         return self._messages_count_cache
 

klaude_code/session/templates/export_session.html

@@ -9,22 +9,36 @@
       href="data:image/svg+xml,<svg xmlns=%22http://www.w3.org/2000/svg%22 viewBox=%220 0 24 24%22 fill=%22none%22 stroke=%22%230851b2%22 stroke-width=%222%22 stroke-linecap=%22round%22 stroke-linejoin=%22round%22><polyline points=%2216 18 22 12 16 6%22></polyline><polyline points=%228 6 2 12 8 18%22></polyline></svg>"
     />
     <link
-      href="https://cdn.jsdelivr.net/npm/@fontsource/geist-sans/latin-400.css"
+      href="https://cdn.jsdelivr.net/npm/@fontsource/source-sans-3/400.css"
       rel="stylesheet"
     />
     <link
-      href="https://cdn.jsdelivr.net/npm/@fontsource/geist-sans/latin-500.css"
+      href="https://cdn.jsdelivr.net/npm/@fontsource/source-sans-3/400-italic.css"
       rel="stylesheet"
     />
     <link
-      href="https://cdn.jsdelivr.net/npm/@fontsource/geist-sans/latin-700.css"
+      href="https://cdn.jsdelivr.net/npm/@fontsource/source-sans-3/700.css"
+      rel="stylesheet"
+    />
+    <link
+      href="https://cdn.jsdelivr.net/npm/@fontsource/source-sans-3/700-italic.css"
+      rel="stylesheet"
+    />
+    <link
+      href="https://cdn.jsdelivr.net/npm/@fontsource/fira-code/400.css"
+      rel="stylesheet"
+    />
+    <link
+      href="https://cdn.jsdelivr.net/npm/@fontsource/fira-code/700.css"
       rel="stylesheet"
     />
     <style>
       :root {
-        --bg-body: #ededed;
-        --bg-container: #f0f0f0;
-        --bg-card: #f0f0f0;
+        --bg-body: #eae9e5;
+        --bg-container: #edece9;
+        --bg-card: #efeeeb;
+        --bg-error: #ffebee;
+        --bg-code: #f2f1ed;
         --border: #c8c8c8;
         --text: #111111;
         --text-dim: #64748b;
@@ -32,13 +46,11 @@
         --accent-dim: rgba(8, 145, 178, 0.08);
         --success: #15803d;
         --error: #dc2626;
-        --bg-error: #ffebee;
-        --bg-code: #f3f3f3;
         --fg-inline-code: #4f4fc7;
-        --font-mono: ui-monospace, SFMono-Regular, Menlo, Monaco, Consolas, Liberation Mono, Courier New, monospace;
-        --font-markdown-mono: ui-monospace, SFMono-Regular, Menlo, Monaco, Consolas, Liberation Mono, Courier New, monospace;
-        --font-markdown: "Geist Sans", system-ui, sans-serif;
-        --font-weight-bold: 800;
+        --font-mono: "Fira Code", ui-monospace, SFMono-Regular, Menlo, Monaco, Consolas, Liberation Mono, Courier New, monospace;
+        --font-markdown-mono: "Fira Code", ui-monospace, SFMono-Regular, Menlo, Monaco, Consolas, Liberation Mono, Courier New, monospace;
+        --font-markdown: "Source Sans 3", system-ui, sans-serif;
+        --font-weight-bold: 700;
         --font-size-xs: 13px;
         --font-size-sm: 14px;
         --font-size-base: 15px;
@@ -62,7 +74,6 @@
         background-color: var(--bg-body);
         color: var(--text);
         font-family: var(--font-mono);
-        font-feature-settings: "ss18";
         line-height: 1.6;
         font-size: var(--font-size-lg);
         -webkit-font-smoothing: antialiased;
@@ -672,8 +683,102 @@
       /* Markdown Elements */
       .markdown-body {
         font-family: var(--font-markdown);
-        line-height: 1.5;
+        line-height: 1.6;
+        font-size: var(--font-size-base);
+      }
+      .markdown-body > *:first-child {
+        margin-top: 0 !important;
+      }
+      .markdown-body > *:last-child {
+        margin-bottom: 0 !important;
+      }
+      
+      .markdown-body h1,
+      .markdown-body h2,
+      .markdown-body h3,
+      .markdown-body h4,
+      .markdown-body h5,
+      .markdown-body h6 {
+        margin-top: 24px;
+        margin-bottom: 16px;
+        font-weight: var(--font-weight-bold);
+        line-height: 1.25;
+      }
+
+      .markdown-body a {
+        color: var(--accent);
+        text-decoration: none;
+        border-bottom: 1px solid rgba(8, 81, 178, 0.2);
+        transition: border-color 0.2s, background-color 0.2s;
+      }
+      .markdown-body a:hover {
+        border-bottom-color: var(--accent);
+        background-color: rgba(8, 81, 178, 0.05);
+        border-radius: 2px;
+      }
+
+      .markdown-body p {
+        margin-bottom: 8px; /* Tighter spacing */
+      }
+      
+      .markdown-body ul,
+      .markdown-body ol {
+        margin-bottom: 8px; /* Tighter spacing */
+        padding-left: 1.5rem;
+        list-style-position: outside;
+      }
+      
+      .markdown-body ul ul,
+      .markdown-body ol ul,
+      .markdown-body ul ol,
+      .markdown-body ol ol {
+        margin-top: 0;
+        margin-bottom: 0;
+      }
+      
+      .markdown-body li > p {
+        margin-bottom: 0;
+      }
+      .markdown-body li + li {
+        margin-top: 0.25em;
+      }
+
+      .markdown-body blockquote {
+        margin: 0 0 16px;
+        padding: 0 1em;
+        color: var(--text-dim);
+        border-left: 0.25em solid var(--border);
       }
+
+      .markdown-body table {
+        border-collapse: collapse;
+        width: 100%;
+        margin-top: 8px;
+        margin-bottom: 16px;
+        display: block;
+        overflow-x: auto;
+      }
+      
+      .markdown-body table tr {
+        background-color: var(--bg-card);
+        border-top: 1px solid var(--border);
+      }
+      
+      .markdown-body table tr:nth-child(2n) {
+        background-color: rgba(0, 0, 0, 0.02);
+      }
+      
+      .markdown-body table th,
+      .markdown-body table td {
+        padding: 6px 13px;
+        border: 1px solid var(--border);
+      }
+      
+      .markdown-body table th {
+        font-weight: var(--font-weight-bold);
+        background-color: rgba(0, 0, 0, 0.04);
+      }
+
       .markdown-body hr {
         height: 0;
         margin: 24px 0;
@@ -682,10 +787,10 @@
       }
       .markdown-body pre {
         background: var(--bg-code);
-        padding: 16px;
+        padding: 12px;
         border-radius: var(--radius-md);
         overflow-x: auto;
-        margin: 12px 0;
+        margin: 8px 0 16px 0;
         border: 1px solid var(--border);
       }
       .markdown-body code {
@@ -694,36 +799,17 @@
         font-size: var(--font-size-sm);
         padding: 2px 4px;
         border-radius: var(--radius-sm);
+        background-color: rgba(0, 0, 0, 0.05); /* Slight bg for inline code */
       }
       .markdown-body pre code {
         background: transparent;
         padding: 0;
         border-radius: 0;
+        color: inherit;
       }
       .markdown-body pre code.hljs {
         background: transparent;
       }
-      .markdown-body p {
-        margin-bottom: 12px;
-      }
-      .markdown-body > *:first-child {
-        margin-top: 0;
-      }
-      .markdown-body > *:last-child {
-        margin-bottom: 0;
-      }
-      .markdown-body ul,
-      .markdown-body ol {
-        margin-bottom: 12px;
-        padding-left: 1.5rem;
-        list-style-position: outside;
-      }
-      .markdown-body ul ul,
-      .markdown-body ol ul,
-      .markdown-body ul ol,
-      .markdown-body ol ol {
-        margin-left: 1rem;
-      }
 
       /* Diff View */
       .diff-view {

klaude_code/trace/__init__.py

@@ -1,3 +1,21 @@
-from .log import DebugType, is_debug_enabled, log, log_debug, logger, prepare_debug_log_file, set_debug_logging
+from .log import (
+    DebugType,
+    get_current_log_file,
+    is_debug_enabled,
+    log,
+    log_debug,
+    logger,
+    prepare_debug_log_file,
+    set_debug_logging,
+)
 
-__all__ = ["DebugType", "is_debug_enabled", "log", "log_debug", "logger", "prepare_debug_log_file", "set_debug_logging"]
+__all__ = [
+    "DebugType",
+    "get_current_log_file",
+    "is_debug_enabled",
+    "log",
+    "log_debug",
+    "logger",
+    "prepare_debug_log_file",
+    "set_debug_logging",
+]

klaude_code/ui/modes/repl/completers.py

@@ -26,6 +26,7 @@ from prompt_toolkit.document import Document
 from prompt_toolkit.formatted_text import HTML
 
 from klaude_code.command import get_commands
+from klaude_code.trace.log import DebugType, log_debug
 
 # Pattern to match @token for completion refresh (used by key bindings).
 # Supports both plain tokens like `@src/file.py` and quoted tokens like
@@ -85,7 +86,7 @@ class _SlashCommandCompleter(Completer):
         matched: list[tuple[str, object, str]] = []
         for cmd_name, cmd_obj in commands.items():
             if cmd_name.startswith(frag):
-                hint = " [instructions]" if cmd_obj.support_addition_params else ""
+                hint = f" [{cmd_obj.placeholder}]" if cmd_obj.support_addition_params else ""
                 matched.append((cmd_name, cmd_obj, hint))
 
         if not matched:
@@ -444,6 +445,8 @@ class _AtFilesCompleter(Completer):
         return items[: min(self._max_results, 100)]
 
     def _run_cmd(self, cmd: list[str], cwd: Path | None = None) -> _CmdResult:
+        cmd_str = " ".join(cmd)
+        start = time.monotonic()
         try:
             p = subprocess.run(
                 cmd,
@@ -453,9 +456,23 @@ class _AtFilesCompleter(Completer):
                 text=True,
                 timeout=1.5,
             )
+            elapsed_ms = (time.monotonic() - start) * 1000
             if p.returncode == 0:
                 lines = [ln.strip() for ln in p.stdout.splitlines() if ln.strip()]
+                log_debug(
+                    f"[completer] cmd={cmd_str} elapsed={elapsed_ms:.1f}ms results={len(lines)}",
+                    debug_type=DebugType.EXECUTION,
+                )
                 return _CmdResult(True, lines)
+            log_debug(
+                f"[completer] cmd={cmd_str} elapsed={elapsed_ms:.1f}ms returncode={p.returncode}",
+                debug_type=DebugType.EXECUTION,
+            )
             return _CmdResult(False, [])
-        except Exception:
+        except Exception as e:
+            elapsed_ms = (time.monotonic() - start) * 1000
+            log_debug(
+                f"[completer] cmd={cmd_str} elapsed={elapsed_ms:.1f}ms error={e!r}",
+                debug_type=DebugType.EXECUTION,
+            )
             return _CmdResult(False, [])

klaude_code/ui/modes/repl/event_handler.py

@@ -9,6 +9,7 @@ from klaude_code import const
 from klaude_code.protocol import events
 from klaude_code.ui.core.stage_manager import Stage, StageManager
 from klaude_code.ui.modes.repl.renderer import REPLRenderer
+from klaude_code.ui.renderers.thinking import normalize_thinking_content
 from klaude_code.ui.rich.markdown import MarkdownStream, ThinkingMarkdown
 from klaude_code.ui.rich.theme import ThemeKey
 from klaude_code.ui.terminal.notifier import Notification, NotificationType, TerminalNotifier
@@ -121,7 +122,7 @@ class ActivityState:
             for name, count in self._tool_calls.items():
                 if not first:
                     activity_text.append(", ")
-                activity_text.append(name)
+                activity_text.append(Text(name, style=ThemeKey.SPINNER_STATUS_TEXT_BOLD))
                 if count > 1:
                     activity_text.append(f" x {count}")
                 first = False
@@ -137,11 +138,13 @@ class SpinnerStatusState:
     Composed of two independent layers:
     - base_status: Set by TodoChange, persistent within a turn
     - activity: Current activity (composing or tool_calls), mutually exclusive
+    - context_percent: Context usage percentage, updated during task execution
 
     Display logic:
     - If activity: show base + activity (if base exists) or activity + "..."
     - Elif base_status: show base_status
     - Else: show "Thinking …"
+    - Context percent is appended at the end if available
     """
 
     DEFAULT_STATUS = "Thinking …"
@@ -149,11 +152,13 @@ class SpinnerStatusState:
     def __init__(self) -> None:
         self._base_status: str | None = None
         self._activity = ActivityState()
+        self._context_percent: float | None = None
 
     def reset(self) -> None:
         """Reset all layers."""
         self._base_status = None
         self._activity.reset()
+        self._context_percent = None
 
     def set_base_status(self, status: str | None) -> None:
         """Set base status from TodoChange."""
@@ -175,12 +180,16 @@ class SpinnerStatusState:
         """Clear activity state for a new turn."""
         self._activity.reset()
 
+    def set_context_percent(self, percent: float) -> None:
+        """Set context usage percentage."""
+        self._context_percent = percent
+
     def get_activity_text(self) -> Text | None:
         """Get current activity text. Returns None if idle."""
         return self._activity.get_activity_text()
 
     def get_status(self) -> Text:
-        """Get current spinner status as rich Text."""
+        """Get current spinner status as rich Text (without context)."""
         activity_text = self._activity.get_activity_text()
 
         if self._base_status:
@@ -188,11 +197,19 @@ class SpinnerStatusState:
             if activity_text:
                 result.append(" | ")
                 result.append_text(activity_text)
-            return result
-        if activity_text:
+        elif activity_text:
             activity_text.append(" …")
-            return activity_text
-        return Text(self.DEFAULT_STATUS)
+            result = activity_text
+        else:
+            result = Text(self.DEFAULT_STATUS)
+
+        return result
+
+    def get_context_text(self) -> Text | None:
+        """Get context usage text for right-aligned display."""
+        if self._context_percent is None:
+            return None
+        return Text(f"{self._context_percent:.1f}%", style=ThemeKey.METADATA_DIM)
 
 
 class DisplayEventHandler:
@@ -247,6 +264,8 @@ class DisplayEventHandler:
                 self._on_task_metadata(e)
             case events.TodoChangeEvent() as e:
                 self._on_todo_change(e)
+            case events.ContextUsageEvent() as e:
+                self._on_context_usage(e)
             case events.TurnEndEvent():
                 pass
             case events.ResponseMetadataEvent():
@@ -330,7 +349,7 @@ class DisplayEventHandler:
         self.thinking_stream.append(event.content)
 
         if first_delta and self.thinking_stream.mdstream is not None:
-            self.thinking_stream.mdstream.update(self.thinking_stream.buffer)
+            self.thinking_stream.mdstream.update(normalize_thinking_content(self.thinking_stream.buffer))
 
         await self.stage_manager.enter_thinking_stage()
         self.thinking_stream.debouncer.schedule()
@@ -397,10 +416,11 @@ class DisplayEventHandler:
             self.renderer.display_tool_call(event)
 
     async def _on_tool_result(self, event: events.ToolResultEvent) -> None:
-        if self.renderer.is_sub_agent_session(event.session_id):
+        if self.renderer.is_sub_agent_session(event.session_id) and event.status == "success":
             return
         await self.stage_manager.transition_to(Stage.TOOL_RESULT)
-        self.renderer.display_tool_call_result(event)
+        with self.renderer.session_print_context(event.session_id):
+            self.renderer.display_tool_call_result(event)
 
     def _on_task_metadata(self, event: events.TaskMetadataEvent) -> None:
         self.renderer.display_task_metadata(event)
@@ -412,6 +432,12 @@ class DisplayEventHandler:
         self.spinner_status.clear_for_new_turn()
         self._update_spinner()
 
+    def _on_context_usage(self, event: events.ContextUsageEvent) -> None:
+        if self.renderer.is_sub_agent_session(event.session_id):
+            return
+        self.spinner_status.set_context_percent(event.context_percent)
+        self._update_spinner()
+
     async def _on_task_finish(self, event: events.TaskFinishEvent) -> None:
         self.renderer.display_task_finish(event)
         if not self.renderer.is_sub_agent_session(event.session_id):
@@ -459,7 +485,10 @@ class DisplayEventHandler:
 
     def _update_spinner(self) -> None:
         """Update spinner text from current status state."""
-        self.renderer.spinner_update(self.spinner_status.get_status())
+        self.renderer.spinner_update(
+            self.spinner_status.get_status(),
+            self.spinner_status.get_context_text(),
+        )
 
     async def _flush_assistant_buffer(self, state: StreamState) -> None:
         if state.is_active:
@@ -471,14 +500,14 @@ class DisplayEventHandler:
         if state.is_active:
             mdstream = state.mdstream
             assert mdstream is not None
-            mdstream.update(state.buffer)
+            mdstream.update(normalize_thinking_content(state.buffer))
 
     async def _finish_thinking_stream(self) -> None:
         if self.thinking_stream.is_active:
             self.thinking_stream.debouncer.cancel()
             mdstream = self.thinking_stream.mdstream
             assert mdstream is not None
-            mdstream.update(self.thinking_stream.buffer, final=True)
+            mdstream.update(normalize_thinking_content(self.thinking_stream.buffer), final=True)
             self.thinking_stream.finish()
             self.renderer.console.pop_theme()
             self.renderer.print()
@@ -525,15 +554,15 @@ class DisplayEventHandler:
         """Calculate max length for base_status based on terminal width.
 
         Reserve space for:
-        - Spinner glyph + space: 2 chars
+        - Spinner glyph + space + context text: 2 chars + context text length 10 chars
         - " | " separator: 3 chars (only if activity text present)
         - Activity text: actual length (only if present)
         - Status hint text (esc to interrupt)
         """
         terminal_width = self.renderer.console.size.width
 
-        # Base reserved space: spinner + status hint
-        reserved_space = 2 + len(const.STATUS_HINT_TEXT)
+        # Base reserved space: spinner + context + status hint
+        reserved_space = 12 + len(const.STATUS_HINT_TEXT)
 
         # Add space for activity text if present
         activity_text = self.spinner_status.get_activity_text()

klaude_code/ui/modes/repl/renderer.py

@@ -264,9 +264,9 @@ class REPLRenderer:
         """Stop the spinner animation."""
         self._spinner.stop()
 
-    def spinner_update(self, status_text: str | Text) -> None:
-        """Update the spinner status text."""
-        self._spinner.update(ShimmerStatusText(status_text, ThemeKey.SPINNER_STATUS_TEXT))
+    def spinner_update(self, status_text: str | Text, right_text: Text | None = None) -> None:
+        """Update the spinner status text with optional right-aligned text."""
+        self._spinner.update(ShimmerStatusText(status_text, ThemeKey.SPINNER_STATUS_TEXT, right_text))
 
     def spinner_renderable(self) -> Spinner:
         """Return the spinner's renderable for embedding in other components."""

klaude_code/ui/renderers/metadata.py

@@ -61,9 +61,7 @@ def _render_task_metadata_block(
     if metadata.usage is not None:
         # Tokens: ↑ 37k cache 5k ↓ 907 think 45k
         token_parts: list[Text] = [
-            Text.assemble(
-                ("↑ ", ThemeKey.METADATA_DIM), (format_number(metadata.usage.input_tokens), ThemeKey.METADATA)
-            )
+            Text.assemble(("↑", ThemeKey.METADATA_DIM), (format_number(metadata.usage.input_tokens), ThemeKey.METADATA))
         ]
         if metadata.usage.cached_tokens > 0:
             token_parts.append(
@@ -74,7 +72,7 @@ def _render_task_metadata_block(
             )
         token_parts.append(
             Text.assemble(
-                ("↓ ", ThemeKey.METADATA_DIM), (format_number(metadata.usage.output_tokens), ThemeKey.METADATA)
+                ("↓", ThemeKey.METADATA_DIM), (format_number(metadata.usage.output_tokens), ThemeKey.METADATA)
             )
         )
         if metadata.usage.reasoning_tokens > 0: