beeai-framework 0.1.21 → 0.1.23

package/dist/{chat-C0s-o6ll.d.ts → chat-DrqwUgD8.d.ts}

@@ -1,8 +1,7 @@
 import { Message, AssistantMessage, MessageContentPart } from './backend/message.js';
-import * as ai from 'ai';
 import { Run, GetRunContext } from './context.js';
 import { Serializable } from './internals/serializable.js';
-import { C as Callback, E as Emitter } from './emitter-CZFbzlUi.js';
+import { C as Callback, E as Emitter } from './emitter-36-9MnvA.js';
 import { FrameworkError } from './errors.js';
 import { Task } from 'promise-based-task';
 import { BaseCache } from './cache/base.js';
@@ -13,6 +12,7 @@ import { ZodSchema, z, ZodType } from 'zod';
 import { SchemaObject } from 'ajv';
 import { PromptTemplate } from './template.js';
 import { Logger } from './logger/logger.js';
+import { ToolCallPart } from 'ai';
 
 /**
  * Copyright 2025 © BeeAI a Series of LF Projects, LLC
@@ -39,6 +39,7 @@ interface ChatModelParameters {
     presencePenalty?: number;
     seed?: number;
     stopSequences?: string[];
+    stream?: boolean;
 }
 interface ResponseObjectJson {
     type: "object-json";
@@ -67,6 +68,7 @@ interface ChatModelInput extends ChatModelParameters {
     responseFormat?: ZodSchema | ResponseObjectJson;
     toolChoice?: ChatModelToolChoice;
     messages: Message[];
+    streamPartialToolCalls?: boolean;
 }
 type ChatModelFinishReason = "stop" | "length" | "content-filter" | "tool-calls" | "error" | "other" | "unknown";
 interface ChatModelUsage {
@@ -113,11 +115,7 @@ declare abstract class ChatModel extends Serializable {
     readonly modelSupportsToolCalling: boolean;
     abstract get modelId(): string;
     abstract get providerId(): string;
-    create(input: ChatModelInput & {
-        stream?: boolean;
-    }): Run<ChatModelOutput, this, readonly [ChatModelInput & {
-        stream?: boolean;
-    }]>;
+    create(input: ChatModelInput): Run<ChatModelOutput, this, readonly [ChatModelInput]>;
     createStructure<T>(input: ChatModelObjectInput<T>): Run<ChatModelObjectOutput<T>, this, readonly [ChatModelObjectInput<T>]>;
     config({ cache, parameters }: ChatConfig): void;
     static fromName(name: FullModelName | ProviderName, options?: ChatModelParameters): Promise<ChatModel>;
@@ -150,7 +148,8 @@ declare class ChatModelOutput extends Serializable {
     constructor(messages: Message[], usage?: ChatModelUsage | undefined, finishReason?: ChatModelFinishReason | undefined);
     static fromChunks(chunks: ChatModelOutput[]): ChatModelOutput;
     merge(other: ChatModelOutput): void;
-    getToolCalls(): ai.ToolCallPart[];
+    dedupe(): void;
+    getToolCalls(): ToolCallPart[];
     getTextMessages(): AssistantMessage[];
     getTextContent(): string;
     toString(): string;

package/dist/context.cjs

@@ -12,6 +12,11 @@ var stream_cjs = require('./internals/helpers/stream.cjs');
 
 var __defProp = Object.defineProperty;
 var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+class Middleware {
+  static {
+    __name(this, "Middleware");
+  }
+}
 class Run extends promise_cjs.LazyPromise {
   static {
     __name(this, "Run");
@@ -33,8 +38,10 @@ class Run extends promise_cjs.LazyPromise {
     });
     return this;
   }
-  middleware(fn) {
-    this.tasks.push(async () => fn(this.runContext));
+  middleware(...fns) {
+    for (const fn of fns) {
+      this.tasks.push(async () => remeda.isFunction(fn) ? fn(this.runContext) : fn.bind(this.runContext));
+    }
     return this;
   }
   async before() {
@@ -117,20 +124,29 @@ class RunContext extends serializable_cjs.Serializable {
           internal: true
         }
       });
+      const finishEvent = {};
       try {
-        await emitter.emit("start", null);
+        const startEvent = {
+          input: runContext.runParams,
+          output: void 0
+        };
+        await emitter.emit("start", startEvent);
+        runContext.runParams = startEvent.input;
+        finishEvent.input = startEvent.input;
         const result = await Promise.race([
-          RunContext.#storage.run(runContext, fn, runContext),
+          RunContext.#storage.run(runContext, startEvent.output === void 0 ? fn : async () => startEvent.output, runContext),
           new Promise((_, reject) => runContext.signal.addEventListener("abort", () => setTimeout(() => reject(runContext.signal.reason), 0)))
         ]);
+        finishEvent.output = result;
         await emitter.emit("success", result);
         return result;
       } catch (_e) {
         const e = errors_cjs.FrameworkError.ensure(_e);
+        finishEvent.output = e;
         await emitter.emit("error", e);
         throw e;
       } finally {
-        await emitter.emit("finish", null);
+        await emitter.emit("finish", finishEvent);
         runContext.destroy();
       }
     }, runContext);
@@ -155,6 +171,7 @@ class RunContext extends serializable_cjs.Serializable {
   }
 }
 
+exports.Middleware = Middleware;
 exports.Run = Run;
 exports.RunContext = RunContext;
 //# sourceMappingURL=context.cjs.map

package/dist/context.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/context.ts"],"names":["Run","LazyPromise","tasks","handler","runContext","Symbol","toStringTag","observe","fn","push","emitter","context","value","Object","assign","middleware","before","executeSequentially","splice","Infinity","asyncIterator","response","then","isAsyncIterable","Error","RunContext","Serializable","AsyncLocalStorage","controller","runId","groupId","parentId","runParams","createdAt","signal","abort","reason","instance","input","parent","Date","params","createRandomHash","omit","AbortController","registerSignals","child","trace","id","parentRunId","pipe","destroy","FrameworkError","enter","getStore","namespace","creator","internal","emit","result","Promise","race","run","_","reject","addEventListener","setTimeout","_e","e","ensure","register","createSnapshot","shallowCopy","loadSnapshot","snapshot"],"mappings":";;;;;;;;;;;;;;AA+BO,MAAMA,YAA+CC,uBAAAA,CAAAA;EA/B5D;;;;AAgCqBC,EAAAA,KAAAA,GAAiC,EAAA;AAEpD,EAAA,WAAA,CACEC,SACmBC,UAAAA,EACnB;AACA,IAAA,KAAA,CAAMD,OAAAA,CAAAA,EAAAA,IAAAA,CAFaC,UAAAA,GAAAA,UAAAA;AAGrB;EAES,CAACC,MAAAA,CAAOC,WAAW,IAAI,SAAA;AAEhCC,EAAAA,OAAAA,CAAQC,EAAAA,EAAmD;AACzD,IAAA,IAAA,CAAKN,MAAMO,IAAAA,CAAK,YAAYD,GAAG,IAAA,CAAKJ,UAAAA,CAAWM,OAAO,CAAA,CAAA;AACtD,IAAA,OAAO,IAAA;AACT;AAEAC,EAAAA,OAAAA,CAAQC,KAAAA,EAAe;AACrB,IAAA,IAAA,CAAKV,KAAAA,CAAMO,KAAK,YAAA;AACdI,MAAAA,MAAAA,CAAOC,MAAAA,CAAO,IAAA,CAAKV,UAAAA,CAAWO,OAAAA,EAASC,KAAAA,CAAAA;AACvCC,MAAAA,MAAAA,CAAOC,MAAAA,CAAO,IAAA,CAAKV,UAAAA,CAAWM,OAAAA,CAAQC,SAASC,KAAAA,CAAAA;KACjD,CAAA;AACA,IAAA,OAAO,IAAA;AACT;AAEAG,EAAAA,UAAAA,CAAWP,EAAAA,EAA4C;AACrD,IAAA,IAAA,CAAKN,MAAMO,IAAAA,CAAK,YAAYD,EAAAA,CAAG,IAAA,CAAKJ,UAAU,CAAA,CAAA;AAC9C,IAAA,OAAO,IAAA;AACT;AAEA,EAAA,MAAgBY,MAAAA,GAAwB;AACtC,IAAA,MAAM,MAAMA,MAAAA,EAAAA;AACZ,IAAA,MAAMC,gCAAoB,IAAA,CAAKf,KAAAA,CAAMgB,MAAAA,CAAO,CAAA,EAAGC,QAAAA,CAAAA,CAAAA;AACjD;;EAGA,QAAQd,MAAAA,CAAOe,aAAa,CAAA,GAAiE;AAC3F,IAAA,MAAMC,QAAAA,GAAW,MAAM,IAAA,CAAKC,IAAAA,EAAI;AAChC,IAAA,IAAIC,0BAAAA,CAAgBF,QAAAA,CAAAA,EAAW;AAC7B,MAAA,OAAOA,QAAAA;KACT,MAAO;AACL,MAAA,MAAM,IAAIG,MAAM,yBAAA,CAAA;AAClB;AACF;AACF;AAOO,MAAMC,mBAAmDC,6BAAAA,CAAAA;EAlFhE;;;;;EAmFE,OAAO,QAAA,GAAW,IAAIC,kCAAAA,EAAAA;AAEHC,EAAAA,UAAAA;AACHC,EAAAA,KAAAA;AACAC,EAAAA,OAAAA;AACAC,EAAAA,QAAAA;AACArB,EAAAA,OAAAA;AACAC,EAAAA,OAAAA;AACAqB,EAAAA,SAAAA;AACAC,EAAAA,SAAAA;AAEhB,EAAA,IAAIC,MAAAA,GAAS;AACX,IAAA,OAAO,KAAKN,UAAAA,CAAWM,MAAAA;AACzB;AAEAC,EAAAA,KAAAA,CAAMC,MAAAA,EAAgB;AACpB,IAAA,IAAA,CAAKR,UAAAA,CAAWO,MAAMC,MAAAA,CAAAA;AACxB;EAEA,WAAA,CACkBC,QAAAA,EACGC,OACnBC,MAAAA,EACA;AACA,IAAA,KAAA,EAAK,EAAA,IAAA,CAJWF,QAAAA,GAAAA,QAAAA,EAAAA,KACGC,KAAAA,GAAAA,KAAAA;AAInB,IAAA,IAAA,CAAKL,SAAAA,uBAAgBO,IAAAA,EAAAA;AACrB,IAAA,IAAA,CAAKR,YAAYM,KAAAA,CAAMG,MAAAA;AACvB,IAAA,IAAA,CAAKZ,KAAAA,GAAQa,0BAAiB,CAAA,CAAA;AAC9B,IAAA,IAAA,CAAKX,WAAWQ,MAAAA,EAAQV,KAAAA;AACxB,IAAA,IAAA,CAAKC,OAAAA,GAAUS,MAAAA,EAAQT,OAAAA,IAAWY,yBAAAA,EAAAA;AAClC,IAAA,IAAA,CAAK/B,OAAAA,GAAUgC,WAAAA,CAAMJ,MAAAA,EAAQ5B,OAAAA,IAAW,EAAC,EAAW;AAAC,MAAA,IAAA;AAAM,MAAA;AAAW,KAAA,CAAA;AAEtE,IAAA,IAAA,CAAKiB,UAAAA,GAAa,IAAIgB,eAAAA,EAAAA;AACtBC,IAAAA,gCAAAA,CAAgB,KAAKjB,UAAAA,EAAY;MAACU,KAAAA,CAAMJ,MAAAA;MAAQK,MAAAA,EAAQL;AAAO,KAAA,CAAA;AAE/D,IAAA,IAAA,CAAKxB,OAAAA,GAAU2B,QAAAA,CAAS3B,OAAAA,CAAQoC,KAAAA,CAAyB;AACvDnC,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;MACdoC,KAAAA,EAAO;AACLC,QAAAA,EAAAA,EAAI,IAAA,CAAKlB,OAAAA;AACTD,QAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AACZoB,QAAAA,WAAAA,EAAaV,MAAAA,EAAQV;AACvB;KACF,CAAA;AACA,IAAA,IAAIU,MAAAA,EAAQ;AACV,MAAA,IAAA,CAAK7B,OAAAA,CAAQwC,IAAAA,CAAKX,MAAAA,CAAO7B,OAAO,CAAA;AAClC;AACF;EAEAyC,OAAAA,GAAU;AACR,IAAA,IAAA,CAAKzC,QAAQyC,OAAAA,EAAO;AACpB,IAAA,IAAA,CAAKvB,UAAAA,CAAWO,KAAAA,CAAM,IAAIiB,yBAAAA,CAAe,oBAAA,CAAA,CAAA;AAC3C;EAEA,OAAOC,KAAAA,CACLhB,QAAAA,EACAC,KAAAA,EACA9B,EAAAA,EACA;AACA,IAAA,MAAM+B,MAAAA,GAASd,UAAAA,CAAW,QAAA,CAAS6B,QAAAA,EAAQ;AAC3C,IAAA,MAAMlD,UAAAA,GAAa,IAAIqB,UAAAA,CAAWY,QAAAA,EAAUC,OAAOC,MAAAA,CAAAA;AAEnD,IAAA,OAAO,IAAIvC,IAAI,YAAA;AACb,MAAA,MAAMU,OAAAA,GAAUN,UAAAA,CAAWM,OAAAA,CAAQoC,KAAAA,CAA2B;QAC5DS,SAAAA,EAAW;AAAC,UAAA;;QACZC,OAAAA,EAASpD,UAAAA;QACTO,OAAAA,EAAS;UAAE8C,QAAAA,EAAU;AAAK;OAC5B,CAAA;AAEA,MAAA,IAAI;AACF,QAAA,MAAM/C,OAAAA,CAAQgD,IAAAA,CAAK,OAAA,EAAS,IAAA,CAAA;AAC5B,QAAA,MAAMC,MAAAA,GAAS,MAAMC,OAAAA,CAAQC,IAAAA,CAAK;AAChCpC,UAAAA,UAAAA,CAAW,QAAA,CAASqC,GAAAA,CAAI1D,UAAAA,EAAYI,EAAAA,EAAIJ,UAAAA,CAAAA;AACxC,UAAA,IAAIwD,QAAe,CAACG,CAAAA,EAAGC,WACrB5D,UAAAA,CAAW8B,MAAAA,CAAO+B,iBAAiB,OAAA,EAAS,MAC1CC,UAAAA,CAAW,MAAMF,OAAO5D,UAAAA,CAAW8B,MAAAA,CAAOE,MAAM,CAAA,EAAG,CAAA,CAAA,CAAA;AAGxD,SAAA,CAAA;AACD,QAAA,MAAM1B,OAAAA,CAAQgD,IAAAA,CAAK,SAAA,EAAWC,MAAAA,CAAAA;AAC9B,QAAA,OAAOA,MAAAA;AACT,OAAA,CAAA,OAASQ,EAAAA,EAAI;AACX,QAAA,MAAMC,CAAAA,GAAIhB,yBAAAA,CAAeiB,MAAAA,CAAOF,EAAAA,CAAAA;AAChC,QAAA,MAAMzD,OAAAA,CAAQgD,IAAAA,CAAK,OAAA,EAASU,CAAAA,CAAAA;AAC5B,QAAA,MAAMA,CAAAA;OACR,SAAA;AACE,QAAA,MAAM1D,OAAAA,CAAQgD,IAAAA,CAAK,QAAA,EAAU,IAAA,CAAA;AAC7BtD,QAAAA,UAAAA,CAAW+C,OAAAA,EAAO;AACpB;AACF,KAAA,EAAG/C,UAAAA,CAAAA;AACL;EAEA;AACE,IAAA,IAAA,CAAKkE,QAAAA,EAAQ;AACf;EAEAC,cAAAA,GAAiB;AACf,IAAA,OAAO;AACL3C,MAAAA,UAAAA,EAAY,IAAA,CAAKA,UAAAA;AACjBC,MAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AACZC,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;AACdC,MAAAA,QAAAA,EAAU,IAAA,CAAKA,QAAAA;AACfrB,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;MACdC,OAAAA,EAAS6D,qBAAAA,CAAY,KAAK7D,OAAO,CAAA;MACjCqB,SAAAA,EAAWwC,qBAAAA,CAAY,KAAKxC,SAAS,CAAA;MACrCC,SAAAA,EAAW,IAAIO,IAAAA,CAAK,IAAA,CAAKP,SAAS;AACpC,KAAA;AACF;AAEAwC,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7D7D,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAM4D,QAAAA,CAAAA;AACtB;AACF","file":"context.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { AsyncLocalStorage } from \"node:async_hooks\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { createRandomHash } from \"@/internals/helpers/hash.js\";\nimport { omit } from \"remeda\";\nimport { Callback } from \"@/emitter/types.js\";\nimport { registerSignals } from \"@/internals/helpers/cancellation.js\";\nimport { Serializable } from \"@/internals/serializable.js\";\nimport { executeSequentially, LazyPromise } from \"@/internals/helpers/promise.js\";\nimport { FrameworkError } from \"@/errors.js\";\nimport { shallowCopy } from \"@/serializer/utils.js\";\nimport { isAsyncIterable } from \"@/internals/helpers/stream.js\";\n\nexport interface RunInstance<T = any> {\n  emitter: Emitter<T>;\n}\n\nexport interface RunContextCallbacks {\n  start: Callback<null>;\n  success: Callback<unknown>;\n  error: Callback<Error>;\n  finish: Callback<null>;\n}\n\nexport type GetRunContext<T, P = any> = T extends RunInstance ? RunContext<T, P> : never;\nexport type GetRunInstance<T> = T extends RunInstance<infer P> ? P : never;\n\nexport class Run<R, I extends RunInstance, P = any> extends LazyPromise<R> {\n  protected readonly tasks: (() => Promise<void>)[] = [];\n\n  constructor(\n    handler: () => Promise<R>,\n    protected readonly runContext: GetRunContext<I, P>,\n  ) {\n    super(handler);\n  }\n\n  readonly [Symbol.toStringTag] = \"Promise\";\n\n  observe(fn: (emitter: Emitter<GetRunInstance<I>>) => void) {\n    this.tasks.push(async () => fn(this.runContext.emitter));\n    return this;\n  }\n\n  context(value: object) {\n    this.tasks.push(async () => {\n      Object.assign(this.runContext.context, value);\n      Object.assign(this.runContext.emitter.context, value);\n    });\n    return this;\n  }\n\n  middleware(fn: (context: GetRunContext<I, P>) => void) {\n    this.tasks.push(async () => fn(this.runContext));\n    return this;\n  }\n\n  protected async before(): Promise<void> {\n    await super.before();\n    await executeSequentially(this.tasks.splice(0, Infinity));\n  }\n\n  // @ts-expect-error too complex\n  async *[Symbol.asyncIterator](): R extends AsyncIterable<infer X> ? AsyncIterator<X> : never {\n    const response = await this.then();\n    if (isAsyncIterable(response)) {\n      yield* response;\n    } else {\n      throw new Error(\"Result is not iterable!\");\n    }\n  }\n}\n\nexport interface RunContextInput<P> {\n  params: P;\n  signal?: AbortSignal;\n}\n\nexport class RunContext<T extends RunInstance, P = any> extends Serializable {\n  static #storage = new AsyncLocalStorage<RunContext<any>>();\n\n  protected readonly controller: AbortController;\n  public readonly runId: string;\n  public readonly groupId: string;\n  public readonly parentId?: string;\n  public readonly emitter;\n  public readonly context: object;\n  public readonly runParams: P;\n  public readonly createdAt: Date;\n\n  get signal() {\n    return this.controller.signal;\n  }\n\n  abort(reason?: Error) {\n    this.controller.abort(reason);\n  }\n\n  constructor(\n    public readonly instance: T,\n    protected readonly input: RunContextInput<P>,\n    parent?: RunContext<any>,\n  ) {\n    super();\n    this.createdAt = new Date();\n    this.runParams = input.params;\n    this.runId = createRandomHash(5);\n    this.parentId = parent?.runId;\n    this.groupId = parent?.groupId ?? createRandomHash();\n    this.context = omit((parent?.context ?? {}) as any, [\"id\", \"parentId\"]);\n\n    this.controller = new AbortController();\n    registerSignals(this.controller, [input.signal, parent?.signal]);\n\n    this.emitter = instance.emitter.child<GetRunInstance<T>>({\n      context: this.context,\n      trace: {\n        id: this.groupId,\n        runId: this.runId,\n        parentRunId: parent?.runId,\n      },\n    });\n    if (parent) {\n      this.emitter.pipe(parent.emitter);\n    }\n  }\n\n  destroy() {\n    this.emitter.destroy();\n    this.controller.abort(new FrameworkError(\"Context destroyed.\"));\n  }\n\n  static enter<C2 extends RunInstance, R2, P2>(\n    instance: C2,\n    input: RunContextInput<P2>,\n    fn: (context: GetRunContext<C2, P2>) => Promise<R2>,\n  ) {\n    const parent = RunContext.#storage.getStore();\n    const runContext = new RunContext(instance, input, parent) as GetRunContext<C2, P2>;\n\n    return new Run(async () => {\n      const emitter = runContext.emitter.child<RunContextCallbacks>({\n        namespace: [\"run\"],\n        creator: runContext,\n        context: { internal: true },\n      });\n\n      try {\n        await emitter.emit(\"start\", null);\n        const result = await Promise.race([\n          RunContext.#storage.run(runContext, fn, runContext),\n          new Promise<never>((_, reject) =>\n            runContext.signal.addEventListener(\"abort\", () =>\n              setTimeout(() => reject(runContext.signal.reason), 0),\n            ),\n          ),\n        ]);\n        await emitter.emit(\"success\", result);\n        return result;\n      } catch (_e) {\n        const e = FrameworkError.ensure(_e);\n        await emitter.emit(\"error\", e);\n        throw e;\n      } finally {\n        await emitter.emit(\"finish\", null);\n        runContext.destroy();\n      }\n    }, runContext);\n  }\n\n  static {\n    this.register();\n  }\n\n  createSnapshot() {\n    return {\n      controller: this.controller,\n      runId: this.runId,\n      groupId: this.groupId,\n      parentId: this.parentId,\n      emitter: this.emitter,\n      context: shallowCopy(this.context),\n      runParams: shallowCopy(this.runParams),\n      createdAt: new Date(this.createdAt),\n    };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n  }\n}\n"]}
+{"version":3,"sources":["../src/context.ts"],"names":["Middleware","Run","LazyPromise","tasks","handler","runContext","Symbol","toStringTag","observe","fn","push","emitter","context","value","Object","assign","middleware","fns","isFunction","bind","before","executeSequentially","splice","Infinity","asyncIterator","response","then","isAsyncIterable","Error","RunContext","Serializable","AsyncLocalStorage","controller","runId","groupId","parentId","runParams","createdAt","signal","abort","reason","instance","input","parent","Date","params","createRandomHash","omit","AbortController","registerSignals","child","trace","id","parentRunId","pipe","destroy","FrameworkError","enter","getStore","namespace","creator","internal","finishEvent","startEvent","output","undefined","emit","result","Promise","race","run","_","reject","addEventListener","setTimeout","_e","e","ensure","register","createSnapshot","shallowCopy","loadSnapshot","snapshot"],"mappings":";;;;;;;;;;;;;;AA4BO,MAAeA,UAAAA,CAAAA;EA5BtB;;;AA8BA;AAOO,MAAMC,YAA+CC,uBAAAA,CAAAA;EArC5D;;;;AAsCqBC,EAAAA,KAAAA,GAAiC,EAAA;AAEpD,EAAA,WAAA,CACEC,SACmBC,UAAAA,EACnB;AACA,IAAA,KAAA,CAAMD,OAAAA,CAAAA,EAAAA,IAAAA,CAFaC,UAAAA,GAAAA,UAAAA;AAGrB;EAES,CAACC,MAAAA,CAAOC,WAAW,IAAI,SAAA;AAEhCC,EAAAA,OAAAA,CAAQC,EAAAA,EAAmD;AACzD,IAAA,IAAA,CAAKN,MAAMO,IAAAA,CAAK,YAAYD,GAAG,IAAA,CAAKJ,UAAAA,CAAWM,OAAO,CAAA,CAAA;AACtD,IAAA,OAAO,IAAA;AACT;AAEAC,EAAAA,OAAAA,CAAQC,KAAAA,EAAe;AACrB,IAAA,IAAA,CAAKV,KAAAA,CAAMO,KAAK,YAAA;AACdI,MAAAA,MAAAA,CAAOC,MAAAA,CAAO,IAAA,CAAKV,UAAAA,CAAWO,OAAAA,EAASC,KAAAA,CAAAA;AACvCC,MAAAA,MAAAA,CAAOC,MAAAA,CAAO,IAAA,CAAKV,UAAAA,CAAWM,OAAAA,CAAQC,SAASC,KAAAA,CAAAA;KACjD,CAAA;AACA,IAAA,OAAO,IAAA;AACT;AAEAG,EAAAA,UAAAA,CAAAA,GAAcC,GAAAA,EAA0B;AACtC,IAAA,KAAA,MAAWR,MAAMQ,GAAAA,EAAK;AACpB,MAAA,IAAA,CAAKd,KAAAA,CAAMO,IAAAA,CAAK,YACdQ,iBAAAA,CAAWT,EAAAA,CAAAA,GAAMA,EAAAA,CAAG,IAAA,CAAKJ,UAAU,CAAA,GAAII,EAAAA,CAAGU,IAAAA,CAAK,IAAA,CAAKd,UAAU,CAAA,CAAA;AAElE;AACA,IAAA,OAAO,IAAA;AACT;AAEA,EAAA,MAAgBe,MAAAA,GAAwB;AACtC,IAAA,MAAM,MAAMA,MAAAA,EAAAA;AACZ,IAAA,MAAMC,gCAAoB,IAAA,CAAKlB,KAAAA,CAAMmB,MAAAA,CAAO,CAAA,EAAGC,QAAAA,CAAAA,CAAAA;AACjD;;EAGA,QAAQjB,MAAAA,CAAOkB,aAAa,CAAA,GAAiE;AAC3F,IAAA,MAAMC,QAAAA,GAAW,MAAM,IAAA,CAAKC,IAAAA,EAAI;AAChC,IAAA,IAAIC,0BAAAA,CAAgBF,QAAAA,CAAAA,EAAW;AAC7B,MAAA,OAAOA,QAAAA;KACT,MAAO;AACL,MAAA,MAAM,IAAIG,MAAM,yBAAA,CAAA;AAClB;AACF;AACF;AAOO,MAAMC,mBAAmDC,6BAAAA,CAAAA;EA5FhE;;;;;EA6FE,OAAO,QAAA,GAAW,IAAIC,kCAAAA,EAAAA;AAEHC,EAAAA,UAAAA;AACHC,EAAAA,KAAAA;AACAC,EAAAA,OAAAA;AACAC,EAAAA,QAAAA;AACAxB,EAAAA,OAAAA;AACAC,EAAAA,OAAAA;AACTwB,EAAAA,SAAAA;AACSC,EAAAA,SAAAA;AAEhB,EAAA,IAAIC,MAAAA,GAAS;AACX,IAAA,OAAO,KAAKN,UAAAA,CAAWM,MAAAA;AACzB;AAEAC,EAAAA,KAAAA,CAAMC,MAAAA,EAAgB;AACpB,IAAA,IAAA,CAAKR,UAAAA,CAAWO,MAAMC,MAAAA,CAAAA;AACxB;EAEA,WAAA,CACkBC,QAAAA,EACGC,OACnBC,MAAAA,EACA;AACA,IAAA,KAAA,EAAK,EAAA,IAAA,CAJWF,QAAAA,GAAAA,QAAAA,EAAAA,KACGC,KAAAA,GAAAA,KAAAA;AAInB,IAAA,IAAA,CAAKL,SAAAA,uBAAgBO,IAAAA,EAAAA;AACrB,IAAA,IAAA,CAAKR,YAAYM,KAAAA,CAAMG,MAAAA;AACvB,IAAA,IAAA,CAAKZ,KAAAA,GAAQa,0BAAiB,CAAA,CAAA;AAC9B,IAAA,IAAA,CAAKX,WAAWQ,MAAAA,EAAQV,KAAAA;AACxB,IAAA,IAAA,CAAKC,OAAAA,GAAUS,MAAAA,EAAQT,OAAAA,IAAWY,yBAAAA,EAAAA;AAClC,IAAA,IAAA,CAAKlC,OAAAA,GAAUmC,WAAAA,CAAMJ,MAAAA,EAAQ/B,OAAAA,IAAW,EAAC,EAAW;AAAC,MAAA,IAAA;AAAM,MAAA;AAAW,KAAA,CAAA;AAEtE,IAAA,IAAA,CAAKoB,UAAAA,GAAa,IAAIgB,eAAAA,EAAAA;AACtBC,IAAAA,gCAAAA,CAAgB,KAAKjB,UAAAA,EAAY;MAACU,KAAAA,CAAMJ,MAAAA;MAAQK,MAAAA,EAAQL;AAAO,KAAA,CAAA;AAE/D,IAAA,IAAA,CAAK3B,OAAAA,GAAU8B,QAAAA,CAAS9B,OAAAA,CAAQuC,KAAAA,CAAyB;AACvDtC,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;MACduC,KAAAA,EAAO;AACLC,QAAAA,EAAAA,EAAI,IAAA,CAAKlB,OAAAA;AACTD,QAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AACZoB,QAAAA,WAAAA,EAAaV,MAAAA,EAAQV;AACvB;KACF,CAAA;AACA,IAAA,IAAIU,MAAAA,EAAQ;AACV,MAAA,IAAA,CAAKhC,OAAAA,CAAQ2C,IAAAA,CAAKX,MAAAA,CAAOhC,OAAO,CAAA;AAClC;AACF;EAEA4C,OAAAA,GAAU;AACR,IAAA,IAAA,CAAK5C,QAAQ4C,OAAAA,EAAO;AACpB,IAAA,IAAA,CAAKvB,UAAAA,CAAWO,KAAAA,CAAM,IAAIiB,yBAAAA,CAAe,oBAAA,CAAA,CAAA;AAC3C;EAEA,OAAOC,KAAAA,CACLhB,QAAAA,EACAC,KAAAA,EACAjC,EAAAA,EACA;AACA,IAAA,MAAMkC,MAAAA,GAASd,UAAAA,CAAW,QAAA,CAAS6B,QAAAA,EAAQ;AAC3C,IAAA,MAAMrD,UAAAA,GAAa,IAAIwB,UAAAA,CAAWY,QAAAA,EAAUC,OAAOC,MAAAA,CAAAA;AAEnD,IAAA,OAAO,IAAI1C,IAAI,YAAA;AACb,MAAA,MAAMU,OAAAA,GAAUN,UAAAA,CAAWM,OAAAA,CAAQuC,KAAAA,CAA2B;QAC5DS,SAAAA,EAAW;AAAC,UAAA;;QACZC,OAAAA,EAASvD,UAAAA;QACTO,OAAAA,EAAS;UAAEiD,QAAAA,EAAU;AAAK;OAC5B,CAAA;AAEA,MAAA,MAAMC,cAAc,EAAC;AACrB,MAAA,IAAI;AACF,QAAA,MAAMC,UAAAA,GAA+D;AACnErB,UAAAA,KAAAA,EAAOrC,UAAAA,CAAW+B,SAAAA;UAClB4B,MAAAA,EAAQC,KAAAA;AACV,SAAA;AACA,QAAA,MAAMtD,OAAAA,CAAQuD,IAAAA,CAAK,OAAA,EAASH,UAAAA,CAAAA;AAG5B1D,QAAAA,UAAAA,CAAW+B,YAAY2B,UAAAA,CAAWrB,KAAAA;AAClCoB,QAAAA,WAAAA,CAAYpB,QAAQqB,UAAAA,CAAWrB,KAAAA;AAE/B,QAAA,MAAMyB,MAAAA,GAAa,MAAMC,OAAAA,CAAQC,IAAAA,CAAK;UACpCxC,UAAAA,CAAW,QAAA,CAASyC,GAAAA,CAClBjE,UAAAA,EACA0D,UAAAA,CAAWC,MAAAA,KAAWC,SAAYxD,EAAAA,GAAK,YAAYsD,UAAAA,CAAWC,MAAAA,EAC9D3D,UAAAA,CAAAA;AAEF,UAAA,IAAI+D,QAAe,CAACG,CAAAA,EAAGC,WACrBnE,UAAAA,CAAWiC,MAAAA,CAAOmC,iBAAiB,OAAA,EAAS,MAC1CC,UAAAA,CAAW,MAAMF,OAAOnE,UAAAA,CAAWiC,MAAAA,CAAOE,MAAM,CAAA,EAAG,CAAA,CAAA,CAAA;AAGxD,SAAA,CAAA;AACDsB,QAAAA,WAAAA,CAAYE,MAAAA,GAASG,MAAAA;AACrB,QAAA,MAAMxD,OAAAA,CAAQuD,IAAAA,CAAK,SAAA,EAAWC,MAAAA,CAAAA;AAC9B,QAAA,OAAOA,MAAAA;AACT,OAAA,CAAA,OAASQ,EAAAA,EAAI;AACX,QAAA,MAAMC,CAAAA,GAAIpB,yBAAAA,CAAeqB,MAAAA,CAAOF,EAAAA,CAAAA;AAChCb,QAAAA,WAAAA,CAAYE,MAAAA,GAASY,CAAAA;AACrB,QAAA,MAAMjE,OAAAA,CAAQuD,IAAAA,CAAK,OAAA,EAASU,CAAAA,CAAAA;AAC5B,QAAA,MAAMA,CAAAA;OACR,SAAA;AACE,QAAA,MAAMjE,OAAAA,CAAQuD,IAAAA,CAAK,QAAA,EAAUJ,WAAAA,CAAAA;AAC7BzD,QAAAA,UAAAA,CAAWkD,OAAAA,EAAO;AACpB;AACF,KAAA,EAAGlD,UAAAA,CAAAA;AACL;EAEA;AACE,IAAA,IAAA,CAAKyE,QAAAA,EAAQ;AACf;EAEAC,cAAAA,GAAiB;AACf,IAAA,OAAO;AACL/C,MAAAA,UAAAA,EAAY,IAAA,CAAKA,UAAAA;AACjBC,MAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AACZC,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;AACdC,MAAAA,QAAAA,EAAU,IAAA,CAAKA,QAAAA;AACfxB,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;MACdC,OAAAA,EAASoE,qBAAAA,CAAY,KAAKpE,OAAO,CAAA;MACjCwB,SAAAA,EAAW4C,qBAAAA,CAAY,KAAK5C,SAAS,CAAA;MACrCC,SAAAA,EAAW,IAAIO,IAAAA,CAAK,IAAA,CAAKP,SAAS;AACpC,KAAA;AACF;AAEA4C,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7DpE,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAMmE,QAAAA,CAAAA;AACtB;AACF","file":"context.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { AsyncLocalStorage } from \"node:async_hooks\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { createRandomHash } from \"@/internals/helpers/hash.js\";\nimport { isFunction, omit } from \"remeda\";\nimport { Callback, InferCallbackValue } from \"@/emitter/types.js\";\nimport { registerSignals } from \"@/internals/helpers/cancellation.js\";\nimport { Serializable } from \"@/internals/serializable.js\";\nimport { executeSequentially, LazyPromise } from \"@/internals/helpers/promise.js\";\nimport { FrameworkError } from \"@/errors.js\";\nimport { shallowCopy } from \"@/serializer/utils.js\";\nimport { isAsyncIterable } from \"@/internals/helpers/stream.js\";\n\nexport interface RunInstance<T = any> {\n  emitter: Emitter<T>;\n}\n\nexport interface RunContextCallbacks {\n  start: Callback<{ input: any; output: any }>;\n  success: Callback<unknown>; // TODO: should be updated to Callback<{ input: any; output: any }>\n  error: Callback<Error>;\n  finish: Callback<{ input: any; output?: any; error?: FrameworkError }>;\n}\n\nexport abstract class Middleware<T extends RunInstance = any> {\n  abstract bind(ctx: RunContext<T>): void;\n}\n\nexport type GetRunContext<T, P = any> = T extends RunInstance ? RunContext<T, P> : never;\nexport type GetRunInstance<T> = T extends RunInstance<infer P> ? P : never;\nexport type MiddlewareFn<T> = (context: GetRunContext<T>) => void;\nexport type MiddlewareType<T extends RunInstance> = Middleware<T> | MiddlewareFn<T>;\n\nexport class Run<R, I extends RunInstance, P = any> extends LazyPromise<R> {\n  protected readonly tasks: (() => Promise<void>)[] = [];\n\n  constructor(\n    handler: () => Promise<R>,\n    protected readonly runContext: GetRunContext<I, P>,\n  ) {\n    super(handler);\n  }\n\n  readonly [Symbol.toStringTag] = \"Promise\";\n\n  observe(fn: (emitter: Emitter<GetRunInstance<I>>) => void) {\n    this.tasks.push(async () => fn(this.runContext.emitter));\n    return this;\n  }\n\n  context(value: object) {\n    this.tasks.push(async () => {\n      Object.assign(this.runContext.context, value);\n      Object.assign(this.runContext.emitter.context, value);\n    });\n    return this;\n  }\n\n  middleware(...fns: MiddlewareType<I>[]) {\n    for (const fn of fns) {\n      this.tasks.push(async () =>\n        isFunction(fn) ? fn(this.runContext) : fn.bind(this.runContext),\n      );\n    }\n    return this;\n  }\n\n  protected async before(): Promise<void> {\n    await super.before();\n    await executeSequentially(this.tasks.splice(0, Infinity));\n  }\n\n  // @ts-expect-error too complex\n  async *[Symbol.asyncIterator](): R extends AsyncIterable<infer X> ? AsyncIterator<X> : never {\n    const response = await this.then();\n    if (isAsyncIterable(response)) {\n      yield* response;\n    } else {\n      throw new Error(\"Result is not iterable!\");\n    }\n  }\n}\n\nexport interface RunContextInput<P> {\n  params: P;\n  signal?: AbortSignal;\n}\n\nexport class RunContext<T extends RunInstance, P = any> extends Serializable {\n  static #storage = new AsyncLocalStorage<RunContext<any>>();\n\n  protected readonly controller: AbortController;\n  public readonly runId: string;\n  public readonly groupId: string;\n  public readonly parentId?: string;\n  public readonly emitter;\n  public readonly context: object;\n  public runParams: P;\n  public readonly createdAt: Date;\n\n  get signal() {\n    return this.controller.signal;\n  }\n\n  abort(reason?: Error) {\n    this.controller.abort(reason);\n  }\n\n  constructor(\n    public readonly instance: T,\n    protected readonly input: RunContextInput<P>,\n    parent?: RunContext<any>,\n  ) {\n    super();\n    this.createdAt = new Date();\n    this.runParams = input.params;\n    this.runId = createRandomHash(5);\n    this.parentId = parent?.runId;\n    this.groupId = parent?.groupId ?? createRandomHash();\n    this.context = omit((parent?.context ?? {}) as any, [\"id\", \"parentId\"]);\n\n    this.controller = new AbortController();\n    registerSignals(this.controller, [input.signal, parent?.signal]);\n\n    this.emitter = instance.emitter.child<GetRunInstance<T>>({\n      context: this.context,\n      trace: {\n        id: this.groupId,\n        runId: this.runId,\n        parentRunId: parent?.runId,\n      },\n    });\n    if (parent) {\n      this.emitter.pipe(parent.emitter);\n    }\n  }\n\n  destroy() {\n    this.emitter.destroy();\n    this.controller.abort(new FrameworkError(\"Context destroyed.\"));\n  }\n\n  static enter<C2 extends RunInstance, R2, P2>(\n    instance: C2,\n    input: RunContextInput<P2>,\n    fn: (context: GetRunContext<C2, P2>) => Promise<R2>,\n  ) {\n    const parent = RunContext.#storage.getStore();\n    const runContext = new RunContext(instance, input, parent) as GetRunContext<C2, P2>;\n\n    return new Run(async () => {\n      const emitter = runContext.emitter.child<RunContextCallbacks>({\n        namespace: [\"run\"],\n        creator: runContext,\n        context: { internal: true },\n      });\n\n      const finishEvent = {} as InferCallbackValue<RunContextCallbacks[\"finish\"]>;\n      try {\n        const startEvent: InferCallbackValue<RunContextCallbacks[\"start\"]> = {\n          input: runContext.runParams,\n          output: undefined,\n        };\n        await emitter.emit(\"start\", startEvent);\n\n        // Copy back any modifications made by middleware to run_params\n        runContext.runParams = startEvent.input;\n        finishEvent.input = startEvent.input;\n\n        const result: R2 = await Promise.race([\n          RunContext.#storage.run(\n            runContext,\n            startEvent.output === undefined ? fn : async () => startEvent.output,\n            runContext,\n          ),\n          new Promise<never>((_, reject) =>\n            runContext.signal.addEventListener(\"abort\", () =>\n              setTimeout(() => reject(runContext.signal.reason), 0),\n            ),\n          ),\n        ]);\n        finishEvent.output = result;\n        await emitter.emit(\"success\", result);\n        return result;\n      } catch (_e) {\n        const e = FrameworkError.ensure(_e);\n        finishEvent.output = e;\n        await emitter.emit(\"error\", e);\n        throw e;\n      } finally {\n        await emitter.emit(\"finish\", finishEvent);\n        runContext.destroy();\n      }\n    }, runContext);\n  }\n\n  static {\n    this.register();\n  }\n\n  createSnapshot() {\n    return {\n      controller: this.controller,\n      runId: this.runId,\n      groupId: this.groupId,\n      parentId: this.parentId,\n      emitter: this.emitter,\n      context: shallowCopy(this.context),\n      runParams: shallowCopy(this.runParams),\n      createdAt: new Date(this.createdAt),\n    };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n  }\n}\n"]}

package/dist/context.d.cts

@@ -1,6 +1,7 @@
-import { E as Emitter, C as Callback } from './emitter-DpqUYjXH.cjs';
+import { E as Emitter, C as Callback } from './emitter-jN55XZZq.cjs';
 import { Serializable } from './internals/serializable.cjs';
 import { LazyPromise } from './internals/helpers/promise.cjs';
+import { FrameworkError } from './errors.cjs';
 import './internals/types.cjs';
 import './internals/helpers/guards.cjs';
 
@@ -13,13 +14,25 @@ interface RunInstance<T = any> {
     emitter: Emitter<T>;
 }
 interface RunContextCallbacks {
-    start: Callback<null>;
+    start: Callback<{
+        input: any;
+        output: any;
+    }>;
     success: Callback<unknown>;
     error: Callback<Error>;
-    finish: Callback<null>;
+    finish: Callback<{
+        input: any;
+        output?: any;
+        error?: FrameworkError;
+    }>;
+}
+declare abstract class Middleware<T extends RunInstance = any> {
+    abstract bind(ctx: RunContext<T>): void;
 }
 type GetRunContext<T, P = any> = T extends RunInstance ? RunContext<T, P> : never;
 type GetRunInstance<T> = T extends RunInstance<infer P> ? P : never;
+type MiddlewareFn<T> = (context: GetRunContext<T>) => void;
+type MiddlewareType<T extends RunInstance> = Middleware<T> | MiddlewareFn<T>;
 declare class Run<R, I extends RunInstance, P = any> extends LazyPromise<R> {
     protected readonly runContext: GetRunContext<I, P>;
     protected readonly tasks: (() => Promise<void>)[];
@@ -27,7 +40,7 @@ declare class Run<R, I extends RunInstance, P = any> extends LazyPromise<R> {
     readonly [Symbol.toStringTag] = "Promise";
     observe(fn: (emitter: Emitter<GetRunInstance<I>>) => void): this;
     context(value: object): this;
-    middleware(fn: (context: GetRunContext<I, P>) => void): this;
+    middleware(...fns: MiddlewareType<I>[]): this;
     protected before(): Promise<void>;
     [Symbol.asyncIterator](): R extends AsyncIterable<infer X> ? AsyncIterator<X> : never;
 }
@@ -45,7 +58,7 @@ declare class RunContext<T extends RunInstance, P = any> extends Serializable {
     readonly parentId?: string;
     readonly emitter: Emitter<GetRunInstance<T>>;
     readonly context: object;
-    readonly runParams: P;
+    runParams: P;
     readonly createdAt: Date;
     get signal(): AbortSignal;
     abort(reason?: Error): void;
@@ -65,4 +78,4 @@ declare class RunContext<T extends RunInstance, P = any> extends Serializable {
     loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>): void;
 }
 
-export { type GetRunContext, type GetRunInstance, Run, RunContext, type RunContextCallbacks, type RunContextInput, type RunInstance };
+export { type GetRunContext, type GetRunInstance, Middleware, type MiddlewareFn, type MiddlewareType, Run, RunContext, type RunContextCallbacks, type RunContextInput, type RunInstance };

package/dist/context.d.ts

@@ -1,6 +1,7 @@
-import { E as Emitter, C as Callback } from './emitter-CZFbzlUi.js';
+import { E as Emitter, C as Callback } from './emitter-36-9MnvA.js';
 import { Serializable } from './internals/serializable.js';
 import { LazyPromise } from './internals/helpers/promise.js';
+import { FrameworkError } from './errors.js';
 import './internals/types.js';
 import './internals/helpers/guards.js';
 
@@ -13,13 +14,25 @@ interface RunInstance<T = any> {
     emitter: Emitter<T>;
 }
 interface RunContextCallbacks {
-    start: Callback<null>;
+    start: Callback<{
+        input: any;
+        output: any;
+    }>;
     success: Callback<unknown>;
     error: Callback<Error>;
-    finish: Callback<null>;
+    finish: Callback<{
+        input: any;
+        output?: any;
+        error?: FrameworkError;
+    }>;
+}
+declare abstract class Middleware<T extends RunInstance = any> {
+    abstract bind(ctx: RunContext<T>): void;
 }
 type GetRunContext<T, P = any> = T extends RunInstance ? RunContext<T, P> : never;
 type GetRunInstance<T> = T extends RunInstance<infer P> ? P : never;
+type MiddlewareFn<T> = (context: GetRunContext<T>) => void;
+type MiddlewareType<T extends RunInstance> = Middleware<T> | MiddlewareFn<T>;
 declare class Run<R, I extends RunInstance, P = any> extends LazyPromise<R> {
     protected readonly runContext: GetRunContext<I, P>;
     protected readonly tasks: (() => Promise<void>)[];
@@ -27,7 +40,7 @@ declare class Run<R, I extends RunInstance, P = any> extends LazyPromise<R> {
     readonly [Symbol.toStringTag] = "Promise";
     observe(fn: (emitter: Emitter<GetRunInstance<I>>) => void): this;
     context(value: object): this;
-    middleware(fn: (context: GetRunContext<I, P>) => void): this;
+    middleware(...fns: MiddlewareType<I>[]): this;
     protected before(): Promise<void>;
     [Symbol.asyncIterator](): R extends AsyncIterable<infer X> ? AsyncIterator<X> : never;
 }
@@ -45,7 +58,7 @@ declare class RunContext<T extends RunInstance, P = any> extends Serializable {
     readonly parentId?: string;
     readonly emitter: Emitter<GetRunInstance<T>>;
     readonly context: object;
-    readonly runParams: P;
+    runParams: P;
     readonly createdAt: Date;
     get signal(): AbortSignal;
     abort(reason?: Error): void;
@@ -65,4 +78,4 @@ declare class RunContext<T extends RunInstance, P = any> extends Serializable {
     loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>): void;
 }
 
-export { type GetRunContext, type GetRunInstance, Run, RunContext, type RunContextCallbacks, type RunContextInput, type RunInstance };
+export { type GetRunContext, type GetRunInstance, Middleware, type MiddlewareFn, type MiddlewareType, Run, RunContext, type RunContextCallbacks, type RunContextInput, type RunInstance };

package/dist/context.js

@@ -1,6 +1,6 @@
 import { AsyncLocalStorage } from 'node:async_hooks';
 import { createRandomHash } from './internals/helpers/hash.js';
-import { omit } from 'remeda';
+import { isFunction, omit } from 'remeda';
 import { registerSignals } from './internals/helpers/cancellation.js';
 import { Serializable } from './internals/serializable.js';
 import { LazyPromise, executeSequentially } from './internals/helpers/promise.js';
@@ -10,6 +10,11 @@ import { isAsyncIterable } from './internals/helpers/stream.js';
 
 var __defProp = Object.defineProperty;
 var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+class Middleware {
+  static {
+    __name(this, "Middleware");
+  }
+}
 class Run extends LazyPromise {
   static {
     __name(this, "Run");
@@ -31,8 +36,10 @@ class Run extends LazyPromise {
     });
     return this;
   }
-  middleware(fn) {
-    this.tasks.push(async () => fn(this.runContext));
+  middleware(...fns) {
+    for (const fn of fns) {
+      this.tasks.push(async () => isFunction(fn) ? fn(this.runContext) : fn.bind(this.runContext));
+    }
     return this;
   }
   async before() {
@@ -115,20 +122,29 @@ class RunContext extends Serializable {
           internal: true
         }
       });
+      const finishEvent = {};
       try {
-        await emitter.emit("start", null);
+        const startEvent = {
+          input: runContext.runParams,
+          output: void 0
+        };
+        await emitter.emit("start", startEvent);
+        runContext.runParams = startEvent.input;
+        finishEvent.input = startEvent.input;
         const result = await Promise.race([
-          RunContext.#storage.run(runContext, fn, runContext),
+          RunContext.#storage.run(runContext, startEvent.output === void 0 ? fn : async () => startEvent.output, runContext),
           new Promise((_, reject) => runContext.signal.addEventListener("abort", () => setTimeout(() => reject(runContext.signal.reason), 0)))
         ]);
+        finishEvent.output = result;
         await emitter.emit("success", result);
         return result;
       } catch (_e) {
         const e = FrameworkError.ensure(_e);
+        finishEvent.output = e;
         await emitter.emit("error", e);
         throw e;
       } finally {
-        await emitter.emit("finish", null);
+        await emitter.emit("finish", finishEvent);
         runContext.destroy();
       }
     }, runContext);
@@ -153,6 +169,6 @@ class RunContext extends Serializable {
   }
 }
 
-export { Run, RunContext };
+export { Middleware, Run, RunContext };
 //# sourceMappingURL=context.js.map
 //# sourceMappingURL=context.js.map

package/dist/context.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/context.ts"],"names":["Run","LazyPromise","tasks","handler","runContext","Symbol","toStringTag","observe","fn","push","emitter","context","value","Object","assign","middleware","before","executeSequentially","splice","Infinity","asyncIterator","response","then","isAsyncIterable","Error","RunContext","Serializable","AsyncLocalStorage","controller","runId","groupId","parentId","runParams","createdAt","signal","abort","reason","instance","input","parent","Date","params","createRandomHash","omit","AbortController","registerSignals","child","trace","id","parentRunId","pipe","destroy","FrameworkError","enter","getStore","namespace","creator","internal","emit","result","Promise","race","run","_","reject","addEventListener","setTimeout","_e","e","ensure","register","createSnapshot","shallowCopy","loadSnapshot","snapshot"],"mappings":";;;;;;;;;;;;AA+BO,MAAMA,YAA+CC,WAAAA,CAAAA;EA/B5D;;;;AAgCqBC,EAAAA,KAAAA,GAAiC,EAAA;AAEpD,EAAA,WAAA,CACEC,SACmBC,UAAAA,EACnB;AACA,IAAA,KAAA,CAAMD,OAAAA,CAAAA,EAAAA,IAAAA,CAFaC,UAAAA,GAAAA,UAAAA;AAGrB;EAES,CAACC,MAAAA,CAAOC,WAAW,IAAI,SAAA;AAEhCC,EAAAA,OAAAA,CAAQC,EAAAA,EAAmD;AACzD,IAAA,IAAA,CAAKN,MAAMO,IAAAA,CAAK,YAAYD,GAAG,IAAA,CAAKJ,UAAAA,CAAWM,OAAO,CAAA,CAAA;AACtD,IAAA,OAAO,IAAA;AACT;AAEAC,EAAAA,OAAAA,CAAQC,KAAAA,EAAe;AACrB,IAAA,IAAA,CAAKV,KAAAA,CAAMO,KAAK,YAAA;AACdI,MAAAA,MAAAA,CAAOC,MAAAA,CAAO,IAAA,CAAKV,UAAAA,CAAWO,OAAAA,EAASC,KAAAA,CAAAA;AACvCC,MAAAA,MAAAA,CAAOC,MAAAA,CAAO,IAAA,CAAKV,UAAAA,CAAWM,OAAAA,CAAQC,SAASC,KAAAA,CAAAA;KACjD,CAAA;AACA,IAAA,OAAO,IAAA;AACT;AAEAG,EAAAA,UAAAA,CAAWP,EAAAA,EAA4C;AACrD,IAAA,IAAA,CAAKN,MAAMO,IAAAA,CAAK,YAAYD,EAAAA,CAAG,IAAA,CAAKJ,UAAU,CAAA,CAAA;AAC9C,IAAA,OAAO,IAAA;AACT;AAEA,EAAA,MAAgBY,MAAAA,GAAwB;AACtC,IAAA,MAAM,MAAMA,MAAAA,EAAAA;AACZ,IAAA,MAAMC,oBAAoB,IAAA,CAAKf,KAAAA,CAAMgB,MAAAA,CAAO,CAAA,EAAGC,QAAAA,CAAAA,CAAAA;AACjD;;EAGA,QAAQd,MAAAA,CAAOe,aAAa,CAAA,GAAiE;AAC3F,IAAA,MAAMC,QAAAA,GAAW,MAAM,IAAA,CAAKC,IAAAA,EAAI;AAChC,IAAA,IAAIC,eAAAA,CAAgBF,QAAAA,CAAAA,EAAW;AAC7B,MAAA,OAAOA,QAAAA;KACT,MAAO;AACL,MAAA,MAAM,IAAIG,MAAM,yBAAA,CAAA;AAClB;AACF;AACF;AAOO,MAAMC,mBAAmDC,YAAAA,CAAAA;EAlFhE;;;;;EAmFE,OAAO,QAAA,GAAW,IAAIC,iBAAAA,EAAAA;AAEHC,EAAAA,UAAAA;AACHC,EAAAA,KAAAA;AACAC,EAAAA,OAAAA;AACAC,EAAAA,QAAAA;AACArB,EAAAA,OAAAA;AACAC,EAAAA,OAAAA;AACAqB,EAAAA,SAAAA;AACAC,EAAAA,SAAAA;AAEhB,EAAA,IAAIC,MAAAA,GAAS;AACX,IAAA,OAAO,KAAKN,UAAAA,CAAWM,MAAAA;AACzB;AAEAC,EAAAA,KAAAA,CAAMC,MAAAA,EAAgB;AACpB,IAAA,IAAA,CAAKR,UAAAA,CAAWO,MAAMC,MAAAA,CAAAA;AACxB;EAEA,WAAA,CACkBC,QAAAA,EACGC,OACnBC,MAAAA,EACA;AACA,IAAA,KAAA,EAAK,EAAA,IAAA,CAJWF,QAAAA,GAAAA,QAAAA,EAAAA,KACGC,KAAAA,GAAAA,KAAAA;AAInB,IAAA,IAAA,CAAKL,SAAAA,uBAAgBO,IAAAA,EAAAA;AACrB,IAAA,IAAA,CAAKR,YAAYM,KAAAA,CAAMG,MAAAA;AACvB,IAAA,IAAA,CAAKZ,KAAAA,GAAQa,iBAAiB,CAAA,CAAA;AAC9B,IAAA,IAAA,CAAKX,WAAWQ,MAAAA,EAAQV,KAAAA;AACxB,IAAA,IAAA,CAAKC,OAAAA,GAAUS,MAAAA,EAAQT,OAAAA,IAAWY,gBAAAA,EAAAA;AAClC,IAAA,IAAA,CAAK/B,OAAAA,GAAUgC,IAAAA,CAAMJ,MAAAA,EAAQ5B,OAAAA,IAAW,EAAC,EAAW;AAAC,MAAA,IAAA;AAAM,MAAA;AAAW,KAAA,CAAA;AAEtE,IAAA,IAAA,CAAKiB,UAAAA,GAAa,IAAIgB,eAAAA,EAAAA;AACtBC,IAAAA,eAAAA,CAAgB,KAAKjB,UAAAA,EAAY;MAACU,KAAAA,CAAMJ,MAAAA;MAAQK,MAAAA,EAAQL;AAAO,KAAA,CAAA;AAE/D,IAAA,IAAA,CAAKxB,OAAAA,GAAU2B,QAAAA,CAAS3B,OAAAA,CAAQoC,KAAAA,CAAyB;AACvDnC,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;MACdoC,KAAAA,EAAO;AACLC,QAAAA,EAAAA,EAAI,IAAA,CAAKlB,OAAAA;AACTD,QAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AACZoB,QAAAA,WAAAA,EAAaV,MAAAA,EAAQV;AACvB;KACF,CAAA;AACA,IAAA,IAAIU,MAAAA,EAAQ;AACV,MAAA,IAAA,CAAK7B,OAAAA,CAAQwC,IAAAA,CAAKX,MAAAA,CAAO7B,OAAO,CAAA;AAClC;AACF;EAEAyC,OAAAA,GAAU;AACR,IAAA,IAAA,CAAKzC,QAAQyC,OAAAA,EAAO;AACpB,IAAA,IAAA,CAAKvB,UAAAA,CAAWO,KAAAA,CAAM,IAAIiB,cAAAA,CAAe,oBAAA,CAAA,CAAA;AAC3C;EAEA,OAAOC,KAAAA,CACLhB,QAAAA,EACAC,KAAAA,EACA9B,EAAAA,EACA;AACA,IAAA,MAAM+B,MAAAA,GAASd,UAAAA,CAAW,QAAA,CAAS6B,QAAAA,EAAQ;AAC3C,IAAA,MAAMlD,UAAAA,GAAa,IAAIqB,UAAAA,CAAWY,QAAAA,EAAUC,OAAOC,MAAAA,CAAAA;AAEnD,IAAA,OAAO,IAAIvC,IAAI,YAAA;AACb,MAAA,MAAMU,OAAAA,GAAUN,UAAAA,CAAWM,OAAAA,CAAQoC,KAAAA,CAA2B;QAC5DS,SAAAA,EAAW;AAAC,UAAA;;QACZC,OAAAA,EAASpD,UAAAA;QACTO,OAAAA,EAAS;UAAE8C,QAAAA,EAAU;AAAK;OAC5B,CAAA;AAEA,MAAA,IAAI;AACF,QAAA,MAAM/C,OAAAA,CAAQgD,IAAAA,CAAK,OAAA,EAAS,IAAA,CAAA;AAC5B,QAAA,MAAMC,MAAAA,GAAS,MAAMC,OAAAA,CAAQC,IAAAA,CAAK;AAChCpC,UAAAA,UAAAA,CAAW,QAAA,CAASqC,GAAAA,CAAI1D,UAAAA,EAAYI,EAAAA,EAAIJ,UAAAA,CAAAA;AACxC,UAAA,IAAIwD,QAAe,CAACG,CAAAA,EAAGC,WACrB5D,UAAAA,CAAW8B,MAAAA,CAAO+B,iBAAiB,OAAA,EAAS,MAC1CC,UAAAA,CAAW,MAAMF,OAAO5D,UAAAA,CAAW8B,MAAAA,CAAOE,MAAM,CAAA,EAAG,CAAA,CAAA,CAAA;AAGxD,SAAA,CAAA;AACD,QAAA,MAAM1B,OAAAA,CAAQgD,IAAAA,CAAK,SAAA,EAAWC,MAAAA,CAAAA;AAC9B,QAAA,OAAOA,MAAAA;AACT,OAAA,CAAA,OAASQ,EAAAA,EAAI;AACX,QAAA,MAAMC,CAAAA,GAAIhB,cAAAA,CAAeiB,MAAAA,CAAOF,EAAAA,CAAAA;AAChC,QAAA,MAAMzD,OAAAA,CAAQgD,IAAAA,CAAK,OAAA,EAASU,CAAAA,CAAAA;AAC5B,QAAA,MAAMA,CAAAA;OACR,SAAA;AACE,QAAA,MAAM1D,OAAAA,CAAQgD,IAAAA,CAAK,QAAA,EAAU,IAAA,CAAA;AAC7BtD,QAAAA,UAAAA,CAAW+C,OAAAA,EAAO;AACpB;AACF,KAAA,EAAG/C,UAAAA,CAAAA;AACL;EAEA;AACE,IAAA,IAAA,CAAKkE,QAAAA,EAAQ;AACf;EAEAC,cAAAA,GAAiB;AACf,IAAA,OAAO;AACL3C,MAAAA,UAAAA,EAAY,IAAA,CAAKA,UAAAA;AACjBC,MAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AACZC,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;AACdC,MAAAA,QAAAA,EAAU,IAAA,CAAKA,QAAAA;AACfrB,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;MACdC,OAAAA,EAAS6D,WAAAA,CAAY,KAAK7D,OAAO,CAAA;MACjCqB,SAAAA,EAAWwC,WAAAA,CAAY,KAAKxC,SAAS,CAAA;MACrCC,SAAAA,EAAW,IAAIO,IAAAA,CAAK,IAAA,CAAKP,SAAS;AACpC,KAAA;AACF;AAEAwC,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7D7D,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAM4D,QAAAA,CAAAA;AACtB;AACF","file":"context.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { AsyncLocalStorage } from \"node:async_hooks\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { createRandomHash } from \"@/internals/helpers/hash.js\";\nimport { omit } from \"remeda\";\nimport { Callback } from \"@/emitter/types.js\";\nimport { registerSignals } from \"@/internals/helpers/cancellation.js\";\nimport { Serializable } from \"@/internals/serializable.js\";\nimport { executeSequentially, LazyPromise } from \"@/internals/helpers/promise.js\";\nimport { FrameworkError } from \"@/errors.js\";\nimport { shallowCopy } from \"@/serializer/utils.js\";\nimport { isAsyncIterable } from \"@/internals/helpers/stream.js\";\n\nexport interface RunInstance<T = any> {\n  emitter: Emitter<T>;\n}\n\nexport interface RunContextCallbacks {\n  start: Callback<null>;\n  success: Callback<unknown>;\n  error: Callback<Error>;\n  finish: Callback<null>;\n}\n\nexport type GetRunContext<T, P = any> = T extends RunInstance ? RunContext<T, P> : never;\nexport type GetRunInstance<T> = T extends RunInstance<infer P> ? P : never;\n\nexport class Run<R, I extends RunInstance, P = any> extends LazyPromise<R> {\n  protected readonly tasks: (() => Promise<void>)[] = [];\n\n  constructor(\n    handler: () => Promise<R>,\n    protected readonly runContext: GetRunContext<I, P>,\n  ) {\n    super(handler);\n  }\n\n  readonly [Symbol.toStringTag] = \"Promise\";\n\n  observe(fn: (emitter: Emitter<GetRunInstance<I>>) => void) {\n    this.tasks.push(async () => fn(this.runContext.emitter));\n    return this;\n  }\n\n  context(value: object) {\n    this.tasks.push(async () => {\n      Object.assign(this.runContext.context, value);\n      Object.assign(this.runContext.emitter.context, value);\n    });\n    return this;\n  }\n\n  middleware(fn: (context: GetRunContext<I, P>) => void) {\n    this.tasks.push(async () => fn(this.runContext));\n    return this;\n  }\n\n  protected async before(): Promise<void> {\n    await super.before();\n    await executeSequentially(this.tasks.splice(0, Infinity));\n  }\n\n  // @ts-expect-error too complex\n  async *[Symbol.asyncIterator](): R extends AsyncIterable<infer X> ? AsyncIterator<X> : never {\n    const response = await this.then();\n    if (isAsyncIterable(response)) {\n      yield* response;\n    } else {\n      throw new Error(\"Result is not iterable!\");\n    }\n  }\n}\n\nexport interface RunContextInput<P> {\n  params: P;\n  signal?: AbortSignal;\n}\n\nexport class RunContext<T extends RunInstance, P = any> extends Serializable {\n  static #storage = new AsyncLocalStorage<RunContext<any>>();\n\n  protected readonly controller: AbortController;\n  public readonly runId: string;\n  public readonly groupId: string;\n  public readonly parentId?: string;\n  public readonly emitter;\n  public readonly context: object;\n  public readonly runParams: P;\n  public readonly createdAt: Date;\n\n  get signal() {\n    return this.controller.signal;\n  }\n\n  abort(reason?: Error) {\n    this.controller.abort(reason);\n  }\n\n  constructor(\n    public readonly instance: T,\n    protected readonly input: RunContextInput<P>,\n    parent?: RunContext<any>,\n  ) {\n    super();\n    this.createdAt = new Date();\n    this.runParams = input.params;\n    this.runId = createRandomHash(5);\n    this.parentId = parent?.runId;\n    this.groupId = parent?.groupId ?? createRandomHash();\n    this.context = omit((parent?.context ?? {}) as any, [\"id\", \"parentId\"]);\n\n    this.controller = new AbortController();\n    registerSignals(this.controller, [input.signal, parent?.signal]);\n\n    this.emitter = instance.emitter.child<GetRunInstance<T>>({\n      context: this.context,\n      trace: {\n        id: this.groupId,\n        runId: this.runId,\n        parentRunId: parent?.runId,\n      },\n    });\n    if (parent) {\n      this.emitter.pipe(parent.emitter);\n    }\n  }\n\n  destroy() {\n    this.emitter.destroy();\n    this.controller.abort(new FrameworkError(\"Context destroyed.\"));\n  }\n\n  static enter<C2 extends RunInstance, R2, P2>(\n    instance: C2,\n    input: RunContextInput<P2>,\n    fn: (context: GetRunContext<C2, P2>) => Promise<R2>,\n  ) {\n    const parent = RunContext.#storage.getStore();\n    const runContext = new RunContext(instance, input, parent) as GetRunContext<C2, P2>;\n\n    return new Run(async () => {\n      const emitter = runContext.emitter.child<RunContextCallbacks>({\n        namespace: [\"run\"],\n        creator: runContext,\n        context: { internal: true },\n      });\n\n      try {\n        await emitter.emit(\"start\", null);\n        const result = await Promise.race([\n          RunContext.#storage.run(runContext, fn, runContext),\n          new Promise<never>((_, reject) =>\n            runContext.signal.addEventListener(\"abort\", () =>\n              setTimeout(() => reject(runContext.signal.reason), 0),\n            ),\n          ),\n        ]);\n        await emitter.emit(\"success\", result);\n        return result;\n      } catch (_e) {\n        const e = FrameworkError.ensure(_e);\n        await emitter.emit(\"error\", e);\n        throw e;\n      } finally {\n        await emitter.emit(\"finish\", null);\n        runContext.destroy();\n      }\n    }, runContext);\n  }\n\n  static {\n    this.register();\n  }\n\n  createSnapshot() {\n    return {\n      controller: this.controller,\n      runId: this.runId,\n      groupId: this.groupId,\n      parentId: this.parentId,\n      emitter: this.emitter,\n      context: shallowCopy(this.context),\n      runParams: shallowCopy(this.runParams),\n      createdAt: new Date(this.createdAt),\n    };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n  }\n}\n"]}
+{"version":3,"sources":["../src/context.ts"],"names":["Middleware","Run","LazyPromise","tasks","handler","runContext","Symbol","toStringTag","observe","fn","push","emitter","context","value","Object","assign","middleware","fns","isFunction","bind","before","executeSequentially","splice","Infinity","asyncIterator","response","then","isAsyncIterable","Error","RunContext","Serializable","AsyncLocalStorage","controller","runId","groupId","parentId","runParams","createdAt","signal","abort","reason","instance","input","parent","Date","params","createRandomHash","omit","AbortController","registerSignals","child","trace","id","parentRunId","pipe","destroy","FrameworkError","enter","getStore","namespace","creator","internal","finishEvent","startEvent","output","undefined","emit","result","Promise","race","run","_","reject","addEventListener","setTimeout","_e","e","ensure","register","createSnapshot","shallowCopy","loadSnapshot","snapshot"],"mappings":";;;;;;;;;;;;AA4BO,MAAeA,UAAAA,CAAAA;EA5BtB;;;AA8BA;AAOO,MAAMC,YAA+CC,WAAAA,CAAAA;EArC5D;;;;AAsCqBC,EAAAA,KAAAA,GAAiC,EAAA;AAEpD,EAAA,WAAA,CACEC,SACmBC,UAAAA,EACnB;AACA,IAAA,KAAA,CAAMD,OAAAA,CAAAA,EAAAA,IAAAA,CAFaC,UAAAA,GAAAA,UAAAA;AAGrB;EAES,CAACC,MAAAA,CAAOC,WAAW,IAAI,SAAA;AAEhCC,EAAAA,OAAAA,CAAQC,EAAAA,EAAmD;AACzD,IAAA,IAAA,CAAKN,MAAMO,IAAAA,CAAK,YAAYD,GAAG,IAAA,CAAKJ,UAAAA,CAAWM,OAAO,CAAA,CAAA;AACtD,IAAA,OAAO,IAAA;AACT;AAEAC,EAAAA,OAAAA,CAAQC,KAAAA,EAAe;AACrB,IAAA,IAAA,CAAKV,KAAAA,CAAMO,KAAK,YAAA;AACdI,MAAAA,MAAAA,CAAOC,MAAAA,CAAO,IAAA,CAAKV,UAAAA,CAAWO,OAAAA,EAASC,KAAAA,CAAAA;AACvCC,MAAAA,MAAAA,CAAOC,MAAAA,CAAO,IAAA,CAAKV,UAAAA,CAAWM,OAAAA,CAAQC,SAASC,KAAAA,CAAAA;KACjD,CAAA;AACA,IAAA,OAAO,IAAA;AACT;AAEAG,EAAAA,UAAAA,CAAAA,GAAcC,GAAAA,EAA0B;AACtC,IAAA,KAAA,MAAWR,MAAMQ,GAAAA,EAAK;AACpB,MAAA,IAAA,CAAKd,KAAAA,CAAMO,IAAAA,CAAK,YACdQ,UAAAA,CAAWT,EAAAA,CAAAA,GAAMA,EAAAA,CAAG,IAAA,CAAKJ,UAAU,CAAA,GAAII,EAAAA,CAAGU,IAAAA,CAAK,IAAA,CAAKd,UAAU,CAAA,CAAA;AAElE;AACA,IAAA,OAAO,IAAA;AACT;AAEA,EAAA,MAAgBe,MAAAA,GAAwB;AACtC,IAAA,MAAM,MAAMA,MAAAA,EAAAA;AACZ,IAAA,MAAMC,oBAAoB,IAAA,CAAKlB,KAAAA,CAAMmB,MAAAA,CAAO,CAAA,EAAGC,QAAAA,CAAAA,CAAAA;AACjD;;EAGA,QAAQjB,MAAAA,CAAOkB,aAAa,CAAA,GAAiE;AAC3F,IAAA,MAAMC,QAAAA,GAAW,MAAM,IAAA,CAAKC,IAAAA,EAAI;AAChC,IAAA,IAAIC,eAAAA,CAAgBF,QAAAA,CAAAA,EAAW;AAC7B,MAAA,OAAOA,QAAAA;KACT,MAAO;AACL,MAAA,MAAM,IAAIG,MAAM,yBAAA,CAAA;AAClB;AACF;AACF;AAOO,MAAMC,mBAAmDC,YAAAA,CAAAA;EA5FhE;;;;;EA6FE,OAAO,QAAA,GAAW,IAAIC,iBAAAA,EAAAA;AAEHC,EAAAA,UAAAA;AACHC,EAAAA,KAAAA;AACAC,EAAAA,OAAAA;AACAC,EAAAA,QAAAA;AACAxB,EAAAA,OAAAA;AACAC,EAAAA,OAAAA;AACTwB,EAAAA,SAAAA;AACSC,EAAAA,SAAAA;AAEhB,EAAA,IAAIC,MAAAA,GAAS;AACX,IAAA,OAAO,KAAKN,UAAAA,CAAWM,MAAAA;AACzB;AAEAC,EAAAA,KAAAA,CAAMC,MAAAA,EAAgB;AACpB,IAAA,IAAA,CAAKR,UAAAA,CAAWO,MAAMC,MAAAA,CAAAA;AACxB;EAEA,WAAA,CACkBC,QAAAA,EACGC,OACnBC,MAAAA,EACA;AACA,IAAA,KAAA,EAAK,EAAA,IAAA,CAJWF,QAAAA,GAAAA,QAAAA,EAAAA,KACGC,KAAAA,GAAAA,KAAAA;AAInB,IAAA,IAAA,CAAKL,SAAAA,uBAAgBO,IAAAA,EAAAA;AACrB,IAAA,IAAA,CAAKR,YAAYM,KAAAA,CAAMG,MAAAA;AACvB,IAAA,IAAA,CAAKZ,KAAAA,GAAQa,iBAAiB,CAAA,CAAA;AAC9B,IAAA,IAAA,CAAKX,WAAWQ,MAAAA,EAAQV,KAAAA;AACxB,IAAA,IAAA,CAAKC,OAAAA,GAAUS,MAAAA,EAAQT,OAAAA,IAAWY,gBAAAA,EAAAA;AAClC,IAAA,IAAA,CAAKlC,OAAAA,GAAUmC,IAAAA,CAAMJ,MAAAA,EAAQ/B,OAAAA,IAAW,EAAC,EAAW;AAAC,MAAA,IAAA;AAAM,MAAA;AAAW,KAAA,CAAA;AAEtE,IAAA,IAAA,CAAKoB,UAAAA,GAAa,IAAIgB,eAAAA,EAAAA;AACtBC,IAAAA,eAAAA,CAAgB,KAAKjB,UAAAA,EAAY;MAACU,KAAAA,CAAMJ,MAAAA;MAAQK,MAAAA,EAAQL;AAAO,KAAA,CAAA;AAE/D,IAAA,IAAA,CAAK3B,OAAAA,GAAU8B,QAAAA,CAAS9B,OAAAA,CAAQuC,KAAAA,CAAyB;AACvDtC,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;MACduC,KAAAA,EAAO;AACLC,QAAAA,EAAAA,EAAI,IAAA,CAAKlB,OAAAA;AACTD,QAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AACZoB,QAAAA,WAAAA,EAAaV,MAAAA,EAAQV;AACvB;KACF,CAAA;AACA,IAAA,IAAIU,MAAAA,EAAQ;AACV,MAAA,IAAA,CAAKhC,OAAAA,CAAQ2C,IAAAA,CAAKX,MAAAA,CAAOhC,OAAO,CAAA;AAClC;AACF;EAEA4C,OAAAA,GAAU;AACR,IAAA,IAAA,CAAK5C,QAAQ4C,OAAAA,EAAO;AACpB,IAAA,IAAA,CAAKvB,UAAAA,CAAWO,KAAAA,CAAM,IAAIiB,cAAAA,CAAe,oBAAA,CAAA,CAAA;AAC3C;EAEA,OAAOC,KAAAA,CACLhB,QAAAA,EACAC,KAAAA,EACAjC,EAAAA,EACA;AACA,IAAA,MAAMkC,MAAAA,GAASd,UAAAA,CAAW,QAAA,CAAS6B,QAAAA,EAAQ;AAC3C,IAAA,MAAMrD,UAAAA,GAAa,IAAIwB,UAAAA,CAAWY,QAAAA,EAAUC,OAAOC,MAAAA,CAAAA;AAEnD,IAAA,OAAO,IAAI1C,IAAI,YAAA;AACb,MAAA,MAAMU,OAAAA,GAAUN,UAAAA,CAAWM,OAAAA,CAAQuC,KAAAA,CAA2B;QAC5DS,SAAAA,EAAW;AAAC,UAAA;;QACZC,OAAAA,EAASvD,UAAAA;QACTO,OAAAA,EAAS;UAAEiD,QAAAA,EAAU;AAAK;OAC5B,CAAA;AAEA,MAAA,MAAMC,cAAc,EAAC;AACrB,MAAA,IAAI;AACF,QAAA,MAAMC,UAAAA,GAA+D;AACnErB,UAAAA,KAAAA,EAAOrC,UAAAA,CAAW+B,SAAAA;UAClB4B,MAAAA,EAAQC,KAAAA;AACV,SAAA;AACA,QAAA,MAAMtD,OAAAA,CAAQuD,IAAAA,CAAK,OAAA,EAASH,UAAAA,CAAAA;AAG5B1D,QAAAA,UAAAA,CAAW+B,YAAY2B,UAAAA,CAAWrB,KAAAA;AAClCoB,QAAAA,WAAAA,CAAYpB,QAAQqB,UAAAA,CAAWrB,KAAAA;AAE/B,QAAA,MAAMyB,MAAAA,GAAa,MAAMC,OAAAA,CAAQC,IAAAA,CAAK;UACpCxC,UAAAA,CAAW,QAAA,CAASyC,GAAAA,CAClBjE,UAAAA,EACA0D,UAAAA,CAAWC,MAAAA,KAAWC,SAAYxD,EAAAA,GAAK,YAAYsD,UAAAA,CAAWC,MAAAA,EAC9D3D,UAAAA,CAAAA;AAEF,UAAA,IAAI+D,QAAe,CAACG,CAAAA,EAAGC,WACrBnE,UAAAA,CAAWiC,MAAAA,CAAOmC,iBAAiB,OAAA,EAAS,MAC1CC,UAAAA,CAAW,MAAMF,OAAOnE,UAAAA,CAAWiC,MAAAA,CAAOE,MAAM,CAAA,EAAG,CAAA,CAAA,CAAA;AAGxD,SAAA,CAAA;AACDsB,QAAAA,WAAAA,CAAYE,MAAAA,GAASG,MAAAA;AACrB,QAAA,MAAMxD,OAAAA,CAAQuD,IAAAA,CAAK,SAAA,EAAWC,MAAAA,CAAAA;AAC9B,QAAA,OAAOA,MAAAA;AACT,OAAA,CAAA,OAASQ,EAAAA,EAAI;AACX,QAAA,MAAMC,CAAAA,GAAIpB,cAAAA,CAAeqB,MAAAA,CAAOF,EAAAA,CAAAA;AAChCb,QAAAA,WAAAA,CAAYE,MAAAA,GAASY,CAAAA;AACrB,QAAA,MAAMjE,OAAAA,CAAQuD,IAAAA,CAAK,OAAA,EAASU,CAAAA,CAAAA;AAC5B,QAAA,MAAMA,CAAAA;OACR,SAAA;AACE,QAAA,MAAMjE,OAAAA,CAAQuD,IAAAA,CAAK,QAAA,EAAUJ,WAAAA,CAAAA;AAC7BzD,QAAAA,UAAAA,CAAWkD,OAAAA,EAAO;AACpB;AACF,KAAA,EAAGlD,UAAAA,CAAAA;AACL;EAEA;AACE,IAAA,IAAA,CAAKyE,QAAAA,EAAQ;AACf;EAEAC,cAAAA,GAAiB;AACf,IAAA,OAAO;AACL/C,MAAAA,UAAAA,EAAY,IAAA,CAAKA,UAAAA;AACjBC,MAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AACZC,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;AACdC,MAAAA,QAAAA,EAAU,IAAA,CAAKA,QAAAA;AACfxB,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;MACdC,OAAAA,EAASoE,WAAAA,CAAY,KAAKpE,OAAO,CAAA;MACjCwB,SAAAA,EAAW4C,WAAAA,CAAY,KAAK5C,SAAS,CAAA;MACrCC,SAAAA,EAAW,IAAIO,IAAAA,CAAK,IAAA,CAAKP,SAAS;AACpC,KAAA;AACF;AAEA4C,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7DpE,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAMmE,QAAAA,CAAAA;AACtB;AACF","file":"context.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { AsyncLocalStorage } from \"node:async_hooks\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { createRandomHash } from \"@/internals/helpers/hash.js\";\nimport { isFunction, omit } from \"remeda\";\nimport { Callback, InferCallbackValue } from \"@/emitter/types.js\";\nimport { registerSignals } from \"@/internals/helpers/cancellation.js\";\nimport { Serializable } from \"@/internals/serializable.js\";\nimport { executeSequentially, LazyPromise } from \"@/internals/helpers/promise.js\";\nimport { FrameworkError } from \"@/errors.js\";\nimport { shallowCopy } from \"@/serializer/utils.js\";\nimport { isAsyncIterable } from \"@/internals/helpers/stream.js\";\n\nexport interface RunInstance<T = any> {\n  emitter: Emitter<T>;\n}\n\nexport interface RunContextCallbacks {\n  start: Callback<{ input: any; output: any }>;\n  success: Callback<unknown>; // TODO: should be updated to Callback<{ input: any; output: any }>\n  error: Callback<Error>;\n  finish: Callback<{ input: any; output?: any; error?: FrameworkError }>;\n}\n\nexport abstract class Middleware<T extends RunInstance = any> {\n  abstract bind(ctx: RunContext<T>): void;\n}\n\nexport type GetRunContext<T, P = any> = T extends RunInstance ? RunContext<T, P> : never;\nexport type GetRunInstance<T> = T extends RunInstance<infer P> ? P : never;\nexport type MiddlewareFn<T> = (context: GetRunContext<T>) => void;\nexport type MiddlewareType<T extends RunInstance> = Middleware<T> | MiddlewareFn<T>;\n\nexport class Run<R, I extends RunInstance, P = any> extends LazyPromise<R> {\n  protected readonly tasks: (() => Promise<void>)[] = [];\n\n  constructor(\n    handler: () => Promise<R>,\n    protected readonly runContext: GetRunContext<I, P>,\n  ) {\n    super(handler);\n  }\n\n  readonly [Symbol.toStringTag] = \"Promise\";\n\n  observe(fn: (emitter: Emitter<GetRunInstance<I>>) => void) {\n    this.tasks.push(async () => fn(this.runContext.emitter));\n    return this;\n  }\n\n  context(value: object) {\n    this.tasks.push(async () => {\n      Object.assign(this.runContext.context, value);\n      Object.assign(this.runContext.emitter.context, value);\n    });\n    return this;\n  }\n\n  middleware(...fns: MiddlewareType<I>[]) {\n    for (const fn of fns) {\n      this.tasks.push(async () =>\n        isFunction(fn) ? fn(this.runContext) : fn.bind(this.runContext),\n      );\n    }\n    return this;\n  }\n\n  protected async before(): Promise<void> {\n    await super.before();\n    await executeSequentially(this.tasks.splice(0, Infinity));\n  }\n\n  // @ts-expect-error too complex\n  async *[Symbol.asyncIterator](): R extends AsyncIterable<infer X> ? AsyncIterator<X> : never {\n    const response = await this.then();\n    if (isAsyncIterable(response)) {\n      yield* response;\n    } else {\n      throw new Error(\"Result is not iterable!\");\n    }\n  }\n}\n\nexport interface RunContextInput<P> {\n  params: P;\n  signal?: AbortSignal;\n}\n\nexport class RunContext<T extends RunInstance, P = any> extends Serializable {\n  static #storage = new AsyncLocalStorage<RunContext<any>>();\n\n  protected readonly controller: AbortController;\n  public readonly runId: string;\n  public readonly groupId: string;\n  public readonly parentId?: string;\n  public readonly emitter;\n  public readonly context: object;\n  public runParams: P;\n  public readonly createdAt: Date;\n\n  get signal() {\n    return this.controller.signal;\n  }\n\n  abort(reason?: Error) {\n    this.controller.abort(reason);\n  }\n\n  constructor(\n    public readonly instance: T,\n    protected readonly input: RunContextInput<P>,\n    parent?: RunContext<any>,\n  ) {\n    super();\n    this.createdAt = new Date();\n    this.runParams = input.params;\n    this.runId = createRandomHash(5);\n    this.parentId = parent?.runId;\n    this.groupId = parent?.groupId ?? createRandomHash();\n    this.context = omit((parent?.context ?? {}) as any, [\"id\", \"parentId\"]);\n\n    this.controller = new AbortController();\n    registerSignals(this.controller, [input.signal, parent?.signal]);\n\n    this.emitter = instance.emitter.child<GetRunInstance<T>>({\n      context: this.context,\n      trace: {\n        id: this.groupId,\n        runId: this.runId,\n        parentRunId: parent?.runId,\n      },\n    });\n    if (parent) {\n      this.emitter.pipe(parent.emitter);\n    }\n  }\n\n  destroy() {\n    this.emitter.destroy();\n    this.controller.abort(new FrameworkError(\"Context destroyed.\"));\n  }\n\n  static enter<C2 extends RunInstance, R2, P2>(\n    instance: C2,\n    input: RunContextInput<P2>,\n    fn: (context: GetRunContext<C2, P2>) => Promise<R2>,\n  ) {\n    const parent = RunContext.#storage.getStore();\n    const runContext = new RunContext(instance, input, parent) as GetRunContext<C2, P2>;\n\n    return new Run(async () => {\n      const emitter = runContext.emitter.child<RunContextCallbacks>({\n        namespace: [\"run\"],\n        creator: runContext,\n        context: { internal: true },\n      });\n\n      const finishEvent = {} as InferCallbackValue<RunContextCallbacks[\"finish\"]>;\n      try {\n        const startEvent: InferCallbackValue<RunContextCallbacks[\"start\"]> = {\n          input: runContext.runParams,\n          output: undefined,\n        };\n        await emitter.emit(\"start\", startEvent);\n\n        // Copy back any modifications made by middleware to run_params\n        runContext.runParams = startEvent.input;\n        finishEvent.input = startEvent.input;\n\n        const result: R2 = await Promise.race([\n          RunContext.#storage.run(\n            runContext,\n            startEvent.output === undefined ? fn : async () => startEvent.output,\n            runContext,\n          ),\n          new Promise<never>((_, reject) =>\n            runContext.signal.addEventListener(\"abort\", () =>\n              setTimeout(() => reject(runContext.signal.reason), 0),\n            ),\n          ),\n        ]);\n        finishEvent.output = result;\n        await emitter.emit(\"success\", result);\n        return result;\n      } catch (_e) {\n        const e = FrameworkError.ensure(_e);\n        finishEvent.output = e;\n        await emitter.emit(\"error\", e);\n        throw e;\n      } finally {\n        await emitter.emit(\"finish\", finishEvent);\n        runContext.destroy();\n      }\n    }, runContext);\n  }\n\n  static {\n    this.register();\n  }\n\n  createSnapshot() {\n    return {\n      controller: this.controller,\n      runId: this.runId,\n      groupId: this.groupId,\n      parentId: this.parentId,\n      emitter: this.emitter,\n      context: shallowCopy(this.context),\n      runParams: shallowCopy(this.runParams),\n      createdAt: new Date(this.createdAt),\n    };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n  }\n}\n"]}

package/dist/emitter/emitter.cjs

@@ -7,6 +7,7 @@ var hash_cjs = require('../internals/helpers/hash.cjs');
 var utils_cjs$1 = require('../serializer/utils.cjs');
 var serializable_cjs = require('../internals/serializable.cjs');
 var remeda = require('remeda');
+var array_cjs = require('../internals/helpers/array.cjs');
 
 var __defProp = Object.defineProperty;
 var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
@@ -25,7 +26,7 @@ class Emitter extends serializable_cjs.Serializable {
   static {
     __name(this, "Emitter");
   }
-  listeners = /* @__PURE__ */ new Set();
+  listeners = [];
   groupId;
   namespace;
   creator;
