File size: 931 Bytes
659433c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 | """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()
|