ai 3.2.30 → 3.2.32

package/dist/index.d.mts

@@ -2098,6 +2098,10 @@ type LangChainMessageContent = string | LangChainMessageContentComplex[];
 type LangChainAIMessageChunk = {
     content: LangChainMessageContent;
 };
+type LangChainStreamEvent = {
+    event: string;
+    data: any;
+};
 /**
 Converts LangChain output streams to AIStream.
 
@@ -2105,7 +2109,7 @@ The following streams are supported:
 - `LangChainAIMessageChunk` streams (LangChain `model.stream` output)
 - `string` streams (LangChain `StringOutputParser` output)
  */
-declare function toAIStream(stream: ReadableStream<LangChainAIMessageChunk> | ReadableStream<string>, callbacks?: AIStreamCallbacksAndOptions): ReadableStream<any>;
+declare function toAIStream(stream: ReadableStream<LangChainStreamEvent> | ReadableStream<LangChainAIMessageChunk> | ReadableStream<string>, callbacks?: AIStreamCallbacksAndOptions): ReadableStream<any>;
 
 declare const langchainAdapter_toAIStream: typeof toAIStream;
 declare namespace langchainAdapter {

package/dist/index.d.ts

@@ -2098,6 +2098,10 @@ type LangChainMessageContent = string | LangChainMessageContentComplex[];
 type LangChainAIMessageChunk = {
     content: LangChainMessageContent;
 };
+type LangChainStreamEvent = {
+    event: string;
+    data: any;
+};
 /**
 Converts LangChain output streams to AIStream.
 
@@ -2105,7 +2109,7 @@ The following streams are supported:
 - `LangChainAIMessageChunk` streams (LangChain `model.stream` output)
 - `string` streams (LangChain `StringOutputParser` output)
  */
-declare function toAIStream(stream: ReadableStream<LangChainAIMessageChunk> | ReadableStream<string>, callbacks?: AIStreamCallbacksAndOptions): ReadableStream<any>;
+declare function toAIStream(stream: ReadableStream<LangChainStreamEvent> | ReadableStream<LangChainAIMessageChunk> | ReadableStream<string>, callbacks?: AIStreamCallbacksAndOptions): ReadableStream<any>;
 
 declare const langchainAdapter_toAIStream: typeof toAIStream;
 declare namespace langchainAdapter {

package/dist/index.js

@@ -3201,23 +3201,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) {