agentpool 2.1.9__py3-none-any.whl → 2.5.0__py3-none-any.whl

agentpool_cli/ui.py

@@ -0,0 +1,557 @@
+"""UI commands for launching interactive interfaces."""
+
+from __future__ import annotations
+
+import signal
+import socket
+import subprocess
+import time
+from typing import Annotated
+
+import typer as t
+
+from agentpool_cli import log
+
+
+logger = log.get_logger(__name__)
+
+# Create UI subcommand group
+ui_app = t.Typer(help="Launch interactive user interfaces")
+
+
+@ui_app.command("opencode")
+def opencode_ui_command(  # noqa: PLR0915
+    config: Annotated[
+        str | None,
+        t.Argument(help="Path to agent configuration (optional, not used with --attach)"),
+    ] = None,
+    host: Annotated[
+        str,
+        t.Option("--host", "-h", help="Host to bind/connect to"),
+    ] = "127.0.0.1",
+    port: Annotated[
+        int,
+        t.Option("--port", "-p", help="Port for server to listen on / connect to"),
+    ] = 4096,
+    agent: Annotated[
+        str | None,
+        t.Option(
+            "--agent",
+            help="Name of specific agent to use (not used with --attach)",
+        ),
+    ] = None,
+    attach: Annotated[
+        bool,
+        t.Option("--attach", help="Only attach TUI to existing server (don't start server)"),
+    ] = False,
+) -> None:
+    """Launch OpenCode TUI with integrated server or attach to existing one.
+
+    By default, starts an OpenCode-compatible server in the background and
+    automatically attaches the OpenCode TUI to it. When you exit the TUI,
+    the server is automatically shut down.
+
+    With --attach, only launches the TUI and connects to an existing server
+    (useful when running the server separately or connecting from multiple clients).
+
+    Examples:
+        # Start server + TUI
+        agentpool ui opencode
+
+        # Use specific config and agent
+        agentpool ui opencode agents.yml --agent myagent
+
+        # Custom port
+        agentpool ui opencode --port 8080
+
+        # Attach to existing server (no server startup)
+        agentpool ui opencode --attach
+        agentpool ui opencode --attach --port 8080
+    """
+    url = f"http://{host}:{port}"
+
+    # Attach-only mode: just launch TUI
+    if attach:
+        logger.info("Attaching to existing OpenCode server", url=url)
+
+        # Clear screen for clean TUI
+        import os
+
+        os.system("clear" if os.name != "nt" else "cls")
+
+        result = subprocess.run(["opencode", "attach", url], check=False)
+        if result.returncode not in {0, 130}:  # 130 = Ctrl+C
+            logger.warning("OpenCode TUI exited with non-zero status", code=result.returncode)
+        return
+
+    # Build server command
+    server_cmd = [
+        "agentpool",
+        "serve-opencode",
+        "--host",
+        host,
+        "--port",
+        str(port),
+    ]
+    if config:
+        server_cmd.append(config)
+    if agent:
+        server_cmd.extend(["--agent", agent])
+
+    logger.info("Starting OpenCode server", url=url)
+
+    # Start server in background with suppressed output
+    server = subprocess.Popen(
+        server_cmd,
+        stdout=subprocess.DEVNULL,
+        stderr=subprocess.DEVNULL,
+    )
+
+    try:
+        # Wait for server to be ready with retry
+        max_retries = 30
+        for i in range(max_retries):
+            try:
+                sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+                sock.settimeout(0.5)
+                sock.connect((host, port))
+                sock.close()
+                logger.info("Server is ready", url=url)
+                break
+            except (TimeoutError, ConnectionRefusedError, OSError):
+                if i == max_retries - 1:
+                    msg = f"Server failed to start after {max_retries} attempts"
+                    raise RuntimeError(msg)  # noqa: B904
+                time.sleep(0.5)
+
+        # Give HTTP layer a moment to be fully ready
+        time.sleep(0.5)
+
+        # Clear screen before launching TUI
+        import os
+
+        os.system("clear" if os.name != "nt" else "cls")
+
+        # Attach TUI
+        result = subprocess.run(["opencode", "attach", url], check=False)
+        if result.returncode != 0:
+            logger.warning("OpenCode TUI exited with non-zero status", code=result.returncode)
+
+    except KeyboardInterrupt:
+        logger.info("UI interrupted by user")
+    except Exception as e:
+        logger.exception("Error running OpenCode UI")
+        raise t.Exit(1) from e
+    finally:
+        # Clean up server
+        logger.info("Shutting down server")
+        server.send_signal(signal.SIGTERM)
+        try:
+            server.wait(timeout=5)
+        except subprocess.TimeoutExpired:
+            logger.warning("Server did not shut down gracefully, killing")
+            server.kill()
+
+
+@ui_app.command("toad")
+def toad_ui_command(
+    config: Annotated[
+        str | None,
+        t.Argument(help="Path to agent configuration (optional)"),
+    ] = None,
+    websocket: Annotated[
+        bool,
+        t.Option("--websocket", "-w", help="Use WebSocket transport (otherwise stdio)"),
+    ] = False,
+    port: Annotated[
+        int,
+        t.Option("--port", "-p", help="Port for WebSocket server (only with --websocket)"),
+    ] = 8765,
+) -> None:
+    """Launch Toad TUI for ACP agents.
+
+    By default uses stdio transport where Toad spawns the agentpool server.
+    With --websocket, starts a WebSocket ACP server in the background first.
+
+    Examples:
+        # Direct stdio (Toad spawns server)
+        agentpool ui toad
+
+        # Use specific config
+        agentpool ui toad agents.yml
+
+        # WebSocket transport
+        agentpool ui toad --websocket
+
+        # WebSocket with custom port
+        agentpool ui toad --websocket --port 9000
+    """
+    if websocket:
+        _run_toad_websocket(config, port)
+    else:
+        _run_toad_stdio(config)
+
+
+def _run_toad_stdio(config: str | None) -> None:
+    """Run Toad with stdio transport (Toad spawns server)."""
+    # Build agentpool command that Toad will spawn
+    agentpool_cmd = "agentpool serve-acp"
+    if config:
+        agentpool_cmd += f" {config}"
+
+    # Clear screen for clean TUI
+    import os
+
+    os.system("clear" if os.name != "nt" else "cls")
+
+    # Run toad with agentpool as subprocess
+    result = subprocess.run(
+        ["uvx", "--from", "batrachian-toad@latest", "toad", "acp", agentpool_cmd],
+        check=False,
+    )
+
+    if result.returncode not in {0, 130}:  # 130 = Ctrl+C
+        logger.warning("Toad TUI exited with non-zero status", code=result.returncode)
+
+
+def _run_toad_websocket(config: str | None, port: int) -> None:
+    """Run Toad with WebSocket transport."""
+    url = f"ws://localhost:{port}"
+
+    # Build server command
+    server_cmd = [
+        "agentpool",
+        "serve-acp",
+        "--transport",
+        "websocket",
+        "--ws-port",
+        str(port),
+    ]
+    if config:
+        server_cmd.append(config)
+
+    logger.info("Starting ACP WebSocket server", url=url)
+
+    # Start server in background
+    server = subprocess.Popen(
+        server_cmd,
+        stdout=subprocess.DEVNULL,
+        stderr=subprocess.DEVNULL,
+    )
+
+    try:
+        # Wait for server startup
+        time.sleep(1.5)
+
+        # Clear screen for clean TUI
+        import os
+
+        os.system("clear" if os.name != "nt" else "cls")
+
+        # Run toad with mcp-ws client
+        result = subprocess.run(
+            ["uvx", "--from", "batrachian-toad@latest", "toad", "acp", f"uvx mcp-ws {url}"],
+            check=False,
+        )
+
+        if result.returncode not in {0, 130}:  # 130 = Ctrl+C
+            logger.warning("Toad TUI exited with non-zero status", code=result.returncode)
+
+    except KeyboardInterrupt:
+        logger.info("UI interrupted by user")
+    except Exception as e:
+        logger.exception("Error running Toad UI")
+        raise t.Exit(1) from e
+    finally:
+        # Clean up server
+        logger.info("Shutting down server")
+        server.send_signal(signal.SIGTERM)
+        try:
+            server.wait(timeout=5)
+        except subprocess.TimeoutExpired:
+            logger.warning("Server did not shut down gracefully, killing")
+            server.kill()
+
+
+@ui_app.command("desktop")
+def opencode_desktop_command(  # noqa: PLR0915
+    config: Annotated[
+        str | None,
+        t.Argument(help="Path to agent configuration (optional, not used with --attach)"),
+    ] = None,
+    host: Annotated[
+        str,
+        t.Option("--host", "-h", help="Host to bind/connect to"),
+    ] = "127.0.0.1",
+    port: Annotated[
+        int,
+        t.Option("--port", "-p", help="Port for server to listen on / connect to"),
+    ] = 4096,
+    agent: Annotated[
+        str | None,
+        t.Option(
+            "--agent",
+            help="Name of specific agent to use (not used with --attach)",
+        ),
+    ] = None,
+    attach: Annotated[
+        bool,
+        t.Option("--attach", help="Connect desktop app to existing server (don't start server)"),
+    ] = False,
+) -> None:
+    """Launch OpenCode desktop app with integrated server or attach to existing one.
+
+    By default, starts an OpenCode-compatible server in the background and
+    configures the desktop app to connect to it. The desktop app will run
+    independently and you can close the terminal.
+
+    With --attach, configures the desktop app to connect to an existing server
+    without starting a new one.
+
+    Note: This command requires the OpenCode desktop app to be installed.
+    The app will be configured to use the specified server URL via its config.
+
+    Examples:
+        # Start server, configure desktop app, and launch it
+        agentpool ui desktop
+
+        # Use specific config and agent
+        agentpool ui desktop agents.yml --agent myagent
+
+        # Custom port
+        agentpool ui desktop --port 8080
+
+        # Configure desktop to attach to existing server
+        agentpool ui desktop --attach
+        agentpool ui desktop --attach --port 8080
+
+        # After using --attach, reset to default (spawn local server)
+        agentpool ui desktop --attach --port 0  # port 0 clears the setting
+    """
+    import json
+    from pathlib import Path
+
+    url = f"http://{host}:{port}"
+
+    # Determine config path based on platform
+    config_dir = Path.home() / ".config" / "opencode"
+    config_file = config_dir / "config.json"
+
+    # Handle attach mode - configure desktop app to use external server
+    if attach:
+        if port == 0:
+            # Special case: port 0 means clear the setting
+            logger.info("Clearing desktop app server configuration")
+            if config_file.exists():
+                try:
+                    with config_file.open() as f:
+                        existing_config = json.load(f)
+                    # Remove server config
+                    if "server" in existing_config:
+                        del existing_config["server"]
+                    with config_file.open("w") as f:
+                        json.dump(existing_config, f, indent=2)
+                    logger.info("Cleared server configuration from config file")
+                except Exception as e:  # noqa: BLE001
+                    logger.warning("Failed to clear config", error=str(e))
+        else:
+            # Configure desktop app to use specified server
+            logger.info("Configuring desktop app to attach to server", url=url)
+
+            config_dir.mkdir(parents=True, exist_ok=True)
+
+            # Read existing config or create new
+            existing_config = {}
+            if config_file.exists():
+                try:
+                    with config_file.open() as f:
+                        existing_config = json.load(f)
+                except Exception as e:  # noqa: BLE001
+                    logger.warning("Failed to read existing config", error=str(e))
+
+            # Update server configuration
+            existing_config["server"] = {
+                "hostname": host,
+                "port": port,
+            }
+
+            try:
+                with config_file.open("w") as f:
+                    json.dump(existing_config, f, indent=2)
+                logger.info("Updated desktop app configuration", config=str(config_file))
+            except Exception as e:
+                logger.exception("Failed to write config", error=str(e))
+                raise t.Exit(1) from e
+
+        # Launch desktop app
+        logger.info("Launching OpenCode desktop app")
+        try:
+            # Try common desktop app launch commands
+            # On macOS: open -a OpenCode
+            # On Linux: OpenCode (capital O) is the desktop app
+            # On Windows: start opencode
+            import platform
+
+            system = platform.system()
+            if system == "Darwin":
+                subprocess.Popen(["open", "-a", "OpenCode"])
+            elif system == "Windows":
+                subprocess.Popen(["start", "opencode"], shell=True)
+            else:  # Linux and others
+                # Try different possible command names - OpenCode (capital O) is the desktop app
+                for cmd in ["OpenCode", "opencode-desktop"]:
+                    try:
+                        subprocess.Popen([cmd])
+                        break
+                    except FileNotFoundError:
+                        continue
+                else:
+                    msg = (
+                        "Could not find OpenCode desktop app. Please install it or launch manually."
+                    )
+                    raise FileNotFoundError(msg)  # noqa: TRY301
+
+            if port != 0:
+                logger.info(
+                    "Desktop app launched and configured to use server",
+                    url=url,
+                    note="The app will connect to the server. You can close this terminal.",
+                )
+            else:
+                logger.info(
+                    "Desktop app launched with default configuration",
+                    note="The app will spawn its own local server. You can close this terminal.",
+                )
+
+        except Exception as e:
+            logger.exception("Failed to launch desktop app", error=str(e))
+            logger.info(
+                "Configuration has been updated. Please launch the OpenCode desktop app manually.",
+                config=str(config_file),
+            )
+            raise t.Exit(1) from e
+
+        return
+
+    # Default mode: Start server + launch desktop app
+    # Build server command
+    server_cmd = [
+        "agentpool",
+        "serve-opencode",
+        "--host",
+        host,
+        "--port",
+        str(port),
+    ]
+    if config:
+        server_cmd.append(config)
+    if agent:
+        server_cmd.extend(["--agent", agent])
+
+    logger.info("Starting OpenCode server for desktop app", url=url)
+
+    # Start server in background
+    server = subprocess.Popen(
+        server_cmd,
+        stdout=subprocess.DEVNULL,
+        stderr=subprocess.DEVNULL,
+    )
+
+    try:
+        # Wait for server to be ready
+        max_retries = 30
+        for i in range(max_retries):
+            try:
+                sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+                sock.settimeout(0.5)
+                sock.connect((host, port))
+                sock.close()
+                logger.info("Server is ready", url=url)
+                break
+            except (TimeoutError, ConnectionRefusedError, OSError):
+                if i == max_retries - 1:
+                    msg = f"Server failed to start after {max_retries} attempts"
+                    raise RuntimeError(msg)  # noqa: B904
+                time.sleep(0.5)
+
+        # Give HTTP layer a moment to be fully ready
+        time.sleep(0.5)
+
+        # Configure desktop app to use this server
+        config_dir.mkdir(parents=True, exist_ok=True)
+
+        existing_config = {}
+        if config_file.exists():
+            try:
+                with config_file.open() as f:
+                    existing_config = json.load(f)
+            except Exception as e:  # noqa: BLE001
+                logger.warning("Failed to read existing config", error=str(e))
+
+        existing_config["server"] = {
+            "hostname": host,
+            "port": port,
+        }
+
+        try:
+            with config_file.open("w") as f:
+                json.dump(existing_config, f, indent=2)
+            logger.info("Configured desktop app", config=str(config_file))
+        except Exception as e:  # noqa: BLE001
+            logger.warning("Failed to write config", error=str(e))
+
+        # Launch desktop app
+        logger.info("Launching OpenCode desktop app")
+        import platform
+
+        system = platform.system()
+        try:
+            if system == "Darwin":
+                subprocess.Popen(["open", "-a", "OpenCode"])
+            elif system == "Windows":
+                subprocess.Popen(["start", "opencode"], shell=True)
+            else:  # Linux
+                # OpenCode (capital O) is the desktop app
+                for cmd in ["OpenCode", "opencode-desktop"]:
+                    try:
+                        subprocess.Popen([cmd])
+                        break
+                    except FileNotFoundError:
+                        continue
+                else:
+                    msg = "Could not find OpenCode desktop app"
+                    raise FileNotFoundError(msg)  # noqa: TRY301
+
+            logger.info(
+                "Desktop app launched",
+                note="Server is running in background. Press Ctrl+C to stop the server.",
+            )
+
+            # Keep server running until interrupted
+            logger.info("Server running. Press Ctrl+C to stop.")
+            server.wait()
+
+        except FileNotFoundError as e:
+            logger.exception(
+                "Desktop app not found. Please install OpenCode desktop app or launch it manually.",
+                config=str(config_file),
+            )
+            raise t.Exit(1) from e
+
+    except KeyboardInterrupt:
+        logger.info("Shutting down server")
+    except Exception as e:
+        logger.exception("Error running desktop app")
+        raise t.Exit(1) from e
+    finally:
+        # Clean up server
+        server.send_signal(signal.SIGTERM)
+        try:
+            server.wait(timeout=5)
+        except subprocess.TimeoutExpired:
+            logger.warning("Server did not shut down gracefully, killing")
+            server.kill()
+
+
+if __name__ == "__main__":
+    t.run(ui_app)

