ai 3.2.29 → 3.2.31

package/dist/index.mjs

@@ -3132,23 +3132,38 @@ __export(langchain_adapter_exports, {
 function toAIStream(stream, callbacks) {
   return stream.pipeThrough(
     new TransformStream({
-      transform: async (chunk, controller) => {
-        if (typeof chunk === "string") {
-          controller.enqueue(chunk);
-        } else if (typeof chunk.content === "string") {
-          controller.enqueue(chunk.content);
-        } else {
-          const content = chunk.content;
-          for (const item of content) {
-            if (item.type === "text") {
-              controller.enqueue(item.text);
-            }
+      transform: async (value, controller) => {
+        var _a;
+        if (typeof value === "string") {
+          controller.enqueue(value);
+          return;
+        }
+        if ("event" in value) {
+          if (value.event === "on_chat_model_stream") {
+            forwardAIMessageChunk(
+              (_a = value.data) == null ? void 0 : _a.chunk,
+              controller
+            );
           }
+          return;
         }
+        forwardAIMessageChunk(value, controller);
       }
     })
   ).pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(createStreamDataTransformer());
 }
+function forwardAIMessageChunk(chunk, controller) {
+  if (typeof chunk.content === "string") {
+    controller.enqueue(chunk.content);
+  } else {
+    const content = chunk.content;
+    for (const item of content) {
+      if (item.type === "text") {
+        controller.enqueue(item.text);
+      }
+    }
+  }
+}
 
 // streams/langchain-stream.ts
 function LangChainStream(callbacks) {