@assethub/cli 0.1.0 → 0.1.1

package/README.md

@@ -11,7 +11,7 @@ assethub files import https://example.com/input.png --media-type image
 assethub image generate --prompt "stylized prop concept" --wait
 assethub mesh generate --file ./input.png --model-id meshGen.hunyuan31 --wait
 assethub jobs watch job_123
-assethub production analyze --image-url https://example.com/input.png --agent-version ah_part_extractor_v65
+assethub production analyze --image-url https://example.com/input.png --part-extractor "V1.5"
 ```
 
 Configuration:
@@ -26,10 +26,15 @@ Agent-friendly login:
 printf '%s' "$ASSETHUB_API_KEY" | assethub auth login --api-key-stdin
 assethub image generate --prompt "small stylized game asset chest" --wait --interval-ms 5000 --timeout-ms 900000
 assethub mesh generate --source-url https://example.com/input.png --model-id meshGen.hunyuan31 --wait --interval-ms 5000 --timeout-ms 900000
+assethub production analyze --image-url https://example.com/input.png --part-extractor "V1.5"
 ```
 
 Image generation defaults to the deployed `v1` endpoint. Pass `--api-version v2`
 only when the `v2` image endpoint is available in the target environment.
 
+Part extraction uses the same public names shown in the AssetHub product:
+`V1`, `V1.5 alpha`, `V1.5`, `V2.0 alpha`, and `V2.1 alpha`. Internal agent
+IDs are intentionally not part of the public CLI interface.
+
 The CLI only uses AssetHub's public API surface and is designed so this package
 can be published independently from the private application monorepo.

package/dist/index.js

@@ -8,6 +8,52 @@ const defaultBaseUrl = 'https://app.assethub.io';
 const defaultImageModelId = 'imageGen.nanoBanana.pro.openrouter';
 const defaultMeshModelId = 'meshGen.hunyuan31';
 const defaultProfileName = 'default';
+const defaultPartExtractorName = 'V1.5';
+const partExtractorOptions = [
+    {
+        internalId: 'ah_part_extractor_v3',
+        publicName: 'Part Extractor V3',
+        aliases: ['part extractor v3', 'v3'],
+    },
+    {
+        internalId: 'ah_part_extractor_v5',
+        publicName: 'V1',
+        aliases: ['v1', '1'],
+    },
+    {
+        internalId: 'ah_part_extractor_v55',
+        publicName: 'V1.5 alpha',
+        aliases: ['v1.5 alpha', 'v1.5-alpha', '1.5 alpha', '1.5-alpha'],
+    },
+    {
+        internalId: 'ah_part_extractor_v65',
+        publicName: 'V1.5',
+        aliases: ['v1.5', '1.5', 'latest', 'default'],
+    },
+    {
+        internalId: 'ah_part_extractor_v20_alpha',
+        publicName: 'V2.0 alpha',
+        aliases: ['v2.0 alpha', 'v2 alpha', 'v2.0-alpha', 'v2-alpha', '2.0 alpha'],
+    },
+    {
+        internalId: 'ah_part_extractor_v21_alpha',
+        publicName: 'V2.1 alpha',
+        aliases: ['v2.1 alpha', 'v2.1-alpha', '2.1 alpha'],
+    },
+];
+const partExtractorPublicNames = partExtractorOptions
+    .map(option => option.publicName)
+    .join(', ');
+const normalizePartExtractorName = (value) => value.trim().toLowerCase().replace(/\s+/g, ' ');
+const publicNameByInternalPartExtractorId = new Map(partExtractorOptions.map(option => [option.internalId, option.publicName]));
+const internalPartExtractorIdByPublicName = new Map(partExtractorOptions.flatMap(option => [
+    [normalizePartExtractorName(option.publicName), option.internalId],
+    ...option.aliases.map(alias => [
+        normalizePartExtractorName(alias),
+        option.internalId,
+    ]),
+]));
+const isInternalPartExtractorId = (value) => value === 'nano_bbox_v1' || value.startsWith('ah_part_extractor_');
 const usage = `AssetHub CLI
 
 Usage:
@@ -25,9 +71,9 @@ Usage:
   assethub mesh retopo (--source-resource-id <id> | --source-url <url> | --file-ref-json <json>) --model-id <id> [--wait]
   assethub jobs get <job-id>
   assethub jobs watch <job-id> [--interval-ms <ms>] [--timeout-ms <ms>]
-  assethub production analyze --image-url <url> --agent-version <id> [--name <name>]
+  assethub production analyze --image-url <url> [--part-extractor <name>] [--name <name>]
   assethub production status <order-id>
-  assethub production execute --order-id <id> --mission-id <id> --task-id <id> [--task-id <id>...] --agent-version <id> [--part-extraction-mode fast|quality]
+  assethub production execute --order-id <id> --mission-id <id> --task-id <id> [--task-id <id>...] [--part-extractor <name>] [--part-extraction-mode fast|high_quality]
 
 Global options:
   --api-key <key>       Overrides saved auth and ASSETHUB_API_KEY.
@@ -43,6 +89,7 @@ Examples:
   assethub files upload ./input.png --media-type image
   assethub image generate --prompt "stylized prop concept" --wait
   assethub mesh generate --file ./input.png --model-id meshGen.hunyuan31 --wait
+  assethub production analyze --image-url https://example.com/input.png --part-extractor "V1.5"
 `;
 const print = (value) => {
     stdout.write(`${JSON.stringify(value, null, 2)}\n`);
@@ -112,6 +159,41 @@ const requireFlag = (flags, name) => {
     }
     return value;
 };
