@ai-sdk/amazon-bedrock 4.0.29 → 4.0.30

package/CHANGELOG.md

@@ -1,5 +1,11 @@
 # @ai-sdk/amazon-bedrock
 
+## 4.0.30
+
+### Patch Changes
+
+- 6ece44c: Fix Cohere embedding model request format on Bedrock by sending the required `input_type` and parsing Cohere-style responses.
+
 ## 4.0.29
 
 ### 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.29" : "0.0.0-test";
+var VERSION = true ? "4.0.30" : "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.29" : "0.0.0-test";
+var VERSION = true ? "4.0.30" : "0.0.0-test";
 
 // src/bedrock-sigv4-fetch.ts
 function createSigV4FetchFunction(getCredentials, fetch = globalThis.fetch) {

package/dist/index.js

@@ -1497,7 +1497,17 @@ var bedrockEmbeddingProviderOptions = import_v44.z.object({
   Flag indicating whether or not to normalize the output embeddings. Defaults to true
   Only supported in amazon.titan-embed-text-v2:0.
    */
-  normalize: import_v44.z.boolean().optional()
+  normalize: import_v44.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: import_v44.z.enum(["search_document", "search_query", "classification", "clustering"]).optional(),
+  /**
+  Truncation behavior for Cohere embedding models on Bedrock.
+     */
+  truncate: import_v44.z.enum(["NONE", "START", "END"]).optional()
 });
 
 // src/bedrock-embedding-model.ts
@@ -1521,7 +1531,7 @@ var BedrockEmbeddingModel = class {
     abortSignal,
     providerOptions
   }) {
-    var _a;
+    var _a, _b;
     if (values.length > this.maxEmbeddingsPerCall) {
       throw new import_provider4.TooManyEmbeddingValuesForCallError({
         provider: this.provider,
@@ -1535,7 +1545,13 @@ var BedrockEmbeddingModel = class {
       providerOptions,
       schema: bedrockEmbeddingProviderOptions
     })) != null ? _a : {};
-    const args = {
+    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: (_b = bedrockOptions.inputType) != null ? _b : "search_query",
+      texts: [values[0]],
+      truncate: bedrockOptions.truncate
+    } : {
       inputText: values[0],
       dimensions: bedrockOptions.dimensions,
       normalize: bedrockOptions.normalize
@@ -1557,17 +1573,33 @@ 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];
     return {
       warnings: [],
-      embeddings: [response.embedding],
-      usage: { tokens: response.inputTextTokenCount }
+      embeddings: [embedding],
+      usage: {
+        tokens: "inputTextTokenCount" in response ? response.inputTextTokenCount : NaN
+      }
     };
   }
 };
-var BedrockEmbeddingResponseSchema = import_v45.z.object({
-  embedding: import_v45.z.array(import_v45.z.number()),
-  inputTextTokenCount: import_v45.z.number()
-});
+var BedrockEmbeddingResponseSchema = import_v45.z.union([
+  // Titan-style response
+  import_v45.z.object({
+    embedding: import_v45.z.array(import_v45.z.number()),
+    inputTextTokenCount: import_v45.z.number()
+  }),
+  // Cohere v3-style response
+  import_v45.z.object({
+    embeddings: import_v45.z.array(import_v45.z.array(import_v45.z.number()))
+  }),
+  // Cohere v4-style response
+  import_v45.z.object({
+    embeddings: import_v45.z.object({
+      float: import_v45.z.array(import_v45.z.array(import_v45.z.number()))
+    })
+  })
+]);
 
 // src/bedrock-image-model.ts
 var import_provider_utils6 = require("@ai-sdk/provider-utils");
@@ -1781,7 +1813,7 @@ var import_provider_utils7 = require("@ai-sdk/provider-utils");
 var import_aws4fetch = require("aws4fetch");
 
 // src/version.ts
-var VERSION = true ? "4.0.29" : "0.0.0-test";
+var VERSION = true ? "4.0.30" : "0.0.0-test";
 
 // src/bedrock-sigv4-fetch.ts
 function createSigV4FetchFunction(getCredentials, fetch = globalThis.fetch) {