Switch to replicate.stream in Bun integration test

.github/workflows/ci.yml

@@ -139,4 +139,4 @@ jobs:
           cd integration/${{ matrix.suite }}
           bun uninstall replicate
           bun install "file:../../${{ needs.build.outputs.tarball-name }}"
-          bun test
+          bun test --timeout 30000

integration/bun/index.test.ts

@@ -19,5 +19,5 @@ import type {
 
 test("main", async () => {
   const output = await main();
-  expect(output as any).toEqual("hello Brünnhilde Bun");
+  expect(output).toContain("bun");
 });

integration/bun/index.ts

@@ -5,12 +5,19 @@ const replicate = new Replicate({
 });
 
 export default async function main() {
-  return await replicate.run(
-    "replicate/hello-world:5c7d5dc6dd8bf75c1acaa8565735e7986bc5b66206b55cca93cb72c9bf15ccaa",
-    {
-      input: {
-        text: "Brünnhilde Bun",
-      },
+  const model = "meta/llama-2-70b-chat";
+  const options = {
+    input: {
+      prompt: "Write a haiku about steam buns",
+    },
+  };
+  const output = [];
+
+  for await (const { event, data } of replicate.stream(model, options)) {
+    if (event === "output") {
+      output.push(data);
     }
-  );
+  }
+
+  return output.join("").trim();
 }

lib/stream.js

@@ -4,6 +4,10 @@ const ApiError = require("./error");
 const {
   EventSourceParserStream,
 } = require("../vendor/eventsource-parser/stream");
+const { TextDecoderStream } =
+  typeof globalThis.TextDecoderStream === "undefined"
+    ? require("../vendor/streams-text-encoding/text-decoder-stream")
+    : globalThis;
 
 /**
  * A server-sent event.

vendor/streams-text-encoding/text-decoder-stream.js

@@ -0,0 +1,95 @@
+// Adapted from  https://github.com/stardazed/sd-streams
+//
+// MIT License
+//
+// Copyright (c) 2018-Present @zenmumbler
+//
+// Permission is hereby granted, free of charge, to any person obtaining a copy
+// of this software and associated documentation files (the "Software"), to deal
+// in the Software without restriction, including without limitation the rights
+// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+// copies of the Software, and to permit persons to whom the Software is
+// furnished to do so, subject to the following conditions:
+//
+// The above copyright notice and this permission notice shall be included in all
+// copies or substantial portions of the Software.
+//
+// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+// SOFTWARE.
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// /input.ts
+var input_exports = {};
+__export(input_exports, {
+  TextDecoderStream: () => TextDecoderStream
+});
+module.exports = __toCommonJS(input_exports);
+
+// http-url:https://unpkg.com/@stardazed/streams-text-encoding@1.0.2/dist/sd-streams-text-encoding.esm.js
+var decDecoder = Symbol("decDecoder");
+var decTransform = Symbol("decTransform");
+var TextDecodeTransformer = class {
+  constructor(decoder) {
+    this.decoder_ = decoder;
+  }
+  transform(chunk, controller) {
+    if (!(chunk instanceof ArrayBuffer || ArrayBuffer.isView(chunk))) {
+      throw new TypeError("Input data must be a BufferSource");
+    }
+    const text = this.decoder_.decode(chunk, { stream: true });
+    if (text.length !== 0) {
+      controller.enqueue(text);
+    }
+  }
+  flush(controller) {
+    const text = this.decoder_.decode();
+    if (text.length !== 0) {
+      controller.enqueue(text);
+    }
+  }
+};
+var TextDecoderStream = class {
+  constructor(label, options) {
+    const decoder = new TextDecoder(label || "utf-8", options || {});
+    this[decDecoder] = decoder;
+    this[decTransform] = new TransformStream(new TextDecodeTransformer(decoder));
+  }
+  get encoding() {
+    return this[decDecoder].encoding;
+  }
+  get fatal() {
+    return this[decDecoder].fatal;
+  }
+  get ignoreBOM() {
+    return this[decDecoder].ignoreBOM;
+  }
+  get readable() {
+    return this[decTransform].readable;
+  }
+  get writable() {
+    return this[decTransform].writable;
+  }
+};
+var encEncoder = Symbol("encEncoder");
+var encTransform = Symbol("encTransform");