klaude-code 2.0.2__py3-none-any.whl → 2.1.0__py3-none-any.whl

klaude_code/{ui/renderers → tui/components}/metadata.py

@@ -5,12 +5,12 @@ from rich.padding import Padding
 from rich.text import Text
 
 from klaude_code.const import DEFAULT_MAX_TOKENS
+from klaude_code.log import is_debug_enabled
 from klaude_code.protocol import events, model
-from klaude_code.trace import is_debug_enabled
-from klaude_code.ui.renderers.common import create_grid
-from klaude_code.ui.rich.quote import Quote
-from klaude_code.ui.rich.theme import ThemeKey
-from klaude_code.ui.utils.common import format_model_params, format_number
+from klaude_code.tui.components.common import create_grid
+from klaude_code.tui.components.rich.quote import Quote
+from klaude_code.tui.components.rich.theme import ThemeKey
+from klaude_code.ui.common import format_model_params, format_number
 
 
 def _get_version() -> str:
@@ -99,7 +99,7 @@ def _render_task_metadata_block(
             )
         )
     if metadata.usage is not None:
-        # Context (only for main agent)
+        # Context usage
         if show_context_and_time and metadata.usage.context_usage_percent is not None:
             context_size = format_number(metadata.usage.context_size or 0)
             # Calculate effective limit (same as Usage.context_usage_percent)
@@ -169,7 +169,7 @@ def render_task_metadata(e: events.TaskMetadataEvent) -> RenderableType:
 
     # Render each sub-agent metadata block
     for meta in e.metadata.sub_agent_task_metadata:
-        renderables.append(_render_task_metadata_block(meta, is_sub_agent=True, show_context_and_time=False))
+        renderables.append(_render_task_metadata_block(meta, is_sub_agent=True, show_context_and_time=True))
 
     # Add total cost line when there are sub-agents
     if e.metadata.sub_agent_task_metadata:

klaude_code/{ui → tui/components}/rich/markdown.py

@@ -24,7 +24,7 @@ from klaude_code.const import (
     MARKDOWN_STREAM_SYNCHRONIZED_OUTPUT_ENABLED,
     UI_REFRESH_RATE_FPS,
 )
-from klaude_code.ui.rich.code_panel import CodePanel
+from klaude_code.tui.components.rich.code_panel import CodePanel
 
 
 class NoInsetCodeBlock(CodeBlock):
@@ -61,7 +61,13 @@ class Divider(MarkdownElement):
 
 class MarkdownTable(TableElement):
     def __rich_console__(self, console: Console, options: ConsoleOptions) -> RenderResult:
-        table = Table(box=box.MARKDOWN, border_style=console.get_style("markdown.table.border"))
+        # rich.box.MARKDOWN intentionally includes a blank top/bottom edge row. Rather than
+        # post-processing rendered segments, disable outer edges to avoid emitting those rows.
+        table = Table(
+            box=box.MARKDOWN,
+            show_edge=False,
+            border_style=console.get_style("markdown.table.border"),
+        )
 
         if self.header is not None and self.header.row is not None:
             for column in self.header.row.cells:
@@ -72,27 +78,7 @@ class MarkdownTable(TableElement):
                 row_content = [element.content for element in row.cells]
                 table.add_row(*row_content)
 
-        # Render table and strip top/bottom blank lines that MARKDOWN box adds
-        segments = list(console.render(table, options))
-
-        # Skip leading blank line (before first newline)
-        first_newline_idx = next((i for i, s in enumerate(segments) if s.text == "\n"), None)
-        if first_newline_idx is not None:
-            first_line = "".join(s.text for s in segments[:first_newline_idx])
-            if not first_line.strip():
-                segments = segments[first_newline_idx + 1 :]
-
-        # Skip trailing blank line (after last newline)
-        while len(segments) >= 2 and segments[-1].text == "\n":
-            prev_newline = next((i for i in range(len(segments) - 2, -1, -1) if segments[i].text == "\n"), None)
-            if prev_newline is not None:
-                between = "".join(s.text for s in segments[prev_newline + 1 : -1])
-                if not between.strip():
-                    segments = segments[: prev_newline + 1]
-                    continue
-            break
-
-        yield from segments
+        yield table
 
 
 class LeftHeading(Heading):

klaude_code/{ui → tui/components}/rich/status.py

