@chat-js/cli 0.1.4 → 0.2.1

package/templates/chat-app/lib/ai/tools/generate-video.ts

@@ -0,0 +1,166 @@
+import { experimental_generateVideo as generateVideo, tool } from "ai";
+import { z } from "zod";
+import { type AppModelId, getAppModelDefinition } from "@/lib/ai/app-models";
+import { getVideoModel } from "@/lib/ai/providers";
+import { uploadFile } from "@/lib/blob";
+import { config } from "@/lib/config";
+import type { CostAccumulator } from "@/lib/credits/cost-accumulator";
+import { createModuleLogger } from "@/lib/logger";
+import { toolsDefinitions } from "./tools-definitions";
+
+interface GenerateVideoProps {
+  costAccumulator?: CostAccumulator;
+  selectedModel?: string;
+}
+
+const log = createModuleLogger("ai.tools.generate-video");
+const DEFAULT_ASPECT_RATIO = "16:9";
+const DEFAULT_DURATION_SECONDS = 5;
+const ALLOWED_EXTENSIONS = new Set(["mp4", "webm", "mov"]);
+
+function resolveVideoExtension(mediaType?: string): string {
+  if (!mediaType) {
+    return "mp4";
+  }
+
+  const [, subtypeWithParams] = mediaType.split("/");
+  if (!subtypeWithParams) {
+    return "mp4";
+  }
+
+  const subtype = subtypeWithParams.split(";")[0]?.trim().toLowerCase();
+  if (!subtype) {
+    return "mp4";
+  }
+
+  const mappedSubtype = subtype === "quicktime" ? "mov" : subtype;
+  return ALLOWED_EXTENSIONS.has(mappedSubtype) ? mappedSubtype : "mp4";
+}
+
+async function resolveVideoModel(selectedModel?: string): Promise<string> {
+  if (selectedModel) {
+    try {
+      const model = await getAppModelDefinition(selectedModel as AppModelId);
+      if (model.output.video) {
+        return selectedModel;
+      }
+    } catch {
+      // Not in app models registry, fall through
+    }
+  }
+  return config.ai.tools.video.default;
+}
+
+export const generateVideoTool = ({
+  selectedModel,
+  costAccumulator,
+}: GenerateVideoProps = {}) =>
+  tool({
+    description:
+      "Generate a short video clip from a text prompt. Use this when the user asks to create, make, or generate a video.",
+    inputSchema: z.object({
+      prompt: z
+        .string()
+        .describe("A descriptive prompt for the video to generate."),
+      aspectRatio: z
+        .enum(["16:9", "9:16", "1:1"])
+        .optional()
+        .describe("Optional output aspect ratio. Defaults to 16:9."),
+      durationSeconds: z
+        .number()
+        .int()
+        .min(1)
+        .max(10)
+        .optional()
+        .describe("Optional video duration in seconds. Defaults to 5."),
+    }),
+    execute: async ({ prompt, aspectRatio, durationSeconds }) => {
+      const startMs = Date.now();
+      const finalAspectRatio = aspectRatio ?? DEFAULT_ASPECT_RATIO;
+      const finalDurationSeconds = durationSeconds ?? DEFAULT_DURATION_SECONDS;
+
+      log.info(
+        {
+          promptLength: prompt.length,
+          selectedModel,
+          aspectRatio: finalAspectRatio,
+          durationSeconds: finalDurationSeconds,
+        },
+        "generateVideo: start"
+      );
+
+      try {
+        const modelId = await resolveVideoModel(selectedModel);
+        const isGoogleModel =
+          modelId.startsWith("google/") || modelId.includes("gemini");
+
+        log.debug({ modelId }, "generateVideo: resolved model");
+
+        const result = await generateVideo({
+          model: getVideoModel(modelId),
+          prompt,
+          aspectRatio: finalAspectRatio,
+          duration: finalDurationSeconds,
+          providerOptions: {
+            ...(isGoogleModel && {
+              google: {
+                aspectRatio: finalAspectRatio,
+              },
+            }),
+          },
+        });
+
+        const video = result.video;
+        if (!video) {
+          throw new Error("No video generated");
+        }
+
+        const buffer = Buffer.from(video.uint8Array);
+        const timestamp = Date.now();
+        const ext = resolveVideoExtension(video.mediaType);
+        const filename = `generated-video-${timestamp}.${ext}`;
+        const uploaded = await uploadFile(filename, buffer);
+
+        costAccumulator?.addAPICost(
+          "generateVideo",
+          toolsDefinitions.generateVideo.cost
+        );
+
+        log.info(
+          {
+            ms: Date.now() - startMs,
+            modelId,
+            videoUrl: uploaded.url,
+          },
+          "generateVideo: success"
+        );
+
+        return { videoUrl: uploaded.url, prompt };
+      } catch (error) {
+        const errorMessage = error instanceof Error ? error.message : "";
+        const isUnsupportedVideoGateway = errorMessage.includes(
+          "does not support video models"
+        );
+
+        log.error(
+          {
+            ms: Date.now() - startMs,
+            selectedModel,
+            error:
+              error instanceof Error
+                ? { message: error.message, name: error.name }
+                : error,
+          },
+          "generateVideo: failure"
+        );
+
+        if (isUnsupportedVideoGateway) {
+          throw new Error(
+            "Video generation is not available for the active gateway."
+          );
+        }
+
+        throw error;
+      }
+    },
+  });

