kimi-cli 0.43__tar.gz → 0.45__tar.gz

{kimi_cli-0.43 → kimi_cli-0.45}/PKG-INFO

@@ -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 → kimi_cli-0.45}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "kimi-cli"
-version = "0.43"
+version = "0.45"
 description = "Kimi CLI is your next CLI agent."
 readme = "README.md"
 requires-python = ">=3.13"
@@ -9,7 +9,7 @@ dependencies = [
     "aiofiles==25.1.0",
     "aiohttp==3.13.2",
     "click==8.3.0",
-    "kosong==0.16.1",
+    "kosong==0.16.2",
     "loguru==0.7.3",
     "patch-ng==1.19.0",
     "prompt-toolkit==3.0.52",

{kimi_cli-0.43 → kimi_cli-0.45}/src/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-0.43/src/kimi_cli/__init__.py → kimi_cli-0.45/src/kimi_cli/app.py

@@ -8,6 +8,7 @@ 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
@@ -15,9 +16,6 @@ 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
 
 
@@ -81,43 +79,6 @@ class KimiCLI:
             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:
@@ -132,17 +93,17 @@ class KimiCLI:
             runtime,
             context=context,
         )
-        return KimiCLI(soul, session, welcome_info)
+        return KimiCLI(soul, runtime, env_overrides)
 
     def __init__(
         self,
-        soul: KimiSoul,
-        session: Session,
-        welcome_info: list[WelcomeInfoItem],
+        _soul: KimiSoul,
+        _runtime: Runtime,
+        _env_overrides: dict[str, str],
     ) -> None:
-        self._soul = soul
-        self._session = session
-        self._welcome_info = welcome_info
+        self._soul = _soul
+        self._runtime = _runtime
+        self._env_overrides = _env_overrides
 
     @property
     def soul(self) -> KimiSoul:
@@ -152,12 +113,12 @@ class KimiCLI:
     @property
     def session(self) -> Session:
         """Get the Session instance."""
-        return self._session
+        return self._runtime.session
 
     @contextlib.contextmanager
     def _app_env(self) -> Generator[None]:
         original_cwd = Path.cwd()
-        os.chdir(self._session.work_dir)
+        os.chdir(self._runtime.session.work_dir)
         try:
             # to ignore possible warnings from dateparser
             warnings.filterwarnings("ignore", category=DeprecationWarning)
@@ -166,9 +127,47 @@ class KimiCLI:
         finally:
             os.chdir(original_cwd)
 
-    async def run_shell_mode(self, command: str | None = None) -> bool:
+    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=self._welcome_info)
+            app = ShellApp(self._soul, welcome_info=welcome_info, markdown=markdown)
             return await app.run(command)
 
     async def run_print_mode(
@@ -177,16 +176,20 @@ class KimiCLI:
         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._session.history_file,
+                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-0.43 → kimi_cli-0.45}/src/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-0.43 → kimi_cli-0.45}/src/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-0.43 → kimi_cli-0.45}/src/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-0.43 → kimi_cli-0.45}/src/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-0.43 → kimi_cli-0.45}/src/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-0.43 → kimi_cli-0.45}/src/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-0.43 → kimi_cli-0.45}/src/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-0.43 → kimi_cli-0.45}/src/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-0.43 → kimi_cli-0.45}/src/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-0.43 → kimi_cli-0.45}/src/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-0.43 → kimi_cli-0.45}/src/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__}]")