@@ -84,21 +85,18 @@ class Emitter extends serializable_cjs.Serializable {
       {
         isBlocking: true,
         once: false,
-        persistent: true
+        persistent: true,
+        matchNested: false
       }
     );
   }
   destroy() {
-    this.listeners.clear();
+    this.listeners.length = 0;
     this.cleanups.forEach((child) => child());
     this.cleanups.length = 0;
   }
   reset() {
-    for (const listener of this.listeners) {
-      if (!listener.options?.persistent) {
-        this.listeners.delete(listener);
-      }
-    }
+    this.listeners = this.listeners.filter((listener) => listener.options?.persistent);
   }
   registerCallbacks(callbacks, options) {
     const listeners = [];
@@ -153,8 +151,16 @@ class Emitter extends serializable_cjs.Serializable {
       raw: matcher,
       match: createMatcher()
     };
-    this.listeners.add(listener);
-    return () => this.listeners.delete(listener);
+    const priority = options?.priority ?? 0;
+    const insertIndex = this.listeners.findIndex((l) => (l.options?.priority ?? 0) >= priority);
+    if (insertIndex === -1) {
+      this.listeners.push(listener);
+    } else {
+      this.listeners.splice(insertIndex, 0, listener);
+    }
+    return () => {
+      array_cjs.removeFromArray(this.listeners, listener);
+    };
   }
   async emit(name, value) {
     utils_cjs.assertValidName(name);
@@ -163,12 +169,13 @@ class Emitter extends serializable_cjs.Serializable {
   }
   async invoke(data, event) {
     const executions = [];
-    for (const listener of this.listeners.values()) {
+    for (let i = this.listeners.length - 1; i >= 0; i--) {
+      const listener = this.listeners[i];
       if (!listener.match(event)) {
         continue;
       }
       if (listener.options?.once) {
-        this.listeners.delete(listener);
+        this.listeners.splice(i, 1);
       }
       const run = /* @__PURE__ */ __name(async () => listener.callback(data, event), "run");
       executions.push(listener.options?.isBlocking ? await run() : run());
@@ -196,7 +203,7 @@ class Emitter extends serializable_cjs.Serializable {
       creator: this.creator,
       context: this.context,
       trace: this.trace,
-      listeners: Array.from(this.listeners).map(remeda.pick([
+      listeners: this.listeners.map(remeda.pick([
         "raw",
         "options",
         "callback"
@@ -206,7 +213,7 @@ class Emitter extends serializable_cjs.Serializable {
   }
   loadSnapshot({ listeners, ...snapshot }) {
     Object.assign(this, snapshot, {
-      listeners: /* @__PURE__ */ new Set()
+      listeners: []
     });
     listeners.forEach(({ raw, callback, options }) => this.match(raw, callback, options));
   }

package/dist/emitter/emitter.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/emitter/emitter.ts"],"names":["Emitter","Serializable","listeners","Set","groupId","namespace","creator","context","trace","cleanups","input","Object","prototype","assertValidNamespace","register","root","create","child","slice","cleanup","pipe","push","target","on","toBoundedFunction","args","invoke","value","name","isBlocking","once","persistent","destroy","clear","forEach","length","reset","listener","options","delete","registerCallbacks","callbacks","entries","key","event","callback","match","matcher","createMatcher","matchers","matchNested","path","createFullPath","RegExp","test","isPath","EmitterError","matchSameRun","runId","unshift","every","raw","add","emit","assertValidName","createEvent","data","executions","values","run","Promise","all","id","createRandomHash","createdAt","Date","source","assign","shallowCopy","createSnapshot","Array","from","map","pick","loadSnapshot","snapshot"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;AA+CO,MAAMA,gBAA0DC,6BAAAA,CAAAA;AAAAA,EAAAA;;;AAC3DC,EAAAA,SAAAA,uBAAgBC,GAAAA,EAAAA;AACPC,EAAAA,OAAAA;AACHC,EAAAA,SAAAA;AACAC,EAAAA,OAAAA;AACAC,EAAAA,OAAAA;AACAC,EAAAA,KAAAA;AACGC,EAAAA,QAAAA,GAAwB,EAAA;EAE3C,WAAA,CAAYC,KAAAA,GAAsB,EAAC,EAAG;AACpC,IAAA,KAAA,EAAK;AACL,IAAA,IAAA,CAAKN,UAAUM,KAAAA,EAAON,OAAAA;AACtB,IAAA,IAAA,CAAKC,SAAAA,GAAYK,KAAAA,EAAOL,SAAAA,IAAa,EAAA;AACrC,IAAA,IAAA,CAAKC,OAAAA,GAAUI,KAAAA,CAAMJ,OAAAA,IAAWK,MAAAA,CAAOC,SAAAA;AACvC,IAAA,IAAA,CAAKL,OAAAA,GAAUG,KAAAA,EAAOH,OAAAA,IAAW,EAAC;AAClC,IAAA,IAAA,CAAKC,QAAQE,KAAAA,CAAMF,KAAAA;AACnBK,IAAAA,8BAAAA,CAAqB,KAAKR,SAAS,CAAA;AACrC;EAEA;AACE,IAAA,IAAA,CAAKS,QAAAA,EAAQ;AACf;AAEA,EAAA,WACWC,IAAAA,GAAO;AAChB,IAAA,OAAO,IAAIf,OAAAA,CAAuC;MAAEM,OAAAA,kBAASK,MAAAA,CAAOK,OAAO,IAAA;KAAM,CAAA;AACnF;EAEAC,KAAAA,CAAaP,KAAAA,GAA2B,EAAC,EAAe;AACtD,IAAA,MAAMO,KAAAA,GAAQ,IAAIjB,OAAAA,CAAW;MAC3BQ,KAAAA,EAAOE,KAAAA,CAAMF,SAAS,IAAA,CAAKA,KAAAA;MAC3BJ,OAAAA,EAASM,KAAAA,EAAON,WAAW,IAAA,CAAKA,OAAAA;MAChCG,OAAAA,EAAS;AAAE,QAAA,GAAG,IAAA,CAAKA,OAAAA;AAAS,QAAA,GAAGG,KAAAA,EAAOH;AAAQ,OAAA;MAC9CD,OAAAA,EAASI,KAAAA,EAAOJ,WAAW,IAAA,CAAKA,OAAAA;AAChCD,MAAAA,SAAAA,EAAWK,OAAOL,SAAAA,GACd;WAAI,IAAA,CAAKA,SAAAA;WAAcK,KAAAA,CAAML;AAC7B,OAAA,GAAA,IAAA,CAAKA,UAAUa,KAAAA;KACrB,CAAA;AAEA,IAAA,MAAMC,OAAAA,GAAUF,KAAAA,CAAMG,IAAAA,CAAK,IAAI,CAAA;AAC/B,IAAA,IAAA,CAAKX,QAAAA,CAASY,KAAKF,OAAAA,CAAAA;AAEnB,IAAA,OAAOF,KAAAA;AACT;AAEAG,EAAAA,IAAAA,CAAKE,MAAAA,EAAiC;AACpC,IAAA,OAAO,IAAA,CAAKC,EAAAA;;AAEV,MAAA,KAAA;AACAC,MAAAA,6BAAAA;;AAEE,QAAA,CAAA,GAAIC,IAAAA,KAASH,MAAAA,CAAOI,MAAAA,CAAM,GAAID,IAAAA,CAAAA;AAC9B,QAAA;AAAC,UAAA;YAAEE,KAAAA,EAAOL,MAAAA;YAAQM,IAAAA,EAAM;AAAS;;AAAE,OAAA;AAErC,MAAA;QACEC,UAAAA,EAAY,IAAA;QACZC,IAAAA,EAAM,KAAA;QACNC,UAAAA,EAAY;AACd;AAAA,KAAA;AAEJ;EAEAC,OAAAA,GAAU;AACR,IAAA,IAAA,CAAK9B,UAAU+B,KAAAA,EAAK;AACpB,IAAA,IAAA,CAAKxB,QAAAA,CAASyB,OAAAA,CAAQ,CAACjB,KAAAA,KAAUA,OAAAA,CAAAA;AACjC,IAAA,IAAA,CAAKR,SAAS0B,MAAAA,GAAS,CAAA;AACzB;EAEAC,KAAAA,GAAQ;AACN,IAAA,KAAA,MAAWC,QAAAA,IAAY,KAAKnC,SAAAA,EAAW;AACrC,MAAA,IAAI,CAACmC,QAAAA,CAASC,OAAAA,EAASP,UAAAA,EAAY;AACjC,QAAA,IAAA,CAAK7B,SAAAA,CAAUqC,OAAOF,QAAAA,CAAAA;AACxB;AACF;AACF;AAEAG,EAAAA,iBAAAA,CACEC,WAGAH,OAAAA,EACW;AACX,IAAA,MAAMpC,YAAyB,EAAA;AAC/BS,IAAAA,MAAAA,CAAO+B,OAAAA,CAAQD,SAAAA,CAAAA,CAAWP,OAAAA,CAAQ,CAAC,CAACS,GAAAA,EAAKhB,KAAAA,CAAAA,KAAM;AAC7C,MAAA,IAAIA,KAAAA,EAAO;AAETzB,QAAAA,SAAAA,CAAUmB,IAAAA,CAAK,KAAKE,EAAAA,CAAGoB,GAAAA,EAAKhB,OAAOW,OAAAA,GAAUK,GAAAA,CAAI,CAAA,CAAA;AACnD;KACF,CAAA;AACA,IAAA,OAAO,MAAMzC,SAAAA,CAAUgC,OAAAA,CAAQ,CAACf,OAAAA,KAAYA,SAAAA,CAAAA;AAC9C;EAEAI,EAAAA,CACEqB,KAAAA,EACAC,UACAP,OAAAA,EACW;AACX,IAAA,OAAO,IAAA,CAAKQ,KAAAA,CAAMF,KAAAA,EAAkBC,QAAAA,EAA2BP,OAAAA,CAAAA;AACjE;EAEAQ,KAAAA,CAASC,OAAAA,EAAkBF,UAAuBP,OAAAA,EAAqC;AACrF,IAAA,MAAMU,gCAAgB,MAAA,CAAA,MAAA;AACpB,MAAA,MAAMC,WAAwB,EAAA;AAC9B,MAAA,IAAIC,cAAcZ,OAAAA,EAASY,WAAAA;AAE3B,MAAA,IAAIH,YAAY,GAAA,EAAK;AACnBG,QAAAA,WAAAA,KAAgB,KAAA;AAChBD,QAAAA,QAAAA,CAAS5B,IAAAA,CAAK,CAACuB,KAAAA,KAAUA,KAAAA,CAAMO,IAAAA,KAASC,yBAAe,IAAA,CAAK/C,SAAAA,EAAWuC,KAAAA,CAAMhB,IAAI,CAAA,CAAA;AACnF,OAAA,MAAA,IAAWmB,YAAY,KAAA,EAAO;AAC5BG,QAAAA,WAAAA,KAAgB,IAAA;AAChBD,QAAAA,QAAAA,CAAS5B,IAAAA,CAAK,MAAM,IAAA,CAAA;AACtB,OAAA,MAAA,IAAW0B,mBAAmBM,MAAAA,EAAQ;AACpCH,QAAAA,WAAAA,KAAgB,IAAA;AAChBD,QAAAA,QAAAA,CAAS5B,KAAK,CAACuB,KAAAA,KAAUG,QAAQO,IAAAA,CAAKV,KAAAA,CAAMO,IAAI,CAAA,CAAA;OAClD,MAAA,IAAW,OAAOJ,YAAY,UAAA,EAAY;AACxCG,QAAAA,WAAAA,KAAgB,KAAA;AAChBD,QAAAA,QAAAA,CAAS5B,KAAK0B,OAAAA,CAAAA;OAChB,MAAA,IAAW,OAAOA,YAAY,QAAA,EAAU;AACtC,QAAA,IAAIQ,gBAAAA,CAAOR,OAAAA,CAAAA,EAAU;AACnBG,UAAAA,WAAAA,KAAgB,IAAA;AAChBD,UAAAA,QAAAA,CAAS5B,IAAAA,CAAK,CAACuB,KAAAA,KAAUA,KAAAA,CAAMO,SAASJ,OAAAA,CAAAA;SAC1C,MAAO;AACLG,UAAAA,WAAAA,KAAgB,KAAA;AAChBD,UAAAA,QAAAA,CAAS5B,IAAAA,CACP,CAACuB,KAAAA,KACCA,KAAAA,CAAMhB,IAAAA,KAASmB,OAAAA,IAAWH,KAAAA,CAAMO,IAAAA,KAASC,wBAAAA,CAAe,IAAA,CAAK/C,SAAAA,EAAWuC,KAAAA,CAAMhB,IAAI,CAAA,CAAA;AAExF;OACF,MAAO;AACL,QAAA,MAAM,IAAI4B,wBAAa,2BAAA,CAAA;AACzB;AAEA,MAAA,IAAI,CAACN,WAAAA,EAAa;AAChB,QAAA,MAAMO,YAAAA,2BAAgBb,KAAAA,KAAAA;AAEpB,UAAA,OAAO,CAAC,KAAKpC,KAAAA,EAAOkD,KAAAA,GAAQ,OAAO,IAAA,CAAKlD,KAAAA,EAAOkD,KAAAA,KAAUd,KAAAA,CAAMpC,KAAAA,EAAOkD,KAAAA;SACxE,EAHqB,cAAA,CAAA;AAIrBT,QAAAA,QAAAA,CAASU,QAAQF,YAAAA,CAAAA;AACnB;AAEA,MAAA,OAAO,CAACb,UAAUK,QAAAA,CAASW,KAAAA,CAAM,CAACb,QAAAA,KAAYA,QAAAA,CAAQH,KAAAA,CAAAA,CAAAA;KACxD,EAxCsB,eAAA,CAAA;AA0CtB,IAAA,MAAMP,QAAAA,GAAqB;AACzBC,MAAAA,OAAAA;AACAO,MAAAA,QAAAA;MACAgB,GAAAA,EAAKd,OAAAA;AACLD,MAAAA,KAAAA,EAAOE,aAAAA;AACT,KAAA;AACA,IAAA,IAAA,CAAK9C,SAAAA,CAAU4D,IAAIzB,QAAAA,CAAAA;AAEnB,IAAA,OAAO,MAAM,IAAA,CAAKnC,SAAAA,CAAUqC,MAAAA,CAAOF,QAAAA,CAAAA;AACrC;EAEA,MAAM0B,IAAAA,CACJnC,MACAD,KAAAA,EACe;AACfqC,IAAAA,yBAAAA,CAAgBpC,IAAAA,CAAAA;AAEhB,IAAA,MAAMgB,KAAAA,GAAQ,IAAA,CAAKqB,WAAAA,CAAYrC,IAAAA,CAAAA;AAC/B,IAAA,OAAO,MAAM,IAAA,CAAKF,MAAAA,CAAOC,KAAAA,EAAOiB,KAAAA,CAAAA;AAClC;EAEA,MAAgBlB,MAAAA,CAAOwC,MAAetB,KAAAA,EAAkB;AACtD,IAAA,MAAMuB,aAAwB,EAAA;AAC9B,IAAA,KAAA,MAAW9B,QAAAA,IAAY,IAAA,CAAKnC,SAAAA,CAAUkE,MAAAA,EAAM,EAAI;AAC9C,MAAA,IAAI,CAAC/B,QAAAA,CAASS,KAAAA,CAAMF,KAAAA,CAAAA,EAAQ;AAC1B,QAAA;AACF;AAEA,MAAA,IAAIP,QAAAA,CAASC,SAASR,IAAAA,EAAM;AAC1B,QAAA,IAAA,CAAK5B,SAAAA,CAAUqC,OAAOF,QAAAA,CAAAA;AACxB;AAEA,MAAA,MAAMgC,sBAAM,MAAA,CAAA,YAAYhC,QAAAA,CAASQ,QAAAA,CAASqB,IAAAA,EAAMtB,KAAAA,CAAAA,EAApC,KAAA,CAAA;AACZuB,MAAAA,UAAAA,CAAW9C,IAAAA,CAAKgB,SAASC,OAAAA,EAAST,UAAAA,GAAa,MAAMwC,GAAAA,EAAAA,GAAQA,KAAAA,CAAAA;AAC/D;AACA,IAAA,MAAMC,OAAAA,CAAQC,IAAIJ,UAAAA,CAAAA;AACpB;AAEUF,EAAAA,WAAAA,CAAYrC,IAAAA,EAAyB;AAC7C,IAAA,OAAO;AACL4C,MAAAA,EAAAA,EAAIC,yBAAAA,EAAAA;AACJrE,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;AACdwB,MAAAA,IAAAA;MACAuB,IAAAA,EAAMC,wBAAAA,CAAe,IAAA,CAAK/C,SAAAA,EAAWuB,IAAAA,CAAAA;AACrC8C,MAAAA,SAAAA,sBAAeC,IAAAA,EAAAA;MACfC,MAAAA,EAAQ,IAAA;AACRtE,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;AACdC,MAAAA,OAAAA,EAASI,OAAOkE,MAAAA,CAAO,IAAI,IAAA,CAAKtE,OAAAA,EAAS,EAAC,CAAA;MAC1CC,KAAAA,EAAOsE,uBAAAA,CAAY,KAAKtE,KAAK;AAC/B,KAAA;AACF;;EAEAuE,cAAAA,GAAiB;AACf,IAAA,OAAO;AACL3E,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;MACdC,SAAAA,EAAWyE,uBAAAA,CAAY,KAAKzE,SAAS,CAAA;AACrCC,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;AACdC,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;AACdC,MAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AACZN,MAAAA,SAAAA,EAAW8E,MAAMC,IAAAA,CAAK,IAAA,CAAK/E,SAAS,CAAA,CAAEgF,IAAIC,WAAAA,CAAK;AAAC,QAAA,KAAA;AAAO,QAAA,SAAA;AAAW,QAAA;OAAW,CAAA,CAAA;AAC7E1E,MAAAA,QAAAA,EAAU,IAAA,CAAKA;AACjB,KAAA;AACF;AAEA2E,EAAAA,YAAAA,CAAa,EAAElF,SAAAA,EAAW,GAAGmF,QAAAA,EAAAA,EAA0D;AACrF1E,IAAAA,MAAAA,CAAOkE,MAAAA,CAAO,MAAMQ,QAAAA,EAAU;AAAEnF,MAAAA,SAAAA,sBAAeC,GAAAA;KAAM,CAAA;AACrDD,IAAAA,SAAAA,CAAUgC,OAAAA,CAAQ,CAAC,EAAE2B,GAAAA,EAAKhB,QAAAA,EAAUP,OAAAA,EAAO,KAAO,IAAA,CAAKQ,KAAAA,CAAMe,GAAAA,EAAKhB,QAAAA,EAAUP,OAAAA,CAAAA,CAAAA;AAC9E;AACF","file":"emitter.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport {\n  Callback,\n  CleanupFn,\n  EmitterOptions,\n  EventMeta,\n  EventTrace,\n  Matcher,\n  MatcherFn,\n  StringKey,\n} from \"@/emitter/types.js\";\nexport type { EventMeta, EventTrace, Callback };\nimport { Cache } from \"@/cache/decoratorCache.js\";\nimport { createFullPath, isPath, assertValidName, assertValidNamespace } from \"@/emitter/utils.js\";\nimport { EmitterError } from \"@/emitter/errors.js\";\nimport { createRandomHash } from \"@/internals/helpers/hash.js\";\nimport { shallowCopy, toBoundedFunction } from \"@/serializer/utils.js\";\nimport { RequiredAll } from \"@/internals/types.js\";\nimport { Serializable } from \"@/internals/serializable.js\";\nimport { pick } from \"remeda\";\n\nexport interface EmitterInput<E extends object = object> {\n  groupId?: string;\n  namespace?: string[];\n  creator?: object;\n  context?: E & object;\n  trace?: EventTrace;\n}\nexport interface EmitterChildInput<E extends object = object> {\n  groupId?: string;\n  namespace?: string[];\n  creator?: object;\n  context?: E & object;\n  trace?: EventTrace;\n}\n\ninterface Listener<T = any> {\n  readonly match: MatcherFn;\n  readonly raw: Matcher;\n  readonly callback: Callback<T>;\n  readonly options?: EmitterOptions;\n}\n\nexport class Emitter<T = Record<keyof any, Callback<unknown>>> extends Serializable {\n  protected listeners = new Set<Listener>();\n  protected readonly groupId?: string;\n  public readonly namespace: string[];\n  public readonly creator?: object;\n  public readonly context: object;\n  public readonly trace?: EventTrace;\n  protected readonly cleanups: CleanupFn[] = [];\n\n  constructor(input: EmitterInput = {}) {\n    super();\n    this.groupId = input?.groupId;\n    this.namespace = input?.namespace ?? [];\n    this.creator = input.creator ?? Object.prototype;\n    this.context = input?.context ?? {};\n    this.trace = input.trace;\n    assertValidNamespace(this.namespace);\n  }\n\n  static {\n    this.register();\n  }\n\n  @Cache()\n  static get root() {\n    return new Emitter<Record<string, Callback<any>>>({ creator: Object.create(null) });\n  }\n\n  child<R = T>(input: EmitterChildInput = {}): Emitter<R> {\n    const child = new Emitter<R>({\n      trace: input.trace ?? this.trace,\n      groupId: input?.groupId ?? this.groupId,\n      context: { ...this.context, ...input?.context },\n      creator: input?.creator ?? this.creator,\n      namespace: input?.namespace\n        ? [...this.namespace, ...input.namespace]\n        : this.namespace.slice(),\n    });\n\n    const cleanup = child.pipe(this);\n    this.cleanups.push(cleanup);\n\n    return child;\n  }\n\n  pipe(target: Emitter<any>): CleanupFn {\n    return this.on(\n      // @ts-expect-error\n      \"*.*\",\n      toBoundedFunction(\n        // @ts-expect-error\n        (...args) => target.invoke(...args),\n        [{ value: target, name: \"target\" }],\n      ),\n      {\n        isBlocking: true,\n        once: false,\n        persistent: true,\n      },\n    );\n  }\n\n  destroy() {\n    this.listeners.clear();\n    this.cleanups.forEach((child) => child());\n    this.cleanups.length = 0;\n  }\n\n  reset() {\n    for (const listener of this.listeners) {\n      if (!listener.options?.persistent) {\n        this.listeners.delete(listener);\n      }\n    }\n  }\n\n  registerCallbacks<K extends StringKey<RequiredAll<T>>>(\n    callbacks: Partial<\n      Record<K, NonNullable<T[K]> extends Callback<infer X> ? Callback<X> : never>\n    >,\n    options?: Partial<Record<K, EmitterOptions>>,\n  ): CleanupFn {\n    const listeners: CleanupFn[] = [];\n    Object.entries(callbacks).forEach(([key, value]) => {\n      if (value) {\n        // @ts-expect-error\n        listeners.push(this.on(key, value, options?.[key]));\n      }\n    });\n    return () => listeners.forEach((cleanup) => cleanup());\n  }\n\n  on<K extends StringKey<RequiredAll<T>>>(\n    event: K,\n    callback: NonNullable<T[K]> extends Callback<any> ? T[K] : never,\n    options?: EmitterOptions,\n  ): CleanupFn {\n    return this.match(event as Matcher, callback as Callback<any>, options);\n  }\n\n  match<L>(matcher: Matcher, callback: Callback<L>, options?: EmitterOptions): CleanupFn {\n    const createMatcher = (): MatcherFn => {\n      const matchers: MatcherFn[] = [];\n      let matchNested = options?.matchNested;\n\n      if (matcher === \"*\") {\n        matchNested ??= false;\n        matchers.push((event) => event.path === createFullPath(this.namespace, event.name));\n      } else if (matcher === \"*.*\") {\n        matchNested ??= true;\n        matchers.push(() => true);\n      } else if (matcher instanceof RegExp) {\n        matchNested ??= true;\n        matchers.push((event) => matcher.test(event.path));\n      } else if (typeof matcher === \"function\") {\n        matchNested ??= false;\n        matchers.push(matcher);\n      } else if (typeof matcher === \"string\") {\n        if (isPath(matcher)) {\n          matchNested ??= true;\n          matchers.push((event) => event.path === matcher);\n        } else {\n          matchNested ??= false;\n          matchers.push(\n            (event) =>\n              event.name === matcher && event.path === createFullPath(this.namespace, event.name),\n          );\n        }\n      } else {\n        throw new EmitterError(\"Invalid matcher provided!\");\n      }\n\n      if (!matchNested) {\n        const matchSameRun = (event: EventMeta) => {\n          // global observability x scoped observability\n          return !this.trace?.runId ? true : this.trace?.runId === event.trace?.runId;\n        };\n        matchers.unshift(matchSameRun);\n      }\n\n      return (event) => matchers.every((matcher) => matcher(event));\n    };\n\n    const listener: Listener = {\n      options,\n      callback,\n      raw: matcher,\n      match: createMatcher(),\n    };\n    this.listeners.add(listener);\n\n    return () => this.listeners.delete(listener);\n  }\n\n  async emit<K extends StringKey<RequiredAll<T>>>(\n    name: K,\n    value: NonNullable<T[K]> extends Callback<infer X> ? X : unknown,\n  ): Promise<void> {\n    assertValidName(name);\n\n    const event = this.createEvent(name);\n    return await this.invoke(value, event);\n  }\n\n  protected async invoke(data: unknown, event: EventMeta) {\n    const executions: unknown[] = [];\n    for (const listener of this.listeners.values()) {\n      if (!listener.match(event)) {\n        continue;\n      }\n\n      if (listener.options?.once) {\n        this.listeners.delete(listener);\n      }\n\n      const run = async () => listener.callback(data, event);\n      executions.push(listener.options?.isBlocking ? await run() : run());\n    }\n    await Promise.all(executions);\n  }\n\n  protected createEvent(name: string): EventMeta {\n    return {\n      id: createRandomHash(),\n      groupId: this.groupId,\n      name,\n      path: createFullPath(this.namespace, name),\n      createdAt: new Date(),\n      source: this,\n      creator: this.creator!,\n      context: Object.assign({}, this.context, {}),\n      trace: shallowCopy(this.trace), // TODO\n    };\n  }\n\n  createSnapshot() {\n    return {\n      groupId: this.groupId,\n      namespace: shallowCopy(this.namespace),\n      creator: this.creator,\n      context: this.context,\n      trace: this.trace,\n      listeners: Array.from(this.listeners).map(pick([\"raw\", \"options\", \"callback\"])),\n      cleanups: this.cleanups,\n    };\n  }\n\n  loadSnapshot({ listeners, ...snapshot }: ReturnType<typeof this.createSnapshot>): void {\n    Object.assign(this, snapshot, { listeners: new Set() });\n    listeners.forEach(({ raw, callback, options }) => this.match(raw, callback, options));\n  }\n}\n"]}
+{"version":3,"sources":["../../src/emitter/emitter.ts"],"names":["Emitter","Serializable","listeners","groupId","namespace","creator","context","trace","cleanups","input","Object","prototype","assertValidNamespace","register","root","create","child","slice","cleanup","pipe","push","target","on","toBoundedFunction","args","invoke","value","name","isBlocking","once","persistent","matchNested","destroy","length","forEach","reset","filter","listener","options","registerCallbacks","callbacks","entries","key","event","callback","match","matcher","createMatcher","matchers","path","createFullPath","RegExp","test","isPath","EmitterError","matchSameRun","runId","unshift","every","raw","priority","insertIndex","findIndex","l","splice","removeFromArray","emit","assertValidName","createEvent","data","executions","i","run","Promise","all","id","createRandomHash","createdAt","Date","source","assign","shallowCopy","createSnapshot","map","pick","loadSnapshot","snapshot"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;AAgDO,MAAMA,gBAA0DC,6BAAAA,CAAAA;AAAAA,EAAAA;;;AAC3DC,EAAAA,SAAAA,GAAwB,EAAA;AACfC,EAAAA,OAAAA;AACHC,EAAAA,SAAAA;AACAC,EAAAA,OAAAA;AACAC,EAAAA,OAAAA;AACAC,EAAAA,KAAAA;AACGC,EAAAA,QAAAA,GAAwB,EAAA;EAE3C,WAAA,CAAYC,KAAAA,GAAsB,EAAC,EAAG;AACpC,IAAA,KAAA,EAAK;AACL,IAAA,IAAA,CAAKN,UAAUM,KAAAA,EAAON,OAAAA;AACtB,IAAA,IAAA,CAAKC,SAAAA,GAAYK,KAAAA,EAAOL,SAAAA,IAAa,EAAA;AACrC,IAAA,IAAA,CAAKC,OAAAA,GAAUI,KAAAA,CAAMJ,OAAAA,IAAWK,MAAAA,CAAOC,SAAAA;AACvC,IAAA,IAAA,CAAKL,OAAAA,GAAUG,KAAAA,EAAOH,OAAAA,IAAW,EAAC;AAClC,IAAA,IAAA,CAAKC,QAAQE,KAAAA,CAAMF,KAAAA;AACnBK,IAAAA,8BAAAA,CAAqB,KAAKR,SAAS,CAAA;AACrC;EAEA;AACE,IAAA,IAAA,CAAKS,QAAAA,EAAQ;AACf;AAEA,EAAA,WACWC,IAAAA,GAAO;AAChB,IAAA,OAAO,IAAId,OAAAA,CAAuC;MAAEK,OAAAA,kBAASK,MAAAA,CAAOK,OAAO,IAAA;KAAM,CAAA;AACnF;EAEAC,KAAAA,CAAaP,KAAAA,GAA2B,EAAC,EAAe;AACtD,IAAA,MAAMO,KAAAA,GAAQ,IAAIhB,OAAAA,CAAW;MAC3BO,KAAAA,EAAOE,KAAAA,CAAMF,SAAS,IAAA,CAAKA,KAAAA;MAC3BJ,OAAAA,EAASM,KAAAA,EAAON,WAAW,IAAA,CAAKA,OAAAA;MAChCG,OAAAA,EAAS;AAAE,QAAA,GAAG,IAAA,CAAKA,OAAAA;AAAS,QAAA,GAAGG,KAAAA,EAAOH;AAAQ,OAAA;MAC9CD,OAAAA,EAASI,KAAAA,EAAOJ,WAAW,IAAA,CAAKA,OAAAA;AAChCD,MAAAA,SAAAA,EAAWK,OAAOL,SAAAA,GACd;WAAI,IAAA,CAAKA,SAAAA;WAAcK,KAAAA,CAAML;AAC7B,OAAA,GAAA,IAAA,CAAKA,UAAUa,KAAAA;KACrB,CAAA;AAEA,IAAA,MAAMC,OAAAA,GAAUF,KAAAA,CAAMG,IAAAA,CAAK,IAAI,CAAA;AAC/B,IAAA,IAAA,CAAKX,QAAAA,CAASY,KAAKF,OAAAA,CAAAA;AAEnB,IAAA,OAAOF,KAAAA;AACT;AAEAG,EAAAA,IAAAA,CAAKE,MAAAA,EAAiC;AACpC,IAAA,OAAO,IAAA,CAAKC,EAAAA;;AAEV,MAAA,KAAA;AACAC,MAAAA,6BAAAA;;AAEE,QAAA,CAAA,GAAIC,IAAAA,KAASH,MAAAA,CAAOI,MAAAA,CAAM,GAAID,IAAAA,CAAAA;AAC9B,QAAA;AAAC,UAAA;YAAEE,KAAAA,EAAOL,MAAAA;YAAQM,IAAAA,EAAM;AAAS;;AAAE,OAAA;AAErC,MAAA;QACEC,UAAAA,EAAY,IAAA;QACZC,IAAAA,EAAM,KAAA;QACNC,UAAAA,EAAY,IAAA;QACZC,WAAAA,EAAa;AACf;AAAA,KAAA;AAEJ;EAEAC,OAAAA,GAAU;AACR,IAAA,IAAA,CAAK9B,UAAU+B,MAAAA,GAAS,CAAA;AACxB,IAAA,IAAA,CAAKzB,QAAAA,CAAS0B,OAAAA,CAAQ,CAAClB,KAAAA,KAAUA,OAAAA,CAAAA;AACjC,IAAA,IAAA,CAAKR,SAASyB,MAAAA,GAAS,CAAA;AACzB;EAEAE,KAAAA,GAAQ;AACN,IAAA,IAAA,CAAKjC,SAAAA,GAAY,KAAKA,SAAAA,CAAUkC,MAAAA,CAAO,CAACC,QAAAA,KAAaA,QAAAA,CAASC,SAASR,UAAAA,CAAAA;AACzE;AAEAS,EAAAA,iBAAAA,CACEC,WAGAF,OAAAA,EACW;AACX,IAAA,MAAMpC,YAAyB,EAAA;AAC/BQ,IAAAA,MAAAA,CAAO+B,OAAAA,CAAQD,SAAAA,CAAAA,CAAWN,OAAAA,CAAQ,CAAC,CAACQ,GAAAA,EAAKhB,KAAAA,CAAAA,KAAM;AAC7C,MAAA,IAAIA,KAAAA,EAAO;AAETxB,QAAAA,SAAAA,CAAUkB,IAAAA,CAAK,KAAKE,EAAAA,CAAGoB,GAAAA,EAAKhB,OAAOY,OAAAA,GAAUI,GAAAA,CAAI,CAAA,CAAA;AACnD;KACF,CAAA;AACA,IAAA,OAAO,MAAMxC,SAAAA,CAAUgC,OAAAA,CAAQ,CAAChB,OAAAA,KAAYA,SAAAA,CAAAA;AAC9C;EAEAI,EAAAA,CACEqB,KAAAA,EACAC,UACAN,OAAAA,EACW;AACX,IAAA,OAAO,IAAA,CAAKO,KAAAA,CAAMF,KAAAA,EAAkBC,QAAAA,EAA2BN,OAAAA,CAAAA;AACjE;EAEAO,KAAAA,CAASC,OAAAA,EAAkBF,UAAuBN,OAAAA,EAAqC;AACrF,IAAA,MAAMS,gCAAgB,MAAA,CAAA,MAAA;AACpB,MAAA,MAAMC,WAAwB,EAAA;AAC9B,MAAA,IAAIjB,cAAcO,OAAAA,EAASP,WAAAA;AAE3B,MAAA,IAAIe,YAAY,GAAA,EAAK;AACnBf,QAAAA,WAAAA,KAAgB,KAAA;AAChBiB,QAAAA,QAAAA,CAAS5B,IAAAA,CAAK,CAACuB,KAAAA,KAAUA,KAAAA,CAAMM,IAAAA,KAASC,yBAAe,IAAA,CAAK9C,SAAAA,EAAWuC,KAAAA,CAAMhB,IAAI,CAAA,CAAA;AACnF,OAAA,MAAA,IAAWmB,YAAY,KAAA,EAAO;AAC5Bf,QAAAA,WAAAA,KAAgB,IAAA;AAChBiB,QAAAA,QAAAA,CAAS5B,IAAAA,CAAK,MAAM,IAAA,CAAA;AACtB,OAAA,MAAA,IAAW0B,mBAAmBK,MAAAA,EAAQ;AACpCpB,QAAAA,WAAAA,KAAgB,IAAA;AAChBiB,QAAAA,QAAAA,CAAS5B,KAAK,CAACuB,KAAAA,KAAUG,QAAQM,IAAAA,CAAKT,KAAAA,CAAMM,IAAI,CAAA,CAAA;OAClD,MAAA,IAAW,OAAOH,YAAY,UAAA,EAAY;AACxCf,QAAAA,WAAAA,KAAgB,KAAA;AAChBiB,QAAAA,QAAAA,CAAS5B,KAAK0B,OAAAA,CAAAA;OAChB,MAAA,IAAW,OAAOA,YAAY,QAAA,EAAU;AACtC,QAAA,IAAIO,gBAAAA,CAAOP,OAAAA,CAAAA,EAAU;AACnBf,UAAAA,WAAAA,KAAgB,IAAA;AAChBiB,UAAAA,QAAAA,CAAS5B,IAAAA,CAAK,CAACuB,KAAAA,KAAUA,KAAAA,CAAMM,SAASH,OAAAA,CAAAA;SAC1C,MAAO;AACLf,UAAAA,WAAAA,KAAgB,KAAA;AAChBiB,UAAAA,QAAAA,CAAS5B,IAAAA,CACP,CAACuB,KAAAA,KACCA,KAAAA,CAAMhB,IAAAA,KAASmB,OAAAA,IAAWH,KAAAA,CAAMM,IAAAA,KAASC,wBAAAA,CAAe,IAAA,CAAK9C,SAAAA,EAAWuC,KAAAA,CAAMhB,IAAI,CAAA,CAAA;AAExF;OACF,MAAO;AACL,QAAA,MAAM,IAAI2B,wBAAa,2BAAA,CAAA;AACzB;AAEA,MAAA,IAAI,CAACvB,WAAAA,EAAa;AAChB,QAAA,MAAMwB,YAAAA,2BAAgBZ,KAAAA,KAAAA;AAEpB,UAAA,OAAO,CAAC,KAAKpC,KAAAA,EAAOiD,KAAAA,GAAQ,OAAO,IAAA,CAAKjD,KAAAA,EAAOiD,KAAAA,KAAUb,KAAAA,CAAMpC,KAAAA,EAAOiD,KAAAA;SACxE,EAHqB,cAAA,CAAA;AAIrBR,QAAAA,QAAAA,CAASS,QAAQF,YAAAA,CAAAA;AACnB;AAEA,MAAA,OAAO,CAACZ,UAAUK,QAAAA,CAASU,KAAAA,CAAM,CAACZ,QAAAA,KAAYA,QAAAA,CAAQH,KAAAA,CAAAA,CAAAA;KACxD,EAxCsB,eAAA,CAAA;AA0CtB,IAAA,MAAMN,QAAAA,GAAqB;AACzBC,MAAAA,OAAAA;AACAM,MAAAA,QAAAA;MACAe,GAAAA,EAAKb,OAAAA;AACLD,MAAAA,KAAAA,EAAOE,aAAAA;AACT,KAAA;AAEA,IAAA,MAAMa,QAAAA,GAAWtB,SAASsB,QAAAA,IAAY,CAAA;AACtC,IAAA,MAAMC,WAAAA,GAAc,IAAA,CAAK3D,SAAAA,CAAU4D,SAAAA,CAAU,CAACC,OAAOA,CAAAA,CAAEzB,OAAAA,EAASsB,QAAAA,IAAY,CAAA,KAAMA,QAAAA,CAAAA;AAClF,IAAA,IAAIC,gBAAgB,EAAA,EAAI;AACtB,MAAA,IAAA,CAAK3D,SAAAA,CAAUkB,KAAKiB,QAAAA,CAAAA;KACtB,MAAO;AACL,MAAA,IAAA,CAAKnC,SAAAA,CAAU8D,MAAAA,CAAOH,WAAAA,EAAa,CAAA,EAAGxB,QAAAA,CAAAA;AACxC;AAEA,IAAA,OAAO,MAAA;AACL4B,MAAAA,yBAAAA,CAAgB,IAAA,CAAK/D,WAAWmC,QAAAA,CAAAA;AAClC,KAAA;AACF;EAEA,MAAM6B,IAAAA,CACJvC,MACAD,KAAAA,EACe;AACfyC,IAAAA,yBAAAA,CAAgBxC,IAAAA,CAAAA;AAEhB,IAAA,MAAMgB,KAAAA,GAAQ,IAAA,CAAKyB,WAAAA,CAAYzC,IAAAA,CAAAA;AAC/B,IAAA,OAAO,MAAM,IAAA,CAAKF,MAAAA,CAAOC,KAAAA,EAAOiB,KAAAA,CAAAA;AAClC;EAEA,MAAgBlB,MAAAA,CAAO4C,MAAe1B,KAAAA,EAAkB;AACtD,IAAA,MAAM2B,aAAwB,EAAA;AAE9B,IAAA,KAAA,IAASC,IAAI,IAAA,CAAKrE,SAAAA,CAAU+B,SAAS,CAAA,EAAGsC,CAAAA,IAAK,GAAGA,CAAAA,EAAAA,EAAK;AACnD,MAAA,MAAMlC,QAAAA,GAAW,IAAA,CAAKnC,SAAAA,CAAUqE,CAAAA,CAAAA;AAChC,MAAA,IAAI,CAAClC,QAAAA,CAASQ,KAAAA,CAAMF,KAAAA,CAAAA,EAAQ;AAC1B,QAAA;AACF;AAEA,MAAA,IAAIN,QAAAA,CAASC,SAAST,IAAAA,EAAM;AAC1B,QAAA,IAAA,CAAK3B,SAAAA,CAAU8D,MAAAA,CAAOO,CAAAA,EAAG,CAAA,CAAA;AAC3B;AAEA,MAAA,MAAMC,sBAAM,MAAA,CAAA,YAAYnC,QAAAA,CAASO,QAAAA,CAASyB,IAAAA,EAAM1B,KAAAA,CAAAA,EAApC,KAAA,CAAA;AACZ2B,MAAAA,UAAAA,CAAWlD,IAAAA,CAAKiB,SAASC,OAAAA,EAASV,UAAAA,GAAa,MAAM4C,GAAAA,EAAAA,GAAQA,KAAAA,CAAAA;AAC/D;AACA,IAAA,MAAMC,OAAAA,CAAQC,IAAIJ,UAAAA,CAAAA;AACpB;AAEUF,EAAAA,WAAAA,CAAYzC,IAAAA,EAAyB;AAC7C,IAAA,OAAO;AACLgD,MAAAA,EAAAA,EAAIC,yBAAAA,EAAAA;AACJzE,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;AACdwB,MAAAA,IAAAA;MACAsB,IAAAA,EAAMC,wBAAAA,CAAe,IAAA,CAAK9C,SAAAA,EAAWuB,IAAAA,CAAAA;AACrCkD,MAAAA,SAAAA,sBAAeC,IAAAA,EAAAA;MACfC,MAAAA,EAAQ,IAAA;AACR1E,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;AACdC,MAAAA,OAAAA,EAASI,OAAOsE,MAAAA,CAAO,IAAI,IAAA,CAAK1E,OAAAA,EAAS,EAAC,CAAA;MAC1CC,KAAAA,EAAO0E,uBAAAA,CAAY,KAAK1E,KAAK;AAC/B,KAAA;AACF;;EAEA2E,cAAAA,GAAiB;AACf,IAAA,OAAO;AACL/E,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;MACdC,SAAAA,EAAW6E,uBAAAA,CAAY,KAAK7E,SAAS,CAAA;AACrCC,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;AACdC,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;AACdC,MAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;MACZL,SAAAA,EAAW,IAAA,CAAKA,SAAAA,CAAUiF,GAAAA,CAAIC,WAAAA,CAAK;AAAC,QAAA,KAAA;AAAO,QAAA,SAAA;AAAW,QAAA;OAAW,CAAA,CAAA;AACjE5E,MAAAA,QAAAA,EAAU,IAAA,CAAKA;AACjB,KAAA;AACF;AAEA6E,EAAAA,YAAAA,CAAa,EAAEnF,SAAAA,EAAW,GAAGoF,QAAAA,EAAAA,EAA0D;AACrF5E,IAAAA,MAAAA,CAAOsE,MAAAA,CAAO,MAAMM,QAAAA,EAAU;AAAEpF,MAAAA,SAAAA,EAAW;KAAG,CAAA;AAC9CA,IAAAA,SAAAA,CAAUgC,OAAAA,CAAQ,CAAC,EAAEyB,GAAAA,EAAKf,QAAAA,EAAUN,OAAAA,EAAO,KAAO,IAAA,CAAKO,KAAAA,CAAMc,GAAAA,EAAKf,QAAAA,EAAUN,OAAAA,CAAAA,CAAAA;AAC9E;AACF","file":"emitter.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport {\n  Callback,\n  CleanupFn,\n  EmitterOptions,\n  EventMeta,\n  EventTrace,\n  Matcher,\n  MatcherFn,\n  StringKey,\n} from \"@/emitter/types.js\";\nexport type { EventMeta, EventTrace, Callback };\nimport { Cache } from \"@/cache/decoratorCache.js\";\nimport { createFullPath, isPath, assertValidName, assertValidNamespace } from \"@/emitter/utils.js\";\nimport { EmitterError } from \"@/emitter/errors.js\";\nimport { createRandomHash } from \"@/internals/helpers/hash.js\";\nimport { shallowCopy, toBoundedFunction } from \"@/serializer/utils.js\";\nimport { RequiredAll } from \"@/internals/types.js\";\nimport { Serializable } from \"@/internals/serializable.js\";\nimport { pick } from \"remeda\";\nimport { removeFromArray } from \"@/internals/helpers/array.js\";\n\nexport interface EmitterInput<E extends object = object> {\n  groupId?: string;\n  namespace?: string[];\n  creator?: object;\n  context?: E & object;\n  trace?: EventTrace;\n}\nexport interface EmitterChildInput<E extends object = object> {\n  groupId?: string;\n  namespace?: string[];\n  creator?: object;\n  context?: E & object;\n  trace?: EventTrace;\n}\n\ninterface Listener<T = any> {\n  readonly match: MatcherFn;\n  readonly raw: Matcher;\n  readonly callback: Callback<T>;\n  readonly options?: EmitterOptions;\n}\n\nexport class Emitter<T = Record<keyof any, Callback<unknown>>> extends Serializable {\n  protected listeners: Listener[] = [];\n  protected readonly groupId?: string;\n  public readonly namespace: string[];\n  public readonly creator?: object;\n  public readonly context: object;\n  public readonly trace?: EventTrace;\n  protected readonly cleanups: CleanupFn[] = [];\n\n  constructor(input: EmitterInput = {}) {\n    super();\n    this.groupId = input?.groupId;\n    this.namespace = input?.namespace ?? [];\n    this.creator = input.creator ?? Object.prototype;\n    this.context = input?.context ?? {};\n    this.trace = input.trace;\n    assertValidNamespace(this.namespace);\n  }\n\n  static {\n    this.register();\n  }\n\n  @Cache()\n  static get root() {\n    return new Emitter<Record<string, Callback<any>>>({ creator: Object.create(null) });\n  }\n\n  child<R = T>(input: EmitterChildInput = {}): Emitter<R> {\n    const child = new Emitter<R>({\n      trace: input.trace ?? this.trace,\n      groupId: input?.groupId ?? this.groupId,\n      context: { ...this.context, ...input?.context },\n      creator: input?.creator ?? this.creator,\n      namespace: input?.namespace\n        ? [...this.namespace, ...input.namespace]\n        : this.namespace.slice(),\n    });\n\n    const cleanup = child.pipe(this);\n    this.cleanups.push(cleanup);\n\n    return child;\n  }\n\n  pipe(target: Emitter<any>): CleanupFn {\n    return this.on(\n      // @ts-expect-error\n      \"*.*\",\n      toBoundedFunction(\n        // @ts-expect-error\n        (...args) => target.invoke(...args),\n        [{ value: target, name: \"target\" }],\n      ),\n      {\n        isBlocking: true,\n        once: false,\n        persistent: true,\n        matchNested: false,\n      },\n    );\n  }\n\n  destroy() {\n    this.listeners.length = 0;\n    this.cleanups.forEach((child) => child());\n    this.cleanups.length = 0;\n  }\n\n  reset() {\n    this.listeners = this.listeners.filter((listener) => listener.options?.persistent);\n  }\n\n  registerCallbacks<K extends StringKey<RequiredAll<T>>>(\n    callbacks: Partial<\n      Record<K, NonNullable<T[K]> extends Callback<infer X> ? Callback<X> : never>\n    >,\n    options?: Partial<Record<K, EmitterOptions>>,\n  ): CleanupFn {\n    const listeners: CleanupFn[] = [];\n    Object.entries(callbacks).forEach(([key, value]) => {\n      if (value) {\n        // @ts-expect-error\n        listeners.push(this.on(key, value, options?.[key]));\n      }\n    });\n    return () => listeners.forEach((cleanup) => cleanup());\n  }\n\n  on<K extends StringKey<RequiredAll<T>>>(\n    event: K,\n    callback: NonNullable<T[K]> extends Callback<any> ? T[K] : never,\n    options?: EmitterOptions,\n  ): CleanupFn {\n    return this.match(event as Matcher, callback as Callback<any>, options);\n  }\n\n  match<L>(matcher: Matcher, callback: Callback<L>, options?: EmitterOptions): CleanupFn {\n    const createMatcher = (): MatcherFn => {\n      const matchers: MatcherFn[] = [];\n      let matchNested = options?.matchNested;\n\n      if (matcher === \"*\") {\n        matchNested ??= false;\n        matchers.push((event) => event.path === createFullPath(this.namespace, event.name));\n      } else if (matcher === \"*.*\") {\n        matchNested ??= true;\n        matchers.push(() => true);\n      } else if (matcher instanceof RegExp) {\n        matchNested ??= true;\n        matchers.push((event) => matcher.test(event.path));\n      } else if (typeof matcher === \"function\") {\n        matchNested ??= false;\n        matchers.push(matcher);\n      } else if (typeof matcher === \"string\") {\n        if (isPath(matcher)) {\n          matchNested ??= true;\n          matchers.push((event) => event.path === matcher);\n        } else {\n          matchNested ??= false;\n          matchers.push(\n            (event) =>\n              event.name === matcher && event.path === createFullPath(this.namespace, event.name),\n          );\n        }\n      } else {\n        throw new EmitterError(\"Invalid matcher provided!\");\n      }\n\n      if (!matchNested) {\n        const matchSameRun = (event: EventMeta) => {\n          // global observability x scoped observability\n          return !this.trace?.runId ? true : this.trace?.runId === event.trace?.runId;\n        };\n        matchers.unshift(matchSameRun);\n      }\n\n      return (event) => matchers.every((matcher) => matcher(event));\n    };\n\n    const listener: Listener = {\n      options,\n      callback,\n      raw: matcher,\n      match: createMatcher(),\n    };\n\n    const priority = options?.priority ?? 0;\n    const insertIndex = this.listeners.findIndex((l) => (l.options?.priority ?? 0) >= priority);\n    if (insertIndex === -1) {\n      this.listeners.push(listener);\n    } else {\n      this.listeners.splice(insertIndex, 0, listener);\n    }\n\n    return () => {\n      removeFromArray(this.listeners, listener);\n    };\n  }\n\n  async emit<K extends StringKey<RequiredAll<T>>>(\n    name: K,\n    value: NonNullable<T[K]> extends Callback<infer X> ? X : unknown,\n  ): Promise<void> {\n    assertValidName(name);\n\n    const event = this.createEvent(name);\n    return await this.invoke(value, event);\n  }\n\n  protected async invoke(data: unknown, event: EventMeta) {\n    const executions: unknown[] = [];\n    // Iterate in reverse to match Python's behavior (reversed list iteration)\n    for (let i = this.listeners.length - 1; i >= 0; i--) {\n      const listener = this.listeners[i];\n      if (!listener.match(event)) {\n        continue;\n      }\n\n      if (listener.options?.once) {\n        this.listeners.splice(i, 1);\n      }\n\n      const run = async () => listener.callback(data, event);\n      executions.push(listener.options?.isBlocking ? await run() : run());\n    }\n    await Promise.all(executions);\n  }\n\n  protected createEvent(name: string): EventMeta {\n    return {\n      id: createRandomHash(),\n      groupId: this.groupId,\n      name,\n      path: createFullPath(this.namespace, name),\n      createdAt: new Date(),\n      source: this,\n      creator: this.creator!,\n      context: Object.assign({}, this.context, {}),\n      trace: shallowCopy(this.trace), // TODO\n    };\n  }\n\n  createSnapshot() {\n    return {\n      groupId: this.groupId,\n      namespace: shallowCopy(this.namespace),\n      creator: this.creator,\n      context: this.context,\n      trace: this.trace,\n      listeners: this.listeners.map(pick([\"raw\", \"options\", \"callback\"])),\n      cleanups: this.cleanups,\n    };\n  }\n\n  loadSnapshot({ listeners, ...snapshot }: ReturnType<typeof this.createSnapshot>): void {\n    Object.assign(this, snapshot, { listeners: [] });\n    listeners.forEach(({ raw, callback, options }) => this.match(raw, callback, options));\n  }\n}\n"]}

package/dist/emitter/emitter.d.cts

@@ -1,4 +1,4 @@
-export { C as Callback, E as Emitter, g as EmitterChildInput, f as EmitterInput, e as EventMeta, d as EventTrace } from '../emitter-DpqUYjXH.cjs';
+export { C as Callback, E as Emitter, g as EmitterChildInput, f as EmitterInput, b as EventMeta, e as EventTrace } from '../emitter-jN55XZZq.cjs';
 import '../internals/types.cjs';
 import '../internals/serializable.cjs';
 import '../internals/helpers/guards.cjs';

package/dist/emitter/emitter.d.ts

@@ -1,4 +1,4 @@
-export { C as Callback, E as Emitter, g as EmitterChildInput, f as EmitterInput, e as EventMeta, d as EventTrace } from '../emitter-CZFbzlUi.js';
+export { C as Callback, E as Emitter, g as EmitterChildInput, f as EmitterInput, b as EventMeta, e as EventTrace } from '../emitter-36-9MnvA.js';
 import '../internals/types.js';
 import '../internals/serializable.js';
 import '../internals/helpers/guards.js';