package/templates/chat-app/lib/ai/tools/get-weather.ts

@@ -23,40 +23,40 @@ export const getWeather = tool({
   },
 });
 
-export type WeatherAtLocation = {
-  latitude: number;
-  longitude: number;
-  generationtime_ms: number;
-  utc_offset_seconds: number;
-  timezone: string;
-  timezone_abbreviation: string;
-  elevation: number;
-  current_units: {
-    time: string;
-    interval: string;
-    temperature_2m: string;
-  };
+export interface WeatherAtLocation {
   current: {
     time: string;
     interval: number;
     temperature_2m: number;
   };
-  hourly_units: {
+  current_units: {
     time: string;
+    interval: string;
     temperature_2m: string;
   };
-  hourly: {
+  daily: {
     time: string[];
-    temperature_2m: number[];
+    sunrise: string[];
+    sunset: string[];
   };
   daily_units: {
     time: string;
     sunrise: string;
     sunset: string;
   };
-  daily: {
+  elevation: number;
+  generationtime_ms: number;
+  hourly: {
     time: string[];
-    sunrise: string[];
-    sunset: string[];
+    temperature_2m: number[];
   };
-};
+  hourly_units: {
+    time: string;
+    temperature_2m: string;
+  };
+  latitude: number;
+  longitude: number;
+  timezone: string;
+  timezone_abbreviation: string;
+  utc_offset_seconds: number;
+}

package/templates/chat-app/lib/ai/tools/read-document.ts

@@ -4,10 +4,10 @@ import type { ToolSession } from "@/lib/ai/tools/types";
 import { getDocumentById } from "@/lib/db/queries";
 import type { StreamWriter } from "../types";
 
-type ReadDocumentProps = {
-  session: ToolSession;
+interface ReadDocumentProps {
   dataStream: StreamWriter;
-};
+  session: ToolSession;
+}
 
 export const readDocument = ({
   session,

package/templates/chat-app/lib/ai/tools/steps/multi-query-web-search.ts

@@ -4,30 +4,30 @@ import { deduplicateByDomainAndUrl } from "./search-utils";
 import type { SearchProviderOptions } from "./web-search";
 import { webSearchStep } from "./web-search";
 
-export type SearchQuery = {
-  query: string;
+export interface SearchQuery {
   maxResults: number;
-};
+  query: string;
+}
 
-export type MultiQuerySearchOptions = {
+export interface MultiQuerySearchOptions {
   baseProviderOptions: SearchProviderOptions;
-  topics?: string[];
   excludeDomains?: string[];
-};
+  topics?: string[];
+}
 
-type MultiQuerySearchResult = {
+interface MultiQuerySearchResult {
   query: SearchQuery;
   results: Array<{
     url: string;
     title: string;
     content: string;
   }>;
-};
+}
 
-export type MultiQuerySearchResponse = {
-  searches: MultiQuerySearchResult[];
+export interface MultiQuerySearchResponse {
   error?: string;
-};
+  searches: MultiQuerySearchResult[];
+}
 
 export async function multiQueryWebSearchStep({
   queries,

package/templates/chat-app/lib/ai/tools/steps/web-search.ts

@@ -11,17 +11,17 @@ export type SearchProviderOptions =
       provider: "firecrawl";
     } & SearchParams);
 
-export type WebSearchResult = {
+export interface WebSearchResult {
+  content: string;
   source: "web";
   title: string;
   url: string;
-  content: string;
-};
+}
 
-export type WebSearchResponse = {
-  results: WebSearchResult[];
+export interface WebSearchResponse {
   error?: string;
-};
+  results: WebSearchResult[];
+}
 
 // Initialize search providers lazily to avoid runtime errors when keys are missing
 const tvly = env.TAVILY_API_KEY ? tavily({ apiKey: env.TAVILY_API_KEY }) : null;

package/templates/chat-app/lib/ai/tools/tools-definitions.ts

@@ -61,7 +61,12 @@ export const toolsDefinitions: Record<ToolName, ToolDefinition> = {
   generateImage: {
     name: "generateImage",
     description: "Generate images from text descriptions",
-    cost: 17, // Nano banana pro ~17¢
+    cost: 0, // LLM cost tracked via token usage
+  },
+  generateVideo: {
+    name: "generateVideo",
+    description: "Generate video clips from text descriptions",
+    cost: 0, // LLM cost tracked via token usage
   },
   deepResearch: {
     name: "deepResearch",
@@ -71,8 +76,8 @@ export const toolsDefinitions: Record<ToolName, ToolDefinition> = {
 };
 
 export const allTools = toolNameSchema.options;
-type ToolDefinition = {
-  name: string;
-  description: string;
+interface ToolDefinition {
   cost: number;
-};
+  description: string;
+  name: string;
+}

package/templates/chat-app/lib/ai/tools/tools.ts

@@ -10,6 +10,7 @@ import { editCodeDocumentTool } from "@/lib/ai/tools/documents/edit-code-documen
 import { editSheetDocumentTool } from "@/lib/ai/tools/documents/edit-sheet-document";
 import { editTextDocumentTool } from "@/lib/ai/tools/documents/edit-text-document";
 import { generateImageTool } from "@/lib/ai/tools/generate-image";
+import { generateVideoTool } from "@/lib/ai/tools/generate-video";
 import { getWeather } from "@/lib/ai/tools/get-weather";
 import { readDocument } from "@/lib/ai/tools/read-document";
 import { retrieveUrl } from "@/lib/ai/tools/retrieve-url";
@@ -62,8 +63,8 @@ export function getTools({
       session,
       dataStream,
     }),
-    ...(config.features.urlRetrieval ? { retrieveUrl } : {}),
-    ...(config.features.webSearch
+    ...(config.ai.tools.urlRetrieval.enabled ? { retrieveUrl } : {}),
+    ...(config.ai.tools.webSearch.enabled
       ? {
           webSearch: tavilyWebSearch({
             dataStream,
@@ -73,10 +74,10 @@ export function getTools({
         }
       : {}),
 
-    ...(config.features.sandbox
+    ...(config.ai.tools.codeExecution.enabled
       ? { codeExecution: codeExecution({ costAccumulator }) }
       : {}),
-    ...(config.features.imageGeneration
+    ...(config.ai.tools.image.enabled
       ? {
           generateImage: generateImageTool({
             attachments,
@@ -86,7 +87,7 @@ export function getTools({
           }),
         }
       : {}),
-    ...(config.features.deepResearch
+    ...(config.ai.tools.deepResearch.enabled
       ? {
           deepResearch: deepResearch({
             session,
@@ -97,6 +98,14 @@ export function getTools({
           }),
         }
       : {}),
+    ...(config.ai.tools.video.enabled
+      ? {
+          generateVideo: generateVideoTool({
+            selectedModel,
+            costAccumulator,
+          }),
+        }
+      : {}),
   };
 }
 
@@ -113,7 +122,7 @@ export async function getMcpTools({
   tools: Record<string, Tool>;
   cleanup: () => Promise<void>;
 }> {
-  if (!config.features.mcp) {
+  if (!config.ai.tools.mcp.enabled) {
     return {
       tools: {},
       cleanup: async () => Promise.resolve(),

package/templates/chat-app/lib/ai/tools/types.ts

@@ -3,6 +3,6 @@ import type { Session } from "@/lib/auth";
 type SessionUser = NonNullable<Session["user"]>;
 
 // Minimal session slice for tools - derived from Better Auth Session to avoid drift
-export type ToolSession = {
+export interface ToolSession {
   user?: Pick<SessionUser, "id">;
-};
+}

package/templates/chat-app/lib/ai/types.ts

@@ -8,6 +8,7 @@ import { z } from "zod";
 import type { codeExecution } from "@/lib/ai/tools/code-execution";
 import type { deepResearch } from "@/lib/ai/tools/deep-research/deep-research";
 import type { generateImageTool as generateImageToolFactory } from "@/lib/ai/tools/generate-image";
+import type { generateVideoTool as generateVideoToolFactory } from "@/lib/ai/tools/generate-video";
 import type { getWeather } from "@/lib/ai/tools/get-weather";
 import type { readDocument } from "@/lib/ai/tools/read-document";
 import type { retrieveUrl } from "@/lib/ai/tools/retrieve-url";
@@ -34,6 +35,7 @@ export const toolNameSchema = z.enum([
   "webSearch",
   "codeExecution",
   "generateImage",
+  "generateVideo",
   "deepResearch",
 ]);
 
@@ -45,6 +47,7 @@ const frontendToolsSchema = z.enum([
   "webSearch",
   "deepResearch",
   "generateImage",
+  "generateVideo",
   "createTextDocument",
   "createCodeDocument",
   "createSheetDocument",
@@ -91,37 +94,43 @@ type readDocumentTool = InferUITool<ReturnType<typeof readDocument>>;
 type generateImageTool = InferUITool<
   ReturnType<typeof generateImageToolFactory>
 >;
+type generateVideoTool = InferUITool<
+  ReturnType<typeof generateVideoToolFactory>
+>;
 type webSearchTool = InferUITool<ReturnType<typeof tavilyWebSearch>>;
 type codeExecutionTool = InferUITool<ReturnType<typeof codeExecution>>;
 type retrieveUrlTool = InferUITool<typeof retrieveUrl>;
 
+// biome-ignore lint/style/useConsistentTypeDefinitions: <explanation>
 export type ChatTools = {
-  getWeather: weatherTool;
-  createTextDocument: createTextDocumentToolType;
+  codeExecution: codeExecutionTool;
   createCodeDocument: createCodeDocumentToolType;
   createSheetDocument: createSheetDocumentToolType;
-  editTextDocument: editTextDocumentToolType;
+  createTextDocument: createTextDocumentToolType;
+  deepResearch: deepResearchTool;
   editCodeDocument: editCodeDocumentToolType;
   editSheetDocument: editSheetDocumentToolType;
-  deepResearch: deepResearchTool;
-  readDocument: readDocumentTool;
+  editTextDocument: editTextDocumentToolType;
   generateImage: generateImageTool;
-  webSearch: webSearchTool;
-  codeExecution: codeExecutionTool;
+  generateVideo: generateVideoTool;
+  getWeather: weatherTool;
+  readDocument: readDocumentTool;
   retrieveUrl: retrieveUrlTool;
+  webSearch: webSearchTool;
 };
 
-type FollowupSuggestions = {
+interface FollowupSuggestions {
   suggestions: string[];
-};
+}
 
+// biome-ignore lint/style/useConsistentTypeDefinitions: <explanation>
 export type CustomUIDataTypes = {
   appendMessage: string;
   chatConfirmed: {
     chatId: string;
   };
-  researchUpdate: ResearchUpdate;
   followupSuggestions: FollowupSuggestions;
+  researchUpdate: ResearchUpdate;
 };
 
 export type ChatMessage = Omit<
@@ -137,8 +146,8 @@ export type ToolOutput<T extends ToolName> = ChatTools[T]["output"];
 
 export type StreamWriter = UIMessageStreamWriter<ChatMessage>;
 
-export type Attachment = {
+export interface Attachment {
+  contentType: string;
   name: string;
   url: string;
-  contentType: string;
-};
+}

package/templates/chat-app/lib/artifacts/code/client.tsx

@@ -56,10 +56,10 @@ function detectRequiredHandlers(code: string): string[] {
   return handlers;
 }
 
-type Metadata = {
-  outputs: ConsoleOutput[];
+interface Metadata {
   language: string;
-};
+  outputs: ConsoleOutput[];
+}
 
 export const codeArtifact = new Artifact<"code", Metadata>({
   kind: "code",
@@ -257,7 +257,7 @@ export const codeArtifact = new Artifact<"code", Metadata>({
             },
           ],
           metadata: {
-            selectedModel: config.models.defaults.codeEdits,
+            selectedModel: config.ai.tools.code.edits,
             createdAt: new Date(),
             parentMessageId: storeApi.getState().getLastMessageId(),
             activeStreamId: null,
@@ -278,7 +278,7 @@ export const codeArtifact = new Artifact<"code", Metadata>({
             },
           ],
           metadata: {
-            selectedModel: config.models.defaults.codeEdits,
+            selectedModel: config.ai.tools.code.edits,
             createdAt: new Date(),
             parentMessageId: storeApi.getState().getLastMessageId(),
             activeStreamId: null,

package/templates/chat-app/lib/artifacts/sheet/client.tsx

@@ -87,7 +87,7 @@ export const sheetArtifact = new Artifact<"sheet", Metadata>({
             { type: "text", text: "Can you please format and clean the data?" },
           ],
           metadata: {
-            selectedModel: config.models.defaults.formatSheet,
+            selectedModel: config.ai.tools.sheet.format,
             createdAt: new Date(),
             parentMessageId: storeApi.getState().getLastMessageId(),
             activeStreamId: null,
@@ -108,7 +108,7 @@ export const sheetArtifact = new Artifact<"sheet", Metadata>({
             },
           ],
           metadata: {
-            selectedModel: config.models.defaults.analyzeSheet,
+            selectedModel: config.ai.tools.sheet.analyze,
             createdAt: new Date(),
             parentMessageId: storeApi.getState().getLastMessageId(),
             activeStreamId: null,

package/templates/chat-app/lib/artifacts/text/client.tsx

@@ -1,10 +1,25 @@
 import { Copy, History, Pen, Redo2, Undo2 } from "lucide-react";
+import dynamic from "next/dynamic";
 import { toast } from "sonner";
 import { Artifact } from "@/components/create-artifact";
-import { DiffView } from "@/components/diffview";
 import { DocumentSkeleton } from "@/components/document-skeleton";
-import { Editor } from "@/components/text-editor";
 import { config } from "@/lib/config";
+
+const DiffView = dynamic(
+  () => import("@/components/diffview").then((m) => ({ default: m.DiffView })),
+  {
+    loading: () => <DocumentSkeleton artifactKind="text" />,
+    ssr: false,
+  }
+);
+
+const Editor = dynamic(
+  () => import("@/components/text-editor").then((m) => ({ default: m.Editor })),
+  {
+    loading: () => <DocumentSkeleton artifactKind="text" />,
+    ssr: false,
+  }
+);
 export const textArtifact = new Artifact<"text">({
   kind: "text",
   description: "Useful for text content, like drafting essays and emails.",
@@ -113,7 +128,7 @@ export const textArtifact = new Artifact<"text">({
             },
           ],
           metadata: {
-            selectedModel: config.models.defaults.polishText,
+            selectedModel: config.ai.tools.text.polish,
             createdAt: new Date(),
             parentMessageId: storeApi.getState().getLastMessageId(),
             activeStreamId: null,

package/templates/chat-app/lib/clone-messages.test.ts

@@ -1,6 +1,11 @@
 import assert from "node:assert/strict";
-import { describe, it } from "vitest";
+import { describe, it, vi } from "vitest";
 import type { ChatMessage } from "./ai/types";
+
+vi.mock("@/lib/config", () => ({
+  config: { appPrefix: "test" },
+}));
+
 import { cloneMessagesWithDocuments } from "./clone-messages";
 
 function createMessage({

package/templates/chat-app/lib/config-requirements.ts

@@ -1,12 +1,16 @@
 import type { GatewayType } from "./ai/gateways/registry";
-import type { AuthenticationConfig, FeaturesConfig } from "./config-schema";
+import type {
+  AiConfig,
+  AuthenticationConfig,
+  FeaturesConfig,
+} from "./config-schema";
 
 type EnvVarName = keyof NodeJS.ProcessEnv;
 
-export type EnvRequirement = {
-  options: EnvVarName[][];
+export interface EnvRequirement {
   description: string;
-};
+  options: EnvVarName[][];
+}
 
 export const gatewayEnvRequirements: Record<GatewayType, EnvRequirement> = {
   openrouter: {
@@ -29,6 +33,15 @@ export const gatewayEnvRequirements: Record<GatewayType, EnvRequirement> = {
 
 export const featureEnvRequirements: Partial<
   Record<keyof FeaturesConfig, EnvRequirement>
+> = {
+  attachments: {
+    options: [["BLOB_READ_WRITE_TOKEN"]],
+    description: "BLOB_READ_WRITE_TOKEN",
+  },
+};
+
+export const aiToolEnvRequirements: Partial<
+  Record<keyof AiConfig["tools"], EnvRequirement>
 > = {
   webSearch: {
     options: [["TAVILY_API_KEY"], ["FIRECRAWL_API_KEY"]],
@@ -46,7 +59,7 @@ export const featureEnvRequirements: Partial<
     options: [["MCP_ENCRYPTION_KEY"]],
     description: "MCP_ENCRYPTION_KEY",
   },
-  sandbox: {
+  codeExecution: {
     options: [
       ["VERCEL_OIDC_TOKEN"],
       ["VERCEL_TEAM_ID", "VERCEL_PROJECT_ID", "VERCEL_TOKEN"],
@@ -54,11 +67,7 @@ export const featureEnvRequirements: Partial<
     description:
       "VERCEL_OIDC_TOKEN (auto on Vercel) or VERCEL_TEAM_ID + VERCEL_PROJECT_ID + VERCEL_TOKEN",
   },
-  imageGeneration: {
-    options: [["BLOB_READ_WRITE_TOKEN"]],
-    description: "BLOB_READ_WRITE_TOKEN",
-  },
-  attachments: {
+  image: {
     options: [["BLOB_READ_WRITE_TOKEN"]],
     description: "BLOB_READ_WRITE_TOKEN",
   },