beeai-framework 0.1.21 → 0.1.23

package/dist/agents/requirement/types.js

@@ -0,0 +1,35 @@
+import { z } from 'zod';
+
+const RequirementAgentSystemPromptInputSchema = z.object({
+  role: z.string().optional(),
+  instructions: z.string().optional(),
+  notes: z.string().optional(),
+  finalAnswerName: z.string(),
+  finalAnswerSchema: z.string().optional(),
+  finalAnswerInstructions: z.string().optional(),
+  tools: z.array(z.object({
+    name: z.string(),
+    description: z.string(),
+    inputSchema: z.string(),
+    allowed: z.string(),
+    reason: z.string().optional()
+  }))
+});
+const RequirementAgentTaskPromptInputSchema = z.object({
+  prompt: z.string(),
+  context: z.string().optional(),
+  expectedOutput: z.string().optional()
+});
+const RequirementAgentToolErrorPromptInputSchema = z.object({
+  reason: z.string()
+});
+const RequirementAgentToolNoResultPromptInputSchema = z.object({
+  toolCall: z.object({
+    tool: z.any().optional(),
+    input: z.unknown()
+  })
+});
+
+export { RequirementAgentSystemPromptInputSchema, RequirementAgentTaskPromptInputSchema, RequirementAgentToolErrorPromptInputSchema, RequirementAgentToolNoResultPromptInputSchema };
+//# sourceMappingURL=types.js.map
+//# sourceMappingURL=types.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/agents/requirement/types.ts"],"names":["RequirementAgentSystemPromptInputSchema","z","object","role","string","optional","instructions","notes","finalAnswerName","finalAnswerSchema","finalAnswerInstructions","tools","array","name","description","inputSchema","allowed","reason","RequirementAgentTaskPromptInputSchema","prompt","context","expectedOutput","RequirementAgentToolErrorPromptInputSchema","RequirementAgentToolNoResultPromptInputSchema","toolCall","tool","any","input","unknown"],"mappings":";;AAeO,MAAMA,uCAAAA,GAA0CC,EAAEC,MAAAA,CAAO;EAC9DC,IAAAA,EAAMF,CAAAA,CAAEG,MAAAA,EAAM,CAAGC,QAAAA,EAAQ;EACzBC,YAAAA,EAAcL,CAAAA,CAAEG,MAAAA,EAAM,CAAGC,QAAAA,EAAQ;EACjCE,KAAAA,EAAON,CAAAA,CAAEG,MAAAA,EAAM,CAAGC,QAAAA,EAAQ;AAC1BG,EAAAA,eAAAA,EAAiBP,EAAEG,MAAAA,EAAM;EACzBK,iBAAAA,EAAmBR,CAAAA,CAAEG,MAAAA,EAAM,CAAGC,QAAAA,EAAQ;EACtCK,uBAAAA,EAAyBT,CAAAA,CAAEG,MAAAA,EAAM,CAAGC,QAAAA,EAAQ;EAC5CM,KAAAA,EAAOV,CAAAA,CAAEW,KAAAA,CACPX,CAAAA,CAAEC,MAAAA,CAAO;AACPW,IAAAA,IAAAA,EAAMZ,EAAEG,MAAAA,EAAM;AACdU,IAAAA,WAAAA,EAAab,EAAEG,MAAAA,EAAM;AACrBW,IAAAA,WAAAA,EAAad,EAAEG,MAAAA,EAAM;AACrBY,IAAAA,OAAAA,EAASf,EAAEG,MAAAA,EAAM;IACjBa,MAAAA,EAAQhB,CAAAA,CAAEG,MAAAA,EAAM,CAAGC,QAAAA;AACrB,GAAA,CAAA;AAEJ,CAAA;AAEO,MAAMa,qCAAAA,GAAwCjB,EAAEC,MAAAA,CAAO;AAC5DiB,EAAAA,MAAAA,EAAQlB,EAAEG,MAAAA,EAAM;EAChBgB,OAAAA,EAASnB,CAAAA,CAAEG,MAAAA,EAAM,CAAGC,QAAAA,EAAQ;EAC5BgB,cAAAA,EAAgBpB,CAAAA,CAAEG,MAAAA,EAAM,CAAGC,QAAAA;AAC7B,CAAA;AAEO,MAAMiB,0CAAAA,GAA6CrB,EAAEC,MAAAA,CAAO;AACjEe,EAAAA,MAAAA,EAAQhB,EAAEG,MAAAA;AACZ,CAAA;AAEO,MAAMmB,6CAAAA,GAAgDtB,EAAEC,MAAAA,CAAO;AACpEsB,EAAAA,QAAAA,EAAUvB,EAAEC,MAAAA,CAAO;IACjBuB,IAAAA,EAAMxB,CAAAA,CAAEyB,GAAAA,EAAG,CAAGrB,QAAAA,EAAQ;AACtBsB,IAAAA,KAAAA,EAAO1B,EAAE2B,OAAAA;GACX;AACF,CAAA","file":"types.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { BaseMemory } from \"@/memory/base.js\";\nimport { AssistantMessage } from \"@/backend/message.js\";\nimport { AnyTool, ToolOutput } from \"@/tools/base.js\";\nimport { FrameworkError } from \"@/errors.js\";\nimport { PromptTemplate } from \"@/template.js\";\nimport { ChatModelOutput, ChatModelToolChoice, ChatModelUsage } from \"@/backend/chat.js\";\nimport { BaseAgentRunOptions } from \"@/agents/base.js\";\nimport { z } from \"zod\";\nimport { FinalAnswerTool } from \"@/agents/requirement/utils/tool.js\";\n\nexport const RequirementAgentSystemPromptInputSchema = z.object({\n  role: z.string().optional(),\n  instructions: z.string().optional(),\n  notes: z.string().optional(),\n  finalAnswerName: z.string(),\n  finalAnswerSchema: z.string().optional(),\n  finalAnswerInstructions: z.string().optional(),\n  tools: z.array(\n    z.object({\n      name: z.string(),\n      description: z.string(),\n      inputSchema: z.string(),\n      allowed: z.string(),\n      reason: z.string().optional(),\n    }),\n  ),\n});\n\nexport const RequirementAgentTaskPromptInputSchema = z.object({\n  prompt: z.string(),\n  context: z.string().optional(),\n  expectedOutput: z.string().optional(),\n});\n\nexport const RequirementAgentToolErrorPromptInputSchema = z.object({\n  reason: z.string(),\n});\n\nexport const RequirementAgentToolNoResultPromptInputSchema = z.object({\n  toolCall: z.object({\n    tool: z.any().optional(),\n    input: z.unknown(),\n  }),\n});\n\n// Template collection\nexport interface RequirementAgentTemplates {\n  system: PromptTemplate<typeof RequirementAgentSystemPromptInputSchema>;\n  task: PromptTemplate<typeof RequirementAgentTaskPromptInputSchema>;\n  toolError: PromptTemplate<typeof RequirementAgentToolErrorPromptInputSchema>;\n  toolNoResult: PromptTemplate<typeof RequirementAgentToolNoResultPromptInputSchema>;\n}\n\n// Run state types\nexport interface RequirementAgentRunStateStep {\n  id: string;\n  iteration: number;\n  tool: AnyTool | null;\n  input: unknown;\n  output: ToolOutput;\n  error: FrameworkError | null;\n}\n\nexport interface RequirementAgentRunState {\n  answer: AssistantMessage | null;\n  result: unknown;\n  memory: BaseMemory;\n  iteration: number;\n  steps: RequirementAgentRunStateStep[];\n  usage: ChatModelUsage;\n  // cost: ChatModelCost; TODO: not supported yet\n}\n\n// Agent output\nexport interface RequirementAgentOutput {\n  result: AssistantMessage;\n  memory: BaseMemory;\n  state: RequirementAgentRunState;\n}\n\n// Internal request structure\nexport interface RequirementAgentRequest {\n  tools: AnyTool[];\n  allowedTools: AnyTool[];\n  reasonByTool: WeakMap<AnyTool, string | undefined>;\n  hiddenTools: AnyTool[];\n  toolChoice: ChatModelToolChoice;\n  finalAnswer: FinalAnswerTool;\n  canStop: boolean;\n}\n\n// Execution configuration\nexport interface RequirementAgentExecutionConfig {\n  maxRetriesPerStep?: number;\n  totalMaxRetries?: number;\n  maxIterations?: number;\n}\n\n// Run input and options\nexport interface RequirementAgentRunInput {\n  prompt: string | null;\n  context?: string;\n  expectedOutput?: string | z.ZodSchema;\n}\n\nexport interface RequirementAgentRunOptions extends BaseAgentRunOptions {\n  execution?: RequirementAgentExecutionConfig;\n}\n\n// Event callback types\nexport interface RequirementAgentCallbacks {\n  start: (data: { state: RequirementAgentRunState; request: RequirementAgentRequest }) => void;\n  success: (data: { state: RequirementAgentRunState; response: ChatModelOutput }) => void;\n  finalAnswer: (data: {\n    state: RequirementAgentRunState;\n    output: string;\n    delta: string;\n    outputStructured: unknown;\n  }) => void;\n}\n"]}

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

