@ai-sdk/amazon-bedrock 4.0.29 → 4.0.30

package/docs/08-amazon-bedrock.mdx

@@ -758,6 +758,13 @@ The following optional provider options are available for Bedrock Titan embeddin
 | `cohere.embed-english-v3`      | 1024               | <Cross size={18} /> |
 | `cohere.embed-multilingual-v3` | 1024               | <Cross size={18} /> |
 
+<Note>
+  Cohere embedding models on Bedrock require an <code>input_type</code>. Set it
+  via
+  <code>providerOptions.bedrock.inputType</code> (defaults to{' '}
+  <code>search_query</code>).
+</Note>
+
 ## Reranking Models
 
 You can create models that call the [Bedrock Rerank API](https://docs.aws.amazon.com/bedrock/latest/userguide/rerank-api.html)

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/amazon-bedrock",
-  "version": "4.0.29",
+  "version": "4.0.30",
   "license": "Apache-2.0",
   "sideEffects": false,
   "main": "./dist/index.js",
@@ -47,8 +47,8 @@
     "tsup": "^8.3.0",
     "typescript": "5.8.3",
     "zod": "3.25.76",
-    "@ai-sdk/test-server": "1.0.3",
-    "@vercel/ai-tsconfig": "0.0.0"
+    "@vercel/ai-tsconfig": "0.0.0",
+    "@ai-sdk/test-server": "1.0.3"
   },
   "peerDependencies": {
     "zod": "^3.25.76 || ^4.1.8"

package/src/bedrock-embedding-model.ts

@@ -67,11 +67,23 @@ export class BedrockEmbeddingModel implements EmbeddingModelV3 {
       })) ?? {};
 
     // https://docs.aws.amazon.com/bedrock/latest/APIReference/API_runtime_InvokeModel.html
-    const args = {
-      inputText: values[0],
-      dimensions: bedrockOptions.dimensions,
-      normalize: bedrockOptions.normalize,
-    };
+    //
+    // Note: Different embedding model families expect different request/response
+    // payloads (e.g. Titan vs Cohere). We keep the public interface stable and
+    // adapt here based on the modelId.
+    const args = this.modelId.startsWith('cohere.embed-')
+      ? {
+          // 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]],
+          truncate: bedrockOptions.truncate,
+        }
+      : {
+          inputText: values[0],
+          dimensions: bedrockOptions.dimensions,
+          normalize: bedrockOptions.normalize,
+        };
     const url = this.getUrl(this.modelId);
     const { value: response } = await postJsonToApi({
       url,
@@ -90,15 +102,40 @@ export class BedrockEmbeddingModel implements EmbeddingModelV3 {
       abortSignal,
     });
 
+    const embedding =
+      'embedding' in response
+        ? response.embedding
+        : Array.isArray(response.embeddings)
+          ? response.embeddings[0]
+          : response.embeddings.float[0];
+
     return {
       warnings: [],
-      embeddings: [response.embedding],
-      usage: { tokens: response.inputTextTokenCount },
+      embeddings: [embedding],
+      usage: {
+        tokens:
+          'inputTextTokenCount' in response
+            ? response.inputTextTokenCount
+            : NaN,
+      },
     };
   }
 }
 
-const BedrockEmbeddingResponseSchema = z.object({
-  embedding: z.array(z.number()),
-  inputTextTokenCount: z.number(),
-});
+const BedrockEmbeddingResponseSchema = z.union([
+  // Titan-style response
+  z.object({
+    embedding: z.array(z.number()),
+    inputTextTokenCount: z.number(),
+  }),
+  // Cohere v3-style response
+  z.object({
+    embeddings: z.array(z.array(z.number())),
+  }),
+  // Cohere v4-style response
+  z.object({
+    embeddings: z.object({
+      float: z.array(z.array(z.number())),
+    }),
+  }),
+]);

package/src/bedrock-embedding-options.ts

@@ -21,4 +21,18 @@ Flag indicating whether or not to normalize the output embeddings. Defaults to t
 Only supported in amazon.titan-embed-text-v2:0.
  */
   normalize: z.boolean().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: z
+    .enum(['search_document', 'search_query', 'classification', 'clustering'])
+    .optional(),
+
+  /**
+Truncation behavior for Cohere embedding models on Bedrock.
+   */
+  truncate: z.enum(['NONE', 'START', 'END']).optional(),
 });