bone-agent 1.4.0 → 2.0.0

package/src/ui/setup_wizard.py

@@ -1,294 +0,0 @@
-"""Interactive first-run setup wizard.
-
-Guides new users through provider selection, API key entry,
-Obsidian vault configuration, and optional settings using
-prompt_toolkit prompts and rich console output.
-"""
-
-from pathlib import Path
-from typing import Optional
-
-from prompt_toolkit import prompt as pt_prompt
-from prompt_toolkit.formatted_text import FormattedText
-from rich.console import Console
-from rich.panel import Panel
-from rich.text import Text
-
-from llm import config as llm_config
-from core.config_manager import ConfigManager
-
-# Provider metadata for wizard display
-WIZARD_PROVIDERS = [
-    ("bone", "bone-agent (free tier — no key needed)", "api"),
-    ("openai", "OpenAI", "api"),
-    ("anthropic", "Anthropic (Claude)", "api"),
-    ("gemini", "Google Gemini", "api"),
-    ("openrouter", "OpenRouter (multi-provider)", "api"),
-    ("glm", "GLM (Zhipu AI)", "api"),
-    ("kimi", "Kimi (Moonshot AI)", "api"),
-    ("minimax", "MiniMax", "api"),
-    ("local", "Local model (llama.cpp)", "local"),
-]
-
-# API providers that need a key (skip bone free tier and local)
-_API_PROVIDERS = [p for p in WIZARD_PROVIDERS if p[2] == "api" and p[0] != "bone"]
-
-
-def _resolve_config_path() -> Path:
-    """Resolve config.yaml path (delegates to llm_config)."""
-    return llm_config.resolve_config_path()
-
-
-def is_first_run() -> bool:
-    """Return True if config.yaml does not exist."""
-    return not _resolve_config_path().exists()
-
-
-def _prompt(console: Console, message: str, default: str = "", password: bool = False) -> str:
-    """Prompt the user using prompt_toolkit with rich-styled message.
-
-    Falls back to simple input() if prompt_toolkit fails.
-    """
-    try:
-        suffix = f" [{default}]" if default else ""
-        if password:
-            return pt_prompt(
-                FormattedText([("class:prompt", f"{message}{suffix}: ")]),
-                is_password=True,
-            ).strip() or default
-        return pt_prompt(
-            FormattedText([("class:prompt", f"{message}{suffix}: ")]),
-        ).strip() or default
-    except (EOFError, KeyboardInterrupt):
-        raise
-    except Exception:
-        # Fallback for environments where prompt_toolkit misbehaves
-        suffix = f" [{default}]" if default else ""
-        if password:
-            import getpass
-            return getpass.getpass(f"{message}{suffix}: ").strip() or default
-        return input(f"{message}{suffix}: ").strip() or default
-
-
-def _select_provider(console: Console) -> str:
-    """Interactive provider selection. Returns provider ID."""
-    console.print()
-    console.print(Panel(
-        Text.from_markup("[bold #5F9EA0]Select a provider[/bold #5F9EA0]\n\n"
-                         "Enter the number or provider name."),
-        title="Step 1 of 3",
-        border_style="grey23",
-        padding=(0, 2),
-        title_align="left",
-    ))
-
-    for i, (pid, label, ptype) in enumerate(WIZARD_PROVIDERS, 1):
-        marker = "[dim]free[/dim]" if pid == "bone" else ""
-        console.print(f"  [bold]{i:>2}[/bold]. {label} {marker}")
-
-    console.print()
-
-    while True:
-        choice = _prompt(console, "Provider", default="1")
-        # Try numeric lookup
-        try:
-            idx = int(choice)
-            if 1 <= idx <= len(WIZARD_PROVIDERS):
-                selected = WIZARD_PROVIDERS[idx - 1]
-                console.print(f"  [green]Selected:[/green] {selected[1]}")
-                return selected[0]
-        except ValueError:
-            pass
-        # Try name lookup
-        choice_lower = choice.lower().strip()
-        for pid, label, ptype in WIZARD_PROVIDERS:
-            if choice_lower in (pid, pid.split("_")[0]):
-                console.print(f"  [green]Selected:[/green] {label}")
-                return pid
-
-        console.print(f"  [red]Invalid choice: '{choice}'. Enter a number or provider name.[/red]")
-
-
-def _prompt_api_key(console: Console, provider_id: str) -> str:
-    """Prompt for API key if the provider requires one."""
-    # bone free tier and local don't need a key
-    if provider_id == "bone":
-        console.print("\n  [dim]bone-agent free tier — no API key required.[/dim]")
-        return ""
-    if provider_id == "local":
-        console.print("\n  [dim]Local model — no API key required.[/dim]")
-        return ""
-
-    provider_label = next((label for pid, label, _ in WIZARD_PROVIDERS if pid == provider_id), provider_id)
-    console.print()
-    console.print(Panel(
-        Text.from_markup(f"Enter your [bold]{provider_label}[/bold] API key."),
-        title="Step 2 of 3",
-        border_style="grey23",
-        padding=(0, 2),
-        title_align="left",
-    ))
-
-    key = _prompt(console, "API key", password=True)
-    if not key:
-        console.print("  [yellow]No key entered — you can set it later with [bold]/key[/bold].[/yellow]")
-    return key
-
-
-def _prompt_obsidian(console: Console) -> tuple[bool, str]:
-    """Prompt for Obsidian vault settings. Returns (enabled, vault_path)."""
-    console.print()
-    console.print(Panel(
-        Text.from_markup("Enable Obsidian vault integration?\n"
-                         "This lets bone-agent create project notes in your vault."),
-        title="Step 3 of 3",
-        border_style="grey23",
-        padding=(0, 2),
-        title_align="left",
-    ))
-
-    enable = _prompt(console, "Enable Obsidian? (y/n)", default="n").lower().startswith("y")
-
-    if not enable:
-        console.print("  [dim]Obsidian integration disabled. Enable later with [bold]/obsidian[/bold].[/dim]")
-        return (False, "")
-
-    while True:
-        vault_path = _prompt(console, "Vault path", default="~/Vault")
-        expanded = Path(vault_path).expanduser().resolve()
-
-        if expanded.is_dir():
-            # Check for .obsidian folder as validation
-            obsidian_marker = expanded / ".obsidian"
-            if obsidian_marker.is_dir():
-                console.print(f"  [green]Valid Obsidian vault detected.[/green]")
-            else:
-                console.print(f"  [yellow]Directory exists but no .obsidian/ folder found.[/yellow]")
-                console.print(f"  [yellow]Make sure this is your Obsidian vault root.[/yellow]")
-        else:
-            console.print(f"  [yellow]Directory does not exist yet: {expanded}[/yellow]")
-            console.print(f"  [yellow]It will be created when needed.[/yellow]")
-
-        confirm = _prompt(console, "Confirm this path? (y/n)", default="y").lower().startswith("y")
-        if confirm:
-            return (True, str(expanded))
-
-        retry = _prompt(console, "Try another path? (y/n)", default="y").lower().startswith("y")
-        if not retry:
-            return (False, "")
-
-
-def write_config(provider_id: str, api_key: str = "",
-                 obsidian_enabled: bool = False, obsidian_path: str = "") -> Path:
-    """Generate and write config.yaml from wizard responses.
-
-    Uses generate_config_template() for the base, then overlays
-    user selections.
-
-    Returns:
-        Path to the written config file.
-    """
-    config_data = llm_config.generate_config_template()
-
-    # Set the chosen provider
-    config_data["LAST_PROVIDER"] = provider_id
-
-    # Set API key if provided
-    if api_key:
-        # Map provider ID to its API key config field
-        key_map = {
-            "openrouter": "OPENROUTER_API_KEY",
-            "glm": "GLM_API_KEY",
-            "openai": "OPENAI_API_KEY",
-            "gemini": "GEMINI_API_KEY",
-            "minimax": "MINIMAX_API_KEY",
-            "anthropic": "ANTHROPIC_API_KEY",
-            "kimi": "KIMI_API_KEY",
-            "bone": "BONE_PROXY_API_KEY",
-        }
-        config_key = key_map.get(provider_id)
-        if config_key:
-            config_data[config_key] = api_key
-
-    # Set Obsidian settings
-    config_data["OBSIDIAN_SETTINGS"] = {
-        "enabled": obsidian_enabled,
-        "vault_path": obsidian_path,
-        "exclude_folders": ".obsidian,.trash,node_modules,.git,__pycache__",
-        "project_base": "Dev",
-    }
-
-    config_path = _resolve_config_path()
-    config_path.parent.mkdir(parents=True, exist_ok=True)
-
-    import yaml
-    with open(config_path, "w", encoding="utf-8-sig") as f:
-        yaml.dump(config_data, f, default_flow_style=False, sort_keys=False, allow_unicode=True)
-
-    return config_path
-
-
-def run_wizard(console: Optional[Console] = None) -> bool:
-    """Run the interactive setup wizard.
-
-    Args:
-        console: Rich console instance. Uses a new one if None.
-
-    Returns:
-        True if wizard completed successfully, False if user aborted.
-    """
-    if console is None:
-        console = Console()
-
-    console.print()
-    console.print(Panel(
-        Text.from_markup(
-            "[bold #5F9EA0]Welcome to bone-agent![/bold #5F9EA0]\n\n"
-            "Let's get you set up. This will take about a minute.\n"
-            "You can always change settings later with [bold]/config[/bold]."
-        ),
-        border_style="grey23",
-        padding=(0, 2),
-        title_align="left",
-    ))
-
-    try:
-        # Step 1: Provider
-        provider_id = _select_provider(console)
-
-        # Step 2: API key
-        api_key = _prompt_api_key(console, provider_id)
-
-        # Step 3: Obsidian
-        obsidian_enabled, obsidian_path = _prompt_obsidian(console)
-
-        # Write config
-        config_path = write_config(
-            provider_id=provider_id,
-            api_key=api_key,
-            obsidian_enabled=obsidian_enabled,
-            obsidian_path=obsidian_path,
-        )
-
-        console.print()
-        console.print(Panel(
-            Text.from_markup(
-                f"[bold green]Setup complete![/bold green]\n\n"
-                f"Config written to: [dim]{config_path}[/dim]\n"
-                f"Provider: [bold]{provider_id}[/bold]\n"
-                f"Obsidian: {'enabled' if obsidian_enabled else 'disabled'}\n\n"
-                f"[dim]Tip: Use [bold]/provider[/bold] to switch providers,[/dim]\n"
-                f"[dim]     [bold]/key[/bold] to update API keys,[/dim]\n"
-                f"[dim]     [bold]/obsidian[/bold] to reconfigure vault.[/dim]"
-            ),
-            border_style="grey23",
-            padding=(0, 2),
-            title_align="left",
-        ))
-        console.print()
-
-        return True
-
-    except (KeyboardInterrupt, EOFError):
-        console.print("\n[yellow]Setup cancelled.[/yellow]")
-        return False

