beeai-framework 0.1.28 → 0.1.30

package/dist/agents/requirement/agent.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/agents/requirement/agent.ts"],"names":["RequirementAgent","BaseAgent","emitter","Emitter","root","child","namespace","creator","input","runner","RequirementAgentRunner","memory","UnconstrainedMemory","tools","requirements","saveIntermediateSteps","finalAnswerAsTool","toolCallChecker","middlewares","push","register","_run","options","run","execution","maxRetriesPerStep","totalMaxRetries","maxIterations","createToolCallChecker","llm","expectedOutput","templates","newMessages","processInput","addMessages","messages","finalState","reset","addMany","lastPair","slice","answer","Error","result","state","prompt","userMessage","UserMessage","task","render","context","undefined","config","Object","assign","checker","ToolCallChecker","enabled","meta","name","description","length","extraDescription","map","tool","join","overrides","finalized","mapObj","system","RequirementAgentSystemPrompt","RequirementAgentTaskPrompt","toolError","RequirementAgentToolErrorPrompt","toolNoResult","RequirementAgentToolNoResultPrompt","key","defaultTemplate","override","PromptTemplate","role","instructions","notes","update","defaults","pickBy","isEmptyish","castArray","v","isDefined","createSnapshot","shallowCopy"],"mappings":";;;;;;;;;;;;;;;;;AAqFO,MAAMA,yBAAyBC,kBAAAA,CAAAA;EArFtC;;;EA0FkBC,OAAAA,GAAUC,mBAAAA,CAAQC,KAAKC,KAAAA,CAAiC;IACtEC,SAAAA,EAAW;AAAC,MAAA,OAAA;AAAS,MAAA;;IACrBC,OAAAA,EAAS;GACX,CAAA;AAEmBC,EAAAA,KAAAA;EAETC,MAAAA,GAEoBC,iCAAAA;AAE9B,EAAA,WAAA,CAAYF,KAAAA,EAA8B;AACxC,IAAA,KAAA,EAAK;AACL,IAAA,IAAA,CAAKA,KAAAA,GAAQ;MACX,GAAGA,KAAAA;MACHG,MAAAA,EAAQH,KAAAA,CAAMG,MAAAA,IAAU,IAAIC,2CAAAA,EAAAA;MAC5BC,KAAAA,EAAOL,KAAAA,CAAMK,SAAS,EAAA;MACtBC,YAAAA,EAAcN,KAAAA,CAAMM,gBAAgB,EAAA;AACpCC,MAAAA,qBAAAA,EAAuBP,MAAMO,qBAAAA,IAAyB,IAAA;AACtDC,MAAAA,iBAAAA,EAAmBR,MAAMQ,iBAAAA,IAAqB,IAAA;AAC9CC,MAAAA,eAAAA,EAAiBT,MAAMS,eAAAA,IAAmB;AAC5C,KAAA;AACA,IAAA,IAAIT,MAAMU,WAAAA,EAAa;AACrB,MAAA,IAAA,CAAKA,WAAAA,CAAYC,IAAAA,CAAI,GAAIX,KAAAA,CAAMU,WAAW,CAAA;AAC5C;AACF;EAEA;AACE,IAAA,IAAA,CAAKE,QAAAA,EAAQ;AACf;AAEA,EAAA,MAAgBC,IAAAA,CACdb,KAAAA,EACAc,OAAAA,GAAsC,IACtCC,GAAAA,EACiC;AACjC,IAAA,MAAMC,SAAAA,GAA6C;MACjDC,iBAAAA,EAAmB,CAAA;MACnBC,eAAAA,EAAiB,EAAA;MACjBC,aAAAA,EAAe,EAAA;AACf,MAAA,GAAG,KAAKnB,KAAAA,CAAMgB,SAAAA;AACd,MAAA,GAAGF,OAAAA,CAAQE;AACb,KAAA;AAEA,IAAA,MAAMP,eAAAA,GAAkB,KAAKW,qBAAAA,EAAqB;AAElD,IAAA,MAAMnB,MAAAA,GAAS,IAAI,IAAA,CAAKA,MAAAA,CACtB,IAAA,CAAKD,KAAAA,CAAMqB,GAAAA,EACXL,SAAAA,EACA,IAAA,CAAKhB,KAAAA,CAAMK,KAAAA,IAAS,EAAA,EACpB,IAAA,CAAKL,KAAAA,CAAMM,YAAAA,IAAgB,EAAA,EAC3BN,KAAAA,CAAMsB,cAAAA,IAAkB,IAAA,EACxBb,eAAAA,EACAM,GAAAA,EACA,IAAA,CAAKf,KAAAA,CAAMQ,iBAAAA,EACX,IAAA,CAAKe,SAAS,CAAA;AAIhB,IAAA,MAAMC,WAAAA,GAAc,IAAA,CAAKC,YAAAA,CAAazB,KAAAA,CAAAA;AACtC,IAAA,MAAMC,OAAOyB,WAAAA,CAAY;AAAI,MAAA,GAAA,IAAA,CAAKvB,MAAAA,CAAOwB;AAAS,KAAA,CAAA;AAClD,IAAA,MAAM1B,MAAAA,CAAOyB,YAAYF,WAAAA,CAAAA;AAGzB,IAAA,MAAMI,UAAAA,GAAa,MAAM3B,MAAAA,CAAOc,GAAAA,EAAG;AAGnC,IAAA,IAAI,IAAA,CAAKf,MAAMO,qBAAAA,EAAuB;AACpC,MAAA,IAAA,CAAKJ,OAAO0B,KAAAA,EAAK;AACjB,MAAA,MAAM,IAAA,CAAK1B,MAAAA,CAAO2B,OAAAA,CAAQF,UAAAA,CAAWzB,OAAOwB,QAAQ,CAAA;KACtD,MAAO;AACL,MAAA,MAAM,IAAA,CAAKxB,MAAAA,CAAO2B,OAAAA,CAAQN,WAAAA,CAAAA;AAE1B,MAAA,MAAMG,QAAAA,GAAWC,WAAWzB,MAAAA,CAAOwB,QAAAA;AACnC,MAAA,MAAMI,QAAAA,GAAWJ,QAAAA,CAASK,KAAAA,CAAM,EAAC,CAAA;AACjC,MAAA,MAAM,IAAA,CAAK7B,MAAAA,CAAO2B,OAAAA,CAAQC,QAAAA,CAAAA;AAC5B;AAEA,IAAA,IAAI,CAACH,WAAWK,MAAAA,EAAQ;AACtB,MAAA,MAAM,IAAIC,MAAM,sCAAA,CAAA;AAClB;AAEA,IAAA,OAAO;AACLC,MAAAA,MAAAA,EAAQP,UAAAA,CAAWK,MAAAA;AACnB9B,MAAAA,MAAAA,EAAQyB,UAAAA,CAAWzB,MAAAA;MACnBiC,KAAAA,EAAOR;AACT,KAAA;AACF;AAEUH,EAAAA,YAAAA,CAAazB,KAAAA,EAA4C;AACjE,IAAA,IAAI,CAACA,MAAMqC,MAAAA,EAAQ;AACjB,MAAA,OAAO,EAAA;AACT;AAEA,IAAA,MAAMC,cAAc,IAAIC,uBAAAA,CACtB,IAAA,CAAKhB,SAAAA,CAAUiB,KAAKC,MAAAA,CAAO;AACzBJ,MAAAA,MAAAA,EAAQrC,KAAAA,CAAMqC,MAAAA;AACdK,MAAAA,OAAAA,EAAS1C,KAAAA,CAAM0C,OAAAA;AACfpB,MAAAA,cAAAA,EAAgB,OAAOtB,KAAAA,CAAMsB,cAAAA,KAAmB,QAAA,GAAWtB,MAAMsB,cAAAA,GAAiBqB;AACpF,KAAA,CAAA,CAAA;AAGF,IAAA,OAAO;AAACL,MAAAA;;AACV;EAEUlB,qBAAAA,GAAyC;AACjD,IAAA,MAAMwB,SAAgC,EAAC;AAEvC,IAAA,IAAI,OAAO,IAAA,CAAK5C,KAAAA,CAAMS,eAAAA,KAAoB,QAAA,EAAU;AAClDoC,MAAAA,MAAAA,CAAOC,MAAAA,CAAOF,MAAAA,EAAQ,IAAA,CAAK5C,KAAAA,CAAMS,eAAe,CAAA;AAClD;AAEA,IAAA,MAAMsC,OAAAA,GAAU,IAAIC,mCAAAA,CAAgBJ,MAAAA,CAAAA;AACpCG,IAAAA,OAAAA,CAAQE,OAAAA,GAAU,IAAA,CAAKjD,KAAAA,CAAMS,eAAAA,KAAoB,KAAA;AACjD,IAAA,OAAOsC,OAAAA;AACT;AAEA,EAAA,IAAIG,IAAAA,GAAkB;AACpB,IAAA,MAAM7C,KAAAA,GAAQ,IAAA,CAAKL,KAAAA,CAAMK,KAAAA,IAAS,EAAA;AAElC,IAAA,OAAO;MACL8C,IAAAA,EAAM,IAAA,CAAKnD,MAAMmD,IAAAA,IAAQ,aAAA;AACzB9C,MAAAA,KAAAA;MACA+C,WAAAA,EACE,IAAA,CAAKpD,MAAMoD,WAAAA,IACX,yLAAA;MAEF,GAAI/C,KAAAA,CAAMgD,SAAS,CAAA,IAAK;QACtBC,gBAAAA,EAAkB;AAChB,UAAA,CAAA,8CAAA,CAAA;aACGjD,KAAAA,CAAMkD,GAAAA,CAAI,CAACC,IAAAA,KAAS,CAAA,MAAA,EAASA,KAAKL,IAAI,CAAA,GAAA,EAAMK,IAAAA,CAAKJ,WAAW,CAAA,CAAA,CAAG;AAClEK,SAAAA,CAAAA,IAAAA,CAAK,IAAA;AACT;AACF,KAAA;AACF;AAEA,EAAA,IAAclC,SAAAA,GAAuC;AACnD,IAAA,MAAMmC,SAAAA,GAAY,IAAA,CAAK1D,KAAAA,CAAMuB,SAAAA,IAAa,EAAC;AAC3C,IAAA,MAAMoC,YAAYC,iBAAAA,CAAO;MACvBC,MAAAA,EAAQC,wCAAAA;MACRtB,IAAAA,EAAMuB,sCAAAA;MACNC,SAAAA,EAAWC,2CAAAA;MACXC,YAAAA,EAAcC;KAChB,CAAA,CAAgC,CAC9BC,GAAAA,EACAC,eAAAA,KAAAA;AAEA,MAAA,MAAMC,QAAAA,GAAWZ,SAAAA,CAAUU,GAAAA,CAAAA,IAAQC,eAAAA;AACnC,MAAA,IAAIC,oBAAoBC,2BAAAA,EAAgB;AACtC,QAAA,OAAOD,QAAAA;AACT;AACA,MAAA,OAAOA,QAAAA,CAASD,eAAAA,CAAAA,IAAoBA,eAAAA;KACtC,CAAA;AAEA,IAAA,IAAI,IAAA,CAAKrE,MAAMwE,IAAAA,IAAQ,IAAA,CAAKxE,MAAMyE,YAAAA,IAAgB,IAAA,CAAKzE,MAAM0E,KAAAA,EAAO;AAClEf,MAAAA,SAAAA,CAAUE,OAAOc,MAAAA,CAAO;AACtBC,QAAAA,QAAAA,EAAUC,aAAAA,CACR;AACEL,UAAAA,IAAAA,EAAM,KAAKxE,KAAAA,CAAMwE,IAAAA;AACjBC,UAAAA,YAAAA,EAAcK,kBAAW,IAAA,CAAK9E,KAAAA,CAAMyE,YAAY,CAAA,GAC5C9B,MAAAA,GACAoC,oBAAU,IAAA,CAAK/E,KAAAA,CAAMyE,YAAY,CAAA,CAC9BlB,GAAAA,CAAI,CAACyB,CAAAA,KAAM,CAAA,EAAA,EAAKA,CAAAA,CAAAA,CAAG,CAAA,CACnBvB,KAAK,IAAA,CAAA;AACZiB,UAAAA,KAAAA,EAAOI,kBAAW,IAAA,CAAK9E,KAAAA,CAAM0E,KAAK,CAAA,GAC9B/B,MAAAA,GACAoC,oBAAU,IAAA,CAAK/E,KAAAA,CAAM0E,KAAK,CAAA,CACvBnB,GAAAA,CAAI,CAACyB,CAAAA,KAAM,CAAA,EAAA,EAAKA,CAAAA,CAAAA,CAAG,CAAA,CACnBvB,KAAK,IAAA;AACd,SAAA,EACAwB,gBAAAA;OAEJ,CAAA;AACF;AAEA,IAAA,OAAOtB,SAAAA;AACT;EAEAuB,cAAAA,GAAiB;AACf,IAAA,OAAO;AACL,MAAA,GAAG,MAAMA,cAAAA,EAAAA;MACTlF,KAAAA,EAAOmF,qBAAAA,CAAY,KAAKnF,KAAK,CAAA;AAC7BN,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;AACdO,MAAAA,MAAAA,EAAQ,IAAA,CAAKA;AACf,KAAA;AACF;AAEA,EAAA,IAAIE,OAAOA,MAAAA,EAAoB;AAC7B,IAAA,IAAA,CAAKH,MAAMG,MAAAA,GAASA,MAAAA;AACtB;AAEA,EAAA,IAAIA,MAAAA,GAAqB;AACvB,IAAA,OAAO,KAAKH,KAAAA,CAAMG,MAAAA;AACpB;AACF","file":"agent.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { BaseAgent } from \"@/agents/base.js\";\nimport { AnyTool } from \"@/tools/base.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\nimport { AgentMeta } from \"@/agents/types.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { ChatModel } from \"@/backend/chat.js\";\nimport type { GetRunContext, MiddlewareType } from \"@/context.js\";\nimport { UnconstrainedMemory } from \"@/memory/unconstrainedMemory.js\";\nimport { Message, UserMessage } from \"@/backend/message.js\";\nimport { shallowCopy } from \"@/serializer/utils.js\";\nimport { mapObj } from \"@/internals/helpers/object.js\";\nimport { PromptTemplate } from \"@/template.js\";\nimport {\n  RequirementAgentCallbacks,\n  RequirementAgentExecutionConfig,\n  RequirementAgentOutput,\n  RequirementAgentRunInput,\n  RequirementAgentRunOptions,\n  RequirementAgentTemplates,\n} from \"./types.js\";\nimport {\n  RequirementAgentSystemPrompt,\n  RequirementAgentTaskPrompt,\n  RequirementAgentToolErrorPrompt,\n  RequirementAgentToolNoResultPrompt,\n} from \"./prompts.js\";\nimport { Requirement } from \"./requirements/requirement.js\";\nimport { RequirementAgentRunner } from \"./runner.js\";\nimport { ToolCallChecker, ToolCallCheckerConfig } from \"./utils/toolCallChecker.js\";\nimport { isDefined, isEmptyish, pickBy } from \"remeda\";\nimport { castArray } from \"@/internals/helpers/array.js\";\nimport type { RequiredExcept } from \"@/internals/types.js\";\n\nexport type RequirementAgentTemplateFactory<K extends keyof RequirementAgentTemplates> = (\n  template: RequirementAgentTemplates[K],\n) => RequirementAgentTemplates[K];\n\nexport interface RequirementAgentInput {\n  llm: ChatModel;\n  memory?: BaseMemory;\n  tools?: AnyTool[];\n  requirements?: Requirement[];\n  name?: string;\n  description?: string;\n  role?: string;\n  instructions?: string | string[];\n  notes?: string | string[];\n  toolCallChecker?: boolean | ToolCallCheckerConfig;\n  finalAnswerAsTool?: boolean;\n  saveIntermediateSteps?: boolean;\n  execution?: RequirementAgentExecutionConfig;\n  templates?: Partial<{\n    [K in keyof RequirementAgentTemplates]:\n      | RequirementAgentTemplates[K]\n      | RequirementAgentTemplateFactory<K>;\n  }>;\n  middlewares?: MiddlewareType<RequirementAgent>[];\n}\n\ntype InternalRequirementAgentInput = RequiredExcept<\n  RequirementAgentInput,\n  | \"role\"\n  | \"instructions\"\n  | \"name\"\n  | \"description\"\n  | \"templates\"\n  | \"notes\"\n  | \"execution\"\n  | \"middlewares\"\n>;\n\n/**\n * RequirementAgent - A declarative AI agent with rule-based constraints\n *\n * The RequirementAgent provides predictable, controlled execution behavior across\n * different language models through rule-based constraints (Requirements).\n * Requirements can be configured as strict or flexible as necessary, adapting to\n * task requirements while ensuring consistent execution regardless of the underlying\n * model's reasoning or tool-calling capabilities.\n */\nexport class RequirementAgent extends BaseAgent<\n  RequirementAgentRunInput,\n  RequirementAgentOutput,\n  RequirementAgentRunOptions\n> {\n  public readonly emitter = Emitter.root.child<RequirementAgentCallbacks>({\n    namespace: [\"agent\", \"requirement\"],\n    creator: this,\n  });\n\n  protected readonly input: InternalRequirementAgentInput;\n\n  protected runner: new (\n    ...args: ConstructorParameters<typeof RequirementAgentRunner>\n  ) => RequirementAgentRunner = RequirementAgentRunner;\n\n  constructor(input: RequirementAgentInput) {\n    super();\n    this.input = {\n      ...input,\n      memory: input.memory || new UnconstrainedMemory(),\n      tools: input.tools || [],\n      requirements: input.requirements || [],\n      saveIntermediateSteps: input.saveIntermediateSteps ?? true,\n      finalAnswerAsTool: input.finalAnswerAsTool ?? true,\n      toolCallChecker: input.toolCallChecker ?? true,\n    };\n    if (input.middlewares) {\n      this.middlewares.push(...input.middlewares);\n    }\n  }\n\n  static {\n    this.register();\n  }\n\n  protected async _run(\n    input: RequirementAgentRunInput,\n    options: RequirementAgentRunOptions = {},\n    run: GetRunContext<typeof this>,\n  ): Promise<RequirementAgentOutput> {\n    const execution: RequirementAgentExecutionConfig = {\n      maxRetriesPerStep: 3,\n      totalMaxRetries: 20,\n      maxIterations: 20,\n      ...this.input.execution,\n      ...options.execution,\n    };\n\n    const toolCallChecker = this.createToolCallChecker();\n\n    const runner = new this.runner(\n      this.input.llm,\n      execution,\n      this.input.tools || [],\n      this.input.requirements || [],\n      input.expectedOutput || null,\n      toolCallChecker,\n      run,\n      this.input.finalAnswerAsTool,\n      this.templates,\n    );\n\n    // Process input messages\n    const newMessages = this.processInput(input);\n    await runner.addMessages([...this.memory.messages]);\n    await runner.addMessages(newMessages);\n\n    // Run the agent\n    const finalState = await runner.run();\n\n    // Update memory\n    if (this.input.saveIntermediateSteps) {\n      this.memory.reset();\n      await this.memory.addMany(finalState.memory.messages);\n    } else {\n      await this.memory.addMany(newMessages);\n      // Add last tool call pair\n      const messages = finalState.memory.messages;\n      const lastPair = messages.slice(-2);\n      await this.memory.addMany(lastPair);\n    }\n\n    if (!finalState.answer) {\n      throw new Error(\"Agent did not produce a final answer\");\n    }\n\n    return {\n      result: finalState.answer,\n      memory: finalState.memory,\n      state: finalState,\n    };\n  }\n\n  protected processInput(input: RequirementAgentRunInput): Message[] {\n    if (!input.prompt) {\n      return [];\n    }\n\n    const userMessage = new UserMessage(\n      this.templates.task.render({\n        prompt: input.prompt,\n        context: input.context,\n        expectedOutput: typeof input.expectedOutput === \"string\" ? input.expectedOutput : undefined,\n      }),\n    );\n\n    return [userMessage];\n  }\n\n  protected createToolCallChecker(): ToolCallChecker {\n    const config: ToolCallCheckerConfig = {};\n\n    if (typeof this.input.toolCallChecker === \"object\") {\n      Object.assign(config, this.input.toolCallChecker);\n    }\n\n    const checker = new ToolCallChecker(config);\n    checker.enabled = this.input.toolCallChecker !== false;\n    return checker;\n  }\n\n  get meta(): AgentMeta {\n    const tools = this.input.tools || [];\n\n    return {\n      name: this.input.name ?? \"Requirement\",\n      tools,\n      description:\n        this.input.description ??\n        \"The RequirementAgent is a declarative AI agent implementation that provides predictable, \" +\n          \"controlled execution behavior across different language models through rule-based constraints.\",\n      ...(tools.length > 0 && {\n        extraDescription: [\n          `Tools that I can use to accomplish given task.`,\n          ...tools.map((tool) => `Tool '${tool.name}': ${tool.description}.`),\n        ].join(\"\\n\"),\n      }),\n    };\n  }\n\n  protected get templates(): RequirementAgentTemplates {\n    const overrides = this.input.templates || {};\n    const finalized = mapObj({\n      system: RequirementAgentSystemPrompt,\n      task: RequirementAgentTaskPrompt,\n      toolError: RequirementAgentToolErrorPrompt,\n      toolNoResult: RequirementAgentToolNoResultPrompt,\n    } as RequirementAgentTemplates)((\n      key,\n      defaultTemplate: RequirementAgentTemplates[typeof key],\n    ) => {\n      const override = overrides[key] ?? defaultTemplate;\n      if (override instanceof PromptTemplate) {\n        return override;\n      }\n      return override(defaultTemplate) ?? defaultTemplate;\n    });\n\n    if (this.input.role || this.input.instructions || this.input.notes) {\n      finalized.system.update({\n        defaults: pickBy(\n          {\n            role: this.input.role,\n            instructions: isEmptyish(this.input.instructions)\n              ? undefined\n              : castArray(this.input.instructions)\n                  .map((v) => `- ${v}`)\n                  .join(\"\\n\"),\n            notes: isEmptyish(this.input.notes)\n              ? undefined\n              : castArray(this.input.notes)\n                  .map((v) => `- ${v}`)\n                  .join(\"\\n\"),\n          },\n          isDefined,\n        ),\n      });\n    }\n\n    return finalized;\n  }\n\n  createSnapshot() {\n    return {\n      ...super.createSnapshot(),\n      input: shallowCopy(this.input),\n      emitter: this.emitter,\n      runner: this.runner,\n    };\n  }\n\n  set memory(memory: BaseMemory) {\n    this.input.memory = memory;\n  }\n\n  get memory(): BaseMemory {\n    return this.input.memory!;\n  }\n}\n"]}
+{"version":3,"sources":["../../../src/agents/requirement/agent.ts"],"names":["RequirementAgent","BaseAgent","emitter","Emitter","root","child","namespace","creator","input","runner","RequirementAgentRunner","memory","UnconstrainedMemory","tools","requirements","saveIntermediateSteps","finalAnswerAsTool","toolCallChecker","middlewares","push","register","_run","options","run","execution","maxRetriesPerStep","totalMaxRetries","maxIterations","createToolCallChecker","llm","expectedOutput","templates","newMessages","processInput","addMessages","messages","finalState","reset","addMany","lastPair","slice","answer","Error","result","state","prompt","userMessage","UserMessage","task","render","context","undefined","config","Object","assign","checker","ToolCallChecker","enabled","meta","name","description","length","extraDescription","map","tool","join","overrides","finalized","mapObj","system","RequirementAgentSystemPrompt","RequirementAgentTaskPrompt","toolError","RequirementAgentToolErrorPrompt","toolNoResult","RequirementAgentToolNoResultPrompt","key","defaultTemplate","override","PromptTemplate","role","instructions","notes","update","defaults","pickBy","isEmptyish","castArray","v","isDefined","createSnapshot","shallowCopy"],"mappings":";;;;;;;;;;;;;;;;;AAqFO,MAAMA,yBAAyBC,kBAAAA,CAAAA;EArFtC;;;EA0FkBC,OAAAA,GAAUC,mBAAAA,CAAQC,KAAKC,KAAAA,CAAiC;IACtEC,SAAAA,EAAW;AAAC,MAAA,OAAA;AAAS,MAAA;;IACrBC,OAAAA,EAAS;GACX,CAAA;AAEmBC,EAAAA,KAAAA;EAETC,MAAAA,GAEoBC,iCAAAA;AAE9B,EAAA,WAAA,CAAYF,KAAAA,EAA8B;AACxC,IAAA,KAAA,EAAK;AACL,IAAA,IAAA,CAAKA,KAAAA,GAAQ;MACX,GAAGA,KAAAA;MACHG,MAAAA,EAAQH,KAAAA,CAAMG,MAAAA,IAAU,IAAIC,2CAAAA,EAAAA;MAC5BC,KAAAA,EAAOL,KAAAA,CAAMK,SAAS,EAAA;MACtBC,YAAAA,EAAcN,KAAAA,CAAMM,gBAAgB,EAAA;AACpCC,MAAAA,qBAAAA,EAAuBP,MAAMO,qBAAAA,IAAyB,IAAA;AACtDC,MAAAA,iBAAAA,EAAmBR,MAAMQ,iBAAAA,IAAqB,IAAA;AAC9CC,MAAAA,eAAAA,EAAiBT,MAAMS,eAAAA,IAAmB;AAC5C,KAAA;AACA,IAAA,IAAIT,MAAMU,WAAAA,EAAa;AACrB,MAAA,IAAA,CAAKA,WAAAA,CAAYC,IAAAA,CAAI,GAAIX,KAAAA,CAAMU,WAAW,CAAA;AAC5C,IAAA;AACF,EAAA;EAEA;AACE,IAAA,IAAA,CAAKE,QAAAA,EAAQ;AACf;AAEA,EAAA,MAAgBC,IAAAA,CACdb,KAAAA,EACAc,OAAAA,GAAsC,IACtCC,GAAAA,EACiC;AACjC,IAAA,MAAMC,SAAAA,GAA6C;MACjDC,iBAAAA,EAAmB,CAAA;MACnBC,eAAAA,EAAiB,EAAA;MACjBC,aAAAA,EAAe,EAAA;AACf,MAAA,GAAG,KAAKnB,KAAAA,CAAMgB,SAAAA;AACd,MAAA,GAAGF,OAAAA,CAAQE;AACb,KAAA;AAEA,IAAA,MAAMP,eAAAA,GAAkB,KAAKW,qBAAAA,EAAqB;AAElD,IAAA,MAAMnB,MAAAA,GAAS,IAAI,IAAA,CAAKA,MAAAA,CACtB,IAAA,CAAKD,KAAAA,CAAMqB,GAAAA,EACXL,SAAAA,EACA,IAAA,CAAKhB,KAAAA,CAAMK,KAAAA,IAAS,EAAA,EACpB,IAAA,CAAKL,KAAAA,CAAMM,YAAAA,IAAgB,EAAA,EAC3BN,KAAAA,CAAMsB,cAAAA,IAAkB,IAAA,EACxBb,eAAAA,EACAM,GAAAA,EACA,IAAA,CAAKf,KAAAA,CAAMQ,iBAAAA,EACX,IAAA,CAAKe,SAAS,CAAA;AAIhB,IAAA,MAAMC,WAAAA,GAAc,IAAA,CAAKC,YAAAA,CAAazB,KAAAA,CAAAA;AACtC,IAAA,MAAMC,OAAOyB,WAAAA,CAAY;AAAI,MAAA,GAAA,IAAA,CAAKvB,MAAAA,CAAOwB;AAAS,KAAA,CAAA;AAClD,IAAA,MAAM1B,MAAAA,CAAOyB,YAAYF,WAAAA,CAAAA;AAGzB,IAAA,MAAMI,UAAAA,GAAa,MAAM3B,MAAAA,CAAOc,GAAAA,EAAG;AAGnC,IAAA,IAAI,IAAA,CAAKf,MAAMO,qBAAAA,EAAuB;AACpC,MAAA,IAAA,CAAKJ,OAAO0B,KAAAA,EAAK;AACjB,MAAA,MAAM,IAAA,CAAK1B,MAAAA,CAAO2B,OAAAA,CAAQF,UAAAA,CAAWzB,OAAOwB,QAAQ,CAAA;IACtD,CAAA,MAAO;AACL,MAAA,MAAM,IAAA,CAAKxB,MAAAA,CAAO2B,OAAAA,CAAQN,WAAAA,CAAAA;AAE1B,MAAA,MAAMG,QAAAA,GAAWC,WAAWzB,MAAAA,CAAOwB,QAAAA;AACnC,MAAA,MAAMI,QAAAA,GAAWJ,QAAAA,CAASK,KAAAA,CAAM,EAAC,CAAA;AACjC,MAAA,MAAM,IAAA,CAAK7B,MAAAA,CAAO2B,OAAAA,CAAQC,QAAAA,CAAAA;AAC5B,IAAA;AAEA,IAAA,IAAI,CAACH,WAAWK,MAAAA,EAAQ;AACtB,MAAA,MAAM,IAAIC,MAAM,sCAAA,CAAA;AAClB,IAAA;AAEA,IAAA,OAAO;AACLC,MAAAA,MAAAA,EAAQP,UAAAA,CAAWK,MAAAA;AACnB9B,MAAAA,MAAAA,EAAQyB,UAAAA,CAAWzB,MAAAA;MACnBiC,KAAAA,EAAOR;AACT,KAAA;AACF,EAAA;AAEUH,EAAAA,YAAAA,CAAazB,KAAAA,EAA4C;AACjE,IAAA,IAAI,CAACA,MAAMqC,MAAAA,EAAQ;AACjB,MAAA,OAAO,EAAA;AACT,IAAA;AAEA,IAAA,MAAMC,cAAc,IAAIC,uBAAAA,CACtB,IAAA,CAAKhB,SAAAA,CAAUiB,KAAKC,MAAAA,CAAO;AACzBJ,MAAAA,MAAAA,EAAQrC,KAAAA,CAAMqC,MAAAA;AACdK,MAAAA,OAAAA,EAAS1C,KAAAA,CAAM0C,OAAAA;AACfpB,MAAAA,cAAAA,EAAgB,OAAOtB,KAAAA,CAAMsB,cAAAA,KAAmB,QAAA,GAAWtB,MAAMsB,cAAAA,GAAiBqB;AACpF,KAAA,CAAA,CAAA;AAGF,IAAA,OAAO;AAACL,MAAAA;;AACV,EAAA;EAEUlB,qBAAAA,GAAyC;AACjD,IAAA,MAAMwB,SAAgC,EAAC;AAEvC,IAAA,IAAI,OAAO,IAAA,CAAK5C,KAAAA,CAAMS,eAAAA,KAAoB,QAAA,EAAU;AAClDoC,MAAAA,MAAAA,CAAOC,MAAAA,CAAOF,MAAAA,EAAQ,IAAA,CAAK5C,KAAAA,CAAMS,eAAe,CAAA;AAClD,IAAA;AAEA,IAAA,MAAMsC,OAAAA,GAAU,IAAIC,mCAAAA,CAAgBJ,MAAAA,CAAAA;AACpCG,IAAAA,OAAAA,CAAQE,OAAAA,GAAU,IAAA,CAAKjD,KAAAA,CAAMS,eAAAA,KAAoB,KAAA;AACjD,IAAA,OAAOsC,OAAAA;AACT,EAAA;AAEA,EAAA,IAAIG,IAAAA,GAAkB;AACpB,IAAA,MAAM7C,KAAAA,GAAQ,IAAA,CAAKL,KAAAA,CAAMK,KAAAA,IAAS,EAAA;AAElC,IAAA,OAAO;MACL8C,IAAAA,EAAM,IAAA,CAAKnD,MAAMmD,IAAAA,IAAQ,aAAA;AACzB9C,MAAAA,KAAAA;MACA+C,WAAAA,EACE,IAAA,CAAKpD,MAAMoD,WAAAA,IACX,yLAAA;MAEF,GAAI/C,KAAAA,CAAMgD,SAAS,CAAA,IAAK;QACtBC,gBAAAA,EAAkB;AAChB,UAAA,CAAA,8CAAA,CAAA;aACGjD,KAAAA,CAAMkD,GAAAA,CAAI,CAACC,IAAAA,KAAS,CAAA,MAAA,EAASA,KAAKL,IAAI,CAAA,GAAA,EAAMK,IAAAA,CAAKJ,WAAW,CAAA,CAAA,CAAG;AAClEK,SAAAA,CAAAA,IAAAA,CAAK,IAAA;AACT;AACF,KAAA;AACF,EAAA;AAEA,EAAA,IAAclC,SAAAA,GAAuC;AACnD,IAAA,MAAMmC,SAAAA,GAAY,IAAA,CAAK1D,KAAAA,CAAMuB,SAAAA,IAAa,EAAC;AAC3C,IAAA,MAAMoC,YAAYC,iBAAAA,CAAO;MACvBC,MAAAA,EAAQC,wCAAAA;MACRtB,IAAAA,EAAMuB,sCAAAA;MACNC,SAAAA,EAAWC,2CAAAA;MACXC,YAAAA,EAAcC;KAChB,CAAA,CAAgC,CAC9BC,GAAAA,EACAC,eAAAA,KAAAA;AAEA,MAAA,MAAMC,QAAAA,GAAWZ,SAAAA,CAAUU,GAAAA,CAAAA,IAAQC,eAAAA;AACnC,MAAA,IAAIC,oBAAoBC,2BAAAA,EAAgB;AACtC,QAAA,OAAOD,QAAAA;AACT,MAAA;AACA,MAAA,OAAOA,QAAAA,CAASD,eAAAA,CAAAA,IAAoBA,eAAAA;IACtC,CAAA,CAAA;AAEA,IAAA,IAAI,IAAA,CAAKrE,MAAMwE,IAAAA,IAAQ,IAAA,CAAKxE,MAAMyE,YAAAA,IAAgB,IAAA,CAAKzE,MAAM0E,KAAAA,EAAO;AAClEf,MAAAA,SAAAA,CAAUE,OAAOc,MAAAA,CAAO;AACtBC,QAAAA,QAAAA,EAAUC,aAAAA,CACR;AACEL,UAAAA,IAAAA,EAAM,KAAKxE,KAAAA,CAAMwE,IAAAA;AACjBC,UAAAA,YAAAA,EAAcK,kBAAW,IAAA,CAAK9E,KAAAA,CAAMyE,YAAY,CAAA,GAC5C9B,MAAAA,GACAoC,oBAAU,IAAA,CAAK/E,KAAAA,CAAMyE,YAAY,CAAA,CAC9BlB,GAAAA,CAAI,CAACyB,CAAAA,KAAM,CAAA,EAAA,EAAKA,CAAAA,CAAAA,CAAG,CAAA,CACnBvB,KAAK,IAAA,CAAA;AACZiB,UAAAA,KAAAA,EAAOI,kBAAW,IAAA,CAAK9E,KAAAA,CAAM0E,KAAK,CAAA,GAC9B/B,MAAAA,GACAoC,oBAAU,IAAA,CAAK/E,KAAAA,CAAM0E,KAAK,CAAA,CACvBnB,GAAAA,CAAI,CAACyB,CAAAA,KAAM,CAAA,EAAA,EAAKA,CAAAA,CAAAA,CAAG,CAAA,CACnBvB,KAAK,IAAA;AACd,SAAA,EACAwB,gBAAAA;OAEJ,CAAA;AACF,IAAA;AAEA,IAAA,OAAOtB,SAAAA;AACT,EAAA;EAEAuB,cAAAA,GAAiB;AACf,IAAA,OAAO;AACL,MAAA,GAAG,MAAMA,cAAAA,EAAAA;MACTlF,KAAAA,EAAOmF,qBAAAA,CAAY,KAAKnF,KAAK,CAAA;AAC7BN,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;AACdO,MAAAA,MAAAA,EAAQ,IAAA,CAAKA;AACf,KAAA;AACF,EAAA;AAEA,EAAA,IAAIE,OAAOA,MAAAA,EAAoB;AAC7B,IAAA,IAAA,CAAKH,MAAMG,MAAAA,GAASA,MAAAA;AACtB,EAAA;AAEA,EAAA,IAAIA,MAAAA,GAAqB;AACvB,IAAA,OAAO,KAAKH,KAAAA,CAAMG,MAAAA;AACpB,EAAA;AACF","file":"agent.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { BaseAgent } from \"@/agents/base.js\";\nimport { AnyTool } from \"@/tools/base.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\nimport { AgentMeta } from \"@/agents/types.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { ChatModel } from \"@/backend/chat.js\";\nimport type { GetRunContext, MiddlewareType } from \"@/context.js\";\nimport { UnconstrainedMemory } from \"@/memory/unconstrainedMemory.js\";\nimport { Message, UserMessage } from \"@/backend/message.js\";\nimport { shallowCopy } from \"@/serializer/utils.js\";\nimport { mapObj } from \"@/internals/helpers/object.js\";\nimport { PromptTemplate } from \"@/template.js\";\nimport {\n  RequirementAgentCallbacks,\n  RequirementAgentExecutionConfig,\n  RequirementAgentOutput,\n  RequirementAgentRunInput,\n  RequirementAgentRunOptions,\n  RequirementAgentTemplates,\n} from \"./types.js\";\nimport {\n  RequirementAgentSystemPrompt,\n  RequirementAgentTaskPrompt,\n  RequirementAgentToolErrorPrompt,\n  RequirementAgentToolNoResultPrompt,\n} from \"./prompts.js\";\nimport { Requirement } from \"./requirements/requirement.js\";\nimport { RequirementAgentRunner } from \"./runner.js\";\nimport { ToolCallChecker, ToolCallCheckerConfig } from \"./utils/toolCallChecker.js\";\nimport { isDefined, isEmptyish, pickBy } from \"remeda\";\nimport { castArray } from \"@/internals/helpers/array.js\";\nimport type { RequiredExcept } from \"@/internals/types.js\";\n\nexport type RequirementAgentTemplateFactory<K extends keyof RequirementAgentTemplates> = (\n  template: RequirementAgentTemplates[K],\n) => RequirementAgentTemplates[K];\n\nexport interface RequirementAgentInput {\n  llm: ChatModel;\n  memory?: BaseMemory;\n  tools?: AnyTool[];\n  requirements?: Requirement[];\n  name?: string;\n  description?: string;\n  role?: string;\n  instructions?: string | string[];\n  notes?: string | string[];\n  toolCallChecker?: boolean | ToolCallCheckerConfig;\n  finalAnswerAsTool?: boolean;\n  saveIntermediateSteps?: boolean;\n  execution?: RequirementAgentExecutionConfig;\n  templates?: Partial<{\n    [K in keyof RequirementAgentTemplates]:\n      | RequirementAgentTemplates[K]\n      | RequirementAgentTemplateFactory<K>;\n  }>;\n  middlewares?: MiddlewareType<RequirementAgent>[];\n}\n\ntype InternalRequirementAgentInput = RequiredExcept<\n  RequirementAgentInput,\n  | \"role\"\n  | \"instructions\"\n  | \"name\"\n  | \"description\"\n  | \"templates\"\n  | \"notes\"\n  | \"execution\"\n  | \"middlewares\"\n>;\n\n/**\n * RequirementAgent - A declarative AI agent with rule-based constraints\n *\n * The RequirementAgent provides predictable, controlled execution behavior across\n * different language models through rule-based constraints (Requirements).\n * Requirements can be configured as strict or flexible as necessary, adapting to\n * task requirements while ensuring consistent execution regardless of the underlying\n * model's reasoning or tool-calling capabilities.\n */\nexport class RequirementAgent extends BaseAgent<\n  RequirementAgentRunInput,\n  RequirementAgentOutput,\n  RequirementAgentRunOptions\n> {\n  public readonly emitter = Emitter.root.child<RequirementAgentCallbacks>({\n    namespace: [\"agent\", \"requirement\"],\n    creator: this,\n  });\n\n  protected readonly input: InternalRequirementAgentInput;\n\n  protected runner: new (\n    ...args: ConstructorParameters<typeof RequirementAgentRunner>\n  ) => RequirementAgentRunner = RequirementAgentRunner;\n\n  constructor(input: RequirementAgentInput) {\n    super();\n    this.input = {\n      ...input,\n      memory: input.memory || new UnconstrainedMemory(),\n      tools: input.tools || [],\n      requirements: input.requirements || [],\n      saveIntermediateSteps: input.saveIntermediateSteps ?? true,\n      finalAnswerAsTool: input.finalAnswerAsTool ?? true,\n      toolCallChecker: input.toolCallChecker ?? true,\n    };\n    if (input.middlewares) {\n      this.middlewares.push(...input.middlewares);\n    }\n  }\n\n  static {\n    this.register();\n  }\n\n  protected async _run(\n    input: RequirementAgentRunInput,\n    options: RequirementAgentRunOptions = {},\n    run: GetRunContext<typeof this>,\n  ): Promise<RequirementAgentOutput> {\n    const execution: RequirementAgentExecutionConfig = {\n      maxRetriesPerStep: 3,\n      totalMaxRetries: 20,\n      maxIterations: 20,\n      ...this.input.execution,\n      ...options.execution,\n    };\n\n    const toolCallChecker = this.createToolCallChecker();\n\n    const runner = new this.runner(\n      this.input.llm,\n      execution,\n      this.input.tools || [],\n      this.input.requirements || [],\n      input.expectedOutput || null,\n      toolCallChecker,\n      run,\n      this.input.finalAnswerAsTool,\n      this.templates,\n    );\n\n    // Process input messages\n    const newMessages = this.processInput(input);\n    await runner.addMessages([...this.memory.messages]);\n    await runner.addMessages(newMessages);\n\n    // Run the agent\n    const finalState = await runner.run();\n\n    // Update memory\n    if (this.input.saveIntermediateSteps) {\n      this.memory.reset();\n      await this.memory.addMany(finalState.memory.messages);\n    } else {\n      await this.memory.addMany(newMessages);\n      // Add last tool call pair\n      const messages = finalState.memory.messages;\n      const lastPair = messages.slice(-2);\n      await this.memory.addMany(lastPair);\n    }\n\n    if (!finalState.answer) {\n      throw new Error(\"Agent did not produce a final answer\");\n    }\n\n    return {\n      result: finalState.answer,\n      memory: finalState.memory,\n      state: finalState,\n    };\n  }\n\n  protected processInput(input: RequirementAgentRunInput): Message[] {\n    if (!input.prompt) {\n      return [];\n    }\n\n    const userMessage = new UserMessage(\n      this.templates.task.render({\n        prompt: input.prompt,\n        context: input.context,\n        expectedOutput: typeof input.expectedOutput === \"string\" ? input.expectedOutput : undefined,\n      }),\n    );\n\n    return [userMessage];\n  }\n\n  protected createToolCallChecker(): ToolCallChecker {\n    const config: ToolCallCheckerConfig = {};\n\n    if (typeof this.input.toolCallChecker === \"object\") {\n      Object.assign(config, this.input.toolCallChecker);\n    }\n\n    const checker = new ToolCallChecker(config);\n    checker.enabled = this.input.toolCallChecker !== false;\n    return checker;\n  }\n\n  get meta(): AgentMeta {\n    const tools = this.input.tools || [];\n\n    return {\n      name: this.input.name ?? \"Requirement\",\n      tools,\n      description:\n        this.input.description ??\n        \"The RequirementAgent is a declarative AI agent implementation that provides predictable, \" +\n          \"controlled execution behavior across different language models through rule-based constraints.\",\n      ...(tools.length > 0 && {\n        extraDescription: [\n          `Tools that I can use to accomplish given task.`,\n          ...tools.map((tool) => `Tool '${tool.name}': ${tool.description}.`),\n        ].join(\"\\n\"),\n      }),\n    };\n  }\n\n  protected get templates(): RequirementAgentTemplates {\n    const overrides = this.input.templates || {};\n    const finalized = mapObj({\n      system: RequirementAgentSystemPrompt,\n      task: RequirementAgentTaskPrompt,\n      toolError: RequirementAgentToolErrorPrompt,\n      toolNoResult: RequirementAgentToolNoResultPrompt,\n    } as RequirementAgentTemplates)((\n      key,\n      defaultTemplate: RequirementAgentTemplates[typeof key],\n    ) => {\n      const override = overrides[key] ?? defaultTemplate;\n      if (override instanceof PromptTemplate) {\n        return override;\n      }\n      return override(defaultTemplate) ?? defaultTemplate;\n    });\n\n    if (this.input.role || this.input.instructions || this.input.notes) {\n      finalized.system.update({\n        defaults: pickBy(\n          {\n            role: this.input.role,\n            instructions: isEmptyish(this.input.instructions)\n              ? undefined\n              : castArray(this.input.instructions)\n                  .map((v) => `- ${v}`)\n                  .join(\"\\n\"),\n            notes: isEmptyish(this.input.notes)\n              ? undefined\n              : castArray(this.input.notes)\n                  .map((v) => `- ${v}`)\n                  .join(\"\\n\"),\n          },\n          isDefined,\n        ),\n      });\n    }\n\n    return finalized;\n  }\n\n  createSnapshot() {\n    return {\n      ...super.createSnapshot(),\n      input: shallowCopy(this.input),\n      emitter: this.emitter,\n      runner: this.runner,\n    };\n  }\n\n  set memory(memory: BaseMemory) {\n    this.input.memory = memory;\n  }\n\n  get memory(): BaseMemory {\n    return this.input.memory!;\n  }\n}\n"]}

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

