klaude-code 1.2.6__py3-none-any.whl → 1.8.0__py3-none-any.whl

klaude_code/core/tool/shell/bash_tool.py

@@ -1,18 +1,44 @@
 import asyncio
+import contextlib
+import os
 import re
+import shlex
+import signal
 import subprocess
 from pathlib import Path
+from typing import Any
 
 from pydantic import BaseModel
 
 from klaude_code import const
 from klaude_code.core.tool.shell.command_safety import is_safe_command
 from klaude_code.core.tool.tool_abc import ToolABC, load_desc
+from klaude_code.core.tool.tool_context import get_current_file_tracker
 from klaude_code.core.tool.tool_registry import register
 from klaude_code.protocol import llm_param, model, tools
 
-# Regex to strip ANSI escape sequences from command output
-_ANSI_ESCAPE_RE = re.compile(r"\x1b\[[0-9;]*m")
+# Regex to strip ANSI and terminal control sequences from command output
+#
+# This is intentionally broader than just SGR color codes (e.g. "\x1b[31m").
+# Many interactive or TUI-style programs emit additional escape sequences
+# that move the cursor, clear the screen, or switch screen buffers
+# (CSI/OSC/DCS/APC/PM, etc). If these reach the Rich console, they can
+# corrupt the REPL layout. We therefore remove all of them before
+# rendering the output.
+_ANSI_ESCAPE_RE = re.compile(
+    r"""
+    \x1B
+    (?:
+        \[[0-?]*[ -/]*[@-~]         |  # CSI sequences
+        \][0-?]*.*?(?:\x07|\x1B\\) |  # OSC sequences
+        P.*?(?:\x07|\x1B\\)       |  # DCS sequences
+        _.*?(?:\x07|\x1B\\)       |  # APC sequences
+        \^.*?(?:\x07|\x1B\\)      |  # PM sequences
+        [@-Z\\-_]                      # 2-char sequences
+    )
+    """,
+    re.VERBOSE | re.DOTALL,
+)
 
 
 @register(tools.BASH)
@@ -67,31 +93,250 @@ class BashTool(ToolABC):
 
         # Run the command using bash -lc so shell semantics work (pipes, &&, etc.)
         # Capture stdout/stderr, respect timeout, and return a ToolMessage.
+        #
+        # Important: this tool is intentionally non-interactive.
+        # - Always detach stdin (DEVNULL) so interactive programs can't steal REPL input.
+        # - Always disable pagers/editors to avoid launching TUI subprocesses that can
+        #   leave the terminal in a bad state.
         cmd = ["bash", "-lc", args.command]
         timeout_sec = max(0.0, args.timeout_ms / 1000.0)
 
