beeai-framework 0.1.16 → 0.1.18

package/dist/workflows/workflow.js.map

@@ -1 +1 @@
-{"version":3,"sources":["workflow.ts"],"names":["WorkflowError","FrameworkError","constructor","message","extra","errors","context","run","isRetryable","isFatal","Workflow","Serializable","START","SELF","PREV","NEXT","END","emitter","steps","startStep","input","Map","undefined","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":";;;;;;;;;;AA0FO,MAAMA,sBAIHC,cAAAA,CAAAA;EA9FV;;;AA+FEC,EAAAA,WAAAA,CAAYC,SAAiBC,KAA2D,EAAA;AACtF,IAAMD,KAAAA,CAAAA,OAAAA,EAASC,OAAOC,MAAQ,EAAA;MAC5BC,OAASF,EAAAA,KAAAA,EAAOG,OAAO,EAAC;MACxBC,WAAa,EAAA,KAAA;MACbC,OAAS,EAAA;KACX,CAAA;AACF;AACF;AAEO,MAAMC,iBAIHC,YAAAA,CAAAA;EA5GV;;;;AA6GE,EAAA,OAAuBC,KAAQ,GAAA,WAAA;AAC/B,EAAA,OAAuBC,IAAO,GAAA,UAAA;AAC9B,EAAA,OAAuBC,IAAO,GAAA,UAAA;AAC9B,EAAA,OAAuBC,IAAO,GAAA,UAAA;AAC9B,EAAA,OAAuBC,GAAM,GAAA,SAAA;AAEbC,EAAAA,OAAAA;AAEGC,EAAAA,KAAAA;AACTC,EAAAA,SAAAA;AAEVjB,EAAAA,WAAAA,CAA+BkB,KAAmC,EAAA;AAChE,IAAK,KAAA,EAAA,EAAA,IADwBA,CAAAA,KAAAA,GAAAA,KAAAA,EAAAA,IAAAA,CAHZF,wBAAYG,IAAAA,GAAAA,EAAAA,EAAAA,IAAAA,CACrBF,SAA+BG,GAAAA,MAAAA;AAIvC,IAAKL,IAAAA,CAAAA,OAAAA,GAAUM,OAAQC,CAAAA,IAAAA,CAAKC,KAAM,CAAA;MAChCC,SAAW,EAAA;AAAC,QAAA,UAAA;QAAYC,WAAYP,CAAAA,KAAAA,EAAOQ,QAAQ,EAAA;AAAKC,OAAAA,CAAAA,MAAAA,CAAOC,OAAAA,CAAAA;MAC/DC,OAAS,EAAA;KACX,CAAA;AACF;EAEAC,QAAW,GAAA;AACT,IAAA,OAAOC,KAAMC,CAAAA,IAAAA,CAAK,IAAKhB,CAAAA,KAAAA,CAAMiB,MAAI,CAAA;AACnC;AAEA,EAAA,IAAIP,IAAO,GAAA;AACT,IAAO,OAAA,IAAA,CAAKR,MAAMQ,IAAQ,IAAA,EAAA;AAC5B;AAEA,EAAA,IAAIQ,OAAU,GAAA;AACZ,IAAOC,OAAAA,IAAAA,CAAK,KAAKjB,KAAO,EAAA;AAAC,MAAA,QAAA;AAAU,MAAA;AAAe,KAAA,CAAA;AACpD;AAUAkB,EAAAA,OAAAA,CACEV,MACAW,IACsC,EAAA;AACtC,IAAO,OAAA,IAAA,CAAKC,QAASZ,CAAAA,IAAAA,EAAMW,IAAAA,CAAAA;AAC7B;EAYAE,aACEb,CAAAA,IAAAA,EACAc,QACAH,IACsC,EAAA;AACtC,IAAA,OAAO,IAAKC,CAAAA,QAAAA,CAASZ,IAAMc,EAAAA,MAAAA,EAAQH,IAAAA,CAAAA;AACrC;EAEUC,QACRZ,CAAAA,IAAAA,EACAe,cACAC,WACsC,EAAA;AACtC,IAAI,IAAA,CAAChB,IAAKiB,CAAAA,IAAAA,EAAQ,EAAA;AAChB,MAAM,MAAA,IAAIC,WAAW,CAA4B,0BAAA,CAAA,CAAA;AACnD;AACA,IAAA,IAAI,IAAK5B,CAAAA,KAAAA,CAAM6B,GAAInB,CAAAA,IAAAA,CAAO,EAAA;AACxB,MAAA,MAAM,IAAIkB,UAAAA,CAAW,CAASlB,MAAAA,EAAAA,IAAAA,CAAuB,iBAAA,CAAA,CAAA;AACvD;AACA,IAAIA,IAAAA,IAAAA,KAASlB,SAASM,GAAK,EAAA;AACzB,MAAA,MAAM,IAAI8B,UAAAA,CAAW,CAAalB,UAAAA,EAAAA,IAAAA,CAAuB,iBAAA,CAAA,CAAA;AAC3D;AAEA,IAAA,MAAMc,MAAUC,GAAAA,YAAAA,IAAgBC,WAAcD,GAAAA,YAAAA,GAAe,KAAKvB,KAAMsB,CAAAA,MAAAA;AACxE,IAAA,MAAMM,aAAaJ,WAAeD,IAAAA,YAAAA;AAElC,IAAKzB,IAAAA,CAAAA,KAAAA,CAAM+B,IAAIrB,IAAM,EAAA;AACnBc,MAAAA,MAAAA;AACAQ,MAAAA,OAAAA,EAASF,sBAAsBtC,QAAWsC,GAAAA,UAAAA,CAAWG,MAAO,CAAA,EAAC,CAAKH,GAAAA;KACpE,CAAA;AAEA,IAAO,OAAA,IAAA;AACT;AAEAI,EAAAA,QAAAA,CAASxB,IAAa,EAAA;AACpB,IAAA,IAAA,CAAKT,SAAYS,GAAAA,IAAAA;AACjB,IAAO,OAAA,IAAA;AACT;EAEArB,GAAI8C,CAAAA,KAAAA,EAAwBC,OAAqC,GAAA,EAAI,EAAA;AACnE,IAAOC,OAAAA,UAAAA,CAAWC,MAChB,IACA,EAAA;AAAEC,MAAAA,MAAAA,EAAQH,OAASG,EAAAA,MAAAA;MAAQC,MAAQ,EAAA;AAACL,QAAAA,KAAAA;AAAOC,QAAAA;;AAAkB,KAAA,EAC7D,OAAOK,UAAAA,KAAAA;AACL,MAAA,MAAMpD,GAA2C,GAAA;AAC/CW,QAAAA,KAAAA,EAAO,EAAA;AACPmC,QAAAA,KAAAA,EAAO,IAAKjC,CAAAA,KAAAA,CAAMsB,MAAOkB,CAAAA,KAAAA,CAAMP,KAAAA,CAAAA;QAC/BQ,MAAQvC,EAAAA;AACV,OAAA;AACA,MAAA,MAAMwC,QAA8C,GAAA;AAClD5C,QAAAA,KAAAA,EAAOX,GAAIW,CAAAA,KAAAA;AACXuC,QAAAA,MAAAA,EAAQE,UAAWF,CAAAA,MAAAA;AACnBM,QAAAA,KAAAA,kBAAQC,MAAAA,CAAAA,CAAAA,MAAAA,KAAWL,UAAWI,CAAAA,KAAAA,CAAMC,MAAAA,CAA7B,EAAA,OAAA;AACT,OAAA;AAEA,MAAA,IAAIC,IACF,GAAA,IAAA,CAAKC,QAASZ,CAAAA,OAAAA,EAASa,SAAS,IAAKhD,CAAAA,SAAAA,IAAa,IAAKa,CAAAA,QAAAA,EAAW,CAAA,CAAA,CAAE,CAAA,CAAEoC,WACtE1D,QAASM,CAAAA,GAAAA;AAEX,MAAOiD,OAAAA,IAAAA,IAAQA,IAASvD,KAAAA,QAAAA,CAASM,GAAK,EAAA;AACpC,QAAA,MAAMY,IAAOqC,GAAAA,IAAAA;AACb,QAAA,MAAM1B,IAAO,GAAA,IAAA,CAAKrB,KAAMmD,CAAAA,GAAAA,CAAIJ,IAAAA,CAAAA;AAC5B,QAAA,IAAI,CAAC1B,IAAM,EAAA;AACT,UAAA,MAAM,IAAIvC,aAAAA,CAAc,CAASiE,MAAAA,EAAAA,IAAAA,CAAwB,gBAAA,CAAA,EAAA;AAAE1D,YAAAA;WAAI,CAAA;AACjE;AACA,QAAMoD,MAAAA,UAAAA,CAAW1C,OAAQqD,CAAAA,IAAAA,CAAK,OAAS,EAAA;AAAE/D,UAAAA,GAAAA;UAAKgC,IAAM0B,EAAAA;SAAK,CAAA;AACzD,QAAI,IAAA;AACF,UAAMZ,MAAAA,MAAAA,GAAQ,MAAMd,IAAAA,CAAKG,MACtB6B,CAAAA,UAAAA,CAAWhE,IAAI8C,KAAK,CAAA,CACpBmB,KAAM,CAAA,OAAOC,GAAAA,KAAAA;AACZ,YAAA,MAAM,IAAIzE,aAAAA,CACR,CAASiE,MAAAA,EAAAA,IAAAA,CACT,oFAAA,CAAA,EAAA;AAAE1D,cAAAA,GAAAA,EAAKmE,YAAYnE,GAAAA,CAAAA;cAAMF,MAAQ,EAAA;AAACoE,gBAAAA;;aAAK,CAAA;WAE3C,CAAA,CACCE,KAAKC,QAAAA,CAAAA;AAER,UAAA,MAAMC,WAAc,GAAA,MAAMtC,IAAKW,CAAAA,OAAAA,CAAQG,QAAOS,QAAAA,CAAAA;AAC9C,UAAIe,IAAAA,WAAAA,KAAgBnE,SAASE,KAAO,EAAA;AAClCqD,YAAAA,IAAAA,GAAO1D,GAAIW,CAAAA,KAAAA,CAAM4D,EAAG,CAAA,CAAA,CAAIlD,EAAAA,IAAAA;WACfiD,MAAAA,IAAAA,WAAAA,KAAgBnE,SAASI,IAAM,EAAA;AACxCmD,YAAO,IAAA,GAAA,IAAA,CAAKC,QAASD,CAAAA,IAAAA,CAAMc,CAAAA,IAAAA;WAClBF,MAAAA,IAAAA,WAAAA,KAAgBnE,SAASG,IAAM,EAAA;AACxCoD,YAAO,IAAA,GAAA,IAAA,CAAKC,QAASD,CAAAA,IAAAA,CAAMG,CAAAA,OAAAA;AAC7B,WAAA,MAAA,IAAW,CAACS,WAAAA,IAAeA,WAAgBnE,KAAAA,QAAAA,CAASK,IAAM,EAAA;AACxDkD,YAAAA,IAAAA,GAAO,IAAKC,CAAAA,QAAAA,CAASD,IAAAA,CAAAA,CAAMA,QAAQvD,QAASM,CAAAA,GAAAA;WACvC,MAAA;AACLiD,YAAOY,IAAAA,GAAAA,WAAAA;AACT;AAEA,UAAA,MAAMG,UAA6C,GAAA;AACjDpD,YAAAA,IAAAA;YACAyB,KAAAA,EAAAA,MAAAA;AACAY,YAAAA;AACF,WAAA;AACA1D,UAAIW,GAAAA,CAAAA,KAAAA,CAAM+D,KAAKD,UAAAA,CAAAA;AACfzE,UAAAA,GAAAA,CAAI8C,QAAQ2B,UAAW3B,CAAAA,KAAAA;AAEvB,UAAMM,MAAAA,UAAAA,CAAW1C,OAAQqD,CAAAA,IAAAA,CAAK,SAAW,EAAA;AACvC/D,YAAAA,GAAAA,EAAKmE,YAAYnE,GAAAA,CAAAA;AACjB8C,YAAAA,KAAAA,EAAO2B,UAAW3B,CAAAA,KAAAA;AAClBd,YAAAA,IAAAA,EAAMyC,UAAWpD,CAAAA,IAAAA;AACjBqC,YAAAA,IAAAA,EAAMe,UAAWf,CAAAA;WACnB,CAAA;AACF,SAAA,CAAA,OAASiB,KAAO,EAAA;AACd,UAAMvB,MAAAA,UAAAA,CAAW1C,OAAQqD,CAAAA,IAAAA,CAAK,OAAS,EAAA;AACrC/D,YAAAA,GAAAA,EAAKmE,YAAYnE,GAAAA,CAAAA;YACjBgC,IAAM0B,EAAAA,IAAAA;AACNiB,YAAAA;WACF,CAAA;AACA,UAAMA,MAAAA,KAAAA;AACR;AACF;AAEA3E,MAAAA,GAAAA,CAAIsD,MAAS,GAAA,MAAA,CAAO,IAAKzC,CAAAA,KAAAA,CAAM+D,gBAAgB,IAAK/D,CAAAA,KAAAA,CAAMsB,MACvD6B,EAAAA,UAAAA,CAAWhE,GAAI8C,CAAAA,KAAK,CACpBmB,CAAAA,KAAAA,CAAM,CAACC,GAAAA,KAAAA;AACN,QAAM,MAAA,IAAIzE,cACR,CACA,kFAAA,CAAA,EAAA;AAAEO,UAAAA,GAAAA,EAAKmE,YAAYnE,GAAAA,CAAAA;UAAMF,MAAQ,EAAA;AAACoE,YAAAA;;SAAK,CAAA;OAE3C,CAAA;AACF,MAAOlE,OAAAA,GAAAA;KACT,CAAA;AAEJ;AAEA6E,EAAAA,OAAAA,CAAyBxD,IAAuD,EAAA;AAC9E,IAAI,IAAA,IAAA,CAAKT,cAAcS,IAAM,EAAA;AAC3B,MAAA,IAAA,CAAKT,SAAYG,GAAAA,MAAAA;AACnB;AACA,IAAKJ,IAAAA,CAAAA,KAAAA,CAAMmE,OAAOzD,IAAAA,CAAAA;AAClB,IAAO,OAAA,IAAA;AACT;AAEAuB,EAAAA,MAAAA,CAIEmC,SAK+C,EAAA;AAC/C,IAAO,OAAA,OAAOjC,OAAOkC,GAAAA,KAAAA;AACnB,MAAA,MAAMC,cAAcF,SAAWlE,EAAAA,KAAAA,GAAQkE,SAAUlE,CAAAA,KAAAA,CAAMiC,KAAAA,CAASA,GAAAA,KAAAA;AAChE,MAAA,MAAMQ,MAAS,GAAA,MAAM,IAAKtD,CAAAA,GAAAA,CAAIiF,WAAa,EAAA;AAAErB,QAAAA,KAAAA,EAAOmB,SAAWnB,EAAAA,KAAAA;AAAOV,QAAAA,MAAAA,EAAQ8B,GAAI9B,CAAAA;OAAO,CAAA;AACzF,MAAMgC,MAAAA,YAAAA,GAAeH,WAAWI,MAASJ,GAAAA,SAAAA,CAAUI,OAAO7B,MAAOR,CAAAA,KAAK,IAAIQ,MAAOR,CAAAA,KAAAA;AACjFsC,MAAOC,MAAAA,CAAAA,MAAAA,CAAOvC,OAAOoC,YAAAA,CAAAA;AACrB,MAAA,OAAOH,SAAUrB,CAAAA,IAAAA;AACnB,KAAA;AACF;AAEUC,EAAAA,QAAAA,CAASE,OAAgB,EAAA;AACjC,IAAMlD,MAAAA,KAAAA,GAAQ,KAAKc,QAAQ,EAAA;AAC3B,IAAM6D,MAAAA,KAAAA,GAAQ3E,KAAM4E,CAAAA,OAAAA,CAAQ1B,OAAAA,CAAAA;AAC5B,IAAO,OAAA;MACLW,IAAM7D,EAAAA,KAAAA,CAAM2E,QAAQ,CAAA,CAAA;AACpBzB,MAAAA,OAAAA,EAASlD,MAAM2E,KAAAA,CAAAA;MACf5B,IAAM/C,EAAAA,KAAAA,CAAM2E,QAAQ,CAAA;AACtB,KAAA;AACF;EAEAE,cAAiB,GAAA;AACf,IAAO,OAAA;MACL3E,KAAO4E,EAAAA,IAAAA,CAAK,KAAK5E,KAAO,EAAA;AAAC,QAAA,QAAA;AAAU,QAAA;AAAe,OAAA,CAAA;AAClDH,MAAAA,OAAAA,EAAS,IAAKA,CAAAA,OAAAA;AACdE,MAAAA,SAAAA,EAAW,IAAKA,CAAAA,SAAAA;AAChBD,MAAAA,KAAAA,EAAO,IAAKA,CAAAA;AACd,KAAA;AACF;AAEA+E,EAAAA,YAAAA,CAAaC,QAAkD,EAAA;AAC7DP,IAAOC,MAAAA,CAAAA,MAAAA,CAAO,MAAMM,QAAAA,CAAAA;AACpB,IAAK9E,IAAAA,CAAAA,KAAAA,CAAMsB,MAAWyD,KAAAA,CAAAA,CAAEC,GAAG,EAAA;AAC3B,IAAKhF,IAAAA,CAAAA,KAAAA,CAAM+D,YAAiBgB,KAAAA,CAAAA,CAAEC,GAAG,EAAA;AACnC;AACF","file":"workflow.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\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;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"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "beeai-framework",
-  "version": "0.1.16",
+  "version": "0.1.18",
   "license": "Apache-2.0",
   "description": "BeeAI Framework - LLM Agent Framework",
   "author": "BeeAI a Series of LF Projects, LLC",
@@ -13,7 +13,7 @@
     "Bee Agent Framework",
     "NodeJS Agent Framework"
   ],
-  "packageManager": "yarn@4.1.1",
+  "packageManager": "yarn@4.9.1",
   "commitlint": {
     "extends": [
       "@commitlint/config-conventional"
@@ -46,11 +46,6 @@
       ]
     }
   },
