a4e 0.1.5__py3-none-any.whl

a4e/cli_commands/update.py

@@ -0,0 +1,285 @@
+# a4e/cli_commands/update.py
+"""
+Commands for updating tools, views, and skills in an agent.
+"""
+
+import typer
+import json
+from pathlib import Path
+from typing import Optional
+
+from rich.console import Console
+from rich.prompt import Prompt, Confirm
+
+console = Console()
+
+# Create a 'Typer' app for the 'update' command group
+app = typer.Typer(
+    no_args_is_help=True,
+    help="Update existing tools, views, or skills.",
+)
+
+TYPE_OPTIONS = ["string", "number", "integer", "boolean", "array", "object"]
+
+
+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 update_tool(
+    tool_name: Optional[str] = typer.Argument(None, help="Name of the tool to update"),
+    description: Optional[str] = typer.Option(
+        None, "--description", "-d", help="New description"
+    ),
+    parameters_json: Optional[str] = typer.Option(
+        None, "--parameters", "-p", help="New parameters as JSON string"
+    ),
+    agent_name: Optional[str] = typer.Option(
+        None, "--agent", "-a", help="Agent name/path"
+    ),
+) -> None:
+    """
+    Update an existing tool's description or parameters.
+
+    Example:
+        a4e update tool calculate_bmi -d "New description"
+        a4e update tool calculate_bmi -p '{"weight": "number", "height": "number"}'
+    """
+    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)
+
+    # List available tools if none specified
+    if not tool_name:
+        tools_dir = agent_dir / "tools"
+        if tools_dir.exists():
+            available = [f.stem for f in tools_dir.glob("*.py") if f.stem != "__init__"]
+            if available:
+                console.print("[bold]Available tools:[/bold]")
+                for t in available:
+                    console.print(f"  • {t}")
+            tool_name = Prompt.ask("[bold]Tool to update[/bold]")
+        else:
+            console.print("[red]No tools directory found[/red]")
+            raise typer.Exit(code=1)
+
+    if not description and not parameters_json:
+        console.print("[yellow]What would you like to update?[/yellow]")
+        update_desc = Confirm.ask("Update description?", default=False)
+        if update_desc:
+            description = Prompt.ask("New description")
+
+        update_params = Confirm.ask("Update parameters?", default=False)
+        if update_params:
+            console.print("[dim]Enter parameters as JSON or use interactive mode[/dim]")
+            parameters_json = Prompt.ask("Parameters JSON", default="{}")
+
+    parameters = None
+    if parameters_json:
+        try:
+            parameters = json.loads(parameters_json)
+        except json.JSONDecodeError:
+            console.print("[red]Invalid JSON for parameters[/red]")
+            raise typer.Exit(code=1)
+
+    try:
+        from ..tools.agent_tools.update_tool import update_tool as mcp_update_tool
+
+        result = mcp_update_tool(
+            tool_name=tool_name,
+            description=description,
+            parameters=parameters,
+            agent_name=str(agent_dir),
+        )
+
+        if result.get("success"):
+            console.print(f"\n[green]✓ Tool '{tool_name}' updated![/green]")
+            console.print(f"  Path: {result.get('path')}")
+        else:
+            console.print(f"\n[red]Error: {result.get('error')}[/red]")
+            if result.get("fix"):
+                console.print(f"  [yellow]Fix: {result.get('fix')}[/yellow]")
+            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 update_view(
+    view_id: Optional[str] = typer.Argument(None, help="ID of the view to update"),
+    description: Optional[str] = typer.Option(
+        None, "--description", "-d", help="New description"
+    ),
+    props_json: Optional[str] = typer.Option(
+        None, "--props", "-p", help="New props as JSON string"
+    ),
+    agent_name: Optional[str] = typer.Option(
+        None, "--agent", "-a", help="Agent name/path"
+    ),
+) -> None:
+    """
+    Update an existing view's description or props.
+
+    Example:
+        a4e update view dashboard -d "New description"
+        a4e update view dashboard -p '{"title": "string", "items": "array"}'
+    """
+    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)
+
+    # List available views if none specified
+    if not view_id:
+        views_dir = agent_dir / "views"
+        if views_dir.exists():
+            available = [d.name for d in views_dir.iterdir() if d.is_dir() and not d.name.startswith("_")]
+            if available:
+                console.print("[bold]Available views:[/bold]")
+                for v in available:
+                    console.print(f"  • {v}")
+            view_id = Prompt.ask("[bold]View to update[/bold]")
+        else:
+            console.print("[red]No views directory found[/red]")
+            raise typer.Exit(code=1)
+
+    if not description and not props_json:
+        console.print("[yellow]What would you like to update?[/yellow]")
+        update_desc = Confirm.ask("Update description?", default=False)
+        if update_desc:
+            description = Prompt.ask("New description")
+
+        update_props = Confirm.ask("Update props?", default=False)
+        if update_props:
+            props_json = Prompt.ask("Props JSON", default="{}")
+
+    props = None
+    if props_json:
+        try:
+            props = json.loads(props_json)
+        except json.JSONDecodeError:
+            console.print("[red]Invalid JSON for props[/red]")
+            raise typer.Exit(code=1)
+
+    try:
+        from ..tools.views.update_view import update_view as mcp_update_view
+
+        result = mcp_update_view(
+            view_id=view_id,
+            description=description,
+            props=props,
+            agent_name=str(agent_dir),
+        )
+
+        if result.get("success"):
+            console.print(f"\n[green]✓ View '{view_id}' updated![/green]")
+            console.print(f"  Path: {result.get('path')}")
+        else:
+            console.print(f"\n[red]Error: {result.get('error')}[/red]")
+            if result.get("fix"):
+                console.print(f"  [yellow]Fix: {result.get('fix')}[/yellow]")
+            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 update_skill(
+    skill_id: Optional[str] = typer.Argument(None, help="ID of the skill to update"),
+    name: Optional[str] = typer.Option(None, "--name", "-n", help="New display name"),
+    description: Optional[str] = typer.Option(None, "--description", "-d", help="New description"),
+    intent_triggers: Optional[str] = typer.Option(
+        None, "--triggers", "-t", help="New comma-separated triggers"
+    ),
+    output_view: Optional[str] = typer.Option(None, "--view", "-v", help="New output view"),
+    internal_tools: Optional[str] = typer.Option(
+        None, "--tools", help="New comma-separated internal tools"
+    ),
+    requires_auth: Optional[bool] = typer.Option(None, "--auth/--no-auth", help="Requires auth"),
+    agent_name: Optional[str] = typer.Option(None, "--agent", "-a", help="Agent name/path"),
+) -> None:
+    """
+    Update an existing skill's properties.
+
+    Example:
+        a4e update skill show_results --name "Display Results" --view new-results
+    """
+    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)
+
+    # List available skills if none specified
+    if not skill_id:
+        skills_file = agent_dir / "skills" / "schemas.json"
+        if skills_file.exists():
+            try:
+                schemas = json.loads(skills_file.read_text())
+                available = list(schemas.keys())
+                if available:
+                    console.print("[bold]Available skills:[/bold]")
+                    for s in available:
+                        console.print(f"  • {s}")
+                skill_id = Prompt.ask("[bold]Skill to update[/bold]")
+            except json.JSONDecodeError:
+                console.print("[red]Could not parse skills/schemas.json[/red]")
+                raise typer.Exit(code=1)
+        else:
+            console.print("[red]No skills/schemas.json found[/red]")
+            raise typer.Exit(code=1)
+
+    # Parse list options
+    triggers_list = [t.strip() for t in intent_triggers.split(",")] if intent_triggers else None
+    tools_list = [t.strip() for t in internal_tools.split(",") if t.strip()] if internal_tools else None
+
+    try:
+        from ..tools.skills.update_skill import update_skill as mcp_update_skill
+
+        result = mcp_update_skill(
+            skill_id=skill_id,
+            name=name,
+            description=description,
+            intent_triggers=triggers_list,
+            output_view=output_view,
+            internal_tools=tools_list,
+            requires_auth=requires_auth,
+            agent_name=str(agent_dir),
+        )
+
+        if result.get("success"):
+            console.print(f"\n[green]✓ Skill '{skill_id}' updated![/green]")
+            console.print(f"  Path: {result.get('path')}")
+            if result.get("warnings"):
+                for w in result["warnings"]:
+                    console.print(f"  [yellow]Warning: {w}[/yellow]")
+        else:
+            console.print(f"\n[red]Error: {result.get('error')}[/red]")
+            if result.get("fix"):
+                console.print(f"  [yellow]Fix: {result.get('fix')}[/yellow]")
+            raise typer.Exit(code=1)
+
+    except ImportError as e:
+        console.print(f"[red]Error importing tools: {e}[/red]")
+        raise typer.Exit(code=1)

