@ai-sdk/amazon-bedrock 4.0.30 → 4.0.31

package/CHANGELOG.md

@@ -1,5 +1,11 @@
 # @ai-sdk/amazon-bedrock
 
+## 4.0.31
+
+### Patch Changes
+
+- 151c4aa: Add support for Amazon Nova embedding models by sending the required `taskType` and `singleEmbeddingParams` payload and parsing Nova-style responses.
+
 ## 4.0.30
 
 ### 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.30" : "0.0.0-test";
+var VERSION = true ? "4.0.31" : "0.0.0-test";
 
 // src/bedrock-sigv4-fetch.ts
 function createSigV4FetchFunction(getCredentials, fetch = globalThis.fetch) {

package/dist/anthropic/index.mjs

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

package/dist/index.js

@@ -1499,13 +1499,35 @@ var bedrockEmbeddingProviderOptions = import_v44.z.object({
    */
   normalize: import_v44.z.boolean().optional(),
   /**
+  The number of dimensions for Nova embedding models (defaults to 1024).
+  Supported values: 256, 384, 1024, 3072.
+  Only supported in amazon.nova-* embedding models.
+   */
+  embeddingDimension: import_v44.z.union([import_v44.z.literal(256), import_v44.z.literal(384), import_v44.z.literal(1024), import_v44.z.literal(3072)]).optional(),
+  /**
+  The purpose of the embedding. Defaults to 'GENERIC_INDEX'.
+  Only supported in amazon.nova-* embedding models.
+   */
+  embeddingPurpose: import_v44.z.enum([
+    "GENERIC_INDEX",
+    "TEXT_RETRIEVAL",
+    "IMAGE_RETRIEVAL",
+    "VIDEO_RETRIEVAL",
+    "DOCUMENT_RETRIEVAL",
+    "AUDIO_RETRIEVAL",
+    "GENERIC_RETRIEVAL",
+    "CLASSIFICATION",
+    "CLUSTERING"
+  ]).optional(),
+  /**
   Input type for Cohere embedding models on Bedrock.
   Common values: `search_document`, `search_query`, `classification`, `clustering`.
   If not set, the provider defaults to `search_query`.
      */
   inputType: import_v44.z.enum(["search_document", "search_query", "classification", "clustering"]).optional(),
   /**
-  Truncation behavior for Cohere embedding models on Bedrock.
+  Truncation behavior when input exceeds the model's context length.
+  Supported in Cohere and Nova embedding models. Defaults to 'END' for Nova models.
      */
   truncate: import_v44.z.enum(["NONE", "START", "END"]).optional()
 });
@@ -1531,7 +1553,7 @@ var BedrockEmbeddingModel = class {
     abortSignal,
     providerOptions
   }) {
-    var _a, _b;
+    var _a, _b, _c, _d, _e, _f;
     if (values.length > this.maxEmbeddingsPerCall) {
       throw new import_provider4.TooManyEmbeddingValuesForCallError({
         provider: this.provider,
@@ -1545,10 +1567,22 @@ var BedrockEmbeddingModel = class {
       providerOptions,
       schema: bedrockEmbeddingProviderOptions
     })) != null ? _a : {};
-    const args = this.modelId.startsWith("cohere.embed-") ? {
+    const isNovaModel = this.modelId.startsWith("amazon.nova-") && this.modelId.includes("embed");
+    const isCohereModel = this.modelId.startsWith("cohere.embed-");
+    const args = isNovaModel ? {
+      taskType: "SINGLE_EMBEDDING",
+      singleEmbeddingParams: {
+        embeddingPurpose: (_b = bedrockOptions.embeddingPurpose) != null ? _b : "GENERIC_INDEX",
+        embeddingDimension: (_c = bedrockOptions.embeddingDimension) != null ? _c : 1024,
+        text: {
+          truncationMode: (_d = bedrockOptions.truncate) != null ? _d : "END",
+          value: values[0]
+        }
+      }
+    } : isCohereModel ? {
       // Cohere embedding models on Bedrock require `input_type`.
       // Without it, the service attempts other schema branches and rejects the request.
-      input_type: (_b = bedrockOptions.inputType) != null ? _b : "search_query",
+      input_type: (_e = bedrockOptions.inputType) != null ? _e : "search_query",
       texts: [values[0]],
       truncate: bedrockOptions.truncate
     } : {
@@ -1573,13 +1607,24 @@ var BedrockEmbeddingModel = class {
       fetch: this.config.fetch,
       abortSignal
     });
-    const embedding = "embedding" in response ? response.embedding : Array.isArray(response.embeddings) ? response.embeddings[0] : response.embeddings.float[0];
+    let embedding;
+    if ("embedding" in response) {
+      embedding = 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;
+      } else {
+        embedding = firstEmbedding;
+      }
+    } else {
+      embedding = response.embeddings.float[0];
+    }
+    const tokens = "inputTextTokenCount" in response ? response.inputTextTokenCount : "inputTokenCount" in response ? (_f = response.inputTokenCount) != null ? _f : 0 : NaN;
     return {
-      warnings: [],
       embeddings: [embedding],
-      usage: {
-        tokens: "inputTextTokenCount" in response ? response.inputTextTokenCount : NaN
-      }
+      usage: { tokens },
+      warnings: []
     };
   }
 };
@@ -1589,6 +1634,16 @@ var BedrockEmbeddingResponseSchema = import_v45.z.union([
     embedding: import_v45.z.array(import_v45.z.number()),
     inputTextTokenCount: import_v45.z.number()
   }),
+  // Nova-style response
+  import_v45.z.object({
+    embeddings: import_v45.z.array(
+      import_v45.z.object({
+        embeddingType: import_v45.z.string(),
+        embedding: import_v45.z.array(import_v45.z.number())
+      })
+    ),
+    inputTokenCount: import_v45.z.number().optional()
+  }),
   // Cohere v3-style response
   import_v45.z.object({
     embeddings: import_v45.z.array(import_v45.z.array(import_v45.z.number()))
@@ -1813,7 +1868,7 @@ var import_provider_utils7 = require("@ai-sdk/provider-utils");
 var import_aws4fetch = require("aws4fetch");
 
 // src/version.ts
-var VERSION = true ? "4.0.30" : "0.0.0-test";
+var VERSION = true ? "4.0.31" : "0.0.0-test";
 
 // src/bedrock-sigv4-fetch.ts
 function createSigV4FetchFunction(getCredentials, fetch = globalThis.fetch) {