klaude-code 1.2.6__py3-none-any.whl

klaude_code/config/config.py

@@ -0,0 +1,177 @@
+import asyncio
+from functools import lru_cache
+from pathlib import Path
+from typing import Any, cast
+
+import yaml
+from pydantic import BaseModel, Field, ValidationError, model_validator
+
+from klaude_code.protocol import llm_param
+from klaude_code.protocol.sub_agent import iter_sub_agent_profiles
+from klaude_code.trace import log
+
+config_path = Path.home() / ".klaude" / "klaude-config.yaml"
+
+
+class ModelConfig(BaseModel):
+    model_name: str
+    provider: str
+    model_params: llm_param.LLMConfigModelParameter
+
+
+class Config(BaseModel):
+    provider_list: list[llm_param.LLMConfigProviderParameter]
+    model_list: list[ModelConfig]
+    main_model: str
+    subagent_models: dict[str, str] = Field(default_factory=dict)
+    theme: str | None = None
+
+    @model_validator(mode="before")
+    @classmethod
+    def _normalize_subagent_models(cls, data: dict[str, Any]) -> dict[str, Any]:
+        raw_val: Any = data.get("subagent_models") or {}
+        raw_models: dict[str, Any] = cast(dict[str, Any], raw_val) if isinstance(raw_val, dict) else {}
+        normalized: dict[str, str] = {}
+        key_map = {p.name.lower(): p.name for p in iter_sub_agent_profiles()}
+        for key, value in dict(raw_models).items():
+            canonical = key_map.get(str(key).lower(), str(key))
+            normalized[canonical] = str(value)
+        data["subagent_models"] = normalized
+        return data
+
+    def get_main_model_config(self) -> llm_param.LLMConfigParameter:
+        return self.get_model_config(self.main_model)
+
+    def get_model_config(self, model_name: str) -> llm_param.LLMConfigParameter:
+        model = next(
+            (model for model in self.model_list if model.model_name == model_name),
+            None,
+        )
+        if model is None:
+            raise ValueError(f"Unknown model: {model_name}")
+
+        provider = next(
+            (provider for provider in self.provider_list if provider.provider_name == model.provider),
+            None,
+        )
+        if provider is None:
+            raise ValueError(f"Unknown provider: {model.provider}")
+
+        return llm_param.LLMConfigParameter(
+            **provider.model_dump(),
+            **model.model_params.model_dump(),
+        )
+
+    async def save(self) -> None:
+        """
+        Save config to file.
+        Notice: it won't preserve comments in the config file.
+        """
+        config_dict = self.model_dump(mode="json", exclude_none=True)
+
+        def _save_config() -> None:
+            config_path.parent.mkdir(parents=True, exist_ok=True)
+            _ = config_path.write_text(yaml.dump(config_dict, default_flow_style=False, sort_keys=False))
+
+        await asyncio.to_thread(_save_config)
+
+
+def get_example_config() -> Config:
+    return Config(
+        main_model="gpt-5.1",
+        subagent_models={"explore": "haiku", "oracle": "gpt-5.1-high"},
+        provider_list=[
+            llm_param.LLMConfigProviderParameter(
+                provider_name="openai",
+                protocol=llm_param.LLMClientProtocol.RESPONSES,
+                api_key="your-openai-api-key",
+                base_url="https://api.openai.com/v1",
+            ),
+            llm_param.LLMConfigProviderParameter(
+                provider_name="openrouter",
+                protocol=llm_param.LLMClientProtocol.OPENROUTER,
+                api_key="your-openrouter-api-key",
+            ),
+        ],
+        model_list=[
+            ModelConfig(
+                model_name="gpt-5.1",
+                provider="openai",
+                model_params=llm_param.LLMConfigModelParameter(
+                    model="gpt-5.1-2025-11-13",
+                    max_tokens=32000,
+                    verbosity="medium",
+                    thinking=llm_param.Thinking(
+                        reasoning_effort="medium",
+                        reasoning_summary="auto",
+                        type="enabled",
+                        budget_tokens=None,
+                    ),
+                    context_limit=368000,
+                ),
+            ),
+            ModelConfig(
+                model_name="gpt-5.1-high",
+                provider="openai",
+                model_params=llm_param.LLMConfigModelParameter(
+                    model="gpt-5.1-2025-11-13",
+                    max_tokens=32000,
+                    verbosity="medium",
+                    thinking=llm_param.Thinking(
+                        reasoning_effort="high",
+                        reasoning_summary="auto",
+                        type="enabled",
+                        budget_tokens=None,
+                    ),
+                    context_limit=368000,
+                ),
+            ),
+            ModelConfig(
+                model_name="haiku",
+                provider="openrouter",
+                model_params=llm_param.LLMConfigModelParameter(
+                    model="anthropic/claude-haiku-4.5",
+                    max_tokens=32000,
+                    provider_routing=llm_param.OpenRouterProviderRouting(
+                        sort="throughput",
+                    ),
+                    context_limit=168000,
+                ),
+            ),
+        ],
+    )
+
+
+@lru_cache(maxsize=1)
+def load_config() -> Config | None:
+    if not config_path.exists():
+        log(f"Config file not found: {config_path}")
+        example_config = get_example_config()
+        config_path.parent.mkdir(parents=True, exist_ok=True)
+        config_dict = example_config.model_dump(mode="json", exclude_none=True)
+
+        # Comment out all example config lines
+        yaml_str = yaml.dump(config_dict, default_flow_style=False, sort_keys=False)
+        commented_yaml = "\n".join(f"# {line}" if line.strip() else "#" for line in yaml_str.splitlines())
+        _ = config_path.write_text(commented_yaml)
+
+        log(f"Example config created at: {config_path}")
+        log("Please edit the config file to set up your models", style="yellow bold")
+        return None
+
+    config_yaml = config_path.read_text()
+    config_dict = yaml.safe_load(config_yaml)
+
+    if config_dict is None:
+        log(f"Config file is empty or all commented: {config_path}", style="red bold")
+        log("Please edit the config file to set up your models", style="yellow bold")
+        return None
+
+    try:
+        config = Config.model_validate(config_dict)
+    except ValidationError as e:
+        log(f"Invalid config file: {config_path}", style="red bold")
+        log(str(e), style="red")
+        raise ValueError(f"Invalid config file: {config_path}") from e
+
+    return config

