@playdrop/playdrop-cli 0.16.12 → 0.17.0

package/dist/commands/generation.js

@@ -34,12 +34,17 @@ var __importStar = (this && this.__importStar) || (function () {
 })();
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.resolveImageAttachment = resolveImageAttachment;
+exports.buildGenerateRequest = buildGenerateRequest;
 exports.generate = generate;
-exports.browseGenerationJobs = browseGenerationJobs;
-exports.showGenerationJob = showGenerationJob;
+exports.browseGenerationTasks = browseGenerationTasks;
+exports.showGenerationTask = showGenerationTask;
+exports.showGenerationTaskEvents = showGenerationTaskEvents;
+exports.cancelGenerationTask = cancelGenerationTask;
+exports.clearGenerationTask = clearGenerationTask;
 exports.generations = generations;
 const config_1 = require("@playdrop/config");
 const types_1 = require("@playdrop/types");
+const node_crypto_1 = require("node:crypto");
 const node_fs_1 = require("node:fs");
 const node_path_1 = require("node:path");
 const commandContext_1 = require("../commandContext");
@@ -62,6 +67,7 @@ const DEFAULT_IMAGE_SIZE = "1K";
 const DEFAULT_VIDEO_ASPECT_RATIO = "16:9";
 const DEFAULT_VIDEO_DURATION_SECONDS = 4;
 const VIDEO_FIRST_LAST_FRAME_DURATION_SECONDS = 8;
+const AI_TASK_INPUT_UPLOAD_MAX_BYTES = 25 * 1024 * 1024;
 const DEFAULT_MODEL3D_TOPOLOGY = "triangle";
 const DEFAULT_MODEL3D_TARGET_POLYCOUNT = 30000;
 const DEFAULT_MODEL3D_SHOULD_REMESH = true;
