main.py

"""Entry point for running the orchestrated multi-agent demo."""

from __future__ import annotations

import argparse
import json
import logging
import sys

from orchestrator.orchestrator import Orchestrator


def configure_logging() -> None:
    logging.basicConfig(
        level=logging.INFO,
        format="%(asctime)s [%(levelname)s] %(name)s: %(message)s",
    )


def parse_args(argv: list[str] | None = None) -> argparse.Namespace:
    parser = argparse.ArgumentParser(description="Run the pure Python multi-agent demo.")
    parser.add_argument(
        "query",
        nargs="?",
        help="User query to process. If omitted, a sample query is used.",
    )
    return parser.parse_args(argv)


def main(argv: list[str] | None = None) -> int:
    configure_logging()
    args = parse_args(argv)
    query = args.query or "What are the key benefits of geometric multi-agent systems?"
    orchestrator = Orchestrator()
    result = orchestrator.run(query)
    print(json.dumps(result, indent=2, default=str))
    return 0


if __name__ == "__main__":  # pragma: no cover - CLI entry
    sys.exit(main())