beeai-framework 0.1.21 → 0.1.22

package/dist/agents/requirement/utils/toolCallChecker.cjs

@@ -0,0 +1,61 @@
+'use strict';
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+class ToolCallChecker {
+  static {
+    __name(this, "ToolCallChecker");
+  }
+  config;
+  strikeHistory = [];
+  occurrencesHistory = [];
+  cycleFound = false;
+  enabled = true;
+  constructor(config = {}) {
+    this.config = {
+      maxStrikeLength: config.maxStrikeLength ?? 1,
+      maxTotalOccurrences: config.maxTotalOccurrences ?? 5,
+      windowSize: config.windowSize ?? 10
+    };
+  }
+  register(value) {
+    if (!this.enabled) {
+      return;
+    }
+    const lastCall = this.strikeHistory.at(-1);
+    if (lastCall && this.isSameToolCall(lastCall, value)) {
+      this.strikeHistory.push(value);
+      if (this.strikeHistory.length > this.config.maxStrikeLength) {
+        this.cycleFound = true;
+      }
+    } else {
+      this.strikeHistory = [
+        value
+      ];
+    }
+    this.occurrencesHistory.push(value);
+    if (this.occurrencesHistory.length > this.config.windowSize) {
+      this.occurrencesHistory.shift();
+    }
+    const occurrences = this.occurrencesHistory.filter((call) => this.isSameToolCall(call, value)).length;
+    if (occurrences > this.config.maxTotalOccurrences) {
+      this.cycleFound = true;
+    }
+  }
+  reset(current) {
+    this.strikeHistory = current ? [
+      current
+    ] : [];
+    this.occurrencesHistory = current ? [
+      current
+    ] : [];
+    this.cycleFound = false;
+  }
+  isSameToolCall(a, b) {
+    return a.toolName === b.toolName && JSON.stringify(a.input) === JSON.stringify(b.input);
+  }
+}
+
+exports.ToolCallChecker = ToolCallChecker;
+//# sourceMappingURL=toolCallChecker.cjs.map
+//# sourceMappingURL=toolCallChecker.cjs.map

