ai-cli 0.2.0 → 0.3.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-cli",
-  "version": "0.2.0",
+  "version": "0.3.0",
   "description": "A tiny, agent-native CLI for generating images, video and text with dead-simple commands, stdin support and predictable artifact outputs",
   "type": "module",
   "license": "Apache-2.0",
@@ -8,11 +8,14 @@
     "type": "git",
     "url": "https://github.com/vercel-labs/ai-cli.git"
   },
+  "engines": {
+    "node": ">=20"
+  },
   "bin": {
-    "ai": "./src/index.ts"
+    "ai": "./dist/index.js"
   },
   "files": [
-    "src",
+    "dist",
     "README.md"
   ],
   "keywords": [
@@ -27,13 +30,14 @@
   ],
   "scripts": {
     "dev": "bun run src/index.ts",
-    "build": "bun build src/index.ts --compile --outfile=dist/ai",
+    "clean": "bun -e \"require('node:fs').rmSync('dist', { recursive: true, force: true })\"",
+    "build": "bun run clean && bun build src/index.ts --target=node --outdir=dist",
     "typecheck": "tsc --noEmit",
     "format": "oxfmt --write src/",
     "format:check": "oxfmt --check src/",
     "lint": "oxlint src/",
     "test": "bun test",
-    "prepublishOnly": "bun run typecheck"
+    "prepack": "bun run typecheck && bun run build"
   },
   "dependencies": {
     "ai": "^6.0.173",

package/src/cli.test.ts

@@ -1,70 +0,0 @@
-import { describe, expect, test } from "bun:test";
-
-const CLI = ["bun", "run", "src/index.ts"];
-const ROOT = import.meta.dir + "/..";
-
-async function run(...args: string[]) {
-  const proc = Bun.spawn([...CLI, ...args], {
-    cwd: ROOT,
-    stdout: "pipe",
-    stderr: "pipe",
-    stdin: "ignore",
-  });
-  const [stdout, stderr] = await Promise.all([
-    new Response(proc.stdout).text(),
-    new Response(proc.stderr).text(),
-  ]);
-  const exitCode = await proc.exited;
-  return { exitCode, stdout, stderr };
-}
-
-describe("cli integration", () => {
-  test("--help exits 0 and lists subcommands", async () => {
-    const { exitCode, stdout } = await run("--help");
-    expect(exitCode).toBe(0);
-    for (const sub of ["text", "image", "video", "models"]) {
-      expect(stdout).toContain(sub);
-    }
-  });
-
-  test("--version exits 0 and prints semver", async () => {
-    const { exitCode, stdout } = await run("--version");
-    expect(exitCode).toBe(0);
-    expect(stdout.trim()).toMatch(/^\d+\.\d+\.\d+/);
-  });
-
-  test("text with no prompt and no stdin exits 1", async () => {
-    const { exitCode, stderr } = await run("text");
-    expect(exitCode).toBe(1);
-    expect(stderr).toContain("prompt is required");
-  });
-
-  test("text --help exits 0 and lists flags", async () => {
-    const { exitCode, stdout } = await run("text", "--help");
-    expect(exitCode).toBe(0);
-    expect(stdout).toContain("--model");
-    expect(stdout).toContain("--format");
-    expect(stdout).toContain("--temperature");
-  });
-
-  test("image --help exits 0 and lists flags", async () => {
-    const { exitCode, stdout } = await run("image", "--help");
-    expect(exitCode).toBe(0);
-    expect(stdout).toContain("--no-preview");
-    expect(stdout).toContain("--size");
-    expect(stdout).toContain("--aspect-ratio");
-  });
-
-  test("video --help exits 0 and lists flags", async () => {
-    const { exitCode, stdout } = await run("video", "--help");
-    expect(exitCode).toBe(0);
-    expect(stdout).toContain("--duration");
-    expect(stdout).toContain("--aspect-ratio");
-  });
-
-  test("models --type invalid exits 1", async () => {
-    const { exitCode, stderr } = await run("models", "--type", "audio");
-    expect(exitCode).toBe(1);
-    expect(stderr).toContain("must be one of");
-  });
-});

package/src/commands/image.ts

@@ -1,189 +0,0 @@
-import { generateImage, generateText, gateway } from "ai";
-import type { Command } from "commander";
-
-import { buildJobs, runJobs } from "../lib/jobs.js";
-import { fetchGatewayModels, resolveModels } from "../lib/models.js";
-import { parsePositiveInt, parseSize, parseAspectRatio } from "../lib/parse.js";
-import { readStdin } from "../lib/stdin.js";
-
-const DEFAULT_CONCURRENCY = 4;
-const DEFAULT_TIMEOUT_MS = 120_000;
-
-interface ImageOptions {
-  model?: string;
-  output?: string;
-  count?: string;
-  size?: string;
-  aspectRatio?: string;
-  quality?: string;
-  style?: string;
-  quiet?: boolean;
-  json?: boolean;
-  concurrency?: string;
-  preview?: boolean;
-}
-
-export function registerImageCommand(program: Command) {
-  program
-    .command("image")
-    .description("Generate an image from a prompt")
-    .argument("[prompt]", "The prompt to generate an image from")
-    .option(
-      "-m, --model <model>",
-      "Model ID (creator/model-name), comma-separated for multi-model"
-    )
-    .option("-o, --output <path>", "Output file path or directory")
-    .option("-n, --count <n>", "Number of images per model (default: 1)")
-    .option("--size <WxH>", "Image size (e.g. 1024x1024)")
-    .option("--aspect-ratio <W:H>", "Aspect ratio (e.g. 16:9)")
-    .option("--quality <level>", "Quality (standard, hd)")
-    .option("--style <style>", "Style (e.g. vivid, natural)")
-    .option("-q, --quiet", "Suppress progress output")
-    .option("--json", "Output metadata as JSON")
-    .option(
-      "--no-preview",
-      "Disable inline image preview in supported terminals"
-    )
-    .option(
-      "-p, --concurrency <n>",
-      `Max parallel generations (default: ${DEFAULT_CONCURRENCY})`
-    )
-    .action(async (rawPrompt: string | undefined, opts: ImageOptions) => {
-      const prompt = rawPrompt?.trim() || undefined;
-      const stdin = await readStdin();
-      if (!prompt && !stdin) {
-        process.stderr.write(
-          "Error: prompt is required (provide as argument or pipe via stdin)\n"
-        );
-        process.exit(1);
-      }
-      let imagePrompt: string | { images: Uint8Array[]; text?: string };
-      if (stdin) {
-        imagePrompt = prompt
-          ? { images: [new Uint8Array(stdin)], text: prompt }
-          : { images: [new Uint8Array(stdin)] };
-      } else {
-        imagePrompt = prompt!;
-      }
-
-      const gatewayModels = await fetchGatewayModels();
-      const models = resolveModels("image", opts.model, gatewayModels.image);
-      const countPerModel = opts.count
-        ? parsePositiveInt(opts.count, "count")
-        : 1;
-      const size = opts.size ? parseSize(opts.size) : undefined;
-      const aspectRatio = opts.aspectRatio
-        ? parseAspectRatio(opts.aspectRatio)
-        : undefined;
-      const provOpts = buildProviderOptions(opts);
-
-      if (
-        (opts.quality || opts.style) &&
-        models.every((m) => !m.startsWith("openai/"))
-      ) {
-        process.stderr.write(
-          "Warning: --quality and --style only apply to OpenAI models\n"
-        );
-      }
-
-      const jobs = buildJobs(models, countPerModel);
-
-      const { total, failed } = await runJobs(
-        jobs,
-        async (modelId) => {
-          const abort = AbortSignal.timeout(DEFAULT_TIMEOUT_MS);
-
-          if (gatewayModels.languageImageModelIds.has(modelId)) {
-            const messageContent: Array<
-              | { type: "text"; text: string }
-              | { type: "image"; image: Uint8Array }
-            > = [];
-            if (typeof imagePrompt === "string") {
-              messageContent.push({ type: "text", text: imagePrompt });
-            } else {
-              for (const img of imagePrompt.images) {
-                messageContent.push({ type: "image", image: img });
-              }
-              if (imagePrompt.text) {
-                messageContent.push({
-                  type: "text",
-                  text: imagePrompt.text,
-                });
-              } else {
-                messageContent.push({
-                  type: "text",
-                  text: "Generate an image",
-                });
-              }
-            }
-            const creator = gatewayModels.all.find(
-              (m) => m.id === modelId
-            )?.creator;
-            const result = await generateText({
-              headers: {
-                "http-referer": "https://github.com/vercel-labs/ai-cli",
-                "x-title": "ai-cli",
-              },
-              model: gateway(modelId),
-              messages: [{ role: "user", content: messageContent }],
-              abortSignal: abort,
-              providerOptions:
-                creator === "google"
-                  ? { google: { responseModalities: ["IMAGE", "TEXT"] } }
-                  : undefined,
-            });
-            const imageFile = result.files?.find((f) =>
-              f.mediaType.startsWith("image/")
-            );
-            if (!imageFile) {
-              throw new Error(
-                `Model ${modelId} did not return an image in the response`
-              );
-            }
-            return Buffer.from(imageFile.uint8Array);
-          }
-
-          const result = await generateImage({
-            headers: {
-              "http-referer": "https://github.com/vercel-labs/ai-cli",
-              "x-title": "ai-cli",
-            },
-            model: gateway.image(modelId),
-            prompt: imagePrompt,
-            abortSignal: abort,
-            n: 1,
-            size,
-            aspectRatio,
-            providerOptions:
-              Object.keys(provOpts).length > 0 ? provOpts : undefined,
-          });
-          return Buffer.from(result.image.uint8Array);
-        },
-        {
-          noun: "image",
-          format: "image",
-          outputPath: opts.output,
-          quiet: opts.quiet,
-          json: opts.json,
-          display: opts.preview,
-          concurrency: opts.concurrency
-            ? parsePositiveInt(opts.concurrency, "concurrency")
-            : DEFAULT_CONCURRENCY,
-        }
-      );
-      if (failed === total) process.exit(1);
-      if (failed > 0) process.exit(2);
-    });
-}
-
-function buildProviderOptions(
-  opts: ImageOptions
-): Record<string, Record<string, string>> {
-  const providerOptions: Record<string, Record<string, string>> = {};
-  if (opts.quality || opts.style) {
-    providerOptions.openai = {};
-    if (opts.quality) providerOptions.openai.quality = opts.quality;
-    if (opts.style) providerOptions.openai.style = opts.style;
-  }
-  return providerOptions;
-}

package/src/commands/models.ts

@@ -1,106 +0,0 @@
-import type { Command } from "commander";
-
-import {
-  fetchGatewayModels,
-  type Modality,
-  type ModelEntry,
-} from "../lib/models.js";
-
-function groupByCreator(models: ModelEntry[]): Map<string, ModelEntry[]> {
-  const groups = new Map<string, ModelEntry[]>();
-  for (const m of models) {
-    if (!groups.has(m.creator)) groups.set(m.creator, []);
-    groups.get(m.creator)!.push(m);
-  }
-  return new Map(
-    [...groups.entries()].sort((a, b) => a[0].localeCompare(b[0]))
-  );
-}
-
-function modelName(id: string): string {
-  const slash = id.indexOf("/");
-  return slash !== -1 ? id.slice(slash + 1) : id;
-}
-
-export function registerModelsCommand(program: Command) {
-  program
-    .command("models")
-    .description("List available models from AI Gateway")
-    .option("--type <type>", "Filter by type: text, image, video")
-    .option("--creator <name>", "Filter by creator (e.g. openai, google)")
-    .option("--json", "Output as JSON (includes descriptions)")
-    .action(
-      async (opts: { type?: string; creator?: string; json?: boolean }) => {
-        const validTypes = ["text", "image", "video"];
-        const filterType = opts.type?.toLowerCase() as Modality | undefined;
-        if (filterType && !validTypes.includes(filterType)) {
-          process.stderr.write(
-            `Error: --type must be one of: ${validTypes.join(", ")} (got "${opts.type}")\n`
-          );
-          process.exit(1);
-        }
-        const filterCreator = opts.creator?.toLowerCase();
-
-        const gatewayModels = await fetchGatewayModels();
-
-        if (opts.json) {
-          let entries = gatewayModels.all;
-          if (filterType) {
-            entries = entries.filter((m) =>
-              m.capabilities.includes(filterType)
-            );
-          }
-          if (filterCreator) {
-            entries = entries.filter(
-              (m) => m.creator.toLowerCase() === filterCreator
-            );
-          }
-          const output = entries.map((m) => ({
-            id: m.id,
-            ...(m.name ? { name: m.name } : {}),
-            ...(m.description ? { description: m.description } : {}),
-            creator: m.creator,
-            capabilities: m.capabilities,
-            ...(m.pricing ? { pricing: m.pricing } : {}),
-          }));
-          process.stdout.write(JSON.stringify(output, null, 2) + "\n");
-          return;
-        }
-
-        const sections: { title: string; entries: ModelEntry[] }[] = [];
-        if (!filterType || filterType === "text")
-          sections.push({ title: "Text", entries: gatewayModels.text });
-        if (!filterType || filterType === "image")
-          sections.push({ title: "Image", entries: gatewayModels.image });
-        if (!filterType || filterType === "video")
-          sections.push({ title: "Video", entries: gatewayModels.video });
-
-        let totalCount = 0;
-        for (const section of sections) {
-          let entries = section.entries;
-          if (filterCreator) {
-            entries = entries.filter(
-              (m) => m.creator.toLowerCase() === filterCreator
-            );
-          }
-          const grouped = groupByCreator(entries);
-          const count = [...grouped.values()].reduce((s, m) => s + m.length, 0);
-          if (count === 0) continue;
-          totalCount += count;
-          process.stdout.write(`\n${section.title} models (${count}):\n`);
-          for (const [creator, models] of grouped) {
-            process.stdout.write(`\n  ${creator}\n`);
-            for (const m of models) {
-              process.stdout.write(`    ${modelName(m.id)}\n`);
-            }
-          }
-        }
-
-        if (totalCount === 0) {
-          process.stderr.write("No models found matching filters\n");
-        } else {
-          process.stdout.write("\n");
-        }
-      }
-    );
-}

package/src/commands/text.ts

@@ -1,118 +0,0 @@
-import { generateText, gateway } from "ai";
-import type { Command } from "commander";
-
-import { buildJobs, runJobs } from "../lib/jobs.js";
-import { fetchGatewayModels, resolveModels } from "../lib/models.js";
-import type { OutputFormat } from "../lib/output.js";
-import { parsePositiveInt, parseTemperature } from "../lib/parse.js";
-import { readStdin, stdinAsText } from "../lib/stdin.js";
-
-const DEFAULT_CONCURRENCY = 4;
-const DEFAULT_TIMEOUT_MS = 120_000;
-
-interface TextOptions {
-  model?: string;
-  output?: string;
-  format?: string;
-  system?: string;
-  maxTokens?: string;
-  temperature?: string;
-  count?: string;
-  concurrency?: string;
-  quiet?: boolean;
-  json?: boolean;
-}
-
-function resolveFormat(fmt?: string): OutputFormat {
-  if (!fmt || fmt === "md") return "md";
-  if (fmt === "txt") return "txt";
-  throw new Error(`--format must be one of: md, txt (got "${fmt}")`);
-}
-
-export function registerTextCommand(program: Command) {
-  program
-    .command("text")
-    .description("Generate text from a prompt")
-    .argument("[prompt]", "The prompt to generate text from")
-    .option(
-      "-m, --model <model>",
-      "Model ID (creator/model-name), comma-separated for multi-model"
-    )
-    .option("-o, --output <path>", "Output file path or directory")
-    .option("-f, --format <fmt>", "Output format: md, txt (default: md)")
-    .option("-n, --count <n>", "Number of generations (default: 1)")
-    .option(
-      "-p, --concurrency <n>",
-      `Max parallel generations (default: ${DEFAULT_CONCURRENCY})`
-    )
-    .option("-s, --system <prompt>", "System prompt")
-    .option("--max-tokens <n>", "Maximum tokens to generate")
-    .option("-t, --temperature <n>", "Temperature (0-2)")
-    .option("-q, --quiet", "Suppress progress output")
-    .option("--json", "Output metadata as JSON")
-    .action(async (rawPrompt: string | undefined, opts: TextOptions) => {
-      const prompt = rawPrompt?.trim() || undefined;
-      const stdin = await readStdin();
-      if (!prompt && !stdin) {
-        process.stderr.write(
-          "Error: prompt is required (provide as argument or pipe via stdin)\n"
-        );
-        process.exit(1);
-      }
-      let fullPrompt: string;
-      if (stdin && prompt) {
-        fullPrompt = `${stdinAsText(stdin)}\n\n---\n\n${prompt}`;
-      } else if (stdin) {
-        fullPrompt = stdinAsText(stdin);
-      } else {
-        fullPrompt = prompt!;
-      }
-
-      const format = resolveFormat(opts.format);
-      const gatewayModels = await fetchGatewayModels();
-      const models = resolveModels("text", opts.model, gatewayModels.text);
-      const countPerModel = opts.count
-        ? parsePositiveInt(opts.count, "count")
-        : 1;
-      const maxTokens = opts.maxTokens
-        ? parsePositiveInt(opts.maxTokens, "max-tokens")
-        : undefined;
-      const temperature = opts.temperature
-        ? parseTemperature(opts.temperature)
-        : undefined;
-
-      const jobs = buildJobs(models, countPerModel);
-
-      const { total, failed } = await runJobs(
-        jobs,
-        async (modelId) => {
-          const abort = AbortSignal.timeout(DEFAULT_TIMEOUT_MS);
-          const result = await generateText({
-            headers: {
-              "http-referer": "https://github.com/vercel-labs/ai-cli",
-              "x-title": "ai-cli",
-            },
-            model: gateway(modelId),
-            prompt: fullPrompt,
-            system: opts.system,
-            maxOutputTokens: maxTokens,
-            temperature,
-            abortSignal: abort,
-          });
-          return result.text;
-        },
-        {
-          noun: "text",
-          format,
-          outputPath: opts.output,
-          quiet: opts.quiet,
-          json: opts.json,
-          concurrency: opts.concurrency
-            ? parsePositiveInt(opts.concurrency, "concurrency")
-            : DEFAULT_CONCURRENCY,
-        }
-      );
-      if (failed === total) process.exit(1);
-      if (failed > 0) process.exit(2);
-    });
-}

package/src/commands/video.ts

@@ -1,114 +0,0 @@
-import { experimental_generateVideo as generateVideo, gateway } from "ai";
-import type { Command } from "commander";
-
-import { buildJobs, runJobs } from "../lib/jobs.js";
-import { fetchGatewayModels, resolveModels } from "../lib/models.js";
-import {
-  parsePositiveInt,
-  parseAspectRatio,
-  parseNonNegativeFloat,
-} from "../lib/parse.js";
-import { readStdin } from "../lib/stdin.js";
-
-const DEFAULT_CONCURRENCY = 2;
-const DEFAULT_TIMEOUT_MS = 300_000;
-
-interface VideoOptions {
-  model?: string;
-  output?: string;
-  count?: string;
-  aspectRatio?: string;
-  duration?: string;
-  quiet?: boolean;
-  json?: boolean;
-  concurrency?: string;
-  preview?: boolean;
-}
-
-export function registerVideoCommand(program: Command) {
-  program
-    .command("video")
-    .description("Generate a video from a prompt")
-    .argument("[prompt]", "The prompt to generate a video from")
-    .option(
-      "-m, --model <model>",
-      "Model ID (creator/model-name), comma-separated for multi-model"
-    )
-    .option("-o, --output <path>", "Output file path or directory")
-    .option("-n, --count <n>", "Number of videos per model (default: 1)")
-    .option("--aspect-ratio <W:H>", "Aspect ratio (e.g. 16:9)")
-    .option("--duration <seconds>", "Video duration in seconds")
-    .option("-q, --quiet", "Suppress progress output")
-    .option("--json", "Output metadata as JSON")
-    .option(
-      "--no-preview",
-      "Disable inline video frame preview in supported terminals"
-    )
-    .option(
-      "-p, --concurrency <n>",
-      `Max parallel generations (default: ${DEFAULT_CONCURRENCY})`
-    )
-    .action(async (rawPrompt: string | undefined, opts: VideoOptions) => {
-      const prompt = rawPrompt?.trim() || undefined;
-      const stdin = await readStdin();
-      if (!prompt && !stdin) {
-        process.stderr.write(
-          "Error: prompt is required (provide as argument or pipe via stdin)\n"
-        );
-        process.exit(1);
-      }
-
-      let videoPrompt: string | { image: Uint8Array; text?: string } = prompt!;
-      if (stdin) {
-        videoPrompt = prompt
-          ? { image: new Uint8Array(stdin), text: prompt }
-          : { image: new Uint8Array(stdin) };
-      }
-
-      const gatewayModels = await fetchGatewayModels();
-      const models = resolveModels("video", opts.model, gatewayModels.video);
-      const countPerModel = opts.count
-        ? parsePositiveInt(opts.count, "count")
-        : 1;
-      const aspectRatio = opts.aspectRatio
-        ? parseAspectRatio(opts.aspectRatio)
-        : undefined;
-      const duration = opts.duration
-        ? parseNonNegativeFloat(opts.duration, "duration")
-        : undefined;
-
-      const jobs = buildJobs(models, countPerModel);
-
-      const { total, failed } = await runJobs(
-        jobs,
-        async (modelId) => {
-          const abort = AbortSignal.timeout(DEFAULT_TIMEOUT_MS);
-          const result = await generateVideo({
-            headers: {
-              "http-referer": "https://github.com/vercel-labs/ai-cli",
-              "x-title": "ai-cli",
-            },
-            model: gateway.video(modelId),
-            prompt: videoPrompt,
-            abortSignal: abort,
-            aspectRatio,
-            duration,
-          });
-          return Buffer.from(result.video.uint8Array);
-        },
-        {
-          noun: "video",
-          format: "video",
-          outputPath: opts.output,
-          quiet: opts.quiet,
-          json: opts.json,
-          display: opts.preview,
-          concurrency: opts.concurrency
-            ? parsePositiveInt(opts.concurrency, "concurrency")
-            : DEFAULT_CONCURRENCY,
-        }
-      );
-      if (failed === total) process.exit(1);
-      if (failed > 0) process.exit(2);
-    });
-}

package/src/index.ts

@@ -1,28 +0,0 @@
-#!/usr/bin/env bun
-import { Command } from "commander";
-
-import pkg from "../package.json";
-import { registerImageCommand } from "./commands/image.js";
-import { registerModelsCommand } from "./commands/models.js";
-import { registerTextCommand } from "./commands/text.js";
-import { registerVideoCommand } from "./commands/video.js";
-
-const program = new Command();
-
-program
-  .name("ai")
-  .description(
-    "A tiny, agent-native CLI for generating images, video and text with dead-simple commands, stdin support and predictable artifact outputs"
-  )
-  .version(pkg.version);
-
-registerTextCommand(program);
-registerImageCommand(program);
-registerVideoCommand(program);
-registerModelsCommand(program);
-
-program.parseAsync(process.argv).catch((err: unknown) => {
-  const msg = err instanceof Error ? err.message : String(err);
-  process.stderr.write(`Error: ${msg}\n`);
-  process.exit(1);
-});

package/src/lib/color.ts

@@ -1,5 +0,0 @@
-export function isColorEnabled(): boolean {
-  if (process.env.NO_COLOR !== undefined) return false;
-  if (process.env.FORCE_COLOR !== undefined) return true;
-  return !!process.stderr.isTTY;
-}