@@ -0,0 +1,202 @@
+'use strict';
+
+var base_cjs$1 = require('../../../tools/base.cjs');
+var base_cjs = require('../../base.cjs');
+var message_cjs = require('../../../backend/message.cjs');
+var remeda = require('remeda');
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+class RequirementsReasoner {
+  static {
+    __name(this, "RequirementsReasoner");
+  }
+  tools;
+  entries = [];
+  context;
+  finalAnswer;
+  constructor(tools, finalAnswer, context) {
+    this.tools = [
+      ...tools,
+      finalAnswer
+    ];
+    this.finalAnswer = finalAnswer;
+    this.context = context;
+  }
+  async update(requirements) {
+    this.entries = [];
+    for (const requirement of requirements) {
+      this.entries.push(requirement);
+    }
+    for (const entry of this.entries) {
+      const emitter = this.context.emitter.child({
+        groupId: remeda.toCamelCase(entry.name),
+        creator: entry
+      });
+      emitter.namespace.push("requirement");
+      const tools = [
+        ...this.tools
+      ];
+      await emitter.emit("init", {
+        tools
+      });
+      await entry.init(tools, this.context);
+    }
+  }
+  findToolByName(name) {
+    const tool = this.tools.find((t) => t.name === name);
+    if (!tool) {
+      throw new Error(`Tool '${name}' not found in (${this.tools.map((t) => t.name).join(", ")}).`);
+    }
+    return tool;
+  }
+  async createRequest(state, forceTool, extraRules = []) {
+    const hidden = [];
+    const allowed = [];
+    const allTools = [
+      ...this.tools
+    ];
+    const reasonByTool = /* @__PURE__ */ new WeakMap();
+    let preventStop = false;
+    const preventStepRefs = [];
+    let forced = null;
+    let forcedLevel = 0;
+    const rulesByTool = /* @__PURE__ */ new Map();
+    for (const tool of this.tools) {
+      rulesByTool.set(tool.name, []);
+    }
+    for (const requirement of this.entries.filter((e) => e.enabled)) {
+      const generatedRules = await requirement.run(state);
+      for (const rule of generatedRules) {
+        const tool = this.findToolByName(rule.target);
+        const rules = rulesByTool.get(tool.name) || [];
+        rules.push({
+          priority: requirement.priority,
+          rule,
+          requirement
+        });
+        rulesByTool.set(tool.name, rules);
+      }
+    }
+    for (const rule of extraRules) {
+      if (!rulesByTool.has(rule.target)) {
+        throw new Error(`Tool '${rule.target}' not found.`);
+      }
+      const rules = rulesByTool.get(rule.target);
+      const priority = rules.length > 0 ? Math.max(...rules.map((r) => r.priority)) + 1 : 1;
+      rules.push({
+        priority,
+        rule,
+        requirement: null
+      });
+    }
+    for (const [toolName, rules] of rulesByTool.entries()) {
+      const tool = this.findToolByName(toolName);
+      rules.sort((a, b) => b.priority - a.priority);
+      const maxPriority = rules.length > 0 ? rules[0].priority : 1;
+      let isAllowed = true;
+      let isForced = false;
+      let isHidden = false;
+      let isPreventStop = false;
+      for (const ruleEntry of rules) {
+        const rule = ruleEntry.rule;
+        if (!rule.allowed) {
+          isAllowed = false;
+        }
+        if (rule.hidden) {
+          isHidden = true;
+        }
+        if (rule.forced) {
+          isForced = true;
+        }
+        if (rule.preventStop) {
+          isPreventStop = true;
+          preventStepRefs.push(ruleEntry);
+        }
+        if (rule.reason) {
+          reasonByTool.set(tool, rule.reason);
+        }
+      }
+      if (isAllowed && isHidden) {
+        isAllowed = false;
+      }
+      if (isAllowed) {
+        if (!allowed.includes(tool)) {
+          allowed.push(tool);
+        }
+        if (isForced && (!forced || forcedLevel < maxPriority)) {
+          forced = tool;
+          forcedLevel = maxPriority;
+        }
+      }
+      if (isHidden) {
+        if (!hidden.includes(tool)) {
+          hidden.push(tool);
+        }
+      }
+      if (isPreventStop) {
+        preventStop = true;
+      }
+    }
+    if (forced) {
+      allowed.length = 0;
+      if (!allowed.includes(forced)) {
+        allowed.push(forced);
+      }
+      if (!allowed.includes(this.finalAnswer)) {
+        allowed.push(this.finalAnswer);
+      }
+    }
+    if (preventStop && forced !== this.finalAnswer) {
+      const finalAnswerIndex = allowed.indexOf(this.finalAnswer);
+      if (finalAnswerIndex !== -1) {
+        allowed.splice(finalAnswerIndex, 1);
+      }
+    }
+    if (allowed.length === 0) {
+      throw new base_cjs.AgentError("One of the generated rules is preventing the agent from continuing. This indicates that the provided requirements may conflict with each other. See the following rules and their attached requirements that are preventing the agent from continuing.\n" + JSON.stringify(preventStepRefs, null, 2));
+    }
+    let toolChoice = forced || "required";
+    if (allowed.length === 1) {
+      toolChoice = allowed[0];
+    }
+    if (!(toolChoice instanceof base_cjs$1.Tool) && !forceTool && !preventStop) {
+      toolChoice = "auto";
+    }
+    return {
+      tools: allTools,
+      allowedTools: allowed,
+      reasonByTool,
+      toolChoice,
+      finalAnswer: this.finalAnswer,
+      hiddenTools: hidden,
+      canStop: !preventStop
+    };
+  }
+}
+async function createSystemMessage(template, request) {
+  return new message_cjs.SystemMessage(template.render({
+    tools: await Promise.all(request.tools.filter((tool) => !request.hiddenTools.includes(tool)).map(async (tool) => ({
+      name: tool.name,
+      description: tool.description,
+      inputSchema: JSON.stringify(await tool.inputSchema(), null, 2),
+      allowed: String(request.allowedTools.includes(tool)),
+      reason: request.reasonByTool.get(tool)
+    }))),
+    finalAnswerName: request.finalAnswer.name,
+    finalAnswerSchema: request.finalAnswer.customSchema ? JSON.stringify(request.finalAnswer.inputSchema, null, 2) : void 0,
+    finalAnswerInstructions: request.finalAnswer.instructions,
+    // default
+    role: void 0,
+    // default
+    instructions: void 0,
+    // default
+    notes: void 0
+  }));
+}
+__name(createSystemMessage, "createSystemMessage");
+
+exports.RequirementsReasoner = RequirementsReasoner;
+exports.createSystemMessage = createSystemMessage;
+//# sourceMappingURL=llm.cjs.map
+//# sourceMappingURL=llm.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../src/agents/requirement/utils/llm.ts"],"names":["RequirementsReasoner","tools","entries","context","finalAnswer","update","requirements","requirement","push","entry","emitter","child","groupId","toCamelCase","name","creator","namespace","emit","init","findToolByName","tool","find","t","Error","map","join","createRequest","state","forceTool","extraRules","hidden","allowed","allTools","reasonByTool","WeakMap","preventStop","preventStepRefs","forced","forcedLevel","rulesByTool","Map","set","filter","e","enabled","generatedRules","run","rule","target","rules","get","priority","has","length","Math","max","r","toolName","sort","a","b","maxPriority","isAllowed","isForced","isHidden","isPreventStop","ruleEntry","reason","includes","finalAnswerIndex","indexOf","splice","AgentError","JSON","stringify","toolChoice","Tool","allowedTools","hiddenTools","canStop","createSystemMessage","template","request","SystemMessage","render","Promise","all","description","inputSchema","String","finalAnswerName","finalAnswerSchema","customSchema","undefined","finalAnswerInstructions","instructions","role","notes"],"mappings":";;;;;;;;;AAiCO,MAAMA,oBAAAA,CAAAA;EAjCb;;;AAkCYC,EAAAA,KAAAA;AACAC,EAAAA,OAAAA,GAAyB,EAAA;AACzBC,EAAAA,OAAAA;AACMC,EAAAA,WAAAA;EAEhB,WAAA,CAAYH,KAAAA,EAAkBG,aAA8BD,OAAAA,EAA0B;AACpF,IAAA,IAAA,CAAKF,KAAAA,GAAQ;AAAIA,MAAAA,GAAAA,KAAAA;AAAOG,MAAAA;;AACxB,IAAA,IAAA,CAAKA,WAAAA,GAAcA,WAAAA;AACnB,IAAA,IAAA,CAAKD,OAAAA,GAAUA,OAAAA;AACjB;AAEA,EAAA,MAAME,OAAOC,YAAAA,EAA4C;AACvD,IAAA,IAAA,CAAKJ,UAAU,EAAA;AAEf,IAAA,KAAA,MAAWK,eAAeD,YAAAA,EAAc;AACtC,MAAA,IAAA,CAAKJ,OAAAA,CAAQM,KAAKD,WAAAA,CAAAA;AACpB;AAEA,IAAA,KAAA,MAAWE,KAAAA,IAAS,KAAKP,OAAAA,EAAS;AAChC,MAAA,MAAMQ,OAAAA,GAAU,IAAA,CAAKP,OAAAA,CAAQO,OAAAA,CAAQC,KAAAA,CAA4B;QAC/DC,OAAAA,EAASC,kBAAAA,CAAYJ,MAAMK,IAAI,CAAA;QAC/BC,OAAAA,EAASN;OACX,CAAA;AACAC,MAAAA,OAAAA,CAAQM,SAAAA,CAAUR,KAAK,aAAA,CAAA;AAEvB,MAAA,MAAMP,KAAAA,GAAQ;WAAI,IAAA,CAAKA;;AACvB,MAAA,MAAMS,OAAAA,CAAQO,KAAK,MAAA,EAAQ;AAAEhB,QAAAA;OAAM,CAAA;AACnC,MAAA,MAAMQ,KAAAA,CAAMS,IAAAA,CAAKjB,KAAAA,EAAO,IAAA,CAAKE,OAAO,CAAA;AACtC;AACF;AAEUgB,EAAAA,cAAAA,CAAeL,IAAAA,EAAuB;AAC9C,IAAA,MAAMM,IAAAA,GAAO,KAAKnB,KAAAA,CAAMoB,IAAAA,CAAK,CAACC,CAAAA,KAAMA,CAAAA,CAAER,SAASA,IAAAA,CAAAA;AAC/C,IAAA,IAAI,CAACM,IAAAA,EAAM;AACT,MAAA,MAAM,IAAIG,KAAAA,CAAM,CAAA,MAAA,EAAST,IAAAA,CAAAA,gBAAAA,EAAuB,KAAKb,KAAAA,CAAMuB,GAAAA,CAAI,CAACF,CAAAA,KAAMA,EAAER,IAAI,CAAA,CAAEW,IAAAA,CAAK,IAAA,CAAA,CAAA,EAAA,CAAS,CAAA;AAC9F;AACA,IAAA,OAAOL,IAAAA;AACT;AAEA,EAAA,MAAMM,aAAAA,CACJC,KAAAA,EACAC,SAAAA,EACAC,UAAAA,GAAqB,EAAA,EACa;AAClC,IAAA,MAAMC,SAAoB,EAAA;AAC1B,IAAA,MAAMC,UAAqB,EAAA;AAC3B,IAAA,MAAMC,QAAAA,GAAsB;SAAI,IAAA,CAAK/B;;AACrC,IAAA,MAAMgC,YAAAA,uBAAmBC,OAAAA,EAAAA;AAEzB,IAAA,IAAIC,WAAAA,GAAc,KAAA;AAClB,IAAA,MAAMC,kBAA+B,EAAA;AAErC,IAAA,IAAIC,MAAAA,GAAyB,IAAA;AAC7B,IAAA,IAAIC,WAAAA,GAAc,CAAA;AAClB,IAAA,MAAMC,WAAAA,uBAAkBC,GAAAA,EAAAA;AAGxB,IAAA,KAAA,MAAWpB,IAAAA,IAAQ,KAAKnB,KAAAA,EAAO;AAC7BsC,MAAAA,WAAAA,CAAYE,GAAAA,CAAIrB,IAAAA,CAAKN,IAAAA,EAAM,EAAE,CAAA;AAC/B;AAGA,IAAA,KAAA,MAAWP,WAAAA,IAAe,KAAKL,OAAAA,CAAQwC,MAAAA,CAAO,CAACC,CAAAA,KAAMA,CAAAA,CAAEC,OAAO,CAAA,EAAG;AAC/D,MAAA,MAAMC,cAAAA,GAAiB,MAAMtC,WAAAA,CAAYuC,GAAAA,CAAInB,KAAAA,CAAAA;AAC7C,MAAA,KAAA,MAAWoB,QAAQF,cAAAA,EAAgB;AACjC,QAAA,MAAMzB,IAAAA,GAAO,IAAA,CAAKD,cAAAA,CAAe4B,IAAAA,CAAKC,MAAM,CAAA;AAC5C,QAAA,MAAMC,QAAQV,WAAAA,CAAYW,GAAAA,CAAI9B,IAAAA,CAAKN,IAAI,KAAK,EAAA;AAC5CmC,QAAAA,KAAAA,CAAMzC,IAAAA,CAAK;AACT2C,UAAAA,QAAAA,EAAU5C,WAAAA,CAAY4C,QAAAA;AACtBJ,UAAAA,IAAAA;AACAxC,UAAAA;SACF,CAAA;AACAgC,QAAAA,WAAAA,CAAYE,GAAAA,CAAIrB,IAAAA,CAAKN,IAAAA,EAAMmC,KAAAA,CAAAA;AAC7B;AACF;AAGA,IAAA,KAAA,MAAWF,QAAQlB,UAAAA,EAAY;AAC7B,MAAA,IAAI,CAACU,WAAAA,CAAYa,GAAAA,CAAIL,IAAAA,CAAKC,MAAM,CAAA,EAAG;AACjC,QAAA,MAAM,IAAIzB,KAAAA,CAAM,CAAA,MAAA,EAASwB,IAAAA,CAAKC,MAAM,CAAA,YAAA,CAAc,CAAA;AACpD;AAEA,MAAA,MAAMC,KAAAA,GAAQV,WAAAA,CAAYW,GAAAA,CAAIH,IAAAA,CAAKC,MAAM,CAAA;AACzC,MAAA,MAAMG,QAAAA,GAAWF,KAAAA,CAAMI,MAAAA,GAAS,CAAA,GAAIC,KAAKC,GAAAA,CAAG,GAAIN,KAAAA,CAAMzB,GAAAA,CAAI,CAACgC,CAAAA,KAAMA,CAAAA,CAAEL,QAAQ,CAAA,IAAK,CAAA,GAAI,CAAA;AACpFF,MAAAA,KAAAA,CAAMzC,IAAAA,CAAK;AACT2C,QAAAA,QAAAA;AACAJ,QAAAA,IAAAA;QACAxC,WAAAA,EAAa;OACf,CAAA;AACF;AAGA,IAAA,KAAA,MAAW,CAACkD,QAAAA,EAAUR,KAAAA,CAAAA,IAAUV,WAAAA,CAAYrC,SAAO,EAAI;AACrD,MAAA,MAAMkB,IAAAA,GAAO,IAAA,CAAKD,cAAAA,CAAesC,QAAAA,CAAAA;AACjCR,MAAAA,KAAAA,CAAMS,KAAK,CAACC,CAAAA,EAAGC,MAAMA,CAAAA,CAAET,QAAAA,GAAWQ,EAAER,QAAQ,CAAA;AAE5C,MAAA,MAAMU,cAAcZ,KAAAA,CAAMI,MAAAA,GAAS,IAAIJ,KAAAA,CAAM,CAAA,EAAGE,QAAAA,GAAW,CAAA;AAC3D,MAAA,IAAIW,SAAAA,GAAY,IAAA;AAChB,MAAA,IAAIC,QAAAA,GAAW,KAAA;AACf,MAAA,IAAIC,QAAAA,GAAW,KAAA;AACf,MAAA,IAAIC,aAAAA,GAAgB,KAAA;AAEpB,MAAA,KAAA,MAAWC,aAAajB,KAAAA,EAAO;AAC7B,QAAA,MAAMF,OAAOmB,SAAAA,CAAUnB,IAAAA;AACvB,QAAA,IAAI,CAACA,KAAKhB,OAAAA,EAAS;AACjB+B,UAAAA,SAAAA,GAAY,KAAA;AACd;AACA,QAAA,IAAIf,KAAKjB,MAAAA,EAAQ;AACfkC,UAAAA,QAAAA,GAAW,IAAA;AACb;AACA,QAAA,IAAIjB,KAAKV,MAAAA,EAAQ;AACf0B,UAAAA,QAAAA,GAAW,IAAA;AACb;AACA,QAAA,IAAIhB,KAAKZ,WAAAA,EAAa;AACpB8B,UAAAA,aAAAA,GAAgB,IAAA;AAChB7B,UAAAA,eAAAA,CAAgB5B,KAAK0D,SAAAA,CAAAA;AACvB;AACA,QAAA,IAAInB,KAAKoB,MAAAA,EAAQ;AACflC,UAAAA,YAAAA,CAAaQ,GAAAA,CAAIrB,IAAAA,EAAM2B,IAAAA,CAAKoB,MAAM,CAAA;AACpC;AACF;AAEA,MAAA,IAAIL,aAAaE,QAAAA,EAAU;AACzBF,QAAAA,SAAAA,GAAY,KAAA;AACd;AAEA,MAAA,IAAIA,SAAAA,EAAW;AACb,QAAA,IAAI,CAAC/B,OAAAA,CAAQqC,QAAAA,CAAShD,IAAAA,CAAAA,EAAO;AAC3BW,UAAAA,OAAAA,CAAQvB,KAAKY,IAAAA,CAAAA;AACf;AACA,QAAA,IAAI2C,QAAAA,KAAa,CAAC1B,MAAAA,IAAUC,WAAAA,GAAcuB,WAAAA,CAAAA,EAAc;AACtDxB,UAAAA,MAAAA,GAASjB,IAAAA;AACTkB,UAAAA,WAAAA,GAAcuB,WAAAA;AAChB;AACF;AACA,MAAA,IAAIG,QAAAA,EAAU;AACZ,QAAA,IAAI,CAAClC,MAAAA,CAAOsC,QAAAA,CAAShD,IAAAA,CAAAA,EAAO;AAC1BU,UAAAA,MAAAA,CAAOtB,KAAKY,IAAAA,CAAAA;AACd;AACF;AACA,MAAA,IAAI6C,aAAAA,EAAe;AACjB9B,QAAAA,WAAAA,GAAc,IAAA;AAChB;AACF;AAGA,IAAA,IAAIE,MAAAA,EAAQ;AACVN,MAAAA,OAAAA,CAAQsB,MAAAA,GAAS,CAAA;AACjB,MAAA,IAAI,CAACtB,OAAAA,CAAQqC,QAAAA,CAAS/B,MAAAA,CAAAA,EAAS;AAC7BN,QAAAA,OAAAA,CAAQvB,KAAK6B,MAAAA,CAAAA;AACf;AACA,MAAA,IAAI,CAACN,OAAAA,CAAQqC,QAAAA,CAAS,IAAA,CAAKhE,WAAW,CAAA,EAAU;AAC9C2B,QAAAA,OAAAA,CAAQvB,IAAAA,CAAK,KAAKJ,WAAW,CAAA;AAC/B;AACF;AAGA,IAAA,IAAI+B,WAAAA,IAAeE,MAAAA,KAAW,IAAA,CAAKjC,WAAAA,EAAa;AAC9C,MAAA,MAAMiE,gBAAAA,GAAmBtC,OAAAA,CAAQuC,OAAAA,CAAQ,IAAA,CAAKlE,WAAW,CAAA;AACzD,MAAA,IAAIiE,qBAAqB,EAAA,EAAI;AAC3BtC,QAAAA,OAAAA,CAAQwC,MAAAA,CAAOF,kBAAkB,CAAA,CAAA;AACnC;AACF;AAEA,IAAA,IAAItC,OAAAA,CAAQsB,WAAW,CAAA,EAAG;AACxB,MAAA,MAAM,IAAImB,oBACR,0PAAA,GAGEC,IAAAA,CAAKC,UAAUtC,eAAAA,EAAiB,IAAA,EAAM,CAAA,CAAA,CAAA;AAE5C;AAEA,IAAA,IAAIuC,aAAkCtC,MAAAA,IAAU,UAAA;AAChD,IAAA,IAAIN,OAAAA,CAAQsB,WAAW,CAAA,EAAG;AACxBsB,MAAAA,UAAAA,GAAa5C,QAAQ,CAAA,CAAA;AACvB;AAGA,IAAA,IAAI,EAAE4C,UAAAA,YAAsBC,eAAAA,CAAAA,IAAS,CAAChD,SAAAA,IAAa,CAACO,WAAAA,EAAa;AAC/DwC,MAAAA,UAAAA,GAAa,MAAA;AACf;AAEA,IAAA,OAAO;MACL1E,KAAAA,EAAO+B,QAAAA;MACP6C,YAAAA,EAAc9C,OAAAA;AACdE,MAAAA,YAAAA;AACA0C,MAAAA,UAAAA;AACAvE,MAAAA,WAAAA,EAAa,IAAA,CAAKA,WAAAA;MAClB0E,WAAAA,EAAahD,MAAAA;AACbiD,MAAAA,OAAAA,EAAS,CAAC5C;AACZ,KAAA;AACF;AACF;AAKA,eAAsB6C,mBAAAA,CACpBC,UACAC,OAAAA,EAAgC;AAEhC,EAAA,OAAO,IAAIC,yBAAAA,CACTF,QAAAA,CAASG,MAAAA,CAAO;AACdnF,IAAAA,KAAAA,EAAO,MAAMoF,OAAAA,CAAQC,GAAAA,CACnBJ,OAAAA,CAAQjF,KAAAA,CACLyC,OAAO,CAACtB,IAAAA,KAAS,CAAC8D,OAAAA,CAAQJ,YAAYV,QAAAA,CAAShD,IAAAA,CAAAA,CAAAA,CAC/CI,GAAAA,CAAI,OAAOJ,IAAAA,MAAU;AACpBN,MAAAA,IAAAA,EAAMM,IAAAA,CAAKN,IAAAA;AACXyE,MAAAA,WAAAA,EAAanE,IAAAA,CAAKmE,WAAAA;AAClBC,MAAAA,WAAAA,EAAaf,KAAKC,SAAAA,CAAU,MAAMtD,KAAKoE,WAAAA,EAAW,EAAI,MAAM,CAAA,CAAA;AAC5DzD,MAAAA,OAAAA,EAAS0D,MAAAA,CAAOP,OAAAA,CAAQL,YAAAA,CAAaT,QAAAA,CAAShD,IAAAA,CAAAA,CAAAA;MAC9C+C,MAAAA,EAAQe,OAAAA,CAAQjD,YAAAA,CAAaiB,GAAAA,CAAI9B,IAAAA;AACnC,KAAA,CAAA,CAAA,CAAA;AAEJsE,IAAAA,eAAAA,EAAiBR,QAAQ9E,WAAAA,CAAYU,IAAAA;IACrC6E,iBAAAA,EAAmBT,OAAAA,CAAQ9E,WAAAA,CAAYwF,YAAAA,GACnCnB,IAAAA,CAAKC,SAAAA,CAAUQ,QAAQ9E,WAAAA,CAAYoF,WAAAA,EAAa,IAAA,EAAM,CAAA,CAAA,GACtDK,MAAAA;AACJC,IAAAA,uBAAAA,EAAyBZ,QAAQ9E,WAAAA,CAAY2F,YAAAA;;IAC7CC,IAAAA,EAAMH,MAAAA;;IACNE,YAAAA,EAAcF,MAAAA;;IACdI,KAAAA,EAAOJ;AACT,GAAA,CAAA,CAAA;AAEJ;AA3BsBb,MAAAA,CAAAA,mBAAAA,EAAAA,qBAAAA,CAAAA","file":"llm.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { AnyTool, Tool } from \"@/tools/base.js\";\nimport { RunContext } from \"@/context.js\";\nimport { AgentError } from \"@/agents/base.js\";\nimport { SystemMessage } from \"@/backend/message.js\";\nimport { PromptTemplate } from \"@/template.js\";\nimport {\n  RequirementAgentRequest,\n  RequirementAgentRunState,\n  RequirementAgentSystemPromptInputSchema,\n} from \"@/agents/requirement/types.js\";\nimport {\n  Requirement,\n  RequirementCallbacks,\n  Rule,\n} from \"@/agents/requirement/requirements/requirement.js\";\nimport { FinalAnswerTool } from \"./tool.js\";\nimport { ChatModelToolChoice } from \"@/backend/chat.js\";\nimport { toCamelCase } from \"remeda\";\n\ninterface RuleEntry {\n  rule: Rule;\n  requirement: Requirement | null;\n  priority: number;\n}\n\n/**\n * Manages requirements and generates requests for the agent\n */\nexport class RequirementsReasoner {\n  protected tools: AnyTool[];\n  protected entries: Requirement[] = [];\n  protected context: RunContext<any>;\n  public readonly finalAnswer: FinalAnswerTool;\n\n  constructor(tools: AnyTool[], finalAnswer: FinalAnswerTool, context: RunContext<any>) {\n    this.tools = [...tools, finalAnswer];\n    this.finalAnswer = finalAnswer;\n    this.context = context;\n  }\n\n  async update(requirements: Requirement[]): Promise<void> {\n    this.entries = [];\n\n    for (const requirement of requirements) {\n      this.entries.push(requirement);\n    }\n\n    for (const entry of this.entries) {\n      const emitter = this.context.emitter.child<RequirementCallbacks>({\n        groupId: toCamelCase(entry.name),\n        creator: entry,\n      });\n      emitter.namespace.push(\"requirement\");\n\n      const tools = [...this.tools];\n      await emitter.emit(\"init\", { tools });\n      await entry.init(tools, this.context);\n    }\n  }\n\n  protected findToolByName(name: string): AnyTool {\n    const tool = this.tools.find((t) => t.name === name);\n    if (!tool) {\n      throw new Error(`Tool '${name}' not found in (${this.tools.map((t) => t.name).join(\", \")}).`);\n    }\n    return tool;\n  }\n\n  async createRequest(\n    state: RequirementAgentRunState,\n    forceTool: boolean,\n    extraRules: Rule[] = [],\n  ): Promise<RequirementAgentRequest> {\n    const hidden: AnyTool[] = [];\n    const allowed: AnyTool[] = [];\n    const allTools: AnyTool[] = [...this.tools];\n    const reasonByTool = new WeakMap<AnyTool, string | undefined>();\n\n    let preventStop = false;\n    const preventStepRefs: RuleEntry[] = [];\n\n    let forced: AnyTool | null = null;\n    let forcedLevel = 0;\n    const rulesByTool = new Map<string, RuleEntry[]>();\n\n    // Initialize rules map\n    for (const tool of this.tools) {\n      rulesByTool.set(tool.name, []);\n    }\n\n    // Group rules from requirements\n    for (const requirement of this.entries.filter((e) => e.enabled)) {\n      const generatedRules = await requirement.run(state);\n      for (const rule of generatedRules) {\n        const tool = this.findToolByName(rule.target);\n        const rules = rulesByTool.get(tool.name) || [];\n        rules.push({\n          priority: requirement.priority,\n          rule,\n          requirement,\n        });\n        rulesByTool.set(tool.name, rules);\n      }\n    }\n\n    // Add extra rules\n    for (const rule of extraRules) {\n      if (!rulesByTool.has(rule.target)) {\n        throw new Error(`Tool '${rule.target}' not found.`);\n      }\n\n      const rules = rulesByTool.get(rule.target)!;\n      const priority = rules.length > 0 ? Math.max(...rules.map((r) => r.priority)) + 1 : 1;\n      rules.push({\n        priority,\n        rule,\n        requirement: null,\n      });\n    }\n\n    // Aggregate rules and infer required tool\n    for (const [toolName, rules] of rulesByTool.entries()) {\n      const tool = this.findToolByName(toolName);\n      rules.sort((a, b) => b.priority - a.priority); // DESC\n\n      const maxPriority = rules.length > 0 ? rules[0].priority : 1;\n      let isAllowed = true;\n      let isForced = false;\n      let isHidden = false;\n      let isPreventStop = false;\n\n      for (const ruleEntry of rules) {\n        const rule = ruleEntry.rule;\n        if (!rule.allowed) {\n          isAllowed = false;\n        }\n        if (rule.hidden) {\n          isHidden = true;\n        }\n        if (rule.forced) {\n          isForced = true;\n        }\n        if (rule.preventStop) {\n          isPreventStop = true;\n          preventStepRefs.push(ruleEntry);\n        }\n        if (rule.reason) {\n          reasonByTool.set(tool, rule.reason);\n        }\n      }\n\n      if (isAllowed && isHidden) {\n        isAllowed = false;\n      }\n\n      if (isAllowed) {\n        if (!allowed.includes(tool)) {\n          allowed.push(tool);\n        }\n        if (isForced && (!forced || forcedLevel < maxPriority)) {\n          forced = tool;\n          forcedLevel = maxPriority;\n        }\n      }\n      if (isHidden) {\n        if (!hidden.includes(tool)) {\n          hidden.push(tool);\n        }\n      }\n      if (isPreventStop) {\n        preventStop = true;\n      }\n    }\n\n    // If forced tool is set, only allow it and final answer\n    if (forced) {\n      allowed.length = 0;\n      if (!allowed.includes(forced)) {\n        allowed.push(forced);\n      }\n      if (!allowed.includes(this.finalAnswer as any)) {\n        allowed.push(this.finalAnswer as any);\n      }\n    }\n\n    // If prevent stop is set and forced is not final answer, remove final answer\n    if (preventStop && forced !== this.finalAnswer) {\n      const finalAnswerIndex = allowed.indexOf(this.finalAnswer);\n      if (finalAnswerIndex !== -1) {\n        allowed.splice(finalAnswerIndex, 1);\n      }\n    }\n\n    if (allowed.length === 0) {\n      throw new AgentError(\n        \"One of the generated rules is preventing the agent from continuing. \" +\n          \"This indicates that the provided requirements may conflict with each other. \" +\n          \"See the following rules and their attached requirements that are preventing the agent from continuing.\\n\" +\n          JSON.stringify(preventStepRefs, null, 2),\n      );\n    }\n\n    let toolChoice: ChatModelToolChoice = forced || \"required\";\n    if (allowed.length === 1) {\n      toolChoice = allowed[0];\n    }\n\n    // Override to auto if not forcing and not preventing stop\n    if (!(toolChoice instanceof Tool) && !forceTool && !preventStop) {\n      toolChoice = \"auto\";\n    }\n\n    return {\n      tools: allTools,\n      allowedTools: allowed,\n      reasonByTool,\n      toolChoice,\n      finalAnswer: this.finalAnswer,\n      hiddenTools: hidden,\n      canStop: !preventStop,\n    };\n  }\n}\n\n/**\n * Create system message from template and request\n */\nexport async function createSystemMessage(\n  template: PromptTemplate<typeof RequirementAgentSystemPromptInputSchema>,\n  request: RequirementAgentRequest,\n): Promise<SystemMessage> {\n  return new SystemMessage(\n    template.render({\n      tools: await Promise.all(\n        request.tools\n          .filter((tool) => !request.hiddenTools.includes(tool))\n          .map(async (tool) => ({\n            name: tool.name,\n            description: tool.description,\n            inputSchema: JSON.stringify(await tool.inputSchema(), null, 2),\n            allowed: String(request.allowedTools.includes(tool)),\n            reason: request.reasonByTool.get(tool),\n          })),\n      ),\n      finalAnswerName: request.finalAnswer.name,\n      finalAnswerSchema: request.finalAnswer.customSchema\n        ? JSON.stringify(request.finalAnswer.inputSchema, null, 2)\n        : undefined,\n      finalAnswerInstructions: request.finalAnswer.instructions, // default\n      role: undefined, // default\n      instructions: undefined, // default\n      notes: undefined, // default\n    }),\n  );\n}\n"]}

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