package/dist/agents/requirement/utils/toolCallChecker.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../src/agents/requirement/utils/toolCallChecker.ts"],"names":["ToolCallChecker","config","strikeHistory","occurrencesHistory","cycleFound","enabled","maxStrikeLength","maxTotalOccurrences","windowSize","register","value","lastCall","at","isSameToolCall","push","length","shift","occurrences","filter","call","reset","current","a","b","toolName","JSON","stringify","input"],"mappings":";;;;AAgBO,MAAMA,eAAAA,CAAAA;EAhBb;;;AAiBYC,EAAAA,MAAAA;AACAC,EAAAA,aAAAA,GAAgC,EAAA;AAChCC,EAAAA,kBAAAA,GAAqC,EAAA;EACxCC,UAAAA,GAAa,KAAA;EACbC,OAAAA,GAAU,IAAA;EAEjB,WAAA,CAAYJ,MAAAA,GAAgC,EAAC,EAAG;AAC9C,IAAA,IAAA,CAAKA,MAAAA,GAAS;AACZK,MAAAA,eAAAA,EAAiBL,OAAOK,eAAAA,IAAmB,CAAA;AAC3CC,MAAAA,mBAAAA,EAAqBN,OAAOM,mBAAAA,IAAuB,CAAA;AACnDC,MAAAA,UAAAA,EAAYP,OAAOO,UAAAA,IAAc;AACnC,KAAA;AACF;AAEAC,EAAAA,QAAAA,CAASC,KAAAA,EAA2B;AAClC,IAAA,IAAI,CAAC,KAAKL,OAAAA,EAAS;AACjB,MAAA;AACF;AAGA,IAAA,MAAMM,QAAAA,GAAW,IAAA,CAAKT,aAAAA,CAAcU,EAAAA,CAAG,EAAC,CAAA;AACxC,IAAA,IAAID,QAAAA,IAAY,IAAA,CAAKE,cAAAA,CAAeF,QAAAA,EAAUD,KAAAA,CAAAA,EAAQ;AACpD,MAAA,IAAA,CAAKR,aAAAA,CAAcY,KAAKJ,KAAAA,CAAAA;AACxB,MAAA,IAAI,IAAA,CAAKR,aAAAA,CAAca,MAAAA,GAAS,IAAA,CAAKd,OAAOK,eAAAA,EAAiB;AAC3D,QAAA,IAAA,CAAKF,UAAAA,GAAa,IAAA;AACpB;KACF,MAAO;AACL,MAAA,IAAA,CAAKF,aAAAA,GAAgB;AAACQ,QAAAA;;AACxB;AAGA,IAAA,IAAA,CAAKP,kBAAAA,CAAmBW,KAAKJ,KAAAA,CAAAA;AAC7B,IAAA,IAAI,IAAA,CAAKP,kBAAAA,CAAmBY,MAAAA,GAAS,IAAA,CAAKd,OAAOO,UAAAA,EAAY;AAC3D,MAAA,IAAA,CAAKL,mBAAmBa,KAAAA,EAAK;AAC/B;AAEA,IAAA,MAAMC,WAAAA,GAAc,IAAA,CAAKd,kBAAAA,CAAmBe,MAAAA,CAAO,CAACC,IAAAA,KAClD,IAAA,CAAKN,cAAAA,CAAeM,IAAAA,EAAMT,KAAAA,CAAAA,CAAAA,CAC1BK,MAAAA;AACF,IAAA,IAAIE,WAAAA,GAAc,IAAA,CAAKhB,MAAAA,CAAOM,mBAAAA,EAAqB;AACjD,MAAA,IAAA,CAAKH,UAAAA,GAAa,IAAA;AACpB;AACF;AAEAgB,EAAAA,KAAAA,CAAMC,OAAAA,EAA8B;AAClC,IAAA,IAAA,CAAKnB,gBAAgBmB,OAAAA,GAAU;AAACA,MAAAA;QAAW,EAAA;AAC3C,IAAA,IAAA,CAAKlB,qBAAqBkB,OAAAA,GAAU;AAACA,MAAAA;QAAW,EAAA;AAChD,IAAA,IAAA,CAAKjB,UAAAA,GAAa,KAAA;AACpB;AAEUS,EAAAA,cAAAA,CAAeS,GAAiBC,CAAAA,EAA0B;AAClE,IAAA,OAAOD,CAAAA,CAAEE,QAAAA,KAAaD,CAAAA,CAAEC,QAAAA,IAAYC,IAAAA,CAAKC,SAAAA,CAAUJ,CAAAA,CAAEK,KAAK,CAAA,KAAMF,IAAAA,CAAKC,SAAAA,CAAUH,CAAAA,CAAEI,KAAK,CAAA;AACxF;AACF","file":"toolCallChecker.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { ToolCallPart } from \"ai\";\n\nexport interface ToolCallCheckerConfig {\n  maxStrikeLength?: number;\n  maxTotalOccurrences?: number;\n  windowSize?: number;\n}\n\n/**\n * Detects cycles in tool calls\n */\nexport class ToolCallChecker {\n  protected config: Required<ToolCallCheckerConfig>;\n  protected strikeHistory: ToolCallPart[] = [];\n  protected occurrencesHistory: ToolCallPart[] = [];\n  public cycleFound = false;\n  public enabled = true;\n\n  constructor(config: ToolCallCheckerConfig = {}) {\n    this.config = {\n      maxStrikeLength: config.maxStrikeLength ?? 1,\n      maxTotalOccurrences: config.maxTotalOccurrences ?? 5,\n      windowSize: config.windowSize ?? 10,\n    };\n  }\n\n  register(value: ToolCallPart): void {\n    if (!this.enabled) {\n      return;\n    }\n\n    // Check for consecutive strikes\n    const lastCall = this.strikeHistory.at(-1);\n    if (lastCall && this.isSameToolCall(lastCall, value)) {\n      this.strikeHistory.push(value);\n      if (this.strikeHistory.length > this.config.maxStrikeLength) {\n        this.cycleFound = true;\n      }\n    } else {\n      this.strikeHistory = [value];\n    }\n\n    // Check for total occurrences in window\n    this.occurrencesHistory.push(value);\n    if (this.occurrencesHistory.length > this.config.windowSize) {\n      this.occurrencesHistory.shift();\n    }\n\n    const occurrences = this.occurrencesHistory.filter((call) =>\n      this.isSameToolCall(call, value),\n    ).length;\n    if (occurrences > this.config.maxTotalOccurrences) {\n      this.cycleFound = true;\n    }\n  }\n\n  reset(current?: ToolCallPart): void {\n    this.strikeHistory = current ? [current] : [];\n    this.occurrencesHistory = current ? [current] : [];\n    this.cycleFound = false;\n  }\n\n  protected isSameToolCall(a: ToolCallPart, b: ToolCallPart): boolean {\n    return a.toolName === b.toolName && JSON.stringify(a.input) === JSON.stringify(b.input);\n  }\n}\n"]}

package/dist/agents/requirement/utils/toolCallChecker.d.cts

