ai 3.4.26 → 3.4.27

package/dist/index.mjs

@@ -1285,8 +1285,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 _a11, _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 ((_a11 = attachment.contentType) == null ? void 0 : _a11.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 AISDKError6 } from "@ai-sdk/provider";
+var name6 = "AI_MessageConversionError";
+var marker6 = `vercel.ai.error.${name6}`;
+var symbol6 = Symbol.for(marker6);
+var _a6;
+var MessageConversionError = class extends AISDKError6 {
+  constructor({
+    originalMessage,
+    message
+  }) {
+    super({ name: name6, message });
+    this[_a6] = true;
+    this.originalMessage = originalMessage;
+  }
+  static isInstance(error) {
+    return AISDKError6.hasMarker(error, marker6);
+  }
+};
+_a6 = symbol6;
+
+// core/prompt/convert-to-core-messages.ts
+function convertToCoreMessages(messages, options) {
+  var _a11;
+  const tools = (_a11 = options == null ? void 0 : options.tools) != null ? _a11 : {};
+  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 tool2 = tools[toolName];
+            return (tool2 == null ? void 0 : tool2.experimental_toToolResultContent) != null ? {
+              type: "tool-result",
+              toolCallId,
+              toolName,
+              result: tool2.experimental_toToolResultContent(result),
+              experimental_content: tool2.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,
@@ -1324,21 +1548,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
     };
   }
