@arbidocs/cli 0.3.11 → 0.3.14

package/dist/index.js

@@ -5,8 +5,8 @@ var commander = require('commander');
 var fs = require('fs');
 var os = require('os');
 var path = require('path');
+var chalk2 = require('chalk');
 var sdk = require('@arbidocs/sdk');
-var chalk = require('chalk');
 var prompts = require('@inquirer/prompts');
 var child_process = require('child_process');
 var client = require('@arbidocs/client');
@@ -18,7 +18,7 @@ function _interopDefault (e) { return e && e.__esModule ? e : { default: e }; }
 var fs__default = /*#__PURE__*/_interopDefault(fs);
 var os__default = /*#__PURE__*/_interopDefault(os);
 var path__default = /*#__PURE__*/_interopDefault(path);
-var chalk__default = /*#__PURE__*/_interopDefault(chalk);
+var chalk2__default = /*#__PURE__*/_interopDefault(chalk2);
 
 var store = new sdk.FileConfigStore();
 function getConfig() {
@@ -38,6 +38,13 @@ function requireConfig() {
     throw err;
   }
 }
+function resolveConfig() {
+  const { config, source } = store.resolveConfigWithFallbacks();
+  if (source !== "config") {
+    console.error(chalk2__default.default.dim(`Using server URL from ${source}: ${config.baseUrl}`));
+  }
+  return config;
+}
 function getCredentials() {
   return store.getCredentials();
 }
@@ -54,32 +61,32 @@ function clearChatSession() {
   store.clearChatSession();
 }
 function success(msg) {
-  console.log(chalk__default.default.green(msg));
+  console.log(chalk2__default.default.green(msg));
 }
 function error(msg) {
-  console.error(chalk__default.default.red(msg));
+  console.error(chalk2__default.default.red(msg));
 }
 function warn(msg) {
-  console.error(chalk__default.default.yellow(msg));
+  console.error(chalk2__default.default.yellow(msg));
 }
 function label(key, value) {
-  console.log(`${chalk__default.default.bold(key)} ${value}`);
+  console.log(`${chalk2__default.default.bold(key)} ${value}`);
 }
 function dim(msg) {
-  console.log(chalk__default.default.dim(msg));
+  console.log(chalk2__default.default.dim(msg));
 }
 function bold(msg) {
-  console.log(chalk__default.default.bold(msg));
+  console.log(chalk2__default.default.bold(msg));
 }
 function status(s) {
   const lower = s.toLowerCase();
-  if (["healthy", "completed", "available", "online", "on"].includes(lower)) return chalk__default.default.green(s);
-  if (["failed", "error", "unavailable", "offline"].includes(lower)) return chalk__default.default.red(s);
-  if (["processing", "pending", "degraded", "warning"].includes(lower)) return chalk__default.default.yellow(s);
+  if (["healthy", "completed", "available", "online", "on"].includes(lower)) return chalk2__default.default.green(s);
+  if (["failed", "error", "unavailable", "offline"].includes(lower)) return chalk2__default.default.red(s);
+  if (["processing", "pending", "degraded", "warning"].includes(lower)) return chalk2__default.default.yellow(s);
   return s;
 }
 function ref(s) {
-  return chalk__default.default.cyan(s);
+  return chalk2__default.default.cyan(s);
 }
 
 // src/commands/config-cmd.ts
