fix(shell): disable kitty keys in VS Code

src/kimi_cli/ui/shell/__init__.py

@@ -40,7 +40,7 @@
 from kimi_cli.utils.signals import install_sigint_handler
 from kimi_cli.utils.slashcmd import SlashCommand, SlashCommandCall, parse_slash_command_call
 from kimi_cli.utils.subprocess_env import get_clean_env
-from kimi_cli.utils.term import ensure_new_line, ensure_tty_sane
+from kimi_cli.utils.term import ensure_new_line, ensure_tty_sane, maybe_disable_kitty_keyboard_protocol
 from kimi_cli.wire.types import ContentPart, StatusUpdate
 
 
@@ -197,37 +197,7 @@ async def _plan_mode_toggle() -> bool:
                 return await self.soul.toggle_plan_mode_from_manual()
             return False
 
-        def _mcp_status_block(columns: int):
-            if not isinstance(self.soul, KimiSoul):
-                return None
-            snapshot = self.soul.status.mcp_status
-            if snapshot is None:
-                return None
-            return render_mcp_prompt(snapshot)
-
-        def _mcp_status_loading() -> bool:
-            if not isinstance(self.soul, KimiSoul):
-                return False
-            snapshot = self.soul.status.mcp_status
-            return bool(snapshot and snapshot.loading)
-
-        @dataclass
-        class _BgCountCache:
-            time: float = 0.0
-            count: int = 0
-
-        _bg_cache = _BgCountCache()
-
-        def _bg_task_count() -> int:
-            if not isinstance(self.soul, KimiSoul):
-                return 0
-            now = time.monotonic()
-            if now - _bg_cache.time < 1.0:
-                return _bg_cache.count
-            views = list_task_views(self.soul.runtime.background_tasks, active_only=True)
-            _bg_cache.count = sum(1 for v in views if v.spec.kind == "bash")
-            _bg_cache.time = now
-            return _bg_cache.count
+        maybe_disable_kitty_keyboard_protocol()
 
         with CustomPromptSession(
             status_provider=lambda: self.soul.status,

src/kimi_cli/utils/term.py

@@ -6,6 +6,8 @@
 import sys
 import time
 
+from kimi_cli.utils.envvar import get_env_bool
+
 
 def ensure_new_line() -> None:
     """Ensure the next prompt starts at column 0 regardless of prior command output."""
@@ -50,6 +52,33 @@ def ensure_tty_sane() -> None:
         termios.tcsetattr(fd, termios.TCSADRAIN, attrs)
 
 
+def maybe_disable_kitty_keyboard_protocol() -> None:
+    """Disable kitty keyboard protocol in terminals that send CSI-u sequences.
+
+    This is primarily a workaround for VS Code's integrated terminal, which can
+    emit CSI-u key sequences that prompt_toolkit doesn't parse.
+    """
+    if sys.platform == "win32":
+        return
+    if not sys.stdout.isatty() or not sys.stdin.isatty():
+        return
+    if not _should_disable_kitty_keyboard_protocol():
+        return
+
+    _write_escape("\x1b[<u")
+
+
+def _should_disable_kitty_keyboard_protocol() -> bool:
+    env_value = os.getenv("KIMI_CLI_DISABLE_KITTY_KEYS")
+    if env_value is not None:
+        return get_env_bool("KIMI_CLI_DISABLE_KITTY_KEYS", default=False)
+    return _is_vscode_terminal()
+
+
+def _is_vscode_terminal() -> bool:
+    return os.getenv("TERM_PROGRAM") == "vscode" or "VSCODE_IPC_HOOK_CLI" in os.environ
+
+
 def _cursor_position_unix() -> tuple[int, int] | None:
     """Get cursor position (row, column) on Unix. Both are 1-indexed."""
     assert sys.platform != "win32"
@@ -149,6 +178,11 @@ def _write_newline() -> None:
     sys.stdout.flush()
 
 
+def _write_escape(value: str) -> None:
+    sys.stdout.write(value)
+    sys.stdout.flush()
+
+
 def get_cursor_row() -> int | None:
     """Get the current cursor row (1-indexed)."""
     if not sys.stdout.isatty() or not sys.stdin.isatty():