cli.py

"""
AgentMesh CLI — typer-based entry point.

Usage:
    agentmesh serve          # start the API server
    agentmesh research       # run research_mesh example
    agentmesh summarise      # run summary_mesh example
    agentmesh mesh           # print current mesh state (requires running server)
"""

import typer
import uvicorn

app = typer.Typer(name="agentmesh", help="AgentMesh AI Orchestration System")


@app.command()
def serve(
    host: str = typer.Option("0.0.0.0", help="Bind host"),
    port: int = typer.Option(8000, help="Bind port"),
    reload: bool = typer.Option(False, help="Auto-reload on code changes"),
):
    """Start the AgentMesh API server."""
    uvicorn.run("api.server:app", host=host, port=port, reload=reload)


@app.command()
def research():
    """Run the 3-agent research pipeline demo."""
    import asyncio
    from examples.research_mesh import main
    asyncio.run(main())


@app.command()
def summarise():
    """Run the hierarchical summarisation demo."""
    import asyncio
    from examples.summary_mesh import main
    asyncio.run(main())


@app.command()
def mesh(
    url: str = typer.Option("http://localhost:8000", help="API base URL"),
):
    """Print the current mesh state from a running server."""
    import httpx
    try:
        resp = httpx.get(f"{url}/mesh", timeout=5)
        import json
        typer.echo(json.dumps(resp.json(), indent=2))
    except Exception as e:
        typer.echo(f"Error: {e}", err=True)
        raise typer.Exit(1)


if __name__ == "__main__":
    app()