vargai 0.3.0

package/src/ai-sdk/video-model.ts

@@ -0,0 +1,185 @@
+/**
+ * VideoModelV3 - Extension to AI SDK for video generation
+ *
+ * This follows the same patterns as ImageModelV3 but adds video-specific
+ * options like duration, fps, and supports multimodal inputs via `files`.
+ *
+ * When upstream merges video support, we can migrate to their types.
+ */
+
+import type {
+  ImageModelV3File,
+  JSONArray,
+  JSONValue,
+  SharedV3ProviderOptions,
+  SharedV3Warning,
+} from "@ai-sdk/provider";
+
+/**
+ * Re-export ImageModelV3File as VideoModelV3File for clarity.
+ * They're the same type - works for image, audio, and video inputs.
+ */
+export type VideoModelV3File = ImageModelV3File;
+
+// ============================================================================
+// Call Options
+// ============================================================================
+
+export type VideoModelV3CallOptions = {
+  /**
+   * Prompt for the video generation.
+   */
+  prompt: string;
+
+  /**
+   * Number of videos to generate.
+   */
+  n: number;
+
+  /**
+   * Resolution of the videos to generate.
+   * Must have the format `{width}x{height}`.
+   * `undefined` will use the provider's default resolution.
+   */
+  resolution: `${number}x${number}` | undefined;
+
+  /**
+   * Aspect ratio of the videos to generate.
+   * Must have the format `{width}:{height}`.
+   * `undefined` will use the provider's default aspect ratio.
+   */
+  aspectRatio: `${number}:${number}` | undefined;
+
+  /**
+   * Duration of the generated video in seconds.
+   * `undefined` will use the provider's default duration.
+   */
+  duration: number | undefined;
+
+  /**
+   * Frames per second of the generated video.
+   * `undefined` will use the provider's default FPS.
+   */
+  fps: number | undefined;
+
+  /**
+   * Seed for the generation.
+   * `undefined` will use the provider's default seed.
+   */
+  seed: number | undefined;
+
+  /**
+   * Input files for video generation.
+   * - Images: for image-to-video generation
+   * - Audio: for audio-reactive video generation
+   * - Video: for video-to-video style transfer, extension, etc.
+   */
+  files: VideoModelV3File[] | undefined;
+
+  /**
+   * Additional provider-specific options.
+   */
+  providerOptions: SharedV3ProviderOptions;
+
+  /**
+   * Abort signal for cancelling the operation.
+   */
+  abortSignal?: AbortSignal;
+
+  /**
+   * Additional HTTP headers to be sent with the request.
+   */
+  headers?: Record<string, string | undefined>;
+};
+
+// ============================================================================
+// Provider Metadata
+// ============================================================================
+
+export type VideoModelV3ProviderMetadata = Record<
+  string,
+  {
+    videos: JSONArray;
+  } & JSONValue
+>;
+
+// ============================================================================
+// Usage
+// ============================================================================
+
+export type VideoModelV3Usage = {
+  /** Input tokens used (if applicable) */
+  inputTokens?: number;
+  /** Output tokens used (if applicable) */
+  outputTokens?: number;
+  /** Total tokens used */
+  totalTokens?: number;
+};
+
+// ============================================================================
+// Model Interface
+// ============================================================================
+
+type GetMaxVideosPerCallFunction = (options: {
+  modelId: string;
+}) => PromiseLike<number | undefined> | number | undefined;
+
+/**
+ * Video generation model specification version 3.
+ * Follows the same patterns as ImageModelV3.
+ */
+export type VideoModelV3 = {
+  /**
+   * The video model must specify which video model interface version it implements.
+   */
+  readonly specificationVersion: "v3";
+
+  /**
+   * Name of the provider for logging purposes.
+   */
+  readonly provider: string;
+
+  /**
+   * Provider-specific model ID for logging purposes.
+   */
+  readonly modelId: string;
+
+  /**
+   * Limit of how many videos can be generated in a single API call.
+   */
+  readonly maxVideosPerCall: number | undefined | GetMaxVideosPerCallFunction;
+
+  /**
+   * Generates videos.
+   */
+  doGenerate(options: VideoModelV3CallOptions): PromiseLike<{
+    /**
+     * Generated videos as base64 encoded strings or binary data.
+     */
+    videos: Array<string> | Array<Uint8Array>;
+
+    /**
+     * Warnings for the call, e.g. unsupported features.
+     */
+    warnings: Array<SharedV3Warning>;
+
+    /**
+     * Additional provider-specific metadata.
+     */
+    providerMetadata?: VideoModelV3ProviderMetadata;
+
+    /**
+     * Response information for telemetry and debugging purposes.
+     */
+    response: {
+      timestamp: Date;
+      modelId: string;
+      headers: Record<string, string> | undefined;
+    };
+
+    /**
+     * Optional token usage for the video generation call.
+     */
+    usage?: VideoModelV3Usage;
+  }>;
+};

