33 lines
853 B
Python
33 lines
853 B
Python
#!/usr/bin/env python3
|
|
"""Run the no-business-side-effect NemoTron shared reservation canary."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import json
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
|
|
ROOT = Path(__file__).resolve().parents[2]
|
|
if str(ROOT) not in sys.path:
|
|
sys.path.insert(0, str(ROOT))
|
|
|
|
from services.nemotron_dispatch_reservation_service import ( # noqa: E402
|
|
PRODUCTION_STATE_PATH,
|
|
run_shared_reservation_canary,
|
|
)
|
|
|
|
|
|
def main() -> int:
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--state-path", default=str(PRODUCTION_STATE_PATH))
|
|
args = parser.parse_args()
|
|
payload = run_shared_reservation_canary(state_path=args.state_path)
|
|
print(json.dumps(payload, ensure_ascii=False, sort_keys=True))
|
|
return 0 if payload["success"] else 1
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|