@assethub/cli 0.1.2 → 0.1.3

package/README.md

@@ -13,6 +13,7 @@ assethub image generate --prompt "stylized prop concept" --wait
 assethub mesh generate --file ./input.png --model-id meshGen.hunyuan31 --wait
 assethub jobs watch job_123 --download --out-dir ./out/job
 assethub parts split --file ./input.png --part-extractor "V1.5" --wait --all-ready --download --out-dir ./out/parts
+assethub parts compare --file ./input.png --preprocess-prompt "clean white background, centered product photo" --part-extractor "V1.5" --wait --all-ready
 ```
 
 Configuration:
@@ -28,6 +29,7 @@ printf '%s' "$ASSETHUB_API_KEY" | assethub auth login --api-key-stdin
 assethub image generate --prompt "small stylized game asset chest" --wait --download --out-dir ./out/image
 assethub mesh generate --source-url https://example.com/input.png --model-id meshGen.hunyuan31 --wait --download --out-dir ./out/mesh
 cat ./character.png | assethub parts split --stdin --file-name character.png --part-extractor "V1.5" --wait --all-ready --download --out-dir ./out/parts
+cat ./character.png | assethub parts compare --stdin --file-name character.png --preprocess-prompt "clean white background" --part-extractor "V1.5" --wait --all-ready
 ```
 
 Agent-oriented input:
@@ -48,6 +50,17 @@ Agent-oriented output:
 - `--download --out-dir <dir>` recursively downloads public artifact URLs from
   job or production status payloads and writes a `manifest.json`.
 
+`assethub parts compare` runs the normal part split first. When
+`--preprocess-prompt <text>` is provided, it then uses the Nano Banana image
+model to generate a preprocessed image, waits for that image job to finish, and
+runs a second split on the generated image. The final JSON contains `direct`
+and `preprocessed` sections so agents can compare both outputs. If
+preprocessing fails, the direct result is still printed with
+`preprocessed.ok: false`; pass `--fail-on-preprocess-error` when a non-zero exit
+is preferred. Use `--all-ready` rather than explicit `--task-id` values when
+preprocessing is enabled, because the generated-image split creates a separate
+production order.
+
 Image generation defaults to the deployed `v1` endpoint. Pass `--api-version v2`
 only when the `v2` image endpoint is available in the target environment.
 

package/dist/index.js

@@ -69,6 +69,7 @@ Usage:
   assethub production status <order-id> [--wait] [--download --out-dir <dir>]
   assethub production execute --order-id <id> --mission-id <id> --task-id <id> [--task-id <id>...] [--part-extractor <name>] [--part-extraction-mode fast|high_quality] [--wait] [--download --out-dir <dir>]
   assethub parts split (--file <path> | --source-url <url> | --source-id <id> | --image-url <url> | --file-ref-json <json> | --stdin | --data-uri <uri> | --order-id <id>) [--part-extractor <name>] [--wait] [--task-id <id>...] [--mission-id <id>] [--all-ready] [--download --out-dir <dir>]
+  assethub parts compare (--file <path> | --source-url <url> | --source-id <id> | --image-url <url> | --file-ref-json <json> | --stdin | --data-uri <uri>) [--preprocess-prompt <text>] [--preprocess-model-id <id>] [--fail-on-preprocess-error] [--part-extractor <name>] [--wait] [--task-id <id>...] [--mission-id <id>] [--all-ready] [--download --out-dir <dir>]
 
 Global options:
   --api-key <key>       Overrides saved auth and ASSETHUB_API_KEY.
@@ -88,6 +89,7 @@ Examples:
   assethub mesh generate --file ./input.png --model-id meshGen.hunyuan31 --wait
   assethub production analyze --file ./input.png --part-extractor "V1.5" --wait
   assethub parts split --file ./input.png --part-extractor "V1.5" --wait --all-ready
