beeai-framework 0.1.21 → 0.1.23

package/dist/agents/requirement/prompts.d.cts

@@ -0,0 +1,107 @@
+import * as zod from 'zod';
+import { PromptTemplate } from '../../template.cjs';
+import '../../errors.cjs';
+import '../../internals/types.cjs';
+import '../../internals/helpers/guards.cjs';
+import '../../internals/serializable.cjs';
+import 'ajv';
+
+declare const RequirementAgentSystemPrompt: PromptTemplate<zod.ZodObject<{
+    role: zod.ZodOptional<zod.ZodString>;
+    instructions: zod.ZodOptional<zod.ZodString>;
+    notes: zod.ZodOptional<zod.ZodString>;
+    finalAnswerName: zod.ZodString;
+    finalAnswerSchema: zod.ZodOptional<zod.ZodString>;
+    finalAnswerInstructions: zod.ZodOptional<zod.ZodString>;
+    tools: zod.ZodArray<zod.ZodObject<{
+        name: zod.ZodString;
+        description: zod.ZodString;
+        inputSchema: zod.ZodString;
+        allowed: zod.ZodString;
+        reason: zod.ZodOptional<zod.ZodString>;
+    }, "strip", zod.ZodTypeAny, {
+        name: string;
+        description: string;
+        inputSchema: string;
+        allowed: string;
+        reason?: string | undefined;
+    }, {
+        name: string;
+        description: string;
+        inputSchema: string;
+        allowed: string;
+        reason?: string | undefined;
+    }>, "many">;
+}, "strip", zod.ZodTypeAny, {
+    finalAnswerName: string;
+    tools: {
+        name: string;
+        description: string;
+        inputSchema: string;
+        allowed: string;
+        reason?: string | undefined;
+    }[];
+    role?: string | undefined;
+    instructions?: string | undefined;
+    notes?: string | undefined;
+    finalAnswerSchema?: string | undefined;
+    finalAnswerInstructions?: string | undefined;
+}, {
+    finalAnswerName: string;
+    tools: {
+        name: string;
+        description: string;
+        inputSchema: string;
+        allowed: string;
+        reason?: string | undefined;
+    }[];
+    role?: string | undefined;
+    instructions?: string | undefined;
+    notes?: string | undefined;
+    finalAnswerSchema?: string | undefined;
+    finalAnswerInstructions?: string | undefined;
+}>>;
+declare const RequirementAgentTaskPrompt: PromptTemplate<zod.ZodObject<{
+    prompt: zod.ZodString;
+    context: zod.ZodOptional<zod.ZodString>;
+    expectedOutput: zod.ZodOptional<zod.ZodString>;
+}, "strip", zod.ZodTypeAny, {
+    prompt: string;
+    context?: string | undefined;
+    expectedOutput?: string | undefined;
+}, {
+    prompt: string;
+    context?: string | undefined;
+    expectedOutput?: string | undefined;
+}>>;
+declare const RequirementAgentToolErrorPrompt: PromptTemplate<zod.ZodObject<{
+    reason: zod.ZodString;
+}, "strip", zod.ZodTypeAny, {
+    reason: string;
+}, {
+    reason: string;
+}>>;
+declare const RequirementAgentToolNoResultPrompt: PromptTemplate<zod.ZodObject<{
+    toolCall: zod.ZodObject<{
+        tool: zod.ZodOptional<zod.ZodAny>;
+        input: zod.ZodUnknown;
+    }, "strip", zod.ZodTypeAny, {
+        tool?: any;
+        input?: unknown;
+    }, {
+        tool?: any;
+        input?: unknown;
+    }>;
+}, "strip", zod.ZodTypeAny, {
+    toolCall: {
+        tool?: any;
+        input?: unknown;
+    };
+}, {
+    toolCall: {
+        tool?: any;
+        input?: unknown;
+    };
+}>>;
+
+export { RequirementAgentSystemPrompt, RequirementAgentTaskPrompt, RequirementAgentToolErrorPrompt, RequirementAgentToolNoResultPrompt };

package/dist/agents/requirement/prompts.d.ts

@@ -0,0 +1,107 @@
+import * as zod from 'zod';
+import { PromptTemplate } from '../../template.js';
+import '../../errors.js';
+import '../../internals/types.js';
+import '../../internals/helpers/guards.js';
+import '../../internals/serializable.js';
+import 'ajv';
+
+declare const RequirementAgentSystemPrompt: PromptTemplate<zod.ZodObject<{
+    role: zod.ZodOptional<zod.ZodString>;
+    instructions: zod.ZodOptional<zod.ZodString>;
+    notes: zod.ZodOptional<zod.ZodString>;
+    finalAnswerName: zod.ZodString;
+    finalAnswerSchema: zod.ZodOptional<zod.ZodString>;
+    finalAnswerInstructions: zod.ZodOptional<zod.ZodString>;
+    tools: zod.ZodArray<zod.ZodObject<{
+        name: zod.ZodString;
+        description: zod.ZodString;
+        inputSchema: zod.ZodString;
+        allowed: zod.ZodString;
+        reason: zod.ZodOptional<zod.ZodString>;
+    }, "strip", zod.ZodTypeAny, {
+        name: string;
+        description: string;
+        inputSchema: string;
+        allowed: string;
+        reason?: string | undefined;
+    }, {
+        name: string;
+        description: string;
+        inputSchema: string;
+        allowed: string;
+        reason?: string | undefined;
+    }>, "many">;
+}, "strip", zod.ZodTypeAny, {
+    finalAnswerName: string;
+    tools: {
+        name: string;
+        description: string;
+        inputSchema: string;
+        allowed: string;
+        reason?: string | undefined;
+    }[];
+    role?: string | undefined;
+    instructions?: string | undefined;
+    notes?: string | undefined;
+    finalAnswerSchema?: string | undefined;
+    finalAnswerInstructions?: string | undefined;
+}, {
+    finalAnswerName: string;
+    tools: {
+        name: string;
+        description: string;
+        inputSchema: string;
+        allowed: string;
+        reason?: string | undefined;
+    }[];
+    role?: string | undefined;
+    instructions?: string | undefined;
+    notes?: string | undefined;
+    finalAnswerSchema?: string | undefined;
+    finalAnswerInstructions?: string | undefined;
+}>>;
+declare const RequirementAgentTaskPrompt: PromptTemplate<zod.ZodObject<{
+    prompt: zod.ZodString;
+    context: zod.ZodOptional<zod.ZodString>;
+    expectedOutput: zod.ZodOptional<zod.ZodString>;
+}, "strip", zod.ZodTypeAny, {
+    prompt: string;
+    context?: string | undefined;
+    expectedOutput?: string | undefined;
+}, {
+    prompt: string;
+    context?: string | undefined;
+    expectedOutput?: string | undefined;
+}>>;
+declare const RequirementAgentToolErrorPrompt: PromptTemplate<zod.ZodObject<{
+    reason: zod.ZodString;
+}, "strip", zod.ZodTypeAny, {
+    reason: string;
+}, {
+    reason: string;
+}>>;
+declare const RequirementAgentToolNoResultPrompt: PromptTemplate<zod.ZodObject<{
+    toolCall: zod.ZodObject<{
+        tool: zod.ZodOptional<zod.ZodAny>;
+        input: zod.ZodUnknown;
+    }, "strip", zod.ZodTypeAny, {
+        tool?: any;
+        input?: unknown;
+    }, {
+        tool?: any;
+        input?: unknown;
+    }>;
+}, "strip", zod.ZodTypeAny, {
+    toolCall: {
+        tool?: any;
+        input?: unknown;
+    };
+}, {
+    toolCall: {
+        tool?: any;
+        input?: unknown;
+    };
+}>>;
+
+export { RequirementAgentSystemPrompt, RequirementAgentTaskPrompt, RequirementAgentToolErrorPrompt, RequirementAgentToolNoResultPrompt };

