@ai-sdk/amazon-bedrock 4.0.40 → 4.0.41

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/amazon-bedrock",
-  "version": "4.0.40",
+  "version": "4.0.41",
   "license": "Apache-2.0",
   "sideEffects": false,
   "main": "./dist/index.js",
@@ -39,8 +39,8 @@
     "@smithy/util-utf8": "^4.0.0",
     "aws4fetch": "^1.0.20",
     "@ai-sdk/anthropic": "3.0.31",
-    "@ai-sdk/provider": "3.0.6",
-    "@ai-sdk/provider-utils": "4.0.11"
+    "@ai-sdk/provider-utils": "4.0.11",
+    "@ai-sdk/provider": "3.0.6"
   },
   "devDependencies": {
     "@types/node": "20.17.24",

package/src/bedrock-embedding-model.ts

@@ -95,6 +95,7 @@ export class BedrockEmbeddingModel implements EmbeddingModelV3 {
             input_type: bedrockOptions.inputType ?? 'search_query',
             texts: [values[0]],
             truncate: bedrockOptions.truncate,
+            output_dimension: bedrockOptions.outputDimension,
           }
         : {
             inputText: values[0],

package/src/bedrock-embedding-options.ts

@@ -17,7 +17,7 @@ export const bedrockEmbeddingProviderOptions = z.object({
     .optional(),
 
   /**
-   * Flag indicating whether or not to normalize the output embeddings. Defaults to true
+   * Flag indicating whether or not to normalize the output embeddings. Defaults to true.
    * Only supported in amazon.titan-embed-text-v2:0.
    */
   normalize: z.boolean().optional(),
@@ -63,4 +63,12 @@ export const bedrockEmbeddingProviderOptions = z.object({
    * Supported in Cohere and Nova embedding models. Defaults to 'END' for Nova models.
    */
   truncate: z.enum(['NONE', 'START', 'END']).optional(),
+
+  /**
+   * The number of dimensions the resulting output embeddings should have (defaults to 1536).
+   * Only supported in cohere.embed-v4:0 and newer Cohere embedding models.
+   */
+  outputDimension: z
+    .union([z.literal(256), z.literal(512), z.literal(1024), z.literal(1536)])
+    .optional(),
 });