beeai-framework 0.1.21 → 0.1.22

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-Cx6f43du.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-CGKd_-kK.cjs';
+import '../../../internals/helpers/promise.cjs';
+import 'ai';
+import '../../../memory/base.cjs';
+import '../../../chat-DB_cf15S.cjs';
+import '../../../backend/constants.cjs';
+import '../../../logger/logger.cjs';
+import 'pino';
+import '../../../base-v-PYIq5G.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-BXkkbABd.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-CN7I4uSV.js';
+import '../../../internals/helpers/promise.js';
+import 'ai';
+import '../../../memory/base.js';
+import '../../../chat-BbCYMAdE.js';
+import '../../../backend/constants.js';
+import '../../../logger/logger.js';
+import 'pino';
+import '../../../base-BurRTBA7.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"]}

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

@@ -0,0 +1,94 @@
+'use strict';
+
+var base_cjs = require('../../../tools/base.cjs');
+var errors_cjs = require('../../../errors.cjs');
+var emitter_cjs = require('../../../emitter/emitter.cjs');
+var zod = require('zod');
+var message_cjs = require('../../../backend/message.cjs');
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+async function runTool(tools, msg, context) {
+  const result = {
+    msg,
+    tool: null,
+    input: msg.input,
+    output: new base_cjs.StringToolOutput(""),
+    error: null
+  };
+  try {
+    result.tool = tools.find((tool) => tool.name === msg.toolName) || null;
+    if (!result.tool) {
+      throw new base_cjs.ToolError(`Tool '${msg.toolName}' does not exist!`);
+    }
+    result.output = await result.tool.run(result.input).context({
+      ...context,
+      toolCallMsg: msg
+    });
+  } catch (e) {
+    if (e instanceof base_cjs.ToolError) {
+      result.error = errors_cjs.FrameworkError.ensure(e);
+    } else {
+      throw e;
+    }
+  }
+  return result;
+}
+__name(runTool, "runTool");
+const FinalAnswerToolSchema = zod.z.object({
+  response: zod.z.string().describe("The final answer to the user")
+});
+class FinalAnswerTool extends base_cjs.Tool {
+  static {
+    __name(this, "FinalAnswerTool");
+  }
+  name = "final_answer";
+  description = "Sends the final answer to the user";
+  instructions;
+  customSchema;
+  emitter;
+  expectedOutput;
+  state;
+  constructor(expectedOutput, state) {
+    super();
+    this.expectedOutput = expectedOutput;
+    this.state = state;
+    this.instructions = typeof expectedOutput === "string" ? expectedOutput : void 0;
+    this.customSchema = expectedOutput instanceof zod.ZodSchema;
+    this.emitter = emitter_cjs.Emitter.root.child({
+      namespace: [
+        "tool",
+        "final_answer"
+      ],
+      creator: this
+    });
+  }
+  inputSchema() {
+    const expectedOutput = this.expectedOutput;
+    if (!expectedOutput) {
+      return FinalAnswerToolSchema;
+    } else if (expectedOutput instanceof zod.ZodSchema) {
+      return expectedOutput;
+    } else if (typeof expectedOutput === "string") {
+      return zod.z.object({
+        response: zod.z.string().describe(expectedOutput)
+      });
+    } else {
+      return FinalAnswerToolSchema;
+    }
+  }
+  async _run(input, _options, _ctx) {
+    this.state.result = input;
+    if (this.expectedOutput instanceof zod.ZodSchema) {
+      this.state.answer = new message_cjs.AssistantMessage(JSON.stringify(input));
+    } else {
+      this.state.answer = new message_cjs.AssistantMessage(input.response);
+    }
+    return new base_cjs.StringToolOutput("Message has been sent");
+  }
+}
+
+exports.FinalAnswerTool = FinalAnswerTool;
+exports.runTool = runTool;
+//# sourceMappingURL=tool.cjs.map
+//# sourceMappingURL=tool.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../src/agents/requirement/utils/tool.ts"],"names":["runTool","tools","msg","context","result","tool","input","output","StringToolOutput","error","find","name","toolName","ToolError","run","toolCallMsg","e","FrameworkError","ensure","FinalAnswerToolSchema","z","object","response","string","describe","FinalAnswerTool","Tool","description","instructions","customSchema","emitter","expectedOutput","state","undefined","ZodSchema","Emitter","root","child","namespace","creator","inputSchema","_run","_options","_ctx","answer","AssistantMessage","JSON","stringify"],"mappings":";;;;;;;;;;AAiCA,eAAsBA,OAAAA,CACpBC,KAAAA,EACAC,GAAAA,EACAC,OAAAA,EAA4B;AAE5B,EAAA,MAAMC,MAAAA,GAA+B;AACnCF,IAAAA,GAAAA;IACAG,IAAAA,EAAM,IAAA;AACNC,IAAAA,KAAAA,EAAOJ,GAAAA,CAAII,KAAAA;IACXC,MAAAA,EAAQ,IAAIC,0BAAiB,EAAA,CAAA;IAC7BC,KAAAA,EAAO;AACT,GAAA;AAEA,EAAA,IAAI;AACFL,IAAAA,MAAAA,CAAOC,IAAAA,GAAOJ,MAAMS,IAAAA,CAAK,CAACL,SAASA,IAAAA,CAAKM,IAAAA,KAAST,GAAAA,CAAIU,QAAQ,CAAA,IAAK,IAAA;AAClE,IAAA,IAAI,CAACR,OAAOC,IAAAA,EAAM;AAChB,MAAA,MAAM,IAAIQ,kBAAAA,CAAU,CAAA,MAAA,EAASX,GAAAA,CAAIU,QAAQ,CAAA,iBAAA,CAAmB,CAAA;AAC9D;AAEAR,IAAAA,MAAAA,CAAOG,MAAAA,GAAS,MAAMH,MAAAA,CAAOC,IAAAA,CAAKS,IAAIV,MAAAA,CAAOE,KAAK,EAAEH,OAAAA,CAAQ;MAC1D,GAAGA,OAAAA;MACHY,WAAAA,EAAab;KACf,CAAA;AACF,GAAA,CAAA,OAASc,CAAAA,EAAG;AACV,IAAA,IAAIA,aAAaH,kBAAAA,EAAW;AAC1BT,MAAAA,MAAAA,CAAOK,KAAAA,GAAQQ,yBAAAA,CAAeC,MAAAA,CAAOF,CAAAA,CAAAA;KACvC,MAAO;AACL,MAAA,MAAMA,CAAAA;AACR;AACF;AAEA,EAAA,OAAOZ,MAAAA;AACT;AAhCsBJ,MAAAA,CAAAA,OAAAA,EAAAA,SAAAA,CAAAA;AAmCtB,MAAMmB,qBAAAA,GAAwBC,MAAEC,MAAAA,CAAO;AACrCC,EAAAA,QAAAA,EAAUF,KAAAA,CAAEG,MAAAA,EAAM,CAAGC,QAAAA,CAAS,8BAAA;AAChC,CAAA,CAAA;AAKO,MAAMC,wBAAwBC,aAAAA,CAAAA;EA3ErC;;;EA4EkBf,IAAAA,GAAO,cAAA;EACPgB,WAAAA,GAAc,oCAAA;AACvBC,EAAAA,YAAAA;AACAC,EAAAA,YAAAA;AACSC,EAAAA,OAAAA;AAENC,EAAAA,cAAAA;AACAC,EAAAA,KAAAA;AAEV,EAAA,WAAA,CAAYD,gBAA2CC,KAAAA,EAAiC;AACtF,IAAA,KAAA,EAAK;AACL,IAAA,IAAA,CAAKD,cAAAA,GAAiBA,cAAAA;AACtB,IAAA,IAAA,CAAKC,KAAAA,GAAQA,KAAAA;AACb,IAAA,IAAA,CAAKJ,YAAAA,GAAe,OAAOG,cAAAA,KAAmB,QAAA,GAAWA,cAAAA,GAAiBE,MAAAA;AAC1E,IAAA,IAAA,CAAKJ,eAAeE,cAAAA,YAA0BG,aAAAA;AAC9C,IAAA,IAAA,CAAKJ,OAAAA,GAAUK,mBAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAAM;MAChCC,SAAAA,EAAW;AAAC,QAAA,MAAA;AAAQ,QAAA;;MACpBC,OAAAA,EAAS;KACX,CAAA;AACF;EAEAC,WAAAA,GAAyB;AACvB,IAAA,MAAMT,iBAAiB,IAAA,CAAKA,cAAAA;AAE5B,IAAA,IAAI,CAACA,cAAAA,EAAgB;AACnB,MAAA,OAAOZ,qBAAAA;AACT,KAAA,MAAA,IAAWY,0BAA0BG,aAAAA,EAAW;AAC9C,MAAA,OAAOH,cAAAA;KACT,MAAA,IAAW,OAAOA,mBAAmB,QAAA,EAAU;AAC7C,MAAA,OAAOX,MAAEC,MAAAA,CAAO;AACdC,QAAAA,QAAAA,EAAUF,KAAAA,CAAEG,MAAAA,EAAM,CAAGC,QAAAA,CAASO,cAAAA;OAChC,CAAA;KACF,MAAO;AACL,MAAA,OAAOZ,qBAAAA;AACT;AACF;EAEA,MAAMsB,IAAAA,CACJnC,KAAAA,EACAoC,QAAAA,EACAC,IAAAA,EAC2B;AAC3B,IAAA,IAAA,CAAKX,MAAM5B,MAAAA,GAASE,KAAAA;AAEpB,IAAA,IAAI,IAAA,CAAKyB,0BAA0BG,aAAAA,EAAW;AAE5C,MAAA,IAAA,CAAKF,MAAMY,MAAAA,GAAS,IAAIC,6BAAiBC,IAAAA,CAAKC,SAAAA,CAAUzC,KAAAA,CAAAA,CAAAA;KAC1D,MAAO;AAEL,MAAA,IAAA,CAAK0B,KAAAA,CAAMY,MAAAA,GAAS,IAAIC,4BAAAA,CAAiBvC,MAAMgB,QAAQ,CAAA;AACzD;AAEA,IAAA,OAAO,IAAId,0BAAiB,uBAAA,CAAA;AAC9B;AACF","file":"tool.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport {\n  AnyTool,\n  BaseToolRunOptions,\n  StringToolOutput,\n  Tool,\n  ToolError,\n  ToolOutput,\n} from \"@/tools/base.js\";\nimport { ToolCallPart } from \"ai\";\nimport { FrameworkError } from \"@/errors.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { z, ZodSchema } from \"zod\";\nimport { RequirementAgentRunState } from \"@/agents/requirement/types.js\";\nimport { AssistantMessage } from \"@/backend/message.js\";\nimport { RunContext } from \"@/context.js\";\n\n// Tool invocation result\nexport interface ToolInvocationResult {\n  msg: ToolCallPart;\n  tool: AnyTool | null;\n  input: unknown;\n  output: ToolOutput;\n  error: FrameworkError | null;\n}\n\n/**\n * Run a single tool with error handling\n */\nexport async function runTool(\n  tools: AnyTool[],\n  msg: ToolCallPart,\n  context: Record<string, any>,\n): Promise<ToolInvocationResult> {\n  const result: ToolInvocationResult = {\n    msg,\n    tool: null,\n    input: msg.input,\n    output: new StringToolOutput(\"\"),\n    error: null,\n  };\n\n  try {\n    result.tool = tools.find((tool) => tool.name === msg.toolName) || null;\n    if (!result.tool) {\n      throw new ToolError(`Tool '${msg.toolName}' does not exist!`);\n    }\n\n    result.output = await result.tool.run(result.input).context({\n      ...context,\n      toolCallMsg: msg,\n    });\n  } catch (e) {\n    if (e instanceof ToolError) {\n      result.error = FrameworkError.ensure(e);\n    } else {\n      throw e;\n    }\n  }\n\n  return result;\n}\n\n// Final answer tool schema\nconst FinalAnswerToolSchema = z.object({\n  response: z.string().describe(\"The final answer to the user\"),\n});\n\n/**\n * Special tool for capturing final answers\n */\nexport class FinalAnswerTool extends Tool {\n  public readonly name = \"final_answer\";\n  public readonly description = \"Sends the final answer to the user\";\n  public instructions?: string;\n  public customSchema: boolean;\n  public readonly emitter: Emitter<any>;\n\n  protected expectedOutput: string | ZodSchema | null;\n  protected state: RequirementAgentRunState;\n\n  constructor(expectedOutput: string | ZodSchema | null, state: RequirementAgentRunState) {\n    super();\n    this.expectedOutput = expectedOutput;\n    this.state = state;\n    this.instructions = typeof expectedOutput === \"string\" ? expectedOutput : undefined;\n    this.customSchema = expectedOutput instanceof ZodSchema;\n    this.emitter = Emitter.root.child({\n      namespace: [\"tool\", \"final_answer\"],\n      creator: this,\n    });\n  }\n\n  inputSchema(): ZodSchema {\n    const expectedOutput = this.expectedOutput;\n\n    if (!expectedOutput) {\n      return FinalAnswerToolSchema;\n    } else if (expectedOutput instanceof ZodSchema) {\n      return expectedOutput;\n    } else if (typeof expectedOutput === \"string\") {\n      return z.object({\n        response: z.string().describe(expectedOutput),\n      });\n    } else {\n      return FinalAnswerToolSchema;\n    }\n  }\n\n  async _run(\n    input: any,\n    _options: BaseToolRunOptions,\n    _ctx: RunContext<typeof this>,\n  ): Promise<StringToolOutput> {\n    this.state.result = input;\n\n    if (this.expectedOutput instanceof ZodSchema) {\n      // For custom schemas, serialize the entire input\n      this.state.answer = new AssistantMessage(JSON.stringify(input));\n    } else {\n      // For string schemas, use the response field\n      this.state.answer = new AssistantMessage(input.response);\n    }\n\n    return new StringToolOutput(\"Message has been sent\");\n  }\n}\n"]}

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

