beeai-framework 0.1.21 → 0.1.22

package/dist/agents/requirement/runner.js

@@ -0,0 +1,250 @@
+import { v4 } from 'uuid';
+import { AgentError } from '../base.js';
+import { AssistantMessage, ToolMessage } from '../../backend/message.js';
+import { UnconstrainedMemory } from '../../memory/unconstrainedMemory.js';
+import { RetryCounter } from '../../internals/helpers/counter.js';
+import { RequirementsReasoner, createSystemMessage } from './utils/llm.js';
+import { FinalAnswerTool, runTool } from './utils/tool.js';
+import { parseBrokenJson } from '../../internals/helpers/schema.js';
+import { mergeTokenUsage } from '../../adapters/vercel/backend/utils.js';
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+const TEMP_MESSAGE_KEY = "tempMessage";
+class RequirementAgentRunner {
+  static {
+    __name(this, "RequirementAgentRunner");
+  }
+  llm;
+  runConfig;
+  requirements;
+  toolCallCycleChecker;
+  ctx;
+  forceFinalAnswerAsTool;
+  templates;
+  state;
+  reasoner;
+  iterationErrorCounter;
+  globalErrorCounter;
+  constructor(llm, runConfig, tools, requirements, expectedOutput, toolCallCycleChecker, ctx, forceFinalAnswerAsTool, templates) {
+    this.llm = llm;
+    this.runConfig = runConfig;
+    this.requirements = requirements;
+    this.toolCallCycleChecker = toolCallCycleChecker;
+    this.ctx = ctx;
+    this.forceFinalAnswerAsTool = forceFinalAnswerAsTool;
+    this.templates = templates;
+    this.state = {
+      answer: null,
+      result: null,
+      memory: new UnconstrainedMemory(),
+      steps: [],
+      iteration: 0,
+      usage: {
+        totalTokens: 0,
+        promptTokens: 0,
+        completionTokens: 0,
+        reasoningTokens: 0,
+        cachedPromptTokens: void 0
+      }
+    };
+    this.requirements = requirements;
+    const finalAnswer = new FinalAnswerTool(expectedOutput, this.state);
+    this.reasoner = new RequirementsReasoner(tools, finalAnswer, ctx);
+    const maxRetriesPerIteration = runConfig.maxRetriesPerStep ?? 0;
+    this.iterationErrorCounter = new RetryCounter(maxRetriesPerIteration, AgentError);
+    const maxRetries = Math.max(maxRetriesPerIteration, runConfig.totalMaxRetries ?? 0);
+    this.globalErrorCounter = new RetryCounter(maxRetries, AgentError);
+  }
+  incrementIteration() {
+    this.state.iteration++;
+    if (this.runConfig.maxIterations && this.state.iteration > this.runConfig.maxIterations) {
+      throw new AgentError(`Agent was not able to resolve the task in ${this.state.iteration} iterations.`);
+    }
+  }
+  async runLLM(request) {
+    const { messages, options } = await this.prepareLLMRequest(request);
+    const response = await this.llm.create({
+      ...options,
+      messages
+    });
+    if (response.usage) {
+      mergeTokenUsage(this.state.usage, response.usage);
+    }
+    return response;
+  }
+  async prepareLLMRequest(request) {
+    const messages = [
+      await createSystemMessage(this.templates.system, request),
+      ...this.state.memory.messages
+    ];
+    const options = {
+      maxRetries: this.runConfig.maxRetriesPerStep,
+      tools: request.allowedTools,
+      toolChoice: request.toolChoice,
+      stream: false
+    };
+    return {
+      messages,
+      options
+    };
+  }
+  async createFinalAnswerToolCall(fullText) {
+    const parsed = parseBrokenJson(fullText, {
+      pair: [
+        "{",
+        "}"
+      ]
+    });
+    if (!parsed) {
+      if (!this.reasoner.finalAnswer.customSchema) {
+        return new AssistantMessage([
+          {
+            type: "tool-call",
+            toolCallId: `call_${v4().substring(0, 8)}`,
+            toolName: this.reasoner.finalAnswer.name,
+            input: this.reasoner.finalAnswer.inputSchema().parse({
+              response: fullText
+            })
+          }
+        ]);
+      }
+      return null;
+    }
+    return new AssistantMessage([
+      {
+        type: "tool-call",
+        toolCallId: `call_${v4().substring(0, 8)}`,
+        toolName: this.reasoner.finalAnswer.name,
+        input: parsed
+      }
+    ]);
+  }
+  async createRequest(extraRules = []) {
+    return await this.reasoner.createRequest(this.state, this.forceFinalAnswerAsTool, extraRules);
+  }
+  async invokeToolCalls(tools, toolCalls) {
+    const toolResults = [];
+    const results = await Promise.all(toolCalls.map((msg) => runTool(tools, msg, {
+      state: this.state
+    })));
+    for (const toolCall of results) {
+      this.state.steps.push({
+        id: v4(),
+        iteration: this.state.iteration,
+        input: toolCall.input,
+        output: toolCall.output,
+        tool: toolCall.tool,
+        error: toolCall.error
+      });
+      let result;
+      if (toolCall.error) {
+        result = this.templates.toolError.render({
+          reason: toolCall.error.explain()
+        });
+      } else {
+        result = !toolCall.output.isEmpty() ? toolCall.output.getTextContent() : this.templates.toolNoResult.render({
+          toolCall: {
+            tool: toolCall.tool,
+            input: toolCall.input
+          }
+        });
+      }
+      toolResults.push(new ToolMessage({
+        type: "tool-result",
+        toolName: toolCall.tool?.name || toolCall.msg.toolName,
+        toolCallId: toolCall.msg.toolCallId,
+        output: {
+          type: "text",
+          value: result
+        }
+      }));
+      if (toolCall.error) {
+        this.iterationErrorCounter.use(toolCall.error);
+        this.globalErrorCounter.use(toolCall.error);
+      }
+    }
+    return toolResults;
+  }
+  async addMessages(messages) {
+    await this.state.memory.addMany(messages);
+  }
+  async run() {
+    if (this.state.answer) {
+      return this.state;
+    }
+    await this.reasoner.update(this.requirements);
+    while (!this.state.answer) {
+      this.incrementIteration();
+      const request = await this.createRequest();
+      await this.ctx.emitter.emit("start", {
+        state: this.state,
+        request
+      });
+      const response = await this.runIteration(request);
+      await this.ctx.emitter.emit("success", {
+        state: this.state,
+        response
+      });
+    }
+    return this.state;
+  }
+  async runIteration(request) {
+    const response = await this.runLLM(request);
+    const toolCalls = response.getToolCalls();
+    if (toolCalls.length === 0) {
+      const textMessages = response.getTextMessages();
+      const text = textMessages.map((m) => m.text).join("\n");
+      if (!text || request.canStop) {
+        throw new AgentError("Model produced an empty response.");
+      }
+      const finalAnswerToolCall = await this.createFinalAnswerToolCall(text);
+      if (!finalAnswerToolCall) {
+        const err = new AgentError("Model produced an invalid final answer tool call.");
+        this.iterationErrorCounter.use(err);
+        this.globalErrorCounter.use(err);
+        await this.reasoner.update([]);
+        const updatedRequest = await this.createRequest([
+          {
+            target: this.reasoner.finalAnswer.name,
+            allowed: true,
+            hidden: false,
+            forced: false,
+            preventStop: false
+          }
+        ]);
+        this.forceFinalAnswerAsTool = true;
+        return await this.runIteration(updatedRequest);
+      }
+      await this.state.memory.add(finalAnswerToolCall);
+      toolCalls.push(...finalAnswerToolCall.getToolCalls());
+    } else {
+      await this.state.memory.addMany(response.messages);
+    }
+    for (const toolCallMsg of toolCalls) {
+      this.toolCallCycleChecker.register(toolCallMsg);
+      if (this.toolCallCycleChecker.cycleFound) {
+        this.toolCallCycleChecker.reset();
+        const updatedRequest = await this.createRequest([
+          {
+            target: toolCallMsg.toolName,
+            allowed: false,
+            hidden: false,
+            forced: true,
+            preventStop: false
+          }
+        ]);
+        return await this.runIteration(updatedRequest);
+      }
+    }
+    const toolResults = await this.invokeToolCalls(request.allowedTools, toolCalls);
+    await this.state.memory.addMany(toolResults);
+    const tempMessages = this.state.memory.messages.filter((msg) => msg.meta[TEMP_MESSAGE_KEY]);
+    await this.state.memory.deleteMany(tempMessages);
+    return response;
+  }
+}
+
+export { RequirementAgentRunner };
+//# sourceMappingURL=runner.js.map
+//# sourceMappingURL=runner.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/agents/requirement/runner.ts"],"names":["TEMP_MESSAGE_KEY","RequirementAgentRunner","state","reasoner","iterationErrorCounter","globalErrorCounter","llm","runConfig","tools","requirements","expectedOutput","toolCallCycleChecker","ctx","forceFinalAnswerAsTool","templates","answer","result","memory","UnconstrainedMemory","steps","iteration","usage","totalTokens","promptTokens","completionTokens","reasoningTokens","cachedPromptTokens","undefined","finalAnswer","FinalAnswerTool","RequirementsReasoner","maxRetriesPerIteration","maxRetriesPerStep","RetryCounter","AgentError","maxRetries","Math","max","totalMaxRetries","incrementIteration","maxIterations","runLLM","request","messages","options","prepareLLMRequest","response","create","mergeTokenUsage","createSystemMessage","system","allowedTools","toolChoice","stream","createFinalAnswerToolCall","fullText","parsed","parseBrokenJson","pair","customSchema","AssistantMessage","type","toolCallId","uuidv4","substring","toolName","name","input","inputSchema","parse","createRequest","extraRules","invokeToolCalls","toolCalls","toolResults","results","Promise","all","map","msg","runTool","toolCall","push","id","output","tool","error","toolError","render","reason","explain","isEmpty","getTextContent","toolNoResult","ToolMessage","value","use","addMessages","addMany","run","update","emitter","emit","runIteration","getToolCalls","length","textMessages","getTextMessages","text","m","join","canStop","finalAnswerToolCall","err","updatedRequest","target","allowed","hidden","forced","preventStop","add","toolCallMsg","register","cycleFound","reset","tempMessages","filter","meta","deleteMany"],"mappings":";;;;;;;;;;;;AA6BA,MAAMA,gBAAAA,GAAmB,aAAA;AAKlB,MAAMC,sBAAAA,CAAAA;EAlCb;;;;;;;;;;AAmCqBC,EAAAA,KAAAA;AACAC,EAAAA,QAAAA;AACAC,EAAAA,qBAAAA;AACAC,EAAAA,kBAAAA;EAEnB,WAAA,CACqBC,GAAAA,EACAC,WACnBC,KAAAA,EACmBC,YAAAA,EACnBC,gBACmBC,oBAAAA,EACAC,GAAAA,EACTC,wBACSC,SAAAA,EACnB;SATmBR,GAAAA,GAAAA,GAAAA;SACAC,SAAAA,GAAAA,SAAAA;SAEAE,YAAAA,GAAAA,YAAAA;SAEAE,oBAAAA,GAAAA,oBAAAA;SACAC,GAAAA,GAAAA,GAAAA;SACTC,sBAAAA,GAAAA,sBAAAA;SACSC,SAAAA,GAAAA,SAAAA;AAEnB,IAAA,IAAA,CAAKZ,KAAAA,GAAQ;MACXa,MAAAA,EAAQ,IAAA;MACRC,MAAAA,EAAQ,IAAA;AACRC,MAAAA,MAAAA,EAAQ,IAAIC,mBAAAA,EAAAA;AACZC,MAAAA,KAAAA,EAAO,EAAA;MACPC,SAAAA,EAAW,CAAA;MACXC,KAAAA,EAAO;QACLC,WAAAA,EAAa,CAAA;QACbC,YAAAA,EAAc,CAAA;QACdC,gBAAAA,EAAkB,CAAA;QAClBC,eAAAA,EAAiB,CAAA;QACjBC,kBAAAA,EAAoBC;AACtB;AACF,KAAA;AAEA,IAAA,IAAA,CAAKlB,YAAAA,GAAeA,YAAAA;AACpB,IAAA,MAAMmB,WAAAA,GAAc,IAAIC,eAAAA,CAAgBnB,cAAAA,EAAgB,KAAKR,KAAK,CAAA;AAClE,IAAA,IAAA,CAAKC,QAAAA,GAAW,IAAI2B,oBAAAA,CAAqBtB,KAAAA,EAAOoB,aAAahB,GAAAA,CAAAA;AAE7D,IAAA,MAAMmB,sBAAAA,GAAyBxB,UAAUyB,iBAAAA,IAAqB,CAAA;AAC9D,IAAA,IAAA,CAAK5B,qBAAAA,GAAwB,IAAI6B,YAAAA,CAAaF,sBAAAA,EAAwBG,UAAAA,CAAAA;AAEtE,IAAA,MAAMC,aAAaC,IAAAA,CAAKC,GAAAA,CAAIN,sBAAAA,EAAwBxB,SAAAA,CAAU+B,mBAAmB,CAAA,CAAA;AACjF,IAAA,IAAA,CAAKjC,kBAAAA,GAAqB,IAAI4B,YAAAA,CAAaE,UAAAA,EAAYD,UAAAA,CAAAA;AACzD;EAEUK,kBAAAA,GAA2B;AACnC,IAAA,IAAA,CAAKrC,KAAAA,CAAMkB,SAAAA,EAAAA;AAEX,IAAA,IAAI,IAAA,CAAKb,UAAUiC,aAAAA,IAAiB,IAAA,CAAKtC,MAAMkB,SAAAA,GAAY,IAAA,CAAKb,UAAUiC,aAAAA,EAAe;AACvF,MAAA,MAAM,IAAIN,UAAAA,CACR,CAAA,0CAAA,EAA6C,IAAA,CAAKhC,KAAAA,CAAMkB,SAAS,CAAA,YAAA,CAAc,CAAA;AAEnF;AACF;AAEA,EAAA,MAAgBqB,OAAOC,OAAAA,EAA4D;AACjF,IAAA,MAAM,EAAEC,QAAAA,EAAUC,OAAAA,KAAY,MAAM,IAAA,CAAKC,kBAAkBH,OAAAA,CAAAA;AAC3D,IAAA,MAAMI,QAAAA,GAAW,MAAM,IAAA,CAAKxC,GAAAA,CAAIyC,MAAAA,CAAO;MAAE,GAAGH,OAAAA;AAASD,MAAAA;KAAS,CAAA;AAE9D,IAAA,IAAIG,SAASzB,KAAAA,EAAO;AAClB2B,MAAAA,eAAAA,CAAgB,IAAA,CAAK9C,KAAAA,CAAMmB,KAAAA,EAAOyB,QAAAA,CAASzB,KAAK,CAAA;AAClD;AAEA,IAAA,OAAOyB,QAAAA;AACT;AAEA,EAAA,MAAgBD,kBAAkBH,OAAAA,EAAkC;AAClE,IAAA,MAAMC,QAAAA,GAAsB;AAC1B,MAAA,MAAMM,mBAAAA,CAAoB,IAAA,CAAKnC,SAAAA,CAAUoC,MAAAA,EAAQR,OAAAA,CAAAA;AAC9C,MAAA,GAAA,IAAA,CAAKxC,MAAMe,MAAAA,CAAO0B;;AAGvB,IAAA,MAAMC,OAAAA,GAAU;AACdT,MAAAA,UAAAA,EAAY,KAAK5B,SAAAA,CAAUyB,iBAAAA;AAC3BxB,MAAAA,KAAAA,EAAOkC,OAAAA,CAAQS,YAAAA;AACfC,MAAAA,UAAAA,EAAYV,OAAAA,CAAQU,UAAAA;MACpBC,MAAAA,EAAQ;AACV,KAAA;AAEA,IAAA,OAAO;AAAEV,MAAAA,QAAAA;AAAUC,MAAAA;AAAQ,KAAA;AAC7B;AAEA,EAAA,MAAgBU,0BAA0BC,QAAAA,EAAoD;AAE5F,IAAA,MAAMC,MAAAA,GAASC,gBAAgBF,QAAAA,EAAU;MAAEG,IAAAA,EAAM;AAAC,QAAA,GAAA;AAAK,QAAA;;KAAK,CAAA;AAC5D,IAAA,IAAI,CAACF,MAAAA,EAAQ;AAEX,MAAA,IAAI,CAAC,IAAA,CAAKrD,QAAAA,CAASyB,WAAAA,CAAY+B,YAAAA,EAAc;AAC3C,QAAA,OAAO,IAAIC,gBAAAA,CAAiB;AAC1B,UAAA;YACEC,IAAAA,EAAM,WAAA;AACNC,YAAAA,UAAAA,EAAY,QAAQC,EAAAA,EAAAA,CAASC,SAAAA,CAAU,CAAA,EAAG,CAAA,CAAA,CAAA,CAAA;YAC1CC,QAAAA,EAAU,IAAA,CAAK9D,SAASyB,WAAAA,CAAYsC,IAAAA;AACpCC,YAAAA,KAAAA,EAAO,IAAA,CAAKhE,QAAAA,CAASyB,WAAAA,CAAYwC,WAAAA,GAAcC,KAAAA,CAAM;cAAEvB,QAAAA,EAAUS;aAAS;AAC5E;AACD,SAAA,CAAA;AACH;AACA,MAAA,OAAO,IAAA;AACT;AAEA,IAAA,OAAO,IAAIK,gBAAAA,CAAiB;AAC1B,MAAA;QACEC,IAAAA,EAAM,WAAA;AACNC,QAAAA,UAAAA,EAAY,QAAQC,EAAAA,EAAAA,CAASC,SAAAA,CAAU,CAAA,EAAG,CAAA,CAAA,CAAA,CAAA;QAC1CC,QAAAA,EAAU,IAAA,CAAK9D,SAASyB,WAAAA,CAAYsC,IAAAA;QACpCC,KAAAA,EAAOX;AACT;AACD,KAAA,CAAA;AACH;EAEA,MAAgBc,aAAAA,CAAcC,UAAAA,GAAqB,EAAA,EAAsC;AACvF,IAAA,OAAO,MAAM,KAAKpE,QAAAA,CAASmE,aAAAA,CAAc,KAAKpE,KAAAA,EAAO,IAAA,CAAKW,wBAAwB0D,UAAAA,CAAAA;AACpF;EAEA,MAAgBC,eAAAA,CACdhE,OACAiE,SAAAA,EACwB;AACxB,IAAA,MAAMC,cAA6B,EAAA;AAEnC,IAAA,MAAMC,OAAAA,GAAU,MAAMC,OAAAA,CAAQC,GAAAA,CAC5BJ,SAAAA,CAAUK,IAAI,CAACC,GAAAA,KAAQC,OAAAA,CAAQxE,KAAAA,EAAOuE,GAAAA,EAAK;AAAE7E,MAAAA,KAAAA,EAAO,IAAA,CAAKA;AAAM,KAAA,CAAA,CAAA,CAAA;AAGjE,IAAA,KAAA,MAAW+E,YAAYN,OAAAA,EAAS;AAC9B,MAAA,IAAA,CAAKzE,KAAAA,CAAMiB,MAAM+D,IAAAA,CAAK;AACpBC,QAAAA,EAAAA,EAAIpB,EAAAA,EAAAA;AACJ3C,QAAAA,SAAAA,EAAW,KAAKlB,KAAAA,CAAMkB,SAAAA;AACtB+C,QAAAA,KAAAA,EAAOc,QAAAA,CAASd,KAAAA;AAChBiB,QAAAA,MAAAA,EAAQH,QAAAA,CAASG,MAAAA;AACjBC,QAAAA,IAAAA,EAAMJ,QAAAA,CAASI,IAAAA;AACfC,QAAAA,KAAAA,EAAOL,QAAAA,CAASK;OAClB,CAAA;AAEA,MAAA,IAAItE,MAAAA;AACJ,MAAA,IAAIiE,SAASK,KAAAA,EAAO;AAClBtE,QAAAA,MAAAA,GAAS,IAAA,CAAKF,SAAAA,CAAUyE,SAAAA,CAAUC,MAAAA,CAAO;UACvCC,MAAAA,EAAQR,QAAAA,CAASK,MAAMI,OAAAA;SACzB,CAAA;OACF,MAAO;AACL1E,QAAAA,MAAAA,GAAS,CAACiE,QAAAA,CAASG,MAAAA,CAAOO,OAAAA,EAAO,GAC7BV,QAAAA,CAASG,MAAAA,CAAOQ,cAAAA,EAAc,GAC9B,IAAA,CAAK9E,SAAAA,CAAU+E,YAAAA,CAAaL,MAAAA,CAAO;UACjCP,QAAAA,EAAU;AAAEI,YAAAA,IAAAA,EAAMJ,QAAAA,CAASI,IAAAA;AAAMlB,YAAAA,KAAAA,EAAOc,QAAAA,CAASd;AAAM;SACzD,CAAA;AACN;AAEAO,MAAAA,WAAAA,CAAYQ,IAAAA,CACV,IAAIY,WAAAA,CAAY;QACdjC,IAAAA,EAAM,aAAA;AACNI,QAAAA,QAAAA,EAAUgB,QAAAA,CAASI,IAAAA,EAAMnB,IAAAA,IAAQe,QAAAA,CAASF,GAAAA,CAAId,QAAAA;AAC9CH,QAAAA,UAAAA,EAAYmB,SAASF,GAAAA,CAAIjB,UAAAA;QACzBsB,MAAAA,EAAQ;UAAEvB,IAAAA,EAAM,MAAA;UAAQkC,KAAAA,EAAO/E;AAAO;AACxC,OAAA,CAAA,CAAA;AAGF,MAAA,IAAIiE,SAASK,KAAAA,EAAO;AAClB,QAAA,IAAA,CAAKlF,qBAAAA,CAAsB4F,GAAAA,CAAIf,QAAAA,CAASK,KAAK,CAAA;AAC7C,QAAA,IAAA,CAAKjF,kBAAAA,CAAmB2F,GAAAA,CAAIf,QAAAA,CAASK,KAAK,CAAA;AAC5C;AACF;AAEA,IAAA,OAAOZ,WAAAA;AACT;AAEA,EAAA,MAAMuB,YAAYtD,QAAAA,EAAoC;AACpD,IAAA,MAAM,IAAA,CAAKzC,KAAAA,CAAMe,MAAAA,CAAOiF,OAAAA,CAAQvD,QAAAA,CAAAA;AAClC;AAEA,EAAA,MAAMwD,GAAAA,GAAyC;AAC7C,IAAA,IAAI,IAAA,CAAKjG,MAAMa,MAAAA,EAAQ;AACrB,MAAA,OAAO,IAAA,CAAKb,KAAAA;AACd;AAGA,IAAA,MAAM,IAAA,CAAKC,QAAAA,CAASiG,MAAAA,CAAO,IAAA,CAAK3F,YAAY,CAAA;AAE5C,IAAA,OAAO,CAAC,IAAA,CAAKP,KAAAA,CAAMa,MAAAA,EAAQ;AACzB,MAAA,IAAA,CAAKwB,kBAAAA,EAAkB;AAEvB,MAAA,MAAMG,OAAAA,GAAU,MAAM,IAAA,CAAK4B,aAAAA,EAAa;AACxC,MAAA,MAAM,IAAA,CAAK1D,GAAAA,CAAIyF,OAAAA,CAAQC,IAAAA,CAAK,OAAA,EAAS;AAAEpG,QAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AAAOwC,QAAAA;OAAQ,CAAA;AAClE,MAAA,MAAMI,QAAAA,GAAW,MAAM,IAAA,CAAKyD,YAAAA,CAAa7D,OAAAA,CAAAA;AACzC,MAAA,MAAM,IAAA,CAAK9B,GAAAA,CAAIyF,OAAAA,CAAQC,IAAAA,CAAK,SAAA,EAAW;AAAEpG,QAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AAAO4C,QAAAA;OAAS,CAAA;AACvE;AAEA,IAAA,OAAO,IAAA,CAAK5C,KAAAA;AACd;AAEA,EAAA,MAAgBqG,aAAa7D,OAAAA,EAA4D;AACvF,IAAA,MAAMI,QAAAA,GAAW,MAAM,IAAA,CAAKL,MAAAA,CAAOC,OAAAA,CAAAA;AAGnC,IAAA,MAAM+B,SAAAA,GAAY3B,SAAS0D,YAAAA,EAAY;AACvC,IAAA,IAAI/B,SAAAA,CAAUgC,WAAW,CAAA,EAAG;AAC1B,MAAA,MAAMC,YAAAA,GAAe5D,SAAS6D,eAAAA,EAAe;AAC7C,MAAA,MAAMC,IAAAA,GAAOF,aAAa5B,GAAAA,CAAI,CAAC+B,MAAMA,CAAAA,CAAED,IAAI,CAAA,CAAEE,IAAAA,CAAK,IAAA,CAAA;AAElD,MAAA,IAAI,CAACF,IAAAA,IAAQlE,OAAAA,CAAQqE,OAAAA,EAAS;AAC5B,QAAA,MAAM,IAAI7E,WAAW,mCAAA,CAAA;AACvB;AAEA,MAAA,MAAM8E,mBAAAA,GAAsB,MAAM,IAAA,CAAK1D,yBAAAA,CAA0BsD,IAAAA,CAAAA;AACjE,MAAA,IAAI,CAACI,mBAAAA,EAAqB;AACxB,QAAA,MAAMC,GAAAA,GAAM,IAAI/E,UAAAA,CAAW,mDAAA,CAAA;AAC3B,QAAA,IAAA,CAAK9B,qBAAAA,CAAsB4F,IAAIiB,GAAAA,CAAAA;AAC/B,QAAA,IAAA,CAAK5G,kBAAAA,CAAmB2F,IAAIiB,GAAAA,CAAAA;AAE5B,QAAA,MAAM,IAAA,CAAK9G,QAAAA,CAASiG,MAAAA,CAAO,EAAE,CAAA;AAC7B,QAAA,MAAMc,cAAAA,GAAiB,MAAM,IAAA,CAAK5C,aAAAA,CAAc;AAC9C,UAAA;YACE6C,MAAAA,EAAQ,IAAA,CAAKhH,SAASyB,WAAAA,CAAYsC,IAAAA;YAClCkD,OAAAA,EAAS,IAAA;YACTC,MAAAA,EAAQ,KAAA;YACRC,MAAAA,EAAQ,KAAA;YACRC,WAAAA,EAAa;AACf;AACD,SAAA,CAAA;AACD,QAAA,IAAA,CAAK1G,sBAAAA,GAAyB,IAAA;AAC9B,QAAA,OAAO,MAAM,IAAA,CAAK0F,YAAAA,CAAaW,cAAAA,CAAAA;AACjC;AAEA,MAAA,MAAM,IAAA,CAAKhH,KAAAA,CAAMe,MAAAA,CAAOuG,GAAAA,CAAIR,mBAAAA,CAAAA;AAC5BvC,MAAAA,SAAAA,CAAUS,IAAAA,CAAI,GAAI8B,mBAAAA,CAAoBR,YAAAA,EAAY,CAAA;KACpD,MAAO;AACL,MAAA,MAAM,IAAA,CAAKtG,KAAAA,CAAMe,MAAAA,CAAOiF,OAAAA,CAAQpD,SAASH,QAAQ,CAAA;AACnD;AAGA,IAAA,KAAA,MAAW8E,eAAehD,SAAAA,EAAW;AACnC,MAAA,IAAA,CAAK9D,oBAAAA,CAAqB+G,SAASD,WAAAA,CAAAA;AACnC,MAAA,IAAI,IAAA,CAAK9G,qBAAqBgH,UAAAA,EAAY;AACxC,QAAA,IAAA,CAAKhH,qBAAqBiH,KAAAA,EAAK;AAC/B,QAAA,MAAMV,cAAAA,GAAiB,MAAM,IAAA,CAAK5C,aAAAA,CAAc;AAC9C,UAAA;AACE6C,YAAAA,MAAAA,EAAQM,WAAAA,CAAYxD,QAAAA;YACpBmD,OAAAA,EAAS,KAAA;YACTC,MAAAA,EAAQ,KAAA;YACRC,MAAAA,EAAQ,IAAA;YACRC,WAAAA,EAAa;AACf;AACD,SAAA,CAAA;AACD,QAAA,OAAO,MAAM,IAAA,CAAKhB,YAAAA,CAAaW,cAAAA,CAAAA;AACjC;AACF;AAEA,IAAA,MAAMxC,cAAc,MAAM,IAAA,CAAKF,eAAAA,CAAgB9B,OAAAA,CAAQS,cAAcsB,SAAAA,CAAAA;AACrE,IAAA,MAAM,IAAA,CAAKvE,KAAAA,CAAMe,MAAAA,CAAOiF,OAAAA,CAAQxB,WAAAA,CAAAA;AAGhC,IAAA,MAAMmD,YAAAA,GAAe,IAAA,CAAK3H,KAAAA,CAAMe,MAAAA,CAAO0B,QAAAA,CAASmF,MAAAA,CAAO,CAAC/C,GAAAA,KAAQA,GAAAA,CAAIgD,IAAAA,CAAK/H,gBAAAA,CAAiB,CAAA;AAC1F,IAAA,MAAM,IAAA,CAAKE,KAAAA,CAAMe,MAAAA,CAAO+G,UAAAA,CAAWH,YAAAA,CAAAA;AAEnC,IAAA,OAAO/E,QAAAA;AACT;AACF","file":"runner.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { v4 as uuidv4 } from \"uuid\";\nimport { AgentError } from \"@/agents/base.js\";\nimport { ChatModel, ChatModelOutput } from \"@/backend/chat.js\";\nimport { AssistantMessage, Message, ToolMessage } from \"@/backend/message.js\";\nimport { ToolCallPart } from \"ai\";\nimport { AnyTool } from \"@/tools/base.js\";\nimport { UnconstrainedMemory } from \"@/memory/unconstrainedMemory.js\";\nimport { RetryCounter } from \"@/internals/helpers/counter.js\";\nimport { RunContext } from \"@/context.js\";\nimport {\n  RequirementAgentExecutionConfig,\n  RequirementAgentRequest,\n  RequirementAgentRunState,\n  RequirementAgentTemplates,\n} from \"./types.js\";\nimport { Requirement, Rule } from \"./requirements/requirement.js\";\nimport { RequirementsReasoner, createSystemMessage } from \"./utils/llm.js\";\nimport { FinalAnswerTool, runTool } from \"./utils/tool.js\";\nimport { ToolCallChecker } from \"./utils/toolCallChecker.js\";\nimport { z } from \"zod\";\nimport { parseBrokenJson } from \"@/internals/helpers/schema.js\";\nimport { RequirementAgent } from \"@/agents/requirement/agent.js\";\nimport { mergeTokenUsage } from \"@/adapters/vercel/backend/utils.js\";\n\nconst TEMP_MESSAGE_KEY = \"tempMessage\";\n\n/**\n * Runner for RequirementAgent execution\n */\nexport class RequirementAgentRunner {\n  protected readonly state: RequirementAgentRunState;\n  protected readonly reasoner: RequirementsReasoner;\n  protected readonly iterationErrorCounter: RetryCounter;\n  protected readonly globalErrorCounter: RetryCounter;\n\n  constructor(\n    protected readonly llm: ChatModel,\n    protected readonly runConfig: RequirementAgentExecutionConfig,\n    tools: AnyTool[],\n    protected readonly requirements: Requirement[],\n    expectedOutput: string | z.ZodSchema | null,\n    protected readonly toolCallCycleChecker: ToolCallChecker,\n    protected readonly ctx: RunContext<RequirementAgent>,\n    protected forceFinalAnswerAsTool: boolean,\n    protected readonly templates: RequirementAgentTemplates,\n  ) {\n    this.state = {\n      answer: null,\n      result: null,\n      memory: new UnconstrainedMemory(),\n      steps: [],\n      iteration: 0,\n      usage: {\n        totalTokens: 0,\n        promptTokens: 0,\n        completionTokens: 0,\n        reasoningTokens: 0,\n        cachedPromptTokens: undefined,\n      },\n    };\n\n    this.requirements = requirements;\n    const finalAnswer = new FinalAnswerTool(expectedOutput, this.state);\n    this.reasoner = new RequirementsReasoner(tools, finalAnswer, ctx);\n\n    const maxRetriesPerIteration = runConfig.maxRetriesPerStep ?? 0;\n    this.iterationErrorCounter = new RetryCounter(maxRetriesPerIteration, AgentError);\n\n    const maxRetries = Math.max(maxRetriesPerIteration, runConfig.totalMaxRetries ?? 0);\n    this.globalErrorCounter = new RetryCounter(maxRetries, AgentError);\n  }\n\n  protected incrementIteration(): void {\n    this.state.iteration++;\n\n    if (this.runConfig.maxIterations && this.state.iteration > this.runConfig.maxIterations) {\n      throw new AgentError(\n        `Agent was not able to resolve the task in ${this.state.iteration} iterations.`,\n      );\n    }\n  }\n\n  protected async runLLM(request: RequirementAgentRequest): Promise<ChatModelOutput> {\n    const { messages, options } = await this.prepareLLMRequest(request);\n    const response = await this.llm.create({ ...options, messages });\n\n    if (response.usage) {\n      mergeTokenUsage(this.state.usage, response.usage);\n    }\n\n    return response;\n  }\n\n  protected async prepareLLMRequest(request: RequirementAgentRequest) {\n    const messages: Message[] = [\n      await createSystemMessage(this.templates.system, request),\n      ...this.state.memory.messages,\n    ];\n\n    const options = {\n      maxRetries: this.runConfig.maxRetriesPerStep,\n      tools: request.allowedTools,\n      toolChoice: request.toolChoice,\n      stream: false,\n    };\n\n    return { messages, options };\n  }\n\n  protected async createFinalAnswerToolCall(fullText: string): Promise<AssistantMessage | null> {\n    // Try to extract JSON from text\n    const parsed = parseBrokenJson(fullText, { pair: [\"{\", \"}\"] });\n    if (!parsed) {\n      // If no JSON and no custom schema, wrap in default schema\n      if (!this.reasoner.finalAnswer.customSchema) {\n        return new AssistantMessage([\n          {\n            type: \"tool-call\",\n            toolCallId: `call_${uuidv4().substring(0, 8)}`,\n            toolName: this.reasoner.finalAnswer.name,\n            input: this.reasoner.finalAnswer.inputSchema().parse({ response: fullText }),\n          },\n        ]);\n      }\n      return null;\n    }\n\n    return new AssistantMessage([\n      {\n        type: \"tool-call\",\n        toolCallId: `call_${uuidv4().substring(0, 8)}`,\n        toolName: this.reasoner.finalAnswer.name,\n        input: parsed,\n      },\n    ]);\n  }\n\n  protected async createRequest(extraRules: Rule[] = []): Promise<RequirementAgentRequest> {\n    return await this.reasoner.createRequest(this.state, this.forceFinalAnswerAsTool, extraRules);\n  }\n\n  protected async invokeToolCalls(\n    tools: AnyTool[],\n    toolCalls: ToolCallPart[],\n  ): Promise<ToolMessage[]> {\n    const toolResults: ToolMessage[] = [];\n\n    const results = await Promise.all(\n      toolCalls.map((msg) => runTool(tools, msg, { state: this.state })),\n    );\n\n    for (const toolCall of results) {\n      this.state.steps.push({\n        id: uuidv4(),\n        iteration: this.state.iteration,\n        input: toolCall.input,\n        output: toolCall.output,\n        tool: toolCall.tool,\n        error: toolCall.error,\n      });\n\n      let result: string;\n      if (toolCall.error) {\n        result = this.templates.toolError.render({\n          reason: toolCall.error.explain(),\n        });\n      } else {\n        result = !toolCall.output.isEmpty()\n          ? toolCall.output.getTextContent()\n          : this.templates.toolNoResult.render({\n              toolCall: { tool: toolCall.tool, input: toolCall.input },\n            });\n      }\n\n      toolResults.push(\n        new ToolMessage({\n          type: \"tool-result\",\n          toolName: toolCall.tool?.name || toolCall.msg.toolName,\n          toolCallId: toolCall.msg.toolCallId,\n          output: { type: \"text\", value: result },\n        }),\n      );\n\n      if (toolCall.error) {\n        this.iterationErrorCounter.use(toolCall.error);\n        this.globalErrorCounter.use(toolCall.error);\n      }\n    }\n\n    return toolResults;\n  }\n\n  async addMessages(messages: Message[]): Promise<void> {\n    await this.state.memory.addMany(messages);\n  }\n\n  async run(): Promise<RequirementAgentRunState> {\n    if (this.state.answer) {\n      return this.state;\n    }\n\n    // Initialize requirements\n    await this.reasoner.update(this.requirements);\n\n    while (!this.state.answer) {\n      this.incrementIteration();\n\n      const request = await this.createRequest();\n      await this.ctx.emitter.emit(\"start\", { state: this.state, request });\n      const response = await this.runIteration(request);\n      await this.ctx.emitter.emit(\"success\", { state: this.state, response });\n    }\n\n    return this.state;\n  }\n\n  protected async runIteration(request: RequirementAgentRequest): Promise<ChatModelOutput> {\n    const response = await this.runLLM(request);\n\n    // Try to cast text message to final answer tool call if allowed\n    const toolCalls = response.getToolCalls();\n    if (toolCalls.length === 0) {\n      const textMessages = response.getTextMessages();\n      const text = textMessages.map((m) => m.text).join(\"\\n\");\n\n      if (!text || request.canStop) {\n        throw new AgentError(\"Model produced an empty response.\");\n      }\n\n      const finalAnswerToolCall = await this.createFinalAnswerToolCall(text);\n      if (!finalAnswerToolCall) {\n        const err = new AgentError(\"Model produced an invalid final answer tool call.\");\n        this.iterationErrorCounter.use(err);\n        this.globalErrorCounter.use(err);\n\n        await this.reasoner.update([]);\n        const updatedRequest = await this.createRequest([\n          {\n            target: this.reasoner.finalAnswer.name,\n            allowed: true,\n            hidden: false,\n            forced: false,\n            preventStop: false,\n          },\n        ]);\n        this.forceFinalAnswerAsTool = true;\n        return await this.runIteration(updatedRequest);\n      }\n\n      await this.state.memory.add(finalAnswerToolCall);\n      toolCalls.push(...finalAnswerToolCall.getToolCalls());\n    } else {\n      await this.state.memory.addMany(response.messages);\n    }\n\n    // Check for cycles\n    for (const toolCallMsg of toolCalls) {\n      this.toolCallCycleChecker.register(toolCallMsg);\n      if (this.toolCallCycleChecker.cycleFound) {\n        this.toolCallCycleChecker.reset();\n        const updatedRequest = await this.createRequest([\n          {\n            target: toolCallMsg.toolName,\n            allowed: false,\n            hidden: false,\n            forced: true,\n            preventStop: false,\n          },\n        ]);\n        return await this.runIteration(updatedRequest);\n      }\n    }\n\n    const toolResults = await this.invokeToolCalls(request.allowedTools, toolCalls);\n    await this.state.memory.addMany(toolResults);\n\n    // Delete temporary messages\n    const tempMessages = this.state.memory.messages.filter((msg) => msg.meta[TEMP_MESSAGE_KEY]);\n    await this.state.memory.deleteMany(tempMessages);\n\n    return response;\n  }\n}\n"]}