@@ -0,0 +1,28 @@
+import { ToolCallPart } from 'ai';
+
+/**
+ * Copyright 2025 © BeeAI a Series of LF Projects, LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+interface ToolCallCheckerConfig {
+    maxStrikeLength?: number;
+    maxTotalOccurrences?: number;
+    windowSize?: number;
+}
+/**
+ * Detects cycles in tool calls
+ */
+declare class ToolCallChecker {
+    protected config: Required<ToolCallCheckerConfig>;
+    protected strikeHistory: ToolCallPart[];
+    protected occurrencesHistory: ToolCallPart[];
+    cycleFound: boolean;
+    enabled: boolean;
+    constructor(config?: ToolCallCheckerConfig);
+    register(value: ToolCallPart): void;
+    reset(current?: ToolCallPart): void;
+    protected isSameToolCall(a: ToolCallPart, b: ToolCallPart): boolean;
+}
+
+export { ToolCallChecker, type ToolCallCheckerConfig };

package/dist/agents/requirement/utils/toolCallChecker.d.ts

@@ -0,0 +1,28 @@
+import { ToolCallPart } from 'ai';
+
+/**
+ * Copyright 2025 © BeeAI a Series of LF Projects, LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+interface ToolCallCheckerConfig {
+    maxStrikeLength?: number;
+    maxTotalOccurrences?: number;
+    windowSize?: number;
+}
+/**
+ * Detects cycles in tool calls
+ */
+declare class ToolCallChecker {
+    protected config: Required<ToolCallCheckerConfig>;
+    protected strikeHistory: ToolCallPart[];
+    protected occurrencesHistory: ToolCallPart[];
+    cycleFound: boolean;
+    enabled: boolean;
+    constructor(config?: ToolCallCheckerConfig);
+    register(value: ToolCallPart): void;
+    reset(current?: ToolCallPart): void;
+    protected isSameToolCall(a: ToolCallPart, b: ToolCallPart): boolean;
+}
+
+export { ToolCallChecker, type ToolCallCheckerConfig };

package/dist/agents/requirement/utils/toolCallChecker.js

@@ -0,0 +1,59 @@
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+class ToolCallChecker {
+  static {
+    __name(this, "ToolCallChecker");
+  }
+  config;
+  strikeHistory = [];
+  occurrencesHistory = [];
+  cycleFound = false;
+  enabled = true;
+  constructor(config = {}) {
+    this.config = {
+      maxStrikeLength: config.maxStrikeLength ?? 1,
+      maxTotalOccurrences: config.maxTotalOccurrences ?? 5,
+      windowSize: config.windowSize ?? 10
+    };
+  }
+  register(value) {
+    if (!this.enabled) {
+      return;
+    }
+    const lastCall = this.strikeHistory.at(-1);
+    if (lastCall && this.isSameToolCall(lastCall, value)) {
+      this.strikeHistory.push(value);
+      if (this.strikeHistory.length > this.config.maxStrikeLength) {
+        this.cycleFound = true;
+      }
+    } else {
+      this.strikeHistory = [
+        value
+      ];
+    }
+    this.occurrencesHistory.push(value);
+    if (this.occurrencesHistory.length > this.config.windowSize) {
+      this.occurrencesHistory.shift();
+    }
+    const occurrences = this.occurrencesHistory.filter((call) => this.isSameToolCall(call, value)).length;
+    if (occurrences > this.config.maxTotalOccurrences) {
+      this.cycleFound = true;
+    }
+  }
+  reset(current) {
+    this.strikeHistory = current ? [
+      current
+    ] : [];
+    this.occurrencesHistory = current ? [
+      current
+    ] : [];
+    this.cycleFound = false;
+  }
+  isSameToolCall(a, b) {
+    return a.toolName === b.toolName && JSON.stringify(a.input) === JSON.stringify(b.input);
+  }
+}
+
+export { ToolCallChecker };
+//# sourceMappingURL=toolCallChecker.js.map
+//# sourceMappingURL=toolCallChecker.js.map

