beeai-framework 0.1.20 → 0.1.22

package/dist/backend/message.js

@@ -2,6 +2,10 @@ import { Serializable } from '../internals/serializable.js';
 import { shallowCopy } from '../serializer/utils.js';
 import { z } from 'zod';
 import { ValueError } from '../errors.js';
+import { popProp, hasProp, safeDefineProperty } from '../internals/helpers/object.js';
+import { Logger } from '../logger/logger.js';
+import { isString } from 'remeda';
+import { watchArray } from '../internals/helpers/array.js';
 
 var __defProp = Object.defineProperty;
 var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
@@ -45,6 +49,14 @@ class Message extends Serializable {
         content
       ];
     }
+    for (const chunk of this.content) {
+      this.assertContent(chunk);
+    }
+    this.content = watchArray(this.content, {
+      onAdd: this.assertContent
+    });
+  }
+  assertContent(content) {
   }
   static of({ role, text, meta }) {
     if (role === "user") {
@@ -111,6 +123,20 @@ class AssistantMessage extends Message {
       text
     };
   }
+  assertContent(content) {
+    if (content.type === "tool-call") {
+      const key = "args";
+      const args = popProp(content, key);
+      if (args !== null && !hasProp(content, "input")) {
+        Logger.root.warn(`The '${key}' property in the AssistantMessage class is deprecated and will be removed. Use the 'input' property for Tool Call content chunks instead.`);
+        content.input = args;
+      }
+      safeDefineProperty(content, key, () => {
+        Logger.root.warn(`The '${key}' property in the Tool Call is deprecated and will be removed. Use the 'input' property instead.`);
+        return content.input;
+      });
+    }
+  }
 }
 class ToolMessage extends Message {
   static {
@@ -124,17 +150,47 @@ class ToolMessage extends Message {
     return this.content.filter(isToolResult);
   }
   fromString(text) {
-    const { success, data } = z.object({
+    const schema = z.object({
       type: z.literal("tool-result"),
-      result: z.any(),
       toolName: z.string(),
       toolCallId: z.string()
-    }).safeParse(text);
+    }).and(z.object({
+      output: z.any()
+    }).or(z.object({
+      result: z.any()
+    })));
+    const { success, data } = schema.safeParse(text);
     if (!success) {
       throw new ValueError(`ToolMessage cannot be created from '${text}'!`);
     }
     return data;
   }
+  assertContent(content) {
+    const result = popProp(content, "result", null);
+    const isError = popProp(content, "isError", false);
+    if (result !== null && !hasProp(content, "output")) {
+      Logger.root.warn("The 'result' property in the ToolMessage class is deprecated and will be removed. Use the 'output' property for content chunks instead.");
+      if (isString(result)) {
+        content.output = {
+          type: isError ? "error-text" : "text",
+          value: result
+        };
+      } else {
+        content.output = {
+          type: isError ? "error-json" : "json",
+          value: result
+        };
+      }
+    }
+    safeDefineProperty(content, "result", () => {
+      Logger.root.warn("The 'result' property of Tool Message is deprecated and will be removed. Use the 'output.value' property instead.");
+      return content.output.value;
+    });
+    safeDefineProperty(content, "isError", () => {
+      Logger.root.warn("The 'isError' property of Tool Message is deprecated and will be removed. Use the 'output.type' property instead.");
+      return content.output.type.includes("error");
+    });
+  }
 }
 class SystemMessage extends Message {
   static {

package/dist/backend/message.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/backend/message.ts"],"names":["isText","content","type","isImage","isFile","isToolCall","isToolResult","Message","Serializable","meta","createdAt","Date","fromString","Array","isArray","of","role","text","UserMessage","AssistantMessage","SystemMessage","ToolMessage","CustomMessage","getTexts","map","c","join","filter","createSnapshot","shallowCopy","loadSnapshot","snapshot","Object","assign","toPlain","merge","other","push","fromChunks","chunks","instance","forEach","chunk","Symbol","iterator","register","getToolCalls","getToolResults","success","data","z","object","literal","result","any","toolName","string","toolCallId","safeParse","ValueError","getImages","getFiles","Role","ASSISTANT","SYSTEM","USER","input"],"mappings":";;;;;;;AAyBA,SAASA,OAAOC,OAAAA,EAA2B;AACzC,EAAA,OAAOA,QAAQC,IAAAA,KAAS,MAAA;AAC1B;AAFSF,MAAAA,CAAAA,MAAAA,EAAAA,QAAAA,CAAAA;AAGT,SAASG,QAAQF,OAAAA,EAA2B;AAC1C,EAAA,OAAOA,QAAQC,IAAAA,KAAS,OAAA;AAC1B;AAFSC,MAAAA,CAAAA,OAAAA,EAAAA,SAAAA,CAAAA;AAGT,SAASC,OAAOH,OAAAA,EAA2B;AACzC,EAAA,OAAOA,QAAQC,IAAAA,KAAS,MAAA;AAC1B;AAFSE,MAAAA,CAAAA,MAAAA,EAAAA,QAAAA,CAAAA;AAGT,SAASC,WAAWJ,OAAAA,EAA2B;AAC7C,EAAA,OAAOA,QAAQC,IAAAA,KAAS,WAAA;AAC1B;AAFSG,MAAAA,CAAAA,UAAAA,EAAAA,YAAAA,CAAAA;AAGT,SAASC,aAAaL,OAAAA,EAA2B;AAC/C,EAAA,OAAOA,QAAQC,IAAAA,KAAS,aAAA;AAC1B;AAFSI,MAAAA,CAAAA,YAAAA,EAAAA,cAAAA,CAAAA;AAIF,MAAeC,gBAGZC,YAAAA,CAAAA;EA5CV;;;;AA8CkBP,EAAAA,OAAAA;EAEhB,WAAA,CACEA,OAAAA,EACgBQ,IAAAA,GAAoB,EAAC,EACrC;AACA,IAAA,KAAA,EAAK,EAAA,KAFWA,IAAAA,GAAAA,IAAAA;AAGhB,IAAA,IAAI,CAACA,MAAMC,SAAAA,EAAW;AACpBD,MAAAA,IAAAA,CAAKC,SAAAA,uBAAgBC,IAAAA,EAAAA;AACvB;AACA,IAAA,IAAI,OAAOV,YAAY,QAAA,EAAU;AAC/B,MAAA,IAAA,CAAKA,OAAAA,GAAU;AAAC,QAAA,IAAA,CAAKW,WAAWX,OAAAA;;KAClC,MAAO;AACL,MAAA,IAAA,CAAKA,OAAAA,GAAUY,KAAAA,CAAMC,OAAAA,CAAQb,OAAAA,IAAWA,OAAAA,GAAU;AAACA,QAAAA;;AACrD;AACF;AAIA,EAAA,OAAOc,EAAAA,CAAG,EAAEC,IAAAA,EAAMC,IAAAA,EAAMR,MAAI,EAA2B;AACrD,IAAA,IAAIO,SAAS,MAAA,EAAQ;AACnB,MAAA,OAAO,IAAIE,WAAAA,CAAYD,IAAAA,EAAMR,IAAAA,CAAAA;AAC/B,KAAA,MAAA,IAAWO,SAAS,WAAA,EAAa;AAC/B,MAAA,OAAO,IAAIG,gBAAAA,CAAiBF,IAAAA,EAAMR,IAAAA,CAAAA;AACpC,KAAA,MAAA,IAAWO,SAAS,QAAA,EAAU;AAC5B,MAAA,OAAO,IAAII,aAAAA,CAAcH,IAAAA,EAAMR,IAAAA,CAAAA;AACjC,KAAA,MAAA,IAAWO,SAAS,MAAA,EAAQ;AAC1B,MAAA,OAAO,IAAIK,WAAAA,CAAYJ,IAAAA,EAAMR,IAAAA,CAAAA;KAC/B,MAAO;AACL,MAAA,OAAO,IAAIa,aAAAA,CAAcN,IAAAA,EAAMC,IAAAA,EAAMR,IAAAA,CAAAA;AACvC;AACF;AAEA,EAAA,IAAIQ,IAAAA,GAAO;AACT,IAAA,OAAO,IAAA,CAAKM,QAAAA,EAAQ,CACjBC,GAAAA,CAAI,CAACC,MAAMA,CAAAA,CAAER,IAAI,CAAA,CACjBS,IAAAA,CAAK,EAAA,CAAA;AACV;EAEAH,QAAAA,GAAW;AACT,IAAA,OAAO,IAAA,CAAKtB,OAAAA,CAAQ0B,MAAAA,CAAO3B,MAAAA,CAAAA;AAC7B;EAEA4B,cAAAA,GAAiB;AACf,IAAA,OAAO;MAAE3B,OAAAA,EAAS4B,WAAAA,CAAY,KAAK5B,OAAO,CAAA;MAAGQ,IAAAA,EAAMoB,WAAAA,CAAY,KAAKpB,IAAI,CAAA;AAAGO,MAAAA,IAAAA,EAAM,IAAA,CAAKA;AAAK,KAAA;AAC7F;AAEAc,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7DC,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAMF,QAAAA,CAAAA;AACtB;EAEAG,OAAAA,GAAU;AACR,IAAA,OAAO;AAAElB,MAAAA,IAAAA,EAAM,IAAA,CAAKA,IAAAA;MAAMf,OAAAA,EAAS4B,WAAAA,CAAY,KAAK5B,OAAO;AAAE,KAAA;AAC/D;AAEAkC,EAAAA,KAAAA,CAAMC,KAAAA,EAAsB;AAC1BJ,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,IAAA,EAAMG,KAAAA,CAAM3B,IAAI,CAAA;AAC9B,IAAA,IAAA,CAAKR,OAAAA,CAAQoC,IAAAA,CAAI,GAAID,KAAAA,CAAMnC,OAAO,CAAA;AACpC;AAEA,EAAA,OAAOqC,WAAiEC,MAAAA,EAAkB;AACxF,IAAA,MAAMC,QAAAA,GAAW,IAAI,IAAA,CAAK,EAAE,CAAA;AAC5BD,IAAAA,MAAAA,CAAOE,QAAQ,CAACC,KAAAA,KAAUF,QAAAA,CAASL,KAAAA,CAAMO,KAAAA,CAAAA,CAAAA;AACzC,IAAA,OAAOF,QAAAA;AACT;EAEA,CAACG,MAAAA,CAAOC,QAAQ,CAAA,GAAiB;AAC/B,IAAA,OAAO,IAAA,CAAK3C,OAAAA,CAAQ0C,MAAAA,CAAOC,QAAQ,CAAA,EAAC;AACtC;AACF;AAEO,MAAMzB,yBAAyBZ,OAAAA,CAAAA;EArHtC;;;EAsHkBS,IAAAA,GAAO,WAAA;EAEvB;AACE,IAAA,IAAA,CAAK6B,QAAAA,EAAQ;AACf;EAEAC,YAAAA,GAAe;AACb,IAAA,OAAO,IAAA,CAAK7C,OAAAA,CAAQ0B,MAAAA,CAAOtB,UAAAA,CAAAA;AAC7B;AAEUO,EAAAA,UAAAA,CAAWK,IAAAA,EAAwB;AAC3C,IAAA,OAAO;MAAEf,IAAAA,EAAM,MAAA;AAAQe,MAAAA;AAAK,KAAA;AAC9B;AACF;AAEO,MAAMI,oBAAoBd,OAAAA,CAAAA;EArIjC;;;EAsIkBS,IAAAA,GAAO,MAAA;EAEvB;AACE,IAAA,IAAA,CAAK6B,QAAAA,EAAQ;AACf;EAEAE,cAAAA,GAAiB;AACf,IAAA,OAAO,IAAA,CAAK9C,OAAAA,CAAQ0B,MAAAA,CAAOrB,YAAAA,CAAAA;AAC7B;AAEUM,EAAAA,UAAAA,CAAWK,IAAAA,EAA8B;AACjD,IAAA,MAAM,EAAE+B,OAAAA,EAASC,IAAAA,EAAI,GAAKC,EACvBC,MAAAA,CAAO;MACNjD,IAAAA,EAAMgD,CAAAA,CAAEE,QAAQ,aAAA,CAAA;AAChBC,MAAAA,MAAAA,EAAQH,EAAEI,GAAAA,EAAG;AACbC,MAAAA,QAAAA,EAAUL,EAAEM,MAAAA,EAAM;AAClBC,MAAAA,UAAAA,EAAYP,EAAEM,MAAAA;KAChB,CAAA,CACCE,UAAUzC,IAAAA,CAAAA;AAEb,IAAA,IAAI,CAAC+B,OAAAA,EAAS;AACZ,MAAA,MAAM,IAAIW,UAAAA,CAAW,CAAA,oCAAA,EAAuC1C,IAAAA,CAAAA,EAAAA,CAAQ,CAAA;AACtE;AAEA,IAAA,OAAOgC,IAAAA;AACT;AACF;AAEO,MAAM7B,sBAAsBb,OAAAA,CAAAA;EAlKnC;;;EAmKkBS,IAAAA,GAAoB,QAAA;EAEpC;AACE,IAAA,IAAA,CAAK6B,QAAAA,EAAQ;AACf;AAEUjC,EAAAA,UAAAA,CAAWK,IAAAA,EAAwB;AAC3C,IAAA,OAAO;MAAEf,IAAAA,EAAM,MAAA;AAAQe,MAAAA;AAAK,KAAA;AAC9B;AACF;AAEO,MAAMC,oBAAoBX,OAAAA,CAAAA;EA9KjC;;;EA+KkBS,IAAAA,GAAO,MAAA;EAEvB;AACE,IAAA,IAAA,CAAK6B,QAAAA,EAAQ;AACf;EAEAe,SAAAA,GAAY;AACV,IAAA,OAAO,IAAA,CAAK3D,OAAAA,CAAQ0B,MAAAA,CAAOxB,OAAAA,CAAAA;AAC7B;EAEA0D,QAAAA,GAAW;AACT,IAAA,OAAO,IAAA,CAAK5D,OAAAA,CAAQ0B,MAAAA,CAAOvB,MAAAA,CAAAA;AAC7B;AAEUQ,EAAAA,UAAAA,CAAWK,IAAAA,EAAwB;AAC3C,IAAA,OAAO;MAAEf,IAAAA,EAAM,MAAA;AAAQe,MAAAA;AAAK,KAAA;AAC9B;AACF;AAEO,MAAM6C,IAAAA,GAAO;EAClBC,SAAAA,EAAW,WAAA;EACXC,MAAAA,EAAQ,QAAA;EACRC,IAAAA,EAAM;AACR;AAEO,MAAM3C,sBAAsBf,OAAAA,CAAAA;EAxMnC;;;AAyMSS,EAAAA,IAAAA;AAEP,EAAA,WAAA,CAAYA,IAAAA,EAAcf,OAAAA,EAAsCQ,IAAAA,GAAoB,EAAC,EAAG;AACtF,IAAA,KAAA,CAAMR,SAASQ,IAAAA,CAAAA;AACf,IAAA,IAAI,CAACO,IAAAA,EAAM;AACT,MAAA,MAAM,IAAI2C,UAAAA,CAAW,CAAA,MAAA,EAAS3C,IAAAA,CAAAA,oBAAAA,CAA0B,CAAA;AAC1D;AACA,IAAA,IAAA,CAAKA,IAAAA,GAAOA,IAAAA;AACd;AAEUJ,EAAAA,UAAAA,CAAWsD,KAAAA,EAAmC;AACtD,IAAA,OAAO;MAAEhE,IAAAA,EAAM,MAAA;MAAQe,IAAAA,EAAMiD;AAAM,KAAA;AACrC;AACF","file":"message.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { Serializable } from \"@/internals/serializable.js\";\nimport { shallowCopy } from \"@/serializer/utils.js\";\nimport { FilePart, ImagePart, TextPart, ToolCallPart, ToolResultPart } from \"ai\";\nimport { z } from \"zod\";\nimport { ValueError } from \"@/errors.js\";\n\nexport type MessageRole = \"user\" | \"system\" | \"tool\" | \"assistant\";\nexport type MessageContentPart = TextPart | ToolCallPart | ImagePart | FilePart | ToolResultPart;\n\nexport interface MessageMeta {\n  [key: string]: any;\n  createdAt?: Date;\n}\n\nexport interface MessageInput {\n  role: MessageRole;\n  text: string; // TODO\n  meta?: MessageMeta;\n}\n\nfunction isText(content: MessageContentPart): content is TextPart {\n  return content.type === \"text\";\n}\nfunction isImage(content: MessageContentPart): content is ImagePart {\n  return content.type === \"image\";\n}\nfunction isFile(content: MessageContentPart): content is FilePart {\n  return content.type === \"file\";\n}\nfunction isToolCall(content: MessageContentPart): content is ToolCallPart {\n  return content.type === \"tool-call\";\n}\nfunction isToolResult(content: MessageContentPart): content is ToolResultPart {\n  return content.type === \"tool-result\";\n}\n\nexport abstract class Message<\n  T extends MessageContentPart = MessageContentPart,\n  R extends string = MessageRole | string,\n> extends Serializable {\n  public abstract readonly role: R;\n  public readonly content: T[];\n\n  constructor(\n    content: T | T[] | string,\n    public readonly meta: MessageMeta = {},\n  ) {\n    super();\n    if (!meta?.createdAt) {\n      meta.createdAt = new Date();\n    }\n    if (typeof content === \"string\") {\n      this.content = [this.fromString(content)];\n    } else {\n      this.content = Array.isArray(content) ? content : [content];\n    }\n  }\n\n  protected abstract fromString(input: string): T;\n\n  static of({ role, text, meta }: MessageInput): Message {\n    if (role === \"user\") {\n      return new UserMessage(text, meta);\n    } else if (role === \"assistant\") {\n      return new AssistantMessage(text, meta);\n    } else if (role === \"system\") {\n      return new SystemMessage(text, meta);\n    } else if (role === \"tool\") {\n      return new ToolMessage(text, meta);\n    } else {\n      return new CustomMessage(role, text, meta);\n    }\n  }\n\n  get text() {\n    return this.getTexts()\n      .map((c) => c.text)\n      .join(\"\");\n  }\n\n  getTexts() {\n    return this.content.filter(isText) as TextPart[];\n  }\n\n  createSnapshot() {\n    return { content: shallowCopy(this.content), meta: shallowCopy(this.meta), role: this.role };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n  }\n\n  toPlain() {\n    return { role: this.role, content: shallowCopy(this.content) } as const;\n  }\n\n  merge(other: Message<T, R>) {\n    Object.assign(this, other.meta);\n    this.content.push(...other.content);\n  }\n\n  static fromChunks<M2 extends Message>(this: new (...args: any[]) => M2, chunks: M2[]): M2 {\n    const instance = new this([]);\n    chunks.forEach((chunk) => instance.merge(chunk));\n    return instance;\n  }\n\n  [Symbol.iterator](): Iterator<T> {\n    return this.content[Symbol.iterator]();\n  }\n}\n\nexport class AssistantMessage extends Message<TextPart | ToolCallPart | FilePart> {\n  public readonly role = \"assistant\";\n\n  static {\n    this.register();\n  }\n\n  getToolCalls() {\n    return this.content.filter(isToolCall);\n  }\n\n  protected fromString(text: string): TextPart {\n    return { type: \"text\", text };\n  }\n}\n\nexport class ToolMessage extends Message<ToolResultPart> {\n  public readonly role = \"tool\";\n\n  static {\n    this.register();\n  }\n\n  getToolResults() {\n    return this.content.filter(isToolResult);\n  }\n\n  protected fromString(text: string): ToolResultPart {\n    const { success, data } = z\n      .object({\n        type: z.literal(\"tool-result\"),\n        result: z.any(),\n        toolName: z.string(),\n        toolCallId: z.string(),\n      })\n      .safeParse(text);\n\n    if (!success) {\n      throw new ValueError(`ToolMessage cannot be created from '${text}'!`);\n    }\n\n    return data as ToolResultPart;\n  }\n}\n\nexport class SystemMessage extends Message<TextPart> {\n  public readonly role: MessageRole = \"system\";\n\n  static {\n    this.register();\n  }\n\n  protected fromString(text: string): TextPart {\n    return { type: \"text\", text };\n  }\n}\n\nexport class UserMessage extends Message<TextPart | ImagePart | FilePart> {\n  public readonly role = \"user\";\n\n  static {\n    this.register();\n  }\n\n  getImages() {\n    return this.content.filter(isImage);\n  }\n\n  getFiles() {\n    return this.content.filter(isFile);\n  }\n\n  protected fromString(text: string): TextPart {\n    return { type: \"text\", text };\n  }\n}\n\nexport const Role = {\n  ASSISTANT: \"assistant\",\n  SYSTEM: \"system\",\n  USER: \"user\",\n} as const;\n\nexport class CustomMessage extends Message<MessageContentPart, string> {\n  public role: string;\n\n  constructor(role: string, content: MessageContentPart | string, meta: MessageMeta = {}) {\n    super(content, meta);\n    if (!role) {\n      throw new ValueError(`Role \"${role}\" must be specified!`);\n    }\n    this.role = role;\n  }\n\n  protected fromString(input: string): MessageContentPart {\n    return { type: \"text\", text: input };\n  }\n}\n"]}
+{"version":3,"sources":["../../src/backend/message.ts"],"names":["isText","content","type","isImage","isFile","isToolCall","isToolResult","Message","Serializable","meta","createdAt","Date","fromString","Array","isArray","chunk","assertContent","watchArray","onAdd","of","role","text","UserMessage","AssistantMessage","SystemMessage","ToolMessage","CustomMessage","getTexts","map","c","join","filter","createSnapshot","shallowCopy","loadSnapshot","snapshot","Object","assign","toPlain","merge","other","push","fromChunks","chunks","instance","forEach","Symbol","iterator","register","getToolCalls","key","args","popProp","hasProp","Logger","root","warn","input","safeDefineProperty","getToolResults","schema","z","object","literal","toolName","string","toolCallId","and","output","any","or","result","success","data","safeParse","ValueError","isError","isString","value","includes","getImages","getFiles","Role","ASSISTANT","SYSTEM","USER"],"mappings":";;;;;;;;;;;AA6BA,SAASA,OAAOC,OAAAA,EAA2B;AACzC,EAAA,OAAOA,QAAQC,IAAAA,KAAS,MAAA;AAC1B;AAFSF,MAAAA,CAAAA,MAAAA,EAAAA,QAAAA,CAAAA;AAGT,SAASG,QAAQF,OAAAA,EAA2B;AAC1C,EAAA,OAAOA,QAAQC,IAAAA,KAAS,OAAA;AAC1B;AAFSC,MAAAA,CAAAA,OAAAA,EAAAA,SAAAA,CAAAA;AAGT,SAASC,OAAOH,OAAAA,EAA2B;AACzC,EAAA,OAAOA,QAAQC,IAAAA,KAAS,MAAA;AAC1B;AAFSE,MAAAA,CAAAA,MAAAA,EAAAA,QAAAA,CAAAA;AAGT,SAASC,WAAWJ,OAAAA,EAA2B;AAC7C,EAAA,OAAOA,QAAQC,IAAAA,KAAS,WAAA;AAC1B;AAFSG,MAAAA,CAAAA,UAAAA,EAAAA,YAAAA,CAAAA;AAGT,SAASC,aAAaL,OAAAA,EAA2B;AAC/C,EAAA,OAAOA,QAAQC,IAAAA,KAAS,aAAA;AAC1B;AAFSI,MAAAA,CAAAA,YAAAA,EAAAA,cAAAA,CAAAA;AAIF,MAAeC,gBAGZC,YAAAA,CAAAA;EAhDV;;;;AAkDkBP,EAAAA,OAAAA;EAEhB,WAAA,CACEA,OAAAA,EACgBQ,IAAAA,GAAoB,EAAC,EACrC;AACA,IAAA,KAAA,EAAK,EAAA,KAFWA,IAAAA,GAAAA,IAAAA;AAGhB,IAAA,IAAI,CAACA,MAAMC,SAAAA,EAAW;AACpBD,MAAAA,IAAAA,CAAKC,SAAAA,uBAAgBC,IAAAA,EAAAA;AACvB;AACA,IAAA,IAAI,OAAOV,YAAY,QAAA,EAAU;AAC/B,MAAA,IAAA,CAAKA,OAAAA,GAAU;AAAC,QAAA,IAAA,CAAKW,WAAWX,OAAAA;;KAClC,MAAO;AACL,MAAA,IAAA,CAAKA,OAAAA,GAAUY,KAAAA,CAAMC,OAAAA,CAAQb,OAAAA,IAAWA,OAAAA,GAAU;AAACA,QAAAA;;AACrD;AACA,IAAA,KAAA,MAAWc,KAAAA,IAAS,KAAKd,OAAAA,EAAS;AAChC,MAAA,IAAA,CAAKe,cAAcD,KAAAA,CAAAA;AACrB;AACA,IAAA,IAAA,CAAKd,OAAAA,GAAUgB,UAAAA,CAAW,IAAA,CAAKhB,OAAAA,EAAS;AACtCiB,MAAAA,KAAAA,EAAO,IAAA,CAAKF;KACd,CAAA;AACF;AAEUA,EAAAA,aAAAA,CAAcf,OAAAA,EAAkC;AAAC;AAI3D,EAAA,OAAOkB,EAAAA,CAAG,EAAEC,IAAAA,EAAMC,IAAAA,EAAMZ,MAAI,EAA2B;AACrD,IAAA,IAAIW,SAAS,MAAA,EAAQ;AACnB,MAAA,OAAO,IAAIE,WAAAA,CAAYD,IAAAA,EAAMZ,IAAAA,CAAAA;AAC/B,KAAA,MAAA,IAAWW,SAAS,WAAA,EAAa;AAC/B,MAAA,OAAO,IAAIG,gBAAAA,CAAiBF,IAAAA,EAAMZ,IAAAA,CAAAA;AACpC,KAAA,MAAA,IAAWW,SAAS,QAAA,EAAU;AAC5B,MAAA,OAAO,IAAII,aAAAA,CAAcH,IAAAA,EAAMZ,IAAAA,CAAAA;AACjC,KAAA,MAAA,IAAWW,SAAS,MAAA,EAAQ;AAC1B,MAAA,OAAO,IAAIK,WAAAA,CAAYJ,IAAAA,EAAMZ,IAAAA,CAAAA;KAC/B,MAAO;AACL,MAAA,OAAO,IAAIiB,aAAAA,CAAcN,IAAAA,EAAMC,IAAAA,EAAMZ,IAAAA,CAAAA;AACvC;AACF;AAEA,EAAA,IAAIY,IAAAA,GAAO;AACT,IAAA,OAAO,IAAA,CAAKM,QAAAA,EAAQ,CACjBC,GAAAA,CAAI,CAACC,MAAMA,CAAAA,CAAER,IAAI,CAAA,CACjBS,IAAAA,CAAK,EAAA,CAAA;AACV;EAEAH,QAAAA,GAAW;AACT,IAAA,OAAO,IAAA,CAAK1B,OAAAA,CAAQ8B,MAAAA,CAAO/B,MAAAA,CAAAA;AAC7B;EAEAgC,cAAAA,GAAiB;AACf,IAAA,OAAO;MAAE/B,OAAAA,EAASgC,WAAAA,CAAY,KAAKhC,OAAO,CAAA;MAAGQ,IAAAA,EAAMwB,WAAAA,CAAY,KAAKxB,IAAI,CAAA;AAAGW,MAAAA,IAAAA,EAAM,IAAA,CAAKA;AAAK,KAAA;AAC7F;AAEAc,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7DC,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAMF,QAAAA,CAAAA;AACtB;EAEAG,OAAAA,GAAU;AACR,IAAA,OAAO;AAAElB,MAAAA,IAAAA,EAAM,IAAA,CAAKA,IAAAA;MAAMnB,OAAAA,EAASgC,WAAAA,CAAY,KAAKhC,OAAO;AAAE,KAAA;AAC/D;AAEAsC,EAAAA,KAAAA,CAAMC,KAAAA,EAAsB;AAC1BJ,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,IAAA,EAAMG,KAAAA,CAAM/B,IAAI,CAAA;AAC9B,IAAA,IAAA,CAAKR,OAAAA,CAAQwC,IAAAA,CAAI,GAAID,KAAAA,CAAMvC,OAAO,CAAA;AACpC;AAEA,EAAA,OAAOyC,WAAiEC,MAAAA,EAAkB;AACxF,IAAA,MAAMC,QAAAA,GAAW,IAAI,IAAA,CAAK,EAAE,CAAA;AAC5BD,IAAAA,MAAAA,CAAOE,QAAQ,CAAC9B,KAAAA,KAAU6B,QAAAA,CAASL,KAAAA,CAAMxB,KAAAA,CAAAA,CAAAA;AACzC,IAAA,OAAO6B,QAAAA;AACT;EAEA,CAACE,MAAAA,CAAOC,QAAQ,CAAA,GAAiB;AAC/B,IAAA,OAAO,IAAA,CAAK9C,OAAAA,CAAQ6C,MAAAA,CAAOC,QAAQ,CAAA,EAAC;AACtC;AACF;AAEO,MAAMxB,yBAAyBhB,OAAAA,CAAAA;EAjItC;;;EAkIkBa,IAAAA,GAAO,WAAA;EAEvB;AACE,IAAA,IAAA,CAAK4B,QAAAA,EAAQ;AACf;EAEAC,YAAAA,GAAe;AACb,IAAA,OAAO,IAAA,CAAKhD,OAAAA,CAAQ8B,MAAAA,CAAO1B,UAAAA,CAAAA;AAC7B;AAEUO,EAAAA,UAAAA,CAAWS,IAAAA,EAAwB;AAC3C,IAAA,OAAO;MAAEnB,IAAAA,EAAM,MAAA;AAAQmB,MAAAA;AAAK,KAAA;AAC9B;AAEUL,EAAAA,aAAAA,CACRf,OAAAA,EACuD;AACvD,IAAA,IAAIA,OAAAA,CAAQC,SAAS,WAAA,EAAa;AAChC,MAAA,MAAMgD,GAAAA,GAAM,MAAA;AACZ,MAAA,MAAMC,IAAAA,GAAOC,OAAAA,CAAQnD,OAAAA,EAAgBiD,GAAAA,CAAAA;AAErC,MAAA,IAAIC,SAAS,IAAA,IAAQ,CAACE,OAAAA,CAAQpD,OAAAA,EAAS,OAAA,CAAA,EAAU;AAC/CqD,QAAAA,MAAAA,CAAOC,IAAAA,CAAKC,IAAAA,CACV,CAAA,KAAA,EAAQN,GAAAA,CAAAA,0IAAAA,CAA+I,CAAA;AAEzJjD,QAAAA,OAAAA,CAAQwD,KAAAA,GAAQN,IAAAA;AAClB;AAEAO,MAAAA,kBAAAA,CAAmBzD,OAAAA,EAASiD,KAA6B,MAAA;AACvDI,QAAAA,MAAAA,CAAOC,IAAAA,CAAKC,IAAAA,CACV,CAAA,KAAA,EAAQN,GAAAA,CAAAA,gGAAAA,CAAqG,CAAA;AAE/G,QAAA,OAAOjD,OAAAA,CAAQwD,KAAAA;OACjB,CAAA;AACF;AACF;AACF;AAEO,MAAMhC,oBAAoBlB,OAAAA,CAAAA;EAxKjC;;;EAyKkBa,IAAAA,GAAO,MAAA;EAEvB;AACE,IAAA,IAAA,CAAK4B,QAAAA,EAAQ;AACf;EAEAW,cAAAA,GAAiB;AACf,IAAA,OAAO,IAAA,CAAK1D,OAAAA,CAAQ8B,MAAAA,CAAOzB,YAAAA,CAAAA;AAC7B;AAEUM,EAAAA,UAAAA,CAAWS,IAAAA,EAA8B;AACjD,IAAA,MAAMuC,MAAAA,GAASC,EACZC,MAAAA,CAAO;MACN5D,IAAAA,EAAM2D,CAAAA,CAAEE,QAAQ,aAAA,CAAA;AAChBC,MAAAA,QAAAA,EAAUH,EAAEI,MAAAA,EAAM;AAClBC,MAAAA,UAAAA,EAAYL,EAAEI,MAAAA;KAChB,CAAA,CACCE,GAAAA,CAAIN,CAAAA,CAAEC,MAAAA,CAAO;AAAEM,MAAAA,MAAAA,EAAQP,EAAEQ,GAAAA;KAAM,CAAA,CAAGC,EAAAA,CAAGT,CAAAA,CAAEC,MAAAA,CAAO;AAAES,MAAAA,MAAAA,EAAQV,EAAEQ,GAAAA;AAAM,KAAA,CAAA,CAAA,CAAA;AAEnE,IAAA,MAAM,EAAEG,OAAAA,EAASC,IAAAA,EAAI,GAAKb,MAAAA,CAAOc,UAAUrD,IAAAA,CAAAA;AAC3C,IAAA,IAAI,CAACmD,OAAAA,EAAS;AACZ,MAAA,MAAM,IAAIG,UAAAA,CAAW,CAAA,oCAAA,EAAuCtD,IAAAA,CAAAA,EAAAA,CAAQ,CAAA;AACtE;AAEA,IAAA,OAAOoD,IAAAA;AACT;AAEUzD,EAAAA,aAAAA,CAAcf,OAAAA,EAA4D;AAClF,IAAA,MAAMsE,MAAAA,GAASnB,OAAAA,CAAQnD,OAAAA,EAAgB,QAAA,EAAU,IAAA,CAAA;AACjD,IAAA,MAAM2E,OAAAA,GAAUxB,OAAAA,CAAQnD,OAAAA,EAAgB,SAAA,EAAW,KAAA,CAAA;AAEnD,IAAA,IAAIsE,WAAW,IAAA,IAAQ,CAAClB,OAAAA,CAAQpD,OAAAA,EAAS,QAAA,CAAA,EAAW;AAClDqD,MAAAA,MAAAA,CAAOC,IAAAA,CAAKC,KACV,yIAAA,CAAA;AAEF,MAAA,IAAIqB,QAAAA,CAASN,MAAAA,CAAAA,EAAS;AACpBtE,QAAAA,OAAAA,CAAQmE,MAAAA,GAAS;AAAElE,UAAAA,IAAAA,EAAM0E,UAAU,YAAA,GAAe,MAAA;UAAQE,KAAAA,EAAOP;AAAO,SAAA;OAC1E,MAAO;AACLtE,QAAAA,OAAAA,CAAQmE,MAAAA,GAAS;AAAElE,UAAAA,IAAAA,EAAM0E,UAAU,YAAA,GAAe,MAAA;UAAQE,KAAAA,EAAOP;AAAO,SAAA;AAC1E;AACF;AAEAb,IAAAA,kBAAAA,CAAmBzD,OAAAA,EAAS,UAAiB,MAAA;AAC3CqD,MAAAA,MAAAA,CAAOC,IAAAA,CAAKC,KACV,mHAAA,CAAA;AAEF,MAAA,OAAOvD,QAAQmE,MAAAA,CAAOU,KAAAA;KACxB,CAAA;AACApB,IAAAA,kBAAAA,CAAmBzD,OAAAA,EAAS,WAAkB,MAAA;AAC5CqD,MAAAA,MAAAA,CAAOC,IAAAA,CAAKC,KACV,mHAAA,CAAA;AAEF,MAAA,OAAOvD,OAAAA,CAAQmE,MAAAA,CAAOlE,IAAAA,CAAK6E,QAAAA,CAAS,OAAA,CAAA;KACtC,CAAA;AACF;AACF;AAEO,MAAMvD,sBAAsBjB,OAAAA,CAAAA;EAlOnC;;;EAmOkBa,IAAAA,GAAoB,QAAA;EAEpC;AACE,IAAA,IAAA,CAAK4B,QAAAA,EAAQ;AACf;AAEUpC,EAAAA,UAAAA,CAAWS,IAAAA,EAAwB;AAC3C,IAAA,OAAO;MAAEnB,IAAAA,EAAM,MAAA;AAAQmB,MAAAA;AAAK,KAAA;AAC9B;AACF;AAEO,MAAMC,oBAAoBf,OAAAA,CAAAA;EA9OjC;;;EA+OkBa,IAAAA,GAAO,MAAA;EAEvB;AACE,IAAA,IAAA,CAAK4B,QAAAA,EAAQ;AACf;EAEAgC,SAAAA,GAAY;AACV,IAAA,OAAO,IAAA,CAAK/E,OAAAA,CAAQ8B,MAAAA,CAAO5B,OAAAA,CAAAA;AAC7B;EAEA8E,QAAAA,GAAW;AACT,IAAA,OAAO,IAAA,CAAKhF,OAAAA,CAAQ8B,MAAAA,CAAO3B,MAAAA,CAAAA;AAC7B;AAEUQ,EAAAA,UAAAA,CAAWS,IAAAA,EAAwB;AAC3C,IAAA,OAAO;MAAEnB,IAAAA,EAAM,MAAA;AAAQmB,MAAAA;AAAK,KAAA;AAC9B;AACF;AAEO,MAAM6D,IAAAA,GAAO;EAClBC,SAAAA,EAAW,WAAA;EACXC,MAAAA,EAAQ,QAAA;EACRC,IAAAA,EAAM;AACR;AAEO,MAAM3D,sBAAsBnB,OAAAA,CAAAA;EAxQnC;;;AAyQSa,EAAAA,IAAAA;AAEP,EAAA,WAAA,CAAYA,IAAAA,EAAcnB,OAAAA,EAAsCQ,IAAAA,GAAoB,EAAC,EAAG;AACtF,IAAA,KAAA,CAAMR,SAASQ,IAAAA,CAAAA;AACf,IAAA,IAAI,CAACW,IAAAA,EAAM;AACT,MAAA,MAAM,IAAIuD,UAAAA,CAAW,CAAA,MAAA,EAASvD,IAAAA,CAAAA,oBAAAA,CAA0B,CAAA;AAC1D;AACA,IAAA,IAAA,CAAKA,IAAAA,GAAOA,IAAAA;AACd;AAEUR,EAAAA,UAAAA,CAAW6C,KAAAA,EAAmC;AACtD,IAAA,OAAO;MAAEvD,IAAAA,EAAM,MAAA;MAAQmB,IAAAA,EAAMoC;AAAM,KAAA;AACrC;AACF","file":"message.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { Serializable } from \"@/internals/serializable.js\";\nimport { shallowCopy } from \"@/serializer/utils.js\";\nimport { FilePart, ImagePart, TextPart, ToolCallPart, ToolResultPart } from \"ai\";\nimport { z } from \"zod\";\nimport { ValueError } from \"@/errors.js\";\nimport { hasProp, popProp, safeDefineProperty } from \"@/internals/helpers/object.js\";\nimport { Logger } from \"@/logger/logger.js\";\nimport { isString } from \"remeda\";\nimport { watchArray } from \"@/internals/helpers/array.js\";\n\nexport type MessageRole = \"user\" | \"system\" | \"tool\" | \"assistant\";\nexport type MessageContentPart = TextPart | ToolCallPart | ImagePart | FilePart | ToolResultPart;\n\nexport interface MessageMeta {\n  [key: string]: any;\n  createdAt?: Date;\n}\n\nexport interface MessageInput {\n  role: MessageRole;\n  text: string; // TODO\n  meta?: MessageMeta;\n}\n\nfunction isText(content: MessageContentPart): content is TextPart {\n  return content.type === \"text\";\n}\nfunction isImage(content: MessageContentPart): content is ImagePart {\n  return content.type === \"image\";\n}\nfunction isFile(content: MessageContentPart): content is FilePart {\n  return content.type === \"file\";\n}\nfunction isToolCall(content: MessageContentPart): content is ToolCallPart {\n  return content.type === \"tool-call\";\n}\nfunction isToolResult(content: MessageContentPart): content is ToolResultPart {\n  return content.type === \"tool-result\";\n}\n\nexport abstract class Message<\n  T extends MessageContentPart = MessageContentPart,\n  R extends string = MessageRole | string,\n> extends Serializable {\n  public abstract readonly role: R;\n  public readonly content: T[];\n\n  constructor(\n    content: T | T[] | string,\n    public readonly meta: MessageMeta = {},\n  ) {\n    super();\n    if (!meta?.createdAt) {\n      meta.createdAt = new Date();\n    }\n    if (typeof content === \"string\") {\n      this.content = [this.fromString(content)];\n    } else {\n      this.content = Array.isArray(content) ? content : [content];\n    }\n    for (const chunk of this.content) {\n      this.assertContent(chunk);\n    }\n    this.content = watchArray(this.content, {\n      onAdd: this.assertContent,\n    });\n  }\n\n  protected assertContent(content: T): asserts content is T {}\n\n  protected abstract fromString(input: string): T;\n\n  static of({ role, text, meta }: MessageInput): Message {\n    if (role === \"user\") {\n      return new UserMessage(text, meta);\n    } else if (role === \"assistant\") {\n      return new AssistantMessage(text, meta);\n    } else if (role === \"system\") {\n      return new SystemMessage(text, meta);\n    } else if (role === \"tool\") {\n      return new ToolMessage(text, meta);\n    } else {\n      return new CustomMessage(role, text, meta);\n    }\n  }\n\n  get text() {\n    return this.getTexts()\n      .map((c) => c.text)\n      .join(\"\");\n  }\n\n  getTexts() {\n    return this.content.filter(isText) as TextPart[];\n  }\n\n  createSnapshot() {\n    return { content: shallowCopy(this.content), meta: shallowCopy(this.meta), role: this.role };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n  }\n\n  toPlain() {\n    return { role: this.role, content: shallowCopy(this.content) } as const;\n  }\n\n  merge(other: Message<T, R>) {\n    Object.assign(this, other.meta);\n    this.content.push(...other.content);\n  }\n\n  static fromChunks<M2 extends Message>(this: new (...args: any[]) => M2, chunks: M2[]): M2 {\n    const instance = new this([]);\n    chunks.forEach((chunk) => instance.merge(chunk));\n    return instance;\n  }\n\n  [Symbol.iterator](): Iterator<T> {\n    return this.content[Symbol.iterator]();\n  }\n}\n\nexport class AssistantMessage extends Message<TextPart | ToolCallPart | FilePart> {\n  public readonly role = \"assistant\";\n\n  static {\n    this.register();\n  }\n\n  getToolCalls() {\n    return this.content.filter(isToolCall);\n  }\n\n  protected fromString(text: string): TextPart {\n    return { type: \"text\", text };\n  }\n\n  protected assertContent(\n    content: TextPart | ToolCallPart | FilePart,\n  ): asserts content is TextPart | ToolCallPart | FilePart {\n    if (content.type === \"tool-call\") {\n      const key = \"args\";\n      const args = popProp(content as any, key);\n\n      if (args !== null && !hasProp(content, \"input\")) {\n        Logger.root.warn(\n          `The '${key}' property in the AssistantMessage class is deprecated and will be removed. Use the 'input' property for Tool Call content chunks instead.`,\n        );\n        content.input = args;\n      }\n\n      safeDefineProperty(content, key as keyof typeof content, () => {\n        Logger.root.warn(\n          `The '${key}' property in the Tool Call is deprecated and will be removed. Use the 'input' property instead.`,\n        );\n        return content.input;\n      });\n    }\n  }\n}\n\nexport class ToolMessage extends Message<ToolResultPart> {\n  public readonly role = \"tool\";\n\n  static {\n    this.register();\n  }\n\n  getToolResults() {\n    return this.content.filter(isToolResult);\n  }\n\n  protected fromString(text: string): ToolResultPart {\n    const schema = z\n      .object({\n        type: z.literal(\"tool-result\"),\n        toolName: z.string(),\n        toolCallId: z.string(),\n      })\n      .and(z.object({ output: z.any() }).or(z.object({ result: z.any() })));\n\n    const { success, data } = schema.safeParse(text);\n    if (!success) {\n      throw new ValueError(`ToolMessage cannot be created from '${text}'!`);\n    }\n\n    return data as ToolResultPart;\n  }\n\n  protected assertContent(content: ToolResultPart): asserts content is ToolResultPart {\n    const result = popProp(content as any, \"result\", null);\n    const isError = popProp(content as any, \"isError\", false);\n\n    if (result !== null && !hasProp(content, \"output\")) {\n      Logger.root.warn(\n        \"The 'result' property in the ToolMessage class is deprecated and will be removed. Use the 'output' property for content chunks instead.\",\n      );\n      if (isString(result)) {\n        content.output = { type: isError ? \"error-text\" : \"text\", value: result };\n      } else {\n        content.output = { type: isError ? \"error-json\" : \"json\", value: result };\n      }\n    }\n\n    safeDefineProperty(content, \"result\" as any, () => {\n      Logger.root.warn(\n        \"The 'result' property of Tool Message is deprecated and will be removed. Use the 'output.value' property instead.\",\n      );\n      return content.output.value as any;\n    });\n    safeDefineProperty(content, \"isError\" as any, () => {\n      Logger.root.warn(\n        \"The 'isError' property of Tool Message is deprecated and will be removed. Use the 'output.type' property instead.\",\n      );\n      return content.output.type.includes(\"error\");\n    });\n  }\n}\n\nexport class SystemMessage extends Message<TextPart> {\n  public readonly role: MessageRole = \"system\";\n\n  static {\n    this.register();\n  }\n\n  protected fromString(text: string): TextPart {\n    return { type: \"text\", text };\n  }\n}\n\nexport class UserMessage extends Message<TextPart | ImagePart | FilePart> {\n  public readonly role = \"user\";\n\n  static {\n    this.register();\n  }\n\n  getImages() {\n    return this.content.filter(isImage);\n  }\n\n  getFiles() {\n    return this.content.filter(isFile);\n  }\n\n  protected fromString(text: string): TextPart {\n    return { type: \"text\", text };\n  }\n}\n\nexport const Role = {\n  ASSISTANT: \"assistant\",\n  SYSTEM: \"system\",\n  USER: \"user\",\n} as const;\n\nexport class CustomMessage extends Message<MessageContentPart, string> {\n  public role: string;\n\n  constructor(role: string, content: MessageContentPart | string, meta: MessageMeta = {}) {\n    super(content, meta);\n    if (!role) {\n      throw new ValueError(`Role \"${role}\" must be specified!`);\n    }\n    this.role = role;\n  }\n\n  protected fromString(input: string): MessageContentPart {\n    return { type: \"text\", text: input };\n  }\n}\n"]}