package/dist/agents/requirement/prompts.js

@@ -0,0 +1,90 @@
+import { PromptTemplate } from '../../template.js';
+import { RequirementAgentSystemPromptInputSchema, RequirementAgentTaskPromptInputSchema, RequirementAgentToolErrorPromptInputSchema, RequirementAgentToolNoResultPromptInputSchema } from './types.js';
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+const RequirementAgentSystemPrompt = new PromptTemplate({
+  schema: RequirementAgentSystemPromptInputSchema,
+  template: `# Role
+Assume the role of {{role}}.
+
+# Instructions
+{{#instructions}}
+{{{.}}}
+{{/instructions}}
+When the user sends a message, figure out a solution and provide a final answer to the user by calling the '{{finalAnswerName}}' tool.
+{{#finalAnswerSchema}}
+The final answer must fulfill the following.
+
+\`\`\`
+{{{finalAnswerSchema}}}
+\`\`\`
+{{/finalAnswerSchema}}
+{{#finalAnswerInstructions}}
+{{{finalAnswerInstructions}}}
+{{/finalAnswerInstructions}}
+
+IMPORTANT: The facts mentioned in the final answer must be backed by evidence provided by relevant tool outputs.
+
+# Tools
+You must use a tool to retrieve factual or historical information.
+Never use the tool twice with the same input if not stated otherwise.
+
+{{#tools.0}}
+{{#tools}}
+Name: {{name}}
+Description: {{description}}
+Allowed: {{allowed}}{{#reason}}
+Reason: {{{.}}}{{/reason}}
+
+{{/tools}}
+{{/tools.0}}
+
+# Notes
+- Use markdown syntax to format code snippets, links, JSON, tables, images, and files.
+- If the provided task is unclear, ask the user for clarification.
+- Do not refer to tools or tool outputs by name when responding.
+- Always take it one step at a time. Don't try to do multiple things at once.
+- When the tool doesn't give you what you were asking for, you must either use another tool or a different tool input.
+- You should always try a few different approaches before declaring the problem unsolvable.
+- If you can't fully answer the user's question, answer partially and describe what you couldn't achieve.
+- You cannot do complex calculations, computations, or data manipulations without using tools.
+- The current date and time is: {{formatDate}}
+{{#notes}}
+{{{.}}}
+{{/notes}}`,
+  defaults: {
+    role: "a helpful AI assistant",
+    instructions: ""
+  },
+  functions: {
+    formatDate: /* @__PURE__ */ __name(() => (/* @__PURE__ */ new Date()).toISOString().split("T")[0], "formatDate")
+  }
+});
+const RequirementAgentTaskPrompt = new PromptTemplate({
+  schema: RequirementAgentTaskPromptInputSchema,
+  template: `{{#context}}This is the context relevant to the task:
+{{{.}}}
+
+{{/context}}
+{{#expectedOutput}}
+This is the expected criteria for your output:
+{{.}}
+
+{{/expectedOutput}}
+Your task: {{prompt}}`
+});
+const RequirementAgentToolErrorPrompt = new PromptTemplate({
+  schema: RequirementAgentToolErrorPromptInputSchema,
+  template: `The tool has failed; the error log is shown below. If the tool cannot accomplish what you want, use a different tool or explain why you can't use it.
+
+{{{reason}}}`
+});
+const RequirementAgentToolNoResultPrompt = new PromptTemplate({
+  schema: RequirementAgentToolNoResultPromptInputSchema,
+  template: `No results were found! Try to reformulate your query or use a different tool.`
+});
+
+export { RequirementAgentSystemPrompt, RequirementAgentTaskPrompt, RequirementAgentToolErrorPrompt, RequirementAgentToolNoResultPrompt };
+//# sourceMappingURL=prompts.js.map
+//# sourceMappingURL=prompts.js.map

