klaude-code 2.0.0__tar.gz → 2.0.1__tar.gz

{klaude_code-2.0.0 → klaude_code-2.0.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: klaude-code
-Version: 2.0.0
+Version: 2.0.1
 Summary: Minimal code agent CLI
 Requires-Dist: anthropic>=0.66.0
 Requires-Dist: chardet>=5.2.0

{klaude_code-2.0.0 → klaude_code-2.0.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "uv_build"
 
 [project]
 name = "klaude-code"
-version = "2.0.0"
+version = "2.0.1"
 description = "Minimal code agent CLI"
 readme = "README.md"
 requires-python = ">=3.13"

{klaude_code-2.0.0 → klaude_code-2.0.1}/src/klaude_code/cli/cost_cmd.py

@@ -179,7 +179,7 @@ def render_cost_table(daily_stats: dict[str, DailyStats]) -> Table:
         show_header=True,
         header_style="bold",
         border_style="bright_black dim",
-        padding=(0, 1, 0, 2),
+        padding=(0, 1, 0, 1),
         box=ASCII_HORIZONAL,
     )
 

{klaude_code-2.0.0 → klaude_code-2.0.1}/src/klaude_code/config/assets/builtin_config.yaml

@@ -347,6 +347,10 @@ provider_list:
         reasoning_effort: medium
       context_limit: 400000
       max_tokens: 128000
+      cost:
+        input: 1.75
+        output: 14.0
+        cache_read: 0.17
 
 sub_agent_models:
   ImageGen: nano-banana-pro@or

{klaude_code-2.0.0 → klaude_code-2.0.1}/src/klaude_code/const.py

@@ -122,13 +122,14 @@ TOOL_OUTPUT_TRUNCATION_DIR = "/tmp/klaude"  # Directory for saving full truncate
 TAB_EXPAND_WIDTH = 8  # Tab expansion width for text rendering
 DIFF_PREFIX_WIDTH = 4  # Width of line number prefix in diff display
 MAX_DIFF_LINES = 500  # Maximum lines to show in diff output
-INVALID_TOOL_CALL_MAX_LENGTH = 500  # Maximum length for invalid tool call display
-TRUNCATE_DISPLAY_MAX_LINE_LENGTH = 1000  # Maximum line length for truncated display
-TRUNCATE_DISPLAY_MAX_LINES = 6  # Maximum lines for truncated display
+INVALID_TOOL_CALL_MAX_LENGTH = 200  # Maximum length for invalid tool call display
+TRUNCATE_DISPLAY_MAX_LINE_LENGTH = 200  # Maximum line length for truncated display
+TRUNCATE_DISPLAY_MAX_LINES = 4  # Maximum lines for truncated display
 MIN_HIDDEN_LINES_FOR_INDICATOR = 5  # Minimum hidden lines before showing truncation indicator
 SUB_AGENT_RESULT_MAX_LINES = 10  # Maximum lines for sub-agent result display
 TRUNCATE_HEAD_MAX_LINES = 2  # Maximum lines for sub-agent error display
 BASH_OUTPUT_PANEL_THRESHOLD = 10  # Bash output line threshold for CodePanel display
+BASH_MULTILINE_STRING_TRUNCATE_MAX_LINES = 2  # Max lines shown for heredoc / multiline string tokens in bash tool calls
 URL_TRUNCATE_MAX_LENGTH = 400  # Maximum length for URL truncation in display
 QUERY_DISPLAY_TRUNCATE_LENGTH = 80  # Maximum length for search query display
 NOTIFY_COMPACT_LIMIT = 160  # Maximum length for notification body text

{klaude_code-2.0.0 → klaude_code-2.0.1}/src/klaude_code/core/executor.py

@@ -176,35 +176,6 @@ class ExecutorContext:
         """Initialize an agent for a session and replay history to UI."""
         await self._ensure_agent(operation.session_id)
 
-    async def handle_user_input(self, operation: op.UserInputOperation) -> None:
-        """Handle a user input operation.
-
-        Core should not parse slash commands. The UI/CLI layer is responsible for
-        turning raw user input into one or more operations.
-        """
-
-        if operation.session_id is None:
-            raise ValueError("session_id cannot be None")
-
-        session_id = operation.session_id
-        agent = await self._ensure_agent(session_id)
-        user_input = operation.input
-
-        await self.emit_event(
-            events.UserMessageEvent(content=user_input.text, session_id=session_id, images=user_input.images)
-        )
-        agent.session.append_history(
-            [message.UserMessage(parts=message.parts_from_text_and_images(user_input.text, user_input.images))]
-        )
-
-        await self.handle_run_agent(
-            op.RunAgentOperation(
-                id=operation.id,
-                session_id=session_id,
-                input=user_input,
-            )
-        )
-
     async def handle_run_agent(self, operation: op.RunAgentOperation) -> None:
         agent = await self._ensure_agent(operation.session_id)
         existing_active = self.task_manager.get(operation.id)

{klaude_code-2.0.0 → klaude_code-2.0.1}/src/klaude_code/core/tool/todo/todo_write_tool.py

@@ -63,9 +63,8 @@ class TodoWriteTool(ToolABC):
                                     "type": "string",
                                     "enum": ["pending", "in_progress", "completed"],
                                 },