klaude_code/config/list_model.py

@@ -0,0 +1,162 @@
+from rich.console import Console, Group
+from rich.panel import Panel
+from rich.table import Table
+from rich.text import Text
+
+from klaude_code.config import Config
+from klaude_code.protocol.sub_agent import iter_sub_agent_profiles
+from klaude_code.ui.rich.theme import ThemeKey, get_theme
+
+
+def mask_api_key(api_key: str | None) -> str:
+    """Mask API key to show only first 6 and last 6 characters with *** in between"""
+    if not api_key or api_key == "N/A":
+        return "N/A"
+
+    if len(api_key) <= 12:
+        return api_key
+
+    return f"{api_key[:6]} … {api_key[-6:]}"
+
+
+def display_models_and_providers(config: Config):
+    """Display models and providers configuration using rich formatting"""
+    themes = get_theme(config.theme)
+    console = Console(theme=themes.app_theme)
+
+    # Display providers section
+    providers_table = Table.grid(padding=(0, 1), expand=True)
+    providers_table.add_column(width=2, no_wrap=True)  # Status
+    providers_table.add_column(overflow="fold")  # Name
+    providers_table.add_column(overflow="fold")  # Protocol
+    providers_table.add_column(overflow="fold")  # Base URL
+    providers_table.add_column(overflow="fold")  # API Key
+
+    # Add header
+    providers_table.add_row(
+        Text("", style="bold"),
+        Text("Name", style=f"bold {ThemeKey.CONFIG_TABLE_HEADER}"),
+        Text("Protocol", style=f"bold {ThemeKey.CONFIG_TABLE_HEADER}"),
+        Text("Base URL", style=f"bold {ThemeKey.CONFIG_TABLE_HEADER}"),
+        Text("API Key", style=f"bold {ThemeKey.CONFIG_TABLE_HEADER}"),
+    )
+
+    # Add providers
+    for provider in config.provider_list:
+        status = Text("✔", style=f"bold {ThemeKey.CONFIG_STATUS_OK}")
+        name = Text(provider.provider_name, style=ThemeKey.CONFIG_ITEM_NAME)
+        protocol = Text(str(provider.protocol.value), style="")
+        base_url = Text(provider.base_url or "N/A", style="")
+        api_key = Text(mask_api_key(provider.api_key), style="")
+
+        providers_table.add_row(status, name, protocol, base_url, api_key)
+
+    # Display models section
+    models_table = Table.grid(padding=(0, 1), expand=True)
+    models_table.add_column(width=2, no_wrap=True)  # Status
+    models_table.add_column(overflow="fold", ratio=1)  # Name
+    models_table.add_column(overflow="fold", ratio=2)  # Model
+    models_table.add_column(overflow="fold", ratio=2)  # Provider
+    models_table.add_column(overflow="fold", ratio=3)  # Params
+
+    # Add header
+    models_table.add_row(
+        Text("", style="bold"),
+        Text("Name", style=f"bold {ThemeKey.CONFIG_TABLE_HEADER}"),
+        Text("Model", style=f"bold {ThemeKey.CONFIG_TABLE_HEADER}"),
+        Text("Provider", style=f"bold {ThemeKey.CONFIG_TABLE_HEADER}"),
+        Text("Params", style=f"bold {ThemeKey.CONFIG_TABLE_HEADER}"),
+    )
+
+    # Add models
+    for model in config.model_list:
+        status = Text("✔", style=f"bold {ThemeKey.CONFIG_STATUS_OK}")
+        if model.model_name == config.main_model:
+            status = Text("★", style=f"bold {ThemeKey.CONFIG_STATUS_PRIMARY}")  # Mark main model
+
+        name = Text(
+            model.model_name,
+            style=ThemeKey.CONFIG_STATUS_PRIMARY
+            if model.model_name == config.main_model
+            else ThemeKey.CONFIG_ITEM_NAME,
+        )
+        model_name = Text(model.model_params.model or "N/A", style="")
+        provider = Text(model.provider, style="")
+        params: list[Text] = []
+        if model.model_params.thinking:
+            if model.model_params.thinking.reasoning_effort is not None:
+                params.append(
+                    Text.assemble(
+                        ("reason-effort", ThemeKey.CONFIG_PARAM_LABEL),
+                        ": ",
+                        model.model_params.thinking.reasoning_effort,
+                    )
+                )
+            if model.model_params.thinking.reasoning_summary is not None:
+                params.append(
+                    Text.assemble(
+                        ("reason-summary", ThemeKey.CONFIG_PARAM_LABEL),
+                        ": ",
+                        model.model_params.thinking.reasoning_summary,
+                    )
+                )
+            if model.model_params.thinking.budget_tokens is not None:
+                params.append(
+                    Text.assemble(
+                        ("thinking-budget-tokens", ThemeKey.CONFIG_PARAM_LABEL),
+                        ": ",
+                        str(model.model_params.thinking.budget_tokens),
+                    )
+                )
+        if model.model_params.provider_routing:
+            params.append(
+                Text.assemble(
+                    ("provider-routing", ThemeKey.CONFIG_PARAM_LABEL),
+                    ": ",
+                    model.model_params.provider_routing.model_dump_json(exclude_none=True),
+                )
+            )
+        if len(params) == 0:
+            params.append(Text("N/A", style=ThemeKey.CONFIG_PARAM_LABEL))
+        models_table.add_row(status, name, model_name, provider, Group(*params))
+
+    # Create panels and display
+    providers_panel = Panel(
+        providers_table,
+        title=Text("Providers Configuration", style="white bold"),
+        border_style=ThemeKey.CONFIG_PANEL_BORDER,
+        padding=(0, 1),
+        title_align="left",
+    )
+
+    models_panel = Panel(
+        models_table,
+        title=Text("Models Configuration", style="white bold"),
+        border_style=ThemeKey.CONFIG_PANEL_BORDER,
+        padding=(0, 1),
+        title_align="left",
+    )
+
+    console.print(providers_panel)
+    console.print()
+    console.print(models_panel)
+
+    # Display main model info
+    console.print()
+    console.print(
+        Text.assemble(
+            ("Default Model: ", "bold"),
+            (config.main_model, ThemeKey.CONFIG_STATUS_PRIMARY),
+        )
+    )
+
+    for profile in iter_sub_agent_profiles():
+        sub_model_name = config.subagent_models.get(profile.name)
+        if not sub_model_name:
+            continue
+        console.print(
+            Text.assemble(
+                (f"{profile.name} Model: ", "bold"),
+                (sub_model_name, ThemeKey.CONFIG_STATUS_PRIMARY),
+            )
+        )

