@ai-sdk/amazon-bedrock 4.0.127 → 4.0.129

package/CHANGELOG.md

@@ -1,5 +1,18 @@
 # @ai-sdk/amazon-bedrock
 
+## 4.0.129
+
+### Patch Changes
+
+- Updated dependencies [4241b49]
+  - @ai-sdk/anthropic@3.0.93
+
+## 4.0.128
+
+### Patch Changes
+
+- 3f9bc0f: feat(amazon-bedrock): increase limit of embeddings in a request for cohere models
+
 ## 4.0.127
 
 ### Patch Changes

package/dist/anthropic/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.129" : "0.0.0-test";
 
 // src/bedrock-sigv4-fetch.ts
 function createSigV4FetchFunction(getCredentials, fetch, service = "bedrock") {

package/dist/anthropic/index.mjs

@@ -24,7 +24,7 @@ import {
 import { AwsV4Signer } from "aws4fetch";
 
 // src/version.ts
-var VERSION = true ? "4.0.127" : "0.0.0-test";
+var VERSION = true ? "4.0.129" : "0.0.0-test";
 
 // src/bedrock-sigv4-fetch.ts
 function createSigV4FetchFunction(getCredentials, fetch, service = "bedrock") {

package/dist/index.js

@@ -1683,9 +1683,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`;
@@ -1710,8 +1712,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: {
@@ -1726,7 +1728,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
     } : {
@@ -1751,30 +1753,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 = import_v46.z.union([
   // Titan-style response
   import_v46.z.object({
@@ -2015,7 +2023,7 @@ var import_provider_utils7 = 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.129" : "0.0.0-test";
 
 // src/bedrock-sigv4-fetch.ts
 function createSigV4FetchFunction(getCredentials, fetch, service = "bedrock") {