package/dist/agents/requirement/types.cjs

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

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

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

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

@@ -0,0 +1,24 @@
+import '../../memory/base.cjs';
+import '../../backend/message.cjs';
+import '../../tools/base.cjs';
+import '../../errors.cjs';
+import '../../template.cjs';
+import '../../chat-DB_cf15S.cjs';
+import '../../base-v-PYIq5G.cjs';
+import 'zod';
+export { h as RequirementAgentCallbacks, c as RequirementAgentExecutionConfig, f as RequirementAgentOutput, a as RequirementAgentRequest, e as RequirementAgentRunInput, g as RequirementAgentRunOptions, R as RequirementAgentRunState, l as RequirementAgentRunStateStep, b as RequirementAgentSystemPromptInputSchema, i as RequirementAgentTaskPromptInputSchema, d as RequirementAgentTemplates, j as RequirementAgentToolErrorPromptInputSchema, k as RequirementAgentToolNoResultPromptInputSchema } from '../../types-Cx6f43du.cjs';
+import '../../internals/serializable.cjs';
+import '../../internals/types.cjs';
+import '../../internals/helpers/guards.cjs';
+import 'ai';
+import 'ajv';
+import '../../context.cjs';
+import '../../emitter-CGKd_-kK.cjs';
+import '../../internals/helpers/promise.cjs';
+import 'promise-based-task';
+import '../../cache/base.cjs';
+import '../../internals/helpers/schema.cjs';
+import 'zod-to-json-schema';
+import '../../backend/constants.cjs';
+import '../../logger/logger.cjs';
+import 'pino';

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

