Live research stats: tool count, tokens, elapsed time in header

agent/tools/research_tool.py

@@ -221,7 +221,25 @@ async def research_handler(
         except Exception:
             pass
 
-    await _log("Starting research sub-agent...")
+    import time as _time
+
+    _start = _time.monotonic()
+    _tool_uses = 0
+    _total_tokens = 0
+
+    def _format_stats() -> str:
+        elapsed = _time.monotonic() - _start
+        if elapsed < 60:
+            time_str = f"{elapsed:.0f}s"
+        else:
+            time_str = f"{elapsed / 60:.0f}m {elapsed % 60:.0f}s"
+        if _total_tokens >= 1000:
+            tok_str = f"{_total_tokens / 1000:.1f}k"
+        else:
+            tok_str = str(_total_tokens)
+        return f"{_tool_uses} tool uses · {tok_str} tokens · {time_str}"
+
+    await _log("stats:" + _format_stats())
 
     # Run the research loop (max 20 iterations — research should be focused)
     max_iterations = 20
@@ -239,11 +257,16 @@ async def research_handler(
             logger.error("Research sub-agent LLM error: %s", e)
             return f"Research agent LLM error: {e}", False
 
+        # Track tokens
+        if response.usage:
+            _total_tokens += response.usage.total_tokens
+
         choice = response.choices[0]
         msg = choice.message
 
         # If no tool calls, we have our final answer
         if not msg.tool_calls:
+            await _log("stats:" + _format_stats())
             await _log("Research complete.")
             content = msg.content or "Research completed but no summary generated."
             return content, True
@@ -285,6 +308,8 @@ async def research_handler(
                 output, _success = await session.tool_router.call_tool(
                     tool_name, tool_args, session=session
                 )
+                _tool_uses += 1
+                await _log("stats:" + _format_stats())
                 # Truncate tool output for the research context
                 if len(output) > 8000:
                     output = output[:4800] + "\n...(truncated)...\n" + output[-3200:]

agent/utils/terminal_display.py

@@ -65,37 +65,55 @@ def print_tool_output(output: str, success: bool, truncate: bool = True) -> None
 
 
 class SubAgentDisplay:
-    """Rolling 2-line gray display of the last 2 sub-agent tool calls."""
+    """Live-updating display: header with stats + rolling 2-line tool calls."""
 
     _MAX_VISIBLE = 2
 
     def __init__(self):
         self._calls: list[str] = []
+        self._stats: str = ""
         self._lines_on_screen = 0
 
-    def update(self, tool_desc: str) -> None:
+    def set_stats(self, stats: str) -> None:
+        """Update the stats line and redraw."""
+        self._stats = stats
+        self._redraw()
+
+    def add_call(self, tool_desc: str) -> None:
+        """Add a tool call and redraw."""
         self._calls.append(tool_desc)
-        visible = self._calls[-self._MAX_VISIBLE:]
-        self._redraw(visible)
+        self._redraw()
 
     def clear(self) -> None:
+        self._erase()
+        self._lines_on_screen = 0
+        self._calls = []
+        self._stats = ""
+
+    def _erase(self) -> None:
         if self._lines_on_screen > 0:
             f = _console.file
             for _ in range(self._lines_on_screen):
                 f.write("\033[A\033[K")
             f.flush()
-        self._lines_on_screen = 0
-        self._calls = []
 
-    def _redraw(self, visible: list[str]) -> None:
+    def _redraw(self) -> None:
         f = _console.file
-        if self._lines_on_screen > 0:
-            for _ in range(self._lines_on_screen):
-                f.write("\033[A\033[K")
+        self._erase()
+        lines = []
+        # Header: ▸ research (stats)
+        header = f"{_I}\033[1;36m▸ research\033[0m"
+        if self._stats:
+            header += f"  \033[2m({self._stats})\033[0m"
+        lines.append(header)
+        # Last 2 tool calls, gray
+        visible = self._calls[-self._MAX_VISIBLE:]
         for desc in visible:
-            f.write(f"{_I}  \033[2m{desc}\033[0m\n")
+            lines.append(f"{_I}  \033[2m{desc}\033[0m")
+        for line in lines:
+            f.write(line + "\n")
         f.flush()
-        self._lines_on_screen = len(visible)
+        self._lines_on_screen = len(lines)
 
 
 _subagent_display = SubAgentDisplay()
@@ -104,13 +122,12 @@ _subagent_display = SubAgentDisplay()
 def print_tool_log(tool: str, log: str) -> None:
     """Handle tool log events — sub-agent calls get the rolling display."""
     if tool == "research":
-        if log == "Starting research sub-agent...":
-            _subagent_display.clear()
-            _console.print(f"{_I}[tool.name]▸ research[/tool.name]")
+        if log.startswith("stats:"):
+            _subagent_display.set_stats(log[6:])
         elif log == "Research complete.":
             _subagent_display.clear()
         else:
-            _subagent_display.update(log)
+            _subagent_display.add_call(log)
     else:
         _console.print(f"{_I}[dim]{tool}: {log}[/dim]")