klaude-code 1.2.23__py3-none-any.whl → 1.2.24__py3-none-any.whl

klaude_code/ui/modes/repl/renderer.py

@@ -5,14 +5,13 @@ from contextlib import contextmanager
 from dataclasses import dataclass
 from typing import Any
 
-from rich import box
-from rich.box import Box
 from rich.console import Console
 from rich.spinner import Spinner
 from rich.status import Status
 from rich.style import Style, StyleType
 from rich.text import Text
 
+from klaude_code import const
 from klaude_code.protocol import events, model
 from klaude_code.ui.renderers import assistant as r_assistant
 from klaude_code.ui.renderers import developer as r_developer
@@ -32,6 +31,7 @@ from klaude_code.ui.rich.theme import ThemeKey, get_theme
 @dataclass
 class SessionStatus:
     color: Style | None = None
+    color_index: int | None = None
     sub_agent_state: model.SubAgentState | None = None
 
 
@@ -43,9 +43,9 @@ class REPLRenderer:
         self.console: Console = Console(theme=self.themes.app_theme)
         self.console.push_theme(self.themes.markdown_theme)
         self._spinner: Status = self.console.status(
-            ShimmerStatusText("Thinking …", ThemeKey.SPINNER_STATUS_TEXT),
+            ShimmerStatusText(const.STATUS_DEFAULT_TEXT),
             spinner=r_status.spinner_name(),
-            spinner_style=ThemeKey.SPINNER_STATUS,
+            spinner_style=ThemeKey.STATUS_SPINNER,
         )
 
         self.session_map: dict[str, SessionStatus] = {}
@@ -57,7 +57,9 @@ class REPLRenderer:
             sub_agent_state=sub_agent_state,
         )
         if sub_agent_state is not None:
-            session_status.color = self.pick_sub_agent_color()
+            color, color_index = self.pick_sub_agent_color()
+            session_status.color = color
+            session_status.color_index = color_index
         self.session_map[session_id] = session_status
 
     def is_sub_agent_session(self, session_id: str) -> bool:
@@ -70,12 +72,12 @@ class REPLRenderer:
             return
         self.sub_agent_color_index = (self.sub_agent_color_index + 1) % palette_size
 
-    def pick_sub_agent_color(self) -> Style:
+    def pick_sub_agent_color(self) -> tuple[Style, int]:
         self._advance_sub_agent_color_index()
         palette = self.themes.sub_agent_colors
         if not palette:
-            return Style()
-        return palette[self.sub_agent_color_index]
+            return Style(), 0
+        return palette[self.sub_agent_color_index], self.sub_agent_color_index
 
     def get_session_sub_agent_color(self, session_id: str) -> Style:
         status = self.session_map.get(session_id)
@@ -83,8 +85,12 @@ class REPLRenderer:
             return status.color
         return Style()
 
-    def box_style(self) -> Box:
-        return box.ROUNDED
+    def get_session_sub_agent_background(self, session_id: str) -> Style:
+        status = self.session_map.get(session_id)
+        backgrounds = self.themes.sub_agent_backgrounds
+        if status and status.color_index is not None and backgrounds:
+            return backgrounds[status.color_index]
+        return Style()
 
     @contextmanager
     def session_print_context(self, session_id: str) -> Iterator[None]:
@@ -114,7 +120,7 @@ class REPLRenderer:
     def display_tool_call_result(self, e: events.ToolResultEvent) -> None:
         if r_tools.is_sub_agent_tool(e.tool_name):
             return
-        renderable = r_tools.render_tool_result(e)
+        renderable = r_tools.render_tool_result(e, code_theme=self.themes.code_theme)
         if renderable is not None:
             self.print(renderable)
 
@@ -152,7 +158,6 @@ class REPLRenderer:
                     case events.ThinkingEvent() as e:
                         if is_sub_agent:
                             continue
-                        self.display_thinking_prefix()
                         self.display_thinking(e.content)
                     case events.DeveloperMessageEvent() as e:
                         self.display_developer_message(e)
@@ -196,7 +201,7 @@ class REPLRenderer:
             self.print()
 
     def display_welcome(self, event: events.WelcomeEvent) -> None:
-        self.print(r_metadata.render_welcome(event, box_style=self.box_style()))
+        self.print(r_metadata.render_welcome(event))
 
     def display_user_message(self, event: events.UserMessageEvent) -> None:
         self.print(r_user_input.render_user_input(event.content))
