public commit

app/src/flows/tasks/presets/SubFlowTask.ts

@@ -0,0 +1,40 @@
+import { Type } from "core/utils";
+import { Flow } from "../../flows/Flow";
+import { Task, dynamic } from "../Task";
+
+export class SubFlowTask<Output extends Record<string, any>> extends Task<
+   typeof SubFlowTask.schema,
+   Output
+> {
+   type = "subflow";
+
+   static override schema = Type.Object({
+      flow: Type.Any(),
+      input: Type.Optional(dynamic(Type.Any(), JSON.parse)),
+      loop: Type.Optional(Type.Boolean())
+   });
+
+   async execute() {
+      const flow = this.params.flow;
+      if (!(flow instanceof Flow)) {
+         throw new Error("Invalid flow provided");
+      }
+
+      if (this.params.loop) {
+         const _input = Array.isArray(this.params.input) ? this.params.input : [this.params.input];
+
+         const results: any[] = [];
+         for (const input of _input) {
+            const execution = flow.createExecution();
+            await execution.start(input);
+            results.push(await execution.getResponse());
+         }
+
+         return results;
+      }
+
+      const execution = flow.createExecution();
+      await execution.start(this.params.input);
+      return execution.getResponse();
+   }
+}