+        env = os.environ.copy()
+        env.update(
+            {
+                # Avoid blocking on git/jj prompts.
+                "GIT_TERMINAL_PROMPT": "0",
+                # Avoid pagers.
+                "PAGER": "cat",
+                "GIT_PAGER": "cat",
+                # Avoid opening editors.
+                "EDITOR": "true",
+                "VISUAL": "true",
+                "GIT_EDITOR": "true",
+                "JJ_EDITOR": "true",
+                # Encourage non-interactive output.
+                "TERM": "dumb",
+            }
+        )
+
+        def _hash_file_content_sha256(file_path: str) -> str | None:
+            try:
+                suffix = Path(file_path).suffix.lower()
+                if suffix in {".png", ".jpg", ".jpeg", ".gif", ".webp"}:
+                    import hashlib
+
+                    with open(file_path, "rb") as f:
+                        return hashlib.sha256(f.read()).hexdigest()
+
+                import hashlib
+
+                hasher = hashlib.sha256()
+                with open(file_path, encoding="utf-8", errors="replace") as f:
+                    for line in f:
+                        hasher.update(line.encode("utf-8"))
+                return hasher.hexdigest()
+            except (FileNotFoundError, IsADirectoryError, OSError, PermissionError, UnicodeDecodeError):
+                return None
+
+        def _resolve_in_dir(base_dir: str, path: str) -> str:
+            if os.path.isabs(path):
+                return os.path.abspath(path)
+            return os.path.abspath(os.path.join(base_dir, path))
+
+        def _track_files_read(file_paths: list[str], *, base_dir: str) -> None:
+            file_tracker = get_current_file_tracker()
+            if file_tracker is None:
+                return
+            for p in file_paths:
+                abs_path = _resolve_in_dir(base_dir, p)
+                if not os.path.exists(abs_path) or os.path.isdir(abs_path):
+                    continue
+                sha = _hash_file_content_sha256(abs_path)
+                if sha is None:
+                    continue
+                existing = file_tracker.get(abs_path)
+                is_mem = existing.is_memory if existing else False
+                with contextlib.suppress(Exception):
+                    file_tracker[abs_path] = model.FileStatus(
+                        mtime=Path(abs_path).stat().st_mtime,
+                        content_sha256=sha,
+                        is_memory=is_mem,
+                    )
+
+        def _track_files_written(file_paths: list[str], *, base_dir: str) -> None:
+            # Same as read tracking, but intentionally kept separate for clarity.
+            _track_files_read(file_paths, base_dir=base_dir)
+
+        def _track_mv(src_paths: list[str], dest_path: str, *, base_dir: str) -> None:
+            file_tracker = get_current_file_tracker()
+            if file_tracker is None:
+                return
+
+            abs_dest = _resolve_in_dir(base_dir, dest_path)
+            dest_is_dir = os.path.isdir(abs_dest)
+
+            for src in src_paths:
+                abs_src = _resolve_in_dir(base_dir, src)
+                abs_new = os.path.join(abs_dest, os.path.basename(abs_src)) if dest_is_dir else abs_dest
+
+                # Remove old entry if present.
+                existing = file_tracker.pop(abs_src, None)
+                is_mem = existing.is_memory if existing else False
+
+                if not os.path.exists(abs_new) or os.path.isdir(abs_new):
+                    continue
+
+                sha = _hash_file_content_sha256(abs_new)
+                if sha is None:
+                    continue
+                with contextlib.suppress(Exception):
+                    file_tracker[abs_new] = model.FileStatus(
+                        mtime=Path(abs_new).stat().st_mtime,
+                        content_sha256=sha,
+                        is_memory=is_mem,
+                    )
+
+        def _best_effort_update_file_tracker(command: str) -> None:
+            # Best-effort heuristics for common shell tools that access/modify files.
+            # We intentionally do not try to interpret complex shell scripts here.
+            try:
+                argv = shlex.split(command, posix=True)
+            except ValueError:
+                return
+            if not argv:
+                return
+
+            # Handle common patterns like: cd subdir && cat file
+            base_dir = os.getcwd()
+            while len(argv) >= 4 and argv[0] == "cd" and argv[2] == "&&":
+                dest = argv[1]
+                if dest != "-":
+                    base_dir = _resolve_in_dir(base_dir, dest)
+                argv = argv[3:]
+                if not argv:
+                    return
+
+            cmd0 = argv[0]
+            if cmd0 == "cat":
+                paths = [a for a in argv[1:] if a and not a.startswith("-") and a != "-"]
+                _track_files_read(paths, base_dir=base_dir)
+                return
+
+            if cmd0 == "sed":
+                # Support: sed [-i ...] 's/old/new/' file1 [file2 ...]
+                # and: sed -n 'Np' file
+                saw_script = False
+                file_paths: list[str] = []
+                for a in argv[1:]:
+                    if not a:
+                        continue
+                    if a == "--":
+                        continue
+                    if a.startswith("-") and not saw_script:
+                        continue
+                    if not saw_script and (a.startswith("s/") or a.startswith("s|") or a.endswith("p")):
+                        saw_script = True
+                        continue
+                    if saw_script and not a.startswith("-"):
+                        file_paths.append(a)
+
+                if file_paths:
+                    _track_files_written(file_paths, base_dir=base_dir)
+                return
+
+            if cmd0 == "mv":
+                # Support: mv [opts] src... dest
+                operands: list[str] = []
+                end_of_opts = False
+                for a in argv[1:]:
+                    if not end_of_opts and a == "--":
+                        end_of_opts = True
+                        continue
+                    if not end_of_opts and a.startswith("-"):
+                        continue
+                    operands.append(a)
+                if len(operands) < 2:
+                    return
+                srcs = operands[:-1]
+                dest = operands[-1]
+                _track_mv(srcs, dest, base_dir=base_dir)
+                return
+
+        async def _terminate_process(proc: asyncio.subprocess.Process) -> None:
+            # Best-effort termination. Ensure we don't hang on cancellation.
+            if proc.returncode is not None:
+                return
+
+            try:
+                if os.name == "posix":
+                    os.killpg(proc.pid, signal.SIGTERM)
+                else:
+                    proc.terminate()
+            except ProcessLookupError:
+                return
+            except OSError:
+                # Fall back to kill below.
+                pass
+
+            with contextlib.suppress(Exception):
+                await asyncio.wait_for(proc.wait(), timeout=1.0)
+                return
+
+            # Escalate to hard kill if it didn't exit quickly.
+            with contextlib.suppress(Exception):
+                if os.name == "posix":
+                    os.killpg(proc.pid, signal.SIGKILL)
+                else:
+                    proc.kill()
+            with contextlib.suppress(Exception):
+                await asyncio.wait_for(proc.wait(), timeout=1.0)
+
         try:
-            completed = await asyncio.to_thread(
-                subprocess.run,
-                cmd,
-                capture_output=True,
-                text=True,
-                timeout=timeout_sec,
-                check=False,
-            )
+            # Create a dedicated process group so we can terminate the whole tree.
+            # (macOS/Linux support start_new_session; Windows does not.)
+            kwargs: dict[str, Any] = {
+                "stdin": asyncio.subprocess.DEVNULL,
+                "stdout": asyncio.subprocess.PIPE,
+                "stderr": asyncio.subprocess.PIPE,
+                "env": env,
+            }
+            if os.name == "posix":
+                kwargs["start_new_session"] = True
+            elif os.name == "nt":  # pragma: no cover
+                kwargs["creationflags"] = getattr(subprocess, "CREATE_NEW_PROCESS_GROUP", 0)
 
-            stdout = _ANSI_ESCAPE_RE.sub("", completed.stdout or "")
-            stderr = _ANSI_ESCAPE_RE.sub("", completed.stderr or "")
-            rc = completed.returncode
+            proc = await asyncio.create_subprocess_exec(*cmd, **kwargs)
+            try:
+                stdout_b, stderr_b = await asyncio.wait_for(proc.communicate(), timeout=timeout_sec)
+            except TimeoutError:
+                with contextlib.suppress(Exception):
+                    await _terminate_process(proc)
+                return model.ToolResultItem(
+                    status="error",
+                    output=f"Timeout after {args.timeout_ms} ms running: {args.command}",
+                )
+            except asyncio.CancelledError:
+                # Ensure subprocess is stopped and propagate cancellation.
+                with contextlib.suppress(Exception):
+                    await asyncio.shield(_terminate_process(proc))
+                raise
+
+            stdout = _ANSI_ESCAPE_RE.sub("", (stdout_b or b"").decode(errors="replace"))
+            stderr = _ANSI_ESCAPE_RE.sub("", (stderr_b or b"").decode(errors="replace"))
+            rc = proc.returncode
 
             if rc == 0:
                 output = stdout if stdout else ""
                 # Include stderr if there is useful diagnostics despite success
                 if stderr.strip():
                     output = (output + ("\n" if output else "")) + f"[stderr]\n{stderr}"