@@ -229,12 +234,17 @@ class REPLRenderer:
 
     def display_task_finish(self, event: events.TaskFinishEvent) -> None:
         if self.is_sub_agent_session(event.session_id):
+            session_status = self.session_map.get(event.session_id)
+            description = session_status.sub_agent_state.sub_agent_desc if session_status and session_status.sub_agent_state else None
+            panel_style = self.get_session_sub_agent_background(event.session_id)
             with self.session_print_context(event.session_id):
                 self.print(
                     r_sub_agent.render_sub_agent_result(
                         event.task_result,
                         code_theme=self.themes.code_theme,
                         has_structured_output=event.has_structured_output,
+                        description=description,
+                        panel_style=panel_style,
                     )
                 )
 
@@ -249,9 +259,6 @@ class REPLRenderer:
             )
         )
 
-    def display_thinking_prefix(self) -> None:
-        self.print(r_thinking.thinking_prefix())
-
     # -------------------------------------------------------------------------
     # Spinner control methods
     # -------------------------------------------------------------------------
@@ -266,7 +273,7 @@ class REPLRenderer:
 
     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))
+        self._spinner.update(ShimmerStatusText(status_text, right_text))
 
     def spinner_renderable(self) -> Spinner:
         """Return the spinner's renderable for embedding in other components."""

klaude_code/ui/renderers/assistant.py

@@ -6,7 +6,7 @@ from klaude_code.ui.renderers.common import create_grid
 from klaude_code.ui.rich.markdown import NoInsetMarkdown
 
 # UI markers
-ASSISTANT_MESSAGE_MARK = "➤"
+ASSISTANT_MESSAGE_MARK = "◆"
 
 
 def render_assistant_message(content: str, *, code_theme: str) -> RenderableType | None:

klaude_code/ui/renderers/metadata.py

@@ -1,7 +1,6 @@
 from importlib.metadata import version
 
 from rich import box
-from rich.box import Box
 from rich.console import Group, RenderableType
 from rich.padding import Padding
 from rich.panel import Panel
@@ -165,11 +164,8 @@ def render_task_metadata(e: events.TaskMetadataEvent) -> RenderableType:
     return Group(*renderables)
 
 
-def render_welcome(e: events.WelcomeEvent, *, box_style: Box | None = None) -> RenderableType:
+def render_welcome(e: events.WelcomeEvent) -> RenderableType:
     """Render the welcome panel with model info and settings."""
-    if box_style is None:
-        box_style = box.ROUNDED
-
     debug_mode = is_debug_enabled()
 
     # First line: Klaude Code version
@@ -219,6 +215,6 @@ def render_welcome(e: events.WelcomeEvent, *, box_style: Box | None = None) -> R
 
     border_style = ThemeKey.WELCOME_DEBUG_BORDER if debug_mode else ThemeKey.LINES
     return Group(
-        Panel.fit(panel_content, border_style=border_style, box=box_style),
+        Panel.fit(panel_content, border_style=border_style, box=box.ROUNDED),
         "",  # empty line
     )

klaude_code/ui/renderers/sub_agent.py

@@ -1,6 +1,7 @@
 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
@@ -58,10 +59,22 @@ def render_sub_agent_call(e: model.SubAgentState, style: Style | None = None) ->
 
 
 def render_sub_agent_result(
-    result: str, *, code_theme: str, style: Style | None = None, has_structured_output: bool = False
+    result: str,
+    *,
+    code_theme: str,
+    style: Style | None = None,
+    has_structured_output: bool = False,
+    description: str | None = None,
+    panel_style: Style | None = None,
 ) -> RenderableType:
     stripped_result = result.strip()
 
+    # Add markdown heading if description is provided
+    if description:
+        stripped_result = f"# {description}\n\n{stripped_result}"
+
+    result_panel_style = panel_style or ThemeKey.SUB_AGENT_RESULT_PANEL
+
     # Use rich JSON for structured output
     if has_structured_output:
         try:
@@ -73,7 +86,9 @@ def render_sub_agent_result(
                     ),
                     JSON(stripped_result),
                 ),
+                box=box.SIMPLE,
                 border_style=ThemeKey.LINES,
+                style=result_panel_style,
             )
         except json.JSONDecodeError:
             # Fall back to markdown if not valid JSON
