@xsai/stream-transcription 0.4.4 → 0.5.0-beta.2

package/dist/index.js

@@ -1,49 +1,9 @@
 import { DelayedPromise, requestURL, requestHeaders, responseCatch } from '@xsai/shared';
-
-const parseChunk = (text) => {
-  if (!text || !text.startsWith("data:"))
-    return [void 0, false];
-  const content = text.slice("data:".length);
-  const data = content.startsWith(" ") ? content.slice(1) : content;
-  if (data.includes("[DONE]")) {
-    return [void 0, true];
-  }
-  if (data.startsWith("{") && data.includes('"error":')) {
-    throw new Error(`Error from server: ${data}`);
-  }
-  const chunk = JSON.parse(data);
-  return [chunk, false];
-};
-const transformChunk = () => {
-  const decoder = new TextDecoder();
-  let buffer = "";
-  return new TransformStream({
-    transform: async (chunk, controller) => {
-      const text = decoder.decode(chunk, { stream: true });
-      buffer += text;
-      const lines = buffer.split("\n");
-      buffer = lines.pop() ?? "";
-      for (const line of lines) {
-        try {
-          const [chunk2, isEnd] = parseChunk(line);
-          if (isEnd)
-            break;
-          if (chunk2) {
-            controller.enqueue(chunk2);
-          }
-        } catch (error) {
-          controller.error(error);
-        }
-      }
-    }
-  });
-};
+import { createControlledStream, closeControllers, errorControllers, EventSourceParserStream, JsonMessageTransformStream } from '@xsai/shared-stream';
 
 const streamTranscription = (options) => {
-  let textStreamCtrl;
-  let fullStreamCtrl;
-  const fullStream = new ReadableStream({ start: (controller) => fullStreamCtrl = controller });
-  const textStream = new ReadableStream({ start: (controller) => textStreamCtrl = controller });
+  const [textStream, textStreamCtrl] = createControlledStream();
+  const [fullStream, fullStreamCtrl] = createControlledStream();
   const fullText = new DelayedPromise();
   let text = "";
   const doStream = async () => {
@@ -65,18 +25,17 @@ const streamTranscription = (options) => {
     });
     await responseCatch(response);
     const { body: stream } = response;
-    await stream.pipeThrough(transformChunk()).pipeTo(new WritableStream({
+    await stream.pipeThrough(new TextDecoderStream()).pipeThrough(new EventSourceParserStream()).pipeThrough(new JsonMessageTransformStream()).pipeTo(new WritableStream({
       abort: (reason) => {
-        fullStreamCtrl?.error(reason);
-        textStreamCtrl?.error(reason);
+        errorControllers(reason, fullStreamCtrl, textStreamCtrl);
       },
       close: () => {
       },
       write: (chunk) => {
         if (chunk.type === "transcript.text.delta") {
-          textStreamCtrl?.enqueue(chunk.delta);
+          textStreamCtrl.current?.enqueue(chunk.delta);
           text += chunk.delta;
-          fullStreamCtrl?.enqueue(chunk);
+          fullStreamCtrl.current?.enqueue(chunk);
         } else if (chunk.type === "transcript.text.done") ;
       }
     }));
@@ -85,11 +44,9 @@ const streamTranscription = (options) => {
     try {
       await doStream();
       fullText.resolve(text);
-      fullStreamCtrl?.close();
-      textStreamCtrl?.close();
+      closeControllers(fullStreamCtrl, textStreamCtrl);
     } catch (err) {
-      fullStreamCtrl?.error(err);
-      textStreamCtrl?.error(err);
+      errorControllers(err, fullStreamCtrl, textStreamCtrl);
       fullText.reject(err);
     }
   })();

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@xsai/stream-transcription",
   "type": "module",
-  "version": "0.4.4",
+  "version": "0.5.0-beta.2",
   "description": "extra-small AI SDK.",
   "author": "Moeru AI",
   "license": "MIT",
@@ -29,7 +29,8 @@
     "dist"
   ],
   "dependencies": {
-    "@xsai/shared": "~0.4.4"
+    "@xsai/shared": "~0.5.0-beta.2",
+    "@xsai/shared-stream": "~0.5.0-beta.2"
   },
   "scripts": {
     "build": "pkgroll",