-                                "activeForm": {"type": "string", "minLength": 1},
                             },
-                            "required": ["content", "status", "activeForm"],
+                            "required": ["content", "status"],
                             "additionalProperties": False,
                         },
                         "description": "The updated todo list",

{klaude_code-2.0.0 → klaude_code-2.0.1}/src/klaude_code/protocol/model.py

@@ -2,7 +2,7 @@ from datetime import datetime
 from enum import Enum
 from typing import Annotated, Any, Literal
 
-from pydantic import BaseModel, ConfigDict, Field, computed_field
+from pydantic import BaseModel, Field, computed_field
 
 from klaude_code.const import DEFAULT_MAX_TOKENS
 from klaude_code.protocol.commands import CommandName
@@ -150,11 +150,8 @@ class TaskMetadataItem(BaseModel):
 
 
 class TodoItem(BaseModel):
-    model_config = ConfigDict(populate_by_name=True)
-
     content: str
     status: TodoStatusType
-    active_form: str = Field(default="", alias="activeForm")
 
 
 class FileStatus(BaseModel):

{klaude_code-2.0.0 → klaude_code-2.0.1}/src/klaude_code/protocol/op.py

@@ -23,7 +23,6 @@ if TYPE_CHECKING:
 class OperationType(Enum):
     """Enumeration of supported operation types."""
 
-    USER_INPUT = "user_input"
     RUN_AGENT = "run_agent"
     CHANGE_MODEL = "change_model"
     CHANGE_THINKING = "change_thinking"
@@ -46,18 +45,6 @@ class Operation(BaseModel):
         raise NotImplementedError("Subclasses must implement execute()")
 
 
-class UserInputOperation(Operation):
-    """Operation for handling user input (text and optional images) that should be processed by an agent."""
-
-    type: OperationType = OperationType.USER_INPUT
-    input: UserInputPayload
-    session_id: str | None = None
-
-    async def execute(self, handler: OperationHandler) -> None:
-        """Execute user input by running it through an agent."""
-        await handler.handle_user_input(self)
-
-
 class RunAgentOperation(Operation):
     """Operation for launching an agent task for a given session."""
 

{klaude_code-2.0.0 → klaude_code-2.0.1}/src/klaude_code/protocol/op_handler.py

@@ -18,17 +18,12 @@ if TYPE_CHECKING:
         InterruptOperation,
         ResumeSessionOperation,
         RunAgentOperation,
-        UserInputOperation,
     )
 
 
 class OperationHandler(Protocol):
     """Protocol defining the interface for handling operations."""
 
-    async def handle_user_input(self, operation: UserInputOperation) -> None:
-        """Handle a user input operation."""
-        ...
-
     async def handle_run_agent(self, operation: RunAgentOperation) -> None:
         """Handle a run agent operation."""
         ...

{klaude_code-2.0.0 → klaude_code-2.0.1}/src/klaude_code/ui/modes/repl/event_handler.py

