ai 3.0.10 → 3.0.12

package/dist/index.d.mts

@@ -1,3 +1,4 @@
+import { AssistantStream } from 'openai/lib/AssistantStream';
 import { ChatCompletionResponseChunk } from '@mistralai/mistralai';
 import { ServerResponse } from 'node:http';
 
@@ -799,11 +800,12 @@ type AssistantResponseSettings = {
     threadId: string;
     messageId: string;
 };
-type AssistantResponseCallback = (stream: {
+type AssistantResponseCallback = (options: {
     threadId: string;
     messageId: string;
     sendMessage: (message: AssistantMessage) => void;
     sendDataMessage: (message: DataMessage) => void;
+    forwardStream: (stream: AssistantStream) => Promise<any>;
 }) => Promise<void>;
 declare function experimental_AssistantResponse({ threadId, messageId }: AssistantResponseSettings, process: AssistantResponseCallback): Response;
 

package/dist/index.d.ts

@@ -1,3 +1,4 @@
+import { AssistantStream } from 'openai/lib/AssistantStream';
 import { ChatCompletionResponseChunk } from '@mistralai/mistralai';
 import { ServerResponse } from 'node:http';
 
@@ -799,11 +800,12 @@ type AssistantResponseSettings = {
     threadId: string;
     messageId: string;
 };
-type AssistantResponseCallback = (stream: {
+type AssistantResponseCallback = (options: {
     threadId: string;
     messageId: string;
     sendMessage: (message: AssistantMessage) => void;
     sendDataMessage: (message: DataMessage) => void;
+    forwardStream: (stream: AssistantStream) => Promise<any>;
 }) => Promise<void>;
 declare function experimental_AssistantResponse({ threadId, messageId }: AssistantResponseSettings, process: AssistantResponseCallback): Response;
 

package/dist/index.js

@@ -550,6 +550,43 @@ function experimental_AssistantResponse({ threadId, messageId }, process2) {
           textEncoder.encode(formatStreamPart("error", errorMessage))
         );
       };
+      const forwardStream = async (stream2) => {
+        var _a2, _b;
+        let result = void 0;
+        for await (const value of stream2) {
+          switch (value.event) {
+            case "thread.message.created": {
+              controller.enqueue(
+                textEncoder.encode(
+                  formatStreamPart("assistant_message", {
+                    id: value.data.id,
+                    role: "assistant",
+                    content: [{ type: "text", text: { value: "" } }]
+                  })
+                )
+              );
+              break;
+            }
+            case "thread.message.delta": {
+              const content = (_a2 = value.data.delta.content) == null ? void 0 : _a2[0];
+              if ((content == null ? void 0 : content.type) === "text" && ((_b = content.text) == null ? void 0 : _b.value) != null) {
+                controller.enqueue(
+                  textEncoder.encode(
+                    formatStreamPart("text", content.text.value)
+                  )
+                );
+              }
+              break;
+            }
+            case "thread.run.completed":
+            case "thread.run.requires_action": {
+              result = value.data;
+              break;
+            }
+          }
+        }
+        return result;
+      };
       controller.enqueue(
         textEncoder.encode(
           formatStreamPart("assistant_control_data", {
@@ -563,7 +600,8 @@ function experimental_AssistantResponse({ threadId, messageId }, process2) {
           threadId,
           messageId,
           sendMessage,
-          sendDataMessage
+          sendDataMessage,
+          forwardStream
         });
       } catch (error) {
         sendError((_a = error.message) != null ? _a : `${error}`);