package/src/ui/sub_agent_panel.py

@@ -1,234 +0,0 @@
-"""Live panel for streaming sub-agent tool output."""
-
-import logging
-import threading
-import time
-
-from rich.panel import Panel
-from rich.text import Text
-from rich.live import Live
-
-from core.tool_feedback import build_panel_tool_message
-
-logger = logging.getLogger(__name__)
-
-
-class SubAgentPanel:
-    """Live panel for streaming sub-agent tool output.
-
-    Displays a Rich panel with animated spinner, tool call log, and
-    completion/error status for sub-agent invocations.
-    """
-
-    _SPINNER_FRAMES = ["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"]
-
-    def __init__(self, query, console):
-        """Initialize the sub-agent panel.
-
-        Args:
-            query: The task query for the sub-agent
-            console: Rich console for display
-        """
-        self.console = console
-        self.query = query
-        self.tool_calls = []  # List of formatted Rich markup strings
-        self.total_tool_calls = 0
-        self.token_info = None  # Live token info string, e.g. "32k / 45k"
-        self._live = None
-        self._spinner_index = 0
-        self._show_spinner = True
-        self._spinner_thread = None
-        self._stop_spinner = threading.Event()
-        self._saved_termios = None
-
-    # ------------------------------------------------------------------
-    # Panel rendering
-    # ------------------------------------------------------------------
-
-    def _get_title(self):
-        """Get panel title with optional spinner and tool call counter."""
-        token_suffix = f" | {self.token_info}" if self.token_info else ""
-        if self._show_spinner:
-            spinner = self._SPINNER_FRAMES[self._spinner_index % len(self._SPINNER_FRAMES)]
-            return f"[#5F9EA0]{spinner} Sub-Agent ({self.total_tool_calls}){token_suffix}[/#5F9EA0]"
-        return f"[#5F9EA0]Sub-Agent ({self.total_tool_calls}){token_suffix}[/#5F9EA0]"
-
-    def _render_panel(self, title=None, border_style="#5F9EA0"):
-        """Render the current panel state.
-
-        Args:
-            title: Optional title override. If None, uses _get_title().
-            border_style: Border style (default: "#5F9EA0")
-
-        Returns:
-            Rich Panel object with current content and title
-        """
-        lines = [f"[bold #5F9EA0]Query:[/bold #5F9EA0] {self.query}", ""]
-
-        if self.tool_calls:
-            content = "\n".join(self.tool_calls)
-            lines.append(content)
-        else:
-            lines.append("[dim]No tools called yet[/dim]")
-
-        content = "\n".join(lines)
-        return Panel(
-            Text.from_markup(content, justify="left"),
-            title=title if title is not None else self._get_title(),
-            title_align="left",
-            border_style=border_style,
-            padding=(0, 1),
-        )
-
-    # ------------------------------------------------------------------
-    # Spinner animation
-    # ------------------------------------------------------------------
-
-    def _spin(self):
-        """Background thread: continuously increment spinner and update display."""
-        while not self._stop_spinner.is_set():
-            self._spinner_index += 1
-            if self._live:
-                self._live.update(self._render_panel())
-            time.sleep(0.1)  # 10 updates per second = smooth animation
-
-    # ------------------------------------------------------------------
-    # Terminal raw mode (suppress keystroke echoes during spinner)
-    # ------------------------------------------------------------------
-
-    @staticmethod
-    def _set_raw_mode():
-        """Switch stdin to raw mode to prevent keystroke echoes during spinner."""
-        import os
-        import sys
-        if os.name == 'nt':
-            return
-        try:
-            import termios
-            fd = sys.stdin.fileno()
-            old = termios.tcgetattr(fd)
-            new = old.copy()
-            new[3] &= ~(termios.ECHO | termios.ICANON | termios.IEXTEN)
-            new[0] &= ~(termios.ICRNL)
-            termios.tcsetattr(fd, termios.TCSANOW, new)
-            return old
-        except Exception:
-            return None
-
-    @staticmethod
-    def _restore_terminal_mode(saved):
-        """Restore terminal mode from saved termios attributes."""
-        import os
-        import sys
-        if saved is None:
-            return
-        try:
-            import os as _os
-            if _os.name == 'nt':
-                return
-        except Exception:
-            pass
-        try:
-            import termios
-            termios.tcsetattr(sys.stdin.fileno(), termios.TCSANOW, saved)
-        except Exception:
-            pass
-
-    # ------------------------------------------------------------------
-    # Context manager (Live display lifecycle)
-    # ------------------------------------------------------------------
-
-    def __enter__(self):
-        """Start Live display context.
-
-        Returns:
-            self for use in with statement
-        """
-        self._saved_termios = self._set_raw_mode()
-        panel = self._render_panel()
-        self._live = Live(panel, console=self.console, refresh_per_second=10)
-        self._live.__enter__()
-
-        # Start background spinner thread
-        self._spinner_thread = threading.Thread(target=self._spin, daemon=True)
-        self._spinner_thread.start()
-
-        return self
-
-    def __exit__(self, *args):
-        """End Live display context."""
-        self._stop_spinner.set()
-        if self._spinner_thread:
-            self._spinner_thread.join(timeout=0.5)
-        if self._live:
-            self._live.__exit__(*args)
-        self._restore_terminal_mode(self._saved_termios)
-        self._saved_termios = None
-
-    # ------------------------------------------------------------------
-    # Public API
-    # ------------------------------------------------------------------
-
-    def add_tool_call(self, tool_name, tool_result=None, command=None):
-        """Add a tool call message to the panel and refresh display.
-
-        Delegates formatting to core.tool_feedback.build_panel_tool_message
-        to avoid duplicating display logic.
-
-        Args:
-            tool_name: Name of the tool (e.g., "read_file", "rg")
-            tool_result: Optional tool result string (for detailed formatting)
-            command: Optional command string for context
-        """
-        self.total_tool_calls += 1
-        message = build_panel_tool_message(tool_name, tool_result, command)
-        self.tool_calls.append(message)
-        # Keep only last 5 tool calls
-        if len(self.tool_calls) > 5:
-            self.tool_calls.pop(0)
-        self._live.update(self._render_panel())
-
-    def append(self, text):
-        """Append text to panel and refresh display (kept for compatibility).
-
-        Args:
-            text: Text to append (may contain Rich markup)
-        """
-        # Just update panel to refresh title counter
-        self._live.update(self._render_panel())
-
-    def set_complete(self, usage=None):
-        """Mark panel as complete with optional token info.
-
-        Args:
-            usage: Optional dict with 'prompt', 'completion', 'total' token counts
-        """
-        self._show_spinner = False  # Stop spinner
-
-        # Build title with token usage: conversation length first, total billed second
-        if usage and usage.get('total_tokens'):
-            ctx_tokens = usage.get('context_tokens', 0)
-            total_tokens = usage.get('total_tokens', 0)
-            title = f"[green]✓ Sub-Agent Complete ({self.total_tool_calls}) - {ctx_tokens:,} curr, {total_tokens:,} total[/green]"
-        else:
-            title = f"[green]✓ Sub-Agent Complete ({self.total_tool_calls})[/green]"
-
-        self._live.update(self._render_panel(
-            title=title,
-            border_style="green"
-        ))
-
-    def set_error(self, message):
-        """Show error in panel with red styling.
-
-        Args:
-            message: Error message to display
-        """
-        self._show_spinner = False  # Stop spinner
-        self._live.update(Panel(
-            message,
-            title="[red]✗ Sub-Agent Error[/red]",
-            title_align="left",
-            border_style="red",
-            padding=(0, 1),
-        ))