From 9515f6035e8b79a5335286c8cc1591d83ee51216 Mon Sep 17 00:00:00 2001
From: Robert Craigie <robert@craigie.dev>
Date: Tue, 23 Jul 2024 10:31:19 +0100
Subject: [PATCH] chore(bedrock): use `chunk` for internal SSE parsing instead
 of `completion` (#472)

---
 packages/bedrock-sdk/src/streaming.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/packages/bedrock-sdk/src/streaming.ts b/packages/bedrock-sdk/src/streaming.ts
index 066ec5cb..01f0956b 100644
--- a/packages/bedrock-sdk/src/streaming.ts
+++ b/packages/bedrock-sdk/src/streaming.ts
@@ -47,7 +47,7 @@ export class Stream<Item> extends CoreStream<Item> {
       for await (const event of eventStream) {
         if (event.chunk && event.chunk.bytes) {
           const s = toUtf8(event.chunk.bytes);
-          yield { event: 'completion', data: s, raw: [] };
+          yield { event: 'chunk', data: s, raw: [] };
         } else if (event.internalServerException) {
           yield { event: 'error', data: 'InternalServerException', raw: [] };
         } else if (event.modelStreamErrorException) {
@@ -69,7 +69,7 @@ export class Stream<Item> extends CoreStream<Item> {
       let done = false;
       try {
         for await (const sse of iterMessages()) {
-          if (sse.event === 'completion') {
+          if (sse.event === 'chunk') {
             try {
               yield JSON.parse(sse.data);
             } catch (e) {