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

package/dist/index.mjs

@@ -623,13 +623,27 @@ var BedrockChatLanguageModel = class {
         setting: "topK"
       });
     }
-    if (responseFormat != null && responseFormat.type !== "text") {
+    if (responseFormat != null && responseFormat.type !== "text" && responseFormat.type !== "json") {
       warnings.push({
         type: "unsupported-setting",
         setting: "responseFormat",
-        details: "JSON response format is not supported."
+        details: "Only text and json response formats are supported."
       });
     }
+    if (tools != null && (responseFormat == null ? void 0 : responseFormat.type) === "json") {
+      if (tools.length > 0) {
+        warnings.push({
+          type: "other",
+          message: "JSON response format does not support tools. The provided tools are ignored."
+        });
+      }
+    }
+    const jsonResponseTool = (responseFormat == null ? void 0 : responseFormat.type) === "json" && responseFormat.schema != null ? {
+      type: "function",
+      name: "json",
+      description: "Respond with a JSON object.",
+      inputSchema: responseFormat.schema
+    } : void 0;
     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;
@@ -670,8 +684,8 @@ var BedrockChatLanguageModel = class {
       });
     }
     const { toolConfig, toolWarnings } = prepareTools({
-      tools,
-      toolChoice,
+      tools: jsonResponseTool != null ? [jsonResponseTool] : tools != null ? tools : [],
+      toolChoice: jsonResponseTool != null ? { type: "tool", toolName: jsonResponseTool.name } : toolChoice,
       prompt
     });
     const { reasoningConfig: _, ...filteredBedrockOptions } = (providerOptions == null ? void 0 : providerOptions.bedrock) || {};
@@ -686,12 +700,17 @@ var BedrockChatLanguageModel = class {
         ...filteredBedrockOptions,
         ...toolConfig.tools !== void 0 ? { toolConfig } : {}
       },