package/dist/agents/requirement/prompts.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/agents/requirement/prompts.ts"],"names":["RequirementAgentSystemPrompt","PromptTemplate","schema","RequirementAgentSystemPromptInputSchema","template","defaults","role","instructions","functions","formatDate","Date","toISOString","split","RequirementAgentTaskPrompt","RequirementAgentTaskPromptInputSchema","RequirementAgentToolErrorPrompt","RequirementAgentToolErrorPromptInputSchema","RequirementAgentToolNoResultPrompt","RequirementAgentToolNoResultPromptInputSchema"],"mappings":";;;;;AAaO,MAAMA,4BAAAA,GAA+B,IAAIC,cAAAA,CAAe;EAC7DC,MAAAA,EAAQC,uCAAAA;EACRC,QAAAA,EAAU,CAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAgDVC,QAAAA,EAAU;IACRC,IAAAA,EAAM,wBAAA;IACNC,YAAAA,EAAc;AAChB,GAAA;EACAC,SAAAA,EAAW;IACTC,UAAAA,kBAAY,MAAA,CAAA,MAAA,iBAAM,IAAIC,IAAAA,EAAAA,EAAOC,WAAAA,GAAcC,KAAAA,CAAM,GAAA,CAAA,CAAK,CAAA,CAAA,EAA1C,YAAA;AACd;AACF,CAAA;AAEO,MAAMC,0BAAAA,GAA6B,IAAIZ,cAAAA,CAAe;EAC3DC,MAAAA,EAAQY,qCAAAA;EACRV,QAAAA,EAAU,CAAA;;;;;;;;;;AAUZ,CAAA;AAEO,MAAMW,+BAAAA,GAAkC,IAAId,cAAAA,CAAe;EAChEC,MAAAA,EAAQc,0CAAAA;EACRZ,QAAAA,EAAU,CAAA;;;AAGZ,CAAA;AAEO,MAAMa,kCAAAA,GAAqC,IAAIhB,cAAAA,CAAe;EACnEC,MAAAA,EAAQgB,6CAAAA;EACRd,QAAAA,EAAU,CAAA,6EAAA;AACZ,CAAA","file":"prompts.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { PromptTemplate } from \"@/template.js\";\nimport {\n  RequirementAgentSystemPromptInputSchema,\n  RequirementAgentTaskPromptInputSchema,\n  RequirementAgentToolErrorPromptInputSchema,\n  RequirementAgentToolNoResultPromptInputSchema,\n} from \"@/agents/requirement/types.js\";\n\nexport const RequirementAgentSystemPrompt = new PromptTemplate({\n  schema: RequirementAgentSystemPromptInputSchema,\n  template: `# Role\nAssume the role of {{role}}.\n\n# Instructions\n{{#instructions}}\n{{{.}}}\n{{/instructions}}\nWhen the user sends a message, figure out a solution and provide a final answer to the user by calling the '{{finalAnswerName}}' tool.\n{{#finalAnswerSchema}}\nThe final answer must fulfill the following.\n\n\\`\\`\\`\n{{{finalAnswerSchema}}}\n\\`\\`\\`\n{{/finalAnswerSchema}}\n{{#finalAnswerInstructions}}\n{{{finalAnswerInstructions}}}\n{{/finalAnswerInstructions}}\n\nIMPORTANT: The facts mentioned in the final answer must be backed by evidence provided by relevant tool outputs.\n\n# Tools\nYou must use a tool to retrieve factual or historical information.\nNever use the tool twice with the same input if not stated otherwise.\n\n{{#tools.0}}\n{{#tools}}\nName: {{name}}\nDescription: {{description}}\nAllowed: {{allowed}}{{#reason}}\nReason: {{{.}}}{{/reason}}\n\n{{/tools}}\n{{/tools.0}}\n\n# Notes\n- Use markdown syntax to format code snippets, links, JSON, tables, images, and files.\n- If the provided task is unclear, ask the user for clarification.\n- Do not refer to tools or tool outputs by name when responding.\n- Always take it one step at a time. Don't try to do multiple things at once.\n- When the tool doesn't give you what you were asking for, you must either use another tool or a different tool input.\n- You should always try a few different approaches before declaring the problem unsolvable.\n- If you can't fully answer the user's question, answer partially and describe what you couldn't achieve.\n- You cannot do complex calculations, computations, or data manipulations without using tools.\n- The current date and time is: {{formatDate}}\n{{#notes}}\n{{{.}}}\n{{/notes}}`,\n  defaults: {\n    role: \"a helpful AI assistant\",\n    instructions: \"\",\n  },\n  functions: {\n    formatDate: () => new Date().toISOString().split(\"T\")[0],\n  },\n});\n\nexport const RequirementAgentTaskPrompt = new PromptTemplate({\n  schema: RequirementAgentTaskPromptInputSchema,\n  template: `{{#context}}This is the context relevant to the task:\n{{{.}}}\n\n{{/context}}\n{{#expectedOutput}}\nThis is the expected criteria for your output:\n{{.}}\n\n{{/expectedOutput}}\nYour task: {{prompt}}`,\n});\n\nexport const RequirementAgentToolErrorPrompt = new PromptTemplate({\n  schema: RequirementAgentToolErrorPromptInputSchema,\n  template: `The tool has failed; the error log is shown below. If the tool cannot accomplish what you want, use a different tool or explain why you can't use it.\n\n{{{reason}}}`,\n});\n\nexport const RequirementAgentToolNoResultPrompt = new PromptTemplate({\n  schema: RequirementAgentToolNoResultPromptInputSchema,\n  template: `No results were found! Try to reformulate your query or use a different tool.`,\n});\n"]}

package/dist/agents/requirement/requirements/conditional.cjs

@@ -0,0 +1,190 @@
+'use strict';
+
+var requirement_cjs = require('./requirement.cjs');
+var utils_cjs = require('./utils.cjs');
+var errors_cjs = require('../../../errors.cjs');
+var remeda = require('remeda');
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+class ConditionalRequirement extends requirement_cjs.Requirement {
+  static {
+    __name(this, "ConditionalRequirement");
+  }
+  source;
+  sourceTool;
+  before;
+  after;
+  forceAfterSet;
+  minInvocations;
+  maxInvocations;
+  forceAtStepValue;
+  onlySuccessInvocations;
+  consecutiveAllowedValue;
+  customChecks;
+  forcePreventStop;
+  reason;
+  constructor(target, options = {}) {
+    super(options.name || `Condition${utils_cjs.extractTargetName(target)}`);
+    this.source = target;
+    this.enabled = options.enabled ?? true;
+    if (options.priority !== void 0) {
+      this.priority = options.priority;
+    }
+    this.before = utils_cjs.extractTargets(options.onlyBefore);
+    this.after = utils_cjs.extractTargets(options.onlyAfter);
+    this.forceAfterSet = utils_cjs.extractTargets(options.forceAfter);
+    this.minInvocations = options.minInvocations ?? 0;
+    this.maxInvocations = options.maxInvocations ?? Infinity;
+    this.forceAtStepValue = options.forceAtStep;
+    this.onlySuccessInvocations = options.onlySuccessInvocations ?? true;
+    this.consecutiveAllowedValue = options.consecutiveAllowed ?? true;
+    this.customChecks = options.customChecks ?? [];
+    this.forcePreventStop = options.forcePreventStop ?? true;
+    this.reason = options.reason;
+    this.checkInvariant();
+  }
+  checkInvariant() {
+    if (this.minInvocations < 0) {
+      throw new errors_cjs.ValueError("The 'minInvocations' argument must be non negative!");
+    }
+    if (this.maxInvocations < 0) {
+      throw new errors_cjs.ValueError("The 'maxInvocations' argument must be non negative!");
+    }
+    if (this.minInvocations > this.maxInvocations) {
+      throw new errors_cjs.ValueError("The 'minInvocations' argument must be less than or equal to 'maxInvocations'!");
+    }
+    const sourceName = utils_cjs.extractTargetName(this.source);
+    if (this.before.has(this.source)) {
+      throw new errors_cjs.ValueError(`Referencing self in 'before' is not allowed (${sourceName})!`);
+    }
+    if (this.forceAfterSet.has(this.source)) {
+      throw new errors_cjs.ValueError(`Referencing self in 'forceAfter' is not allowed (${sourceName})!`);
+    }
+    const beforeAfterIntersection = new Set([
+      ...this.before
+    ].filter((x) => this.after.has(x)));
+    if (beforeAfterIntersection.size > 0) {
+      throw new errors_cjs.ValueError(`Tool specified as 'before' and 'after' at the same time: ${Array.from(beforeAfterIntersection).join(", ")}!`);
+    }
+    const beforeForceAfterIntersection = new Set([
+      ...this.before
+    ].filter((x) => this.forceAfterSet.has(x)));
+    if (beforeForceAfterIntersection.size > 0) {
+      throw new errors_cjs.ValueError(`Tool specified as 'before' and 'forceAfter' at the same time: ${Array.from(beforeForceAfterIntersection).join(", ")}!`);
+    }
+    if (this.forceAtStepValue !== void 0 && this.forceAtStepValue < 1) {
+      throw new errors_cjs.ValueError("The 'forceAtStep' argument must be >= 1!");
+    }
+  }
+  async init(tools, ctx) {
+    await super.init(tools, ctx);
+    const allTargets = /* @__PURE__ */ new Set([
+      ...this.before,
+      ...this.after,
+      ...this.forceAfterSet,
+      this.source
+    ]);
+    utils_cjs.assertAllRulesFound(allTargets, tools);
+    for (const tool of tools) {
+      if (utils_cjs.targetSeenIn(tool, this.source)) {
+        if (this.sourceTool && this.sourceTool !== tool) {
+          throw new errors_cjs.ValueError(`More than one occurrence of ${utils_cjs.extractTargetName(this.source)} has been found!`);
+        }
+        this.sourceTool = tool;
+      }
+    }
+    if (!this.sourceTool) {
+      throw new errors_cjs.ValueError(`Source tool ${utils_cjs.extractTargetName(this.source)} was not found!`);
+    }
+    if (utils_cjs.targetSeenIn(this.sourceTool, this.before)) {
+      throw new errors_cjs.ValueError(`Referencing self in 'before' is not allowed: ${this.sourceTool.name}!`);
+    }
+    if (this.consecutiveAllowedValue && utils_cjs.targetSeenIn(this.sourceTool, this.forceAfterSet)) {
+      throw new errors_cjs.ValueError(`Referencing self in 'forceAfter' is not allowed: ${this.sourceTool.name}. It would prevent an infinite loop. Consider setting 'consecutiveAllowed' to false.`);
+    }
+  }
+  reset() {
+    this.before.clear();
+    this.after.clear();
+    this.forceAfterSet.clear();
+    return this;
+  }
+  async _run(state, _) {
+    const sourceTool = this.sourceTool;
+    if (!sourceTool) {
+      throw new requirement_cjs.RequirementError("Source was not found!", this);
+    }
+    const steps = this.onlySuccessInvocations ? state.steps.filter((step) => !step.error) : state.steps;
+    const lastStepTool = steps.at(-1)?.tool;
+    const invocations = steps.filter((step) => step.tool === sourceTool).length;
+    const resolve = /* @__PURE__ */ __name((allowed) => {
+      const currentStep = steps.length + 1;
+      if (!allowed && this.forceAtStepValue === currentStep) {
+        throw new requirement_cjs.RequirementError(`Tool '${sourceTool.name}' cannot be executed at step ${this.forceAtStepValue} because it has not met all requirements.`, this);
+      }
+      const forced = allowed ? Boolean(utils_cjs.targetSeenIn(lastStepTool, this.forceAfterSet) || this.forceAtStepValue === currentStep) : false;
+      return [
+        {
+          target: sourceTool.name,
+          allowed,
+          forced,
+          hidden: false,
+          preventStop: this.minInvocations > invocations || forced && this.forcePreventStop,
+          reason: !allowed ? this.reason : void 0
+        }
+      ];
+    }, "resolve");
+    if (!this.consecutiveAllowedValue && sourceTool === lastStepTool) {
+      return resolve(false);
+    }
+    if (invocations >= this.maxInvocations) {
+      return resolve(false);
+    }
+    if (this.after.size > 0) {
+      const stepsAsToolCalls = steps.map((step) => step.tool).filter(remeda.isTruthy);
+      const afterToolsRemaining = new Set(this.after);
+      for (const stepTool of stepsAsToolCalls) {
+        if (utils_cjs.targetSeenIn(stepTool, this.forceAfterSet)) {
+          return resolve(false);
+        }
+        const matcher = utils_cjs.targetSeenIn(stepTool, this.after);
+        if (matcher) {
+          afterToolsRemaining.delete(matcher);
+        }
+      }
+      if (afterToolsRemaining.size > 0) {
+        return resolve(false);
+      }
+    }
+    for (const check of this.customChecks) {
+      if (!check(state)) {
+        return resolve(false);
+      }
+    }
+    return resolve(true);
+  }
+  async clone() {
+    const instance = await super.clone();
+    instance.before = new Set(this.before);
+    instance.after = new Set(this.after);
+    instance.forceAfterSet = new Set(this.forceAfterSet);
+    instance.minInvocations = this.minInvocations;
+    instance.maxInvocations = this.maxInvocations;
+    instance.customChecks = [
+      ...this.customChecks
+    ];
+    instance.onlySuccessInvocations = this.onlySuccessInvocations;
+    instance.forceAtStepValue = this.forceAtStepValue;
+    instance.consecutiveAllowedValue = this.consecutiveAllowedValue;
+    instance.source = this.source;
+    instance.sourceTool = this.sourceTool;
+    instance.forcePreventStop = this.forcePreventStop;
+    instance.reason = this.reason;
+    return instance;
+  }
+}
+
+exports.ConditionalRequirement = ConditionalRequirement;
+//# sourceMappingURL=conditional.cjs.map
+//# sourceMappingURL=conditional.cjs.map

