ai 5.0.0-beta.21 → 5.0.0-beta.22

package/CHANGELOG.md

@@ -1,5 +1,19 @@
 # ai
 
+## 5.0.0-beta.22
+
+### Patch Changes
+
+- de2d2ab: feat(ai): add provider and provider registry middleware functionality
+- 6c42e56: feat (ai): validate ui stream data chunks
+- c93a8bc: chore(ai): export AsyncIterableStream type from async-iterable-stream module
+- 20398f2: feat: ai sdk cli documentation + adjusted default model
+- 86293e5: fix (ai): use correct generateMessageId in streamText toUIMessageStream
+- 205077b: fix: improve Zod compatibility
+- Updated dependencies [205077b]
+  - @ai-sdk/provider-utils@3.0.0-beta.4
+  - @ai-sdk/gateway@1.0.0-beta.9
+
 ## 5.0.0-beta.21
 
 ### Patch Changes

package/dist/bin/ai.js

@@ -751,24 +751,24 @@ function processUIMessageStream({
 }) {
   return stream.pipeThrough(
     new TransformStream({
-      async transform(part, controller) {
+      async transform(chunk, controller) {
         await runUpdateMessageJob(async ({ state, write }) => {
           var _a9, _b, _c, _d;
           function updateToolInvocationPart(options) {
             var _a10;
-            const part2 = state.message.parts.find(
-              (part3) => isToolUIPart(part3) && part3.toolCallId === options.toolCallId
+            const part = state.message.parts.find(
+              (part2) => isToolUIPart(part2) && part2.toolCallId === options.toolCallId
             );
             const anyOptions = options;
-            const anyPart = part2;
-            if (part2 != null) {
-              part2.state = options.state;
+            const anyPart = part;
+            if (part != null) {
+              part.state = options.state;
               anyPart.input = anyOptions.input;
               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;
+              anyPart.providerExecuted = (_a10 = anyOptions.providerExecuted) != null ? _a10 : part.providerExecuted;
+              if (anyOptions.providerMetadata != null && part.state === "input-available") {
+                part.callProviderMetadata = anyOptions.providerMetadata;
               }
             } else {
               state.message.parts.push({
@@ -795,31 +795,31 @@ function processUIMessageStream({
               state.message.metadata = mergedMetadata;
             }
           }
-          switch (part.type) {
+          switch (chunk.type) {
             case "text-start": {
               const textPart = {
                 type: "text",
                 text: "",
-                providerMetadata: part.providerMetadata,
+                providerMetadata: chunk.providerMetadata,
                 state: "streaming"
               };
-              state.activeTextParts[part.id] = textPart;
+              state.activeTextParts[chunk.id] = textPart;
               state.message.parts.push(textPart);
               write();
               break;
             }
             case "text-delta": {
-              const textPart = state.activeTextParts[part.id];
-              textPart.text += part.delta;
-              textPart.providerMetadata = (_a9 = part.providerMetadata) != null ? _a9 : textPart.providerMetadata;
+              const textPart = state.activeTextParts[chunk.id];
+              textPart.text += chunk.delta;
+              textPart.providerMetadata = (_a9 = chunk.providerMetadata) != null ? _a9 : textPart.providerMetadata;
               write();
               break;
             }
             case "text-end": {
-              const textPart = state.activeTextParts[part.id];
+              const textPart = state.activeTextParts[chunk.id];
               textPart.state = "done";
-              textPart.providerMetadata = (_b = part.providerMetadata) != null ? _b : textPart.providerMetadata;
-              delete state.activeTextParts[part.id];
+              textPart.providerMetadata = (_b = chunk.providerMetadata) != null ? _b : textPart.providerMetadata;
+              delete state.activeTextParts[chunk.id];
               write();
               break;
             }
@@ -827,34 +827,34 @@ function processUIMessageStream({
               const reasoningPart = {
                 type: "reasoning",
                 text: "",
-                providerMetadata: part.providerMetadata,
+                providerMetadata: chunk.providerMetadata,
                 state: "streaming"
               };
-              state.activeReasoningParts[part.id] = reasoningPart;
+              state.activeReasoningParts[chunk.id] = reasoningPart;
               state.message.parts.push(reasoningPart);
               write();
               break;
             }
             case "reasoning-delta": {
-              const reasoningPart = state.activeReasoningParts[part.id];
-              reasoningPart.text += part.delta;
-              reasoningPart.providerMetadata = (_c = part.providerMetadata) != null ? _c : reasoningPart.providerMetadata;
+              const reasoningPart = state.activeReasoningParts[chunk.id];
+              reasoningPart.text += chunk.delta;
+              reasoningPart.providerMetadata = (_c = chunk.providerMetadata) != null ? _c : reasoningPart.providerMetadata;
               write();
               break;
             }
             case "reasoning-end": {
-              const reasoningPart = state.activeReasoningParts[part.id];
-              reasoningPart.providerMetadata = (_d = part.providerMetadata) != null ? _d : reasoningPart.providerMetadata;
+              const reasoningPart = state.activeReasoningParts[chunk.id];
+              reasoningPart.providerMetadata = (_d = chunk.providerMetadata) != null ? _d : reasoningPart.providerMetadata;
               reasoningPart.state = "done";
-              delete state.activeReasoningParts[part.id];
+              delete state.activeReasoningParts[chunk.id];
               write();
               break;
             }
             case "file": {
               state.message.parts.push({
                 type: "file",
-                mediaType: part.mediaType,
-                url: part.url
+                mediaType: chunk.mediaType,
+                url: chunk.url
               });
               write();
               break;
@@ -862,10 +862,10 @@ function processUIMessageStream({
             case "source-url": {
               state.message.parts.push({
                 type: "source-url",
-                sourceId: part.sourceId,
-                url: part.url,
-                title: part.title,
-                providerMetadata: part.providerMetadata
+                sourceId: chunk.sourceId,
+                url: chunk.url,
+                title: chunk.title,
+                providerMetadata: chunk.providerMetadata
               });
               write();
               break;
@@ -873,40 +873,40 @@ function processUIMessageStream({
             case "source-document": {
               state.message.parts.push({
                 type: "source-document",
-                sourceId: part.sourceId,
-                mediaType: part.mediaType,
-                title: part.title,
-                filename: part.filename,
-                providerMetadata: part.providerMetadata
+                sourceId: chunk.sourceId,
+                mediaType: chunk.mediaType,
+                title: chunk.title,
+                filename: chunk.filename,
+                providerMetadata: chunk.providerMetadata
               });
               write();
               break;
             }
             case "tool-input-start": {
               const toolInvocations = state.message.parts.filter(isToolUIPart);
-              state.partialToolCalls[part.toolCallId] = {
+              state.partialToolCalls[chunk.toolCallId] = {
                 text: "",
-                toolName: part.toolName,
+                toolName: chunk.toolName,
                 index: toolInvocations.length
               };
               updateToolInvocationPart({
-                toolCallId: part.toolCallId,
-                toolName: part.toolName,
+                toolCallId: chunk.toolCallId,
+                toolName: chunk.toolName,
                 state: "input-streaming",
                 input: void 0,
-                providerExecuted: part.providerExecuted
+                providerExecuted: chunk.providerExecuted
               });
               write();
               break;
             }
             case "tool-input-delta": {
-              const partialToolCall = state.partialToolCalls[part.toolCallId];
-              partialToolCall.text += part.inputTextDelta;
+              const partialToolCall = state.partialToolCalls[chunk.toolCallId];
+              partialToolCall.text += chunk.inputTextDelta;
               const { value: partialArgs } = await parsePartialJson(
                 partialToolCall.text
               );
               updateToolInvocationPart({
-                toolCallId: part.toolCallId,
+                toolCallId: chunk.toolCallId,
                 toolName: partialToolCall.toolName,
                 state: "input-streaming",
                 input: partialArgs
@@ -916,24 +916,24 @@ function processUIMessageStream({
             }
             case "tool-input-available": {
               updateToolInvocationPart({
-                toolCallId: part.toolCallId,
-                toolName: part.toolName,
+                toolCallId: chunk.toolCallId,
+                toolName: chunk.toolName,
                 state: "input-available",
-                input: part.input,
-                providerExecuted: part.providerExecuted,
-                providerMetadata: part.providerMetadata
+                input: chunk.input,
+                providerExecuted: chunk.providerExecuted,
+                providerMetadata: chunk.providerMetadata
               });
               write();
-              if (onToolCall && !part.providerExecuted) {
+              if (onToolCall && !chunk.providerExecuted) {
                 const result = await onToolCall({
-                  toolCall: part
+                  toolCall: chunk
                 });
                 if (result != null) {
                   updateToolInvocationPart({
-                    toolCallId: part.toolCallId,
-                    toolName: part.toolName,
+                    toolCallId: chunk.toolCallId,
+                    toolName: chunk.toolName,
                     state: "output-available",
-                    input: part.input,
+                    input: chunk.input,
                     output: result
                   });
                   write();
@@ -947,7 +947,7 @@ function processUIMessageStream({
                 throw new Error("tool_result must be preceded by a tool_call");
               }
               const toolInvocationIndex = toolInvocations.findIndex(
-                (invocation) => invocation.toolCallId === part.toolCallId
+                (invocation) => invocation.toolCallId === chunk.toolCallId
               );
               if (toolInvocationIndex === -1) {
                 throw new Error(
@@ -958,12 +958,12 @@ function processUIMessageStream({
                 toolInvocations[toolInvocationIndex]
               );
               updateToolInvocationPart({
-                toolCallId: part.toolCallId,
+                toolCallId: chunk.toolCallId,
                 toolName,
                 state: "output-available",
                 input: toolInvocations[toolInvocationIndex].input,
-                output: part.output,
-                providerExecuted: part.providerExecuted
+                output: chunk.output,
+                providerExecuted: chunk.providerExecuted
               });
               write();
               break;
@@ -974,7 +974,7 @@ function processUIMessageStream({
                 throw new Error("tool_result must be preceded by a tool_call");
               }
               const toolInvocationIndex = toolInvocations.findIndex(
-                (invocation) => invocation.toolCallId === part.toolCallId
+                (invocation) => invocation.toolCallId === chunk.toolCallId
               );
               if (toolInvocationIndex === -1) {
                 throw new Error(
@@ -985,12 +985,12 @@ function processUIMessageStream({
                 toolInvocations[toolInvocationIndex]
               );
               updateToolInvocationPart({
-                toolCallId: part.toolCallId,
+                toolCallId: chunk.toolCallId,
                 toolName,
                 state: "output-error",
                 input: toolInvocations[toolInvocationIndex].input,
-                errorText: part.errorText,
-                providerExecuted: part.providerExecuted
+                errorText: chunk.errorText,
+                providerExecuted: chunk.providerExecuted
               });
               write();
               break;
@@ -1005,62 +1005,65 @@ function processUIMessageStream({
               break;
             }
             case "start": {
-              if (part.messageId != null) {
-                state.message.id = part.messageId;
+              if (chunk.messageId != null) {
+                state.message.id = chunk.messageId;
               }
-              await updateMessageMetadata(part.messageMetadata);
-              if (part.messageId != null || part.messageMetadata != null) {
+              await updateMessageMetadata(chunk.messageMetadata);
+              if (chunk.messageId != null || chunk.messageMetadata != null) {
                 write();
               }
               break;
             }
             case "finish": {
-              await updateMessageMetadata(part.messageMetadata);
-              if (part.messageMetadata != null) {
+              await updateMessageMetadata(chunk.messageMetadata);
+              if (chunk.messageMetadata != null) {
                 write();
               }
               break;
             }
             case "message-metadata": {
-              await updateMessageMetadata(part.messageMetadata);
-              if (part.messageMetadata != null) {
+              await updateMessageMetadata(chunk.messageMetadata);
+              if (chunk.messageMetadata != null) {
                 write();
               }
               break;
             }
             case "error": {
-              onError == null ? void 0 : onError(new Error(part.errorText));
+              onError == null ? void 0 : onError(new Error(chunk.errorText));
               break;
             }
             default: {
-              if (isDataUIMessageChunk(part)) {
-                const dataPart = part;
-                if (dataPart.transient) {
-                  onData == null ? void 0 : onData(dataPart);
+              if (isDataUIMessageChunk(chunk)) {
+                if ((dataPartSchemas == null ? void 0 : dataPartSchemas[chunk.type]) != null) {
+                  await (0, import_provider_utils2.validateTypes)({
+                    value: chunk.data,
+                    schema: dataPartSchemas[chunk.type]
+                  });
+                }
+                const dataChunk = chunk;
+                if (dataChunk.transient) {
+                  onData == null ? void 0 : onData(dataChunk);
                   break;
                 }
-                const existingPart = dataPart.id != null ? state.message.parts.find(
-                  (partArg) => dataPart.type === partArg.type && dataPart.id === partArg.id
+                const existingUIPart = dataChunk.id != null ? state.message.parts.find(
+                  (chunkArg) => dataChunk.type === chunkArg.type && dataChunk.id === chunkArg.id
                 ) : void 0;
-                if (existingPart != null) {
-                  existingPart.data = isObject(existingPart.data) && isObject(dataPart.data) ? mergeObjects(existingPart.data, dataPart.data) : dataPart.data;
+                if (existingUIPart != null) {
+                  existingUIPart.data = dataChunk.data;
                 } else {
-                  state.message.parts.push(dataPart);
+                  state.message.parts.push(dataChunk);
                 }
-                onData == null ? void 0 : onData(dataPart);
+                onData == null ? void 0 : onData(dataChunk);
                 write();
               }
             }
           }
-          controller.enqueue(part);
+          controller.enqueue(chunk);
         });
       }
     })
   );
 }
-function isObject(value) {
-  return typeof value === "object" && value !== null;
-}
 
 // src/ui-message-stream/handle-ui-message-stream-finish.ts
 function handleUIMessageStreamFinish({
@@ -4071,10 +4074,10 @@ var DefaultStreamTextResult = class {
     sendFinish = true,
     onError = import_provider16.getErrorMessage
   } = {}) {
-    const responseMessageId = getResponseUIMessageId({
+    const responseMessageId = generateMessageId != null ? getResponseUIMessageId({
       originalMessages,
-      responseMessageId: this.generateId
-    });
+      responseMessageId: generateMessageId
+    }) : void 0;
     const baseStream = this.fullStream.pipeThrough(
       new TransformStream({
         transform: async (part, controller) => {
@@ -4228,8 +4231,8 @@ var DefaultStreamTextResult = class {
               if (sendStart) {
                 controller.enqueue({
                   type: "start",
-                  messageId: responseMessageId,
-                  messageMetadata: messageMetadataValue
+                  ...messageMetadataValue != null ? { messageMetadata: messageMetadataValue } : {},
+                  ...responseMessageId != null ? { messageId: responseMessageId } : {}
                 });
               }
               break;
@@ -4238,7 +4241,7 @@ var DefaultStreamTextResult = class {
               if (sendFinish) {
                 controller.enqueue({
                   type: "finish",
-                  messageMetadata: messageMetadataValue
+                  ...messageMetadataValue != null ? { messageMetadata: messageMetadataValue } : {}
                 });
               }
               break;
@@ -4412,7 +4415,7 @@ function readFileContent(filePath) {
 function parseArgs() {
   const args = process.argv.slice(2);
   const options = {
-    model: process.env.AI_MODEL || "openai/gpt-4",
+    model: process.env.AI_DEFAULT_MODEL || "openai/gpt-4",
     files: [],
     help: false,
     version: false,
@@ -4493,10 +4496,9 @@ Options:
 
 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)
-
+  
 Environment Variables:
-  AI_MODEL: Default model to use
+  AI_DEFAULT_MODEL: Default model to use
   AI_SYSTEM: Default system message
   AI_VERBOSE: Set to 'true' for detailed output
 
@@ -4513,7 +4515,9 @@ Unix-style piping:
   cat README.md | npx ai "Summarize this"
   curl -s https://api.github.com/repos/vercel/ai | npx ai "What is this repository about?"
   
-  The gateway supports OpenAI, Anthropic, Google, Groq, and more providers.`);
+  The gateway supports OpenAI, Anthropic, Google, Groq, and more providers.
+  
+  For detailed setup instructions, visit: https://ai-sdk.dev/docs/cli/authentication`);
 }
 function showVersion() {
   console.log("1.0.0");
@@ -4597,7 +4601,7 @@ ${prompt}` : stdinContent;
       }
       console.error("");
     }
-    const hasApiKey = process.env.AI_GATEWAY_API_KEY || process.env.VERCEL_OIDC_TOKEN;
+    const hasApiKey = process.env.AI_GATEWAY_API_KEY;
     if (!hasApiKey) {
       console.error(`Error: Authentication required.
 
@@ -4605,12 +4609,10 @@ 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"
+  export AI_DEFAULT_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.`);