+const getPartExtractorFlag = (flags) => getFlag(flags, 'part-extractor') ??
+    getFlag(flags, 'part-extractor-version') ??
+    getFlag(flags, 'agent-version');
+const resolvePartExtractorInternalId = (value) => {
+    if (isInternalPartExtractorId(value)) {
+        throw new Error(`Internal part extractor IDs are not accepted by the public CLI. Use one of: ${partExtractorPublicNames}.`);
+    }
+    const internalId = internalPartExtractorIdByPublicName.get(normalizePartExtractorName(value));
+    if (internalId == null) {
+        throw new Error(`Unknown part extractor: ${value}. Use one of: ${partExtractorPublicNames}.`);
+    }
+    return internalId;
+};
+const resolvePartExtractorForAnalyze = (flags) => resolvePartExtractorInternalId(getPartExtractorFlag(flags) ?? defaultPartExtractorName);
+const publicPartExtractorName = (value) => {
+    if (value == null) {
+        return null;
+    }
+    return (publicNameByInternalPartExtractorId.get(value) ??
+        (isInternalPartExtractorId(value) ? 'Unknown' : value));
+};
+const sanitizeProductionOutput = (value) => {
+    if (Array.isArray(value)) {
+        return value.map(item => sanitizeProductionOutput(item));
+    }
+    if (value == null || typeof value !== 'object') {
+        return value;
+    }
+    return Object.fromEntries(Object.entries(value).map(([key, entry]) => [
+        key,
+        key === 'agentVersion' && typeof entry === 'string'
+            ? publicPartExtractorName(entry)
+            : sanitizeProductionOutput(entry),
+    ]));
+};
 const parsePositiveIntegerFlag = (flags, name, fallback) => {
     const raw = getFlag(flags, name);
     if (raw == null) {
@@ -509,18 +591,30 @@ const commandJobs = async (subcommand, positionals, ctx) => {
     }
     throw new Error('Unknown jobs command. Use "jobs get" or "jobs watch".');
 };
+const resolvePartExtractorFromMission = async (ctx, orderId, missionId) => {
+    const status = await ctx.client.v1.getProductionStatus(orderId);
+    const mission = status.missions.find(item => item.id === missionId);
+    if (mission?.agentVersion == null || mission.agentVersion.trim() === '') {
+        throw new Error('Could not infer the part extractor from the mission. Pass --part-extractor with one of: ' +
+            partExtractorPublicNames);
+    }
+    if (isInternalPartExtractorId(mission.agentVersion)) {
+        return mission.agentVersion;
+    }
+    return resolvePartExtractorInternalId(mission.agentVersion);
+};
 const commandProduction = async (subcommand, positionals, ctx) => {
     if (subcommand === 'analyze') {
-        print(await ctx.client.v1.analyzeProduction({
+        print(sanitizeProductionOutput(await ctx.client.v1.analyzeProduction({
             imageUrl: requireFlag(ctx.flags, 'image-url'),
-            agentVersion: requireFlag(ctx.flags, 'agent-version'),
+            agentVersion: resolvePartExtractorForAnalyze(ctx.flags),
             name: getFlag(ctx.flags, 'name'),
-        }));
+        })));
         return;
     }
     if (subcommand === 'status') {
         const orderId = requirePositional(positionals, 2, 'order-id');
-        print(await ctx.client.v1.getProductionStatus(orderId));
+        print(sanitizeProductionOutput(await ctx.client.v1.getProductionStatus(orderId)));
         return;
     }
     if (subcommand === 'execute') {
@@ -531,16 +625,22 @@ const commandProduction = async (subcommand, positionals, ctx) => {
         const partExtractionMode = getFlag(ctx.flags, 'part-extraction-mode');
         if (partExtractionMode != null &&
             partExtractionMode !== 'fast' &&
-            partExtractionMode !== 'quality') {
-            throw new Error('--part-extraction-mode must be "fast" or "quality"');
+            partExtractionMode !== 'high_quality') {
+            throw new Error('--part-extraction-mode must be "fast" or "high_quality"');
         }
-        print(await ctx.client.v1.executeProduction({
-            orderId: requireFlag(ctx.flags, 'order-id'),
-            missionId: requireFlag(ctx.flags, 'mission-id'),
+        const orderId = requireFlag(ctx.flags, 'order-id');
+        const missionId = requireFlag(ctx.flags, 'mission-id');
+        const partExtractorFlag = getPartExtractorFlag(ctx.flags);
+        const agentVersion = partExtractorFlag == null
+            ? await resolvePartExtractorFromMission(ctx, orderId, missionId)
+            : resolvePartExtractorInternalId(partExtractorFlag);
+        print(sanitizeProductionOutput(await ctx.client.v1.executeProduction({
+            orderId,
+            missionId,
             confirmedTaskIds,
-            agentVersion: requireFlag(ctx.flags, 'agent-version'),
+            agentVersion,
             partExtractionMode,
-        }));
+        })));
         return;
     }
     throw new Error('Unknown production command. Use "production analyze", "production status", or "production execute".');

package/package.json

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