agentpool_commands/__init__.py

@@ -5,6 +5,7 @@ from __future__ import annotations
 from agentpool_commands.base import NodeCommand
 from agentpool_commands.agents import (
     CreateAgentCommand,
+    CreateTeamCommand,
     ListAgentsCommand,
     ShowAgentCommand,
     # SwitchAgentCommand,
@@ -15,6 +16,12 @@ from agentpool_commands.connections import (
     ListConnectionsCommand,
     DisconnectAllCommand,
 )
+from agentpool_commands.history import SearchHistoryCommand
+from agentpool_commands.mcp import (
+    AddMCPServerCommand,
+    AddRemoteMCPServerCommand,
+    ListMCPServersCommand,
+)
 from agentpool_commands.models import SetModelCommand
 from agentpool_commands.prompts import ListPromptsCommand, ShowPromptCommand
 from agentpool_commands.resources import (
@@ -28,19 +35,22 @@ from agentpool_commands.tools import (
     DisableToolCommand,
     EnableToolCommand,
     ListToolsCommand,
+    RegisterCodeToolCommand,
     RegisterToolCommand,
     ShowToolCommand,
 )
-from agentpool_commands.workers import (
-    AddWorkerCommand,
-    RemoveWorkerCommand,
-    ListWorkersCommand,
-)
+from agentpool_commands.workers import AddWorkerCommand, RemoveWorkerCommand, ListWorkersCommand
 from agentpool_commands.utils import (
     CopyClipboardCommand,
     EditAgentFileCommand,
+    GetLogsCommand,
     ShareHistoryCommand,
 )
+from agentpool_commands.pool import ListPoolsCommand, SpawnCommand
+
+# CompactCommand is only for Native Agent (has its own history)
+# Other agents (ClaudeCode, ACP, AGUI) don't control their own history
+from agentpool_commands.pool import CompactCommand  # noqa: F401
 from typing import TYPE_CHECKING, Any
 
 if TYPE_CHECKING:
@@ -64,6 +74,7 @@ def get_agent_commands(**kwargs: Any) -> Sequence[BaseCommand | type[SlashedComm
         "enable_enable_tool": EnableToolCommand,
         "enable_disable_tool": DisableToolCommand,
         "enable_register_tool": RegisterToolCommand,
+        "enable_register_code_tool": RegisterCodeToolCommand,
         "enable_list_resources": ListResourcesCommand,
         "enable_show_resource": ShowResourceCommand,
         "enable_add_resource": AddResourceCommand,
@@ -77,6 +88,11 @@ def get_agent_commands(**kwargs: Any) -> Sequence[BaseCommand | type[SlashedComm
         "enable_list_connections": ListConnectionsCommand,
         "enable_disconnect_all": DisconnectAllCommand,
         "enable_read": ReadCommand,
+        "enable_add_mcp_server": AddMCPServerCommand,
+        "enable_add_remote_mcp_server": AddRemoteMCPServerCommand,
+        "enable_list_mcp_servers": ListMCPServersCommand,
+        "enable_search_history": SearchHistoryCommand,
+        "enable_get_logs": GetLogsCommand,
     }
     return [command for flag, command in command_map.items() if kwargs.get(flag, True)]
 
@@ -85,9 +101,12 @@ def get_pool_commands(**kwargs: Any) -> Sequence[BaseCommand | type[SlashedComma
     """Get commands that operate on multiple agents or the pool itself."""
     command_map = {
         "enable_create_agent": CreateAgentCommand,
+        "enable_create_team": CreateTeamCommand,
         "enable_list_agents": ListAgentsCommand,
         "enable_show_agent": ShowAgentCommand,
         "enable_edit_agent_file": EditAgentFileCommand,
+        "enable_list_pools": ListPoolsCommand,
+        "enable_spawn": SpawnCommand,
     }
     return [command for flag, command in command_map.items() if kwargs.get(flag, True)]
 
@@ -123,6 +142,7 @@ def get_commands(
     enable_enable_tool: bool = True,
     enable_disable_tool: bool = True,
     enable_register_tool: bool = True,
+    enable_register_code_tool: bool = True,
     enable_list_resources: bool = True,
     enable_show_resource: bool = True,
     enable_add_resource: bool = True,
@@ -136,10 +156,18 @@ def get_commands(
     enable_list_connections: bool = True,
     enable_disconnect_all: bool = True,
     enable_read: bool = True,
+    enable_add_mcp_server: bool = True,
+    enable_add_remote_mcp_server: bool = True,
+    enable_list_mcp_servers: bool = True,
+    enable_search_history: bool = True,
+    enable_get_logs: bool = True,
     enable_create_agent: bool = True,
+    enable_create_team: bool = True,
     enable_list_agents: bool = True,
     enable_show_agent: bool = True,
     enable_edit_agent_file: bool = True,
+    enable_list_pools: bool = True,
+    enable_spawn: bool = True,
 ) -> list[BaseCommand | type[SlashedCommand]]:
     """Get all built-in commands."""
     agent_kwargs = {
@@ -153,6 +181,7 @@ def get_commands(
         "enable_enable_tool": enable_enable_tool,
         "enable_disable_tool": enable_disable_tool,
         "enable_register_tool": enable_register_tool,
+        "enable_register_code_tool": enable_register_code_tool,
         "enable_list_resources": enable_list_resources,
         "enable_show_resource": enable_show_resource,
         "enable_add_resource": enable_add_resource,
@@ -166,12 +195,20 @@ def get_commands(
         "enable_list_connections": enable_list_connections,
         "enable_disconnect_all": enable_disconnect_all,
         "enable_read": enable_read,
+        "enable_add_mcp_server": enable_add_mcp_server,
+        "enable_add_remote_mcp_server": enable_add_remote_mcp_server,
+        "enable_list_mcp_servers": enable_list_mcp_servers,
+        "enable_search_history": enable_search_history,
+        "enable_get_logs": enable_get_logs,
     }
     pool_kwargs = {
         "enable_create_agent": enable_create_agent,
+        "enable_create_team": enable_create_team,
         "enable_list_agents": enable_list_agents,
         "enable_show_agent": enable_show_agent,
         "enable_edit_agent_file": enable_edit_agent_file,
+        "enable_list_pools": enable_list_pools,
+        "enable_spawn": enable_spawn,
     }
 
     return [