a4e/cli_commands/validate.py

@@ -0,0 +1,117 @@
+# a4e/cli_commands/validate.py
+"""
+Command for validating an agent project.
+"""
+
+import typer
+from pathlib import Path
+from typing import Optional
+
+from rich.console import Console
+from rich.table import Table
+from rich.panel import Panel
+
+console = Console()
+
+# Create a 'Typer' app for the 'validate' command
+app = typer.Typer(
+    no_args_is_help=False,
+    help="Validate your agent project for errors and warnings.",
+)
+
+
+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.callback(invoke_without_command=True)
+def validate(
+    ctx: typer.Context,
+    agent_name: Optional[str] = typer.Option(
+        None, "--agent", "-a", help="Agent name/path (defaults to current directory)"
+    ),
+    strict: bool = typer.Option(
+        False, "--strict", "-s", help="Treat warnings as errors"
+    ),
+) -> None:
+    """
+    Validate the agent project structure and code.
+
+    Checks for:
+    - Required files (agent.py, metadata.json, prompts/agent.md, etc.)
+    - Python syntax errors
+    - Type hints on public functions
+    - Schema existence
+    - Skill dependencies (views, tools)
+
+    Example:
+        a4e validate
+        a4e validate --agent my-agent --strict
+    """
+    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)
+
+    console.print(Panel.fit(
+        f"[bold]Validating agent:[/bold] {agent_dir.name}",
+        border_style="blue"
+    ))
+
+    try:
+        from ..tools.validation.validate import validate as mcp_validate
+
+        result = mcp_validate(agent_name=str(agent_dir))
+
+        if result.get("success"):
+            errors = result.get("errors", [])
+            warnings = result.get("warnings", [])
+
+            # Show errors
+            if errors:
+                console.print(f"\n[red bold]Errors ({len(errors)}):[/red bold]")
+                for error in errors:
+                    console.print(f"  [red]✗[/red] {error}")
+
+            # Show warnings
+            if warnings:
+                console.print(f"\n[yellow bold]Warnings ({len(warnings)}):[/yellow bold]")
+                for warning in warnings:
+                    console.print(f"  [yellow]⚠[/yellow] {warning}")
+
+            # Summary
+            console.print("")
+            if errors:
+                console.print(f"[red bold]Validation failed with {len(errors)} error(s)[/red bold]")
+                raise typer.Exit(code=1)
+            elif warnings and strict:
+                console.print(f"[yellow bold]Validation failed with {len(warnings)} warning(s) (strict mode)[/yellow bold]")
+                raise typer.Exit(code=1)
+            elif warnings:
+                console.print(f"[green]✓ Validation passed with {len(warnings)} warning(s)[/green]")
+            else:
+                console.print("[green bold]✓ Validation passed - no issues found![/green bold]")
+
+        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)