package/dist/backend/utils.d.cts

@@ -1,12 +1,12 @@
 import '../internals/types.cjs';
 import './constants.cjs';
 import '../tools/base.cjs';
-export { F as FullModelName, r as filterToolsByToolChoice, q as generateToolUnionSchema, o as loadModel, p as parseModel } from '../chat-CR4yfOR9.cjs';
+export { F as FullModelName, r as filterToolsByToolChoice, q as generateToolUnionSchema, o as loadModel, p as parseModel } from '../chat-DB_cf15S.cjs';
 import 'zod';
 import '../internals/helpers/guards.cjs';
 import 'ajv';
 import '../context.cjs';
-import '../emitter-DWEsd-ud.cjs';
+import '../emitter-CGKd_-kK.cjs';
 import '../internals/serializable.cjs';
 import '../internals/helpers/promise.cjs';
 import '../errors.cjs';

package/dist/backend/utils.d.ts

@@ -1,12 +1,12 @@
 import '../internals/types.js';
 import './constants.js';
 import '../tools/base.js';
-export { F as FullModelName, r as filterToolsByToolChoice, q as generateToolUnionSchema, o as loadModel, p as parseModel } from '../chat-W9j8T6UU.js';
+export { F as FullModelName, r as filterToolsByToolChoice, q as generateToolUnionSchema, o as loadModel, p as parseModel } from '../chat-BbCYMAdE.js';
 import 'zod';
 import '../internals/helpers/guards.js';
 import 'ajv';
 import '../context.js';
