ai-sdk-graph 0.1.3 → 0.1.4

package/dist/index.js

@@ -98,8 +98,12 @@ class Graph {
     let context;
     return createUIMessageStream({
       execute: async ({ writer }) => {
-        context = await this.createExecutionContext(runId, initialState, writer);
-        await this.onStart({ state: context.state, writer });
+        const result = await this.createExecutionContext(runId, initialState, writer);
+        context = result.context;
+        const firstTime = result.firstTime;
+        if (firstTime) {
+          await this.onStart({ state: context.state, writer });
+        }
         await this.runExecutionLoop(context);
       },
       onFinish: async () => {
@@ -110,7 +114,7 @@ class Graph {
     });
   }
   async executeInternal(runId, initialState, writer) {
-    const context = await this.createExecutionContext(runId, initialState, writer);
+    const { context } = await this.createExecutionContext(runId, initialState, writer);
     await this.runExecutionLoopInternal(context);
     return context.state;
   }
@@ -124,8 +128,8 @@ class Graph {
     this.edgeRegistry.set(edge.from, existingEdges);
   }
   async createExecutionContext(runId, initialState, writer) {
-    const restored = await this.restoreCheckpoint(runId, initialState);
-    return { runId, writer, ...restored };
+    const { context, firstTime } = await this.restoreCheckpoint(runId, initialState);
+    return { context: { ...context, runId, writer }, firstTime };
   }
   async runExecutionLoop(context) {
     await this.executeWithStrategy(context, "return");
@@ -177,9 +181,9 @@ class Graph {
   async restoreCheckpoint(runId, initialState) {
     const checkpoint = await this.storage.load(runId);
     if (this.isValidCheckpoint(checkpoint)) {
-      return this.restoreFromCheckpoint(checkpoint, initialState);
+      return { context: this.restoreFromCheckpoint(checkpoint, initialState), firstTime: false };
     }
-    return this.createFreshExecution(initialState);
+    return { context: this.createFreshExecution(initialState), firstTime: true };
   }
   isValidCheckpoint(checkpoint) {
     return this.hasNodeIds(checkpoint) && this.hasAtLeastOneNode(checkpoint);

package/package.json

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