Fix stream consumption in the browser environment

Browsers don't yet support async iteration of ReadableStream so we need to do this manually

Author: aron

index.js

@@ -296,7 +296,19 @@ class Replicate {
         fetch: this.fetch,
         options: { signal },
       });
-      yield* stream;
+
+      const reader = stream.getReader();
+
+      while (true) {
+        const { done, value } = await reader.read();
+
+        if (done) {
+          break;
+        }
+
+        yield value;
+      }
+      reader.releaseLock();
     } else {
       throw new Error("Prediction does not support streaming");
     }

lib/stream.js

@@ -73,7 +73,15 @@ function createReadableStream({ url, fetch, options = {} }) {
         .pipeThrough(new TextDecoderStream())
         .pipeThrough(new EventSourceParserStream());
 
-      for await (const event of stream) {
+      const reader = stream.getReader();
+
+      while (true) {
+        const { done, value: event } = await reader.read();
+
+        if (done) {
+          break;
+        }
+
         if (event.event === "error") {
           controller.error(new Error(event.data));
           break;
@@ -88,6 +96,7 @@ function createReadableStream({ url, fetch, options = {} }) {
         }
       }
 
+      reader.releaseLock();
       controller.close();
     },
   });