embedding-studio-cli 0.1.0 → 0.2.0

package/package.json

@@ -1,9 +1,9 @@
 {
   "name": "embedding-studio-cli",
-  "version": "0.1.0",
+  "version": "0.2.0",
   "description": "The Embedding Studio API from a terminal: push labelled data, fit and calibrate, place the review band, decide, and download the calibration file. JSON out, non-zero exit on refusal.",
   "bin": {
-    "studio": "./studio.js"
+    "studio": "studio.js"
   },
   "files": [
     "studio.js"

package/studio.js

@@ -79,11 +79,29 @@ function readStdin() {
   });
 }
 
+/* The record, whichever way it arrived: --file sends the bytes (an image, recording or clip; the
+ * server sniffs what they are, so there is no --type flag to get wrong), --text or stdin the
+ * words. --simple asks for plain values back: {answers: {qid: value}, route, confidence}. */
+async function recordBody(a) {
+  const body = {};
+  if (a.file !== undefined) {
+    body.input = { data: fs.readFileSync(a.file).toString("base64") };
+  } else {
+    body.text = a.text !== undefined ? a.text : await readStdin();
+  }
+  if (a.simple) body.simple = true;
+  return body;
+}
+
 /* ---- a small flag parser: --name value, --flag, and positionals, nothing cleverer ---- */
 function parse(argv, spec) {
   const out = { _: [] };
   for (let i = 0; i < argv.length; i++) {
     const a = argv[i];
+    if (a === "-o") {                      /* the one short flag argparse gives the Python side */
+      out.o = argv[++i];
+      continue;
+    }
     if (a.startsWith("--")) {
       const name = a.slice(2);
       if (!(name in spec)) fail(`unknown flag --${name}`, 2);
@@ -181,12 +199,21 @@ const COMMANDS = {
     } else process.stdout.write(buf.toString("utf8"));
   },
   "decide": async (argv) => {
-    const a = parse(argv, { "text": "value", "idempotency-key": "value" });
-    const text = a.text !== undefined ? a.text : await readStdin();
-    const body = { text };
+    const a = parse(argv, { "text": "value", "file": "value", "simple": "flag",
+                            "idempotency-key": "value" });
+    const body = await recordBody(a);
     if (a["idempotency-key"]) body.idempotency_key = a["idempotency-key"];
     say(await call("POST", `/v1/projects/${a._[0]}/decide`, { body }));
   },
+  "ask": async (argv) => {
+    const a = parse(argv, { "text": "value", "file": "value", "simple": "flag",
+                            "template": "value", "questions": "value" });
+    if (!a.template === !a.questions) fail("ask needs exactly one of --template or --questions", 2);
+    const body = await recordBody(a);
+    if (a.template) body.template = a.template;
+    else body.questions = JSON.parse(fs.readFileSync(a.questions, "utf8"));
+    say(await call("POST", "/v1/decide", { body }));
+  },
   "decisions": async (argv) =>
     say(await call("GET", `/v1/projects/${parse(argv, {})._[0]}/decisions`)),
   "audit": async (argv) => say(await call("GET", `/v1/projects/${parse(argv, {})._[0]}/audit`)),