@@ -118,7 +125,9 @@ function registerConfigCommand(program2) {
     label("Server URL:", cfg.baseUrl);
     label("Domain:", cfg.deploymentDomain);
     label("Auto-update:", cfg.autoUpdate ? "on" : "off");
-    label("Notifications:", cfg.notifications ? "on" : "off");
+    label("Verbose:", cfg.verbose !== false ? "on" : "off");
+    label("Watch:", cfg.watch !== false ? "on" : "off");
+    label("Notifications:", cfg.notifications !== false ? "on" : "off");
     if (cfg.selectedWorkspaceId) {
       label("Workspace:", cfg.selectedWorkspaceId);
     }
@@ -126,7 +135,7 @@ function registerConfigCommand(program2) {
   config.command("notifications [on|off]").description("Toggle background WebSocket notifications").action((toggle) => {
     const cfg = getConfig();
     if (!toggle) {
-      label("Notifications:", cfg?.notifications ? "on" : "off");
+      label("Notifications:", cfg?.notifications !== false ? "on" : "off");
       return;
     }
     if (toggle !== "on" && toggle !== "off") {
@@ -136,6 +145,32 @@ function registerConfigCommand(program2) {
     updateConfig({ notifications: toggle === "on" });
     success(`Notifications: ${toggle}`);
   });
+  config.command("verbose [on|off]").description("Toggle verbose agent steps in ask (default: on)").action((toggle) => {
+    const cfg = getConfig();
+    if (!toggle) {
+      label("Verbose:", cfg?.verbose !== false ? "on" : "off");
+      return;
+    }
+    if (toggle !== "on" && toggle !== "off") {
+      error("Usage: arbi config verbose [on|off]");
+      process.exit(1);
+    }
+    updateConfig({ verbose: toggle === "on" });
+    success(`Verbose: ${toggle}`);
+  });
+  config.command("watch [on|off]").description("Toggle upload watch (default: on)").action((toggle) => {
+    const cfg = getConfig();
+    if (!toggle) {
+      label("Watch:", cfg?.watch !== false ? "on" : "off");
+      return;
+    }
+    if (toggle !== "on" && toggle !== "off") {
+      error("Usage: arbi config watch [on|off]");
+      process.exit(1);
+    }
+    updateConfig({ watch: toggle === "on" });
+    success(`Watch: ${toggle}`);
+  });
   config.command("alias").description('Set up shell alias A for "arbi ask"').action(() => {
     const rcPath = getShellRcPath();
     if (isAliasInstalled(rcPath)) {
@@ -3502,7 +3537,7 @@ function getLatestVersion(skipCache = false) {
   }
 }
 function getCurrentVersion() {
-  return "0.3.11";
+  return "0.3.14";
 }
 function readChangelog(fromVersion, toVersion) {
   try {
@@ -3555,17 +3590,17 @@ function showChangelog(fromVersion, toVersion) {
 async function checkForUpdates(autoUpdate) {
   try {
     const latest = getLatestVersion();
-    if (!latest || latest === "0.3.11") return;
+    if (!latest || latest === "0.3.14") return;
     if (autoUpdate) {
       warn(`
-Your arbi version is out of date (${"0.3.11"} \u2192 ${latest}). Updating...`);
+Your arbi version is out of date (${"0.3.14"} \u2192 ${latest}). Updating...`);
       child_process.execSync("npm install -g @arbidocs/cli@latest", { stdio: "inherit" });
-      showChangelog("0.3.11", latest);
+      showChangelog("0.3.14", latest);
       console.log(`Updated to ${latest}.`);
     } else {
       warn(
         `
-Your arbi version is out of date (${"0.3.11"} \u2192 ${latest}).
+Your arbi version is out of date (${"0.3.14"} \u2192 ${latest}).
 Run "arbi update" to upgrade, or "arbi update auto" to always stay up to date.`
       );
     }
@@ -3575,9 +3610,9 @@ Run "arbi update" to upgrade, or "arbi update auto" to always stay up to date.`
 function hintUpdateOnError() {
   try {
     const cached = readCache();
-    if (cached && cached.latest !== "0.3.11") {
+    if (cached && cached.latest !== "0.3.14") {
       warn(
-        `Your arbi version is out of date (${"0.3.11"} \u2192 ${cached.latest}). Run "arbi update".`
+        `Your arbi version is out of date (${"0.3.14"} \u2192 ${cached.latest}). Run "arbi update".`
       );
     }
   } catch {
@@ -3658,7 +3693,11 @@ function registerRegisterCommand(program2) {
   });
 }
 async function smartRegister(config, opts) {
-  const email = opts.email || process.env.ARBI_EMAIL || await promptInput("Email");
+  let email = opts.email || process.env.ARBI_EMAIL || await promptInput("Email");
+  if ((opts.email || process.env.ARBI_EMAIL) && !email.includes("@")) {
+    email = `${email}@${config.deploymentDomain}`;
+    console.log(`Using email: ${email}`);
+  }
   const arbi = client.createArbiClient({
     baseUrl: config.baseUrl,
     deploymentDomain: config.deploymentDomain,
@@ -3724,13 +3763,17 @@ Registered successfully as ${email}`);
   }
 }
 async function nonInteractiveRegister(config, opts) {
-  const email = opts.email || process.env.ARBI_EMAIL;
+  let email = opts.email || process.env.ARBI_EMAIL;
   const password2 = opts.password || process.env.ARBI_PASSWORD;
   const supportApiKey = process.env.SUPPORT_API_KEY;
   if (!email) {
     error("Email required. Use --email <email> or set ARBI_EMAIL");
     process.exit(1);
   }
+  if (!email.includes("@")) {
+    email = `${email}@${config.deploymentDomain}`;
+    console.log(`Using email: ${email}`);
+  }
   if (!password2) {
     error("Password required. Use --password <password> or set ARBI_PASSWORD");
     process.exit(1);
@@ -3837,14 +3880,62 @@ function registerStatusCommand(program2) {
     }
   });
 }
+var MAX_TASKS = 50;
+var MAX_AGE_MS = 7 * 24 * 60 * 60 * 1e3;
+function getTasksFile() {
+  const configDir = process.env.ARBI_CONFIG_DIR ?? path__default.default.join(os__default.default.homedir(), ".arbi");
+  return path__default.default.join(configDir, "tasks.json");
+}
+function ensureDir(filePath) {
+  const dir = path__default.default.dirname(filePath);
+  if (!fs__default.default.existsSync(dir)) {
+    fs__default.default.mkdirSync(dir, { recursive: true, mode: 448 });
+  }
+}
+function readTasks() {
+  try {
+    const content = fs__default.default.readFileSync(getTasksFile(), "utf-8");
+    return JSON.parse(content);
+  } catch {
+    return [];
+  }
+}
+function writeTasks(tasks) {
+  const filePath = getTasksFile();
+  ensureDir(filePath);
+  fs__default.default.writeFileSync(filePath, JSON.stringify(tasks, null, 2) + "\n", { mode: 384 });
+}
+function getTasks() {
+  const now = Date.now();
+  const tasks = readTasks().filter((t) => now - new Date(t.submittedAt).getTime() < MAX_AGE_MS);
+  return tasks;
+}
+function addTask(task) {
+  const tasks = [task, ...getTasks().filter((t) => t.id !== task.id)].slice(0, MAX_TASKS);
+  writeTasks(tasks);
+}
+function updateTaskStatus(id, status2) {
+  const tasks = readTasks();
+  const task = tasks.find((t) => t.id === id);
+  if (task) {
+    task.status = status2;
+    writeTasks(tasks);
+  }
+}
+function getLatestTask() {
+  const tasks = getTasks();
+  return tasks[0] ?? null;
+}
+
+// src/notifications.ts
 var activeConnection = null;
 function timestamp() {
   return (/* @__PURE__ */ new Date()).toLocaleTimeString("en-GB", { hour12: false });
 }
 function colorize(level, text) {
-  if (level === "success") return chalk__default.default.green(text);
-  if (level === "error") return chalk__default.default.red(text);
-  if (level === "warning") return chalk__default.default.yellow(text);
+  if (level === "success") return chalk2__default.default.green(text);
+  if (level === "error") return chalk2__default.default.red(text);
+  if (level === "warning") return chalk2__default.default.yellow(text);
   return text;
 }
 async function startBackgroundNotifications(baseUrl, accessToken) {
@@ -3854,6 +3945,9 @@ async function startBackgroundNotifications(baseUrl, accessToken) {
       baseUrl,
       accessToken,
       onMessage: (msg) => {
+        if (client.isMessageType(msg, "response_complete")) {
+          updateTaskStatus(msg.response_id, msg.status);
+        }
         const { text, level } = sdk.formatWsMessage(msg);
         process.stderr.write(`
 ${colorize(level, `[${timestamp()}] ${text}`)}
@@ -3865,18 +3959,18 @@ ${colorize(level, `[${timestamp()}] ${text}`)}
       onReconnecting: (attempt, maxRetries) => {
         process.stderr.write(
           `
-${chalk__default.default.yellow(`[${timestamp()}] Reconnecting... (${attempt}/${maxRetries})`)}
+${chalk2__default.default.yellow(`[${timestamp()}] Reconnecting... (${attempt}/${maxRetries})`)}
 `
         );
       },
       onReconnected: () => {
         process.stderr.write(`
-${chalk__default.default.green(`[${timestamp()}] Reconnected`)}
+${chalk2__default.default.green(`[${timestamp()}] Reconnected`)}
 `);
       },
       onReconnectFailed: () => {
         process.stderr.write(`
-${chalk__default.default.red(`[${timestamp()}] Reconnection failed`)}
+${chalk2__default.default.red(`[${timestamp()}] Reconnection failed`)}
 `);
       }
     });
@@ -3894,7 +3988,31 @@ process.on("SIGINT", () => {
 });
 
 // src/helpers.ts
+var CONNECTION_ERROR_HINTS = {
+  ECONNREFUSED: "Connection refused. Is the backend running?",
+  ECONNRESET: "Connection reset by server. The backend may have restarted.",
+  ENOTFOUND: "DNS resolution failed. Check the server URL.",
+  ETIMEDOUT: "Connection timed out. Check network connectivity.",
+  UNABLE_TO_VERIFY_LEAF_SIGNATURE: "TLS certificate cannot be verified. The cert may be expired or self-signed.",
+  CERT_HAS_EXPIRED: "TLS certificate has expired. Renew with manage-deployment.",
+  ERR_TLS_CERT_ALTNAME_INVALID: "TLS certificate hostname mismatch. Check the server URL.",
+  DEPTH_ZERO_SELF_SIGNED_CERT: "Self-signed TLS certificate. The cert may need to be renewed.",
+  SELF_SIGNED_CERT_IN_CHAIN: "Self-signed certificate in chain. The cert may need to be renewed."
+};
+function diagnoseConnectionError(err) {
+  const code = sdk.getErrorCode(err);
+  if (code && code in CONNECTION_ERROR_HINTS) {
+    return CONNECTION_ERROR_HINTS[code];
+  }
+  const msg = err instanceof Error ? err.message : "";
+  if (msg === "fetch failed" || msg.includes("fetch failed")) {
+    return "Network error connecting to the server. Run `arbi health` to diagnose.";
+  }
+  return void 0;
+}
 function formatCliError(err) {
+  const connectionHint = diagnoseConnectionError(err);
+  if (connectionHint) return connectionHint;
   if (err instanceof sdk.ArbiApiError && err.apiError && typeof err.apiError === "object") {
     const base = err.message;
     const apiErr = err.apiError;
@@ -3920,6 +4038,7 @@ function runAction(fn) {
 }
 async function resolveAuth() {
   try {
+    resolveConfig();
     return await sdk.resolveAuth(store);
   } catch (err) {
     if (err instanceof sdk.ArbiError) {
@@ -3931,8 +4050,9 @@ async function resolveAuth() {
 }
 async function resolveWorkspace(workspaceOpt) {
   try {
+    resolveConfig();
     const ctx = await sdk.resolveWorkspace(store, workspaceOpt);
-    if (getConfig()?.notifications) {
+    if (getConfig()?.notifications !== false) {
       startBackgroundNotifications(ctx.config.baseUrl, ctx.accessToken).catch(() => {
       });
     }
@@ -3946,7 +4066,7 @@ async function resolveWorkspace(workspaceOpt) {
   }
 }
 function printTable(columns, rows) {
-  console.log(chalk__default.default.bold(columns.map((c) => c.header.padEnd(c.width)).join("")));
+  console.log(chalk2__default.default.bold(columns.map((c) => c.header.padEnd(c.width)).join("")));
   for (const row of rows) {
     console.log(
       columns.map((c) => {
@@ -4234,11 +4354,11 @@ function registerDocsCommand(program2) {
           const reason = raw.error_reason || raw.status_details || raw.error || null;
           const docId = raw.external_id;
           if (reason) {
-            console.error(chalk__default.default.red(`
+            console.error(chalk2__default.default.red(`
 \u26A0 ${docId} processing failed: ${reason}`));
           } else {
             console.error(
-              chalk__default.default.red(`
+              chalk2__default.default.red(`
 \u26A0 ${docId} processing failed (no error details available)`)
             );
           }
@@ -4326,11 +4446,11 @@ function registerDocsCommand(program2) {
   );
 }
 function registerUploadCommand(program2) {
-  program2.command("upload <files...>").description("Upload documents to the active workspace").option("-w, --workspace <id>", "Workspace ID (defaults to selected workspace)").option("-W, --watch", "Watch document processing progress after upload").action(
-    (files, opts) => runAction(async () => {
-      for (const f of files) {
-        if (!fs__default.default.existsSync(f)) {
-          error(`File not found: ${f}`);
+  program2.command("add <paths...>").alias("upload").description("Add files, directories, or zip archives to the active workspace").option("-w, --workspace <id>", "Workspace ID (defaults to selected workspace)").option("-W, --watch", "Watch document processing progress after upload").option("--no-watch", "Skip watching document processing").action(
+    (paths, opts) => runAction(async () => {
+      for (const p of paths) {
+        if (!fs__default.default.existsSync(p)) {
+          error(`Path not found: ${p}`);
           process.exit(1);
         }
       }
@@ -4339,16 +4459,54 @@ function registerUploadCommand(program2) {
       );
       const uploadedDocs = /* @__PURE__ */ new Map();
       const auth = { baseUrl: config.baseUrl, accessToken, workspaceKeyHeader };
-      for (const filePath of files) {
-        const result = await sdk.documents.uploadLocalFile(auth, workspaceId, filePath);
-        success(`Uploaded: ${result.fileName} (${result.doc_ext_ids.join(", ")})`);
-        if (result.duplicates && result.duplicates.length > 0) {
-          warn(`  Duplicates: ${result.duplicates.join(", ")}`);
+      for (const filePath of paths) {
+        const stat = fs__default.default.statSync(filePath);
+        if (stat.isDirectory()) {
+          const result = await sdk.documentsNode.uploadDirectory(auth, workspaceId, filePath);
+          if (result.doc_ext_ids.length === 0) {
+            warn(`No supported files found in directory: ${filePath}`);
+            continue;
+          }
+          for (const [folder, info] of result.folders) {
+            success(
+              `  ${folder}: ${info.fileCount} file(s) \u2192 ${info.doc_ext_ids.length} uploaded`
+            );
+            if (info.duplicates.length > 0) {
+              warn(`    Duplicates: ${info.duplicates.join(", ")}`);
+            }
+          }
+          success(
+            `Uploaded directory: ${filePath} (${result.doc_ext_ids.length} document(s) total)`
+          );
+          for (const id of result.doc_ext_ids) uploadedDocs.set(id, filePath);
+        } else if (filePath.toLowerCase().endsWith(".zip")) {
+          const result = await sdk.documentsNode.uploadZip(auth, workspaceId, filePath);
+          if (result.doc_ext_ids.length === 0) {
+            warn(`No supported files found in zip: ${filePath}`);
+            continue;
+          }
+          for (const [folder, info] of result.folders) {
+            success(
+              `  ${folder}: ${info.fileCount} file(s) \u2192 ${info.doc_ext_ids.length} uploaded`
+            );
+            if (info.duplicates.length > 0) {
+              warn(`    Duplicates: ${info.duplicates.join(", ")}`);
+            }
+          }
+          success(`Uploaded zip: ${filePath} (${result.doc_ext_ids.length} document(s) total)`);
+          for (const id of result.doc_ext_ids) uploadedDocs.set(id, filePath);
+        } else {
+          const result = await sdk.documentsNode.uploadLocalFile(auth, workspaceId, filePath);
+          success(`Uploaded: ${result.fileName} (${result.doc_ext_ids.join(", ")})`);
+          if (result.duplicates && result.duplicates.length > 0) {
+            warn(`  Duplicates: ${result.duplicates.join(", ")}`);
+          }
+          for (const id of result.doc_ext_ids) uploadedDocs.set(id, result.fileName);
         }
-        for (const id of result.doc_ext_ids) uploadedDocs.set(id, result.fileName);
       }
       const isInteractive = process.stdout.isTTY === true;
-      const shouldWatch = opts.watch === true || files.length === 1 && isInteractive;
+      const watchPref = getConfig()?.watch !== false;
+      const shouldWatch = opts.watch === false ? false : opts.watch === true || watchPref && isInteractive;
       if (shouldWatch && uploadedDocs.size > 0) {
         const pending = new Set(uploadedDocs.keys());
         const failed = /* @__PURE__ */ new Map();
@@ -4446,22 +4604,55 @@ function registerDownloadCommand(program2) {
   );
 }
 function registerAskCommand(program2) {
-  program2.command("ask <question...>").description("Ask the RAG assistant a question (no quotes needed)").option("-w, --workspace <id>", "Workspace ID (defaults to selected workspace)").option("-c, --config <id>", "Config ext_id to use (e.g. cfg-xxx)").option("-n, --new", "Start a new conversation (ignore previous context)").option("-v, --verbose", "Show agent steps and tool calls").action(
+  program2.command("ask <question...>").description("Ask the RAG assistant a question (no quotes needed)").option("-w, --workspace <id>", "Workspace ID (defaults to selected workspace)").option("-c, --continue <msg-id>", "Continue from a specific message ID").option("--config <id>", "Config ext_id to use (e.g. cfg-xxx)").option("-n, --new", "Start a new conversation (ignore previous context)").option("-b, --background", "Submit as background task (fire and forget)").option("-q, --quiet", "Suppress agent steps and tool calls").option("--json", "Output in JSON format (background mode only)").action(
     (words, opts) => runAction(async () => {
       const question = words.join(" ");
       const { arbi, accessToken, workspaceKeyHeader, workspaceId, config } = await resolveWorkspace(opts.workspace);
-      const session = getChatSession();
-      const workspaceChanged = session.lastMessageExtId && session.workspaceId && session.workspaceId !== workspaceId;
       let previousResponseId = null;
-      if (opts.new) {
+      if (opts.continue) {
+        previousResponseId = opts.continue;
+      } else if (opts.new) {
         clearChatSession();
-      } else if (workspaceChanged) {
-        clearChatSession();
-      } else if (session.lastMessageExtId) {
-        previousResponseId = session.lastMessageExtId;
+      } else {
+        const session = getChatSession();
+        const workspaceChanged = session.lastMessageExtId && session.workspaceId && session.workspaceId !== workspaceId;
+        if (workspaceChanged) {
+          clearChatSession();
+        } else if (session.lastMessageExtId) {
+          previousResponseId = session.lastMessageExtId;
+        }
       }
       const docs = await sdk.documents.listDocuments(arbi);
       const docIds = docs.map((d) => d.external_id);
+      if (opts.background) {
+        const result2 = await sdk.responses.submitBackgroundQuery({
+          baseUrl: config.baseUrl,
+          accessToken,
+          workspaceKeyHeader,
+          workspaceId,
+          question,
+          docIds,
+          previousResponseId,
+          model: opts.config
+        });
+        addTask({
+          id: result2.id,
+          question: question.slice(0, 200),
+          workspaceId,
+          submittedAt: (/* @__PURE__ */ new Date()).toISOString(),
+          status: "queued",
+          model: result2.model
+        });
+        if (opts.json) {
+          console.log(JSON.stringify({ id: result2.id, status: result2.status }));
+        } else {
+          console.log(`Task queued: ${chalk2__default.default.cyan(result2.id)}`);
+          console.log(
+            chalk2__default.default.dim("Use `arbi task status` or `arbi task result` to check progress.")
+          );
+        }
+        return;
+      }
       let res;
       try {
         res = await sdk.assistant.queryAssistant({
@@ -4489,19 +4680,47 @@ function registerAskCommand(program2) {
           model: opts.config
         });
       }
+      const verbose = opts.quiet === true ? false : getConfig()?.verbose !== false;
+      let elapsedTime = null;
       const result = await sdk.streamSSE(res, {
         onToken: (content) => process.stdout.write(content),
         onAgentStep: (data) => {
-          if (opts.verbose) {
-            const focus = data.focus || data.step || "";
-            console.error(chalk__default.default.dim(`
-[agent] ${focus}`));
+          if (verbose) {
+            const label2 = sdk.formatAgentStepLabel(data);
+            if (label2) console.error(chalk2__default.default.dim(`
+[agent] ${label2}`));
           }
         },
-        onError: (message) => console.error(chalk__default.default.red(`
+        onElapsedTime: (t) => {
+          elapsedTime = t;
+        },
+        onError: (message) => console.error(chalk2__default.default.red(`
 Error: ${message}`))
       });
       process.stdout.write("\n");
+      const parts = [];
+      if (result.agentSteps.length > 0) {
+        let stepLabel = `${result.agentSteps.length} step${result.agentSteps.length === 1 ? "" : "s"}`;
+        if (result.toolCallCount > 0) {
+          stepLabel += ` (${result.toolCallCount} tool call${result.toolCallCount === 1 ? "" : "s"})`;
+        }
+        parts.push(stepLabel);
+      }
+      if (result.usage) {
+        parts.push(`${result.usage.total_tokens.toLocaleString()} tokens`);
+      }
+      if (result.context && result.context.context_window > 0) {
+        const used = result.context.all_llm_calls?.last_input_tokens ?? result.context.total_input;
+        parts.push(
+          `${used.toLocaleString()}/${result.context.context_window.toLocaleString()} context`
+        );
+      }
+      if (elapsedTime != null) {
+        parts.push(`${elapsedTime.toFixed(1)}s`);
+      }
+      if (parts.length > 0) {
+        console.error(chalk2__default.default.dim(`[${parts.join(" \xB7 ")}]`));
+      }
       if (result.assistantMessageExtId) {
         const updates = {
           lastMessageExtId: result.assistantMessageExtId,
@@ -4517,36 +4736,73 @@ Error: ${message}`))
   );
 }
 function colorize2(level, text) {
-  if (level === "success") return chalk__default.default.green(text);
-  if (level === "error") return chalk__default.default.red(text);
-  if (level === "warning") return chalk__default.default.yellow(text);
+  if (level === "success") return chalk2__default.default.green(text);
+  if (level === "error") return chalk2__default.default.red(text);
+  if (level === "warning") return chalk2__default.default.yellow(text);
   return text;
 }
 function registerWatchCommand(program2) {
-  program2.command("watch").description("Watch workspace activity in real time").option("-w, --workspace <id>", "Workspace ID (defaults to selected workspace)").action(
+  program2.command("watch").description("Watch workspace activity in real time").option("-w, --workspace <id>", "Workspace ID (defaults to selected workspace)").option("-t, --timeout <seconds>", "Auto-close after N seconds").option("-n, --count <n>", "Stop after N messages").option("--json", "Output NDJSON (one JSON object per line)").action(
     (opts) => runAction(async () => {
       const { config, accessToken, workspaceId } = await resolveWorkspace(opts.workspace);
-      console.log(`Watching workspace ${workspaceId}... (Ctrl+C to stop)`);
+      const timeoutSec = opts.timeout ? parseInt(opts.timeout, 10) : void 0;
+      const maxCount = opts.count ? parseInt(opts.count, 10) : void 0;
+      const jsonMode = opts.json ?? false;
+      if (!jsonMode) {
+        const parts = [`Watching workspace ${workspaceId}...`];
+        if (timeoutSec) parts.push(`(timeout: ${timeoutSec}s)`);
+        if (maxCount) parts.push(`(max: ${maxCount} messages)`);
+        parts.push("(Ctrl+C to stop)");
+        console.log(parts.join(" "));
+      }
+      let messageCount = 0;
       let onDone;
       const done = new Promise((r) => {
         onDone = r;
       });
-      await sdk.connectWebSocket({
+      const conn = await sdk.connectWebSocket({
         baseUrl: config.baseUrl,
         accessToken,
         onMessage: (msg) => {
-          const { text, level } = sdk.formatWsMessage(msg);
-          console.log(colorize2(level, text));
+          messageCount++;
+          if (jsonMode) {
+            console.log(JSON.stringify(msg));
+          } else {
+            const { text, level } = sdk.formatWsMessage(msg);
+            console.log(colorize2(level, text));
+          }
+          if (maxCount && messageCount >= maxCount) {
+            if (!jsonMode) console.log(chalk2__default.default.dim(`
+Reached ${maxCount} messages, closing.`));
+            conn.close();
+          }
         },
         onClose: (code, reason) => {
-          console.log(
-            chalk__default.default.yellow(`
+          if (!jsonMode) {
+            console.log(
+              chalk2__default.default.yellow(`
 Connection closed (code ${code}${reason ? ": " + reason : ""})`)
-          );
+            );
+          }
           onDone();
         }
       });
+      let timer;
+      if (timeoutSec) {
+        timer = setTimeout(() => {
+          if (!jsonMode) console.log(chalk2__default.default.dim(`
+Timeout (${timeoutSec}s), closing.`));
+          conn.close();
+        }, timeoutSec * 1e3);
+      }
+      const sigintHandler = () => {
+        if (timer) clearTimeout(timer);
+        conn.close();
+      };
+      process.on("SIGINT", sigintHandler);
       await done;
+      if (timer) clearTimeout(timer);
+      process.removeListener("SIGINT", sigintHandler);
     })()
   );
 }
@@ -5373,6 +5629,326 @@ Updated to ${latest}.`);
     success("Auto-update enabled. ARBI CLI will update automatically on login.");
   });
 }
+function registerQuickstartCommand(program2) {
+  program2.command("quickstart").description("Interactive setup wizard \u2014 configure, register/login, and select a workspace").argument("[url]", "Server URL (auto-detected if omitted)").action(async (url) => {
+    console.log("\nWelcome to ARBI CLI setup!\n");
+    const { config, source } = store.resolveConfigWithFallbacks();
+    if (url) {
+      const domain = new URL(url).hostname;
+      config.baseUrl = url.replace(/\/$/, "");
+      config.deploymentDomain = domain;
+      store.saveConfig(config);
+    }
+    const useDetected = await promptConfirm(`Use server ${config.baseUrl}?`, true);
+    if (!useDetected) {
+      const customUrl = await promptInput("Server URL");
+      try {
+        const domain = new URL(customUrl).hostname;
+        config.baseUrl = customUrl.replace(/\/$/, "");
+        config.deploymentDomain = domain;
+        store.saveConfig(config);
+      } catch {
+        error("Invalid URL");
+        process.exit(1);
+      }
+    } else if (source !== "config") {
+      store.saveConfig(config);
+    }
+    dim(`Server: ${config.baseUrl}`);
+    const action = await promptSelect("Do you have an account?", [
+      { name: "Yes, log me in", value: "login" },
+      { name: "No, register a new account", value: "register" }
+    ]);
+    const arbiClient = client.createArbiClient({
+      baseUrl: config.baseUrl,
+      deploymentDomain: config.deploymentDomain,
+      credentials: "omit"
+    });
+    await arbiClient.crypto.initSodium();
+    let email;
+    let password2;
+    if (action === "register") {
+      email = await promptInput("Email");
+      password2 = await promptPassword("Password");
+      const confirmPw = await promptPassword("Confirm password");
+      if (password2 !== confirmPw) {
+        error("Passwords do not match.");
+        process.exit(1);
+      }
+      const codeMethod = await promptSelect("Verification method", [
+        { name: "I have an invitation code", value: "code" },
+        { name: "Send me a verification email", value: "email" }
+      ]);
+      let verificationCode;
+      if (codeMethod === "code") {
+        verificationCode = await promptInput("Invitation code");
+      } else {
+        console.log("Sending verification email...");
+        const verifyResponse = await arbiClient.fetch.POST("/v1/user/verify-email", {
+          body: { email }
+        });
+        if (verifyResponse.error) {
+          error(`Failed to send verification email: ${JSON.stringify(verifyResponse.error)}`);
+          process.exit(1);
+        }
+        success("Verification email sent. Check your inbox.");
+        verificationCode = await promptInput("Verification code");
+      }
+      try {
+        const firstName = await promptInput("First name", false) || "User";
+        const lastName = await promptInput("Last name", false) || "";
+        await arbiClient.auth.register({
+          email,
+          password: password2,
+          verificationCode,
+          firstName,
+          lastName
+        });
+        success(`Registered as ${email}`);
+      } catch (err) {
+        error(`Registration failed: ${sdk.getErrorMessage(err)}`);
+        process.exit(1);
+      }
+    } else {
+      email = await promptInput("Email");
+      password2 = await promptPassword("Password");
+    }
+    try {
+      const { arbi } = await sdk.performPasswordLogin(config, email, password2, store);
+      success(`Logged in as ${email}`);
+      const { data: workspaces2 } = await arbi.fetch.GET("/v1/user/workspaces");
+      const wsList = workspaces2 || [];
+      const memberWorkspaces = wsList.filter((ws) => ws.users?.some((u) => u.email === email));
+      let workspaceId;
+      let workspaceName;
+      if (memberWorkspaces.length === 0) {
+        console.log("Creating your first workspace...");
+        const ws = await sdk.workspaces.createWorkspace(arbi, "My Workspace");
+        workspaceId = ws.external_id;
+        workspaceName = ws.name;
+      } else {
+        const choices = [
+          ...sdk.formatWorkspaceChoices(memberWorkspaces),
+          { name: "+ Create new workspace", value: "__new__", description: "" }
+        ];
+        const selected = await promptSelect("Select workspace", choices);
+        if (selected === "__new__") {
+          const name = await promptInput("Workspace name", false) || "My Workspace";
+          const ws = await sdk.workspaces.createWorkspace(arbi, name);
+          workspaceId = ws.external_id;
+          workspaceName = ws.name;
+        } else {
+          workspaceId = selected;
+          workspaceName = memberWorkspaces.find((w) => w.external_id === selected)?.name || "";
+        }
+      }
+      updateConfig({ selectedWorkspaceId: workspaceId });
+      console.log("");
+      success("Setup complete!");
+      console.log("");
+      console.log(`  Server:    ${ref(config.baseUrl)}`);
+      console.log(`  Account:   ${ref(email)}`);
+      console.log(`  Workspace: ${workspaceName} (${ref(workspaceId)})`);
+      console.log("");
+      dim('Try: arbi ask "hello"');
+    } catch (err) {
+      error(`Login failed: ${sdk.getErrorMessage(err)}`);
+      process.exit(1);
+    }
+  });
+}
+var CENTRAL_API_URL2 = "https://central.arbi.work";
+var DEFAULT_PASSWORD = "agent-dev-1234";
+async function getVerificationCode2(email, apiKey) {
+  const params = new URLSearchParams({ email });
+  const res = await fetch(`${CENTRAL_API_URL2}/license-management/verify-ci?${params.toString()}`, {
+    method: "GET",
+    headers: { "x-api-key": apiKey }
+  });
+  if (!res.ok) {
+    const body = await res.text().catch(() => "");
+    throw new Error(`Failed to get verification code: ${res.status} ${body}`);
+  }
+  const data = await res.json();
+  const words = data?.verification_words ?? data?.verification_code ?? null;
+  if (!words) throw new Error("No verification code in response");
+  return Array.isArray(words) ? words.join(" ") : String(words);
+}
+function registerAgentCreateCommand(program2) {
+  program2.command("agent-create").description("Create a bot/test account (requires SUPPORT_API_KEY)").argument("[url]", "Server URL (auto-detected if omitted)").option("-p, --password <password>", "Account password", DEFAULT_PASSWORD).option("--workspace-name <name>", "Workspace name", "Agent Workspace").option("--email <email>", "Custom email (default: agent-{timestamp}@{domain})").action(
+    async (url, opts) => {
+      const { config, source } = store.resolveConfigWithFallbacks();
+      if (url) {
+        const domain = new URL(url).hostname;
+        config.baseUrl = url.replace(/\/$/, "");
+        config.deploymentDomain = domain;
+        store.saveConfig(config);
+        dim(`Server: ${config.baseUrl}`);
+      } else {
+        dim(`Server: ${config.baseUrl} (from ${source})`);
+      }
+      const supportApiKey = process.env.SUPPORT_API_KEY;
+      if (!supportApiKey) {
+        error(
+          "SUPPORT_API_KEY is required.\nSet it with: export SUPPORT_API_KEY=<key>\nOr source from .env: source .env && arbi agent-create"
+        );
+        process.exit(1);
+      }
+      const timestamp2 = Date.now();
+      let email = opts.email || `agent-${timestamp2}@${config.deploymentDomain}`;
+      if (!email.includes("@")) {
+        email = `${email}@${config.deploymentDomain}`;
+      }
+      dim(`Email: ${email}`);
+      const arbiClient = client.createArbiClient({
+        baseUrl: config.baseUrl,
+        deploymentDomain: config.deploymentDomain,
+        credentials: "omit"
+      });
+      await arbiClient.crypto.initSodium();
+      const verifyResponse = await arbiClient.fetch.POST("/v1/user/verify-email", {
+        body: { email }
+      });
+      if (verifyResponse.error) {
+        error(`verify-email failed: ${JSON.stringify(verifyResponse.error)}`);
+        process.exit(1);
+      }
+      let verificationCode;
+      try {
+        verificationCode = await getVerificationCode2(email, supportApiKey);
+      } catch (err) {
+        error(`Failed to get verification code: ${sdk.getErrorMessage(err)}`);
+        process.exit(1);
+      }
+      try {
+        await arbiClient.auth.register({
+          email,
+          password: opts.password,
+          verificationCode,
+          firstName: "Agent",
+          lastName: `${timestamp2}`
+        });
+      } catch (err) {
+        error(`Registration failed: ${sdk.getErrorMessage(err)}`);
+        process.exit(1);
+      }
+      try {
+        const { arbi } = await sdk.performPasswordLogin(config, email, opts.password, store);
+        const ws = await sdk.workspaces.createWorkspace(arbi, opts.workspaceName);
+        updateConfig({ selectedWorkspaceId: ws.external_id });
+        console.log("");
+        success("Agent account created!");
+        console.log("");
+        console.log(`  Email:     ${ref(email)}`);
+        console.log(`  Password:  ${ref(opts.password)}`);
+        console.log(`  Workspace: ${ws.name} (${ref(ws.external_id)})`);
+        console.log(`  Server:    ${ref(config.baseUrl)}`);
+        console.log("");
+        dim('Ready to use: arbi ask "hello"');
+      } catch (err) {
+        error(`Post-registration setup failed: ${sdk.getErrorMessage(err)}`);
+        process.exit(1);
+      }
+    }
+  );
+}
+function formatAge(isoDate) {
+  const ms = Date.now() - new Date(isoDate).getTime();
+  if (ms < 6e4) return `${Math.round(ms / 1e3)}s`;
+  if (ms < 36e5) return `${Math.round(ms / 6e4)}m`;
+  if (ms < 864e5) return `${Math.round(ms / 36e5)}h`;
+  return `${Math.round(ms / 864e5)}d`;
+}
+function resolveTaskId(taskIdArg) {
+  if (taskIdArg) return taskIdArg;
+  const latest = getLatestTask();
+  if (!latest) {
+    console.error(chalk2__default.default.red('No tasks found. Submit one with: arbi ask -b "your question"'));
+    process.exit(1);
+  }
+  return latest.id;
+}
+function statusColor(s) {
+  if (s === "completed") return chalk2__default.default.green(s);
+  if (s === "failed") return chalk2__default.default.red(s);
+  if (s === "queued" || s === "in_progress") return chalk2__default.default.yellow(s);
+  return s;
+}
+function registerTaskCommand(program2) {
+  const task = program2.command("task").description("Manage background tasks");
+  task.action(async (_opts, cmd) => {
+    await cmd.commands.find((c) => c.name() === "list").parseAsync([], { from: "user" });
+  });
+  task.command("list").description("List background tasks").action(
+    () => runAction(async () => {
+      const tasks = getTasks();
+      if (tasks.length === 0) {
+        console.log('No tasks. Submit one with: arbi ask -b "your question"');
+        return;
+      }
+      printTable(
+        [
+          { header: "ID", width: 20, value: (r) => r.id },
+          { header: "STATUS", width: 14, value: (r) => statusColor(r.status) },
+          { header: "QUESTION", width: 42, value: (r) => r.question },
+          { header: "AGE", width: 6, value: (r) => formatAge(r.submittedAt) }
+        ],
+        tasks
+      );
+    })()
+  );
+  task.command("status [task-id]").description("Check current task status (defaults to most recent)").option("-w, --workspace <id>", "Workspace ID").option("--json", "Output in JSON format").action(
+    (taskId, opts) => runAction(async () => {
+      const id = resolveTaskId(taskId);
+      const { accessToken, workspaceKeyHeader, config } = await resolveWorkspace(opts?.workspace);
+      const result = await sdk.responses.getResponse(
+        { baseUrl: config.baseUrl, accessToken, workspaceKeyHeader },
+        id
+      );
+      if (result.status === "completed" || result.status === "failed") {
+        updateTaskStatus(id, result.status);
+      }
+      if (opts?.json) {
+        console.log(JSON.stringify(result, null, 2));
+      } else {
+        console.log(`${chalk2__default.default.bold("ID:")}     ${result.id}`);
+        console.log(`${chalk2__default.default.bold("Status:")} ${statusColor(result.status)}`);
+        if (result.model) console.log(`${chalk2__default.default.bold("Model:")}  ${result.model}`);
+        if (result.usage) {
+          console.log(`${chalk2__default.default.bold("Tokens:")} ${result.usage.total_tokens.toLocaleString()}`);
+        }
+      }
+    })()
+  );
+  task.command("result [task-id]").description("Fetch and print the completed task result").option("-w, --workspace <id>", "Workspace ID").option("--json", "Output in JSON format").action(
+    (taskId, opts) => runAction(async () => {
+      const id = resolveTaskId(taskId);
+      const { accessToken, workspaceKeyHeader, config } = await resolveWorkspace(opts?.workspace);
+      const result = await sdk.responses.getResponse(
+        { baseUrl: config.baseUrl, accessToken, workspaceKeyHeader },
+        id
+      );
+      if (result.status === "completed" || result.status === "failed") {
+        updateTaskStatus(id, result.status);
+      }
+      if (opts?.json) {
+        console.log(JSON.stringify(result, null, 2));
+        return;
+      }
+      if (result.status !== "completed") {
+        console.error(
+          chalk2__default.default.yellow(`Task is ${result.status}. Use \`arbi task status\` to check progress.`)
+        );
+        process.exit(1);
+      }
+      const text = sdk.responses.extractResponseText(result);
+      process.stdout.write(text + "\n");
+      if (result.usage) {
+        console.error(chalk2__default.default.dim(`[${result.usage.total_tokens.toLocaleString()} tokens]`));
+      }
+    })()
+  );
+}
 
 // src/index.ts
 console.debug = () => {
@@ -5383,7 +5959,7 @@ console.info = (...args) => {
   _origInfo(...args);
 };
 var program = new commander.Command();
-program.name("arbi").description("ARBI CLI \u2014 interact with ARBI from the terminal").version("0.3.11");
+program.name("arbi").description("ARBI CLI \u2014 interact with ARBI from the terminal").version("0.3.14");
 registerConfigCommand(program);
 registerLoginCommand(program);
 registerRegisterCommand(program);
@@ -5405,6 +5981,9 @@ registerAgentconfigCommand(program);
 registerHealthCommand(program);
 registerTuiCommand(program);
 registerUpdateCommand(program);
+registerQuickstartCommand(program);
+registerAgentCreateCommand(program);
+registerTaskCommand(program);
 program.parse();
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map