replay_cli.py

Go to the documentation of this file.

# Copyright (c) 2025-present Aquin Labs Private Limited. All Rights Reserved.
"""aquin replay simulation"""
 
import sys
 
from aquin.cli_flags import reject_legacy_output_flags
 
 
def _print_help() -> None:
    print("Replay stored simulation runs.")
    print("")
    print("Usage: aquin replay simulation --run_id <id>")
    print("       aquin replay simulation <run_id>")
    print("")
    print("List: aquin list simulation")
    print("Docs: https://aquin.app/docs/simulation/llm")
 
 
def cmd_replay(args: list[str]) -> None:
    if not args or args[0] in ("-h", "--help", "help"):
        _print_help()
        return
 
    sub = args[0]
    rest = args[1:]
 
    if sub == "watch":
        print("Removed. Training watch is no longer part of the CLI.", file=sys.stderr)
        print("Use the SDK: aquin.init() / run.log() / run.checkpoint() → ./aquin_run/", file=sys.stderr)
        sys.exit(1)
 
    if sub == "simulation":
        _cmd_replay_simulation(rest)
        return
 
    print(f"Unknown replay subcommand: {sub}")
    print("Run: aquin replay --help")
    sys.exit(1)
 
 
def _cmd_replay_simulation(args: list[str]) -> None:
    if not args or args[0] in ("-h", "--help", "help"):
        print("Usage: aquin replay simulation --run_id <id>")
        print("       aquin replay simulation <run_id>")
        sys.exit(0 if args and args[0] in ("-h", "--help", "help") else 1)
 
    reject_legacy_output_flags(args)
 
    from aquin.cli import _cmd_simulation_catalog
 
    # Accept positional run_id like replay watch.
    if args and not args[0].startswith("-"):
        rest = ["--run_id", args[0], *args[1:]]
    else:
        rest = args
 
    _cmd_simulation_catalog("replay simulation", "load_simulation_run", rest)