-import '../emitter-C9nwGLt6.js';
+import '../emitter-CN7I4uSV.js';
 import '../internals/serializable.js';
 import '../internals/helpers/promise.js';
 import '../errors.js';

package/dist/{base-BsY2baWy.d.ts → base-BurRTBA7.d.ts}

@@ -1,7 +1,7 @@
-import { Run, GetRunContext } from './context.js';
+import { MiddlewareType, Run, GetRunContext } from './context.js';
 import { FrameworkError } from './errors.js';
 import { Serializable } from './internals/serializable.js';
-import { E as Emitter } from './emitter-C9nwGLt6.js';
+import { E as Emitter } from './emitter-CN7I4uSV.js';
 import { BaseMemory } from './memory/base.js';
 import { OmitEmpty } from './internals/types.js';
 import { AnyTool } from './tools/base.js';
@@ -39,6 +39,7 @@ interface BaseAgentRunOptions {
 declare abstract class BaseAgent<TInput, TOutput, TOptions extends BaseAgentRunOptions = BaseAgentRunOptions> extends Serializable {
     protected isRunning: boolean;
     abstract readonly emitter: Emitter<unknown>;
+    readonly middlewares: MiddlewareType<typeof this>[];
     run(...[input, options]: Partial<TOptions> extends TOptions ? [input: TInput, options?: TOptions] : [input: TInput, options: TOptions]): Run<TOutput, this, readonly [TInput, TOptions | undefined]>;
     protected abstract _run(input: TInput, options: TOptions, run: GetRunContext<typeof this>): Promise<TOutput>;
     destroy(): void;
@@ -48,6 +49,7 @@ declare abstract class BaseAgent<TInput, TOutput, TOptions extends BaseAgentRunO
     createSnapshot(): {
         isRunning: boolean;
         emitter: Emitter<unknown>;
+        middlewares: MiddlewareType<any>[];
     };
     loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>): void;
 }

