@baimingtao/lbs-agent 1.0.0

package/interfaces/cli/_impl/main.py

@@ -0,0 +1,727 @@
+"""CLI 入口与命令调度器。
+
+命令实现位于 ``cli.commands``。本文件应保持精简：解析 argv、加载配置、调度。
+"""
+
+from __future__ import annotations
+
+import json
+import logging
+import sys
+from typing import Any, Callable
+
+from application.container import AgentSystemContainer
+from config.config import load_config
+from interfaces.cli._impl.commands.common import bar as _bar, mask as _mask, project_root
+from interfaces.cli._impl.commands.agents import cmd_agents, cmd_config, cmd_status
+from interfaces.cli._impl.commands.doctor import cmd_doctor
+from interfaces.cli._impl.commands.gateway import cmd_gateway
+from interfaces.cli._impl.commands.skill import cmd_skill
+from interfaces.cli._impl.commands.llm import (
+    cmd_llm_diagnostics, cmd_llm_inspect, cmd_llm_route, cmd_llm_validate,
+    llm_validate_command_result,
+)
+from interfaces.cli._impl.commands.models import (
+    cmd_model_current, cmd_model_health, cmd_model_keys,
+    cmd_model_list, cmd_model_switch, cmd_models, cmd_models_fetch,
+    cmd_key_add, cmd_key_rotate, cmd_config_set,
+)
+from interfaces.cli._impl.commands.orchestrator import cmd_orchestrator_dry_run, cmd_orchestrator_inspect
+from interfaces.cli._impl.commands.chat_repl import cmd_chat
+from interfaces.cli._impl.commands.run_agent import cmd_compress, cmd_run, cmd_run_agent
+from interfaces.cli._impl.commands.sessions import (
+    cmd_sessions_delete, cmd_sessions_export, cmd_sessions_list,
+    cmd_sessions_prune, cmd_sessions_report, cmd_sessions_show,
+    cmd_sessions_stats, cmd_workers_list, cmd_workers_show,
+    cmd_session_rename, cmd_session_import,
+)
+from interfaces.cli._impl.commands.tools import (
+    cmd_mcp_list, cmd_mcp_add, cmd_mcp_remove, cmd_mcp_test,
+    cmd_skills_list, cmd_skills_view, cmd_tools_inspect,
+    cmd_tools_list, cmd_tools_profiles, skills_list_command_result, tools_inspect_command_result,
+    tools_list_command_result, tools_profiles_command_result,
+)
+from interfaces.cli._impl.commands.architecture import architecture_command_result, print_architecture_report
+from interfaces.cli._impl.renderers import render_json
+from interfaces.cli._impl.commands.cron import (
+    cmd_cron_list, cmd_cron_create, cmd_cron_remove, cmd_cron_pause,
+    cmd_cron_resume, cmd_cron_trigger, cmd_cron_tick, cmd_cron_status,
+    cmd_cron_logs,
+)
+from interfaces.cli._impl.commands.soul import cmd_soul_list, cmd_soul_switch, cmd_soul_show
+from interfaces.cli._impl.commands.smoke_test import main as smoke_main
+
+
+# --- 命令注册表 ---
+
+
+class CommandRegistry:
+    """命令注册表。映射命令名 to (handler, help_text)。"""
+
+    def __init__(self) -> None:
+        self._cmds: dict[str, tuple[Callable[..., int], str]] = {}
+        self._aliases: dict[str, str] = {}
+
+    def register(self, name: str, handler: Callable[..., int], *,
+                 aliases: list[str] | None = None, help: str = "") -> None:
+        normalized_name = name.strip()
+        normalized_aliases = tuple(alias.strip() for alias in (aliases or []) if alias.strip())
+        if not normalized_name:
+            raise ValueError("命令名不能为空")
+        if normalized_name in self._cmds:
+            raise ValueError(f"重复命令名: {normalized_name}")
+        if normalized_name in self._aliases:
+            raise ValueError(f"命令名与既有 alias 冲突: {normalized_name}")
+        if len(set(normalized_aliases)) != len(normalized_aliases):
+            raise ValueError(f"同一命令存在重复 alias: {normalized_name}")
+        for alias in normalized_aliases:
+            if alias == normalized_name:
+                raise ValueError(f"alias 不能等于命令名: {normalized_name}")
+            if alias in self._cmds or alias in self._aliases:
+                raise ValueError(f"alias 冲突: {alias}")
+        self._cmds[normalized_name] = (handler, help)
+        for alias in normalized_aliases:
+            self._aliases[alias] = normalized_name
+
+    def public_inventory(self) -> dict[str, tuple[str, ...]]:
+        """返回 canonical command 到 aliases 的稳定公开映射。"""
+        return {
+            name: tuple(alias for alias, target in self._aliases.items() if target == name)
+            for name in self._cmds
+        }
+
+    def resolve(self, name: str) -> Callable[..., int] | None:
+        real = self._aliases.get(name, name)
+        entry = self._cmds.get(real)
+        return entry[0] if entry else None
+
+    def list_commands(self) -> list[str]:
+        return list(self._cmds.keys())
+
+
+def cmd_help() -> int:
+    lines = [
+        "",
+        "  LBS Agent CLI",
+        "",
+        "  用法:",
+        "    lbs <command> [args]",
+        "",
+        "  常用命令:",
+    ]
+    for name in sorted(_REGISTRY.list_commands()):
+        entry = _REGISTRY._cmds.get(name)
+        if entry:
+            _, help_text = entry
+            if help_text:
+                lines.append(f"    {name:<25s} {help_text}")
+    lines.extend([
+        "",
+        "  编排预览:",
+        "    lbs orchestrator dry-run <agent> <task>",
+    ])
+    print("\n".join(lines))
+    return 0
+
+
+def _cmd_model_handler(config: Any) -> int:
+    if len(sys.argv) < 3:
+        print("  用法: lbs model <list|current|switch|health|keys> ...", file=sys.stderr)
+        return 2
+    sub = sys.argv[2]
+    if sub == "list":
+        use_live = "--config" not in sys.argv
+        provider = next((a for a in sys.argv[3:] if not a.startswith("--")), "")
+        cmd_model_list(config, live=use_live, provider_filter=provider)
+    elif sub in ("current", "cur"):
+        cmd_model_current(config, live="--live" in sys.argv)
+    elif sub == "switch":
+        provider = sys.argv[3] if len(sys.argv) > 3 else ""
+        model = sys.argv[4] if len(sys.argv) > 4 else ""
+        cmd_model_switch(config, provider, model)
+    elif sub == "health":
+        provider = sys.argv[3] if len(sys.argv) > 3 else ""
+        cmd_model_health(config, provider)
+    elif sub == "keys":
+        provider = sys.argv[3] if len(sys.argv) > 3 else ""
+        cmd_model_keys(config, provider)
+    else:
+        print(f"  未知子命令: model {sub}", file=sys.stderr)
+        return 2
+    return 0
+
+
+def _cmd_models_handler(config: Any) -> int:
+    if len(sys.argv) > 2:
+        sub = sys.argv[2]
+        if sub == "fetch":
+            provider = sys.argv[3] if len(sys.argv) > 3 else ""
+            cmd_models_fetch(config, provider)
+        elif sub == "list":
+            provider = next((a for a in sys.argv[3:] if not a.startswith("--")), "")
+            cmd_model_list(config, live="--config" not in sys.argv, provider_filter=provider)
+        elif sub in ("current", "cur"):
+            cmd_model_current(config, live="--live" in sys.argv)
+        elif sub == "switch":
+            provider = sys.argv[3] if len(sys.argv) > 3 else ""
+            model = sys.argv[4] if len(sys.argv) > 4 else ""
+            cmd_model_switch(config, provider, model)
+        elif sub == "health":
+            provider = sys.argv[3] if len(sys.argv) > 3 else ""
+            cmd_model_health(config, provider)
+        elif sub == "keys":
+            provider = sys.argv[3] if len(sys.argv) > 3 else ""
+            cmd_model_keys(config, provider)
+        else:
+            cmd_models(config)
+    else:
+        cmd_models(config)
+    return 0
+
+
+def _cmd_llm_handler(config: Any) -> int:
+    sub = sys.argv[2] if len(sys.argv) > 2 else "inspect"
+    if sub == "inspect":
+        cmd_llm_inspect(config)
+    elif sub == "validate":
+        if "--json" in sys.argv:
+            return int(render_json(llm_validate_command_result(config)))
+        cmd_llm_validate(config)
+    elif sub == "diagnostics":
+        cmd_llm_diagnostics(config, json_output="--json" in sys.argv)
+    elif sub == "route":
+        if len(sys.argv) < 4:
+            print("  用法: lbs llm route <agent_name>", file=sys.stderr)
+            return 2
+        cmd_llm_route(config, sys.argv[3])
+    else:
+        print(f"  未知子命令: llm {sub}", file=sys.stderr)
+        return 2
+    return 0
+
+
+def _cmd_orchestrator_handler(config: Any) -> int:
+    sub = sys.argv[2] if len(sys.argv) > 2 else "inspect"
+    if sub == "inspect":
+        cmd_orchestrator_inspect(config)
+    elif sub == "dry-run":
+        if len(sys.argv) < 5:
+            print('  用法: lbs orchestrator dry-run <agent_name> "<task>"', file=sys.stderr)
+            return 2
+        cmd_orchestrator_dry_run(config, sys.argv[3], sys.argv[4], json_output="--json" in sys.argv)
+    else:
+        print(f"  未知子命令: orchestrator {sub}", file=sys.stderr)
+        return 2
+    return 0
+
+
+def _cmd_run_handler(config: Any) -> int:
+    if len(sys.argv) < 3:
+        print('  用法: lbs run "<task>"', file=sys.stderr)
+        return 2
+    cmd_run(config, sys.argv[2])
+    return 0
+
+
+def _cmd_chat_handler(config: Any) -> int:
+    agent_name = ""
+    provider_name = ""
+    model_name = ""
+    query = ""
+    resume_session = ""
+    args = sys.argv[2:]
+    i = 0
+    while i < len(args):
+        a = args[i]
+        if a == "--agent" and i + 1 < len(args):
+            agent_name = args[i + 1]
+            i += 2
+        elif a == "--provider" and i + 1 < len(args):
+            provider_name = args[i + 1]
+            i += 2
+        elif a == "--model" and i + 1 < len(args):
+            model_name = args[i + 1]
+            i += 2
+        elif a in ("--query", "-q") and i + 1 < len(args):
+            query = args[i + 1]
+            i += 2
+        elif a in ("--session", "--resume") and i + 1 < len(args):
+            resume_session = args[i + 1]
+            i += 2
+        else:
+            i += 1
+    return cmd_chat(config, agent_name=agent_name, provider_name=provider_name,
+                    model_name=model_name, query=query, session_id=resume_session)
+
+
+def _cmd_run_agent_handler(config: Any) -> int:
+    if len(sys.argv) < 4:
+        print('  用法: lbs run-agent <agent_name> "<task>"', file=sys.stderr)
+        print(f"  可用: {', '.join(sorted(config.agents.agents.keys()))}", file=sys.stderr)
+        return 2
+    cmd_run_agent(config, sys.argv[2], sys.argv[3])
+    return 0
+
+
+def _cmd_tools_handler(config: Any) -> int:
+    sub = sys.argv[2] if len(sys.argv) > 2 else "list"
+    if sub == "inspect":
+        tools_config_path = ""
+        if "--tools-config" in sys.argv:
+            idx = sys.argv.index("--tools-config")
+            if idx + 1 >= len(sys.argv):
+                print("  用法: lbs tools inspect [--tools-config <path>]", file=sys.stderr)
+                return 2
+            tools_config_path = sys.argv[idx + 1]
+        if "--json" in sys.argv:
+            return int(render_json(tools_inspect_command_result(config, tools_config_path)))
+        cmd_tools_inspect(config, tools_config_path=tools_config_path)
+    elif sub == "list":
+        agent_name = ""
+        if "--agent" in sys.argv:
+            idx = sys.argv.index("--agent")
+            if idx + 1 < len(sys.argv):
+                agent_name = sys.argv[idx + 1]
+        if "--json" in sys.argv:
+            return int(render_json(tools_list_command_result(config, agent_name)))
+        cmd_tools_list(config, agent_name=agent_name)
+    elif sub == "profiles":
+        tools_config_path = ""
+        if "--tools-config" in sys.argv:
+            idx = sys.argv.index("--tools-config")
+            if idx + 1 >= len(sys.argv):
+                print("  用法: lbs tools profiles [--tools-config <path>]", file=sys.stderr)
+                return 2
+            tools_config_path = sys.argv[idx + 1]
+        if "--json" in sys.argv:
+            return int(render_json(tools_profiles_command_result(config, tools_config_path)))
+        cmd_tools_profiles(config, tools_config_path=tools_config_path)
+    else:
+        print(f"  未知子命令: tools {sub}", file=sys.stderr)
+        return 2
+    return 0
+
+
+def _cmd_skills_handler(config: Any) -> int:
+    sub = sys.argv[2] if len(sys.argv) > 2 else "list"
+    if sub == "list":
+        if "--json" in sys.argv:
+            return int(render_json(skills_list_command_result(config)))
+        cmd_skills_list(config)
+    elif sub == "view":
+        if len(sys.argv) < 4:
+            print("  用法: lbs skills view <name>", file=sys.stderr)
+            return 2
+        cmd_skills_view(config, sys.argv[3])
+    else:
+        print(f"  未知子命令: skills {sub}", file=sys.stderr)
+        return 2
+    return 0
+
+
+def _cmd_mcp_handler(config: Any) -> int:
+    sub = sys.argv[2] if len(sys.argv) > 2 else "list"
+    if sub in ("list", "ls"):
+        cmd_mcp_list(config)
+    elif sub == "add":
+        if len(sys.argv) < 5:
+            print("  用法: lbs mcp add <name> <type> <config_json>", file=sys.stderr)
+            return 2
+        cmd_mcp_add(config, sys.argv[3], sys.argv[4], " ".join(sys.argv[5:]))
+    elif sub in ("remove", "rm"):
+        if len(sys.argv) < 4:
+            print("  用法: lbs mcp remove <name>", file=sys.stderr)
+            return 2
+        return cmd_mcp_remove(config, sys.argv[3])
+    elif sub in ("test", "ping"):
+        if len(sys.argv) < 4:
+            print("  用法: lbs mcp test <name>", file=sys.stderr)
+            return 2
+        return cmd_mcp_test(config, sys.argv[3])
+    elif sub == "status":
+        from application.container import AgentSystemContainer
+        from interfaces.cli._impl.commands.common import project_root
+        container = AgentSystemContainer(config=config, project_root=project_root)
+        result = container.mcp_service().status()
+        print("MCP Status")
+        print("  --------------------------------------------------------")
+        for item in result:
+            if isinstance(item, dict):
+                name = item.get("name", "?")
+                running = item.get("running", False)
+                tool_count = item.get("tool_count", 0)
+                print(f"  - {name} [{'running' if running else 'stopped'}] tools={tool_count}")
+            else:
+                print(f"  - {item}")
+    elif sub == "connect":
+        if len(sys.argv) < 4:
+            print("  用法: lbs mcp connect <name>", file=sys.stderr); return 2
+        from application.container import AgentSystemContainer
+        from interfaces.cli._impl.commands.common import project_root
+        container = AgentSystemContainer(config=config, project_root=project_root)
+        result = container.mcp_service().connect(sys.argv[3])
+        if result.get("ok"):
+            tools = result.get("tools", [])
+            print(f"  连接成功: {sys.argv[3]} ({len(tools)} tools)")
+            for t in tools:
+                print(f"    - {t.get('name', '?')}: {t.get('description', '')[:60]}")
+        else:
+            print(f"  连接失败: {sys.argv[3]}", file=sys.stderr); return 1
+    elif sub == "toggle":
+        if len(sys.argv) < 5:
+            print("  用法: lbs mcp toggle <name> <on|off>", file=sys.stderr); return 2
+        from application.container import AgentSystemContainer
+        from interfaces.cli._impl.commands.common import project_root
+        container = AgentSystemContainer(config=config, project_root=project_root)
+        enabled = sys.argv[4] in ("on", "true", "1", "enable")
+        container.mcp_service().set_enabled(sys.argv[3], enabled)
+        print(f"  {sys.argv[3]}: {'enabled' if enabled else 'disabled'}")
+    else:
+        print(f"  未知子命令: mcp {sub}", file=sys.stderr)
+        return 2
+    return 0
+
+
+def _cmd_sessions_handler(config: Any) -> int:
+    sub = sys.argv[2] if len(sys.argv) > 2 else "list"
+    if sub == "list":
+        cmd_sessions_list(config)
+    elif sub == "show":
+        if len(sys.argv) < 4:
+            print("  用法: lbs sessions show <session_id>", file=sys.stderr)
+            return 2
+        cmd_sessions_show(config, sys.argv[3])
+    elif sub == "export":
+        if len(sys.argv) < 5:
+            print("  用法: lbs sessions export <session_id> <output.json>", file=sys.stderr)
+            return 2
+        cmd_sessions_export(config, sys.argv[3], sys.argv[4])
+    elif sub == "report":
+        cmd_sessions_report(config, sys.argv[3] if len(sys.argv) >= 4 else "")
+    elif sub == "delete":
+        if len(sys.argv) < 4:
+            print("  用法: lbs sessions delete <session_id>", file=sys.stderr)
+            return 2
+        cmd_sessions_delete(config, sys.argv[3])
+    elif sub == "prune":
+        try:
+            days = int(sys.argv[3]) if len(sys.argv) > 3 else 30
+        except ValueError:
+            print("  用法: lbs sessions prune [days]  # days 必须是整数", file=sys.stderr)
+            return 2
+        cmd_sessions_prune(config, days)
+    elif sub == "stats":
+        cmd_sessions_stats(config)
+    elif sub == "rename":
+        if len(sys.argv) < 5:
+            print("  用法: lbs sessions rename <session_id> <new_name>", file=sys.stderr)
+            return 2
+        cmd_session_rename(config, sys.argv[3], sys.argv[4])
+    elif sub == "import":
+        if len(sys.argv) < 4:
+            print("  用法: lbs sessions import <file.json>", file=sys.stderr)
+            return 2
+        cmd_session_import(config, sys.argv[3])
+    else:
+        print(f"  未知子命令: sessions {sub}", file=sys.stderr)
+        return 2
+    return 0
+
+
+def _cmd_workers_handler(config: Any) -> int:
+    sub = sys.argv[2] if len(sys.argv) > 2 else "list"
+    if sub == "list":
+        cmd_workers_list(config)
+    elif sub == "show":
+        if len(sys.argv) < 4:
+            print("  用法: lbs workers show <session_id>", file=sys.stderr)
+            return 2
+        cmd_workers_show(config, sys.argv[3])
+    else:
+        print(f"  未知子命令: workers {sub}", file=sys.stderr)
+        return 2
+    return 0
+
+
+def _cmd_compress_handler(config: Any) -> int:
+    target = sys.argv[2] if len(sys.argv) > 2 else "current"
+    cmd_compress(config, target)
+    return 0
+
+
+def _cmd_architecture_handler(config: Any) -> int:
+    sub = sys.argv[2] if len(sys.argv) > 2 else "inspect"
+    if sub != "inspect":
+        print(f"  未知子命令: architecture {sub}", file=sys.stderr)
+        return 2
+    if "--json" in sys.argv:
+        return int(render_json(architecture_command_result(config)))
+    print_architecture_report(config)
+    return 0
+
+
+def _cmd_cron_handler(config: Any) -> int:
+    sub = sys.argv[2] if len(sys.argv) > 2 else "list"
+    if sub == "list":
+        cmd_cron_list()
+    elif sub == "create":
+        import argparse as _ap
+        _p = _ap.ArgumentParser()
+        _p.add_argument("--name", default="")
+        _p.add_argument("--schedule", default="")
+        _p.add_argument("--agent", default="")
+        _p.add_argument("--task", default="")
+        _p.add_argument("--context", default="")
+        _args = _p.parse_args(sys.argv[3:])
+        cmd_cron_create(_args)
+    elif sub == "remove":
+        cmd_cron_remove(type('A', (), {'job_id': sys.argv[3] if len(sys.argv) > 3 else ''})())
+    elif sub == "pause":
+        cmd_cron_pause(type('A', (), {'job_id': sys.argv[3] if len(sys.argv) > 3 else ''})())
+    elif sub == "resume":
+        cmd_cron_resume(type('A', (), {'job_id': sys.argv[3] if len(sys.argv) > 3 else ''})())
+    elif sub == "trigger":
+        cmd_cron_trigger(type('A', (), {'job_id': sys.argv[3] if len(sys.argv) > 3 else ''})())
+    elif sub == "tick":
+        cmd_cron_tick()
+    elif sub == "status":
+        cmd_cron_status()
+    elif sub in ("logs", "log"):
+        cmd_cron_logs(" ".join(sys.argv[3:]))
+    else:
+        print(f"  未知子命令: cron {sub}", file=sys.stderr)
+        return 2
+    return 0
+
+
+def _cmd_soul_handler(config: Any) -> int:
+    sub = sys.argv[2] if len(sys.argv) > 2 else "list"
+    if sub == "list":
+        cmd_soul_list()
+    elif sub == "switch":
+        cmd_soul_switch(type('A', (), {'name': sys.argv[3] if len(sys.argv) > 3 else ''})())
+    elif sub == "show":
+        cmd_soul_show(type('A', (), {'name': sys.argv[3] if len(sys.argv) > 3 else ''})())
+    else:
+        print(f"  未知子命令: soul {sub}", file=sys.stderr)
+        return 2
+    return 0
+
+
+def _cmd_usage_handler(config: Any) -> int:
+    container = AgentSystemContainer(config=config)
+    summary = container.model_service().usage_summary()
+    is_cost = len(sys.argv) > 2 and sys.argv[2] == "cost"
+
+    if is_cost:
+        total = float(summary.get("total_cost_usd", 0))
+        print()
+        print("  成本分析")
+        _bar()
+        print(f"  总成本: ${total:.4f}")
+        print()
+        by_model = summary.get("by_model", {})
+        if by_model:
+            print("  按模型:")
+            print(f"  {'模型':<25} {'调用':>6} {'输入':>10} {'输出':>10} {'成本':>10}")
+            for m, d in sorted(by_model.items(), key=lambda x: -float(x[1].get("cost_usd", 0))):
+                cost = float(d.get("cost_usd", 0))
+                print(f"  {m:<25} {d.get('calls',0):>6} {d.get('input_tokens',0):>10,} {d.get('output_tokens',0):>10,} ${cost:.4f}")
+        by_agent = summary.get("by_agent", {})
+        if by_agent:
+            print()
+            print("  按 Agent:")
+            print(f"  {'Agent':<25} {'调用':>6} {'成本':>10}")
+            for agent, d in sorted(by_agent.items(), key=lambda x: -float(x[1].get("cost_usd", 0))):
+                cost = float(d.get("cost_usd", 0))
+                print(f"  {agent:<25} {d.get('calls',0):>6} ${cost:.4f}")
+        print()
+    else:
+        stats = container.session_service().store.session_stats()
+        print()
+        print("  用量分析")
+        _bar()
+        print(f"  LLM 调用:     {summary.get('total_calls', 0)}")
+        print(f"  输入 tokens:  {summary.get('total_input_tokens', 0):,}")
+        print(f"  输出 tokens:  {summary.get('total_output_tokens', 0):,}")
+        print(f"  总 tokens:    {summary.get('total_tokens', 0):,}")
+        print(f"  总成本:       ${summary.get('total_cost_usd', '0')}")
+        by_model = summary.get("by_model", {})
+        if by_model:
+            print(f"\n  按模型:")
+            for model, d in sorted(by_model.items()):
+                print(f"    {model:25s} {d.get('calls',0):3d} calls  {d.get('input_tokens',0):>8,}+{d.get('output_tokens',0):>8,}  ${d.get('cost_usd', '0')}")
+        by_agent = summary.get("by_agent", {})
+        if by_agent:
+            print(f"\n  按 Agent:")
+            for agent, d in sorted(by_agent.items()):
+                print(f"    {agent:25s} {d.get('calls',0):3d} calls  ${d.get('cost_usd', '0')}")
+        print(f"\n  会话统计: {stats.get('total', 0)} 个会话, {stats.get('total_iterations', 0)} 迭代, {stats.get('total_tool_calls', 0)} 工具调用")
+        print()
+    return 0
+
+
+def _cmd_keys_handler(config: Any) -> int:
+    sub = sys.argv[2] if len(sys.argv) > 2 else "list"
+    if sub in ("list", "ls"):
+        cmd_model_keys(config)
+    elif sub == "add":
+        if len(sys.argv) < 6:
+            print("  用法: lbs keys add <provider> <key_name> <key_value>", file=sys.stderr)
+            return 2
+        cmd_key_add(config, sys.argv[3], sys.argv[4], sys.argv[5])
+    elif sub == "rotate":
+        if len(sys.argv) < 5:
+            print("  用法: lbs keys rotate <provider> <key_name>", file=sys.stderr)
+            return 2
+        cmd_key_rotate(config, sys.argv[3], sys.argv[4])
+    else:
+        print(f"  未知子命令: keys {sub}", file=sys.stderr)
+        return 2
+    return 0
+
+
+def _cmd_config_handler(config: Any) -> int:
+    sub = sys.argv[2] if len(sys.argv) > 2 else "show"
+    if sub == "show":
+        cmd_config(config)
+    elif sub == "set":
+        if len(sys.argv) < 5:
+            print("  用法: lbs config set <key> <value>", file=sys.stderr)
+            print("  示例: lbs config set default_model gpt-4", file=sys.stderr)
+            return 2
+        cmd_config_set(config, sys.argv[3], sys.argv[4])
+    else:
+        print(f"  未知子命令: config {sub}", file=sys.stderr)
+        return 2
+    return 0
+
+
+def _cmd_skill_handler(config: Any) -> int:
+    sub = sys.argv[2] if len(sys.argv) > 2 else "list"
+    query = sys.argv[3] if len(sys.argv) > 3 else ""
+    category = sys.argv[3] if len(sys.argv) > 3 and sub in ("activate", "deactivate") else ""
+    name = sys.argv[4] if len(sys.argv) > 4 and sub in ("activate", "deactivate") else ""
+    source = sys.argv[5] if len(sys.argv) > 5 and sub == "activate" else "local"
+    repo_url = ""
+    alias = ""
+    sync_mode = "clone"
+    if sub == "sync":
+        args = sys.argv[3:]
+        if args and args[0] == "--pull":
+            sync_mode = "pull"
+            alias = args[1] if len(args) > 1 else ""
+        elif args and args[0] == "--remove":
+            sync_mode = "remove"
+            alias = args[1] if len(args) > 1 else ""
+        else:
+            repo_url = args[0] if len(args) > 0 else ""
+            alias = args[1] if len(args) > 1 else ""
+    return cmd_skill(
+        config,
+        sub=sub,
+        query=query,
+        category=category,
+        name=name,
+        source=source,
+        repo_url=repo_url,
+        alias=alias,
+        sync_mode=sync_mode,
+    )
+
+
+def _cmd_gateway_handler(config: Any) -> int:
+    sub = sys.argv[2] if len(sys.argv) > 2 else "list"
+    platform = sys.argv[3] if len(sys.argv) > 3 else ""
+    target = sys.argv[4] if len(sys.argv) > 4 else ""
+    text = " ".join(sys.argv[5:]) if len(sys.argv) > 5 else ""
+    return cmd_gateway(config, sub=sub, platform=platform, target=target, text=text)
+
+
+def _cmd_doctor_handler(config: Any) -> int:
+    _json = "--json" in sys.argv
+    _live_provider = "--live-provider" in sys.argv
+    tools_config_path = ""
+    if "--tools-config" in sys.argv:
+        idx = sys.argv.index("--tools-config")
+        if idx + 1 >= len(sys.argv):
+            print("  用法: lbs doctor [--json] [--live-provider] [--tools-config <path>]", file=sys.stderr)
+            return 2
+        tools_config_path = sys.argv[idx + 1]
+    report = cmd_doctor(
+        config,
+        json_mode=_json,
+        live_provider=_live_provider,
+        tools_config_path=tools_config_path,
+    )
+    return 1 if report.get("failed", 0) else 0
+
+
+# --- 注册表实例 ---
+
+_REGISTRY = CommandRegistry()
+
+
+def _build_registry() -> None:
+    _REGISTRY.register("help", lambda _config: cmd_help(), aliases=["-h", "--help"], help="显示帮助")
+    _REGISTRY.register("model", _cmd_model_handler, help="模型操作（list/switch/health/keys）")
+    _REGISTRY.register("models", _cmd_models_handler, aliases=["m"], help="查看模型供应商")
+    _REGISTRY.register("llm", _cmd_llm_handler, help="LLM 巡检/验证/诊断/路由")
+    _REGISTRY.register("orchestrator", _cmd_orchestrator_handler, aliases=["orch"], help="编排层巡检/预览（inspect/dry-run <agent> <task>）")
+    _REGISTRY.register("run", _cmd_run_handler, help="运行默认 Agent")
+    _REGISTRY.register("chat", _cmd_chat_handler, help="Chat 模式")
+    _REGISTRY.register("run-agent", _cmd_run_agent_handler, help="启动指定 agent 自循环")
+    _REGISTRY.register("tools", _cmd_tools_handler, help="工具层操作")
+    _REGISTRY.register("skills", _cmd_skills_handler, help="Skills 查看")
+    _REGISTRY.register("mcp", _cmd_mcp_handler, help="MCP server 管理")
+    _REGISTRY.register("sessions", _cmd_sessions_handler, aliases=["session"], help="会话管理")
+    _REGISTRY.register("workers", _cmd_workers_handler, aliases=["worker"], help="Worker 管理")
+    _REGISTRY.register("status", cmd_status, help="运行状态快照")
+    _REGISTRY.register("compress", _cmd_compress_handler, help="上下文压缩")
+    _REGISTRY.register("agents", cmd_agents, aliases=["a"], help="查看子代理配置")
+    _REGISTRY.register("architecture", _cmd_architecture_handler, help="架构巡检")
+    _REGISTRY.register("cron", _cmd_cron_handler, help="定时任务管理")
+    _REGISTRY.register("soul", _cmd_soul_handler, help="Soul 管理")
+    _REGISTRY.register("usage", _cmd_usage_handler, help="用量/成本分析")
+    _REGISTRY.register("keys", _cmd_keys_handler, aliases=["key"], help="API Key 管理")
+    _REGISTRY.register("config", _cmd_config_handler, aliases=["cfg"], help="配置查看/设置")
+    _REGISTRY.register("skill", _cmd_skill_handler, help="项目级 Skill 管理")
+    _REGISTRY.register("gateway", _cmd_gateway_handler, help="网关操作")
+    _REGISTRY.register("doctor", _cmd_doctor_handler, help="诊断配置问题")
+    _REGISTRY.register("smoke-test", lambda _config: smoke_main(), aliases=["smoke"], help="快速冒烟测试")
+
+
+# --- 入口 ---
+
+
+def main() -> int:
+    _build_registry()
+
+    try:
+        from application.onboarding import show_onboarding_if_needed
+        hint = show_onboarding_if_needed("cli")
+        if hint:
+            print(hint)
+    except Exception:
+        pass
+
+    logging.basicConfig(level=logging.WARNING, format="  %(message)s")
+
+    if len(sys.argv) < 2:
+        cmd_help()
+        return 0
+
+    cmd = sys.argv[1]
+    config = load_config()
+
+    handler = _REGISTRY.resolve(cmd)
+    if handler is not None:
+        return handler(config)
+
+    print(f"  未知命令: {cmd}", file=sys.stderr)
+    cmd_help()
+    return 2
+
+
+if __name__ == "__main__":
+    sys.exit(main())