@xsai/stream-text 0.4.0-beta.9 → 0.4.1

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

@@ -46,14 +46,17 @@ const streamText = (options) => {
   const maxSteps = options.maxSteps ?? 1;
   let usage;
   let totalUsage;
+  let reasoningField;
   const resultSteps = new DelayedPromise();
   const resultMessages = new DelayedPromise();
   const resultUsage = new DelayedPromise();
   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);
@@ -76,13 +79,20 @@ const streamText = (options) => {
         completion_tokens: totalUsage.completion_tokens + u.completion_tokens,
         prompt_tokens: totalUsage.prompt_tokens + u.prompt_tokens,
         total_tokens: totalUsage.total_tokens + u.total_tokens
-      } : { ...u };
+      } : 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 +111,17 @@ 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 != null) {
+          if (reasoningField !== "reasoning")
+            reasoningField = "reasoning";
+          pushEvent({ text: choice.delta.reasoning, type: "reasoning-delta" });
+          pushReasoningText(choice.delta.reasoning);
+        } else if (choice.delta.reasoning_content != null) {
+          if (reasoningField !== "reasoning_content")
+            reasoningField = "reasoning_content";
           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) {
@@ -122,19 +141,29 @@ const streamText = (options) => {
                 ...toolCall,
                 function: {
                   ...toolCall.function,
-                  arguments: toolCall.function.arguments
+                  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 ?? tool_calls[index].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({
+      ...reasoningField != null ? { [reasoningField]: reasoningText } : {},
+      content: text,
+      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)
@@ -174,9 +203,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);
@@ -192,6 +223,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.9",
+  "version": "0.4.1",
   "description": "extra-small AI SDK.",
   "author": "Moeru AI",
   "license": "MIT",
@@ -29,12 +29,12 @@
     "dist"
   ],
   "dependencies": {
-    "@xsai/shared": "~0.4.0-beta.9",
-    "@xsai/shared-chat": "~0.4.0-beta.9"
+    "@xsai/shared": "~0.4.1",
+    "@xsai/shared-chat": "~0.4.1"
   },
   "devDependencies": {
     "valibot": "^1.0.0",
-    "@xsai/tool": "~0.4.0-beta.9"
+    "@xsai/tool": "~0.4.1"
   },
   "scripts": {
     "build": "pkgroll",