@ai-sdk/amazon-bedrock 4.0.158 → 4.0.159

package/dist/index.mjs

@@ -147,7 +147,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);
@@ -166,8 +166,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() {
@@ -192,29 +196,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
-            });
-          }
         }
       }
     )
@@ -1372,6 +1378,10 @@ var BedrockChatLanguageModel = class {
               enqueueError(value.modelStreamErrorException);
               return;
             }
+            if (value.serviceUnavailableException) {
+              enqueueError(value.serviceUnavailableException);
+              return;
+            }
             if (value.throttlingException) {
               enqueueError(value.throttlingException);
               return;
@@ -1815,6 +1825,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()
 });
@@ -2244,7 +2255,7 @@ import {
 import { AwsV4Signer } from "aws4fetch";
 
 // src/version.ts
-var VERSION = true ? "4.0.158" : "0.0.0-test";
+var VERSION = true ? "4.0.159" : "0.0.0-test";
 
 // src/bedrock-sigv4-fetch.ts
 function createSigV4FetchFunction(getCredentials, fetch, service = "bedrock") {