| """JSONL inference CLI; compiles but never executes tools.""" | |
| import argparse | |
| import json | |
| import sys | |
| from core import compile_js | |
| from model import Model | |
| def main(): | |
| ap = argparse.ArgumentParser(description=__doc__) | |
| ap.add_argument( | |
| "--checkpoint", default=None, help="NPZ weights (default: packaged CNN)" | |
| ) | |
| ap.add_argument("--js", action="store_true") | |
| args = ap.parse_args() | |
| model = Model(args.checkpoint) | |
| for line in sys.stdin: | |
| try: | |
| item = json.loads(line) | |
| plan = model.predict(item) | |
| print( | |
| json.dumps( | |
| {"plan": plan, "js": compile_js(plan, item["tools"])} | |
| if args.js | |
| else {"plan": plan} | |
| ) | |
| ) | |
| except (ValueError, KeyError, IndexError) as e: | |
| print(json.dumps({"diagnostic": str(e)})) | |
| if __name__ == "__main__": | |
| main() | |