@ai-sdk/amazon-bedrock 3.0.0-canary.7 → 3.0.0-canary.9

package/dist/index.mjs

@@ -11,9 +11,9 @@ import {
   combineHeaders,
   createJsonErrorResponseHandler,
   createJsonResponseHandler,
+  parseProviderOptions,
   postJsonToApi,
-  resolve,
-  parseProviderOptions
+  resolve
 } from "@ai-sdk/provider-utils";
 import { z as z3 } from "zod";
 
@@ -495,8 +495,6 @@ var BedrockChatLanguageModel = class {
     this.config = config;
     this.specificationVersion = "v2";
     this.provider = "amazon-bedrock";
-    this.defaultObjectGenerationMode = "tool";
-    this.supportsImageUrls = false;
   }
   getArgs({
     prompt,
@@ -605,8 +603,13 @@ var BedrockChatLanguageModel = class {
       warnings: [...warnings, ...toolWarnings]
     };
   }
+  async getSupportedUrls() {
+    return {
+      // no supported urls for bedrock
+    };
+  }
   async doGenerate(options) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n;
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m;
     const { command: args, warnings } = this.getArgs(options);
     const url = `${this.getUrl(this.modelId)}/converse`;
     const { value: response, responseHeaders } = await postJsonToApi({
@@ -629,71 +632,68 @@ var BedrockChatLanguageModel = class {
       abortSignal: options.abortSignal,
       fetch: this.config.fetch
     });
-    const providerMetadata = response.trace || response.usage ? {
-      bedrock: {
-        ...response.trace && typeof response.trace === "object" ? { trace: response.trace } : {},
-        ...response.usage && {
-          usage: {
-            cacheReadInputTokens: (_b = (_a = response.usage) == null ? void 0 : _a.cacheReadInputTokens) != null ? _b : Number.NaN,
-            cacheWriteInputTokens: (_d = (_c = response.usage) == null ? void 0 : _c.cacheWriteInputTokens) != null ? _d : Number.NaN
-          }
-        }
+    const content = [];
+    for (const part of response.output.message.content) {
+      if (part.text) {
+        content.push({ type: "text", text: part.text });
       }
-    } : void 0;
-    const reasoning = [];
-    for (const content of response.output.message.content) {
-      if (content.reasoningContent) {
-        if ("reasoningText" in content.reasoningContent) {
-          reasoning.push({
+      if (part.reasoningContent) {
+        if ("reasoningText" in part.reasoningContent) {
+          content.push({
             type: "reasoning",
             reasoningType: "text",
-            text: content.reasoningContent.reasoningText.text
+            text: part.reasoningContent.reasoningText.text
           });
-          if (content.reasoningContent.reasoningText.signature) {
-            reasoning.push({
+          if (part.reasoningContent.reasoningText.signature) {
+            content.push({
               type: "reasoning",
               reasoningType: "signature",
-              signature: content.reasoningContent.reasoningText.signature
+              signature: part.reasoningContent.reasoningText.signature
             });
           }
-        } else if ("redactedReasoning" in content.reasoningContent) {
-          reasoning.push({
+        } else if ("redactedReasoning" in part.reasoningContent) {
+          content.push({
             type: "reasoning",
             reasoningType: "redacted",
-            data: (_e = content.reasoningContent.redactedReasoning.data) != null ? _e : ""
+            data: (_a = part.reasoningContent.redactedReasoning.data) != null ? _a : ""
           });
         }
       }
-    }
-    const text = (_h = (_g = (_f = response.output) == null ? void 0 : _f.message) == null ? void 0 : _g.content) == null ? void 0 : _h.map((part) => {
-      var _a2;
-      return (_a2 = part.text) != null ? _a2 : "";
-    }).join("");
-    return {
-      text: text != null ? { type: "text", text } : void 0,
-      toolCalls: (_l = (_k = (_j = (_i = response.output) == null ? void 0 : _i.message) == null ? void 0 : _j.content) == null ? void 0 : _k.filter((part) => !!part.toolUse)) == null ? void 0 : _l.map((part) => {
-        var _a2, _b2, _c2, _d2, _e2, _f2;
-        return {
+      if (part.toolUse) {
+        content.push({
           type: "tool-call",
           toolCallType: "function",
-          toolCallId: (_b2 = (_a2 = part.toolUse) == null ? void 0 : _a2.toolUseId) != null ? _b2 : this.config.generateId(),
-          toolName: (_d2 = (_c2 = part.toolUse) == null ? void 0 : _c2.name) != null ? _d2 : `tool-${this.config.generateId()}`,
-          args: JSON.stringify((_f2 = (_e2 = part.toolUse) == null ? void 0 : _e2.input) != null ? _f2 : "")
-        };
-      }),
+          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()}`,
+          args: JSON.stringify((_g = (_f = part.toolUse) == null ? void 0 : _f.input) != null ? _g : "")
+        });
+      }
+    }
+    const providerMetadata = response.trace || response.usage ? {
+      bedrock: {
+        ...response.trace && typeof response.trace === "object" ? { trace: response.trace } : {},
+        ...response.usage && {
+          usage: {
+            cacheReadInputTokens: (_i = (_h = response.usage) == null ? void 0 : _h.cacheReadInputTokens) != null ? _i : Number.NaN,
+            cacheWriteInputTokens: (_k = (_j = response.usage) == null ? void 0 : _j.cacheWriteInputTokens) != null ? _k : Number.NaN
+          }
+        }
+      }
+    } : void 0;
+    return {
+      content,
       finishReason: mapBedrockFinishReason(
         response.stopReason
       ),
       usage: {
-        inputTokens: (_m = response.usage) == null ? void 0 : _m.inputTokens,
-        outputTokens: (_n = response.usage) == null ? void 0 : _n.outputTokens
+        inputTokens: (_l = response.usage) == null ? void 0 : _l.inputTokens,
+        outputTokens: (_m = response.usage) == null ? void 0 : _m.outputTokens
       },
       response: {
         // TODO add id, timestamp, etc
         headers: responseHeaders
       },
       warnings,
-      reasoning: reasoning.length > 0 ? reasoning : void 0,
       ...providerMetadata && { providerMetadata }
     };
   }
@@ -725,6 +725,9 @@ var BedrockChatLanguageModel = class {
     return {
       stream: response.pipeThrough(
         new TransformStream({
+          start(controller) {
+            controller.enqueue({ type: "stream-start", warnings });
+          },
           transform(chunk, controller) {
             var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n;
             function enqueueError(bedrockError) {
@@ -854,8 +857,8 @@ var BedrockChatLanguageModel = class {
           }
         })
       ),
-      response: { headers: responseHeaders },
-      warnings
+      // TODO request?
+      response: { headers: responseHeaders }
     };
   }
   getUrl(modelId) {
@@ -964,14 +967,31 @@ import {
   combineHeaders as combineHeaders2,
   createJsonErrorResponseHandler as createJsonErrorResponseHandler2,
   createJsonResponseHandler as createJsonResponseHandler2,
+  parseProviderOptions as parseProviderOptions2,
   postJsonToApi as postJsonToApi2,
   resolve as resolve2
 } from "@ai-sdk/provider-utils";
+
+// src/bedrock-embedding-options.ts
 import { z as z4 } from "zod";
+var bedrockEmbeddingProviderOptions = z4.object({
+  /**
+  The number of dimensions the resulting output embeddings should have (defaults to 1024).
+  Only supported in amazon.titan-embed-text-v2:0.
+     */
+  dimensions: z4.union([z4.literal(1024), z4.literal(512), z4.literal(256)]).optional(),
+  /**
+  Flag indicating whether or not to normalize the output embeddings. Defaults to true
+  Only supported in amazon.titan-embed-text-v2:0.
+   */
+  normalize: z4.boolean().optional()
+});
+
+// src/bedrock-embedding-model.ts
+import { z as z5 } from "zod";
 var BedrockEmbeddingModel = class {
-  constructor(modelId, settings, config) {
+  constructor(modelId, config) {
     this.modelId = modelId;
-    this.settings = settings;
     this.config = config;
     this.specificationVersion = "v2";
     this.provider = "amazon-bedrock";
@@ -985,13 +1005,20 @@ var BedrockEmbeddingModel = class {
   async doEmbed({
     values,
     headers,
-    abortSignal
+    abortSignal,
+    providerOptions
   }) {
+    var _a;
+    const bedrockOptions = (_a = parseProviderOptions2({
+      provider: "bedrock",
+      providerOptions,
+      schema: bedrockEmbeddingProviderOptions
+    })) != null ? _a : {};
     const embedSingleText = async (inputText) => {
       const args = {
         inputText,
-        dimensions: this.settings.dimensions,
-        normalize: this.settings.normalize
+        dimensions: bedrockOptions.dimensions,
+        normalize: bedrockOptions.normalize
       };
       const url = this.getUrl(this.modelId);
       const { value: response } = await postJsonToApi2({
@@ -1026,9 +1053,9 @@ var BedrockEmbeddingModel = class {
     );
   }
 };
-var BedrockEmbeddingResponseSchema = z4.object({
-  embedding: z4.array(z4.number()),
-  inputTextTokenCount: z4.number()
+var BedrockEmbeddingResponseSchema = z5.object({
+  embedding: z5.array(z5.number()),
+  inputTextTokenCount: z5.number()
 });
 
 // src/bedrock-image-model.ts
@@ -1046,7 +1073,7 @@ var modelMaxImagesPerCall = {
 };
 
 // src/bedrock-image-model.ts
-import { z as z5 } from "zod";
+import { z as z6 } from "zod";
 var BedrockImageModel = class {
   constructor(modelId, settings, config) {
     this.modelId = modelId;
@@ -1128,8 +1155,8 @@ var BedrockImageModel = class {
     };
   }
 };
-var bedrockImageResponseSchema = z5.object({
-  images: z5.array(z5.string())
+var bedrockImageResponseSchema = z6.object({
+  images: z6.array(z6.string())
 });
 
 // src/headers-utils.ts
@@ -1271,9 +1298,9 @@ function createAmazonBedrock(options = {}) {
     }
     return createChatModel(modelId);
   };
-  const createEmbeddingModel = (modelId, settings = {}) => {
+  const createEmbeddingModel = (modelId) => {
     var _a;
-    return new BedrockEmbeddingModel(modelId, settings, {
+    return new BedrockEmbeddingModel(modelId, {
       baseUrl: getBaseUrl,
       headers: (_a = options.headers) != null ? _a : {},
       fetch: sigv4Fetch