klaude-code 2.8.1__py3-none-any.whl → 2.9.0__py3-none-any.whl

klaude_code/tui/machine.py

@@ -49,7 +49,7 @@ from klaude_code.tui.commands import (
 from klaude_code.tui.components.rich import status as r_status
 from klaude_code.tui.components.rich.theme import ThemeKey
 from klaude_code.tui.components.thinking import extract_last_bold_header, normalize_thinking_content
-from klaude_code.tui.components.tools import get_tool_active_form, is_sub_agent_tool
+from klaude_code.tui.components.tools import get_task_active_form, get_tool_active_form, is_sub_agent_tool
 
 # Tools that complete quickly and don't benefit from streaming activity display.
 # For models without fine-grained tool JSON streaming (e.g., Gemini), showing these
@@ -97,10 +97,6 @@ class ActivityState:
         self._sub_agent_tool_calls: dict[str, int] = {}
         self._sub_agent_tool_calls_by_id: dict[str, str] = {}
 
-    @property
-    def is_composing(self) -> bool:
-        return self._composing and not self._tool_calls and not self._sub_agent_tool_calls
-
     def set_composing(self, composing: bool) -> None:
         self._composing = composing
         if not composing:
@@ -321,7 +317,7 @@ class _SessionState:
 
     @property
     def should_show_sub_agent_thinking_header(self) -> bool:
-        return bool(self.sub_agent_state and self.sub_agent_state.sub_agent_type == "ImageGen")
+        return bool(self.sub_agent_state and self.sub_agent_state.sub_agent_type == tools.IMAGE_GEN)
 
     @property
     def should_extract_reasoning_header(self) -> bool:
@@ -607,11 +603,14 @@ class DisplayStateMachine:
                 # Skip activity state for fast tools on non-streaming models (e.g., Gemini)
                 # to avoid flash-and-disappear effect
                 if not is_replay and not s.should_skip_tool_activity(e.tool_name):
-                    tool_active_form = get_tool_active_form(e.tool_name)
-                    if is_sub_agent_tool(e.tool_name):
-                        self._spinner.add_sub_agent_tool_call(e.tool_call_id, tool_active_form)
+                    if e.tool_name == tools.TASK:
+                        pass
                     else:
-                        self._spinner.add_tool_call(tool_active_form)
+                        tool_active_form = get_tool_active_form(e.tool_name)
+                        if is_sub_agent_tool(e.tool_name):
+                            self._spinner.add_sub_agent_tool_call(e.tool_call_id, tool_active_form)
+                        else:
+                            self._spinner.add_tool_call(tool_active_form)
 
                 if not is_replay:
                     cmds.extend(self._spinner_update_commands())
@@ -629,12 +628,17 @@ class DisplayStateMachine:
                         primary.thinking_stream_active = False
                         cmds.append(EndThinkingStream(session_id=primary.session_id))
 
+                if not is_replay and e.tool_name == tools.TASK and not s.should_skip_tool_activity(e.tool_name):
+                    tool_active_form = get_task_active_form(e.arguments)
+                    self._spinner.add_sub_agent_tool_call(e.tool_call_id, tool_active_form)
+                    cmds.extend(self._spinner_update_commands())
+
                 cmds.append(RenderToolCall(e))
                 return cmds
 
             case events.ToolResultEvent() as e:
                 if not is_replay and is_sub_agent_tool(e.tool_name):
-                    self._spinner.finish_sub_agent_tool_call(e.tool_call_id, get_tool_active_form(e.tool_name))
+                    self._spinner.finish_sub_agent_tool_call(e.tool_call_id)
                     cmds.extend(self._spinner_update_commands())
 
                 if s.is_sub_agent and not e.is_error:

klaude_code/tui/renderer.py

@@ -184,13 +184,6 @@ class TUICommandRenderer:
     def is_sub_agent_session(self, session_id: str) -> bool:
         return session_id in self._sessions and self._sessions[session_id].sub_agent_state is not None
 
-    def _should_display_sub_agent_thinking_header(self, session_id: str) -> bool:
-        # Hardcoded: only show sub-agent thinking headers for ImageGen.
-        st = self._sessions.get(session_id)
-        if st is None or st.sub_agent_state is None:
-            return False
-        return st.sub_agent_state.sub_agent_type == "ImageGen"
-
     def _advance_sub_agent_color_index(self) -> None:
         palette_size = len(self.themes.sub_agent_colors)
         if palette_size == 0:
@@ -417,21 +410,12 @@ class TUICommandRenderer:
             if image_path is not None:
                 self.display_image(str(image_path))
 
-        renderable = c_tools.render_tool_result(e, code_theme=self.themes.code_theme, session_id=e.session_id)
-        if renderable is not None:
-            self.print(renderable)
+        if not is_sub_agent and isinstance(e.ui_extra, model.ImageUIExtra):
+            self.display_image(e.ui_extra.file_path)
 
-    def display_thinking(self, content: str) -> None:
-        renderable = c_thinking.render_thinking(
-            content,
-            code_theme=self.themes.code_theme,
-            style=ThemeKey.THINKING,
-        )
+        renderable = c_tools.render_tool_result(e, code_theme=self.themes.code_theme, session_id=e.session_id)
         if renderable is not None:
-            self.console.push_theme(theme=self.themes.thinking_markdown_theme)
             self.print(renderable)
-            self.console.pop_theme()
-            self.print()
 
     def display_thinking_header(self, header: str) -> None:
         stripped = header.strip()
@@ -451,6 +435,13 @@ class TUICommandRenderer:
         with self.session_print_context(e.session_id):
             self.print(c_developer.render_developer_message(e))
 
+        # Display images from @ file references and user attachments
+        if e.item.ui_extra:
+            for ui_item in e.item.ui_extra.items:
+                if isinstance(ui_item, (model.AtFileImagesUIItem, model.UserImagesUIItem)):
+                    for image_path in ui_item.paths:
+                        self.display_image(image_path)
+
     def display_command_output(self, e: events.CommandOutputEvent) -> None:
         with self.session_print_context(e.session_id):
             self.print(c_command_output.render_command_output(e))
@@ -690,7 +681,7 @@ class TUICommandRenderer:
                 case PrintBlankLine():
                     self.print()
                 case PrintRuleLine():
-                    self.console.print(Rule(characters="─", style=ThemeKey.LINES))
+                    self.console.print(Rule(characters="─", style=ThemeKey.LINES_DIM))
                 case EmitOsc94Error():
                     emit_osc94(OSC94States.ERROR)
                 case EmitTmuxSignal():

klaude_code/tui/runner.py

@@ -327,5 +327,6 @@ async def run_interactive(init_config: AppInitConfig, session_id: str | None = N
         if not exit_hint_printed:
             active_session_id = components.executor.context.current_session_id()
             if active_session_id and Session.exists(active_session_id):
+                short_id = Session.shortest_unique_prefix(active_session_id)
                 log(f"Session ID: {active_session_id}")
-                log(f"Resume with: klaude --resume {active_session_id}")
+                log(f"Resume with: klaude -r {short_id}")

klaude_code/tui/terminal/image.py

@@ -2,7 +2,6 @@ from __future__ import annotations
 
 import base64
 import shutil
-import struct
 import subprocess
 import sys
 import tempfile
@@ -12,8 +11,8 @@ from typing import IO
 # Kitty graphics protocol chunk size (4096 is the recommended max)
 _CHUNK_SIZE = 4096
 
-# Max columns for non-wide images
-_MAX_COLS = 120
+# Max columns for image display
+_MAX_COLS = 80
 
 # Image formats that need conversion to PNG
 _NEEDS_CONVERSION = {".jpg", ".jpeg", ".gif", ".bmp", ".webp", ".tiff", ".tif"}
@@ -40,26 +39,10 @@ def _convert_to_png(path: Path) -> bytes | None:
     return None
 
 
-def _get_png_dimensions(data: bytes) -> tuple[int, int] | None:
-    """Extract width and height from PNG file header."""
-    # PNG: 8-byte signature + IHDR chunk (4 len + 4 type + 4 width + 4 height)
-    if len(data) < 24 or data[:8] != b"\x89PNG\r\n\x1a\n":
-        return None
-    width, height = struct.unpack(">II", data[16:24])
-    return width, height
-
-
 def print_kitty_image(file_path: str | Path, *, file: IO[str] | None = None) -> None:
     """Print an image to the terminal using Kitty graphics protocol.
 
-    This intentionally bypasses Rich rendering to avoid interleaving Live refreshes
-    with raw escape sequences. Image size adapts based on aspect ratio:
-    - Landscape images: fill terminal width
-    - Portrait images: limit height to avoid oversized display
-
-    Args:
-        file_path: Path to the image file (PNG recommended).
-        file: Output file stream. Defaults to stdout.
+    Only specifies column width; Kitty auto-scales height to preserve aspect ratio.
     """
     path = Path(file_path) if isinstance(file_path, str) else file_path
     if not path.exists():
@@ -80,20 +63,9 @@ def print_kitty_image(file_path: str | Path, *, file: IO[str] | None = None) ->
         out = file or sys.stdout
 
         term_size = shutil.get_terminal_size()
-        dimensions = _get_png_dimensions(data)
-
-        # Determine sizing strategy based on aspect ratio
-        if dimensions is not None:
-            img_width, img_height = dimensions
-            if img_width > 2 * img_height:
-                # Wide landscape (width > 2x height): fill terminal width
-                size_param = f"c={term_size.columns}"
-            else:
-                # Other images: limit width to 80% of terminal
-                size_param = f"c={min(_MAX_COLS, term_size.columns * 4 // 5)}"
-        else:
-            # Fallback: limit width to 80% of terminal
-            size_param = f"c={min(_MAX_COLS, term_size.columns * 4 // 5)}"
+        # Only specify columns, let Kitty auto-scale height to preserve aspect ratio
+        target_cols = min(_MAX_COLS, term_size.columns)
+        size_param = f"c={target_cols}"
         print("", file=out)
         _write_kitty_graphics(out, encoded, size_param=size_param)
         print("", file=out)

klaude_code/ui/common.py

@@ -1,17 +1,8 @@
-import re
-import subprocess
-from pathlib import Path
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
     from klaude_code.protocol.llm_param import LLMConfigModelParameter, OpenRouterProviderRouting
 
-LEADING_NEWLINES_REGEX = re.compile(r"^\n{2,}")
-
-
-def remove_leading_newlines(text: str) -> str:
-    return text.lstrip("\n")
-
 
 def format_number(tokens: int) -> str:
     if tokens < 1000:
@@ -33,67 +24,6 @@ def format_number(tokens: int) -> str:
             return f"{m}M{remaining}k"
 
 
-def get_current_git_branch(path: Path | None = None) -> str | None:
-    """Get current git branch name, return None if not in a git repository"""
-    if path is None:
-        path = Path.cwd()
-
-    try:
-        # Check if in git repository
-        git_dir = subprocess.run(
-            ["git", "rev-parse", "--git-dir"],
-            cwd=path,
-            capture_output=True,
-            text=True,
-            timeout=2,
-        )
-
-        if git_dir.returncode != 0:
-            return None
-
-        # Get current branch name
-        result = subprocess.run(
-            ["git", "branch", "--show-current"],
-            cwd=path,
-            capture_output=True,
-            text=True,
-            timeout=2,
-        )
-
-        if result.returncode == 0:
-            branch = result.stdout.strip()
-            return branch if branch else None
-
-        # Fallback: get HEAD reference
-        head_file = subprocess.run(
-            ["git", "rev-parse", "--abbrev-ref", "HEAD"],
-            cwd=path,
-            capture_output=True,
-            text=True,
-            timeout=2,
-        )
-
-        if head_file.returncode == 0:
-            branch = head_file.stdout.strip()
-            return branch if branch and branch != "HEAD" else None
-
-    except (subprocess.TimeoutExpired, subprocess.SubprocessError, FileNotFoundError):
-        pass
-
-    return None
-
-
-def show_path_with_tilde(path: Path | None = None):
-    if path is None:
-        path = Path.cwd()
-
-    try:
-        relative_path = path.relative_to(Path.home())
-        return f"~/{relative_path}"
-    except ValueError:
-        return str(path)
-
-
 def format_model_params(model_params: "LLMConfigModelParameter") -> list[str]:
     """Format model parameters in a concise style.
 

{klaude_code-2.8.1.dist-info → klaude_code-2.9.0.dist-info}/METADATA

@@ -1,11 +1,12 @@
 Metadata-Version: 2.3
 Name: klaude-code
-Version: 2.8.1
+Version: 2.9.0
 Summary: Minimal code agent CLI
 Requires-Dist: anthropic>=0.66.0
 Requires-Dist: chardet>=5.2.0
 Requires-Dist: ddgs>=9.9.3
 Requires-Dist: diff-match-patch>=20241021
+Requires-Dist: filelock>=3.20.3
 Requires-Dist: google-genai>=1.56.0
 Requires-Dist: markdown-it-py>=4.0.0
 Requires-Dist: openai>=1.102.0
@@ -23,7 +24,7 @@ Description-Content-Type: text/markdown
 Minimal code agent CLI.
 
 ## Features
-- **Multi-provider**: Anthropic Message API, OpenAI Responses API, OpenRouter, Claude Max OAuth and ChatGPT Codex OAuth etc.
+- **Multi-provider**: Anthropic Message API, OpenAI Responses API, OpenRouter, ChatGPT Codex OAuth etc.
 - **Keep reasoning item in context**: Interleaved thinking support
 - **Model-aware tools**: Claude Code tool set for Opus, `apply_patch` for GPT-5/Codex
 - **Reminders**: Cooldown-based todo tracking, instruction reinforcement and external file change reminder
@@ -107,7 +108,6 @@ On first run, you'll be prompted to select a model. Your choice is saved as `mai
 | Provider    | Env Variable          | Models                                                                        |
 |-------------|-----------------------|-------------------------------------------------------------------------------|
 | anthropic   | `ANTHROPIC_API_KEY`   | sonnet, opus                                                                  |
-| claude      | N/A (OAuth)           | sonnet@claude, opus@claude (requires Claude Pro/Max subscription)             |
 | openai      | `OPENAI_API_KEY`      | gpt-5.2                                                                       |
 | openrouter  | `OPENROUTER_API_KEY`  | gpt-5.2, gpt-5.2-fast, gpt-5.1-codex-max, sonnet, opus, haiku, kimi, gemini-* |
 | deepseek    | `DEEPSEEK_API_KEY`    | deepseek                                                                      |
@@ -139,7 +139,6 @@ klaude auth login deepseek    # Set DEEPSEEK_API_KEY
 klaude auth login moonshot    # Set MOONSHOT_API_KEY
 
 # OAuth login for subscription-based providers
-klaude auth login claude      # Claude Pro/Max subscription
 klaude auth login codex       # ChatGPT Pro subscription
 ```
 
@@ -148,7 +147,6 @@ API keys are stored in `~/.klaude/klaude-auth.json` and used as fallback when en
 To logout from OAuth providers:
 
 ```bash
-klaude auth logout claude
 klaude auth logout codex
 ```
 
@@ -201,7 +199,6 @@ provider_list:
 ##### Supported Protocols
 
 - `anthropic` - Anthropic Messages API
-- `claude_oauth` - Claude OAuth (for Claude Pro/Max subscribers)
 - `openai` - OpenAI Chat Completion API
 - `responses` - OpenAI Responses API (for o-series, GPT-5, Codex)
 - `codex_oauth` - OpenAI Codex CLI (OAuth-based, for ChatGPT Pro subscribers)