@flatkey-ai/cli 0.2.1 → 0.2.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@flatkey-ai/cli",
-  "version": "0.2.1",
+  "version": "0.2.2",
   "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}`);
   }

package/src/cli.js

@@ -1051,9 +1051,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,
+  };
+}