a4e/core.py

@@ -0,0 +1,109 @@
+"""
+Core module with shared utilities and MCP instance.
+"""
+
+from mcp.server.fastmcp import FastMCP
+from pathlib import Path
+from typing import Optional
+import os
+import re
+from jinja2 import Environment, FileSystemLoader
+
+# Global project directory (set by CLI args)
+_PROJECT_DIR: Optional[Path] = None
+
+# Initialize MCP server
+mcp = FastMCP(name="a4e-agent-creator")
+
+# Load templates
+template_dir = Path(__file__).parent / "templates"
+jinja_env = Environment(
+    loader=FileSystemLoader(template_dir),
+    autoescape=False,  # Explicit for code generation
+)
+
+
+def set_project_dir(path: Path) -> None:
+    """Set the global project directory."""
+    global _PROJECT_DIR
+    _PROJECT_DIR = path
+
+
+def get_configured_project_dir() -> Optional[Path]:
+    """Get the configured project directory."""
+    return _PROJECT_DIR
+
+
+def sanitize_input(value: str, allowed_chars: str = r"a-zA-Z0-9_-") -> str:
+    """
+    Sanitize user input to prevent template injection.
+
+    Args:
+        value: Input string to sanitize
+        allowed_chars: Regex character class of allowed characters
+
+    Returns:
+        Sanitized string with only allowed characters
+    """
+    pattern = f"[^{allowed_chars}]"
+    return re.sub(pattern, "", value)
+
+
+def get_project_dir(agent_name: Optional[str] = None) -> Path:
+    """
+    Resolve the agent project directory.
+
+    Priority (highest to lowest):
+    1. --project-dir CLI arg (explicit override)
+    2. A4E_WORKSPACE env var (set by editor via ${workspaceFolder})
+    3. Path.cwd() (fallback)
+
+    Note: Tools should prefer using explicit project_path parameter when available,
+    as this function relies on environment context that may not be available
+    in all MCP clients.
+
+    Args:
+        agent_name: Optional agent ID to resolve path for
+
+    Returns:
+        Path to agent directory or project root
+
+    Raises:
+        ValueError: If agent creation attempted in invalid location
+    """
+    global _PROJECT_DIR
+
+    root = None
+    workspace_env = os.environ.get("A4E_WORKSPACE", "")
+
+    # Priority 1: Explicit CLI override
+    if _PROJECT_DIR:
+        root = _PROJECT_DIR
+    # Priority 2: Workspace from editor (if properly expanded)
+    elif workspace_env and "${" not in workspace_env:
+        # Only use if the variable was expanded and path exists
+        workspace_path = Path(workspace_env)
+        if workspace_path.exists() and workspace_path.is_dir():
+            root = workspace_path.resolve()
+    
+    # Priority 3: Fallback to cwd
+    if root is None:
+        root = Path.cwd()
+
+    if not agent_name:
+        return root
+
+    # Safety: Warn if creating in HOME directory without explicit path
+    if root == Path.home():
+        raise ValueError(
+            f"Cannot create agent in HOME directory.\n"
+            f"\n"
+            f"Please specify project_path in your tool call:\n"
+            f'  initialize_project(name="{agent_name}", project_path="/path/to/your/project", ...)\n'
+            f"\n"
+            f"The agent will be created at: {{project_path}}/{agent_name}/"
+        )
+
+    # Create agent directly in the workspace root
+    return root / agent_name
+