@flatkey-ai/cli 0.2.1 → 0.2.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@flatkey-ai/cli",
-  "version": "0.2.1",
+  "version": "0.2.3",
   "description": "Flatkey media generation CLI for image, video, audio, credits, status, and models.",
   "type": "module",
   "bin": {

package/src/api.js

@@ -1,3 +1,5 @@
+import { withCliRequestHeaders } from "./requestHeaders.js";
+
 export const DEFAULT_BASE_URL = "https://router.flatkey.ai";
 export const DEFAULT_MODELS_BASE_URL = "https://console.flatkey.ai";
 export const DEFAULT_CONSOLE_URL = "https://console.flatkey.ai";
@@ -231,7 +233,7 @@ function planRequest(options, path, init = {}) {
   return {
     url: buildUrl(options.baseUrl, path),
     method: init.method ?? "GET",
-    headers: init.headers ?? authHeaders(options.apiKey),
+    headers: withCliRequestHeaders(init.headers ?? authHeaders(options.apiKey)),
     body: init.body,
   };
 }

package/src/artifacts.js

@@ -2,6 +2,8 @@ import { mkdir, writeFile } from "node:fs/promises";
 import { homedir } from "node:os";
 import { dirname, extname, join } from "node:path";
 
+import { CLI_REQUEST_HEADERS } from "./requestHeaders.js";
+
 const DEFAULT_EXTENSIONS = {
   audio: "mp3",
   image: "png",
@@ -80,7 +82,7 @@ async function persistItem({ kind, item, outDir, output, index, fetchImpl }) {
 }
 
 async function downloadArtifact({ url, path, fetchImpl }) {
-  const response = await fetchImpl(url);
+  const response = await fetchImpl(url, { headers: CLI_REQUEST_HEADERS });
   if (!response.ok) {
     throw new Error(`Failed to download artifact ${url}: HTTP ${response.status}`);
   }
@@ -99,6 +101,8 @@ function extractItems(response) {
   if (typeof response?.metadata?.url === "string") return [{ url: response.metadata.url }];
   if (typeof response?.url === "string") return [response];
   if (typeof response?.video_url?.url === "string") return [{ url: response.video_url.url }];
+  if (typeof response?.video?.url === "string") return [{ url: response.video.url }];
+  if (typeof response?.video?.temp_url === "string") return [{ url: response.video.temp_url }];
   return [];
 }
 

package/src/cli.js

@@ -1,3 +1,5 @@
+import { join } from "node:path";
+
 const COMMANDS = new Set([
   "audio",
   "credits",
@@ -548,11 +550,14 @@ async function handleGenerate(command, deps) {
       const output = await writeTextOutput(text, command.options.output);
       return { kind: command.group, text, output, response };
     }
+    const outDir = command.options.out ?? "flatkey-output";
+    const output = command.options.output
+      ?? (command.group === "video" ? join(outDir, "video-01.mp4") : undefined);
     const artifacts = await persistArtifacts({
       kind: command.group,
       response: artifactResponse,
-      outDir: command.options.out ?? "flatkey-output",
-      output: command.options.output,
+      outDir,
+      output,
       fetch: deps.fetch,
     });
     return { kind: command.group, artifacts, response: scrubArtifactResponse(artifactResponse) };
@@ -649,6 +654,8 @@ function hasArtifact(response) {
       || response?.url
       || response?.temp_url
       || response?.video_url?.url
+      || response?.video?.url
+      || response?.video?.temp_url
       || response?.data?.some?.((item) => item?.url || item?.temp_url || item?.b64_json || item?.base64 || item?.data)
       || response?.artifacts?.some?.((item) => item?.url || item?.temp_url || item?.path)
       || response?.content?.some?.((item) => item?.url || item?.video_url?.url || item?.data_url || item?.b64_json || item?.base64 || item?.data),
@@ -660,7 +667,7 @@ function videoTaskId(response) {
 }
 
 function isVideoDone(response) {
-  return ["completed", "succeeded", "success"].includes(String(response?.status ?? "").toLowerCase());
+  return ["completed", "done", "succeeded", "success"].includes(String(response?.status ?? "").toLowerCase());
 }
 
 function isVideoFailed(response) {
@@ -1051,9 +1058,11 @@ async function maybeWriteUpdateNotice(command, deps = {}) {
 }
 
 async function fetchLatestPackageVersion(deps = {}) {
+  const { CLI_REQUEST_HEADERS } = await import("./requestHeaders.js");
   const fetchImpl = deps.fetch ?? globalThis.fetch;
   if (typeof fetchImpl !== "function") return undefined;
   const response = await fetchImpl("https://registry.npmjs.org/@flatkey-ai%2fcli/latest", {
+    headers: CLI_REQUEST_HEADERS,
     signal: typeof AbortSignal !== "undefined" && typeof AbortSignal.timeout === "function"
       ? AbortSignal.timeout(750)
       : undefined,

package/src/requestHeaders.js

@@ -0,0 +1,10 @@
+export const CLI_REQUEST_HEADERS = {
+  "x-flatkey-client": "cli",
+};
+
+export function withCliRequestHeaders(headers = {}) {
+  return {
+    ...headers,
+    ...CLI_REQUEST_HEADERS,
+  };
+}