beeai-framework 0.1.28 → 0.1.30

package/dist/workflows/agent.d.ts

@@ -4,10 +4,10 @@ import { Message, MessageContentPart } from '../backend/message.js';
 import { AnyTool } from '../tools/base.js';
 import { ReadOnlyMemory } from '../memory/base.js';
 import { z } from 'zod';
-import { a as BaseAgent } from '../base-BfhNzUYf.js';
-import { m as ChatModel } from '../chat-DTxmaKP0.js';
+import { B as BaseAgent } from '../base-BjzyQ8k6.js';
+import { a as ChatModel } from '../chat-CfTNfR60.js';
 import { ToolCallingAgentRunInput, ToolCallingAgentRunOutput, ToolCallingAgentRunOptions, ToolCallingAgentExecutionConfig } from '../agents/toolCalling/types.js';
-import '../emitter-CL56qab4.js';
+import '../emitter-kHxkUxco.js';
 import '../internals/types.js';
 import '../internals/helpers/guards.js';
 import '../internals/serializable.js';
@@ -42,28 +42,28 @@ declare class AgentWorkflow {
             context: z.ZodOptional<z.ZodString>;
             expectedOutput: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodType<z.ZodType<unknown, z.ZodTypeDef, unknown>, z.ZodTypeDef, z.ZodType<unknown, z.ZodTypeDef, unknown>>]>>;
         }, "strip", z.ZodTypeAny, {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }, {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }>, "many">;
         finalAnswer: z.ZodOptional<z.ZodString>;
         newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
     }, "strip", z.ZodTypeAny, {
         inputs: {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }[];
         newMessages: Message<MessageContentPart, string>[];
         finalAnswer?: string | undefined;
     }, {
         inputs: {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }[];
         finalAnswer?: string | undefined;
@@ -74,28 +74,28 @@ declare class AgentWorkflow {
             context: z.ZodOptional<z.ZodString>;
             expectedOutput: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodType<z.ZodType<unknown, z.ZodTypeDef, unknown>, z.ZodTypeDef, z.ZodType<unknown, z.ZodTypeDef, unknown>>]>>;
         }, "strip", z.ZodTypeAny, {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }, {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }>, "many">;
         finalAnswer: z.ZodString;
         newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
     }, "strip", z.ZodTypeAny, {
         inputs: {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }[];
         finalAnswer: string;
         newMessages: Message<MessageContentPart, string>[];
     }, {
         inputs: {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }[];
         finalAnswer: string;
@@ -107,28 +107,28 @@ declare class AgentWorkflow {
             context: z.ZodOptional<z.ZodString>;
             expectedOutput: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodType<z.ZodType<unknown, z.ZodTypeDef, unknown>, z.ZodTypeDef, z.ZodType<unknown, z.ZodTypeDef, unknown>>]>>;
         }, "strip", z.ZodTypeAny, {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }, {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }>, "many">;
         finalAnswer: z.ZodOptional<z.ZodString>;
         newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
     }, "strip", z.ZodTypeAny, {
         inputs: {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }[];
         newMessages: Message<MessageContentPart, string>[];
         finalAnswer?: string | undefined;
     }, {
         inputs: {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }[];
         finalAnswer?: string | undefined;
@@ -141,28 +141,28 @@ declare class AgentWorkflow {
             context: z.ZodOptional<z.ZodString>;
             expectedOutput: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodType<z.ZodType<unknown, z.ZodTypeDef, unknown>, z.ZodTypeDef, z.ZodType<unknown, z.ZodTypeDef, unknown>>]>>;
         }, "strip", z.ZodTypeAny, {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }, {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }>, "many">;
         finalAnswer: z.ZodOptional<z.ZodString>;
         newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
     }, "strip", z.ZodTypeAny, {
         inputs: {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }[];
         newMessages: Message<MessageContentPart, string>[];
         finalAnswer?: string | undefined;
     }, {
         inputs: {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }[];
         finalAnswer?: string | undefined;
@@ -173,28 +173,28 @@ declare class AgentWorkflow {
             context: z.ZodOptional<z.ZodString>;
             expectedOutput: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodType<z.ZodType<unknown, z.ZodTypeDef, unknown>, z.ZodTypeDef, z.ZodType<unknown, z.ZodTypeDef, unknown>>]>>;
         }, "strip", z.ZodTypeAny, {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }, {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }>, "many">;
         finalAnswer: z.ZodString;
         newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
     }, "strip", z.ZodTypeAny, {
         inputs: {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }[];
         finalAnswer: string;
         newMessages: Message<MessageContentPart, string>[];
     }, {
         inputs: {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }[];
         finalAnswer: string;
@@ -205,28 +205,28 @@ declare class AgentWorkflow {
             context: z.ZodOptional<z.ZodString>;
             expectedOutput: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodType<z.ZodType<unknown, z.ZodTypeDef, unknown>, z.ZodTypeDef, z.ZodType<unknown, z.ZodTypeDef, unknown>>]>>;
         }, "strip", z.ZodTypeAny, {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }, {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }>, "many">;
         finalAnswer: z.ZodOptional<z.ZodString>;
         newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
     }, "strip", z.ZodTypeAny, {
         inputs: {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }[];
         newMessages: Message<MessageContentPart, string>[];
         finalAnswer?: string | undefined;
     }, {
         inputs: {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }[];
         finalAnswer?: string | undefined;
@@ -237,36 +237,36 @@ declare class AgentWorkflow {
             context: z.ZodOptional<z.ZodString>;
             expectedOutput: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodType<z.ZodType<unknown, z.ZodTypeDef, unknown>, z.ZodTypeDef, z.ZodType<unknown, z.ZodTypeDef, unknown>>]>>;
         }, "strip", z.ZodTypeAny, {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }, {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }>, "many">;
         finalAnswer: z.ZodString;
         newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
     }, "strip", z.ZodTypeAny, {
         inputs: {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }[];
         finalAnswer: string;
         newMessages: Message<MessageContentPart, string>[];
     }, {
         inputs: {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }[];
         finalAnswer: string;
         newMessages?: Message<MessageContentPart, string>[] | undefined;
     }>, string>, readonly [{
         inputs: {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }[];
         finalAnswer?: string | undefined;
@@ -280,28 +280,28 @@ declare class AgentWorkflow {
             context: z.ZodOptional<z.ZodString>;
             expectedOutput: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodType<z.ZodType<unknown, z.ZodTypeDef, unknown>, z.ZodTypeDef, z.ZodType<unknown, z.ZodTypeDef, unknown>>]>>;
         }, "strip", z.ZodTypeAny, {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }, {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }>, "many">;
         finalAnswer: z.ZodOptional<z.ZodString>;
         newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
     }, "strip", z.ZodTypeAny, {
         inputs: {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }[];
         newMessages: Message<MessageContentPart, string>[];
         finalAnswer?: string | undefined;
     }, {
         inputs: {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }[];
         finalAnswer?: string | undefined;
@@ -312,28 +312,28 @@ declare class AgentWorkflow {
             context: z.ZodOptional<z.ZodString>;
             expectedOutput: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodType<z.ZodType<unknown, z.ZodTypeDef, unknown>, z.ZodTypeDef, z.ZodType<unknown, z.ZodTypeDef, unknown>>]>>;
         }, "strip", z.ZodTypeAny, {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }, {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }>, "many">;
         finalAnswer: z.ZodString;
         newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
     }, "strip", z.ZodTypeAny, {
         inputs: {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }[];
         finalAnswer: string;
         newMessages: Message<MessageContentPart, string>[];
     }, {
         inputs: {
-            context?: string | undefined;
             prompt?: string | undefined;
+            context?: string | undefined;
             expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
         }[];
         finalAnswer: string;

package/dist/workflows/agent.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/workflows/agent.ts"],"names":["AgentWorkflow","workflow","schema","z","object","inputs","array","prompt","string","optional","context","expectedOutput","union","instanceof","ZodSchema","finalAnswer","newMessages","Message","default","name","Workflow","outputSchema","required","run","options","map","input","text","addAgent","agent","BaseAgent","clone","then","factory","memory","_add","meta","randomString","isFunction","_createFactory","delAgent","delStep","ToolCallingAgent","llm","tools","description","instructions","execution","templates","system","template","fork","config","defaults","role","addStep","state","ctx","UnconstrainedMemory","addMany","runInput","shift","undefined","asReadOnly","result","signal","push","UserMessage"],"mappings":";;;;;;;;;;AAqCO,MAAMA,aAAAA,CAAAA;EArCb;;;AAsCqBC,EAAAA,QAAAA;AAEVC,EAAAA,MAAAA,GAASC,EAAEC,MAAAA,CAAO;IACzBC,MAAAA,EAAQF,CAAAA,CAAEG,KAAAA,CACRH,CAAAA,CAAEC,MAAAA,CAAO;MACPG,MAAAA,EAAQJ,CAAAA,CAAEK,MAAAA,EAAM,CAAGC,QAAAA,EAAQ;MAC3BC,OAAAA,EAASP,CAAAA,CAAEK,MAAAA,EAAM,CAAGC,QAAAA,EAAQ;AAC5BE,MAAAA,cAAAA,EAAgBR,EAAES,KAAAA,CAAM;AAACT,QAAAA,CAAAA,CAAEK,MAAAA,EAAM;QAAIL,CAAAA,CAAEU,UAAAA,CAAWV,EAAEW,SAAS;AAAE,OAAA,CAAA,CAAEL,QAAAA;AACnE,KAAA,CAAA,CAAA;IAGFM,WAAAA,EAAaZ,CAAAA,CAAEK,MAAAA,EAAM,CAAGC,QAAAA,EAAQ;IAChCO,WAAAA,EAAab,CAAAA,CAAEG,MAAMH,CAAAA,CAAEU,UAAAA,CAAWI,OAAAA,CAAAA,CAAAA,CAAUC,OAAAA,CAAQ,EAAE;GACxD,CAAA;AAEA,EAAA,WAAA,CAAYC,OAAO,eAAA,EAAiB;AAClC,IAAA,IAAA,CAAKlB,QAAAA,GAAW,IAAImB,QAAAA,CAAS;AAC3BD,MAAAA,IAAAA;AACAjB,MAAAA,MAAAA,EAAQ,IAAA,CAAKA,MAAAA;MACbmB,YAAAA,EAAc,IAAA,CAAKnB,OAAOoB,QAAAA;KAC5B,CAAA;AACF;EAEAC,GAAAA,CAAIlB,MAAAA,EAAgDmB,OAAAA,GAAsC,EAAC,EAAG;AAC5F,IAAA,OAAO,IAAA,CAAKvB,SAASsB,GAAAA,CACnB;AACElB,MAAAA,MAAAA,EAAQA,MAAAA,CAAOoB,GAAAA,CAAI,CAACC,KAAAA,KAAWA,iBAAiBT,OAAAA,GAAU;AAAEV,QAAAA,MAAAA,EAAQmB,KAAAA,CAAMC;AAAK,OAAA,GAAID,KAAAA;AACrF,KAAA,EACAF,OAAAA,CAAAA;AAEJ;AAIAI,EAAAA,QAAAA,CAASC,KAAAA,EAA+E;AACtF,IAAA,IAAIA,iBAAiBC,SAAAA,EAAW;AAC9B,MAAA,OAAOD,KAAAA,CAAME,KAAAA,EAAK,CAAGC,IAAAA,CAAK,CAACD,KAAAA,KAAAA;AACzB,QAAA,MAAME,OAAAA,2BAAyBC,MAAAA,KAAAA;AAC7BH,UAAAA,KAAAA,CAAMG,MAAAA,GAASA,MAAAA;AACf,UAAA,OAAOH,KAAAA;SACT,EAH8B,SAAA,CAAA;AAI9B,QAAA,OAAO,IAAA,CAAKI,IAAAA,CAAKJ,KAAAA,CAAMK,IAAAA,CAAKjB,MAAMc,OAAAA,CAAAA;OACpC,CAAA;AACF;AAEA,IAAA,MAAMd,OAAOU,KAAAA,CAAMV,IAAAA,IAAQ,CAAA,KAAA,EAAQkB,YAAAA,CAAa,CAAA,CAAA,CAAA,CAAA;AAChD,IAAA,OAAO,IAAA,CAAKF,IAAAA,CAAKhB,IAAAA,EAAMmB,UAAAA,CAAWT,KAAAA,IAASA,KAAAA,GAAQ,IAAA,CAAKU,cAAAA,CAAeV,KAAAA,CAAAA,CAAAA;AACzE;AAEAW,EAAAA,QAAAA,CAASrB,IAAAA,EAAc;AACrB,IAAA,OAAO,IAAA,CAAKlB,QAAAA,CAASwC,OAAAA,CAAQtB,IAAAA,CAAAA;AAC/B;AAEUoB,EAAAA,cAAAA,CAAeb,KAAAA,EAAwC;AAC/D,IAAA,OAAO,CAACQ,MAAAA,KACN,IAAIQ,gBAAAA,CAAiB;AACnBC,MAAAA,GAAAA,EAAKjB,KAAAA,CAAMiB,GAAAA;MACXC,KAAAA,EAAOlB,KAAAA,CAAMkB,SAAS,EAAA;AACtBV,MAAAA,MAAAA;MACAE,IAAAA,EAAM;AACJjB,QAAAA,IAAAA,EAAMO,KAAAA,CAAMP,IAAAA,IAAQ,CAAA,KAAA,EAAQkB,YAAAA,CAAa,CAAA,CAAA,CAAA,CAAA;AACzCQ,QAAAA,WAAAA,EAAanB,MAAMoB,YAAAA,IAAgB;AACrC,OAAA;AACAC,MAAAA,SAAAA,EAAWrB,KAAAA,CAAMqB,SAAAA;AACjB,MAAA,GAAIrB,MAAMoB,YAAAA,IAAgB;QACxBE,SAAAA,EAAW;AACTC,UAAAA,MAAAA,kBAAQ,MAAA,CAAA,CAACC,QAAAA,KACPA,QAAAA,CAASC,IAAAA,CAAK,CAACC,MAAAA,KAAAA;AACbA,YAAAA,MAAAA,CAAOC,QAAAA,CAASP,YAAAA,GAAepB,KAAAA,CAAMoB,YAAAA,IAAgBM,OAAOC,QAAAA,CAASP,YAAAA;AACrEM,YAAAA,MAAAA,CAAOC,QAAAA,CAASC,IAAAA,GAAO5B,KAAAA,CAAM4B,IAAAA,IAAQF,OAAOC,QAAAA,CAASC,IAAAA;AACvD,WAAA,CAAA,EAJM,QAAA;AAKV;AACF;KACF,CAAA;AACJ;AAEUnB,EAAAA,IAAAA,CAAKhB,MAAcc,OAAAA,EAAuB;AAClD,IAAA,IAAA,CAAKhC,QAAAA,CAASsD,OAAAA,CAAQpC,IAAAA,EAAM,OAAOqC,OAAOC,GAAAA,KAAAA;AACxC,MAAA,MAAMvB,MAAAA,GAAS,IAAIwB,mBAAAA,EAAAA;AACnB,MAAA,MAAMxB,MAAAA,CAAOyB,OAAAA,CAAQH,KAAAA,CAAMxC,WAAW,CAAA;AAEtC,MAAA,MAAM4C,QAAAA,GAAWJ,KAAAA,CAAMnD,MAAAA,CAAOwD,KAAAA,EAAK,IAAM;QAAEtD,MAAAA,EAAQuD;AAAU,OAAA;AAC7D,MAAA,MAAMjC,KAAAA,GAAQ,MAAMI,OAAAA,CAAQC,MAAAA,CAAO6B,YAAU,CAAA;AAC7C,MAAA,MAAM,EAAEC,MAAAA,EAAM,GAAK,MAAMnC,KAAAA,CAAMN,IAAIqC,QAAAA,EAAU;AAAEK,QAAAA,MAAAA,EAAQR,GAAAA,CAAIQ;OAAO,CAAA;AAClET,MAAAA,KAAAA,CAAMzC,cAAciD,MAAAA,CAAOrC,IAAAA;AAC3B,MAAA,IAAIiC,SAASrD,MAAAA,EAAQ;AACnBiD,QAAAA,KAAAA,CAAMxC,YAAYkD,IAAAA,CAAK,IAAIC,WAAAA,CAAYP,QAAAA,CAASrD,MAAM,CAAA,CAAA;AACxD;AACAiD,MAAAA,KAAAA,CAAMxC,WAAAA,CAAYkD,KAAKF,MAAAA,CAAAA;KACzB,CAAA;AACA,IAAA,OAAO,IAAA;AACT;AACF","file":"agent.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { Workflow, WorkflowRunOptions } from \"@/workflows/workflow.js\";\nimport { Message, UserMessage } from \"@/backend/message.js\";\nimport { AnyTool } from \"@/tools/base.js\";\nimport { BaseMemory, ReadOnlyMemory } from \"@/memory/base.js\";\nimport { z } from \"zod\";\nimport { UnconstrainedMemory } from \"@/memory/unconstrainedMemory.js\";\nimport { BaseAgent } from \"@/agents/base.js\";\nimport { isFunction, randomString } from \"remeda\";\nimport { ChatModel } from \"@/backend/chat.js\";\nimport { ToolCallingAgent } from \"@/agents/toolCalling/agent.js\";\nimport {\n  ToolCallingAgentExecutionConfig,\n  ToolCallingAgentRunInput,\n  ToolCallingAgentRunOptions,\n  ToolCallingAgentRunOutput,\n} from \"@/agents/toolCalling/types.js\";\n\ntype AgentInstance = BaseAgent<\n  ToolCallingAgentRunInput,\n  ToolCallingAgentRunOutput,\n  ToolCallingAgentRunOptions\n>;\ntype AgentFactory = (memory: ReadOnlyMemory) => AgentInstance | Promise<AgentInstance>;\ninterface AgentFactoryInput {\n  name?: string;\n  role?: string;\n  llm: ChatModel;\n  instructions?: string;\n  tools?: AnyTool[];\n  execution?: ToolCallingAgentExecutionConfig;\n}\n\nexport class AgentWorkflow {\n  protected readonly workflow;\n\n  readonly schema = z.object({\n    inputs: z.array(\n      z.object({\n        prompt: z.string().optional(),\n        context: z.string().optional(),\n        expectedOutput: z.union([z.string(), z.instanceof(z.ZodSchema)]).optional(),\n      }),\n    ),\n\n    finalAnswer: z.string().optional(),\n    newMessages: z.array(z.instanceof(Message)).default([]),\n  });\n\n  constructor(name = \"AgentWorkflow\") {\n    this.workflow = new Workflow({\n      name,\n      schema: this.schema,\n      outputSchema: this.schema.required(),\n    });\n  }\n\n  run(inputs: (ToolCallingAgentRunInput | Message)[], options: WorkflowRunOptions<string> = {}) {\n    return this.workflow.run(\n      {\n        inputs: inputs.map((input) => (input instanceof Message ? { prompt: input.text } : input)),\n      },\n      options,\n    );\n  }\n\n  addAgent(agent: AgentFactory | AgentFactoryInput): this;\n  addAgent(agent: AgentInstance): Promise<this>;\n  addAgent(agent: AgentInstance | AgentFactory | AgentFactoryInput): this | Promise<this> {\n    if (agent instanceof BaseAgent) {\n      return agent.clone().then((clone) => {\n        const factory: AgentFactory = (memory) => {\n          clone.memory = memory;\n          return clone;\n        };\n        return this._add(clone.meta.name, factory);\n      });\n    }\n\n    const name = agent.name || `Agent${randomString(4)}`;\n    return this._add(name, isFunction(agent) ? agent : this._createFactory(agent));\n  }\n\n  delAgent(name: string) {\n    return this.workflow.delStep(name);\n  }\n\n  protected _createFactory(input: AgentFactoryInput): AgentFactory {\n    return (memory: BaseMemory) =>\n      new ToolCallingAgent({\n        llm: input.llm,\n        tools: input.tools ?? [],\n        memory,\n        meta: {\n          name: input.name || `Agent${randomString(4)}`,\n          description: input.instructions ?? \"\",\n        },\n        execution: input.execution,\n        ...(input.instructions && {\n          templates: {\n            system: (template) =>\n              template.fork((config) => {\n                config.defaults.instructions = input.instructions || config.defaults.instructions;\n                config.defaults.role = input.role || config.defaults.role;\n              }),\n          },\n        }),\n      });\n  }\n\n  protected _add(name: string, factory: AgentFactory) {\n    this.workflow.addStep(name, async (state, ctx) => {\n      const memory = new UnconstrainedMemory();\n      await memory.addMany(state.newMessages);\n\n      const runInput = state.inputs.shift() ?? { prompt: undefined };\n      const agent = await factory(memory.asReadOnly());\n      const { result } = await agent.run(runInput, { signal: ctx.signal });\n      state.finalAnswer = result.text;\n      if (runInput.prompt) {\n        state.newMessages.push(new UserMessage(runInput.prompt));\n      }\n      state.newMessages.push(result);\n    });\n    return this;\n  }\n}\n"]}
+{"version":3,"sources":["../../src/workflows/agent.ts"],"names":["AgentWorkflow","workflow","schema","z","object","inputs","array","prompt","string","optional","context","expectedOutput","union","instanceof","ZodSchema","finalAnswer","newMessages","Message","default","name","Workflow","outputSchema","required","run","options","map","input","text","addAgent","agent","BaseAgent","clone","then","factory","memory","_add","meta","randomString","isFunction","_createFactory","delAgent","delStep","ToolCallingAgent","llm","tools","description","instructions","execution","templates","system","template","fork","config","defaults","role","addStep","state","ctx","UnconstrainedMemory","addMany","runInput","shift","undefined","asReadOnly","result","signal","push","UserMessage"],"mappings":";;;;;;;;;;AAqCO,MAAMA,aAAAA,CAAAA;EArCb;;;AAsCqBC,EAAAA,QAAAA;AAEVC,EAAAA,MAAAA,GAASC,EAAEC,MAAAA,CAAO;IACzBC,MAAAA,EAAQF,CAAAA,CAAEG,KAAAA,CACRH,CAAAA,CAAEC,MAAAA,CAAO;MACPG,MAAAA,EAAQJ,CAAAA,CAAEK,MAAAA,EAAM,CAAGC,QAAAA,EAAQ;MAC3BC,OAAAA,EAASP,CAAAA,CAAEK,MAAAA,EAAM,CAAGC,QAAAA,EAAQ;AAC5BE,MAAAA,cAAAA,EAAgBR,EAAES,KAAAA,CAAM;AAACT,QAAAA,CAAAA,CAAEK,MAAAA,EAAM;QAAIL,CAAAA,CAAEU,UAAAA,CAAWV,EAAEW,SAAS;AAAE,OAAA,CAAA,CAAEL,QAAAA;AACnE,KAAA,CAAA,CAAA;IAGFM,WAAAA,EAAaZ,CAAAA,CAAEK,MAAAA,EAAM,CAAGC,QAAAA,EAAQ;IAChCO,WAAAA,EAAab,CAAAA,CAAEG,MAAMH,CAAAA,CAAEU,UAAAA,CAAWI,OAAAA,CAAAA,CAAAA,CAAUC,OAAAA,CAAQ,EAAE;GACxD,CAAA;AAEA,EAAA,WAAA,CAAYC,OAAO,eAAA,EAAiB;AAClC,IAAA,IAAA,CAAKlB,QAAAA,GAAW,IAAImB,QAAAA,CAAS;AAC3BD,MAAAA,IAAAA;AACAjB,MAAAA,MAAAA,EAAQ,IAAA,CAAKA,MAAAA;MACbmB,YAAAA,EAAc,IAAA,CAAKnB,OAAOoB,QAAAA;KAC5B,CAAA;AACF,EAAA;EAEAC,GAAAA,CAAIlB,MAAAA,EAAgDmB,OAAAA,GAAsC,EAAC,EAAG;AAC5F,IAAA,OAAO,IAAA,CAAKvB,SAASsB,GAAAA,CACnB;AACElB,MAAAA,MAAAA,EAAQA,MAAAA,CAAOoB,GAAAA,CAAI,CAACC,KAAAA,KAAWA,iBAAiBT,OAAAA,GAAU;AAAEV,QAAAA,MAAAA,EAAQmB,KAAAA,CAAMC;AAAK,OAAA,GAAID,KAAAA;AACrF,KAAA,EACAF,OAAAA,CAAAA;AAEJ,EAAA;AAIAI,EAAAA,QAAAA,CAASC,KAAAA,EAA+E;AACtF,IAAA,IAAIA,iBAAiBC,SAAAA,EAAW;AAC9B,MAAA,OAAOD,KAAAA,CAAME,KAAAA,EAAK,CAAGC,IAAAA,CAAK,CAACD,KAAAA,KAAAA;AACzB,QAAA,MAAME,OAAAA,2BAAyBC,MAAAA,KAAAA;AAC7BH,UAAAA,KAAAA,CAAMG,MAAAA,GAASA,MAAAA;AACf,UAAA,OAAOH,KAAAA;QACT,CAAA,EAH8B,SAAA,CAAA;AAI9B,QAAA,OAAO,IAAA,CAAKI,IAAAA,CAAKJ,KAAAA,CAAMK,IAAAA,CAAKjB,MAAMc,OAAAA,CAAAA;MACpC,CAAA,CAAA;AACF,IAAA;AAEA,IAAA,MAAMd,OAAOU,KAAAA,CAAMV,IAAAA,IAAQ,CAAA,KAAA,EAAQkB,YAAAA,CAAa,CAAA,CAAA,CAAA,CAAA;AAChD,IAAA,OAAO,IAAA,CAAKF,IAAAA,CAAKhB,IAAAA,EAAMmB,UAAAA,CAAWT,KAAAA,IAASA,KAAAA,GAAQ,IAAA,CAAKU,cAAAA,CAAeV,KAAAA,CAAAA,CAAAA;AACzE,EAAA;AAEAW,EAAAA,QAAAA,CAASrB,IAAAA,EAAc;AACrB,IAAA,OAAO,IAAA,CAAKlB,QAAAA,CAASwC,OAAAA,CAAQtB,IAAAA,CAAAA;AAC/B,EAAA;AAEUoB,EAAAA,cAAAA,CAAeb,KAAAA,EAAwC;AAC/D,IAAA,OAAO,CAACQ,MAAAA,KACN,IAAIQ,gBAAAA,CAAiB;AACnBC,MAAAA,GAAAA,EAAKjB,KAAAA,CAAMiB,GAAAA;MACXC,KAAAA,EAAOlB,KAAAA,CAAMkB,SAAS,EAAA;AACtBV,MAAAA,MAAAA;MACAE,IAAAA,EAAM;AACJjB,QAAAA,IAAAA,EAAMO,KAAAA,CAAMP,IAAAA,IAAQ,CAAA,KAAA,EAAQkB,YAAAA,CAAa,CAAA,CAAA,CAAA,CAAA;AACzCQ,QAAAA,WAAAA,EAAanB,MAAMoB,YAAAA,IAAgB;AACrC,OAAA;AACAC,MAAAA,SAAAA,EAAWrB,KAAAA,CAAMqB,SAAAA;AACjB,MAAA,GAAIrB,MAAMoB,YAAAA,IAAgB;QACxBE,SAAAA,EAAW;AACTC,UAAAA,MAAAA,kBAAQ,MAAA,CAAA,CAACC,QAAAA,KACPA,QAAAA,CAASC,IAAAA,CAAK,CAACC,MAAAA,KAAAA;AACbA,YAAAA,MAAAA,CAAOC,QAAAA,CAASP,YAAAA,GAAepB,KAAAA,CAAMoB,YAAAA,IAAgBM,OAAOC,QAAAA,CAASP,YAAAA;AACrEM,YAAAA,MAAAA,CAAOC,QAAAA,CAASC,IAAAA,GAAO5B,KAAAA,CAAM4B,IAAAA,IAAQF,OAAOC,QAAAA,CAASC,IAAAA;AACvD,UAAA,CAAA,CAAA,EAJM,QAAA;AAKV;AACF;KACF,CAAA;AACJ,EAAA;AAEUnB,EAAAA,IAAAA,CAAKhB,MAAcc,OAAAA,EAAuB;AAClD,IAAA,IAAA,CAAKhC,QAAAA,CAASsD,OAAAA,CAAQpC,IAAAA,EAAM,OAAOqC,OAAOC,GAAAA,KAAAA;AACxC,MAAA,MAAMvB,MAAAA,GAAS,IAAIwB,mBAAAA,EAAAA;AACnB,MAAA,MAAMxB,MAAAA,CAAOyB,OAAAA,CAAQH,KAAAA,CAAMxC,WAAW,CAAA;AAEtC,MAAA,MAAM4C,QAAAA,GAAWJ,KAAAA,CAAMnD,MAAAA,CAAOwD,KAAAA,EAAK,IAAM;QAAEtD,MAAAA,EAAQuD;AAAU,OAAA;AAC7D,MAAA,MAAMjC,KAAAA,GAAQ,MAAMI,OAAAA,CAAQC,MAAAA,CAAO6B,YAAU,CAAA;AAC7C,MAAA,MAAM,EAAEC,MAAAA,EAAM,GAAK,MAAMnC,KAAAA,CAAMN,IAAIqC,QAAAA,EAAU;AAAEK,QAAAA,MAAAA,EAAQR,GAAAA,CAAIQ;OAAO,CAAA;AAClET,MAAAA,KAAAA,CAAMzC,cAAciD,MAAAA,CAAOrC,IAAAA;AAC3B,MAAA,IAAIiC,SAASrD,MAAAA,EAAQ;AACnBiD,QAAAA,KAAAA,CAAMxC,YAAYkD,IAAAA,CAAK,IAAIC,WAAAA,CAAYP,QAAAA,CAASrD,MAAM,CAAA,CAAA;AACxD,MAAA;AACAiD,MAAAA,KAAAA,CAAMxC,WAAAA,CAAYkD,KAAKF,MAAAA,CAAAA;IACzB,CAAA,CAAA;AACA,IAAA,OAAO,IAAA;AACT,EAAA;AACF","file":"agent.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { Workflow, WorkflowRunOptions } from \"@/workflows/workflow.js\";\nimport { Message, UserMessage } from \"@/backend/message.js\";\nimport { AnyTool } from \"@/tools/base.js\";\nimport { BaseMemory, ReadOnlyMemory } from \"@/memory/base.js\";\nimport { z } from \"zod\";\nimport { UnconstrainedMemory } from \"@/memory/unconstrainedMemory.js\";\nimport { BaseAgent } from \"@/agents/base.js\";\nimport { isFunction, randomString } from \"remeda\";\nimport { ChatModel } from \"@/backend/chat.js\";\nimport { ToolCallingAgent } from \"@/agents/toolCalling/agent.js\";\nimport {\n  ToolCallingAgentExecutionConfig,\n  ToolCallingAgentRunInput,\n  ToolCallingAgentRunOptions,\n  ToolCallingAgentRunOutput,\n} from \"@/agents/toolCalling/types.js\";\n\ntype AgentInstance = BaseAgent<\n  ToolCallingAgentRunInput,\n  ToolCallingAgentRunOutput,\n  ToolCallingAgentRunOptions\n>;\ntype AgentFactory = (memory: ReadOnlyMemory) => AgentInstance | Promise<AgentInstance>;\ninterface AgentFactoryInput {\n  name?: string;\n  role?: string;\n  llm: ChatModel;\n  instructions?: string;\n  tools?: AnyTool[];\n  execution?: ToolCallingAgentExecutionConfig;\n}\n\nexport class AgentWorkflow {\n  protected readonly workflow;\n\n  readonly schema = z.object({\n    inputs: z.array(\n      z.object({\n        prompt: z.string().optional(),\n        context: z.string().optional(),\n        expectedOutput: z.union([z.string(), z.instanceof(z.ZodSchema)]).optional(),\n      }),\n    ),\n\n    finalAnswer: z.string().optional(),\n    newMessages: z.array(z.instanceof(Message)).default([]),\n  });\n\n  constructor(name = \"AgentWorkflow\") {\n    this.workflow = new Workflow({\n      name,\n      schema: this.schema,\n      outputSchema: this.schema.required(),\n    });\n  }\n\n  run(inputs: (ToolCallingAgentRunInput | Message)[], options: WorkflowRunOptions<string> = {}) {\n    return this.workflow.run(\n      {\n        inputs: inputs.map((input) => (input instanceof Message ? { prompt: input.text } : input)),\n      },\n      options,\n    );\n  }\n\n  addAgent(agent: AgentFactory | AgentFactoryInput): this;\n  addAgent(agent: AgentInstance): Promise<this>;\n  addAgent(agent: AgentInstance | AgentFactory | AgentFactoryInput): this | Promise<this> {\n    if (agent instanceof BaseAgent) {\n      return agent.clone().then((clone) => {\n        const factory: AgentFactory = (memory) => {\n          clone.memory = memory;\n          return clone;\n        };\n        return this._add(clone.meta.name, factory);\n      });\n    }\n\n    const name = agent.name || `Agent${randomString(4)}`;\n    return this._add(name, isFunction(agent) ? agent : this._createFactory(agent));\n  }\n\n  delAgent(name: string) {\n    return this.workflow.delStep(name);\n  }\n\n  protected _createFactory(input: AgentFactoryInput): AgentFactory {\n    return (memory: BaseMemory) =>\n      new ToolCallingAgent({\n        llm: input.llm,\n        tools: input.tools ?? [],\n        memory,\n        meta: {\n          name: input.name || `Agent${randomString(4)}`,\n          description: input.instructions ?? \"\",\n        },\n        execution: input.execution,\n        ...(input.instructions && {\n          templates: {\n            system: (template) =>\n              template.fork((config) => {\n                config.defaults.instructions = input.instructions || config.defaults.instructions;\n                config.defaults.role = input.role || config.defaults.role;\n              }),\n          },\n        }),\n      });\n  }\n\n  protected _add(name: string, factory: AgentFactory) {\n    this.workflow.addStep(name, async (state, ctx) => {\n      const memory = new UnconstrainedMemory();\n      await memory.addMany(state.newMessages);\n\n      const runInput = state.inputs.shift() ?? { prompt: undefined };\n      const agent = await factory(memory.asReadOnly());\n      const { result } = await agent.run(runInput, { signal: ctx.signal });\n      state.finalAnswer = result.text;\n      if (runInput.prompt) {\n        state.newMessages.push(new UserMessage(runInput.prompt));\n      }\n      state.newMessages.push(result);\n    });\n    return this;\n  }\n}\n"]}