@@ -82,20 +97,28 @@ def render_sub_agent_result(
     lines = stripped_result.splitlines()
     if len(lines) > const.SUB_AGENT_RESULT_MAX_LINES:
         hidden_count = len(lines) - const.SUB_AGENT_RESULT_MAX_LINES
-        truncated_text = "\n".join(lines[-const.SUB_AGENT_RESULT_MAX_LINES :])
+        head_count = const.SUB_AGENT_RESULT_MAX_LINES // 2
+        tail_count = const.SUB_AGENT_RESULT_MAX_LINES - head_count
+        head_text = "\n".join(lines[:head_count])
+        tail_text = "\n".join(lines[-tail_count:])
         return Panel.fit(
             Group(
+                NoInsetMarkdown(head_text, code_theme=code_theme, style=style or ""),
                 Text(
-                    f"… more {hidden_count} lines — use /export to view full output",
-                    style=ThemeKey.TOOL_RESULT,
+                    f"\n… more {hidden_count} lines — use /export to view full output\n",
+                    style=ThemeKey.TOOL_RESULT_TRUNCATED,
                 ),
-                NoInsetMarkdown(truncated_text, code_theme=code_theme, style=style or ""),
+                NoInsetMarkdown(tail_text, code_theme=code_theme, style=style or ""),
             ),
+            box=box.SIMPLE,
             border_style=ThemeKey.LINES,
+            style=result_panel_style,
         )
     return Panel.fit(
         NoInsetMarkdown(stripped_result, code_theme=code_theme),
+        box=box.SIMPLE,
         border_style=ThemeKey.LINES,
+        style=result_panel_style,
     )
 
 

klaude_code/ui/renderers/thinking.py

@@ -4,12 +4,13 @@ from rich.console import RenderableType
 from rich.padding import Padding
 from rich.text import Text
 
+from klaude_code import const
+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
 
-
-def thinking_prefix() -> Text:
-    return Text.from_markup("[not italic]⸫[/not italic] Thinking …", style=ThemeKey.THINKING_BOLD)
+# UI markers
+THINKING_MESSAGE_MARK = "⠶"
 
 
 def normalize_thinking_content(content: str) -> str:
@@ -37,7 +38,7 @@ def normalize_thinking_content(content: str) -> str:
 
 
 def render_thinking(content: str, *, code_theme: str, style: str) -> RenderableType | None:
-    """Render thinking content as indented markdown.
+    """Render thinking content as markdown with left mark.
 
     Returns None if content is empty.
     Note: Caller should push thinking_markdown_theme before printing.
@@ -45,11 +46,16 @@ def render_thinking(content: str, *, code_theme: str, style: str) -> RenderableT
     if len(content.strip()) == 0:
         return None
 
-    return Padding.indent(
-        ThinkingMarkdown(
-            normalize_thinking_content(content),
-            code_theme=code_theme,
-            style=style,
+    grid = create_grid()
+    grid.add_row(
+        Text(THINKING_MESSAGE_MARK, style=ThemeKey.THINKING),
+        Padding(
+            ThinkingMarkdown(
+                normalize_thinking_content(content),
+                code_theme=code_theme,
+                style=style,
+            ),
+            (0, const.MARKDOWN_RIGHT_MARGIN, 0, 0),
         ),
-        level=2,
     )
+    return grid

klaude_code/ui/renderers/tools.py

@@ -2,8 +2,10 @@ import json
 from pathlib import Path
 from typing import Any, cast
 
+from rich import box
 from rich.console import Group, RenderableType
 from rich.padding import Padding
+from rich.panel import Panel
 from rich.text import Text
 
 from klaude_code import const
@@ -11,6 +13,7 @@ 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.common import create_grid, truncate_display
+from klaude_code.ui.rich.markdown import NoInsetMarkdown
 from klaude_code.ui.rich.theme import ThemeKey
 
 # Tool markers (Unicode symbols for UI display)
@@ -528,7 +531,23 @@ def _extract_diff(ui_extra: model.ToolResultUIExtra | None) -> model.DiffUIExtra
     return None
 
 
-def render_tool_result(e: events.ToolResultEvent) -> RenderableType | None:
+def _extract_markdown_doc(ui_extra: model.ToolResultUIExtra | None) -> model.MarkdownDocUIExtra | None:
+    if isinstance(ui_extra, model.MarkdownDocUIExtra):
+        return ui_extra
+    return None
+
+
+def render_markdown_doc(md_ui: model.MarkdownDocUIExtra, *, code_theme: str) -> RenderableType:
+    """Render markdown document content in a panel."""
+    return Panel.fit(
+        NoInsetMarkdown(md_ui.content, code_theme=code_theme),
+        box=box.SIMPLE,
+        border_style=ThemeKey.LINES,
+        style=ThemeKey.WRITE_MARKDOWN_PANEL,
+    )
+
+
+def render_tool_result(e: events.ToolResultEvent, *, code_theme: str = "monokai") -> RenderableType | None:
     """Unified entry point for rendering tool results.
 
     Returns a Rich Renderable or None if the tool result should not be rendered.
@@ -549,11 +568,16 @@ def render_tool_result(e: events.ToolResultEvent) -> RenderableType | None:
         return Group(render_truncation_info(truncation_info), render_generic_tool_result(e.result))
 
     diff_ui = _extract_diff(e.ui_extra)
+    md_ui = _extract_markdown_doc(e.ui_extra)
 
     match e.tool_name:
         case tools.READ:
             return None
-        case tools.EDIT | tools.WRITE:
+        case tools.EDIT:
+            return Padding.indent(r_diffs.render_structured_diff(diff_ui) if diff_ui else Text(""), level=2)
+        case tools.WRITE:
+            if md_ui:
+                return Padding.indent(render_markdown_doc(md_ui, code_theme=code_theme), level=2)
             return Padding.indent(r_diffs.render_structured_diff(diff_ui) if diff_ui else Text(""), level=2)
         case tools.APPLY_PATCH:
             if diff_ui:

klaude_code/ui/rich/markdown.py

@@ -12,7 +12,7 @@ from rich.live import Live
 from rich.markdown import CodeBlock, Heading, Markdown, MarkdownElement
 from rich.rule import Rule
 from rich.spinner import Spinner
-from rich.style import Style
+from rich.style import Style, StyleType
 from rich.syntax import Syntax
 from rich.text import Text
 from rich.theme import Theme
@@ -64,7 +64,7 @@ class LeftHeading(Heading):
             yield h1_text
         elif self.tag == "h2":
             text.stylize(Style(bold=True, underline=False))
-            yield Rule(title=text, characters="-", style="markdown.h2.border", align="left")
+            yield Rule(title=text, characters="·", style="markdown.h2.border", align="left")
         else:
             yield text
 
@@ -108,6 +108,7 @@ class MarkdownStream:
         console: Console | None = None,
         spinner: Spinner | None = None,
         mark: str | None = None,
+        mark_style: StyleType | None = None,
         left_margin: int = 0,
         right_margin: int = const.MARKDOWN_RIGHT_MARGIN,
         markdown_class: Callable[..., Markdown] | None = None,
@@ -119,6 +120,7 @@ class MarkdownStream:
             theme (Theme, optional): Theme for rendering markdown
             console (Console, optional): External console to use for rendering
             mark (str | None, optional): Marker shown before the first non-empty line when left_margin >= 2
+            mark_style (StyleType | None, optional): Style to apply to the mark
             left_margin (int, optional): Number of columns to reserve on the left side
             right_margin (int, optional): Number of columns to reserve on the right side
             markdown_class: Markdown class to use for rendering (defaults to NoInsetMarkdown)
@@ -142,6 +144,7 @@ class MarkdownStream:
         self.console = console
         self.spinner: Spinner | None = spinner
         self.mark: str | None = mark
+        self.mark_style: StyleType | None = mark_style
 
         self.left_margin: int = max(left_margin, 0)
 
@@ -194,12 +197,24 @@ class MarkdownStream:
         mark_applied = False
         use_mark = bool(self.mark) and self.left_margin >= 2
 
+        # Pre-render styled mark if needed
+        styled_mark: str | None = None
+        if use_mark and self.mark:
+            if self.mark_style:
+                mark_text = Text(self.mark, style=self.mark_style)
+                mark_buffer = io.StringIO()
+                mark_console = Console(file=mark_buffer, force_terminal=True, theme=self.theme)
+                mark_console.print(mark_text, end="")
+                styled_mark = mark_buffer.getvalue()
+            else:
+                styled_mark = self.mark
+
         for line in lines:
             stripped = line.rstrip()
 
             # Apply mark to the first non-empty line only when left_margin is at least 2.
             if use_mark and not mark_applied and stripped:
-                stripped = f"{self.mark} {stripped}"
+                stripped = f"{styled_mark} {stripped}"
                 mark_applied = True
             elif indent_prefix:
                 stripped = indent_prefix + stripped

klaude_code/ui/rich/status.py

@@ -169,9 +169,19 @@ class ShimmerStatusText:
     Supports optional right-aligned text that stays fixed at the right edge.
     """
 
-    def __init__(self, main_text: str | Text, main_style: ThemeKey, right_text: Text | None = None) -> None:
-        self._main_text = main_text if isinstance(main_text, Text) else Text(main_text)
-        self._main_style = main_style
+    def __init__(
+        self,
+        main_text: str | Text,
+        right_text: Text | None = None,
+        main_style: ThemeKey = ThemeKey.STATUS_TEXT,
+    ) -> None:
+        if isinstance(main_text, Text):
+            text = main_text.copy()
+            if not text.style:
+                text.style = str(main_style)
+            self._main_text = text
+        else:
+            self._main_text = Text(main_text, style=main_style)
         self._hint_text = Text(const.STATUS_HINT_TEXT)
         self._hint_style = ThemeKey.STATUS_HINT
         self._right_text = right_text
@@ -193,13 +203,11 @@ class ShimmerStatusText:
     def _render_left_text(self, console: Console) -> Text:
         """Render the left part with shimmer effect on main text only."""
         result = Text()
-        main_style = console.get_style(str(self._main_style))
         hint_style = console.get_style(str(self._hint_style))
 
         # Apply shimmer only to main text
         for index, (ch, intensity) in enumerate(_shimmer_profile(self._main_text.plain)):
-            char_style = self._main_text.get_style_at_offset(console, index)
-            base_style = main_style + char_style
+            base_style = self._main_text.get_style_at_offset(console, index)
             style = _shimmer_style(console, base_style, intensity)
             result.append(ch, style=style)
 

klaude_code/ui/rich/theme.py

@@ -25,7 +25,18 @@ class Palette:
     diff_remove: str
     diff_remove_char: str
     code_theme: str
-    text_background: str
+    code_background: str
+    green_background: str
+    blue_grey_background: str
+    # Sub-agent backgrounds (corresponding to sub_agent_colors order)
+    cyan_background: str
+    green_sub_background: str
+    blue_sub_background: str
+    purple_background: str
+    orange_background: str
+    red_background: str
+    grey_background: str
+    yellow_background: str
 
 
 LIGHT_PALETTE = Palette(
@@ -47,7 +58,17 @@ LIGHT_PALETTE = Palette(
     diff_remove="#82071e on #ffecec",
     diff_remove_char="#82071e on #ffcfcf",
     code_theme="ansi_light",
-    text_background="#e0e0e0",
+    code_background="#e0e0e0",
+    green_background="#e8f1e9",
+    blue_grey_background="#e8e9f1",
+    cyan_background="#e0f0f0",
+    green_sub_background="#e0f0e0",
+    blue_sub_background="#e0e8f5",
+    purple_background="#ede0f5",
+    orange_background="#f5ebe0",
+    red_background="#f5e0e0",
+    grey_background="#e8e8e8",
+    yellow_background="#f5f5e0",
 )
 
 DARK_PALETTE = Palette(
@@ -69,12 +90,26 @@ DARK_PALETTE = Palette(
     diff_remove="#ffcdd2 on #3d1f23",
     diff_remove_char="#ffcdd2 on #7a3a42",
     code_theme="ansi_dark",
-    text_background="#2f3440",
+    code_background="#2f3440",
+    green_background="#23342c",
+    blue_grey_background="#313848",
+    cyan_background="#1a3333",
+    green_sub_background="#1b3928",
+    blue_sub_background="#1a2a3d",
+    purple_background="#2a2640",
+    orange_background="#3d2a1a",
+    red_background="#3d1f23",
+    grey_background="#2a2d30",
+    yellow_background="#3d3a1a",
 )
 
 
 class ThemeKey(str, Enum):
     LINES = "lines"
+
+    # PANEL
+    SUB_AGENT_RESULT_PANEL = "panel.sub_agent_result"
+    WRITE_MARKDOWN_PANEL = "panel.write_markdown"
     # DIFF
     DIFF_FILE_NAME = "diff.file_name"
     DIFF_REMOVE = "diff.remove"
@@ -92,9 +127,9 @@ class ThemeKey(str, Enum):
     METADATA_DIM = "metadata.dim"
     METADATA_BOLD = "metadata.bold"
     # SPINNER_STATUS
-    SPINNER_STATUS = "spinner.status"
-    SPINNER_STATUS_TEXT = "spinner.status.text"
-    SPINNER_STATUS_TEXT_BOLD = "spinner.status.text.bold"
+    STATUS_SPINNER = "spinner.status"
+    STATUS_TEXT = "spinner.status.text"
+    STATUS_TEXT_BOLD = "spinner.status.text.bold"
     # STATUS
     STATUS_HINT = "status.hint"
     # USER_INPUT
@@ -162,6 +197,7 @@ class Themes:
     thinking_markdown_theme: Theme
     code_theme: str
     sub_agent_colors: list[Style]
+    sub_agent_backgrounds: list[Style]
 
 
 def get_theme(theme: str | None = None) -> Themes:
@@ -170,6 +206,9 @@ def get_theme(theme: str | None = None) -> Themes:
         app_theme=Theme(
             styles={
                 ThemeKey.LINES.value: palette.grey3,
+                # PANEL
+                ThemeKey.SUB_AGENT_RESULT_PANEL.value: f"on {palette.blue_grey_background}",
+                ThemeKey.WRITE_MARKDOWN_PANEL.value: f"on {palette.green_background}",
                 # DIFF
                 ThemeKey.DIFF_FILE_NAME.value: palette.blue,
                 ThemeKey.DIFF_REMOVE.value: palette.diff_remove,
@@ -183,7 +222,7 @@ def get_theme(theme: str | None = None) -> Themes:
                 ThemeKey.ERROR_BOLD.value: "bold " + palette.red,
                 ThemeKey.INTERRUPT.value: "reverse bold " + palette.red,
                 # USER_INPUT
-                ThemeKey.USER_INPUT.value: palette.magenta,
+                ThemeKey.USER_INPUT.value: "bold " + palette.magenta,
                 ThemeKey.USER_INPUT_PROMPT.value: "bold " + palette.magenta,
                 ThemeKey.USER_INPUT_AT_PATTERN.value: palette.purple,
                 ThemeKey.USER_INPUT_SLASH_COMMAND.value: "bold reverse " + palette.blue,
@@ -192,11 +231,10 @@ def get_theme(theme: str | None = None) -> Themes:
                 ThemeKey.METADATA.value: palette.lavender,
                 ThemeKey.METADATA_DIM.value: "dim " + palette.lavender,
                 ThemeKey.METADATA_BOLD.value: "bold " + palette.lavender,
-                # SPINNER_STATUS
-                ThemeKey.SPINNER_STATUS.value: palette.blue,
-                ThemeKey.SPINNER_STATUS_TEXT.value: palette.blue,
-                ThemeKey.SPINNER_STATUS_TEXT_BOLD.value: "bold " + palette.blue,
                 # STATUS
+                ThemeKey.STATUS_SPINNER.value: palette.blue,
+                ThemeKey.STATUS_TEXT.value: palette.blue,
+                ThemeKey.STATUS_TEXT_BOLD.value: "bold italic " + palette.blue,
                 ThemeKey.STATUS_HINT.value: palette.grey2,
                 # REMINDER
                 ThemeKey.REMINDER.value: palette.grey1,
@@ -265,7 +303,10 @@ def get_theme(theme: str | None = None) -> Themes:
         ),
         thinking_markdown_theme=Theme(
             styles={
-                "markdown.code": palette.grey1 + " italic on " + palette.text_background,
+                # THINKING (used for left-side mark in thinking output)
+                ThemeKey.THINKING.value: "italic " + palette.grey2,
+                ThemeKey.THINKING_BOLD.value: "bold italic " + palette.grey1,
+                "markdown.code": palette.grey1 + " italic on " + palette.code_background,
                 "markdown.code.block": palette.grey1,
                 "markdown.code.border": palette.grey3,
                 "markdown.h1": "bold reverse",
@@ -292,4 +333,14 @@ def get_theme(theme: str | None = None) -> Themes:
             Style(color=palette.grey1),
             Style(color=palette.yellow),
         ],
+        sub_agent_backgrounds=[
+            Style(bgcolor=palette.cyan_background),
+            Style(bgcolor=palette.green_sub_background),
+            Style(bgcolor=palette.blue_sub_background),
+            Style(bgcolor=palette.purple_background),
+            Style(bgcolor=palette.orange_background),
+            Style(bgcolor=palette.red_background),
+            Style(bgcolor=palette.grey_background),
+            Style(bgcolor=palette.yellow_background),
+        ],
     )

{klaude_code-1.2.23.dist-info → klaude_code-1.2.24.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: klaude-code
-Version: 1.2.23
+Version: 1.2.24
 Summary: Add your description here
 Requires-Dist: anthropic>=0.66.0
 Requires-Dist: chardet>=5.2.0