@@ -0,0 +1,24 @@
+import '../../../tools/base.cjs';
+import 'ai';
+import '../../../errors.cjs';
+import '../../../emitter-CGKd_-kK.cjs';
+import 'zod';
+export { F as FinalAnswerTool, T as ToolInvocationResult, r as runTool } from '../../../types-Cx6f43du.cjs';
+import '../../../context.cjs';
+import 'ajv';
+import '../../../internals/serializable.cjs';
+import '../../../internals/types.cjs';
+import '../../../internals/helpers/guards.cjs';
+import 'promise-based-task';
+import '../../../cache/base.cjs';
+import '../../../internals/helpers/schema.cjs';
+import 'zod-to-json-schema';
+import '../../../internals/helpers/promise.cjs';
+import '../../../memory/base.cjs';
+import '../../../backend/message.cjs';
+import '../../../template.cjs';
+import '../../../chat-DB_cf15S.cjs';
+import '../../../backend/constants.cjs';
+import '../../../logger/logger.cjs';
+import 'pino';
+import '../../../base-v-PYIq5G.cjs';

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

@@ -0,0 +1,24 @@
+import '../../../tools/base.js';
+import 'ai';
+import '../../../errors.js';
+import '../../../emitter-CN7I4uSV.js';
+import 'zod';
+export { F as FinalAnswerTool, T as ToolInvocationResult, r as runTool } from '../../../types-BXkkbABd.js';
+import '../../../context.js';
+import 'ajv';
+import '../../../internals/serializable.js';
+import '../../../internals/types.js';
+import '../../../internals/helpers/guards.js';
+import 'promise-based-task';
+import '../../../cache/base.js';
+import '../../../internals/helpers/schema.js';
+import 'zod-to-json-schema';
+import '../../../internals/helpers/promise.js';
+import '../../../memory/base.js';
+import '../../../backend/message.js';
+import '../../../template.js';
+import '../../../chat-BbCYMAdE.js';
+import '../../../backend/constants.js';
+import '../../../logger/logger.js';
+import 'pino';
+import '../../../base-BurRTBA7.js';

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