-  "lint-staged": {
-    "{src,tests,scripts}/**/*": "yarn copyright",
-    "*": "prettier --ignore-unknown --write",
-    "*.{ts,js}": "eslint --fix"
-  },
   "prettier": {
     "printWidth": 100,
     "quoteProps": "consistent"
@@ -84,16 +79,11 @@
       "publish": true
     },
     "hooks": {
-      "before:init": [
-        "yarn lint",
-        "yarn ts:check",
-        "[ \"$RELEASE_SKIP_TESTS\" = \"true\" ] && echo 'Skipping tests.' || yarn test:all"
-      ],
       "after:bump": [
-        "yarn build"
+        "mise typescript:build"
       ],
       "before:release": [
-        "yarn lint:fix && yarn format:fix && git add . --update"
+        "mise typescript:fix && git add . --update"
       ]
     },
     "github": {
@@ -147,42 +137,18 @@
   "bugs": {
     "url": "https://github.com/i-am-bee/beeai-framework/issues"
   },
-  "scripts": {
-    "clean": "rimraf dist",
-    "build": "yarn clean && yarn ts:check && NODE_OPTIONS='--max-old-space-size=8192' tsup",
-    "ts:check": "tsc --noEmit && tsc -p tsconfig.examples.json --noEmit",
-    "start": "tsx --tsconfig tsconfig.examples.json",
-    "start:bee": "yarn start -- examples/agents/bee.ts",
-    "lint": "yarn eslint",
-    "prepare": "../.githooks/install.sh",
-    "lint:fix": "yarn eslint --fix",
-    "format": "yarn prettier --check .",
-    "format:fix": "yarn prettier --write .",
-    "test:unit": "vitest run src",
-    "test:unit:watch": "vitest run src",
-    "test:e2e": "vitest run tests/e2e",
-    "test:e2e:watch": "vitest watch tests/e2e",
-    "test:all": "yarn test:unit && yarn test:e2e && yarn test:examples",
-    "test:examples": "vitest --config ./examples/vitest.examples.config.ts run tests/examples",
-    "test:examples:watch": "vitest --config ./examples/vitest.examples.config.ts watch tests/examples",
-    "copyright": "./scripts/copyright.sh",
-    "copyright:check": "TYPE=check ./scripts/copyright.sh",
-    "release": "release-it",
-    "_ensure_env": "cp -n .env.template .env || true",
-    "git:commit-msg": "yarn commitlint --edit --",
-    "git:pre-commit": "yarn copyright:check && yarn lint && yarn format && yarn ts:check && yarn run test:unit"
-  },
   "dependencies": {
+    "@ai-sdk/xai": "^1.2.18",
     "@ai-zen/node-fetch-event-source": "^2.1.4",
     "@streamparser/json": "^0.0.22",
-    "ai": "^4.3.13",
+    "ai": "^4.3.19",
     "ajv": "^8.17.1",
     "ajv-formats": "^3.0.1",
     "duck-duck-scrape": "^2.2.7",
-    "fast-xml-parser": "^5.0.8",
-    "header-generator": "^2.1.62",
+    "fast-xml-parser": "^5.2.5",
+    "header-generator": "^2.1.69",
     "joplin-turndown-plugin-gfm": "^1.0.12",
-    "jsonrepair": "^3.12.0",
+    "jsonrepair": "^3.13.0",
     "mathjs": "^14.0.0",
     "mustache": "^4.2.0",
     "object-hash": "^3.0.0",
@@ -193,33 +159,37 @@
     "remeda": "^2.21.0",
     "serialize-error-cjs": "^0.2.0",
     "string-comparison": "^1.3.0",
-    "string-strip-html": "^13.4.12",
+    "string-strip-html": "^13.4.13",
     "turndown": "^7.2.0",
     "wikipedia": "^2.1.2",
-    "zod": "^3.24.2",
-    "zod-to-json-schema": "^3.24.3"
+    "zod": "^3.24.4",
+    "zod-to-json-schema": "^3.24.6"
   },
   "peerDependencies": {
-    "@ai-sdk/amazon-bedrock": "^2.2.8",
-    "@ai-sdk/anthropic": "^1.2.10",
-    "@ai-sdk/azure": "^1.3.14",
-    "@ai-sdk/google-vertex": "^2.2.15",
-    "@ai-sdk/groq": "^1.2.8",
-    "@ai-sdk/openai": "^1.3.21",
-    "@aws-sdk/client-bedrock-runtime": "^3.772.0",
+    "@a2a-js/sdk": "^0.3.4",
+    "@ai-sdk/amazon-bedrock": "^2.2.12",
+    "@ai-sdk/anthropic": "^1.2.12",
+    "@ai-sdk/azure": "^1.3.24",
+    "@ai-sdk/google-vertex": "^2.2.27",
+    "@ai-sdk/groq": "^1.2.9",
+    "@ai-sdk/openai": "^1.3.23",
+    "@aws-sdk/client-bedrock-runtime": "^3.848.0",
     "@elastic/elasticsearch": "^8.0.0",
     "@googleapis/customsearch": "^3.2.0",
     "@langchain/community": ">=0.2.28",
     "@langchain/core": ">=0.2.27",
-    "@modelcontextprotocol/sdk": "^1.12.0",
-    "@qdrant/js-client-rest": "^1.14.0",
-    "@zilliz/milvus2-sdk-node": "^2.4.9",
+    "@modelcontextprotocol/sdk": "^1.17.0",
+    "@qdrant/js-client-rest": "^1.15.0",
+    "@zilliz/milvus2-sdk-node": "^2.6.0",
     "express": "^5.0.0",
     "ollama-ai-provider": "^1.2.0",
-    "sequelize": "^6.37.3",
+    "sequelize": "^6.37.7",
     "yaml": "^2.6.1"
   },
   "peerDependenciesMeta": {
+    "@a2a-js/sdk": {
+      "optional": true
+    },
     "@ai-sdk/amazon-bedrock": {
       "optional": true
     },
@@ -276,22 +246,23 @@
     }
   },
   "devDependencies": {
-    "@ai-sdk/amazon-bedrock": "^2.2.8",
-    "@ai-sdk/anthropic": "^1.2.10",
-    "@ai-sdk/azure": "^1.3.14",
-    "@ai-sdk/google-vertex": "^2.2.15",
-    "@ai-sdk/groq": "^1.2.8",
-    "@ai-sdk/openai": "^1.3.13",
+    "@a2a-js/sdk": "^0.3.4",
+    "@ai-sdk/amazon-bedrock": "^2.2.12",
+    "@ai-sdk/anthropic": "^1.2.12",
+    "@ai-sdk/azure": "^1.3.24",
+    "@ai-sdk/google-vertex": "^2.2.27",
+    "@ai-sdk/groq": "^1.2.9",
+    "@ai-sdk/openai": "^1.3.23",
     "@commitlint/cli": "^19.7.1",
     "@commitlint/config-conventional": "^19.7.1",
     "@elastic/elasticsearch": "^8.16.2",
     "@eslint/js": "^9.16.0",
     "@eslint/markdown": "^6.2.1",
     "@googleapis/customsearch": "^3.2.0",
-    "@ibm-cloud/watsonx-ai": "^1.6.0",
-    "@langchain/community": "0.3.34",
-    "@langchain/core": "0.3.42",
-    "@langchain/langgraph": "0.2.55",
+    "@ibm-cloud/watsonx-ai": "^1.6.9",
+    "@langchain/community": "^0.3.49",
+    "@langchain/core": "^0.3.59",
+    "@langchain/langgraph": "^0.3.11",
     "@langchain/ollama": "^0.2.0",
     "@modelcontextprotocol/sdk": "^1.12.0",
     "@qdrant/js-client-rest": "^1.14.0",
@@ -318,23 +289,21 @@
     "eslint-plugin-unused-imports": "^4.1.4",
     "express": "^5.0.0",
     "glob": "^11.0.0",
-    "ibm-cloud-sdk-core": "^5.3.2",
+    "ibm-cloud-sdk-core": "^5.4.2",
     "langchain": "0.3.19",
-    "linkinator": "^6.1.2",
-    "lint-staged": "^15.2.10",
     "ollama-ai-provider": "^1.2.0",
     "picocolors": "^1.1.1",
     "pino-pretty": "^13.0.0",
     "pino-test": "^1.1.0",
-    "prettier": "^3.4.2",
+    "prettier": "^3.6.2",
     "release-it": "^17.10.0",
     "rimraf": "^6.0.1",
     "sequelize": "^6.37.5",
     "sqlite3": "^5.1.7",
     "strip-ansi": "^7.1.0",
     "tsup": "^8.4.0",
-    "tsx": "^4.19.2",
-    "typescript": "^5.8.2",
+    "tsx": "^4.20.3",
+    "typescript": "^5.8.3",
     "typescript-eslint": "^8.26.1",
     "vite-tsconfig-paths": "^5.1.4",
     "vitest": "^2.1.9",