package/dist/workflows/workflow.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/workflows/workflow.ts"],"names":["WorkflowError","FrameworkError","message","extra","errors","context","run","isRetryable","isFatal","Workflow","Serializable","START","SELF","PREV","NEXT","END","emitter","steps","Map","startStep","undefined","input","Emitter","root","child","namespace","toCamelCase","name","filter","Boolean","creator","getSteps","Array","from","keys","schemas","pick","addStep","step","_addStep","addStrictStep","schema","schemaOrStep","stepOrEmpty","trim","ValueError","has","stepOrFlow","set","handler","asStep","setStart","state","options","RunContext","enter","signal","params","runContext","parse","result","handlers","abort","reason","next","findStep","start","current","get","emit","parseAsync","catch","err","shallowCopy","then","deepCopy","nextStepRaw","at","prev","stepResult","push","error","outputSchema","delStep","delete","overrides","ctx","mappedInput","mappedOutput","output","Object","assign","index","indexOf","createSnapshot","omit","loadSnapshot","snapshot","z","any"],"mappings":";;;;;;;;;;;;AA+EO,MAAMA,sBAIHC,yBAAAA,CAAAA;EAnFV;;;AAoFE,EAAA,WAAA,CAAYC,SAAiBC,KAAAA,EAA2D;AACtF,IAAA,KAAA,CAAMD,OAAAA,EAASC,OAAOC,MAAAA,EAAQ;MAC5BC,OAAAA,EAASF,KAAAA,EAAOG,OAAO,EAAC;MACxBC,WAAAA,EAAa,KAAA;MACbC,OAAAA,EAAS;KACX,CAAA;AACF;AACF;AAEO,MAAMC,iBAIHC,6BAAAA,CAAAA;EAjGV;;;;AAkGE,EAAA,OAAuBC,KAAAA,GAAQ,WAAA;AAC/B,EAAA,OAAuBC,IAAAA,GAAO,UAAA;AAC9B,EAAA,OAAuBC,IAAAA,GAAO,UAAA;AAC9B,EAAA,OAAuBC,IAAAA,GAAO,UAAA;AAC9B,EAAA,OAAuBC,GAAAA,GAAM,SAAA;AAEbC,EAAAA,OAAAA;AAEGC,EAAAA,KAAAA,uBAAYC,GAAAA,EAAAA;EACrBC,SAAAA,GAA+BC,MAAAA;AAEzC,EAAA,WAAA,CAA+BC,KAAAA,EAAmC;AAChE,IAAA,KAAA,EAAK,EAAA,KADwBA,KAAAA,GAAAA,KAAAA;AAE7B,IAAA,IAAA,CAAKL,OAAAA,GAAUM,mBAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAAM;MAChCC,SAAAA,EAAW;AAAC,QAAA,UAAA;QAAYC,kBAAAA,CAAYL,KAAAA,EAAOM,QAAQ,EAAA;AAAKC,OAAAA,CAAAA,MAAAA,CAAOC,OAAAA,CAAAA;MAC/DC,OAAAA,EAAS;KACX,CAAA;AACF;EAEAC,QAAAA,GAAW;AACT,IAAA,OAAOC,KAAAA,CAAMC,IAAAA,CAAK,IAAA,CAAKhB,KAAAA,CAAMiB,MAAI,CAAA;AACnC;AAEA,EAAA,IAAIP,IAAAA,GAAO;AACT,IAAA,OAAO,IAAA,CAAKN,MAAMM,IAAAA,IAAQ,EAAA;AAC5B;AAEA,EAAA,IAAIQ,OAAAA,GAAU;AACZ,IAAA,OAAOC,WAAAA,CAAK,KAAKf,KAAAA,EAAO;AAAC,MAAA,QAAA;AAAU,MAAA;AAAe,KAAA,CAAA;AACpD;AAUAgB,EAAAA,OAAAA,CACEV,MACAW,IAAAA,EACsC;AACtC,IAAA,OAAO,IAAA,CAAKC,QAAAA,CAASZ,IAAAA,EAAMW,IAAAA,CAAAA;AAC7B;EAYAE,aAAAA,CACEb,IAAAA,EACAc,QACAH,IAAAA,EACsC;AACtC,IAAA,OAAO,IAAA,CAAKC,QAAAA,CAASZ,IAAAA,EAAMc,MAAAA,EAAQH,IAAAA,CAAAA;AACrC;EAEUC,QAAAA,CACRZ,IAAAA,EACAe,cACAC,WAAAA,EACsC;AACtC,IAAA,IAAI,CAAChB,IAAAA,CAAKiB,IAAAA,EAAI,EAAI;AAChB,MAAA,MAAM,IAAIC,sBAAW,CAAA,0BAAA,CAA4B,CAAA;AACnD;AACA,IAAA,IAAI,IAAA,CAAK5B,KAAAA,CAAM6B,GAAAA,CAAInB,IAAAA,CAAAA,EAAO;AACxB,MAAA,MAAM,IAAIkB,qBAAAA,CAAW,CAAA,MAAA,EAASlB,IAAAA,CAAAA,iBAAAA,CAAuB,CAAA;AACvD;AACA,IAAA,IAAIA,IAAAA,KAASlB,SAASM,GAAAA,EAAK;AACzB,MAAA,MAAM,IAAI8B,qBAAAA,CAAW,CAAA,UAAA,EAAalB,IAAAA,CAAAA,iBAAAA,CAAuB,CAAA;AAC3D;AAEA,IAAA,MAAMc,MAAAA,GAAUC,YAAAA,IAAgBC,WAAAA,GAAcD,YAAAA,GAAe,KAAKrB,KAAAA,CAAMoB,MAAAA;AACxE,IAAA,MAAMM,aAAaJ,WAAAA,IAAeD,YAAAA;AAElC,IAAA,IAAA,CAAKzB,KAAAA,CAAM+B,IAAIrB,IAAAA,EAAM;AACnBc,MAAAA,MAAAA;AACAQ,MAAAA,OAAAA,EAASF,sBAAsBtC,QAAAA,GAAWsC,UAAAA,CAAWG,MAAAA,CAAO,EAAC,CAAA,GAAKH;KACpE,CAAA;AAEA,IAAA,OAAO,IAAA;AACT;AAEAI,EAAAA,QAAAA,CAASxB,IAAAA,EAAa;AACpB,IAAA,IAAA,CAAKR,SAAAA,GAAYQ,IAAAA;AACjB,IAAA,OAAO,IAAA;AACT;EAEArB,GAAAA,CAAI8C,KAAAA,EAAwBC,OAAAA,GAAqC,EAAC,EAAG;AACnE,IAAA,OAAOC,sBAAAA,CAAWC,MAChB,IAAA,EACA;AAAEC,MAAAA,MAAAA,EAAQH,OAAAA,EAASG,MAAAA;MAAQC,MAAAA,EAAQ;AAACL,QAAAA,KAAAA;AAAOC,QAAAA;;AAAkB,KAAA,EAC7D,OAAOK,UAAAA,KAAAA;AACL,MAAA,MAAMpD,GAAAA,GAA2C;AAC/CW,QAAAA,KAAAA,EAAO,EAAA;AACPmC,QAAAA,KAAAA,EAAO,IAAA,CAAK/B,KAAAA,CAAMoB,MAAAA,CAAOkB,KAAAA,CAAMP,KAAAA,CAAAA;QAC/BQ,MAAAA,EAAQxC;AACV,OAAA;AACA,MAAA,MAAMyC,QAAAA,GAA8C;AAClD5C,QAAAA,KAAAA,EAAOX,GAAAA,CAAIW,KAAAA;AACXuC,QAAAA,MAAAA,EAAQE,UAAAA,CAAWF,MAAAA;AACnBM,QAAAA,KAAAA,kBAAO,MAAA,CAAA,CAACC,MAAAA,KAAWL,UAAAA,CAAWI,KAAAA,CAAMC,MAAAA,CAAAA,EAA7B,OAAA;AACT,OAAA;AAEA,MAAA,IAAIC,IAAAA,GACF,IAAA,CAAKC,QAAAA,CAASZ,OAAAA,EAASa,SAAS,IAAA,CAAK/C,SAAAA,IAAa,IAAA,CAAKY,QAAAA,EAAQ,CAAG,CAAA,CAAE,CAAA,CAAEoC,WACtE1D,QAAAA,CAASM,GAAAA;AAEX,MAAA,OAAOiD,IAAAA,IAAQA,IAAAA,KAASvD,QAAAA,CAASM,GAAAA,EAAK;AACpC,QAAA,MAAMY,IAAAA,GAAOqC,IAAAA;AACb,QAAA,MAAM1B,IAAAA,GAAO,IAAA,CAAKrB,KAAAA,CAAMmD,GAAAA,CAAIJ,IAAAA,CAAAA;AAC5B,QAAA,IAAI,CAAC1B,IAAAA,EAAM;AACT,UAAA,MAAM,IAAItC,aAAAA,CAAc,CAAA,MAAA,EAASgE,IAAAA,CAAAA,gBAAAA,CAAAA,EAAwB;AAAE1D,YAAAA;WAAI,CAAA;AACjE;AACA,QAAA,MAAMoD,UAAAA,CAAW1C,OAAAA,CAAQqD,IAAAA,CAAK,OAAA,EAAS;AAAE/D,UAAAA,GAAAA;UAAKgC,IAAAA,EAAM0B;SAAK,CAAA;AACzD,QAAA,IAAI;AACF,UAAA,MAAMZ,MAAAA,GAAQ,MAAMd,IAAAA,CAAKG,MAAAA,CACtB6B,UAAAA,CAAWhE,IAAI8C,KAAK,CAAA,CACpBmB,KAAAA,CAAM,OAAOC,GAAAA,KAAAA;AACZ,YAAA,MAAM,IAAIxE,aAAAA,CACR,CAAA,MAAA,EAASgE,IAAAA,CAAAA,oFAAAA,CAAAA,EACT;AAAE1D,cAAAA,GAAAA,EAAKmE,sBAAYnE,GAAAA,CAAAA;cAAMF,MAAAA,EAAQ;AAACoE,gBAAAA;;aAAK,CAAA;WAE3C,CAAA,CACCE,KAAKC,kBAAAA,CAAAA;AAER,UAAA,MAAMC,WAAAA,GAAc,MAAMtC,IAAAA,CAAKW,OAAAA,CAAQG,QAAOS,QAAAA,CAAAA;AAC9C,UAAA,IAAIe,WAAAA,KAAgBnE,SAASE,KAAAA,EAAO;AAClCqD,YAAAA,IAAAA,GAAO1D,GAAAA,CAAIW,KAAAA,CAAM4D,EAAAA,CAAG,CAAA,CAAA,EAAIlD,IAAAA;WAC1B,MAAA,IAAWiD,WAAAA,KAAgBnE,SAASI,IAAAA,EAAM;AACxCmD,YAAAA,IAAAA,GAAO,IAAA,CAAKC,QAAAA,CAASD,IAAAA,CAAAA,CAAMc,IAAAA;WAC7B,MAAA,IAAWF,WAAAA,KAAgBnE,SAASG,IAAAA,EAAM;AACxCoD,YAAAA,IAAAA,GAAO,IAAA,CAAKC,QAAAA,CAASD,IAAAA,CAAAA,CAAMG,OAAAA;AAC7B,WAAA,MAAA,IAAW,CAACS,WAAAA,IAAeA,WAAAA,KAAgBnE,QAAAA,CAASK,IAAAA,EAAM;AACxDkD,YAAAA,IAAAA,GAAO,IAAA,CAAKC,QAAAA,CAASD,IAAAA,CAAAA,CAAMA,QAAQvD,QAAAA,CAASM,GAAAA;WAC9C,MAAO;AACLiD,YAAAA,IAAAA,GAAOY,WAAAA;AACT;AAEA,UAAA,MAAMG,UAAAA,GAA6C;AACjDpD,YAAAA,IAAAA;YACAyB,KAAAA,EAAAA,MAAAA;AACAY,YAAAA;AACF,WAAA;AACA1D,UAAAA,GAAAA,CAAIW,KAAAA,CAAM+D,KAAKD,UAAAA,CAAAA;AACfzE,UAAAA,GAAAA,CAAI8C,QAAQ2B,UAAAA,CAAW3B,KAAAA;AAEvB,UAAA,MAAMM,UAAAA,CAAW1C,OAAAA,CAAQqD,IAAAA,CAAK,SAAA,EAAW;AACvC/D,YAAAA,GAAAA,EAAKmE,sBAAYnE,GAAAA,CAAAA;AACjB8C,YAAAA,KAAAA,EAAO2B,UAAAA,CAAW3B,KAAAA;AAClBd,YAAAA,IAAAA,EAAMyC,UAAAA,CAAWpD,IAAAA;AACjBqC,YAAAA,IAAAA,EAAMe,UAAAA,CAAWf;WACnB,CAAA;AACF,SAAA,CAAA,OAASiB,KAAAA,EAAO;AACd,UAAA,MAAMvB,UAAAA,CAAW1C,OAAAA,CAAQqD,IAAAA,CAAK,OAAA,EAAS;AACrC/D,YAAAA,GAAAA,EAAKmE,sBAAYnE,GAAAA,CAAAA;YACjBgC,IAAAA,EAAM0B,IAAAA;AACNiB,YAAAA;WACF,CAAA;AACA,UAAA,MAAMA,KAAAA;AACR;AACF;AAEA3E,MAAAA,GAAAA,CAAIsD,MAAAA,GAAS,MAAA,CAAO,IAAA,CAAKvC,KAAAA,CAAM6D,gBAAgB,IAAA,CAAK7D,KAAAA,CAAMoB,MAAAA,EACvD6B,UAAAA,CAAWhE,GAAAA,CAAI8C,KAAK,CAAA,CACpBmB,KAAAA,CAAM,CAACC,GAAAA,KAAAA;AACN,QAAA,MAAM,IAAIxE,cACR,CAAA,kFAAA,CAAA,EACA;AAAEM,UAAAA,GAAAA,EAAKmE,sBAAYnE,GAAAA,CAAAA;UAAMF,MAAAA,EAAQ;AAACoE,YAAAA;;SAAK,CAAA;OAE3C,CAAA;AACF,MAAA,OAAOlE,GAAAA;KACT,CAAA;AAEJ;AAEA6E,EAAAA,OAAAA,CAAyBxD,IAAAA,EAAuD;AAC9E,IAAA,IAAI,IAAA,CAAKR,cAAcQ,IAAAA,EAAM;AAC3B,MAAA,IAAA,CAAKR,SAAAA,GAAYC,MAAAA;AACnB;AACA,IAAA,IAAA,CAAKH,KAAAA,CAAMmE,OAAOzD,IAAAA,CAAAA;AAClB,IAAA,OAAO,IAAA;AACT;AAEAuB,EAAAA,MAAAA,CAIEmC,SAAAA,EAK+C;AAC/C,IAAA,OAAO,OAAOjC,OAAOkC,GAAAA,KAAAA;AACnB,MAAA,MAAMC,cAAcF,SAAAA,EAAWhE,KAAAA,GAAQgE,SAAAA,CAAUhE,KAAAA,CAAM+B,KAAAA,CAAAA,GAASA,KAAAA;AAChE,MAAA,MAAMQ,MAAAA,GAAS,MAAM,IAAA,CAAKtD,GAAAA,CAAIiF,WAAAA,EAAa;AAAErB,QAAAA,KAAAA,EAAOmB,SAAAA,EAAWnB,KAAAA;AAAOV,QAAAA,MAAAA,EAAQ8B,GAAAA,CAAI9B;OAAO,CAAA;AACzF,MAAA,MAAMgC,YAAAA,GAAeH,WAAWI,MAAAA,GAASJ,SAAAA,CAAUI,OAAO7B,MAAAA,CAAOR,KAAK,IAAIQ,MAAAA,CAAOR,KAAAA;AACjFsC,MAAAA,MAAAA,CAAOC,MAAAA,CAAOvC,OAAOoC,YAAAA,CAAAA;AACrB,MAAA,OAAOH,SAAAA,CAAUrB,IAAAA;AACnB,KAAA;AACF;AAEUC,EAAAA,QAAAA,CAASE,OAAAA,EAAgB;AACjC,IAAA,MAAMlD,KAAAA,GAAQ,KAAKc,QAAAA,EAAQ;AAC3B,IAAA,MAAM6D,KAAAA,GAAQ3E,KAAAA,CAAM4E,OAAAA,CAAQ1B,OAAAA,CAAAA;AAC5B,IAAA,OAAO;MACLW,IAAAA,EAAM7D,KAAAA,CAAM2E,QAAQ,CAAA,CAAA;AACpBzB,MAAAA,OAAAA,EAASlD,MAAM2E,KAAAA,CAAAA;MACf5B,IAAAA,EAAM/C,KAAAA,CAAM2E,QAAQ,CAAA;AACtB,KAAA;AACF;EAEAE,cAAAA,GAAiB;AACf,IAAA,OAAO;MACLzE,KAAAA,EAAO0E,WAAAA,CAAK,KAAK1E,KAAAA,EAAO;AAAC,QAAA,QAAA;AAAU,QAAA;AAAe,OAAA,CAAA;AAClDL,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;AACdG,MAAAA,SAAAA,EAAW,IAAA,CAAKA,SAAAA;AAChBF,MAAAA,KAAAA,EAAO,IAAA,CAAKA;AACd,KAAA;AACF;AAEA+E,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7DP,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAMM,QAAAA,CAAAA;AACpB,IAAA,IAAA,CAAK5E,KAAAA,CAAMoB,MAAAA,KAAWyD,KAAAA,CAAEC,GAAAA,EAAG;AAC3B,IAAA,IAAA,CAAK9E,KAAAA,CAAM6D,YAAAA,KAAiBgB,KAAAA,CAAEC,GAAAA,EAAG;AACnC;AACF","file":"workflow.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { z, ZodSchema } from \"zod\";\nimport { Serializable } from \"@/internals/serializable.js\";\nimport { Callback, Emitter } from \"@/emitter/emitter.js\";\nimport { RunContext } from \"@/context.js\";\nimport { omit, pick, toCamelCase } from \"remeda\";\nimport { deepCopy, shallowCopy } from \"@/serializer/utils.js\";\nimport { FrameworkError, ValueError } from \"@/errors.js\";\n\n// eslint-disable-next-line @typescript-eslint/no-invalid-void-type\ntype WorkflowStepHandlerReturn<K extends string> = K | ReservedStep | void;\n\nexport interface WorkflowRun<T extends ZodSchema, T2 extends ZodSchema, K extends string> {\n  result: z.output<T2>;\n  steps: WorkflowStepRes<T, K>[];\n  state: z.output<T>;\n}\n\nexport interface WorkflowRunOptions<K extends string> {\n  start?: K;\n  signal?: AbortSignal;\n}\n\nexport interface WorkflowStepDef<T extends ZodSchema, K extends string> {\n  schema: T;\n  handler: WorkflowStepHandler<T, K>;\n}\n\nexport interface WorkflowStepRes<T extends ZodSchema, K extends string> {\n  name: K;\n  next: K | ReservedStep;\n  state: z.output<T>;\n}\n\nexport interface WorkflowRunContext<T extends ZodSchema, K extends string> {\n  steps: WorkflowStepRes<T, K>[];\n  signal: AbortSignal;\n  abort: (reason?: Error) => void;\n}\n\nexport type WorkflowStepHandler<T extends ZodSchema, K extends string> = (\n  state: z.output<T>,\n  context: WorkflowRunContext<T, K>,\n) => Promise<WorkflowStepHandlerReturn<K>> | WorkflowStepHandlerReturn<K>;\n\nexport interface WorkflowEvents<T extends ZodSchema, T2 extends ZodSchema, K extends string> {\n  start: Callback<{ step: K; run: WorkflowRun<T, T2, K> }>;\n  error: Callback<{\n    step: K;\n    error: Error;\n    run: WorkflowRun<T, T2, K>;\n  }>;\n  success: Callback<{\n    step: K;\n    next: K | ReservedStep;\n    state: z.output<T>;\n    run: WorkflowRun<T, T2, K>;\n  }>;\n}\n\ninterface FlowInput<TS extends ZodSchema, TS2 extends ZodSchema = TS> {\n  name?: string;\n  schema: TS;\n  outputSchema?: TS2;\n}\n\ntype RunStep<K extends string> = K | typeof Workflow.END;\ntype ReservedStep =\n  | typeof Workflow.START\n  | typeof Workflow.NEXT\n  | typeof Workflow.PREV\n  | typeof Workflow.END\n  | typeof Workflow.SELF;\nexport type FlowStepName<K extends string> = K | ReservedStep;\n\nexport class WorkflowError<\n  T extends ZodSchema,\n  T2 extends ZodSchema,\n  K extends string,\n> extends FrameworkError {\n  constructor(message: string, extra?: { run?: WorkflowRun<T, T2, K>; errors?: Error[] }) {\n    super(message, extra?.errors, {\n      context: extra?.run ?? {},\n      isRetryable: false,\n      isFatal: true,\n    });\n  }\n}\n\nexport class Workflow<\n  TInput extends ZodSchema,\n  TOutput extends ZodSchema = TInput,\n  TKeys extends string = string,\n> extends Serializable {\n  public static readonly START = \"__start__\";\n  public static readonly SELF = \"__self__\";\n  public static readonly PREV = \"__prev__\";\n  public static readonly NEXT = \"__next__\";\n  public static readonly END = \"__end__\";\n\n  public readonly emitter: Emitter<WorkflowEvents<TInput, TOutput, TKeys>>;\n\n  protected readonly steps = new Map<string, WorkflowStepDef<TInput, TKeys>>();\n  protected startStep: TKeys | undefined = undefined;\n\n  constructor(protected readonly input: FlowInput<TInput, TOutput>) {\n    super();\n    this.emitter = Emitter.root.child({\n      namespace: [\"workflow\", toCamelCase(input?.name ?? \"\")].filter(Boolean),\n      creator: this,\n    });\n  }\n\n  getSteps() {\n    return Array.from(this.steps.keys()) as TKeys[];\n  }\n\n  get name() {\n    return this.input.name ?? \"\";\n  }\n\n  get schemas() {\n    return pick(this.input, [\"schema\", \"outputSchema\"]);\n  }\n\n  addStep<L extends string>(\n    name: L,\n    handler: WorkflowStepHandler<TInput | TOutput, TKeys>,\n  ): Workflow<TInput, TOutput, L | TKeys>;\n  addStep<L extends string, TFlow extends AnyWorkflowFlow>(\n    name: L,\n    flow: Workflow.pipeable<this, TFlow>,\n  ): Workflow<TInput, TOutput, L | TKeys>;\n  addStep<L extends string>(\n    name: L,\n    step: WorkflowStepHandler<TInput | TOutput, TKeys> | AnyWorkflowFlow,\n  ): Workflow<TInput, TOutput, L | TKeys> {\n    return this._addStep(name, step);\n  }\n\n  addStrictStep<L extends string, TI2 extends ZodSchema>(\n    name: L,\n    schema: TI2,\n    handler: WorkflowStepHandler<TI2, TKeys>,\n  ): Workflow<TInput, TOutput, L | TKeys>;\n  addStrictStep<L extends string, TI2 extends ZodSchema, TFlow extends AnyWorkflowFlow>(\n    name: L,\n    schema: TI2,\n    flow: Workflow.pipeable<Workflow<TI2, TOutput, TKeys>, TFlow>,\n  ): Workflow<TInput, TOutput, L | TKeys>;\n  addStrictStep<L extends string, TI2 extends ZodSchema>(\n    name: L,\n    schema: TI2,\n    step: WorkflowStepHandler<TI2, TKeys> | AnyWorkflowFlow,\n  ): Workflow<TInput, TOutput, L | TKeys> {\n    return this._addStep(name, schema, step);\n  }\n\n  protected _addStep<TI2 extends ZodSchema = TInput, L extends string = TKeys>(\n    name: L,\n    schemaOrStep: TI2 | WorkflowStepHandler<TInput, TKeys> | AnyWorkflowFlow,\n    stepOrEmpty?: WorkflowStepHandler<TI2, TKeys> | AnyWorkflowFlow,\n  ): Workflow<TInput, TOutput, L | TKeys> {\n    if (!name.trim()) {\n      throw new ValueError(`Step name cannot be empty!`);\n    }\n    if (this.steps.has(name)) {\n      throw new ValueError(`Step '${name}' already exists!`);\n    }\n    if (name === Workflow.END) {\n      throw new ValueError(`The name '${name}' cannot be used!`);\n    }\n\n    const schema = (schemaOrStep && stepOrEmpty ? schemaOrStep : this.input.schema) as TInput;\n    const stepOrFlow = stepOrEmpty || schemaOrStep;\n\n    this.steps.set(name, {\n      schema,\n      handler: stepOrFlow instanceof Workflow ? stepOrFlow.asStep({}) : stepOrFlow,\n    } as WorkflowStepDef<TInput, TKeys>);\n\n    return this as unknown as Workflow<TInput, TOutput, L | TKeys>;\n  }\n\n  setStart(name: TKeys) {\n    this.startStep = name;\n    return this;\n  }\n\n  run(state: z.input<TInput>, options: WorkflowRunOptions<TKeys> = {}) {\n    return RunContext.enter(\n      this,\n      { signal: options?.signal, params: [state, options] as const },\n      async (runContext): Promise<WorkflowRun<TInput, TOutput, TKeys>> => {\n        const run: WorkflowRun<TInput, TOutput, TKeys> = {\n          steps: [],\n          state: this.input.schema.parse(state),\n          result: undefined as z.output<TOutput>,\n        };\n        const handlers: WorkflowRunContext<TInput, TKeys> = {\n          steps: run.steps,\n          signal: runContext.signal,\n          abort: (reason) => runContext.abort(reason),\n        };\n\n        let next: RunStep<TKeys> =\n          this.findStep(options?.start || this.startStep || this.getSteps()[0]).current ??\n          Workflow.END;\n\n        while (next && next !== Workflow.END) {\n          const name = next;\n          const step = this.steps.get(next);\n          if (!step) {\n            throw new WorkflowError(`Step '${next}' was not found.`, { run });\n          }\n          await runContext.emitter.emit(\"start\", { run, step: next });\n          try {\n            const state = await step.schema\n              .parseAsync(run.state)\n              .catch(async (err: Error) => {\n                throw new WorkflowError(\n                  `Step '${next}' cannot be executed because the provided input doesn't adhere to the step's schema.`,\n                  { run: shallowCopy(run), errors: [err] },\n                );\n              })\n              .then(deepCopy);\n\n            const nextStepRaw = await step.handler(state, handlers);\n            if (nextStepRaw === Workflow.START) {\n              next = run.steps.at(0)?.name!;\n            } else if (nextStepRaw === Workflow.PREV) {\n              next = this.findStep(next).prev;\n            } else if (nextStepRaw === Workflow.SELF) {\n              next = this.findStep(next).current;\n            } else if (!nextStepRaw || nextStepRaw === Workflow.NEXT) {\n              next = this.findStep(next).next || Workflow.END;\n            } else {\n              next = nextStepRaw;\n            }\n\n            const stepResult: WorkflowStepRes<TInput, TKeys> = {\n              name,\n              state,\n              next,\n            };\n            run.steps.push(stepResult);\n            run.state = stepResult.state;\n\n            await runContext.emitter.emit(\"success\", {\n              run: shallowCopy(run),\n              state: stepResult.state,\n              step: stepResult.name,\n              next: stepResult.next,\n            });\n          } catch (error) {\n            await runContext.emitter.emit(\"error\", {\n              run: shallowCopy(run),\n              step: next as TKeys,\n              error,\n            });\n            throw error;\n          }\n        }\n\n        run.result = await (this.input.outputSchema ?? this.input.schema)\n          .parseAsync(run.state)\n          .catch((err) => {\n            throw new WorkflowError(\n              `Workflow has ended but it's state does not adhere to the workflow's output schema.`,\n              { run: shallowCopy(run), errors: [err] },\n            );\n          });\n        return run;\n      },\n    );\n  }\n\n  delStep<L extends TKeys>(name: L): Workflow<TInput, TOutput, Exclude<TKeys, L>> {\n    if (this.startStep === name) {\n      this.startStep = undefined;\n    }\n    this.steps.delete(name);\n    return this as unknown as Workflow<TInput, TOutput, Exclude<TKeys, L>>;\n  }\n\n  asStep<\n    TInput2 extends ZodSchema = TInput,\n    TOutput2 extends ZodSchema = TOutput,\n    TKeys2 extends string = TKeys,\n  >(overrides: {\n    input?: (input: z.output<TInput2>) => z.output<TInput> | z.input<TInput>;\n    output?: (output: z.output<TOutput>) => z.output<TOutput2> | z.input<TOutput2>;\n    start?: TKeys;\n    next?: FlowStepName<TKeys2>;\n  }): WorkflowStepHandler<TInput2, TKeys | TKeys2> {\n    return async (state, ctx) => {\n      const mappedInput = overrides?.input ? overrides.input(state) : state;\n      const result = await this.run(mappedInput, { start: overrides?.start, signal: ctx.signal });\n      const mappedOutput = overrides?.output ? overrides.output(result.state) : result.state;\n      Object.assign(state, mappedOutput);\n      return overrides.next;\n    };\n  }\n\n  protected findStep(current: TKeys) {\n    const steps = this.getSteps();\n    const index = steps.indexOf(current);\n    return {\n      prev: steps[index - 1],\n      current: steps[index],\n      next: steps[index + 1],\n    };\n  }\n\n  createSnapshot() {\n    return {\n      input: omit(this.input, [\"schema\", \"outputSchema\"]),\n      emitter: this.emitter,\n      startStep: this.startStep,\n      steps: this.steps,\n    };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n    this.input.schema ??= z.any() as unknown as TInput;\n    this.input.outputSchema ??= z.any() as unknown as TOutput;\n  }\n}\n\n// eslint-disable-next-line @typescript-eslint/no-namespace\nexport namespace Workflow {\n  export type pipeable<T extends AnyWorkflowFlow, T2 extends AnyWorkflowFlow> =\n    Workflow.state<T> extends Workflow.input<T2> ? T2 : never;\n  export type input<T> = T extends Workflow<infer A, any, any> ? z.input<A> : never;\n  export type state<T> = T extends Workflow<infer A, any, any> ? z.output<A> : never;\n  export type output<T> = T extends Workflow<any, infer A, any> ? z.output<A> : never;\n  export type run<T> = T extends Workflow<infer A, infer B, infer C> ? WorkflowRun<A, B, C> : never;\n}\n\nexport type AnyWorkflowFlow = Workflow<any, any, any>;\n"]}
+{"version":3,"sources":["../../src/workflows/workflow.ts"],"names":["WorkflowError","FrameworkError","message","extra","errors","context","run","isRetryable","isFatal","Workflow","Serializable","START","SELF","PREV","NEXT","END","emitter","steps","Map","startStep","undefined","input","Emitter","root","child","namespace","toCamelCase","name","filter","Boolean","creator","getSteps","Array","from","keys","schemas","pick","addStep","step","_addStep","addStrictStep","schema","schemaOrStep","stepOrEmpty","trim","ValueError","has","stepOrFlow","set","handler","asStep","setStart","state","options","RunContext","enter","signal","params","runContext","parse","result","handlers","abort","reason","next","findStep","start","current","get","emit","parseAsync","catch","err","shallowCopy","then","deepCopy","nextStepRaw","at","prev","stepResult","push","error","outputSchema","delStep","delete","overrides","ctx","mappedInput","mappedOutput","output","Object","assign","index","indexOf","createSnapshot","omit","loadSnapshot","snapshot","z","any"],"mappings":";;;;;;;;;;;;AA+EO,MAAMA,sBAIHC,yBAAAA,CAAAA;EAnFV;;;AAoFE,EAAA,WAAA,CAAYC,SAAiBC,KAAAA,EAA2D;AACtF,IAAA,KAAA,CAAMD,OAAAA,EAASC,OAAOC,MAAAA,EAAQ;MAC5BC,OAAAA,EAASF,KAAAA,EAAOG,OAAO,EAAC;MACxBC,WAAAA,EAAa,KAAA;MACbC,OAAAA,EAAS;KACX,CAAA;AACF,EAAA;AACF;AAEO,MAAMC,iBAIHC,6BAAAA,CAAAA;EAjGV;;;;AAkGE,EAAA,OAAuBC,KAAAA,GAAQ,WAAA;AAC/B,EAAA,OAAuBC,IAAAA,GAAO,UAAA;AAC9B,EAAA,OAAuBC,IAAAA,GAAO,UAAA;AAC9B,EAAA,OAAuBC,IAAAA,GAAO,UAAA;AAC9B,EAAA,OAAuBC,GAAAA,GAAM,SAAA;AAEbC,EAAAA,OAAAA;AAEGC,EAAAA,KAAAA,uBAAYC,GAAAA,EAAAA;EACrBC,SAAAA,GAA+BC,MAAAA;AAEzC,EAAA,WAAA,CAA+BC,KAAAA,EAAmC;AAChE,IAAA,KAAA,EAAK,EAAA,KADwBA,KAAAA,GAAAA,KAAAA;AAE7B,IAAA,IAAA,CAAKL,OAAAA,GAAUM,mBAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAAM;MAChCC,SAAAA,EAAW;AAAC,QAAA,UAAA;QAAYC,kBAAAA,CAAYL,KAAAA,EAAOM,QAAQ,EAAA;AAAKC,OAAAA,CAAAA,MAAAA,CAAOC,OAAAA,CAAAA;MAC/DC,OAAAA,EAAS;KACX,CAAA;AACF,EAAA;EAEAC,QAAAA,GAAW;AACT,IAAA,OAAOC,KAAAA,CAAMC,IAAAA,CAAK,IAAA,CAAKhB,KAAAA,CAAMiB,MAAI,CAAA;AACnC,EAAA;AAEA,EAAA,IAAIP,IAAAA,GAAO;AACT,IAAA,OAAO,IAAA,CAAKN,MAAMM,IAAAA,IAAQ,EAAA;AAC5B,EAAA;AAEA,EAAA,IAAIQ,OAAAA,GAAU;AACZ,IAAA,OAAOC,WAAAA,CAAK,KAAKf,KAAAA,EAAO;AAAC,MAAA,QAAA;AAAU,MAAA;AAAe,KAAA,CAAA;AACpD,EAAA;AAUAgB,EAAAA,OAAAA,CACEV,MACAW,IAAAA,EACsC;AACtC,IAAA,OAAO,IAAA,CAAKC,QAAAA,CAASZ,IAAAA,EAAMW,IAAAA,CAAAA;AAC7B,EAAA;EAYAE,aAAAA,CACEb,IAAAA,EACAc,QACAH,IAAAA,EACsC;AACtC,IAAA,OAAO,IAAA,CAAKC,QAAAA,CAASZ,IAAAA,EAAMc,MAAAA,EAAQH,IAAAA,CAAAA;AACrC,EAAA;EAEUC,QAAAA,CACRZ,IAAAA,EACAe,cACAC,WAAAA,EACsC;AACtC,IAAA,IAAI,CAAChB,IAAAA,CAAKiB,IAAAA,EAAI,EAAI;AAChB,MAAA,MAAM,IAAIC,sBAAW,CAAA,0BAAA,CAA4B,CAAA;AACnD,IAAA;AACA,IAAA,IAAI,IAAA,CAAK5B,KAAAA,CAAM6B,GAAAA,CAAInB,IAAAA,CAAAA,EAAO;AACxB,MAAA,MAAM,IAAIkB,qBAAAA,CAAW,CAAA,MAAA,EAASlB,IAAAA,CAAAA,iBAAAA,CAAuB,CAAA;AACvD,IAAA;AACA,IAAA,IAAIA,IAAAA,KAASlB,SAASM,GAAAA,EAAK;AACzB,MAAA,MAAM,IAAI8B,qBAAAA,CAAW,CAAA,UAAA,EAAalB,IAAAA,CAAAA,iBAAAA,CAAuB,CAAA;AAC3D,IAAA;AAEA,IAAA,MAAMc,MAAAA,GAAUC,YAAAA,IAAgBC,WAAAA,GAAcD,YAAAA,GAAe,KAAKrB,KAAAA,CAAMoB,MAAAA;AACxE,IAAA,MAAMM,aAAaJ,WAAAA,IAAeD,YAAAA;AAElC,IAAA,IAAA,CAAKzB,KAAAA,CAAM+B,IAAIrB,IAAAA,EAAM;AACnBc,MAAAA,MAAAA;AACAQ,MAAAA,OAAAA,EAASF,sBAAsBtC,QAAAA,GAAWsC,UAAAA,CAAWG,MAAAA,CAAO,EAAC,CAAA,GAAKH;KACpE,CAAA;AAEA,IAAA,OAAO,IAAA;AACT,EAAA;AAEAI,EAAAA,QAAAA,CAASxB,IAAAA,EAAa;AACpB,IAAA,IAAA,CAAKR,SAAAA,GAAYQ,IAAAA;AACjB,IAAA,OAAO,IAAA;AACT,EAAA;EAEArB,GAAAA,CAAI8C,KAAAA,EAAwBC,OAAAA,GAAqC,EAAC,EAAG;AACnE,IAAA,OAAOC,sBAAAA,CAAWC,MAChB,IAAA,EACA;AAAEC,MAAAA,MAAAA,EAAQH,OAAAA,EAASG,MAAAA;MAAQC,MAAAA,EAAQ;AAACL,QAAAA,KAAAA;AAAOC,QAAAA;;AAAkB,KAAA,EAC7D,OAAOK,UAAAA,KAAAA;AACL,MAAA,MAAMpD,GAAAA,GAA2C;AAC/CW,QAAAA,KAAAA,EAAO,EAAA;AACPmC,QAAAA,KAAAA,EAAO,IAAA,CAAK/B,KAAAA,CAAMoB,MAAAA,CAAOkB,KAAAA,CAAMP,KAAAA,CAAAA;QAC/BQ,MAAAA,EAAQxC;AACV,OAAA;AACA,MAAA,MAAMyC,QAAAA,GAA8C;AAClD5C,QAAAA,KAAAA,EAAOX,GAAAA,CAAIW,KAAAA;AACXuC,QAAAA,MAAAA,EAAQE,UAAAA,CAAWF,MAAAA;AACnBM,QAAAA,KAAAA,kBAAO,MAAA,CAAA,CAACC,MAAAA,KAAWL,UAAAA,CAAWI,KAAAA,CAAMC,MAAAA,CAAAA,EAA7B,OAAA;AACT,OAAA;AAEA,MAAA,IAAIC,IAAAA,GACF,IAAA,CAAKC,QAAAA,CAASZ,OAAAA,EAASa,SAAS,IAAA,CAAK/C,SAAAA,IAAa,IAAA,CAAKY,QAAAA,EAAQ,CAAG,CAAA,CAAE,CAAA,CAAEoC,WACtE1D,QAAAA,CAASM,GAAAA;AAEX,MAAA,OAAOiD,IAAAA,IAAQA,IAAAA,KAASvD,QAAAA,CAASM,GAAAA,EAAK;AACpC,QAAA,MAAMY,IAAAA,GAAOqC,IAAAA;AACb,QAAA,MAAM1B,IAAAA,GAAO,IAAA,CAAKrB,KAAAA,CAAMmD,GAAAA,CAAIJ,IAAAA,CAAAA;AAC5B,QAAA,IAAI,CAAC1B,IAAAA,EAAM;AACT,UAAA,MAAM,IAAItC,aAAAA,CAAc,CAAA,MAAA,EAASgE,IAAAA,CAAAA,gBAAAA,CAAAA,EAAwB;AAAE1D,YAAAA;WAAI,CAAA;AACjE,QAAA;AACA,QAAA,MAAMoD,UAAAA,CAAW1C,OAAAA,CAAQqD,IAAAA,CAAK,OAAA,EAAS;AAAE/D,UAAAA,GAAAA;UAAKgC,IAAAA,EAAM0B;SAAK,CAAA;AACzD,QAAA,IAAI;AACF,UAAA,MAAMZ,MAAAA,GAAQ,MAAMd,IAAAA,CAAKG,MAAAA,CACtB6B,UAAAA,CAAWhE,IAAI8C,KAAK,CAAA,CACpBmB,KAAAA,CAAM,OAAOC,GAAAA,KAAAA;AACZ,YAAA,MAAM,IAAIxE,aAAAA,CACR,CAAA,MAAA,EAASgE,IAAAA,CAAAA,oFAAAA,CAAAA,EACT;AAAE1D,cAAAA,GAAAA,EAAKmE,sBAAYnE,GAAAA,CAAAA;cAAMF,MAAAA,EAAQ;AAACoE,gBAAAA;;aAAK,CAAA;UAE3C,CAAA,CAAA,CACCE,KAAKC,kBAAAA,CAAAA;AAER,UAAA,MAAMC,WAAAA,GAAc,MAAMtC,IAAAA,CAAKW,OAAAA,CAAQG,QAAOS,QAAAA,CAAAA;AAC9C,UAAA,IAAIe,WAAAA,KAAgBnE,SAASE,KAAAA,EAAO;AAClCqD,YAAAA,IAAAA,GAAO1D,GAAAA,CAAIW,KAAAA,CAAM4D,EAAAA,CAAG,CAAA,CAAA,EAAIlD,IAAAA;UAC1B,CAAA,MAAA,IAAWiD,WAAAA,KAAgBnE,SAASI,IAAAA,EAAM;AACxCmD,YAAAA,IAAAA,GAAO,IAAA,CAAKC,QAAAA,CAASD,IAAAA,CAAAA,CAAMc,IAAAA;UAC7B,CAAA,MAAA,IAAWF,WAAAA,KAAgBnE,SAASG,IAAAA,EAAM;AACxCoD,YAAAA,IAAAA,GAAO,IAAA,CAAKC,QAAAA,CAASD,IAAAA,CAAAA,CAAMG,OAAAA;AAC7B,UAAA,CAAA,MAAA,IAAW,CAACS,WAAAA,IAAeA,WAAAA,KAAgBnE,QAAAA,CAASK,IAAAA,EAAM;AACxDkD,YAAAA,IAAAA,GAAO,IAAA,CAAKC,QAAAA,CAASD,IAAAA,CAAAA,CAAMA,QAAQvD,QAAAA,CAASM,GAAAA;UAC9C,CAAA,MAAO;AACLiD,YAAAA,IAAAA,GAAOY,WAAAA;AACT,UAAA;AAEA,UAAA,MAAMG,UAAAA,GAA6C;AACjDpD,YAAAA,IAAAA;YACAyB,KAAAA,EAAAA,MAAAA;AACAY,YAAAA;AACF,WAAA;AACA1D,UAAAA,GAAAA,CAAIW,KAAAA,CAAM+D,KAAKD,UAAAA,CAAAA;AACfzE,UAAAA,GAAAA,CAAI8C,QAAQ2B,UAAAA,CAAW3B,KAAAA;AAEvB,UAAA,MAAMM,UAAAA,CAAW1C,OAAAA,CAAQqD,IAAAA,CAAK,SAAA,EAAW;AACvC/D,YAAAA,GAAAA,EAAKmE,sBAAYnE,GAAAA,CAAAA;AACjB8C,YAAAA,KAAAA,EAAO2B,UAAAA,CAAW3B,KAAAA;AAClBd,YAAAA,IAAAA,EAAMyC,UAAAA,CAAWpD,IAAAA;AACjBqC,YAAAA,IAAAA,EAAMe,UAAAA,CAAWf;WACnB,CAAA;AACF,QAAA,CAAA,CAAA,OAASiB,KAAAA,EAAO;AACd,UAAA,MAAMvB,UAAAA,CAAW1C,OAAAA,CAAQqD,IAAAA,CAAK,OAAA,EAAS;AACrC/D,YAAAA,GAAAA,EAAKmE,sBAAYnE,GAAAA,CAAAA;YACjBgC,IAAAA,EAAM0B,IAAAA;AACNiB,YAAAA;WACF,CAAA;AACA,UAAA,MAAMA,KAAAA;AACR,QAAA;AACF,MAAA;AAEA3E,MAAAA,GAAAA,CAAIsD,MAAAA,GAAS,MAAA,CAAO,IAAA,CAAKvC,KAAAA,CAAM6D,gBAAgB,IAAA,CAAK7D,KAAAA,CAAMoB,MAAAA,EACvD6B,UAAAA,CAAWhE,GAAAA,CAAI8C,KAAK,CAAA,CACpBmB,KAAAA,CAAM,CAACC,GAAAA,KAAAA;AACN,QAAA,MAAM,IAAIxE,cACR,CAAA,kFAAA,CAAA,EACA;AAAEM,UAAAA,GAAAA,EAAKmE,sBAAYnE,GAAAA,CAAAA;UAAMF,MAAAA,EAAQ;AAACoE,YAAAA;;SAAK,CAAA;MAE3C,CAAA,CAAA;AACF,MAAA,OAAOlE,GAAAA;IACT,CAAA,CAAA;AAEJ,EAAA;AAEA6E,EAAAA,OAAAA,CAAyBxD,IAAAA,EAAuD;AAC9E,IAAA,IAAI,IAAA,CAAKR,cAAcQ,IAAAA,EAAM;AAC3B,MAAA,IAAA,CAAKR,SAAAA,GAAYC,MAAAA;AACnB,IAAA;AACA,IAAA,IAAA,CAAKH,KAAAA,CAAMmE,OAAOzD,IAAAA,CAAAA;AAClB,IAAA,OAAO,IAAA;AACT,EAAA;AAEAuB,EAAAA,MAAAA,CAIEmC,SAAAA,EAK+C;AAC/C,IAAA,OAAO,OAAOjC,OAAOkC,GAAAA,KAAAA;AACnB,MAAA,MAAMC,cAAcF,SAAAA,EAAWhE,KAAAA,GAAQgE,SAAAA,CAAUhE,KAAAA,CAAM+B,KAAAA,CAAAA,GAASA,KAAAA;AAChE,MAAA,MAAMQ,MAAAA,GAAS,MAAM,IAAA,CAAKtD,GAAAA,CAAIiF,WAAAA,EAAa;AAAErB,QAAAA,KAAAA,EAAOmB,SAAAA,EAAWnB,KAAAA;AAAOV,QAAAA,MAAAA,EAAQ8B,GAAAA,CAAI9B;OAAO,CAAA;AACzF,MAAA,MAAMgC,YAAAA,GAAeH,WAAWI,MAAAA,GAASJ,SAAAA,CAAUI,OAAO7B,MAAAA,CAAOR,KAAK,IAAIQ,MAAAA,CAAOR,KAAAA;AACjFsC,MAAAA,MAAAA,CAAOC,MAAAA,CAAOvC,OAAOoC,YAAAA,CAAAA;AACrB,MAAA,OAAOH,SAAAA,CAAUrB,IAAAA;AACnB,IAAA,CAAA;AACF,EAAA;AAEUC,EAAAA,QAAAA,CAASE,OAAAA,EAAgB;AACjC,IAAA,MAAMlD,KAAAA,GAAQ,KAAKc,QAAAA,EAAQ;AAC3B,IAAA,MAAM6D,KAAAA,GAAQ3E,KAAAA,CAAM4E,OAAAA,CAAQ1B,OAAAA,CAAAA;AAC5B,IAAA,OAAO;MACLW,IAAAA,EAAM7D,KAAAA,CAAM2E,QAAQ,CAAA,CAAA;AACpBzB,MAAAA,OAAAA,EAASlD,MAAM2E,KAAAA,CAAAA;MACf5B,IAAAA,EAAM/C,KAAAA,CAAM2E,QAAQ,CAAA;AACtB,KAAA;AACF,EAAA;EAEAE,cAAAA,GAAiB;AACf,IAAA,OAAO;MACLzE,KAAAA,EAAO0E,WAAAA,CAAK,KAAK1E,KAAAA,EAAO;AAAC,QAAA,QAAA;AAAU,QAAA;AAAe,OAAA,CAAA;AAClDL,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;AACdG,MAAAA,SAAAA,EAAW,IAAA,CAAKA,SAAAA;AAChBF,MAAAA,KAAAA,EAAO,IAAA,CAAKA;AACd,KAAA;AACF,EAAA;AAEA+E,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7DP,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAMM,QAAAA,CAAAA;AACpB,IAAA,IAAA,CAAK5E,KAAAA,CAAMoB,MAAAA,KAAWyD,KAAAA,CAAEC,GAAAA,EAAG;AAC3B,IAAA,IAAA,CAAK9E,KAAAA,CAAM6D,YAAAA,KAAiBgB,KAAAA,CAAEC,GAAAA,EAAG;AACnC,EAAA;AACF","file":"workflow.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { z, ZodSchema } from \"zod\";\nimport { Serializable } from \"@/internals/serializable.js\";\nimport { Callback, Emitter } from \"@/emitter/emitter.js\";\nimport { RunContext } from \"@/context.js\";\nimport { omit, pick, toCamelCase } from \"remeda\";\nimport { deepCopy, shallowCopy } from \"@/serializer/utils.js\";\nimport { FrameworkError, ValueError } from \"@/errors.js\";\n\n// eslint-disable-next-line @typescript-eslint/no-invalid-void-type\ntype WorkflowStepHandlerReturn<K extends string> = K | ReservedStep | void;\n\nexport interface WorkflowRun<T extends ZodSchema, T2 extends ZodSchema, K extends string> {\n  result: z.output<T2>;\n  steps: WorkflowStepRes<T, K>[];\n  state: z.output<T>;\n}\n\nexport interface WorkflowRunOptions<K extends string> {\n  start?: K;\n  signal?: AbortSignal;\n}\n\nexport interface WorkflowStepDef<T extends ZodSchema, K extends string> {\n  schema: T;\n  handler: WorkflowStepHandler<T, K>;\n}\n\nexport interface WorkflowStepRes<T extends ZodSchema, K extends string> {\n  name: K;\n  next: K | ReservedStep;\n  state: z.output<T>;\n}\n\nexport interface WorkflowRunContext<T extends ZodSchema, K extends string> {\n  steps: WorkflowStepRes<T, K>[];\n  signal: AbortSignal;\n  abort: (reason?: Error) => void;\n}\n\nexport type WorkflowStepHandler<T extends ZodSchema, K extends string> = (\n  state: z.output<T>,\n  context: WorkflowRunContext<T, K>,\n) => Promise<WorkflowStepHandlerReturn<K>> | WorkflowStepHandlerReturn<K>;\n\nexport interface WorkflowEvents<T extends ZodSchema, T2 extends ZodSchema, K extends string> {\n  start: Callback<{ step: K; run: WorkflowRun<T, T2, K> }>;\n  error: Callback<{\n    step: K;\n    error: Error;\n    run: WorkflowRun<T, T2, K>;\n  }>;\n  success: Callback<{\n    step: K;\n    next: K | ReservedStep;\n    state: z.output<T>;\n    run: WorkflowRun<T, T2, K>;\n  }>;\n}\n\ninterface FlowInput<TS extends ZodSchema, TS2 extends ZodSchema = TS> {\n  name?: string;\n  schema: TS;\n  outputSchema?: TS2;\n}\n\ntype RunStep<K extends string> = K | typeof Workflow.END;\ntype ReservedStep =\n  | typeof Workflow.START\n  | typeof Workflow.NEXT\n  | typeof Workflow.PREV\n  | typeof Workflow.END\n  | typeof Workflow.SELF;\nexport type FlowStepName<K extends string> = K | ReservedStep;\n\nexport class WorkflowError<\n  T extends ZodSchema,\n  T2 extends ZodSchema,\n  K extends string,\n> extends FrameworkError {\n  constructor(message: string, extra?: { run?: WorkflowRun<T, T2, K>; errors?: Error[] }) {\n    super(message, extra?.errors, {\n      context: extra?.run ?? {},\n      isRetryable: false,\n      isFatal: true,\n    });\n  }\n}\n\nexport class Workflow<\n  TInput extends ZodSchema,\n  TOutput extends ZodSchema = TInput,\n  TKeys extends string = string,\n> extends Serializable {\n  public static readonly START = \"__start__\";\n  public static readonly SELF = \"__self__\";\n  public static readonly PREV = \"__prev__\";\n  public static readonly NEXT = \"__next__\";\n  public static readonly END = \"__end__\";\n\n  public readonly emitter: Emitter<WorkflowEvents<TInput, TOutput, TKeys>>;\n\n  protected readonly steps = new Map<string, WorkflowStepDef<TInput, TKeys>>();\n  protected startStep: TKeys | undefined = undefined;\n\n  constructor(protected readonly input: FlowInput<TInput, TOutput>) {\n    super();\n    this.emitter = Emitter.root.child({\n      namespace: [\"workflow\", toCamelCase(input?.name ?? \"\")].filter(Boolean),\n      creator: this,\n    });\n  }\n\n  getSteps() {\n    return Array.from(this.steps.keys()) as TKeys[];\n  }\n\n  get name() {\n    return this.input.name ?? \"\";\n  }\n\n  get schemas() {\n    return pick(this.input, [\"schema\", \"outputSchema\"]);\n  }\n\n  addStep<L extends string>(\n    name: L,\n    handler: WorkflowStepHandler<TInput | TOutput, TKeys>,\n  ): Workflow<TInput, TOutput, L | TKeys>;\n  addStep<L extends string, TFlow extends AnyWorkflowFlow>(\n    name: L,\n    flow: Workflow.pipeable<this, TFlow>,\n  ): Workflow<TInput, TOutput, L | TKeys>;\n  addStep<L extends string>(\n    name: L,\n    step: WorkflowStepHandler<TInput | TOutput, TKeys> | AnyWorkflowFlow,\n  ): Workflow<TInput, TOutput, L | TKeys> {\n    return this._addStep(name, step);\n  }\n\n  addStrictStep<L extends string, TI2 extends ZodSchema>(\n    name: L,\n    schema: TI2,\n    handler: WorkflowStepHandler<TI2, TKeys>,\n  ): Workflow<TInput, TOutput, L | TKeys>;\n  addStrictStep<L extends string, TI2 extends ZodSchema, TFlow extends AnyWorkflowFlow>(\n    name: L,\n    schema: TI2,\n    flow: Workflow.pipeable<Workflow<TI2, TOutput, TKeys>, TFlow>,\n  ): Workflow<TInput, TOutput, L | TKeys>;\n  addStrictStep<L extends string, TI2 extends ZodSchema>(\n    name: L,\n    schema: TI2,\n    step: WorkflowStepHandler<TI2, TKeys> | AnyWorkflowFlow,\n  ): Workflow<TInput, TOutput, L | TKeys> {\n    return this._addStep(name, schema, step);\n  }\n\n  protected _addStep<TI2 extends ZodSchema = TInput, L extends string = TKeys>(\n    name: L,\n    schemaOrStep: TI2 | WorkflowStepHandler<TInput, TKeys> | AnyWorkflowFlow,\n    stepOrEmpty?: WorkflowStepHandler<TI2, TKeys> | AnyWorkflowFlow,\n  ): Workflow<TInput, TOutput, L | TKeys> {\n    if (!name.trim()) {\n      throw new ValueError(`Step name cannot be empty!`);\n    }\n    if (this.steps.has(name)) {\n      throw new ValueError(`Step '${name}' already exists!`);\n    }\n    if (name === Workflow.END) {\n      throw new ValueError(`The name '${name}' cannot be used!`);\n    }\n\n    const schema = (schemaOrStep && stepOrEmpty ? schemaOrStep : this.input.schema) as TInput;\n    const stepOrFlow = stepOrEmpty || schemaOrStep;\n\n    this.steps.set(name, {\n      schema,\n      handler: stepOrFlow instanceof Workflow ? stepOrFlow.asStep({}) : stepOrFlow,\n    } as WorkflowStepDef<TInput, TKeys>);\n\n    return this as unknown as Workflow<TInput, TOutput, L | TKeys>;\n  }\n\n  setStart(name: TKeys) {\n    this.startStep = name;\n    return this;\n  }\n\n  run(state: z.input<TInput>, options: WorkflowRunOptions<TKeys> = {}) {\n    return RunContext.enter(\n      this,\n      { signal: options?.signal, params: [state, options] as const },\n      async (runContext): Promise<WorkflowRun<TInput, TOutput, TKeys>> => {\n        const run: WorkflowRun<TInput, TOutput, TKeys> = {\n          steps: [],\n          state: this.input.schema.parse(state),\n          result: undefined as z.output<TOutput>,\n        };\n        const handlers: WorkflowRunContext<TInput, TKeys> = {\n          steps: run.steps,\n          signal: runContext.signal,\n          abort: (reason) => runContext.abort(reason),\n        };\n\n        let next: RunStep<TKeys> =\n          this.findStep(options?.start || this.startStep || this.getSteps()[0]).current ??\n          Workflow.END;\n\n        while (next && next !== Workflow.END) {\n          const name = next;\n          const step = this.steps.get(next);\n          if (!step) {\n            throw new WorkflowError(`Step '${next}' was not found.`, { run });\n          }\n          await runContext.emitter.emit(\"start\", { run, step: next });\n          try {\n            const state = await step.schema\n              .parseAsync(run.state)\n              .catch(async (err: Error) => {\n                throw new WorkflowError(\n                  `Step '${next}' cannot be executed because the provided input doesn't adhere to the step's schema.`,\n                  { run: shallowCopy(run), errors: [err] },\n                );\n              })\n              .then(deepCopy);\n\n            const nextStepRaw = await step.handler(state, handlers);\n            if (nextStepRaw === Workflow.START) {\n              next = run.steps.at(0)?.name!;\n            } else if (nextStepRaw === Workflow.PREV) {\n              next = this.findStep(next).prev;\n            } else if (nextStepRaw === Workflow.SELF) {\n              next = this.findStep(next).current;\n            } else if (!nextStepRaw || nextStepRaw === Workflow.NEXT) {\n              next = this.findStep(next).next || Workflow.END;\n            } else {\n              next = nextStepRaw;\n            }\n\n            const stepResult: WorkflowStepRes<TInput, TKeys> = {\n              name,\n              state,\n              next,\n            };\n            run.steps.push(stepResult);\n            run.state = stepResult.state;\n\n            await runContext.emitter.emit(\"success\", {\n              run: shallowCopy(run),\n              state: stepResult.state,\n              step: stepResult.name,\n              next: stepResult.next,\n            });\n          } catch (error) {\n            await runContext.emitter.emit(\"error\", {\n              run: shallowCopy(run),\n              step: next as TKeys,\n              error,\n            });\n            throw error;\n          }\n        }\n\n        run.result = await (this.input.outputSchema ?? this.input.schema)\n          .parseAsync(run.state)\n          .catch((err) => {\n            throw new WorkflowError(\n              `Workflow has ended but it's state does not adhere to the workflow's output schema.`,\n              { run: shallowCopy(run), errors: [err] },\n            );\n          });\n        return run;\n      },\n    );\n  }\n\n  delStep<L extends TKeys>(name: L): Workflow<TInput, TOutput, Exclude<TKeys, L>> {\n    if (this.startStep === name) {\n      this.startStep = undefined;\n    }\n    this.steps.delete(name);\n    return this as unknown as Workflow<TInput, TOutput, Exclude<TKeys, L>>;\n  }\n\n  asStep<\n    TInput2 extends ZodSchema = TInput,\n    TOutput2 extends ZodSchema = TOutput,\n    TKeys2 extends string = TKeys,\n  >(overrides: {\n    input?: (input: z.output<TInput2>) => z.output<TInput> | z.input<TInput>;\n    output?: (output: z.output<TOutput>) => z.output<TOutput2> | z.input<TOutput2>;\n    start?: TKeys;\n    next?: FlowStepName<TKeys2>;\n  }): WorkflowStepHandler<TInput2, TKeys | TKeys2> {\n    return async (state, ctx) => {\n      const mappedInput = overrides?.input ? overrides.input(state) : state;\n      const result = await this.run(mappedInput, { start: overrides?.start, signal: ctx.signal });\n      const mappedOutput = overrides?.output ? overrides.output(result.state) : result.state;\n      Object.assign(state, mappedOutput);\n      return overrides.next;\n    };\n  }\n\n  protected findStep(current: TKeys) {\n    const steps = this.getSteps();\n    const index = steps.indexOf(current);\n    return {\n      prev: steps[index - 1],\n      current: steps[index],\n      next: steps[index + 1],\n    };\n  }\n\n  createSnapshot() {\n    return {\n      input: omit(this.input, [\"schema\", \"outputSchema\"]),\n      emitter: this.emitter,\n      startStep: this.startStep,\n      steps: this.steps,\n    };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n    this.input.schema ??= z.any() as unknown as TInput;\n    this.input.outputSchema ??= z.any() as unknown as TOutput;\n  }\n}\n\n// eslint-disable-next-line @typescript-eslint/no-namespace\nexport namespace Workflow {\n  export type pipeable<T extends AnyWorkflowFlow, T2 extends AnyWorkflowFlow> =\n    Workflow.state<T> extends Workflow.input<T2> ? T2 : never;\n  export type input<T> = T extends Workflow<infer A, any, any> ? z.input<A> : never;\n  export type state<T> = T extends Workflow<infer A, any, any> ? z.output<A> : never;\n  export type output<T> = T extends Workflow<any, infer A, any> ? z.output<A> : never;\n  export type run<T> = T extends Workflow<infer A, infer B, infer C> ? WorkflowRun<A, B, C> : never;\n}\n\nexport type AnyWorkflowFlow = Workflow<any, any, any>;\n"]}

