ai-sdk-graph 0.3.0 → 0.3.2

package/dist/index.d.ts

@@ -1,3 +1,4 @@
 export * from './graph';
 export * from './storage';
 export * from './types';
+export * from './utils';

package/dist/index.js

@@ -407,8 +407,34 @@ class MermaidGenerator {
     });
   }
 }
+// src/utils.ts
+async function consumeAndMergeStream(stream, writer, options) {
+  let resolver;
+  let rejecter;
+  const messages = new Promise((resolve, reject) => {
+    resolver = resolve;
+    rejecter = reject;
+  });
+  try {
+    const uiMessageStream = stream.toUIMessageStream({
+      ...options,
+      onFinish: ({ messages: messages2 }) => {
+        resolver(messages2);
+      },
+      onError: (error) => {
+        rejecter(error);
+        return error instanceof Error ? error.message : "An error occurred.";
+      }
+    });
+    writer.merge(uiMessageStream);
+  } catch (error) {
+    rejecter(error);
+  }
+  return messages;
+}
 export {
   graph,
+  consumeAndMergeStream,
   SuspenseError,
   RedisStorage,
   InMemoryStorage,

package/dist/utils.d.ts

@@ -1,3 +1,3 @@
-import type { StreamTextResult, ToolSet, UIMessage } from 'ai';
+import type { StreamTextResult, UIMessage } from 'ai';
 import type { GraphSDK } from './types';
-export declare function consumeAndMergeStream<Stream extends StreamTextResult<ToolSet, any>>(stream: Stream, writer: GraphSDK.Writer, options?: Omit<Parameters<Stream['toUIMessageStream']>[0], 'onFinish' | 'onError'>): Promise<UIMessage<unknown, import("ai").UIDataTypes, import("ai").UITools>[]>;
+export declare function consumeAndMergeStream<Stream extends StreamTextResult<any, any>>(stream: Stream, writer: GraphSDK.Writer, options?: Omit<Parameters<Stream['toUIMessageStream']>[0], 'onFinish' | 'onError'>): Promise<UIMessage<unknown, import("ai").UIDataTypes, import("ai").UITools>[]>;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-sdk-graph",
-  "version": "0.3.0",
+  "version": "0.3.2",
   "description": "Graph-based workflows for the AI SDK",
   "type": "module",
   "main": "dist/index.js",