embedding-studio-cli 0.1.2 → 0.2.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "embedding-studio-cli",
-  "version": "0.1.2",
+  "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"

package/studio.js

@@ -79,6 +79,20 @@ 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 = { _: [] };
@@ -185,17 +199,17 @@ 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", "template": "value", "questions": "value" });
+    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 text = a.text !== undefined ? a.text : await readStdin();
-    const body = { text };
+    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 }));