@@ -1,14 +1,14 @@
 import '../../context.cjs';
 import 'zod';
-import '../../base-BqzN2QwB.cjs';
+import '../../base-s3ioVIE9.cjs';
 import '../../tools/base.cjs';
 import '../../memory/base.cjs';
-import '../../emitter-UkIkNrek.cjs';
-import '../../chat-DaqV3woF.cjs';
+import '../../emitter-D4OcelZ9.cjs';
+import '../../chat-C20nYj46.cjs';
 import '../../backend/message.cjs';
-import '../../types-YoNQSxQH.cjs';
+import '../../types-BNsa3W7Z.cjs';
 import './requirements/requirement.cjs';
-export { b as RequirementAgent, a as RequirementAgentInput, R as RequirementAgentTemplateFactory } from './runner.cjs';
+export { R as RequirementAgent, a as RequirementAgentInput, b as RequirementAgentTemplateFactory } from './runner.cjs';
 import './utils/toolCallChecker.cjs';
 import '../../internals/types.cjs';
 import '../../internals/serializable.cjs';

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

@@ -1,14 +1,14 @@
 import '../../context.js';
 import 'zod';
-import '../../base-BfhNzUYf.js';
+import '../../base-BjzyQ8k6.js';
 import '../../tools/base.js';
 import '../../memory/base.js';