@@ -23,8 +23,8 @@ from klaude_code.const import (
     STATUS_SHIMMER_BAND_HALF_WIDTH,
     STATUS_SHIMMER_PADDING,
 )
-from klaude_code.ui.rich.theme import ThemeKey
-from klaude_code.ui.terminal.color import get_last_terminal_background_rgb
+from klaude_code.tui.components.rich.theme import ThemeKey
+from klaude_code.tui.terminal.color import get_last_terminal_background_rgb
 
 # Use an existing Rich spinner name; BreathingSpinner overrides its rendering
 BREATHING_SPINNER_NAME = "dots"

klaude_code/{ui → tui/components}/rich/theme.py

@@ -138,6 +138,7 @@ class ThemeKey(str, Enum):
     STATUS_TEXT = "spinner.status.text"
     STATUS_TEXT_BOLD = "spinner.status.text.bold"
     STATUS_TEXT_BOLD_ITALIC = "spinner.status.text.bold_italic"
+    STATUS_TOAST = "spinner.status.toast"
     # STATUS
     STATUS_HINT = "status.hint"
     # USER_INPUT
@@ -262,6 +263,7 @@ def get_theme(theme: str | None = None) -> Themes:
                 ThemeKey.STATUS_TEXT.value: palette.blue,
                 ThemeKey.STATUS_TEXT_BOLD.value: "bold " + palette.blue,
                 ThemeKey.STATUS_TEXT_BOLD_ITALIC.value: "bold italic " + palette.blue,
+                ThemeKey.STATUS_TOAST.value: "bold " + palette.yellow,
                 ThemeKey.STATUS_HINT.value: palette.grey2,
                 # REMINDER
                 ThemeKey.REMINDER.value: palette.grey1,
@@ -275,7 +277,7 @@ def get_theme(theme: str | None = None) -> Themes:
                 ThemeKey.TOOL_RESULT.value: palette.grey_green,
                 ThemeKey.TOOL_RESULT_TREE_PREFIX.value: palette.grey3 + " dim",
                 ThemeKey.TOOL_RESULT_BOLD.value: "bold " + palette.grey_green,
-                ThemeKey.TOOL_RESULT_TRUNCATED.value: palette.grey1,
+                ThemeKey.TOOL_RESULT_TRUNCATED.value: palette.grey1 + " dim",
                 ThemeKey.TOOL_MARK.value: "bold",
                 ThemeKey.TOOL_APPROVED.value: palette.green + " bold reverse",
                 ThemeKey.TOOL_REJECTED.value: palette.red + " bold reverse",

klaude_code/{ui/renderers → tui/components}/sub_agent.py

@@ -1,19 +1,16 @@
 import json
 from typing import Any, cast
 
-from rich import box
 from rich.console import Group, RenderableType
 from rich.json import JSON
-from rich.panel import Panel
-from rich.style import Style
+from rich.style import Style, StyleType
 from rich.text import Text
 
 from klaude_code.const import SUB_AGENT_RESULT_MAX_LINES
 from klaude_code.protocol import events, model
 from klaude_code.protocol.sub_agent import get_sub_agent_profile_by_tool
-from klaude_code.ui.renderers.common import truncate_head
-from klaude_code.ui.rich.markdown import NoInsetMarkdown
-from klaude_code.ui.rich.theme import ThemeKey
+from klaude_code.tui.components.common import truncate_head
+from klaude_code.tui.components.rich.theme import ThemeKey
 
 
 def _compact_schema_value(value: dict[str, Any]) -> str | list[Any] | dict[str, Any]:
@@ -83,13 +80,11 @@ def render_sub_agent_result(
     result: str,
     *,
     code_theme: str,
-    style: Style | None = None,
+    style: StyleType | None = None,
     has_structured_output: bool = False,
     description: str | None = None,
-    panel_style: Style | None = None,
 ) -> RenderableType:
     stripped_result = result.strip()
-    result_panel_style = panel_style or ThemeKey.SUB_AGENT_RESULT_PANEL
 
     # Extract agentId footer for separate styling
     main_content, agent_id_footer = _extract_agent_id_footer(stripped_result)
@@ -106,15 +101,10 @@ def render_sub_agent_result(
                 JSON(stripped_result),
             ]
             if description:
-                group_elements.insert(0, NoInsetMarkdown(f"# {description}", code_theme=code_theme, style=style or ""))
+                group_elements.insert(0, Text(f"\n{description}", style=style or ""))
             if agent_id_footer:
                 group_elements.append(Text(agent_id_footer, style=ThemeKey.SUB_AGENT_FOOTER))
-            return Panel.fit(
-                Group(*group_elements),
-                box=box.SIMPLE,
-                border_style=ThemeKey.LINES,
-                style=result_panel_style,
-            )
+            return Group(*group_elements)
         except json.JSONDecodeError:
             # Fall back to markdown if not valid JSON
             pass
@@ -122,42 +112,32 @@ def render_sub_agent_result(
     if not stripped_result:
         return Text()
 
-    # Add markdown heading if description is provided for non-structured output
-    if description:
-        stripped_result = f"# {description}\n\n{stripped_result}"
-
     lines = stripped_result.splitlines()
     if len(lines) > SUB_AGENT_RESULT_MAX_LINES:
         hidden_count = len(lines) - SUB_AGENT_RESULT_MAX_LINES
-        head_count = SUB_AGENT_RESULT_MAX_LINES // 2
-        tail_count = SUB_AGENT_RESULT_MAX_LINES - head_count
-        head_text = "\n".join(lines[:head_count])
-        tail_text = "\n".join(lines[-tail_count:])
-        truncated_elements: list[RenderableType] = [
-            NoInsetMarkdown(head_text, code_theme=code_theme, style=style or ""),
+        tail_text = "\n".join(lines[-SUB_AGENT_RESULT_MAX_LINES:])
+        truncated_elements: list[RenderableType] = []
+        # Add description heading separately so it won't be truncated
+        if description:
+            truncated_elements.append(Text(f"---\n{description}", style=style or ""))
+        truncated_elements.append(
             Text(
-                f"( … more {hidden_count} lines — use /export to view full output )",
+                f"( … more {hidden_count} lines)",
                 style=ThemeKey.TOOL_RESULT_TRUNCATED,
-            ),
-            NoInsetMarkdown(tail_text, code_theme=code_theme, style=style or ""),
-        ]
+            )
+        )
+        truncated_elements.append(Text(tail_text, style=style or ""))
         if agent_id_footer:
             truncated_elements.append(Text(agent_id_footer, style=ThemeKey.SUB_AGENT_FOOTER))
-        return Panel.fit(
-            Group(*truncated_elements),
-            box=box.SIMPLE,
-            border_style=ThemeKey.LINES,
-            style=result_panel_style,
-        )
-    normal_elements: list[RenderableType] = [NoInsetMarkdown(stripped_result, code_theme=code_theme)]
+        return Group(*truncated_elements)
+
+    # No truncation needed - add description heading if provided
+    if description:
+        stripped_result = f"\n# {description}\n\n{stripped_result}"
+    normal_elements: list[RenderableType] = [Text(stripped_result)]
     if agent_id_footer:
         normal_elements.append(Text(agent_id_footer, style=ThemeKey.SUB_AGENT_FOOTER))
-    return Panel.fit(
-        Group(*normal_elements),
-        box=box.SIMPLE,
-        border_style=ThemeKey.LINES,
-        style=result_panel_style,
-    )
+    return Group(*normal_elements)
 
 
 def build_sub_agent_state_from_tool_call(e: events.ToolCallEvent) -> model.SubAgentState | None:

klaude_code/{ui/renderers → tui/components}/thinking.py

@@ -5,9 +5,9 @@ from rich.padding import Padding
 from rich.text import Text
 
 from klaude_code.const import MARKDOWN_RIGHT_MARGIN
-from klaude_code.ui.renderers.common import create_grid
-from klaude_code.ui.rich.markdown import ThinkingMarkdown
-from klaude_code.ui.rich.theme import ThemeKey
+from klaude_code.tui.components.common import create_grid
+from klaude_code.tui.components.rich.markdown import ThinkingMarkdown
+from klaude_code.tui.components.rich.theme import ThemeKey
 
 # UI markers
 THINKING_MESSAGE_MARK = "∴"

klaude_code/{ui/renderers → tui/components}/tools.py

@@ -17,14 +17,14 @@ from klaude_code.const import (
 )
 from klaude_code.protocol import events, model, tools
 from klaude_code.protocol.sub_agent import is_sub_agent_tool as _is_sub_agent_tool