@@ -83,7 +89,7 @@ function handleGenerateBuildRequestError(message) {
     const errorByCode = {
         missing_type: {
             title: "A generation type is required.",
-            details: ["Use one of: image, music, sfx, speech, video, model_3d."],
+            details: ["Use one of: text, json, image, music, sfx, speech, video, model_3d."],
         },
         missing_prompt: {
             title: "A prompt is required.",
@@ -91,7 +97,19 @@ function handleGenerateBuildRequestError(message) {
         },
         invalid_type: {
             title: "The generation type is invalid.",
-            details: ["Use one of: image, music, sfx, speech, video, model_3d."],
+            details: ["Use one of: text, json, image, music, sfx, speech, video, model_3d."],
+        },
+        missing_json_schema: {
+            title: "JSON generation requires a response schema.",
+            details: ['Pass an inline JSON object or file path with --schema, for example --schema \'{"type":"object"}\'.'],
+        },
+        invalid_json_schema: {
+            title: "The JSON response schema is invalid.",
+            details: ["Use a JSON object or a path to a JSON file."],
+        },
+        invalid_text_size: {
+            title: "The text generation size is invalid.",
+            details: ["Use small, medium, or large with --text-size."],
         },
         invalid_duration: {
             title: "The --duration value is invalid.",
@@ -127,7 +145,7 @@ function handleGenerateBuildRequestError(message) {
         },
         speech_disallows_subcategory: {
             title: "Speech does not accept --subcategory.",
-            details: ["Speech jobs always create AUDIO assets with subcategory speech."],
+            details: ["Speech tasks always create AUDIO assets with subcategory speech."],
         },
         invalid_speech_voice_selection: {
             title: "Speech voice options are invalid.",
@@ -210,6 +228,10 @@ function normalizeModality(value) {
         .trim()
         .toUpperCase()
         .replace(/[-\s]+/g, "_");
+    if (normalized === "TEXT")
+        return "TEXT";
+    if (normalized === "JSON")
+        return "JSON";
     if (normalized === "IMAGE")
         return "IMAGE";
     if (normalized === "MUSIC")
@@ -224,14 +246,29 @@ function normalizeModality(value) {
         return "MODEL_3D";
     return null;
 }
+function parseResponseSchema(value) {
+    const normalized = typeof value === "string" ? value.trim() : "";
+    if (!normalized)
+        throw new Error("missing_json_schema");
+    const source = (0, node_fs_1.existsSync)(normalized) ? (0, node_fs_1.readFileSync)(normalized, "utf8") : normalized;
+    let parsed;
+    try {
+        parsed = JSON.parse(source);
+    }
+    catch {
+        throw new Error("invalid_json_schema");
+    }
+    if (!parsed || typeof parsed !== "object" || Array.isArray(parsed))
+        throw new Error("invalid_json_schema");
+    return parsed;
+}
 function normalizeVisibility(value) {
     if (typeof value !== "string" || value.trim().length === 0) {
         return undefined;
     }
     const normalized = value.trim().toUpperCase();
-    if (normalized === "PRIVATE" || normalized === "PUBLIC") {
+    if (normalized === "PRIVATE")
         return normalized;
-    }
     return undefined;
 }
 function parseAssetSubcategory(value) {
@@ -465,15 +502,33 @@ function buildGenerateRequest(typeInput, promptInput, options) {
     if (modality === "SPEECH" && (0, types_1.countSpeechWords)(prompt) > types_1.PLAYDROP_SPEECH_MAX_WORDS) {
         throw new Error("speech_input_too_long");
     }
-    const visibility = normalizeVisibility(options.visibility);
-    if (options.visibility && !visibility) {
-        throw new Error("invalid_visibility");
-    }
     const timeoutSeconds = parseOptionalPositiveInt(options.timeoutSeconds, "timeout_seconds") ?? GENERATE_DEFAULT_TIMEOUT_SECONDS;
     const pollIntervalMs = parseOptionalPositiveInt(options.pollIntervalMs, "poll_interval_ms") ?? GENERATE_DEFAULT_POLL_INTERVAL_MS;
     if (pollIntervalMs < 250) {
         throw new Error("invalid_poll_interval_ms");
     }
+    if (modality === "TEXT" || modality === "JSON") {
+        const textSize = typeof options.textSize === "string" ? options.textSize.trim().toLowerCase() : "small";
+        if (textSize !== "small" && textSize !== "medium" && textSize !== "large") {
+            throw new Error("invalid_text_size");
+        }
+        return {
+            modality,
+            request: {
+                modality,
+                input: prompt,
+                size: textSize,
+                ...(typeof options.system === "string" && options.system.trim() ? { system: options.system.trim() } : {}),
+                ...(modality === "JSON" ? { responseSchema: parseResponseSchema(options.responseSchema) } : {}),
+            },
+            timeoutSeconds,
+            pollIntervalMs,
+        };
+    }
+    const visibility = normalizeVisibility(options.visibility);
+    if (options.visibility && !visibility) {
+        throw new Error("invalid_visibility");
+    }
     const duration = parseOptionalDuration(options.duration);
     const request = {
         modality,
@@ -487,7 +542,7 @@ function buildGenerateRequest(typeInput, promptInput, options) {
         assetDescription: typeof options.assetDescription === "string" && options.assetDescription.trim().length > 0
             ? options.assetDescription.trim()
             : undefined,
-        assetVisibility: visibility ?? "PUBLIC",
+        assetVisibility: "PRIVATE",
         attachAppVersionId: parseOptionalPositiveInt(options.attachAppVersionId, "attach_app_version_id"),
     };
     if (modality !== "SPEECH") {
@@ -582,15 +637,28 @@ function buildGenerateRequest(typeInput, promptInput, options) {
     }
     if (modality === "MODEL_3D") {
         const sourceModeInput = typeof options.sourceMode === "string" ? options.sourceMode.trim().toUpperCase() : "";
-        if (sourceModeInput.length > 0 && sourceModeInput !== "TEXT" && sourceModeInput !== "IMAGE") {
+        if (sourceModeInput.length > 0 &&
+            sourceModeInput !== "TEXT" &&
+            sourceModeInput !== "IMAGE" &&
+            sourceModeInput !== "MULTI_IMAGE") {
             throw new Error("invalid_source_mode");
         }
         request.image1 = resolveImageAttachment(options.image1, "image1");
         request.image2 = resolveImageAttachment(options.image2, "image2");
         const hasInputImages = Boolean(request.image1 || request.image2);
+        const hasMultipleInputImages = Boolean(request.image1 && request.image2);
         request.sourceMode =
-            sourceModeInput.length > 0 ? sourceModeInput : hasInputImages ? "IMAGE" : "TEXT";
-        if (request.sourceMode === "IMAGE" && !hasInputImages) {
+            sourceModeInput.length > 0
+                ? sourceModeInput
+                : hasMultipleInputImages
+                    ? "MULTI_IMAGE"
+                    : hasInputImages
+                        ? "IMAGE"
+                        : "TEXT";
+        if (request.sourceMode === "IMAGE" && (!hasInputImages || hasMultipleInputImages)) {
+            throw new Error("missing_model3d_images");
+        }
+        if (request.sourceMode === "MULTI_IMAGE" && !hasMultipleInputImages) {
             throw new Error("missing_model3d_images");
         }
         if (request.sourceMode === "TEXT" && hasInputImages) {
@@ -609,33 +677,35 @@ function buildGenerateRequest(typeInput, promptInput, options) {
     return { modality, request, timeoutSeconds, pollIntervalMs };
 }
 function generationPrompt(record) {
-    if ("inputPrompt" in record && typeof record.inputPrompt === "string") {
-        return record.inputPrompt;
-    }
-    return "";
+    const input = record.inputData && typeof record.inputData === "object" && !Array.isArray(record.inputData)
+        ? record.inputData
+        : null;
+    return typeof input?.input === "string" ? input.input : "";
 }
 function generationAssetRef(record) {
-    const candidate = record.generatedAsset;
-    if (candidate &&
-        typeof candidate === "object" &&
-        typeof candidate.ref === "string" &&
-        candidate.ref.trim().length > 0) {
-        return candidate.ref.trim();
-    }
-    return null;
+    const output = record.outputData && typeof record.outputData === "object" && !Array.isArray(record.outputData)
+        ? record.outputData
+        : null;
+    return typeof output?.assetRef === "string" && output.assetRef.trim() ? output.assetRef.trim() : null;
 }
 function generationAssetSubcategory(record) {
-    const candidate = record.generatedAsset;
-    if (candidate &&
-        typeof candidate === "object" &&
-        typeof candidate.subcategory === "string" &&
-        candidate.subcategory.trim().length > 0) {
-        return candidate.subcategory.trim();
-    }
-    return null;
+    const input = record.inputData && typeof record.inputData === "object" && !Array.isArray(record.inputData)
+        ? record.inputData
+        : null;
+    return typeof input?.assetSubcategory === "string" && input.assetSubcategory.trim()
+        ? input.assetSubcategory.trim()
+        : null;
 }
 function printGenerateResult(modality, payload) {
     console.log(`Modality: ${modality}`);
+    if (modality === "TEXT") {
+        console.log(typeof payload?.text === "string" ? payload.text : "");
+        return;
+    }
+    if (modality === "JSON") {
+        console.log(JSON.stringify(payload?.json ?? null, null, 2));
+        return;
+    }
     if (payload?.billing?.generationId) {
         console.log(`Generation ID: ${payload.billing.generationId}`);
     }
@@ -645,13 +715,9 @@ function printGenerateResult(modality, payload) {
     if (typeof payload?.createdAt === "string") {
         console.log(`Created At: ${payload.createdAt}`);
     }
-    const ref = typeof payload?.generatedAsset?.ref === "string" ? payload.generatedAsset.ref : null;
-    const subcategory = typeof payload?.generatedAsset?.subcategory === "string" ? payload.generatedAsset.subcategory : null;
+    const ref = typeof payload?.assetRef === "string" ? payload.assetRef : null;
     if (ref) {
         console.log(`Generated Asset: ${ref}`);
-        if (subcategory) {
-            console.log(`Generated Subcategory: ${subcategory}`);
-        }
         console.log(`Next: run "playdrop detail ${ref}" to inspect the generated asset.`);
     }
     else {
@@ -861,12 +927,67 @@ async function optimizeGeneratedPngOutputBuffer(buffer, loadedSharp) {
     }
     throw new Error(`ai_generation_image_output_too_large:${smallest?.length ?? buffer.length}:${GENERATED_PNG_OUTPUT_MAX_BYTES}`);
 }
-function formatJobStatus(job) {
-    const progress = typeof job.progressPercent === "number"
-        ? ` progress=${Math.max(0, Math.min(100, Math.round(job.progressPercent)))}%`
-        : "";
-    const eta = typeof job.etaSecondsRemaining === "number" ? ` eta=${Math.max(0, Math.round(job.etaSecondsRemaining))}s` : "";
-    return `${job.status}${progress}${eta}`;
+function formatTaskStatus(task) {
+    return `${task.status}${task.queuePosition ? ` queue=${task.queuePosition}` : ""}`;
+}
+async function uploadImageAttachment(client, attachment) {
+    let buffer;
+    let mimeType;
+    if ("url" in attachment) {
+        const response = await fetch(attachment.url);
+        if (!response.ok)
+            throw new Error(`image_ref_download_failed:${response.status}`);
+        if (new URL(response.url).protocol !== "https:")
+            throw new Error("image_ref_download_insecure_redirect");
+        const contentLength = Number(response.headers.get("content-length"));
+        if (Number.isFinite(contentLength) && contentLength > AI_TASK_INPUT_UPLOAD_MAX_BYTES) {
+            throw new Error(`image_ref_too_large:${contentLength}:${AI_TASK_INPUT_UPLOAD_MAX_BYTES}`);
+        }
+        buffer = Buffer.from(await response.arrayBuffer());
+        if (buffer.length === 0 || buffer.length > AI_TASK_INPUT_UPLOAD_MAX_BYTES) {
+            throw new Error(`image_ref_too_large:${buffer.length}:${AI_TASK_INPUT_UPLOAD_MAX_BYTES}`);
+        }
+        const detected = detectLocalImageMimeType(buffer);
+        const declared = response.headers.get("content-type")?.split(";", 1)[0]?.trim().toLowerCase();
+        if (!detected || declared !== detected)
+            throw new Error("image_ref_download_type_invalid");
+        mimeType = detected;
+    }
+    else {
+        buffer = Buffer.from(attachment.data, "base64");
+        mimeType = attachment.mimeType;
+    }
+    const extension = mimeType === "image/png" ? "png" : mimeType === "image/webp" ? "webp" : "jpg";
+    const prepared = await client.prepareAiTaskInputUpload({
+        fileName: `reference.${extension}`,
+        contentType: mimeType,
+        sizeBytes: buffer.length,
+        sha256: (0, node_crypto_1.createHash)("sha256").update(buffer).digest("hex"),
+        md5Base64: (0, node_crypto_1.createHash)("md5").update(buffer).digest("base64"),
+    });
+    const response = await fetch(prepared.uploadUrl, {
+        method: "PUT",
+        headers: prepared.uploadHeaders,
+        body: new Uint8Array(buffer),
+    });
+    if (!response.ok)
+        throw new Error(`ai_task_input_upload_failed:${response.status}`);
+    await client.finalizeAiTaskInputUpload(prepared.uploadId);
+    return { type: "TASK_UPLOAD", uploadId: prepared.uploadId };
+}
+async function prepareTaskRequest(client, request) {
+    const { modality, image1, image2, ...fields } = request;
+    if (modality === "TEXT" || modality === "JSON") {
+        return { ...fields, kind: modality };
+    }
+    return {
+        ...fields,
+        kind: modality,
+        assetVisibility: "PRIVATE",
+        ...(image1 ? { image1: await uploadImageAttachment(client, image1) } : {}),
+        ...(image2 ? { image2: await uploadImageAttachment(client, image2) } : {}),
+        ...(modality === "MODEL_3D" ? { outputFormat: "GLB" } : {}),
+    };
 }
 async function generate(typeInput, promptInput, options = {}) {
     let parsed;
@@ -895,41 +1016,43 @@ async function generate(typeInput, promptInput, options = {}) {
         process.exitCode = 1;
         return;
     }
-    await (0, commandContext_1.withEnvironment)("ai create", "Generating AI assets", async ({ aiClient }) => {
+    await (0, commandContext_1.withEnvironment)("ai create", "Generating AI assets", async ({ client }) => {
         try {
-            const created = await aiClient.createGenerationJob(parsed.request);
-            let job = created.job;
-            const jobId = String(job.id || "").trim();
-            if (!jobId) {
-                throw new Error("ai_generation_job_missing_id");
-            }
+            const request = await prepareTaskRequest(client, parsed.request);
+            const created = await client.createAiGenerationTask(request);
+            let task = created.task;
+            const taskId = task.id;
             if (!options.json) {
-                console.log(`Job ID: ${jobId}`);
-                console.log(`Status: ${formatJobStatus(job)}`);
+                console.log(`Task ID: ${taskId}`);
+                console.log(`Status: ${formatTaskStatus(task)}`);
             }
             const deadlineMs = Date.now() + parsed.timeoutSeconds * 1000;
-            while (job.status === "PENDING" || job.status === "RUNNING") {
+            while (task.status === "QUEUED" || task.status === "RUNNING") {
                 if (Date.now() >= deadlineMs) {
-                    throw new Error(`ai_generation_job_timeout: Timed out waiting for job ${jobId}`);
+                    throw new Error(`ai_generation_task_timeout: Timed out waiting for task ${taskId}`);
                 }
                 await sleep(parsed.pollIntervalMs);
-                const latest = await aiClient.getGenerationJob(jobId);
-                job = latest.job;
+                const latest = await client.getAiGenerationTask(taskId);
+                task = latest.task;
                 if (!options.json) {
-                    console.log(`Status: ${formatJobStatus(job)}`);
+                    console.log(`Status: ${formatTaskStatus(task)}`);
                 }
             }
-            if (job.status !== "SUCCESS") {
-                const errorCode = typeof job.errorCode === "string" && job.errorCode.length > 0 ? job.errorCode : "ai_generation_failed";
-                const errorMessage = typeof job.errorMessage === "string" && job.errorMessage.length > 0 ? job.errorMessage : "Generation failed.";
-                throw new Error(`${errorCode}: ${errorMessage}`);
+            if (task.status !== "DONE") {
+                throw new Error(`${task.error ?? "ai_generation_failed"}: Generation failed.`);
             }
-            const payload = job.resultData && typeof job.resultData === "object" ? job.resultData : {};
+            const payload = task.result && typeof task.result === "object" ? task.result : {};
             if (outputPath) {
-                await writeGeneratedImageOutput(payload, outputPath, parsed.request.input);
+                const generationId = "generationId" in payload && typeof payload.generationId === "string"
+                    ? payload.generationId
+                    : "";
+                const history = await client.listAiGenerations({ type: "IMAGE", limit: 100, offset: 0 });
+                const generation = history.generations.find((entry) => entry.id === generationId);
+                const imageUrl = generation?.outputAssets[0];
+                await writeGeneratedImageOutput({ ...payload, imageUrl }, outputPath, parsed.request.input);
             }
             if (options.json) {
-                (0, output_1.printJson)({ job, result: payload });
+                (0, output_1.printJson)({ task, result: payload });
                 return;
             }
             printGenerateResult(parsed.modality, payload);
@@ -941,7 +1064,7 @@ async function generate(typeInput, promptInput, options = {}) {
         }
     });
 }
-function parseJobModality(raw) {
+function parseTaskModality(raw) {
     if (!raw || raw.trim().length === 0 || raw.trim().toLowerCase() === "all") {
         return undefined;
     }
@@ -980,11 +1103,6 @@ function handleAiFailure(error, command, context) {
         return true;
     }
     const message = error instanceof Error ? error.message : String(error);
-    if (message === "ai_generation_job_missing_id") {
-        (0, messages_1.printErrorWithHelp)("AI create failed because the server did not return a job id.", ["Retry the command. If the problem persists, contact Playdrop support."], { command });
-        process.exitCode = 1;
-        return true;
-    }
     if (message === "ai_generation_image_output_unsupported_type") {
         (0, messages_1.printErrorWithHelp)("AI create returned an image format that Playdrop CLI could not identify.", ["Retry the command. If the problem persists, contact Playdrop support."], { command });
         process.exitCode = 1;
@@ -1005,9 +1123,9 @@ function handleAiFailure(error, command, context) {
         process.exitCode = 1;
         return true;
     }
-    if (message.startsWith("ai_generation_job_timeout:")) {
+    if (message.startsWith("ai_generation_task_timeout:")) {
         const detail = message.split(":").slice(1).join(":").trim();
-        (0, messages_1.printErrorWithHelp)(detail || "AI create timed out while waiting for the job to finish.", ["Retry the command with a higher --timeout value if the job normally takes longer."], { command });
+        (0, messages_1.printErrorWithHelp)(detail || "AI create timed out while waiting for the task to finish.", ["Retry the command with a higher --timeout value if the task normally takes longer."], { command });
         process.exitCode = 1;
         return true;
     }
@@ -1022,11 +1140,11 @@ function handleAiFailure(error, command, context) {
     }
     return false;
 }
-async function browseGenerationJobs(options = {}) {
-    const modality = parseJobModality(options.type);
+async function browseGenerationTasks(options = {}) {
+    const modality = parseTaskModality(options.type);
     if (options.type !== undefined && modality === null) {
         (0, messages_1.printErrorWithHelp)("The --type value is invalid.", ["Use one of: image, music, sfx, video, model_3d, all."], {
-            command: "ai jobs browse",
+            command: "ai tasks browse",
         });
         process.exitCode = 1;
         return;
@@ -1039,78 +1157,149 @@ async function browseGenerationJobs(options = {}) {
     }
     catch (error) {
         const message = typeof error?.message === "string" ? error.message : "invalid_pagination";
-        (0, messages_1.printErrorWithHelp)(`Invalid ai jobs input: ${message}.`, ["Use positive integers for --limit and --offset."], {
-            command: "ai jobs browse",
+        (0, messages_1.printErrorWithHelp)(`Invalid ai tasks input: ${message}.`, ["Use positive integers for --limit and --offset."], {
+            command: "ai tasks browse",
         });
         process.exitCode = 1;
         return;
     }
-    await (0, commandContext_1.withEnvironment)("ai jobs browse", "Browsing AI jobs", async ({ aiClient }) => {
+    await (0, commandContext_1.withEnvironment)("ai tasks browse", "Browsing AI tasks", async ({ client }) => {
         try {
-            const response = await aiClient.listGenerationJobs({
-                modality: modality ?? undefined,
+            const response = await client.listAiGenerationTasks({
+                kind: modality ?? undefined,
                 limit,
                 offset,
             });
             if (options.json) {
-                (0, output_1.printJson)({ items: response.jobs, pagination: response.pagination });
+                (0, output_1.printJson)({ items: response.tasks, count: response.count });
                 return;
             }
-            if (!Array.isArray(response.jobs) || response.jobs.length === 0) {
-                console.log("No AI jobs found.");
-                console.log('Next: run "playdrop ai create <type> <prompt>" to start a new job.');
+            if (!Array.isArray(response.tasks) || response.tasks.length === 0) {
+                console.log("No AI tasks found.");
+                console.log('Next: run "playdrop ai create <type> <prompt>" to start a new task.');
                 return;
             }
-            console.log("AI jobs:\n");
-            for (const job of response.jobs) {
-                console.log(`${job.id} | ${job.modality.toLowerCase()} | ${formatJobStatus(job)} | ${job.createdAt}`);
+            console.log("AI tasks:\n");
+            for (const task of response.tasks) {
+                console.log(`${task.id} | ${task.kind.toLowerCase()} | ${formatTaskStatus(task)} | ${task.createdAt}`);
             }
-            console.log('\nNext: run "playdrop ai jobs detail <id>" to inspect one job.');
+            console.log('\nNext: run "playdrop ai tasks detail <id>" to inspect one task.');
         }
         catch (error) {
-            if (!handleAiFailure(error, "ai jobs browse", "AI job lookup")) {
+            if (!handleAiFailure(error, "ai tasks browse", "AI task lookup")) {
                 throw error;
             }
         }
     });
 }
-async function showGenerationJob(id, options = {}) {
-    const jobId = typeof id === "string" ? id.trim() : "";
-    if (!jobId) {
-        (0, messages_1.printErrorWithHelp)("A job id is required.", ["Example: playdrop ai jobs detail job_123"], {
-            command: "ai jobs detail",
+async function showGenerationTask(id, options = {}) {
+    const taskId = Number(id);
+    if (!Number.isInteger(taskId) || taskId <= 0) {
+        (0, messages_1.printErrorWithHelp)("A task id is required.", ["Example: playdrop ai tasks detail 123"], {
+            command: "ai tasks detail",
         });
         process.exitCode = 1;
         return;
     }
-    await (0, commandContext_1.withEnvironment)("ai jobs detail", "Inspecting an AI job", async ({ aiClient }) => {
+    await (0, commandContext_1.withEnvironment)("ai tasks detail", "Inspecting an AI task", async ({ client }) => {
         try {
-            const response = await aiClient.getGenerationJob(jobId);
+            const response = await client.getAiGenerationTask(taskId);
             if (options.json) {
-                (0, output_1.printJson)({ job: response.job });
+                (0, output_1.printJson)({ task: response.task });
                 return;
             }
-            const job = response.job;
-            console.log(`Job: ${job.id}`);
-            console.log(`Type: ${job.modality.toLowerCase()}`);
-            console.log(`Status: ${formatJobStatus(job)}`);
-            console.log(`Created: ${job.createdAt}`);
-            console.log(`Updated: ${job.updatedAt}`);
-            if (job.errorMessage) {
-                console.log(`Error: ${job.errorMessage}`);
+            const task = response.task;
+            console.log(`Task: ${task.id}`);
+            console.log(`Type: ${task.kind.toLowerCase()}`);
+            console.log(`Status: ${formatTaskStatus(task)}`);
+            console.log(`Created: ${task.createdAt}`);
+            console.log(`Updated: ${task.updatedAt}`);
+            if (task.error) {
+                console.log(`Error: ${task.error}`);
             }
-            if (job.linkedGenerationId) {
-                console.log(`Linked generation: ${job.linkedGenerationId}`);
+            if (typeof task.result?.generationId === "string") {
+                console.log(`Linked generation: ${task.result.generationId}`);
             }
-            console.log('\nNext: run "playdrop ai jobs browse" to review other jobs.');
+            console.log('\nNext: run "playdrop ai tasks browse" to review other tasks.');
         }
         catch (error) {
-            if (!handleAiFailure(error, "ai jobs detail", "AI job detail lookup")) {
+            if (!handleAiFailure(error, "ai tasks detail", "AI task detail lookup")) {
                 throw error;
             }
         }
     });
 }
+function parseGenerationTaskId(id, command) {
+    const taskId = Number(id);
+    if (Number.isInteger(taskId) && taskId > 0)
+        return taskId;
+    (0, messages_1.printErrorWithHelp)("A task id is required.", [`Example: playdrop ai tasks ${command} 123`], {
+        command: `ai tasks ${command}`,
+    });
+    process.exitCode = 1;
+    return null;
+}
+async function showGenerationTaskEvents(id, options = {}) {
+    const taskId = parseGenerationTaskId(id, "events");
+    if (!taskId)
+        return;
+    await (0, commandContext_1.withEnvironment)("ai tasks events", "Inspecting AI task events", async ({ client }) => {
+        try {
+            const response = await client.listAiGenerationTaskEvents(taskId);
+            if (options.json) {
+                (0, output_1.printJson)(response);
+                return;
+            }
+            if (response.events.length === 0) {
+                console.log(`Task ${taskId} has no events.`);
+                return;
+            }
+            for (const event of response.events) {
+                console.log(`${event.createdAt} | ${event.kind} | ${event.message}`);
+            }
+        }
+        catch (error) {
+            if (!handleAiFailure(error, "ai tasks events", "AI task events lookup"))
+                throw error;
+        }
+    });
+}
+async function cancelGenerationTask(id, options = {}) {
+    const taskId = parseGenerationTaskId(id, "cancel");
+    if (!taskId)
+        return;
+    await (0, commandContext_1.withEnvironment)("ai tasks cancel", "Cancelling an AI task", async ({ client }) => {
+        try {
+            const response = await client.cancelAiGenerationTask(taskId);
+            if (options.json)
+                (0, output_1.printJson)({ task: response.task });
+            else
+                console.log(`Task ${taskId}: ${formatTaskStatus(response.task)}`);
+        }
+        catch (error) {
+            if (!handleAiFailure(error, "ai tasks cancel", "AI task cancellation"))
+                throw error;
+        }
+    });
+}
+async function clearGenerationTask(id, options = {}) {
+    const taskId = parseGenerationTaskId(id, "clear");
+    if (!taskId)
+        return;
+    await (0, commandContext_1.withEnvironment)("ai tasks clear", "Clearing an AI task", async ({ client }) => {
+        try {
+            await client.clearAiGenerationTask(taskId);
+            if (options.json)
+                (0, output_1.printJson)({ taskId, cleared: true });
+            else
+                console.log(`Task ${taskId} cleared.`);
+        }
+        catch (error) {
+            if (!handleAiFailure(error, "ai tasks clear", "AI task clearing"))
+                throw error;
+        }
+    });
+}
 async function generations(options = {}) {
     const format = parseListFormat(options.format);
     if (!format) {
@@ -1140,8 +1329,8 @@ async function generations(options = {}) {
         process.exitCode = 1;
         return;
     }
-    await (0, commandContext_1.withEnvironment)("generations", "Listing AI generations", async ({ aiClient }) => {
-        const response = await aiClient.listGenerations({ type: type ?? undefined, limit, offset });
+    await (0, commandContext_1.withEnvironment)("generations", "Listing AI generations", async ({ client }) => {
+        const response = await client.listAiGenerations({ type: type ?? undefined, limit, offset });
         if (format === "json") {
             console.log(JSON.stringify(response, null, 2));
             return;
@@ -1154,7 +1343,7 @@ async function generations(options = {}) {
             const prompt = generationPrompt(generation);
             const ref = generationAssetRef(generation);
             const subcategory = generationAssetSubcategory(generation);
-            const parts = [generation.id, generation.type, generation.createdAt, `cost=${generation.cost}`];
+            const parts = [generation.id, generation.type, generation.createdAt, `cost=${generation.amount}`];
             if (prompt.length > 0) {
                 parts.push(`prompt=${prompt}`);
             }