@erdoai/cli 0.23.0 → 0.24.0

package/dist/index.js

@@ -490,9 +490,11 @@ var ErdoClient = class {
       "/v1/datasets"
     );
   }
-  queryDataset(slug, query, timezone) {
+  // The endpoint's field is `question` (QueryDataNaturalLanguageInput). Sending
+  // `query` made every `erdo datasets query` fail with "question is required".
+  queryDataset(slug, question, timezone) {
     return this.request("POST", `/v1/datasets/${encodeURIComponent(slug)}/query-nl`, {
-      query,
+      question,
       timezone
     });
   }
@@ -920,13 +922,13 @@ function awaitingApprovalMessage(threadID) {
     "the run paused for approval.",
     "  erdo approvals list --status pending",
     "  erdo approvals decide <id> --approve",
-    `  erdo agent messages ${threadID}`
+    `  erdo agent wait ${threadID}`
   ].join("\n");
 }
 function timedOutMessage(threadID) {
   return [
     "timed out waiting for the run to finish \u2014 it is still running on its thread.",
-    `  erdo agent messages ${threadID}`,
+    `  erdo agent wait ${threadID}`,
     `  erdo runs list --thread ${threadID}`
   ].join("\n");
 }
@@ -1829,6 +1831,22 @@ agentCmd.command("send <threadId> <message>").description("Send a message to a t
     fail(e);
   }
 });
+agentCmd.command("wait <threadId>").description("Re-attach to a thread's latest run and wait for it to finish; prints the answer").action(async (threadId) => {
+  try {
+    const client = new ErdoClient();
+    process.stderr.write("waiting\u2026");
+    const run = await client.waitForThreadRun(threadId, {
+      onTick: () => process.stderr.write(".")
+    });
+    process.stderr.write("\n");
+    if (!run) throw new Error(timedOutMessage(threadId));
+    if (run.status === "awaiting_approval") throw new Error(awaitingApprovalMessage(threadId));
+    if (run.error) throw new Error(run.error);
+    console.log(run.output || `(status: ${run.status})`);
+  } catch (e) {
+    fail(e);
+  }
+});
 agentCmd.command("thread").description("Create a thread; prints its id").option("-n, --name <name>", "thread name").option("-d, --datasets <csv>", "dataset UUIDs to attach").action(async (opts) => {
   try {
     const res = await new ErdoClient().createThread({

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@erdoai/cli",
-  "version": "0.23.0",
-  "description": "Erdo CLI — drive datasets, pages, and evals from the terminal or CI",
+  "version": "0.24.0",
+  "description": "Erdo CLI \u2014 drive datasets, pages, and evals from the terminal or CI",
   "type": "module",
   "bin": {
     "erdo": "dist/index.js"