-from klaude_code.ui.renderers import diffs as r_diffs
-from klaude_code.ui.renderers import mermaid_viewer as r_mermaid_viewer
-from klaude_code.ui.renderers.bash_syntax import highlight_bash_command
-from klaude_code.ui.renderers.common import create_grid, truncate_middle
-from klaude_code.ui.rich.code_panel import CodePanel
-from klaude_code.ui.rich.markdown import NoInsetMarkdown
-from klaude_code.ui.rich.quote import TreeQuote
-from klaude_code.ui.rich.theme import ThemeKey
+from klaude_code.tui.components import diffs as r_diffs
+from klaude_code.tui.components import mermaid_viewer as r_mermaid_viewer
+from klaude_code.tui.components.bash_syntax import highlight_bash_command
+from klaude_code.tui.components.common import create_grid, truncate_middle
+from klaude_code.tui.components.rich.code_panel import CodePanel
+from klaude_code.tui.components.rich.markdown import NoInsetMarkdown
+from klaude_code.tui.components.rich.quote import TreeQuote
+from klaude_code.tui.components.rich.theme import ThemeKey
 
 # Tool markers (Unicode symbols for UI display)
 MARK_GENERIC = "⚒"
@@ -487,7 +487,7 @@ def render_mermaid_tool_result(
     *,
     session_id: str | None = None,
 ) -> RenderableType:
-    from klaude_code.ui.terminal import supports_osc8_hyperlinks
+    from klaude_code.tui.terminal import supports_osc8_hyperlinks
 
     link_info = _extract_mermaid_link(tr.ui_extra)
     if link_info is None:

klaude_code/{ui/renderers → tui/components}/user_input.py

@@ -1,29 +1,12 @@
 import re
-from collections.abc import Callable
 
 from rich.console import Group, RenderableType
 from rich.text import Text
 
 from klaude_code.skill import get_available_skills
-from klaude_code.ui.renderers.common import create_grid
-from klaude_code.ui.rich.theme import ThemeKey
-
-# Module-level command name checker. Set by cli/runtime.py on startup.
-_command_name_checker: Callable[[str], bool] | None = None
-
-
-def set_command_name_checker(checker: Callable[[str], bool]) -> None:
-    """Set the command name validation function (called from runtime layer)."""
-    global _command_name_checker
-    _command_name_checker = checker
-
-
-def is_slash_command_name(name: str) -> bool:
-    """Check if name is a valid slash command using the injected checker."""
-    if _command_name_checker is None:
-        return False
-    return _command_name_checker(name)
-
+from klaude_code.tui.command import is_slash_command_name
+from klaude_code.tui.components.common import create_grid
+from klaude_code.tui.components.rich.theme import ThemeKey
 
 # Match @-file patterns only when they appear at the beginning of the line
 # or immediately after whitespace, to avoid treating mid-word email-like

klaude_code/tui/display.py