+
+                _best_effort_update_file_tracker(args.command)
                 return model.ToolResultItem(
                     status="success",
-                    output=output.strip(),
+                    # Preserve leading whitespace for tools like `nl -ba`.
+                    # Only trim trailing newlines to avoid adding an extra blank line in the UI.
+                    output=output.rstrip("\n"),
                 )
             else:
                 combined = ""
@@ -103,20 +348,18 @@ class BashTool(ToolABC):
                     combined = f"Command exited with code {rc}"
                 return model.ToolResultItem(
                     status="error",
-                    output=combined.strip(),
+                    # Preserve leading whitespace; only trim trailing newlines.
+                    output=combined.rstrip("\n"),
                 )
-
-        except subprocess.TimeoutExpired:
-            return model.ToolResultItem(
-                status="error",
-                output=f"Timeout after {args.timeout_ms} ms running: {args.command}",
-            )
         except FileNotFoundError:
             return model.ToolResultItem(
                 status="error",
                 output="bash not found on system path",
             )
-        except Exception as e:  # safeguard against unexpected failures
+        except asyncio.CancelledError:
+            # Propagate cooperative cancellation so outer layers can handle interrupts correctly.
+            raise
+        except OSError as e:  # safeguard: catch remaining OS-level errors (permissions, resources, etc.)
             return model.ToolResultItem(
                 status="error",
                 output=f"Execution error: {e}",

klaude_code/core/tool/shell/command_safety.py

@@ -275,12 +275,10 @@ def _is_safe_argv(argv: list[str]) -> SafetyCheckResult:
             "tag",
             "clone",
             "worktree",
+            "push",
+            "pull",
+            "remote",
         }
-        # Block remote operations
-        blocked_git_cmds = {"push", "pull", "remote"}
-
-        if sub in blocked_git_cmds:
-            return SafetyCheckResult(False, f"git: Remote operation '{sub}' not allowed")
         if sub not in allowed_git_cmds:
             return SafetyCheckResult(False, f"git: Subcommand '{sub}' not in allow list")
         return SafetyCheckResult(True)
@@ -293,7 +291,6 @@ def _is_safe_argv(argv: list[str]) -> SafetyCheckResult:
         "ruff",
         "pyright",
         "make",
-        "isort",
         "npm",
         "pnpm",
         "bun",

klaude_code/core/tool/{memory → skill}/skill_tool.py

@@ -1,27 +1,22 @@
+"""SkillTool - Tool for agent to activate and load skills."""
+
 from pathlib import Path
 
 from pydantic import BaseModel
 
-from klaude_code.core.tool.memory.skill_loader import SkillLoader
 from klaude_code.core.tool.tool_abc import ToolABC, load_desc
 from klaude_code.core.tool.tool_registry import register
 from klaude_code.protocol import llm_param, model, tools
+from klaude_code.skill import get_available_skills, get_skill, list_skill_names
 
 
 @register(tools.SKILL)
 class SkillTool(ToolABC):
-    """Tool to execute/load a skill within the main conversation"""
-
-    _skill_loader: SkillLoader | None = None
-
-    @classmethod
-    def set_skill_loader(cls, loader: SkillLoader) -> None:
-        """Set the skill loader instance"""
-        cls._skill_loader = loader
+    """Tool to execute/load a skill within the main conversation."""
 
     @classmethod
     def schema(cls) -> llm_param.ToolSchema:
-        """Generate schema with embedded available skills metadata"""
+        """Generate schema with embedded available skills metadata."""
         skills_xml = cls._generate_skills_xml()
 
         return llm_param.ToolSchema(
@@ -42,16 +37,17 @@ class SkillTool(ToolABC):
 
     @classmethod
     def _generate_skills_xml(cls) -> str:
-        """Generate XML format skills metadata"""
-        if not cls._skill_loader:
+        """Generate XML format skills metadata."""
+        skills = get_available_skills()
+        if not skills:
             return ""
 
         xml_parts: list[str] = []
-        for skill in cls._skill_loader.loaded_skills.values():
+        for name, description, location in skills:
             xml_parts.append(f"""<skill>
-<name>{skill.name}</name>
-<description>{skill.description}</description>
-<location>{skill.location}</location>
+<name>{name}</name>
+<description>{description}</description>
+<location>{location}</location>
 </skill>""")
         return "\n".join(xml_parts)
 
@@ -60,7 +56,7 @@ class SkillTool(ToolABC):
 
     @classmethod
     async def call(cls, arguments: str) -> model.ToolResultItem:
-        """Load and return full skill content"""
+        """Load and return full skill content."""
         try:
             args = cls.SkillArguments.model_validate_json(arguments)
         except ValueError as e:
@@ -69,16 +65,10 @@ class SkillTool(ToolABC):
                 output=f"Invalid arguments: {e}",
             )
 
-        if not cls._skill_loader:
-            return model.ToolResultItem(
-                status="error",
-                output="Skill loader not initialized",
-            )
-
-        skill = cls._skill_loader.get_skill(args.command)
+        skill = get_skill(args.command)
 
         if not skill:
-            available = ", ".join(cls._skill_loader.list_skills())
+            available = ", ".join(list_skill_names())
             return model.ToolResultItem(
                 status="error",
                 output=f"Skill '{args.command}' does not exist. Available skills: {available}",
@@ -88,7 +78,7 @@ class SkillTool(ToolABC):
         base_dir = str(skill.skill_path.parent) if skill.skill_path else "unknown"
 
         # Return with loading message format
-        result = f"""<command-message>The "{skill.name}" skill is running</command-message>
+        result = f"""<command-message>The "{skill.name}" skill is activated</command-message>
 <command-name>{skill.name}</command-name>
 
 Base directory for this skill: {base_dir}

klaude_code/core/tool/sub_agent_tool.py

@@ -10,7 +10,7 @@ import asyncio
 import json
 from typing import TYPE_CHECKING, ClassVar
 
-from klaude_code.core.tool.tool_abc import ToolABC
+from klaude_code.core.tool.tool_abc import ToolABC, ToolConcurrencyPolicy, ToolMetadata
 from klaude_code.core.tool.tool_context import current_run_subtask_callback
 from klaude_code.protocol import llm_param, model
 
@@ -36,6 +36,10 @@ class SubAgentTool(ToolABC):
             {"_profile": profile},
         )
 
+    @classmethod
+    def metadata(cls) -> ToolMetadata:
+        return ToolMetadata(concurrency_policy=ToolConcurrencyPolicy.CONCURRENT, has_side_effects=True)
+
     @classmethod
     def schema(cls) -> llm_param.ToolSchema:
         profile = cls._profile
@@ -63,12 +67,18 @@ class SubAgentTool(ToolABC):
         prompt = profile.prompt_builder(args)
         description = args.get("description", "")
 
+        # Extract output_schema if configured
+        output_schema = None
+        if profile.output_schema_arg:
+            output_schema = args.get(profile.output_schema_arg)
+
         try:
             result = await runner(
                 model.SubAgentState(
                     sub_agent_type=profile.name,
                     sub_agent_desc=description,
                     sub_agent_prompt=prompt,
+                    output_schema=output_schema,
                 )
             )
         except asyncio.CancelledError:
@@ -79,5 +89,6 @@ class SubAgentTool(ToolABC):
         return model.ToolResultItem(
             status="success" if not result.error else "error",
             output=result.task_result or "",
-            ui_extra=model.ToolResultUIExtra(type=model.ToolResultUIExtraType.SESSION_ID, session_id=result.session_id),
+            ui_extra=model.SessionIdUIExtra(session_id=result.session_id),
+            task_metadata=result.task_metadata,
         )

