klaude-code 1.4.3__py3-none-any.whl → 1.5.0__py3-none-any.whl

klaude_code/ui/modes/repl/key_bindings.py

@@ -12,6 +12,7 @@ from collections.abc import Callable
 from typing import cast
 
 from prompt_toolkit.buffer import Buffer
+from prompt_toolkit.filters import Always, 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
@@ -21,6 +22,10 @@ def create_key_bindings(
     capture_clipboard_tag: Callable[[], str | None],
     copy_to_clipboard: Callable[[str], None],
     at_token_pattern: re.Pattern[str],
+    *,
+    input_enabled: Filter | None = None,
+    open_model_picker: Callable[[], None] | None = None,
+    open_thinking_picker: Callable[[], None] | None = None,
 ) -> KeyBindings:
     """Create REPL key bindings with injected dependencies.
 
@@ -33,6 +38,7 @@ def create_key_bindings(
         KeyBindings instance with all REPL handlers configured
     """
     kb = KeyBindings()
+    enabled = input_enabled if input_enabled is not None else Always()
 
     def _should_submit_instead_of_accepting_completion(buf: Buffer) -> bool:
         """Return True when Enter should submit even if completions are visible.
@@ -111,7 +117,7 @@ def create_key_bindings(
         buf.apply_completion(completion)
         return True
 
-    @kb.add("c-v")
+    @kb.add("c-v", filter=enabled)
     def _(event: KeyPressEvent) -> None:
         """Paste image from clipboard as [Image #N]."""
         tag = capture_clipboard_tag()
@@ -119,7 +125,7 @@ def create_key_bindings(
             with contextlib.suppress(Exception):
                 event.current_buffer.insert_text(tag)  # pyright: ignore[reportUnknownMemberType]
 
-    @kb.add("enter")
+    @kb.add("enter", filter=enabled)
     def _(event: KeyPressEvent) -> None:
         buf = event.current_buffer
         doc = buf.document  # type: ignore
@@ -150,29 +156,29 @@ def create_key_bindings(
         # No need to persist manifest anymore - iter_inputs will handle image extraction
         buf.validate_and_handle()  # type: ignore
 
-    @kb.add("tab", filter=has_completions)
+    @kb.add("tab", filter=enabled & has_completions)
     def _(event: KeyPressEvent) -> None:
         buf = event.current_buffer
         if _accept_current_completion(buf):
             event.app.invalidate()  # type: ignore[reportUnknownMemberType]
 
-    @kb.add("down", filter=has_completions)
+    @kb.add("down", filter=enabled & has_completions)
     def _(event: KeyPressEvent) -> None:
         buf = event.current_buffer
         _cycle_completion(buf, delta=1)
         event.app.invalidate()  # type: ignore[reportUnknownMemberType]
 
-    @kb.add("up", filter=has_completions)
+    @kb.add("up", filter=enabled & has_completions)
     def _(event: KeyPressEvent) -> None:
         buf = event.current_buffer
         _cycle_completion(buf, delta=-1)
         event.app.invalidate()  # type: ignore[reportUnknownMemberType]
 
-    @kb.add("c-j")
+    @kb.add("c-j", filter=enabled)
     def _(event: KeyPressEvent) -> None:
         event.current_buffer.insert_text("\n")  # type: ignore
 
-    @kb.add("c")
+    @kb.add("c", filter=enabled)
     def _(event: KeyPressEvent) -> None:
         """Copy selected text to system clipboard, or insert 'c' if no selection."""
         buf = event.current_buffer  # type: ignore
@@ -187,7 +193,7 @@ def create_key_bindings(
         else:
             buf.insert_text("c")  # type: ignore[reportUnknownMemberType]
 
-    @kb.add("backspace")
+    @kb.add("backspace", filter=enabled)
     def _(event: KeyPressEvent) -> None:
         """Ensure completions refresh on backspace when editing an @token.
 
@@ -218,7 +224,7 @@ def create_key_bindings(
         except (AttributeError, TypeError):
             pass
 
-    @kb.add("left")
+    @kb.add("left", filter=enabled)
     def _(event: KeyPressEvent) -> None:
         """Support wrapping to previous line when pressing left at column 0."""
         buf = event.current_buffer  # type: ignore
@@ -243,7 +249,7 @@ def create_key_bindings(
         except (AttributeError, IndexError, TypeError):
             pass
 
-    @kb.add("right")
+    @kb.add("right", filter=enabled)
     def _(event: KeyPressEvent) -> None:
         """Support wrapping to next line when pressing right at line end."""
         buf = event.current_buffer  # type: ignore
@@ -270,4 +276,18 @@ def create_key_bindings(
         except (AttributeError, IndexError, TypeError):
             pass
 
+    @kb.add("c-l", filter=enabled, eager=True)
+    def _(event: KeyPressEvent) -> None:
+        del event
+        if open_model_picker is not None:
+            with contextlib.suppress(Exception):
+                open_model_picker()
+
+    @kb.add("c-t", filter=enabled, eager=True)
+    def _(event: KeyPressEvent) -> None:
+        del event
+        if open_thinking_picker is not None:
+            with contextlib.suppress(Exception):
+                open_thinking_picker()
+
     return kb

klaude_code/ui/renderers/metadata.py

@@ -148,18 +148,15 @@ def _render_task_metadata_block(
 
 
 def render_task_metadata(e: events.TaskMetadataEvent) -> RenderableType:
-    """Render task metadata including main agent and sub-agents, aggregated by model+provider."""
+    """Render task metadata including main agent and sub-agents."""
     renderables: list[RenderableType] = []
 
     renderables.append(
         _render_task_metadata_block(e.metadata.main_agent, is_sub_agent=False, show_context_and_time=True)
     )
 
-    # Aggregate by (model_name, provider), sorted by total_cost descending
-    sorted_items = model.TaskMetadata.aggregate_by_model(e.metadata.sub_agent_task_metadata)
-
-    # Render each aggregated model block
-    for meta in sorted_items:
+    # 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))
 
     return Group(*renderables)

klaude_code/ui/renderers/user_input.py

@@ -1,13 +1,30 @@
 import re
+from collections.abc import Callable
 
 from rich.console import Group, RenderableType
 from rich.text import Text
 
-from klaude_code.command import is_slash_command_name
 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)
+
+
 # Match @-file patterns only when they appear at the beginning of the line
 # or immediately after whitespace, to avoid treating mid-word email-like
 # patterns such as foo@bar.com as file references.

klaude_code/ui/rich/theme.py

@@ -44,7 +44,7 @@ LIGHT_PALETTE = Palette(
     red="red",
     yellow="yellow",
     green="#00875f",
-    grey_yellow="#a89a85",
+    grey_yellow="#5f9f7a",
     cyan="cyan",
     blue="#3078C5",
     orange="#d77757",
@@ -77,7 +77,7 @@ DARK_PALETTE = Palette(
     red="#d75f5f",
     yellow="yellow",
     green="#5fd787",
-    grey_yellow="#c0b095",
+    grey_yellow="#8ac89a",
     cyan="cyan",
     blue="#00afff",
     orange="#e6704e",

klaude_code/ui/terminal/notifier.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import os
+import subprocess
 import sys
 from dataclasses import dataclass
 from enum import Enum
@@ -8,6 +9,9 @@ from typing import TextIO, cast
 
 from klaude_code.trace import DebugType, log_debug
 
+# Environment variable for tmux test signal channel
+TMUX_SIGNAL_ENV = "KLAUDE_TEST_SIGNAL"
+
 ST = "\033\\"
 BEL = "\a"
 
@@ -103,3 +107,41 @@ def _compact(text: str, limit: int = 160) -> str:
     if len(squashed) > limit:
         return squashed[: limit - 3] + "…"
     return squashed
+
+
+def emit_tmux_signal(channel: str | None = None) -> bool:
+    """Send a tmux wait-for signal when a task completes.
+
+    This enables synchronous testing by allowing external scripts to block
+    until a task finishes, eliminating the need for polling or sleep.
+
+    Usage:
+        KLAUDE_TEST_SIGNAL=done klaude  # In tmux session
+        tmux wait-for done              # Blocks until task completes
+
+    Args:
+        channel: Signal channel name. If None, reads from KLAUDE_TEST_SIGNAL env var.
+
+    Returns:
+        True if signal was sent successfully, False otherwise.
+    """
+    channel = channel or os.getenv(TMUX_SIGNAL_ENV)
+    if not channel:
+        return False
+
+    # Check if we're in a tmux session
+    if not os.getenv("TMUX"):
+        log_debug("tmux signal skipped: not in tmux session", debug_type=DebugType.TERMINAL)
+        return False
+
+    try:
+        subprocess.run(
+            ["tmux", "wait-for", "-S", channel],
+            check=True,
+            capture_output=True,
+        )
+        log_debug(f"tmux signal sent: {channel}", debug_type=DebugType.TERMINAL)
+        return True
+    except (subprocess.CalledProcessError, FileNotFoundError) as e:
+        log_debug(f"tmux signal failed: {e}", debug_type=DebugType.TERMINAL)
+        return False