-      warnings: [...warnings, ...toolWarnings]
+      warnings: [...warnings, ...toolWarnings],
+      usesJsonResponseTool: jsonResponseTool != null
     };
   }
   async doGenerate(options) {
     var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n;
-    const { command: args, warnings } = await this.getArgs(options);
+    const {
+      command: args,
+      warnings,
+      usesJsonResponseTool
+    } = await this.getArgs(options);
     const url = `${this.getUrl(this.modelId)}/converse`;
     const { value: response, responseHeaders } = await postJsonToApi({
       url,
@@ -716,7 +735,9 @@ var BedrockChatLanguageModel = class {
     const content = [];
     for (const part of response.output.message.content) {
       if (part.text) {
-        content.push({ type: "text", text: part.text });
+        if (!usesJsonResponseTool) {
+          content.push({ type: "text", text: part.text });
+        }
       }
       if (part.reasoningContent) {
         if ("reasoningText" in part.reasoningContent) {
@@ -745,22 +766,29 @@ var BedrockChatLanguageModel = class {
         }
       }
       if (part.toolUse) {
-        content.push({
-          type: "tool-call",
-          toolCallId: (_c = (_b = part.toolUse) == null ? void 0 : _b.toolUseId) != null ? _c : this.config.generateId(),
-          toolName: (_e = (_d = part.toolUse) == null ? void 0 : _d.name) != null ? _e : `tool-${this.config.generateId()}`,
-          input: JSON.stringify((_g = (_f = part.toolUse) == null ? void 0 : _f.input) != null ? _g : "")
-        });
+        content.push(
+          // when a json response tool is used, the tool call becomes the text:
+          usesJsonResponseTool ? {
+            type: "text",
+            text: JSON.stringify(part.toolUse.input)
+          } : {
+            type: "tool-call",
+            toolCallId: (_c = (_b = part.toolUse) == null ? void 0 : _b.toolUseId) != null ? _c : this.config.generateId(),
+            toolName: (_e = (_d = part.toolUse) == null ? void 0 : _d.name) != null ? _e : `tool-${this.config.generateId()}`,
+            input: JSON.stringify((_g = (_f = part.toolUse) == null ? void 0 : _f.input) != null ? _g : "")
+          }
+        );
       }
     }
-    const providerMetadata = response.trace || response.usage ? {
+    const providerMetadata = response.trace || response.usage || usesJsonResponseTool ? {
       bedrock: {
         ...response.trace && typeof response.trace === "object" ? { trace: response.trace } : {},
         ...((_h = response.usage) == null ? void 0 : _h.cacheWriteInputTokens) != null && {
           usage: {
             cacheWriteInputTokens: response.usage.cacheWriteInputTokens
           }
-        }
+        },
+        ...usesJsonResponseTool && { isJsonResponseFromTool: true }
       }
     } : void 0;
     return {
@@ -783,7 +811,11 @@ var BedrockChatLanguageModel = class {
     };
   }
   async doStream(options) {
-    const { command: args, warnings } = await this.getArgs(options);
+    const {
+      command: args,
+      warnings,
+      usesJsonResponseTool
+    } = await this.getArgs(options);
     const url = `${this.getUrl(this.modelId)}/converse-stream`;
     const { value: response, responseHeaders } = await postJsonToApi({
       url,
@@ -862,11 +894,14 @@ var BedrockChatLanguageModel = class {
               const trace = value.metadata.trace ? {
                 trace: value.metadata.trace
               } : void 0;
-              if (cacheUsage || trace) {
+              if (cacheUsage || trace || usesJsonResponseTool) {
                 providerMetadata = {
                   bedrock: {
                     ...cacheUsage,
-                    ...trace
+                    ...trace,
+                    ...usesJsonResponseTool && {
+                      isJsonResponseFromTool: true
+                    }
                   }
                 };
               }
@@ -883,16 +918,20 @@ var BedrockChatLanguageModel = class {
               const blockIndex = value.contentBlockDelta.contentBlockIndex || 0;
               if (contentBlocks[blockIndex] == null) {
                 contentBlocks[blockIndex] = { type: "text" };
+                if (!usesJsonResponseTool) {
+                  controller.enqueue({
+                    type: "text-start",
+                    id: String(blockIndex)
+                  });
+                }
+              }
+              if (!usesJsonResponseTool) {
                 controller.enqueue({
-                  type: "text-start",
-                  id: String(blockIndex)
+                  type: "text-delta",
+                  id: String(blockIndex),
+                  delta: value.contentBlockDelta.delta.text
                 });
               }
-              controller.enqueue({
-                type: "text-delta",
-                id: String(blockIndex),
-                delta: value.contentBlockDelta.delta.text
-              });
             }
             if (((_n = value.contentBlockStop) == null ? void 0 : _n.contentBlockIndex) != null) {
               const blockIndex = value.contentBlockStop.contentBlockIndex;
@@ -904,21 +943,39 @@ var BedrockChatLanguageModel = class {
                     id: String(blockIndex)
                   });
                 } else if (contentBlock.type === "text") {
-                  controller.enqueue({
-                    type: "text-end",
-                    id: String(blockIndex)
-                  });
+                  if (!usesJsonResponseTool) {
+                    controller.enqueue({
+                      type: "text-end",
+                      id: String(blockIndex)
+                    });
+                  }
                 } else if (contentBlock.type === "tool-call") {
-                  controller.enqueue({
-                    type: "tool-input-end",
-                    id: contentBlock.toolCallId
-                  });
-                  controller.enqueue({
-                    type: "tool-call",
-                    toolCallId: contentBlock.toolCallId,
-                    toolName: contentBlock.toolName,
-                    input: contentBlock.jsonText
-                  });
+                  if (usesJsonResponseTool) {
+                    controller.enqueue({
+                      type: "text-start",
+                      id: String(blockIndex)
+                    });
+                    controller.enqueue({
+                      type: "text-delta",
+                      id: String(blockIndex),
+                      delta: contentBlock.jsonText
+                    });
+                    controller.enqueue({
+                      type: "text-end",
+                      id: String(blockIndex)
+                    });
+                  } else {
+                    controller.enqueue({
+                      type: "tool-input-end",
+                      id: contentBlock.toolCallId
+                    });
+                    controller.enqueue({
+                      type: "tool-call",
+                      toolCallId: contentBlock.toolCallId,
+                      toolName: contentBlock.toolName,
+                      input: contentBlock.jsonText
+                    });
+                  }
                 }
                 delete contentBlocks[blockIndex];
               }
@@ -973,11 +1030,13 @@ var BedrockChatLanguageModel = class {
                 toolName: toolUse.name,
                 jsonText: ""
               };
-              controller.enqueue({
-                type: "tool-input-start",
-                id: toolUse.toolUseId,
-                toolName: toolUse.name
-              });
+              if (!usesJsonResponseTool) {
+                controller.enqueue({
+                  type: "tool-input-start",
+                  id: toolUse.toolUseId,
+                  toolName: toolUse.name
+                });
+              }
             }
             const contentBlockDelta = value.contentBlockDelta;
             if ((contentBlockDelta == null ? void 0 : contentBlockDelta.delta) && "toolUse" in contentBlockDelta.delta && contentBlockDelta.delta.toolUse) {
@@ -985,11 +1044,13 @@ var BedrockChatLanguageModel = class {
               const contentBlock = contentBlocks[blockIndex];
               if ((contentBlock == null ? void 0 : contentBlock.type) === "tool-call") {
                 const delta = (_q = contentBlockDelta.delta.toolUse.input) != null ? _q : "";
-                controller.enqueue({
-                  type: "tool-input-delta",
-                  id: contentBlock.toolCallId,
-                  delta
-                });
+                if (!usesJsonResponseTool) {
+                  controller.enqueue({
+                    type: "tool-input-delta",
+                    id: contentBlock.toolCallId,
+                    delta
+                  });
+                }
                 contentBlock.jsonText += delta;
               }
             }