@ai-sdk/amazon-bedrock 4.0.126 → 4.0.128

package/dist/index.mjs

@@ -1692,9 +1692,11 @@ var BedrockEmbeddingModel = class {
     this.config = config;
     this.specificationVersion = "v3";
     this.provider = "amazon-bedrock";
-    this.maxEmbeddingsPerCall = 1;
     this.supportsParallelCalls = true;
   }
+  get maxEmbeddingsPerCall() {
+    return isCohereEmbeddingModel(this.modelId) ? 96 : 1;
+  }
   getUrl(modelId) {
     const encodedModelId = encodeURIComponent(modelId);
     return `${this.config.baseUrl()}/model/${encodedModelId}/invoke`;
@@ -1719,8 +1721,8 @@ var BedrockEmbeddingModel = class {
       providerOptions,
       schema: amazonBedrockEmbeddingModelOptionsSchema
     })) != null ? _a : {};
-    const isNovaModel = this.modelId.startsWith("amazon.nova-") && this.modelId.includes("embed");
-    const isCohereModel = this.modelId.includes("cohere.embed-");
+    const isNovaModel = isNovaEmbeddingModel(this.modelId);
+    const isCohereModel = isCohereEmbeddingModel(this.modelId);
     const args = isNovaModel ? {
       taskType: "SINGLE_EMBEDDING",
       singleEmbeddingParams: {
@@ -1735,7 +1737,7 @@ var BedrockEmbeddingModel = class {
       // Cohere embedding models on Bedrock require `input_type`.
       // Without it, the service attempts other schema branches and rejects the request.
       input_type: (_e = bedrockOptions.inputType) != null ? _e : "search_query",
-      texts: [values[0]],
+      texts: values,
       truncate: bedrockOptions.truncate,
       output_dimension: bedrockOptions.outputDimension
     } : {
@@ -1760,30 +1762,36 @@ var BedrockEmbeddingModel = class {
       fetch: this.config.fetch,
       abortSignal
     });
-    let embedding;
+    let embeddings;
     if ("embedding" in response) {
-      embedding = response.embedding;
+      embeddings = [response.embedding];
     } else if (Array.isArray(response.embeddings)) {
       const firstEmbedding = response.embeddings[0];
       if (typeof firstEmbedding === "object" && firstEmbedding !== null && "embeddingType" in firstEmbedding) {
-        embedding = firstEmbedding.embedding;
+        embeddings = [firstEmbedding.embedding];
       } else {
-        embedding = firstEmbedding;
+        embeddings = response.embeddings;
       }
     } else {
-      embedding = response.embeddings.float[0];
+      embeddings = response.embeddings.float;
     }
     const headerTokenCount = Number(
       responseHeaders == null ? void 0 : responseHeaders["x-amzn-bedrock-input-token-count"]
     );
     const tokens = "inputTextTokenCount" in response ? response.inputTextTokenCount : "inputTokenCount" in response ? (_f = response.inputTokenCount) != null ? _f : 0 : headerTokenCount;
     return {
-      embeddings: [embedding],
+      embeddings,
       usage: { tokens },
       warnings: []
     };
   }
 };
+function isCohereEmbeddingModel(modelId) {
+  return modelId.includes("cohere.embed-");
+}
+function isNovaEmbeddingModel(modelId) {
+  return modelId.startsWith("amazon.nova-") && modelId.includes("embed");
+}
 var BedrockEmbeddingResponseSchema = z6.union([
   // Titan-style response
   z6.object({
@@ -2036,7 +2044,7 @@ import {
 import { AwsV4Signer } from "aws4fetch";
 
 // src/version.ts
-var VERSION = true ? "4.0.126" : "0.0.0-test";
+var VERSION = true ? "4.0.128" : "0.0.0-test";
 
 // src/bedrock-sigv4-fetch.ts
 function createSigV4FetchFunction(getCredentials, fetch, service = "bedrock") {