code-puppy 0.0.169__py3-none-any.whl → 0.0.366__py3-none-any.whl

code_puppy/command_line/mcp/add_command.py

@@ -1,183 +0,0 @@
-"""
-MCP Add Command - Adds new MCP servers from JSON configuration or wizard.
-"""
-
-import json
-import logging
-import os
-from typing import List, Optional
-
-from code_puppy.messaging import emit_info
-from code_puppy.state_management import is_tui_mode
-
-from .base import MCPCommandBase
-from .wizard_utils import run_interactive_install_wizard
-
-# Configure logging
-logger = logging.getLogger(__name__)
-
-
-class AddCommand(MCPCommandBase):
-    """
-    Command handler for adding MCP servers.
-
-    Adds new MCP servers from JSON configuration or interactive wizard.
-    """
-
-    def execute(self, args: List[str], group_id: Optional[str] = None) -> None:
-        """
-        Add a new MCP server from JSON configuration or launch wizard.
-
-        Usage:
-            /mcp add                    - Launch interactive wizard
-            /mcp add <json>             - Add server from JSON config
-
-        Example JSON:
-            /mcp add {"name": "test", "type": "stdio", "command": "echo", "args": ["hello"]}
-
-        Args:
-            args: Command arguments - JSON config or empty for wizard
-            group_id: Optional message group ID for grouping related messages
-        """
-        if group_id is None:
-            group_id = self.generate_group_id()
-
-        # Check if in TUI mode and guide user to use Ctrl+T instead
-        if is_tui_mode() and not args:
-            emit_info(
-                "💡 In TUI mode, press Ctrl+T to open the MCP Install Wizard",
-                message_group=group_id,
-            )
-            emit_info(
-                "   The wizard provides a better interface for browsing and installing MCP servers.",
-                message_group=group_id,
-            )
-            return
-
-        try:
-            if args:
-                # Parse JSON from arguments
-                json_str = " ".join(args)
-
-                try:
-                    config_dict = json.loads(json_str)
-                except json.JSONDecodeError as e:
-                    emit_info(f"Invalid JSON: {e}", message_group=group_id)
-                    emit_info(
-                        "Usage: /mcp add <json> or /mcp add (for wizard)",
-                        message_group=group_id,
-                    )
-                    emit_info(
-                        'Example: /mcp add {"name": "test", "type": "stdio", "command": "echo"}',
-                        message_group=group_id,
-                    )
-                    return
-
-                # Validate required fields
-                if "name" not in config_dict:
-                    emit_info("Missing required field: 'name'", message_group=group_id)
-                    return
-                if "type" not in config_dict:
-                    emit_info("Missing required field: 'type'", message_group=group_id)
-                    return
-
-                # Add the server
-                success = self._add_server_from_json(config_dict, group_id)
-
-                if success:
-                    # Reload MCP servers
-                    try:
-                        from code_puppy.agent import reload_mcp_servers
-
-                        reload_mcp_servers()
-                    except ImportError:
-                        pass
-
-                    emit_info(
-                        "Use '/mcp list' to see all servers", message_group=group_id
-                    )
-
-            else:
-                # No arguments - launch interactive wizard with server templates
-                success = run_interactive_install_wizard(self.manager, group_id)
-
-                if success:
-                    # Reload the agent to pick up new server
-                    try:
-                        from code_puppy.agent import reload_mcp_servers
-
-                        reload_mcp_servers()
-                    except ImportError:
-                        pass
-
-        except ImportError as e:
-            logger.error(f"Failed to import: {e}")
-            emit_info("Required module not available", message_group=group_id)
-        except Exception as e:
-            logger.error(f"Error in add command: {e}")
-            emit_info(f"[red]Error adding server: {e}[/red]", message_group=group_id)
-
-    def _add_server_from_json(self, config_dict: dict, group_id: str) -> bool:
-        """
-        Add a server from JSON configuration.
-
-        Args:
-            config_dict: Server configuration dictionary
-            group_id: Message group ID
-
-        Returns:
-            True if successful, False otherwise
-        """
-        try:
-            from code_puppy.config import MCP_SERVERS_FILE
-            from code_puppy.mcp.managed_server import ServerConfig
-
-            # Extract required fields
-            name = config_dict.pop("name")
-            server_type = config_dict.pop("type")
-            enabled = config_dict.pop("enabled", True)
-
-            # Everything else goes into config
-            server_config = ServerConfig(
-                id=f"{name}_{hash(name)}",
-                name=name,
-                type=server_type,
-                enabled=enabled,
-                config=config_dict,  # Remaining fields are server-specific config
-            )
-
-            # Register the server
-            server_id = self.manager.register_server(server_config)
-
-            if not server_id:
-                emit_info(f"Failed to add server '{name}'", message_group=group_id)
-                return False
-
-            emit_info(
-                f"✅ Added server '{name}' (ID: {server_id})", message_group=group_id
-            )
-
-            # Save to mcp_servers.json for persistence
-            if os.path.exists(MCP_SERVERS_FILE):
-                with open(MCP_SERVERS_FILE, "r") as f:
-                    data = json.load(f)
-                    servers = data.get("mcp_servers", {})
-            else:
-                servers = {}
-                data = {"mcp_servers": servers}
-
-            # Add new server
-            servers[name] = config_dict.copy()
-            servers[name]["type"] = server_type
-
-            # Save back
-            os.makedirs(os.path.dirname(MCP_SERVERS_FILE), exist_ok=True)
-            with open(MCP_SERVERS_FILE, "w") as f:
-                json.dump(data, f, indent=2)
-
-            return True
-
-        except Exception as e:
-            logger.error(f"Error adding server from JSON: {e}")
-            emit_info(f"[red]Failed to add server: {e}[/red]", message_group=group_id)
-            return False