+  assethub parts compare --file ./input.png --preprocess-prompt "clean white background, centered product photo" --part-extractor "V1.5" --wait --all-ready
 `;
 const print = (value) => {
     stdout.write(`${JSON.stringify(value, null, 2)}\n`);
@@ -458,6 +460,24 @@ const resolveProductionAnalyzeSource = async (ctx) => {
     }
     return { imageBlobLocation: source.fileRef };
 };
+const productionAnalyzeSourceFromSource = (source) => {
+    if (source.resourceId != null) {
+        return { imageAssetId: source.resourceId };
+    }
+    if (source.url != null) {
+        return { imageUrl: source.url };
+    }
+    return { imageBlobLocation: source.fileRef };
+};
+const sourceFromProductionAnalyzeSource = (source) => {
+    if (source.imageAssetId != null) {
+        return { resourceId: source.imageAssetId };
+    }
+    if (source.imageUrl != null) {
+        return { url: source.imageUrl };
+    }
+    return { fileRef: source.imageBlobLocation };
+};
 const productionAnalyzeTerminalStatuses = new Set(['ready', 'failed']);
 const productionExecuteTerminalStatuses = new Set(['completed', 'failed']);
 const pollProductionStatus = async ({ ctx, orderId, terminalStatuses, }) => {
@@ -885,6 +905,106 @@ const withOptionalDownloads = async (flags, value) => {
     const downloads = await maybeDownloadUrls(flags, value);
     return downloads == null ? value : { ...value, downloads };
 };
+const firstStringValueForKeys = (value, keys) => {
+    if (value == null || typeof value !== 'object') {
+        return undefined;
+    }
+    if (Array.isArray(value)) {
+        for (const item of value) {
+            const found = firstStringValueForKeys(item, keys);
+            if (found != null) {
+                return found;
+            }
+        }
+        return undefined;
+    }
+    for (const [key, entry] of Object.entries(value)) {
+        if (keys.has(key) && typeof entry === 'string' && entry.trim() !== '') {
+            return entry;
+        }
+        const found = firstStringValueForKeys(entry, keys);
+        if (found != null) {
+            return found;
+        }
+    }
+    return undefined;
+};
+const firstBlobLocation = (value) => {
+    if (value == null || typeof value !== 'object') {
+        return undefined;
+    }
+    if (Array.isArray(value)) {
+        for (const item of value) {
+            const found = firstBlobLocation(item);
+            if (found != null) {
+                return found;
+            }
+        }
+        return undefined;
+    }
+    const record = value;
+    if (record.type === 'supabase' && typeof record.bucket === 'string' && typeof record.path === 'string') {
+        return {
+            type: 'supabase',
+            bucket: record.bucket,
+            path: record.path,
+        };
+    }
+    if (record.type === 'signedSupabase' && typeof record.signedUrl === 'string') {
+        return {
+            type: 'signedSupabase',
+            signedUrl: record.signedUrl,
+        };
+    }
+    if (record.type === 'asset' && typeof record.assetId === 'string') {
+        return {
+            type: 'asset',
+            assetId: record.assetId,
+        };
+    }
+    for (const entry of Object.values(record)) {
+        const found = firstBlobLocation(entry);
+        if (found != null) {
+            return found;
+        }
+    }
+    return undefined;
+};
+const imageSourceFromCompletedJob = (job) => {
+    if (job.resourceId != null && job.resourceId.trim() !== '') {
+        return { imageAssetId: job.resourceId };
+    }
+    const resourceId = firstStringValueForKeys(job, new Set(['resourceId', 'resource_id', 'assetId', 'asset_id', 'imageAssetId']));
+    if (resourceId != null) {
+        return { imageAssetId: resourceId };
+    }
+    const imageUrl = firstStringValueForKeys(job, new Set(['imageUrl', 'image_url', 'url', 'publicUrl', 'public_url']));
+    if (imageUrl != null) {
+        return { imageUrl };
+    }
+    const blobLocation = firstBlobLocation(job);
+    if (blobLocation != null) {
+        return { imageBlobLocation: blobLocation };
+    }
+    throw new Error('Preprocessing completed, but the image job did not include an image source for part splitting.');
+};
+const errorToOutput = (error) => {
+    if (error instanceof AssetHubApiError) {
+        return {
+            name: error.name,
+            message: error.message,
+            status: error.status,
+            code: error.code,
+        };
+    }
+    if (error instanceof Error) {
+        return {
+            name: error.name,
+            message: error.message,
+        };
+    }
+    return { message: String(error) };
+};
 const readyTaskIdsFromStatus = (status) => {
     const missions = status != null && typeof status === 'object' && 'missions' in status
         ? status.missions
@@ -996,83 +1116,163 @@ const commandProduction = async (subcommand, positionals, ctx) => {
     throw new Error('Unknown production command. Use "production analyze", "production status", or "production execute".');
 };
 const commandParts = async (subcommand, ctx) => {
-    if (subcommand !== 'split') {
-        throw new Error('Unknown parts command. Use "parts split".');
-    }
-    let orderId = getFlag(ctx.flags, 'order-id');
-    let analyzed;
-    let status;
-    if (orderId == null) {
-        const result = await ctx.client.v1.analyzeProduction({
-            ...(await resolveProductionAnalyzeSource(ctx)),
-            agentVersion: resolvePartExtractorForAnalyze(ctx.flags),
-            name: getFlag(ctx.flags, 'name'),
-        });
-        analyzed = result;
-        orderId = result.orderId;
-        if (hasFlag(ctx.flags, 'wait') || hasFlag(ctx.flags, 'all-ready')) {
+    const runPartSplit = async (source) => {
+        let orderId = source == null ? getFlag(ctx.flags, 'order-id') : undefined;
+        let analyzed;
+        let status;
+        if (orderId == null) {
+            const result = await ctx.client.v1.analyzeProduction({
+                ...(source ?? (await resolveProductionAnalyzeSource(ctx))),
+                agentVersion: resolvePartExtractorForAnalyze(ctx.flags),
+                name: getFlag(ctx.flags, 'name'),
+            });
+            analyzed = result;
+            orderId = result.orderId;
+            if (hasFlag(ctx.flags, 'wait') || hasFlag(ctx.flags, 'all-ready')) {
+                status = await pollProductionStatus({
+                    ctx,
+                    orderId,
+                    terminalStatuses: productionAnalyzeTerminalStatuses,
+                });
+            }
+        }
+        else if (hasFlag(ctx.flags, 'wait') || hasFlag(ctx.flags, 'all-ready')) {
             status = await pollProductionStatus({
                 ctx,
                 orderId,
                 terminalStatuses: productionAnalyzeTerminalStatuses,
             });
         }
-    }
-    else if (hasFlag(ctx.flags, 'wait') || hasFlag(ctx.flags, 'all-ready')) {
-        status = await pollProductionStatus({
-            ctx,
-            orderId,
-            terminalStatuses: productionAnalyzeTerminalStatuses,
+        let taskIds = getFlagValues(ctx.flags, 'task-id');
+        if (hasFlag(ctx.flags, 'all-ready')) {
+            if (status == null) {
+                status = await ctx.client.v1.getProductionStatus(orderId);
+            }
+            taskIds = readyTaskIdsFromStatus(status);
+            if (taskIds.length === 0) {
+                throw new Error('No ready tasks were found for --all-ready.');
+            }
+        }
+        let executed;
+        let executionStatus;
+        if (taskIds.length > 0) {
+            if (status == null) {
+                status = await ctx.client.v1.getProductionStatus(orderId);
+            }
+            const missionId = getFlag(ctx.flags, 'mission-id') ?? singleMissionIdFromStatus(status);
+            const partExtractorFlag = getPartExtractorFlag(ctx.flags);
+            const agentVersion = partExtractorFlag == null
+                ? await resolvePartExtractorFromMission(ctx, orderId, missionId)
+                : resolvePartExtractorPublicName(partExtractorFlag);
+            const partExtractionMode = getFlag(ctx.flags, 'part-extraction-mode');
+            if (partExtractionMode != null &&
+                partExtractionMode !== 'fast' &&
+                partExtractionMode !== 'high_quality') {
+                throw new Error('--part-extraction-mode must be "fast" or "high_quality"');
+            }
+            executed = await ctx.client.v1.executeProduction({
+                orderId,
+                missionId,
+                confirmedTaskIds: taskIds,
+                agentVersion,
+                partExtractionMode,
+            });
+            if (hasFlag(ctx.flags, 'wait')) {
+                executionStatus = await pollProductionStatus({
+                    ctx,
+                    orderId,
+                    terminalStatuses: productionExecuteTerminalStatuses,
+                });
+            }
+        }
+        return sanitizeProductionOutput({
+            analyzed,
+            status,
+            executed,
+            executionStatus,
         });
+    };
+    if (subcommand === 'split') {
+        print(await withOptionalDownloads(ctx.flags, await runPartSplit()));
+        return;
     }
-    let taskIds = getFlagValues(ctx.flags, 'task-id');
-    if (hasFlag(ctx.flags, 'all-ready')) {
-        if (status == null) {
-            status = await ctx.client.v1.getProductionStatus(orderId);
+    if (subcommand === 'compare') {
+        if (getFlag(ctx.flags, 'order-id') != null) {
+            throw new Error('parts compare requires an image source. Use --file, --source-url, --source-id, --image-url, --file-ref-json, --stdin, or --data-uri.');
         }
-        taskIds = readyTaskIdsFromStatus(status);
-        if (taskIds.length === 0) {
-            throw new Error('No ready tasks were found for --all-ready.');
+        const preprocessPrompt = getFlag(ctx.flags, 'preprocess-prompt');
+        if (preprocessPrompt === 'true') {
+            throw new Error('Missing required flag value: --preprocess-prompt <text>');
         }
-    }
-    let executed;
-    let executionStatus;
-    if (taskIds.length > 0) {
-        if (status == null) {
-            status = await ctx.client.v1.getProductionStatus(orderId);
+        if (preprocessPrompt != null && getFlagValues(ctx.flags, 'task-id').length > 0) {
+            throw new Error('parts compare preprocessing creates a second order. Use --all-ready instead of --task-id.');
         }
-        const missionId = getFlag(ctx.flags, 'mission-id') ?? singleMissionIdFromStatus(status);
-        const partExtractorFlag = getPartExtractorFlag(ctx.flags);
-        const agentVersion = partExtractorFlag == null
-            ? await resolvePartExtractorFromMission(ctx, orderId, missionId)
-            : resolvePartExtractorPublicName(partExtractorFlag);
-        const partExtractionMode = getFlag(ctx.flags, 'part-extraction-mode');
-        if (partExtractionMode != null &&
-            partExtractionMode !== 'fast' &&
-            partExtractionMode !== 'high_quality') {
-            throw new Error('--part-extraction-mode must be "fast" or "high_quality"');
+        if (preprocessPrompt != null && getFlag(ctx.flags, 'mission-id') != null) {
+            throw new Error('parts compare preprocessing creates a second order. Omit --mission-id and let the CLI infer it.');
         }
-        executed = await ctx.client.v1.executeProduction({
-            orderId,
-            missionId,
-            confirmedTaskIds: taskIds,
-            agentVersion,
-            partExtractionMode,
-        });
-        if (hasFlag(ctx.flags, 'wait')) {
-            executionStatus = await pollProductionStatus({
-                ctx,
-                orderId,
-                terminalStatuses: productionExecuteTerminalStatuses,
-            });
+        const imageUrlFlag = getFlag(ctx.flags, 'image-url');
+        if (imageUrlFlag === 'true') {
+            throw new Error('Missing required flag value: --image-url <url>');
+        }
+        const inputSource = imageUrlFlag == null
+            ? productionAnalyzeSourceFromSource(await requireSource({
+                flags: ctx.flags,
+                client: ctx.client,
+                fileMediaType: 'image',
+            }))
+            : { imageUrl: imageUrlFlag };
+        const direct = await runPartSplit(inputSource);
+        let preprocessed = null;
+        if (preprocessPrompt != null && preprocessPrompt !== 'true') {
+            try {
+                const queued = await ctx.client.v2.generateImage({
+                    source: sourceFromProductionAnalyzeSource(inputSource),
+                    prompt: preprocessPrompt,
+                    modelId: getFlag(ctx.flags, 'preprocess-model-id') ?? defaultImageModelId,
+                    batchSize: 1,
+                    negativePrompt: getFlag(ctx.flags, 'preprocess-negative-prompt'),
+                    seed: getFlag(ctx.flags, 'preprocess-seed') == null
+                        ? undefined
+                        : parsePositiveIntegerFlag(ctx.flags, 'preprocess-seed', 1),
+                });
+                const job = await ctx.client.v2.pollJob(queued.jobId, {
+                    intervalMs: parsePositiveIntegerFlag(ctx.flags, 'interval-ms', 5000),
+                    timeoutMs: parsePositiveIntegerFlag(ctx.flags, 'timeout-ms', 900000),
+                    onPoll: current => {
+                        stderr.write(`[poll] preprocess job=${current.id} status=${current.status}\n`);
+                    },
+                });
+                if (job.status !== 'completed') {
+                    throw new Error(`Preprocessing failed: job=${job.id} status=${job.status}`);
+                }
+                const generatedSource = imageSourceFromCompletedJob(job);
+                preprocessed = {
+                    ok: true,
+                    image: {
+                        queued,
+                        job,
+                        source: generatedSource,
+                    },
+                    split: await runPartSplit(generatedSource),
+                };
+            }
+            catch (error) {
+                if (hasFlag(ctx.flags, 'fail-on-preprocess-error')) {
+                    throw error;
+                }
+                preprocessed = {
+                    ok: false,
+                    error: errorToOutput(error),
+                };
+            }
         }
+        print(await withOptionalDownloads(ctx.flags, {
+            direct,
+            preprocessed,
+        }));
+        return;
     }
-    print(await withOptionalDownloads(ctx.flags, sanitizeProductionOutput({
-        analyzed,
-        status,
-        executed,
-        executionStatus,
-    })));
+    throw new Error('Unknown parts command. Use "parts split" or "parts compare".');
 };
 const run = async () => {
     const parsed = parseArgs(argv.slice(2));

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assethub/cli",
-  "version": "0.1.2",
+  "version": "0.1.3",
   "description": "Command line interface for AssetHub.",
   "type": "module",
   "bin": {