package/dist/agents/requirement/utils/toolCallChecker.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../src/agents/requirement/utils/toolCallChecker.ts"],"names":["ToolCallChecker","config","strikeHistory","occurrencesHistory","cycleFound","enabled","maxStrikeLength","maxTotalOccurrences","windowSize","register","value","lastCall","at","isSameToolCall","push","length","shift","occurrences","filter","call","reset","current","a","b","toolName","JSON","stringify","input"],"mappings":";;AAgBO,MAAMA,eAAAA,CAAAA;EAhBb;;;AAiBYC,EAAAA,MAAAA;AACAC,EAAAA,aAAAA,GAAgC,EAAA;AAChCC,EAAAA,kBAAAA,GAAqC,EAAA;EACxCC,UAAAA,GAAa,KAAA;EACbC,OAAAA,GAAU,IAAA;EAEjB,WAAA,CAAYJ,MAAAA,GAAgC,EAAC,EAAG;AAC9C,IAAA,IAAA,CAAKA,MAAAA,GAAS;AACZK,MAAAA,eAAAA,EAAiBL,OAAOK,eAAAA,IAAmB,CAAA;AAC3CC,MAAAA,mBAAAA,EAAqBN,OAAOM,mBAAAA,IAAuB,CAAA;AACnDC,MAAAA,UAAAA,EAAYP,OAAOO,UAAAA,IAAc;AACnC,KAAA;AACF;AAEAC,EAAAA,QAAAA,CAASC,KAAAA,EAA2B;AAClC,IAAA,IAAI,CAAC,KAAKL,OAAAA,EAAS;AACjB,MAAA;AACF;AAGA,IAAA,MAAMM,QAAAA,GAAW,IAAA,CAAKT,aAAAA,CAAcU,EAAAA,CAAG,EAAC,CAAA;AACxC,IAAA,IAAID,QAAAA,IAAY,IAAA,CAAKE,cAAAA,CAAeF,QAAAA,EAAUD,KAAAA,CAAAA,EAAQ;AACpD,MAAA,IAAA,CAAKR,aAAAA,CAAcY,KAAKJ,KAAAA,CAAAA;AACxB,MAAA,IAAI,IAAA,CAAKR,aAAAA,CAAca,MAAAA,GAAS,IAAA,CAAKd,OAAOK,eAAAA,EAAiB;AAC3D,QAAA,IAAA,CAAKF,UAAAA,GAAa,IAAA;AACpB;KACF,MAAO;AACL,MAAA,IAAA,CAAKF,aAAAA,GAAgB;AAACQ,QAAAA;;AACxB;AAGA,IAAA,IAAA,CAAKP,kBAAAA,CAAmBW,KAAKJ,KAAAA,CAAAA;AAC7B,IAAA,IAAI,IAAA,CAAKP,kBAAAA,CAAmBY,MAAAA,GAAS,IAAA,CAAKd,OAAOO,UAAAA,EAAY;AAC3D,MAAA,IAAA,CAAKL,mBAAmBa,KAAAA,EAAK;AAC/B;AAEA,IAAA,MAAMC,WAAAA,GAAc,IAAA,CAAKd,kBAAAA,CAAmBe,MAAAA,CAAO,CAACC,IAAAA,KAClD,IAAA,CAAKN,cAAAA,CAAeM,IAAAA,EAAMT,KAAAA,CAAAA,CAAAA,CAC1BK,MAAAA;AACF,IAAA,IAAIE,WAAAA,GAAc,IAAA,CAAKhB,MAAAA,CAAOM,mBAAAA,EAAqB;AACjD,MAAA,IAAA,CAAKH,UAAAA,GAAa,IAAA;AACpB;AACF;AAEAgB,EAAAA,KAAAA,CAAMC,OAAAA,EAA8B;AAClC,IAAA,IAAA,CAAKnB,gBAAgBmB,OAAAA,GAAU;AAACA,MAAAA;QAAW,EAAA;AAC3C,IAAA,IAAA,CAAKlB,qBAAqBkB,OAAAA,GAAU;AAACA,MAAAA;QAAW,EAAA;AAChD,IAAA,IAAA,CAAKjB,UAAAA,GAAa,KAAA;AACpB;AAEUS,EAAAA,cAAAA,CAAeS,GAAiBC,CAAAA,EAA0B;AAClE,IAAA,OAAOD,CAAAA,CAAEE,QAAAA,KAAaD,CAAAA,CAAEC,QAAAA,IAAYC,IAAAA,CAAKC,SAAAA,CAAUJ,CAAAA,CAAEK,KAAK,CAAA,KAAMF,IAAAA,CAAKC,SAAAA,CAAUH,CAAAA,CAAEI,KAAK,CAAA;AACxF;AACF","file":"toolCallChecker.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { ToolCallPart } from \"ai\";\n\nexport interface ToolCallCheckerConfig {\n  maxStrikeLength?: number;\n  maxTotalOccurrences?: number;\n  windowSize?: number;\n}\n\n/**\n * Detects cycles in tool calls\n */\nexport class ToolCallChecker {\n  protected config: Required<ToolCallCheckerConfig>;\n  protected strikeHistory: ToolCallPart[] = [];\n  protected occurrencesHistory: ToolCallPart[] = [];\n  public cycleFound = false;\n  public enabled = true;\n\n  constructor(config: ToolCallCheckerConfig = {}) {\n    this.config = {\n      maxStrikeLength: config.maxStrikeLength ?? 1,\n      maxTotalOccurrences: config.maxTotalOccurrences ?? 5,\n      windowSize: config.windowSize ?? 10,\n    };\n  }\n\n  register(value: ToolCallPart): void {\n    if (!this.enabled) {\n      return;\n    }\n\n    // Check for consecutive strikes\n    const lastCall = this.strikeHistory.at(-1);\n    if (lastCall && this.isSameToolCall(lastCall, value)) {\n      this.strikeHistory.push(value);\n      if (this.strikeHistory.length > this.config.maxStrikeLength) {\n        this.cycleFound = true;\n      }\n    } else {\n      this.strikeHistory = [value];\n    }\n\n    // Check for total occurrences in window\n    this.occurrencesHistory.push(value);\n    if (this.occurrencesHistory.length > this.config.windowSize) {\n      this.occurrencesHistory.shift();\n    }\n\n    const occurrences = this.occurrencesHistory.filter((call) =>\n      this.isSameToolCall(call, value),\n    ).length;\n    if (occurrences > this.config.maxTotalOccurrences) {\n      this.cycleFound = true;\n    }\n  }\n\n  reset(current?: ToolCallPart): void {\n    this.strikeHistory = current ? [current] : [];\n    this.occurrencesHistory = current ? [current] : [];\n    this.cycleFound = false;\n  }\n\n  protected isSameToolCall(a: ToolCallPart, b: ToolCallPart): boolean {\n    return a.toolName === b.toolName && JSON.stringify(a.input) === JSON.stringify(b.input);\n  }\n}\n"]}

