a4e 0.1.5__py3-none-any.whl

a4e/cli_commands/mcp.py

@@ -0,0 +1,504 @@
+"""
+MCP configuration commands for A4E.
+
+This module provides commands to configure the A4E MCP server
+for different IDEs like Cursor, Claude Code, and Antigravity.
+
+Usage:
+    a4e mcp setup cursor
+    a4e mcp setup claude-code
+    a4e mcp setup antigravity
+    a4e mcp show cursor
+    a4e mcp remove cursor
+    a4e mcp test
+"""
+
+import json
+import sys
+from pathlib import Path
+
+import typer
+
+app = typer.Typer(
+    help="Configure MCP server for IDEs",
+    no_args_is_help=True,
+)
+
+
+# ============================================================================
+# Configuration Paths
+# ============================================================================
+
+
+def _get_platform() -> str:
+    """Get the current platform: darwin, win32, or linux."""
+    return sys.platform
+
+
+def _get_ide_configs() -> dict:
+    """
+    Get IDE configuration paths and formats.
+
+    Returns dict with IDE name as key and:
+        - path: Path to config file
+        - servers_key: Key name for MCP servers in config
+        - name: Display name for the IDE
+    """
+    platform = _get_platform()
+    home = Path.home()
+
+    # Antigravity config path varies by OS
+    if platform == "win32":
+        antigravity_path = home / ".gemini" / "antigravity" / "mcp_config.json"
+    else:  # macOS and Linux
+        antigravity_path = home / ".gemini" / "antigravity" / "mcp_config.json"
+
+    return {
+        "cursor": {
+            "path": home / ".cursor" / "mcp.json",
+            "servers_key": "mcpServers",
+            "name": "Cursor",
+        },
+        "claude-code": {
+            "path": home / ".claude.json",
+            "servers_key": "mcpServers",
+            "name": "Claude Code",
+        },
+        "antigravity": {
+            "path": antigravity_path,
+            "servers_key": "mcpServers",
+            "name": "Antigravity",
+        },
+    }
+
+
+def _get_a4e_server_config(ide: str) -> dict:
+    """
+    Generate the MCP server configuration for A4E.
+
+    Uses sys.executable to get the exact Python that has a4e installed.
+    This ensures the config works regardless of how Python was installed
+    (pyenv, system, venv, homebrew, etc.)
+
+    The env configuration varies by IDE:
+    - Cursor: Uses ${workspaceFolder} which it expands
+    - Claude Code / Antigravity: No workspace variable, uses cwd
+
+    Args:
+        ide: The IDE identifier (cursor, claude-code, antigravity)
+    """
+    base_config = {
+        "command": sys.executable,
+        "args": ["-m", "a4e.server"],
+    }
+
+    # IDEs that support ${workspaceFolder}
+    if ide == "cursor":
+        base_config["env"] = {
+            "A4E_WORKSPACE": "${workspaceFolder}"
+        }
+    # Claude Code and Antigravity: no workspace variable
+    # The server will use cwd or require explicit project_path in tool calls
+
+    return base_config
+
+
+def _get_supported_ides() -> list[str]:
+    """Get list of supported IDE names."""
+    return list(_get_ide_configs().keys())
+
+
+# ============================================================================
+# Helper Functions
+# ============================================================================
+
+
+def _load_config(config_path: Path) -> tuple[dict, bool]:
+    """
+    Load existing config from path.
+
+    Returns:
+        Tuple of (config_dict, had_error)
+        If file doesn't exist, returns empty dict with no error.
+        If file has invalid JSON, returns empty dict with error=True.
+    """
+    if not config_path.exists():
+        return {}, False
+
+    try:
+        content = config_path.read_text().strip()
+        if not content:
+            return {}, False
+        return json.loads(content), False
+    except json.JSONDecodeError:
+        return {}, True
+
+
+def _save_config(config_path: Path, config: dict) -> None:
+    """Save config to path, creating parent directories if needed."""
+    config_path.parent.mkdir(parents=True, exist_ok=True)
+    config_path.write_text(json.dumps(config, indent=2) + "\n")
+
+
+# ============================================================================
+# Commands
+# ============================================================================
+
+
+@app.command()
+def setup(
+    ide: str = typer.Argument(
+        ...,
+        help="IDE to configure: cursor, claude-code, antigravity"
+    ),
+    dry_run: bool = typer.Option(
+        False, "--dry-run", "-n",
+        help="Show what would be done without making changes"
+    ),
+    force: bool = typer.Option(
+        False, "--force", "-f",
+        help="Overwrite existing A4E configuration"
+    ),
+):
+    """
+    Configure A4E MCP server for your IDE.
+
+    This writes the necessary configuration so your IDE can use
+    A4E tools through the Model Context Protocol (MCP).
+
+    Examples:
+
+        a4e mcp setup cursor
+
+        a4e mcp setup claude-code
+
+        a4e mcp setup antigravity
+
+        a4e mcp setup cursor --dry-run
+
+        a4e mcp setup cursor --force
+    """
+    ide_configs = _get_ide_configs()
+    ide_lower = ide.lower()
+
+    if ide_lower not in ide_configs:
+        supported = ", ".join(_get_supported_ides())
+        typer.echo(f"Error: Unknown IDE '{ide}'", err=True)
+        typer.echo(f"Supported IDEs: {supported}", err=True)
+        raise typer.Exit(1)
+
+    config_info = ide_configs[ide_lower]
+    config_path: Path = config_info["path"]
+    servers_key: str = config_info["servers_key"]
+    ide_name: str = config_info["name"]
+    a4e_config = _get_a4e_server_config(ide_lower)
+
+    # Dry run mode - show what would happen
+    if dry_run:
+        typer.echo(f"Would configure A4E for {ide_name}:")
+        typer.echo()
+        typer.echo(f"  Config file: {config_path}")
+        typer.echo(f"  Python path: {sys.executable}")
+        typer.echo()
+        typer.echo("  Configuration to add:")
+        preview = {servers_key: {"a4e": a4e_config}}
+        typer.echo(json.dumps(preview, indent=2))
+        return
+
+    # Load existing config
+    existing, had_json_error = _load_config(config_path)
+
+    # Always create backup if file exists (before any modification)
+    if config_path.exists() and not had_json_error:
+        backup_path = config_path.with_suffix(".json.backup")
+        backup_path.write_text(config_path.read_text())
+        typer.echo(f"Backup created: {backup_path}")
+
+    if had_json_error:
+        # Backup corrupted config
+        backup_path = config_path.with_suffix(".json.backup")
+        typer.echo(f"Warning: Invalid JSON in {config_path}", err=True)
+        typer.echo(f"Creating backup at {backup_path}", err=True)
+
+        if config_path.exists():
+            backup_path.write_text(config_path.read_text())
+
+        existing = {}
+
+    # Ensure servers key exists
+    existing.setdefault(servers_key, {})
+
+    # Check if a4e already configured
+    if "a4e" in existing[servers_key]:
+        if not force:
+            typer.echo(f"A4E is already configured for {ide_name}")
+            typer.echo()
+            typer.echo("Current configuration:")
+            typer.echo(json.dumps(existing[servers_key]["a4e"], indent=2))
+            typer.echo()
+            typer.echo("Use --force to overwrite, or --dry-run to preview")
+            raise typer.Exit(1)
+        else:
+            typer.echo("Overwriting existing A4E configuration...")
+
+    # Add/update a4e config
+    existing[servers_key]["a4e"] = a4e_config
+
+    # Save config
+    _save_config(config_path, existing)
+
+    typer.echo(f"A4E MCP configured for {ide_name}")
+    typer.echo()
+    typer.echo(f"  Config file: {config_path}")
+    typer.echo(f"  Python: {sys.executable}")
+    typer.echo()
+
+    # IDE-specific notes
+    if ide_lower == "cursor":
+        typer.echo("  Workspace: Uses ${workspaceFolder} (auto-detected)")
+    else:
+        typer.echo("  Workspace: Uses current directory or specify project_path in tool calls")
+
+    typer.echo()
+    typer.echo(f"Restart {ide_name} to activate the MCP server.")
+
+
+@app.command()
+def show(
+    ide: str = typer.Argument(
+        ...,
+        help="IDE to show config for: cursor, claude-code, antigravity"
+    )
+):
+    """
+    Show current MCP configuration for an IDE.
+
+    Example:
+
+        a4e mcp show cursor
+    """
+    ide_configs = _get_ide_configs()
+    ide_lower = ide.lower()
+
+    if ide_lower not in ide_configs:
+        supported = ", ".join(_get_supported_ides())
+        typer.echo(f"Error: Unknown IDE '{ide}'", err=True)
+        typer.echo(f"Supported IDEs: {supported}", err=True)
+        raise typer.Exit(1)
+
+    config_info = ide_configs[ide_lower]
+    config_path: Path = config_info["path"]
+    servers_key: str = config_info["servers_key"]
+    ide_name: str = config_info["name"]
+
+    if not config_path.exists():
+        typer.echo(f"No MCP config found for {ide_name}")
+        typer.echo(f"Expected path: {config_path}")
+        typer.echo()
+        typer.echo(f"Run: a4e mcp setup {ide}")
+        return
+
+    config, had_error = _load_config(config_path)
+
+    if had_error:
+        typer.echo(f"Error: Invalid JSON in {config_path}", err=True)
+        raise typer.Exit(1)
+
+    typer.echo(f"MCP configuration for {ide_name}")
+    typer.echo(f"File: {config_path}")
+    typer.echo()
+    typer.echo(json.dumps(config, indent=2))
+
+    # Check A4E status
+    if servers_key in config and "a4e" in config[servers_key]:
+        typer.echo()
+        typer.echo("A4E status: Configured")
+    else:
+        typer.echo()
+        typer.echo("A4E status: Not configured")
+        typer.echo(f"Run: a4e mcp setup {ide}")
+
+
+@app.command()
+def remove(
+    ide: str = typer.Argument(
+        ...,
+        help="IDE to remove A4E config from"
+    ),
+):
+    """
+    Remove A4E MCP configuration from an IDE.
+
+    This removes only the A4E configuration, preserving other MCP servers.
+
+    Example:
+
+        a4e mcp remove cursor
+    """
+    ide_configs = _get_ide_configs()
+    ide_lower = ide.lower()
+
+    if ide_lower not in ide_configs:
+        supported = ", ".join(_get_supported_ides())
+        typer.echo(f"Error: Unknown IDE '{ide}'", err=True)
+        typer.echo(f"Supported IDEs: {supported}", err=True)
+        raise typer.Exit(1)
+
+    config_info = ide_configs[ide_lower]
+    config_path: Path = config_info["path"]
+    servers_key: str = config_info["servers_key"]
+    ide_name: str = config_info["name"]
+
+    if not config_path.exists():
+        typer.echo(f"No MCP config found for {ide_name}")
+        return
+
+    config, had_error = _load_config(config_path)
+
+    if had_error:
+        typer.echo(f"Error: Invalid JSON in {config_path}", err=True)
+        raise typer.Exit(1)
+
+    # Check if a4e is configured
+    if servers_key not in config or "a4e" not in config[servers_key]:
+        typer.echo(f"A4E is not configured for {ide_name}")
+        return
+
+    # Remove a4e
+    del config[servers_key]["a4e"]
+
+    # Clean up empty servers dict
+    if not config[servers_key]:
+        del config[servers_key]
+
+    # Save config
+    _save_config(config_path, config)
+
+    typer.echo(f"A4E removed from {ide_name} configuration")
+    typer.echo()
+    typer.echo(f"Restart {ide_name} to apply changes.")
+
+
+@app.command()
+def test():
+    """
+    Test that the A4E MCP server can start correctly.
+
+    This verifies that:
+    - The server module can be imported
+    - The Python path is correct
+    - Dependencies are installed
+
+    Useful for debugging configuration issues.
+    """
+    import subprocess
+
+    typer.echo("Testing A4E MCP server...")
+    typer.echo()
+    typer.echo(f"  Python: {sys.executable}")
+    typer.echo(f"  Command: {sys.executable} -m a4e.server")
+    typer.echo()
+
+    # Test 1: Module import
+    typer.echo("1. Testing module import...")
+    try:
+        # Test that the module can be imported
+        import importlib
+        importlib.import_module("a4e.server")
+        typer.echo("   Module imports correctly")
+    except ImportError as e:
+        typer.echo(f"   Error: Failed to import a4e.server: {e}", err=True)
+        raise typer.Exit(1)
+
+    # Test 2: FastMCP instance
+    typer.echo("2. Testing MCP server instance...")
+    try:
+        from a4e.core import mcp
+        tools = list(mcp._tool_manager._tools.keys()) if hasattr(mcp, '_tool_manager') else []
+        typer.echo("   MCP server initialized")
+        if tools:
+            typer.echo(f"   Tools registered: {len(tools)}")
+    except Exception as e:
+        typer.echo(f"   Warning: Could not inspect MCP instance: {e}", err=True)
+
+    # Test 3: Server process start
+    typer.echo("3. Testing server process...")
+    try:
+        # Run with --help to quickly test if it starts
+        result = subprocess.run(
+            [sys.executable, "-c", "from a4e.server import main; print('OK')"],
+            capture_output=True,
+            text=True,
+            timeout=10,
+        )
+
+        if result.returncode == 0 and "OK" in result.stdout:
+            typer.echo("   Server process starts correctly")
+        else:
+            typer.echo("   Warning: Unexpected output", err=True)
+            if result.stderr:
+                typer.echo(f"   stderr: {result.stderr}", err=True)
+    except subprocess.TimeoutExpired:
+        typer.echo("   Server process timed out (may be normal for stdio server)")
+    except Exception as e:
+        typer.echo(f"   Error testing server process: {e}", err=True)
+        raise typer.Exit(1)
+
+    typer.echo()
+    typer.echo("All tests passed. MCP server is ready.")
+    typer.echo()
+    typer.echo("Next steps:")
+    typer.echo("  1. Run: a4e mcp setup <ide>")
+    typer.echo("  2. Restart your IDE")
+    typer.echo("  3. Start using A4E tools!")
+
+
+@app.command()
+def path(
+    ide: str = typer.Argument(
+        ...,
+        help="IDE to show config path for"
+    )
+):
+    """
+    Show the config file path for an IDE.
+
+    Useful for manual configuration or debugging.
+
+    Example:
+
+        a4e mcp path cursor
+    """
+    ide_configs = _get_ide_configs()
+    ide_lower = ide.lower()
+
+    if ide_lower not in ide_configs:
+        supported = ", ".join(_get_supported_ides())
+        typer.echo(f"Error: Unknown IDE '{ide}'", err=True)
+        typer.echo(f"Supported IDEs: {supported}", err=True)
+        raise typer.Exit(1)
+
+    config_path = ide_configs[ide_lower]["path"]
+    typer.echo(str(config_path))
+
+
+@app.command(name="list")
+def list_ides():
+    """
+    List all supported IDEs.
+
+    Example:
+
+        a4e mcp list
+    """
+    ide_configs = _get_ide_configs()
+
+    typer.echo("Supported IDEs:")
+    typer.echo()
+
+    for ide_key, config in ide_configs.items():
+        path_exists = "exists" if config["path"].exists() else "not found"
+        typer.echo(f"  {ide_key}")
+        typer.echo(f"    Name: {config['name']}")
+        typer.echo(f"    Config: {config['path']} ({path_exists})")
+        typer.echo()