package/dist/{base-BP7v-NSL.d.cts → base-v-PYIq5G.d.cts}

@@ -1,7 +1,7 @@
-import { Run, GetRunContext } from './context.cjs';
+import { MiddlewareType, Run, GetRunContext } from './context.cjs';
 import { FrameworkError } from './errors.cjs';
 import { Serializable } from './internals/serializable.cjs';
-import { E as Emitter } from './emitter-DWEsd-ud.cjs';
+import { E as Emitter } from './emitter-CGKd_-kK.cjs';
 import { BaseMemory } from './memory/base.cjs';
 import { OmitEmpty } from './internals/types.cjs';
 import { AnyTool } from './tools/base.cjs';
@@ -39,6 +39,7 @@ interface BaseAgentRunOptions {
 declare abstract class BaseAgent<TInput, TOutput, TOptions extends BaseAgentRunOptions = BaseAgentRunOptions> extends Serializable {
     protected isRunning: boolean;
     abstract readonly emitter: Emitter<unknown>;
+    readonly middlewares: MiddlewareType<typeof this>[];
     run(...[input, options]: Partial<TOptions> extends TOptions ? [input: TInput, options?: TOptions] : [input: TInput, options: TOptions]): Run<TOutput, this, readonly [TInput, TOptions | undefined]>;
     protected abstract _run(input: TInput, options: TOptions, run: GetRunContext<typeof this>): Promise<TOutput>;
     destroy(): void;
@@ -48,6 +49,7 @@ declare abstract class BaseAgent<TInput, TOutput, TOptions extends BaseAgentRunO
     createSnapshot(): {
         isRunning: boolean;
         emitter: Emitter<unknown>;
+        middlewares: MiddlewareType<any>[];
     };
     loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>): void;
 }

