format

Author: wilbeibi

src/kimi_cli/ui/shell/prompt.py

@@ -394,6 +394,8 @@ def toast(message: str, duration: float = 5.0) -> None:
     _toast_queue.put_nowait((message, duration))
 
 
+# Matches both image and text placeholders:
+# [image:id,640x480] or [text:id,60 lines]
 _ATTACHMENT_PLACEHOLDER_RE = re.compile(
     r"\[(?P<type>image|text):(?P<id>[a-zA-Z0-9_\-\.]+)"
     r"(?:,(?P<width>\d+)x(?P<height>\d+)|,(?P<line_count>\d+) lines)?\]"
@@ -474,10 +476,7 @@ def _delete_placeholder_at_cursor(buff, is_backspace: bool) -> bool:
 
                 # Backspace: delete if cursor is inside OR right after placeholder
                 # Delete: delete if cursor is inside OR right before placeholder
-                if is_backspace:
-                    should_delete = start < cursor <= end
-                else:
-                    should_delete = start <= cursor < end
+                should_delete = start < cursor <= end if is_backspace else start <= cursor < end
 
                 if should_delete:
                     attachment_id = match.group("id")
@@ -490,14 +489,14 @@ def _delete_placeholder_at_cursor(buff, is_backspace: bool) -> bool:
 
         @_kb.add("backspace", eager=True)
         def _smart_backspace(event: KeyPressEvent) -> None:
-            """Delete entire placeholder if cursor is within/after one, otherwise backspace normally."""
+            """Delete entire placeholder if cursor is within/after one, else backspace normally."""
             if not _delete_placeholder_at_cursor(event.current_buffer, is_backspace=True):
                 event.current_buffer.delete_before_cursor(1)
             self._last_buffer_text = event.current_buffer.text
 
         @_kb.add("delete", eager=True)
         def _smart_delete(event: KeyPressEvent) -> None:
-            """Delete entire placeholder if cursor is within/before one, otherwise delete normally."""
+            """Delete entire placeholder if cursor is within/before one, else delete normally."""
             if not _delete_placeholder_at_cursor(event.current_buffer, is_backspace=False):
                 event.current_buffer.delete(1)
             self._last_buffer_text = event.current_buffer.text
@@ -663,8 +662,8 @@ def _check_and_replace_large_paste(self, buffer) -> None:
         if not current_text.startswith(previous_text):
             return
 
-        inserted_text = current_text[len(previous_text):]
-        inserted_line_count = inserted_text.count('\n') + 1
+        inserted_text = current_text[len(previous_text) :]
+        inserted_line_count = inserted_text.count("\n") + 1
 
         if inserted_line_count <= LARGE_PASTE_LINE_THRESHOLD:
             return

src/kimi_cli/ui/shell/replay.py

@@ -42,7 +42,8 @@ async def replay_recent_history(history: Sequence[Message]) -> None:
     for run in runs:
         wire = Wire()
         console.print(
-            f"{getpass.getuser()}{PROMPT_SYMBOL} {message_stringify(run.user_message, context='replay')}"
+            f"{getpass.getuser()}{PROMPT_SYMBOL} "
+            f"{message_stringify(run.user_message, context='replay')}"
         )
         ui_task = asyncio.create_task(
             visualize(wire.ui_side, initial_status=StatusSnapshot(context_usage=0.0))

src/kimi_cli/utils/message.py

@@ -18,7 +18,7 @@ def _maybe_collapse(text: str) -> str:
         """Collapse text if it exceeds threshold (except in replay context)."""
         if context == "replay":
             return text
-        line_count = text.count('\n') + 1
+        line_count = text.count("\n") + 1
         if line_count > LARGE_PASTE_LINE_THRESHOLD:
             return f"[pasted {line_count} lines]"
         return text

tests/test_message_utils.py

@@ -103,7 +103,6 @@ def test_extract_text_from_empty_string():
     assert result == ""
 
 
-# New tests for large paste collapse feature
 def test_stringify_collapses_large_text_by_default():
     """Test that large text is collapsed in default context."""
     large_text = "\n".join(["line"] * 60)

tests/test_prompt_placeholders.py

@@ -2,8 +2,6 @@
 
 import re
 
-from kimi_cli.utils.message import LARGE_PASTE_LINE_THRESHOLD
-
 # The regex pattern from prompt.py
 _ATTACHMENT_PLACEHOLDER_RE = re.compile(
     r"\[(?P<type>image|text):(?P<id>[a-zA-Z0-9_\-\.]+)"
@@ -18,6 +16,7 @@ def test_backspace_boundary_conditions(self):
         """Test backspace deletes when cursor is after placeholder, not before."""
         text = "prefix [text:abc12345,60 lines]"
         match = _ATTACHMENT_PLACEHOLDER_RE.search(text)
+        assert match is not None
         start, end = match.span()
 
         # Backspace with cursor right after placeholder should delete (the bug fix)
@@ -34,6 +33,7 @@ def test_delete_boundary_conditions(self):
         """Test delete key deletes when cursor is before placeholder, not after."""
         text = "prefix [text:abc12345,60 lines]"
         match = _ATTACHMENT_PLACEHOLDER_RE.search(text)
+        assert match is not None
         start, end = match.span()
 
         # Delete with cursor before placeholder should delete