import argparse
from nato1000_runtime.config import RuntimeConfig
from nato1000_runtime.runtime import NATO1000Runtime

parser = argparse.ArgumentParser(description="NATO-1000 runtime")
parser.add_argument("prompt")
parser.add_argument("--model-id", default="NATO1000-SENTINEL")
args = parser.parse_args()
cfg = RuntimeConfig(model_id=args.model_id)
cfg.validate()
runtime = NATO1000Runtime(cfg)
print(runtime.run(args.prompt).text)
print(runtime.health())