@ai-sdk/amazon-bedrock 3.0.0-canary.10 → 3.0.0-canary.11

package/dist/index.mjs

@@ -11,7 +11,7 @@ import {
   combineHeaders,
   createJsonErrorResponseHandler,
   createJsonResponseHandler,
-  parseProviderOptions,
+  parseProviderOptions as parseProviderOptions2,
   postJsonToApi,
   resolve
 } from "@ai-sdk/provider-utils";
@@ -77,7 +77,7 @@ var createBedrockEventStreamResponseHandler = (chunkSchema) => async ({ response
     responseHeaders,
     value: response.body.pipeThrough(
       new TransformStream({
-        transform(chunk, controller) {
+        async transform(chunk, controller) {
           var _a, _b;
           const newBuffer = new Uint8Array(buffer.length + chunk.length);
           newBuffer.set(buffer);
@@ -98,7 +98,7 @@ var createBedrockEventStreamResponseHandler = (chunkSchema) => async ({ response
               buffer = buffer.slice(totalLength);
               if (((_a = decoded.headers[":message-type"]) == null ? void 0 : _a.value) === "event") {
                 const data = textDecoder.decode(decoded.body);
-                const parsedDataResult = safeParseJSON({ text: data });
+                const parsedDataResult = await safeParseJSON({ text: data });
                 if (!parsedDataResult.success) {
                   controller.enqueue(parsedDataResult);
                   break;
@@ -107,7 +107,7 @@ var createBedrockEventStreamResponseHandler = (chunkSchema) => async ({ response
                 let wrappedData = {
                   [(_b = decoded.headers[":event-type"]) == null ? void 0 : _b.value]: parsedDataResult.value
                 };
-                const validatedWrappedData = safeValidateTypes({
+                const validatedWrappedData = await safeValidateTypes({
                   value: wrappedData,
                   schema: chunkSchema
                 });
@@ -207,13 +207,17 @@ function prepareTools({
 import {
   UnsupportedFunctionalityError as UnsupportedFunctionalityError2
 } from "@ai-sdk/provider";
-import { convertToBase64, createIdGenerator } from "@ai-sdk/provider-utils";
+import {
+  convertToBase64,
+  createIdGenerator,
+  parseProviderOptions
+} from "@ai-sdk/provider-utils";
 var generateFileId = createIdGenerator({ prefix: "file", size: 16 });
 function getCachePoint(providerMetadata) {
   var _a;
   return (_a = providerMetadata == null ? void 0 : providerMetadata.bedrock) == null ? void 0 : _a.cachePoint;
 }
-function convertToBedrockChatMessages(prompt) {
+async function convertToBedrockChatMessages(prompt) {
   var _a, _b, _c, _d;
   const blocks = groupIntoBlocks(prompt);
   let system = [];
@@ -362,32 +366,39 @@ function convertToBedrockChatMessages(prompt) {
                 break;
               }
               case "reasoning": {
-                bedrockContent.push({
-                  reasoningContent: {
-                    reasoningText: {
-                      // trim the last text part if it's the last message in the block
-                      // because Bedrock does not allow trailing whitespace
-                      // in pre-filled assistant responses
-                      text: trimIfLast(
-                        isLastBlock,
-                        isLastMessage,
-                        isLastContentPart,
-                        part.text
-                      ),
-                      signature: part.signature
-                    }
-                  }
+                const reasoningMetadata = await parseProviderOptions({
+                  provider: "bedrock",
+                  providerOptions: part.providerOptions,
+                  schema: bedrockReasoningMetadataSchema
                 });
-                break;
-              }
-              case "redacted-reasoning": {
-                bedrockContent.push({
-                  reasoningContent: {
-                    redactedReasoning: {
-                      data: part.data
-                    }
+                if (reasoningMetadata != null) {
+                  if (reasoningMetadata.signature != null) {
+                    bedrockContent.push({
+                      reasoningContent: {
+                        reasoningText: {
+                          // trim the last text part if it's the last message in the block
+                          // because Bedrock does not allow trailing whitespace
+                          // in pre-filled assistant responses
+                          text: trimIfLast(
+                            isLastBlock,
+                            isLastMessage,
+                            isLastContentPart,
+                            part.text
+                          ),
+                          signature: reasoningMetadata.signature
+                        }
+                      }
+                    });
+                  } else if (reasoningMetadata.redactedData != null) {
+                    bedrockContent.push({
+                      reasoningContent: {
+                        redactedReasoning: {
+                          data: reasoningMetadata.redactedData
+                        }
+                      }
+                    });
                   }
-                });
+                }
                 break;
               }
               case "tool-call": {
@@ -496,7 +507,7 @@ var BedrockChatLanguageModel = class {
     this.specificationVersion = "v2";
     this.provider = "amazon-bedrock";
   }
-  getArgs({
+  async getArgs({
     prompt,
     maxOutputTokens,
     temperature,
@@ -512,7 +523,7 @@ var BedrockChatLanguageModel = class {
     providerOptions
   }) {
     var _a, _b, _c, _d, _e;
-    const bedrockOptions = (_a = parseProviderOptions({
+    const bedrockOptions = (_a = await parseProviderOptions2({
       provider: "bedrock",
       providerOptions,
       schema: bedrockProviderOptions
@@ -549,7 +560,7 @@ var BedrockChatLanguageModel = class {
         details: "JSON response format is not supported."
       });
     }
-    const { system, messages } = convertToBedrockChatMessages(prompt);
+    const { system, messages } = await convertToBedrockChatMessages(prompt);
     const isThinking = ((_b = bedrockOptions.reasoningConfig) == null ? void 0 : _b.type) === "enabled";
     const thinkingBudget = (_c = bedrockOptions.reasoningConfig) == null ? void 0 : _c.budgetTokens;
     const inferenceConfig = {
@@ -610,7 +621,7 @@ var BedrockChatLanguageModel = class {
   }
   async doGenerate(options) {
     var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m;
-    const { command: args, warnings } = this.getArgs(options);
+    const { command: args, warnings } = await this.getArgs(options);
     const url = `${this.getUrl(this.modelId)}/converse`;
     const { value: response, responseHeaders } = await postJsonToApi({
       url,
@@ -639,23 +650,27 @@ var BedrockChatLanguageModel = class {
       }
       if (part.reasoningContent) {
         if ("reasoningText" in part.reasoningContent) {
-          content.push({
+          const reasoning = {
             type: "reasoning",
-            reasoningType: "text",
             text: part.reasoningContent.reasoningText.text
-          });
+          };
           if (part.reasoningContent.reasoningText.signature) {
-            content.push({
-              type: "reasoning",
-              reasoningType: "signature",
-              signature: part.reasoningContent.reasoningText.signature
-            });
+            reasoning.providerMetadata = {
+              bedrock: {
+                signature: part.reasoningContent.reasoningText.signature
+              }
+            };
           }
+          content.push(reasoning);
         } else if ("redactedReasoning" in part.reasoningContent) {
           content.push({
             type: "reasoning",
-            reasoningType: "redacted",
-            data: (_a = part.reasoningContent.redactedReasoning.data) != null ? _a : ""
+            text: "",
+            providerMetadata: {
+              bedrock: {
+                redactedData: (_a = part.reasoningContent.redactedReasoning.data) != null ? _a : ""
+              }
+            }
           });
         }
       }
@@ -698,7 +713,7 @@ var BedrockChatLanguageModel = class {
     };
   }
   async doStream(options) {
-    const { command: args, warnings } = this.getArgs(options);
+    const { command: args, warnings } = await this.getArgs(options);
     const url = `${this.getUrl(this.modelId)}/converse-stream`;
     const { value: response, responseHeaders } = await postJsonToApi({
       url,
@@ -792,21 +807,30 @@ var BedrockChatLanguageModel = class {
               if ("text" in reasoningContent && reasoningContent.text) {
                 controller.enqueue({
                   type: "reasoning",
-                  reasoningType: "text",
                   text: reasoningContent.text
                 });
               } else if ("signature" in reasoningContent && reasoningContent.signature) {
                 controller.enqueue({
                   type: "reasoning",
-                  reasoningType: "signature",
-                  signature: reasoningContent.signature
+                  text: "",
+                  providerMetadata: {
+                    bedrock: {
+                      signature: reasoningContent.signature
+                    }
+                  }
                 });
+                controller.enqueue({ type: "reasoning-part-finish" });
               } else if ("data" in reasoningContent && reasoningContent.data) {
                 controller.enqueue({
                   type: "reasoning",
-                  reasoningType: "redacted",
-                  data: reasoningContent.data
+                  text: "",
+                  providerMetadata: {
+                    bedrock: {
+                      redactedData: reasoningContent.data
+                    }
+                  }
                 });
+                controller.enqueue({ type: "reasoning-part-finish" });
               }
             }
             const contentBlockStart = value.contentBlockStart;
@@ -961,13 +985,17 @@ var BedrockStreamSchema = z3.object({
   throttlingException: z3.record(z3.unknown()).nullish(),
   validationException: z3.record(z3.unknown()).nullish()
 });
+var bedrockReasoningMetadataSchema = z3.object({
+  signature: z3.string().optional(),
+  redactedData: z3.string().optional()
+});
 
 // src/bedrock-embedding-model.ts
 import {
   combineHeaders as combineHeaders2,
   createJsonErrorResponseHandler as createJsonErrorResponseHandler2,
   createJsonResponseHandler as createJsonResponseHandler2,
-  parseProviderOptions as parseProviderOptions2,
+  parseProviderOptions as parseProviderOptions3,
   postJsonToApi as postJsonToApi2,
   resolve as resolve2
 } from "@ai-sdk/provider-utils";
@@ -1009,7 +1037,7 @@ var BedrockEmbeddingModel = class {
     providerOptions
   }) {
     var _a;
-    const bedrockOptions = (_a = parseProviderOptions2({
+    const bedrockOptions = (_a = await parseProviderOptions3({
       provider: "bedrock",
       providerOptions,
       schema: bedrockEmbeddingProviderOptions