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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: kimi-cli
-Version: 0.43
+Version: 0.44
 Summary: Kimi CLI is your next CLI agent.
 Requires-Dist: agent-client-protocol==0.6.2
 Requires-Dist: aiofiles==25.1.0

{kimi_cli-0.43 → kimi_cli-0.44}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "kimi-cli"
-version = "0.43"
+version = "0.44"
 description = "Kimi CLI is your next CLI agent."
 readme = "README.md"
 requires-python = ">=3.13"

{kimi_cli-0.43 → kimi_cli-0.44}/src/kimi_cli/CHANGELOG.md

@@ -9,6 +9,16 @@ Internal builds may append content to the Unreleased section.
 Only write entries that are worth mentioning to users.
 -->
 
+## [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.44/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)
@@ -167,8 +128,46 @@ class KimiCLI:
             os.chdir(original_cwd)
 
     async def run_shell_mode(self, command: str | None = None) -> 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)
             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.44}/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"]))
@@ -156,6 +153,10 @@ def kimi(
     yolo: 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

{kimi_cli-0.43 → kimi_cli-0.44}/src/kimi_cli/llm.py

@@ -2,9 +2,6 @@ import os
 from typing import NamedTuple
 
 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
@@ -68,6 +65,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,6 +80,8 @@ 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,
@@ -88,6 +89,8 @@ def create_llm(
                 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.44}/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.44}/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.44}/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.44}/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.44}/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",
     ),