fix(shell): keep image paths literal in shell mode

Author: he-yufeng

src/kimi_cli/ui/shell/placeholders.py

@@ -583,7 +583,9 @@ def maybe_placeholderize_pasted_text(self, text: str) -> str:
     def create_image_placeholder(self, image: Image.Image) -> str | None:
         return self._image_handler.create_placeholder(image)
 
-    def resolve_command(self, command: str) -> ResolvedPromptCommand:
+    def resolve_command(
+        self, command: str, *, attach_literal_images: bool = True
+    ) -> ResolvedPromptCommand:
         content: list[ContentPart] = []
         resolved_chunks: list[str] = []
         cursor = 0
@@ -592,12 +594,22 @@ def resolve_command(self, command: str) -> ResolvedPromptCommand:
         while match := self._find_next_match(command, cursor):
             if match.start > cursor:
                 literal = command[cursor : match.start]
-                self._append_literal_content(literal, content, attached_image_paths)
+                self._append_literal_content(
+                    literal,
+                    content,
+                    attached_image_paths,
+                    attach_images=attach_literal_images,
+                )
                 resolved_chunks.append(literal)
 
             resolved_content = match.handler.resolve_content(match)
             if resolved_content is None:
-                self._append_literal_content(match.raw, content, attached_image_paths)
+                self._append_literal_content(
+                    match.raw,
+                    content,
+                    attached_image_paths,
+                    attach_images=attach_literal_images,
+                )
                 resolved_chunks.append(match.raw)
             else:
                 content.extend(resolved_content)
@@ -608,7 +620,12 @@ def resolve_command(self, command: str) -> ResolvedPromptCommand:
 
         if cursor < len(command):
             literal = command[cursor:]
-            self._append_literal_content(literal, content, attached_image_paths)
+            self._append_literal_content(
+                literal,
+                content,
+                attached_image_paths,
+                attach_images=attach_literal_images,
+            )
             resolved_chunks.append(literal)
 
         return ResolvedPromptCommand(
@@ -670,10 +687,12 @@ def _append_literal_content(
         literal: str,
         content: list[ContentPart],
         attached_image_paths: set[Path],
+        *,
+        attach_images: bool,
     ) -> None:
         if not literal:
             return
-        if not self._supports_image_input():
+        if not attach_images or not self._supports_image_input():
             content.append(TextPart(text=literal))
             return
 

src/kimi_cli/ui/shell/prompt.py

@@ -2066,7 +2066,10 @@ async def _prompt_once(self, *, append_history: bool | None) -> UserInput:
     def _build_user_input(self, command: str) -> UserInput:
         manager = self._get_placeholder_manager()
         manager.update_model_capabilities(set(self._model_capabilities))
-        resolved = manager.resolve_command(command)
+        resolved = manager.resolve_command(
+            command,
+            attach_literal_images=self._mode == PromptMode.AGENT,
+        )
 
         return UserInput(
             mode=self._mode,

tests/ui_and_conv/test_prompt_tips.py

@@ -933,6 +933,22 @@ def build_user_input(command: str) -> UserInput:
     assert history == []
 
 
+def test_build_user_input_keeps_shell_image_paths_literal(tmp_path) -> None:
+    prompt_session = object.__new__(CustomPromptSession)
+    prompt_session._mode = PromptMode.SHELL
+    prompt_session._model_capabilities = {"image_in"}
+
+    command = f"cat {tmp_path / 'missing.png'}"
+
+    user_input = prompt_session._build_user_input(command)
+
+    assert user_input.mode == PromptMode.SHELL
+    assert user_input.command == command
+    assert user_input.resolved_command == command
+    assert len(user_input.content) == 1
+    assert user_input.content[0].text == command
+
+
 @pytest.mark.asyncio
 async def test_prompt_next_skips_history_for_running_submission() -> None:
     prompt_session = object.__new__(CustomPromptSession)