kimi-cli 0.52__tar.gz → 0.56__tar.gz

{kimi_cli-0.52 → kimi_cli-0.56}/PKG-INFO

@@ -1,12 +1,12 @@
 Metadata-Version: 2.3
 Name: kimi-cli
-Version: 0.52
+Version: 0.56
 Summary: Kimi CLI is your next CLI agent.
 Requires-Dist: agent-client-protocol==0.6.3
 Requires-Dist: aiofiles==25.1.0
 Requires-Dist: aiohttp==3.13.2
 Requires-Dist: typer==0.20.0
-Requires-Dist: kosong==0.22.0
+Requires-Dist: kosong==0.26.1
 Requires-Dist: loguru==0.7.3
 Requires-Dist: patch-ng==1.19.0
 Requires-Dist: prompt-toolkit==3.0.52

{kimi_cli-0.52 → kimi_cli-0.56}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "kimi-cli"
-version = "0.52"
+version = "0.56"
 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",
     "typer==0.20.0",
-    "kosong==0.22.0",
+    "kosong==0.26.1",
     "loguru==0.7.3",
     "patch-ng==1.19.0",
     "prompt-toolkit==3.0.52",
@@ -30,8 +30,8 @@ dev = [
     "inline-snapshot[black]>=0.31.1",
     "pyinstaller>=6.16.0",
     "pyright>=1.1.407",
-    "pytest>=8.4.2",
-    "pytest-asyncio>=1.2.0",
+    "pytest>=9.0.1",
+    "pytest-asyncio>=1.3.0",
     "ruff>=0.14.4",
 ]
 
@@ -40,7 +40,7 @@ requires = ["uv_build>=0.8.5,<0.9.0"]
 build-backend = "uv_build"
 
 [tool.uv.build-backend]
-module-name = ["kimi_cli"]
+module-name = "kimi_cli"
 source-exclude = ["tests/**/*", "src/kimi_cli/deps/**/*"]
 
 [project.scripts]

{kimi_cli-0.52 → kimi_cli-0.56}/src/kimi_cli/CHANGELOG.md

@@ -9,6 +9,39 @@ Internal builds may append content to the Unreleased section.
 Only write entries that are worth mentioning to users.
 -->
 
+## [0.56] - 2025-11-19
+
+- LLM: Add support for Google GenAI provider
+
+## [0.55] - 2025-11-18
+
+- Lib: Add `kimi_cli.app.enable_logging` function to enable logging when directly using `KimiCLI` class
+- Core: Fix relative path resolution in agent spec files
+- Core: Prevent from panic when LLM API connection failed
+- Tool: Optimize `FetchURL` tool for better content extraction
+- Tool: Increase MCP tool call timeout to 60 seconds
+- Tool: Provide better error message in `Glob` tool when pattern is `**`
+- ACP: Fix thinking content not displayed properly
+- UI: Minor UI improvements in shell mode
+
+## [0.54] - 2025-11-13
+
+- Lib: Move `WireMessage` from `kimi_cli.wire.message` to `kimi_cli.wire`
+- Print: Fix `stream-json` output format missing the last assistant message
+- UI: Add warning when API key is overridden by `KIMI_API_KEY` environment variable
+- UI: Make a bell sound when there's an approval request
+- Core: Fix context compaction and clearing on Windows
+
+## [0.53] - 2025-11-12
+
+- UI: Remove unnecessary trailing spaces in console output
+- Core: Throw error when there are unsupported message parts
+- MetaCmd: Add `/yolo` meta command to enable YOLO mode after startup
+- Tool: Add approval request for MCP tools
+- Tool: Disable `Think` tool in default agent
+- CLI: Restore thinking mode from last time when `--thinking` is not specified
+- CLI: Fix `/reload` not working in binary packed by PyInstaller
+
 ## [0.52] - 2025-11-10
 
 - CLI: Remove `--ui` option in favor of `--print`, `--acp`, and `--wire` flags (shell is still the default)