package/dist/agents/requirement/requirements/conditional.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../src/agents/requirement/requirements/conditional.ts"],"names":["ConditionalRequirement","Requirement","source","sourceTool","before","after","forceAfterSet","minInvocations","maxInvocations","forceAtStepValue","onlySuccessInvocations","consecutiveAllowedValue","customChecks","forcePreventStop","reason","target","options","name","extractTargetName","enabled","priority","undefined","extractTargets","onlyBefore","onlyAfter","forceAfter","Infinity","forceAtStep","consecutiveAllowed","checkInvariant","ValueError","sourceName","has","beforeAfterIntersection","Set","filter","x","size","Array","from","join","beforeForceAfterIntersection","init","tools","ctx","allTargets","assertAllRulesFound","tool","targetSeenIn","reset","clear","_run","state","_","RequirementError","steps","step","error","lastStepTool","at","invocations","length","resolve","allowed","currentStep","forced","Boolean","hidden","preventStop","stepsAsToolCalls","map","isTruthy","afterToolsRemaining","stepTool","matcher","delete","check","clone","instance"],"mappings":";;;;;;;;;AAqCO,MAAMA,+BAA+BC,2BAAAA,CAAAA;EArC5C;;;AAsCYC,EAAAA,MAAAA;AACAC,EAAAA,UAAAA;AACAC,EAAAA,MAAAA;AACAC,EAAAA,KAAAA;AACAC,EAAAA,aAAAA;AACAC,EAAAA,cAAAA;AACAC,EAAAA,cAAAA;AACAC,EAAAA,gBAAAA;AACAC,EAAAA,sBAAAA;AACAC,EAAAA,uBAAAA;AACAC,EAAAA,YAAAA;AACAC,EAAAA,gBAAAA;AACAC,EAAAA,MAAAA;EAEV,WAAA,CAAYC,MAAAA,EAAoBC,OAAAA,GAAyC,EAAC,EAAG;AAC3E,IAAA,KAAA,CAAMA,QAAQC,IAAAA,IAAQ,CAAA,SAAA,EAAYC,2BAAAA,CAAkBH,MAAAA,CAAAA,CAAAA,CAAS,CAAA;AAE7D,IAAA,IAAA,CAAKb,MAAAA,GAASa,MAAAA;AACd,IAAA,IAAA,CAAKI,OAAAA,GAAUH,QAAQG,OAAAA,IAAW,IAAA;AAElC,IAAA,IAAIH,OAAAA,CAAQI,aAAaC,MAAAA,EAAW;AAClC,MAAA,IAAA,CAAKD,WAAWJ,OAAAA,CAAQI,QAAAA;AAC1B;AAEA,IAAA,IAAA,CAAKhB,MAAAA,GAASkB,wBAAAA,CAAeN,OAAAA,CAAQO,UAAU,CAAA;AAC/C,IAAA,IAAA,CAAKlB,KAAAA,GAAQiB,wBAAAA,CAAeN,OAAAA,CAAQQ,SAAS,CAAA;AAC7C,IAAA,IAAA,CAAKlB,aAAAA,GAAgBgB,wBAAAA,CAAeN,OAAAA,CAAQS,UAAU,CAAA;AACtD,IAAA,IAAA,CAAKlB,cAAAA,GAAiBS,QAAQT,cAAAA,IAAkB,CAAA;AAChD,IAAA,IAAA,CAAKC,cAAAA,GAAiBQ,QAAQR,cAAAA,IAAkBkB,QAAAA;AAChD,IAAA,IAAA,CAAKjB,mBAAmBO,OAAAA,CAAQW,WAAAA;AAChC,IAAA,IAAA,CAAKjB,sBAAAA,GAAyBM,QAAQN,sBAAAA,IAA0B,IAAA;AAChE,IAAA,IAAA,CAAKC,uBAAAA,GAA0BK,QAAQY,kBAAAA,IAAsB,IAAA;AAC7D,IAAA,IAAA,CAAKhB,YAAAA,GAAeI,OAAAA,CAAQJ,YAAAA,IAAgB,EAAA;AAC5C,IAAA,IAAA,CAAKC,gBAAAA,GAAmBG,QAAQH,gBAAAA,IAAoB,IAAA;AACpD,IAAA,IAAA,CAAKC,SAASE,OAAAA,CAAQF,MAAAA;AAEtB,IAAA,IAAA,CAAKe,cAAAA,EAAc;AACrB;EAEUA,cAAAA,GAAuB;AAC/B,IAAA,IAAI,IAAA,CAAKtB,iBAAiB,CAAA,EAAG;AAC3B,MAAA,MAAM,IAAIuB,sBAAW,qDAAA,CAAA;AACvB;AAEA,IAAA,IAAI,IAAA,CAAKtB,iBAAiB,CAAA,EAAG;AAC3B,MAAA,MAAM,IAAIsB,sBAAW,qDAAA,CAAA;AACvB;AAEA,IAAA,IAAI,IAAA,CAAKvB,cAAAA,GAAiB,IAAA,CAAKC,cAAAA,EAAgB;AAC7C,MAAA,MAAM,IAAIsB,sBACR,+EAAA,CAAA;AAEJ;AAEA,IAAA,MAAMC,UAAAA,GAAab,2BAAAA,CAAkB,IAAA,CAAKhB,MAAM,CAAA;AAChD,IAAA,IAAI,IAAA,CAAKE,MAAAA,CAAO4B,GAAAA,CAAI,IAAA,CAAK9B,MAAM,CAAA,EAAG;AAChC,MAAA,MAAM,IAAI4B,qBAAAA,CAAW,CAAA,6CAAA,EAAgDC,UAAAA,CAAAA,EAAAA,CAAc,CAAA;AACrF;AAEA,IAAA,IAAI,IAAA,CAAKzB,aAAAA,CAAc0B,GAAAA,CAAI,IAAA,CAAK9B,MAAM,CAAA,EAAG;AACvC,MAAA,MAAM,IAAI4B,qBAAAA,CAAW,CAAA,iDAAA,EAAoDC,UAAAA,CAAAA,EAAAA,CAAc,CAAA;AACzF;AAEA,IAAA,MAAME,uBAAAA,GAA0B,IAAIC,GAAAA,CAAI;SAAI,IAAA,CAAK9B;AAAQ+B,KAAAA,CAAAA,MAAAA,CAAO,CAACC,CAAAA,KAAM,IAAA,CAAK/B,MAAM2B,GAAAA,CAAII,CAAAA,CAAAA,CAAAA,CAAAA;AACtF,IAAA,IAAIH,uBAAAA,CAAwBI,OAAO,CAAA,EAAG;AACpC,MAAA,MAAM,IAAIP,qBAAAA,CACR,CAAA,yDAAA,EAA4DQ,KAAAA,CAAMC,IAAAA,CAAKN,uBAAAA,CAAAA,CAAyBO,IAAAA,CAAK,IAAA,CAAA,CAAA,CAAA,CAAQ,CAAA;AAEjH;AAEA,IAAA,MAAMC,4BAAAA,GAA+B,IAAIP,GAAAA,CACvC;SAAI,IAAA,CAAK9B;AAAQ+B,KAAAA,CAAAA,MAAAA,CAAO,CAACC,CAAAA,KAAM,IAAA,CAAK9B,cAAc0B,GAAAA,CAAII,CAAAA,CAAAA,CAAAA,CAAAA;AAExD,IAAA,IAAIK,4BAAAA,CAA6BJ,OAAO,CAAA,EAAG;AACzC,MAAA,MAAM,IAAIP,qBAAAA,CACR,CAAA,8DAAA,EAAiEQ,KAAAA,CAAMC,IAAAA,CAAKE,4BAAAA,CAAAA,CAA8BD,IAAAA,CAAK,IAAA,CAAA,CAAA,CAAA,CAAQ,CAAA;AAE3H;AAEA,IAAA,IAAI,IAAA,CAAK/B,gBAAAA,KAAqBY,MAAAA,IAAa,IAAA,CAAKZ,mBAAmB,CAAA,EAAG;AACpE,MAAA,MAAM,IAAIqB,sBAAW,0CAAA,CAAA;AACvB;AACF;EAEA,MAAMY,IAAAA,CAAKC,OAAkBC,GAAAA,EAAqC;AAChE,IAAA,MAAM,KAAA,CAAMF,IAAAA,CAAKC,KAAAA,EAAOC,GAAAA,CAAAA;AAExB,IAAA,MAAMC,UAAAA,uBAAiBX,GAAAA,CAAI;SAAI,IAAA,CAAK9B,MAAAA;SAAW,IAAA,CAAKC,KAAAA;SAAU,IAAA,CAAKC,aAAAA;MAAe,IAAA,CAAKJ;AAAO,KAAA,CAAA;AAC9F4C,IAAAA,6BAAAA,CAAoBD,YAAYF,KAAAA,CAAAA;AAEhC,IAAA,KAAA,MAAWI,QAAQJ,KAAAA,EAAO;AACxB,MAAA,IAAIK,sBAAAA,CAAaD,IAAAA,EAAM,IAAA,CAAK7C,MAAM,CAAA,EAAG;AACnC,QAAA,IAAI,IAAA,CAAKC,UAAAA,IAAc,IAAA,CAAKA,UAAAA,KAAe4C,IAAAA,EAAM;AAC/C,UAAA,MAAM,IAAIjB,qBAAAA,CACR,CAAA,4BAAA,EAA+BZ,4BAAkB,IAAA,CAAKhB,MAAM,CAAA,CAAA,gBAAA,CAAmB,CAAA;AAEnF;AACA,QAAA,IAAA,CAAKC,UAAAA,GAAa4C,IAAAA;AACpB;AACF;AAEA,IAAA,IAAI,CAAC,KAAK5C,UAAAA,EAAY;AACpB,MAAA,MAAM,IAAI2B,qBAAAA,CAAW,CAAA,YAAA,EAAeZ,4BAAkB,IAAA,CAAKhB,MAAM,CAAA,CAAA,eAAA,CAAkB,CAAA;AACrF;AAEA,IAAA,IAAI8C,sBAAAA,CAAa,IAAA,CAAK7C,UAAAA,EAAY,IAAA,CAAKC,MAAM,CAAA,EAAG;AAC9C,MAAA,MAAM,IAAI0B,qBAAAA,CAAW,CAAA,6CAAA,EAAgD,IAAA,CAAK3B,UAAAA,CAAWc,IAAI,CAAA,CAAA,CAAG,CAAA;AAC9F;AAEA,IAAA,IAAI,KAAKN,uBAAAA,IAA2BqC,sBAAAA,CAAa,KAAK7C,UAAAA,EAAY,IAAA,CAAKG,aAAa,CAAA,EAAG;AACrF,MAAA,MAAM,IAAIwB,qBAAAA,CACR,CAAA,iDAAA,EAAoD,IAAA,CAAK3B,UAAAA,CAAWc,IAAI,CAAA,oFAAA,CACc,CAAA;AAE1F;AACF;EAEAgC,KAAAA,GAAc;AACZ,IAAA,IAAA,CAAK7C,OAAO8C,KAAAA,EAAK;AACjB,IAAA,IAAA,CAAK7C,MAAM6C,KAAAA,EAAK;AAChB,IAAA,IAAA,CAAK5C,cAAc4C,KAAAA,EAAK;AACxB,IAAA,OAAO,IAAA;AACT;EAEA,MAAMC,IAAAA,CAAKC,OAAiCC,CAAAA,EAA6C;AACvF,IAAA,MAAMlD,aAAa,IAAA,CAAKA,UAAAA;AACxB,IAAA,IAAI,CAACA,UAAAA,EAAY;AACf,MAAA,MAAM,IAAImD,gCAAAA,CAAiB,uBAAA,EAAyB,IAAI,CAAA;AAC1D;AAGA,IAAA,MAAMC,KAAAA,GAAQ,IAAA,CAAK7C,sBAAAA,GACf0C,KAAAA,CAAMG,KAAAA,CAAMpB,MAAAA,CAAO,CAACqB,IAAAA,KAAS,CAACA,IAAAA,CAAKC,KAAK,CAAA,GACxCL,KAAAA,CAAMG,KAAAA;AAEV,IAAA,MAAMG,YAAAA,GAAeH,KAAAA,CAAMI,EAAAA,CAAG,EAAC,CAAA,EAAIZ,IAAAA;AAEnC,IAAA,MAAMa,WAAAA,GAAcL,MAAMpB,MAAAA,CAAO,CAACqB,SAASA,IAAAA,CAAKT,IAAAA,KAAS5C,UAAAA,CAAAA,CAAY0D,MAAAA;AAErE,IAAA,MAAMC,OAAAA,2BAAWC,OAAAA,KAAAA;AACf,MAAA,MAAMC,WAAAA,GAAcT,MAAMM,MAAAA,GAAS,CAAA;AACnC,MAAA,IAAI,CAACE,OAAAA,IAAW,IAAA,CAAKtD,gBAAAA,KAAqBuD,WAAAA,EAAa;AACrD,QAAA,MAAM,IAAIV,iCACR,CAAA,MAAA,EAASnD,UAAAA,CAAWc,IAAI,CAAA,6BAAA,EAAgC,IAAA,CAAKR,gBAAgB,CAAA,yCAAA,CAAA,EAE7E,IAAI,CAAA;AAER;AAEA,MAAA,MAAMwD,MAAAA,GAASF,OAAAA,GACXG,OAAAA,CACElB,sBAAAA,CAAaU,YAAAA,EAAc,IAAA,CAAKpD,aAAa,CAAA,IAAK,IAAA,CAAKG,gBAAAA,KAAqBuD,WAAAA,CAAAA,GAE9E,KAAA;AAEJ,MAAA,OAAO;AACL,QAAA;AACEjD,UAAAA,MAAAA,EAAQZ,UAAAA,CAAWc,IAAAA;AACnB8C,UAAAA,OAAAA;AACAE,UAAAA,MAAAA;UACAE,MAAAA,EAAQ,KAAA;AACRC,UAAAA,WAAAA,EAAa,IAAA,CAAK7D,cAAAA,GAAiBqD,WAAAA,IAAgBK,MAAAA,IAAU,IAAA,CAAKpD,gBAAAA;UAClEC,MAAAA,EAAQ,CAACiD,OAAAA,GAAU,IAAA,CAAKjD,MAAAA,GAASO;AACnC;;KAEJ,EA1BgB,SAAA,CAAA;AA6BhB,IAAA,IAAI,CAAC,IAAA,CAAKV,uBAAAA,IAA2BR,UAAAA,KAAeuD,YAAAA,EAAc;AAChE,MAAA,OAAOI,QAAQ,KAAA,CAAA;AACjB;AAGA,IAAA,IAAIF,WAAAA,IAAe,KAAKpD,cAAAA,EAAgB;AACtC,MAAA,OAAOsD,QAAQ,KAAA,CAAA;AACjB;AAGA,IAAA,IAAI,IAAA,CAAKzD,KAAAA,CAAMgC,IAAAA,GAAO,CAAA,EAAG;AACvB,MAAA,MAAMgC,gBAAAA,GAAmBd,MAAMe,GAAAA,CAAI,CAACd,SAASA,IAAAA,CAAKT,IAAI,CAAA,CAAEZ,MAAAA,CAAOoC,eAAAA,CAAAA;AAC/D,MAAA,MAAMC,mBAAAA,GAAsB,IAAItC,GAAAA,CAAI,IAAA,CAAK7B,KAAK,CAAA;AAE9C,MAAA,KAAA,MAAWoE,YAAYJ,gBAAAA,EAAkB;AACvC,QAAA,IAAIrB,sBAAAA,CAAayB,QAAAA,EAAU,IAAA,CAAKnE,aAAa,CAAA,EAAG;AAC9C,UAAA,OAAOwD,QAAQ,KAAA,CAAA;AACjB;AAEA,QAAA,MAAMY,OAAAA,GAAU1B,sBAAAA,CAAayB,QAAAA,EAAiB,IAAA,CAAKpE,KAAK,CAAA;AACxD,QAAA,IAAIqE,OAAAA,EAAS;AACXF,UAAAA,mBAAAA,CAAoBG,OAAOD,OAAAA,CAAAA;AAC7B;AACF;AAEA,MAAA,IAAIF,mBAAAA,CAAoBnC,OAAO,CAAA,EAAG;AAChC,QAAA,OAAOyB,QAAQ,KAAA,CAAA;AACjB;AACF;AAGA,IAAA,KAAA,MAAWc,KAAAA,IAAS,KAAKhE,YAAAA,EAAc;AACrC,MAAA,IAAI,CAACgE,KAAAA,CAAMxB,KAAAA,CAAAA,EAAQ;AACjB,QAAA,OAAOU,QAAQ,KAAA,CAAA;AACjB;AACF;AAEA,IAAA,OAAOA,QAAQ,IAAA,CAAA;AACjB;AAEA,EAAA,MAAMe,KAAAA,GAAuB;AAC3B,IAAA,MAAMC,QAAAA,GAAW,MAAM,KAAA,CAAMD,KAAAA,EAAAA;AAC7BC,IAAAA,QAAAA,CAAS1E,MAAAA,GAAS,IAAI8B,GAAAA,CAAI,IAAA,CAAK9B,MAAM,CAAA;AACrC0E,IAAAA,QAAAA,CAASzE,KAAAA,GAAQ,IAAI6B,GAAAA,CAAI,IAAA,CAAK7B,KAAK,CAAA;AACnCyE,IAAAA,QAAAA,CAASxE,aAAAA,GAAgB,IAAI4B,GAAAA,CAAI,IAAA,CAAK5B,aAAa,CAAA;AACnDwE,IAAAA,QAAAA,CAASvE,iBAAiB,IAAA,CAAKA,cAAAA;AAC/BuE,IAAAA,QAAAA,CAAStE,iBAAiB,IAAA,CAAKA,cAAAA;AAC/BsE,IAAAA,QAAAA,CAASlE,YAAAA,GAAe;SAAI,IAAA,CAAKA;;AACjCkE,IAAAA,QAAAA,CAASpE,yBAAyB,IAAA,CAAKA,sBAAAA;AACvCoE,IAAAA,QAAAA,CAASrE,mBAAmB,IAAA,CAAKA,gBAAAA;AACjCqE,IAAAA,QAAAA,CAASnE,0BAA0B,IAAA,CAAKA,uBAAAA;AACxCmE,IAAAA,QAAAA,CAAS5E,SAAS,IAAA,CAAKA,MAAAA;AACvB4E,IAAAA,QAAAA,CAAS3E,aAAa,IAAA,CAAKA,UAAAA;AAC3B2E,IAAAA,QAAAA,CAASjE,mBAAmB,IAAA,CAAKA,gBAAAA;AACjCiE,IAAAA,QAAAA,CAAShE,SAAS,IAAA,CAAKA,MAAAA;AACvB,IAAA,OAAOgE,QAAAA;AACT;AACF","file":"conditional.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { Requirement, RequirementError, Rule } from \"./requirement.js\";\nimport {\n  TargetType,\n  MultiTargetType,\n  extractTargets,\n  extractTargetName,\n  targetSeenIn,\n  assertAllRulesFound,\n} from \"./utils.js\";\nimport { AnyTool } from \"@/tools/base.js\";\nimport { RunContext } from \"@/context.js\";\nimport { ValueError } from \"@/errors.js\";\nimport type { RequirementAgentRunState } from \"@/agents/requirement/types.js\";\nimport { isTruthy } from \"remeda\";\n\nexport interface ConditionalRequirementOptions {\n  name?: string;\n  forceAtStep?: number;\n  onlyBefore?: MultiTargetType;\n  onlyAfter?: MultiTargetType;\n  forceAfter?: MultiTargetType;\n  forcePreventStop?: boolean;\n  minInvocations?: number;\n  maxInvocations?: number;\n  onlySuccessInvocations?: boolean;\n  consecutiveAllowed?: boolean;\n  priority?: number;\n  customChecks?: ((state: RequirementAgentRunState) => boolean)[];\n  enabled?: boolean;\n  reason?: string;\n}\n\nexport class ConditionalRequirement extends Requirement {\n  protected source: TargetType;\n  protected sourceTool?: AnyTool;\n  protected before: Set<TargetType>;\n  protected after: Set<TargetType>;\n  protected forceAfterSet: Set<TargetType>;\n  protected minInvocations: number;\n  protected maxInvocations: number;\n  protected forceAtStepValue?: number;\n  protected onlySuccessInvocations: boolean;\n  protected consecutiveAllowedValue: boolean;\n  protected customChecks: ((state: RequirementAgentRunState) => boolean)[];\n  protected forcePreventStop: boolean;\n  protected reason?: string;\n\n  constructor(target: TargetType, options: ConditionalRequirementOptions = {}) {\n    super(options.name || `Condition${extractTargetName(target)}`);\n\n    this.source = target;\n    this.enabled = options.enabled ?? true;\n\n    if (options.priority !== undefined) {\n      this.priority = options.priority;\n    }\n\n    this.before = extractTargets(options.onlyBefore);\n    this.after = extractTargets(options.onlyAfter);\n    this.forceAfterSet = extractTargets(options.forceAfter);\n    this.minInvocations = options.minInvocations ?? 0;\n    this.maxInvocations = options.maxInvocations ?? Infinity;\n    this.forceAtStepValue = options.forceAtStep;\n    this.onlySuccessInvocations = options.onlySuccessInvocations ?? true;\n    this.consecutiveAllowedValue = options.consecutiveAllowed ?? true;\n    this.customChecks = options.customChecks ?? [];\n    this.forcePreventStop = options.forcePreventStop ?? true;\n    this.reason = options.reason;\n\n    this.checkInvariant();\n  }\n\n  protected checkInvariant(): void {\n    if (this.minInvocations < 0) {\n      throw new ValueError(\"The 'minInvocations' argument must be non negative!\");\n    }\n\n    if (this.maxInvocations < 0) {\n      throw new ValueError(\"The 'maxInvocations' argument must be non negative!\");\n    }\n\n    if (this.minInvocations > this.maxInvocations) {\n      throw new ValueError(\n        \"The 'minInvocations' argument must be less than or equal to 'maxInvocations'!\",\n      );\n    }\n\n    const sourceName = extractTargetName(this.source);\n    if (this.before.has(this.source)) {\n      throw new ValueError(`Referencing self in 'before' is not allowed (${sourceName})!`);\n    }\n\n    if (this.forceAfterSet.has(this.source)) {\n      throw new ValueError(`Referencing self in 'forceAfter' is not allowed (${sourceName})!`);\n    }\n\n    const beforeAfterIntersection = new Set([...this.before].filter((x) => this.after.has(x)));\n    if (beforeAfterIntersection.size > 0) {\n      throw new ValueError(\n        `Tool specified as 'before' and 'after' at the same time: ${Array.from(beforeAfterIntersection).join(\", \")}!`,\n      );\n    }\n\n    const beforeForceAfterIntersection = new Set(\n      [...this.before].filter((x) => this.forceAfterSet.has(x)),\n    );\n    if (beforeForceAfterIntersection.size > 0) {\n      throw new ValueError(\n        `Tool specified as 'before' and 'forceAfter' at the same time: ${Array.from(beforeForceAfterIntersection).join(\", \")}!`,\n      );\n    }\n\n    if (this.forceAtStepValue !== undefined && this.forceAtStepValue < 1) {\n      throw new ValueError(\"The 'forceAtStep' argument must be >= 1!\");\n    }\n  }\n\n  async init(tools: AnyTool[], ctx: RunContext<any>): Promise<void> {\n    await super.init(tools, ctx);\n\n    const allTargets = new Set([...this.before, ...this.after, ...this.forceAfterSet, this.source]);\n    assertAllRulesFound(allTargets, tools);\n\n    for (const tool of tools) {\n      if (targetSeenIn(tool, this.source)) {\n        if (this.sourceTool && this.sourceTool !== tool) {\n          throw new ValueError(\n            `More than one occurrence of ${extractTargetName(this.source)} has been found!`,\n          );\n        }\n        this.sourceTool = tool;\n      }\n    }\n\n    if (!this.sourceTool) {\n      throw new ValueError(`Source tool ${extractTargetName(this.source)} was not found!`);\n    }\n\n    if (targetSeenIn(this.sourceTool, this.before)) {\n      throw new ValueError(`Referencing self in 'before' is not allowed: ${this.sourceTool.name}!`);\n    }\n\n    if (this.consecutiveAllowedValue && targetSeenIn(this.sourceTool, this.forceAfterSet)) {\n      throw new ValueError(\n        `Referencing self in 'forceAfter' is not allowed: ${this.sourceTool.name}. ` +\n          `It would prevent an infinite loop. Consider setting 'consecutiveAllowed' to false.`,\n      );\n    }\n  }\n\n  reset(): this {\n    this.before.clear();\n    this.after.clear();\n    this.forceAfterSet.clear();\n    return this;\n  }\n\n  async _run(state: RequirementAgentRunState, _: RunContext<typeof this>): Promise<Rule[]> {\n    const sourceTool = this.sourceTool;\n    if (!sourceTool) {\n      throw new RequirementError(\"Source was not found!\", this);\n    }\n\n    // Get steps from state (assuming state has steps property)\n    const steps = this.onlySuccessInvocations\n      ? state.steps.filter((step) => !step.error)\n      : state.steps;\n\n    const lastStepTool = steps.at(-1)?.tool;\n\n    const invocations = steps.filter((step) => step.tool === sourceTool).length;\n\n    const resolve = (allowed: boolean): Rule[] => {\n      const currentStep = steps.length + 1;\n      if (!allowed && this.forceAtStepValue === currentStep) {\n        throw new RequirementError(\n          `Tool '${sourceTool.name}' cannot be executed at step ${this.forceAtStepValue} ` +\n            `because it has not met all requirements.`,\n          this,\n        );\n      }\n\n      const forced = allowed\n        ? Boolean(\n            targetSeenIn(lastStepTool, this.forceAfterSet) || this.forceAtStepValue === currentStep,\n          )\n        : false;\n\n      return [\n        {\n          target: sourceTool.name,\n          allowed,\n          forced,\n          hidden: false,\n          preventStop: this.minInvocations > invocations || (forced && this.forcePreventStop),\n          reason: !allowed ? this.reason : undefined,\n        },\n      ];\n    };\n\n    // Check consecutive constraint\n    if (!this.consecutiveAllowedValue && sourceTool === lastStepTool) {\n      return resolve(false);\n    }\n\n    // Check max invocations\n    if (invocations >= this.maxInvocations) {\n      return resolve(false);\n    }\n\n    // Check after/before constraints\n    if (this.after.size > 0) {\n      const stepsAsToolCalls = steps.map((step) => step.tool).filter(isTruthy);\n      const afterToolsRemaining = new Set(this.after);\n\n      for (const stepTool of stepsAsToolCalls) {\n        if (targetSeenIn(stepTool, this.forceAfterSet)) {\n          return resolve(false);\n        }\n\n        const matcher = targetSeenIn(stepTool as any, this.after);\n        if (matcher) {\n          afterToolsRemaining.delete(matcher);\n        }\n      }\n\n      if (afterToolsRemaining.size > 0) {\n        return resolve(false);\n      }\n    }\n\n    // Check custom checks\n    for (const check of this.customChecks) {\n      if (!check(state)) {\n        return resolve(false);\n      }\n    }\n\n    return resolve(true);\n  }\n\n  async clone(): Promise<this> {\n    const instance = await super.clone();\n    instance.before = new Set(this.before);\n    instance.after = new Set(this.after);\n    instance.forceAfterSet = new Set(this.forceAfterSet);\n    instance.minInvocations = this.minInvocations;\n    instance.maxInvocations = this.maxInvocations;\n    instance.customChecks = [...this.customChecks];\n    instance.onlySuccessInvocations = this.onlySuccessInvocations;\n    instance.forceAtStepValue = this.forceAtStepValue;\n    instance.consecutiveAllowedValue = this.consecutiveAllowedValue;\n    instance.source = this.source;\n    instance.sourceTool = this.sourceTool;\n    instance.forcePreventStop = this.forcePreventStop;\n    instance.reason = this.reason;\n    return instance;\n  }\n}\n"]}