package/dist/agents/toolCalling/agent.d.cts

@@ -1,21 +1,21 @@
-import { b as AgentMeta, a as BaseAgent } from '../../base-CwcxwybB.cjs';
+import { GetRunContext, MiddlewareType } from '../../context.cjs';
+import { b as AgentMeta, a as BaseAgent } from '../../base-v-PYIq5G.cjs';
 import { AnyTool } from '../../tools/base.cjs';
 import { BaseMemory } from '../../memory/base.cjs';
-import { E as Emitter } from '../../emitter-DpqUYjXH.cjs';
+import { E as Emitter } from '../../emitter-CGKd_-kK.cjs';
 import { ToolCallingAgentTemplates, ToolCallingAgentExecutionConfig, ToolCallingAgentRunInput, ToolCallingAgentRunOutput, ToolCallingAgentRunOptions, ToolCallingAgentCallbacks } from './types.cjs';
-import { GetRunContext } from '../../context.cjs';
-import { m as ChatModel } from '../../chat-CRb3vUVg.cjs';
-import '../../errors.cjs';
+import { m as ChatModel } from '../../chat-DB_cf15S.cjs';
+import '../../internals/serializable.cjs';
 import '../../internals/types.cjs';
 import '../../internals/helpers/guards.cjs';
-import '../../internals/serializable.cjs';
+import '../../internals/helpers/promise.cjs';
+import '../../errors.cjs';
 import 'ajv';
 import 'promise-based-task';
 import '../../cache/base.cjs';
 import '../../internals/helpers/schema.cjs';
 import 'zod';
 import 'zod-to-json-schema';
-import '../../internals/helpers/promise.cjs';
 import '../../backend/message.cjs';
 import 'ai';
 import './prompts.cjs';
@@ -24,11 +24,6 @@ import '../../backend/constants.cjs';
 import '../../logger/logger.cjs';
 import 'pino';
 