@@ -0,0 +1,50 @@
+import { AnyTool } from '../../../tools/base.cjs';
+import { RunContext } from '../../../context.cjs';
+import { SystemMessage } from '../../../backend/message.cjs';
+import { PromptTemplate } from '../../../template.cjs';
+import { F as FinalAnswerTool, R as RequirementAgentRunState, a as RequirementAgentRequest, b as RequirementAgentSystemPromptInputSchema } from '../../../types-W0QttuP_.cjs';
+import { Requirement, Rule } from '../requirements/requirement.cjs';
+import 'ajv';
+import '../../../errors.cjs';
+import '../../../internals/types.cjs';
+import '../../../internals/helpers/guards.cjs';
+import '../../../internals/serializable.cjs';
+import 'promise-based-task';
+import '../../../cache/base.cjs';
+import '../../../internals/helpers/schema.cjs';
+import 'zod';
+import 'zod-to-json-schema';
+import '../../../emitter-jN55XZZq.cjs';
+import '../../../internals/helpers/promise.cjs';
+import 'ai';
+import '../../../memory/base.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
+ */
+
+/**
+ * Manages requirements and generates requests for the agent
+ */
+declare class RequirementsReasoner {
+    protected tools: AnyTool[];
+    protected entries: Requirement[];
+    protected context: RunContext<any>;
+    readonly finalAnswer: FinalAnswerTool;
+    constructor(tools: AnyTool[], finalAnswer: FinalAnswerTool, context: RunContext<any>);
+    update(requirements: Requirement[]): Promise<void>;
+    protected findToolByName(name: string): AnyTool;
+    createRequest(state: RequirementAgentRunState, forceTool: boolean, extraRules?: Rule[]): Promise<RequirementAgentRequest>;
+}
+/**
+ * Create system message from template and request
+ */
+declare function createSystemMessage(template: PromptTemplate<typeof RequirementAgentSystemPromptInputSchema>, request: RequirementAgentRequest): Promise<SystemMessage>;
+
+export { RequirementsReasoner, createSystemMessage };

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

