kimi-cli 0.43__py3-none-any.whl → 0.45__py3-none-any.whl

kimi_cli/CHANGELOG.md

@@ -9,6 +9,28 @@ Internal builds may append content to the Unreleased section.
 Only write entries that are worth mentioning to users.
 -->
 
+## [0.45] - 2025-10-31
+
+### Added
+
+- Allow `KIMI_MODEL_CAPABILITIES` environment variable to override model capabilities
+- Add `--no-markdown` option to disable markdown rendering
+- Support `openai_responses` LLM provider type
+
+### Fixed
+
+- Fix crash when continuing a session
+
+## [0.44] - 2025-10-30
+
+### Changed
+
+- Improve startup time
+
+### Fixed
+
+- Fix potential invalid bytes in user input
+
 ## [0.43] - 2025-10-30
 
 ### Added

kimi_cli/__init__.py

@@ -1,192 +0,0 @@
-import contextlib
-import os
-import warnings
-from collections.abc import Generator
-from pathlib import Path
-from typing import Any
-
-from pydantic import SecretStr
-
-from kimi_cli.agentspec import DEFAULT_AGENT_FILE
-from kimi_cli.config import LLMModel, LLMProvider, load_config
-from kimi_cli.llm import augment_provider_with_env_vars, create_llm
-from kimi_cli.session import Session
-from kimi_cli.soul.agent import load_agent
-from kimi_cli.soul.context import Context
-from kimi_cli.soul.kimisoul import KimiSoul
-from kimi_cli.soul.runtime import Runtime
-from kimi_cli.ui.acp import ACPServer
-from kimi_cli.ui.print import InputFormat, OutputFormat, PrintApp
-from kimi_cli.ui.shell import ShellApp, WelcomeInfoItem
-from kimi_cli.utils.logging import StreamToLogger, logger
-
-
-class KimiCLI:
-    @staticmethod
-    async def create(
-        session: Session,
-        *,
-        yolo: bool = False,
-        stream: bool = True,  # TODO: remove this when we have a correct print mode impl
-        mcp_configs: list[dict[str, Any]] | None = None,
-        config_file: Path | None = None,
-        model_name: str | None = None,
-        agent_file: Path | None = None,
-    ) -> "KimiCLI":
-        """
-        Create a KimiCLI instance.
-
-        Args:
-            session (Session): A session created by `Session.create` or `Session.continue_`.
-            yolo (bool, optional): Approve all actions without confirmation. Defaults to False.
-            stream (bool, optional): Use stream mode when calling LLM API. Defaults to True.
-            config_file (Path | None, optional): Path to the configuration file. Defaults to None.
-            model_name (str | None, optional): Name of the model to use. Defaults to None.
-            agent_file (Path | None, optional): Path to the agent file. Defaults to None.
-
-        Raises:
-            FileNotFoundError: When the agent file is not found.
-            ConfigError(KimiCLIException): When the configuration is invalid.
-            AgentSpecError(KimiCLIException): When the agent specification is invalid.
-        """
-        config = load_config(config_file)
-        logger.info("Loaded config: {config}", config=config)
-
-        model: LLMModel | None = None
-        provider: LLMProvider | None = None
-
-        # try to use config file
-        if not model_name and config.default_model:
-            # no --model specified && default model is set in config
-            model = config.models[config.default_model]
-            provider = config.providers[model.provider]
-        if model_name and model_name in config.models:
-            # --model specified && model is set in config
-            model = config.models[model_name]
-            provider = config.providers[model.provider]
-
-        if not model:
-            model = LLMModel(provider="", model="", max_context_size=100_000)
-            provider = LLMProvider(type="kimi", base_url="", api_key=SecretStr(""))
-
-        # try overwrite with environment variables
-        assert provider is not None
-        assert model is not None
-        env_overrides = augment_provider_with_env_vars(provider, model)
-
-        if not provider.base_url or not model.model:
-            llm = None
-        else:
-            logger.info("Using LLM provider: {provider}", provider=provider)
-            logger.info("Using LLM model: {model}", model=model)
-            llm = create_llm(provider, model, stream=stream, session_id=session.id)
-
-        welcome_info = [
-            WelcomeInfoItem(name="Directory", value=str(session.work_dir)),
-            WelcomeInfoItem(name="Session", value=session.id),
-        ]
-        if base_url := env_overrides.get("KIMI_BASE_URL"):
-            welcome_info.append(
-                WelcomeInfoItem(
-                    name="API URL",
-                    value=f"{base_url} (from KIMI_BASE_URL)",
-                    level=WelcomeInfoItem.Level.WARN,
-                )
-            )
-        if not llm:
-            welcome_info.append(
-                WelcomeInfoItem(
-                    name="Model",
-                    value="not set, send /setup to configure",
-                    level=WelcomeInfoItem.Level.WARN,
-                )
-            )
-        elif "KIMI_MODEL_NAME" in env_overrides:
-            welcome_info.append(
-                WelcomeInfoItem(
-                    name="Model",
-                    value=f"{model.model} (from KIMI_MODEL_NAME)",
-                    level=WelcomeInfoItem.Level.WARN,
-                )
-            )
-        else:
-            welcome_info.append(
-                WelcomeInfoItem(
-                    name="Model",
-                    value=model.model,
-                    level=WelcomeInfoItem.Level.INFO,
-                )
-            )
-
-        runtime = await Runtime.create(config, llm, session, yolo)
-
-        if agent_file is None:
-            agent_file = DEFAULT_AGENT_FILE
-        agent = await load_agent(agent_file, runtime, mcp_configs=mcp_configs or [])
-
-        context = Context(session.history_file)
-        await context.restore()
-
-        soul = KimiSoul(
-            agent,
-            runtime,
-            context=context,
-        )
-        return KimiCLI(soul, session, welcome_info)
-
-    def __init__(
-        self,
-        soul: KimiSoul,
-        session: Session,
-        welcome_info: list[WelcomeInfoItem],
-    ) -> None:
-        self._soul = soul
-        self._session = session
-        self._welcome_info = welcome_info
-
-    @property
-    def soul(self) -> KimiSoul:
-        """Get the KimiSoul instance."""
-        return self._soul
-
-    @property
-    def session(self) -> Session:
-        """Get the Session instance."""
-        return self._session
-
-    @contextlib.contextmanager
-    def _app_env(self) -> Generator[None]:
-        original_cwd = Path.cwd()
-        os.chdir(self._session.work_dir)
-        try:
-            # to ignore possible warnings from dateparser
-            warnings.filterwarnings("ignore", category=DeprecationWarning)
-            with contextlib.redirect_stderr(StreamToLogger()):
-                yield
-        finally:
-            os.chdir(original_cwd)
-
-    async def run_shell_mode(self, command: str | None = None) -> bool:
-        with self._app_env():
-            app = ShellApp(self._soul, welcome_info=self._welcome_info)
-            return await app.run(command)
-
-    async def run_print_mode(
-        self,
-        input_format: InputFormat,
-        output_format: OutputFormat,
-        command: str | None = None,
-    ) -> bool:
-        with self._app_env():
-            app = PrintApp(
-                self._soul,
-                input_format,
-                output_format,
-                self._session.history_file,
-            )
-            return await app.run(command)
-
-    async def run_acp_server(self) -> bool:
-        with self._app_env():
-            app = ACPServer(self._soul)
-            return await app.run()