{kimi_cli-0.52 → kimi_cli-0.56}/src/kimi_cli/agents/default/agent.yaml

@@ -7,7 +7,7 @@ agent:
   tools:
     - "kimi_cli.tools.task:Task"
     # - "kimi_cli.tools.dmail:SendDMail"
-    - "kimi_cli.tools.think:Think"
+    # - "kimi_cli.tools.think:Think"
     - "kimi_cli.tools.todo:SetTodoList"
     - "kimi_cli.tools.bash:Bash"
     - "kimi_cli.tools.file:ReadFile"

{kimi_cli-0.52 → kimi_cli-0.56}/src/kimi_cli/agentspec.py

@@ -1,5 +1,8 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
 from pathlib import Path
-from typing import Any, NamedTuple
+from typing import Any
 
 import yaml
 from pydantic import BaseModel, Field
@@ -27,7 +30,7 @@ class AgentSpec(BaseModel):
     )
     tools: list[str] | None = Field(default=None, description="Tools")  # required
     exclude_tools: list[str] | None = Field(default=None, description="Tools to exclude")
-    subagents: dict[str, "SubagentSpec"] | None = Field(default=None, description="Subagents")
+    subagents: dict[str, SubagentSpec] | None = Field(default=None, description="Subagents")
 
 
 class SubagentSpec(BaseModel):
@@ -37,7 +40,8 @@ class SubagentSpec(BaseModel):
     description: str = Field(description="Subagent description")
 
 
-class ResolvedAgentSpec(NamedTuple):
+@dataclass(frozen=True, slots=True, kw_only=True)
+class ResolvedAgentSpec:
     """Resolved agent specification."""
 
     name: str
@@ -45,7 +49,7 @@ class ResolvedAgentSpec(NamedTuple):
     system_prompt_args: dict[str, str]
     tools: list[str]
     exclude_tools: list[str]
-    subagents: dict[str, "SubagentSpec"]
+    subagents: dict[str, SubagentSpec]
 
 
 def load_agent_spec(agent_file: Path) -> ResolvedAgentSpec:
@@ -75,7 +79,10 @@ def load_agent_spec(agent_file: Path) -> ResolvedAgentSpec:
 
 
 def _load_agent_spec(agent_file: Path) -> AgentSpec:
-    assert agent_file.is_file(), "expect agent file to exist"
+    if not agent_file.exists():
+        raise AgentSpecError(f"Agent spec file not found: {agent_file}")
+    if not agent_file.is_file():
+        raise AgentSpecError(f"Agent spec path is not a file: {agent_file}")
     try:
         with open(agent_file, encoding="utf-8") as f:
             data: dict[str, Any] = yaml.safe_load(f)
@@ -88,15 +95,17 @@ def _load_agent_spec(agent_file: Path) -> AgentSpec:
 
     agent_spec = AgentSpec(**data.get("agent", {}))
     if agent_spec.system_prompt_path is not None:
-        agent_spec.system_prompt_path = agent_file.parent / agent_spec.system_prompt_path
+        agent_spec.system_prompt_path = (
+            agent_file.parent / agent_spec.system_prompt_path
+        ).absolute()
     if agent_spec.subagents is not None:
         for v in agent_spec.subagents.values():
-            v.path = agent_file.parent / v.path
+            v.path = (agent_file.parent / v.path).absolute()
     if agent_spec.extend:
         if agent_spec.extend == "default":
             base_agent_file = DEFAULT_AGENT_FILE
         else:
-            base_agent_file = agent_file.parent / agent_spec.extend
+            base_agent_file = (agent_file.parent / agent_spec.extend).absolute()
         base_agent_spec = _load_agent_spec(base_agent_file)
         if agent_spec.name is not None:
             base_agent_spec.name = agent_spec.name

{kimi_cli-0.52 → kimi_cli-0.56}/src/kimi_cli/app.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import contextlib
 import os
 import warnings