package/src/cli/commands/find.tsx

@@ -0,0 +1,137 @@
+/**
+ * varg find command
+ * Ink-based search view
+ */
+
+import { defineCommand } from "citty";
+import { Box, Text } from "ink";
+import { registry } from "../../core/registry/index.ts";
+import { DataTable, Header, VargBox, VargText } from "../ui/index.ts";
+import { renderStatic } from "../ui/render.ts";
+
+interface FindViewProps {
+  query: string;
+  results: Array<{ name: string; description: string; type: string }>;
+}
+
+function FindView({ query, results }: FindViewProps) {
+  return (
+    <VargBox title={`search: ${query}`}>
+      <DataTable
+        rows={results.map((r) => ({
+          name: r.name,
+          description: r.description,
+          type: r.type as "model" | "action" | "skill",
+        }))}
+        showType
+      />
+      <Box marginTop={1}>
+        <VargText variant="muted">
+          {results.length} result{results.length > 1 ? "s" : ""}
+        </VargText>
+      </Box>
+    </VargBox>
+  );
+}
+
+function NoResultsView({ query }: { query: string }) {
+  return (
+    <Box flexDirection="column" padding={1}>
+      <VargText variant="warning">no results found for</VargText>
+      <Text> '{query}'</Text>
+      <Box marginTop={1}>
+        <VargText variant="muted">try </VargText>
+        <VargText variant="accent">varg list</VargText>
+        <VargText variant="muted"> to see all available items</VargText>
+      </Box>
+    </Box>
+  );
+}
+
+/** Help view for find command */
+function FindHelpView() {
+  return (
+    <VargBox title="varg find">
+      <Box marginBottom={1}>
+        <Text>search for models, actions, and skills</Text>
+      </Box>
+
+      <Header>USAGE</Header>
+      <Box paddingLeft={2} marginBottom={1}>
+        <VargText variant="accent">
+          varg find {"<query>"} [--type type]
+        </VargText>
+      </Box>
+
+      <Header>ARGUMENTS</Header>
+      <Box flexDirection="column" paddingLeft={2} marginBottom={1}>
+        <Text>query search term to find</Text>
+      </Box>
+
+      <Header>OPTIONS</Header>
+      <Box flexDirection="column" paddingLeft={2} marginBottom={1}>
+        <Text>--type filter by type: model, action, skill</Text>
+      </Box>
+
+      <Header>EXAMPLES</Header>
+      <Box flexDirection="column" paddingLeft={2}>
+        <Box flexDirection="column" marginBottom={1}>
+          <Text dimColor># search for video-related items</Text>
+          <VargText variant="accent">varg find video</VargText>
+        </Box>
+        <Box flexDirection="column">
+          <Text dimColor># search only in models</Text>
+          <VargText variant="accent">varg find flux --type model</VargText>
+        </Box>
+      </Box>
+    </VargBox>
+  );
+}
+
+/** Show find command help */
+export function showFindHelp() {
+  renderStatic(<FindHelpView />);
+}
+
+export const findCmd = defineCommand({
+  meta: {
+    name: "find",
+    description: "search for models, actions, and skills",
+  },
+  args: {
+    query: {
+      type: "positional",
+      description: "search query",
+      required: false,
+    },
+    type: {
+      type: "string",
+      description: "filter by type (model, action, skill)",
+    },
+  },
+  async run({ args, rawArgs }) {
+    // Handle --help
+    if (rawArgs.includes("--help") || rawArgs.includes("-h")) {
+      showFindHelp();
+      return;
+    }
+
+    const query = args.query as string;
+
+    if (!query) {
+      showFindHelp();
+      return;
+    }
+
+    const type = args.type as "model" | "action" | "skill" | undefined;
+
+    const results = registry.search(query, { type });
+
+    if (results.length === 0) {
+      renderStatic(<NoResultsView query={query} />);
+      return;
+    }
+
+    renderStatic(<FindView query={query} results={results} />);
+  },
+});

package/src/cli/commands/help.tsx