@@ -0,0 +1,50 @@
+import { AnyTool } from '../../../tools/base.js';
+import { RunContext } from '../../../context.js';
+import { SystemMessage } from '../../../backend/message.js';
+import { PromptTemplate } from '../../../template.js';
+import { F as FinalAnswerTool, R as RequirementAgentRunState, a as RequirementAgentRequest, b as RequirementAgentSystemPromptInputSchema } from '../../../types-CJwiMH3u.js';
+import { Requirement, Rule } from '../requirements/requirement.js';
+import 'ajv';
+import '../../../errors.js';
+import '../../../internals/types.js';
+import '../../../internals/helpers/guards.js';
+import '../../../internals/serializable.js';
+import 'promise-based-task';
+import '../../../cache/base.js';
+import '../../../internals/helpers/schema.js';
+import 'zod';
+import 'zod-to-json-schema';
+import '../../../emitter-36-9MnvA.js';
+import '../../../internals/helpers/promise.js';
+import 'ai';
+import '../../../memory/base.js';
+import '../../../chat-DrqwUgD8.js';
+import '../../../backend/constants.js';
+import '../../../logger/logger.js';
+import 'pino';
+import '../../../base-CyFx7BRL.js';
+
+/**
+ * Copyright 2025 © BeeAI a Series of LF Projects, LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+/**
+ * Manages requirements and generates requests for the agent
+ */
+declare class RequirementsReasoner {
+    protected tools: AnyTool[];
+    protected entries: Requirement[];
+    protected context: RunContext<any>;
+    readonly finalAnswer: FinalAnswerTool;
+    constructor(tools: AnyTool[], finalAnswer: FinalAnswerTool, context: RunContext<any>);
+    update(requirements: Requirement[]): Promise<void>;
+    protected findToolByName(name: string): AnyTool;
+    createRequest(state: RequirementAgentRunState, forceTool: boolean, extraRules?: Rule[]): Promise<RequirementAgentRequest>;
+}
+/**
+ * Create system message from template and request
+ */
+declare function createSystemMessage(template: PromptTemplate<typeof RequirementAgentSystemPromptInputSchema>, request: RequirementAgentRequest): Promise<SystemMessage>;
+
+export { RequirementsReasoner, createSystemMessage };

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

