X-iZhang
03f4134190
- Introduced UserMessage widget for displaying user input with a styled prompt. - Updated onboarding steps to include UI backend selection (Rich CLI or Textual TUI). - Modified EvoScientistConfig to store selected UI backend. - Enhanced configuration handling to support UI backend environment variable. - Updated README with new UI backend options and commands. - Added tests for new UI backend functionality and UserMessage widget. - Removed obsolete test files and ensured existing tests are updated accordingly.
83 lines
2.4 KiB
Python
83 lines
2.4 KiB
Python
"""Thinking panel widget for extended thinking display.
|
|
|
|
Renders a Rich Panel matching the Rich CLI's style:
|
|
blue border, "Thinking" title with spinner, dim content.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from rich.panel import Panel
|
|
from rich.text import Text
|
|
|
|
from textual.widgets import Static
|
|
|
|
_SPINNER_FRAMES = "\u280b\u2819\u2839\u2838\u283c\u2834\u2826\u2827\u2807\u280f"
|
|
_MAX_DISPLAY_CHARS = 1000
|
|
|
|
|
|
class ThinkingWidget(Static):
|
|
"""Collapsible panel showing the model's extended thinking.
|
|
|
|
Uses Rich ``Panel`` for rendering, matching the Rich CLI output exactly.
|
|
|
|
Usage::
|
|
|
|
w = ThinkingWidget(show_thinking=True)
|
|
await container.mount(w)
|
|
w.append_text("reasoning chunk...")
|
|
w.finalize() # stop spinner
|
|
"""
|
|
|
|
DEFAULT_CSS = """
|
|
ThinkingWidget {
|
|
height: auto;
|
|
margin: 0 0 1 0;
|
|
}
|
|
"""
|
|
|
|
def __init__(self, *, show_thinking: bool = True) -> None:
|
|
super().__init__("")
|
|
self._content = ""
|
|
self._is_active = True
|
|
self._show = show_thinking
|
|
self._frame = 0
|
|
self._timer_handle = None
|
|
if not show_thinking:
|
|
self.display = False
|
|
|
|
def on_mount(self) -> None:
|
|
self._timer_handle = self.set_interval(0.1, self._tick)
|
|
self._refresh_display()
|
|
|
|
def _tick(self) -> None:
|
|
if self._is_active:
|
|
self._frame = (self._frame + 1) % len(_SPINNER_FRAMES)
|
|
self._refresh_display()
|
|
|
|
def _refresh_display(self) -> None:
|
|
if self._is_active:
|
|
char = _SPINNER_FRAMES[self._frame]
|
|
title = f"Thinking {char}"
|
|
else:
|
|
title = "Thinking"
|
|
|
|
display = self._content.rstrip()
|
|
if len(display) > _MAX_DISPLAY_CHARS:
|
|
display = "..." + display[-_MAX_DISPLAY_CHARS:]
|
|
|
|
body = Text(display, style="dim") if display else Text("...", style="dim")
|
|
self.update(Panel(body, title=title, border_style="blue", padding=(0, 1)))
|
|
|
|
def append_text(self, chunk: str) -> None:
|
|
"""Append a chunk of thinking text."""
|
|
self._content += chunk
|
|
self._refresh_display()
|
|
|
|
def finalize(self) -> None:
|
|
"""Mark thinking as complete — stop spinner, dim border."""
|
|
self._is_active = False
|
|
if self._timer_handle is not None:
|
|
self._timer_handle.stop()
|
|
self._timer_handle = None
|
|
self._refresh_display()
|