@xsai/stream-text 0.4.0-beta.8 → 0.4.0

package/README.md

@@ -0,0 +1 @@
+https://xsai.js.org/docs/packages/stream/text

package/dist/index.d.ts

@@ -50,6 +50,7 @@ interface StreamTextOptions extends ChatOptions {
 interface StreamTextResult {
     fullStream: ReadableStream<StreamTextEvent>;
     messages: Promise<Message[]>;
+    reasoningTextStream: ReadableStream<string>;
     steps: Promise<CompletionStep[]>;
     textStream: ReadableStream<string>;
     totalUsage: Promise<undefined | Usage>;

package/dist/index.js

@@ -52,8 +52,10 @@ const streamText = (options) => {
   const resultTotalUsage = new DelayedPromise();
   let eventCtrl;
   let textCtrl;
+  let reasoningTextCtrl;
   const eventStream = new ReadableStream({ start: (controller) => eventCtrl = controller });
   const textStream = new ReadableStream({ start: (controller) => textCtrl = controller });
+  const reasoningTextStream = new ReadableStream({ start: (controller) => reasoningTextCtrl = controller });
   const pushEvent = (stepEvent) => {
     eventCtrl?.enqueue(stepEvent);
     void options.onEvent?.(stepEvent);
@@ -79,10 +81,17 @@ const streamText = (options) => {
       } : { ...u };
     };
     let text = "";
+    let reasoningText;
     const pushText = (content) => {
       textCtrl?.enqueue(content);
       text += content;
     };
+    const pushReasoningText = (reasoningContent) => {
+      if (reasoningText == null)
+        reasoningText = "";
+      reasoningTextCtrl?.enqueue(reasoningContent);
+      reasoningText += reasoningContent;
+    };
     const tool_calls = [];
     const toolCalls = [];
     const toolResults = [];
@@ -101,8 +110,10 @@ const streamText = (options) => {
         if (chunk.choices == null || chunk.choices.length === 0)
           return;
         const choice = chunk.choices[0];
-        if (choice.delta.reasoning_content != null)
+        if (choice.delta.reasoning_content != null) {
           pushEvent({ text: choice.delta.reasoning_content, type: "reasoning-delta" });
+          pushReasoningText(choice.delta.reasoning_content);
+        }
         if (choice.finish_reason != null)
           finishReason = choice.finish_reason;
         if (choice.delta.tool_calls?.length === 0 || choice.delta.tool_calls == null) {
@@ -118,17 +129,33 @@ const streamText = (options) => {
           for (const toolCall of choice.delta.tool_calls) {
             const { index } = toolCall;
             if (!tool_calls.at(index)) {
-              tool_calls[index] = toolCall;
+              tool_calls[index] = {
+                ...toolCall,
+                function: {
+                  ...toolCall.function,
+                  arguments: toolCall.function.arguments ?? ""
+                }
+              };
               pushEvent({ toolCallId: toolCall.id, toolName: toolCall.function.name, type: "tool-call-streaming-start" });
             } else {
               tool_calls[index].function.arguments += toolCall.function.arguments;
-              pushEvent({ argsTextDelta: toolCall.function.arguments, toolCallId: toolCall.id, toolName: toolCall.function.name, type: "tool-call-delta" });
+              pushEvent({
+                argsTextDelta: toolCall.function.arguments,
+                toolCallId: toolCall.id,
+                toolName: toolCall.function.name ?? tool_calls[index].function.name,
+                type: "tool-call-delta"
+              });
             }
           }
         }
       }
     }));
-    messages.push({ content: text, role: "assistant", tool_calls });
+    messages.push({
+      content: text,
+      reasoning_content: reasoningText,
+      role: "assistant",
+      tool_calls: tool_calls.length > 0 ? tool_calls : void 0
+    });
     if (tool_calls.length !== 0) {
       for (const toolCall of tool_calls) {
         if (toolCall == null)
@@ -168,9 +195,11 @@ const streamText = (options) => {
       await trampoline(async () => doStream());
       eventCtrl?.close();
       textCtrl?.close();
+      reasoningTextCtrl?.close();
     } catch (err) {
       eventCtrl?.error(err);
       textCtrl?.error(err);
+      reasoningTextCtrl?.error(err);
       resultSteps.reject(err);
       resultMessages.reject(err);
       resultUsage.reject(err);
@@ -186,6 +215,7 @@ const streamText = (options) => {
   return {
     fullStream: eventStream,
     messages: resultMessages.promise,
+    reasoningTextStream,
     steps: resultSteps.promise,
     textStream,
     totalUsage: resultTotalUsage.promise,

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@xsai/stream-text",
   "type": "module",
-  "version": "0.4.0-beta.8",
+  "version": "0.4.0",
   "description": "extra-small AI SDK.",
   "author": "Moeru AI",
   "license": "MIT",
@@ -29,12 +29,12 @@
     "dist"
   ],
   "dependencies": {
-    "@xsai/shared": "~0.4.0-beta.8",
-    "@xsai/shared-chat": "~0.4.0-beta.8"
+    "@xsai/shared": "~0.4.0",
+    "@xsai/shared-chat": "~0.4.0"
   },
   "devDependencies": {
     "valibot": "^1.0.0",
-    "@xsai/tool": "~0.4.0-beta.8"
+    "@xsai/tool": "~0.4.0"
   },
   "scripts": {
     "build": "pkgroll",