klaude_code/config/select_model.py

@@ -0,0 +1,67 @@
+from klaude_code.config.config import load_config
+from klaude_code.trace import log
+from klaude_code.ui.rich.searchable_text import SearchableFormattedList
+
+
+def select_model_from_config(preferred: str | None = None) -> str | None:
+    """
+    Interactive single-choice model selector.
+    for `--select-model`
+    """
+    config = load_config()
+    assert config is not None
+    models = sorted(config.model_list, key=lambda m: m.model_name.lower())
+
+    if not models:
+        raise ValueError("No models configured. Please update your config.yaml")
+
+    names: list[str] = [m.model_name for m in models]
+    default_name: str | None = (
+        preferred if preferred in names else (config.main_model if config.main_model in names else None)
+    )
+
+    try:
+        import questionary
+
+        choices: list[questionary.Choice] = []
+
+        max_model_name_length = max(len(m.model_name) for m in models)
+        for m in models:
+            star = "★ " if m.model_name == config.main_model else "  "
+            fragments = [
+                ("class:t", f"{star}{m.model_name:<{max_model_name_length}}   → "),
+                ("class:b", m.model_params.model or "N/A"),
+                ("class:d", f" {m.provider}"),
+            ]
+            # Provide a formatted title for display and a plain text for search.
+            title = SearchableFormattedList(fragments)
+            choices.append(questionary.Choice(title=title, value=m.model_name))
+
+        try:
+            result = questionary.select(
+                message="Select a model:",
+                choices=choices,
+                default=default_name,
+                pointer="→",
+                instruction="↑↓ to move • Enter to select",
+                use_jk_keys=False,
+                use_search_filter=True,
+                style=questionary.Style(
+                    [
+                        ("t", ""),
+                        ("b", "bold"),
+                        ("d", "dim"),
+                        # search filter colors at the bottom
+                        ("search_success", "noinherit fg:ansigreen"),
+                        ("search_none", "noinherit fg:ansired"),
+                        ("question-mark", "fg:ansigreen"),
+                    ]
+                ),
+            ).ask()
+            if isinstance(result, str) and result in names:
+                return result
+        except KeyboardInterrupt:
+            return None
+    except Exception as e:
+        log(f"Failed to use questionary, falling back to default model, {e}")
+        return preferred