@@ -624,11 +624,8 @@ class DisplayEventHandler:
     def _extract_active_form_text(self, todo_event: events.TodoChangeEvent) -> str | None:
         status_text: str | None = None
         for todo in todo_event.todos:
-            if todo.status == "in_progress":
-                if len(todo.active_form) > 0:
-                    status_text = todo.active_form
-                if len(todo.content) > 0:
-                    status_text = todo.content
+            if todo.status == "in_progress" and len(todo.content) > 0:
+                status_text = todo.content
 
         if status_text is None:
             return None

{klaude_code-2.0.0 → klaude_code-2.0.1}/src/klaude_code/ui/modes/repl/key_bindings.py

@@ -11,8 +11,9 @@ import re
 from collections.abc import Callable
 from typing import cast
 
+from prompt_toolkit.application.current import get_app
 from prompt_toolkit.buffer import Buffer
-from prompt_toolkit.filters import Always, Filter
+from prompt_toolkit.filters import Always, Condition, Filter
 from prompt_toolkit.filters.app import has_completions
 from prompt_toolkit.key_binding import KeyBindings
 from prompt_toolkit.key_binding.key_processor import KeyPressEvent
@@ -40,6 +41,119 @@ def create_key_bindings(
     kb = KeyBindings()
     enabled = input_enabled if input_enabled is not None else Always()
 
+    def _can_move_cursor_visually_within_wrapped_line(delta_visible_y: int) -> bool:
+        """Return True when Up/Down should move within a wrapped visual line.
+
+        prompt_toolkit's default Up/Down behavior operates on logical lines
+        (split by '\n'). When a single logical line wraps across terminal
+        rows, pressing Up/Down should move within those wrapped rows instead of
+        triggering history navigation.
+
+        We only intercept when the cursor can move to an adjacent *visible*
+        line that maps to the same input line.
+        """
+
+        try:
+            app = get_app()
+            window = app.layout.current_window
+            ri = window.render_info
+            if ri is None:
+                return False
+
+            current_visible_y = int(ri.cursor_position.y)
+            target_visible_y = current_visible_y + delta_visible_y
+            if target_visible_y < 0:
+                return False
+
+            current_input_line = ri.visible_line_to_input_line.get(current_visible_y)
+            target_input_line = ri.visible_line_to_input_line.get(target_visible_y)
+            return current_input_line is not None and current_input_line == target_input_line
+        except Exception:
+            return False
+
+    def _move_cursor_visually_within_wrapped_line(event: KeyPressEvent, *, delta_visible_y: int) -> None:
+        """Move the cursor Up/Down by one wrapped screen row, keeping column."""
+
+        buf = event.current_buffer
+        try:
+            window = event.app.layout.current_window
+            ri = window.render_info
+            if ri is None:
+                return
+
+            rowcol_to_yx = getattr(ri, "_rowcol_to_yx", None)
+            x_offset = getattr(ri, "_x_offset", None)
+            y_offset = getattr(ri, "_y_offset", None)
+            if not isinstance(rowcol_to_yx, dict) or not isinstance(x_offset, int) or not isinstance(y_offset, int):
+                return
+            rowcol_to_yx_typed = cast(dict[tuple[int, int], tuple[int, int]], rowcol_to_yx)
+
+            current_visible_y = int(ri.cursor_position.y)
+            target_visible_y = current_visible_y + delta_visible_y
+            mapping = ri.visible_line_to_row_col
+            if current_visible_y not in mapping or target_visible_y not in mapping:
+                return
+
+            current_row, _ = mapping[current_visible_y]
+            target_row, _ = mapping[target_visible_y]
+
+            # Only handle wrapped rows within the same input line.
+            if current_row != target_row:
+                return
+
+            current_abs_y = y_offset + current_visible_y
+            target_abs_y = y_offset + target_visible_y
+            cursor_abs_x = x_offset + int(ri.cursor_position.x)
+
+            def _segment_start_abs_x(row: int, abs_y: int) -> int | None:
+                xs: list[int] = []
+                for (r, _col), (y, x) in rowcol_to_yx_typed.items():
+                    if r == row and y == abs_y:
+                        xs.append(x)
+                return min(xs) if xs else None
+
+            current_start_x = _segment_start_abs_x(current_row, current_abs_y)
+            target_start_x = _segment_start_abs_x(target_row, target_abs_y)
+            if current_start_x is None or target_start_x is None:
+                return
+
+            offset_in_segment_cells = max(0, cursor_abs_x - current_start_x)
+            desired_abs_x = target_start_x + offset_in_segment_cells
+
+            candidates: list[tuple[int, int]] = []
+            for (r, col), (y, x) in rowcol_to_yx_typed.items():
+                if r == target_row and y == target_abs_y:
+                    candidates.append((col, x))
+            if not candidates:
+                return
+
+            # Pick the closest column at/before the desired X. If the desired
+            # position is before the first character, snap to the first.
+            candidates.sort(key=lambda t: t[1])
+            chosen_display_col = candidates[0][0]
+            for col, x in candidates:
+                if x <= desired_abs_x:
+                    chosen_display_col = col
+                else:
+                    break
+
+            control = event.app.layout.current_control
+            get_processed_line = getattr(control, "_last_get_processed_line", None)
+            target_source_col = chosen_display_col
+            if callable(get_processed_line):
+                processed_line = get_processed_line(target_row)
+                display_to_source = getattr(processed_line, "display_to_source", None)
+                if callable(display_to_source):
+                    display_to_source_fn = cast(Callable[[int], int], display_to_source)
+                    target_source_col = display_to_source_fn(chosen_display_col)
+
+            doc = buf.document  # type: ignore[reportUnknownMemberType]
+            new_index = doc.translate_row_col_to_index(target_row, target_source_col)  # type: ignore[reportUnknownMemberType]
+            buf.cursor_position = new_index  # type: ignore[reportUnknownMemberType]
+            event.app.invalidate()  # type: ignore[reportUnknownMemberType]
+        except Exception:
+            return
+
     def _should_submit_instead_of_accepting_completion(buf: Buffer) -> bool:
         """Return True when Enter should submit even if completions are visible.
 
@@ -174,6 +288,26 @@ def create_key_bindings(
         _cycle_completion(buf, delta=-1)
         event.app.invalidate()  # type: ignore[reportUnknownMemberType]
 
+    @kb.add(
+        "up",
+        filter=enabled
+        & ~has_completions
+        & Condition(lambda: _can_move_cursor_visually_within_wrapped_line(delta_visible_y=-1)),
+        eager=True,
+    )
+    def _(event: KeyPressEvent) -> None:
+        _move_cursor_visually_within_wrapped_line(event, delta_visible_y=-1)
+
+    @kb.add(
+        "down",
+        filter=enabled
+        & ~has_completions
+        & Condition(lambda: _can_move_cursor_visually_within_wrapped_line(delta_visible_y=1)),
+        eager=True,
+    )
+    def _(event: KeyPressEvent) -> None:
+        _move_cursor_visually_within_wrapped_line(event, delta_visible_y=1)
+
     @kb.add("c-j", filter=enabled)
     def _(event: KeyPressEvent) -> None:
         event.current_buffer.insert_text("\n")  # type: ignore

{klaude_code-2.0.0 → klaude_code-2.0.1}/src/klaude_code/ui/renderers/bash_syntax.py

@@ -7,6 +7,8 @@ from pygments.lexers import BashLexer  # pyright: ignore[reportUnknownVariableTy
 from pygments.token import Token
 from rich.text import Text
 
+from klaude_code.const import BASH_MULTILINE_STRING_TRUNCATE_MAX_LINES
+from klaude_code.ui.renderers.common import truncate_head
 from klaude_code.ui.rich.theme import ThemeKey
 
 # Token types for bash syntax highlighting
@@ -110,13 +112,34 @@ def _append_heredoc(result: Text, token_value: str) -> None:
         # Extra content on first line (e.g., "> file.py")
         if extra:
             result.append(extra, style=ThemeKey.BASH_ARGUMENT)
-        # Body content
-        result.append(body, style=ThemeKey.BASH_STRING)
+
+        # Body content (truncate to keep tool call rendering compact)
+        body_inner = body.strip("\n")
+        result.append("\n")
+        if body_inner:
+            body_text = truncate_head(
+                body_inner,
+                max_lines=BASH_MULTILINE_STRING_TRUNCATE_MAX_LINES,
+                base_style=ThemeKey.BASH_STRING,
+                truncated_style=ThemeKey.TOOL_RESULT_TRUNCATED,
+            )
+            result.append_text(body_text)
+            result.append("\n")
+
         # End delimiter
         result.append(end_delimiter, style=ThemeKey.BASH_HEREDOC_DELIMITER)
     else:
         # Fallback: couldn't parse heredoc structure
-        result.append(token_value, style=ThemeKey.BASH_STRING)
+        if "\n" in token_value and len(token_value.splitlines()) > BASH_MULTILINE_STRING_TRUNCATE_MAX_LINES:
+            truncated = truncate_head(
+                token_value,
+                max_lines=BASH_MULTILINE_STRING_TRUNCATE_MAX_LINES,
+                base_style=ThemeKey.BASH_STRING,
+                truncated_style=ThemeKey.TOOL_RESULT_TRUNCATED,
+            )
+            result.append_text(truncated)
+        else:
+            result.append(token_value, style=ThemeKey.BASH_STRING)
 
 
 def highlight_bash_command(command: str) -> Text:
@@ -145,7 +168,16 @@ def highlight_bash_command(command: str) -> Text:
             if token_value.startswith("<<"):
                 _append_heredoc(result, token_value)
             else:
-                result.append(token_value, style=ThemeKey.BASH_STRING)
+                if "\n" in token_value and len(token_value.splitlines()) > BASH_MULTILINE_STRING_TRUNCATE_MAX_LINES:
+                    truncated = truncate_head(
+                        token_value,
+                        max_lines=BASH_MULTILINE_STRING_TRUNCATE_MAX_LINES,
+                        base_style=ThemeKey.BASH_STRING,
+                        truncated_style=ThemeKey.TOOL_RESULT_TRUNCATED,
+                    )
+                    result.append_text(truncated)
+                else:
+                    result.append(token_value, style=ThemeKey.BASH_STRING)
             expect_subcommand = False
         elif token_type in _OPERATOR_TOKENS:
             result.append(token_value, style=ThemeKey.BASH_OPERATOR)

{klaude_code-2.0.0 → klaude_code-2.0.1}/src/klaude_code/ui/renderers/common.py

@@ -1,3 +1,5 @@
+from typing import Literal
+
 from rich.style import Style
 from rich.table import Table
 from rich.text import Text
@@ -12,10 +14,10 @@ from klaude_code.const import (
 from klaude_code.ui.rich.theme import ThemeKey
 
 
-def create_grid() -> Table:
+def create_grid(*, overflow: Literal["fold", "crop", "ellipsis", "ignore"] = "fold") -> Table:
     grid = Table.grid(padding=(0, 1))
     grid.add_column(no_wrap=True)
-    grid.add_column(overflow="fold")
+    grid.add_column(overflow=overflow)
     return grid
 
 
@@ -36,7 +38,7 @@ def truncate_middle(
     if max_lines <= 0:
         truncated_lines = text.split("\n")
         remaining = max(0, len(truncated_lines))
-        return Text(f"… (more {remaining} lines)", style=ThemeKey.TOOL_RESULT_TRUNCATED)
+        return Text(f"  … (more {remaining} lines)", style=ThemeKey.TOOL_RESULT_TRUNCATED)
 
     lines = text.split("\n")
     truncated_lines = 0
@@ -65,7 +67,7 @@ def truncate_middle(
             out.append(line[:max_line_length])
             out.append_text(
                 Text(
-                    f"… (more {extra_chars} characters in this line)",
+                    f" … (more {extra_chars} characters in this line)",
                     style=ThemeKey.TOOL_RESULT_TRUNCATED,
                 )
             )
@@ -82,7 +84,7 @@ def truncate_middle(
             out.append("\n")
 
     if truncated_lines > 0:
-        out.append_text(Text(f"⋮ (more {truncated_lines} lines)\n", style=ThemeKey.TOOL_RESULT_TRUNCATED))
+        out.append_text(Text(f"  … (more {truncated_lines} lines)\n", style=ThemeKey.TOOL_RESULT_TRUNCATED))
 
     for idx, line in enumerate(tail_lines):
         append_line(out, line)
@@ -139,6 +141,6 @@ def truncate_head(
         out.append("\n")
 
     remaining = len(lines) - max_lines
-    out.append_text(Text(f"… more {remaining} lines", style=truncated_style or ThemeKey.TOOL_RESULT_TRUNCATED))
+    out.append_text(Text(f"  … (more {remaining} lines)", style=truncated_style or ThemeKey.TOOL_RESULT_TRUNCATED))
 
     return out

{klaude_code-2.0.0 → klaude_code-2.0.1}/src/klaude_code/ui/renderers/tools.py

@@ -68,9 +68,7 @@ def _render_tool_call_tree(
     tool_name: str,
     details: RenderableType | None,
 ) -> RenderableType:
-    # Keep the original 2-column layout (tool name on the left, details on the right),
-    # but move the tool mark into the tree prefix so it can connect to the tool result.
-    grid = create_grid()
+    grid = create_grid(overflow="ellipsis")
     grid.add_row(
         Text(tool_name, style=ThemeKey.TOOL_NAME),
         details if details is not None else Text(""),
@@ -138,15 +136,13 @@ def render_bash_tool_call(arguments: str) -> RenderableType:
     command = payload.get("command")
     timeout_ms = payload.get("timeout_ms")
 
-    # Build the command display with optional timeout suffix
     if isinstance(command, str) and command.strip():
         cmd_str = command.strip()
-        line_count = len(cmd_str.splitlines())
-
         highlighted = highlight_bash_command(cmd_str)
 
-        # For commands > threshold lines, use CodePanel for better display
-        if line_count > BASH_OUTPUT_PANEL_THRESHOLD:
+        display_line_count = len(highlighted.plain.splitlines())
+
+        if display_line_count > BASH_OUTPUT_PANEL_THRESHOLD:
             code_panel = CodePanel(highlighted, border_style=ThemeKey.LINES)
             if isinstance(timeout_ms, int):
                 if timeout_ms >= 1000 and timeout_ms % 1000 == 0:
@@ -298,7 +294,6 @@ def render_move_tool_call(arguments: str) -> RenderableType:
     start_line = payload.get("start_line", "")
     end_line = payload.get("end_line", "")
 
-    # Build display: source:start-end -> target
     parts = Text()
     if source_path:
         parts.append_text(render_path(source_path, ThemeKey.TOOL_PARAM_FILE_PATH))
@@ -386,15 +381,17 @@ def render_todo(tr: events.ToolResultEvent) -> RenderableType:
 def render_generic_tool_result(result: str, *, is_error: bool = False) -> RenderableType:
     """Render a generic tool result as truncated text."""
     style = ThemeKey.ERROR if is_error else ThemeKey.TOOL_RESULT
-    return truncate_middle(result, base_style=style)
+    text = truncate_middle(result, base_style=style)
+    # Tool results should not reflow/wrap; use ellipsis when exceeding terminal width.
+    text.no_wrap = True
+    text.overflow = "ellipsis"
+    return text
 
 
 def _extract_mermaid_link(
     ui_extra: model.ToolResultUIExtra | None,
 ) -> model.MermaidLinkUIExtra | None:
-    if isinstance(ui_extra, model.MermaidLinkUIExtra):
-        return ui_extra
-    return None
+    return ui_extra if isinstance(ui_extra, model.MermaidLinkUIExtra) else None
 
 
 def render_mermaid_tool_call(arguments: str) -> RenderableType:
@@ -443,7 +440,7 @@ def _render_mermaid_viewer_link(
 ) -> RenderableType:
     viewer_path = r_mermaid_viewer.build_viewer(code=link_info.code, link=link_info.link, tool_call_id=tr.tool_call_id)
     if viewer_path is None:
-        return Text(link_info.link, style=ThemeKey.TOOL_RESULT_MERMAID, overflow="fold")
+        return Text(link_info.link, style=ThemeKey.TOOL_RESULT_MERMAID, overflow="ellipsis", no_wrap=True)
 
     display_path = str(viewer_path)
 
@@ -534,9 +531,7 @@ def render_mermaid_tool_result(
 def _extract_truncation(
     ui_extra: model.ToolResultUIExtra | None,
 ) -> model.TruncationUIExtra | None:
-    if isinstance(ui_extra, model.TruncationUIExtra):
-        return ui_extra
-    return None
+    return ui_extra if isinstance(ui_extra, model.TruncationUIExtra) else None
 
 
 def render_truncation_info(ui_extra: model.TruncationUIExtra) -> RenderableType:
@@ -548,6 +543,8 @@ def render_truncation_info(ui_extra: model.TruncationUIExtra) -> RenderableType:
         (ui_extra.saved_file_path, ThemeKey.TOOL_RESULT_TRUNCATED),
         (f", {truncated_kb:.1f}KB truncated", ThemeKey.TOOL_RESULT_TRUNCATED),
     )
+    text.no_wrap = True
+    text.overflow = "ellipsis"
     return text
 
 
@@ -685,7 +682,7 @@ def render_tool_result(
 
     # Handle error case
     if e.status == "error" and e.ui_extra is None:
-        return wrap(truncate_middle(e.result, base_style=ThemeKey.ERROR))
+        return wrap(render_generic_tool_result(e.result, is_error=True))
 
     # Render multiple ui blocks if present
     if isinstance(e.ui_extra, model.MultiUIExtra) and e.ui_extra.items:
@@ -737,8 +734,6 @@ def render_tool_result(
         case tools.MERMAID:
             return wrap(render_mermaid_tool_result(e, session_id=session_id))
         case tools.BASH:
-            if e.result.startswith("diff --git"):
-                return wrap(r_diffs.render_diff_panel(e.result, show_file_name=True))
             return _render_fallback()
         case _:
             return _render_fallback()

{klaude_code-2.0.0 → klaude_code-2.0.1}/src/klaude_code/ui/rich/theme.py

@@ -60,17 +60,17 @@ LIGHT_PALETTE = Palette(
     diff_remove="#82071e on #ffecec",
     diff_remove_char="#82071e on #ffcfcf",
     code_theme="ansi_light",
-    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",
+    code_background="#ebebeb",
+    green_background="#f0f7f1",
+    blue_grey_background="#f0f1f7",
+    cyan_background="#ecf7f7",
+    green_sub_background="#ecf7ec",
+    blue_sub_background="#ecf1f9",
+    purple_background="#f5ecf9",
+    orange_background="#f9f3ec",
+    red_background="#f9ecec",
+    grey_background="#f0f0f0",
+    yellow_background="#f9f9ec",
 )
 
 DARK_PALETTE = Palette(
@@ -270,11 +270,11 @@ def get_theme(theme: str | None = None) -> Themes:
                 ThemeKey.TOOL_RESULT.value: palette.grey_green,
                 ThemeKey.TOOL_RESULT_TREE_PREFIX.value: palette.grey_green + " dim",
                 ThemeKey.TOOL_RESULT_BOLD.value: "bold " + palette.grey_green,
-                ThemeKey.TOOL_RESULT_TRUNCATED.value: palette.yellow + " dim",
+                ThemeKey.TOOL_RESULT_TRUNCATED.value: palette.grey1,
                 ThemeKey.TOOL_MARK.value: "bold",
                 ThemeKey.TOOL_APPROVED.value: palette.green + " bold reverse",
                 ThemeKey.TOOL_REJECTED.value: palette.red + " bold reverse",
-                ThemeKey.TOOL_TIMEOUT.value: palette.yellow,
+                ThemeKey.TOOL_TIMEOUT.value: palette.grey2,
                 ThemeKey.TOOL_RESULT_MERMAID: palette.blue + " underline",
                 ThemeKey.SUB_AGENT_FOOTER.value: "dim " + palette.grey2,
                 # BASH SYNTAX