@@ -0,0 +1,199 @@
+import { Tool } from '../../../tools/base.js';
+import { AgentError } from '../../base.js';
+import { SystemMessage } from '../../../backend/message.js';
+import { toCamelCase } from 'remeda';
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+class RequirementsReasoner {
+  static {
+    __name(this, "RequirementsReasoner");
+  }
+  tools;
+  entries = [];
+  context;
+  finalAnswer;
+  constructor(tools, finalAnswer, context) {
+    this.tools = [
+      ...tools,
+      finalAnswer
+    ];
+    this.finalAnswer = finalAnswer;
+    this.context = context;
+  }
+  async update(requirements) {
+    this.entries = [];
+    for (const requirement of requirements) {
+      this.entries.push(requirement);
+    }
+    for (const entry of this.entries) {
+      const emitter = this.context.emitter.child({
+        groupId: toCamelCase(entry.name),
+        creator: entry
+      });
+      emitter.namespace.push("requirement");
+      const tools = [
+        ...this.tools
+      ];
+      await emitter.emit("init", {
+        tools
+      });
+      await entry.init(tools, this.context);
+    }
+  }
+  findToolByName(name) {
+    const tool = this.tools.find((t) => t.name === name);
+    if (!tool) {
+      throw new Error(`Tool '${name}' not found in (${this.tools.map((t) => t.name).join(", ")}).`);
+    }
+    return tool;
+  }
+  async createRequest(state, forceTool, extraRules = []) {
+    const hidden = [];
+    const allowed = [];
+    const allTools = [
+      ...this.tools
+    ];
+    const reasonByTool = /* @__PURE__ */ new WeakMap();
+    let preventStop = false;
+    const preventStepRefs = [];
+    let forced = null;
+    let forcedLevel = 0;
+    const rulesByTool = /* @__PURE__ */ new Map();
+    for (const tool of this.tools) {
+      rulesByTool.set(tool.name, []);
+    }
+    for (const requirement of this.entries.filter((e) => e.enabled)) {
+      const generatedRules = await requirement.run(state);
+      for (const rule of generatedRules) {
+        const tool = this.findToolByName(rule.target);
+        const rules = rulesByTool.get(tool.name) || [];
+        rules.push({
+          priority: requirement.priority,
+          rule,
+          requirement
+        });
+        rulesByTool.set(tool.name, rules);
+      }
+    }
+    for (const rule of extraRules) {
+      if (!rulesByTool.has(rule.target)) {
+        throw new Error(`Tool '${rule.target}' not found.`);
+      }
+      const rules = rulesByTool.get(rule.target);
+      const priority = rules.length > 0 ? Math.max(...rules.map((r) => r.priority)) + 1 : 1;
+      rules.push({
+        priority,
+        rule,
+        requirement: null
+      });
+    }
+    for (const [toolName, rules] of rulesByTool.entries()) {
+      const tool = this.findToolByName(toolName);
+      rules.sort((a, b) => b.priority - a.priority);
+      const maxPriority = rules.length > 0 ? rules[0].priority : 1;
+      let isAllowed = true;
+      let isForced = false;
+      let isHidden = false;
+      let isPreventStop = false;
+      for (const ruleEntry of rules) {
+        const rule = ruleEntry.rule;
+        if (!rule.allowed) {
+          isAllowed = false;
+        }
+        if (rule.hidden) {
+          isHidden = true;
+        }
+        if (rule.forced) {
+          isForced = true;
+        }
+        if (rule.preventStop) {
+          isPreventStop = true;
+          preventStepRefs.push(ruleEntry);
+        }
+        if (rule.reason) {
+          reasonByTool.set(tool, rule.reason);
+        }
+      }
+      if (isAllowed && isHidden) {
+        isAllowed = false;
+      }
+      if (isAllowed) {
+        if (!allowed.includes(tool)) {
+          allowed.push(tool);
+        }
+        if (isForced && (!forced || forcedLevel < maxPriority)) {
+          forced = tool;
+          forcedLevel = maxPriority;
+        }
+      }
+      if (isHidden) {
+        if (!hidden.includes(tool)) {
+          hidden.push(tool);
+        }
+      }
+      if (isPreventStop) {
+        preventStop = true;
+      }
+    }
+    if (forced) {
+      allowed.length = 0;
+      if (!allowed.includes(forced)) {
+        allowed.push(forced);
+      }
+      if (!allowed.includes(this.finalAnswer)) {
+        allowed.push(this.finalAnswer);
+      }
+    }
+    if (preventStop && forced !== this.finalAnswer) {
+      const finalAnswerIndex = allowed.indexOf(this.finalAnswer);
+      if (finalAnswerIndex !== -1) {
+        allowed.splice(finalAnswerIndex, 1);
+      }
+    }
+    if (allowed.length === 0) {
+      throw new AgentError("One of the generated rules is preventing the agent from continuing. This indicates that the provided requirements may conflict with each other. See the following rules and their attached requirements that are preventing the agent from continuing.\n" + JSON.stringify(preventStepRefs, null, 2));
+    }
+    let toolChoice = forced || "required";
+    if (allowed.length === 1) {
+      toolChoice = allowed[0];
+    }
+    if (!(toolChoice instanceof Tool) && !forceTool && !preventStop) {
+      toolChoice = "auto";
+    }
+    return {
+      tools: allTools,
+      allowedTools: allowed,
+      reasonByTool,
+      toolChoice,
+      finalAnswer: this.finalAnswer,
+      hiddenTools: hidden,
+      canStop: !preventStop
+    };
+  }
+}
+async function createSystemMessage(template, request) {
+  return new SystemMessage(template.render({
+    tools: await Promise.all(request.tools.filter((tool) => !request.hiddenTools.includes(tool)).map(async (tool) => ({
+      name: tool.name,
+      description: tool.description,
+      inputSchema: JSON.stringify(await tool.inputSchema(), null, 2),
+      allowed: String(request.allowedTools.includes(tool)),
+      reason: request.reasonByTool.get(tool)
+    }))),
+    finalAnswerName: request.finalAnswer.name,
+    finalAnswerSchema: request.finalAnswer.customSchema ? JSON.stringify(request.finalAnswer.inputSchema, null, 2) : void 0,
+    finalAnswerInstructions: request.finalAnswer.instructions,
+    // default
+    role: void 0,
+    // default
+    instructions: void 0,
+    // default
+    notes: void 0
+  }));
+}
+__name(createSystemMessage, "createSystemMessage");
+
+export { RequirementsReasoner, createSystemMessage };
+//# sourceMappingURL=llm.js.map
+//# sourceMappingURL=llm.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../src/agents/requirement/utils/llm.ts"],"names":["RequirementsReasoner","tools","entries","context","finalAnswer","update","requirements","requirement","push","entry","emitter","child","groupId","toCamelCase","name","creator","namespace","emit","init","findToolByName","tool","find","t","Error","map","join","createRequest","state","forceTool","extraRules","hidden","allowed","allTools","reasonByTool","WeakMap","preventStop","preventStepRefs","forced","forcedLevel","rulesByTool","Map","set","filter","e","enabled","generatedRules","run","rule","target","rules","get","priority","has","length","Math","max","r","toolName","sort","a","b","maxPriority","isAllowed","isForced","isHidden","isPreventStop","ruleEntry","reason","includes","finalAnswerIndex","indexOf","splice","AgentError","JSON","stringify","toolChoice","Tool","allowedTools","hiddenTools","canStop","createSystemMessage","template","request","SystemMessage","render","Promise","all","description","inputSchema","String","finalAnswerName","finalAnswerSchema","customSchema","undefined","finalAnswerInstructions","instructions","role","notes"],"mappings":";;;;;;;AAiCO,MAAMA,oBAAAA,CAAAA;EAjCb;;;AAkCYC,EAAAA,KAAAA;AACAC,EAAAA,OAAAA,GAAyB,EAAA;AACzBC,EAAAA,OAAAA;AACMC,EAAAA,WAAAA;EAEhB,WAAA,CAAYH,KAAAA,EAAkBG,aAA8BD,OAAAA,EAA0B;AACpF,IAAA,IAAA,CAAKF,KAAAA,GAAQ;AAAIA,MAAAA,GAAAA,KAAAA;AAAOG,MAAAA;;AACxB,IAAA,IAAA,CAAKA,WAAAA,GAAcA,WAAAA;AACnB,IAAA,IAAA,CAAKD,OAAAA,GAAUA,OAAAA;AACjB;AAEA,EAAA,MAAME,OAAOC,YAAAA,EAA4C;AACvD,IAAA,IAAA,CAAKJ,UAAU,EAAA;AAEf,IAAA,KAAA,MAAWK,eAAeD,YAAAA,EAAc;AACtC,MAAA,IAAA,CAAKJ,OAAAA,CAAQM,KAAKD,WAAAA,CAAAA;AACpB;AAEA,IAAA,KAAA,MAAWE,KAAAA,IAAS,KAAKP,OAAAA,EAAS;AAChC,MAAA,MAAMQ,OAAAA,GAAU,IAAA,CAAKP,OAAAA,CAAQO,OAAAA,CAAQC,KAAAA,CAA4B;QAC/DC,OAAAA,EAASC,WAAAA,CAAYJ,MAAMK,IAAI,CAAA;QAC/BC,OAAAA,EAASN;OACX,CAAA;AACAC,MAAAA,OAAAA,CAAQM,SAAAA,CAAUR,KAAK,aAAA,CAAA;AAEvB,MAAA,MAAMP,KAAAA,GAAQ;WAAI,IAAA,CAAKA;;AACvB,MAAA,MAAMS,OAAAA,CAAQO,KAAK,MAAA,EAAQ;AAAEhB,QAAAA;OAAM,CAAA;AACnC,MAAA,MAAMQ,KAAAA,CAAMS,IAAAA,CAAKjB,KAAAA,EAAO,IAAA,CAAKE,OAAO,CAAA;AACtC;AACF;AAEUgB,EAAAA,cAAAA,CAAeL,IAAAA,EAAuB;AAC9C,IAAA,MAAMM,IAAAA,GAAO,KAAKnB,KAAAA,CAAMoB,IAAAA,CAAK,CAACC,CAAAA,KAAMA,CAAAA,CAAER,SAASA,IAAAA,CAAAA;AAC/C,IAAA,IAAI,CAACM,IAAAA,EAAM;AACT,MAAA,MAAM,IAAIG,KAAAA,CAAM,CAAA,MAAA,EAAST,IAAAA,CAAAA,gBAAAA,EAAuB,KAAKb,KAAAA,CAAMuB,GAAAA,CAAI,CAACF,CAAAA,KAAMA,EAAER,IAAI,CAAA,CAAEW,IAAAA,CAAK,IAAA,CAAA,CAAA,EAAA,CAAS,CAAA;AAC9F;AACA,IAAA,OAAOL,IAAAA;AACT;AAEA,EAAA,MAAMM,aAAAA,CACJC,KAAAA,EACAC,SAAAA,EACAC,UAAAA,GAAqB,EAAA,EACa;AAClC,IAAA,MAAMC,SAAoB,EAAA;AAC1B,IAAA,MAAMC,UAAqB,EAAA;AAC3B,IAAA,MAAMC,QAAAA,GAAsB;SAAI,IAAA,CAAK/B;;AACrC,IAAA,MAAMgC,YAAAA,uBAAmBC,OAAAA,EAAAA;AAEzB,IAAA,IAAIC,WAAAA,GAAc,KAAA;AAClB,IAAA,MAAMC,kBAA+B,EAAA;AAErC,IAAA,IAAIC,MAAAA,GAAyB,IAAA;AAC7B,IAAA,IAAIC,WAAAA,GAAc,CAAA;AAClB,IAAA,MAAMC,WAAAA,uBAAkBC,GAAAA,EAAAA;AAGxB,IAAA,KAAA,MAAWpB,IAAAA,IAAQ,KAAKnB,KAAAA,EAAO;AAC7BsC,MAAAA,WAAAA,CAAYE,GAAAA,CAAIrB,IAAAA,CAAKN,IAAAA,EAAM,EAAE,CAAA;AAC/B;AAGA,IAAA,KAAA,MAAWP,WAAAA,IAAe,KAAKL,OAAAA,CAAQwC,MAAAA,CAAO,CAACC,CAAAA,KAAMA,CAAAA,CAAEC,OAAO,CAAA,EAAG;AAC/D,MAAA,MAAMC,cAAAA,GAAiB,MAAMtC,WAAAA,CAAYuC,GAAAA,CAAInB,KAAAA,CAAAA;AAC7C,MAAA,KAAA,MAAWoB,QAAQF,cAAAA,EAAgB;AACjC,QAAA,MAAMzB,IAAAA,GAAO,IAAA,CAAKD,cAAAA,CAAe4B,IAAAA,CAAKC,MAAM,CAAA;AAC5C,QAAA,MAAMC,QAAQV,WAAAA,CAAYW,GAAAA,CAAI9B,IAAAA,CAAKN,IAAI,KAAK,EAAA;AAC5CmC,QAAAA,KAAAA,CAAMzC,IAAAA,CAAK;AACT2C,UAAAA,QAAAA,EAAU5C,WAAAA,CAAY4C,QAAAA;AACtBJ,UAAAA,IAAAA;AACAxC,UAAAA;SACF,CAAA;AACAgC,QAAAA,WAAAA,CAAYE,GAAAA,CAAIrB,IAAAA,CAAKN,IAAAA,EAAMmC,KAAAA,CAAAA;AAC7B;AACF;AAGA,IAAA,KAAA,MAAWF,QAAQlB,UAAAA,EAAY;AAC7B,MAAA,IAAI,CAACU,WAAAA,CAAYa,GAAAA,CAAIL,IAAAA,CAAKC,MAAM,CAAA,EAAG;AACjC,QAAA,MAAM,IAAIzB,KAAAA,CAAM,CAAA,MAAA,EAASwB,IAAAA,CAAKC,MAAM,CAAA,YAAA,CAAc,CAAA;AACpD;AAEA,MAAA,MAAMC,KAAAA,GAAQV,WAAAA,CAAYW,GAAAA,CAAIH,IAAAA,CAAKC,MAAM,CAAA;AACzC,MAAA,MAAMG,QAAAA,GAAWF,KAAAA,CAAMI,MAAAA,GAAS,CAAA,GAAIC,KAAKC,GAAAA,CAAG,GAAIN,KAAAA,CAAMzB,GAAAA,CAAI,CAACgC,CAAAA,KAAMA,CAAAA,CAAEL,QAAQ,CAAA,IAAK,CAAA,GAAI,CAAA;AACpFF,MAAAA,KAAAA,CAAMzC,IAAAA,CAAK;AACT2C,QAAAA,QAAAA;AACAJ,QAAAA,IAAAA;QACAxC,WAAAA,EAAa;OACf,CAAA;AACF;AAGA,IAAA,KAAA,MAAW,CAACkD,QAAAA,EAAUR,KAAAA,CAAAA,IAAUV,WAAAA,CAAYrC,SAAO,EAAI;AACrD,MAAA,MAAMkB,IAAAA,GAAO,IAAA,CAAKD,cAAAA,CAAesC,QAAAA,CAAAA;AACjCR,MAAAA,KAAAA,CAAMS,KAAK,CAACC,CAAAA,EAAGC,MAAMA,CAAAA,CAAET,QAAAA,GAAWQ,EAAER,QAAQ,CAAA;AAE5C,MAAA,MAAMU,cAAcZ,KAAAA,CAAMI,MAAAA,GAAS,IAAIJ,KAAAA,CAAM,CAAA,EAAGE,QAAAA,GAAW,CAAA;AAC3D,MAAA,IAAIW,SAAAA,GAAY,IAAA;AAChB,MAAA,IAAIC,QAAAA,GAAW,KAAA;AACf,MAAA,IAAIC,QAAAA,GAAW,KAAA;AACf,MAAA,IAAIC,aAAAA,GAAgB,KAAA;AAEpB,MAAA,KAAA,MAAWC,aAAajB,KAAAA,EAAO;AAC7B,QAAA,MAAMF,OAAOmB,SAAAA,CAAUnB,IAAAA;AACvB,QAAA,IAAI,CAACA,KAAKhB,OAAAA,EAAS;AACjB+B,UAAAA,SAAAA,GAAY,KAAA;AACd;AACA,QAAA,IAAIf,KAAKjB,MAAAA,EAAQ;AACfkC,UAAAA,QAAAA,GAAW,IAAA;AACb;AACA,QAAA,IAAIjB,KAAKV,MAAAA,EAAQ;AACf0B,UAAAA,QAAAA,GAAW,IAAA;AACb;AACA,QAAA,IAAIhB,KAAKZ,WAAAA,EAAa;AACpB8B,UAAAA,aAAAA,GAAgB,IAAA;AAChB7B,UAAAA,eAAAA,CAAgB5B,KAAK0D,SAAAA,CAAAA;AACvB;AACA,QAAA,IAAInB,KAAKoB,MAAAA,EAAQ;AACflC,UAAAA,YAAAA,CAAaQ,GAAAA,CAAIrB,IAAAA,EAAM2B,IAAAA,CAAKoB,MAAM,CAAA;AACpC;AACF;AAEA,MAAA,IAAIL,aAAaE,QAAAA,EAAU;AACzBF,QAAAA,SAAAA,GAAY,KAAA;AACd;AAEA,MAAA,IAAIA,SAAAA,EAAW;AACb,QAAA,IAAI,CAAC/B,OAAAA,CAAQqC,QAAAA,CAAShD,IAAAA,CAAAA,EAAO;AAC3BW,UAAAA,OAAAA,CAAQvB,KAAKY,IAAAA,CAAAA;AACf;AACA,QAAA,IAAI2C,QAAAA,KAAa,CAAC1B,MAAAA,IAAUC,WAAAA,GAAcuB,WAAAA,CAAAA,EAAc;AACtDxB,UAAAA,MAAAA,GAASjB,IAAAA;AACTkB,UAAAA,WAAAA,GAAcuB,WAAAA;AAChB;AACF;AACA,MAAA,IAAIG,QAAAA,EAAU;AACZ,QAAA,IAAI,CAAClC,MAAAA,CAAOsC,QAAAA,CAAShD,IAAAA,CAAAA,EAAO;AAC1BU,UAAAA,MAAAA,CAAOtB,KAAKY,IAAAA,CAAAA;AACd;AACF;AACA,MAAA,IAAI6C,aAAAA,EAAe;AACjB9B,QAAAA,WAAAA,GAAc,IAAA;AAChB;AACF;AAGA,IAAA,IAAIE,MAAAA,EAAQ;AACVN,MAAAA,OAAAA,CAAQsB,MAAAA,GAAS,CAAA;AACjB,MAAA,IAAI,CAACtB,OAAAA,CAAQqC,QAAAA,CAAS/B,MAAAA,CAAAA,EAAS;AAC7BN,QAAAA,OAAAA,CAAQvB,KAAK6B,MAAAA,CAAAA;AACf;AACA,MAAA,IAAI,CAACN,OAAAA,CAAQqC,QAAAA,CAAS,IAAA,CAAKhE,WAAW,CAAA,EAAU;AAC9C2B,QAAAA,OAAAA,CAAQvB,IAAAA,CAAK,KAAKJ,WAAW,CAAA;AAC/B;AACF;AAGA,IAAA,IAAI+B,WAAAA,IAAeE,MAAAA,KAAW,IAAA,CAAKjC,WAAAA,EAAa;AAC9C,MAAA,MAAMiE,gBAAAA,GAAmBtC,OAAAA,CAAQuC,OAAAA,CAAQ,IAAA,CAAKlE,WAAW,CAAA;AACzD,MAAA,IAAIiE,qBAAqB,EAAA,EAAI;AAC3BtC,QAAAA,OAAAA,CAAQwC,MAAAA,CAAOF,kBAAkB,CAAA,CAAA;AACnC;AACF;AAEA,IAAA,IAAItC,OAAAA,CAAQsB,WAAW,CAAA,EAAG;AACxB,MAAA,MAAM,IAAImB,WACR,0PAAA,GAGEC,IAAAA,CAAKC,UAAUtC,eAAAA,EAAiB,IAAA,EAAM,CAAA,CAAA,CAAA;AAE5C;AAEA,IAAA,IAAIuC,aAAkCtC,MAAAA,IAAU,UAAA;AAChD,IAAA,IAAIN,OAAAA,CAAQsB,WAAW,CAAA,EAAG;AACxBsB,MAAAA,UAAAA,GAAa5C,QAAQ,CAAA,CAAA;AACvB;AAGA,IAAA,IAAI,EAAE4C,UAAAA,YAAsBC,IAAAA,CAAAA,IAAS,CAAChD,SAAAA,IAAa,CAACO,WAAAA,EAAa;AAC/DwC,MAAAA,UAAAA,GAAa,MAAA;AACf;AAEA,IAAA,OAAO;MACL1E,KAAAA,EAAO+B,QAAAA;MACP6C,YAAAA,EAAc9C,OAAAA;AACdE,MAAAA,YAAAA;AACA0C,MAAAA,UAAAA;AACAvE,MAAAA,WAAAA,EAAa,IAAA,CAAKA,WAAAA;MAClB0E,WAAAA,EAAahD,MAAAA;AACbiD,MAAAA,OAAAA,EAAS,CAAC5C;AACZ,KAAA;AACF;AACF;AAKA,eAAsB6C,mBAAAA,CACpBC,UACAC,OAAAA,EAAgC;AAEhC,EAAA,OAAO,IAAIC,aAAAA,CACTF,QAAAA,CAASG,MAAAA,CAAO;AACdnF,IAAAA,KAAAA,EAAO,MAAMoF,OAAAA,CAAQC,GAAAA,CACnBJ,OAAAA,CAAQjF,KAAAA,CACLyC,OAAO,CAACtB,IAAAA,KAAS,CAAC8D,OAAAA,CAAQJ,YAAYV,QAAAA,CAAShD,IAAAA,CAAAA,CAAAA,CAC/CI,GAAAA,CAAI,OAAOJ,IAAAA,MAAU;AACpBN,MAAAA,IAAAA,EAAMM,IAAAA,CAAKN,IAAAA;AACXyE,MAAAA,WAAAA,EAAanE,IAAAA,CAAKmE,WAAAA;AAClBC,MAAAA,WAAAA,EAAaf,KAAKC,SAAAA,CAAU,MAAMtD,KAAKoE,WAAAA,EAAW,EAAI,MAAM,CAAA,CAAA;AAC5DzD,MAAAA,OAAAA,EAAS0D,MAAAA,CAAOP,OAAAA,CAAQL,YAAAA,CAAaT,QAAAA,CAAShD,IAAAA,CAAAA,CAAAA;MAC9C+C,MAAAA,EAAQe,OAAAA,CAAQjD,YAAAA,CAAaiB,GAAAA,CAAI9B,IAAAA;AACnC,KAAA,CAAA,CAAA,CAAA;AAEJsE,IAAAA,eAAAA,EAAiBR,QAAQ9E,WAAAA,CAAYU,IAAAA;IACrC6E,iBAAAA,EAAmBT,OAAAA,CAAQ9E,WAAAA,CAAYwF,YAAAA,GACnCnB,IAAAA,CAAKC,SAAAA,CAAUQ,QAAQ9E,WAAAA,CAAYoF,WAAAA,EAAa,IAAA,EAAM,CAAA,CAAA,GACtDK,MAAAA;AACJC,IAAAA,uBAAAA,EAAyBZ,QAAQ9E,WAAAA,CAAY2F,YAAAA;;IAC7CC,IAAAA,EAAMH,MAAAA;;IACNE,YAAAA,EAAcF,MAAAA;;IACdI,KAAAA,EAAOJ;AACT,GAAA,CAAA,CAAA;AAEJ;AA3BsBb,MAAAA,CAAAA,mBAAAA,EAAAA,qBAAAA,CAAAA","file":"llm.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { AnyTool, Tool } from \"@/tools/base.js\";\nimport { RunContext } from \"@/context.js\";\nimport { AgentError } from \"@/agents/base.js\";\nimport { SystemMessage } from \"@/backend/message.js\";\nimport { PromptTemplate } from \"@/template.js\";\nimport {\n  RequirementAgentRequest,\n  RequirementAgentRunState,\n  RequirementAgentSystemPromptInputSchema,\n} from \"@/agents/requirement/types.js\";\nimport {\n  Requirement,\n  RequirementCallbacks,\n  Rule,\n} from \"@/agents/requirement/requirements/requirement.js\";\nimport { FinalAnswerTool } from \"./tool.js\";\nimport { ChatModelToolChoice } from \"@/backend/chat.js\";\nimport { toCamelCase } from \"remeda\";\n\ninterface RuleEntry {\n  rule: Rule;\n  requirement: Requirement | null;\n  priority: number;\n}\n\n/**\n * Manages requirements and generates requests for the agent\n */\nexport class RequirementsReasoner {\n  protected tools: AnyTool[];\n  protected entries: Requirement[] = [];\n  protected context: RunContext<any>;\n  public readonly finalAnswer: FinalAnswerTool;\n\n  constructor(tools: AnyTool[], finalAnswer: FinalAnswerTool, context: RunContext<any>) {\n    this.tools = [...tools, finalAnswer];\n    this.finalAnswer = finalAnswer;\n    this.context = context;\n  }\n\n  async update(requirements: Requirement[]): Promise<void> {\n    this.entries = [];\n\n    for (const requirement of requirements) {\n      this.entries.push(requirement);\n    }\n\n    for (const entry of this.entries) {\n      const emitter = this.context.emitter.child<RequirementCallbacks>({\n        groupId: toCamelCase(entry.name),\n        creator: entry,\n      });\n      emitter.namespace.push(\"requirement\");\n\n      const tools = [...this.tools];\n      await emitter.emit(\"init\", { tools });\n      await entry.init(tools, this.context);\n    }\n  }\n\n  protected findToolByName(name: string): AnyTool {\n    const tool = this.tools.find((t) => t.name === name);\n    if (!tool) {\n      throw new Error(`Tool '${name}' not found in (${this.tools.map((t) => t.name).join(\", \")}).`);\n    }\n    return tool;\n  }\n\n  async createRequest(\n    state: RequirementAgentRunState,\n    forceTool: boolean,\n    extraRules: Rule[] = [],\n  ): Promise<RequirementAgentRequest> {\n    const hidden: AnyTool[] = [];\n    const allowed: AnyTool[] = [];\n    const allTools: AnyTool[] = [...this.tools];\n    const reasonByTool = new WeakMap<AnyTool, string | undefined>();\n\n    let preventStop = false;\n    const preventStepRefs: RuleEntry[] = [];\n\n    let forced: AnyTool | null = null;\n    let forcedLevel = 0;\n    const rulesByTool = new Map<string, RuleEntry[]>();\n\n    // Initialize rules map\n    for (const tool of this.tools) {\n      rulesByTool.set(tool.name, []);\n    }\n\n    // Group rules from requirements\n    for (const requirement of this.entries.filter((e) => e.enabled)) {\n      const generatedRules = await requirement.run(state);\n      for (const rule of generatedRules) {\n        const tool = this.findToolByName(rule.target);\n        const rules = rulesByTool.get(tool.name) || [];\n        rules.push({\n          priority: requirement.priority,\n          rule,\n          requirement,\n        });\n        rulesByTool.set(tool.name, rules);\n      }\n    }\n\n    // Add extra rules\n    for (const rule of extraRules) {\n      if (!rulesByTool.has(rule.target)) {\n        throw new Error(`Tool '${rule.target}' not found.`);\n      }\n\n      const rules = rulesByTool.get(rule.target)!;\n      const priority = rules.length > 0 ? Math.max(...rules.map((r) => r.priority)) + 1 : 1;\n      rules.push({\n        priority,\n        rule,\n        requirement: null,\n      });\n    }\n\n    // Aggregate rules and infer required tool\n    for (const [toolName, rules] of rulesByTool.entries()) {\n      const tool = this.findToolByName(toolName);\n      rules.sort((a, b) => b.priority - a.priority); // DESC\n\n      const maxPriority = rules.length > 0 ? rules[0].priority : 1;\n      let isAllowed = true;\n      let isForced = false;\n      let isHidden = false;\n      let isPreventStop = false;\n\n      for (const ruleEntry of rules) {\n        const rule = ruleEntry.rule;\n        if (!rule.allowed) {\n          isAllowed = false;\n        }\n        if (rule.hidden) {\n          isHidden = true;\n        }\n        if (rule.forced) {\n          isForced = true;\n        }\n        if (rule.preventStop) {\n          isPreventStop = true;\n          preventStepRefs.push(ruleEntry);\n        }\n        if (rule.reason) {\n          reasonByTool.set(tool, rule.reason);\n        }\n      }\n\n      if (isAllowed && isHidden) {\n        isAllowed = false;\n      }\n\n      if (isAllowed) {\n        if (!allowed.includes(tool)) {\n          allowed.push(tool);\n        }\n        if (isForced && (!forced || forcedLevel < maxPriority)) {\n          forced = tool;\n          forcedLevel = maxPriority;\n        }\n      }\n      if (isHidden) {\n        if (!hidden.includes(tool)) {\n          hidden.push(tool);\n        }\n      }\n      if (isPreventStop) {\n        preventStop = true;\n      }\n    }\n\n    // If forced tool is set, only allow it and final answer\n    if (forced) {\n      allowed.length = 0;\n      if (!allowed.includes(forced)) {\n        allowed.push(forced);\n      }\n      if (!allowed.includes(this.finalAnswer as any)) {\n        allowed.push(this.finalAnswer as any);\n      }\n    }\n\n    // If prevent stop is set and forced is not final answer, remove final answer\n    if (preventStop && forced !== this.finalAnswer) {\n      const finalAnswerIndex = allowed.indexOf(this.finalAnswer);\n      if (finalAnswerIndex !== -1) {\n        allowed.splice(finalAnswerIndex, 1);\n      }\n    }\n\n    if (allowed.length === 0) {\n      throw new AgentError(\n        \"One of the generated rules is preventing the agent from continuing. \" +\n          \"This indicates that the provided requirements may conflict with each other. \" +\n          \"See the following rules and their attached requirements that are preventing the agent from continuing.\\n\" +\n          JSON.stringify(preventStepRefs, null, 2),\n      );\n    }\n\n    let toolChoice: ChatModelToolChoice = forced || \"required\";\n    if (allowed.length === 1) {\n      toolChoice = allowed[0];\n    }\n\n    // Override to auto if not forcing and not preventing stop\n    if (!(toolChoice instanceof Tool) && !forceTool && !preventStop) {\n      toolChoice = \"auto\";\n    }\n\n    return {\n      tools: allTools,\n      allowedTools: allowed,\n      reasonByTool,\n      toolChoice,\n      finalAnswer: this.finalAnswer,\n      hiddenTools: hidden,\n      canStop: !preventStop,\n    };\n  }\n}\n\n/**\n * Create system message from template and request\n */\nexport async function createSystemMessage(\n  template: PromptTemplate<typeof RequirementAgentSystemPromptInputSchema>,\n  request: RequirementAgentRequest,\n): Promise<SystemMessage> {\n  return new SystemMessage(\n    template.render({\n      tools: await Promise.all(\n        request.tools\n          .filter((tool) => !request.hiddenTools.includes(tool))\n          .map(async (tool) => ({\n            name: tool.name,\n            description: tool.description,\n            inputSchema: JSON.stringify(await tool.inputSchema(), null, 2),\n            allowed: String(request.allowedTools.includes(tool)),\n            reason: request.reasonByTool.get(tool),\n          })),\n      ),\n      finalAnswerName: request.finalAnswer.name,\n      finalAnswerSchema: request.finalAnswer.customSchema\n        ? JSON.stringify(request.finalAnswer.inputSchema, null, 2)\n        : undefined,\n      finalAnswerInstructions: request.finalAnswer.instructions, // default\n      role: undefined, // default\n      instructions: undefined, // default\n      notes: undefined, // default\n    }),\n  );\n}\n"]}