@ai-sdk/amazon-bedrock 4.0.127 → 4.0.128

package/dist/mantle/index.js

@@ -35,7 +35,7 @@ var import_provider_utils = require("@ai-sdk/provider-utils");
 var import_aws4fetch = require("aws4fetch");
 
 // src/version.ts
-var VERSION = true ? "4.0.127" : "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") {

package/dist/mantle/index.mjs

@@ -23,7 +23,7 @@ import {
 import { AwsV4Signer } from "aws4fetch";
 
 // src/version.ts
-var VERSION = true ? "4.0.127" : "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") {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/amazon-bedrock",
-  "version": "4.0.127",
+  "version": "4.0.128",
   "license": "Apache-2.0",
   "sideEffects": false,
   "main": "./dist/index.js",
@@ -44,10 +44,10 @@
     "@smithy/eventstream-codec": "^4.0.1",
     "@smithy/util-utf8": "^4.0.0",
     "aws4fetch": "^1.0.20",
-    "@ai-sdk/anthropic": "3.0.92",
-    "@ai-sdk/openai": "3.0.80",
     "@ai-sdk/provider": "3.0.13",
-    "@ai-sdk/provider-utils": "4.0.35"
+    "@ai-sdk/anthropic": "3.0.92",
+    "@ai-sdk/provider-utils": "4.0.35",
+    "@ai-sdk/openai": "3.0.80"
   },
   "devDependencies": {
     "@types/node": "20.17.24",

package/src/bedrock-embedding-model.ts

@@ -30,9 +30,12 @@ type DoEmbedResponse = Awaited<ReturnType<EmbeddingModelV3['doEmbed']>>;
 export class BedrockEmbeddingModel implements EmbeddingModelV3 {
   readonly specificationVersion = 'v3';
   readonly provider = 'amazon-bedrock';
-  readonly maxEmbeddingsPerCall = 1;
   readonly supportsParallelCalls = true;
 
+  get maxEmbeddingsPerCall() {
+    return isCohereEmbeddingModel(this.modelId) ? 96 : 1;
+  }
+
   constructor(
     readonly modelId: BedrockEmbeddingModelId,
     private readonly config: BedrockEmbeddingConfig,
@@ -71,11 +74,8 @@ export class BedrockEmbeddingModel implements EmbeddingModelV3 {
     // Note: Different embedding model families expect different request/response
     // payloads (e.g. Titan vs Cohere vs Nova). We keep the public interface stable and
     // adapt here based on the modelId.
-    const isNovaModel =
-      this.modelId.startsWith('amazon.nova-') && this.modelId.includes('embed');
-    // Use `includes` so cross-region inference profile ids (e.g.
-    // `us.cohere.embed-v4:0`, `global.cohere.embed-v4:0`) are detected too.
-    const isCohereModel = this.modelId.includes('cohere.embed-');
+    const isNovaModel = isNovaEmbeddingModel(this.modelId);
+    const isCohereModel = isCohereEmbeddingModel(this.modelId);
 
     const args = isNovaModel
       ? {
@@ -95,7 +95,7 @@ export class BedrockEmbeddingModel implements EmbeddingModelV3 {
             // Cohere embedding models on Bedrock require `input_type`.
             // Without it, the service attempts other schema branches and rejects the request.
             input_type: bedrockOptions.inputType ?? 'search_query',
-            texts: [values[0]],
+            texts: values,
             truncate: bedrockOptions.truncate,
             output_dimension: bedrockOptions.outputDimension,
           }
@@ -123,11 +123,11 @@ export class BedrockEmbeddingModel implements EmbeddingModelV3 {
       abortSignal,
     });
 
-    // Extract embedding based on response format
-    let embedding: number[];
+    // Extract embeddings based on response format
+    let embeddings: number[][];
     if ('embedding' in response) {
       // Titan response
-      embedding = response.embedding;
+      embeddings = [response.embedding];
     } else if (Array.isArray(response.embeddings)) {
       const firstEmbedding = response.embeddings[0];
       if (
@@ -136,14 +136,14 @@ export class BedrockEmbeddingModel implements EmbeddingModelV3 {
         'embeddingType' in firstEmbedding
       ) {
         // Nova response
-        embedding = firstEmbedding.embedding;
+        embeddings = [firstEmbedding.embedding];
       } else {
         // Cohere v3 response
-        embedding = firstEmbedding as number[];
+        embeddings = response.embeddings as number[][];
       }
     } else {
       // Cohere v4 response
-      embedding = response.embeddings.float[0];
+      embeddings = response.embeddings.float;
     }
 
     // Extract token count based on response format
@@ -158,13 +158,23 @@ export class BedrockEmbeddingModel implements EmbeddingModelV3 {
           : headerTokenCount;
 
     return {
-      embeddings: [embedding],
+      embeddings,
       usage: { tokens },
       warnings: [],
     };
   }
 }
 
+function isCohereEmbeddingModel(modelId: string) {
+  // Use `includes` so cross-region inference profile ids (e.g.
+  // `us.cohere.embed-v4:0`, `global.cohere.embed-v4:0`) are detected too.
+  return modelId.includes('cohere.embed-');
+}
+
+function isNovaEmbeddingModel(modelId: string) {
+  return modelId.startsWith('amazon.nova-') && modelId.includes('embed');
+}
+
 const BedrockEmbeddingResponseSchema = z.union([
   // Titan-style response
   z.object({