TTI / Release /scripts /check_vlac_services.py
JosephBai's picture
Upload folder using huggingface_hub
857c2e9 verified
#!/usr/bin/env python3
from __future__ import annotations
import argparse
import sys
import requests
def main() -> None:
parser = argparse.ArgumentParser(description="Check VLAC service /healthcheck endpoints.")
parser.add_argument(
"--urls",
required=True,
help="Comma-separated base URLs, e.g. http://localhost:8111,http://localhost:8112",
)
parser.add_argument("--timeout", type=float, default=10.0, help="Request timeout in seconds")
args = parser.parse_args()
urls = [u.strip().rstrip("/") for u in args.urls.split(",") if u.strip()]
if not urls:
print("No URLs provided.", file=sys.stderr)
sys.exit(2)
ok_any = False
for base in urls:
try:
resp = requests.post(f"{base}/healthcheck", timeout=args.timeout)
if resp.ok:
print(f"OK {base}")
ok_any = True
else:
print(f"BAD {base} status={resp.status_code}", file=sys.stderr)
except Exception as exc:
print(f"ERR {base} {exc}", file=sys.stderr)
sys.exit(0 if ok_any else 1)
if __name__ == "__main__":
main()