ai 3.1.35 → 3.1.36

package/dist/index.mjs

@@ -2067,21 +2067,24 @@ var InvalidModelIdError = class extends Error {
 var NoSuchModelError = class extends Error {
   constructor({
     modelId,
-    message = `No such model: ${modelId}`
+    modelType,
+    message = `No such ${modelType}: ${modelId}`
   }) {
     super(message);
     this.name = "AI_NoSuchModelError";
     this.modelId = modelId;
+    this.modelType = modelType;
   }
   static isNoSuchModelError(error) {
-    return error instanceof Error && error.name === "AI_NoSuchModelError" && typeof error.modelId === "string";
+    return error instanceof Error && error.name === "AI_NoSuchModelError" && typeof error.modelId === "string" && typeof error.modelType === "string";
   }
   toJSON() {
     return {
       name: this.name,
       message: this.message,
       stack: this.stack,
-      modelId: this.modelId
+      modelId: this.modelId,
+      modelType: this.modelType
     };
   }
 };
@@ -2090,75 +2093,78 @@ var NoSuchModelError = class extends Error {
 var NoSuchProviderError = class extends Error {
   constructor({
     providerId,
-    message = `No such provider: ${providerId}`
+    availableProviders,
+    message = `No such provider: ${providerId} (available providers: ${availableProviders.join()})`
   }) {
     super(message);
     this.name = "AI_NoSuchProviderError";
     this.providerId = providerId;
+    this.availableProviders = availableProviders;
   }
   static isNoSuchProviderError(error) {
-    return error instanceof Error && error.name === "AI_NoSuchProviderError" && typeof error.providerId === "string";
+    return error instanceof Error && error.name === "AI_NoSuchProviderError" && typeof error.providerId === "string" && Array.isArray(error.availableProviders);
   }
   toJSON() {
     return {
       name: this.name,
       message: this.message,
       stack: this.stack,
-      providerId: this.providerId
+      providerId: this.providerId,
+      availableProviders: this.availableProviders
     };
   }
 };
 
-// core/registry/model-registry.ts
-function experimental_createModelRegistry(providers) {
-  const registry = new DefaultModelRegistry();
+// core/registry/provider-registry.ts
+function experimental_createProviderRegistry(providers) {
+  const registry = new DefaultProviderRegistry();
   for (const [id, provider] of Object.entries(providers)) {
-    registry.registerLanguageModelProvider({ id, provider });
+    registry.registerProvider({ id, provider });
   }
   return registry;
 }
-var DefaultModelRegistry = class {
+var experimental_createModelRegistry = experimental_createProviderRegistry;
+var DefaultProviderRegistry = 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
-  }) {
+  registerProvider({ 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) {
+  getProvider(id) {
+    const provider = this.providers[id];
+    if (provider == null) {
+      throw new NoSuchProviderError({
+        providerId: id,
+        availableProviders: Object.keys(this.providers)
+      });
+    }
+    return provider;
+  }
+  splitId(id) {
     if (!id.includes(":")) {
       throw new InvalidModelIdError({ id });
     }
-    const [providerId, modelId] = id.split(":");
-    const provider = this.providers[providerId];
-    if (!provider) {
-      throw new NoSuchProviderError({ providerId });
+    return id.split(":");
+  }
+  languageModel(id) {
+    var _a, _b;
+    const [providerId, modelId] = this.splitId(id);
+    const model = (_b = (_a = this.getProvider(providerId)).languageModel) == null ? void 0 : _b.call(_a, modelId);
+    if (model == null) {
+      throw new NoSuchModelError({ modelId: id, modelType: "language model" });
     }
-    const model = provider(modelId);
-    if (!model) {
-      throw new NoSuchModelError({ modelId: id });
+    return model;
+  }
+  textEmbeddingModel(id) {
+    var _a, _b;
+    const [providerId, modelId] = this.splitId(id);
+    const model = (_b = (_a = this.getProvider(providerId)).textEmbedding) == null ? void 0 : _b.call(_a, modelId);
+    if (model == null) {
+      throw new NoSuchModelError({
+        modelId: id,
+        modelType: "text embedding model"
+      });
     }
     return model;
   }
@@ -3366,6 +3372,7 @@ export {
   experimental_AssistantResponse,
   experimental_StreamData,
   experimental_createModelRegistry,
+  experimental_createProviderRegistry,
   experimental_generateObject,
   experimental_generateText,
   experimental_streamObject,