ai 3.1.32 → 3.1.33

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";
@@ -1974,6 +1974,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;
@@ -3095,6 +3219,10 @@ var StreamingTextResponse = class extends Response {
     });
   }
 };
+
+// streams/index.ts
+var generateId2 = generateIdImpl;
+var nanoid = generateIdImpl;
 export {
   AIStream,
   APICallError2 as APICallError,
@@ -3116,6 +3244,7 @@ export {
   InkeepStream,
   InvalidArgumentError2 as InvalidArgumentError,
   InvalidDataContentError2 as InvalidDataContentError,
+  InvalidModelIdError,
   InvalidPromptError2 as InvalidPromptError,
   InvalidResponseDataError,
   InvalidToolArgumentsError2 as InvalidToolArgumentsError,
@@ -3125,6 +3254,8 @@ export {
   LoadAPIKeyError,
   MistralStream,
   NoObjectGeneratedError2 as NoObjectGeneratedError,
+  NoSuchModelError,
+  NoSuchProviderError,
   NoSuchToolError3 as NoSuchToolError,
   OpenAIStream,
   ReplicateStream,
@@ -3147,6 +3278,7 @@ export {
   embedMany,
   experimental_AssistantResponse,
   experimental_StreamData,
+  experimental_createModelRegistry,
   experimental_generateObject,
   experimental_generateText,
   experimental_streamObject,
@@ -3155,7 +3287,7 @@ export {
   generateId2 as generateId,
   generateObject,
   generateText,
-  generateId3 as nanoid,
+  nanoid,
   parseComplexResponse,
   parseStreamPart,
   readDataStream,