klaude_code/const/__init__.py

@@ -0,0 +1,133 @@
+"""Centralized configuration constants for klaude_code.
+
+This module consolidates all magic numbers and configuration values
+that were previously scattered across the codebase.
+"""
+
+# =============================================================================
+# Agent Configuration
+# =============================================================================
+
+# Timeout for waiting for the first event from LLM (seconds)
+FIRST_EVENT_TIMEOUT_S = 200.0
+
+# Maximum number of retry attempts for failed turns
+MAX_FAILED_TURN_RETRIES = 10
+
+# Initial delay before retrying a failed turn (seconds)
+INITIAL_RETRY_DELAY_S = 1.0
+
+# Maximum delay between retries (seconds)
+MAX_RETRY_DELAY_S = 30.0
+
+# Message shown when a tool call is cancelled by the user
+CANCEL_OUTPUT = "[Request interrupted by user for tool use]"
+
+# Default maximum tokens for LLM responses
+DEFAULT_MAX_TOKENS = 32000
+
+# Default temperature for LLM requests
+DEFAULT_TEMPERATURE = 1.0
+
+# Default thinking budget tokens for Anthropic models
+DEFAULT_ANTHROPIC_THINKING_BUDGET_TOKENS = 2048
+
+# Tool call count threshold for todo reminder
+TODO_REMINDER_TOOL_CALL_THRESHOLD = 10
+
+
+# =============================================================================
+# Tool
+# =============================================================================
+
+# -- Read Tool --
+# Maximum characters per line before truncation
+READ_CHAR_LIMIT_PER_LINE = 2000
+
+# Maximum number of lines to read from a file
+READ_GLOBAL_LINE_CAP = 2000
+
+# Maximum total characters to read
+READ_MAX_CHARS = 60000
+
+# Maximum file size in KB for text files
+READ_MAX_KB = 256
+
+# Maximum image file size in bytes (4MB)
+READ_MAX_IMAGE_BYTES = 4 * 1024 * 1024
+
+# -- Bash Tool --
+# Default timeout for bash commands (milliseconds)
+BASH_DEFAULT_TIMEOUT_MS = 120000
+
+# -- Tool Output --
+# Maximum length for tool output before truncation
+TOOL_OUTPUT_MAX_LENGTH = 50000
+
+# Characters to show from the beginning of truncated output
+TOOL_OUTPUT_DISPLAY_HEAD = 10000
+
+# Characters to show from the end of truncated output
+TOOL_OUTPUT_DISPLAY_TAIL = 10000
+
+# Directory for saving full truncated output
+TOOL_OUTPUT_TRUNCATION_DIR = "/tmp/klaude"
+
+
+# =============================================================================
+# UI
+# =============================================================================
+
+# Width of line number prefix in diff display
+DIFF_PREFIX_WIDTH = 4
+
+# Maximum lines to show in diff output
+MAX_DIFF_LINES = 1000
+
+# Maximum length for invalid tool call display
+INVALID_TOOL_CALL_MAX_LENGTH = 500
+
+# Maximum line length for truncated display output
+TRUNCATE_DISPLAY_MAX_LINE_LENGTH = 1000
+
+# Maximum lines for truncated display output
+TRUNCATE_DISPLAY_MAX_LINES = 10
+
+# Maximum lines for sub-agent result display
+SUB_AGENT_RESULT_MAX_LINES = 12
+
+
+# UI refresh rate (frames per second) for debounced content streaming
+UI_REFRESH_RATE_FPS = 20
+
+# Number of lines to keep visible at bottom of markdown streaming window
+MARKDOWN_STREAM_LIVE_WINDOW = 20
+
+# Status shimmer animation
+# Horizontal padding used when computing shimmer band position
+STATUS_SHIMMER_PADDING = 10
+# Duration in seconds for one full shimmer sweep across the text
+STATUS_SHIMMER_SWEEP_SECONDS = 2
+# Half-width of the shimmer band in characters
+STATUS_SHIMMER_BAND_HALF_WIDTH = 5.0
+# Scale factor applied to shimmer intensity when blending colors
+STATUS_SHIMMER_ALPHA_SCALE = 0.7
+
+# Spinner breathing animation
+# Duration in seconds for one full breathe-in + breathe-out cycle
+# Keep in sync with STATUS_SHIMMER_SWEEP_SECONDS for visual consistency
+SPINNER_BREATH_PERIOD_SECONDS = 2
+
+
+# =============================================================================
+# Debug / Logging
+# =============================================================================
+
+# Default debug log file path
+DEFAULT_DEBUG_LOG_FILE = "debug.log"
+
+# Maximum log file size before rotation (10MB)
+LOG_MAX_BYTES = 10 * 1024 * 1024
+
+# Number of backup log files to keep
+LOG_BACKUP_COUNT = 3