Use the fetch() function provided in the Stream implementation

Author: aron

index.js

@@ -289,7 +289,11 @@ class Replicate {
 
     if (prediction.urls && prediction.urls.stream) {
       const { signal } = options;
-      const stream = new Stream(prediction.urls.stream, { signal });
+      const stream = new Stream({
+        url: prediction.urls.stream,
+        fetch: this.fetch,
+        options: { signal },
+      });
       yield* stream;
     } else {
       throw new Error("Prediction does not support streaming");

lib/stream.js

@@ -48,10 +48,12 @@ class Stream extends Readable {
   /**
    * Create a new stream of server-sent events.
    *
-   * @param {string} url The URL to connect to.
-   * @param {object} options The fetch options.
+   * @param {object} config
+   * @param {string} config.url The URL to connect to.
+   * @param {Function} [config.fetch] The fetch implemention to use.
+   * @param {object} [config.options] The fetch options.
    */
-  constructor(url, options) {
+  constructor({ url, fetch = globalThis.fetch, options = {} }) {
     if (!Readable) {
       throw new Error(
         "Readable streams are not supported. Please use Node.js 18 or later, or install the readable-stream package."
@@ -60,6 +62,7 @@ class Stream extends Readable {
 
     super();
     this.url = url;
+    this.fetch = fetch;
     this.options = options;
 
     this.event = null;
@@ -104,7 +107,7 @@ class Stream extends Readable {
   }
 
   async *[Symbol.asyncIterator]() {
-    const response = await fetch(this.url, {
+    const response = await this.fetch(this.url, {
       ...this.options,
       headers: {
         Accept: "text/event-stream",