klaude_code/core/tool/todo/todo_write_tool.md

@@ -22,161 +22,4 @@ Skip using this tool when:
 
 NOTE that you should not use this tool if there is only one trivial task to do. In this case you are better off just doing the task directly.
 
-#### Examples of When to Use the Todo List
-
-<example>
-User: I want to add a dark mode toggle to the application settings. Make sure you run the tests and build when you're done!
-Assistant: I'll help add a dark mode toggle to your application settings. Let me create a todo list to track this implementation.
-*Creates todo list with the following items:*
-1. Creating dark mode toggle component in Settings page
-2. Adding dark mode state management (context/store)
-3. Implementing CSS-in-JS styles for dark theme
-4. Updating existing components to support theme switching
-5. Running tests and build process, addressing any failures or errors that occur
-*Begins working on the first task*
-
-<reasoning>
-The assistant used the todo list because:
-1. Adding dark mode is a multi-step feature requiring UI, state management, and styling changes
-2. The user explicitly requested tests and build be run afterward
-3. The assistant inferred that tests and build need to pass by adding "Ensure tests and build succeed" as the final task
-</reasoning>
-</example>
-
-<example>
-User: Help me rename the function getCwd to getCurrentWorkingDirectory across my project
-Assistant: Let me first search through your codebase to find all occurrences of 'getCwd'.
-*Uses grep or search tools to locate all instances of getCwd in the codebase*
-Assistant: I've found 15 instances of 'getCwd' across 8 different files. Let me create a todo list to track these changes.
-*Creates todo list with specific items for each file that needs updating*
-
-<reasoning>
-The assistant used the todo list because:
-1. First, the assistant searched to understand the scope of the task
-2. Upon finding multiple occurrences across different files, it determined this was a complex task with multiple steps
-3. The todo list helps ensure every instance is tracked and updated systematically
-4. This approach prevents missing any occurrences and maintains code consistency
-</reasoning>
-</example>
-
-
-<example>
-User: I need to implement these features for my e-commerce site: user registration, product catalog, shopping cart, and checkout flow.
-Assistant: I'll help implement these features. First, let's add all the features to the todo list.
-*Creates a todo list breaking down each feature into specific tasks based on the project architecture*
-Assistant: Let's start with user registration. This will involve creating database models, API endpoints, and frontend forms that integrate with your existing authentication system.
-
-<reasoning>
-The assistant used the todo list because:
-1. The user provided multiple complex features to implement in a comma separated list
-2. The todo list helps organize these large features into manageable tasks
-3. This approach allows for tracking progress across the entire implementation
-</reasoning>
-</example>
-
-<example>
-User: Can you help optimize my React application? It's rendering slowly and has performance issues.</user>
-Assistant: I'll help optimize your React application. First, let me examine your codebase to identify potential performance bottlenecks.
-*Reviews component structure, render patterns, state management, and data fetching*
-Assistant: After analyzing your codebase, I've identified several performance issues. Let me create a todo list to track our optimization efforts.
-*Creates todo list with items like: 1) Implementing memoization for expensive calculations in ProductList, 2) Adding virtualization for long lists in Dashboard, 3) Optimizing image loading in Gallery component, 4) Fixing state update loops in ShoppingCart, 5) Reviewing bundle size and implementing code splitting*
-Let's start by implementing memoization for the expensive calculations in your ProductList component.</assistant>
-
-<reasoning>
-The assistant used the todo list because:
-1. First, the assistant examined the codebase to identify specific performance issues
-2. Based on this analysis, it identified multiple optimization opportunities
-3. Performance optimization is a non-trivial task requiring multiple steps
-4. The todo list helps methodically track improvements across different components
-5. This systematic approach ensures all performance bottlenecks are addressed
-</reasoning>
-</example>
-
-#### Examples of When NOT to Use the Todo List
-
-<example>
-User: How do I print 'Hello World' in Python?
-Assistant: In Python, you can print "Hello World" with this simple code:
-
-python
-print("Hello World")
-
-This will output the text "Hello World" to the console when executed.</assistant>
-
-<reasoning>
-The assistant did not use the todo list because this is a single, trivial task that can be completed in one step. There's no need to track multiple tasks or steps for such a straightforward request.
-</reasoning>
-</example>
-
-<example>
-User: What does the git status command do?
-Assistant: The git status command shows the current state of your working directory and staging area. It displays which changes have been staged, which haven't, and which files aren't being tracked by Git. This is a fundamental command for understanding the current state of your repository before making commits.
-
-<reasoning>
-The assistant did not use the todo list because this is an informational request with no actual coding task to complete. The user is simply asking for an explanation, not for the assistant to perform multiple steps or tasks.
-</reasoning>
-</example>
-
-<example>
-User: Can you add a comment to the calculateTotal function to explain what it does?
-Assistant: Sure, let me add a comment to the calculateTotal function to explain what it does.
-* Uses the Edit tool to add a comment to the calculateTotal function *
-
-<reasoning>
-The assistant did not use the todo list because this is a single, straightforward task confined to one location in the code. Adding a comment doesn't require tracking multiple steps or systematic organization.
-</reasoning>
-</example>
-
-<example>
-User: Run npm install for me and tell me what happens.
-Assistant: I'll run the npm install command for you.
-
-*Executes: npm install*
-
-The command completed successfully. Here's the output:
-[Output of npm install command]
-
-All dependencies have been installed according to your package.json file.
-
-<reasoning>
-The assistant did not use the todo list because this is a single command execution with immediate results. There are no multiple steps to track or organize, making the todo list unnecessary for this straightforward task.
-</reasoning>
-</example>
-
-#### Task States and Management
-
-1. **Task States**: Use these states to track progress:
-   - pending: Task not yet started
-   - in_progress: Currently working on (limit to ONE task at a time)
-   - completed: Task finished successfully
-
-   **IMPORTANT**: Task descriptions must have two forms:
-   - content: The imperative form describing what needs to be done (e.g., "Run tests", "Build the project")
-   - activeForm: The present continuous form shown during execution (e.g., "Running tests", "Building the project")
-
-2. **Task Management**:
-   - Update task status in real-time as you work
-   - Mark tasks complete IMMEDIATELY after finishing (don't batch completions)
-   - Exactly ONE task must be in_progress at any time (not less, not more)
-   - Complete current tasks before starting new ones
-   - Remove tasks that are no longer relevant from the list entirely
-
-3. **Task Completion Requirements**:
-   - ONLY mark a task as completed when you have FULLY accomplished it
-   - If you encounter errors, blockers, or cannot finish, keep the task as in_progress
-   - When blocked, create a new task describing what needs to be resolved
-   - Never mark a task as completed if:
-     - Tests are failing
-     - Implementation is partial
-     - You encountered unresolved errors
-     - You couldn't find necessary files or dependencies
-
-4. **Task Breakdown**:
-   - Create specific, actionable items
-   - Break complex tasks into smaller, manageable steps
-   - Use clear, descriptive task names
-   - Always provide both forms:
-     - content: "Fix authentication bug"
-     - activeForm: "Fixing authentication bug"
-
 When in doubt, use this tool. Being proactive with task management demonstrates attentiveness and ensures you complete all requirements successfully.

klaude_code/core/tool/todo/todo_write_tool.py

@@ -116,6 +116,6 @@ Your todo list has changed. DO NOT mention this explicitly to the user. Here are
         return model.ToolResultItem(
             status="success",
             output=response,
-            ui_extra=model.ToolResultUIExtra(type=model.ToolResultUIExtraType.TODO_LIST, todo_list=ui_extra),
+            ui_extra=model.TodoListUIExtra(todo_list=ui_extra),
             side_effects=[model.ToolSideEffect.TODO_CHANGE],
         )