-import '../../emitter-CL56qab4.js';
-import '../../chat-DTxmaKP0.js';
+import '../../emitter-kHxkUxco.js';
+import '../../chat-CfTNfR60.js';
 import '../../backend/message.js';
-import '../../types-YT-WdK3m.js';
+import '../../types-DYtzF6ya.js';
 import './requirements/requirement.js';
-export { b as RequirementAgent, a as RequirementAgentInput, R as RequirementAgentTemplateFactory } from './runner.js';
+export { R as RequirementAgent, a as RequirementAgentInput, b as RequirementAgentTemplateFactory } from './runner.js';
 import './utils/toolCallChecker.js';
 import '../../internals/types.js';
 import '../../internals/serializable.js';

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

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/agents/requirement/agent.ts"],"names":["RequirementAgent","BaseAgent","emitter","Emitter","root","child","namespace","creator","input","runner","RequirementAgentRunner","memory","UnconstrainedMemory","tools","requirements","saveIntermediateSteps","finalAnswerAsTool","toolCallChecker","middlewares","push","register","_run","options","run","execution","maxRetriesPerStep","totalMaxRetries","maxIterations","createToolCallChecker","llm","expectedOutput","templates","newMessages","processInput","addMessages","messages","finalState","reset","addMany","lastPair","slice","answer","Error","result","state","prompt","userMessage","UserMessage","task","render","context","undefined","config","Object","assign","checker","ToolCallChecker","enabled","meta","name","description","length","extraDescription","map","tool","join","overrides","finalized","mapObj","system","RequirementAgentSystemPrompt","RequirementAgentTaskPrompt","toolError","RequirementAgentToolErrorPrompt","toolNoResult","RequirementAgentToolNoResultPrompt","key","defaultTemplate","override","PromptTemplate","role","instructions","notes","update","defaults","pickBy","isEmptyish","castArray","v","isDefined","createSnapshot","shallowCopy"],"mappings":";;;;;;;;;;;;;;;AAqFO,MAAMA,yBAAyBC,SAAAA,CAAAA;EArFtC;;;EA0FkBC,OAAAA,GAAUC,OAAAA,CAAQC,KAAKC,KAAAA,CAAiC;IACtEC,SAAAA,EAAW;AAAC,MAAA,OAAA;AAAS,MAAA;;IACrBC,OAAAA,EAAS;GACX,CAAA;AAEmBC,EAAAA,KAAAA;EAETC,MAAAA,GAEoBC,sBAAAA;AAE9B,EAAA,WAAA,CAAYF,KAAAA,EAA8B;AACxC,IAAA,KAAA,EAAK;AACL,IAAA,IAAA,CAAKA,KAAAA,GAAQ;MACX,GAAGA,KAAAA;MACHG,MAAAA,EAAQH,KAAAA,CAAMG,MAAAA,IAAU,IAAIC,mBAAAA,EAAAA;MAC5BC,KAAAA,EAAOL,KAAAA,CAAMK,SAAS,EAAA;MACtBC,YAAAA,EAAcN,KAAAA,CAAMM,gBAAgB,EAAA;AACpCC,MAAAA,qBAAAA,EAAuBP,MAAMO,qBAAAA,IAAyB,IAAA;AACtDC,MAAAA,iBAAAA,EAAmBR,MAAMQ,iBAAAA,IAAqB,IAAA;AAC9CC,MAAAA,eAAAA,EAAiBT,MAAMS,eAAAA,IAAmB;AAC5C,KAAA;AACA,IAAA,IAAIT,MAAMU,WAAAA,EAAa;AACrB,MAAA,IAAA,CAAKA,WAAAA,CAAYC,IAAAA,CAAI,GAAIX,KAAAA,CAAMU,WAAW,CAAA;AAC5C;AACF;EAEA;AACE,IAAA,IAAA,CAAKE,QAAAA,EAAQ;AACf;AAEA,EAAA,MAAgBC,IAAAA,CACdb,KAAAA,EACAc,OAAAA,GAAsC,IACtCC,GAAAA,EACiC;AACjC,IAAA,MAAMC,SAAAA,GAA6C;MACjDC,iBAAAA,EAAmB,CAAA;MACnBC,eAAAA,EAAiB,EAAA;MACjBC,aAAAA,EAAe,EAAA;AACf,MAAA,GAAG,KAAKnB,KAAAA,CAAMgB,SAAAA;AACd,MAAA,GAAGF,OAAAA,CAAQE;AACb,KAAA;AAEA,IAAA,MAAMP,eAAAA,GAAkB,KAAKW,qBAAAA,EAAqB;AAElD,IAAA,MAAMnB,MAAAA,GAAS,IAAI,IAAA,CAAKA,MAAAA,CACtB,IAAA,CAAKD,KAAAA,CAAMqB,GAAAA,EACXL,SAAAA,EACA,IAAA,CAAKhB,KAAAA,CAAMK,KAAAA,IAAS,EAAA,EACpB,IAAA,CAAKL,KAAAA,CAAMM,YAAAA,IAAgB,EAAA,EAC3BN,KAAAA,CAAMsB,cAAAA,IAAkB,IAAA,EACxBb,eAAAA,EACAM,GAAAA,EACA,IAAA,CAAKf,KAAAA,CAAMQ,iBAAAA,EACX,IAAA,CAAKe,SAAS,CAAA;AAIhB,IAAA,MAAMC,WAAAA,GAAc,IAAA,CAAKC,YAAAA,CAAazB,KAAAA,CAAAA;AACtC,IAAA,MAAMC,OAAOyB,WAAAA,CAAY;AAAI,MAAA,GAAA,IAAA,CAAKvB,MAAAA,CAAOwB;AAAS,KAAA,CAAA;AAClD,IAAA,MAAM1B,MAAAA,CAAOyB,YAAYF,WAAAA,CAAAA;AAGzB,IAAA,MAAMI,UAAAA,GAAa,MAAM3B,MAAAA,CAAOc,GAAAA,EAAG;AAGnC,IAAA,IAAI,IAAA,CAAKf,MAAMO,qBAAAA,EAAuB;AACpC,MAAA,IAAA,CAAKJ,OAAO0B,KAAAA,EAAK;AACjB,MAAA,MAAM,IAAA,CAAK1B,MAAAA,CAAO2B,OAAAA,CAAQF,UAAAA,CAAWzB,OAAOwB,QAAQ,CAAA;KACtD,MAAO;AACL,MAAA,MAAM,IAAA,CAAKxB,MAAAA,CAAO2B,OAAAA,CAAQN,WAAAA,CAAAA;AAE1B,MAAA,MAAMG,QAAAA,GAAWC,WAAWzB,MAAAA,CAAOwB,QAAAA;AACnC,MAAA,MAAMI,QAAAA,GAAWJ,QAAAA,CAASK,KAAAA,CAAM,EAAC,CAAA;AACjC,MAAA,MAAM,IAAA,CAAK7B,MAAAA,CAAO2B,OAAAA,CAAQC,QAAAA,CAAAA;AAC5B;AAEA,IAAA,IAAI,CAACH,WAAWK,MAAAA,EAAQ;AACtB,MAAA,MAAM,IAAIC,MAAM,sCAAA,CAAA;AAClB;AAEA,IAAA,OAAO;AACLC,MAAAA,MAAAA,EAAQP,UAAAA,CAAWK,MAAAA;AACnB9B,MAAAA,MAAAA,EAAQyB,UAAAA,CAAWzB,MAAAA;MACnBiC,KAAAA,EAAOR;AACT,KAAA;AACF;AAEUH,EAAAA,YAAAA,CAAazB,KAAAA,EAA4C;AACjE,IAAA,IAAI,CAACA,MAAMqC,MAAAA,EAAQ;AACjB,MAAA,OAAO,EAAA;AACT;AAEA,IAAA,MAAMC,cAAc,IAAIC,WAAAA,CACtB,IAAA,CAAKhB,SAAAA,CAAUiB,KAAKC,MAAAA,CAAO;AACzBJ,MAAAA,MAAAA,EAAQrC,KAAAA,CAAMqC,MAAAA;AACdK,MAAAA,OAAAA,EAAS1C,KAAAA,CAAM0C,OAAAA;AACfpB,MAAAA,cAAAA,EAAgB,OAAOtB,KAAAA,CAAMsB,cAAAA,KAAmB,QAAA,GAAWtB,MAAMsB,cAAAA,GAAiBqB;AACpF,KAAA,CAAA,CAAA;AAGF,IAAA,OAAO;AAACL,MAAAA;;AACV;EAEUlB,qBAAAA,GAAyC;AACjD,IAAA,MAAMwB,SAAgC,EAAC;AAEvC,IAAA,IAAI,OAAO,IAAA,CAAK5C,KAAAA,CAAMS,eAAAA,KAAoB,QAAA,EAAU;AAClDoC,MAAAA,MAAAA,CAAOC,MAAAA,CAAOF,MAAAA,EAAQ,IAAA,CAAK5C,KAAAA,CAAMS,eAAe,CAAA;AAClD;AAEA,IAAA,MAAMsC,OAAAA,GAAU,IAAIC,eAAAA,CAAgBJ,MAAAA,CAAAA;AACpCG,IAAAA,OAAAA,CAAQE,OAAAA,GAAU,IAAA,CAAKjD,KAAAA,CAAMS,eAAAA,KAAoB,KAAA;AACjD,IAAA,OAAOsC,OAAAA;AACT;AAEA,EAAA,IAAIG,IAAAA,GAAkB;AACpB,IAAA,MAAM7C,KAAAA,GAAQ,IAAA,CAAKL,KAAAA,CAAMK,KAAAA,IAAS,EAAA;AAElC,IAAA,OAAO;MACL8C,IAAAA,EAAM,IAAA,CAAKnD,MAAMmD,IAAAA,IAAQ,aAAA;AACzB9C,MAAAA,KAAAA;MACA+C,WAAAA,EACE,IAAA,CAAKpD,MAAMoD,WAAAA,IACX,yLAAA;MAEF,GAAI/C,KAAAA,CAAMgD,SAAS,CAAA,IAAK;QACtBC,gBAAAA,EAAkB;AAChB,UAAA,CAAA,8CAAA,CAAA;aACGjD,KAAAA,CAAMkD,GAAAA,CAAI,CAACC,IAAAA,KAAS,CAAA,MAAA,EAASA,KAAKL,IAAI,CAAA,GAAA,EAAMK,IAAAA,CAAKJ,WAAW,CAAA,CAAA,CAAG;AAClEK,SAAAA,CAAAA,IAAAA,CAAK,IAAA;AACT;AACF,KAAA;AACF;AAEA,EAAA,IAAclC,SAAAA,GAAuC;AACnD,IAAA,MAAMmC,SAAAA,GAAY,IAAA,CAAK1D,KAAAA,CAAMuB,SAAAA,IAAa,EAAC;AAC3C,IAAA,MAAMoC,YAAYC,MAAAA,CAAO;MACvBC,MAAAA,EAAQC,4BAAAA;MACRtB,IAAAA,EAAMuB,0BAAAA;MACNC,SAAAA,EAAWC,+BAAAA;MACXC,YAAAA,EAAcC;KAChB,CAAA,CAAgC,CAC9BC,GAAAA,EACAC,eAAAA,KAAAA;AAEA,MAAA,MAAMC,QAAAA,GAAWZ,SAAAA,CAAUU,GAAAA,CAAAA,IAAQC,eAAAA;AACnC,MAAA,IAAIC,oBAAoBC,cAAAA,EAAgB;AACtC,QAAA,OAAOD,QAAAA;AACT;AACA,MAAA,OAAOA,QAAAA,CAASD,eAAAA,CAAAA,IAAoBA,eAAAA;KACtC,CAAA;AAEA,IAAA,IAAI,IAAA,CAAKrE,MAAMwE,IAAAA,IAAQ,IAAA,CAAKxE,MAAMyE,YAAAA,IAAgB,IAAA,CAAKzE,MAAM0E,KAAAA,EAAO;AAClEf,MAAAA,SAAAA,CAAUE,OAAOc,MAAAA,CAAO;AACtBC,QAAAA,QAAAA,EAAUC,MAAAA,CACR;AACEL,UAAAA,IAAAA,EAAM,KAAKxE,KAAAA,CAAMwE,IAAAA;AACjBC,UAAAA,YAAAA,EAAcK,WAAW,IAAA,CAAK9E,KAAAA,CAAMyE,YAAY,CAAA,GAC5C9B,MAAAA,GACAoC,UAAU,IAAA,CAAK/E,KAAAA,CAAMyE,YAAY,CAAA,CAC9BlB,GAAAA,CAAI,CAACyB,CAAAA,KAAM,CAAA,EAAA,EAAKA,CAAAA,CAAAA,CAAG,CAAA,CACnBvB,KAAK,IAAA,CAAA;AACZiB,UAAAA,KAAAA,EAAOI,WAAW,IAAA,CAAK9E,KAAAA,CAAM0E,KAAK,CAAA,GAC9B/B,MAAAA,GACAoC,UAAU,IAAA,CAAK/E,KAAAA,CAAM0E,KAAK,CAAA,CACvBnB,GAAAA,CAAI,CAACyB,CAAAA,KAAM,CAAA,EAAA,EAAKA,CAAAA,CAAAA,CAAG,CAAA,CACnBvB,KAAK,IAAA;AACd,SAAA,EACAwB,SAAAA;OAEJ,CAAA;AACF;AAEA,IAAA,OAAOtB,SAAAA;AACT;EAEAuB,cAAAA,GAAiB;AACf,IAAA,OAAO;AACL,MAAA,GAAG,MAAMA,cAAAA,EAAAA;MACTlF,KAAAA,EAAOmF,WAAAA,CAAY,KAAKnF,KAAK,CAAA;AAC7BN,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;AACdO,MAAAA,MAAAA,EAAQ,IAAA,CAAKA;AACf,KAAA;AACF;AAEA,EAAA,IAAIE,OAAOA,MAAAA,EAAoB;AAC7B,IAAA,IAAA,CAAKH,MAAMG,MAAAA,GAASA,MAAAA;AACtB;AAEA,EAAA,IAAIA,MAAAA,GAAqB;AACvB,IAAA,OAAO,KAAKH,KAAAA,CAAMG,MAAAA;AACpB;AACF","file":"agent.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { BaseAgent } from \"@/agents/base.js\";\nimport { AnyTool } from \"@/tools/base.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\nimport { AgentMeta } from \"@/agents/types.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { ChatModel } from \"@/backend/chat.js\";\nimport type { GetRunContext, MiddlewareType } from \"@/context.js\";\nimport { UnconstrainedMemory } from \"@/memory/unconstrainedMemory.js\";\nimport { Message, UserMessage } from \"@/backend/message.js\";\nimport { shallowCopy } from \"@/serializer/utils.js\";\nimport { mapObj } from \"@/internals/helpers/object.js\";\nimport { PromptTemplate } from \"@/template.js\";\nimport {\n  RequirementAgentCallbacks,\n  RequirementAgentExecutionConfig,\n  RequirementAgentOutput,\n  RequirementAgentRunInput,\n  RequirementAgentRunOptions,\n  RequirementAgentTemplates,\n} from \"./types.js\";\nimport {\n  RequirementAgentSystemPrompt,\n  RequirementAgentTaskPrompt,\n  RequirementAgentToolErrorPrompt,\n  RequirementAgentToolNoResultPrompt,\n} from \"./prompts.js\";\nimport { Requirement } from \"./requirements/requirement.js\";\nimport { RequirementAgentRunner } from \"./runner.js\";\nimport { ToolCallChecker, ToolCallCheckerConfig } from \"./utils/toolCallChecker.js\";\nimport { isDefined, isEmptyish, pickBy } from \"remeda\";\nimport { castArray } from \"@/internals/helpers/array.js\";\nimport type { RequiredExcept } from \"@/internals/types.js\";\n\nexport type RequirementAgentTemplateFactory<K extends keyof RequirementAgentTemplates> = (\n  template: RequirementAgentTemplates[K],\n) => RequirementAgentTemplates[K];\n\nexport interface RequirementAgentInput {\n  llm: ChatModel;\n  memory?: BaseMemory;\n  tools?: AnyTool[];\n  requirements?: Requirement[];\n  name?: string;\n  description?: string;\n  role?: string;\n  instructions?: string | string[];\n  notes?: string | string[];\n  toolCallChecker?: boolean | ToolCallCheckerConfig;\n  finalAnswerAsTool?: boolean;\n  saveIntermediateSteps?: boolean;\n  execution?: RequirementAgentExecutionConfig;\n  templates?: Partial<{\n    [K in keyof RequirementAgentTemplates]:\n      | RequirementAgentTemplates[K]\n      | RequirementAgentTemplateFactory<K>;\n  }>;\n  middlewares?: MiddlewareType<RequirementAgent>[];\n}\n\ntype InternalRequirementAgentInput = RequiredExcept<\n  RequirementAgentInput,\n  | \"role\"\n  | \"instructions\"\n  | \"name\"\n  | \"description\"\n  | \"templates\"\n  | \"notes\"\n  | \"execution\"\n  | \"middlewares\"\n>;\n\n/**\n * RequirementAgent - A declarative AI agent with rule-based constraints\n *\n * The RequirementAgent provides predictable, controlled execution behavior across\n * different language models through rule-based constraints (Requirements).\n * Requirements can be configured as strict or flexible as necessary, adapting to\n * task requirements while ensuring consistent execution regardless of the underlying\n * model's reasoning or tool-calling capabilities.\n */\nexport class RequirementAgent extends BaseAgent<\n  RequirementAgentRunInput,\n  RequirementAgentOutput,\n  RequirementAgentRunOptions\n> {\n  public readonly emitter = Emitter.root.child<RequirementAgentCallbacks>({\n    namespace: [\"agent\", \"requirement\"],\n    creator: this,\n  });\n\n  protected readonly input: InternalRequirementAgentInput;\n\n  protected runner: new (\n    ...args: ConstructorParameters<typeof RequirementAgentRunner>\n  ) => RequirementAgentRunner = RequirementAgentRunner;\n\n  constructor(input: RequirementAgentInput) {\n    super();\n    this.input = {\n      ...input,\n      memory: input.memory || new UnconstrainedMemory(),\n      tools: input.tools || [],\n      requirements: input.requirements || [],\n      saveIntermediateSteps: input.saveIntermediateSteps ?? true,\n      finalAnswerAsTool: input.finalAnswerAsTool ?? true,\n      toolCallChecker: input.toolCallChecker ?? true,\n    };\n    if (input.middlewares) {\n      this.middlewares.push(...input.middlewares);\n    }\n  }\n\n  static {\n    this.register();\n  }\n\n  protected async _run(\n    input: RequirementAgentRunInput,\n    options: RequirementAgentRunOptions = {},\n    run: GetRunContext<typeof this>,\n  ): Promise<RequirementAgentOutput> {\n    const execution: RequirementAgentExecutionConfig = {\n      maxRetriesPerStep: 3,\n      totalMaxRetries: 20,\n      maxIterations: 20,\n      ...this.input.execution,\n      ...options.execution,\n    };\n\n    const toolCallChecker = this.createToolCallChecker();\n\n    const runner = new this.runner(\n      this.input.llm,\n      execution,\n      this.input.tools || [],\n      this.input.requirements || [],\n      input.expectedOutput || null,\n      toolCallChecker,\n      run,\n      this.input.finalAnswerAsTool,\n      this.templates,\n    );\n\n    // Process input messages\n    const newMessages = this.processInput(input);\n    await runner.addMessages([...this.memory.messages]);\n    await runner.addMessages(newMessages);\n\n    // Run the agent\n    const finalState = await runner.run();\n\n    // Update memory\n    if (this.input.saveIntermediateSteps) {\n      this.memory.reset();\n      await this.memory.addMany(finalState.memory.messages);\n    } else {\n      await this.memory.addMany(newMessages);\n      // Add last tool call pair\n      const messages = finalState.memory.messages;\n      const lastPair = messages.slice(-2);\n      await this.memory.addMany(lastPair);\n    }\n\n    if (!finalState.answer) {\n      throw new Error(\"Agent did not produce a final answer\");\n    }\n\n    return {\n      result: finalState.answer,\n      memory: finalState.memory,\n      state: finalState,\n    };\n  }\n\n  protected processInput(input: RequirementAgentRunInput): Message[] {\n    if (!input.prompt) {\n      return [];\n    }\n\n    const userMessage = new UserMessage(\n      this.templates.task.render({\n        prompt: input.prompt,\n        context: input.context,\n        expectedOutput: typeof input.expectedOutput === \"string\" ? input.expectedOutput : undefined,\n      }),\n    );\n\n    return [userMessage];\n  }\n\n  protected createToolCallChecker(): ToolCallChecker {\n    const config: ToolCallCheckerConfig = {};\n\n    if (typeof this.input.toolCallChecker === \"object\") {\n      Object.assign(config, this.input.toolCallChecker);\n    }\n\n    const checker = new ToolCallChecker(config);\n    checker.enabled = this.input.toolCallChecker !== false;\n    return checker;\n  }\n\n  get meta(): AgentMeta {\n    const tools = this.input.tools || [];\n\n    return {\n      name: this.input.name ?? \"Requirement\",\n      tools,\n      description:\n        this.input.description ??\n        \"The RequirementAgent is a declarative AI agent implementation that provides predictable, \" +\n          \"controlled execution behavior across different language models through rule-based constraints.\",\n      ...(tools.length > 0 && {\n        extraDescription: [\n          `Tools that I can use to accomplish given task.`,\n          ...tools.map((tool) => `Tool '${tool.name}': ${tool.description}.`),\n        ].join(\"\\n\"),\n      }),\n    };\n  }\n\n  protected get templates(): RequirementAgentTemplates {\n    const overrides = this.input.templates || {};\n    const finalized = mapObj({\n      system: RequirementAgentSystemPrompt,\n      task: RequirementAgentTaskPrompt,\n      toolError: RequirementAgentToolErrorPrompt,\n      toolNoResult: RequirementAgentToolNoResultPrompt,\n    } as RequirementAgentTemplates)((\n      key,\n      defaultTemplate: RequirementAgentTemplates[typeof key],\n    ) => {\n      const override = overrides[key] ?? defaultTemplate;\n      if (override instanceof PromptTemplate) {\n        return override;\n      }\n      return override(defaultTemplate) ?? defaultTemplate;\n    });\n\n    if (this.input.role || this.input.instructions || this.input.notes) {\n      finalized.system.update({\n        defaults: pickBy(\n          {\n            role: this.input.role,\n            instructions: isEmptyish(this.input.instructions)\n              ? undefined\n              : castArray(this.input.instructions)\n                  .map((v) => `- ${v}`)\n                  .join(\"\\n\"),\n            notes: isEmptyish(this.input.notes)\n              ? undefined\n              : castArray(this.input.notes)\n                  .map((v) => `- ${v}`)\n                  .join(\"\\n\"),\n          },\n          isDefined,\n        ),\n      });\n    }\n\n    return finalized;\n  }\n\n  createSnapshot() {\n    return {\n      ...super.createSnapshot(),\n      input: shallowCopy(this.input),\n      emitter: this.emitter,\n      runner: this.runner,\n    };\n  }\n\n  set memory(memory: BaseMemory) {\n    this.input.memory = memory;\n  }\n\n  get memory(): BaseMemory {\n    return this.input.memory!;\n  }\n}\n"]}
+{"version":3,"sources":["../../../src/agents/requirement/agent.ts"],"names":["RequirementAgent","BaseAgent","emitter","Emitter","root","child","namespace","creator","input","runner","RequirementAgentRunner","memory","UnconstrainedMemory","tools","requirements","saveIntermediateSteps","finalAnswerAsTool","toolCallChecker","middlewares","push","register","_run","options","run","execution","maxRetriesPerStep","totalMaxRetries","maxIterations","createToolCallChecker","llm","expectedOutput","templates","newMessages","processInput","addMessages","messages","finalState","reset","addMany","lastPair","slice","answer","Error","result","state","prompt","userMessage","UserMessage","task","render","context","undefined","config","Object","assign","checker","ToolCallChecker","enabled","meta","name","description","length","extraDescription","map","tool","join","overrides","finalized","mapObj","system","RequirementAgentSystemPrompt","RequirementAgentTaskPrompt","toolError","RequirementAgentToolErrorPrompt","toolNoResult","RequirementAgentToolNoResultPrompt","key","defaultTemplate","override","PromptTemplate","role","instructions","notes","update","defaults","pickBy","isEmptyish","castArray","v","isDefined","createSnapshot","shallowCopy"],"mappings":";;;;;;;;;;;;;;;AAqFO,MAAMA,yBAAyBC,SAAAA,CAAAA;EArFtC;;;EA0FkBC,OAAAA,GAAUC,OAAAA,CAAQC,KAAKC,KAAAA,CAAiC;IACtEC,SAAAA,EAAW;AAAC,MAAA,OAAA;AAAS,MAAA;;IACrBC,OAAAA,EAAS;GACX,CAAA;AAEmBC,EAAAA,KAAAA;EAETC,MAAAA,GAEoBC,sBAAAA;AAE9B,EAAA,WAAA,CAAYF,KAAAA,EAA8B;AACxC,IAAA,KAAA,EAAK;AACL,IAAA,IAAA,CAAKA,KAAAA,GAAQ;MACX,GAAGA,KAAAA;MACHG,MAAAA,EAAQH,KAAAA,CAAMG,MAAAA,IAAU,IAAIC,mBAAAA,EAAAA;MAC5BC,KAAAA,EAAOL,KAAAA,CAAMK,SAAS,EAAA;MACtBC,YAAAA,EAAcN,KAAAA,CAAMM,gBAAgB,EAAA;AACpCC,MAAAA,qBAAAA,EAAuBP,MAAMO,qBAAAA,IAAyB,IAAA;AACtDC,MAAAA,iBAAAA,EAAmBR,MAAMQ,iBAAAA,IAAqB,IAAA;AAC9CC,MAAAA,eAAAA,EAAiBT,MAAMS,eAAAA,IAAmB;AAC5C,KAAA;AACA,IAAA,IAAIT,MAAMU,WAAAA,EAAa;AACrB,MAAA,IAAA,CAAKA,WAAAA,CAAYC,IAAAA,CAAI,GAAIX,KAAAA,CAAMU,WAAW,CAAA;AAC5C,IAAA;AACF,EAAA;EAEA;AACE,IAAA,IAAA,CAAKE,QAAAA,EAAQ;AACf;AAEA,EAAA,MAAgBC,IAAAA,CACdb,KAAAA,EACAc,OAAAA,GAAsC,IACtCC,GAAAA,EACiC;AACjC,IAAA,MAAMC,SAAAA,GAA6C;MACjDC,iBAAAA,EAAmB,CAAA;MACnBC,eAAAA,EAAiB,EAAA;MACjBC,aAAAA,EAAe,EAAA;AACf,MAAA,GAAG,KAAKnB,KAAAA,CAAMgB,SAAAA;AACd,MAAA,GAAGF,OAAAA,CAAQE;AACb,KAAA;AAEA,IAAA,MAAMP,eAAAA,GAAkB,KAAKW,qBAAAA,EAAqB;AAElD,IAAA,MAAMnB,MAAAA,GAAS,IAAI,IAAA,CAAKA,MAAAA,CACtB,IAAA,CAAKD,KAAAA,CAAMqB,GAAAA,EACXL,SAAAA,EACA,IAAA,CAAKhB,KAAAA,CAAMK,KAAAA,IAAS,EAAA,EACpB,IAAA,CAAKL,KAAAA,CAAMM,YAAAA,IAAgB,EAAA,EAC3BN,KAAAA,CAAMsB,cAAAA,IAAkB,IAAA,EACxBb,eAAAA,EACAM,GAAAA,EACA,IAAA,CAAKf,KAAAA,CAAMQ,iBAAAA,EACX,IAAA,CAAKe,SAAS,CAAA;AAIhB,IAAA,MAAMC,WAAAA,GAAc,IAAA,CAAKC,YAAAA,CAAazB,KAAAA,CAAAA;AACtC,IAAA,MAAMC,OAAOyB,WAAAA,CAAY;AAAI,MAAA,GAAA,IAAA,CAAKvB,MAAAA,CAAOwB;AAAS,KAAA,CAAA;AAClD,IAAA,MAAM1B,MAAAA,CAAOyB,YAAYF,WAAAA,CAAAA;AAGzB,IAAA,MAAMI,UAAAA,GAAa,MAAM3B,MAAAA,CAAOc,GAAAA,EAAG;AAGnC,IAAA,IAAI,IAAA,CAAKf,MAAMO,qBAAAA,EAAuB;AACpC,MAAA,IAAA,CAAKJ,OAAO0B,KAAAA,EAAK;AACjB,MAAA,MAAM,IAAA,CAAK1B,MAAAA,CAAO2B,OAAAA,CAAQF,UAAAA,CAAWzB,OAAOwB,QAAQ,CAAA;IACtD,CAAA,MAAO;AACL,MAAA,MAAM,IAAA,CAAKxB,MAAAA,CAAO2B,OAAAA,CAAQN,WAAAA,CAAAA;AAE1B,MAAA,MAAMG,QAAAA,GAAWC,WAAWzB,MAAAA,CAAOwB,QAAAA;AACnC,MAAA,MAAMI,QAAAA,GAAWJ,QAAAA,CAASK,KAAAA,CAAM,EAAC,CAAA;AACjC,MAAA,MAAM,IAAA,CAAK7B,MAAAA,CAAO2B,OAAAA,CAAQC,QAAAA,CAAAA;AAC5B,IAAA;AAEA,IAAA,IAAI,CAACH,WAAWK,MAAAA,EAAQ;AACtB,MAAA,MAAM,IAAIC,MAAM,sCAAA,CAAA;AAClB,IAAA;AAEA,IAAA,OAAO;AACLC,MAAAA,MAAAA,EAAQP,UAAAA,CAAWK,MAAAA;AACnB9B,MAAAA,MAAAA,EAAQyB,UAAAA,CAAWzB,MAAAA;MACnBiC,KAAAA,EAAOR;AACT,KAAA;AACF,EAAA;AAEUH,EAAAA,YAAAA,CAAazB,KAAAA,EAA4C;AACjE,IAAA,IAAI,CAACA,MAAMqC,MAAAA,EAAQ;AACjB,MAAA,OAAO,EAAA;AACT,IAAA;AAEA,IAAA,MAAMC,cAAc,IAAIC,WAAAA,CACtB,IAAA,CAAKhB,SAAAA,CAAUiB,KAAKC,MAAAA,CAAO;AACzBJ,MAAAA,MAAAA,EAAQrC,KAAAA,CAAMqC,MAAAA;AACdK,MAAAA,OAAAA,EAAS1C,KAAAA,CAAM0C,OAAAA;AACfpB,MAAAA,cAAAA,EAAgB,OAAOtB,KAAAA,CAAMsB,cAAAA,KAAmB,QAAA,GAAWtB,MAAMsB,cAAAA,GAAiBqB;AACpF,KAAA,CAAA,CAAA;AAGF,IAAA,OAAO;AAACL,MAAAA;;AACV,EAAA;EAEUlB,qBAAAA,GAAyC;AACjD,IAAA,MAAMwB,SAAgC,EAAC;AAEvC,IAAA,IAAI,OAAO,IAAA,CAAK5C,KAAAA,CAAMS,eAAAA,KAAoB,QAAA,EAAU;AAClDoC,MAAAA,MAAAA,CAAOC,MAAAA,CAAOF,MAAAA,EAAQ,IAAA,CAAK5C,KAAAA,CAAMS,eAAe,CAAA;AAClD,IAAA;AAEA,IAAA,MAAMsC,OAAAA,GAAU,IAAIC,eAAAA,CAAgBJ,MAAAA,CAAAA;AACpCG,IAAAA,OAAAA,CAAQE,OAAAA,GAAU,IAAA,CAAKjD,KAAAA,CAAMS,eAAAA,KAAoB,KAAA;AACjD,IAAA,OAAOsC,OAAAA;AACT,EAAA;AAEA,EAAA,IAAIG,IAAAA,GAAkB;AACpB,IAAA,MAAM7C,KAAAA,GAAQ,IAAA,CAAKL,KAAAA,CAAMK,KAAAA,IAAS,EAAA;AAElC,IAAA,OAAO;MACL8C,IAAAA,EAAM,IAAA,CAAKnD,MAAMmD,IAAAA,IAAQ,aAAA;AACzB9C,MAAAA,KAAAA;MACA+C,WAAAA,EACE,IAAA,CAAKpD,MAAMoD,WAAAA,IACX,yLAAA;MAEF,GAAI/C,KAAAA,CAAMgD,SAAS,CAAA,IAAK;QACtBC,gBAAAA,EAAkB;AAChB,UAAA,CAAA,8CAAA,CAAA;aACGjD,KAAAA,CAAMkD,GAAAA,CAAI,CAACC,IAAAA,KAAS,CAAA,MAAA,EAASA,KAAKL,IAAI,CAAA,GAAA,EAAMK,IAAAA,CAAKJ,WAAW,CAAA,CAAA,CAAG;AAClEK,SAAAA,CAAAA,IAAAA,CAAK,IAAA;AACT;AACF,KAAA;AACF,EAAA;AAEA,EAAA,IAAclC,SAAAA,GAAuC;AACnD,IAAA,MAAMmC,SAAAA,GAAY,IAAA,CAAK1D,KAAAA,CAAMuB,SAAAA,IAAa,EAAC;AAC3C,IAAA,MAAMoC,YAAYC,MAAAA,CAAO;MACvBC,MAAAA,EAAQC,4BAAAA;MACRtB,IAAAA,EAAMuB,0BAAAA;MACNC,SAAAA,EAAWC,+BAAAA;MACXC,YAAAA,EAAcC;KAChB,CAAA,CAAgC,CAC9BC,GAAAA,EACAC,eAAAA,KAAAA;AAEA,MAAA,MAAMC,QAAAA,GAAWZ,SAAAA,CAAUU,GAAAA,CAAAA,IAAQC,eAAAA;AACnC,MAAA,IAAIC,oBAAoBC,cAAAA,EAAgB;AACtC,QAAA,OAAOD,QAAAA;AACT,MAAA;AACA,MAAA,OAAOA,QAAAA,CAASD,eAAAA,CAAAA,IAAoBA,eAAAA;IACtC,CAAA,CAAA;AAEA,IAAA,IAAI,IAAA,CAAKrE,MAAMwE,IAAAA,IAAQ,IAAA,CAAKxE,MAAMyE,YAAAA,IAAgB,IAAA,CAAKzE,MAAM0E,KAAAA,EAAO;AAClEf,MAAAA,SAAAA,CAAUE,OAAOc,MAAAA,CAAO;AACtBC,QAAAA,QAAAA,EAAUC,MAAAA,CACR;AACEL,UAAAA,IAAAA,EAAM,KAAKxE,KAAAA,CAAMwE,IAAAA;AACjBC,UAAAA,YAAAA,EAAcK,WAAW,IAAA,CAAK9E,KAAAA,CAAMyE,YAAY,CAAA,GAC5C9B,MAAAA,GACAoC,UAAU,IAAA,CAAK/E,KAAAA,CAAMyE,YAAY,CAAA,CAC9BlB,GAAAA,CAAI,CAACyB,CAAAA,KAAM,CAAA,EAAA,EAAKA,CAAAA,CAAAA,CAAG,CAAA,CACnBvB,KAAK,IAAA,CAAA;AACZiB,UAAAA,KAAAA,EAAOI,WAAW,IAAA,CAAK9E,KAAAA,CAAM0E,KAAK,CAAA,GAC9B/B,MAAAA,GACAoC,UAAU,IAAA,CAAK/E,KAAAA,CAAM0E,KAAK,CAAA,CACvBnB,GAAAA,CAAI,CAACyB,CAAAA,KAAM,CAAA,EAAA,EAAKA,CAAAA,CAAAA,CAAG,CAAA,CACnBvB,KAAK,IAAA;AACd,SAAA,EACAwB,SAAAA;OAEJ,CAAA;AACF,IAAA;AAEA,IAAA,OAAOtB,SAAAA;AACT,EAAA;EAEAuB,cAAAA,GAAiB;AACf,IAAA,OAAO;AACL,MAAA,GAAG,MAAMA,cAAAA,EAAAA;MACTlF,KAAAA,EAAOmF,WAAAA,CAAY,KAAKnF,KAAK,CAAA;AAC7BN,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;AACdO,MAAAA,MAAAA,EAAQ,IAAA,CAAKA;AACf,KAAA;AACF,EAAA;AAEA,EAAA,IAAIE,OAAOA,MAAAA,EAAoB;AAC7B,IAAA,IAAA,CAAKH,MAAMG,MAAAA,GAASA,MAAAA;AACtB,EAAA;AAEA,EAAA,IAAIA,MAAAA,GAAqB;AACvB,IAAA,OAAO,KAAKH,KAAAA,CAAMG,MAAAA;AACpB,EAAA;AACF","file":"agent.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { BaseAgent } from \"@/agents/base.js\";\nimport { AnyTool } from \"@/tools/base.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\nimport { AgentMeta } from \"@/agents/types.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { ChatModel } from \"@/backend/chat.js\";\nimport type { GetRunContext, MiddlewareType } from \"@/context.js\";\nimport { UnconstrainedMemory } from \"@/memory/unconstrainedMemory.js\";\nimport { Message, UserMessage } from \"@/backend/message.js\";\nimport { shallowCopy } from \"@/serializer/utils.js\";\nimport { mapObj } from \"@/internals/helpers/object.js\";\nimport { PromptTemplate } from \"@/template.js\";\nimport {\n  RequirementAgentCallbacks,\n  RequirementAgentExecutionConfig,\n  RequirementAgentOutput,\n  RequirementAgentRunInput,\n  RequirementAgentRunOptions,\n  RequirementAgentTemplates,\n} from \"./types.js\";\nimport {\n  RequirementAgentSystemPrompt,\n  RequirementAgentTaskPrompt,\n  RequirementAgentToolErrorPrompt,\n  RequirementAgentToolNoResultPrompt,\n} from \"./prompts.js\";\nimport { Requirement } from \"./requirements/requirement.js\";\nimport { RequirementAgentRunner } from \"./runner.js\";\nimport { ToolCallChecker, ToolCallCheckerConfig } from \"./utils/toolCallChecker.js\";\nimport { isDefined, isEmptyish, pickBy } from \"remeda\";\nimport { castArray } from \"@/internals/helpers/array.js\";\nimport type { RequiredExcept } from \"@/internals/types.js\";\n\nexport type RequirementAgentTemplateFactory<K extends keyof RequirementAgentTemplates> = (\n  template: RequirementAgentTemplates[K],\n) => RequirementAgentTemplates[K];\n\nexport interface RequirementAgentInput {\n  llm: ChatModel;\n  memory?: BaseMemory;\n  tools?: AnyTool[];\n  requirements?: Requirement[];\n  name?: string;\n  description?: string;\n  role?: string;\n  instructions?: string | string[];\n  notes?: string | string[];\n  toolCallChecker?: boolean | ToolCallCheckerConfig;\n  finalAnswerAsTool?: boolean;\n  saveIntermediateSteps?: boolean;\n  execution?: RequirementAgentExecutionConfig;\n  templates?: Partial<{\n    [K in keyof RequirementAgentTemplates]:\n      | RequirementAgentTemplates[K]\n      | RequirementAgentTemplateFactory<K>;\n  }>;\n  middlewares?: MiddlewareType<RequirementAgent>[];\n}\n\ntype InternalRequirementAgentInput = RequiredExcept<\n  RequirementAgentInput,\n  | \"role\"\n  | \"instructions\"\n  | \"name\"\n  | \"description\"\n  | \"templates\"\n  | \"notes\"\n  | \"execution\"\n  | \"middlewares\"\n>;\n\n/**\n * RequirementAgent - A declarative AI agent with rule-based constraints\n *\n * The RequirementAgent provides predictable, controlled execution behavior across\n * different language models through rule-based constraints (Requirements).\n * Requirements can be configured as strict or flexible as necessary, adapting to\n * task requirements while ensuring consistent execution regardless of the underlying\n * model's reasoning or tool-calling capabilities.\n */\nexport class RequirementAgent extends BaseAgent<\n  RequirementAgentRunInput,\n  RequirementAgentOutput,\n  RequirementAgentRunOptions\n> {\n  public readonly emitter = Emitter.root.child<RequirementAgentCallbacks>({\n    namespace: [\"agent\", \"requirement\"],\n    creator: this,\n  });\n\n  protected readonly input: InternalRequirementAgentInput;\n\n  protected runner: new (\n    ...args: ConstructorParameters<typeof RequirementAgentRunner>\n  ) => RequirementAgentRunner = RequirementAgentRunner;\n\n  constructor(input: RequirementAgentInput) {\n    super();\n    this.input = {\n      ...input,\n      memory: input.memory || new UnconstrainedMemory(),\n      tools: input.tools || [],\n      requirements: input.requirements || [],\n      saveIntermediateSteps: input.saveIntermediateSteps ?? true,\n      finalAnswerAsTool: input.finalAnswerAsTool ?? true,\n      toolCallChecker: input.toolCallChecker ?? true,\n    };\n    if (input.middlewares) {\n      this.middlewares.push(...input.middlewares);\n    }\n  }\n\n  static {\n    this.register();\n  }\n\n  protected async _run(\n    input: RequirementAgentRunInput,\n    options: RequirementAgentRunOptions = {},\n    run: GetRunContext<typeof this>,\n  ): Promise<RequirementAgentOutput> {\n    const execution: RequirementAgentExecutionConfig = {\n      maxRetriesPerStep: 3,\n      totalMaxRetries: 20,\n      maxIterations: 20,\n      ...this.input.execution,\n      ...options.execution,\n    };\n\n    const toolCallChecker = this.createToolCallChecker();\n\n    const runner = new this.runner(\n      this.input.llm,\n      execution,\n      this.input.tools || [],\n      this.input.requirements || [],\n      input.expectedOutput || null,\n      toolCallChecker,\n      run,\n      this.input.finalAnswerAsTool,\n      this.templates,\n    );\n\n    // Process input messages\n    const newMessages = this.processInput(input);\n    await runner.addMessages([...this.memory.messages]);\n    await runner.addMessages(newMessages);\n\n    // Run the agent\n    const finalState = await runner.run();\n\n    // Update memory\n    if (this.input.saveIntermediateSteps) {\n      this.memory.reset();\n      await this.memory.addMany(finalState.memory.messages);\n    } else {\n      await this.memory.addMany(newMessages);\n      // Add last tool call pair\n      const messages = finalState.memory.messages;\n      const lastPair = messages.slice(-2);\n      await this.memory.addMany(lastPair);\n    }\n\n    if (!finalState.answer) {\n      throw new Error(\"Agent did not produce a final answer\");\n    }\n\n    return {\n      result: finalState.answer,\n      memory: finalState.memory,\n      state: finalState,\n    };\n  }\n\n  protected processInput(input: RequirementAgentRunInput): Message[] {\n    if (!input.prompt) {\n      return [];\n    }\n\n    const userMessage = new UserMessage(\n      this.templates.task.render({\n        prompt: input.prompt,\n        context: input.context,\n        expectedOutput: typeof input.expectedOutput === \"string\" ? input.expectedOutput : undefined,\n      }),\n    );\n\n    return [userMessage];\n  }\n\n  protected createToolCallChecker(): ToolCallChecker {\n    const config: ToolCallCheckerConfig = {};\n\n    if (typeof this.input.toolCallChecker === \"object\") {\n      Object.assign(config, this.input.toolCallChecker);\n    }\n\n    const checker = new ToolCallChecker(config);\n    checker.enabled = this.input.toolCallChecker !== false;\n    return checker;\n  }\n\n  get meta(): AgentMeta {\n    const tools = this.input.tools || [];\n\n    return {\n      name: this.input.name ?? \"Requirement\",\n      tools,\n      description:\n        this.input.description ??\n        \"The RequirementAgent is a declarative AI agent implementation that provides predictable, \" +\n          \"controlled execution behavior across different language models through rule-based constraints.\",\n      ...(tools.length > 0 && {\n        extraDescription: [\n          `Tools that I can use to accomplish given task.`,\n          ...tools.map((tool) => `Tool '${tool.name}': ${tool.description}.`),\n        ].join(\"\\n\"),\n      }),\n    };\n  }\n\n  protected get templates(): RequirementAgentTemplates {\n    const overrides = this.input.templates || {};\n    const finalized = mapObj({\n      system: RequirementAgentSystemPrompt,\n      task: RequirementAgentTaskPrompt,\n      toolError: RequirementAgentToolErrorPrompt,\n      toolNoResult: RequirementAgentToolNoResultPrompt,\n    } as RequirementAgentTemplates)((\n      key,\n      defaultTemplate: RequirementAgentTemplates[typeof key],\n    ) => {\n      const override = overrides[key] ?? defaultTemplate;\n      if (override instanceof PromptTemplate) {\n        return override;\n      }\n      return override(defaultTemplate) ?? defaultTemplate;\n    });\n\n    if (this.input.role || this.input.instructions || this.input.notes) {\n      finalized.system.update({\n        defaults: pickBy(\n          {\n            role: this.input.role,\n            instructions: isEmptyish(this.input.instructions)\n              ? undefined\n              : castArray(this.input.instructions)\n                  .map((v) => `- ${v}`)\n                  .join(\"\\n\"),\n            notes: isEmptyish(this.input.notes)\n              ? undefined\n              : castArray(this.input.notes)\n                  .map((v) => `- ${v}`)\n                  .join(\"\\n\"),\n          },\n          isDefined,\n        ),\n      });\n    }\n\n    return finalized;\n  }\n\n  createSnapshot() {\n    return {\n      ...super.createSnapshot(),\n      input: shallowCopy(this.input),\n      emitter: this.emitter,\n      runner: this.runner,\n    };\n  }\n\n  set memory(memory: BaseMemory) {\n    this.input.memory = memory;\n  }\n\n  get memory(): BaseMemory {\n    return this.input.memory!;\n  }\n}\n"]}

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