@@ -0,0 +1,91 @@
+import { Tool, StringToolOutput, ToolError } from '../../../tools/base.js';
+import { FrameworkError } from '../../../errors.js';
+import { Emitter } from '../../../emitter/emitter.js';
+import { z, ZodSchema } from 'zod';
+import { AssistantMessage } from '../../../backend/message.js';
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+async function runTool(tools, msg, context) {
+  const result = {
+    msg,
+    tool: null,
+    input: msg.input,
+    output: new StringToolOutput(""),
+    error: null
+  };
+  try {
+    result.tool = tools.find((tool) => tool.name === msg.toolName) || null;
+    if (!result.tool) {
+      throw new ToolError(`Tool '${msg.toolName}' does not exist!`);
+    }
+    result.output = await result.tool.run(result.input).context({
+      ...context,
+      toolCallMsg: msg
+    });
+  } catch (e) {
+    if (e instanceof ToolError) {
+      result.error = FrameworkError.ensure(e);
+    } else {
+      throw e;
+    }
+  }
+  return result;
+}
+__name(runTool, "runTool");
+const FinalAnswerToolSchema = z.object({
+  response: z.string().describe("The final answer to the user")
+});
+class FinalAnswerTool extends Tool {
+  static {
+    __name(this, "FinalAnswerTool");
+  }
+  name = "final_answer";
+  description = "Sends the final answer to the user";
+  instructions;
+  customSchema;
+  emitter;
+  expectedOutput;
+  state;
+  constructor(expectedOutput, state) {
+    super();
+    this.expectedOutput = expectedOutput;
+    this.state = state;
+    this.instructions = typeof expectedOutput === "string" ? expectedOutput : void 0;
+    this.customSchema = expectedOutput instanceof ZodSchema;
+    this.emitter = Emitter.root.child({
+      namespace: [
+        "tool",
+        "final_answer"
+      ],
+      creator: this
+    });
+  }
+  inputSchema() {
+    const expectedOutput = this.expectedOutput;
+    if (!expectedOutput) {
+      return FinalAnswerToolSchema;
+    } else if (expectedOutput instanceof ZodSchema) {
+      return expectedOutput;
+    } else if (typeof expectedOutput === "string") {
+      return z.object({
+        response: z.string().describe(expectedOutput)
+      });
+    } else {
+      return FinalAnswerToolSchema;
+    }
+  }
+  async _run(input, _options, _ctx) {
+    this.state.result = input;
+    if (this.expectedOutput instanceof ZodSchema) {
+      this.state.answer = new AssistantMessage(JSON.stringify(input));
+    } else {
+      this.state.answer = new AssistantMessage(input.response);
+    }
+    return new StringToolOutput("Message has been sent");
+  }
+}
+
+export { FinalAnswerTool, runTool };
+//# sourceMappingURL=tool.js.map
+//# sourceMappingURL=tool.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../src/agents/requirement/utils/tool.ts"],"names":["runTool","tools","msg","context","result","tool","input","output","StringToolOutput","error","find","name","toolName","ToolError","run","toolCallMsg","e","FrameworkError","ensure","FinalAnswerToolSchema","z","object","response","string","describe","FinalAnswerTool","Tool","description","instructions","customSchema","emitter","expectedOutput","state","undefined","ZodSchema","Emitter","root","child","namespace","creator","inputSchema","_run","_options","_ctx","answer","AssistantMessage","JSON","stringify"],"mappings":";;;;;;;;AAiCA,eAAsBA,OAAAA,CACpBC,KAAAA,EACAC,GAAAA,EACAC,OAAAA,EAA4B;AAE5B,EAAA,MAAMC,MAAAA,GAA+B;AACnCF,IAAAA,GAAAA;IACAG,IAAAA,EAAM,IAAA;AACNC,IAAAA,KAAAA,EAAOJ,GAAAA,CAAII,KAAAA;IACXC,MAAAA,EAAQ,IAAIC,iBAAiB,EAAA,CAAA;IAC7BC,KAAAA,EAAO;AACT,GAAA;AAEA,EAAA,IAAI;AACFL,IAAAA,MAAAA,CAAOC,IAAAA,GAAOJ,MAAMS,IAAAA,CAAK,CAACL,SAASA,IAAAA,CAAKM,IAAAA,KAAST,GAAAA,CAAIU,QAAQ,CAAA,IAAK,IAAA;AAClE,IAAA,IAAI,CAACR,OAAOC,IAAAA,EAAM;AAChB,MAAA,MAAM,IAAIQ,SAAAA,CAAU,CAAA,MAAA,EAASX,GAAAA,CAAIU,QAAQ,CAAA,iBAAA,CAAmB,CAAA;AAC9D;AAEAR,IAAAA,MAAAA,CAAOG,MAAAA,GAAS,MAAMH,MAAAA,CAAOC,IAAAA,CAAKS,IAAIV,MAAAA,CAAOE,KAAK,EAAEH,OAAAA,CAAQ;MAC1D,GAAGA,OAAAA;MACHY,WAAAA,EAAab;KACf,CAAA;AACF,GAAA,CAAA,OAASc,CAAAA,EAAG;AACV,IAAA,IAAIA,aAAaH,SAAAA,EAAW;AAC1BT,MAAAA,MAAAA,CAAOK,KAAAA,GAAQQ,cAAAA,CAAeC,MAAAA,CAAOF,CAAAA,CAAAA;KACvC,MAAO;AACL,MAAA,MAAMA,CAAAA;AACR;AACF;AAEA,EAAA,OAAOZ,MAAAA;AACT;AAhCsBJ,MAAAA,CAAAA,OAAAA,EAAAA,SAAAA,CAAAA;AAmCtB,MAAMmB,qBAAAA,GAAwBC,EAAEC,MAAAA,CAAO;AACrCC,EAAAA,QAAAA,EAAUF,CAAAA,CAAEG,MAAAA,EAAM,CAAGC,QAAAA,CAAS,8BAAA;AAChC,CAAA,CAAA;AAKO,MAAMC,wBAAwBC,IAAAA,CAAAA;EA3ErC;;;EA4EkBf,IAAAA,GAAO,cAAA;EACPgB,WAAAA,GAAc,oCAAA;AACvBC,EAAAA,YAAAA;AACAC,EAAAA,YAAAA;AACSC,EAAAA,OAAAA;AAENC,EAAAA,cAAAA;AACAC,EAAAA,KAAAA;AAEV,EAAA,WAAA,CAAYD,gBAA2CC,KAAAA,EAAiC;AACtF,IAAA,KAAA,EAAK;AACL,IAAA,IAAA,CAAKD,cAAAA,GAAiBA,cAAAA;AACtB,IAAA,IAAA,CAAKC,KAAAA,GAAQA,KAAAA;AACb,IAAA,IAAA,CAAKJ,YAAAA,GAAe,OAAOG,cAAAA,KAAmB,QAAA,GAAWA,cAAAA,GAAiBE,MAAAA;AAC1E,IAAA,IAAA,CAAKJ,eAAeE,cAAAA,YAA0BG,SAAAA;AAC9C,IAAA,IAAA,CAAKJ,OAAAA,GAAUK,OAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAAM;MAChCC,SAAAA,EAAW;AAAC,QAAA,MAAA;AAAQ,QAAA;;MACpBC,OAAAA,EAAS;KACX,CAAA;AACF;EAEAC,WAAAA,GAAyB;AACvB,IAAA,MAAMT,iBAAiB,IAAA,CAAKA,cAAAA;AAE5B,IAAA,IAAI,CAACA,cAAAA,EAAgB;AACnB,MAAA,OAAOZ,qBAAAA;AACT,KAAA,MAAA,IAAWY,0BAA0BG,SAAAA,EAAW;AAC9C,MAAA,OAAOH,cAAAA;KACT,MAAA,IAAW,OAAOA,mBAAmB,QAAA,EAAU;AAC7C,MAAA,OAAOX,EAAEC,MAAAA,CAAO;AACdC,QAAAA,QAAAA,EAAUF,CAAAA,CAAEG,MAAAA,EAAM,CAAGC,QAAAA,CAASO,cAAAA;OAChC,CAAA;KACF,MAAO;AACL,MAAA,OAAOZ,qBAAAA;AACT;AACF;EAEA,MAAMsB,IAAAA,CACJnC,KAAAA,EACAoC,QAAAA,EACAC,IAAAA,EAC2B;AAC3B,IAAA,IAAA,CAAKX,MAAM5B,MAAAA,GAASE,KAAAA;AAEpB,IAAA,IAAI,IAAA,CAAKyB,0BAA0BG,SAAAA,EAAW;AAE5C,MAAA,IAAA,CAAKF,MAAMY,MAAAA,GAAS,IAAIC,iBAAiBC,IAAAA,CAAKC,SAAAA,CAAUzC,KAAAA,CAAAA,CAAAA;KAC1D,MAAO;AAEL,MAAA,IAAA,CAAK0B,KAAAA,CAAMY,MAAAA,GAAS,IAAIC,gBAAAA,CAAiBvC,MAAMgB,QAAQ,CAAA;AACzD;AAEA,IAAA,OAAO,IAAId,iBAAiB,uBAAA,CAAA;AAC9B;AACF","file":"tool.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport {\n  AnyTool,\n  BaseToolRunOptions,\n  StringToolOutput,\n  Tool,\n  ToolError,\n  ToolOutput,\n} from \"@/tools/base.js\";\nimport { ToolCallPart } from \"ai\";\nimport { FrameworkError } from \"@/errors.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { z, ZodSchema } from \"zod\";\nimport { RequirementAgentRunState } from \"@/agents/requirement/types.js\";\nimport { AssistantMessage } from \"@/backend/message.js\";\nimport { RunContext } from \"@/context.js\";\n\n// Tool invocation result\nexport interface ToolInvocationResult {\n  msg: ToolCallPart;\n  tool: AnyTool | null;\n  input: unknown;\n  output: ToolOutput;\n  error: FrameworkError | null;\n}\n\n/**\n * Run a single tool with error handling\n */\nexport async function runTool(\n  tools: AnyTool[],\n  msg: ToolCallPart,\n  context: Record<string, any>,\n): Promise<ToolInvocationResult> {\n  const result: ToolInvocationResult = {\n    msg,\n    tool: null,\n    input: msg.input,\n    output: new StringToolOutput(\"\"),\n    error: null,\n  };\n\n  try {\n    result.tool = tools.find((tool) => tool.name === msg.toolName) || null;\n    if (!result.tool) {\n      throw new ToolError(`Tool '${msg.toolName}' does not exist!`);\n    }\n\n    result.output = await result.tool.run(result.input).context({\n      ...context,\n      toolCallMsg: msg,\n    });\n  } catch (e) {\n    if (e instanceof ToolError) {\n      result.error = FrameworkError.ensure(e);\n    } else {\n      throw e;\n    }\n  }\n\n  return result;\n}\n\n// Final answer tool schema\nconst FinalAnswerToolSchema = z.object({\n  response: z.string().describe(\"The final answer to the user\"),\n});\n\n/**\n * Special tool for capturing final answers\n */\nexport class FinalAnswerTool extends Tool {\n  public readonly name = \"final_answer\";\n  public readonly description = \"Sends the final answer to the user\";\n  public instructions?: string;\n  public customSchema: boolean;\n  public readonly emitter: Emitter<any>;\n\n  protected expectedOutput: string | ZodSchema | null;\n  protected state: RequirementAgentRunState;\n\n  constructor(expectedOutput: string | ZodSchema | null, state: RequirementAgentRunState) {\n    super();\n    this.expectedOutput = expectedOutput;\n    this.state = state;\n    this.instructions = typeof expectedOutput === \"string\" ? expectedOutput : undefined;\n    this.customSchema = expectedOutput instanceof ZodSchema;\n    this.emitter = Emitter.root.child({\n      namespace: [\"tool\", \"final_answer\"],\n      creator: this,\n    });\n  }\n\n  inputSchema(): ZodSchema {\n    const expectedOutput = this.expectedOutput;\n\n    if (!expectedOutput) {\n      return FinalAnswerToolSchema;\n    } else if (expectedOutput instanceof ZodSchema) {\n      return expectedOutput;\n    } else if (typeof expectedOutput === \"string\") {\n      return z.object({\n        response: z.string().describe(expectedOutput),\n      });\n    } else {\n      return FinalAnswerToolSchema;\n    }\n  }\n\n  async _run(\n    input: any,\n    _options: BaseToolRunOptions,\n    _ctx: RunContext<typeof this>,\n  ): Promise<StringToolOutput> {\n    this.state.result = input;\n\n    if (this.expectedOutput instanceof ZodSchema) {\n      // For custom schemas, serialize the entire input\n      this.state.answer = new AssistantMessage(JSON.stringify(input));\n    } else {\n      // For string schemas, use the response field\n      this.state.answer = new AssistantMessage(input.response);\n    }\n\n    return new StringToolOutput(\"Message has been sent\");\n  }\n}\n"]}