kimi_cli/app.py

@@ -0,0 +1,195 @@
+import contextlib
+import os
+import warnings
+from collections.abc import Generator
+from pathlib import Path
+from typing import Any
+
+from pydantic import SecretStr
+
+from kimi_cli.agentspec import DEFAULT_AGENT_FILE
+from kimi_cli.cli import InputFormat, OutputFormat
+from kimi_cli.config import LLMModel, LLMProvider, load_config
+from kimi_cli.llm import augment_provider_with_env_vars, create_llm
+from kimi_cli.session import Session
+from kimi_cli.soul.agent import load_agent
+from kimi_cli.soul.context import Context
+from kimi_cli.soul.kimisoul import KimiSoul
+from kimi_cli.soul.runtime import Runtime
+from kimi_cli.utils.logging import StreamToLogger, logger
+
+
+class KimiCLI:
+    @staticmethod
+    async def create(
+        session: Session,
+        *,
+        yolo: bool = False,
+        stream: bool = True,  # TODO: remove this when we have a correct print mode impl
+        mcp_configs: list[dict[str, Any]] | None = None,
+        config_file: Path | None = None,
+        model_name: str | None = None,
+        agent_file: Path | None = None,
+    ) -> "KimiCLI":
+        """
+        Create a KimiCLI instance.
+
+        Args:
+            session (Session): A session created by `Session.create` or `Session.continue_`.
+            yolo (bool, optional): Approve all actions without confirmation. Defaults to False.
+            stream (bool, optional): Use stream mode when calling LLM API. Defaults to True.
+            config_file (Path | None, optional): Path to the configuration file. Defaults to None.
+            model_name (str | None, optional): Name of the model to use. Defaults to None.
+            agent_file (Path | None, optional): Path to the agent file. Defaults to None.
+
+        Raises:
+            FileNotFoundError: When the agent file is not found.
+            ConfigError(KimiCLIException): When the configuration is invalid.
+            AgentSpecError(KimiCLIException): When the agent specification is invalid.
+        """
+        config = load_config(config_file)
+        logger.info("Loaded config: {config}", config=config)
+
+        model: LLMModel | None = None
+        provider: LLMProvider | None = None
+
+        # try to use config file
+        if not model_name and config.default_model:
+            # no --model specified && default model is set in config
+            model = config.models[config.default_model]
+            provider = config.providers[model.provider]
+        if model_name and model_name in config.models:
+            # --model specified && model is set in config
+            model = config.models[model_name]
+            provider = config.providers[model.provider]
+
+        if not model:
+            model = LLMModel(provider="", model="", max_context_size=100_000)
+            provider = LLMProvider(type="kimi", base_url="", api_key=SecretStr(""))
+
+        # try overwrite with environment variables
+        assert provider is not None
+        assert model is not None
+        env_overrides = augment_provider_with_env_vars(provider, model)
+
+        if not provider.base_url or not model.model:
+            llm = None
+        else:
+            logger.info("Using LLM provider: {provider}", provider=provider)
+            logger.info("Using LLM model: {model}", model=model)
+            llm = create_llm(provider, model, stream=stream, session_id=session.id)
+
+        runtime = await Runtime.create(config, llm, session, yolo)
+
+        if agent_file is None:
+            agent_file = DEFAULT_AGENT_FILE
+        agent = await load_agent(agent_file, runtime, mcp_configs=mcp_configs or [])
+
+        context = Context(session.history_file)
+        await context.restore()
+
+        soul = KimiSoul(
+            agent,
+            runtime,
+            context=context,
+        )
+        return KimiCLI(soul, runtime, env_overrides)
+
+    def __init__(
+        self,
+        _soul: KimiSoul,
+        _runtime: Runtime,
+        _env_overrides: dict[str, str],
+    ) -> None:
+        self._soul = _soul
+        self._runtime = _runtime
+        self._env_overrides = _env_overrides
+
+    @property
+    def soul(self) -> KimiSoul:
+        """Get the KimiSoul instance."""
+        return self._soul
+
+    @property
+    def session(self) -> Session:
+        """Get the Session instance."""
+        return self._runtime.session
+
+    @contextlib.contextmanager
+    def _app_env(self) -> Generator[None]:
+        original_cwd = Path.cwd()
+        os.chdir(self._runtime.session.work_dir)
+        try:
+            # to ignore possible warnings from dateparser
+            warnings.filterwarnings("ignore", category=DeprecationWarning)
+            with contextlib.redirect_stderr(StreamToLogger()):
+                yield
+        finally:
+            os.chdir(original_cwd)
+
+    async def run_shell_mode(self, command: str | None = None, markdown: bool = True) -> bool:
+        from kimi_cli.ui.shell import ShellApp, WelcomeInfoItem
+
+        welcome_info = [
+            WelcomeInfoItem(name="Directory", value=str(self._runtime.session.work_dir)),
+            WelcomeInfoItem(name="Session", value=self._runtime.session.id),
+        ]
+        if base_url := self._env_overrides.get("KIMI_BASE_URL"):
+            welcome_info.append(
+                WelcomeInfoItem(
+                    name="API URL",
+                    value=f"{base_url} (from KIMI_BASE_URL)",
+                    level=WelcomeInfoItem.Level.WARN,
+                )
+            )
+        if not self._runtime.llm:
+            welcome_info.append(
+                WelcomeInfoItem(
+                    name="Model",
+                    value="not set, send /setup to configure",
+                    level=WelcomeInfoItem.Level.WARN,
+                )
+            )
+        elif "KIMI_MODEL_NAME" in self._env_overrides:
+            welcome_info.append(
+                WelcomeInfoItem(
+                    name="Model",
+                    value=f"{self._soul.model} (from KIMI_MODEL_NAME)",
+                    level=WelcomeInfoItem.Level.WARN,
+                )
+            )
+        else:
+            welcome_info.append(
+                WelcomeInfoItem(
+                    name="Model",
+                    value=self._soul.model,
+                    level=WelcomeInfoItem.Level.INFO,
+                )
+            )
+        with self._app_env():
+            app = ShellApp(self._soul, welcome_info=welcome_info, markdown=markdown)
+            return await app.run(command)
+
+    async def run_print_mode(
+        self,
+        input_format: InputFormat,
+        output_format: OutputFormat,
+        command: str | None = None,
+    ) -> bool:
+        from kimi_cli.ui.print import PrintApp
+
+        with self._app_env():
+            app = PrintApp(
+                self._soul,
+                input_format,
+                output_format,
+                self._runtime.session.history_file,
+            )
+            return await app.run(command)
+
+    async def run_acp_server(self) -> bool:
+        from kimi_cli.ui.acp import ACPServer
+
+        with self._app_env():
+            app = ACPServer(self._soul)
+            return await app.run()