-/**
- * Copyright 2025 © BeeAI a Series of LF Projects, LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-
 type ToolCallingAgentTemplateFactory<K extends keyof ToolCallingAgentTemplates> = (template: ToolCallingAgentTemplates[K]) => ToolCallingAgentTemplates[K];
 interface ToolCallingAgentInput {
     llm: ChatModel;
@@ -52,6 +47,7 @@ declare class ToolCallingAgent extends BaseAgent<ToolCallingAgentRunInput, ToolC
         input: ToolCallingAgentInput;
         emitter: Emitter<ToolCallingAgentCallbacks>;
         isRunning: boolean;
+        middlewares: MiddlewareType<any>[];
     };
     set memory(memory: BaseMemory);
     get memory(): BaseMemory;

package/dist/agents/toolCalling/agent.d.ts

@@ -1,21 +1,21 @@
-import { b as AgentMeta, a as BaseAgent } from '../../base-BQgDlcGa.js';
+import { GetRunContext, MiddlewareType } from '../../context.js';
+import { b as AgentMeta, a as BaseAgent } from '../../base-BurRTBA7.js';
 import { AnyTool } from '../../tools/base.js';
 import { BaseMemory } from '../../memory/base.js';
-import { E as Emitter } from '../../emitter-CZFbzlUi.js';
+import { E as Emitter } from '../../emitter-CN7I4uSV.js';
 import { ToolCallingAgentTemplates, ToolCallingAgentExecutionConfig, ToolCallingAgentRunInput, ToolCallingAgentRunOutput, ToolCallingAgentRunOptions, ToolCallingAgentCallbacks } from './types.js';
-import { GetRunContext } from '../../context.js';
-import { m as ChatModel } from '../../chat-C0s-o6ll.js';
-import '../../errors.js';
+import { m as ChatModel } from '../../chat-BbCYMAdE.js';
+import '../../internals/serializable.js';
 import '../../internals/types.js';
 import '../../internals/helpers/guards.js';
-import '../../internals/serializable.js';
+import '../../internals/helpers/promise.js';
+import '../../errors.js';
 import 'ajv';
 import 'promise-based-task';
 import '../../cache/base.js';
 import '../../internals/helpers/schema.js';
 import 'zod';
 import 'zod-to-json-schema';
-import '../../internals/helpers/promise.js';
 import '../../backend/message.js';
 import 'ai';
 import './prompts.js';
@@ -24,11 +24,6 @@ import '../../backend/constants.js';
 import '../../logger/logger.js';
 import 'pino';
 
-/**
- * Copyright 2025 © BeeAI a Series of LF Projects, LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-
 type ToolCallingAgentTemplateFactory<K extends keyof ToolCallingAgentTemplates> = (template: ToolCallingAgentTemplates[K]) => ToolCallingAgentTemplates[K];
 interface ToolCallingAgentInput {
     llm: ChatModel;
@@ -52,6 +47,7 @@ declare class ToolCallingAgent extends BaseAgent<ToolCallingAgentRunInput, ToolC
         input: ToolCallingAgentInput;
         emitter: Emitter<ToolCallingAgentCallbacks>;
         isRunning: boolean;
+        middlewares: MiddlewareType<any>[];
     };
     set memory(memory: BaseMemory);
     get memory(): BaseMemory;

package/dist/agents/toolCalling/types.d.cts

@@ -1,6 +1,6 @@
 import { BaseMemory } from '../../memory/base.cjs';
 import { AssistantMessage } from '../../backend/message.cjs';
-import { C as Callback } from '../../emitter-DpqUYjXH.cjs';
+import { C as Callback } from '../../emitter-CGKd_-kK.cjs';
 import { ToolCallingAgentSystemPrompt, ToolCallingAgentTaskPrompt } from './prompts.cjs';
 import { ZodSchema } from 'zod';
 import '../../errors.cjs';

package/dist/agents/toolCalling/types.d.ts

@@ -1,6 +1,6 @@
 import { BaseMemory } from '../../memory/base.js';
 import { AssistantMessage } from '../../backend/message.js';
-import { C as Callback } from '../../emitter-CZFbzlUi.js';
+import { C as Callback } from '../../emitter-CN7I4uSV.js';
 import { ToolCallingAgentSystemPrompt, ToolCallingAgentTaskPrompt } from './prompts.js';
 import { ZodSchema } from 'zod';
 import '../../errors.js';

package/dist/agents/types.d.cts

@@ -1,8 +1,8 @@
-export { e as AgentCallback, d as AgentCallbackValue, b as AgentMeta, c as AnyAgent, G as GetAgentInput, f as GetAgentOutput, I as InternalAgentCallbackValue, P as PublicAgentCallbackValue } from '../base-CwcxwybB.cjs';
+export { e as AgentCallback, d as AgentCallbackValue, b as AgentMeta, c as AnyAgent, G as GetAgentInput, f as GetAgentOutput, I as InternalAgentCallbackValue, P as PublicAgentCallbackValue } from '../base-v-PYIq5G.cjs';
 import '../internals/types.cjs';
 import '../tools/base.cjs';
 import '../context.cjs';
-import '../emitter-DpqUYjXH.cjs';
+import '../emitter-CGKd_-kK.cjs';
 import '../internals/serializable.cjs';
 import '../internals/helpers/guards.cjs';
 import '../internals/helpers/promise.cjs';

package/dist/agents/types.d.ts

@@ -1,8 +1,8 @@
-export { e as AgentCallback, d as AgentCallbackValue, b as AgentMeta, c as AnyAgent, G as GetAgentInput, f as GetAgentOutput, I as InternalAgentCallbackValue, P as PublicAgentCallbackValue } from '../base-BQgDlcGa.js';
+export { e as AgentCallback, d as AgentCallbackValue, b as AgentMeta, c as AnyAgent, G as GetAgentInput, f as GetAgentOutput, I as InternalAgentCallbackValue, P as PublicAgentCallbackValue } from '../base-BurRTBA7.js';
 import '../internals/types.js';
 import '../tools/base.js';
 import '../context.js';
-import '../emitter-CZFbzlUi.js';
+import '../emitter-CN7I4uSV.js';
 import '../internals/serializable.js';
 import '../internals/helpers/guards.js';
 import '../internals/helpers/promise.js';

package/dist/backend/backend.d.cts

@@ -1,12 +1,12 @@
 import { Serializable } from '../internals/serializable.cjs';
-import { m as ChatModel, F as FullModelName } from '../chat-CRb3vUVg.cjs';
+import { m as ChatModel, F as FullModelName } from '../chat-DB_cf15S.cjs';
 import { EmbeddingModel } from './embedding.cjs';
 import { ProviderName } from './constants.cjs';
 import { OptionalExcept } from '../internals/types.cjs';
 import './message.cjs';
 import 'ai';
 import '../context.cjs';
-import '../emitter-DpqUYjXH.cjs';
+import '../emitter-CGKd_-kK.cjs';
 import '../internals/helpers/promise.cjs';
 import '../errors.cjs';
 import 'promise-based-task';

package/dist/backend/backend.d.ts

@@ -1,12 +1,12 @@
 import { Serializable } from '../internals/serializable.js';
-import { m as ChatModel, F as FullModelName } from '../chat-C0s-o6ll.js';
+import { m as ChatModel, F as FullModelName } from '../chat-BbCYMAdE.js';
 import { EmbeddingModel } from './embedding.js';
 import { ProviderName } from './constants.js';
 import { OptionalExcept } from '../internals/types.js';
 import './message.js';
 import 'ai';
 import '../context.js';
-import '../emitter-CZFbzlUi.js';
+import '../emitter-CN7I4uSV.js';
 import '../internals/helpers/promise.js';
 import '../errors.js';
 import 'promise-based-task';

package/dist/backend/chat.d.cts

@@ -2,11 +2,11 @@ import './message.cjs';
 import 'ai';
 import '../context.cjs';
 import '../internals/serializable.cjs';
-import '../emitter-DpqUYjXH.cjs';
+import '../emitter-CGKd_-kK.cjs';
 import '../errors.cjs';
 import 'promise-based-task';
 import '../cache/base.cjs';
-export { k as ChatConfig, m as ChatModel, i as ChatModelCache, h as ChatModelEmitter, g as ChatModelEvents, e as ChatModelFinishReason, d as ChatModelInput, a as ChatModelObjectInput, b as ChatModelObjectOutput, n as ChatModelOutput, C as ChatModelParameters, c as ChatModelToolChoice, l as ChatModelToolChoiceSupport, f as ChatModelUsage, j as ConfigFn } from '../chat-CRb3vUVg.cjs';
+export { k as ChatConfig, m as ChatModel, i as ChatModelCache, h as ChatModelEmitter, g as ChatModelEvents, e as ChatModelFinishReason, d as ChatModelInput, a as ChatModelObjectInput, b as ChatModelObjectOutput, n as ChatModelOutput, C as ChatModelParameters, c as ChatModelToolChoice, l as ChatModelToolChoiceSupport, f as ChatModelUsage, j as ConfigFn } from '../chat-DB_cf15S.cjs';
 import './constants.cjs';
 import '../tools/base.cjs';
 import 'zod';

package/dist/backend/chat.d.ts

@@ -2,11 +2,11 @@ import './message.js';
 import 'ai';
 import '../context.js';
 import '../internals/serializable.js';
-import '../emitter-CZFbzlUi.js';
+import '../emitter-CN7I4uSV.js';
 import '../errors.js';
 import 'promise-based-task';
 import '../cache/base.js';
-export { k as ChatConfig, m as ChatModel, i as ChatModelCache, h as ChatModelEmitter, g as ChatModelEvents, e as ChatModelFinishReason, d as ChatModelInput, a as ChatModelObjectInput, b as ChatModelObjectOutput, n as ChatModelOutput, C as ChatModelParameters, c as ChatModelToolChoice, l as ChatModelToolChoiceSupport, f as ChatModelUsage, j as ConfigFn } from '../chat-C0s-o6ll.js';
+export { k as ChatConfig, m as ChatModel, i as ChatModelCache, h as ChatModelEmitter, g as ChatModelEvents, e as ChatModelFinishReason, d as ChatModelInput, a as ChatModelObjectInput, b as ChatModelObjectOutput, n as ChatModelOutput, C as ChatModelParameters, c as ChatModelToolChoice, l as ChatModelToolChoiceSupport, f as ChatModelUsage, j as ConfigFn } from '../chat-BbCYMAdE.js';
 import './constants.js';
 import '../tools/base.js';
 import 'zod';

package/dist/backend/core.d.cts

@@ -1,5 +1,5 @@
 export { Backend } from './backend.cjs';
-export { k as ChatConfig, m as ChatModel, i as ChatModelCache, h as ChatModelEmitter, g as ChatModelEvents, e as ChatModelFinishReason, d as ChatModelInput, a as ChatModelObjectInput, b as ChatModelObjectOutput, n as ChatModelOutput, C as ChatModelParameters, c as ChatModelToolChoice, l as ChatModelToolChoiceSupport, f as ChatModelUsage, j as ConfigFn } from '../chat-CRb3vUVg.cjs';
+export { k as ChatConfig, m as ChatModel, i as ChatModelCache, h as ChatModelEmitter, g as ChatModelEvents, e as ChatModelFinishReason, d as ChatModelInput, a as ChatModelObjectInput, b as ChatModelObjectOutput, n as ChatModelOutput, C as ChatModelParameters, c as ChatModelToolChoice, l as ChatModelToolChoiceSupport, f as ChatModelUsage, j as ConfigFn } from '../chat-DB_cf15S.cjs';
 export { EmbeddingModel, EmbeddingModelEmitter, EmbeddingModelEvents, EmbeddingModelInput, EmbeddingModelOutput } from './embedding.cjs';
 export { AssistantMessage, CustomMessage, Message, MessageContentPart, MessageInput, MessageMeta, MessageRole, Role, SystemMessage, ToolMessage, UserMessage } from './message.cjs';
 export { BackendError, ChatModelError, EmbeddingModelError } from './errors.cjs';
@@ -9,7 +9,7 @@ import '../internals/helpers/guards.cjs';
 import './constants.cjs';
 import 'ai';
 import '../context.cjs';
-import '../emitter-DpqUYjXH.cjs';
+import '../emitter-CGKd_-kK.cjs';
 import '../internals/helpers/promise.cjs';
 import '../errors.cjs';
 import 'promise-based-task';

package/dist/backend/core.d.ts

@@ -1,5 +1,5 @@
 export { Backend } from './backend.js';
-export { k as ChatConfig, m as ChatModel, i as ChatModelCache, h as ChatModelEmitter, g as ChatModelEvents, e as ChatModelFinishReason, d as ChatModelInput, a as ChatModelObjectInput, b as ChatModelObjectOutput, n as ChatModelOutput, C as ChatModelParameters, c as ChatModelToolChoice, l as ChatModelToolChoiceSupport, f as ChatModelUsage, j as ConfigFn } from '../chat-C0s-o6ll.js';
+export { k as ChatConfig, m as ChatModel, i as ChatModelCache, h as ChatModelEmitter, g as ChatModelEvents, e as ChatModelFinishReason, d as ChatModelInput, a as ChatModelObjectInput, b as ChatModelObjectOutput, n as ChatModelOutput, C as ChatModelParameters, c as ChatModelToolChoice, l as ChatModelToolChoiceSupport, f as ChatModelUsage, j as ConfigFn } from '../chat-BbCYMAdE.js';
 export { EmbeddingModel, EmbeddingModelEmitter, EmbeddingModelEvents, EmbeddingModelInput, EmbeddingModelOutput } from './embedding.js';
 export { AssistantMessage, CustomMessage, Message, MessageContentPart, MessageInput, MessageMeta, MessageRole, Role, SystemMessage, ToolMessage, UserMessage } from './message.js';
 export { BackendError, ChatModelError, EmbeddingModelError } from './errors.js';
@@ -9,7 +9,7 @@ import '../internals/helpers/guards.js';
 import './constants.js';
 import 'ai';
 import '../context.js';
-import '../emitter-CZFbzlUi.js';
+import '../emitter-CN7I4uSV.js';
 import '../internals/helpers/promise.js';
 import '../errors.js';
 import 'promise-based-task';

package/dist/backend/embedding.d.cts

@@ -1,8 +1,8 @@
 import { Run, GetRunContext } from '../context.cjs';
 import { Serializable } from '../internals/serializable.cjs';
-import { C as Callback, E as Emitter } from '../emitter-DpqUYjXH.cjs';
+import { C as Callback, E as Emitter } from '../emitter-CGKd_-kK.cjs';
 import { FrameworkError } from '../errors.cjs';
-import { F as FullModelName } from '../chat-CRb3vUVg.cjs';
+import { F as FullModelName } from '../chat-DB_cf15S.cjs';
 import { ProviderName } from './constants.cjs';
 import '../internals/helpers/promise.cjs';
 import '../internals/types.cjs';

package/dist/backend/embedding.d.ts

@@ -1,8 +1,8 @@
 import { Run, GetRunContext } from '../context.js';
 import { Serializable } from '../internals/serializable.js';
-import { C as Callback, E as Emitter } from '../emitter-CZFbzlUi.js';
+import { C as Callback, E as Emitter } from '../emitter-CN7I4uSV.js';
 import { FrameworkError } from '../errors.js';
-import { F as FullModelName } from '../chat-C0s-o6ll.js';
+import { F as FullModelName } from '../chat-BbCYMAdE.js';
 import { ProviderName } from './constants.js';
 import '../internals/helpers/promise.js';
 import '../internals/types.js';

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-CRb3vUVg.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-DpqUYjXH.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-C0s-o6ll.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-CZFbzlUi.js';
+import '../emitter-CN7I4uSV.js';
 import '../internals/serializable.js';
 import '../internals/helpers/promise.js';
 import '../errors.js';

package/dist/{base-BQgDlcGa.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-CZFbzlUi.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-CwcxwybB.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-DpqUYjXH.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/{chat-C0s-o6ll.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-CZFbzlUi.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';

package/dist/{chat-CRb3vUVg.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-DpqUYjXH.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';

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