ai 3.4.25 → 3.4.27

package/rsc/dist/rsc-server.mjs

@@ -81,7 +81,7 @@ function getMutableAIState(...args) {
     store.mutationDeltaResolve = resolve;
   }
   function doUpdate(newState, done) {
-    var _a8, _b;
+    var _a9, _b;
     if (args.length > 0) {
       if (typeof store.currentState !== "object") {
         const key = args[0];
@@ -105,7 +105,7 @@ function getMutableAIState(...args) {
         store.currentState = newState;
       }
     }
-    (_b = (_a8 = store.options).onSetAIState) == null ? void 0 : _b.call(_a8, {
+    (_b = (_a9 = store.options).onSetAIState) == null ? void 0 : _b.call(_a9, {
       key: args.length > 0 ? args[0] : void 0,
       state: store.currentState,
       done
@@ -172,20 +172,20 @@ function createAI({
   onGetUIState
 }) {
   const wrappedActions = {};
-  for (const name8 in actions) {
-    wrappedActions[name8] = wrapAction(actions[name8], {
+  for (const name9 in actions) {
+    wrappedActions[name9] = wrapAction(actions[name9], {
       onSetAIState
     });
   }
   const wrappedSyncUIState = onGetUIState ? wrapAction(onGetUIState, {}) : void 0;
   const AI = async (props) => {
-    var _a8, _b;
+    var _a9, _b;
     if ("useState" in React) {
       throw new Error(
         "This component can only be used inside Server Components."
       );
     }
-    let uiState = (_a8 = props.initialUIState) != null ? _a8 : initialUIState;
+    let uiState = (_a9 = props.initialUIState) != null ? _a9 : initialUIState;
     let aiState = (_b = props.initialAIState) != null ? _b : initialAIState;
     let aiStateDelta = void 0;
     if (wrappedSyncUIState) {
@@ -263,7 +263,7 @@ async function download({
   url,
   fetchImplementation = fetch
 }) {
-  var _a8;
+  var _a9;
   const urlText = url.toString();
   try {
     const response = await fetchImplementation(urlText);
@@ -276,7 +276,7 @@ async function download({
     }
     return {
       data: new Uint8Array(await response.arrayBuffer()),
-      mimeType: (_a8 = response.headers.get("content-type")) != null ? _a8 : void 0
+      mimeType: (_a9 = response.headers.get("content-type")) != null ? _a9 : void 0
     };
   } catch (error) {
     if (DownloadError.isInstance(error)) {
@@ -357,8 +357,8 @@ var dataContentSchema = z.union([
   z.custom(
     // Buffer might not be available in some environments such as CloudFlare:
     (value) => {
-      var _a8, _b;
-      return (_b = (_a8 = globalThis.Buffer) == null ? void 0 : _a8.isBuffer(value)) != null ? _b : false;
+      var _a9, _b;
+      return (_b = (_a9 = globalThis.Buffer) == null ? void 0 : _a9.isBuffer(value)) != null ? _b : false;
     },
     { message: "Must be a Buffer" }
   )
@@ -392,6 +392,13 @@ function convertDataContentToUint8Array(content) {
   }
   throw new InvalidDataContentError({ content });
 }
+function convertUint8ArrayToText(uint8Array) {
+  try {
+    return new TextDecoder().decode(uint8Array);
+  } catch (error) {
+    throw new Error("Error decoding Uint8Array to text");
+  }
+}
 
 // core/prompt/invalid-message-role-error.ts
 import { AISDKError as AISDKError3 } from "@ai-sdk/provider";
@@ -805,24 +812,24 @@ function prepareToolsAndToolChoice({
     };
   }
   const filteredTools = activeTools != null ? Object.entries(tools).filter(
-    ([name8]) => activeTools.includes(name8)
+    ([name9]) => activeTools.includes(name9)
   ) : Object.entries(tools);
   return {
-    tools: filteredTools.map(([name8, tool]) => {
+    tools: filteredTools.map(([name9, tool]) => {
       const toolType = tool.type;
       switch (toolType) {
         case void 0:
         case "function":
           return {
             type: "function",
-            name: name8,
+            name: name9,
             description: tool.description,
             parameters: asSchema(tool.parameters).jsonSchema
           };
         case "provider-defined":
           return {
             type: "provider-defined",
-            name: name8,
+            name: name9,
             id: tool.id,
             args: tool.args
           };
@@ -950,8 +957,232 @@ var coreMessageSchema = z6.union([
   coreToolMessageSchema
 ]);
 
+// core/prompt/detect-prompt-type.ts
+function detectPromptType(prompt) {
+  if (!Array.isArray(prompt)) {
+    return "other";
+  }
+  if (prompt.length === 0) {
+    return "messages";
+  }
+  const characteristics = prompt.map(detectSingleMessageCharacteristics);
+  if (characteristics.some((c) => c === "has-ui-specific-parts")) {
+    return "ui-messages";
+  } else if (characteristics.every(
+    (c) => c === "has-core-specific-parts" || c === "message"
+  )) {
+    return "messages";
+  } else {
+    return "other";
+  }
+}
+function detectSingleMessageCharacteristics(message) {
+  if (typeof message === "object" && message !== null && (message.role === "function" || // UI-only role
+  message.role === "data" || // UI-only role
+  "toolInvocations" in message || // UI-specific field
+  "experimental_attachments" in message)) {
+    return "has-ui-specific-parts";
+  } else if (typeof message === "object" && message !== null && "content" in message && (Array.isArray(message.content) || // Core messages can have array content
+  "experimental_providerMetadata" in message)) {
+    return "has-core-specific-parts";
+  } else if (typeof message === "object" && message !== null && "role" in message && "content" in message && typeof message.content === "string" && ["system", "user", "assistant", "tool"].includes(message.role)) {
+    return "message";
+  } else {
+    return "other";
+  }
+}
+
+// core/prompt/attachments-to-parts.ts
+function attachmentsToParts(attachments) {
+  var _a9, _b, _c;
+  const parts = [];
+  for (const attachment of attachments) {
+    let url;
+    try {
+      url = new URL(attachment.url);
+    } catch (error) {
+      throw new Error(`Invalid URL: ${attachment.url}`);
+    }
+    switch (url.protocol) {
+      case "http:":
+      case "https:": {
+        if ((_a9 = attachment.contentType) == null ? void 0 : _a9.startsWith("image/")) {
+          parts.push({ type: "image", image: url });
+        } else {
+          if (!attachment.contentType) {
+            throw new Error(
+              "If the attachment is not an image, it must specify a content type"
+            );
+          }
+          parts.push({
+            type: "file",
+            data: url,
+            mimeType: attachment.contentType
+          });
+        }
+        break;
+      }
+      case "data:": {
+        let header;
+        let base64Content;
+        let mimeType;
+        try {
+          [header, base64Content] = attachment.url.split(",");
+          mimeType = header.split(";")[0].split(":")[1];
+        } catch (error) {
+          throw new Error(`Error processing data URL: ${attachment.url}`);
+        }
+        if (mimeType == null || base64Content == null) {
+          throw new Error(`Invalid data URL format: ${attachment.url}`);
+        }
+        if ((_b = attachment.contentType) == null ? void 0 : _b.startsWith("image/")) {
+          parts.push({
+            type: "image",
+            image: convertDataContentToUint8Array(base64Content)
+          });
+        } else if ((_c = attachment.contentType) == null ? void 0 : _c.startsWith("text/")) {
+          parts.push({
+            type: "text",
+            text: convertUint8ArrayToText(
+              convertDataContentToUint8Array(base64Content)
+            )
+          });
+        } else {
+          if (!attachment.contentType) {
+            throw new Error(
+              "If the attachment is not an image or text, it must specify a content type"
+            );
+          }
+          parts.push({
+            type: "file",
+            data: base64Content,
+            mimeType: attachment.contentType
+          });
+        }
+        break;
+      }
+      default: {
+        throw new Error(`Unsupported URL protocol: ${url.protocol}`);
+      }
+    }
+  }
+  return parts;
+}
+
+// core/prompt/message-conversion-error.ts
+import { AISDKError as AISDKError5 } from "@ai-sdk/provider";
+var name5 = "AI_MessageConversionError";
+var marker5 = `vercel.ai.error.${name5}`;
+var symbol5 = Symbol.for(marker5);
+var _a5;
+var MessageConversionError = class extends AISDKError5 {
+  constructor({
+    originalMessage,
+    message
+  }) {
+    super({ name: name5, message });
+    this[_a5] = true;
+    this.originalMessage = originalMessage;
+  }
+  static isInstance(error) {
+    return AISDKError5.hasMarker(error, marker5);
+  }
+};
+_a5 = symbol5;
+
+// core/prompt/convert-to-core-messages.ts
+function convertToCoreMessages(messages, options) {
+  var _a9;
+  const tools = (_a9 = options == null ? void 0 : options.tools) != null ? _a9 : {};
+  const coreMessages = [];
+  for (const message of messages) {
+    const { role, content, toolInvocations, experimental_attachments } = message;
+    switch (role) {
+      case "system": {
+        coreMessages.push({
+          role: "system",
+          content
+        });
+        break;
+      }
+      case "user": {
+        coreMessages.push({
+          role: "user",
+          content: experimental_attachments ? [
+            { type: "text", text: content },
+            ...attachmentsToParts(experimental_attachments)
+          ] : content
+        });
+        break;
+      }
+      case "assistant": {
+        if (toolInvocations == null) {
+          coreMessages.push({ role: "assistant", content });
+          break;
+        }
+        coreMessages.push({
+          role: "assistant",
+          content: [
+            { type: "text", text: content },
+            ...toolInvocations.map(
+              ({ toolCallId, toolName, args }) => ({
+                type: "tool-call",
+                toolCallId,
+                toolName,
+                args
+              })
+            )
+          ]
+        });
+        coreMessages.push({
+          role: "tool",
+          content: toolInvocations.map((toolInvocation) => {
+            if (!("result" in toolInvocation)) {
+              throw new MessageConversionError({
+                originalMessage: message,
+                message: "ToolInvocation must have a result: " + JSON.stringify(toolInvocation)
+              });
+            }
+            const { toolCallId, toolName, result } = toolInvocation;
+            const tool = tools[toolName];
+            return (tool == null ? void 0 : tool.experimental_toToolResultContent) != null ? {
+              type: "tool-result",
+              toolCallId,
+              toolName,
+              result: tool.experimental_toToolResultContent(result),
+              experimental_content: tool.experimental_toToolResultContent(result)
+            } : {
+              type: "tool-result",
+              toolCallId,
+              toolName,
+              result
+            };
+          })
+        });
+        break;
+      }
+      case "function":
+      case "data":
+      case "tool": {
+        break;
+      }
+      default: {
+        const _exhaustiveCheck = role;
+        throw new MessageConversionError({
+          originalMessage: message,
+          message: `Unsupported role: ${_exhaustiveCheck}`
+        });
+      }
+    }
+  }
+  return coreMessages;
+}
+
 // core/prompt/standardize-prompt.ts
-function standardizePrompt(prompt) {
+function standardizePrompt({
+  prompt,
+  tools
+}) {
   if (prompt.prompt == null && prompt.messages == null) {
     throw new InvalidPromptError({
       prompt,
@@ -989,21 +1220,30 @@ function standardizePrompt(prompt) {
     };
   }
   if (prompt.messages != null) {
+    const promptType = detectPromptType(prompt.messages);
+    if (promptType === "other") {
+      throw new InvalidPromptError({
+        prompt,
+        message: "messages must be an array of CoreMessage or UIMessage"
+      });
+    }
+    const messages = promptType === "ui-messages" ? convertToCoreMessages(prompt.messages, {
+      tools
+    }) : prompt.messages;
     const validationResult = safeValidateTypes({
-      value: prompt.messages,
+      value: messages,
       schema: z7.array(coreMessageSchema)
     });
     if (!validationResult.success) {
       throw new InvalidPromptError({
         prompt,
-        message: "messages must be an array of CoreMessage",
+        message: "messages must be an array of CoreMessage or UIMessage",
         cause: validationResult.error
       });
     }
     return {
       type: "messages",
-      messages: prompt.messages,
-      // only possible case bc of checks above
+      messages,
       system: prompt.system
     };
   }
@@ -1020,12 +1260,12 @@ function calculateLanguageModelUsage(usage) {
 }
 
 // errors/invalid-tool-arguments-error.ts
-import { AISDKError as AISDKError5, getErrorMessage } from "@ai-sdk/provider";
-var name5 = "AI_InvalidToolArgumentsError";
-var marker5 = `vercel.ai.error.${name5}`;
-var symbol5 = Symbol.for(marker5);
-var _a5;
-var InvalidToolArgumentsError = class extends AISDKError5 {
+import { AISDKError as AISDKError6, getErrorMessage } from "@ai-sdk/provider";
+var name6 = "AI_InvalidToolArgumentsError";
+var marker6 = `vercel.ai.error.${name6}`;
+var symbol6 = Symbol.for(marker6);
+var _a6;
+var InvalidToolArgumentsError = class extends AISDKError6 {
   constructor({
     toolArgs,
     toolName,
@@ -1034,19 +1274,19 @@ var InvalidToolArgumentsError = class extends AISDKError5 {
       cause
     )}`
   }) {
-    super({ name: name5, message, cause });
-    this[_a5] = true;
+    super({ name: name6, message, cause });
+    this[_a6] = true;
     this.toolArgs = toolArgs;
     this.toolName = toolName;
   }
   static isInstance(error) {
-    return AISDKError5.hasMarker(error, marker5);
+    return AISDKError6.hasMarker(error, marker6);
   }
   /**
    * @deprecated use `isInstance` instead
    */
   static isInvalidToolArgumentsError(error) {
-    return error instanceof Error && error.name === name5 && typeof error.toolName === "string" && typeof error.toolArgs === "string";
+    return error instanceof Error && error.name === name6 && typeof error.toolName === "string" && typeof error.toolArgs === "string";
   }
   /**
    * @deprecated Do not use this method. It will be removed in the next major version.
@@ -1062,33 +1302,33 @@ var InvalidToolArgumentsError = class extends AISDKError5 {
     };
   }
 };
-_a5 = symbol5;
+_a6 = symbol6;
 
 // errors/no-such-tool-error.ts
-import { AISDKError as AISDKError6 } from "@ai-sdk/provider";
-var name6 = "AI_NoSuchToolError";
-var marker6 = `vercel.ai.error.${name6}`;
-var symbol6 = Symbol.for(marker6);
-var _a6;
-var NoSuchToolError = class extends AISDKError6 {
+import { AISDKError as AISDKError7 } from "@ai-sdk/provider";
+var name7 = "AI_NoSuchToolError";
+var marker7 = `vercel.ai.error.${name7}`;
+var symbol7 = Symbol.for(marker7);
+var _a7;
+var NoSuchToolError = class extends AISDKError7 {
   constructor({
     toolName,
     availableTools = void 0,
     message = `Model tried to call unavailable tool '${toolName}'. ${availableTools === void 0 ? "No tools are available." : `Available tools: ${availableTools.join(", ")}.`}`
   }) {
-    super({ name: name6, message });
-    this[_a6] = true;
+    super({ name: name7, message });
+    this[_a7] = true;
     this.toolName = toolName;
     this.availableTools = availableTools;
   }
   static isInstance(error) {
-    return AISDKError6.hasMarker(error, marker6);
+    return AISDKError7.hasMarker(error, marker7);
   }
   /**
    * @deprecated use `isInstance` instead
    */
   static isNoSuchToolError(error) {
-    return error instanceof Error && error.name === name6 && "toolName" in error && error.toolName != void 0 && typeof error.name === "string";
+    return error instanceof Error && error.name === name7 && "toolName" in error && error.toolName != void 0 && typeof error.name === "string";
   }
   /**
    * @deprecated Do not use this method. It will be removed in the next major version.
@@ -1103,7 +1343,7 @@ var NoSuchToolError = class extends AISDKError6 {
     };
   }
 };
-_a6 = symbol6;
+_a7 = symbol7;
 
 // util/is-async-generator.ts
 function isAsyncGenerator(value) {
@@ -1125,31 +1365,31 @@ async function delay(delayInMs) {
 }
 
 // util/retry-error.ts
-import { AISDKError as AISDKError7 } from "@ai-sdk/provider";
-var name7 = "AI_RetryError";
-var marker7 = `vercel.ai.error.${name7}`;
-var symbol7 = Symbol.for(marker7);
-var _a7;
-var RetryError = class extends AISDKError7 {
+import { AISDKError as AISDKError8 } from "@ai-sdk/provider";
+var name8 = "AI_RetryError";
+var marker8 = `vercel.ai.error.${name8}`;
+var symbol8 = Symbol.for(marker8);
+var _a8;
+var RetryError = class extends AISDKError8 {
   constructor({
     message,
     reason,
     errors
   }) {
-    super({ name: name7, message });
-    this[_a7] = true;
+    super({ name: name8, message });
+    this[_a8] = true;
     this.reason = reason;
     this.errors = errors;
     this.lastError = errors[errors.length - 1];
   }
   static isInstance(error) {
-    return AISDKError7.hasMarker(error, marker7);
+    return AISDKError8.hasMarker(error, marker8);
   }
   /**
    * @deprecated use `isInstance` instead
    */
   static isRetryError(error) {
-    return error instanceof Error && error.name === name7 && typeof error.reason === "string" && Array.isArray(error.errors);
+    return error instanceof Error && error.name === name8 && typeof error.reason === "string" && Array.isArray(error.errors);
   }
   /**
    * @deprecated Do not use this method. It will be removed in the next major version.
@@ -1164,7 +1404,7 @@ var RetryError = class extends AISDKError7 {
     };
   }
 };
-_a7 = symbol7;
+_a8 = symbol8;
 
 // util/retry-with-exponential-backoff.ts
 var retryWithExponentialBackoff = ({
@@ -1362,10 +1602,10 @@ async function streamUI({
     );
   }
   if (tools) {
-    for (const [name8, tool] of Object.entries(tools)) {
+    for (const [name9, tool] of Object.entries(tools)) {
       if ("render" in tool) {
         throw new Error(
-          "Tool definition in `streamUI` should not have `render` property. Use `generate` instead. Found in tool: " + name8
+          "Tool definition in `streamUI` should not have `render` property. Use `generate` instead. Found in tool: " + name9
         );
       }
     }
@@ -1408,7 +1648,11 @@ async function streamUI({
     renderFinished.resolve(void 0);
   }
   const retry = retryWithExponentialBackoff({ maxRetries });
-  const validatedPrompt = standardizePrompt({ system, prompt, messages });
+  const validatedPrompt = standardizePrompt({
+    prompt: { system, prompt, messages },
+    tools: void 0
+    // streamUI tools don't support multi-modal tool result conversion
+  });
   const result = await retry(
     async () => model.doStream({
       mode: {
@@ -1654,8 +1898,8 @@ function readableFromAsyncIterable(iterable) {
         controller.enqueue(value);
     },
     async cancel(reason) {
-      var _a8;
-      await ((_a8 = it.return) == null ? void 0 : _a8.call(it, reason));
+      var _a9;
+      await ((_a9 = it.return) == null ? void 0 : _a9.call(it, reason));
     }
   });
 }
@@ -1694,10 +1938,10 @@ async function* streamable(stream) {
         model: chunk.model,
         // not exposed by Azure API
         choices: chunk.choices.map((choice) => {
-          var _a8, _b, _c, _d, _e, _f, _g;
+          var _a9, _b, _c, _d, _e, _f, _g;
           return {
             delta: {
-              content: (_a8 = choice.delta) == null ? void 0 : _a8.content,
+              content: (_a9 = choice.delta) == null ? void 0 : _a9.content,
               function_call: (_b = choice.delta) == null ? void 0 : _b.functionCall,
               role: (_c = choice.delta) == null ? void 0 : _c.role,
               tool_calls: ((_e = (_d = choice.delta) == null ? void 0 : _d.toolCalls) == null ? void 0 : _e.length) ? (_g = (_f = choice.delta) == null ? void 0 : _f.toolCalls) == null ? void 0 : _g.map((toolCall, index) => ({
@@ -1722,9 +1966,9 @@ function chunkToText() {
   const trimStartOfStream = trimStartOfStreamHelper();
   let isFunctionStreamingIn;
   return (json) => {
-    var _a8, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p, _q, _r;
+    var _a9, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p, _q, _r;
     if (isChatCompletionChunk(json)) {
-      const delta = (_a8 = json.choices[0]) == null ? void 0 : _a8.delta;
+      const delta = (_a9 = json.choices[0]) == null ? void 0 : _a9.delta;
       if ((_b = delta.function_call) == null ? void 0 : _b.name) {
         isFunctionStreamingIn = true;
         return {
@@ -2010,20 +2254,20 @@ function render(options) {
   const ui = createStreamableUI(options.initial);
   const text = options.text ? options.text : ({ content }) => content;
   const functions = options.functions ? Object.entries(options.functions).map(
-    ([name8, { description, parameters }]) => {
+    ([name9, { description, parameters }]) => {
       return {
-        name: name8,
+        name: name9,
         description,
         parameters: zodToJsonSchema(parameters)
       };
     }
   ) : void 0;
   const tools = options.tools ? Object.entries(options.tools).map(
-    ([name8, { description, parameters }]) => {
+    ([name9, { description, parameters }]) => {
       return {
         type: "function",
         function: {
-          name: name8,
+          name: name9,
           description,
           parameters: zodToJsonSchema(parameters)
         }
@@ -2093,23 +2337,23 @@ function render(options) {
         {
           ...functions ? {
             async experimental_onFunctionCall(functionCallPayload) {
-              var _a8, _b;
+              var _a9, _b;
               hasFunction = true;
               handleRender(
                 functionCallPayload.arguments,
-                (_b = (_a8 = options.functions) == null ? void 0 : _a8[functionCallPayload.name]) == null ? void 0 : _b.render,
+                (_b = (_a9 = options.functions) == null ? void 0 : _a9[functionCallPayload.name]) == null ? void 0 : _b.render,
                 ui
               );
             }
           } : {},
           ...tools ? {
             async experimental_onToolCall(toolCallPayload) {
-              var _a8, _b;
+              var _a9, _b;
               hasFunction = true;
               for (const tool of toolCallPayload.tools) {
                 handleRender(
                   tool.func.arguments,
-                  (_b = (_a8 = options.tools) == null ? void 0 : _a8[tool.func.name]) == null ? void 0 : _b.render,
+                  (_b = (_a9 = options.tools) == null ? void 0 : _a9[tool.func.name]) == null ? void 0 : _b.render,
                   ui
                 );
               }