@@ -33,6 +33,7 @@ declare const RequirementAgentSystemPrompt: PromptTemplate<zod.ZodObject<{
         reason?: string | undefined;
     }>, "many">;
 }, "strip", zod.ZodTypeAny, {
+    finalAnswerName: string;
     tools: {
         name: string;
         description: string;
@@ -40,13 +41,13 @@ declare const RequirementAgentSystemPrompt: PromptTemplate<zod.ZodObject<{
         allowed: string;
         reason?: string | undefined;
     }[];
-    finalAnswerName: string;
     role?: string | undefined;
     instructions?: string | undefined;
     notes?: string | undefined;
     finalAnswerSchema?: string | undefined;
     finalAnswerInstructions?: string | undefined;
 }, {
+    finalAnswerName: string;
     tools: {
         name: string;
         description: string;
@@ -54,7 +55,6 @@ declare const RequirementAgentSystemPrompt: PromptTemplate<zod.ZodObject<{
         allowed: string;
         reason?: string | undefined;
     }[];
-    finalAnswerName: string;
     role?: string | undefined;
     instructions?: string | undefined;
     notes?: string | undefined;

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

@@ -33,6 +33,7 @@ declare const RequirementAgentSystemPrompt: PromptTemplate<zod.ZodObject<{
         reason?: string | undefined;
     }>, "many">;
 }, "strip", zod.ZodTypeAny, {
+    finalAnswerName: string;
     tools: {
         name: string;
         description: string;
@@ -40,13 +41,13 @@ declare const RequirementAgentSystemPrompt: PromptTemplate<zod.ZodObject<{
         allowed: string;
         reason?: string | undefined;
     }[];
-    finalAnswerName: string;
     role?: string | undefined;
     instructions?: string | undefined;
     notes?: string | undefined;
     finalAnswerSchema?: string | undefined;
     finalAnswerInstructions?: string | undefined;
 }, {
+    finalAnswerName: string;
     tools: {
         name: string;
         description: string;
@@ -54,7 +55,6 @@ declare const RequirementAgentSystemPrompt: PromptTemplate<zod.ZodObject<{
         allowed: string;
         reason?: string | undefined;
     }[];
-    finalAnswerName: string;
     role?: string | undefined;
     instructions?: string | undefined;
     notes?: string | undefined;

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

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

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

@@ -1,9 +1,9 @@
 import { Requirement, Rule } from './requirement.cjs';
-import { MultiTargetType, TargetType } from './utils.cjs';
+import { TargetType, MultiTargetType } from './utils.cjs';
 import { AnyTool } from '../../../tools/base.cjs';
 import { RunContext } from '../../../context.cjs';
-import { R as RequirementAgentRunState } from '../../../types-YoNQSxQH.cjs';
-import '../../../emitter-UkIkNrek.cjs';
+import { R as RequirementAgentRunState } from '../../../types-BNsa3W7Z.cjs';
+import '../../../emitter-D4OcelZ9.cjs';
 import '../../../internals/types.cjs';
 import '../../../internals/helpers/guards.cjs';
 import '../../../internals/serializable.cjs';
@@ -19,11 +19,11 @@ import '../../../memory/base.cjs';
 import '../../../backend/message.cjs';
 import 'ai';
 import '../../../template.cjs';
-import '../../../chat-DaqV3woF.cjs';
+import '../../../chat-C20nYj46.cjs';
 import '../../../backend/constants.cjs';
 import '../../../logger/logger.cjs';
 import 'pino';
-import '../../../base-BqzN2QwB.cjs';
+import '../../../base-s3ioVIE9.cjs';
 
 /**
  * Copyright 2025 © BeeAI a Series of LF Projects, LLC

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

@@ -1,9 +1,9 @@
 import { Requirement, Rule } from './requirement.js';
-import { MultiTargetType, TargetType } from './utils.js';
+import { TargetType, MultiTargetType } from './utils.js';
 import { AnyTool } from '../../../tools/base.js';
 import { RunContext } from '../../../context.js';
-import { R as RequirementAgentRunState } from '../../../types-YT-WdK3m.js';
-import '../../../emitter-CL56qab4.js';
+import { R as RequirementAgentRunState } from '../../../types-DYtzF6ya.js';
+import '../../../emitter-kHxkUxco.js';
 import '../../../internals/types.js';
 import '../../../internals/helpers/guards.js';
 import '../../../internals/serializable.js';
@@ -19,11 +19,11 @@ import '../../../memory/base.js';
 import '../../../backend/message.js';
 import 'ai';
 import '../../../template.js';
-import '../../../chat-DTxmaKP0.js';
+import '../../../chat-CfTNfR60.js';
 import '../../../backend/constants.js';
 import '../../../logger/logger.js';
 import 'pino';
-import '../../../base-BfhNzUYf.js';
+import '../../../base-BjzyQ8k6.js';
 
 /**
  * Copyright 2025 © BeeAI a Series of LF Projects, LLC

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

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

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

@@ -1 +1 @@
-{"version":3,"sources":["../../../../src/agents/requirement/requirements/requirement.ts"],"names":["Requirement","name","state","enabled","middlewares","_priority","priority","value","ValueError","emitter","Emitter","root","child","namespace","toCamelCase","creator","run","RunContext","enter","signal","undefined","params","context","_run","init","tools","_","emit","clone","instance","Object","create","getPrototypeOf","enumerable","RequirementError","FrameworkError","message","requirement","cause","isFatal","isRetryable"],"mappings":";;;;;;;;;;;;;;;;;;;;;AAwBO,MAAeA,WAAAA,CAAAA;AAAAA,EAAAA;;;AACbC,EAAAA,IAAAA;AACAC,EAAAA,KAAAA,GAA6B,EAAC;EAC9BC,OAAAA,GAAU,IAAA;AACVC,EAAAA,WAAAA,GAA6C,EAAA;EAE1CC,SAAAA,GAAY,EAAA;AAEtB,EAAA,WAAA,CAAYJ,IAAAA,EAAe;AACzB,IAAA,IAAA,CAAKA,IAAAA,GAAOA,IAAAA,IAAQ,IAAA,CAAK,WAAA,CAAYA,IAAAA;AACvC;AAEA,EAAA,IAAIK,QAAAA,GAAmB;AACrB,IAAA,OAAO,IAAA,CAAKD,SAAAA;AACd;AAEA,EAAA,IAAIC,SAASC,KAAAA,EAAe;AAC1B,IAAA,IAAIA,SAAS,CAAA,EAAG;AACd,MAAA,MAAM,IAAIC,sBAAW,sCAAA,CAAA;AACvB;AACA,IAAA,IAAA,CAAKH,SAAAA,GAAYE,KAAAA;AACnB;AAEA,EAAA,IACIE,OAAAA,GAAyC;AAC3C,IAAA,OAAOC,mBAAAA,CAAQC,KAAKC,KAAAA,CAAM;MACxBC,SAAAA,EAAW;AAAC,QAAA,aAAA;AAAeC,QAAAA,kBAAAA,CAAY,KAAKb,IAAI;;MAChDc,OAAAA,EAAS;KACX,CAAA;AACF;AAEAC,EAAAA,GAAAA,CAAId,KAAAA,EAAiC;AACnC,IAAA,OAAOe,sBAAAA,CAAWC,MAChB,IAAA,EACA;MAAEC,MAAAA,EAAQC,MAAAA;MAAWC,MAAAA,EAAQ;AAACnB,QAAAA;;AAAgB,KAAA,EAC9C,OAAOoB,OAAAA,KAAAA;AACL,MAAA,OAAO,MAAM,IAAA,CAAKC,IAAAA,CAAKrB,KAAAA,EAAOoB,OAAAA,CAAAA;KAChC,CAAA;AAEJ;EAIA,MAAME,IAAAA,CAAKC,OAAkBC,CAAAA,EAAmC;AAC9D,IAAA,MAAM,IAAA,CAAKjB,OAAAA,CAAQkB,IAAAA,CAAK,MAAA,EAAQ;AAAEF,MAAAA;KAAM,CAAA;AAC1C;AAEA,EAAA,MAAMG,KAAAA,GAAuB;AAC3B,IAAA,MAAMC,WAAWC,MAAAA,CAAOC,MAAAA,CAAOD,MAAAA,CAAOE,cAAAA,CAAe,IAAI,CAAA,CAAA;AACzDH,IAAAA,QAAAA,CAAS5B,OAAO,IAAA,CAAKA,IAAAA;AACrB4B,IAAAA,QAAAA,CAASvB,WAAW,IAAA,CAAKA,QAAAA;AACzBuB,IAAAA,QAAAA,CAAS1B,UAAU,IAAA,CAAKA,OAAAA;AACxB0B,IAAAA,QAAAA,CAAS3B,KAAAA,GAAQ;AAAE,MAAA,GAAG,IAAA,CAAKA;AAAM,KAAA;AACjC2B,IAAAA,QAAAA,CAASzB,WAAAA,GAAc;SAAI,IAAA,CAAKA;;AAChC,IAAA,OAAOyB,QAAAA;AACT;AACF;;;IAjCWI,UAAAA,EAAY;;;;;AAoChB,MAAMC,yBAAyBC,yBAAAA,CAAAA;AAAAA,EAAAA;;;;EACpC,WAAA,CACEC,OAAAA,EACgBC,WAAAA,EAChBC,KAAAA,EACAhB,OAAAA,EACA;AACA,IAAA,KAAA,CAAMc,SAASE,KAAAA,GAAQ;AAACA,MAAAA;QAASlB,MAAAA,EAAW;MAC1CmB,OAAAA,EAAS,IAAA;MACTC,WAAAA,EAAa,KAAA;AACblB,MAAAA;KACF,CAAA,EAAA,KARgBe,WAAAA,GAAAA,WAAAA;AASlB;AACF","file":"requirement.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { AnyTool } from \"@/tools/base.js\";\nimport { MiddlewareType, RunContext } from \"@/context.js\";\nimport { Callback, Emitter } from \"@/emitter/emitter.js\";\nimport { FrameworkError, ValueError } from \"@/errors.js\";\nimport { toCamelCase } from \"remeda\";\nimport type { RequirementAgentRunState } from \"@/agents/requirement/types.js\";\nimport { Cache } from \"@/cache/decoratorCache.js\";\n\n// Rule definition\nexport interface Rule {\n  target: string;\n  allowed: boolean;\n  reason?: string;\n  preventStop: boolean;\n  forced: boolean;\n  hidden: boolean;\n}\n\n// Base Requirement class\nexport abstract class Requirement {\n  public name: string;\n  public state: Record<string, any> = {};\n  public enabled = true;\n  public middlewares: MiddlewareType<typeof this>[] = [];\n\n  protected _priority = 10;\n\n  constructor(name?: string) {\n    this.name = name || this.constructor.name;\n  }\n\n  get priority(): number {\n    return this._priority;\n  }\n\n  set priority(value: number) {\n    if (value <= 0) {\n      throw new ValueError(\"Priority must be a positive integer.\");\n    }\n    this._priority = value;\n  }\n\n  @Cache({ enumerable: false })\n  get emitter(): Emitter<RequirementCallbacks> {\n    return Emitter.root.child({\n      namespace: [\"requirement\", toCamelCase(this.name)],\n      creator: this,\n    });\n  }\n\n  run(state: RequirementAgentRunState) {\n    return RunContext.enter(\n      this,\n      { signal: undefined, params: [state] as const },\n      async (context) => {\n        return await this._run(state, context);\n      },\n    );\n  }\n\n  abstract _run(state: RequirementAgentRunState, _: RunContext<typeof this>): Promise<Rule[]>;\n\n  async init(tools: AnyTool[], _: RunContext<any>): Promise<void> {\n    await this.emitter.emit(\"init\", { tools });\n  }\n\n  async clone(): Promise<this> {\n    const instance = Object.create(Object.getPrototypeOf(this));\n    instance.name = this.name;\n    instance.priority = this.priority;\n    instance.enabled = this.enabled;\n    instance.state = { ...this.state };\n    instance.middlewares = [...this.middlewares];\n    return instance;\n  }\n}\n\n// Requirement error\nexport class RequirementError extends FrameworkError {\n  constructor(\n    message: string,\n    public readonly requirement?: Requirement,\n    cause?: Error,\n    context?: Record<string, any>,\n  ) {\n    super(message, cause ? [cause] : undefined, {\n      isFatal: true,\n      isRetryable: false,\n      context,\n    });\n  }\n}\n\nexport interface RequirementCallbacks {\n  init: Callback<{ tools: AnyTool[] }>;\n}\n"]}
+{"version":3,"sources":["../../../../src/agents/requirement/requirements/requirement.ts"],"names":["Requirement","name","state","enabled","middlewares","_priority","priority","value","ValueError","emitter","Emitter","root","child","namespace","toCamelCase","creator","run","RunContext","enter","signal","undefined","params","context","_run","init","tools","_","emit","clone","instance","Object","create","getPrototypeOf","enumerable","RequirementError","FrameworkError","message","requirement","cause","isFatal","isRetryable"],"mappings":";;;;;;;;;;;;;;;;;;;;;AAwBO,MAAeA,WAAAA,CAAAA;AAAAA,EAAAA;;;AACbC,EAAAA,IAAAA;AACAC,EAAAA,KAAAA,GAA6B,EAAC;EAC9BC,OAAAA,GAAU,IAAA;AACVC,EAAAA,WAAAA,GAA6C,EAAA;EAE1CC,SAAAA,GAAY,EAAA;AAEtB,EAAA,WAAA,CAAYJ,IAAAA,EAAe;AACzB,IAAA,IAAA,CAAKA,IAAAA,GAAOA,IAAAA,IAAQ,IAAA,CAAK,WAAA,CAAYA,IAAAA;AACvC,EAAA;AAEA,EAAA,IAAIK,QAAAA,GAAmB;AACrB,IAAA,OAAO,IAAA,CAAKD,SAAAA;AACd,EAAA;AAEA,EAAA,IAAIC,SAASC,KAAAA,EAAe;AAC1B,IAAA,IAAIA,SAAS,CAAA,EAAG;AACd,MAAA,MAAM,IAAIC,sBAAW,sCAAA,CAAA;AACvB,IAAA;AACA,IAAA,IAAA,CAAKH,SAAAA,GAAYE,KAAAA;AACnB,EAAA;AAEA,EAAA,IACIE,OAAAA,GAAyC;AAC3C,IAAA,OAAOC,mBAAAA,CAAQC,KAAKC,KAAAA,CAAM;MACxBC,SAAAA,EAAW;AAAC,QAAA,aAAA;AAAeC,QAAAA,kBAAAA,CAAY,KAAKb,IAAI;;MAChDc,OAAAA,EAAS;KACX,CAAA;AACF,EAAA;AAEAC,EAAAA,GAAAA,CAAId,KAAAA,EAAiC;AACnC,IAAA,OAAOe,sBAAAA,CAAWC,MAChB,IAAA,EACA;MAAEC,MAAAA,EAAQC,MAAAA;MAAWC,MAAAA,EAAQ;AAACnB,QAAAA;;AAAgB,KAAA,EAC9C,OAAOoB,OAAAA,KAAAA;AACL,MAAA,OAAO,MAAM,IAAA,CAAKC,IAAAA,CAAKrB,KAAAA,EAAOoB,OAAAA,CAAAA;IAChC,CAAA,CAAA;AAEJ,EAAA;EAIA,MAAME,IAAAA,CAAKC,OAAkBC,CAAAA,EAAmC;AAC9D,IAAA,MAAM,IAAA,CAAKjB,OAAAA,CAAQkB,IAAAA,CAAK,MAAA,EAAQ;AAAEF,MAAAA;KAAM,CAAA;AAC1C,EAAA;AAEA,EAAA,MAAMG,KAAAA,GAAuB;AAC3B,IAAA,MAAMC,WAAWC,MAAAA,CAAOC,MAAAA,CAAOD,MAAAA,CAAOE,cAAAA,CAAe,IAAI,CAAA,CAAA;AACzDH,IAAAA,QAAAA,CAAS5B,OAAO,IAAA,CAAKA,IAAAA;AACrB4B,IAAAA,QAAAA,CAASvB,WAAW,IAAA,CAAKA,QAAAA;AACzBuB,IAAAA,QAAAA,CAAS1B,UAAU,IAAA,CAAKA,OAAAA;AACxB0B,IAAAA,QAAAA,CAAS3B,KAAAA,GAAQ;AAAE,MAAA,GAAG,IAAA,CAAKA;AAAM,KAAA;AACjC2B,IAAAA,QAAAA,CAASzB,WAAAA,GAAc;SAAI,IAAA,CAAKA;;AAChC,IAAA,OAAOyB,QAAAA;AACT,EAAA;AACF;;;IAjCWI,UAAAA,EAAY;;;;;AAoChB,MAAMC,yBAAyBC,yBAAAA,CAAAA;AAAAA,EAAAA;;;;EACpC,WAAA,CACEC,OAAAA,EACgBC,WAAAA,EAChBC,KAAAA,EACAhB,OAAAA,EACA;AACA,IAAA,KAAA,CAAMc,SAASE,KAAAA,GAAQ;AAACA,MAAAA;QAASlB,MAAAA,EAAW;MAC1CmB,OAAAA,EAAS,IAAA;MACTC,WAAAA,EAAa,KAAA;AACblB,MAAAA;KACF,CAAA,EAAA,KARgBe,WAAAA,GAAAA,WAAAA;AASlB,EAAA;AACF","file":"requirement.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { AnyTool } from \"@/tools/base.js\";\nimport { MiddlewareType, RunContext } from \"@/context.js\";\nimport { Callback, Emitter } from \"@/emitter/emitter.js\";\nimport { FrameworkError, ValueError } from \"@/errors.js\";\nimport { toCamelCase } from \"remeda\";\nimport type { RequirementAgentRunState } from \"@/agents/requirement/types.js\";\nimport { Cache } from \"@/cache/decoratorCache.js\";\n\n// Rule definition\nexport interface Rule {\n  target: string;\n  allowed: boolean;\n  reason?: string;\n  preventStop: boolean;\n  forced: boolean;\n  hidden: boolean;\n}\n\n// Base Requirement class\nexport abstract class Requirement {\n  public name: string;\n  public state: Record<string, any> = {};\n  public enabled = true;\n  public middlewares: MiddlewareType<typeof this>[] = [];\n\n  protected _priority = 10;\n\n  constructor(name?: string) {\n    this.name = name || this.constructor.name;\n  }\n\n  get priority(): number {\n    return this._priority;\n  }\n\n  set priority(value: number) {\n    if (value <= 0) {\n      throw new ValueError(\"Priority must be a positive integer.\");\n    }\n    this._priority = value;\n  }\n\n  @Cache({ enumerable: false })\n  get emitter(): Emitter<RequirementCallbacks> {\n    return Emitter.root.child({\n      namespace: [\"requirement\", toCamelCase(this.name)],\n      creator: this,\n    });\n  }\n\n  run(state: RequirementAgentRunState) {\n    return RunContext.enter(\n      this,\n      { signal: undefined, params: [state] as const },\n      async (context) => {\n        return await this._run(state, context);\n      },\n    );\n  }\n\n  abstract _run(state: RequirementAgentRunState, _: RunContext<typeof this>): Promise<Rule[]>;\n\n  async init(tools: AnyTool[], _: RunContext<any>): Promise<void> {\n    await this.emitter.emit(\"init\", { tools });\n  }\n\n  async clone(): Promise<this> {\n    const instance = Object.create(Object.getPrototypeOf(this));\n    instance.name = this.name;\n    instance.priority = this.priority;\n    instance.enabled = this.enabled;\n    instance.state = { ...this.state };\n    instance.middlewares = [...this.middlewares];\n    return instance;\n  }\n}\n\n// Requirement error\nexport class RequirementError extends FrameworkError {\n  constructor(\n    message: string,\n    public readonly requirement?: Requirement,\n    cause?: Error,\n    context?: Record<string, any>,\n  ) {\n    super(message, cause ? [cause] : undefined, {\n      isFatal: true,\n      isRetryable: false,\n      context,\n    });\n  }\n}\n\nexport interface RequirementCallbacks {\n  init: Callback<{ tools: AnyTool[] }>;\n}\n"]}

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

@@ -1,8 +1,8 @@
 import { MiddlewareType, Run, RunContext } from '../../../context.cjs';
 import { AnyTool } from '../../../tools/base.cjs';
-import { E as Emitter, C as Callback } from '../../../emitter-UkIkNrek.cjs';
+import { E as Emitter, C as Callback } from '../../../emitter-D4OcelZ9.cjs';
 import { FrameworkError } from '../../../errors.cjs';
-import { R as RequirementAgentRunState } from '../../../types-YoNQSxQH.cjs';
+import { R as RequirementAgentRunState } from '../../../types-BNsa3W7Z.cjs';
 import '../../../internals/serializable.cjs';
 import '../../../internals/types.cjs';
 import '../../../internals/helpers/guards.cjs';
@@ -17,11 +17,11 @@ import '../../../memory/base.cjs';
 import '../../../backend/message.cjs';
 import 'ai';
 import '../../../template.cjs';
-import '../../../chat-DaqV3woF.cjs';
+import '../../../chat-C20nYj46.cjs';
 import '../../../backend/constants.cjs';
 import '../../../logger/logger.cjs';
 import 'pino';
-import '../../../base-BqzN2QwB.cjs';
+import '../../../base-s3ioVIE9.cjs';
 
 interface Rule {
     target: string;

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

@@ -1,8 +1,8 @@
 import { MiddlewareType, Run, RunContext } from '../../../context.js';
 import { AnyTool } from '../../../tools/base.js';
-import { E as Emitter, C as Callback } from '../../../emitter-CL56qab4.js';
+import { E as Emitter, C as Callback } from '../../../emitter-kHxkUxco.js';
 import { FrameworkError } from '../../../errors.js';
-import { R as RequirementAgentRunState } from '../../../types-YT-WdK3m.js';
+import { R as RequirementAgentRunState } from '../../../types-DYtzF6ya.js';
 import '../../../internals/serializable.js';
 import '../../../internals/types.js';
 import '../../../internals/helpers/guards.js';
@@ -17,11 +17,11 @@ import '../../../memory/base.js';
 import '../../../backend/message.js';
 import 'ai';
 import '../../../template.js';
-import '../../../chat-DTxmaKP0.js';
+import '../../../chat-CfTNfR60.js';
 import '../../../backend/constants.js';
 import '../../../logger/logger.js';
 import 'pino';
-import '../../../base-BfhNzUYf.js';
+import '../../../base-BjzyQ8k6.js';
 
 interface Rule {
     target: string;