@cadenza.io/core 1.11.16 → 1.12.0

package/dist/index.d.mts

@@ -232,12 +232,15 @@ declare class GraphNode extends SignalEmitter implements Graph {
     emitWithMetadata(signal: string, ctx: AnyObject): void;
     emitMetricsWithMetadata(signal: string, ctx: AnyObject): void;
     onProgress(progress: number): void;
-    postProcess(): void;
+    postProcess(): GraphNode[] | Promise<GraphNode[]>;
+    postProcessAsync(nextNodes: Promise<GraphNode[]>): Promise<GraphNode[]>;
+    finalize(): void;
     onError(error: unknown, errorData?: AnyObject): void;
     retry(prevResult?: any): Promise<TaskResult>;
     retryAsync(prevResult?: any): Promise<TaskResult>;
     delayRetry(): Promise<void>;
-    divide(): GraphNode[];
+    divide(): GraphNode[] | Promise<GraphNode[]>;
+    divideAsync(current: Promise<IteratorResult<any>>): Promise<GraphNode[]>;
     generateNewNodes(result: any): GraphNode[];
     differentiate(task: Task): GraphNode;
     migrate(ctx: any): GraphNode;

package/dist/index.d.ts

@@ -232,12 +232,15 @@ declare class GraphNode extends SignalEmitter implements Graph {
     emitWithMetadata(signal: string, ctx: AnyObject): void;
     emitMetricsWithMetadata(signal: string, ctx: AnyObject): void;
     onProgress(progress: number): void;
-    postProcess(): void;
+    postProcess(): GraphNode[] | Promise<GraphNode[]>;
+    postProcessAsync(nextNodes: Promise<GraphNode[]>): Promise<GraphNode[]>;
+    finalize(): void;
     onError(error: unknown, errorData?: AnyObject): void;
     retry(prevResult?: any): Promise<TaskResult>;
     retryAsync(prevResult?: any): Promise<TaskResult>;
     delayRetry(): Promise<void>;
-    divide(): GraphNode[];
+    divide(): GraphNode[] | Promise<GraphNode[]>;
+    divideAsync(current: Promise<IteratorResult<any>>): Promise<GraphNode[]>;
     generateNewNodes(result: any): GraphNode[];
     differentiate(task: Task): GraphNode;
     migrate(ctx: any): GraphNode;

package/dist/index.js

@@ -934,7 +934,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;
   }
@@ -950,7 +954,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() {
@@ -1036,7 +1044,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();
     }
@@ -1093,6 +1114,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) {
@@ -1136,6 +1160,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 = (0, import_uuid3.v4)();
     const newNodes = [];