kimi_cli/cli.py

@@ -7,12 +7,7 @@ from typing import Any, Literal, get_args
 
 import click
 
-from kimi_cli import KimiCLI
 from kimi_cli.constant import VERSION
-from kimi_cli.session import Session
-from kimi_cli.share import get_share_dir
-from kimi_cli.ui.print import InputFormat, OutputFormat
-from kimi_cli.utils.logging import logger
 
 
 class Reload(Exception):
@@ -22,6 +17,8 @@ class Reload(Exception):
 
 
 UIMode = Literal["shell", "print", "acp"]
+InputFormat = Literal["text", "stream-json"]
+OutputFormat = Literal["text", "stream-json"]
 
 
 @click.command(context_settings=dict(help_option_names=["-h", "--help"]))
@@ -140,6 +137,12 @@ UIMode = Literal["shell", "print", "acp"]
     default=False,
     help="Automatically approve all actions. Default: no.",
 )
+@click.option(
+    "--markdown/--no-markdown",
+    is_flag=True,
+    default=True,
+    help="Enable/disable markdown rendering in shell UI. Default: yes.",
+)
 def kimi(
     verbose: bool,
     debug: bool,
@@ -154,8 +157,13 @@ def kimi(
     mcp_config_file: list[Path],
     mcp_config: list[str],
     yolo: bool,
+    markdown: bool,
 ):
     """Kimi, your next CLI agent."""
+    from kimi_cli.app import KimiCLI
+    from kimi_cli.session import Session
+    from kimi_cli.share import get_share_dir
+    from kimi_cli.utils.logging import logger
 
     def _noop_echo(*args: Any, **kwargs: Any):
         pass
@@ -219,7 +227,7 @@ def kimi(
         )
         match ui:
             case "shell":
-                return await instance.run_shell_mode(command)
+                return await instance.run_shell_mode(command, markdown=markdown)
             case "print":
                 return await instance.run_print_mode(
                     input_format or "text",

kimi_cli/config.py

@@ -12,7 +12,7 @@ from kimi_cli.utils.logging import logger
 class LLMProvider(BaseModel):
     """LLM provider configuration."""
 
-    type: Literal["kimi", "openai_legacy", "_chaos"]
+    type: Literal["kimi", "openai_legacy", "openai_responses", "_chaos"]
     """Provider type"""
     base_url: str
     """API base URL"""

kimi_cli/llm.py

@@ -1,10 +1,7 @@
 import os
-from typing import NamedTuple
+from typing import NamedTuple, cast, get_args
 
 from kosong.base.chat_provider import ChatProvider
-from kosong.chat_provider.chaos import ChaosChatProvider, ChaosConfig
-from kosong.chat_provider.kimi import Kimi
-from kosong.chat_provider.openai_legacy import OpenAILegacy
 from pydantic import SecretStr
 
 from kimi_cli.config import LLMModel, LLMModelCapability, LLMProvider
@@ -44,11 +41,19 @@ def augment_provider_with_env_vars(provider: LLMProvider, model: LLMModel) -> di
                 applied["KIMI_API_KEY"] = "******"
             if model_name := os.getenv("KIMI_MODEL_NAME"):
                 model.model = model_name
-                applied["KIMI_MODEL_NAME"] = model.model
+                applied["KIMI_MODEL_NAME"] = model_name
             if max_context_size := os.getenv("KIMI_MODEL_MAX_CONTEXT_SIZE"):
                 model.max_context_size = int(max_context_size)
-                applied["KIMI_MODEL_MAX_CONTEXT_SIZE"] = str(model.max_context_size)
-        case "openai_legacy":
+                applied["KIMI_MODEL_MAX_CONTEXT_SIZE"] = max_context_size
+            if capabilities := os.getenv("KIMI_MODEL_CAPABILITIES"):
+                caps_lower = (cap.strip().lower() for cap in capabilities.split(",") if cap.strip())
+                model.capabilities = set(
+                    cast(LLMModelCapability, cap)
+                    for cap in caps_lower
+                    if cap in get_args(LLMModelCapability)
+                )
+                applied["KIMI_MODEL_CAPABILITIES"] = capabilities
+        case "openai_legacy" | "openai_responses":
             if base_url := os.getenv("OPENAI_BASE_URL"):
                 provider.base_url = base_url
             if api_key := os.getenv("OPENAI_API_KEY"):
@@ -68,6 +73,8 @@ def create_llm(
 ) -> LLM:
     match provider.type:
         case "kimi":
+            from kosong.chat_provider.kimi import Kimi
+
             chat_provider = Kimi(
                 model=model.model,
                 base_url=provider.base_url,
@@ -81,13 +88,26 @@ def create_llm(
             if session_id:
                 chat_provider = chat_provider.with_generation_kwargs(prompt_cache_key=session_id)
         case "openai_legacy":
+            from kosong.chat_provider.openai_legacy import OpenAILegacy
+
             chat_provider = OpenAILegacy(
                 model=model.model,
                 base_url=provider.base_url,
                 api_key=provider.api_key.get_secret_value(),
                 stream=stream,
             )
+        case "openai_responses":
+            from kosong.chat_provider.openai_responses import OpenAIResponses
+
+            chat_provider = OpenAIResponses(
+                model=model.model,
+                base_url=provider.base_url,
+                api_key=provider.api_key.get_secret_value(),
+                stream=stream,
+            )
         case "_chaos":
+            from kosong.chat_provider.chaos import ChaosChatProvider, ChaosConfig
+
             chat_provider = ChaosChatProvider(
                 model=model.model,
                 base_url=provider.base_url,

kimi_cli/soul/agent.py

@@ -4,7 +4,6 @@ import string
 from pathlib import Path
 from typing import Any, NamedTuple
 
-import fastmcp
 from kosong.tooling import CallableTool, CallableTool2, Toolset
 
 from kimi_cli.agentspec import ResolvedAgentSpec, load_agent_spec
@@ -14,7 +13,6 @@ from kimi_cli.soul.approval import Approval
 from kimi_cli.soul.denwarenji import DenwaRenji
 from kimi_cli.soul.runtime import BuiltinSystemPromptArgs, Runtime
 from kimi_cli.soul.toolset import CustomToolset
-from kimi_cli.tools.mcp import MCPTool
 from kimi_cli.utils.logging import logger
 
 
@@ -143,6 +141,10 @@ async def _load_mcp_tools(
         ValueError: If the MCP config is not valid.
         RuntimeError: If the MCP server cannot be connected.
     """
+    import fastmcp
+
+    from kimi_cli.tools.mcp import MCPTool
+
     for mcp_config in mcp_configs:
         logger.info("Loading MCP tools from: {mcp_config}", mcp_config=mcp_config)
         client = fastmcp.Client(mcp_config)

kimi_cli/soul/runtime.py

@@ -1,3 +1,4 @@
+import asyncio
 import subprocess
 import sys
 from datetime import datetime
@@ -75,9 +76,10 @@ class Runtime(NamedTuple):
         session: Session,
         yolo: bool,
     ) -> "Runtime":
-        # FIXME: do these asynchronously
-        ls_output = _list_work_dir(session.work_dir)
-        agents_md = load_agents_md(session.work_dir) or ""
+        ls_output, agents_md = await asyncio.gather(
+            asyncio.to_thread(_list_work_dir, session.work_dir),
+            asyncio.to_thread(load_agents_md, session.work_dir),
+        )
 
         return Runtime(
             config=config,
@@ -87,7 +89,7 @@ class Runtime(NamedTuple):
                 KIMI_NOW=datetime.now().astimezone().isoformat(),
                 KIMI_WORK_DIR=session.work_dir,
                 KIMI_WORK_DIR_LS=ls_output,
-                KIMI_AGENTS_MD=agents_md,
+                KIMI_AGENTS_MD=agents_md or "",
             ),
             denwa_renji=DenwaRenji(),
             approval=Approval(yolo=yolo),

kimi_cli/ui/print/__init__.py

@@ -3,13 +3,13 @@ import json
 import sys
 from functools import partial
 from pathlib import Path
-from typing import Literal
 
 import aiofiles
 from kosong.base.message import Message
 from kosong.chat_provider import ChatProviderError
 from rich import print
 
+from kimi_cli.cli import InputFormat, OutputFormat
 from kimi_cli.soul import LLMNotSet, MaxStepsReached, RunCancelled, Soul, run_soul
 from kimi_cli.utils.logging import logger
 from kimi_cli.utils.message import message_extract_text
@@ -17,9 +17,6 @@ from kimi_cli.utils.signals import install_sigint_handler
 from kimi_cli.wire import WireUISide
 from kimi_cli.wire.message import StepInterrupted
 
-InputFormat = Literal["text", "stream-json"]
-OutputFormat = Literal["text", "stream-json"]
-
 
 class PrintApp:
     """

kimi_cli/ui/shell/__init__.py

@@ -24,10 +24,16 @@ from kimi_cli.utils.signals import install_sigint_handler
 
 
 class ShellApp:
-    def __init__(self, soul: Soul, welcome_info: list["WelcomeInfoItem"] | None = None):
+    def __init__(
+        self,
+        soul: Soul,
+        welcome_info: list["WelcomeInfoItem"] | None = None,
+        markdown: bool = True,
+    ):
         self.soul = soul
         self._welcome_info = list(welcome_info or [])
         self._background_tasks: set[asyncio.Task[Any]] = set()
+        self._markdown = markdown
 
     async def run(self, command: str | None = None) -> bool:
         if command is not None:
@@ -168,7 +174,10 @@ class ShellApp:
                 self.soul,
                 user_input,
                 lambda wire: visualize(
-                    wire, initial_status=self.soul.status, cancel_event=cancel_event
+                    wire,
+                    initial_status=self.soul.status,
+                    cancel_event=cancel_event,
+                    markdown=self._markdown,
                 ),
                 cancel_event,
             )

kimi_cli/ui/shell/liveview.py

@@ -1,5 +1,6 @@
 import asyncio
 from collections import deque
+from typing import Literal
 
 import streamingjson
 from kosong.base.message import ToolCall, ToolCallPart
@@ -129,6 +130,7 @@ class StepLiveView:
     def __init__(self, status: StatusSnapshot, cancel_event: asyncio.Event | None = None):
         # message content
         self._line_buffer = Text("")
+        self._last_text_mode: Literal["text", "think", ""] = ""
 
         # tool call
         self._tool_calls: dict[str, _ToolCallDisplay] = {}
@@ -187,7 +189,21 @@ class StepLiveView:
         """
         console.print(renderable)
 
-    def append_text(self, text: str):
+    def append_text(self, text: str, mode: Literal["text", "think"] = "text"):
+        if not text:
+            # Ignore empty message
+            return
+        if self._last_text_mode != mode:
+            if self._line_buffer:
+                self._push_out(self._line_buffer)
+                self._push_out("")  # Add extra line between different modes
+                self._line_buffer.plain = ""
+            self._last_text_mode = mode
+            match mode:
+                case "text":
+                    self._line_buffer.style = ""
+                case "think":
+                    self._line_buffer.style = "grey50 italic"
         lines = text.split("\n")
         prev_is_empty = not self._line_buffer
         for line in lines[:-1]:
@@ -313,7 +329,14 @@ class StepLiveViewWithMarkdown(StepLiveView):
         self._buffer_status_active = False
         self._buffer_status_obj: Status | None = None
 
-    def append_text(self, text: str):
+    def append_text(self, text: str, mode: Literal["text", "think"] = "text"):
+        if not text:
+            # Ignore empty message
+            return
+        if self._last_text_mode != mode:
+            if self._flush_markdown():
+                self._push_out("")  # Add extra line between different modes
+            self._last_text_mode = mode
         if not self._pending_markdown_parts:
             self._show_thinking_status()
         self._pending_markdown_parts.append(text)
@@ -334,14 +357,22 @@ class StepLiveViewWithMarkdown(StepLiveView):
         self._flush_markdown()
         return super().__exit__(exc_type, exc_value, traceback)
 
-    def _flush_markdown(self):
+    def _flush_markdown(self) -> bool:
         self._hide_thinking_status()
         if not self._pending_markdown_parts:
-            return
+            return False
         markdown_text = "".join(self._pending_markdown_parts)
         self._pending_markdown_parts.clear()
         if markdown_text.strip():
-            self._push_out(_LeftAlignedMarkdown(markdown_text, justify="left"))
+            self._push_out(
+                _LeftAlignedMarkdown(
+                    markdown_text,
+                    justify="left",
+                    style="grey50 italic" if self._last_text_mode == "think" else "none",
+                )
+            )
+            return True
+        return False
 
     def _show_thinking_status(self):
         if self._buffer_status_active:

kimi_cli/ui/shell/prompt.py

@@ -572,6 +572,7 @@ class CustomPromptSession:
     async def prompt(self) -> UserInput:
         with patch_stdout():
             command = str(await self._session.prompt_async()).strip()
+            command = command.replace("\x00", "")  # just in case null bytes are somehow inserted
         self._append_history_entry(command)
 
         # Parse rich content parts

kimi_cli/ui/shell/setup.py

@@ -26,7 +26,7 @@ class _Platform(NamedTuple):
 _PLATFORMS = [
     _Platform(
         id="kimi-for-coding",
-        name="Kimi For Coding (CN)",
+        name="Kimi For Coding",
         base_url="https://api.kimi.com/coding/v1",
         search_url="https://api.kimi.com/coding/v1/search",
     ),

kimi_cli/ui/shell/visualize.py

@@ -1,7 +1,7 @@
 import asyncio
 from contextlib import asynccontextmanager, suppress
 
-from kosong.base.message import ContentPart, TextPart, ToolCall, ToolCallPart
+from kosong.base.message import ContentPart, TextPart, ThinkPart, ToolCall, ToolCallPart
 from kosong.tooling import ToolResult
 
 from kimi_cli.soul import StatusSnapshot
@@ -42,6 +42,7 @@ async def visualize(
     *,
     initial_status: StatusSnapshot,
     cancel_event: asyncio.Event | None = None,
+    markdown: bool = True,
 ):
     """
     A loop to consume agent events and visualize the agent behavior.
@@ -60,7 +61,8 @@ async def visualize(
     while True:
         # TODO: Maybe we can always have a StepLiveView here.
         #       No need to recreate for each step.
-        with StepLiveViewWithMarkdown(latest_status, cancel_event) as step:
+        LiveView = StepLiveViewWithMarkdown if markdown else StepLiveView
+        with LiveView(latest_status, cancel_event) as step:
             async with _keyboard_listener(step):
                 # spin the moon at the beginning of each step
                 with console.status("", spinner="moon"):
@@ -78,7 +80,9 @@ async def visualize(
                 while True:
                     match msg:
                         case TextPart(text=text):
-                            step.append_text(text)
+                            step.append_text(text, mode="text")
+                        case ThinkPart(think=think):
+                            step.append_text(think, mode="think")
                         case ContentPart():
                             # TODO: support more content parts
                             step.append_text(f"[{msg.__class__.__name__}]")

{kimi_cli-0.43.dist-info → kimi_cli-0.45.dist-info}/METADATA

@@ -1,12 +1,12 @@
 Metadata-Version: 2.3
 Name: kimi-cli
-Version: 0.43
+Version: 0.45
 Summary: Kimi CLI is your next CLI agent.
 Requires-Dist: agent-client-protocol==0.6.2
 Requires-Dist: aiofiles==25.1.0
 Requires-Dist: aiohttp==3.13.2
 Requires-Dist: click==8.3.0
-Requires-Dist: kosong==0.16.1
+Requires-Dist: kosong==0.16.2
 Requires-Dist: loguru==0.7.3
 Requires-Dist: patch-ng==1.19.0
 Requires-Dist: prompt-toolkit==3.0.52

{kimi_cli-0.43.dist-info → kimi_cli-0.45.dist-info}/RECORD

@@ -1,14 +1,15 @@
-kimi_cli/CHANGELOG.md,sha256=079d43c65c761148f411503690acb8ebf8b2e87b7a430778bd010752b22c5a05,8208
-kimi_cli/__init__.py,sha256=fcd01c57f5222328008dd982640f931a84eda4a2afea357aab79aad9a893aa61,6876
+kimi_cli/CHANGELOG.md,sha256=0affdbd19f99b6dedb4cc2cac63eb2f9f942d8cfab44f01b445374764b572a10,8603
+kimi_cli/__init__.py,sha256=e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855,0
 kimi_cli/agents/default/agent.yaml,sha256=6e5c51987ef5cfc0c4c4e34cc20b6fc975953ee219623fccae81a19155aab7b3,709
 kimi_cli/agents/default/sub.yaml,sha256=e0c1ea34fdb04b0d6dc635709f0f130aff25d7f9fb97e238470143c8145be251,634
 kimi_cli/agents/default/system.md,sha256=1d8fd4956b2442215396b5e9651771c9da8f9505ccbd3b6d5e91b1ac4ff35418,5001
 kimi_cli/agentspec.py,sha256=1148b5184ca610b2fb261ce365a63eb2fc9d09497330fe0ea4b2567fc98d5657,4307
-kimi_cli/cli.py,sha256=5db63b9299b7ede52c7d758e627579d9082e4466f34fbe5e8bcd0910e1e98e67,6658
-kimi_cli/config.py,sha256=1bdd90554d33c45848d9ed9499a7914aecdf51e9166abe1b211d95aaa05a6382,4992
+kimi_cli/app.py,sha256=34fde6911d3b4fa6ada013f5ae714cdf354be479da9adb26c6460de0a9ba10c3,7053
+kimi_cli/cli.py,sha256=5731276d9c13e58962c39cdf9db3a57bd875a7a09d560c64498454df62b717a0,6910
+kimi_cli/config.py,sha256=5524c2dd7d3d10e7965901253c754185a4e4b1d30937598ff2b321a070c760be,5012
 kimi_cli/constant.py,sha256=78e25b9304cca7b6f70bb08bf0e1fee4b066297a05386e56dd6935ba42027cd9,110
 kimi_cli/exception.py,sha256=a3fec07566da7d2d34be8cc454fb825f34109bbde3cddf69e1ece6ab21b4b219,259
-kimi_cli/llm.py,sha256=0f62baadd7b6e685372f3d781fed04a2f7dcc5a96c2695e0c259eb071ab65165,3589
+kimi_cli/llm.py,sha256=fdb4f944d6ccec9fe5b8bc1f80bb283df629663659fcdbffe405af2f61f73733,4442
 kimi_cli/metadata.py,sha256=9e9d4bc12ff26fc34e0e09d9068be989f2ff3c8b682ef453de69e442f8f672a1,1557
 kimi_cli/prompts/__init__.py,sha256=6dc5ed2d841f145c09550075f30853cdc51f00d2f5d9aa1097f8edea770536e7,174
 kimi_cli/prompts/compact.md,sha256=6655bd7d8270b24d8f97b51ef7c471cf71d686c56f8ec9a5cc9e47caa3aae87c,1877
@@ -17,14 +18,14 @@ kimi_cli/py.typed,sha256=e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991
 kimi_cli/session.py,sha256=c0623b0cec8ce311f1ddc7bb47e1e452499e13f816b7c3023342499a436dcfc5,2853
 kimi_cli/share.py,sha256=4292df7f44177419c45c772a933b5f36e2b7533f8cef9842629a438bc7856dc0,204
 kimi_cli/soul/__init__.py,sha256=9888a937570bf8bc6e49087a0694f121957210eaa14a7efb4042eb513e136caa,5499
-kimi_cli/soul/agent.py,sha256=637aa6c8c52fd334fb20f15db6f54ef47dc1731eec14efd54d42c29f53bbe13b,4833
+kimi_cli/soul/agent.py,sha256=776e92f46cff2ac5ec0acd06acb445ddabbee05fcc358f9cde54d439e505d300,4843
 kimi_cli/soul/approval.py,sha256=48cd230dff81dfd70bd85f1ad2b99604d5569cf617a4c79c444f9772bbc89ce6,2552
 kimi_cli/soul/compaction.py,sha256=dab17979060fceeed4a7a344373833022dc7abac04282364f2a1b20e6edd4581,3558
 kimi_cli/soul/context.py,sha256=541759a65f8f87a3424a6da160ffb2043046e6f6b714124d94d82a77635df9bc,5855
 kimi_cli/soul/denwarenji.py,sha256=66b95f052a1fa844e2347972d34e1916a7be24d3e493701b451f5380b0375c9f,1384
 kimi_cli/soul/kimisoul.py,sha256=68c65fe0f97700832a42018ae00892d38dc79f14759265042f47b707dbc0c749,11822
 kimi_cli/soul/message.py,sha256=7a52a6d4d63ef1a3621d93d5ff86887baa7e67019bf2e9a08c374fc130b8d152,2541
-kimi_cli/soul/runtime.py,sha256=5104f501e39f082115ff1a09da797f0dc9ae7763b76719fc4af73420e4a98634,2629
+kimi_cli/soul/runtime.py,sha256=9421a3ce6882587a95ecdf77b3d75f3b7ecab55cf68dc57e3e563b93e5a02e46,2690
 kimi_cli/soul/toolset.py,sha256=60166d89ef0efac690fa6866e88afe70fbe80ad862ba2524d70ddf657a730d14,744
 kimi_cli/tools/__init__.py,sha256=4d612402814eede7182e0a55e7dd21c4532b5dd44700dc744763a8308c2f74f8,3280
 kimi_cli/tools/bash/__init__.py,sha256=de21b19c714bda53f6c89e3348c4c82fb4278040130fed1d261b3ab203054e8c,3028
@@ -60,18 +61,18 @@ kimi_cli/tools/web/search.md,sha256=24049f9e90d37083e0fc78b8b2e3a5f6fadf09bea00f
 kimi_cli/tools/web/search.py,sha256=85de343b20bc9e58de8a09aba7c3aac619d6fc6d30a6a5b565108faeb4500faf,4517
 kimi_cli/ui/__init__.py,sha256=e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855,0
 kimi_cli/ui/acp/__init__.py,sha256=a35e17273e943168882865f90d922180c5a1f01de0d128c899ffcfe55a9db3c3,17120
-kimi_cli/ui/print/__init__.py,sha256=cea17f8db2b52eee6ccb632636f3e9588ae0193a7a2f723b42c515ed0a2d28aa,5614
-kimi_cli/ui/shell/__init__.py,sha256=aa30b512004be5ad9c8b8ab420bbdb574532dc90968067682dab58ff19e8b660,11289
+kimi_cli/ui/print/__init__.py,sha256=ca402bec701a253acd6de5b57d59635ac0b05d4013cebc26877c5aa4aa2c27c7,5546
+kimi_cli/ui/shell/__init__.py,sha256=c9644dbc27f899baea38aeee1b4b5ae3b82f84d0321f939477bd03f2ad7532b9,11471
 kimi_cli/ui/shell/console.py,sha256=bcbf7efd214cba3d2259f2a2c1842250cde96d49e4f9f1e0b60273cf1c366be3,842
 kimi_cli/ui/shell/debug.py,sha256=cd4e7259c83f099b5c6519713be5306580f30d3fa4944e07916d4468e960c9c7,5562
 kimi_cli/ui/shell/keyboard.py,sha256=23e5fbc4b6acda4c0f3b5297a0ae6eb09a90f4b5b37b2e95b7ce86a2da0d5dca,5160
-kimi_cli/ui/shell/liveview.py,sha256=f4e6ac37c446740b5c55cf37d5ebd327b5d41334d41d5e54ad8ad15445c1a492,14239
+kimi_cli/ui/shell/liveview.py,sha256=2f780323dfc7d9b35491a17ccc2b4f6320176ae00c596b426ea7d7f513d3e42c,15478
 kimi_cli/ui/shell/metacmd.py,sha256=a0e52e9cbd8758c1ba13f025599341aa59dd5bc5e244840da2ff9bb71f952a20,7678
-kimi_cli/ui/shell/prompt.py,sha256=f33a27646ddafdd1afb6f16a63a6c164f39019893e96764243d9e58650ff8d9b,25710
+kimi_cli/ui/shell/prompt.py,sha256=d18003de33da0c9711bf54496cb4d5f6f5aa6b2179b1618e4475b1f020208f6c,25808
 kimi_cli/ui/shell/replay.py,sha256=e54f58acebc46ad944e1a2cdf54d81559262d2cf8baf5da391ed903926f1ccf1,3767
-kimi_cli/ui/shell/setup.py,sha256=36be348c8cdbb908f1b896f7dfd93802df4d76eeb8572d39b81a34dfd5ee2a3c,5374
+kimi_cli/ui/shell/setup.py,sha256=8fbf2935fc5b972d2c3946e8dc9f4a7e9d2953810b57c0fb6f22172abf3e6fb5,5369
 kimi_cli/ui/shell/update.py,sha256=56dcb0bd1da82b98c22bfdddca717a2805bd8ac3e93bf23fb3b508549c41fae8,7340
-kimi_cli/ui/shell/visualize.py,sha256=1abaa53cf78836f71b1f83085b6bfe3f86624951fa3297d5086a06bafa97f960,3884
+kimi_cli/ui/shell/visualize.py,sha256=1ca7a1e766f96f108318640d48552111d42b433e4ddb551af869d331c074d950,4112
 kimi_cli/utils/aiohttp.py,sha256=f8f61e3beaf6439e949c33c3a10db3035bf88136e882b09c858ea92a4c888e00,245
 kimi_cli/utils/changelog.py,sha256=bfcf5a5a360b13648bb7a6abc83e427270caa502646b5acc950d62148510641c,3402
 kimi_cli/utils/logging.py,sha256=129298ac214ecd8d913c3431cc05d754f9c4c8c4042c458618bf9e8ddebdb763,399
@@ -82,7 +83,7 @@ kimi_cli/utils/signals.py,sha256=20e0d158a1043189d44815fe3624cd0bfe41e99620a18ac
 kimi_cli/utils/string.py,sha256=0d437d3633199df1051813af8b49a2f808c6525547310cc5c3d427710d2eae06,593
 kimi_cli/wire/__init__.py,sha256=9f1d7eb58f76885edaf76f769371c363ec801b46cada03883eeb3536fa2677f7,1896
 kimi_cli/wire/message.py,sha256=72222d3f3d7228a323dbba7b1084f35018104c58e4bb2aa51d0827984791841d,2398
-kimi_cli-0.43.dist-info/WHEEL,sha256=70ab3c2925fe316809860cb034f99ba13c4b49819b339959274aab755cc084a8,78
-kimi_cli-0.43.dist-info/entry_points.txt,sha256=97e051756296e9db3167f6dce61d6c88e58d170314a2d63d18c84c73a5c1333b,44
-kimi_cli-0.43.dist-info/METADATA,sha256=7319672d68a983b488b93d9b4e21bde14f5b4c23069bff3316df481f2856f61f,5217
-kimi_cli-0.43.dist-info/RECORD,,
+kimi_cli-0.45.dist-info/WHEEL,sha256=70ab3c2925fe316809860cb034f99ba13c4b49819b339959274aab755cc084a8,78
+kimi_cli-0.45.dist-info/entry_points.txt,sha256=97e051756296e9db3167f6dce61d6c88e58d170314a2d63d18c84c73a5c1333b,44
+kimi_cli-0.45.dist-info/METADATA,sha256=bac942d1519d1a5f2650199cb6add467e99274392d3e9b839707915a1acd58d0,5217
+kimi_cli-0.45.dist-info/RECORD,,