klaude-code 2.2.0__py3-none-any.whl → 2.3.0__py3-none-any.whl

klaude_code/tui/command/sub_agent_model_cmd.py

@@ -0,0 +1,190 @@
+"""Command for changing sub-agent models."""
+
+from __future__ import annotations
+
+import asyncio
+
+from prompt_toolkit.styles import Style
+
+from klaude_code.config.config import load_config
+from klaude_code.config.sub_agent_model_helper import SubAgentModelHelper, SubAgentModelInfo
+from klaude_code.protocol import commands, events, message, model, op
+from klaude_code.tui.terminal.selector import SelectItem, build_model_select_items, select_one
+
+from .command_abc import Agent, CommandABC, CommandResult
+
+SELECT_STYLE = Style(
+    [
+        ("instruction", "ansibrightblack"),
+        ("pointer", "ansigreen"),
+        ("highlighted", "ansigreen"),
+        ("text", "ansibrightblack"),
+        ("question", "bold"),
+        ("meta", "fg:ansibrightblack"),
+        ("msg", ""),
+    ]
+)
+
+USE_DEFAULT_BEHAVIOR = "__default__"
+
+
+def _build_sub_agent_select_items(
+    sub_agents: list[SubAgentModelInfo],
+    helper: SubAgentModelHelper,
+    main_model_name: str,
+) -> list[SelectItem[str]]:
+    """Build SelectItem list for sub-agent selection."""
+    items: list[SelectItem[str]] = []
+    max_name_len = max(len(sa.profile.name) for sa in sub_agents) if sub_agents else 0
+
+    for sa in sub_agents:
+        name = sa.profile.name
+
+        if sa.configured_model:
+            model_display = sa.configured_model
+        else:
+            behavior = helper.describe_empty_model_config_behavior(name, main_model_name=main_model_name)
+            model_display = f"({behavior.description})"
+
+        title = [
+            ("class:msg", f"{name:<{max_name_len}}"),
+            ("class:meta", f"  current: {model_display}\n"),
+        ]
+        items.append(SelectItem(title=title, value=name, search_text=name))
+
+    return items
+
+
+def _select_sub_agent_sync(
+    sub_agents: list[SubAgentModelInfo],
+    helper: SubAgentModelHelper,
+    main_model_name: str,
+) -> str | None:
+    """Synchronous sub-agent type selection."""
+    items = _build_sub_agent_select_items(sub_agents, helper, main_model_name)
+    if not items:
+        return None
+
+    try:
+        result = select_one(
+            message="Select sub-agent to configure:",
+            items=items,
+            pointer="->",
+            style=SELECT_STYLE,
+            use_search_filter=False,
+        )
+        return result if isinstance(result, str) else None
+    except KeyboardInterrupt:
+        return None
+
+
+def _select_model_for_sub_agent_sync(
+    helper: SubAgentModelHelper,
+    sub_agent_type: str,
+    main_model_name: str,
+) -> str | None:
+    """Synchronous model selection for a sub-agent."""
+    models = helper.get_selectable_models(sub_agent_type)
+
+    default_behavior = helper.describe_empty_model_config_behavior(sub_agent_type, main_model_name=main_model_name)
+
+    inherit_item = SelectItem[str](
+        title=[
+            ("class:msg", "(Use default behavior)"),
+            ("class:meta", f"  -> {default_behavior.description}\n"),
+        ],
+        value=USE_DEFAULT_BEHAVIOR,
+        search_text="default unset",
+    )
+    model_items = build_model_select_items(models)
+    all_items = [inherit_item, *model_items]
+
+    try:
+        result = select_one(
+            message=f"Select model for {sub_agent_type}:",
+            items=all_items,
+            pointer="->",
+            style=SELECT_STYLE,
+            use_search_filter=True,
+        )
+        return result if isinstance(result, str) else None
+    except KeyboardInterrupt:
+        return None
+
+
+class SubAgentModelCommand(CommandABC):
+    """Configure models for sub-agents (Task, Explore, WebAgent, ImageGen)."""
+
+    @property
+    def name(self) -> commands.CommandName:
+        return commands.CommandName.SUB_AGENT_MODEL
+
+    @property
+    def summary(self) -> str:
+        return "Change sub-agent models"
+
+    @property
+    def is_interactive(self) -> bool:
+        return True
+
+    async def run(self, agent: Agent, user_input: message.UserInputPayload) -> CommandResult:
+        config = load_config()
+        helper = SubAgentModelHelper(config)
+        main_model_name = agent.get_llm_client().model_name
+
+        sub_agents = helper.get_available_sub_agents()
+        if not sub_agents:
+            return CommandResult(
+                events=[
+                    events.DeveloperMessageEvent(
+                        session_id=agent.session.id,
+                        item=message.DeveloperMessage(
+                            parts=message.text_parts_from_str("No sub-agents available"),
+                            ui_extra=model.build_command_output_extra(self.name),
+                        ),
+                    )
+                ]
+            )
+
+        selected_sub_agent = await asyncio.to_thread(_select_sub_agent_sync, sub_agents, helper, main_model_name)
+        if selected_sub_agent is None:
+            return CommandResult(
+                events=[
+                    events.DeveloperMessageEvent(
+                        session_id=agent.session.id,
+                        item=message.DeveloperMessage(
+                            parts=message.text_parts_from_str("(cancelled)"),
+                            ui_extra=model.build_command_output_extra(self.name),
+                        ),
+                    )
+                ]
+            )
+
+        selected_model = await asyncio.to_thread(
+            _select_model_for_sub_agent_sync, helper, selected_sub_agent, main_model_name
+        )
+        if selected_model is None:
+            return CommandResult(
+                events=[
+                    events.DeveloperMessageEvent(
+                        session_id=agent.session.id,
+                        item=message.DeveloperMessage(
+                            parts=message.text_parts_from_str("(cancelled)"),
+                            ui_extra=model.build_command_output_extra(self.name),
+                        ),
+                    )
+                ]
+            )
+
+        model_name: str | None = None if selected_model == USE_DEFAULT_BEHAVIOR else selected_model
+
+        return CommandResult(
+            operations=[
+                op.ChangeSubAgentModelOperation(
+                    session_id=agent.session.id,
+                    sub_agent_type=selected_sub_agent,
+                    model_name=model_name,
+                    save_as_default=True,
+                )
+            ]
+        )

