scripts/modal_smoke_enabled_models.py

from __future__ import annotations

import argparse
import subprocess
import sys
from pathlib import Path


sys.path.insert(0, str(Path(__file__).resolve().parents[1]))

from token_holdem.agents import ROSTER
from token_holdem.model_runtime import configured_modal_model_names


def main() -> None:
    parser = argparse.ArgumentParser(description="Smoke-test Modal inference for enabled Token Hold'em model seats.")
    parser.add_argument("--run", action="store_true", help="Run commands instead of printing them.")
    args = parser.parse_args()

    enabled = configured_modal_model_names()
    names = [profile.name for profile in ROSTER if profile.name in enabled]
    if not names:
        raise SystemExit("No Modal models are enabled by TOKEN_HOLDEM_MODAL_MODEL_NAMES.")

    for name in names:
        command = ["uv", "run", "modal", "run", "modal_inference.py::smoke", "--model-name", name]
        if args.run:
            subprocess.run(command, check=True)
        else:
            print(" ".join(command))


if __name__ == "__main__":
    main()