package/dist/agents/requirement/requirements/conditional.d.cts

@@ -0,0 +1,71 @@
+import { Requirement, Rule } from './requirement.cjs';
+import { MultiTargetType, TargetType } from './utils.cjs';
+import { AnyTool } from '../../../tools/base.cjs';
+import { RunContext } from '../../../context.cjs';
+import { R as RequirementAgentRunState } from '../../../types-W0QttuP_.cjs';
+import '../../../emitter-jN55XZZq.cjs';
+import '../../../internals/types.cjs';
+import '../../../internals/helpers/guards.cjs';
+import '../../../internals/serializable.cjs';
+import '../../../errors.cjs';
+import '../../../internals/helpers/promise.cjs';
+import 'ajv';
+import 'promise-based-task';
+import '../../../cache/base.cjs';
+import '../../../internals/helpers/schema.cjs';
+import 'zod';
+import 'zod-to-json-schema';
+import '../../../memory/base.cjs';
+import '../../../backend/message.cjs';
+import 'ai';
+import '../../../template.cjs';
+import '../../../chat-BOb9n7AJ.cjs';
+import '../../../backend/constants.cjs';
+import '../../../logger/logger.cjs';
+import 'pino';
+import '../../../base-XfWeXQ-S.cjs';
+
+/**
+ * Copyright 2025 © BeeAI a Series of LF Projects, LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+interface ConditionalRequirementOptions {
+    name?: string;
+    forceAtStep?: number;
+    onlyBefore?: MultiTargetType;
+    onlyAfter?: MultiTargetType;
+    forceAfter?: MultiTargetType;
+    forcePreventStop?: boolean;
+    minInvocations?: number;
+    maxInvocations?: number;
+    onlySuccessInvocations?: boolean;
+    consecutiveAllowed?: boolean;
+    priority?: number;
+    customChecks?: ((state: RequirementAgentRunState) => boolean)[];
+    enabled?: boolean;
+    reason?: string;
+}
+declare class ConditionalRequirement extends Requirement {
+    protected source: TargetType;
+    protected sourceTool?: AnyTool;
+    protected before: Set<TargetType>;
+    protected after: Set<TargetType>;
+    protected forceAfterSet: Set<TargetType>;
+    protected minInvocations: number;
+    protected maxInvocations: number;
+    protected forceAtStepValue?: number;
+    protected onlySuccessInvocations: boolean;
+    protected consecutiveAllowedValue: boolean;
+    protected customChecks: ((state: RequirementAgentRunState) => boolean)[];
+    protected forcePreventStop: boolean;
+    protected reason?: string;
+    constructor(target: TargetType, options?: ConditionalRequirementOptions);
+    protected checkInvariant(): void;
+    init(tools: AnyTool[], ctx: RunContext<any>): Promise<void>;
+    reset(): this;
+    _run(state: RequirementAgentRunState, _: RunContext<typeof this>): Promise<Rule[]>;
+    clone(): Promise<this>;
+}
+
+export { ConditionalRequirement, type ConditionalRequirementOptions };