@@ -0,0 +1,85 @@
+/**
+ * varg help command
+ * Ink-based help display
+ */
+
+import { defineCommand } from "citty";
+import { Box, Text } from "ink";
+import { Header, HelpBlock, VargBox, VargText } from "../ui/index.ts";
+import { renderStatic } from "../ui/render.ts";
+
+interface CommandRowProps {
+  name: string;
+  description: string;
+}
+
+function CommandRow({ name, description }: CommandRowProps) {
+  return (
+    <Box paddingLeft={2}>
+      <VargText variant="accent">{name.padEnd(12)}</VargText>
+      <Text>{description}</Text>
+    </Box>
+  );
+}
+
+function HelpView() {
+  const examples = [
+    {
+      command: 'varg run video --prompt "ocean waves"',
+      description: "generate a video from text",
+    },
+    {
+      command: 'varg run video --prompt "person talking" --image photo.jpg',
+      description: "generate video from image",
+    },
+    {
+      command: 'varg run voice --text "hello world" --voice sam',
+      description: "text to speech",
+    },
+    { command: "varg list", description: "see all available" },
+    {
+      command: "varg which video",
+      description: "inspect an action or model",
+    },
+  ];
+
+  return (
+    <VargBox title="varg">
+      <Box marginBottom={1}>
+        <Text>ai video infrastructure from your terminal</Text>
+      </Box>
+
+      <Header>COMMANDS</Header>
+      <Box flexDirection="column" marginY={1}>
+        <CommandRow name="run" description="run a model, action, or skill" />
+        <CommandRow name="list" description="discover what's available" />
+        <CommandRow
+          name="find"
+          description="search for models, actions, skills"
+        />
+        <CommandRow name="which" description="inspect a specific item" />
+        <CommandRow name="help" description="show this help" />
+      </Box>
+
+      <Header>EXAMPLES</Header>
+      <Box marginTop={1}>
+        <HelpBlock examples={examples} />
+      </Box>
+    </VargBox>
+  );
+}
+
+/** Render help view - can be called directly */
+export function showHelp() {
+  renderStatic(<HelpView />);
+}
+
+export const helpCmd = defineCommand({
+  meta: {
+    name: "help",
+    description: "show help",
+  },
+  async run() {
+    showHelp();
+  },
+});

package/src/cli/commands/index.ts

@@ -0,0 +1,9 @@
+/**
+ * CLI commands exports
+ */
+
+export { findCmd, showFindHelp } from "./find.tsx";
+export { helpCmd, showHelp } from "./help.tsx";
+export { listCmd, showListHelp } from "./list.tsx";
+export { runCmd, showRunHelp, showTargetHelp } from "./run.tsx";
+export { showWhichHelp, whichCmd } from "./which.tsx";

package/src/cli/commands/list.tsx