@@ -12,12 +14,26 @@ 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.share import get_share_dir
 from kimi_cli.soul import LLMNotSet, LLMNotSupported
 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
+from kimi_cli.utils.path import shorten_home
+
+
+def enable_logging(debug: bool = False) -> None:
+    if debug:
+        logger.enable("kosong")
+    logger.add(
+        get_share_dir() / "logs" / "kimi.log",
+        # FIXME: configure level for different modules
+        level="TRACE" if debug else "INFO",
+        rotation="06:00",
+        retention="10 days",
+    )
 
 
 class KimiCLI:
@@ -26,20 +42,18 @@ class KimiCLI:
         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,
         thinking: bool = False,
         agent_file: Path | None = None,
-    ) -> "KimiCLI":
+    ) -> 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.
@@ -79,7 +93,7 @@ class KimiCLI:
         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)
+            llm = create_llm(provider, model, session_id=session.id)
 
         runtime = await Runtime.create(config, llm, session, yolo)
 
@@ -137,7 +151,9 @@ class KimiCLI:
         from kimi_cli.ui.shell import ShellApp, WelcomeInfoItem
 
         welcome_info = [
-            WelcomeInfoItem(name="Directory", value=str(self._runtime.session.work_dir)),
+            WelcomeInfoItem(
+                name="Directory", value=str(shorten_home(self._runtime.session.work_dir))
+            ),
             WelcomeInfoItem(name="Session", value=self._runtime.session.id),
         ]
         if base_url := self._env_overrides.get("KIMI_BASE_URL"):
@@ -148,6 +164,14 @@ class KimiCLI:
                     level=WelcomeInfoItem.Level.WARN,
                 )
             )
+        if self._env_overrides.get("KIMI_API_KEY"):
+            welcome_info.append(
+                WelcomeInfoItem(
+                    name="API Key",
+                    value="****** (from KIMI_API_KEY)",
+                    level=WelcomeInfoItem.Level.WARN,
+                )
+            )
         if not self._runtime.llm:
             welcome_info.append(
                 WelcomeInfoItem(

{kimi_cli-0.52 → kimi_cli-0.56}/src/kimi_cli/cli.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import asyncio
 import json
 import sys
@@ -185,19 +187,18 @@ def kimi(
         ),
     ] = False,
     thinking: Annotated[
-        bool,
+        bool | None,
         typer.Option(
             "--thinking",
-            help="Enable thinking mode if supported. Default: no.",
+            help="Enable thinking mode if supported. Default: same as last time.",
         ),
-    ] = False,
+    ] = None,
 ):
     """Kimi, your next CLI agent."""
     del version  # handled in the callback
 
-    from kimi_cli.app import KimiCLI
+    from kimi_cli.app import KimiCLI, enable_logging
     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):
