@xsai/stream-text 0.0.11 → 0.0.13

package/dist/index.d.ts

@@ -16,6 +16,7 @@ interface StreamTextResponseUsage {
     total_tokens: number;
 }
 interface StreamTextResult {
+    chunkStream: ReadableStream<StreamTextResponse>;
     finishReason?: FinishReason;
     textStream: ReadableStream<string>;
     usage?: StreamTextResponseUsage;

package/dist/index.js

@@ -10,12 +10,12 @@ const streamText = async (options) => await chatCompletion({
   const decoder = new TextDecoder();
   let finishReason;
   let usage;
-  const textStream = res.body.pipeThrough(new TransformStream({
+  const rawChunkStream = res.body.pipeThrough(new TransformStream({
     transform: (chunk, controller) => {
       for (const line of decoder.decode(chunk).split("\n").filter((line2) => line2)) {
         if (line !== "data: [DONE]") {
           const data = JSON.parse(line.slice(6));
-          controller.enqueue(data.choices[0].delta.content);
+          controller.enqueue(data);
           if (data.choices[0].finish_reason) {
             finishReason = data.choices[0].finish_reason;
           }
@@ -27,7 +27,11 @@ const streamText = async (options) => await chatCompletion({
       }
     }
   }));
-  return { finishReason, textStream, usage };
+  const [chunkStream, rawTextStream] = rawChunkStream.tee();
+  const textStream = rawTextStream.pipeThrough(new TransformStream({
+    transform: (chunk, controller) => controller.enqueue(chunk.choices[0].delta.content)
+  }));
+  return { chunkStream, finishReason, textStream, usage };
 });
 
 export { streamText as default, streamText };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@xsai/stream-text",
-  "version": "0.0.11",
+  "version": "0.0.13",
   "type": "module",
   "author": "Moeru AI",
   "license": "MIT",
@@ -25,6 +25,10 @@
   "dependencies": {
     "@xsai/shared-chat-completion": ""
   },
+  "devDependencies": {
+    "@xsai/providers": "",
+    "@xsai/shared": ""
+  },
   "scripts": {
     "build": "pkgroll",
     "build:watch": "pkgroll --watch",