ai 5.0.0-beta.20 → 5.0.0-beta.21

package/CHANGELOG.md

@@ -1,5 +1,15 @@
 # ai
 
+## 5.0.0-beta.21
+
+### Patch Changes
+
+- 38ae5cc: feat (ai): export InferUIMessageChunk type
+- faea29f: fix (provider/openai): multi-step reasoning with text
+- 90ac328: fix (ui): tool part metadata support in ui messages
+- 4a1e0c8: fix(ai-cli): fix bundling and improve authentication error handling
+- 30ac566: fix (ui): text message metadata support in ui messages
+
 ## 5.0.0-beta.20
 
 ### Patch Changes

package/dist/bin/ai.js

@@ -24,6 +24,7 @@ __export(ai_exports, {
   formatAttachedFiles: () => formatAttachedFiles,
   getMediaType: () => getMediaType,
   isStdinAvailable: () => isStdinAvailable,
+  main: () => main,
   parseArgs: () => parseArgs,
   readFileContent: () => readFileContent,
   resolveModel: () => resolveModel,
@@ -192,125 +193,152 @@ function getResponseUIMessageId({
 var import_provider_utils2 = require("@ai-sdk/provider-utils");
 
 // src/ui-message-stream/ui-message-chunks.ts
+var import_v43 = require("zod/v4");
+
+// src/types/provider-metadata.ts
+var import_v42 = require("zod/v4");
+
+// src/types/json-value.ts
 var import_v4 = require("zod/v4");
-var uiMessageChunkSchema = import_v4.z.union([
-  import_v4.z.strictObject({
-    type: import_v4.z.literal("text-start"),
-    id: import_v4.z.string()
+var jsonValueSchema = import_v4.z.lazy(
+  () => import_v4.z.union([
+    import_v4.z.null(),
+    import_v4.z.string(),
+    import_v4.z.number(),
+    import_v4.z.boolean(),
+    import_v4.z.record(import_v4.z.string(), jsonValueSchema),
+    import_v4.z.array(jsonValueSchema)
+  ])
+);
+
+// src/types/provider-metadata.ts
+var providerMetadataSchema = import_v42.z.record(
+  import_v42.z.string(),
+  import_v42.z.record(import_v42.z.string(), jsonValueSchema)
+);
+
+// src/ui-message-stream/ui-message-chunks.ts
+var uiMessageChunkSchema = import_v43.z.union([
+  import_v43.z.strictObject({
+    type: import_v43.z.literal("text-start"),
+    id: import_v43.z.string(),
+    providerMetadata: providerMetadataSchema.optional()
   }),
-  import_v4.z.strictObject({
-    type: import_v4.z.literal("text-delta"),
-    id: import_v4.z.string(),
-    delta: import_v4.z.string()
+  import_v43.z.strictObject({
+    type: import_v43.z.literal("text-delta"),
+    id: import_v43.z.string(),
+    delta: import_v43.z.string(),
+    providerMetadata: providerMetadataSchema.optional()
   }),
-  import_v4.z.strictObject({
-    type: import_v4.z.literal("text-end"),
-    id: import_v4.z.string()
+  import_v43.z.strictObject({
+    type: import_v43.z.literal("text-end"),
+    id: import_v43.z.string(),
+    providerMetadata: providerMetadataSchema.optional()
   }),
-  import_v4.z.strictObject({
-    type: import_v4.z.literal("error"),
-    errorText: import_v4.z.string()
+  import_v43.z.strictObject({
+    type: import_v43.z.literal("error"),
+    errorText: import_v43.z.string()
   }),
-  import_v4.z.strictObject({
-    type: import_v4.z.literal("tool-input-start"),
-    toolCallId: import_v4.z.string(),
-    toolName: import_v4.z.string(),
-    providerExecuted: import_v4.z.boolean().optional()
+  import_v43.z.strictObject({
+    type: import_v43.z.literal("tool-input-start"),
+    toolCallId: import_v43.z.string(),
+    toolName: import_v43.z.string(),
+    providerExecuted: import_v43.z.boolean().optional()
   }),
-  import_v4.z.strictObject({
-    type: import_v4.z.literal("tool-input-delta"),
-    toolCallId: import_v4.z.string(),
-    inputTextDelta: import_v4.z.string()
+  import_v43.z.strictObject({
+    type: import_v43.z.literal("tool-input-delta"),
+    toolCallId: import_v43.z.string(),
+    inputTextDelta: import_v43.z.string()
   }),
-  import_v4.z.strictObject({
-    type: import_v4.z.literal("tool-input-available"),
-    toolCallId: import_v4.z.string(),
-    toolName: import_v4.z.string(),
-    input: import_v4.z.unknown(),
-    providerExecuted: import_v4.z.boolean().optional()
+  import_v43.z.strictObject({
+    type: import_v43.z.literal("tool-input-available"),
+    toolCallId: import_v43.z.string(),
+    toolName: import_v43.z.string(),
+    input: import_v43.z.unknown(),
+    providerExecuted: import_v43.z.boolean().optional(),
+    providerMetadata: providerMetadataSchema.optional()
   }),
-  import_v4.z.strictObject({
-    type: import_v4.z.literal("tool-output-available"),
-    toolCallId: import_v4.z.string(),
-    output: import_v4.z.unknown(),
-    providerExecuted: import_v4.z.boolean().optional()
+  import_v43.z.strictObject({
+    type: import_v43.z.literal("tool-output-available"),
+    toolCallId: import_v43.z.string(),
+    output: import_v43.z.unknown(),
+    providerExecuted: import_v43.z.boolean().optional()
   }),
-  import_v4.z.strictObject({
-    type: import_v4.z.literal("tool-output-error"),
-    toolCallId: import_v4.z.string(),
-    errorText: import_v4.z.string(),
-    providerExecuted: import_v4.z.boolean().optional()
+  import_v43.z.strictObject({
+    type: import_v43.z.literal("tool-output-error"),
+    toolCallId: import_v43.z.string(),
+    errorText: import_v43.z.string(),
+    providerExecuted: import_v43.z.boolean().optional()
   }),
-  import_v4.z.strictObject({
-    type: import_v4.z.literal("reasoning"),
-    text: import_v4.z.string(),
-    providerMetadata: import_v4.z.record(import_v4.z.string(), import_v4.z.any()).optional()
+  import_v43.z.strictObject({
+    type: import_v43.z.literal("reasoning"),
+    text: import_v43.z.string(),
+    providerMetadata: providerMetadataSchema.optional()
   }),
-  import_v4.z.strictObject({
-    type: import_v4.z.literal("reasoning-start"),
-    id: import_v4.z.string(),
-    providerMetadata: import_v4.z.record(import_v4.z.string(), import_v4.z.any()).optional()
+  import_v43.z.strictObject({
+    type: import_v43.z.literal("reasoning-start"),
+    id: import_v43.z.string(),
+    providerMetadata: providerMetadataSchema.optional()
   }),
-  import_v4.z.strictObject({
-    type: import_v4.z.literal("reasoning-delta"),
-    id: import_v4.z.string(),
-    delta: import_v4.z.string(),
-    providerMetadata: import_v4.z.record(import_v4.z.string(), import_v4.z.any()).optional()
+  import_v43.z.strictObject({
+    type: import_v43.z.literal("reasoning-delta"),
+    id: import_v43.z.string(),
+    delta: import_v43.z.string(),
+    providerMetadata: providerMetadataSchema.optional()
   }),
-  import_v4.z.strictObject({
-    type: import_v4.z.literal("reasoning-end"),
-    id: import_v4.z.string(),
-    providerMetadata: import_v4.z.record(import_v4.z.string(), import_v4.z.any()).optional()
+  import_v43.z.strictObject({
+    type: import_v43.z.literal("reasoning-end"),
+    id: import_v43.z.string(),
+    providerMetadata: providerMetadataSchema.optional()
   }),
-  import_v4.z.strictObject({
-    type: import_v4.z.literal("reasoning-part-finish")
+  import_v43.z.strictObject({
+    type: import_v43.z.literal("reasoning-part-finish")
   }),
-  import_v4.z.strictObject({
-    type: import_v4.z.literal("source-url"),
-    sourceId: import_v4.z.string(),
-    url: import_v4.z.string(),
-    title: import_v4.z.string().optional(),
-    providerMetadata: import_v4.z.any().optional()
-    // Use z.any() for generic metadata
+  import_v43.z.strictObject({
+    type: import_v43.z.literal("source-url"),
+    sourceId: import_v43.z.string(),
+    url: import_v43.z.string(),
+    title: import_v43.z.string().optional(),
+    providerMetadata: providerMetadataSchema.optional()
   }),
-  import_v4.z.strictObject({
-    type: import_v4.z.literal("source-document"),
-    sourceId: import_v4.z.string(),
-    mediaType: import_v4.z.string(),
-    title: import_v4.z.string(),
-    filename: import_v4.z.string().optional(),
-    providerMetadata: import_v4.z.any().optional()
-    // Use z.any() for generic metadata
+  import_v43.z.strictObject({
+    type: import_v43.z.literal("source-document"),
+    sourceId: import_v43.z.string(),
+    mediaType: import_v43.z.string(),
+    title: import_v43.z.string(),
+    filename: import_v43.z.string().optional(),
+    providerMetadata: providerMetadataSchema.optional()
   }),
-  import_v4.z.strictObject({
-    type: import_v4.z.literal("file"),
-    url: import_v4.z.string(),
-    mediaType: import_v4.z.string()
+  import_v43.z.strictObject({
+    type: import_v43.z.literal("file"),
+    url: import_v43.z.string(),
+    mediaType: import_v43.z.string(),
+    providerMetadata: providerMetadataSchema.optional()
   }),
-  import_v4.z.strictObject({
-    type: import_v4.z.string().startsWith("data-"),
-    id: import_v4.z.string().optional(),
-    data: import_v4.z.unknown(),
-    transient: import_v4.z.boolean().optional()
+  import_v43.z.strictObject({
+    type: import_v43.z.string().startsWith("data-"),
+    id: import_v43.z.string().optional(),
+    data: import_v43.z.unknown(),
+    transient: import_v43.z.boolean().optional()
   }),
-  import_v4.z.strictObject({
-    type: import_v4.z.literal("start-step")
+  import_v43.z.strictObject({
+    type: import_v43.z.literal("start-step")
   }),
-  import_v4.z.strictObject({
-    type: import_v4.z.literal("finish-step")
+  import_v43.z.strictObject({
+    type: import_v43.z.literal("finish-step")
   }),
-  import_v4.z.strictObject({
-    type: import_v4.z.literal("start"),
-    messageId: import_v4.z.string().optional(),
-    messageMetadata: import_v4.z.unknown().optional()
+  import_v43.z.strictObject({
+    type: import_v43.z.literal("start"),
+    messageId: import_v43.z.string().optional(),
+    messageMetadata: import_v43.z.unknown().optional()
   }),
-  import_v4.z.strictObject({
-    type: import_v4.z.literal("finish"),
-    messageMetadata: import_v4.z.unknown().optional()
+  import_v43.z.strictObject({
+    type: import_v43.z.literal("finish"),
+    messageMetadata: import_v43.z.unknown().optional()
   }),
-  import_v4.z.strictObject({
-    type: import_v4.z.literal("message-metadata"),
-    messageMetadata: import_v4.z.unknown()
+  import_v43.z.strictObject({
+    type: import_v43.z.literal("message-metadata"),
+    messageMetadata: import_v43.z.unknown()
   })
 ]);
 function isDataUIMessageChunk(chunk) {
@@ -725,7 +753,7 @@ function processUIMessageStream({
     new TransformStream({
       async transform(part, controller) {
         await runUpdateMessageJob(async ({ state, write }) => {
-          var _a9, _b;
+          var _a9, _b, _c, _d;
           function updateToolInvocationPart(options) {
             var _a10;
             const part2 = state.message.parts.find(
@@ -739,6 +767,9 @@ function processUIMessageStream({
               anyPart.output = anyOptions.output;
               anyPart.errorText = anyOptions.errorText;
               anyPart.providerExecuted = (_a10 = anyOptions.providerExecuted) != null ? _a10 : part2.providerExecuted;
+              if (anyOptions.providerMetadata != null && part2.state === "input-available") {
+                part2.callProviderMetadata = anyOptions.providerMetadata;
+              }
             } else {
               state.message.parts.push({
                 type: `tool-${options.toolName}`,
@@ -747,7 +778,8 @@ function processUIMessageStream({
                 input: anyOptions.input,
                 output: anyOptions.output,
                 errorText: anyOptions.errorText,
-                providerExecuted: anyOptions.providerExecuted
+                providerExecuted: anyOptions.providerExecuted,
+                ...anyOptions.providerMetadata != null ? { callProviderMetadata: anyOptions.providerMetadata } : {}
               });
             }
           }
@@ -768,6 +800,7 @@ function processUIMessageStream({
               const textPart = {
                 type: "text",
                 text: "",
+                providerMetadata: part.providerMetadata,
                 state: "streaming"
               };
               state.activeTextParts[part.id] = textPart;
@@ -776,13 +809,16 @@ function processUIMessageStream({
               break;
             }
             case "text-delta": {
-              state.activeTextParts[part.id].text += part.delta;
+              const textPart = state.activeTextParts[part.id];
+              textPart.text += part.delta;
+              textPart.providerMetadata = (_a9 = part.providerMetadata) != null ? _a9 : textPart.providerMetadata;
               write();
               break;
             }
             case "text-end": {
               const textPart = state.activeTextParts[part.id];
               textPart.state = "done";
+              textPart.providerMetadata = (_b = part.providerMetadata) != null ? _b : textPart.providerMetadata;
               delete state.activeTextParts[part.id];
               write();
               break;
@@ -802,13 +838,13 @@ function processUIMessageStream({
             case "reasoning-delta": {
               const reasoningPart = state.activeReasoningParts[part.id];
               reasoningPart.text += part.delta;
-              reasoningPart.providerMetadata = (_a9 = part.providerMetadata) != null ? _a9 : reasoningPart.providerMetadata;
+              reasoningPart.providerMetadata = (_c = part.providerMetadata) != null ? _c : reasoningPart.providerMetadata;
               write();
               break;
             }
             case "reasoning-end": {
               const reasoningPart = state.activeReasoningParts[part.id];
-              reasoningPart.providerMetadata = (_b = part.providerMetadata) != null ? _b : reasoningPart.providerMetadata;
+              reasoningPart.providerMetadata = (_d = part.providerMetadata) != null ? _d : reasoningPart.providerMetadata;
               reasoningPart.state = "done";
               delete state.activeReasoningParts[part.id];
               write();
@@ -884,7 +920,8 @@ function processUIMessageStream({
                 toolName: part.toolName,
                 state: "input-available",
                 input: part.input,
-                providerExecuted: part.providerExecuted
+                providerExecuted: part.providerExecuted,
+                providerMetadata: part.providerMetadata
               });
               write();
               if (onToolCall && !part.providerExecuted) {
@@ -1615,7 +1652,7 @@ async function download({ url }) {
 // src/prompt/data-content.ts
 var import_provider6 = require("@ai-sdk/provider");
 var import_provider_utils5 = require("@ai-sdk/provider-utils");
-var import_v42 = require("zod/v4");
+var import_v44 = require("zod/v4");
 
 // src/prompt/split-data-url.ts
 function splitDataUrl(dataUrl) {
@@ -1634,11 +1671,11 @@ function splitDataUrl(dataUrl) {
 }
 
 // src/prompt/data-content.ts
-var dataContentSchema = import_v42.z.union([
-  import_v42.z.string(),
-  import_v42.z.instanceof(Uint8Array),
-  import_v42.z.instanceof(ArrayBuffer),
-  import_v42.z.custom(
+var dataContentSchema = import_v44.z.union([
+  import_v44.z.string(),
+  import_v44.z.instanceof(Uint8Array),
+  import_v44.z.instanceof(ArrayBuffer),
+  import_v44.z.custom(
     // Buffer might not be available in some environments such as CloudFlare:
     (value) => {
       var _a9, _b;
@@ -2186,28 +2223,6 @@ var import_v47 = require("zod/v4");
 // src/prompt/message.ts
 var import_v46 = require("zod/v4");
 
-// src/types/provider-metadata.ts
-var import_v44 = require("zod/v4");
-
-// src/types/json-value.ts
-var import_v43 = require("zod/v4");
-var jsonValueSchema = import_v43.z.lazy(
-  () => import_v43.z.union([
-    import_v43.z.null(),
-    import_v43.z.string(),
-    import_v43.z.number(),
-    import_v43.z.boolean(),
-    import_v43.z.record(import_v43.z.string(), jsonValueSchema),
-    import_v43.z.array(jsonValueSchema)
-  ])
-);
-
-// src/types/provider-metadata.ts
-var providerMetadataSchema = import_v44.z.record(
-  import_v44.z.string(),
-  import_v44.z.record(import_v44.z.string(), jsonValueSchema)
-);
-
 // src/prompt/content-part.ts
 var import_v45 = require("zod/v4");
 var textPartSchema = import_v45.z.object({
@@ -3038,7 +3053,11 @@ function toResponseMessages({
   ).filter((part) => part.type !== "text" || part.text.length > 0).map((part) => {
     switch (part.type) {
       case "text":
-        return part;
+        return {
+          type: "text",
+          text: part.text,
+          providerOptions: part.providerMetadata
+        };
       case "reasoning":
         return {
           type: "reasoning",
@@ -3293,7 +3312,7 @@ var DefaultStreamTextResult = class {
     let activeReasoningContent = {};
     const eventProcessor = new TransformStream({
       async transform(chunk, controller) {
-        var _a9, _b;
+        var _a9, _b, _c;
         controller.enqueue(chunk);
         const { part } = chunk;
         if (part.type === "text" || part.type === "reasoning" || part.type === "source" || part.type === "tool-call" || part.type === "tool-result" || part.type === "tool-input-start" || part.type === "tool-input-delta" || part.type === "raw") {
@@ -3323,7 +3342,7 @@ var DefaultStreamTextResult = class {
             return;
           }
           activeText.text += part.text;
-          activeText.providerMetadata = part.providerMetadata;
+          activeText.providerMetadata = (_a9 = part.providerMetadata) != null ? _a9 : activeText.providerMetadata;
         }
         if (part.type === "text-end") {
           delete activeTextContent[part.id];
@@ -3349,7 +3368,7 @@ var DefaultStreamTextResult = class {
             return;
           }
           activeReasoning.text += part.text;
-          activeReasoning.providerMetadata = (_a9 = part.providerMetadata) != null ? _a9 : activeReasoning.providerMetadata;
+          activeReasoning.providerMetadata = (_b = part.providerMetadata) != null ? _b : activeReasoning.providerMetadata;
         }
         if (part.type === "reasoning-end") {
           const activeReasoning = activeReasoningContent[part.id];
@@ -3363,7 +3382,7 @@ var DefaultStreamTextResult = class {
             });
             return;
           }
-          activeReasoning.providerMetadata = (_b = part.providerMetadata) != null ? _b : activeReasoning.providerMetadata;
+          activeReasoning.providerMetadata = (_c = part.providerMetadata) != null ? _c : activeReasoning.providerMetadata;
           delete activeReasoningContent[part.id];
         }
         if (part.type === "file") {
@@ -4063,26 +4082,35 @@ var DefaultStreamTextResult = class {
           const partType = part.type;
           switch (partType) {
             case "text-start": {
-              controller.enqueue({ type: "text-start", id: part.id });
+              controller.enqueue({
+                type: "text-start",
+                id: part.id,
+                ...part.providerMetadata != null ? { providerMetadata: part.providerMetadata } : {}
+              });
               break;
             }
             case "text": {
               controller.enqueue({
                 type: "text-delta",
                 id: part.id,
-                delta: part.text
+                delta: part.text,
+                ...part.providerMetadata != null ? { providerMetadata: part.providerMetadata } : {}
               });
               break;
             }
             case "text-end": {
-              controller.enqueue({ type: "text-end", id: part.id });
+              controller.enqueue({
+                type: "text-end",
+                id: part.id,
+                ...part.providerMetadata != null ? { providerMetadata: part.providerMetadata } : {}
+              });
               break;
             }
             case "reasoning-start": {
               controller.enqueue({
                 type: "reasoning-start",
                 id: part.id,
-                providerMetadata: part.providerMetadata
+                ...part.providerMetadata != null ? { providerMetadata: part.providerMetadata } : {}
               });
               break;
             }
@@ -4092,7 +4120,7 @@ var DefaultStreamTextResult = class {
                   type: "reasoning-delta",
                   id: part.id,
                   delta: part.text,
-                  providerMetadata: part.providerMetadata
+                  ...part.providerMetadata != null ? { providerMetadata: part.providerMetadata } : {}
                 });
               }
               break;
@@ -4101,7 +4129,7 @@ var DefaultStreamTextResult = class {
               controller.enqueue({
                 type: "reasoning-end",
                 id: part.id,
-                providerMetadata: part.providerMetadata
+                ...part.providerMetadata != null ? { providerMetadata: part.providerMetadata } : {}
               });
               break;
             }
@@ -4120,7 +4148,7 @@ var DefaultStreamTextResult = class {
                   sourceId: part.id,
                   url: part.url,
                   title: part.title,
-                  providerMetadata: part.providerMetadata
+                  ...part.providerMetadata != null ? { providerMetadata: part.providerMetadata } : {}
                 });
               }
               if (sendSources && part.sourceType === "document") {
@@ -4130,7 +4158,7 @@ var DefaultStreamTextResult = class {
                   mediaType: part.mediaType,
                   title: part.title,
                   filename: part.filename,
-                  providerMetadata: part.providerMetadata
+                  ...part.providerMetadata != null ? { providerMetadata: part.providerMetadata } : {}
                 });
               }
               break;
@@ -4158,7 +4186,8 @@ var DefaultStreamTextResult = class {
                 toolCallId: part.toolCallId,
                 toolName: part.toolName,
                 input: part.input,
-                providerExecuted: part.providerExecuted
+                providerExecuted: part.providerExecuted,
+                providerMetadata: part.providerMetadata
               });
               break;
             }
@@ -4453,43 +4482,27 @@ function showHelp() {
 
 AI CLI - Stream text generation from various AI models
 
-Arguments:
-  prompt                   The prompt to send to the AI model (optional if using stdin)
-
 Options:
-  -m, --model <model>      Model to use. Format: provider/model or just model name.
-                           Examples: openai/gpt-4o, anthropic/claude-3-5-sonnet-20241022
-                           (default: "openai/gpt-4")
-  -f, --file <file>        Attach a file to the prompt (can be used multiple times)
-  -s, --system <message>   System message to set context
-  -v, --verbose            Show detailed information (model, usage, etc.)
-  -h, --help               Display help for command
-  -V, --version            Output the version number
-
-Environment Variables:
-  - AI_MODEL: Default model to use
-  - AI_SYSTEM: Default system message  
-  - AI_VERBOSE: Set to 'true' to enable verbose output
+  -m, --model <model>      Model to use (default: "openai/gpt-4")
+                           Format: provider/model (e.g., anthropic/claude-3-5-sonnet)
+  -f, --file <file>        Attach file(s) to prompt
+  -s, --system <message>   System message
+  -v, --verbose            Show detailed output
+  -h, --help               Show help
+  -V, --version            Show version
 
-Authentication (choose one):
-  - VERCEL_OIDC_TOKEN: Vercel OIDC token (for Vercel projects)
-  - AI_GATEWAY_API_KEY: AI Gateway API key
+Authentication (required):
+  export AI_GATEWAY_API_KEY="your-key"     # Get from Vercel Dashboard (AI tab)
+  export VERCEL_OIDC_TOKEN="your-token"   # For Vercel projects (or run: vercel env pull)
 
-Setting Environment Variables:
-  # Option 1: Export in current session
-  export AI_GATEWAY_API_KEY="your-key-here"
-  export AI_MODEL="anthropic/claude-3-5-sonnet-20241022"
-
-  # Option 2: Inline for single command
-  AI_GATEWAY_API_KEY="your-key" ai "Hello world"
-
-  # Option 3: Add to shell profile (~/.bashrc, ~/.zshrc)
-  echo 'export AI_GATEWAY_API_KEY="your-key"' >> ~/.bashrc
+Environment Variables:
+  AI_MODEL: Default model to use
+  AI_SYSTEM: Default system message
+  AI_VERBOSE: Set to 'true' for detailed output
 
 Examples:
   npx ai "Hello, world!"
-  npx ai "Write a poem" -m anthropic/claude-3-5-sonnet-20241022
-  npx ai "Explain quantum physics" -m groq/llama-3.1-8b-instant
+  npx ai "Write a poem" -m anthropic/claude-3-5-sonnet
   npx ai "Explain this code" -f script.js -f README.md
   echo "What is life?" | npx ai
   cat file.txt | npx ai "Summarize this content"
@@ -4499,17 +4512,6 @@ Unix-style piping:
   echo "Hello world" | npx ai "Translate to French"
   cat README.md | npx ai "Summarize this"
   curl -s https://api.github.com/repos/vercel/ai | npx ai "What is this repository about?"
-
-Authentication Setup:
-  This CLI uses the Vercel AI Gateway. You need ONE of these for authentication:
-
-  OIDC Token (for Vercel projects):
-  - Automatically available in Vercel deployments  
-  - For local development: run 'vercel env pull' or use 'vercel dev'
-  
-  API Key (for any environment):
-  - Get your key from the AI Gateway dashboard
-  - Set: export AI_GATEWAY_API_KEY="your-key-here"
   
   The gateway supports OpenAI, Anthropic, Google, Groq, and more providers.`);
 }
@@ -4595,6 +4597,25 @@ ${prompt}` : stdinContent;
       }
       console.error("");
     }
+    const hasApiKey = process.env.AI_GATEWAY_API_KEY || process.env.VERCEL_OIDC_TOKEN;
+    if (!hasApiKey) {
+      console.error(`Error: Authentication required.
+
+Set up authentication with one of these options:
+
+  # Option 1: Export in current session
+  export AI_GATEWAY_API_KEY="your-key-here"
+  export VERCEL_OIDC_TOKEN="your-oidc-token"
+  export AI_MODEL="anthropic/claude-3-5-sonnet"
+
+  # Option 2: Add to shell profile (~/.bashrc, ~/.zshrc)
+  echo 'export AI_GATEWAY_API_KEY="your-key"' >> ~/.bashrc
+  # Or run: vercel env pull
+  
+Get your API key from the Vercel Dashboard (AI tab > API keys).
+Use --help for more details and examples.`);
+      process.exit(1);
+    }
     const model = resolveModel(options.model);
     let messages;
     if (imageFiles.length > 0) {
@@ -4655,6 +4676,7 @@ main().catch((error) => {
   formatAttachedFiles,
   getMediaType,
   isStdinAvailable,
+  main,
   parseArgs,
   readFileContent,
   resolveModel,