@@ -224,16 +225,7 @@ def kimi(
         ui = "wire"
 
     echo: Callable[..., None] = typer.echo if verbose else _noop_echo
-
-    if debug:
-        logger.enable("kosong")
-    logger.add(
-        get_share_dir() / "logs" / "kimi.log",
-        # FIXME: configure level for different modules
-        level="TRACE" if debug else "INFO",
-        rotation="06:00",
-        retention="10 days",
-    )
+    enable_logging(debug)
 
     work_dir = (work_dir or Path.cwd()).absolute()
     if continue_:
@@ -279,20 +271,27 @@ def kimi(
         raise typer.BadParameter(f"Invalid JSON: {e}", param_hint="--mcp-config") from e
 
     async def _run() -> bool:
+        from kimi_cli.metadata import WorkDirMeta, load_metadata, save_metadata
+
+        if thinking is None:
+            metadata = load_metadata()
+            thinking_mode = metadata.thinking
+        else:
+            thinking_mode = thinking
+
         instance = await KimiCLI.create(
             session,
             yolo=yolo or (ui == "print"),  # print mode implies yolo
-            stream=ui != "print",  # use non-streaming mode only for print UI
             mcp_configs=mcp_configs,
             model_name=model_name,
-            thinking=thinking,
+            thinking=thinking_mode,
             agent_file=agent_file,
         )
         match ui:
             case "shell":
-                return await instance.run_shell_mode(command)
+                succeeded = await instance.run_shell_mode(command)
             case "print":
-                return await instance.run_print_mode(
+                succeeded = await instance.run_print_mode(
                     input_format or "text",
                     output_format or "text",
                     command,
@@ -300,17 +299,41 @@ def kimi(
             case "acp":
                 if command is not None:
                     logger.warning("ACP server ignores command argument")
-                return await instance.run_acp_server()
+                succeeded = await instance.run_acp_server()
             case "wire":
                 if command is not None:
                     logger.warning("Wire server ignores command argument")
-                return await instance.run_wire_server()
+                succeeded = await instance.run_wire_server()
+
+        if succeeded:
+            metadata = load_metadata()
+
+            # Update work_dir metadata with last session
+            work_dir_meta = next(
+                (wd for wd in metadata.work_dirs if wd.path == str(session.work_dir)), None
+            )
+
+            if work_dir_meta is None:
+                logger.warning(
+                    "Work dir metadata missing when marking last session, recreating: {work_dir}",
+                    work_dir=session.work_dir,
+                )
+                work_dir_meta = WorkDirMeta(path=str(session.work_dir))
+                metadata.work_dirs.append(work_dir_meta)
+
+            work_dir_meta.last_session_id = session.id
+
+            # Update thinking mode
+            metadata.thinking = instance.soul.thinking
+
+            save_metadata(metadata)
+
+        return succeeded
 
     while True:
         try:
             succeeded = asyncio.run(_run())
             if succeeded:
-                session.mark_as_last()
                 break
             sys.exit(1)
         except Reload:
@@ -318,4 +341,7 @@ def kimi(
 
 
 if __name__ == "__main__":
-    cli()
+    if "kimi_cli.cli" not in sys.modules:
+        sys.modules["kimi_cli.cli"] = sys.modules[__name__]
+
+    sys.exit(cli())

{kimi_cli-0.52 → kimi_cli-0.56}/src/kimi_cli/config.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import json
 from pathlib import Path
 from typing import Self

{kimi_cli-0.52 → kimi_cli-0.56}/src/kimi_cli/constant.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import importlib.metadata
 
 VERSION = importlib.metadata.version("kimi-cli")

{kimi_cli-0.52 → kimi_cli-0.56}/src/kimi_cli/exception.py

@@ -1,3 +1,6 @@
+from __future__ import annotations
+
+
 class KimiCLIException(Exception):
     """Base exception class for Kimi CLI."""
 

{kimi_cli-0.52 → kimi_cli-0.56}/src/kimi_cli/llm.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import os
 from dataclasses import dataclass
 from typing import TYPE_CHECKING, Literal, cast, get_args
@@ -10,7 +12,14 @@ from kimi_cli.constant import USER_AGENT
 if TYPE_CHECKING:
     from kimi_cli.config import LLMModel, LLMProvider
 
-type ProviderType = Literal["kimi", "openai_legacy", "openai_responses", "anthropic", "_chaos"]
+type ProviderType = Literal[
+    "kimi",
+    "openai_legacy",
+    "openai_responses",
+    "anthropic",
+    "google_genai",
+    "_chaos",
+]
 
 type ModelCapability = Literal["image_in", "thinking"]
 ALL_MODEL_CAPABILITIES: set[ModelCapability] = set(get_args(ModelCapability))
@@ -27,7 +36,7 @@ class LLM:
         return self.chat_provider.model_name
 
 
-def augment_provider_with_env_vars(provider: "LLMProvider", model: "LLMModel") -> dict[str, str]:
+def augment_provider_with_env_vars(provider: LLMProvider, model: LLMModel) -> dict[str, str]:
     """Override provider/model settings from environment variables.
 
     Returns:
@@ -69,10 +78,9 @@ def augment_provider_with_env_vars(provider: "LLMProvider", model: "LLMModel") -
 
 
 def create_llm(
-    provider: "LLMProvider",
-    model: "LLMModel",
+    provider: LLMProvider,
+    model: LLMModel,
     *,
-    stream: bool = True,
     session_id: str | None = None,
 ) -> LLM:
     match provider.type:
@@ -83,7 +91,6 @@ def create_llm(
                 model=model.model,
                 base_url=provider.base_url,
                 api_key=provider.api_key.get_secret_value(),
-                stream=stream,
                 default_headers={
                     "User-Agent": USER_AGENT,
                     **(provider.custom_headers or {}),
@@ -98,7 +105,6 @@ def create_llm(
                 model=model.model,
                 base_url=provider.base_url,
                 api_key=provider.api_key.get_secret_value(),
-                stream=stream,
             )
         case "openai_responses":
             from kosong.contrib.chat_provider.openai_responses import OpenAIResponses
@@ -107,7 +113,6 @@ def create_llm(
                 model=model.model,
                 base_url=provider.base_url,
                 api_key=provider.api_key.get_secret_value(),
-                stream=stream,
             )
         case "anthropic":
             from kosong.contrib.chat_provider.anthropic import Anthropic
@@ -116,9 +121,16 @@ def create_llm(
                 model=model.model,
                 base_url=provider.base_url,
                 api_key=provider.api_key.get_secret_value(),
-                stream=stream,
                 default_max_tokens=50000,
             )
+        case "google_genai":
+            from kosong.contrib.chat_provider.google_genai import GoogleGenAI
+
+            chat_provider = GoogleGenAI(
+                model=model.model,
+                base_url=provider.base_url,
+                api_key=provider.api_key.get_secret_value(),
+            )
         case "_chaos":
             from kosong.chat_provider.chaos import ChaosChatProvider, ChaosConfig
 
@@ -139,7 +151,7 @@ def create_llm(
     )
 
 
-def _derive_capabilities(provider: "LLMProvider", model: "LLMModel") -> set[ModelCapability]:
+def _derive_capabilities(provider: LLMProvider, model: LLMModel) -> set[ModelCapability]:
     capabilities = model.capabilities or set()
     if provider.type != "kimi":
         return capabilities

{kimi_cli-0.52 → kimi_cli-0.56}/src/kimi_cli/metadata.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import json
 from hashlib import md5
 from pathlib import Path
@@ -31,9 +33,11 @@ class WorkDirMeta(BaseModel):
 class Metadata(BaseModel):
     """Kimi metadata structure."""
 
-    work_dirs: list[WorkDirMeta] = Field(
-        default_factory=list[WorkDirMeta], description="Work directory list"
-    )
+    work_dirs: list[WorkDirMeta] = Field(default_factory=list[WorkDirMeta])
+    """Work directory list."""
+
+    thinking: bool = False
+    """Whether the last session was in thinking mode."""
 
 
 def load_metadata() -> Metadata:

{kimi_cli-0.52 → kimi_cli-0.56}/src/kimi_cli/prompts/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from pathlib import Path
 
 INIT = (Path(__file__).parent / "init.md").read_text(encoding="utf-8")

{kimi_cli-0.52 → kimi_cli-0.56}/src/kimi_cli/session.py

@@ -1,12 +1,15 @@
+from __future__ import annotations
+
 import uuid
+from dataclasses import dataclass
 from pathlib import Path
-from typing import NamedTuple
 
 from kimi_cli.metadata import WorkDirMeta, load_metadata, save_metadata
 from kimi_cli.utils.logging import logger
 
 
-class Session(NamedTuple):
+@dataclass(frozen=True, slots=True, kw_only=True)
+class Session:
     """A session of a work directory."""
 
     id: str
@@ -14,7 +17,7 @@ class Session(NamedTuple):
     history_file: Path
 
     @staticmethod
-    def create(work_dir: Path, _history_file: Path | None = None) -> "Session":
+    def create(work_dir: Path, _history_file: Path | None = None) -> Session:
         """Create a new session for a work directory."""
         logger.debug("Creating new session for work directory: {work_dir}", work_dir=work_dir)
 
@@ -53,7 +56,7 @@ class Session(NamedTuple):
         )
 
     @staticmethod
-    def continue_(work_dir: Path) -> "Session | None":
+    def continue_(work_dir: Path) -> Session | None:
         """Get the last session for a work directory."""
         logger.debug("Continuing session for work directory: {work_dir}", work_dir=work_dir)
 
@@ -78,26 +81,3 @@ class Session(NamedTuple):
             work_dir=work_dir,
             history_file=history_file,
         )
-
-    def mark_as_last(self) -> None:
-        """Mark this session as the last completed session for its work directory."""
-        metadata = load_metadata()
-        work_dir_meta = next(
-            (wd for wd in metadata.work_dirs if wd.path == str(self.work_dir)), None
-        )
-
-        if work_dir_meta is None:
-            logger.warning(
-                "Work directory metadata missing when marking last session, recreating: {work_dir}",
-                work_dir=self.work_dir,
-            )
-            work_dir_meta = WorkDirMeta(path=str(self.work_dir))
-            metadata.work_dirs.append(work_dir_meta)
-
-        work_dir_meta.last_session_id = self.id
-        logger.debug(
-            "Updated last session for work directory: {work_dir} -> {session_id}",
-            work_dir=self.work_dir,
-            session_id=self.id,
-        )
-        save_metadata(metadata)

{kimi_cli-0.52 → kimi_cli-0.56}/src/kimi_cli/share.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from pathlib import Path
 
 

{kimi_cli-0.52 → kimi_cli-0.56}/src/kimi_cli/soul/__init__.py

@@ -1,14 +1,16 @@
+from __future__ import annotations
+
 import asyncio
 import contextlib
 from collections.abc import Callable, Coroutine
 from contextvars import ContextVar
-from typing import TYPE_CHECKING, Any, NamedTuple, Protocol, runtime_checkable
+from dataclasses import dataclass
+from typing import TYPE_CHECKING, Any, Protocol, runtime_checkable
 
 from kosong.message import ContentPart
 
 from kimi_cli.utils.logging import logger
-from kimi_cli.wire import Wire, WireUISide
-from kimi_cli.wire.message import WireMessage
+from kimi_cli.wire import Wire, WireMessage, WireUISide
 
 if TYPE_CHECKING:
     from kimi_cli.llm import LLM, ModelCapability
@@ -23,7 +25,7 @@ class LLMNotSet(Exception):
 class LLMNotSupported(Exception):
     """Raised when the LLM does not have required capabilities."""
 
-    def __init__(self, llm: "LLM", capabilities: "list[ModelCapability]"):
+    def __init__(self, llm: LLM, capabilities: list[ModelCapability]):
         self.llm = llm
         self.capabilities = capabilities
         capabilities_str = "capability" if len(capabilities) == 1 else "capabilities"
@@ -43,7 +45,8 @@ class MaxStepsReached(Exception):
         self.n_steps = n_steps
 
 
-class StatusSnapshot(NamedTuple):
+@dataclass(frozen=True, slots=True)
+class StatusSnapshot:
     context_usage: float
     """The usage of the context, in percentage."""
 
@@ -61,7 +64,7 @@ class Soul(Protocol):
         ...
 
     @property
-    def model_capabilities(self) -> "set[ModelCapability] | None":
+    def model_capabilities(self) -> set[ModelCapability] | None:
         """The capabilities of the LLM model used by the soul. None indicates no LLM configured."""
         ...
 
@@ -96,7 +99,7 @@ class RunCancelled(Exception):
 
 
 async def run_soul(
-    soul: "Soul",
+    soul: Soul,
     user_input: str | list[ContentPart],
     ui_loop_fn: UILoopFn,
     cancel_event: asyncio.Event,