package/dist/cache/fileCache.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/cache/fileCache.ts"],"names":["FileCache","BaseCache","input","register","source","fullPath","fromProvider","provider","fs","promises","writeFile","serialize","getProvider","exists","stat","then","r","isFile","catch","serialized","readFile","target","snapshot","Serializer","deserialize","Target","getFactory","ref","instance","fromSnapshot","TypeError","SlidingCache","size","Infinity","ttl","reload","Cache","getInstance","clear","save","set","key","value","get","finally","has","delete","result","createSnapshot","loadSnapshot","Object","assign"],"mappings":";;;;;;;;;;;;;;AAGC,SAAA,YAAA,CAAA,UAAA,EAAA,MAAA,EAAA,GAAA,EAAA,IAAA,EAAA;;;;;;AAAA,MAAA,CAAA,YAAA,EAAA,cAAA,CAAA;;;;;AAaM,MAAMA,kBAAqBC,kBAAAA,CAAAA;EAhBlC;;;;AAiBE,EAAA,WAAA,CAA+BC,KAAAA,EAAc;AAC3C,IAAA,KAAA,EAAK,EAAA,KADwBA,KAAAA,GAAAA,KAAAA;AAE/B;EAEA;AACE,IAAA,IAAA,CAAKC,QAAAA,EAAQ;AACf;AAEA,EAAA,IAAIC,MAAAA,GAAS;AACX,IAAA,OAAO,KAAKF,KAAAA,CAAMG,QAAAA;AACpB;EAEA,aAAaC,YAAAA,CAAgBC,UAAwBL,KAAAA,EAAc;AACjE,IAAA,MAAMM,mBAAAA,CAAGC,SAASC,SAAAA,CAAUR,KAAAA,CAAMG,UAAU,MAAME,QAAAA,CAASI,WAAS,CAAA;AACpE,IAAA,OAAO,IAAIX,UAAaE,KAAAA,CAAAA;AAC1B;AAEA,EAAA,MACgBU,WAAAA,GAAqC;AACnD,IAAA,MAAMC,SAAS,MAAML,mBAAAA,CAAGC,SACrBK,IAAAA,CAAK,IAAA,CAAKZ,MAAMG,QAAQ,CAAA,CACxBU,IAAAA,CAAK,CAACC,MAAMA,CAAAA,CAAEC,MAAAA,EAAM,CAAA,CACpBC,KAAAA,CAAM,MAAM,KAAA,CAAA;AAEf,IAAA,IAAIL,MAAAA,EAAQ;AACV,MAAA,MAAMM,UAAAA,GAAa,MAAMX,mBAAAA,CAAGC,QAAAA,CAASW,SAAS,IAAA,CAAKlB,KAAAA,CAAMG,UAAU,MAAA,CAAA;AACnE,MAAA,MAAM,EAAEgB,MAAAA,EAAQC,QAAAA,KAAa,MAAMC,yBAAAA,CAAWC,YAAiBL,UAAAA,CAAAA;AAC/D,MAAA,MAAMM,MAAAA,GAASF,yBAAAA,CAAWG,UAAAA,CAAWL,MAAAA,CAAAA,CAAQM,GAAAA;AAC7C,MAAA,MAAMC,QAAAA,GAAW,MAAMH,MAAAA,CAAOI,YAAAA,CAAaP,QAAAA,CAAAA;AAC3C,MAAA,IAAI,EAAEM,oBAAoB3B,kBAAAA,CAAAA,EAAY;AACpC,QAAA,MAAM,IAAI6B,UAAU,mEAAA,CAAA;AACtB;AACA,MAAA,OAAOF,QAAAA;KACT,MAAO;AACL,MAAA,OAAO,IAAIG,6BAAAA,CAAa;QACtBC,IAAAA,EAAMC,QAAAA;QACNC,GAAAA,EAAKD;OACP,CAAA;AACF;AACF;AAEA,EAAA,MAAME,MAAAA,GAAS;AAEbC,IAAAA,wBAAAA,CAAMC,WAAAA,CAAY,IAAA,EAAM,aAAA,CAAA,CAAeC,KAAAA,EAAK;AAC5C,IAAA,KAAM,MAAM,KAAK1B,WAAAA,EAAW;AAC9B;AAEA,EAAA,MAAgB2B,IAAAA,GAAO;AACrB,IAAA,MAAMhC,QAAAA,GAAW,MAAM,IAAA,CAAKK,WAAAA,EAAW;AACvC,IAAA,OAAO,MAAMJ,mBAAAA,CAAGC,QAAAA,CAASC,SAAAA,CAAU,IAAA,CAAKR,MAAMG,QAAAA,EAAU,MAAME,QAAAA,CAASI,SAAAA,EAAS,CAAA;AAClF;AAEA,EAAA,MAAMqB,IAAAA,GAAO;AACX,IAAA,MAAMzB,QAAAA,GAAW,MAAM,IAAA,CAAKK,WAAAA,EAAW;AACvC,IAAA,OAAOL,SAASyB,IAAAA,EAAI;AACtB;EAEA,MAAMQ,GAAAA,CAAIC,KAAaC,KAAAA,EAAU;AAC/B,IAAA,MAAMnC,QAAAA,GAAW,MAAM,IAAA,CAAKK,WAAAA,EAAW;AACvC,IAAA,MAAML,QAAAA,CAASiC,GAAAA,CAAIC,GAAAA,EAAKC,KAAAA,CAAAA;AACxB,IAAA,KAAKnC,QAAAA,CAASoC,GAAAA,CAAIF,GAAAA,CAAAA,CAAKG,QAAQ,MAAA;AAC7B,MAAA,KAAK,KAAKL,IAAAA,EAAI;KAChB,CAAA;AACF;AAEA,EAAA,MAAMI,IAAIF,GAAAA,EAAa;AACrB,IAAA,MAAMlC,QAAAA,GAAW,MAAM,IAAA,CAAKK,WAAAA,EAAW;AACvC,IAAA,OAAO,MAAML,QAAAA,CAASoC,GAAAA,CAAIF,GAAAA,CAAAA;AAC5B;AAEA,EAAA,MAAMI,IAAIJ,GAAAA,EAAa;AACrB,IAAA,MAAMlC,QAAAA,GAAW,MAAM,IAAA,CAAKK,WAAAA,EAAW;AACvC,IAAA,OAAO,MAAML,QAAAA,CAASsC,GAAAA,CAAIJ,GAAAA,CAAAA;AAC5B;AAEA,EAAA,MAAMK,OAAOL,GAAAA,EAAa;AACxB,IAAA,MAAMlC,QAAAA,GAAW,MAAM,IAAA,CAAKK,WAAAA,EAAW;AACvC,IAAA,MAAMmC,MAAAA,GAAS,MAAMxC,QAAAA,CAASuC,MAAAA,CAAOL,GAAAA,CAAAA;AACrC,IAAA,MAAM,KAAKF,IAAAA,EAAI;AACf,IAAA,OAAOQ,MAAAA;AACT;AAEA,EAAA,MAAMT,KAAAA,GAAQ;AACZ,IAAA,MAAM/B,QAAAA,GAAW,MAAM,IAAA,CAAKK,WAAAA,EAAW;AACvC,IAAA,MAAML,SAAS+B,KAAAA,EAAK;AACpB,IAAA,MAAM,KAAKC,IAAAA,EAAI;AACjB;AAEA,EAAA,MAAMS,cAAAA,GAAiB;AACrB,IAAA,OAAO;MACL9C,KAAAA,EAAO;AAAEG,QAAAA,QAAAA,EAAU,KAAKH,KAAAA,CAAMG;AAAS,OAAA;MACvCE,QAAAA,EAAU,MAAM,KAAKK,WAAAA;AACvB,KAAA;AACF;AAEAqC,EAAAA,YAAAA,CAAa3B,QAAAA,EAAiE;AAC5E4B,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAM7B,QAAAA,CAAAA;AACtB;AACF","file":"fileCache.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { BaseCache } from \"@/cache/base.js\";\nimport fs from \"node:fs\";\nimport { SlidingCache } from \"@/cache/slidingCache.js\";\nimport { Cache } from \"@/cache/decoratorCache.js\";\nimport { Serializer } from \"@/serializer/serializer.js\";\nimport { SerializableClass } from \"@/internals/serializable.js\";\n\ninterface Input {\n  fullPath: string;\n}\n\nexport class FileCache<T> extends BaseCache<T> {\n  constructor(protected readonly input: Input) {\n    super();\n  }\n\n  static {\n    this.register();\n  }\n\n  get source() {\n    return this.input.fullPath;\n  }\n\n  static async fromProvider<A>(provider: BaseCache<A>, input: Input) {\n    await fs.promises.writeFile(input.fullPath, await provider.serialize());\n    return new FileCache<A>(input);\n  }\n\n  @Cache()\n  protected async getProvider(): Promise<BaseCache<T>> {\n    const exists = await fs.promises\n      .stat(this.input.fullPath)\n      .then((r) => r.isFile())\n      .catch(() => false);\n\n    if (exists) {\n      const serialized = await fs.promises.readFile(this.input.fullPath, \"utf8\");\n      const { target, snapshot } = await Serializer.deserialize<any>(serialized);\n      const Target = Serializer.getFactory(target).ref as SerializableClass<BaseCache<T>>;\n      const instance = await Target.fromSnapshot(snapshot);\n      if (!(instance instanceof BaseCache)) {\n        throw new TypeError(\"Provided file does not serialize any instance of BaseCache class.\");\n      }\n      return instance;\n    } else {\n      return new SlidingCache({\n        size: Infinity,\n        ttl: Infinity,\n      });\n    }\n  }\n\n  async reload() {\n    // @ts-expect-error protected property\n    Cache.getInstance(this, \"getProvider\").clear();\n    void (await this.getProvider());\n  }\n\n  protected async save() {\n    const provider = await this.getProvider();\n    return await fs.promises.writeFile(this.input.fullPath, await provider.serialize());\n  }\n\n  async size() {\n    const provider = await this.getProvider();\n    return provider.size();\n  }\n\n  async set(key: string, value: T) {\n    const provider = await this.getProvider();\n    await provider.set(key, value);\n    void provider.get(key).finally(() => {\n      void this.save();\n    });\n  }\n\n  async get(key: string) {\n    const provider = await this.getProvider();\n    return await provider.get(key);\n  }\n\n  async has(key: string) {\n    const provider = await this.getProvider();\n    return await provider.has(key);\n  }\n\n  async delete(key: string) {\n    const provider = await this.getProvider();\n    const result = await provider.delete(key);\n    await this.save();\n    return result;\n  }\n\n  async clear() {\n    const provider = await this.getProvider();\n    await provider.clear();\n    await this.save();\n  }\n\n  async createSnapshot() {\n    return {\n      input: { fullPath: this.input.fullPath },\n      provider: await this.getProvider(),\n    };\n  }\n\n  loadSnapshot(snapshot: Awaited<ReturnType<typeof this.createSnapshot>>): void {\n    Object.assign(this, snapshot);\n  }\n}\n"]}
+{"version":3,"sources":["../../src/cache/fileCache.ts"],"names":["FileCache","BaseCache","input","register","source","fullPath","fromProvider","provider","fs","promises","writeFile","serialize","getProvider","exists","stat","then","r","isFile","catch","serialized","readFile","target","snapshot","Serializer","deserialize","Target","getFactory","ref","instance","fromSnapshot","TypeError","SlidingCache","size","Infinity","ttl","reload","Cache","getInstance","clear","save","set","key","value","get","finally","has","delete","result","createSnapshot","loadSnapshot","Object","assign"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;AAgBO,MAAMA,kBAAqBC,kBAAAA,CAAAA;AAAAA,EAAAA;;;;AAChC,EAAA,WAAA,CAA+BC,KAAAA,EAAc;AAC3C,IAAA,KAAA,EAAK,EAAA,KADwBA,KAAAA,GAAAA,KAAAA;AAE/B;EAEA;AACE,IAAA,IAAA,CAAKC,QAAAA,EAAQ;AACf;AAEA,EAAA,IAAIC,MAAAA,GAAS;AACX,IAAA,OAAO,KAAKF,KAAAA,CAAMG,QAAAA;AACpB;EAEA,aAAaC,YAAAA,CAAgBC,UAAwBL,KAAAA,EAAc;AACjE,IAAA,MAAMM,mBAAAA,CAAGC,SAASC,SAAAA,CAAUR,KAAAA,CAAMG,UAAU,MAAME,QAAAA,CAASI,WAAS,CAAA;AACpE,IAAA,OAAO,IAAIX,UAAaE,KAAAA,CAAAA;AAC1B;AAEA,EAAA,MACgBU,WAAAA,GAAqC;AACnD,IAAA,MAAMC,SAAS,MAAML,mBAAAA,CAAGC,SACrBK,IAAAA,CAAK,IAAA,CAAKZ,MAAMG,QAAQ,CAAA,CACxBU,IAAAA,CAAK,CAACC,MAAMA,CAAAA,CAAEC,MAAAA,EAAM,CAAA,CACpBC,KAAAA,CAAM,MAAM,KAAA,CAAA;AAEf,IAAA,IAAIL,MAAAA,EAAQ;AACV,MAAA,MAAMM,UAAAA,GAAa,MAAMX,mBAAAA,CAAGC,QAAAA,CAASW,SAAS,IAAA,CAAKlB,KAAAA,CAAMG,UAAU,MAAA,CAAA;AACnE,MAAA,MAAM,EAAEgB,MAAAA,EAAQC,QAAAA,KAAa,MAAMC,yBAAAA,CAAWC,YAAiBL,UAAAA,CAAAA;AAC/D,MAAA,MAAMM,MAAAA,GAASF,yBAAAA,CAAWG,UAAAA,CAAWL,MAAAA,CAAAA,CAAQM,GAAAA;AAC7C,MAAA,MAAMC,QAAAA,GAAW,MAAMH,MAAAA,CAAOI,YAAAA,CAAaP,QAAAA,CAAAA;AAC3C,MAAA,IAAI,EAAEM,oBAAoB3B,kBAAAA,CAAAA,EAAY;AACpC,QAAA,MAAM,IAAI6B,UAAU,mEAAA,CAAA;AACtB;AACA,MAAA,OAAOF,QAAAA;KACT,MAAO;AACL,MAAA,OAAO,IAAIG,6BAAAA,CAAa;QACtBC,IAAAA,EAAMC,QAAAA;QACNC,GAAAA,EAAKD;OACP,CAAA;AACF;AACF;AAEA,EAAA,MAAME,MAAAA,GAAS;AAEbC,IAAAA,wBAAAA,CAAMC,WAAAA,CAAY,IAAA,EAAM,aAAA,CAAA,CAAeC,KAAAA,EAAK;AAC5C,IAAA,KAAM,MAAM,KAAK1B,WAAAA,EAAW;AAC9B;AAEA,EAAA,MAAgB2B,IAAAA,GAAO;AACrB,IAAA,MAAMhC,QAAAA,GAAW,MAAM,IAAA,CAAKK,WAAAA,EAAW;AACvC,IAAA,OAAO,MAAMJ,mBAAAA,CAAGC,QAAAA,CAASC,SAAAA,CAAU,IAAA,CAAKR,MAAMG,QAAAA,EAAU,MAAME,QAAAA,CAASI,SAAAA,EAAS,CAAA;AAClF;AAEA,EAAA,MAAMqB,IAAAA,GAAO;AACX,IAAA,MAAMzB,QAAAA,GAAW,MAAM,IAAA,CAAKK,WAAAA,EAAW;AACvC,IAAA,OAAOL,SAASyB,IAAAA,EAAI;AACtB;EAEA,MAAMQ,GAAAA,CAAIC,KAAaC,KAAAA,EAAU;AAC/B,IAAA,MAAMnC,QAAAA,GAAW,MAAM,IAAA,CAAKK,WAAAA,EAAW;AACvC,IAAA,MAAML,QAAAA,CAASiC,GAAAA,CAAIC,GAAAA,EAAKC,KAAAA,CAAAA;AACxB,IAAA,KAAKnC,QAAAA,CAASoC,GAAAA,CAAIF,GAAAA,CAAAA,CAAKG,QAAQ,MAAA;AAC7B,MAAA,KAAK,KAAKL,IAAAA,EAAI;KAChB,CAAA;AACF;AAEA,EAAA,MAAMI,IAAIF,GAAAA,EAAa;AACrB,IAAA,MAAMlC,QAAAA,GAAW,MAAM,IAAA,CAAKK,WAAAA,EAAW;AACvC,IAAA,OAAO,MAAML,QAAAA,CAASoC,GAAAA,CAAIF,GAAAA,CAAAA;AAC5B;AAEA,EAAA,MAAMI,IAAIJ,GAAAA,EAAa;AACrB,IAAA,MAAMlC,QAAAA,GAAW,MAAM,IAAA,CAAKK,WAAAA,EAAW;AACvC,IAAA,OAAO,MAAML,QAAAA,CAASsC,GAAAA,CAAIJ,GAAAA,CAAAA;AAC5B;AAEA,EAAA,MAAMK,OAAOL,GAAAA,EAAa;AACxB,IAAA,MAAMlC,QAAAA,GAAW,MAAM,IAAA,CAAKK,WAAAA,EAAW;AACvC,IAAA,MAAMmC,MAAAA,GAAS,MAAMxC,QAAAA,CAASuC,MAAAA,CAAOL,GAAAA,CAAAA;AACrC,IAAA,MAAM,KAAKF,IAAAA,EAAI;AACf,IAAA,OAAOQ,MAAAA;AACT;AAEA,EAAA,MAAMT,KAAAA,GAAQ;AACZ,IAAA,MAAM/B,QAAAA,GAAW,MAAM,IAAA,CAAKK,WAAAA,EAAW;AACvC,IAAA,MAAML,SAAS+B,KAAAA,EAAK;AACpB,IAAA,MAAM,KAAKC,IAAAA,EAAI;AACjB;AAEA,EAAA,MAAMS,cAAAA,GAAiB;AACrB,IAAA,OAAO;MACL9C,KAAAA,EAAO;AAAEG,QAAAA,QAAAA,EAAU,KAAKH,KAAAA,CAAMG;AAAS,OAAA;MACvCE,QAAAA,EAAU,MAAM,KAAKK,WAAAA;AACvB,KAAA;AACF;AAEAqC,EAAAA,YAAAA,CAAa3B,QAAAA,EAAiE;AAC5E4B,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAM7B,QAAAA,CAAAA;AACtB;AACF","file":"fileCache.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { BaseCache } from \"@/cache/base.js\";\nimport fs from \"node:fs\";\nimport { SlidingCache } from \"@/cache/slidingCache.js\";\nimport { Cache } from \"@/cache/decoratorCache.js\";\nimport { Serializer } from \"@/serializer/serializer.js\";\nimport { SerializableClass } from \"@/internals/serializable.js\";\n\ninterface Input {\n  fullPath: string;\n}\n\nexport class FileCache<T> extends BaseCache<T> {\n  constructor(protected readonly input: Input) {\n    super();\n  }\n\n  static {\n    this.register();\n  }\n\n  get source() {\n    return this.input.fullPath;\n  }\n\n  static async fromProvider<A>(provider: BaseCache<A>, input: Input) {\n    await fs.promises.writeFile(input.fullPath, await provider.serialize());\n    return new FileCache<A>(input);\n  }\n\n  @Cache()\n  protected async getProvider(): Promise<BaseCache<T>> {\n    const exists = await fs.promises\n      .stat(this.input.fullPath)\n      .then((r) => r.isFile())\n      .catch(() => false);\n\n    if (exists) {\n      const serialized = await fs.promises.readFile(this.input.fullPath, \"utf8\");\n      const { target, snapshot } = await Serializer.deserialize<any>(serialized);\n      const Target = Serializer.getFactory(target).ref as SerializableClass<BaseCache<T>>;\n      const instance = await Target.fromSnapshot(snapshot);\n      if (!(instance instanceof BaseCache)) {\n        throw new TypeError(\"Provided file does not serialize any instance of BaseCache class.\");\n      }\n      return instance;\n    } else {\n      return new SlidingCache({\n        size: Infinity,\n        ttl: Infinity,\n      });\n    }\n  }\n\n  async reload() {\n    // @ts-expect-error protected property\n    Cache.getInstance(this, \"getProvider\").clear();\n    void (await this.getProvider());\n  }\n\n  protected async save() {\n    const provider = await this.getProvider();\n    return await fs.promises.writeFile(this.input.fullPath, await provider.serialize());\n  }\n\n  async size() {\n    const provider = await this.getProvider();\n    return provider.size();\n  }\n\n  async set(key: string, value: T) {\n    const provider = await this.getProvider();\n    await provider.set(key, value);\n    void provider.get(key).finally(() => {\n      void this.save();\n    });\n  }\n\n  async get(key: string) {\n    const provider = await this.getProvider();\n    return await provider.get(key);\n  }\n\n  async has(key: string) {\n    const provider = await this.getProvider();\n    return await provider.has(key);\n  }\n\n  async delete(key: string) {\n    const provider = await this.getProvider();\n    const result = await provider.delete(key);\n    await this.save();\n    return result;\n  }\n\n  async clear() {\n    const provider = await this.getProvider();\n    await provider.clear();\n    await this.save();\n  }\n\n  async createSnapshot() {\n    return {\n      input: { fullPath: this.input.fullPath },\n      provider: await this.getProvider(),\n    };\n  }\n\n  loadSnapshot(snapshot: Awaited<ReturnType<typeof this.createSnapshot>>): void {\n    Object.assign(this, snapshot);\n  }\n}\n"]}