@@ -0,0 +1,85 @@
+from __future__ import annotations
+
+import asyncio
+import contextlib
+from collections.abc import Coroutine
+from typing import Any, override
+
+from klaude_code.protocol import events
+from klaude_code.tui.machine import DisplayStateMachine
+from klaude_code.tui.renderer import TUICommandRenderer
+from klaude_code.tui.terminal.notifier import TerminalNotifier
+from klaude_code.ui.core.display import DisplayABC
+
+
+class TUIDisplay(DisplayABC):
+    """Interactive terminal display using Rich for rendering."""
+
+    def __init__(self, theme: str | None = None, notifier: TerminalNotifier | None = None):
+        self._notifier = notifier or TerminalNotifier()
+        self._machine = DisplayStateMachine()
+        self._renderer = TUICommandRenderer(theme=theme, notifier=self._notifier)
+
+        self._sigint_toast_clear_handle: asyncio.Handle | None = None
+        self._bg_tasks: set[asyncio.Task[None]] = set()
+
+    def _create_bg_task(self, coro: Coroutine[Any, Any, None]) -> None:
+        task = asyncio.create_task(coro)
+        self._bg_tasks.add(task)
+        task.add_done_callback(self._bg_tasks.discard)
+
+    @override
+    async def consume_event(self, event: events.Event) -> None:
+        commands = self._machine.transition(event)
+        await self._renderer.execute(commands)
+
+    @override
+    async def start(self) -> None:
+        pass
+
+    @override
+    async def stop(self) -> None:
+        if self._sigint_toast_clear_handle is not None:
+            with contextlib.suppress(Exception):
+                self._sigint_toast_clear_handle.cancel()
+            self._sigint_toast_clear_handle = None
+
+        for task in list(self._bg_tasks):
+            with contextlib.suppress(Exception):
+                task.cancel()
+        self._bg_tasks.clear()
+
+        await self._renderer.stop()
+
+        with contextlib.suppress(Exception):
+            self._renderer.stop_bottom_live()
+
+    def show_sigint_exit_toast(self, *, window_seconds: float = 2.0) -> None:
+        """Show a transient Ctrl+C hint in the TUI status line."""
+
+        async def _apply_show() -> None:
+            await self._renderer.execute(self._machine.show_sigint_exit_toast())
+
+        async def _apply_clear() -> None:
+            await self._renderer.execute(self._machine.clear_sigint_exit_toast())
+
+        loop = asyncio.get_running_loop()
+        self._create_bg_task(_apply_show())
+
+        if self._sigint_toast_clear_handle is not None:
+            with contextlib.suppress(Exception):
+                self._sigint_toast_clear_handle.cancel()
+            self._sigint_toast_clear_handle = None
+
+        def _schedule_clear() -> None:
+            self._create_bg_task(_apply_clear())
+
+        self._sigint_toast_clear_handle = loop.call_later(window_seconds, _schedule_clear)
+
+    def hide_progress_ui(self) -> None:
+        """Stop transient Rich UI elements before prompt-toolkit takes control."""
+
+        with contextlib.suppress(Exception):
+            self._renderer.spinner_stop()
+        with contextlib.suppress(Exception):
+            self._renderer.stop_bottom_live()

klaude_code/{ui/modes/repl → tui/input}/__init__.py

@@ -1,4 +1,4 @@
-from klaude_code.ui.modes.repl.input_prompt_toolkit import REPLStatusSnapshot
+from klaude_code.tui.input.prompt_toolkit import REPLStatusSnapshot
 
 
 def build_repl_status_snapshot(update_message: str | None) -> REPLStatusSnapshot:

klaude_code/{ui/modes/repl → tui/input}/completers.py

@@ -28,8 +28,8 @@ from prompt_toolkit.document import Document
 from prompt_toolkit.formatted_text import FormattedText
 
 from klaude_code.const import COMPLETER_CACHE_TTL_SEC, COMPLETER_CMD_TIMEOUT_SEC, COMPLETER_DEBOUNCE_SEC
+from klaude_code.log import DebugType, log_debug
 from klaude_code.protocol.commands import CommandInfo
-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

klaude_code/{ui/modes/repl/input_prompt_toolkit.py → tui/input/prompt_toolkit.py}

@@ -35,13 +35,13 @@ from klaude_code.config.thinking import (
 from klaude_code.protocol import llm_param
 from klaude_code.protocol.commands import CommandInfo
 from klaude_code.protocol.message import UserInputPayload
+from klaude_code.tui.components.user_input import USER_MESSAGE_MARK
+from klaude_code.tui.input.clipboard import capture_clipboard_tag, copy_to_clipboard, extract_images_from_text
+from klaude_code.tui.input.completers import AT_TOKEN_PATTERN, create_repl_completer
+from klaude_code.tui.input.key_bindings import create_key_bindings
+from klaude_code.tui.terminal.color import is_light_terminal_background
+from klaude_code.tui.terminal.selector import SelectItem, SelectOverlay, build_model_select_items
 from klaude_code.ui.core.input import InputProviderABC
-from klaude_code.ui.modes.repl.clipboard import capture_clipboard_tag, copy_to_clipboard, extract_images_from_text
-from klaude_code.ui.modes.repl.completers import AT_TOKEN_PATTERN, create_repl_completer
-from klaude_code.ui.modes.repl.key_bindings import create_key_bindings
-from klaude_code.ui.renderers.user_input import USER_MESSAGE_MARK
-from klaude_code.ui.terminal.color import is_light_terminal_background
-from klaude_code.ui.terminal.selector import SelectItem, SelectOverlay, build_model_select_items
 
 
 class REPLStatusSnapshot(NamedTuple):