@@ -1391,25 +1624,25 @@ function injectJsonInstruction({
 }
 
 // core/generate-object/no-object-generated-error.ts
-import { AISDKError as AISDKError6 } from "@ai-sdk/provider";
-var name6 = "AI_NoObjectGeneratedError";
-var marker6 = `vercel.ai.error.${name6}`;
-var symbol6 = Symbol.for(marker6);
-var _a6;
-var NoObjectGeneratedError = class extends AISDKError6 {
+import { AISDKError as AISDKError7 } from "@ai-sdk/provider";
+var name7 = "AI_NoObjectGeneratedError";
+var marker7 = `vercel.ai.error.${name7}`;
+var symbol7 = Symbol.for(marker7);
+var _a7;
+var NoObjectGeneratedError = class extends AISDKError7 {
   // used in isInstance
   constructor({ message = "No object generated." } = {}) {
-    super({ name: name6, message });
-    this[_a6] = true;
+    super({ name: name7, message });
+    this[_a7] = true;
   }
   static isInstance(error) {
-    return AISDKError6.hasMarker(error, marker6);
+    return AISDKError7.hasMarker(error, marker7);
   }
   /**
    * @deprecated Use isInstance instead.
    */
   static isNoObjectGeneratedError(error) {
-    return error instanceof Error && error.name === name6;
+    return error instanceof Error && error.name === name7;
   }
   /**
    * @deprecated Do not use this method. It will be removed in the next major version.
@@ -1423,7 +1656,7 @@ var NoObjectGeneratedError = class extends AISDKError6 {
     };
   }
 };
-_a6 = symbol6;
+_a7 = symbol7;
 
 // core/generate-object/output-strategy.ts
 import {
@@ -1877,16 +2110,19 @@ async function generateObject({
       let resultProviderMetadata;
       switch (mode) {
         case "json": {
-          const standardPrompt = standardizePrompt({
-            system: outputStrategy.jsonSchema == null ? injectJsonInstruction({ prompt: system }) : model.supportsStructuredOutputs ? system : injectJsonInstruction({
-              prompt: system,
-              schema: outputStrategy.jsonSchema
-            }),
-            prompt,
-            messages
+          const standardizedPrompt = standardizePrompt({
+            prompt: {
+              system: outputStrategy.jsonSchema == null ? injectJsonInstruction({ prompt: system }) : model.supportsStructuredOutputs ? system : injectJsonInstruction({
+                prompt: system,
+                schema: outputStrategy.jsonSchema
+              }),
+              prompt,
+              messages
+            },
+            tools: void 0
           });
           const promptMessages = await convertToLanguageModelPrompt({
-            prompt: standardPrompt,
+            prompt: standardizedPrompt,
             modelSupportsImageUrls: model.supportsImageUrls,
             modelSupportsUrl: model.supportsUrl
           });
@@ -1902,7 +2138,7 @@ async function generateObject({
                   }),
                   ...baseTelemetryAttributes,
                   "ai.prompt.format": {
-                    input: () => standardPrompt.type
+                    input: () => standardizedPrompt.type
                   },
                   "ai.prompt.messages": {
                     input: () => JSON.stringify(promptMessages)
@@ -1930,7 +2166,7 @@ async function generateObject({
                     description: schemaDescription
                   },
                   ...prepareCallSettings(settings),
-                  inputFormat: standardPrompt.type,
+                  inputFormat: standardizedPrompt.type,
                   prompt: promptMessages,
                   providerMetadata,
                   abortSignal,
@@ -1983,17 +2219,16 @@ async function generateObject({
           break;
         }
         case "tool": {
-          const validatedPrompt = standardizePrompt({
-            system,
-            prompt,
-            messages
+          const standardizedPrompt = standardizePrompt({
+            prompt: { system, prompt, messages },
+            tools: void 0
           });
           const promptMessages = await convertToLanguageModelPrompt({
-            prompt: validatedPrompt,
+            prompt: standardizedPrompt,
             modelSupportsImageUrls: model.supportsImageUrls,
             modelSupportsUrl: model.supportsUrl
           });
-          const inputFormat = validatedPrompt.type;
+          const inputFormat = standardizedPrompt.type;
           const generateResult = await retry(
             () => recordSpan({
               name: "ai.generateObject.doGenerate",
@@ -2358,13 +2593,16 @@ async function streamObject({
       let transformer;
       switch (mode) {
         case "json": {
-          const standardPrompt = standardizePrompt({
-            system: outputStrategy.jsonSchema == null ? injectJsonInstruction({ prompt: system }) : model.supportsStructuredOutputs ? system : injectJsonInstruction({
-              prompt: system,
-              schema: outputStrategy.jsonSchema
-            }),
-            prompt,
-            messages
+          const standardizedPrompt = standardizePrompt({
+            prompt: {
+              system: outputStrategy.jsonSchema == null ? injectJsonInstruction({ prompt: system }) : model.supportsStructuredOutputs ? system : injectJsonInstruction({
+                prompt: system,
+                schema: outputStrategy.jsonSchema
+              }),
+              prompt,
+              messages
+            },
+            tools: void 0
           });
           callOptions = {
             mode: {
@@ -2374,9 +2612,9 @@ async function streamObject({
               description: schemaDescription
             },
             ...prepareCallSettings(settings),
-            inputFormat: standardPrompt.type,
+            inputFormat: standardizedPrompt.type,
             prompt: await convertToLanguageModelPrompt({
-              prompt: standardPrompt,
+              prompt: standardizedPrompt,
               modelSupportsImageUrls: model.supportsImageUrls,
               modelSupportsUrl: model.supportsUrl
             }),
@@ -2401,10 +2639,9 @@ async function streamObject({
           break;
         }
         case "tool": {
-          const validatedPrompt = standardizePrompt({
-            system,
-            prompt,
-            messages
+          const standardizedPrompt = standardizePrompt({
+            prompt: { system, prompt, messages },
+            tools: void 0
           });
           callOptions = {
             mode: {
@@ -2417,9 +2654,9 @@ async function streamObject({
               }
             },
             ...prepareCallSettings(settings),
-            inputFormat: validatedPrompt.type,
+            inputFormat: standardizedPrompt.type,
             prompt: await convertToLanguageModelPrompt({
-              prompt: validatedPrompt,
+              prompt: standardizedPrompt,
               modelSupportsImageUrls: model.supportsImageUrls,
               modelSupportsUrl: model.supportsUrl
             }),
@@ -2810,12 +3047,12 @@ import {
 } from "@ai-sdk/provider";
 
 // errors/invalid-tool-arguments-error.ts
-import { AISDKError as AISDKError7, getErrorMessage as getErrorMessage2 } from "@ai-sdk/provider";
-var name7 = "AI_InvalidToolArgumentsError";
-var marker7 = `vercel.ai.error.${name7}`;
-var symbol7 = Symbol.for(marker7);
-var _a7;
-var InvalidToolArgumentsError = class extends AISDKError7 {
+import { AISDKError as AISDKError8, getErrorMessage as getErrorMessage2 } from "@ai-sdk/provider";
+var name8 = "AI_InvalidToolArgumentsError";
+var marker8 = `vercel.ai.error.${name8}`;
+var symbol8 = Symbol.for(marker8);
+var _a8;
+var InvalidToolArgumentsError = class extends AISDKError8 {
   constructor({
     toolArgs,
     toolName,
@@ -2824,19 +3061,19 @@ var InvalidToolArgumentsError = class extends AISDKError7 {
       cause
     )}`
   }) {
-    super({ name: name7, message, cause });
-    this[_a7] = true;
+    super({ name: name8, message, cause });
+    this[_a8] = true;
     this.toolArgs = toolArgs;
     this.toolName = toolName;
   }
   static isInstance(error) {
-    return AISDKError7.hasMarker(error, marker7);
+    return AISDKError8.hasMarker(error, marker8);
   }
   /**
    * @deprecated use `isInstance` instead
    */
   static isInvalidToolArgumentsError(error) {
-    return error instanceof Error && error.name === name7 && typeof error.toolName === "string" && typeof error.toolArgs === "string";
+    return error instanceof Error && error.name === name8 && typeof error.toolName === "string" && typeof error.toolArgs === "string";
   }
   /**
    * @deprecated Do not use this method. It will be removed in the next major version.
@@ -2852,33 +3089,33 @@ var InvalidToolArgumentsError = class extends AISDKError7 {
     };
   }
 };
-_a7 = symbol7;
+_a8 = symbol8;
 
 // errors/no-such-tool-error.ts
-import { AISDKError as AISDKError8 } from "@ai-sdk/provider";
-var name8 = "AI_NoSuchToolError";
-var marker8 = `vercel.ai.error.${name8}`;
-var symbol8 = Symbol.for(marker8);
-var _a8;
-var NoSuchToolError = class extends AISDKError8 {
+import { AISDKError as AISDKError9 } from "@ai-sdk/provider";
+var name9 = "AI_NoSuchToolError";
+var marker9 = `vercel.ai.error.${name9}`;
+var symbol9 = Symbol.for(marker9);
+var _a9;
+var NoSuchToolError = class extends AISDKError9 {
   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: name8, message });
-    this[_a8] = true;
+    super({ name: name9, message });
+    this[_a9] = true;
     this.toolName = toolName;
     this.availableTools = availableTools;
   }
   static isInstance(error) {
-    return AISDKError8.hasMarker(error, marker8);
+    return AISDKError9.hasMarker(error, marker9);
   }
   /**
    * @deprecated use `isInstance` instead
    */
   static isNoSuchToolError(error) {
-    return error instanceof Error && error.name === name8 && "toolName" in error && error.toolName != void 0 && typeof error.name === "string";
+    return error instanceof Error && error.name === name9 && "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.
@@ -2893,27 +3130,6 @@ var NoSuchToolError = class extends AISDKError8 {
     };
   }
 };
-_a8 = symbol8;
-
-// core/prompt/message-conversion-error.ts
-import { AISDKError as AISDKError9 } from "@ai-sdk/provider";
-var name9 = "AI_MessageConversionError";
-var marker9 = `vercel.ai.error.${name9}`;
-var symbol9 = Symbol.for(marker9);
-var _a9;
-var MessageConversionError = class extends AISDKError9 {
-  constructor({
-    originalMessage,
-    message
-  }) {
-    super({ name: name9, message });
-    this[_a9] = true;
-    this.originalMessage = originalMessage;
-  }
-  static isInstance(error) {
-    return AISDKError9.hasMarker(error, marker9);
-  }
-};
 _a9 = symbol9;
 
 // core/prompt/prepare-tools-and-tool-choice.ts
@@ -3093,7 +3309,10 @@ async function generateText({
     headers,
     settings: { ...settings, maxRetries }
   });
-  const initialPrompt = standardizePrompt({ system, prompt, messages });
+  const initialPrompt = standardizePrompt({
+    prompt: { system, prompt, messages },
+    tools
+  });
   const tracer = getTracer(telemetry);
   return recordSpan({
     name: "ai.generateText",
@@ -3819,7 +4038,10 @@ async function streamText({
     settings: { ...settings, maxRetries }
   });
   const tracer = getTracer(telemetry);
-  const initialPrompt = standardizePrompt({ system, prompt, messages });
+  const initialPrompt = standardizePrompt({
+    prompt: { system, prompt, messages },
+    tools
+  });
   return recordSpan({
     name: "ai.streamText",
     attributes: selectTelemetryAttributes({
@@ -4643,171 +4865,6 @@ var experimental_wrapLanguageModel = ({
   };
 };
 
-// core/prompt/attachments-to-parts.ts
-function attachmentsToParts(attachments) {
-  var _a11, _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 ((_a11 = attachment.contentType) == null ? void 0 : _a11.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/convert-to-core-messages.ts
-function convertToCoreMessages(messages, options) {
-  var _a11;
-  const tools = (_a11 = options == null ? void 0 : options.tools) != null ? _a11 : {};
-  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 tool2 = tools[toolName];
-            return (tool2 == null ? void 0 : tool2.experimental_toToolResultContent) != null ? {
-              type: "tool-result",
-              toolCallId,
-              toolName,
-              result: tool2.experimental_toToolResultContent(result),
-              experimental_content: tool2.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/registry/custom-provider.ts
 import { NoSuchModelError as NoSuchModelError2 } from "@ai-sdk/provider";
 function experimental_customProvider({