@cadenza.io/core 1.11.16 → 1.12.0

package/dist/index.mjs

@@ -896,7 +896,11 @@ var GraphNode = class _GraphNode extends SignalEmitter {
       if (this.result instanceof Promise) {
         return this.executeAsync();
       }
-      this.postProcess();
+      const nextNodes = this.postProcess();
+      if (nextNodes instanceof Promise) {
+        return nextNodes;
+      }
+      this.nextNodes = nextNodes;
     }
     return this.nextNodes;
   }
@@ -912,7 +916,11 @@ var GraphNode = class _GraphNode extends SignalEmitter {
   }
   async executeAsync() {
     await this.workAsync();
-    this.postProcess();
+    const nextNodes = this.postProcess();
+    if (nextNodes instanceof Promise) {
+      return nextNodes;
+    }
+    this.nextNodes = nextNodes;
     return this.nextNodes;
   }
   work() {
@@ -998,7 +1006,20 @@ var GraphNode = class _GraphNode extends SignalEmitter {
     if (Array.isArray(this.result)) {
       this.onError(`Returning arrays is not allowed. Returned: ${this.result}`);
     }
-    this.nextNodes = this.divide();
+    const nextNodes = this.divide();
+    if (nextNodes instanceof Promise) {
+      return this.postProcessAsync(nextNodes);
+    }
+    this.nextNodes = nextNodes;
+    this.finalize();
+    return this.nextNodes;
+  }
+  async postProcessAsync(nextNodes) {
+    this.nextNodes = await nextNodes;
+    this.finalize();
+    return this.nextNodes;
+  }
+  finalize() {
     if (this.nextNodes.length === 0) {
       this.completeSubgraph();
     }
@@ -1055,6 +1076,9 @@ var GraphNode = class _GraphNode extends SignalEmitter {
     if (((_a = this.result) == null ? void 0 : _a.next) && typeof this.result.next === "function") {
       const generator = this.result;
       let current = generator.next();
+      if (current instanceof Promise) {
+        return this.divideAsync(current);
+      }
       while (!current.done && current.value !== void 0) {
         const outputValidation = this.task.validateOutput(current.value);
         if (outputValidation !== true) {
@@ -1098,6 +1122,28 @@ var GraphNode = class _GraphNode extends SignalEmitter {
     });
     return newNodes;
   }
+  async divideAsync(current) {
+    const nextNodes = [];
+    const _current = await current;
+    const outputValidation = this.task.validateOutput(_current.value);
+    if (outputValidation !== true) {
+      this.onError(outputValidation.__validationErrors);
+      return nextNodes;
+    } else {
+      nextNodes.push(...this.generateNewNodes(_current.value));
+    }
+    for await (const result of this.result) {
+      const outputValidation2 = this.task.validateOutput(result);
+      if (outputValidation2 !== true) {
+        this.onError(outputValidation2.__validationErrors);
+        return [];
+      } else {
+        nextNodes.push(...this.generateNewNodes(result));
+      }
+    }
+    this.divided = true;
+    return nextNodes;
+  }
   generateNewNodes(result) {
     const groupId = uuid3();
     const newNodes = [];