ai 3.1.35 → 3.1.36

package/dist/index.d.mts

@@ -1234,32 +1234,27 @@ declare class StreamTextResult<TOOLS extends Record<string, CoreTool>> {
  */
 declare const experimental_streamText: typeof streamText;
 
-/**
-Registry for managing models. It enables getting a model with a string id.
- */
-type experimental_ModelRegistry = {
-    /**
-  Returns the language model with the given id in the format `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: string): LanguageModel;
-};
-/**
- * Creates a model registry for the given providers.
- */
-declare function experimental_createModelRegistry(providers: Record<string, (id: string) => LanguageModel>): experimental_ModelRegistry;
+declare class InvalidModelIdError extends Error {
+    readonly id: string;
+    constructor({ id, message, }: {
+        id: string;
+        message?: string;
+    });
+    static isInvalidModelIdError(error: unknown): error is InvalidModelIdError;
+    toJSON(): {
+        name: string;
+        message: string;
+        stack: string | undefined;
+        id: string;
+    };
+}
 
 declare class NoSuchModelError extends Error {
     readonly modelId: string;
-    constructor({ modelId, message, }: {
+    readonly modelType: string;
+    constructor({ modelId, modelType, message, }: {
         modelId: string;
+        modelType: string;
         message?: string;
     });
     static isNoSuchModelError(error: unknown): error is NoSuchModelError;
@@ -1268,13 +1263,16 @@ declare class NoSuchModelError extends Error {
         message: string;
         stack: string | undefined;
         modelId: string;
+        modelType: string;
     };
 }
 
 declare class NoSuchProviderError extends Error {
     readonly providerId: string;
-    constructor({ providerId, message, }: {
+    readonly availableProviders: string[];
+    constructor({ providerId, availableProviders, message, }: {
         providerId: string;
+        availableProviders: string[];
         message?: string;
     });
     static isNoSuchProviderError(error: unknown): error is NoSuchProviderError;
@@ -1283,23 +1281,65 @@ declare class NoSuchProviderError extends Error {
         message: string;
         stack: string | undefined;
         providerId: string;
+        availableProviders: string[];
     };
 }
 
-declare class InvalidModelIdError extends Error {
-    readonly id: string;
-    constructor({ id, message, }: {
-        id: string;
-        message?: string;
-    });
-    static isInvalidModelIdError(error: unknown): error is InvalidModelIdError;
-    toJSON(): {
-        name: string;
-        message: string;
-        stack: string | undefined;
-        id: string;
-    };
+/**
+Registry for managing models. It enables getting a model with a string id.
+ */
+type experimental_ProviderRegistry = {
+    /**
+  Returns the language model with the given id in the format `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: string): LanguageModel;
+    /**
+  Returns the text embedding model with the given id in the format `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.
+     */
+    textEmbeddingModel(id: string): EmbeddingModel<string>;
+};
+/**
+ * @deprecated Use `experimental_ProviderRegistry` instead.
+ */
+type experimental_ModelRegistry = experimental_ProviderRegistry;
+/**
+ * Provider for language and text embedding models. Compatible with the
+ * provider registry.
+ */
+interface Provider {
+    /**
+     * Returns a language model with the given id.
+     */
+    languageModel?: (modelId: string) => LanguageModel;
+    /**
+     * Returns a text embedding model with the given id.
+     */
+    textEmbedding?: (modelId: string) => EmbeddingModel<string>;
 }
+/**
+ * Creates a registry for the given providers.
+ */
+declare function experimental_createProviderRegistry(providers: Record<string, Provider>): experimental_ProviderRegistry;
+/**
+ * @deprecated Use `experimental_createProviderRegistry` instead.
+ */
+declare const experimental_createModelRegistry: typeof experimental_createProviderRegistry;
 
 /**
  * Calculates the cosine similarity between two vectors. This is a useful metric for
@@ -2041,4 +2081,4 @@ declare const generateId: (size?: number | undefined) => string;
  */
 declare const nanoid: (size?: number | undefined) => string;
 
-export { AIStream, AIStreamCallbacksAndOptions, AIStreamParser, AIStreamParserOptions, AWSBedrockAnthropicMessagesStream, AWSBedrockAnthropicStream, AWSBedrockCohereStream, AWSBedrockLlama2Stream, AWSBedrockStream, AnthropicStream, AssistantContent, AssistantResponse, CallWarning, CohereStream, CompletionUsage, CoreAssistantMessage, CoreMessage, CoreSystemMessage, CoreTool, CoreToolChoice, CoreToolMessage, CoreUserMessage, DataContent, DeepPartial, EmbedManyResult, EmbedResult, Embedding, EmbeddingModel, ExperimentalAssistantMessage, ExperimentalMessage, ExperimentalTool, ExperimentalToolMessage, ExperimentalUserMessage, FinishReason, FunctionCallPayload, GenerateObjectResult, GenerateTextResult, GoogleGenerativeAIStream, HuggingFaceStream, ImagePart, InkeepAIStreamCallbacksAndOptions, InkeepChatResultCallbacks, InkeepOnFinalMetadata, InkeepStream, InvalidMessageRoleError, InvalidModelIdError, langchainAdapter as LangChainAdapter, LangChainStream, LanguageModel, LogProbs, MistralStream, NoSuchModelError, NoSuchProviderError, ObjectStreamInputPart, ObjectStreamPart, OpenAIStream, OpenAIStreamCallbacks, ReplicateStream, StreamData, StreamObjectResult, StreamTextResult, StreamingTextResponse, TextPart$1 as TextPart, TextStreamPart, TokenUsage, ToolCallPart, ToolCallPayload, ToolContent, ToolResultPart, UserContent, convertDataContentToBase64String, convertDataContentToUint8Array, convertToCoreMessages, cosineSimilarity, createCallbacksTransformer, createEventStreamTransformer, createStreamDataTransformer, embed, embedMany, experimental_AssistantResponse, experimental_ModelRegistry, experimental_StreamData, experimental_createModelRegistry, experimental_generateObject, experimental_generateText, experimental_streamObject, experimental_streamText, generateId, generateObject, generateText, nanoid, readableFromAsyncIterable, streamObject, streamText, streamToResponse, tool, trimStartOfStreamHelper };
+export { AIStream, AIStreamCallbacksAndOptions, AIStreamParser, AIStreamParserOptions, AWSBedrockAnthropicMessagesStream, AWSBedrockAnthropicStream, AWSBedrockCohereStream, AWSBedrockLlama2Stream, AWSBedrockStream, AnthropicStream, AssistantContent, AssistantResponse, CallWarning, CohereStream, CompletionUsage, CoreAssistantMessage, CoreMessage, CoreSystemMessage, CoreTool, CoreToolChoice, CoreToolMessage, CoreUserMessage, DataContent, DeepPartial, EmbedManyResult, EmbedResult, Embedding, EmbeddingModel, ExperimentalAssistantMessage, ExperimentalMessage, ExperimentalTool, ExperimentalToolMessage, ExperimentalUserMessage, FinishReason, FunctionCallPayload, GenerateObjectResult, GenerateTextResult, GoogleGenerativeAIStream, HuggingFaceStream, ImagePart, InkeepAIStreamCallbacksAndOptions, InkeepChatResultCallbacks, InkeepOnFinalMetadata, InkeepStream, InvalidMessageRoleError, InvalidModelIdError, langchainAdapter as LangChainAdapter, LangChainStream, LanguageModel, LogProbs, MistralStream, NoSuchModelError, NoSuchProviderError, ObjectStreamInputPart, ObjectStreamPart, OpenAIStream, OpenAIStreamCallbacks, ReplicateStream, StreamData, StreamObjectResult, StreamTextResult, StreamingTextResponse, TextPart$1 as TextPart, TextStreamPart, TokenUsage, ToolCallPart, ToolCallPayload, ToolContent, ToolResultPart, UserContent, convertDataContentToBase64String, convertDataContentToUint8Array, convertToCoreMessages, cosineSimilarity, createCallbacksTransformer, createEventStreamTransformer, createStreamDataTransformer, embed, embedMany, experimental_AssistantResponse, experimental_ModelRegistry, experimental_ProviderRegistry, experimental_StreamData, experimental_createModelRegistry, experimental_createProviderRegistry, experimental_generateObject, experimental_generateText, experimental_streamObject, experimental_streamText, generateId, generateObject, generateText, nanoid, readableFromAsyncIterable, streamObject, streamText, streamToResponse, tool, trimStartOfStreamHelper };

package/dist/index.d.ts

@@ -1234,32 +1234,27 @@ declare class StreamTextResult<TOOLS extends Record<string, CoreTool>> {
  */
 declare const experimental_streamText: typeof streamText;
 
-/**
-Registry for managing models. It enables getting a model with a string id.
- */
-type experimental_ModelRegistry = {
-    /**
-  Returns the language model with the given id in the format `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: string): LanguageModel;
-};
-/**
- * Creates a model registry for the given providers.
- */
-declare function experimental_createModelRegistry(providers: Record<string, (id: string) => LanguageModel>): experimental_ModelRegistry;
+declare class InvalidModelIdError extends Error {
+    readonly id: string;
+    constructor({ id, message, }: {
+        id: string;
+        message?: string;
+    });
+    static isInvalidModelIdError(error: unknown): error is InvalidModelIdError;
+    toJSON(): {
+        name: string;
+        message: string;
+        stack: string | undefined;
+        id: string;
+    };
+}
 
 declare class NoSuchModelError extends Error {
     readonly modelId: string;
-    constructor({ modelId, message, }: {
+    readonly modelType: string;
+    constructor({ modelId, modelType, message, }: {
         modelId: string;
+        modelType: string;
         message?: string;
     });
     static isNoSuchModelError(error: unknown): error is NoSuchModelError;
@@ -1268,13 +1263,16 @@ declare class NoSuchModelError extends Error {
         message: string;
         stack: string | undefined;
         modelId: string;
+        modelType: string;
     };
 }
 
 declare class NoSuchProviderError extends Error {
     readonly providerId: string;
-    constructor({ providerId, message, }: {
+    readonly availableProviders: string[];
+    constructor({ providerId, availableProviders, message, }: {
         providerId: string;
+        availableProviders: string[];
         message?: string;
     });
     static isNoSuchProviderError(error: unknown): error is NoSuchProviderError;
@@ -1283,23 +1281,65 @@ declare class NoSuchProviderError extends Error {
         message: string;
         stack: string | undefined;
         providerId: string;
+        availableProviders: string[];
     };
 }
 
-declare class InvalidModelIdError extends Error {
-    readonly id: string;
-    constructor({ id, message, }: {
-        id: string;
-        message?: string;
-    });
-    static isInvalidModelIdError(error: unknown): error is InvalidModelIdError;
-    toJSON(): {
-        name: string;
-        message: string;
-        stack: string | undefined;
-        id: string;
-    };
+/**
+Registry for managing models. It enables getting a model with a string id.
+ */
+type experimental_ProviderRegistry = {
+    /**
+  Returns the language model with the given id in the format `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: string): LanguageModel;
+    /**
+  Returns the text embedding model with the given id in the format `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.
+     */
+    textEmbeddingModel(id: string): EmbeddingModel<string>;
+};
+/**
+ * @deprecated Use `experimental_ProviderRegistry` instead.
+ */
+type experimental_ModelRegistry = experimental_ProviderRegistry;
+/**
+ * Provider for language and text embedding models. Compatible with the
+ * provider registry.
+ */
+interface Provider {
+    /**
+     * Returns a language model with the given id.
+     */
+    languageModel?: (modelId: string) => LanguageModel;
+    /**
+     * Returns a text embedding model with the given id.
+     */
+    textEmbedding?: (modelId: string) => EmbeddingModel<string>;
 }
+/**
+ * Creates a registry for the given providers.
+ */
+declare function experimental_createProviderRegistry(providers: Record<string, Provider>): experimental_ProviderRegistry;
+/**
+ * @deprecated Use `experimental_createProviderRegistry` instead.
+ */
+declare const experimental_createModelRegistry: typeof experimental_createProviderRegistry;
 
 /**
  * Calculates the cosine similarity between two vectors. This is a useful metric for
@@ -2041,4 +2081,4 @@ declare const generateId: (size?: number | undefined) => string;
  */
 declare const nanoid: (size?: number | undefined) => string;
 
-export { AIStream, AIStreamCallbacksAndOptions, AIStreamParser, AIStreamParserOptions, AWSBedrockAnthropicMessagesStream, AWSBedrockAnthropicStream, AWSBedrockCohereStream, AWSBedrockLlama2Stream, AWSBedrockStream, AnthropicStream, AssistantContent, AssistantResponse, CallWarning, CohereStream, CompletionUsage, CoreAssistantMessage, CoreMessage, CoreSystemMessage, CoreTool, CoreToolChoice, CoreToolMessage, CoreUserMessage, DataContent, DeepPartial, EmbedManyResult, EmbedResult, Embedding, EmbeddingModel, ExperimentalAssistantMessage, ExperimentalMessage, ExperimentalTool, ExperimentalToolMessage, ExperimentalUserMessage, FinishReason, FunctionCallPayload, GenerateObjectResult, GenerateTextResult, GoogleGenerativeAIStream, HuggingFaceStream, ImagePart, InkeepAIStreamCallbacksAndOptions, InkeepChatResultCallbacks, InkeepOnFinalMetadata, InkeepStream, InvalidMessageRoleError, InvalidModelIdError, langchainAdapter as LangChainAdapter, LangChainStream, LanguageModel, LogProbs, MistralStream, NoSuchModelError, NoSuchProviderError, ObjectStreamInputPart, ObjectStreamPart, OpenAIStream, OpenAIStreamCallbacks, ReplicateStream, StreamData, StreamObjectResult, StreamTextResult, StreamingTextResponse, TextPart$1 as TextPart, TextStreamPart, TokenUsage, ToolCallPart, ToolCallPayload, ToolContent, ToolResultPart, UserContent, convertDataContentToBase64String, convertDataContentToUint8Array, convertToCoreMessages, cosineSimilarity, createCallbacksTransformer, createEventStreamTransformer, createStreamDataTransformer, embed, embedMany, experimental_AssistantResponse, experimental_ModelRegistry, experimental_StreamData, experimental_createModelRegistry, experimental_generateObject, experimental_generateText, experimental_streamObject, experimental_streamText, generateId, generateObject, generateText, nanoid, readableFromAsyncIterable, streamObject, streamText, streamToResponse, tool, trimStartOfStreamHelper };
+export { AIStream, AIStreamCallbacksAndOptions, AIStreamParser, AIStreamParserOptions, AWSBedrockAnthropicMessagesStream, AWSBedrockAnthropicStream, AWSBedrockCohereStream, AWSBedrockLlama2Stream, AWSBedrockStream, AnthropicStream, AssistantContent, AssistantResponse, CallWarning, CohereStream, CompletionUsage, CoreAssistantMessage, CoreMessage, CoreSystemMessage, CoreTool, CoreToolChoice, CoreToolMessage, CoreUserMessage, DataContent, DeepPartial, EmbedManyResult, EmbedResult, Embedding, EmbeddingModel, ExperimentalAssistantMessage, ExperimentalMessage, ExperimentalTool, ExperimentalToolMessage, ExperimentalUserMessage, FinishReason, FunctionCallPayload, GenerateObjectResult, GenerateTextResult, GoogleGenerativeAIStream, HuggingFaceStream, ImagePart, InkeepAIStreamCallbacksAndOptions, InkeepChatResultCallbacks, InkeepOnFinalMetadata, InkeepStream, InvalidMessageRoleError, InvalidModelIdError, langchainAdapter as LangChainAdapter, LangChainStream, LanguageModel, LogProbs, MistralStream, NoSuchModelError, NoSuchProviderError, ObjectStreamInputPart, ObjectStreamPart, OpenAIStream, OpenAIStreamCallbacks, ReplicateStream, StreamData, StreamObjectResult, StreamTextResult, StreamingTextResponse, TextPart$1 as TextPart, TextStreamPart, TokenUsage, ToolCallPart, ToolCallPayload, ToolContent, ToolResultPart, UserContent, convertDataContentToBase64String, convertDataContentToUint8Array, convertToCoreMessages, cosineSimilarity, createCallbacksTransformer, createEventStreamTransformer, createStreamDataTransformer, embed, embedMany, experimental_AssistantResponse, experimental_ModelRegistry, experimental_ProviderRegistry, experimental_StreamData, experimental_createModelRegistry, experimental_createProviderRegistry, experimental_generateObject, experimental_generateText, experimental_streamObject, experimental_streamText, generateId, generateObject, generateText, nanoid, readableFromAsyncIterable, streamObject, streamText, streamToResponse, tool, trimStartOfStreamHelper };

package/dist/index.js

@@ -87,6 +87,7 @@ __export(streams_exports, {
   experimental_AssistantResponse: () => experimental_AssistantResponse,
   experimental_StreamData: () => experimental_StreamData,
   experimental_createModelRegistry: () => experimental_createModelRegistry,
+  experimental_createProviderRegistry: () => experimental_createProviderRegistry,
   experimental_generateObject: () => experimental_generateObject,
   experimental_generateText: () => experimental_generateText,
   experimental_streamObject: () => experimental_streamObject,
@@ -2158,21 +2159,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
     };
   }
 };
@@ -2181,75 +2185,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;
   }
@@ -3434,6 +3441,7 @@ var nanoid = import_provider_utils7.generateId;
   experimental_AssistantResponse,
   experimental_StreamData,
   experimental_createModelRegistry,
+  experimental_createProviderRegistry,
   experimental_generateObject,
   experimental_generateText,
   experimental_streamObject,