@ai-sdk/amazon-bedrock 3.0.118 → 3.0.120

package/dist/index.mjs

@@ -8,7 +8,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "3.0.118" : "0.0.0-test";
+var VERSION = true ? "3.0.120" : "0.0.0-test";
 
 // src/bedrock-provider.ts
 import { anthropicTools as anthropicTools2 } from "@ai-sdk/anthropic/internal";
@@ -131,7 +131,7 @@ function createBedrockEventStreamDecoder(body, processEvent) {
   return body.pipeThrough(
     new TransformStream({
       async transform(chunk, controller) {
-        var _a, _b;
+        var _a, _b, _c;
         const newBuffer = new Uint8Array(buffer.length + chunk.length);
         newBuffer.set(buffer);
         newBuffer.set(chunk, buffer.length);
@@ -150,8 +150,12 @@ function createBedrockEventStreamDecoder(body, processEvent) {
           buffer = buffer.slice(totalLength);
           const messageType = (_a = decoded.headers[":message-type"]) == null ? void 0 : _a.value;
           const eventType = (_b = decoded.headers[":event-type"]) == null ? void 0 : _b.value;
+          const exceptionType = (_c = decoded.headers[":exception-type"]) == null ? void 0 : _c.value;
           const data = textDecoder.decode(decoded.body);
-          await processEvent({ messageType, eventType, data }, controller);
+          await processEvent(
+            { messageType, eventType, exceptionType, data },
+            controller
+          );
         }
       },
       flush() {
@@ -176,29 +180,31 @@ var createBedrockEventStreamResponseHandler = (chunkSchema) => async ({ response
     value: createBedrockEventStreamDecoder(
       response.body,
       async (event, controller) => {
-        if (event.messageType === "event") {
-          const parsedDataResult = await safeParseJSON({ text: event.data });
-          if (!parsedDataResult.success) {
-            controller.enqueue(parsedDataResult);
-            return;
-          }
-          delete parsedDataResult.value.p;
-          const wrappedData = {
-            [event.eventType]: parsedDataResult.value
-          };
-          const validatedWrappedData = await safeValidateTypes({
-            value: wrappedData,
-            schema: chunkSchema
+        const payloadType = event.messageType === "event" ? event.eventType : event.messageType === "exception" ? event.exceptionType : void 0;
+        if (payloadType == null) {
+          return;
+        }
+        const parsedDataResult = await safeParseJSON({ text: event.data });
+        if (!parsedDataResult.success) {
+          controller.enqueue(parsedDataResult);
+          return;
+        }
+        delete parsedDataResult.value.p;
+        const wrappedData = {
+          [payloadType]: parsedDataResult.value
+        };
+        const validatedWrappedData = await safeValidateTypes({
+          value: wrappedData,
+          schema: chunkSchema
+        });
+        if (!validatedWrappedData.success) {
+          controller.enqueue(validatedWrappedData);
+        } else {
+          controller.enqueue({
+            success: true,
+            value: validatedWrappedData.value,
+            rawValue: wrappedData
           });
-          if (!validatedWrappedData.success) {
-            controller.enqueue(validatedWrappedData);
-          } else {
-            controller.enqueue({
-              success: true,
-              value: validatedWrappedData.value,
-              rawValue: wrappedData
-            });
-          }
         }
       }
     )
@@ -1176,6 +1182,10 @@ var BedrockChatLanguageModel = class {
               enqueueError(value.modelStreamErrorException);
               return;
             }
+            if (value.serviceUnavailableException) {
+              enqueueError(value.serviceUnavailableException);
+              return;
+            }
             if (value.throttlingException) {
               enqueueError(value.throttlingException);
               return;
@@ -1545,6 +1555,7 @@ var BedrockStreamSchema = z4.object({
     }).nullish()
   }).nullish(),
   modelStreamErrorException: z4.record(z4.string(), z4.unknown()).nullish(),
+  serviceUnavailableException: z4.record(z4.string(), z4.unknown()).nullish(),
   throttlingException: z4.record(z4.string(), z4.unknown()).nullish(),
   validationException: z4.record(z4.string(), z4.unknown()).nullish()
 });