@xsai/stream-text 0.0.20 → 0.0.22

package/dist/index.d.ts

@@ -1,4 +1,4 @@
-import { ChatCompletionOptions, FinishReason } from '@xsai/shared-chat-completion';
+import { ChatCompletionOptions, FinishReason } from '@xsai/shared-chat';
 
 interface StreamTextOptions extends ChatCompletionOptions {
     /** if you want to disable stream, use `@xsai/generate-text` */

package/dist/index.js

@@ -1,4 +1,4 @@
-import { chatCompletion } from '@xsai/shared-chat-completion';
+import { chatCompletion } from '@xsai/shared-chat';
 
 const dataHeaderPrefix = "data: ";
 const dataErrorPrefix = `{"error":`;
@@ -12,9 +12,13 @@ const streamText = async (options) => await chatCompletion({
   const decoder = new TextDecoder();
   let finishReason;
   let usage;
+  let buffer = "";
   const rawChunkStream = res.body.pipeThrough(new TransformStream({
     transform: (chunk, controller) => {
-      for (const line of decoder.decode(chunk).split("\n").filter((line2) => line2)) {
+      buffer += decoder.decode(chunk);
+      const lines = buffer.split("\n\n");
+      buffer = lines.pop() || "";
+      for (const line of lines) {
         if (!line || !line.startsWith(dataHeaderPrefix)) {
           continue;
         }
@@ -27,7 +31,7 @@ const streamText = async (options) => await chatCompletion({
           controller.terminate();
           continue;
         }
-        const data = JSON.parse(line.slice(6));
+        const data = JSON.parse(lineWithoutPrefix);
         controller.enqueue(data);
         if (data.choices[0].finish_reason) {
           finishReason = data.choices[0].finish_reason;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@xsai/stream-text",
-  "version": "0.0.20",
+  "version": "0.0.22",
   "type": "module",
   "author": "Moeru AI",
   "license": "MIT",
@@ -30,7 +30,7 @@
     "dist"
   ],
   "dependencies": {
-    "@xsai/shared-chat-completion": ""
+    "@xsai/shared-chat": ""
   },
   "devDependencies": {
     "@xsai/providers": "",