ai 3.1.32 → 3.1.34

package/dist/index.mjs

@@ -11,7 +11,7 @@ import {
   readDataStream,
   parseComplexResponse
 } from "@ai-sdk/ui-utils";
-import { generateId as generateId2, generateId as generateId3 } from "@ai-sdk/provider-utils";
+import { generateId as generateIdImpl } from "@ai-sdk/provider-utils";
 
 // core/util/retry-with-exponential-backoff.ts
 import { APICallError, RetryError } from "@ai-sdk/provider";
@@ -215,6 +215,29 @@ 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
 function convertToLanguageModelPrompt(prompt) {
   const languageModelMessages = [];
@@ -244,7 +267,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 };
     }
@@ -303,8 +327,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 });
     }
   }
 }
@@ -1974,6 +1998,130 @@ function convertToCoreMessages(messages) {
   return coreMessages;
 }
 
+// core/registry/invalid-model-id-error.ts
+var InvalidModelIdError = class extends Error {
+  constructor({
+    id,
+    message = `Invalid model id: ${id}`
+  }) {
+    super(message);
+    this.name = "AI_InvalidModelIdError";
+    this.id = id;
+  }
+  static isInvalidModelIdError(error) {
+    return error instanceof Error && error.name === "AI_InvalidModelIdError" && typeof error.id === "string";
+  }
+  toJSON() {
+    return {
+      name: this.name,
+      message: this.message,
+      stack: this.stack,
+      id: this.id
+    };
+  }
+};
+
+// core/registry/no-such-model-error.ts
+var NoSuchModelError = class extends Error {
+  constructor({
+    modelId,
+    message = `No such model: ${modelId}`
+  }) {
+    super(message);
+    this.name = "AI_NoSuchModelError";
+    this.modelId = modelId;
+  }
+  static isNoSuchModelError(error) {
+    return error instanceof Error && error.name === "AI_NoSuchModelError" && typeof error.modelId === "string";
+  }
+  toJSON() {
+    return {
+      name: this.name,
+      message: this.message,
+      stack: this.stack,
+      modelId: this.modelId
+    };
+  }
+};
+
+// core/registry/no-such-provider-error.ts
+var NoSuchProviderError = class extends Error {
+  constructor({
+    providerId,
+    message = `No such provider: ${providerId}`
+  }) {
+    super(message);
+    this.name = "AI_NoSuchProviderError";
+    this.providerId = providerId;
+  }
+  static isNoSuchProviderError(error) {
+    return error instanceof Error && error.name === "AI_NoSuchProviderError" && typeof error.providerId === "string";
+  }
+  toJSON() {
+    return {
+      name: this.name,
+      message: this.message,
+      stack: this.stack,
+      providerId: this.providerId
+    };
+  }
+};
+
+// core/registry/model-registry.ts
+function experimental_createModelRegistry(providers) {
+  const registry = new DefaultModelRegistry();
+  for (const [id, provider] of Object.entries(providers)) {
+    registry.registerLanguageModelProvider({ id, provider });
+  }
+  return registry;
+}
+var DefaultModelRegistry = class {
+  constructor() {
+    // Mapping of provider id to provider
+    this.providers = {};
+  }
+  /**
+  Registers a language model provider with a given id.
+  
+  @param {string} id - The id of the provider.
+  @param {(id: string) => LanguageModel} provider - The provider function to register.
+     */
+  registerLanguageModelProvider({
+    id,
+    provider
+  }) {
+    this.providers[id] = provider;
+  }
+  /**
+  Returns the language model with the given id.
+  The id can either be a registered model id or use a provider prefix.
+  Provider ids are separated from the model id by a colon: `providerId:modelId`.
+  The model id is then passed to the provider function to get the model.
+  
+  @param {string} id - The id of the model to return.
+  
+  @throws {NoSuchModelError} If no model with the given id exists.
+  @throws {NoSuchProviderError} If no provider with the given id exists.
+  
+  @returns {LanguageModel} The language model associated with the id.
+     */
+  languageModel(id) {
+    if (!id.includes(":")) {
+      throw new InvalidModelIdError({ id });
+    }
+    const [providerId, modelId] = id.split(":");
+    const provider = this.providers[providerId];
+    if (!provider) {
+      throw new NoSuchProviderError({ providerId });
+    }
+    const model = provider(modelId);
+    if (!model) {
+      throw new NoSuchModelError({ modelId: id });
+    }
+    return model;
+  }
+};
+
 // core/tool/tool.ts
 function tool(tool2) {
   return tool2;
@@ -1999,6 +2147,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
@@ -3095,6 +3262,10 @@ var StreamingTextResponse = class extends Response {
     });
   }
 };
+
+// streams/index.ts
+var generateId2 = generateIdImpl;
+var nanoid = generateIdImpl;
 export {
   AIStream,
   APICallError2 as APICallError,
@@ -3116,6 +3287,8 @@ export {
   InkeepStream,
   InvalidArgumentError2 as InvalidArgumentError,
   InvalidDataContentError2 as InvalidDataContentError,
+  InvalidMessageRoleError,
+  InvalidModelIdError,
   InvalidPromptError2 as InvalidPromptError,
   InvalidResponseDataError,
   InvalidToolArgumentsError2 as InvalidToolArgumentsError,
@@ -3125,6 +3298,8 @@ export {
   LoadAPIKeyError,
   MistralStream,
   NoObjectGeneratedError2 as NoObjectGeneratedError,
+  NoSuchModelError,
+  NoSuchProviderError,
   NoSuchToolError3 as NoSuchToolError,
   OpenAIStream,
   ReplicateStream,
@@ -3140,6 +3315,7 @@ export {
   convertDataContentToBase64String,
   convertDataContentToUint8Array,
   convertToCoreMessages,
+  cosineSimilarity,
   createCallbacksTransformer,
   createEventStreamTransformer,
   createStreamDataTransformer,
@@ -3147,6 +3323,7 @@ export {
   embedMany,
   experimental_AssistantResponse,
   experimental_StreamData,
+  experimental_createModelRegistry,
   experimental_generateObject,
   experimental_generateText,
   experimental_streamObject,
@@ -3155,7 +3332,7 @@ export {
   generateId2 as generateId,
   generateObject,
   generateText,
-  generateId3 as nanoid,
+  nanoid,
   parseComplexResponse,
   parseStreamPart,
   readDataStream,