@gobi-ai/cli 0.9.0 → 0.9.1

package/.claude-plugin/marketplace.json

@@ -4,12 +4,12 @@
     "name": "gobi-ai"
   },
   "description": "Claude Code plugin for the Gobi collaborative knowledge platform CLI",
-  "version": "0.8.0",
+  "version": "0.9.0",
   "plugins": [
     {
       "name": "gobi",
       "description": "Manage the Gobi collaborative knowledge platform from the command line. Search and ask brains, publish brain documents, create threads, manage sessions, generate images and videos.",
-      "version": "0.8.0",
+      "version": "0.9.0",
       "author": {
         "name": "gobi-ai"
       },

package/.claude-plugin/plugin.json

@@ -1,7 +1,7 @@
 {
   "name": "gobi",
   "description": "Manage the Gobi collaborative knowledge platform from the command line",
-  "version": "0.8.0",
+  "version": "0.9.0",
   "author": {
     "name": "gobi-ai"
   },

package/dist/commands/media.js

@@ -3,7 +3,7 @@ import { BASE_URL, POLL_MAX_DURATION_MS } from "../constants.js";
 import { getValidToken } from "../auth/manager.js";
 import { ApiError } from "../errors.js";
 import { isJsonMode, jsonOut, unwrapResp } from "./utils.js";
-// ── Polling helper ──
+// ── Helpers ──
 async function pollStatus(path, terminalStates, intervalMs = 3000) {
     const start = Date.now();
     while (Date.now() - start < POLL_MAX_DURATION_MS) {
@@ -12,10 +12,16 @@ async function pollStatus(path, terminalStates, intervalMs = 3000) {
         const status = data.status || "";
         if (terminalStates.includes(status))
             return data;
+        // If no status field but downloadUrl exists, treat as completed
+        if (!status && extractImageUrl(data))
+            return data;
         await new Promise((r) => setTimeout(r, intervalMs));
     }
     throw new Error(`Polling timed out after ${POLL_MAX_DURATION_MS / 1000}s`);
 }
+function extractImageUrl(data) {
+    return (data.downloadUrl || data.download_url || data.url);
+}
 export function registerMediaCommand(program) {
     const media = program
         .command("media")
@@ -278,10 +284,17 @@ export function registerMediaCommand(program) {
             jsonOut(data);
             return;
         }
+        const imgUrl = extractImageUrl(data);
+        const status = data.status || "queued";
         console.log(`Image generation started!\n` +
             `  Job ID: ${jobId}\n` +
-            `  Status: ${data.status || "queued"}\n` +
-            `  Check:  gobi media image-status ${jobId}`);
+            `  Status: ${status}`);
+        if (imgUrl) {
+            console.log(`  Download URL: ${imgUrl}`);
+        }
+        else if (status === "queued" || status === "inference_started" || status === "inference_working") {
+            console.log(`  Check:  gobi media image-status ${jobId}`);
+        }
     });
     media
         .command("image-edit")
@@ -311,9 +324,13 @@ export function registerMediaCommand(program) {
             jsonOut(data);
             return;
         }
+        const imgUrl = extractImageUrl(data);
         console.log(`Image edit started!\n` +
             `  Job ID: ${jobId}\n` +
             `  Status: ${data.status || "queued"}`);
+        if (imgUrl) {
+            console.log(`  Download URL: ${imgUrl}`);
+        }
     });
     media
         .command("image-inpaint")
@@ -345,35 +362,82 @@ export function registerMediaCommand(program) {
             jsonOut(data);
             return;
         }
+        const imgUrl = extractImageUrl(data);
         console.log(`Inpainting started!\n` +
             `  Job ID: ${jobId}\n` +
             `  Status: ${data.status || "queued"}`);
+        if (imgUrl) {
+            console.log(`  Download URL: ${imgUrl}`);
+        }
     });
     media
         .command("image-status <jobId>")
         .description("Check image generation job status.")
         .option("--wait", "Poll until a terminal state is reached")
         .action(async (jobId, opts) => {
+        let data;
         if (opts.wait) {
-            const data = await pollStatus(`/media-gen/images/${jobId}`, [
+            data = await pollStatus(`/media-gen/images/${jobId}`, [
                 "completed",
                 "failed",
                 "inference_complete",
                 "inference_failed",
             ]);
-            if (isJsonMode(media)) {
-                jsonOut(data);
-                return;
-            }
-            console.log(`Image job ${jobId} — status: ${data.status}`);
-            return;
         }
-        const resp = (await apiGet(`/media-gen/images/${jobId}`));
-        const data = unwrapResp(resp);
+        else {
+            const resp = (await apiGet(`/media-gen/images/${jobId}`));
+            data = unwrapResp(resp);
+        }
         if (isJsonMode(media)) {
             jsonOut(data);
             return;
         }
+        const imgUrl = extractImageUrl(data);
         console.log(`Image job ${jobId} — status: ${data.status || "unknown"}`);
+        if (imgUrl) {
+            console.log(`  Download URL: ${imgUrl}`);
+        }
+    });
+    media
+        .command("image-download <jobId>")
+        .description("Download a generated image.")
+        .option("--wait", "Poll until generation completes before downloading")
+        .option("--type <type>", "Image type (image, thumbnail, asset)")
+        .action(async (jobId, opts) => {
+        if (opts.wait) {
+            console.log(`Waiting for image job ${jobId} to complete…`);
+            await pollStatus(`/media-gen/images/${jobId}`, [
+                "completed",
+                "failed",
+                "inference_complete",
+                "inference_failed",
+            ]);
+        }
+        const token = await getValidToken();
+        const query = opts.type ? `?type=${opts.type}` : "";
+        const url = `${BASE_URL}/media-gen/images/${jobId}/download${query}`;
+        const res = await fetch(url, {
+            headers: { Authorization: `Bearer ${token}` },
+        });
+        if (!res.ok) {
+            const text = (await res.text()) || "(no body)";
+            throw new ApiError(res.status, `/media-gen/images/${jobId}/download`, text);
+        }
+        const contentType = res.headers.get("content-type") || "image/png";
+        const ext = contentType.includes("jpeg") || contentType.includes("jpg") ? "jpg"
+            : contentType.includes("webp") ? "webp"
+                : "png";
+        const filename = `${jobId}.${ext}`;
+        if (isJsonMode(media)) {
+            // In JSON mode, return base64-encoded image
+            const buffer = Buffer.from(await res.arrayBuffer());
+            jsonOut({ filename, contentType, size: buffer.length, base64: buffer.toString("base64") });
+            return;
+        }
+        // Write to file
+        const { writeFile } = await import("fs/promises");
+        const buffer = Buffer.from(await res.arrayBuffer());
+        await writeFile(filename, buffer);
+        console.log(`Image saved to ${filename} (${buffer.length} bytes)`);
     });
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@gobi-ai/cli",
-  "version": "0.9.0",
+  "version": "0.9.1",
   "description": "CLI client for the Gobi collaborative knowledge platform",
   "license": "MIT",
   "type": "module",