@@ -0,0 +1,24 @@
+import '../../memory/base.js';
+import '../../backend/message.js';
+import '../../tools/base.js';
+import '../../errors.js';
+import '../../template.js';
+import '../../chat-BbCYMAdE.js';
+import '../../base-BurRTBA7.js';
+import 'zod';
+export { h as RequirementAgentCallbacks, c as RequirementAgentExecutionConfig, f as RequirementAgentOutput, a as RequirementAgentRequest, e as RequirementAgentRunInput, g as RequirementAgentRunOptions, R as RequirementAgentRunState, l as RequirementAgentRunStateStep, b as RequirementAgentSystemPromptInputSchema, i as RequirementAgentTaskPromptInputSchema, d as RequirementAgentTemplates, j as RequirementAgentToolErrorPromptInputSchema, k as RequirementAgentToolNoResultPromptInputSchema } from '../../types-BXkkbABd.js';
+import '../../internals/serializable.js';
+import '../../internals/types.js';
+import '../../internals/helpers/guards.js';
+import 'ai';
+import 'ajv';
+import '../../context.js';
+import '../../emitter-CN7I4uSV.js';
+import '../../internals/helpers/promise.js';
+import 'promise-based-task';
+import '../../cache/base.js';
+import '../../internals/helpers/schema.js';
+import 'zod-to-json-schema';
+import '../../backend/constants.js';
+import '../../logger/logger.js';
+import 'pino';

package/dist/agents/requirement/types.js

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

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

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

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

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