package/dist/cache/fileCache.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/cache/fileCache.ts"],"names":["FileCache","BaseCache","input","register","source","fullPath","fromProvider","provider","fs","promises","writeFile","serialize","getProvider","exists","stat","then","r","isFile","catch","serialized","readFile","target","snapshot","Serializer","deserialize","Target","getFactory","ref","instance","fromSnapshot","TypeError","SlidingCache","size","Infinity","ttl","reload","Cache","getInstance","clear","save","set","key","value","get","finally","has","delete","result","createSnapshot","loadSnapshot","Object","assign"],"mappings":";;;;;;;;AAGC,SAAA,YAAA,CAAA,UAAA,EAAA,MAAA,EAAA,GAAA,EAAA,IAAA,EAAA;;;;;;AAAA,MAAA,CAAA,YAAA,EAAA,cAAA,CAAA;;;;;AAaM,MAAMA,kBAAqBC,SAAAA,CAAAA;EAhBlC;;;;AAiBE,EAAA,WAAA,CAA+BC,KAAAA,EAAc;AAC3C,IAAA,KAAA,EAAK,EAAA,KADwBA,KAAAA,GAAAA,KAAAA;AAE/B;EAEA;AACE,IAAA,IAAA,CAAKC,QAAAA,EAAQ;AACf;AAEA,EAAA,IAAIC,MAAAA,GAAS;AACX,IAAA,OAAO,KAAKF,KAAAA,CAAMG,QAAAA;AACpB;EAEA,aAAaC,YAAAA,CAAgBC,UAAwBL,KAAAA,EAAc;AACjE,IAAA,MAAMM,EAAAA,CAAGC,SAASC,SAAAA,CAAUR,KAAAA,CAAMG,UAAU,MAAME,QAAAA,CAASI,WAAS,CAAA;AACpE,IAAA,OAAO,IAAIX,UAAaE,KAAAA,CAAAA;AAC1B;AAEA,EAAA,MACgBU,WAAAA,GAAqC;AACnD,IAAA,MAAMC,SAAS,MAAML,EAAAA,CAAGC,SACrBK,IAAAA,CAAK,IAAA,CAAKZ,MAAMG,QAAQ,CAAA,CACxBU,IAAAA,CAAK,CAACC,MAAMA,CAAAA,CAAEC,MAAAA,EAAM,CAAA,CACpBC,KAAAA,CAAM,MAAM,KAAA,CAAA;AAEf,IAAA,IAAIL,MAAAA,EAAQ;AACV,MAAA,MAAMM,UAAAA,GAAa,MAAMX,EAAAA,CAAGC,QAAAA,CAASW,SAAS,IAAA,CAAKlB,KAAAA,CAAMG,UAAU,MAAA,CAAA;AACnE,MAAA,MAAM,EAAEgB,MAAAA,EAAQC,QAAAA,KAAa,MAAMC,UAAAA,CAAWC,YAAiBL,UAAAA,CAAAA;AAC/D,MAAA,MAAMM,MAAAA,GAASF,UAAAA,CAAWG,UAAAA,CAAWL,MAAAA,CAAAA,CAAQM,GAAAA;AAC7C,MAAA,MAAMC,QAAAA,GAAW,MAAMH,MAAAA,CAAOI,YAAAA,CAAaP,QAAAA,CAAAA;AAC3C,MAAA,IAAI,EAAEM,oBAAoB3B,SAAAA,CAAAA,EAAY;AACpC,QAAA,MAAM,IAAI6B,UAAU,mEAAA,CAAA;AACtB;AACA,MAAA,OAAOF,QAAAA;KACT,MAAO;AACL,MAAA,OAAO,IAAIG,YAAAA,CAAa;QACtBC,IAAAA,EAAMC,QAAAA;QACNC,GAAAA,EAAKD;OACP,CAAA;AACF;AACF;AAEA,EAAA,MAAME,MAAAA,GAAS;AAEbC,IAAAA,KAAAA,CAAMC,WAAAA,CAAY,IAAA,EAAM,aAAA,CAAA,CAAeC,KAAAA,EAAK;AAC5C,IAAA,KAAM,MAAM,KAAK1B,WAAAA,EAAW;AAC9B;AAEA,EAAA,MAAgB2B,IAAAA,GAAO;AACrB,IAAA,MAAMhC,QAAAA,GAAW,MAAM,IAAA,CAAKK,WAAAA,EAAW;AACvC,IAAA,OAAO,MAAMJ,EAAAA,CAAGC,QAAAA,CAASC,SAAAA,CAAU,IAAA,CAAKR,MAAMG,QAAAA,EAAU,MAAME,QAAAA,CAASI,SAAAA,EAAS,CAAA;AAClF;AAEA,EAAA,MAAMqB,IAAAA,GAAO;AACX,IAAA,MAAMzB,QAAAA,GAAW,MAAM,IAAA,CAAKK,WAAAA,EAAW;AACvC,IAAA,OAAOL,SAASyB,IAAAA,EAAI;AACtB;EAEA,MAAMQ,GAAAA,CAAIC,KAAaC,KAAAA,EAAU;AAC/B,IAAA,MAAMnC,QAAAA,GAAW,MAAM,IAAA,CAAKK,WAAAA,EAAW;AACvC,IAAA,MAAML,QAAAA,CAASiC,GAAAA,CAAIC,GAAAA,EAAKC,KAAAA,CAAAA;AACxB,IAAA,KAAKnC,QAAAA,CAASoC,GAAAA,CAAIF,GAAAA,CAAAA,CAAKG,QAAQ,MAAA;AAC7B,MAAA,KAAK,KAAKL,IAAAA,EAAI;KAChB,CAAA;AACF;AAEA,EAAA,MAAMI,IAAIF,GAAAA,EAAa;AACrB,IAAA,MAAMlC,QAAAA,GAAW,MAAM,IAAA,CAAKK,WAAAA,EAAW;AACvC,IAAA,OAAO,MAAML,QAAAA,CAASoC,GAAAA,CAAIF,GAAAA,CAAAA;AAC5B;AAEA,EAAA,MAAMI,IAAIJ,GAAAA,EAAa;AACrB,IAAA,MAAMlC,QAAAA,GAAW,MAAM,IAAA,CAAKK,WAAAA,EAAW;AACvC,IAAA,OAAO,MAAML,QAAAA,CAASsC,GAAAA,CAAIJ,GAAAA,CAAAA;AAC5B;AAEA,EAAA,MAAMK,OAAOL,GAAAA,EAAa;AACxB,IAAA,MAAMlC,QAAAA,GAAW,MAAM,IAAA,CAAKK,WAAAA,EAAW;AACvC,IAAA,MAAMmC,MAAAA,GAAS,MAAMxC,QAAAA,CAASuC,MAAAA,CAAOL,GAAAA,CAAAA;AACrC,IAAA,MAAM,KAAKF,IAAAA,EAAI;AACf,IAAA,OAAOQ,MAAAA;AACT;AAEA,EAAA,MAAMT,KAAAA,GAAQ;AACZ,IAAA,MAAM/B,QAAAA,GAAW,MAAM,IAAA,CAAKK,WAAAA,EAAW;AACvC,IAAA,MAAML,SAAS+B,KAAAA,EAAK;AACpB,IAAA,MAAM,KAAKC,IAAAA,EAAI;AACjB;AAEA,EAAA,MAAMS,cAAAA,GAAiB;AACrB,IAAA,OAAO;MACL9C,KAAAA,EAAO;AAAEG,QAAAA,QAAAA,EAAU,KAAKH,KAAAA,CAAMG;AAAS,OAAA;MACvCE,QAAAA,EAAU,MAAM,KAAKK,WAAAA;AACvB,KAAA;AACF;AAEAqC,EAAAA,YAAAA,CAAa3B,QAAAA,EAAiE;AAC5E4B,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAM7B,QAAAA,CAAAA;AACtB;AACF","file":"fileCache.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { BaseCache } from \"@/cache/base.js\";\nimport fs from \"node:fs\";\nimport { SlidingCache } from \"@/cache/slidingCache.js\";\nimport { Cache } from \"@/cache/decoratorCache.js\";\nimport { Serializer } from \"@/serializer/serializer.js\";\nimport { SerializableClass } from \"@/internals/serializable.js\";\n\ninterface Input {\n  fullPath: string;\n}\n\nexport class FileCache<T> extends BaseCache<T> {\n  constructor(protected readonly input: Input) {\n    super();\n  }\n\n  static {\n    this.register();\n  }\n\n  get source() {\n    return this.input.fullPath;\n  }\n\n  static async fromProvider<A>(provider: BaseCache<A>, input: Input) {\n    await fs.promises.writeFile(input.fullPath, await provider.serialize());\n    return new FileCache<A>(input);\n  }\n\n  @Cache()\n  protected async getProvider(): Promise<BaseCache<T>> {\n    const exists = await fs.promises\n      .stat(this.input.fullPath)\n      .then((r) => r.isFile())\n      .catch(() => false);\n\n    if (exists) {\n      const serialized = await fs.promises.readFile(this.input.fullPath, \"utf8\");\n      const { target, snapshot } = await Serializer.deserialize<any>(serialized);\n      const Target = Serializer.getFactory(target).ref as SerializableClass<BaseCache<T>>;\n      const instance = await Target.fromSnapshot(snapshot);\n      if (!(instance instanceof BaseCache)) {\n        throw new TypeError(\"Provided file does not serialize any instance of BaseCache class.\");\n      }\n      return instance;\n    } else {\n      return new SlidingCache({\n        size: Infinity,\n        ttl: Infinity,\n      });\n    }\n  }\n\n  async reload() {\n    // @ts-expect-error protected property\n    Cache.getInstance(this, \"getProvider\").clear();\n    void (await this.getProvider());\n  }\n\n  protected async save() {\n    const provider = await this.getProvider();\n    return await fs.promises.writeFile(this.input.fullPath, await provider.serialize());\n  }\n\n  async size() {\n    const provider = await this.getProvider();\n    return provider.size();\n  }\n\n  async set(key: string, value: T) {\n    const provider = await this.getProvider();\n    await provider.set(key, value);\n    void provider.get(key).finally(() => {\n      void this.save();\n    });\n  }\n\n  async get(key: string) {\n    const provider = await this.getProvider();\n    return await provider.get(key);\n  }\n\n  async has(key: string) {\n    const provider = await this.getProvider();\n    return await provider.has(key);\n  }\n\n  async delete(key: string) {\n    const provider = await this.getProvider();\n    const result = await provider.delete(key);\n    await this.save();\n    return result;\n  }\n\n  async clear() {\n    const provider = await this.getProvider();\n    await provider.clear();\n    await this.save();\n  }\n\n  async createSnapshot() {\n    return {\n      input: { fullPath: this.input.fullPath },\n      provider: await this.getProvider(),\n    };\n  }\n\n  loadSnapshot(snapshot: Awaited<ReturnType<typeof this.createSnapshot>>): void {\n    Object.assign(this, snapshot);\n  }\n}\n"]}
+{"version":3,"sources":["../../src/cache/fileCache.ts"],"names":["FileCache","BaseCache","input","register","source","fullPath","fromProvider","provider","fs","promises","writeFile","serialize","getProvider","exists","stat","then","r","isFile","catch","serialized","readFile","target","snapshot","Serializer","deserialize","Target","getFactory","ref","instance","fromSnapshot","TypeError","SlidingCache","size","Infinity","ttl","reload","Cache","getInstance","clear","save","set","key","value","get","finally","has","delete","result","createSnapshot","loadSnapshot","Object","assign"],"mappings":";;;;;;;;;;;;;;;;;;;AAgBO,MAAMA,kBAAqBC,SAAAA,CAAAA;AAAAA,EAAAA;;;;AAChC,EAAA,WAAA,CAA+BC,KAAAA,EAAc;AAC3C,IAAA,KAAA,EAAK,EAAA,KADwBA,KAAAA,GAAAA,KAAAA;AAE/B;EAEA;AACE,IAAA,IAAA,CAAKC,QAAAA,EAAQ;AACf;AAEA,EAAA,IAAIC,MAAAA,GAAS;AACX,IAAA,OAAO,KAAKF,KAAAA,CAAMG,QAAAA;AACpB;EAEA,aAAaC,YAAAA,CAAgBC,UAAwBL,KAAAA,EAAc;AACjE,IAAA,MAAMM,EAAAA,CAAGC,SAASC,SAAAA,CAAUR,KAAAA,CAAMG,UAAU,MAAME,QAAAA,CAASI,WAAS,CAAA;AACpE,IAAA,OAAO,IAAIX,UAAaE,KAAAA,CAAAA;AAC1B;AAEA,EAAA,MACgBU,WAAAA,GAAqC;AACnD,IAAA,MAAMC,SAAS,MAAML,EAAAA,CAAGC,SACrBK,IAAAA,CAAK,IAAA,CAAKZ,MAAMG,QAAQ,CAAA,CACxBU,IAAAA,CAAK,CAACC,MAAMA,CAAAA,CAAEC,MAAAA,EAAM,CAAA,CACpBC,KAAAA,CAAM,MAAM,KAAA,CAAA;AAEf,IAAA,IAAIL,MAAAA,EAAQ;AACV,MAAA,MAAMM,UAAAA,GAAa,MAAMX,EAAAA,CAAGC,QAAAA,CAASW,SAAS,IAAA,CAAKlB,KAAAA,CAAMG,UAAU,MAAA,CAAA;AACnE,MAAA,MAAM,EAAEgB,MAAAA,EAAQC,QAAAA,KAAa,MAAMC,UAAAA,CAAWC,YAAiBL,UAAAA,CAAAA;AAC/D,MAAA,MAAMM,MAAAA,GAASF,UAAAA,CAAWG,UAAAA,CAAWL,MAAAA,CAAAA,CAAQM,GAAAA;AAC7C,MAAA,MAAMC,QAAAA,GAAW,MAAMH,MAAAA,CAAOI,YAAAA,CAAaP,QAAAA,CAAAA;AAC3C,MAAA,IAAI,EAAEM,oBAAoB3B,SAAAA,CAAAA,EAAY;AACpC,QAAA,MAAM,IAAI6B,UAAU,mEAAA,CAAA;AACtB;AACA,MAAA,OAAOF,QAAAA;KACT,MAAO;AACL,MAAA,OAAO,IAAIG,YAAAA,CAAa;QACtBC,IAAAA,EAAMC,QAAAA;QACNC,GAAAA,EAAKD;OACP,CAAA;AACF;AACF;AAEA,EAAA,MAAME,MAAAA,GAAS;AAEbC,IAAAA,KAAAA,CAAMC,WAAAA,CAAY,IAAA,EAAM,aAAA,CAAA,CAAeC,KAAAA,EAAK;AAC5C,IAAA,KAAM,MAAM,KAAK1B,WAAAA,EAAW;AAC9B;AAEA,EAAA,MAAgB2B,IAAAA,GAAO;AACrB,IAAA,MAAMhC,QAAAA,GAAW,MAAM,IAAA,CAAKK,WAAAA,EAAW;AACvC,IAAA,OAAO,MAAMJ,EAAAA,CAAGC,QAAAA,CAASC,SAAAA,CAAU,IAAA,CAAKR,MAAMG,QAAAA,EAAU,MAAME,QAAAA,CAASI,SAAAA,EAAS,CAAA;AAClF;AAEA,EAAA,MAAMqB,IAAAA,GAAO;AACX,IAAA,MAAMzB,QAAAA,GAAW,MAAM,IAAA,CAAKK,WAAAA,EAAW;AACvC,IAAA,OAAOL,SAASyB,IAAAA,EAAI;AACtB;EAEA,MAAMQ,GAAAA,CAAIC,KAAaC,KAAAA,EAAU;AAC/B,IAAA,MAAMnC,QAAAA,GAAW,MAAM,IAAA,CAAKK,WAAAA,EAAW;AACvC,IAAA,MAAML,QAAAA,CAASiC,GAAAA,CAAIC,GAAAA,EAAKC,KAAAA,CAAAA;AACxB,IAAA,KAAKnC,QAAAA,CAASoC,GAAAA,CAAIF,GAAAA,CAAAA,CAAKG,QAAQ,MAAA;AAC7B,MAAA,KAAK,KAAKL,IAAAA,EAAI;KAChB,CAAA;AACF;AAEA,EAAA,MAAMI,IAAIF,GAAAA,EAAa;AACrB,IAAA,MAAMlC,QAAAA,GAAW,MAAM,IAAA,CAAKK,WAAAA,EAAW;AACvC,IAAA,OAAO,MAAML,QAAAA,CAASoC,GAAAA,CAAIF,GAAAA,CAAAA;AAC5B;AAEA,EAAA,MAAMI,IAAIJ,GAAAA,EAAa;AACrB,IAAA,MAAMlC,QAAAA,GAAW,MAAM,IAAA,CAAKK,WAAAA,EAAW;AACvC,IAAA,OAAO,MAAML,QAAAA,CAASsC,GAAAA,CAAIJ,GAAAA,CAAAA;AAC5B;AAEA,EAAA,MAAMK,OAAOL,GAAAA,EAAa;AACxB,IAAA,MAAMlC,QAAAA,GAAW,MAAM,IAAA,CAAKK,WAAAA,EAAW;AACvC,IAAA,MAAMmC,MAAAA,GAAS,MAAMxC,QAAAA,CAASuC,MAAAA,CAAOL,GAAAA,CAAAA;AACrC,IAAA,MAAM,KAAKF,IAAAA,EAAI;AACf,IAAA,OAAOQ,MAAAA;AACT;AAEA,EAAA,MAAMT,KAAAA,GAAQ;AACZ,IAAA,MAAM/B,QAAAA,GAAW,MAAM,IAAA,CAAKK,WAAAA,EAAW;AACvC,IAAA,MAAML,SAAS+B,KAAAA,EAAK;AACpB,IAAA,MAAM,KAAKC,IAAAA,EAAI;AACjB;AAEA,EAAA,MAAMS,cAAAA,GAAiB;AACrB,IAAA,OAAO;MACL9C,KAAAA,EAAO;AAAEG,QAAAA,QAAAA,EAAU,KAAKH,KAAAA,CAAMG;AAAS,OAAA;MACvCE,QAAAA,EAAU,MAAM,KAAKK,WAAAA;AACvB,KAAA;AACF;AAEAqC,EAAAA,YAAAA,CAAa3B,QAAAA,EAAiE;AAC5E4B,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAM7B,QAAAA,CAAAA;AACtB;AACF","file":"fileCache.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { BaseCache } from \"@/cache/base.js\";\nimport fs from \"node:fs\";\nimport { SlidingCache } from \"@/cache/slidingCache.js\";\nimport { Cache } from \"@/cache/decoratorCache.js\";\nimport { Serializer } from \"@/serializer/serializer.js\";\nimport { SerializableClass } from \"@/internals/serializable.js\";\n\ninterface Input {\n  fullPath: string;\n}\n\nexport class FileCache<T> extends BaseCache<T> {\n  constructor(protected readonly input: Input) {\n    super();\n  }\n\n  static {\n    this.register();\n  }\n\n  get source() {\n    return this.input.fullPath;\n  }\n\n  static async fromProvider<A>(provider: BaseCache<A>, input: Input) {\n    await fs.promises.writeFile(input.fullPath, await provider.serialize());\n    return new FileCache<A>(input);\n  }\n\n  @Cache()\n  protected async getProvider(): Promise<BaseCache<T>> {\n    const exists = await fs.promises\n      .stat(this.input.fullPath)\n      .then((r) => r.isFile())\n      .catch(() => false);\n\n    if (exists) {\n      const serialized = await fs.promises.readFile(this.input.fullPath, \"utf8\");\n      const { target, snapshot } = await Serializer.deserialize<any>(serialized);\n      const Target = Serializer.getFactory(target).ref as SerializableClass<BaseCache<T>>;\n      const instance = await Target.fromSnapshot(snapshot);\n      if (!(instance instanceof BaseCache)) {\n        throw new TypeError(\"Provided file does not serialize any instance of BaseCache class.\");\n      }\n      return instance;\n    } else {\n      return new SlidingCache({\n        size: Infinity,\n        ttl: Infinity,\n      });\n    }\n  }\n\n  async reload() {\n    // @ts-expect-error protected property\n    Cache.getInstance(this, \"getProvider\").clear();\n    void (await this.getProvider());\n  }\n\n  protected async save() {\n    const provider = await this.getProvider();\n    return await fs.promises.writeFile(this.input.fullPath, await provider.serialize());\n  }\n\n  async size() {\n    const provider = await this.getProvider();\n    return provider.size();\n  }\n\n  async set(key: string, value: T) {\n    const provider = await this.getProvider();\n    await provider.set(key, value);\n    void provider.get(key).finally(() => {\n      void this.save();\n    });\n  }\n\n  async get(key: string) {\n    const provider = await this.getProvider();\n    return await provider.get(key);\n  }\n\n  async has(key: string) {\n    const provider = await this.getProvider();\n    return await provider.has(key);\n  }\n\n  async delete(key: string) {\n    const provider = await this.getProvider();\n    const result = await provider.delete(key);\n    await this.save();\n    return result;\n  }\n\n  async clear() {\n    const provider = await this.getProvider();\n    await provider.clear();\n    await this.save();\n  }\n\n  async createSnapshot() {\n    return {\n      input: { fullPath: this.input.fullPath },\n      provider: await this.getProvider(),\n    };\n  }\n\n  loadSnapshot(snapshot: Awaited<ReturnType<typeof this.createSnapshot>>): void {\n    Object.assign(this, snapshot);\n  }\n}\n"]}