a4e/cli_commands/remove.py

@@ -0,0 +1,197 @@
+# a4e/cli_commands/remove.py
+"""
+Commands for removing tools, views, and skills from an agent.
+"""
+
+import typer
+from pathlib import Path
+from typing import Optional
+
+from rich.console import Console
+from rich.prompt import Confirm
+
+console = Console()
+
+# Create a 'Typer' app for the 'remove' command group
+app = typer.Typer(
+    no_args_is_help=True,
+    help="Remove tools, views, or skills from your agent.",
+)
+
+
+def find_agent_dir(agent_name: Optional[str] = None) -> Optional[Path]:
+    """Find the agent directory from name or current working directory."""
+    cwd = Path.cwd()
+
+    if agent_name:
+        if Path(agent_name).is_absolute() and Path(agent_name).exists():
+            return Path(agent_name)
+        if (cwd / agent_name).exists():
+            return cwd / agent_name
+        agent_store = cwd / "file-store" / "agent-store" / agent_name
+        if agent_store.exists():
+            return agent_store
+        return None
+
+    if (cwd / "agent.py").exists() and (cwd / "metadata.json").exists():
+        return cwd
+
+    return None
+
+
+@app.command("tool")
+def remove_tool(
+    tool_name: str = typer.Argument(..., help="Name of the tool to remove"),
+    agent_name: Optional[str] = typer.Option(
+        None, "--agent", "-a", help="Agent name/path (defaults to current directory)"
+    ),
+    yes: bool = typer.Option(
+        False, "--yes", "-y", help="Skip confirmation prompt"
+    ),
+) -> None:
+    """
+    Remove a tool from the agent.
+
+    Example:
+        a4e remove tool calculate_bmi
+        a4e remove tool calculate_bmi --yes
+    """
+    agent_dir = find_agent_dir(agent_name)
+    if not agent_dir:
+        console.print("[red]Error: Not in an agent directory. Use --agent to specify the agent.[/red]")
+        raise typer.Exit(code=1)
+
+    # Confirm removal
+    if not yes:
+        if not Confirm.ask(f"Remove tool '{tool_name}'?", default=False):
+            console.print("[yellow]Cancelled.[/yellow]")
+            raise typer.Exit(code=0)
+
+    try:
+        from ..tools.agent_tools.remove_tool import remove_tool as mcp_remove_tool
+
+        result = mcp_remove_tool(
+            tool_name=tool_name,
+            agent_name=str(agent_dir),
+        )
+
+        if result.get("success"):
+            console.print(f"[green]✓ Tool '{tool_name}' removed![/green]")
+            console.print(f"  Removed: {result.get('removed_file')}")
+        else:
+            console.print(f"[red]Error: {result.get('error')}[/red]")
+            raise typer.Exit(code=1)
+
+    except ImportError as e:
+        console.print(f"[red]Error importing tools: {e}[/red]")
+        raise typer.Exit(code=1)
+
+
+@app.command("view")
+def remove_view(
+    view_id: str = typer.Argument(..., help="ID of the view to remove"),
+    agent_name: Optional[str] = typer.Option(
+        None, "--agent", "-a", help="Agent name/path (defaults to current directory)"
+    ),
+    yes: bool = typer.Option(
+        False, "--yes", "-y", help="Skip confirmation prompt"
+    ),
+) -> None:
+    """
+    Remove a view from the agent.
+
+    Note: The 'welcome' view cannot be removed as it's required.
+
+    Example:
+        a4e remove view results-display
+        a4e remove view results-display --yes
+    """
+    agent_dir = find_agent_dir(agent_name)
+    if not agent_dir:
+        console.print("[red]Error: Not in an agent directory. Use --agent to specify the agent.[/red]")
+        raise typer.Exit(code=1)
+
+    # Warn about welcome view
+    if view_id == "welcome":
+        console.print("[red]Error: Cannot remove the 'welcome' view - it is required for all agents.[/red]")
+        raise typer.Exit(code=1)
+
+    # Confirm removal
+    if not yes:
+        if not Confirm.ask(f"Remove view '{view_id}'?", default=False):
+            console.print("[yellow]Cancelled.[/yellow]")
+            raise typer.Exit(code=0)
+
+    try:
+        from ..tools.views.remove_view import remove_view as mcp_remove_view
+
+        result = mcp_remove_view(
+            view_id=view_id,
+            agent_name=str(agent_dir),
+        )
+
+        if result.get("success"):
+            console.print(f"[green]✓ View '{view_id}' removed![/green]")
+            console.print(f"  Removed: {result.get('removed_folder')}")
+        else:
+            console.print(f"[red]Error: {result.get('error')}[/red]")
+            raise typer.Exit(code=1)
+
+    except ImportError as e:
+        console.print(f"[red]Error importing tools: {e}[/red]")
+        raise typer.Exit(code=1)
+
+
+@app.command("skill")
+def remove_skill(
+    skill_id: str = typer.Argument(..., help="ID of the skill to remove"),
+    agent_name: Optional[str] = typer.Option(
+        None, "--agent", "-a", help="Agent name/path (defaults to current directory)"
+    ),
+    yes: bool = typer.Option(
+        False, "--yes", "-y", help="Skip confirmation prompt"
+    ),
+) -> None:
+    """
+    Remove a skill from the agent.
+
+    Note: The 'show_welcome' skill cannot be removed as it's required.
+
+    Example:
+        a4e remove skill show_results
+        a4e remove skill show_results --yes
+    """
+    agent_dir = find_agent_dir(agent_name)
+    if not agent_dir:
+        console.print("[red]Error: Not in an agent directory. Use --agent to specify the agent.[/red]")
+        raise typer.Exit(code=1)
+
+    # Warn about show_welcome skill
+    if skill_id == "show_welcome":
+        console.print("[red]Error: Cannot remove the 'show_welcome' skill - it is required for all agents.[/red]")
+        raise typer.Exit(code=1)
+
+    # Confirm removal
+    if not yes:
+        if not Confirm.ask(f"Remove skill '{skill_id}'?", default=False):
+            console.print("[yellow]Cancelled.[/yellow]")
+            raise typer.Exit(code=0)
+
+    try:
+        from ..tools.skills.remove_skill import remove_skill as mcp_remove_skill
+
+        result = mcp_remove_skill(
+            skill_id=skill_id,
+            agent_name=str(agent_dir),
+        )
+
+        if result.get("success"):
+            console.print(f"[green]✓ Skill '{skill_id}' removed![/green]")
+            console.print(f"  Removed: {result.get('removed_folder')}")
+        else:
+            console.print(f"[red]Error: {result.get('error')}[/red]")
+            raise typer.Exit(code=1)
+
+    except ImportError as e:
+        console.print(f"[red]Error importing tools: {e}[/red]")
+        raise typer.Exit(code=1)