@@ -0,0 +1,238 @@
+/**
+ * varg list command
+ * Ink-based discovery view
+ */
+
+import { defineCommand } from "citty";
+import { Box, Text } from "ink";
+import { registry } from "../../core/registry/index.ts";
+import { toJsonSchema } from "../../core/schema/helpers.ts";
+import type { Definition } from "../../core/schema/types.ts";
+import {
+  DataTable,
+  Header,
+  OptionRow,
+  Separator,
+  VargBox,
+  VargText,
+} from "../ui/index.ts";
+import { renderStatic } from "../ui/render.ts";
+
+interface ListViewProps {
+  filterType?: "model" | "action" | "skill";
+}
+
+function ListView({ filterType }: ListViewProps) {
+  const definitions = registry.list(filterType);
+
+  const models = definitions.filter((d) => d.type === "model");
+  const actions = definitions.filter((d) => d.type === "action");
+  const skills = definitions.filter((d) => d.type === "skill");
+  const stats = registry.stats;
+
+  return (
+    <VargBox title="varg">
+      {/* Models section */}
+      {(!filterType || filterType === "model") && (
+        <Box flexDirection="column" marginBottom={1}>
+          <Header>MODELS</Header>
+          <Box marginY={1}>
+            {models.length > 0 ? (
+              <DataTable
+                rows={models.map((m) => ({
+                  name: m.name,
+                  description: m.description,
+                  type: m.type as "model",
+                }))}
+              />
+            ) : (
+              <Box paddingLeft={2}>
+                <VargText variant="muted">no models registered</VargText>
+              </Box>
+            )}
+          </Box>
+        </Box>
+      )}
+
+      {/* Actions section */}
+      {(!filterType || filterType === "action") && (
+        <Box flexDirection="column" marginBottom={1}>
+          <Header>ACTIONS</Header>
+          <Box marginY={1}>
+            {actions.length > 0 ? (
+              <DataTable
+                rows={actions.map((a) => ({
+                  name: a.name,
+                  description: a.description,
+                  type: a.type as "action",
+                }))}
+              />
+            ) : (
+              <Box paddingLeft={2}>
+                <VargText variant="muted">no actions registered</VargText>
+              </Box>
+            )}
+          </Box>
+        </Box>
+      )}
+
+      {/* Skills section */}
+      {(!filterType || filterType === "skill") && (
+        <Box flexDirection="column" marginBottom={1}>
+          <Header>SKILLS</Header>
+          <Box marginY={1}>
+            {skills.length > 0 ? (
+              <DataTable
+                rows={skills.map((s) => ({
+                  name: s.name,
+                  description: s.description,
+                  type: s.type as "skill",
+                }))}
+              />
+            ) : (
+              <Box paddingLeft={2}>
+                <VargText variant="muted">no skills registered</VargText>
+              </Box>
+            )}
+          </Box>
+        </Box>
+      )}
+
+      {/* Footer */}
+      <Separator />
+      <Box marginTop={1}>
+        <Text>
+          {stats.models} models · {stats.actions} actions · {stats.skills}{" "}
+          skills
+        </Text>
+      </Box>
+      <Box marginTop={1}>
+        <VargText variant="muted">run </VargText>
+        <VargText variant="accent">varg which {"<name>"}</VargText>
+        <VargText variant="muted"> for details</VargText>
+      </Box>
+    </VargBox>
+  );
+}
+
+/**
+ * Convert a definition to an AI agent tool schema
+ */
+function definitionToToolSchema(def: Definition) {
+  const inputSchema = toJsonSchema(def.schema.input);
+
+  return {
+    type: "function" as const,
+    function: {
+      name: def.name,
+      description: def.description,
+      parameters: {
+        type: "object",
+        properties: inputSchema.properties || {},
+        required: inputSchema.required || [],
+      },
+    },
+  };
+}
+
+/**
+ * Generate all tools as JSON schemas for AI agents
+ */
+function generateToolsSchema(filterType?: "model" | "action" | "skill") {
+  const definitions = registry.list(filterType);
+
+  return {
+    tools: definitions.map(definitionToToolSchema),
+  };
+}
+
+/** Help view for list command */
+function ListHelpView() {
+  return (
+    <VargBox title="varg list">
+      <Box marginBottom={1}>
+        <Text>discover what's available</Text>
+      </Box>
+
+      <Header>USAGE</Header>
+      <Box paddingLeft={2} marginBottom={1}>
+        <VargText variant="accent">varg list [type] [options]</VargText>
+      </Box>
+
+      <Header>ARGUMENTS</Header>
+      <Box flexDirection="column" paddingLeft={2} marginBottom={1}>
+        <Text>type filter by type: model, action, skill</Text>
+      </Box>
+
+      <Header>OPTIONS</Header>
+      <Box flexDirection="column" paddingLeft={2} marginBottom={1}>
+        <OptionRow
+          name="schema"
+          description="output all tools as JSON schemas for AI agents"
+        />
+        <OptionRow name="help, -h" description="show this help" />
+      </Box>
+
+      <Header>EXAMPLES</Header>
+      <Box flexDirection="column" paddingLeft={2}>
+        <Box flexDirection="column" marginBottom={1}>
+          <Text dimColor># list everything</Text>
+          <VargText variant="accent">varg list</VargText>
+        </Box>
+        <Box flexDirection="column" marginBottom={1}>
+          <Text dimColor># list only models</Text>
+          <VargText variant="accent">varg list model</VargText>
+        </Box>
+        <Box flexDirection="column" marginBottom={1}>
+          <Text dimColor># list only actions</Text>
+          <VargText variant="accent">varg list action</VargText>
+        </Box>
+        <Box flexDirection="column">
+          <Text dimColor># export tools for AI agent</Text>
+          <VargText variant="accent">varg list --schema</VargText>
+        </Box>
+      </Box>
+    </VargBox>
+  );
+}
+
+/** Show list command help */
+export function showListHelp() {
+  renderStatic(<ListHelpView />);
+}
+
+export const listCmd = defineCommand({
+  meta: {
+    name: "list",
+    description: "discover what's available",
+  },
+  args: {
+    type: {
+      type: "positional",
+      description: "filter by type (model, action, skill)",
+      required: false,
+    },
+    schema: {
+      type: "boolean",
+      description: "output all tools as JSON schemas for AI agents",
+    },
+  },
+  async run({ args, rawArgs }) {
+    // Handle --help
+    if (rawArgs.includes("--help") || rawArgs.includes("-h")) {
+      showListHelp();
+      return;
+    }
+
+    const filterType = args.type as "model" | "action" | "skill" | undefined;
+
+    // Output JSON schemas for AI agents
+    if (args.schema) {
+      const toolsSchema = generateToolsSchema(filterType);
+      console.log(JSON.stringify(toolsSchema, null, 2));
+      return;
+    }
+
+    renderStatic(<ListView filterType={filterType} />);
+  },
+});