@ai-sdk/amazon-bedrock 4.0.24 → 4.0.26

package/src/bedrock-event-stream-response-handler.ts

@@ -0,0 +1,57 @@
+import { EmptyResponseBodyError } from '@ai-sdk/provider';
+import {
+  ParseResult,
+  safeParseJSON,
+  extractResponseHeaders,
+  ResponseHandler,
+  safeValidateTypes,
+} from '@ai-sdk/provider-utils';
+import { ZodType } from 'zod/v4';
+import { createBedrockEventStreamDecoder } from './bedrock-event-stream-decoder';
+
+export const createBedrockEventStreamResponseHandler =
+  <T>(
+    chunkSchema: ZodType<T, any>,
+  ): ResponseHandler<ReadableStream<ParseResult<T>>> =>
+  async ({ response }: { response: Response }) => {
+    const responseHeaders = extractResponseHeaders(response);
+
+    if (response.body == null) {
+      throw new EmptyResponseBodyError({});
+    }
+
+    return {
+      responseHeaders,
+      value: createBedrockEventStreamDecoder<ParseResult<T>>(
+        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 as any).p;
+            const wrappedData = {
+              [event.eventType]: parsedDataResult.value,
+            };
+
+            const validatedWrappedData = await safeValidateTypes<T>({
+              value: wrappedData,
+              schema: chunkSchema,
+            });
+            if (!validatedWrappedData.success) {
+              controller.enqueue(validatedWrappedData);
+            } else {
+              controller.enqueue({
+                success: true,
+                value: validatedWrappedData.value,
+                rawValue: wrappedData,
+              });
+            }
+          }
+        },
+      ),
+    };
+  };