ai 3.1.33 → 3.1.35

package/dist/index.mjs

@@ -215,7 +215,31 @@ function convertDataContentToUint8Array(content) {
   throw new InvalidDataContentError({ content });
 }
 
+// core/prompt/invalid-message-role-error.ts
+var InvalidMessageRoleError = class extends Error {
+  constructor({
+    role,
+    message = `Invalid message role: '${role}'. Must be one of: "system", "user", "assistant", "tool".`
+  }) {
+    super(message);
+    this.name = "AI_InvalidMessageRoleError";
+    this.role = role;
+  }
+  static isInvalidMessageRoleError(error) {
+    return error instanceof Error && error.name === "AI_InvalidMessageRoleError" && typeof error.role === "string";
+  }
+  toJSON() {
+    return {
+      name: this.name,
+      message: this.message,
+      stack: this.stack,
+      role: this.role
+    };
+  }
+};
+
 // core/prompt/convert-to-language-model-prompt.ts
+import { getErrorMessage as getErrorMessage2 } from "@ai-sdk/provider-utils";
 function convertToLanguageModelPrompt(prompt) {
   const languageModelMessages = [];
   if (prompt.system != null) {
@@ -244,7 +268,8 @@ function convertToLanguageModelPrompt(prompt) {
   return languageModelMessages;
 }
 function convertToLanguageModelMessage(message) {
-  switch (message.role) {
+  const role = message.role;
+  switch (role) {
     case "system": {
       return { role: "system", content: message.content };
     }
@@ -272,6 +297,47 @@ function convertToLanguageModelMessage(message) {
                     mimeType: part.mimeType
                   };
                 }
+                if (typeof part.image === "string") {
+                  try {
+                    const url = new URL(part.image);
+                    switch (url.protocol) {
+                      case "http:":
+                      case "https:": {
+                        return {
+                          type: "image",
+                          image: url,
+                          mimeType: part.mimeType
+                        };
+                      }
+                      case "data:": {
+                        try {
+                          const [header, base64Content] = part.image.split(",");
+                          const mimeType = header.split(";")[0].split(":")[1];
+                          if (mimeType == null || base64Content == null) {
+                            throw new Error("Invalid data URL format");
+                          }
+                          return {
+                            type: "image",
+                            image: convertDataContentToUint8Array(base64Content),
+                            mimeType
+                          };
+                        } catch (error) {
+                          throw new Error(
+                            `Error processing data URL: ${getErrorMessage2(
+                              message
+                            )}`
+                          );
+                        }
+                      }
+                      default: {
+                        throw new Error(
+                          `Unsupported URL protocol: ${url.protocol}`
+                        );
+                      }
+                    }
+                  } catch (_ignored) {
+                  }
+                }
                 const imageUint8 = convertDataContentToUint8Array(part.image);
                 return {
                   type: "image",
@@ -303,8 +369,8 @@ function convertToLanguageModelMessage(message) {
       return message;
     }
     default: {
-      const _exhaustiveCheck = message;
-      throw new Error(`Unsupported message role: ${_exhaustiveCheck}`);
+      const _exhaustiveCheck = role;
+      throw new InvalidMessageRoleError({ role: _exhaustiveCheck });
     }
   }
 }
@@ -2123,6 +2189,25 @@ import {
   UnsupportedJSONSchemaError
 } from "@ai-sdk/provider";
 
+// core/util/cosine-similarity.ts
+function cosineSimilarity(vector1, vector2) {
+  if (vector1.length !== vector2.length) {
+    throw new Error(
+      `Vectors must have the same length (vector1: ${vector1.length} elements, vector2: ${vector2.length} elements)`
+    );
+  }
+  return dotProduct(vector1, vector2) / (magnitude(vector1) * magnitude(vector2));
+}
+function dotProduct(vector1, vector2) {
+  return vector1.reduce(
+    (accumulator, value, index) => accumulator + value * vector2[index],
+    0
+  );
+}
+function magnitude(vector) {
+  return Math.sqrt(dotProduct(vector, vector));
+}
+
 // streams/ai-stream.ts
 import {
   createParser
@@ -3244,6 +3329,7 @@ export {
   InkeepStream,
   InvalidArgumentError2 as InvalidArgumentError,
   InvalidDataContentError2 as InvalidDataContentError,
+  InvalidMessageRoleError,
   InvalidModelIdError,
   InvalidPromptError2 as InvalidPromptError,
   InvalidResponseDataError,
@@ -3271,6 +3357,7 @@ export {
   convertDataContentToBase64String,
   convertDataContentToUint8Array,
   convertToCoreMessages,
+  cosineSimilarity,
   createCallbacksTransformer,
   createEventStreamTransformer,
   createStreamDataTransformer,