klaude_code/tui/components/bash_syntax.py

@@ -3,7 +3,7 @@
 import re
 from typing import Any
 
-from pygments.lexers import BashLexer  # pyright: ignore[reportUnknownVariableType]
+from pygments.lexers.shell import BashLexer  # pyright: ignore[reportMissingTypeStubs, reportUnknownVariableType]
 from pygments.token import Token
 from rich.text import Text
 

klaude_code/tui/components/common.py

@@ -106,7 +106,7 @@ def truncate_head(
     text = text.expandtabs(TAB_EXPAND_WIDTH)
     lines = [line for line in text.split("\n") if line.strip()]
 
-    out = Text()
+    out = Text(overflow="fold")
     if base_style is not None:
         out.style = base_style
 

klaude_code/tui/components/developer.py

@@ -5,7 +5,6 @@ from rich.text import Text
 
 from klaude_code.protocol import commands, events, message, model
 from klaude_code.tui.components.common import create_grid, truncate_middle
-from klaude_code.tui.components.rich.markdown import NoInsetMarkdown
 from klaude_code.tui.components.rich.theme import ThemeKey
 from klaude_code.tui.components.tools import render_path
 
@@ -142,12 +141,8 @@ def render_command_output(e: events.DeveloperMessageEvent) -> RenderableType:
 
     content = message.join_text_parts(e.item.parts)
     match command_output.command_name:
-        case commands.CommandName.HELP:
-            return Padding.indent(Text.from_markup(content or "", style=ThemeKey.TOOL_RESULT), level=2)
         case commands.CommandName.STATUS:
             return _render_status_output(command_output)
-        case commands.CommandName.RELEASE_NOTES:
-            return Padding.indent(NoInsetMarkdown(content or ""), level=2)
         case commands.CommandName.FORK_SESSION:
             return _render_fork_session_output(command_output)
         case _:

klaude_code/tui/components/metadata.py

@@ -1,24 +1,12 @@
-from importlib.metadata import PackageNotFoundError, version
-
 from rich.console import Group, RenderableType
 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.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:
-    """Get the current version of klaude-code."""
-    try:
-        return version("klaude-code")
-    except PackageNotFoundError:
-        return "unknown"
+from klaude_code.ui.common import format_number
 
 
 def _render_task_metadata_block(
@@ -195,53 +183,3 @@ def render_task_metadata(e: events.TaskMetadataEvent) -> RenderableType:
         renderables.append(Padding(grid, (0, 0, 0, 2)))
 
     return Group(*renderables)
-
-
-def render_welcome(e: events.WelcomeEvent) -> RenderableType:
-    """Render the welcome panel with model info and settings.
-
-    Args:
-        e: The welcome event.
-    """
-    debug_mode = is_debug_enabled()
-
-    panel_content = Text()
-
-    if e.show_klaude_code_info:
-        # First line: Klaude Code version
-        klaude_code_style = ThemeKey.WELCOME_DEBUG_TITLE if debug_mode else ThemeKey.WELCOME_HIGHLIGHT_BOLD
-        panel_content.append_text(Text("Klaude Code", style=klaude_code_style))
-        panel_content.append_text(Text(f" v{_get_version()}", style=ThemeKey.WELCOME_INFO))
-        panel_content.append_text(Text("\n"))
-
-    # Model line: model @ provider · params...
-    panel_content.append_text(
-        Text.assemble(
-            (str(e.llm_config.model), ThemeKey.WELCOME_HIGHLIGHT),
-            (" @ ", ThemeKey.WELCOME_INFO),
-            (e.llm_config.provider_name, ThemeKey.WELCOME_INFO),
-        )
-    )
-
-    # Use format_model_params for consistent formatting
-    param_strings = format_model_params(e.llm_config)
-
-    # Render config items with tree-style prefixes
-    for i, param_str in enumerate(param_strings):
-        is_last = i == len(param_strings) - 1
-        prefix = "└─ " if is_last else "├─ "
-        panel_content.append_text(
-            Text.assemble(
-                ("\n", ThemeKey.WELCOME_INFO),
-                (prefix, ThemeKey.LINES),
-                (param_str, ThemeKey.WELCOME_INFO),
-            )
-        )
-
-    border_style = ThemeKey.WELCOME_DEBUG_BORDER if debug_mode else ThemeKey.LINES
-
-    if e.show_klaude_code_info:
-        groups = ["", Quote(panel_content, style=border_style, prefix="▌ "), ""]
-    else:
-        groups = [Quote(panel_content, style=border_style, prefix="▌ "), ""]
-    return Group(*groups)

klaude_code/tui/components/rich/cjk_wrap.py

@@ -4,6 +4,7 @@ from __future__ import annotations
 
 import unicodedata
 from collections.abc import Callable
+from typing import Any, cast
 
 
 def _is_cjk_char(ch: str) -> bool:
@@ -222,7 +223,7 @@ def install_rich_cjk_wrap_patch() -> bool:
 
         return break_positions
 
-    _wrap.divide_line = divide_line_patched  # pyright: ignore[reportPrivateImportUsage]
-    _text.divide_line = divide_line_patched  # pyright: ignore[reportPrivateImportUsage]
+    cast(Any, _wrap).divide_line = divide_line_patched  # pyright: ignore[reportPrivateImportUsage]
+    cast(Any, _text).divide_line = divide_line_patched  # pyright: ignore[reportPrivateImportUsage]
     _rich_cjk_wrap_patch_installed = True
     return True

klaude_code/tui/components/rich/status.py

@@ -263,6 +263,53 @@ def _breathing_style(console: Console, base_style: Style, intensity: float) -> S
     return base_style + Style(color=breathing_color)
 
 
+def truncate_left(text: Text, max_cells: int, *, console: Console, ellipsis: str = "…") -> Text:
+    """Left-truncate Text to fit within max_cells.
+
+    Keeps the rightmost part of the text and prepends an ellipsis when truncation occurs.
+    Uses cell width so wide characters are handled reasonably.
+    """
+
+    max_cells = max(0, int(max_cells))
+    if max_cells == 0:
+        return Text("")
+
+    if cell_len(text.plain) <= max_cells:
+        return text
+
+    ellipsis_cells = cell_len(ellipsis)
+    if max_cells <= ellipsis_cells:
+        # Not enough space to show any meaningful suffix.
+        clipped = Text(ellipsis, style=text.style)
+        clipped.truncate(max_cells, overflow="crop", pad=False)
+        return clipped
+
+    suffix_budget = max_cells - ellipsis_cells
+    plain = text.plain
+
+    suffix_cells = 0
+    start_index = len(plain)
+    for i in range(len(plain) - 1, -1, -1):
+        ch_cells = cell_len(plain[i])
+        if suffix_cells + ch_cells > suffix_budget:
+            break
+        suffix_cells += ch_cells
+        start_index = i
+        if suffix_cells == suffix_budget:
+            break
+
+    if start_index >= len(plain):
+        return Text(ellipsis, style=text.style)
+
+    suffix = text[start_index:]
+    try:
+        ellipsis_style = suffix.get_style_at_offset(console, 0)
+    except Exception:
+        ellipsis_style = suffix.style or text.style
+
+    return Text.assemble(Text(ellipsis, style=ellipsis_style), suffix)
+
+
 class ShimmerStatusText:
     """Renderable status line with shimmer effect on the main text and hint.
 
@@ -322,12 +369,11 @@ class _StatusLeftText:
         # If the hint itself can't fit, fall back to truncating the combined text.
         if max_width <= hint_cells:
             combined = Text.assemble(main_text, hint_text)
-            combined.truncate(max(1, max_width), overflow="ellipsis", pad=False)
-            yield combined
+            yield truncate_left(combined, max(1, max_width), console=console)
             return
 
         main_budget = max_width - hint_cells
-        main_text.truncate(max(1, main_budget), overflow="ellipsis", pad=False)
+        main_text = truncate_left(main_text, max(1, main_budget), console=console)
         yield Text.assemble(main_text, hint_text)
 
 

klaude_code/tui/components/rich/theme.py

@@ -191,6 +191,7 @@ class ThemeKey(str, Enum):
     WELCOME_HIGHLIGHT_BOLD = "welcome.highlight.bold"
     WELCOME_HIGHLIGHT = "welcome.highlight"
     WELCOME_INFO = "welcome.info"
+    WELCOME_INFO_BOLD = "welcome.info.bold"
     # WELCOME DEBUG
     WELCOME_DEBUG_TITLE = "welcome.debug.title"
     WELCOME_DEBUG_BORDER = "welcome.debug.border"
@@ -307,6 +308,7 @@ def get_theme(theme: str | None = None) -> Themes:
                 ThemeKey.WELCOME_HIGHLIGHT_BOLD.value: "bold",
                 ThemeKey.WELCOME_HIGHLIGHT.value: palette.blue,
                 ThemeKey.WELCOME_INFO.value: palette.grey1,
+                ThemeKey.WELCOME_INFO_BOLD.value: "bold " + palette.grey1,
                 # WELCOME DEBUG
                 ThemeKey.WELCOME_DEBUG_TITLE.value: "bold " + palette.red,
                 ThemeKey.WELCOME_DEBUG_BORDER.value: palette.red,

klaude_code/tui/components/sub_agent.py

@@ -3,7 +3,7 @@ from typing import Any, cast
 
 from rich.console import Group, RenderableType
 from rich.json import JSON
-from rich.style import Style, StyleType
+from rich.style import Style
 from rich.text import Text
 
 from klaude_code.const import SUB_AGENT_RESULT_MAX_LINES
@@ -79,65 +79,44 @@ def _extract_agent_id_footer(text: str) -> tuple[str, str | None]:
 def render_sub_agent_result(
     result: str,
     *,
-    code_theme: str,
-    style: StyleType | None = None,
     has_structured_output: bool = False,
     description: str | None = None,
 ) -> RenderableType:
     stripped_result = result.strip()
-
-    # Extract agentId footer for separate styling
     main_content, agent_id_footer = _extract_agent_id_footer(stripped_result)
     stripped_result = main_content.strip()
 
-    # Use rich JSON for structured output
+    elements: list[RenderableType] = []
+    if description:
+        elements.append(Text(f"---\n{description}", style=ThemeKey.TOOL_RESULT))
+
+    # Try structured JSON output first
+    use_text_rendering = True
     if has_structured_output:
         try:
-            group_elements: list[RenderableType] = [
-                Text(
-                    "use /export to view full output",
-                    style=ThemeKey.TOOL_RESULT,
-                ),
-                JSON(stripped_result),
-            ]
-            if description:
-                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 Group(*group_elements)
+            elements.append(Text("use /export to view full output", style=ThemeKey.TOOL_RESULT_TRUNCATED))
+            elements.append(JSON(stripped_result))
+            use_text_rendering = False
         except json.JSONDecodeError:
-            # Fall back to markdown if not valid JSON
             pass
 
-    if not stripped_result:
-        return Text()
-
-    lines = stripped_result.splitlines()
-    if len(lines) > SUB_AGENT_RESULT_MAX_LINES:
-        hidden_count = len(lines) - SUB_AGENT_RESULT_MAX_LINES
-        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)",
-                style=ThemeKey.TOOL_RESULT_TRUNCATED,
-            )
-        )
-        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 Group(*truncated_elements)
+    # Text rendering (either fallback or non-structured)
+    if use_text_rendering:
+        if not stripped_result:
+            return Text()
+
+        lines = stripped_result.splitlines()
+        if len(lines) > SUB_AGENT_RESULT_MAX_LINES:
+            hidden_count = len(lines) - SUB_AGENT_RESULT_MAX_LINES
+            elements.append(Text(f"( ... more {hidden_count} lines)", style=ThemeKey.TOOL_RESULT_TRUNCATED))
+            elements.append(Text("\n".join(lines[-SUB_AGENT_RESULT_MAX_LINES:]), style=ThemeKey.TOOL_RESULT))
+        else:
+            elements.append(Text(stripped_result, style=ThemeKey.TOOL_RESULT))
 
-    # 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 Group(*normal_elements)
+        elements.append(Text(agent_id_footer, style=ThemeKey.SUB_AGENT_FOOTER))
+
+    return Group(*elements)
 
 
 def build_sub_agent_state_from_tool_call(e: events.ToolCallEvent) -> model.SubAgentState | None:

klaude_code/tui/components/welcome.py

@@ -0,0 +1,99 @@
+from importlib.metadata import PackageNotFoundError, version
+
+from rich.console import Group, RenderableType
+from rich.text import Text
+
+from klaude_code.log import is_debug_enabled
+from klaude_code.protocol import events
+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
+
+
+def _get_version() -> str:
+    """Get the current version of klaude-code."""
+    try:
+        return version("klaude-code")
+    except PackageNotFoundError:
+        return "unknown"
+
+
+def render_welcome(e: events.WelcomeEvent) -> RenderableType:
+    """Render the welcome panel with model info and settings.
+
+    Args:
+        e: The welcome event.
+    """
+    debug_mode = is_debug_enabled()
+
+    panel_content = Text()
+
+    if e.show_klaude_code_info:
+        # First line: Klaude Code version
+        klaude_code_style = ThemeKey.WELCOME_DEBUG_TITLE if debug_mode else ThemeKey.WELCOME_HIGHLIGHT_BOLD
+        panel_content.append_text(Text("Klaude Code", style=klaude_code_style))
+        panel_content.append_text(Text(f" v{_get_version()}", style=ThemeKey.WELCOME_INFO))
+        panel_content.append_text(Text("\n"))
+
+    # Model line: model @ provider · params...
+    panel_content.append_text(
+        Text.assemble(
+            (str(e.llm_config.model), ThemeKey.WELCOME_HIGHLIGHT),
+            (" @ ", ThemeKey.WELCOME_INFO),
+            (e.llm_config.provider_name, ThemeKey.WELCOME_INFO),
+        )
+    )
+
+    # Use format_model_params for consistent formatting
+    param_strings = format_model_params(e.llm_config)
+
+    # Check if we have sub-agent models to show
+    has_sub_agents = e.show_sub_agent_models and e.sub_agent_models
+
+    # Render config items with tree-style prefixes
+    for i, param_str in enumerate(param_strings):
+        is_last = i == len(param_strings) - 1 and not has_sub_agents
+        prefix = "└─ " if is_last else "├─ "
+        panel_content.append_text(
+            Text.assemble(
+                ("\n", ThemeKey.WELCOME_INFO),
+                (prefix, ThemeKey.LINES),
+                (param_str, ThemeKey.WELCOME_INFO),
+            )
+        )
+
+    # Render sub-agent models
+    if has_sub_agents:
+        # Add sub-agents header with tree prefix
+        panel_content.append_text(
+            Text.assemble(
+                ("\n", ThemeKey.WELCOME_INFO),
+                ("└─ ", ThemeKey.LINES),
+                ("sub-agents:", ThemeKey.WELCOME_INFO),
+            )
+        )
+        sub_agent_items = list(e.sub_agent_models.items())
+        max_type_len = max(len(t) for t in e.sub_agent_models)
+        for i, (sub_agent_type, sub_llm_config) in enumerate(sub_agent_items):
+            is_last = i == len(sub_agent_items) - 1
+            prefix = "└─ " if is_last else "├─ "
+            panel_content.append_text(
+                Text.assemble(
+                    ("\n", ThemeKey.WELCOME_INFO),
+                    ("   ", ThemeKey.WELCOME_INFO),  # Indentation for sub-items
+                    (prefix, ThemeKey.LINES),
+                    (sub_agent_type.lower().ljust(max_type_len), ThemeKey.WELCOME_INFO),
+                    (": ", ThemeKey.LINES),
+                    (str(sub_llm_config.model), ThemeKey.WELCOME_HIGHLIGHT),
+                    (" @ ", ThemeKey.WELCOME_INFO),
+                    (sub_llm_config.provider_name, ThemeKey.WELCOME_INFO),
+                )
+            )
+
+    border_style = ThemeKey.WELCOME_DEBUG_BORDER if debug_mode else ThemeKey.LINES
+
+    if e.show_klaude_code_info:
+        groups = ["", Quote(panel_content, style=border_style, prefix="▌ "), ""]
+    else:
+        groups = [Quote(panel_content, style=border_style, prefix="▌ "), ""]
+    return Group(*groups)

klaude_code/tui/input/prompt_toolkit.py

@@ -455,7 +455,7 @@ class PromptToolkitInput(InputProviderABC):
         config = load_config()
         models: list[ModelEntry] = sorted(
             config.iter_model_entries(only_available=True),
-            key=lambda m: m.model_name.lower(),
+            key=lambda m: (m.model_name.lower(), m.provider.lower()),
         )
         if not models:
             return [], None
@@ -468,6 +468,15 @@ class PromptToolkitInput(InputProviderABC):
                 initial = self._get_current_model_config_name()
         if initial is None:
             initial = config.main_model
+        if isinstance(initial, str) and initial and "@" not in initial:
+            try:
+                resolved = config.resolve_model_location_prefer_available(initial) or config.resolve_model_location(
+                    initial
+                )
+            except ValueError:
+                resolved = None
+            if resolved is not None:
+                initial = f"{resolved[0]}@{resolved[1]}"
         return items, initial
 
     def _open_model_picker(self) -> None:
@@ -608,6 +617,10 @@ class PromptToolkitInput(InputProviderABC):
                 (symbol_style, " ctrl-t "),
                 (text_style, " "),
                 (text_style, "think"),
+                (text_style, "  "),
+                (symbol_style, " ctrl-v "),
+                (text_style, " "),
+                (text_style, "paste image"),
             ]
         )
 

klaude_code/tui/renderer.py

@@ -61,6 +61,7 @@ from klaude_code.tui.components import sub_agent as c_sub_agent
 from klaude_code.tui.components import thinking as c_thinking
 from klaude_code.tui.components import tools as c_tools
 from klaude_code.tui.components import user_input as c_user_input
+from klaude_code.tui.components import welcome as c_welcome
 from klaude_code.tui.components.common import truncate_head, truncate_middle
 from klaude_code.tui.components.rich import status as r_status
 from klaude_code.tui.components.rich.live import CropAboveLive, SingleLine
@@ -511,7 +512,7 @@ class TUICommandRenderer:
             self.print()
 
     def display_welcome(self, event: events.WelcomeEvent) -> None:
-        self.print(c_metadata.render_welcome(event))
+        self.print(c_welcome.render_welcome(event))
 
     def display_user_message(self, event: events.UserMessageEvent) -> None:
         self.print(c_user_input.render_user_input(event.content))
@@ -567,10 +568,8 @@ class TUICommandRenderer:
                 self.print(
                     c_sub_agent.render_sub_agent_result(
                         event.task_result,
-                        code_theme=self.themes.code_theme,
                         has_structured_output=event.has_structured_output,
                         description=description,
-                        style=ThemeKey.TOOL_RESULT,
                     )
                 )