package/dist/{chat-W9j8T6UU.d.ts → chat-BbCYMAdE.d.ts}

@@ -2,7 +2,7 @@ import { Message, AssistantMessage, MessageContentPart } from './backend/message
 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-C9nwGLt6.js';
+import { C as Callback, E as Emitter } from './emitter-CN7I4uSV.js';
 import { FrameworkError } from './errors.js';
 import { Task } from 'promise-based-task';
 import { BaseCache } from './cache/base.js';
@@ -47,7 +47,7 @@ interface ResponseObjectJson {
     description?: string;
 }
 interface ChatModelObjectInput<T> extends ChatModelParameters {
-    schema: z.ZodSchema<T> | ResponseObjectJson;
+    schema: z.ZodSchema<T, any, any> | ResponseObjectJson;
     systemPromptTemplate?: PromptTemplate<ZodType<{
         schema: string;
     }>>;
@@ -73,6 +73,8 @@ interface ChatModelUsage {
     promptTokens: number;
     completionTokens: number;
     totalTokens: number;
+    reasoningTokens?: number;
+    cachedPromptTokens?: number;
 }
 interface ChatModelEvents {
     newToken?: Callback<{

package/dist/{chat-CR4yfOR9.d.cts → chat-DB_cf15S.d.cts}

@@ -2,7 +2,7 @@ import { Message, AssistantMessage, MessageContentPart } from './backend/message
 import * as ai from 'ai';
 import { Run, GetRunContext } from './context.cjs';
 import { Serializable } from './internals/serializable.cjs';
-import { C as Callback, E as Emitter } from './emitter-DWEsd-ud.cjs';
+import { C as Callback, E as Emitter } from './emitter-CGKd_-kK.cjs';
 import { FrameworkError } from './errors.cjs';
 import { Task } from 'promise-based-task';
 import { BaseCache } from './cache/base.cjs';
@@ -47,7 +47,7 @@ interface ResponseObjectJson {
     description?: string;
 }
 interface ChatModelObjectInput<T> extends ChatModelParameters {
-    schema: z.ZodSchema<T> | ResponseObjectJson;
+    schema: z.ZodSchema<T, any, any> | ResponseObjectJson;
     systemPromptTemplate?: PromptTemplate<ZodType<{
         schema: string;
     }>>;
@@ -73,6 +73,8 @@ interface ChatModelUsage {
     promptTokens: number;
     completionTokens: number;
     totalTokens: number;
+    reasoningTokens?: number;
+    cachedPromptTokens?: number;
 }
 interface ChatModelEvents {
     newToken?: Callback<{

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-DWEsd-ud.cjs';
+import { E as Emitter, C as Callback } from './emitter-CGKd_-kK.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 };