klaude-code 1.2.1__py3-none-any.whl → 1.2.3__py3-none-any.whl

klaude_code/command/model_cmd.py

@@ -1,15 +1,10 @@
 import asyncio
 
-from klaude_code.command.command_abc import CommandABC, CommandResult
+from klaude_code.command.command_abc import CommandABC, CommandResult, InputAction
 from klaude_code.command.registry import register_command
-from klaude_code.config import load_config
-from klaude_code.config.select_model import select_model_from_config
-from klaude_code.core import Agent
-from klaude_code.llm.registry import create_llm_client
-from klaude_code.protocol.commands import CommandName
-from klaude_code.protocol.events import DeveloperMessageEvent, WelcomeEvent
-from klaude_code.protocol.model import CommandOutput, DeveloperMessageItem
-from klaude_code.trace import DebugType, log_debug
+from klaude_code.config import select_model_from_config
+from klaude_code.core.agent import Agent
+from klaude_code.protocol import commands, events, model
 
 
 @register_command
@@ -17,8 +12,8 @@ class ModelCommand(CommandABC):
     """Display or change the model configuration."""
 
     @property
-    def name(self) -> CommandName:
-        return CommandName.MODEL
+    def name(self) -> commands.CommandName:
+        return commands.CommandName.MODEL
 
     @property
     def summary(self) -> str:
@@ -35,38 +30,14 @@ class ModelCommand(CommandABC):
         if selected_model is None or selected_model == current_model:
             return CommandResult(
                 events=[
-                    DeveloperMessageEvent(
+                    events.DeveloperMessageEvent(
                         session_id=agent.session.id,
-                        item=DeveloperMessageItem(
-                            content="(no change)", command_output=CommandOutput(command_name=self.name)
+                        item=model.DeveloperMessageItem(
+                            content="(no change)",
+                            command_output=model.CommandOutput(command_name=self.name),
                         ),
                     )
                 ]
             )
 
-        config = load_config()
-        assert config is not None
-        llm_config = config.get_model_config(selected_model)
-
-        log_debug(
-            "Updated LLM config",
-            llm_config.model_dump_json(exclude_none=True),
-            style="yellow",
-            debug_type=DebugType.LLM_CONFIG,
-        )
-
-        llm_client = create_llm_client(llm_config)
-        agent.set_model_profile(agent.build_model_profile(llm_client))
-
-        return CommandResult(
-            events=[
-                DeveloperMessageEvent(
-                    session_id=agent.session.id,
-                    item=DeveloperMessageItem(
-                        content=f"switched to model: {selected_model}",
-                        command_output=CommandOutput(command_name=self.name),
-                    ),
-                ),
-                WelcomeEvent(llm_config=llm_config, work_dir=str(agent.session.work_dir)),
-            ]
-        )
+        return CommandResult(actions=[InputAction.change_model(selected_model)])

klaude_code/command/{prompt-update-dev-doc.md → prompt-dev-docs-update.md}

@@ -1,6 +1,6 @@
 ---
 description: Update dev documentation before context compaction
-argument-hint: Optional - specific context or tasks to focus on (leave empty for comprehensive update)
+from: https://github.com/diet103/claude-code-infrastructure-showcase/blob/main/.claude/commands/dev-docs-update.md
 ---
 
 We're approaching context limits. Please update the development documentation to ensure seamless continuation after context reset.
@@ -50,6 +50,7 @@ If switching to a new conversation:
 - Any uncommitted changes that need attention
 - Test commands to verify work
 
-## Additional Context: $ARGUMENTS
+## Additional Context:
+$ARGUMENTS
 
 **Priority**: Focus on capturing information that would be hard to rediscover or reconstruct from code alone.

klaude_code/command/{prompt-dev-doc.md → prompt-dev-docs.md}

@@ -1,9 +1,10 @@
 ---
 description: Create a comprehensive strategic plan with structured task breakdown
-argument-hint: Describe what you need planned (e.g., "refactor authentication system", "implement microservices")
+from: https://github.com/diet103/claude-code-infrastructure-showcase/blob/main/.claude/commands/dev-docs.md
 ---
 
-You are an elite strategic planning specialist. Create a comprehensive, actionable plan for: $ARGUMENTS
+Create a comprehensive, actionable plan for: 
+$ARGUMENTS
 
 ## Instructions
 

klaude_code/command/prompt-init.md

@@ -1,6 +1,6 @@
 ---
 description: Create an AGENTS.md file with instructions for agent
-argument-hint: Additional user instructions
+from: https://github.com/openai/codex/blob/main/codex-rs/tui/prompt_for_init_command.md
 ---
 
 Generate/Update a file named AGENTS.md that serves as a contributor guide for this repository.
@@ -42,7 +42,4 @@ Commit & Pull Request Guidelines
 - Summarize commit message conventions found in the project’s Git history.
 - Outline pull request requirements (descriptions, linked issues, screenshots, etc.).
 
-(Optional) Add other sections if relevant, such as Security & Configuration Tips, Architecture Overview, or Agent-Specific Instructions.
-
-Additional Instructions:
-$ARGUMENTS
+(Optional) Add other sections if relevant, such as Security & Configuration Tips, Architecture Overview, or Agent-Specific Instructions.

klaude_code/command/prompt_command.py

@@ -2,9 +2,9 @@ from importlib.resources import files
 
 import yaml
 
-from klaude_code.command.command_abc import CommandABC, CommandResult
-from klaude_code.core import Agent
-from klaude_code.protocol.commands import CommandName
+from klaude_code.command.command_abc import CommandABC, CommandResult, InputAction
+from klaude_code.core.agent import Agent
+from klaude_code.protocol import commands
 
 
 class PromptCommand(CommandABC):
@@ -17,7 +17,7 @@ class PromptCommand(CommandABC):
         self._metadata: dict[str, str] = {}
 
     @property
-    def name(self) -> str | CommandName:
+    def name(self) -> str | commands.CommandName:
         return self._command_name
 
     @property
@@ -57,7 +57,7 @@ class PromptCommand(CommandABC):
     async def run(self, raw: str, agent: Agent) -> CommandResult:
         self._ensure_loaded()
         template_content = self._content or ""
-        user_input = raw.strip()
+        user_input = raw.strip() or "<none>"
 
         if "$ARGUMENTS" in template_content:
             final_prompt = template_content.replace("$ARGUMENTS", user_input)
@@ -66,4 +66,4 @@ class PromptCommand(CommandABC):
             if user_input:
                 final_prompt += f"\n\nAdditional Instructions:\n{user_input}"
 
-        return CommandResult(agent_input=final_prompt)
+        return CommandResult(actions=[InputAction.run_agent(final_prompt)])

klaude_code/command/refresh_cmd.py

@@ -1,8 +1,7 @@
 from klaude_code.command.command_abc import CommandABC, CommandResult
 from klaude_code.command.registry import register_command
-from klaude_code.core import Agent
-from klaude_code.protocol import events
-from klaude_code.protocol.commands import CommandName
+from klaude_code.core.agent import Agent
+from klaude_code.protocol import commands, events
 
 
 @register_command
@@ -10,8 +9,8 @@ class RefreshTerminalCommand(CommandABC):
     """Refresh terminal display"""
 
     @property
-    def name(self) -> CommandName:
-        return CommandName.REFRESH_TERMINAL
+    def name(self) -> commands.CommandName:
+        return commands.CommandName.REFRESH_TERMINAL
 
     @property
     def summary(self) -> str:

klaude_code/command/registry.py

@@ -1,17 +1,15 @@
 from importlib.resources import files
 from typing import TYPE_CHECKING, TypeVar
 
-from klaude_code.command.command_abc import CommandResult
+from klaude_code.command.command_abc import CommandResult, InputAction
 from klaude_code.command.prompt_command import PromptCommand
 from klaude_code.core.agent import Agent
-from klaude_code.protocol.commands import CommandName
-from klaude_code.protocol.events import DeveloperMessageEvent
-from klaude_code.protocol.model import CommandOutput, DeveloperMessageItem
+from klaude_code.protocol import commands, events, model
 
 if TYPE_CHECKING:
     from .command_abc import CommandABC
 
-_COMMANDS: dict[CommandName | str, "CommandABC"] = {}
+_COMMANDS: dict[commands.CommandName | str, "CommandABC"] = {}
 
 T = TypeVar("T", bound="CommandABC")
 
@@ -37,20 +35,19 @@ def load_prompt_commands():
         pass
 
 
-def get_commands() -> dict[CommandName | str, "CommandABC"]:
+def get_commands() -> dict[commands.CommandName | str, "CommandABC"]:
     """Get all registered commands."""
     return _COMMANDS.copy()
 
 
-def get_command_names() -> list[str]:
-    """Get all registered command names for completion."""
-    return [str(k) for k in _COMMANDS.keys()]
+def is_slash_command_name(name: str) -> bool:
+    return name in _COMMANDS
 
 
 async def dispatch_command(raw: str, agent: Agent) -> CommandResult:
     # Detect command name
     if not raw.startswith("/"):
-        return CommandResult(agent_input=raw)
+        return CommandResult(actions=[InputAction.run_agent(raw)])
 
     splits = raw.split(" ", maxsplit=1)
     command_name_raw = splits[0][1:]
@@ -65,31 +62,31 @@ async def dispatch_command(raw: str, agent: Agent) -> CommandResult:
     else:
         # Then try Enum conversion for standard commands
         try:
-            enum_key = CommandName(command_name_raw)
+            enum_key = commands.CommandName(command_name_raw)
             if enum_key in _COMMANDS:
                 command_key = enum_key
         except ValueError:
             pass
 
     if command_key is None:
-        return CommandResult(agent_input=raw)
+        return CommandResult(actions=[InputAction.run_agent(raw)])
 
     command = _COMMANDS[command_key]
-    command_identifier: CommandName | str = command.name
+    command_identifier: commands.CommandName | str = command.name
 
     try:
         return await command.run(rest, agent)
     except Exception as e:
         command_output = (
-            CommandOutput(command_name=command_identifier, is_error=True)
-            if isinstance(command_identifier, CommandName)
+            model.CommandOutput(command_name=command_identifier, is_error=True)
+            if isinstance(command_identifier, commands.CommandName)
             else None
         )
         return CommandResult(
             events=[
-                DeveloperMessageEvent(
+                events.DeveloperMessageEvent(
                     session_id=agent.session.id,
-                    item=DeveloperMessageItem(
+                    item=model.DeveloperMessageItem(
                         content=f"Command {command_identifier} error: [{e.__class__.__name__}] {str(e)}",
                         command_output=command_output,
                     ),
@@ -98,13 +95,13 @@ async def dispatch_command(raw: str, agent: Agent) -> CommandResult:
         )
 
 
-def is_interactive_command(raw: str) -> bool:
+def has_interactive_command(raw: str) -> bool:
     if not raw.startswith("/"):
         return False
     splits = raw.split(" ", maxsplit=1)
     command_name_raw = splits[0][1:]
     try:
-        command_name = CommandName(command_name_raw)
+        command_name = commands.CommandName(command_name_raw)
     except ValueError:
         return False
     if command_name not in _COMMANDS:

klaude_code/command/terminal_setup_cmd.py

@@ -4,10 +4,8 @@ from pathlib import Path
 
 from klaude_code.command.command_abc import CommandABC, CommandResult
 from klaude_code.command.registry import register_command
-from klaude_code.core import Agent
-from klaude_code.protocol import events
-from klaude_code.protocol.commands import CommandName
-from klaude_code.protocol.model import CommandOutput, DeveloperMessageItem
+from klaude_code.core.agent import Agent
+from klaude_code.protocol import commands, events, model
 
 
 @register_command
@@ -15,8 +13,8 @@ class TerminalSetupCommand(CommandABC):
     """Setup shift+enter newline functionality in terminal"""
 
     @property
-    def name(self) -> CommandName:
-        return CommandName.TERMINAL_SETUP
+    def name(self) -> commands.CommandName:
+        return commands.CommandName.TERMINAL_SETUP
 
     @property
     def summary(self) -> str:
@@ -197,7 +195,10 @@ class TerminalSetupCommand(CommandABC):
             return True, "appended mapping"
 
         # Not an array – avoid modifying to prevent corrupting user config
-        return False, "unsupported keybindings.json format (not an array); please add mapping manually"
+        return (
+            False,
+            "unsupported keybindings.json format (not an array); please add mapping manually",
+        )
 
     def _setup_generic(self, term_program: str) -> str:
         """Provide generic manual configuration guide for unknown or unsupported terminals"""
@@ -228,9 +229,9 @@ class TerminalSetupCommand(CommandABC):
             events=[
                 events.DeveloperMessageEvent(
                     session_id=agent.session.id,
-                    item=DeveloperMessageItem(
+                    item=model.DeveloperMessageItem(
                         content=message,
-                        command_output=CommandOutput(command_name=self.name, is_error=False),
+                        command_output=model.CommandOutput(command_name=self.name, is_error=False),
                     ),
                 )
             ]
@@ -242,9 +243,9 @@ class TerminalSetupCommand(CommandABC):
             events=[
                 events.DeveloperMessageEvent(
                     session_id=agent.session.id,
-                    item=DeveloperMessageItem(
+                    item=model.DeveloperMessageItem(
                         content=message,
-                        command_output=CommandOutput(command_name=self.name, is_error=True),
+                        command_output=model.CommandOutput(command_name=self.name, is_error=True),
                     ),
                 )
             ]

klaude_code/config/__init__.py

@@ -1,7 +1,11 @@
 from .config import Config, config_path, load_config
+from .list_model import display_models_and_providers
+from .select_model import select_model_from_config
 
 __all__ = [
     "Config",
     "load_config",
     "config_path",
+    "display_models_and_providers",
+    "select_model_from_config",
 ]

klaude_code/config/config.py

@@ -6,15 +6,8 @@ from typing import Any, cast
 import yaml
 from pydantic import BaseModel, Field, ValidationError, model_validator
 
-from klaude_code.core.sub_agent import iter_sub_agent_profiles
-from klaude_code.protocol.llm_parameter import (
-    LLMClientProtocol,
-    LLMConfigModelParameter,
-    LLMConfigParameter,
-    LLMConfigProviderParameter,
-    OpenRouterProviderRouting,
-    Thinking,
-)
+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"
@@ -23,11 +16,11 @@ config_path = Path.home() / ".klaude" / "klaude-config.yaml"
 class ModelConfig(BaseModel):
     model_name: str
     provider: str
-    model_params: LLMConfigModelParameter
+    model_params: llm_param.LLMConfigModelParameter
 
 
 class Config(BaseModel):
-    provider_list: list[LLMConfigProviderParameter]
+    provider_list: list[llm_param.LLMConfigProviderParameter]
     model_list: list[ModelConfig]
     main_model: str
     subagent_models: dict[str, str] = Field(default_factory=dict)
@@ -46,10 +39,10 @@ class Config(BaseModel):
         data["subagent_models"] = normalized
         return data
 
-    def get_main_model_config(self) -> LLMConfigParameter:
+    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) -> LLMConfigParameter:
+    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,
@@ -64,7 +57,7 @@ class Config(BaseModel):
         if provider is None:
             raise ValueError(f"Unknown provider: {model.provider}")
 
-        return LLMConfigParameter(
+        return llm_param.LLMConfigParameter(
             **provider.model_dump(),
             **model.model_params.model_dump(),
         )
@@ -88,15 +81,15 @@ def get_example_config() -> Config:
         main_model="gpt-5.1",
         subagent_models={"explore": "haiku", "oracle": "gpt-5.1-high"},
         provider_list=[
-            LLMConfigProviderParameter(
+            llm_param.LLMConfigProviderParameter(
                 provider_name="openai",
-                protocol=LLMClientProtocol.RESPONSES,
+                protocol=llm_param.LLMClientProtocol.RESPONSES,
                 api_key="your-openai-api-key",
                 base_url="https://api.openai.com/v1",
             ),
-            LLMConfigProviderParameter(
+            llm_param.LLMConfigProviderParameter(
                 provider_name="openrouter",
-                protocol=LLMClientProtocol.OPENROUTER,
+                protocol=llm_param.LLMClientProtocol.OPENROUTER,
                 api_key="your-openrouter-api-key",
             ),
         ],
@@ -104,12 +97,15 @@ def get_example_config() -> Config:
             ModelConfig(
                 model_name="gpt-5.1",
                 provider="openai",
-                model_params=LLMConfigModelParameter(
+                model_params=llm_param.LLMConfigModelParameter(
                     model="gpt-5.1-2025-11-13",
                     max_tokens=32000,
                     verbosity="medium",
-                    thinking=Thinking(
-                        reasoning_effort="medium", reasoning_summary="auto", type="enabled", budget_tokens=None
+                    thinking=llm_param.Thinking(
+                        reasoning_effort="medium",
+                        reasoning_summary="auto",
+                        type="enabled",
+                        budget_tokens=None,
                     ),
                     context_limit=368000,
                 ),
@@ -117,12 +113,15 @@ def get_example_config() -> Config:
             ModelConfig(
                 model_name="gpt-5.1-high",
                 provider="openai",
-                model_params=LLMConfigModelParameter(
+                model_params=llm_param.LLMConfigModelParameter(
                     model="gpt-5.1-2025-11-13",
                     max_tokens=32000,
                     verbosity="medium",
-                    thinking=Thinking(
-                        reasoning_effort="high", reasoning_summary="auto", type="enabled", budget_tokens=None
+                    thinking=llm_param.Thinking(
+                        reasoning_effort="high",
+                        reasoning_summary="auto",
+                        type="enabled",
+                        budget_tokens=None,
                     ),
                     context_limit=368000,
                 ),
@@ -130,10 +129,10 @@ def get_example_config() -> Config:
             ModelConfig(
                 model_name="haiku",
                 provider="openrouter",
-                model_params=LLMConfigModelParameter(
+                model_params=llm_param.LLMConfigModelParameter(
                     model="anthropic/claude-haiku-4.5",
                     max_tokens=32000,
-                    provider_routing=OpenRouterProviderRouting(
+                    provider_routing=llm_param.OpenRouterProviderRouting(
                         sort="throughput",
                     ),
                     context_limit=168000,

klaude_code/config/list_model.py

@@ -4,8 +4,8 @@ from rich.table import Table
 from rich.text import Text
 
 from klaude_code.config import Config
-from klaude_code.core.sub_agent import iter_sub_agent_profiles
-from klaude_code.ui.base.theme import ThemeKey, get_theme
+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:
@@ -143,7 +143,12 @@ def display_models_and_providers(config: Config):
 
     # Display main model info
     console.print()
-    console.print(Text.assemble(("Default Model: ", "bold"), (config.main_model, ThemeKey.CONFIG_STATUS_PRIMARY)))
+    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)

klaude_code/config/select_model.py

@@ -1,6 +1,6 @@
 from klaude_code.config.config import load_config
 from klaude_code.trace import log
-from klaude_code.ui.rich_ext.searchable_text import SearchableFormattedList
+from klaude_code.ui.rich.searchable_text import SearchableFormattedList
 
 
 def select_model_from_config(preferred: str | None = None) -> str | None:

klaude_code/const/__init__.py

@@ -109,7 +109,7 @@ 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 = 3.0
+STATUS_SHIMMER_BAND_HALF_WIDTH = 5.0
 # Scale factor applied to shimmer intensity when blending colors
 STATUS_SHIMMER_ALPHA_SCALE = 0.7
 

klaude_code/core/__init__.py

@@ -1,3 +0,0 @@
-from .agent import Agent
-
-__all__ = ["Agent"]