package/dist/workflows/workflow.d.cts

@@ -1,7 +1,7 @@
 import { Run } from '../context.cjs';
 import { ZodSchema, z } from 'zod';
 import { Serializable } from '../internals/serializable.cjs';
-import { E as Emitter, C as Callback } from '../emitter-UkIkNrek.cjs';
+import { E as Emitter, C as Callback } from '../emitter-D4OcelZ9.cjs';
 import { FrameworkError } from '../errors.cjs';
 import '../internals/helpers/promise.cjs';
 import '../internals/types.cjs';

package/dist/workflows/workflow.d.ts

@@ -1,7 +1,7 @@
 import { Run } from '../context.js';
 import { ZodSchema, z } from 'zod';
 import { Serializable } from '../internals/serializable.js';
-import { E as Emitter, C as Callback } from '../emitter-CL56qab4.js';
+import { E as Emitter, C as Callback } from '../emitter-kHxkUxco.js';
 import { FrameworkError } from '../errors.js';
 import '../internals/helpers/promise.js';
 import '../internals/types.js';

package/dist/workflows/workflow.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/workflows/workflow.ts"],"names":["WorkflowError","FrameworkError","message","extra","errors","context","run","isRetryable","isFatal","Workflow","Serializable","START","SELF","PREV","NEXT","END","emitter","steps","Map","startStep","undefined","input","Emitter","root","child","namespace","toCamelCase","name","filter","Boolean","creator","getSteps","Array","from","keys","schemas","pick","addStep","step","_addStep","addStrictStep","schema","schemaOrStep","stepOrEmpty","trim","ValueError","has","stepOrFlow","set","handler","asStep","setStart","state","options","RunContext","enter","signal","params","runContext","parse","result","handlers","abort","reason","next","findStep","start","current","get","emit","parseAsync","catch","err","shallowCopy","then","deepCopy","nextStepRaw","at","prev","stepResult","push","error","outputSchema","delStep","delete","overrides","ctx","mappedInput","mappedOutput","output","Object","assign","index","indexOf","createSnapshot","omit","loadSnapshot","snapshot","z","any"],"mappings":";;;;;;;;;;AA+EO,MAAMA,sBAIHC,cAAAA,CAAAA;EAnFV;;;AAoFE,EAAA,WAAA,CAAYC,SAAiBC,KAAAA,EAA2D;AACtF,IAAA,KAAA,CAAMD,OAAAA,EAASC,OAAOC,MAAAA,EAAQ;MAC5BC,OAAAA,EAASF,KAAAA,EAAOG,OAAO,EAAC;MACxBC,WAAAA,EAAa,KAAA;MACbC,OAAAA,EAAS;KACX,CAAA;AACF;AACF;AAEO,MAAMC,iBAIHC,YAAAA,CAAAA;EAjGV;;;;AAkGE,EAAA,OAAuBC,KAAAA,GAAQ,WAAA;AAC/B,EAAA,OAAuBC,IAAAA,GAAO,UAAA;AAC9B,EAAA,OAAuBC,IAAAA,GAAO,UAAA;AAC9B,EAAA,OAAuBC,IAAAA,GAAO,UAAA;AAC9B,EAAA,OAAuBC,GAAAA,GAAM,SAAA;AAEbC,EAAAA,OAAAA;AAEGC,EAAAA,KAAAA,uBAAYC,GAAAA,EAAAA;EACrBC,SAAAA,GAA+BC,MAAAA;AAEzC,EAAA,WAAA,CAA+BC,KAAAA,EAAmC;AAChE,IAAA,KAAA,EAAK,EAAA,KADwBA,KAAAA,GAAAA,KAAAA;AAE7B,IAAA,IAAA,CAAKL,OAAAA,GAAUM,OAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAAM;MAChCC,SAAAA,EAAW;AAAC,QAAA,UAAA;QAAYC,WAAAA,CAAYL,KAAAA,EAAOM,QAAQ,EAAA;AAAKC,OAAAA,CAAAA,MAAAA,CAAOC,OAAAA,CAAAA;MAC/DC,OAAAA,EAAS;KACX,CAAA;AACF;EAEAC,QAAAA,GAAW;AACT,IAAA,OAAOC,KAAAA,CAAMC,IAAAA,CAAK,IAAA,CAAKhB,KAAAA,CAAMiB,MAAI,CAAA;AACnC;AAEA,EAAA,IAAIP,IAAAA,GAAO;AACT,IAAA,OAAO,IAAA,CAAKN,MAAMM,IAAAA,IAAQ,EAAA;AAC5B;AAEA,EAAA,IAAIQ,OAAAA,GAAU;AACZ,IAAA,OAAOC,IAAAA,CAAK,KAAKf,KAAAA,EAAO;AAAC,MAAA,QAAA;AAAU,MAAA;AAAe,KAAA,CAAA;AACpD;AAUAgB,EAAAA,OAAAA,CACEV,MACAW,IAAAA,EACsC;AACtC,IAAA,OAAO,IAAA,CAAKC,QAAAA,CAASZ,IAAAA,EAAMW,IAAAA,CAAAA;AAC7B;EAYAE,aAAAA,CACEb,IAAAA,EACAc,QACAH,IAAAA,EACsC;AACtC,IAAA,OAAO,IAAA,CAAKC,QAAAA,CAASZ,IAAAA,EAAMc,MAAAA,EAAQH,IAAAA,CAAAA;AACrC;EAEUC,QAAAA,CACRZ,IAAAA,EACAe,cACAC,WAAAA,EACsC;AACtC,IAAA,IAAI,CAAChB,IAAAA,CAAKiB,IAAAA,EAAI,EAAI;AAChB,MAAA,MAAM,IAAIC,WAAW,CAAA,0BAAA,CAA4B,CAAA;AACnD;AACA,IAAA,IAAI,IAAA,CAAK5B,KAAAA,CAAM6B,GAAAA,CAAInB,IAAAA,CAAAA,EAAO;AACxB,MAAA,MAAM,IAAIkB,UAAAA,CAAW,CAAA,MAAA,EAASlB,IAAAA,CAAAA,iBAAAA,CAAuB,CAAA;AACvD;AACA,IAAA,IAAIA,IAAAA,KAASlB,SAASM,GAAAA,EAAK;AACzB,MAAA,MAAM,IAAI8B,UAAAA,CAAW,CAAA,UAAA,EAAalB,IAAAA,CAAAA,iBAAAA,CAAuB,CAAA;AAC3D;AAEA,IAAA,MAAMc,MAAAA,GAAUC,YAAAA,IAAgBC,WAAAA,GAAcD,YAAAA,GAAe,KAAKrB,KAAAA,CAAMoB,MAAAA;AACxE,IAAA,MAAMM,aAAaJ,WAAAA,IAAeD,YAAAA;AAElC,IAAA,IAAA,CAAKzB,KAAAA,CAAM+B,IAAIrB,IAAAA,EAAM;AACnBc,MAAAA,MAAAA;AACAQ,MAAAA,OAAAA,EAASF,sBAAsBtC,QAAAA,GAAWsC,UAAAA,CAAWG,MAAAA,CAAO,EAAC,CAAA,GAAKH;KACpE,CAAA;AAEA,IAAA,OAAO,IAAA;AACT;AAEAI,EAAAA,QAAAA,CAASxB,IAAAA,EAAa;AACpB,IAAA,IAAA,CAAKR,SAAAA,GAAYQ,IAAAA;AACjB,IAAA,OAAO,IAAA;AACT;EAEArB,GAAAA,CAAI8C,KAAAA,EAAwBC,OAAAA,GAAqC,EAAC,EAAG;AACnE,IAAA,OAAOC,UAAAA,CAAWC,MAChB,IAAA,EACA;AAAEC,MAAAA,MAAAA,EAAQH,OAAAA,EAASG,MAAAA;MAAQC,MAAAA,EAAQ;AAACL,QAAAA,KAAAA;AAAOC,QAAAA;;AAAkB,KAAA,EAC7D,OAAOK,UAAAA,KAAAA;AACL,MAAA,MAAMpD,GAAAA,GAA2C;AAC/CW,QAAAA,KAAAA,EAAO,EAAA;AACPmC,QAAAA,KAAAA,EAAO,IAAA,CAAK/B,KAAAA,CAAMoB,MAAAA,CAAOkB,KAAAA,CAAMP,KAAAA,CAAAA;QAC/BQ,MAAAA,EAAQxC;AACV,OAAA;AACA,MAAA,MAAMyC,QAAAA,GAA8C;AAClD5C,QAAAA,KAAAA,EAAOX,GAAAA,CAAIW,KAAAA;AACXuC,QAAAA,MAAAA,EAAQE,UAAAA,CAAWF,MAAAA;AACnBM,QAAAA,KAAAA,kBAAO,MAAA,CAAA,CAACC,MAAAA,KAAWL,UAAAA,CAAWI,KAAAA,CAAMC,MAAAA,CAAAA,EAA7B,OAAA;AACT,OAAA;AAEA,MAAA,IAAIC,IAAAA,GACF,IAAA,CAAKC,QAAAA,CAASZ,OAAAA,EAASa,SAAS,IAAA,CAAK/C,SAAAA,IAAa,IAAA,CAAKY,QAAAA,EAAQ,CAAG,CAAA,CAAE,CAAA,CAAEoC,WACtE1D,QAAAA,CAASM,GAAAA;AAEX,MAAA,OAAOiD,IAAAA,IAAQA,IAAAA,KAASvD,QAAAA,CAASM,GAAAA,EAAK;AACpC,QAAA,MAAMY,IAAAA,GAAOqC,IAAAA;AACb,QAAA,MAAM1B,IAAAA,GAAO,IAAA,CAAKrB,KAAAA,CAAMmD,GAAAA,CAAIJ,IAAAA,CAAAA;AAC5B,QAAA,IAAI,CAAC1B,IAAAA,EAAM;AACT,UAAA,MAAM,IAAItC,aAAAA,CAAc,CAAA,MAAA,EAASgE,IAAAA,CAAAA,gBAAAA,CAAAA,EAAwB;AAAE1D,YAAAA;WAAI,CAAA;AACjE;AACA,QAAA,MAAMoD,UAAAA,CAAW1C,OAAAA,CAAQqD,IAAAA,CAAK,OAAA,EAAS;AAAE/D,UAAAA,GAAAA;UAAKgC,IAAAA,EAAM0B;SAAK,CAAA;AACzD,QAAA,IAAI;AACF,UAAA,MAAMZ,MAAAA,GAAQ,MAAMd,IAAAA,CAAKG,MAAAA,CACtB6B,UAAAA,CAAWhE,IAAI8C,KAAK,CAAA,CACpBmB,KAAAA,CAAM,OAAOC,GAAAA,KAAAA;AACZ,YAAA,MAAM,IAAIxE,aAAAA,CACR,CAAA,MAAA,EAASgE,IAAAA,CAAAA,oFAAAA,CAAAA,EACT;AAAE1D,cAAAA,GAAAA,EAAKmE,YAAYnE,GAAAA,CAAAA;cAAMF,MAAAA,EAAQ;AAACoE,gBAAAA;;aAAK,CAAA;WAE3C,CAAA,CACCE,KAAKC,QAAAA,CAAAA;AAER,UAAA,MAAMC,WAAAA,GAAc,MAAMtC,IAAAA,CAAKW,OAAAA,CAAQG,QAAOS,QAAAA,CAAAA;AAC9C,UAAA,IAAIe,WAAAA,KAAgBnE,SAASE,KAAAA,EAAO;AAClCqD,YAAAA,IAAAA,GAAO1D,GAAAA,CAAIW,KAAAA,CAAM4D,EAAAA,CAAG,CAAA,CAAA,EAAIlD,IAAAA;WAC1B,MAAA,IAAWiD,WAAAA,KAAgBnE,SAASI,IAAAA,EAAM;AACxCmD,YAAAA,IAAAA,GAAO,IAAA,CAAKC,QAAAA,CAASD,IAAAA,CAAAA,CAAMc,IAAAA;WAC7B,MAAA,IAAWF,WAAAA,KAAgBnE,SAASG,IAAAA,EAAM;AACxCoD,YAAAA,IAAAA,GAAO,IAAA,CAAKC,QAAAA,CAASD,IAAAA,CAAAA,CAAMG,OAAAA;AAC7B,WAAA,MAAA,IAAW,CAACS,WAAAA,IAAeA,WAAAA,KAAgBnE,QAAAA,CAASK,IAAAA,EAAM;AACxDkD,YAAAA,IAAAA,GAAO,IAAA,CAAKC,QAAAA,CAASD,IAAAA,CAAAA,CAAMA,QAAQvD,QAAAA,CAASM,GAAAA;WAC9C,MAAO;AACLiD,YAAAA,IAAAA,GAAOY,WAAAA;AACT;AAEA,UAAA,MAAMG,UAAAA,GAA6C;AACjDpD,YAAAA,IAAAA;YACAyB,KAAAA,EAAAA,MAAAA;AACAY,YAAAA;AACF,WAAA;AACA1D,UAAAA,GAAAA,CAAIW,KAAAA,CAAM+D,KAAKD,UAAAA,CAAAA;AACfzE,UAAAA,GAAAA,CAAI8C,QAAQ2B,UAAAA,CAAW3B,KAAAA;AAEvB,UAAA,MAAMM,UAAAA,CAAW1C,OAAAA,CAAQqD,IAAAA,CAAK,SAAA,EAAW;AACvC/D,YAAAA,GAAAA,EAAKmE,YAAYnE,GAAAA,CAAAA;AACjB8C,YAAAA,KAAAA,EAAO2B,UAAAA,CAAW3B,KAAAA;AAClBd,YAAAA,IAAAA,EAAMyC,UAAAA,CAAWpD,IAAAA;AACjBqC,YAAAA,IAAAA,EAAMe,UAAAA,CAAWf;WACnB,CAAA;AACF,SAAA,CAAA,OAASiB,KAAAA,EAAO;AACd,UAAA,MAAMvB,UAAAA,CAAW1C,OAAAA,CAAQqD,IAAAA,CAAK,OAAA,EAAS;AACrC/D,YAAAA,GAAAA,EAAKmE,YAAYnE,GAAAA,CAAAA;YACjBgC,IAAAA,EAAM0B,IAAAA;AACNiB,YAAAA;WACF,CAAA;AACA,UAAA,MAAMA,KAAAA;AACR;AACF;AAEA3E,MAAAA,GAAAA,CAAIsD,MAAAA,GAAS,MAAA,CAAO,IAAA,CAAKvC,KAAAA,CAAM6D,gBAAgB,IAAA,CAAK7D,KAAAA,CAAMoB,MAAAA,EACvD6B,UAAAA,CAAWhE,GAAAA,CAAI8C,KAAK,CAAA,CACpBmB,KAAAA,CAAM,CAACC,GAAAA,KAAAA;AACN,QAAA,MAAM,IAAIxE,cACR,CAAA,kFAAA,CAAA,EACA;AAAEM,UAAAA,GAAAA,EAAKmE,YAAYnE,GAAAA,CAAAA;UAAMF,MAAAA,EAAQ;AAACoE,YAAAA;;SAAK,CAAA;OAE3C,CAAA;AACF,MAAA,OAAOlE,GAAAA;KACT,CAAA;AAEJ;AAEA6E,EAAAA,OAAAA,CAAyBxD,IAAAA,EAAuD;AAC9E,IAAA,IAAI,IAAA,CAAKR,cAAcQ,IAAAA,EAAM;AAC3B,MAAA,IAAA,CAAKR,SAAAA,GAAYC,MAAAA;AACnB;AACA,IAAA,IAAA,CAAKH,KAAAA,CAAMmE,OAAOzD,IAAAA,CAAAA;AAClB,IAAA,OAAO,IAAA;AACT;AAEAuB,EAAAA,MAAAA,CAIEmC,SAAAA,EAK+C;AAC/C,IAAA,OAAO,OAAOjC,OAAOkC,GAAAA,KAAAA;AACnB,MAAA,MAAMC,cAAcF,SAAAA,EAAWhE,KAAAA,GAAQgE,SAAAA,CAAUhE,KAAAA,CAAM+B,KAAAA,CAAAA,GAASA,KAAAA;AAChE,MAAA,MAAMQ,MAAAA,GAAS,MAAM,IAAA,CAAKtD,GAAAA,CAAIiF,WAAAA,EAAa;AAAErB,QAAAA,KAAAA,EAAOmB,SAAAA,EAAWnB,KAAAA;AAAOV,QAAAA,MAAAA,EAAQ8B,GAAAA,CAAI9B;OAAO,CAAA;AACzF,MAAA,MAAMgC,YAAAA,GAAeH,WAAWI,MAAAA,GAASJ,SAAAA,CAAUI,OAAO7B,MAAAA,CAAOR,KAAK,IAAIQ,MAAAA,CAAOR,KAAAA;AACjFsC,MAAAA,MAAAA,CAAOC,MAAAA,CAAOvC,OAAOoC,YAAAA,CAAAA;AACrB,MAAA,OAAOH,SAAAA,CAAUrB,IAAAA;AACnB,KAAA;AACF;AAEUC,EAAAA,QAAAA,CAASE,OAAAA,EAAgB;AACjC,IAAA,MAAMlD,KAAAA,GAAQ,KAAKc,QAAAA,EAAQ;AAC3B,IAAA,MAAM6D,KAAAA,GAAQ3E,KAAAA,CAAM4E,OAAAA,CAAQ1B,OAAAA,CAAAA;AAC5B,IAAA,OAAO;MACLW,IAAAA,EAAM7D,KAAAA,CAAM2E,QAAQ,CAAA,CAAA;AACpBzB,MAAAA,OAAAA,EAASlD,MAAM2E,KAAAA,CAAAA;MACf5B,IAAAA,EAAM/C,KAAAA,CAAM2E,QAAQ,CAAA;AACtB,KAAA;AACF;EAEAE,cAAAA,GAAiB;AACf,IAAA,OAAO;MACLzE,KAAAA,EAAO0E,IAAAA,CAAK,KAAK1E,KAAAA,EAAO;AAAC,QAAA,QAAA;AAAU,QAAA;AAAe,OAAA,CAAA;AAClDL,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;AACdG,MAAAA,SAAAA,EAAW,IAAA,CAAKA,SAAAA;AAChBF,MAAAA,KAAAA,EAAO,IAAA,CAAKA;AACd,KAAA;AACF;AAEA+E,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7DP,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAMM,QAAAA,CAAAA;AACpB,IAAA,IAAA,CAAK5E,KAAAA,CAAMoB,MAAAA,KAAWyD,CAAAA,CAAEC,GAAAA,EAAG;AAC3B,IAAA,IAAA,CAAK9E,KAAAA,CAAM6D,YAAAA,KAAiBgB,CAAAA,CAAEC,GAAAA,EAAG;AACnC;AACF","file":"workflow.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { z, ZodSchema } from \"zod\";\nimport { Serializable } from \"@/internals/serializable.js\";\nimport { Callback, Emitter } from \"@/emitter/emitter.js\";\nimport { RunContext } from \"@/context.js\";\nimport { omit, pick, toCamelCase } from \"remeda\";\nimport { deepCopy, shallowCopy } from \"@/serializer/utils.js\";\nimport { FrameworkError, ValueError } from \"@/errors.js\";\n\n// eslint-disable-next-line @typescript-eslint/no-invalid-void-type\ntype WorkflowStepHandlerReturn<K extends string> = K | ReservedStep | void;\n\nexport interface WorkflowRun<T extends ZodSchema, T2 extends ZodSchema, K extends string> {\n  result: z.output<T2>;\n  steps: WorkflowStepRes<T, K>[];\n  state: z.output<T>;\n}\n\nexport interface WorkflowRunOptions<K extends string> {\n  start?: K;\n  signal?: AbortSignal;\n}\n\nexport interface WorkflowStepDef<T extends ZodSchema, K extends string> {\n  schema: T;\n  handler: WorkflowStepHandler<T, K>;\n}\n\nexport interface WorkflowStepRes<T extends ZodSchema, K extends string> {\n  name: K;\n  next: K | ReservedStep;\n  state: z.output<T>;\n}\n\nexport interface WorkflowRunContext<T extends ZodSchema, K extends string> {\n  steps: WorkflowStepRes<T, K>[];\n  signal: AbortSignal;\n  abort: (reason?: Error) => void;\n}\n\nexport type WorkflowStepHandler<T extends ZodSchema, K extends string> = (\n  state: z.output<T>,\n  context: WorkflowRunContext<T, K>,\n) => Promise<WorkflowStepHandlerReturn<K>> | WorkflowStepHandlerReturn<K>;\n\nexport interface WorkflowEvents<T extends ZodSchema, T2 extends ZodSchema, K extends string> {\n  start: Callback<{ step: K; run: WorkflowRun<T, T2, K> }>;\n  error: Callback<{\n    step: K;\n    error: Error;\n    run: WorkflowRun<T, T2, K>;\n  }>;\n  success: Callback<{\n    step: K;\n    next: K | ReservedStep;\n    state: z.output<T>;\n    run: WorkflowRun<T, T2, K>;\n  }>;\n}\n\ninterface FlowInput<TS extends ZodSchema, TS2 extends ZodSchema = TS> {\n  name?: string;\n  schema: TS;\n  outputSchema?: TS2;\n}\n\ntype RunStep<K extends string> = K | typeof Workflow.END;\ntype ReservedStep =\n  | typeof Workflow.START\n  | typeof Workflow.NEXT\n  | typeof Workflow.PREV\n  | typeof Workflow.END\n  | typeof Workflow.SELF;\nexport type FlowStepName<K extends string> = K | ReservedStep;\n\nexport class WorkflowError<\n  T extends ZodSchema,\n  T2 extends ZodSchema,\n  K extends string,\n> extends FrameworkError {\n  constructor(message: string, extra?: { run?: WorkflowRun<T, T2, K>; errors?: Error[] }) {\n    super(message, extra?.errors, {\n      context: extra?.run ?? {},\n      isRetryable: false,\n      isFatal: true,\n    });\n  }\n}\n\nexport class Workflow<\n  TInput extends ZodSchema,\n  TOutput extends ZodSchema = TInput,\n  TKeys extends string = string,\n> extends Serializable {\n  public static readonly START = \"__start__\";\n  public static readonly SELF = \"__self__\";\n  public static readonly PREV = \"__prev__\";\n  public static readonly NEXT = \"__next__\";\n  public static readonly END = \"__end__\";\n\n  public readonly emitter: Emitter<WorkflowEvents<TInput, TOutput, TKeys>>;\n\n  protected readonly steps = new Map<string, WorkflowStepDef<TInput, TKeys>>();\n  protected startStep: TKeys | undefined = undefined;\n\n  constructor(protected readonly input: FlowInput<TInput, TOutput>) {\n    super();\n    this.emitter = Emitter.root.child({\n      namespace: [\"workflow\", toCamelCase(input?.name ?? \"\")].filter(Boolean),\n      creator: this,\n    });\n  }\n\n  getSteps() {\n    return Array.from(this.steps.keys()) as TKeys[];\n  }\n\n  get name() {\n    return this.input.name ?? \"\";\n  }\n\n  get schemas() {\n    return pick(this.input, [\"schema\", \"outputSchema\"]);\n  }\n\n  addStep<L extends string>(\n    name: L,\n    handler: WorkflowStepHandler<TInput | TOutput, TKeys>,\n  ): Workflow<TInput, TOutput, L | TKeys>;\n  addStep<L extends string, TFlow extends AnyWorkflowFlow>(\n    name: L,\n    flow: Workflow.pipeable<this, TFlow>,\n  ): Workflow<TInput, TOutput, L | TKeys>;\n  addStep<L extends string>(\n    name: L,\n    step: WorkflowStepHandler<TInput | TOutput, TKeys> | AnyWorkflowFlow,\n  ): Workflow<TInput, TOutput, L | TKeys> {\n    return this._addStep(name, step);\n  }\n\n  addStrictStep<L extends string, TI2 extends ZodSchema>(\n    name: L,\n    schema: TI2,\n    handler: WorkflowStepHandler<TI2, TKeys>,\n  ): Workflow<TInput, TOutput, L | TKeys>;\n  addStrictStep<L extends string, TI2 extends ZodSchema, TFlow extends AnyWorkflowFlow>(\n    name: L,\n    schema: TI2,\n    flow: Workflow.pipeable<Workflow<TI2, TOutput, TKeys>, TFlow>,\n  ): Workflow<TInput, TOutput, L | TKeys>;\n  addStrictStep<L extends string, TI2 extends ZodSchema>(\n    name: L,\n    schema: TI2,\n    step: WorkflowStepHandler<TI2, TKeys> | AnyWorkflowFlow,\n  ): Workflow<TInput, TOutput, L | TKeys> {\n    return this._addStep(name, schema, step);\n  }\n\n  protected _addStep<TI2 extends ZodSchema = TInput, L extends string = TKeys>(\n    name: L,\n    schemaOrStep: TI2 | WorkflowStepHandler<TInput, TKeys> | AnyWorkflowFlow,\n    stepOrEmpty?: WorkflowStepHandler<TI2, TKeys> | AnyWorkflowFlow,\n  ): Workflow<TInput, TOutput, L | TKeys> {\n    if (!name.trim()) {\n      throw new ValueError(`Step name cannot be empty!`);\n    }\n    if (this.steps.has(name)) {\n      throw new ValueError(`Step '${name}' already exists!`);\n    }\n    if (name === Workflow.END) {\n      throw new ValueError(`The name '${name}' cannot be used!`);\n    }\n\n    const schema = (schemaOrStep && stepOrEmpty ? schemaOrStep : this.input.schema) as TInput;\n    const stepOrFlow = stepOrEmpty || schemaOrStep;\n\n    this.steps.set(name, {\n      schema,\n      handler: stepOrFlow instanceof Workflow ? stepOrFlow.asStep({}) : stepOrFlow,\n    } as WorkflowStepDef<TInput, TKeys>);\n\n    return this as unknown as Workflow<TInput, TOutput, L | TKeys>;\n  }\n\n  setStart(name: TKeys) {\n    this.startStep = name;\n    return this;\n  }\n\n  run(state: z.input<TInput>, options: WorkflowRunOptions<TKeys> = {}) {\n    return RunContext.enter(\n      this,\n      { signal: options?.signal, params: [state, options] as const },\n      async (runContext): Promise<WorkflowRun<TInput, TOutput, TKeys>> => {\n        const run: WorkflowRun<TInput, TOutput, TKeys> = {\n          steps: [],\n          state: this.input.schema.parse(state),\n          result: undefined as z.output<TOutput>,\n        };\n        const handlers: WorkflowRunContext<TInput, TKeys> = {\n          steps: run.steps,\n          signal: runContext.signal,\n          abort: (reason) => runContext.abort(reason),\n        };\n\n        let next: RunStep<TKeys> =\n          this.findStep(options?.start || this.startStep || this.getSteps()[0]).current ??\n          Workflow.END;\n\n        while (next && next !== Workflow.END) {\n          const name = next;\n          const step = this.steps.get(next);\n          if (!step) {\n            throw new WorkflowError(`Step '${next}' was not found.`, { run });\n          }\n          await runContext.emitter.emit(\"start\", { run, step: next });\n          try {\n            const state = await step.schema\n              .parseAsync(run.state)\n              .catch(async (err: Error) => {\n                throw new WorkflowError(\n                  `Step '${next}' cannot be executed because the provided input doesn't adhere to the step's schema.`,\n                  { run: shallowCopy(run), errors: [err] },\n                );\n              })\n              .then(deepCopy);\n\n            const nextStepRaw = await step.handler(state, handlers);\n            if (nextStepRaw === Workflow.START) {\n              next = run.steps.at(0)?.name!;\n            } else if (nextStepRaw === Workflow.PREV) {\n              next = this.findStep(next).prev;\n            } else if (nextStepRaw === Workflow.SELF) {\n              next = this.findStep(next).current;\n            } else if (!nextStepRaw || nextStepRaw === Workflow.NEXT) {\n              next = this.findStep(next).next || Workflow.END;\n            } else {\n              next = nextStepRaw;\n            }\n\n            const stepResult: WorkflowStepRes<TInput, TKeys> = {\n              name,\n              state,\n              next,\n            };\n            run.steps.push(stepResult);\n            run.state = stepResult.state;\n\n            await runContext.emitter.emit(\"success\", {\n              run: shallowCopy(run),\n              state: stepResult.state,\n              step: stepResult.name,\n              next: stepResult.next,\n            });\n          } catch (error) {\n            await runContext.emitter.emit(\"error\", {\n              run: shallowCopy(run),\n              step: next as TKeys,\n              error,\n            });\n            throw error;\n          }\n        }\n\n        run.result = await (this.input.outputSchema ?? this.input.schema)\n          .parseAsync(run.state)\n          .catch((err) => {\n            throw new WorkflowError(\n              `Workflow has ended but it's state does not adhere to the workflow's output schema.`,\n              { run: shallowCopy(run), errors: [err] },\n            );\n          });\n        return run;\n      },\n    );\n  }\n\n  delStep<L extends TKeys>(name: L): Workflow<TInput, TOutput, Exclude<TKeys, L>> {\n    if (this.startStep === name) {\n      this.startStep = undefined;\n    }\n    this.steps.delete(name);\n    return this as unknown as Workflow<TInput, TOutput, Exclude<TKeys, L>>;\n  }\n\n  asStep<\n    TInput2 extends ZodSchema = TInput,\n    TOutput2 extends ZodSchema = TOutput,\n    TKeys2 extends string = TKeys,\n  >(overrides: {\n    input?: (input: z.output<TInput2>) => z.output<TInput> | z.input<TInput>;\n    output?: (output: z.output<TOutput>) => z.output<TOutput2> | z.input<TOutput2>;\n    start?: TKeys;\n    next?: FlowStepName<TKeys2>;\n  }): WorkflowStepHandler<TInput2, TKeys | TKeys2> {\n    return async (state, ctx) => {\n      const mappedInput = overrides?.input ? overrides.input(state) : state;\n      const result = await this.run(mappedInput, { start: overrides?.start, signal: ctx.signal });\n      const mappedOutput = overrides?.output ? overrides.output(result.state) : result.state;\n      Object.assign(state, mappedOutput);\n      return overrides.next;\n    };\n  }\n\n  protected findStep(current: TKeys) {\n    const steps = this.getSteps();\n    const index = steps.indexOf(current);\n    return {\n      prev: steps[index - 1],\n      current: steps[index],\n      next: steps[index + 1],\n    };\n  }\n\n  createSnapshot() {\n    return {\n      input: omit(this.input, [\"schema\", \"outputSchema\"]),\n      emitter: this.emitter,\n      startStep: this.startStep,\n      steps: this.steps,\n    };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n    this.input.schema ??= z.any() as unknown as TInput;\n    this.input.outputSchema ??= z.any() as unknown as TOutput;\n  }\n}\n\n// eslint-disable-next-line @typescript-eslint/no-namespace\nexport namespace Workflow {\n  export type pipeable<T extends AnyWorkflowFlow, T2 extends AnyWorkflowFlow> =\n    Workflow.state<T> extends Workflow.input<T2> ? T2 : never;\n  export type input<T> = T extends Workflow<infer A, any, any> ? z.input<A> : never;\n  export type state<T> = T extends Workflow<infer A, any, any> ? z.output<A> : never;\n  export type output<T> = T extends Workflow<any, infer A, any> ? z.output<A> : never;\n  export type run<T> = T extends Workflow<infer A, infer B, infer C> ? WorkflowRun<A, B, C> : never;\n}\n\nexport type AnyWorkflowFlow = Workflow<any, any, any>;\n"]}
+{"version":3,"sources":["../../src/workflows/workflow.ts"],"names":["WorkflowError","FrameworkError","message","extra","errors","context","run","isRetryable","isFatal","Workflow","Serializable","START","SELF","PREV","NEXT","END","emitter","steps","Map","startStep","undefined","input","Emitter","root","child","namespace","toCamelCase","name","filter","Boolean","creator","getSteps","Array","from","keys","schemas","pick","addStep","step","_addStep","addStrictStep","schema","schemaOrStep","stepOrEmpty","trim","ValueError","has","stepOrFlow","set","handler","asStep","setStart","state","options","RunContext","enter","signal","params","runContext","parse","result","handlers","abort","reason","next","findStep","start","current","get","emit","parseAsync","catch","err","shallowCopy","then","deepCopy","nextStepRaw","at","prev","stepResult","push","error","outputSchema","delStep","delete","overrides","ctx","mappedInput","mappedOutput","output","Object","assign","index","indexOf","createSnapshot","omit","loadSnapshot","snapshot","z","any"],"mappings":";;;;;;;;;;AA+EO,MAAMA,sBAIHC,cAAAA,CAAAA;EAnFV;;;AAoFE,EAAA,WAAA,CAAYC,SAAiBC,KAAAA,EAA2D;AACtF,IAAA,KAAA,CAAMD,OAAAA,EAASC,OAAOC,MAAAA,EAAQ;MAC5BC,OAAAA,EAASF,KAAAA,EAAOG,OAAO,EAAC;MACxBC,WAAAA,EAAa,KAAA;MACbC,OAAAA,EAAS;KACX,CAAA;AACF,EAAA;AACF;AAEO,MAAMC,iBAIHC,YAAAA,CAAAA;EAjGV;;;;AAkGE,EAAA,OAAuBC,KAAAA,GAAQ,WAAA;AAC/B,EAAA,OAAuBC,IAAAA,GAAO,UAAA;AAC9B,EAAA,OAAuBC,IAAAA,GAAO,UAAA;AAC9B,EAAA,OAAuBC,IAAAA,GAAO,UAAA;AAC9B,EAAA,OAAuBC,GAAAA,GAAM,SAAA;AAEbC,EAAAA,OAAAA;AAEGC,EAAAA,KAAAA,uBAAYC,GAAAA,EAAAA;EACrBC,SAAAA,GAA+BC,MAAAA;AAEzC,EAAA,WAAA,CAA+BC,KAAAA,EAAmC;AAChE,IAAA,KAAA,EAAK,EAAA,KADwBA,KAAAA,GAAAA,KAAAA;AAE7B,IAAA,IAAA,CAAKL,OAAAA,GAAUM,OAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAAM;MAChCC,SAAAA,EAAW;AAAC,QAAA,UAAA;QAAYC,WAAAA,CAAYL,KAAAA,EAAOM,QAAQ,EAAA;AAAKC,OAAAA,CAAAA,MAAAA,CAAOC,OAAAA,CAAAA;MAC/DC,OAAAA,EAAS;KACX,CAAA;AACF,EAAA;EAEAC,QAAAA,GAAW;AACT,IAAA,OAAOC,KAAAA,CAAMC,IAAAA,CAAK,IAAA,CAAKhB,KAAAA,CAAMiB,MAAI,CAAA;AACnC,EAAA;AAEA,EAAA,IAAIP,IAAAA,GAAO;AACT,IAAA,OAAO,IAAA,CAAKN,MAAMM,IAAAA,IAAQ,EAAA;AAC5B,EAAA;AAEA,EAAA,IAAIQ,OAAAA,GAAU;AACZ,IAAA,OAAOC,IAAAA,CAAK,KAAKf,KAAAA,EAAO;AAAC,MAAA,QAAA;AAAU,MAAA;AAAe,KAAA,CAAA;AACpD,EAAA;AAUAgB,EAAAA,OAAAA,CACEV,MACAW,IAAAA,EACsC;AACtC,IAAA,OAAO,IAAA,CAAKC,QAAAA,CAASZ,IAAAA,EAAMW,IAAAA,CAAAA;AAC7B,EAAA;EAYAE,aAAAA,CACEb,IAAAA,EACAc,QACAH,IAAAA,EACsC;AACtC,IAAA,OAAO,IAAA,CAAKC,QAAAA,CAASZ,IAAAA,EAAMc,MAAAA,EAAQH,IAAAA,CAAAA;AACrC,EAAA;EAEUC,QAAAA,CACRZ,IAAAA,EACAe,cACAC,WAAAA,EACsC;AACtC,IAAA,IAAI,CAAChB,IAAAA,CAAKiB,IAAAA,EAAI,EAAI;AAChB,MAAA,MAAM,IAAIC,WAAW,CAAA,0BAAA,CAA4B,CAAA;AACnD,IAAA;AACA,IAAA,IAAI,IAAA,CAAK5B,KAAAA,CAAM6B,GAAAA,CAAInB,IAAAA,CAAAA,EAAO;AACxB,MAAA,MAAM,IAAIkB,UAAAA,CAAW,CAAA,MAAA,EAASlB,IAAAA,CAAAA,iBAAAA,CAAuB,CAAA;AACvD,IAAA;AACA,IAAA,IAAIA,IAAAA,KAASlB,SAASM,GAAAA,EAAK;AACzB,MAAA,MAAM,IAAI8B,UAAAA,CAAW,CAAA,UAAA,EAAalB,IAAAA,CAAAA,iBAAAA,CAAuB,CAAA;AAC3D,IAAA;AAEA,IAAA,MAAMc,MAAAA,GAAUC,YAAAA,IAAgBC,WAAAA,GAAcD,YAAAA,GAAe,KAAKrB,KAAAA,CAAMoB,MAAAA;AACxE,IAAA,MAAMM,aAAaJ,WAAAA,IAAeD,YAAAA;AAElC,IAAA,IAAA,CAAKzB,KAAAA,CAAM+B,IAAIrB,IAAAA,EAAM;AACnBc,MAAAA,MAAAA;AACAQ,MAAAA,OAAAA,EAASF,sBAAsBtC,QAAAA,GAAWsC,UAAAA,CAAWG,MAAAA,CAAO,EAAC,CAAA,GAAKH;KACpE,CAAA;AAEA,IAAA,OAAO,IAAA;AACT,EAAA;AAEAI,EAAAA,QAAAA,CAASxB,IAAAA,EAAa;AACpB,IAAA,IAAA,CAAKR,SAAAA,GAAYQ,IAAAA;AACjB,IAAA,OAAO,IAAA;AACT,EAAA;EAEArB,GAAAA,CAAI8C,KAAAA,EAAwBC,OAAAA,GAAqC,EAAC,EAAG;AACnE,IAAA,OAAOC,UAAAA,CAAWC,MAChB,IAAA,EACA;AAAEC,MAAAA,MAAAA,EAAQH,OAAAA,EAASG,MAAAA;MAAQC,MAAAA,EAAQ;AAACL,QAAAA,KAAAA;AAAOC,QAAAA;;AAAkB,KAAA,EAC7D,OAAOK,UAAAA,KAAAA;AACL,MAAA,MAAMpD,GAAAA,GAA2C;AAC/CW,QAAAA,KAAAA,EAAO,EAAA;AACPmC,QAAAA,KAAAA,EAAO,IAAA,CAAK/B,KAAAA,CAAMoB,MAAAA,CAAOkB,KAAAA,CAAMP,KAAAA,CAAAA;QAC/BQ,MAAAA,EAAQxC;AACV,OAAA;AACA,MAAA,MAAMyC,QAAAA,GAA8C;AAClD5C,QAAAA,KAAAA,EAAOX,GAAAA,CAAIW,KAAAA;AACXuC,QAAAA,MAAAA,EAAQE,UAAAA,CAAWF,MAAAA;AACnBM,QAAAA,KAAAA,kBAAO,MAAA,CAAA,CAACC,MAAAA,KAAWL,UAAAA,CAAWI,KAAAA,CAAMC,MAAAA,CAAAA,EAA7B,OAAA;AACT,OAAA;AAEA,MAAA,IAAIC,IAAAA,GACF,IAAA,CAAKC,QAAAA,CAASZ,OAAAA,EAASa,SAAS,IAAA,CAAK/C,SAAAA,IAAa,IAAA,CAAKY,QAAAA,EAAQ,CAAG,CAAA,CAAE,CAAA,CAAEoC,WACtE1D,QAAAA,CAASM,GAAAA;AAEX,MAAA,OAAOiD,IAAAA,IAAQA,IAAAA,KAASvD,QAAAA,CAASM,GAAAA,EAAK;AACpC,QAAA,MAAMY,IAAAA,GAAOqC,IAAAA;AACb,QAAA,MAAM1B,IAAAA,GAAO,IAAA,CAAKrB,KAAAA,CAAMmD,GAAAA,CAAIJ,IAAAA,CAAAA;AAC5B,QAAA,IAAI,CAAC1B,IAAAA,EAAM;AACT,UAAA,MAAM,IAAItC,aAAAA,CAAc,CAAA,MAAA,EAASgE,IAAAA,CAAAA,gBAAAA,CAAAA,EAAwB;AAAE1D,YAAAA;WAAI,CAAA;AACjE,QAAA;AACA,QAAA,MAAMoD,UAAAA,CAAW1C,OAAAA,CAAQqD,IAAAA,CAAK,OAAA,EAAS;AAAE/D,UAAAA,GAAAA;UAAKgC,IAAAA,EAAM0B;SAAK,CAAA;AACzD,QAAA,IAAI;AACF,UAAA,MAAMZ,MAAAA,GAAQ,MAAMd,IAAAA,CAAKG,MAAAA,CACtB6B,UAAAA,CAAWhE,IAAI8C,KAAK,CAAA,CACpBmB,KAAAA,CAAM,OAAOC,GAAAA,KAAAA;AACZ,YAAA,MAAM,IAAIxE,aAAAA,CACR,CAAA,MAAA,EAASgE,IAAAA,CAAAA,oFAAAA,CAAAA,EACT;AAAE1D,cAAAA,GAAAA,EAAKmE,YAAYnE,GAAAA,CAAAA;cAAMF,MAAAA,EAAQ;AAACoE,gBAAAA;;aAAK,CAAA;UAE3C,CAAA,CAAA,CACCE,KAAKC,QAAAA,CAAAA;AAER,UAAA,MAAMC,WAAAA,GAAc,MAAMtC,IAAAA,CAAKW,OAAAA,CAAQG,QAAOS,QAAAA,CAAAA;AAC9C,UAAA,IAAIe,WAAAA,KAAgBnE,SAASE,KAAAA,EAAO;AAClCqD,YAAAA,IAAAA,GAAO1D,GAAAA,CAAIW,KAAAA,CAAM4D,EAAAA,CAAG,CAAA,CAAA,EAAIlD,IAAAA;UAC1B,CAAA,MAAA,IAAWiD,WAAAA,KAAgBnE,SAASI,IAAAA,EAAM;AACxCmD,YAAAA,IAAAA,GAAO,IAAA,CAAKC,QAAAA,CAASD,IAAAA,CAAAA,CAAMc,IAAAA;UAC7B,CAAA,MAAA,IAAWF,WAAAA,KAAgBnE,SAASG,IAAAA,EAAM;AACxCoD,YAAAA,IAAAA,GAAO,IAAA,CAAKC,QAAAA,CAASD,IAAAA,CAAAA,CAAMG,OAAAA;AAC7B,UAAA,CAAA,MAAA,IAAW,CAACS,WAAAA,IAAeA,WAAAA,KAAgBnE,QAAAA,CAASK,IAAAA,EAAM;AACxDkD,YAAAA,IAAAA,GAAO,IAAA,CAAKC,QAAAA,CAASD,IAAAA,CAAAA,CAAMA,QAAQvD,QAAAA,CAASM,GAAAA;UAC9C,CAAA,MAAO;AACLiD,YAAAA,IAAAA,GAAOY,WAAAA;AACT,UAAA;AAEA,UAAA,MAAMG,UAAAA,GAA6C;AACjDpD,YAAAA,IAAAA;YACAyB,KAAAA,EAAAA,MAAAA;AACAY,YAAAA;AACF,WAAA;AACA1D,UAAAA,GAAAA,CAAIW,KAAAA,CAAM+D,KAAKD,UAAAA,CAAAA;AACfzE,UAAAA,GAAAA,CAAI8C,QAAQ2B,UAAAA,CAAW3B,KAAAA;AAEvB,UAAA,MAAMM,UAAAA,CAAW1C,OAAAA,CAAQqD,IAAAA,CAAK,SAAA,EAAW;AACvC/D,YAAAA,GAAAA,EAAKmE,YAAYnE,GAAAA,CAAAA;AACjB8C,YAAAA,KAAAA,EAAO2B,UAAAA,CAAW3B,KAAAA;AAClBd,YAAAA,IAAAA,EAAMyC,UAAAA,CAAWpD,IAAAA;AACjBqC,YAAAA,IAAAA,EAAMe,UAAAA,CAAWf;WACnB,CAAA;AACF,QAAA,CAAA,CAAA,OAASiB,KAAAA,EAAO;AACd,UAAA,MAAMvB,UAAAA,CAAW1C,OAAAA,CAAQqD,IAAAA,CAAK,OAAA,EAAS;AACrC/D,YAAAA,GAAAA,EAAKmE,YAAYnE,GAAAA,CAAAA;YACjBgC,IAAAA,EAAM0B,IAAAA;AACNiB,YAAAA;WACF,CAAA;AACA,UAAA,MAAMA,KAAAA;AACR,QAAA;AACF,MAAA;AAEA3E,MAAAA,GAAAA,CAAIsD,MAAAA,GAAS,MAAA,CAAO,IAAA,CAAKvC,KAAAA,CAAM6D,gBAAgB,IAAA,CAAK7D,KAAAA,CAAMoB,MAAAA,EACvD6B,UAAAA,CAAWhE,GAAAA,CAAI8C,KAAK,CAAA,CACpBmB,KAAAA,CAAM,CAACC,GAAAA,KAAAA;AACN,QAAA,MAAM,IAAIxE,cACR,CAAA,kFAAA,CAAA,EACA;AAAEM,UAAAA,GAAAA,EAAKmE,YAAYnE,GAAAA,CAAAA;UAAMF,MAAAA,EAAQ;AAACoE,YAAAA;;SAAK,CAAA;MAE3C,CAAA,CAAA;AACF,MAAA,OAAOlE,GAAAA;IACT,CAAA,CAAA;AAEJ,EAAA;AAEA6E,EAAAA,OAAAA,CAAyBxD,IAAAA,EAAuD;AAC9E,IAAA,IAAI,IAAA,CAAKR,cAAcQ,IAAAA,EAAM;AAC3B,MAAA,IAAA,CAAKR,SAAAA,GAAYC,MAAAA;AACnB,IAAA;AACA,IAAA,IAAA,CAAKH,KAAAA,CAAMmE,OAAOzD,IAAAA,CAAAA;AAClB,IAAA,OAAO,IAAA;AACT,EAAA;AAEAuB,EAAAA,MAAAA,CAIEmC,SAAAA,EAK+C;AAC/C,IAAA,OAAO,OAAOjC,OAAOkC,GAAAA,KAAAA;AACnB,MAAA,MAAMC,cAAcF,SAAAA,EAAWhE,KAAAA,GAAQgE,SAAAA,CAAUhE,KAAAA,CAAM+B,KAAAA,CAAAA,GAASA,KAAAA;AAChE,MAAA,MAAMQ,MAAAA,GAAS,MAAM,IAAA,CAAKtD,GAAAA,CAAIiF,WAAAA,EAAa;AAAErB,QAAAA,KAAAA,EAAOmB,SAAAA,EAAWnB,KAAAA;AAAOV,QAAAA,MAAAA,EAAQ8B,GAAAA,CAAI9B;OAAO,CAAA;AACzF,MAAA,MAAMgC,YAAAA,GAAeH,WAAWI,MAAAA,GAASJ,SAAAA,CAAUI,OAAO7B,MAAAA,CAAOR,KAAK,IAAIQ,MAAAA,CAAOR,KAAAA;AACjFsC,MAAAA,MAAAA,CAAOC,MAAAA,CAAOvC,OAAOoC,YAAAA,CAAAA;AACrB,MAAA,OAAOH,SAAAA,CAAUrB,IAAAA;AACnB,IAAA,CAAA;AACF,EAAA;AAEUC,EAAAA,QAAAA,CAASE,OAAAA,EAAgB;AACjC,IAAA,MAAMlD,KAAAA,GAAQ,KAAKc,QAAAA,EAAQ;AAC3B,IAAA,MAAM6D,KAAAA,GAAQ3E,KAAAA,CAAM4E,OAAAA,CAAQ1B,OAAAA,CAAAA;AAC5B,IAAA,OAAO;MACLW,IAAAA,EAAM7D,KAAAA,CAAM2E,QAAQ,CAAA,CAAA;AACpBzB,MAAAA,OAAAA,EAASlD,MAAM2E,KAAAA,CAAAA;MACf5B,IAAAA,EAAM/C,KAAAA,CAAM2E,QAAQ,CAAA;AACtB,KAAA;AACF,EAAA;EAEAE,cAAAA,GAAiB;AACf,IAAA,OAAO;MACLzE,KAAAA,EAAO0E,IAAAA,CAAK,KAAK1E,KAAAA,EAAO;AAAC,QAAA,QAAA;AAAU,QAAA;AAAe,OAAA,CAAA;AAClDL,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;AACdG,MAAAA,SAAAA,EAAW,IAAA,CAAKA,SAAAA;AAChBF,MAAAA,KAAAA,EAAO,IAAA,CAAKA;AACd,KAAA;AACF,EAAA;AAEA+E,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7DP,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAMM,QAAAA,CAAAA;AACpB,IAAA,IAAA,CAAK5E,KAAAA,CAAMoB,MAAAA,KAAWyD,CAAAA,CAAEC,GAAAA,EAAG;AAC3B,IAAA,IAAA,CAAK9E,KAAAA,CAAM6D,YAAAA,KAAiBgB,CAAAA,CAAEC,GAAAA,EAAG;AACnC,EAAA;AACF","file":"workflow.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { z, ZodSchema } from \"zod\";\nimport { Serializable } from \"@/internals/serializable.js\";\nimport { Callback, Emitter } from \"@/emitter/emitter.js\";\nimport { RunContext } from \"@/context.js\";\nimport { omit, pick, toCamelCase } from \"remeda\";\nimport { deepCopy, shallowCopy } from \"@/serializer/utils.js\";\nimport { FrameworkError, ValueError } from \"@/errors.js\";\n\n// eslint-disable-next-line @typescript-eslint/no-invalid-void-type\ntype WorkflowStepHandlerReturn<K extends string> = K | ReservedStep | void;\n\nexport interface WorkflowRun<T extends ZodSchema, T2 extends ZodSchema, K extends string> {\n  result: z.output<T2>;\n  steps: WorkflowStepRes<T, K>[];\n  state: z.output<T>;\n}\n\nexport interface WorkflowRunOptions<K extends string> {\n  start?: K;\n  signal?: AbortSignal;\n}\n\nexport interface WorkflowStepDef<T extends ZodSchema, K extends string> {\n  schema: T;\n  handler: WorkflowStepHandler<T, K>;\n}\n\nexport interface WorkflowStepRes<T extends ZodSchema, K extends string> {\n  name: K;\n  next: K | ReservedStep;\n  state: z.output<T>;\n}\n\nexport interface WorkflowRunContext<T extends ZodSchema, K extends string> {\n  steps: WorkflowStepRes<T, K>[];\n  signal: AbortSignal;\n  abort: (reason?: Error) => void;\n}\n\nexport type WorkflowStepHandler<T extends ZodSchema, K extends string> = (\n  state: z.output<T>,\n  context: WorkflowRunContext<T, K>,\n) => Promise<WorkflowStepHandlerReturn<K>> | WorkflowStepHandlerReturn<K>;\n\nexport interface WorkflowEvents<T extends ZodSchema, T2 extends ZodSchema, K extends string> {\n  start: Callback<{ step: K; run: WorkflowRun<T, T2, K> }>;\n  error: Callback<{\n    step: K;\n    error: Error;\n    run: WorkflowRun<T, T2, K>;\n  }>;\n  success: Callback<{\n    step: K;\n    next: K | ReservedStep;\n    state: z.output<T>;\n    run: WorkflowRun<T, T2, K>;\n  }>;\n}\n\ninterface FlowInput<TS extends ZodSchema, TS2 extends ZodSchema = TS> {\n  name?: string;\n  schema: TS;\n  outputSchema?: TS2;\n}\n\ntype RunStep<K extends string> = K | typeof Workflow.END;\ntype ReservedStep =\n  | typeof Workflow.START\n  | typeof Workflow.NEXT\n  | typeof Workflow.PREV\n  | typeof Workflow.END\n  | typeof Workflow.SELF;\nexport type FlowStepName<K extends string> = K | ReservedStep;\n\nexport class WorkflowError<\n  T extends ZodSchema,\n  T2 extends ZodSchema,\n  K extends string,\n> extends FrameworkError {\n  constructor(message: string, extra?: { run?: WorkflowRun<T, T2, K>; errors?: Error[] }) {\n    super(message, extra?.errors, {\n      context: extra?.run ?? {},\n      isRetryable: false,\n      isFatal: true,\n    });\n  }\n}\n\nexport class Workflow<\n  TInput extends ZodSchema,\n  TOutput extends ZodSchema = TInput,\n  TKeys extends string = string,\n> extends Serializable {\n  public static readonly START = \"__start__\";\n  public static readonly SELF = \"__self__\";\n  public static readonly PREV = \"__prev__\";\n  public static readonly NEXT = \"__next__\";\n  public static readonly END = \"__end__\";\n\n  public readonly emitter: Emitter<WorkflowEvents<TInput, TOutput, TKeys>>;\n\n  protected readonly steps = new Map<string, WorkflowStepDef<TInput, TKeys>>();\n  protected startStep: TKeys | undefined = undefined;\n\n  constructor(protected readonly input: FlowInput<TInput, TOutput>) {\n    super();\n    this.emitter = Emitter.root.child({\n      namespace: [\"workflow\", toCamelCase(input?.name ?? \"\")].filter(Boolean),\n      creator: this,\n    });\n  }\n\n  getSteps() {\n    return Array.from(this.steps.keys()) as TKeys[];\n  }\n\n  get name() {\n    return this.input.name ?? \"\";\n  }\n\n  get schemas() {\n    return pick(this.input, [\"schema\", \"outputSchema\"]);\n  }\n\n  addStep<L extends string>(\n    name: L,\n    handler: WorkflowStepHandler<TInput | TOutput, TKeys>,\n  ): Workflow<TInput, TOutput, L | TKeys>;\n  addStep<L extends string, TFlow extends AnyWorkflowFlow>(\n    name: L,\n    flow: Workflow.pipeable<this, TFlow>,\n  ): Workflow<TInput, TOutput, L | TKeys>;\n  addStep<L extends string>(\n    name: L,\n    step: WorkflowStepHandler<TInput | TOutput, TKeys> | AnyWorkflowFlow,\n  ): Workflow<TInput, TOutput, L | TKeys> {\n    return this._addStep(name, step);\n  }\n\n  addStrictStep<L extends string, TI2 extends ZodSchema>(\n    name: L,\n    schema: TI2,\n    handler: WorkflowStepHandler<TI2, TKeys>,\n  ): Workflow<TInput, TOutput, L | TKeys>;\n  addStrictStep<L extends string, TI2 extends ZodSchema, TFlow extends AnyWorkflowFlow>(\n    name: L,\n    schema: TI2,\n    flow: Workflow.pipeable<Workflow<TI2, TOutput, TKeys>, TFlow>,\n  ): Workflow<TInput, TOutput, L | TKeys>;\n  addStrictStep<L extends string, TI2 extends ZodSchema>(\n    name: L,\n    schema: TI2,\n    step: WorkflowStepHandler<TI2, TKeys> | AnyWorkflowFlow,\n  ): Workflow<TInput, TOutput, L | TKeys> {\n    return this._addStep(name, schema, step);\n  }\n\n  protected _addStep<TI2 extends ZodSchema = TInput, L extends string = TKeys>(\n    name: L,\n    schemaOrStep: TI2 | WorkflowStepHandler<TInput, TKeys> | AnyWorkflowFlow,\n    stepOrEmpty?: WorkflowStepHandler<TI2, TKeys> | AnyWorkflowFlow,\n  ): Workflow<TInput, TOutput, L | TKeys> {\n    if (!name.trim()) {\n      throw new ValueError(`Step name cannot be empty!`);\n    }\n    if (this.steps.has(name)) {\n      throw new ValueError(`Step '${name}' already exists!`);\n    }\n    if (name === Workflow.END) {\n      throw new ValueError(`The name '${name}' cannot be used!`);\n    }\n\n    const schema = (schemaOrStep && stepOrEmpty ? schemaOrStep : this.input.schema) as TInput;\n    const stepOrFlow = stepOrEmpty || schemaOrStep;\n\n    this.steps.set(name, {\n      schema,\n      handler: stepOrFlow instanceof Workflow ? stepOrFlow.asStep({}) : stepOrFlow,\n    } as WorkflowStepDef<TInput, TKeys>);\n\n    return this as unknown as Workflow<TInput, TOutput, L | TKeys>;\n  }\n\n  setStart(name: TKeys) {\n    this.startStep = name;\n    return this;\n  }\n\n  run(state: z.input<TInput>, options: WorkflowRunOptions<TKeys> = {}) {\n    return RunContext.enter(\n      this,\n      { signal: options?.signal, params: [state, options] as const },\n      async (runContext): Promise<WorkflowRun<TInput, TOutput, TKeys>> => {\n        const run: WorkflowRun<TInput, TOutput, TKeys> = {\n          steps: [],\n          state: this.input.schema.parse(state),\n          result: undefined as z.output<TOutput>,\n        };\n        const handlers: WorkflowRunContext<TInput, TKeys> = {\n          steps: run.steps,\n          signal: runContext.signal,\n          abort: (reason) => runContext.abort(reason),\n        };\n\n        let next: RunStep<TKeys> =\n          this.findStep(options?.start || this.startStep || this.getSteps()[0]).current ??\n          Workflow.END;\n\n        while (next && next !== Workflow.END) {\n          const name = next;\n          const step = this.steps.get(next);\n          if (!step) {\n            throw new WorkflowError(`Step '${next}' was not found.`, { run });\n          }\n          await runContext.emitter.emit(\"start\", { run, step: next });\n          try {\n            const state = await step.schema\n              .parseAsync(run.state)\n              .catch(async (err: Error) => {\n                throw new WorkflowError(\n                  `Step '${next}' cannot be executed because the provided input doesn't adhere to the step's schema.`,\n                  { run: shallowCopy(run), errors: [err] },\n                );\n              })\n              .then(deepCopy);\n\n            const nextStepRaw = await step.handler(state, handlers);\n            if (nextStepRaw === Workflow.START) {\n              next = run.steps.at(0)?.name!;\n            } else if (nextStepRaw === Workflow.PREV) {\n              next = this.findStep(next).prev;\n            } else if (nextStepRaw === Workflow.SELF) {\n              next = this.findStep(next).current;\n            } else if (!nextStepRaw || nextStepRaw === Workflow.NEXT) {\n              next = this.findStep(next).next || Workflow.END;\n            } else {\n              next = nextStepRaw;\n            }\n\n            const stepResult: WorkflowStepRes<TInput, TKeys> = {\n              name,\n              state,\n              next,\n            };\n            run.steps.push(stepResult);\n            run.state = stepResult.state;\n\n            await runContext.emitter.emit(\"success\", {\n              run: shallowCopy(run),\n              state: stepResult.state,\n              step: stepResult.name,\n              next: stepResult.next,\n            });\n          } catch (error) {\n            await runContext.emitter.emit(\"error\", {\n              run: shallowCopy(run),\n              step: next as TKeys,\n              error,\n            });\n            throw error;\n          }\n        }\n\n        run.result = await (this.input.outputSchema ?? this.input.schema)\n          .parseAsync(run.state)\n          .catch((err) => {\n            throw new WorkflowError(\n              `Workflow has ended but it's state does not adhere to the workflow's output schema.`,\n              { run: shallowCopy(run), errors: [err] },\n            );\n          });\n        return run;\n      },\n    );\n  }\n\n  delStep<L extends TKeys>(name: L): Workflow<TInput, TOutput, Exclude<TKeys, L>> {\n    if (this.startStep === name) {\n      this.startStep = undefined;\n    }\n    this.steps.delete(name);\n    return this as unknown as Workflow<TInput, TOutput, Exclude<TKeys, L>>;\n  }\n\n  asStep<\n    TInput2 extends ZodSchema = TInput,\n    TOutput2 extends ZodSchema = TOutput,\n    TKeys2 extends string = TKeys,\n  >(overrides: {\n    input?: (input: z.output<TInput2>) => z.output<TInput> | z.input<TInput>;\n    output?: (output: z.output<TOutput>) => z.output<TOutput2> | z.input<TOutput2>;\n    start?: TKeys;\n    next?: FlowStepName<TKeys2>;\n  }): WorkflowStepHandler<TInput2, TKeys | TKeys2> {\n    return async (state, ctx) => {\n      const mappedInput = overrides?.input ? overrides.input(state) : state;\n      const result = await this.run(mappedInput, { start: overrides?.start, signal: ctx.signal });\n      const mappedOutput = overrides?.output ? overrides.output(result.state) : result.state;\n      Object.assign(state, mappedOutput);\n      return overrides.next;\n    };\n  }\n\n  protected findStep(current: TKeys) {\n    const steps = this.getSteps();\n    const index = steps.indexOf(current);\n    return {\n      prev: steps[index - 1],\n      current: steps[index],\n      next: steps[index + 1],\n    };\n  }\n\n  createSnapshot() {\n    return {\n      input: omit(this.input, [\"schema\", \"outputSchema\"]),\n      emitter: this.emitter,\n      startStep: this.startStep,\n      steps: this.steps,\n    };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n    this.input.schema ??= z.any() as unknown as TInput;\n    this.input.outputSchema ??= z.any() as unknown as TOutput;\n  }\n}\n\n// eslint-disable-next-line @typescript-eslint/no-namespace\nexport namespace Workflow {\n  export type pipeable<T extends AnyWorkflowFlow, T2 extends AnyWorkflowFlow> =\n    Workflow.state<T> extends Workflow.input<T2> ? T2 : never;\n  export type input<T> = T extends Workflow<infer A, any, any> ? z.input<A> : never;\n  export type state<T> = T extends Workflow<infer A, any, any> ? z.output<A> : never;\n  export type output<T> = T extends Workflow<any, infer A, any> ? z.output<A> : never;\n  export type run<T> = T extends Workflow<infer A, infer B, infer C> ? WorkflowRun<A, B, C> : never;\n}\n\nexport type AnyWorkflowFlow = Workflow<any, any, any>;\n"]}