code_puppy/command_line/meta_command_handler.py

@@ -1,153 +0,0 @@
-import os
-
-from rich.console import Console
-
-from code_puppy.command_line.model_picker_completion import (
-    load_model_names,
-    update_model_in_input,
-)
-from code_puppy.config import get_config_keys
-from code_puppy.command_line.utils import make_directory_table
-from code_puppy.command_line.motd import print_motd
-
-META_COMMANDS_HELP = """
-[bold magenta]Meta Commands Help[/bold magenta]
-~help, ~h             Show this help message
-~cd <dir>             Change directory or show directories
-~m <model>            Set active model
-~motd                 Show the latest message of the day (MOTD)
-~show                 Show puppy config key-values
-~set                  Set puppy config key-values (message_limit, protected_token_count, compaction_threshold, allow_recursion, etc.)
-~<unknown>            Show unknown meta command warning
-"""
-
-
-def handle_meta_command(command: str, console: Console) -> bool:
-    """
-    Handle meta/config commands prefixed with '~'.
-    Returns True if the command was handled (even if just an error/help), False if not.
-    """
-    command = command.strip()
-
-    if command.strip().startswith("~motd"):
-        print_motd(console, force=True)
-        return True
-
-    if command.startswith("~cd"):
-        tokens = command.split()
-        if len(tokens) == 1:
-            try:
-                table = make_directory_table()
-                console.print(table)
-            except Exception as e:
-                console.print(f"[red]Error listing directory:[/red] {e}")
-            return True
-        elif len(tokens) == 2:
-            dirname = tokens[1]
-            target = os.path.expanduser(dirname)
-            if not os.path.isabs(target):
-                target = os.path.join(os.getcwd(), target)
-            if os.path.isdir(target):
-                os.chdir(target)
-                console.print(
-                    f"[bold green]Changed directory to:[/bold green] [cyan]{target}[/cyan]"
-                )
-            else:
-                console.print(f"[red]Not a directory:[/red] [bold]{dirname}[/bold]")
-            return True
-
-    if command.strip().startswith("~show"):
-        from code_puppy.command_line.model_picker_completion import get_active_model
-        from code_puppy.config import (
-            get_owner_name,
-            get_puppy_name,
-            get_yolo_mode,
-            get_message_limit,
-        )
-
-        puppy_name = get_puppy_name()
-        owner_name = get_owner_name()
-        model = get_active_model()
-        yolo_mode = get_yolo_mode()
-        msg_limit = get_message_limit()
-        console.print(f"""[bold magenta]🐶 Puppy Status[/bold magenta]
-
-[bold]puppy_name:[/bold]     [cyan]{puppy_name}[/cyan]
-[bold]owner_name:[/bold]     [cyan]{owner_name}[/cyan]
-[bold]model:[/bold]          [green]{model}[/green]
-[bold]YOLO_MODE:[/bold]      {"[red]ON[/red]" if yolo_mode else "[yellow]off[/yellow]"}
-[bold]message_limit:[/bold]   [cyan]{msg_limit}[/cyan] requests per minute
-""")
-        return True
-
-    if command.startswith("~set"):
-        # Syntax: ~set KEY=VALUE or ~set KEY VALUE
-        from code_puppy.config import set_config_value
-
-        tokens = command.split(None, 2)
-        argstr = command[len("~set") :].strip()
-        key = None
-        value = None
-        if "=" in argstr:
-            key, value = argstr.split("=", 1)
-            key = key.strip()
-            value = value.strip()
-        elif len(tokens) >= 3:
-            key = tokens[1]
-            value = tokens[2]
-        elif len(tokens) == 2:
-            key = tokens[1]
-            value = ""
-        else:
-            console.print(
-                f"[yellow]Usage:[/yellow] ~set KEY=VALUE or ~set KEY VALUE\nConfig keys: {', '.join(get_config_keys())}"
-            )
-            return True
-        if key:
-            set_config_value(key, value)
-            console.print(
-                f'[green]🌶 Set[/green] [cyan]{key}[/cyan] = "{value}" in puppy.cfg!'
-            )
-        else:
-            console.print("[red]You must supply a key.[/red]")
-        return True
-
-    if command.startswith("~m"):
-        # Try setting model and show confirmation
-        new_input = update_model_in_input(command)
-        if new_input is not None:
-            from code_puppy.command_line.model_picker_completion import get_active_model
-            from code_puppy.agents.runtime_manager import get_runtime_agent_manager
-
-            model = get_active_model()
-            # Make sure this is called for the test
-            manager = get_runtime_agent_manager()
-            manager.reload_agent()
-            console.print(
-                f"[bold green]Active model set and loaded:[/bold green] [cyan]{model}[/cyan]"
-            )
-            return True
-        # If no model matched, show available models
-        model_names = load_model_names()
-        console.print("[yellow]Usage:[/yellow] ~m <model-name>")
-        console.print(f"[yellow]Available models:[/yellow] {', '.join(model_names)}")
-        return True
-    if command in ("~help", "~h"):
-        console.print(META_COMMANDS_HELP)
-        return True
-    if command.startswith("~"):
-        name = command[1:].split()[0] if len(command) > 1 else ""
-        if name:
-            console.print(
-                f"[yellow]Unknown meta command:[/yellow] {command}\n[dim]Type ~help for options.[/dim]"
-            )
-        else:
-            # Show current model ONLY here
-            from code_puppy.command_line.model_picker_completion import get_active_model
-
-            current_model = get_active_model()
-            console.print(
-                f"[bold green]Current Model:[/bold green] [cyan]{current_model}[/cyan]"
-            )
-        return True
-    return False