beeai-framework 0.1.24 → 0.1.25

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

@@ -1,12 +1,12 @@
 import '../../context.cjs';
 import 'zod';
-import '../../base-XfWeXQ-S.cjs';
+import '../../base-BqzN2QwB.cjs';
 import '../../tools/base.cjs';
 import '../../memory/base.cjs';
-import '../../emitter-jN55XZZq.cjs';
-import '../../chat-CC0s8z23.cjs';
+import '../../emitter-UkIkNrek.cjs';
+import '../../chat-DaqV3woF.cjs';
 import '../../backend/message.cjs';
-import '../../types-D9pn62yF.cjs';
+import '../../types-YoNQSxQH.cjs';
 import './requirements/requirement.cjs';
 export { b as RequirementAgent, a as RequirementAgentInput, R as RequirementAgentTemplateFactory } from './runner.cjs';
 import './utils/toolCallChecker.cjs';

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

@@ -1,12 +1,12 @@
 import '../../context.js';
 import 'zod';
-import '../../base-CyFx7BRL.js';
+import '../../base-BfhNzUYf.js';
 import '../../tools/base.js';
 import '../../memory/base.js';
-import '../../emitter-36-9MnvA.js';
-import '../../chat-D8SVtw5Q.js';
+import '../../emitter-CL56qab4.js';
+import '../../chat-DTxmaKP0.js';
 import '../../backend/message.js';
-import '../../types-Dyd4e1G6.js';
+import '../../types-YT-WdK3m.js';
 import './requirements/requirement.js';
 export { b as RequirementAgent, a as RequirementAgentInput, R as RequirementAgentTemplateFactory } from './runner.js';
 import './utils/toolCallChecker.js';

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

@@ -2,8 +2,8 @@ import { Requirement, Rule } from './requirement.cjs';
 import { MultiTargetType, TargetType } from './utils.cjs';
 import { AnyTool } from '../../../tools/base.cjs';
 import { RunContext } from '../../../context.cjs';
-import { R as RequirementAgentRunState } from '../../../types-D9pn62yF.cjs';
-import '../../../emitter-jN55XZZq.cjs';
+import { R as RequirementAgentRunState } from '../../../types-YoNQSxQH.cjs';
+import '../../../emitter-UkIkNrek.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-CC0s8z23.cjs';
+import '../../../chat-DaqV3woF.cjs';
 import '../../../backend/constants.cjs';
 import '../../../logger/logger.cjs';
 import 'pino';
-import '../../../base-XfWeXQ-S.cjs';
+import '../../../base-BqzN2QwB.cjs';
 
 /**
  * Copyright 2025 © BeeAI a Series of LF Projects, LLC

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

@@ -2,8 +2,8 @@ import { Requirement, Rule } from './requirement.js';
 import { MultiTargetType, TargetType } from './utils.js';
 import { AnyTool } from '../../../tools/base.js';
 import { RunContext } from '../../../context.js';
-import { R as RequirementAgentRunState } from '../../../types-Dyd4e1G6.js';
-import '../../../emitter-36-9MnvA.js';
+import { R as RequirementAgentRunState } from '../../../types-YT-WdK3m.js';
+import '../../../emitter-CL56qab4.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-D8SVtw5Q.js';
+import '../../../chat-DTxmaKP0.js';
 import '../../../backend/constants.js';
 import '../../../logger/logger.js';
 import 'pino';
-import '../../../base-CyFx7BRL.js';
+import '../../../base-BfhNzUYf.js';
 
 /**
  * Copyright 2025 © BeeAI a Series of LF Projects, LLC

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-jN55XZZq.cjs';
+import { E as Emitter, C as Callback } from '../../../emitter-UkIkNrek.cjs';
 import { FrameworkError } from '../../../errors.cjs';
-import { R as RequirementAgentRunState } from '../../../types-D9pn62yF.cjs';
+import { R as RequirementAgentRunState } from '../../../types-YoNQSxQH.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-CC0s8z23.cjs';
+import '../../../chat-DaqV3woF.cjs';
 import '../../../backend/constants.cjs';
 import '../../../logger/logger.cjs';
 import 'pino';
-import '../../../base-XfWeXQ-S.cjs';
+import '../../../base-BqzN2QwB.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-36-9MnvA.js';
+import { E as Emitter, C as Callback } from '../../../emitter-CL56qab4.js';
 import { FrameworkError } from '../../../errors.js';
-import { R as RequirementAgentRunState } from '../../../types-Dyd4e1G6.js';
+import { R as RequirementAgentRunState } from '../../../types-YT-WdK3m.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-D8SVtw5Q.js';
+import '../../../chat-DTxmaKP0.js';
 import '../../../backend/constants.js';
 import '../../../logger/logger.js';
 import 'pino';
-import '../../../base-CyFx7BRL.js';
+import '../../../base-BfhNzUYf.js';
 
 interface Rule {
     target: string;

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

@@ -1,7 +1,7 @@
 import { AnyTool, AnyToolClass } from '../../../tools/base.cjs';
 import 'ajv';
 import '../../../context.cjs';
-import '../../../emitter-jN55XZZq.cjs';
+import '../../../emitter-UkIkNrek.cjs';
 import '../../../internals/types.cjs';
 import '../../../internals/helpers/guards.cjs';
 import '../../../internals/serializable.cjs';

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

@@ -1,7 +1,7 @@
 import { AnyTool, AnyToolClass } from '../../../tools/base.js';
 import 'ajv';
 import '../../../context.js';
-import '../../../emitter-36-9MnvA.js';
+import '../../../emitter-CL56qab4.js';
 import '../../../internals/types.js';
 import '../../../internals/helpers/guards.js';
 import '../../../internals/serializable.js';

package/dist/agents/requirement/runner.cjs

@@ -2,6 +2,7 @@
 
 var uuid = require('uuid');
 var base_cjs = require('../base.cjs');
+var chat_cjs = require('../../backend/chat.cjs');
 var message_cjs = require('../../backend/message.cjs');
 var unconstrainedMemory_cjs = require('../../memory/unconstrainedMemory.cjs');
 var counter_cjs = require('../../internals/helpers/counter.cjs');
@@ -87,7 +88,8 @@ class RequirementAgentRunner {
       messages,
       tools: request.allowedTools,
       toolChoice: request.toolChoice,
-      streamPartialToolCalls: true
+      streamPartialToolCalls: true,
+      maxRetries: this.runConfig.maxRetriesPerStep ?? 0
     };
   }
   async createFinalAnswerToolCall(fullText) {
@@ -192,18 +194,22 @@ class RequirementAgentRunner {
   }
   async runIteration(request) {
     const response = await this.runLLM(request);
-    const toolCalls = response.getToolCalls();
-    if (toolCalls.length === 0) {
+    if (response.getToolCalls().length === 0) {
       const textMessages = response.getTextMessages();
       const text = textMessages.map((m) => m.text).join("\n");
-      if (!text || request.canStop) {
-        throw new base_cjs.AgentError("Model produced an empty response.");
-      }
-      const finalAnswerToolCall = await this.createFinalAnswerToolCall(text);
-      if (!finalAnswerToolCall) {
+      const finalAnswerToolCall = text && request.canStop ? await this.createFinalAnswerToolCall(text) : null;
+      if (finalAnswerToolCall) {
+        const stream = this.createFinalAnswerStream(request.finalAnswer);
+        await stream.add(new chat_cjs.ChatModelOutput([
+          finalAnswerToolCall
+        ]));
+      } else {
         const err = new base_cjs.AgentError("Model produced an invalid final answer tool call.");
         this.iterationErrorCounter.use(err);
         this.globalErrorCounter.use(err);
+        if (!request.canStop) {
+          return await this.runIteration(request);
+        }
         await this.reasoner.update([]);
         const updatedRequest = await this.createRequest([
           {
@@ -217,11 +223,10 @@ class RequirementAgentRunner {
         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);
+      response.messages.length = 0;
+      response.messages.push(finalAnswerToolCall);
     }
+    const toolCalls = response.getToolCalls();
     for (const toolCallMsg of toolCalls) {
       this.toolCallCycleChecker.register(toolCallMsg);
       if (this.toolCallCycleChecker.cycleFound) {
@@ -239,7 +244,10 @@ class RequirementAgentRunner {
       }
     }
     const toolResults = await this.invokeToolCalls(request.allowedTools, toolCalls);
-    await this.state.memory.addMany(toolResults);
+    await this.state.memory.addMany([
+      ...response.messages,
+      ...toolResults
+    ]);
     const tempMessages = this.state.memory.messages.filter((msg) => msg.meta[TEMP_MESSAGE_KEY]);
     await this.state.memory.deleteMany(tempMessages);
     return response;

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

@@ -1 +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","streamMiddleware","createFinalAnswerStream","input","prepareLLMRequest","response","create","middleware","mergeTokenUsage","unbind","messages","createSystemMessage","system","allowedTools","toolChoice","streamPartialToolCalls","createFinalAnswerToolCall","fullText","parsed","parseBrokenJson","pair","customSchema","AssistantMessage","type","toolCallId","uuidv4","substring","toolName","name","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","StreamToolCallMiddleware","forceStreaming","key","matchNested","on","data","_","delta","outputStructured"],"mappings":";;;;;;;;;;;;;;;AA8BA,MAAMA,gBAAAA,GAAmB,aAAA;AAKlB,MAAMC,sBAAAA,CAAAA;EAnCb;;;;;;;;;;AAoCqBC,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,2CAAAA,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,wBAAAA,CAAgBnB,cAAAA,EAAgB,KAAKR,KAAK,CAAA;AAClE,IAAA,IAAA,CAAKC,QAAAA,GAAW,IAAI2B,4BAAAA,CAAqBtB,KAAAA,EAAOoB,aAAahB,GAAAA,CAAAA;AAE7D,IAAA,MAAMmB,sBAAAA,GAAyBxB,UAAUyB,iBAAAA,IAAqB,CAAA;AAC9D,IAAA,IAAA,CAAK5B,qBAAAA,GAAwB,IAAI6B,wBAAAA,CAAaF,sBAAAA,EAAwBG,mBAAAA,CAAAA;AAEtE,IAAA,MAAMC,aAAaC,IAAAA,CAAKC,GAAAA,CAAIN,sBAAAA,EAAwBxB,SAAAA,CAAU+B,mBAAmB,CAAA,CAAA;AACjF,IAAA,IAAA,CAAKjC,kBAAAA,GAAqB,IAAI4B,wBAAAA,CAAaE,UAAAA,EAAYD,mBAAAA,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,mBAAAA,CACR,CAAA,0CAAA,EAA6C,IAAA,CAAKhC,KAAAA,CAAMkB,SAAS,CAAA,YAAA,CAAc,CAAA;AAEnF;AACF;AAEA,EAAA,MAAgBqB,OAAOC,OAAAA,EAA4D;AACjF,IAAA,MAAMC,gBAAAA,GAAmB,IAAA,CAAKC,uBAAAA,CAAwBF,OAAAA,CAAQd,WAAW,CAAA;AACzE,IAAA,IAAI;AACF,MAAA,MAAMiB,KAAAA,GAAQ,MAAM,IAAA,CAAKC,iBAAAA,CAAkBJ,OAAAA,CAAAA;AAC3C,MAAA,MAAMK,QAAAA,GAAW,MAAM,IAAA,CAAKzC,GAAAA,CAAI0C,OAAOH,KAAAA,CAAAA,CAAOI,WAAWN,gBAAAA,CAAAA;AAEzD,MAAA,IAAII,SAAS1B,KAAAA,EAAO;AAClB6B,QAAAA,yBAAAA,CAAgB,IAAA,CAAKhD,KAAAA,CAAMmB,KAAAA,EAAO0B,QAAAA,CAAS1B,KAAK,CAAA;AAClD;AAEA,MAAA,OAAO0B,QAAAA;KACT,SAAA;AACEJ,MAAAA,gBAAAA,CAAiBQ,MAAAA,EAAM;AACzB;AACF;AAEA,EAAA,MAAgBL,kBAAkBJ,OAAAA,EAA2D;AAC3F,IAAA,MAAMU,QAAAA,GAAsB;AAC1B,MAAA,MAAMC,2BAAAA,CAAoB,IAAA,CAAKvC,SAAAA,CAAUwC,MAAAA,EAAQZ,OAAAA,CAAAA;AAC9C,MAAA,GAAA,IAAA,CAAKxC,MAAMe,MAAAA,CAAOmC;;AAGvB,IAAA,OAAO;AACLA,MAAAA,QAAAA;AACA5C,MAAAA,KAAAA,EAAOkC,OAAAA,CAAQa,YAAAA;AACfC,MAAAA,UAAAA,EAAYd,OAAAA,CAAQc,UAAAA;MACpBC,sBAAAA,EAAwB;AAC1B,KAAA;AACF;AAEA,EAAA,MAAgBC,0BAA0BC,QAAAA,EAAoD;AAE5F,IAAA,MAAMC,MAAAA,GAASC,2BAAgBF,QAAAA,EAAU;MAAEG,IAAAA,EAAM;AAAC,QAAA,GAAA;AAAK,QAAA;;KAAK,CAAA;AAC5D,IAAA,IAAI,CAACF,MAAAA,EAAQ;AAEX,MAAA,IAAI,CAAC,IAAA,CAAKzD,QAAAA,CAASyB,WAAAA,CAAYmC,YAAAA,EAAc;AAC3C,QAAA,OAAO,IAAIC,4BAAAA,CAAiB;AAC1B,UAAA;YACEC,IAAAA,EAAM,WAAA;AACNC,YAAAA,UAAAA,EAAY,QAAQC,OAAAA,EAAAA,CAASC,SAAAA,CAAU,CAAA,EAAG,CAAA,CAAA,CAAA,CAAA;YAC1CC,QAAAA,EAAU,IAAA,CAAKlE,SAASyB,WAAAA,CAAY0C,IAAAA;AACpCzB,YAAAA,KAAAA,EAAO,IAAA,CAAK1C,QAAAA,CAASyB,WAAAA,CAAY2C,WAAAA,GAAcC,KAAAA,CAAM;cAAEzB,QAAAA,EAAUY;aAAS;AAC5E;AACD,SAAA,CAAA;AACH;AACA,MAAA,OAAO,IAAA;AACT;AAEA,IAAA,OAAO,IAAIK,4BAAAA,CAAiB;AAC1B,MAAA;QACEC,IAAAA,EAAM,WAAA;AACNC,QAAAA,UAAAA,EAAY,QAAQC,OAAAA,EAAAA,CAASC,SAAAA,CAAU,CAAA,EAAG,CAAA,CAAA,CAAA,CAAA;QAC1CC,QAAAA,EAAU,IAAA,CAAKlE,SAASyB,WAAAA,CAAY0C,IAAAA;QACpCzB,KAAAA,EAAOe;AACT;AACD,KAAA,CAAA;AACH;EAEA,MAAgBa,aAAAA,CAAcC,UAAAA,GAAqB,EAAA,EAAsC;AACvF,IAAA,OAAO,MAAM,KAAKvE,QAAAA,CAASsE,aAAAA,CAAc,KAAKvE,KAAAA,EAAO,IAAA,CAAKW,wBAAwB6D,UAAAA,CAAAA;AACpF;EAEA,MAAgBC,eAAAA,CACdnE,OACAoE,SAAAA,EACwB;AACxB,IAAA,MAAMC,cAA6B,EAAA;AAEnC,IAAA,MAAMC,OAAAA,GAAU,MAAMC,OAAAA,CAAQC,GAAAA,CAC5BJ,SAAAA,CAAUK,IAAI,CAACC,GAAAA,KAAQC,gBAAAA,CAAQ3E,KAAAA,EAAO0E,GAAAA,EAAK;AAAEhF,MAAAA,KAAAA,EAAO,IAAA,CAAKA;AAAM,KAAA,CAAA,CAAA,CAAA;AAGjE,IAAA,KAAA,MAAWkF,YAAYN,OAAAA,EAAS;AAC9B,MAAA,IAAA,CAAK5E,KAAAA,CAAMiB,MAAMkE,IAAAA,CAAK;AACpBC,QAAAA,EAAAA,EAAInB,OAAAA,EAAAA;AACJ/C,QAAAA,SAAAA,EAAW,KAAKlB,KAAAA,CAAMkB,SAAAA;AACtByB,QAAAA,KAAAA,EAAOuC,QAAAA,CAASvC,KAAAA;AAChB0C,QAAAA,MAAAA,EAAQH,QAAAA,CAASG,MAAAA;AACjBC,QAAAA,IAAAA,EAAMJ,QAAAA,CAASI,IAAAA;AACfC,QAAAA,KAAAA,EAAOL,QAAAA,CAASK;OAClB,CAAA;AAEA,MAAA,IAAIzE,MAAAA;AACJ,MAAA,IAAIoE,SAASK,KAAAA,EAAO;AAClBzE,QAAAA,MAAAA,GAAS,IAAA,CAAKF,SAAAA,CAAU4E,SAAAA,CAAUC,MAAAA,CAAO;UACvCC,MAAAA,EAAQR,QAAAA,CAASK,MAAMI,OAAAA;SACzB,CAAA;OACF,MAAO;AACL7E,QAAAA,MAAAA,GAAS,CAACoE,QAAAA,CAASG,MAAAA,CAAOO,OAAAA,EAAO,GAC7BV,QAAAA,CAASG,MAAAA,CAAOQ,cAAAA,EAAc,GAC9B,IAAA,CAAKjF,SAAAA,CAAUkF,YAAAA,CAAaL,MAAAA,CAAO;UACjCP,QAAAA,EAAU;AAAEI,YAAAA,IAAAA,EAAMJ,QAAAA,CAASI,IAAAA;AAAM3C,YAAAA,KAAAA,EAAOuC,QAAAA,CAASvC;AAAM;SACzD,CAAA;AACN;AACAgC,MAAAA,WAAAA,CAAYQ,IAAAA,CACV,IAAIY,uBAAAA,CAAY;QACdhC,IAAAA,EAAM,aAAA;AACNI,QAAAA,QAAAA,EAAUe,QAAAA,CAASI,IAAAA,EAAMlB,IAAAA,IAAQc,QAAAA,CAASF,GAAAA,CAAIb,QAAAA;AAC9CH,QAAAA,UAAAA,EAAYkB,SAASF,GAAAA,CAAIhB,UAAAA;QACzBqB,MAAAA,EAAQ;UAAEtB,IAAAA,EAAM,MAAA;UAAQiC,KAAAA,EAAOlF;AAAO;AACxC,OAAA,CAAA,CAAA;AAGF,MAAA,IAAIoE,SAASK,KAAAA,EAAO;AAClB,QAAA,IAAA,CAAKrF,qBAAAA,CAAsB+F,GAAAA,CAAIf,QAAAA,CAASK,KAAK,CAAA;AAC7C,QAAA,IAAA,CAAKpF,kBAAAA,CAAmB8F,GAAAA,CAAIf,QAAAA,CAASK,KAAK,CAAA;AAC5C;AACF;AAEA,IAAA,OAAOZ,WAAAA;AACT;AAEA,EAAA,MAAMuB,YAAYhD,QAAAA,EAAoC;AACpD,IAAA,MAAM,IAAA,CAAKlD,KAAAA,CAAMe,MAAAA,CAAOoF,OAAAA,CAAQjD,QAAAA,CAAAA;AAClC;AAEA,EAAA,MAAMkD,GAAAA,GAAyC;AAC7C,IAAA,IAAI,IAAA,CAAKpG,MAAMa,MAAAA,EAAQ;AACrB,MAAA,OAAO,IAAA,CAAKb,KAAAA;AACd;AAGA,IAAA,MAAM,IAAA,CAAKC,QAAAA,CAASoG,MAAAA,CAAO,IAAA,CAAK9F,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,CAAK+B,aAAAA,EAAa;AACxC,MAAA,MAAM,IAAA,CAAK7D,GAAAA,CAAI4F,OAAAA,CAAQC,IAAAA,CAAK,OAAA,EAAS;AAAEvG,QAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AAAOwC,QAAAA;OAAQ,CAAA;AAClE,MAAA,MAAMK,QAAAA,GAAW,MAAM,IAAA,CAAK2D,YAAAA,CAAahE,OAAAA,CAAAA;AACzC,MAAA,MAAM,IAAA,CAAK9B,GAAAA,CAAI4F,OAAAA,CAAQC,IAAAA,CAAK,SAAA,EAAW;AAAEvG,QAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AAAO6C,QAAAA;OAAS,CAAA;AACvE;AAEA,IAAA,OAAO,IAAA,CAAK7C,KAAAA;AACd;AAEA,EAAA,MAAgBwG,aAAahE,OAAAA,EAA4D;AACvF,IAAA,MAAMK,QAAAA,GAAW,MAAM,IAAA,CAAKN,MAAAA,CAAOC,OAAAA,CAAAA;AAGnC,IAAA,MAAMkC,SAAAA,GAAY7B,SAAS4D,YAAAA,EAAY;AACvC,IAAA,IAAI/B,SAAAA,CAAUgC,WAAW,CAAA,EAAG;AAC1B,MAAA,MAAMC,YAAAA,GAAe9D,SAAS+D,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,IAAQrE,OAAAA,CAAQwE,OAAAA,EAAS;AAC5B,QAAA,MAAM,IAAIhF,oBAAW,mCAAA,CAAA;AACvB;AAEA,MAAA,MAAMiF,mBAAAA,GAAsB,MAAM,IAAA,CAAKzD,yBAAAA,CAA0BqD,IAAAA,CAAAA;AACjE,MAAA,IAAI,CAACI,mBAAAA,EAAqB;AACxB,QAAA,MAAMC,GAAAA,GAAM,IAAIlF,mBAAAA,CAAW,mDAAA,CAAA;AAC3B,QAAA,IAAA,CAAK9B,qBAAAA,CAAsB+F,IAAIiB,GAAAA,CAAAA;AAC/B,QAAA,IAAA,CAAK/G,kBAAAA,CAAmB8F,IAAIiB,GAAAA,CAAAA;AAE5B,QAAA,MAAM,IAAA,CAAKjH,QAAAA,CAASoG,MAAAA,CAAO,EAAE,CAAA;AAC7B,QAAA,MAAMc,cAAAA,GAAiB,MAAM,IAAA,CAAK5C,aAAAA,CAAc;AAC9C,UAAA;YACE6C,MAAAA,EAAQ,IAAA,CAAKnH,SAASyB,WAAAA,CAAY0C,IAAAA;YAClCiD,OAAAA,EAAS,IAAA;YACTC,MAAAA,EAAQ,KAAA;YACRC,MAAAA,EAAQ,KAAA;YACRC,WAAAA,EAAa;AACf;AACD,SAAA,CAAA;AACD,QAAA,IAAA,CAAK7G,sBAAAA,GAAyB,IAAA;AAC9B,QAAA,OAAO,MAAM,IAAA,CAAK6F,YAAAA,CAAaW,cAAAA,CAAAA;AACjC;AAEA,MAAA,MAAM,IAAA,CAAKnH,KAAAA,CAAMe,MAAAA,CAAO0G,GAAAA,CAAIR,mBAAAA,CAAAA;AAC5BvC,MAAAA,SAAAA,CAAUS,IAAAA,CAAI,GAAI8B,mBAAAA,CAAoBR,YAAAA,EAAY,CAAA;KACpD,MAAO;AACL,MAAA,MAAM,IAAA,CAAKzG,KAAAA,CAAMe,MAAAA,CAAOoF,OAAAA,CAAQtD,SAASK,QAAQ,CAAA;AACnD;AAGA,IAAA,KAAA,MAAWwE,eAAehD,SAAAA,EAAW;AACnC,MAAA,IAAA,CAAKjE,oBAAAA,CAAqBkH,SAASD,WAAAA,CAAAA;AACnC,MAAA,IAAI,IAAA,CAAKjH,qBAAqBmH,UAAAA,EAAY;AACxC,QAAA,IAAA,CAAKnH,qBAAqBoH,KAAAA,EAAK;AAC/B,QAAA,MAAMV,cAAAA,GAAiB,MAAM,IAAA,CAAK5C,aAAAA,CAAc;AAC9C,UAAA;AACE6C,YAAAA,MAAAA,EAAQM,WAAAA,CAAYvD,QAAAA;YACpBkD,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,CAAgBjC,OAAAA,CAAQa,cAAcqB,SAAAA,CAAAA;AACrE,IAAA,MAAM,IAAA,CAAK1E,KAAAA,CAAMe,MAAAA,CAAOoF,OAAAA,CAAQxB,WAAAA,CAAAA;AAGhC,IAAA,MAAMmD,YAAAA,GAAe,IAAA,CAAK9H,KAAAA,CAAMe,MAAAA,CAAOmC,QAAAA,CAAS6E,MAAAA,CAAO,CAAC/C,GAAAA,KAAQA,GAAAA,CAAIgD,IAAAA,CAAKlI,gBAAAA,CAAiB,CAAA;AAC1F,IAAA,MAAM,IAAA,CAAKE,KAAAA,CAAMe,MAAAA,CAAOkH,UAAAA,CAAWH,YAAAA,CAAAA;AAEnC,IAAA,OAAOjF,QAAAA;AACT;AAEQH,EAAAA,uBAAAA,CAAwBhB,WAAAA,EAA8B;AAC5D,IAAA,MAAMqB,UAAAA,GAAa,IAAImF,2CAAAA,CAAyB;MAC9Cd,MAAAA,EAAQ1F,WAAAA;MACRyG,cAAAA,EAAgB,KAAA;MAChBC,GAAAA,EAAK,UAAA;MACLC,WAAAA,EAAa;KACf,CAAA;AACAtF,IAAAA,UAAAA,CAAWuD,OAAAA,CAAQgC,EAAAA,CAAG,QAAA,EAAU,OAAOC,MAAMC,CAAAA,KAAAA;AAC3C,MAAA,MAAM,IAAA,CAAK9H,GAAAA,CAAI4F,OAAAA,CAAQC,IAAAA,CAAK,aAAA,EAAe;AACzCvG,QAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AACZqF,QAAAA,MAAAA,EAAQkD,IAAAA,CAAKlD,MAAAA;AACboD,QAAAA,KAAAA,EAAOF,IAAAA,CAAKE,KAAAA;QACZC,gBAAAA,EAAkBjH;OACpB,CAAA;KACF,CAAA;AACA,IAAA,OAAOsB,UAAAA;AACT;AACF","file":"runner.cjs","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, ChatModelInput, 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\";\nimport { StreamToolCallMiddleware } from \"@/middleware/streamToolCall.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 streamMiddleware = this.createFinalAnswerStream(request.finalAnswer);\n    try {\n      const input = await this.prepareLLMRequest(request);\n      const response = await this.llm.create(input).middleware(streamMiddleware);\n\n      if (response.usage) {\n        mergeTokenUsage(this.state.usage, response.usage);\n      }\n\n      return response;\n    } finally {\n      streamMiddleware.unbind();\n    }\n  }\n\n  protected async prepareLLMRequest(request: RequirementAgentRequest): Promise<ChatModelInput> {\n    const messages: Message[] = [\n      await createSystemMessage(this.templates.system, request),\n      ...this.state.memory.messages,\n    ];\n\n    return {\n      messages,\n      tools: request.allowedTools,\n      toolChoice: request.toolChoice,\n      streamPartialToolCalls: true,\n    };\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      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  private createFinalAnswerStream(finalAnswer: FinalAnswerTool) {\n    const middleware = new StreamToolCallMiddleware({\n      target: finalAnswer,\n      forceStreaming: false,\n      key: \"response\",\n      matchNested: false,\n    });\n    middleware.emitter.on(\"update\", async (data, _) => {\n      await this.ctx.emitter.emit(\"finalAnswer\", {\n        state: this.state,\n        output: data.output,\n        delta: data.delta,\n        outputStructured: undefined,\n      });\n    });\n    return middleware;\n  }\n}\n"]}
+{"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","streamMiddleware","createFinalAnswerStream","input","prepareLLMRequest","response","create","middleware","mergeTokenUsage","unbind","messages","createSystemMessage","system","allowedTools","toolChoice","streamPartialToolCalls","createFinalAnswerToolCall","fullText","parsed","parseBrokenJson","pair","customSchema","AssistantMessage","type","toolCallId","uuidv4","substring","toolName","name","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","finalAnswerToolCall","canStop","stream","add","ChatModelOutput","err","updatedRequest","target","allowed","hidden","forced","preventStop","toolCallMsg","register","cycleFound","reset","tempMessages","filter","meta","deleteMany","StreamToolCallMiddleware","forceStreaming","key","matchNested","on","data","_","delta","outputStructured"],"mappings":";;;;;;;;;;;;;;;;AA8BA,MAAMA,gBAAAA,GAAmB,aAAA;AAKlB,MAAMC,sBAAAA,CAAAA;EAnCb;;;;;;;;;;AAoCqBC,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,2CAAAA,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,wBAAAA,CAAgBnB,cAAAA,EAAgB,KAAKR,KAAK,CAAA;AAClE,IAAA,IAAA,CAAKC,QAAAA,GAAW,IAAI2B,4BAAAA,CAAqBtB,KAAAA,EAAOoB,aAAahB,GAAAA,CAAAA;AAE7D,IAAA,MAAMmB,sBAAAA,GAAyBxB,UAAUyB,iBAAAA,IAAqB,CAAA;AAC9D,IAAA,IAAA,CAAK5B,qBAAAA,GAAwB,IAAI6B,wBAAAA,CAAaF,sBAAAA,EAAwBG,mBAAAA,CAAAA;AAEtE,IAAA,MAAMC,aAAaC,IAAAA,CAAKC,GAAAA,CAAIN,sBAAAA,EAAwBxB,SAAAA,CAAU+B,mBAAmB,CAAA,CAAA;AACjF,IAAA,IAAA,CAAKjC,kBAAAA,GAAqB,IAAI4B,wBAAAA,CAAaE,UAAAA,EAAYD,mBAAAA,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,mBAAAA,CACR,CAAA,0CAAA,EAA6C,IAAA,CAAKhC,KAAAA,CAAMkB,SAAS,CAAA,YAAA,CAAc,CAAA;AAEnF;AACF;AAEA,EAAA,MAAgBqB,OAAOC,OAAAA,EAA4D;AACjF,IAAA,MAAMC,gBAAAA,GAAmB,IAAA,CAAKC,uBAAAA,CAAwBF,OAAAA,CAAQd,WAAW,CAAA;AACzE,IAAA,IAAI;AACF,MAAA,MAAMiB,KAAAA,GAAQ,MAAM,IAAA,CAAKC,iBAAAA,CAAkBJ,OAAAA,CAAAA;AAC3C,MAAA,MAAMK,QAAAA,GAAW,MAAM,IAAA,CAAKzC,GAAAA,CAAI0C,OAAOH,KAAAA,CAAAA,CAAOI,WAAWN,gBAAAA,CAAAA;AAEzD,MAAA,IAAII,SAAS1B,KAAAA,EAAO;AAClB6B,QAAAA,yBAAAA,CAAgB,IAAA,CAAKhD,KAAAA,CAAMmB,KAAAA,EAAO0B,QAAAA,CAAS1B,KAAK,CAAA;AAClD;AAEA,MAAA,OAAO0B,QAAAA;KACT,SAAA;AACEJ,MAAAA,gBAAAA,CAAiBQ,MAAAA,EAAM;AACzB;AACF;AAEA,EAAA,MAAgBL,kBAAkBJ,OAAAA,EAA2D;AAC3F,IAAA,MAAMU,QAAAA,GAAsB;AAC1B,MAAA,MAAMC,2BAAAA,CAAoB,IAAA,CAAKvC,SAAAA,CAAUwC,MAAAA,EAAQZ,OAAAA,CAAAA;AAC9C,MAAA,GAAA,IAAA,CAAKxC,MAAMe,MAAAA,CAAOmC;;AAGvB,IAAA,OAAO;AACLA,MAAAA,QAAAA;AACA5C,MAAAA,KAAAA,EAAOkC,OAAAA,CAAQa,YAAAA;AACfC,MAAAA,UAAAA,EAAYd,OAAAA,CAAQc,UAAAA;MACpBC,sBAAAA,EAAwB,IAAA;MACxBtB,UAAAA,EAAY,IAAA,CAAK5B,UAAUyB,iBAAAA,IAAqB;AAClD,KAAA;AACF;AAEA,EAAA,MAAgB0B,0BAA0BC,QAAAA,EAAoD;AAE5F,IAAA,MAAMC,MAAAA,GAASC,2BAAgBF,QAAAA,EAAU;MAAEG,IAAAA,EAAM;AAAC,QAAA,GAAA;AAAK,QAAA;;KAAK,CAAA;AAC5D,IAAA,IAAI,CAACF,MAAAA,EAAQ;AAEX,MAAA,IAAI,CAAC,IAAA,CAAKzD,QAAAA,CAASyB,WAAAA,CAAYmC,YAAAA,EAAc;AAC3C,QAAA,OAAO,IAAIC,4BAAAA,CAAiB;AAC1B,UAAA;YACEC,IAAAA,EAAM,WAAA;AACNC,YAAAA,UAAAA,EAAY,QAAQC,OAAAA,EAAAA,CAASC,SAAAA,CAAU,CAAA,EAAG,CAAA,CAAA,CAAA,CAAA;YAC1CC,QAAAA,EAAU,IAAA,CAAKlE,SAASyB,WAAAA,CAAY0C,IAAAA;AACpCzB,YAAAA,KAAAA,EAAO,IAAA,CAAK1C,QAAAA,CAASyB,WAAAA,CAAY2C,WAAAA,GAAcC,KAAAA,CAAM;cAAEzB,QAAAA,EAAUY;aAAS;AAC5E;AACD,SAAA,CAAA;AACH;AACA,MAAA,OAAO,IAAA;AACT;AAEA,IAAA,OAAO,IAAIK,4BAAAA,CAAiB;AAC1B,MAAA;QACEC,IAAAA,EAAM,WAAA;AACNC,QAAAA,UAAAA,EAAY,QAAQC,OAAAA,EAAAA,CAASC,SAAAA,CAAU,CAAA,EAAG,CAAA,CAAA,CAAA,CAAA;QAC1CC,QAAAA,EAAU,IAAA,CAAKlE,SAASyB,WAAAA,CAAY0C,IAAAA;QACpCzB,KAAAA,EAAOe;AACT;AACD,KAAA,CAAA;AACH;EAEA,MAAgBa,aAAAA,CAAcC,UAAAA,GAAqB,EAAA,EAAsC;AACvF,IAAA,OAAO,MAAM,KAAKvE,QAAAA,CAASsE,aAAAA,CAAc,KAAKvE,KAAAA,EAAO,IAAA,CAAKW,wBAAwB6D,UAAAA,CAAAA;AACpF;EAEA,MAAgBC,eAAAA,CACdnE,OACAoE,SAAAA,EACwB;AACxB,IAAA,MAAMC,cAA6B,EAAA;AAEnC,IAAA,MAAMC,OAAAA,GAAU,MAAMC,OAAAA,CAAQC,GAAAA,CAC5BJ,SAAAA,CAAUK,IAAI,CAACC,GAAAA,KAAQC,gBAAAA,CAAQ3E,KAAAA,EAAO0E,GAAAA,EAAK;AAAEhF,MAAAA,KAAAA,EAAO,IAAA,CAAKA;AAAM,KAAA,CAAA,CAAA,CAAA;AAGjE,IAAA,KAAA,MAAWkF,YAAYN,OAAAA,EAAS;AAC9B,MAAA,IAAA,CAAK5E,KAAAA,CAAMiB,MAAMkE,IAAAA,CAAK;AACpBC,QAAAA,EAAAA,EAAInB,OAAAA,EAAAA;AACJ/C,QAAAA,SAAAA,EAAW,KAAKlB,KAAAA,CAAMkB,SAAAA;AACtByB,QAAAA,KAAAA,EAAOuC,QAAAA,CAASvC,KAAAA;AAChB0C,QAAAA,MAAAA,EAAQH,QAAAA,CAASG,MAAAA;AACjBC,QAAAA,IAAAA,EAAMJ,QAAAA,CAASI,IAAAA;AACfC,QAAAA,KAAAA,EAAOL,QAAAA,CAASK;OAClB,CAAA;AAEA,MAAA,IAAIzE,MAAAA;AACJ,MAAA,IAAIoE,SAASK,KAAAA,EAAO;AAClBzE,QAAAA,MAAAA,GAAS,IAAA,CAAKF,SAAAA,CAAU4E,SAAAA,CAAUC,MAAAA,CAAO;UACvCC,MAAAA,EAAQR,QAAAA,CAASK,MAAMI,OAAAA;SACzB,CAAA;OACF,MAAO;AACL7E,QAAAA,MAAAA,GAAS,CAACoE,QAAAA,CAASG,MAAAA,CAAOO,OAAAA,EAAO,GAC7BV,QAAAA,CAASG,MAAAA,CAAOQ,cAAAA,EAAc,GAC9B,IAAA,CAAKjF,SAAAA,CAAUkF,YAAAA,CAAaL,MAAAA,CAAO;UACjCP,QAAAA,EAAU;AAAEI,YAAAA,IAAAA,EAAMJ,QAAAA,CAASI,IAAAA;AAAM3C,YAAAA,KAAAA,EAAOuC,QAAAA,CAASvC;AAAM;SACzD,CAAA;AACN;AACAgC,MAAAA,WAAAA,CAAYQ,IAAAA,CACV,IAAIY,uBAAAA,CAAY;QACdhC,IAAAA,EAAM,aAAA;AACNI,QAAAA,QAAAA,EAAUe,QAAAA,CAASI,IAAAA,EAAMlB,IAAAA,IAAQc,QAAAA,CAASF,GAAAA,CAAIb,QAAAA;AAC9CH,QAAAA,UAAAA,EAAYkB,SAASF,GAAAA,CAAIhB,UAAAA;QACzBqB,MAAAA,EAAQ;UAAEtB,IAAAA,EAAM,MAAA;UAAQiC,KAAAA,EAAOlF;AAAO;AACxC,OAAA,CAAA,CAAA;AAGF,MAAA,IAAIoE,SAASK,KAAAA,EAAO;AAClB,QAAA,IAAA,CAAKrF,qBAAAA,CAAsB+F,GAAAA,CAAIf,QAAAA,CAASK,KAAK,CAAA;AAC7C,QAAA,IAAA,CAAKpF,kBAAAA,CAAmB8F,GAAAA,CAAIf,QAAAA,CAASK,KAAK,CAAA;AAC5C;AACF;AAEA,IAAA,OAAOZ,WAAAA;AACT;AAEA,EAAA,MAAMuB,YAAYhD,QAAAA,EAAoC;AACpD,IAAA,MAAM,IAAA,CAAKlD,KAAAA,CAAMe,MAAAA,CAAOoF,OAAAA,CAAQjD,QAAAA,CAAAA;AAClC;AAEA,EAAA,MAAMkD,GAAAA,GAAyC;AAC7C,IAAA,IAAI,IAAA,CAAKpG,MAAMa,MAAAA,EAAQ;AACrB,MAAA,OAAO,IAAA,CAAKb,KAAAA;AACd;AAGA,IAAA,MAAM,IAAA,CAAKC,QAAAA,CAASoG,MAAAA,CAAO,IAAA,CAAK9F,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,CAAK+B,aAAAA,EAAa;AACxC,MAAA,MAAM,IAAA,CAAK7D,GAAAA,CAAI4F,OAAAA,CAAQC,IAAAA,CAAK,OAAA,EAAS;AAAEvG,QAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AAAOwC,QAAAA;OAAQ,CAAA;AAClE,MAAA,MAAMK,QAAAA,GAAW,MAAM,IAAA,CAAK2D,YAAAA,CAAahE,OAAAA,CAAAA;AACzC,MAAA,MAAM,IAAA,CAAK9B,GAAAA,CAAI4F,OAAAA,CAAQC,IAAAA,CAAK,SAAA,EAAW;AAAEvG,QAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AAAO6C,QAAAA;OAAS,CAAA;AACvE;AAEA,IAAA,OAAO,IAAA,CAAK7C,KAAAA;AACd;AAEA,EAAA,MAAgBwG,aAAahE,OAAAA,EAA4D;AACvF,IAAA,MAAMK,QAAAA,GAAW,MAAM,IAAA,CAAKN,MAAAA,CAAOC,OAAAA,CAAAA;AAGnC,IAAA,IAAIK,QAAAA,CAAS4D,YAAAA,EAAY,CAAGC,MAAAA,KAAW,CAAA,EAAG;AACxC,MAAA,MAAMC,YAAAA,GAAe9D,SAAS+D,eAAAA,EAAe;AAC7C,MAAA,MAAMC,IAAAA,GAAOF,aAAa5B,GAAAA,CAAI,CAAC+B,MAAMA,CAAAA,CAAED,IAAI,CAAA,CAAEE,IAAAA,CAAK,IAAA,CAAA;AAElD,MAAA,MAAMC,mBAAAA,GACJH,QAAQrE,OAAAA,CAAQyE,OAAAA,GAAU,MAAM,IAAA,CAAKzD,yBAAAA,CAA0BqD,IAAAA,CAAAA,GAAQ,IAAA;AACzE,MAAA,IAAIG,mBAAAA,EAAqB;AACvB,QAAA,MAAME,MAAAA,GAAS,IAAA,CAAKxE,uBAAAA,CAAwBF,OAAAA,CAAQd,WAAW,CAAA;AAC/D,QAAA,MAAMwF,MAAAA,CAAOC,GAAAA,CAAI,IAAIC,wBAAAA,CAAgB;AAACJ,UAAAA;SAAoB,CAAA,CAAA;OAC5D,MAAO;AACL,QAAA,MAAMK,GAAAA,GAAM,IAAIrF,mBAAAA,CAAW,mDAAA,CAAA;AAC3B,QAAA,IAAA,CAAK9B,qBAAAA,CAAsB+F,IAAIoB,GAAAA,CAAAA;AAC/B,QAAA,IAAA,CAAKlH,kBAAAA,CAAmB8F,IAAIoB,GAAAA,CAAAA;AAE5B,QAAA,IAAI,CAAC7E,QAAQyE,OAAAA,EAAS;AACpB,UAAA,OAAO,MAAM,IAAA,CAAKT,YAAAA,CAAahE,OAAAA,CAAAA;AACjC;AAEA,QAAA,MAAM,IAAA,CAAKvC,QAAAA,CAASoG,MAAAA,CAAO,EAAE,CAAA;AAC7B,QAAA,MAAMiB,cAAAA,GAAiB,MAAM,IAAA,CAAK/C,aAAAA,CAAc;AAC9C,UAAA;YACEgD,MAAAA,EAAQ,IAAA,CAAKtH,SAASyB,WAAAA,CAAY0C,IAAAA;YAClCoD,OAAAA,EAAS,IAAA;YACTC,MAAAA,EAAQ,KAAA;YACRC,MAAAA,EAAQ,KAAA;YACRC,WAAAA,EAAa;AACf;AACD,SAAA,CAAA;AACD,QAAA,IAAA,CAAKhH,sBAAAA,GAAyB,IAAA;AAC9B,QAAA,OAAO,MAAM,IAAA,CAAK6F,YAAAA,CAAac,cAAAA,CAAAA;AACjC;AAEAzE,MAAAA,QAAAA,CAASK,SAASwD,MAAAA,GAAS,CAAA;AAC3B7D,MAAAA,QAAAA,CAASK,QAAAA,CAASiC,KAAK6B,mBAAAA,CAAAA;AACzB;AAGA,IAAA,MAAMtC,SAAAA,GAAY7B,SAAS4D,YAAAA,EAAY;AACvC,IAAA,KAAA,MAAWmB,eAAelD,SAAAA,EAAW;AACnC,MAAA,IAAA,CAAKjE,oBAAAA,CAAqBoH,SAASD,WAAAA,CAAAA;AACnC,MAAA,IAAI,IAAA,CAAKnH,qBAAqBqH,UAAAA,EAAY;AACxC,QAAA,IAAA,CAAKrH,qBAAqBsH,KAAAA,EAAK;AAC/B,QAAA,MAAMT,cAAAA,GAAiB,MAAM,IAAA,CAAK/C,aAAAA,CAAc;AAC9C,UAAA;AACEgD,YAAAA,MAAAA,EAAQK,WAAAA,CAAYzD,QAAAA;YACpBqD,OAAAA,EAAS,KAAA;YACTC,MAAAA,EAAQ,KAAA;YACRC,MAAAA,EAAQ,IAAA;YACRC,WAAAA,EAAa;AACf;AACD,SAAA,CAAA;AACD,QAAA,OAAO,MAAM,IAAA,CAAKnB,YAAAA,CAAac,cAAAA,CAAAA;AACjC;AACF;AAEA,IAAA,MAAM3C,cAAc,MAAM,IAAA,CAAKF,eAAAA,CAAgBjC,OAAAA,CAAQa,cAAcqB,SAAAA,CAAAA;AACrE,IAAA,MAAM,IAAA,CAAK1E,KAAAA,CAAMe,MAAAA,CAAOoF,OAAAA,CAAQ;SAAItD,QAAAA,CAASK,QAAAA;AAAayB,MAAAA,GAAAA;AAAY,KAAA,CAAA;AAGtE,IAAA,MAAMqD,YAAAA,GAAe,IAAA,CAAKhI,KAAAA,CAAMe,MAAAA,CAAOmC,QAAAA,CAAS+E,MAAAA,CAAO,CAACjD,GAAAA,KAAQA,GAAAA,CAAIkD,IAAAA,CAAKpI,gBAAAA,CAAiB,CAAA;AAC1F,IAAA,MAAM,IAAA,CAAKE,KAAAA,CAAMe,MAAAA,CAAOoH,UAAAA,CAAWH,YAAAA,CAAAA;AAEnC,IAAA,OAAOnF,QAAAA;AACT;AAEQH,EAAAA,uBAAAA,CAAwBhB,WAAAA,EAA8B;AAC5D,IAAA,MAAMqB,UAAAA,GAAa,IAAIqF,2CAAAA,CAAyB;MAC9Cb,MAAAA,EAAQ7F,WAAAA;MACR2G,cAAAA,EAAgB,KAAA;MAChBC,GAAAA,EAAK,UAAA;MACLC,WAAAA,EAAa;KACf,CAAA;AACAxF,IAAAA,UAAAA,CAAWuD,OAAAA,CAAQkC,EAAAA,CAAG,QAAA,EAAU,OAAOC,MAAMC,CAAAA,KAAAA;AAC3C,MAAA,MAAM,IAAA,CAAKhI,GAAAA,CAAI4F,OAAAA,CAAQC,IAAAA,CAAK,aAAA,EAAe;AACzCvG,QAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AACZqF,QAAAA,MAAAA,EAAQoD,IAAAA,CAAKpD,MAAAA;AACbsD,QAAAA,KAAAA,EAAOF,IAAAA,CAAKE,KAAAA;QACZC,gBAAAA,EAAkBnH;OACpB,CAAA;KACF,CAAA;AACA,IAAA,OAAOsB,UAAAA;AACT;AACF","file":"runner.cjs","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, ChatModelInput, 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\";\nimport { StreamToolCallMiddleware } from \"@/middleware/streamToolCall.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 streamMiddleware = this.createFinalAnswerStream(request.finalAnswer);\n    try {\n      const input = await this.prepareLLMRequest(request);\n      const response = await this.llm.create(input).middleware(streamMiddleware);\n\n      if (response.usage) {\n        mergeTokenUsage(this.state.usage, response.usage);\n      }\n\n      return response;\n    } finally {\n      streamMiddleware.unbind();\n    }\n  }\n\n  protected async prepareLLMRequest(request: RequirementAgentRequest): Promise<ChatModelInput> {\n    const messages: Message[] = [\n      await createSystemMessage(this.templates.system, request),\n      ...this.state.memory.messages,\n    ];\n\n    return {\n      messages,\n      tools: request.allowedTools,\n      toolChoice: request.toolChoice,\n      streamPartialToolCalls: true,\n      maxRetries: this.runConfig.maxRetriesPerStep ?? 0,\n    };\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      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    if (response.getToolCalls().length === 0) {\n      const textMessages = response.getTextMessages();\n      const text = textMessages.map((m) => m.text).join(\"\\n\");\n\n      const finalAnswerToolCall =\n        text && request.canStop ? await this.createFinalAnswerToolCall(text) : null;\n      if (finalAnswerToolCall) {\n        const stream = this.createFinalAnswerStream(request.finalAnswer);\n        await stream.add(new ChatModelOutput([finalAnswerToolCall]));\n      } else {\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        if (!request.canStop) {\n          return await this.runIteration(request);\n        }\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      response.messages.length = 0;\n      response.messages.push(finalAnswerToolCall);\n    }\n\n    // Check for cycles\n    const toolCalls = response.getToolCalls();\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([...response.messages, ...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  private createFinalAnswerStream(finalAnswer: FinalAnswerTool) {\n    const middleware = new StreamToolCallMiddleware({\n      target: finalAnswer,\n      forceStreaming: false,\n      key: \"response\",\n      matchNested: false,\n    });\n    middleware.emitter.on(\"update\", async (data, _) => {\n      await this.ctx.emitter.emit(\"finalAnswer\", {\n        state: this.state,\n        output: data.output,\n        delta: data.delta,\n        outputStructured: undefined,\n      });\n    });\n    return middleware;\n  }\n}\n"]}

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

@@ -1,13 +1,13 @@
 import { RunContext, MiddlewareType, GetRunContext } from '../../context.cjs';
 import * as zod from 'zod';
 import { z } from 'zod';
-import { a as BaseAgent, c as AgentMeta } from '../../base-XfWeXQ-S.cjs';
+import { a as BaseAgent, c as AgentMeta } from '../../base-BqzN2QwB.cjs';
 import { AnyTool } from '../../tools/base.cjs';
 import { BaseMemory } from '../../memory/base.cjs';
-import { E as Emitter } from '../../emitter-jN55XZZq.cjs';
-import { m as ChatModel, n as ChatModelOutput, d as ChatModelInput } from '../../chat-CC0s8z23.cjs';
+import { E as Emitter } from '../../emitter-UkIkNrek.cjs';
+import { m as ChatModel, n as ChatModelOutput, d as ChatModelInput } from '../../chat-DaqV3woF.cjs';
 import { AssistantMessage, ToolMessage, Message } from '../../backend/message.cjs';
-import { c as RequirementAgentExecutionConfig, d as RequirementAgentTemplates, R as RequirementAgentRunState, a as RequirementAgentRequest, e as RequirementAgentRunInput, f as RequirementAgentOutput, g as RequirementAgentRunOptions, h as RequirementAgentCallbacks } from '../../types-D9pn62yF.cjs';
+import { c as RequirementAgentExecutionConfig, d as RequirementAgentTemplates, R as RequirementAgentRunState, a as RequirementAgentRequest, e as RequirementAgentRunInput, f as RequirementAgentOutput, g as RequirementAgentRunOptions, h as RequirementAgentCallbacks } from '../../types-YoNQSxQH.cjs';
 import { Requirement, Rule } from './requirements/requirement.cjs';
 import { ToolCallPart } from 'ai';
 import { RetryCounter } from '../../internals/helpers/counter.cjs';

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

@@ -1,13 +1,13 @@
 import { RunContext, MiddlewareType, GetRunContext } from '../../context.js';
 import * as zod from 'zod';
 import { z } from 'zod';
-import { a as BaseAgent, c as AgentMeta } from '../../base-CyFx7BRL.js';
+import { a as BaseAgent, c as AgentMeta } from '../../base-BfhNzUYf.js';
 import { AnyTool } from '../../tools/base.js';
 import { BaseMemory } from '../../memory/base.js';
-import { E as Emitter } from '../../emitter-36-9MnvA.js';
-import { m as ChatModel, n as ChatModelOutput, d as ChatModelInput } from '../../chat-D8SVtw5Q.js';
+import { E as Emitter } from '../../emitter-CL56qab4.js';
+import { m as ChatModel, n as ChatModelOutput, d as ChatModelInput } from '../../chat-DTxmaKP0.js';
 import { AssistantMessage, ToolMessage, Message } from '../../backend/message.js';
-import { c as RequirementAgentExecutionConfig, d as RequirementAgentTemplates, R as RequirementAgentRunState, a as RequirementAgentRequest, e as RequirementAgentRunInput, f as RequirementAgentOutput, g as RequirementAgentRunOptions, h as RequirementAgentCallbacks } from '../../types-Dyd4e1G6.js';
+import { c as RequirementAgentExecutionConfig, d as RequirementAgentTemplates, R as RequirementAgentRunState, a as RequirementAgentRequest, e as RequirementAgentRunInput, f as RequirementAgentOutput, g as RequirementAgentRunOptions, h as RequirementAgentCallbacks } from '../../types-YT-WdK3m.js';
 import { Requirement, Rule } from './requirements/requirement.js';
 import { ToolCallPart } from 'ai';
 import { RetryCounter } from '../../internals/helpers/counter.js';

package/dist/agents/requirement/runner.js

@@ -1,5 +1,6 @@
 import { v4 } from 'uuid';
 import { AgentError } from '../base.js';
+import { ChatModelOutput } from '../../backend/chat.js';
 import { AssistantMessage, ToolMessage } from '../../backend/message.js';
 import { UnconstrainedMemory } from '../../memory/unconstrainedMemory.js';
 import { RetryCounter } from '../../internals/helpers/counter.js';
@@ -85,7 +86,8 @@ class RequirementAgentRunner {
       messages,
       tools: request.allowedTools,
       toolChoice: request.toolChoice,
-      streamPartialToolCalls: true
+      streamPartialToolCalls: true,
+      maxRetries: this.runConfig.maxRetriesPerStep ?? 0
     };
   }
   async createFinalAnswerToolCall(fullText) {
@@ -190,18 +192,22 @@ class RequirementAgentRunner {
   }
   async runIteration(request) {
     const response = await this.runLLM(request);
-    const toolCalls = response.getToolCalls();
-    if (toolCalls.length === 0) {
+    if (response.getToolCalls().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 finalAnswerToolCall = text && request.canStop ? await this.createFinalAnswerToolCall(text) : null;
+      if (finalAnswerToolCall) {
+        const stream = this.createFinalAnswerStream(request.finalAnswer);
+        await stream.add(new ChatModelOutput([
+          finalAnswerToolCall
+        ]));
+      } else {
         const err = new AgentError("Model produced an invalid final answer tool call.");
         this.iterationErrorCounter.use(err);
         this.globalErrorCounter.use(err);
+        if (!request.canStop) {
+          return await this.runIteration(request);
+        }
         await this.reasoner.update([]);
         const updatedRequest = await this.createRequest([
           {
@@ -215,11 +221,10 @@ class RequirementAgentRunner {
         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);
+      response.messages.length = 0;
+      response.messages.push(finalAnswerToolCall);
     }
+    const toolCalls = response.getToolCalls();
     for (const toolCallMsg of toolCalls) {
       this.toolCallCycleChecker.register(toolCallMsg);
       if (this.toolCallCycleChecker.cycleFound) {
@@ -237,7 +242,10 @@ class RequirementAgentRunner {
       }
     }
     const toolResults = await this.invokeToolCalls(request.allowedTools, toolCalls);
-    await this.state.memory.addMany(toolResults);
+    await this.state.memory.addMany([
+      ...response.messages,
+      ...toolResults
+    ]);
     const tempMessages = this.state.memory.messages.filter((msg) => msg.meta[TEMP_MESSAGE_KEY]);
     await this.state.memory.deleteMany(tempMessages);
     return response;

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

@@ -1 +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","streamMiddleware","createFinalAnswerStream","input","prepareLLMRequest","response","create","middleware","mergeTokenUsage","unbind","messages","createSystemMessage","system","allowedTools","toolChoice","streamPartialToolCalls","createFinalAnswerToolCall","fullText","parsed","parseBrokenJson","pair","customSchema","AssistantMessage","type","toolCallId","uuidv4","substring","toolName","name","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","StreamToolCallMiddleware","forceStreaming","key","matchNested","on","data","_","delta","outputStructured"],"mappings":";;;;;;;;;;;;;AA8BA,MAAMA,gBAAAA,GAAmB,aAAA;AAKlB,MAAMC,sBAAAA,CAAAA;EAnCb;;;;;;;;;;AAoCqBC,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,MAAMC,gBAAAA,GAAmB,IAAA,CAAKC,uBAAAA,CAAwBF,OAAAA,CAAQd,WAAW,CAAA;AACzE,IAAA,IAAI;AACF,MAAA,MAAMiB,KAAAA,GAAQ,MAAM,IAAA,CAAKC,iBAAAA,CAAkBJ,OAAAA,CAAAA;AAC3C,MAAA,MAAMK,QAAAA,GAAW,MAAM,IAAA,CAAKzC,GAAAA,CAAI0C,OAAOH,KAAAA,CAAAA,CAAOI,WAAWN,gBAAAA,CAAAA;AAEzD,MAAA,IAAII,SAAS1B,KAAAA,EAAO;AAClB6B,QAAAA,eAAAA,CAAgB,IAAA,CAAKhD,KAAAA,CAAMmB,KAAAA,EAAO0B,QAAAA,CAAS1B,KAAK,CAAA;AAClD;AAEA,MAAA,OAAO0B,QAAAA;KACT,SAAA;AACEJ,MAAAA,gBAAAA,CAAiBQ,MAAAA,EAAM;AACzB;AACF;AAEA,EAAA,MAAgBL,kBAAkBJ,OAAAA,EAA2D;AAC3F,IAAA,MAAMU,QAAAA,GAAsB;AAC1B,MAAA,MAAMC,mBAAAA,CAAoB,IAAA,CAAKvC,SAAAA,CAAUwC,MAAAA,EAAQZ,OAAAA,CAAAA;AAC9C,MAAA,GAAA,IAAA,CAAKxC,MAAMe,MAAAA,CAAOmC;;AAGvB,IAAA,OAAO;AACLA,MAAAA,QAAAA;AACA5C,MAAAA,KAAAA,EAAOkC,OAAAA,CAAQa,YAAAA;AACfC,MAAAA,UAAAA,EAAYd,OAAAA,CAAQc,UAAAA;MACpBC,sBAAAA,EAAwB;AAC1B,KAAA;AACF;AAEA,EAAA,MAAgBC,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,CAAKzD,QAAAA,CAASyB,WAAAA,CAAYmC,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,CAAKlE,SAASyB,WAAAA,CAAY0C,IAAAA;AACpCzB,YAAAA,KAAAA,EAAO,IAAA,CAAK1C,QAAAA,CAASyB,WAAAA,CAAY2C,WAAAA,GAAcC,KAAAA,CAAM;cAAEzB,QAAAA,EAAUY;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,CAAKlE,SAASyB,WAAAA,CAAY0C,IAAAA;QACpCzB,KAAAA,EAAOe;AACT;AACD,KAAA,CAAA;AACH;EAEA,MAAgBa,aAAAA,CAAcC,UAAAA,GAAqB,EAAA,EAAsC;AACvF,IAAA,OAAO,MAAM,KAAKvE,QAAAA,CAASsE,aAAAA,CAAc,KAAKvE,KAAAA,EAAO,IAAA,CAAKW,wBAAwB6D,UAAAA,CAAAA;AACpF;EAEA,MAAgBC,eAAAA,CACdnE,OACAoE,SAAAA,EACwB;AACxB,IAAA,MAAMC,cAA6B,EAAA;AAEnC,IAAA,MAAMC,OAAAA,GAAU,MAAMC,OAAAA,CAAQC,GAAAA,CAC5BJ,SAAAA,CAAUK,IAAI,CAACC,GAAAA,KAAQC,OAAAA,CAAQ3E,KAAAA,EAAO0E,GAAAA,EAAK;AAAEhF,MAAAA,KAAAA,EAAO,IAAA,CAAKA;AAAM,KAAA,CAAA,CAAA,CAAA;AAGjE,IAAA,KAAA,MAAWkF,YAAYN,OAAAA,EAAS;AAC9B,MAAA,IAAA,CAAK5E,KAAAA,CAAMiB,MAAMkE,IAAAA,CAAK;AACpBC,QAAAA,EAAAA,EAAInB,EAAAA,EAAAA;AACJ/C,QAAAA,SAAAA,EAAW,KAAKlB,KAAAA,CAAMkB,SAAAA;AACtByB,QAAAA,KAAAA,EAAOuC,QAAAA,CAASvC,KAAAA;AAChB0C,QAAAA,MAAAA,EAAQH,QAAAA,CAASG,MAAAA;AACjBC,QAAAA,IAAAA,EAAMJ,QAAAA,CAASI,IAAAA;AACfC,QAAAA,KAAAA,EAAOL,QAAAA,CAASK;OAClB,CAAA;AAEA,MAAA,IAAIzE,MAAAA;AACJ,MAAA,IAAIoE,SAASK,KAAAA,EAAO;AAClBzE,QAAAA,MAAAA,GAAS,IAAA,CAAKF,SAAAA,CAAU4E,SAAAA,CAAUC,MAAAA,CAAO;UACvCC,MAAAA,EAAQR,QAAAA,CAASK,MAAMI,OAAAA;SACzB,CAAA;OACF,MAAO;AACL7E,QAAAA,MAAAA,GAAS,CAACoE,QAAAA,CAASG,MAAAA,CAAOO,OAAAA,EAAO,GAC7BV,QAAAA,CAASG,MAAAA,CAAOQ,cAAAA,EAAc,GAC9B,IAAA,CAAKjF,SAAAA,CAAUkF,YAAAA,CAAaL,MAAAA,CAAO;UACjCP,QAAAA,EAAU;AAAEI,YAAAA,IAAAA,EAAMJ,QAAAA,CAASI,IAAAA;AAAM3C,YAAAA,KAAAA,EAAOuC,QAAAA,CAASvC;AAAM;SACzD,CAAA;AACN;AACAgC,MAAAA,WAAAA,CAAYQ,IAAAA,CACV,IAAIY,WAAAA,CAAY;QACdhC,IAAAA,EAAM,aAAA;AACNI,QAAAA,QAAAA,EAAUe,QAAAA,CAASI,IAAAA,EAAMlB,IAAAA,IAAQc,QAAAA,CAASF,GAAAA,CAAIb,QAAAA;AAC9CH,QAAAA,UAAAA,EAAYkB,SAASF,GAAAA,CAAIhB,UAAAA;QACzBqB,MAAAA,EAAQ;UAAEtB,IAAAA,EAAM,MAAA;UAAQiC,KAAAA,EAAOlF;AAAO;AACxC,OAAA,CAAA,CAAA;AAGF,MAAA,IAAIoE,SAASK,KAAAA,EAAO;AAClB,QAAA,IAAA,CAAKrF,qBAAAA,CAAsB+F,GAAAA,CAAIf,QAAAA,CAASK,KAAK,CAAA;AAC7C,QAAA,IAAA,CAAKpF,kBAAAA,CAAmB8F,GAAAA,CAAIf,QAAAA,CAASK,KAAK,CAAA;AAC5C;AACF;AAEA,IAAA,OAAOZ,WAAAA;AACT;AAEA,EAAA,MAAMuB,YAAYhD,QAAAA,EAAoC;AACpD,IAAA,MAAM,IAAA,CAAKlD,KAAAA,CAAMe,MAAAA,CAAOoF,OAAAA,CAAQjD,QAAAA,CAAAA;AAClC;AAEA,EAAA,MAAMkD,GAAAA,GAAyC;AAC7C,IAAA,IAAI,IAAA,CAAKpG,MAAMa,MAAAA,EAAQ;AACrB,MAAA,OAAO,IAAA,CAAKb,KAAAA;AACd;AAGA,IAAA,MAAM,IAAA,CAAKC,QAAAA,CAASoG,MAAAA,CAAO,IAAA,CAAK9F,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,CAAK+B,aAAAA,EAAa;AACxC,MAAA,MAAM,IAAA,CAAK7D,GAAAA,CAAI4F,OAAAA,CAAQC,IAAAA,CAAK,OAAA,EAAS;AAAEvG,QAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AAAOwC,QAAAA;OAAQ,CAAA;AAClE,MAAA,MAAMK,QAAAA,GAAW,MAAM,IAAA,CAAK2D,YAAAA,CAAahE,OAAAA,CAAAA;AACzC,MAAA,MAAM,IAAA,CAAK9B,GAAAA,CAAI4F,OAAAA,CAAQC,IAAAA,CAAK,SAAA,EAAW;AAAEvG,QAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AAAO6C,QAAAA;OAAS,CAAA;AACvE;AAEA,IAAA,OAAO,IAAA,CAAK7C,KAAAA;AACd;AAEA,EAAA,MAAgBwG,aAAahE,OAAAA,EAA4D;AACvF,IAAA,MAAMK,QAAAA,GAAW,MAAM,IAAA,CAAKN,MAAAA,CAAOC,OAAAA,CAAAA;AAGnC,IAAA,MAAMkC,SAAAA,GAAY7B,SAAS4D,YAAAA,EAAY;AACvC,IAAA,IAAI/B,SAAAA,CAAUgC,WAAW,CAAA,EAAG;AAC1B,MAAA,MAAMC,YAAAA,GAAe9D,SAAS+D,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,IAAQrE,OAAAA,CAAQwE,OAAAA,EAAS;AAC5B,QAAA,MAAM,IAAIhF,WAAW,mCAAA,CAAA;AACvB;AAEA,MAAA,MAAMiF,mBAAAA,GAAsB,MAAM,IAAA,CAAKzD,yBAAAA,CAA0BqD,IAAAA,CAAAA;AACjE,MAAA,IAAI,CAACI,mBAAAA,EAAqB;AACxB,QAAA,MAAMC,GAAAA,GAAM,IAAIlF,UAAAA,CAAW,mDAAA,CAAA;AAC3B,QAAA,IAAA,CAAK9B,qBAAAA,CAAsB+F,IAAIiB,GAAAA,CAAAA;AAC/B,QAAA,IAAA,CAAK/G,kBAAAA,CAAmB8F,IAAIiB,GAAAA,CAAAA;AAE5B,QAAA,MAAM,IAAA,CAAKjH,QAAAA,CAASoG,MAAAA,CAAO,EAAE,CAAA;AAC7B,QAAA,MAAMc,cAAAA,GAAiB,MAAM,IAAA,CAAK5C,aAAAA,CAAc;AAC9C,UAAA;YACE6C,MAAAA,EAAQ,IAAA,CAAKnH,SAASyB,WAAAA,CAAY0C,IAAAA;YAClCiD,OAAAA,EAAS,IAAA;YACTC,MAAAA,EAAQ,KAAA;YACRC,MAAAA,EAAQ,KAAA;YACRC,WAAAA,EAAa;AACf;AACD,SAAA,CAAA;AACD,QAAA,IAAA,CAAK7G,sBAAAA,GAAyB,IAAA;AAC9B,QAAA,OAAO,MAAM,IAAA,CAAK6F,YAAAA,CAAaW,cAAAA,CAAAA;AACjC;AAEA,MAAA,MAAM,IAAA,CAAKnH,KAAAA,CAAMe,MAAAA,CAAO0G,GAAAA,CAAIR,mBAAAA,CAAAA;AAC5BvC,MAAAA,SAAAA,CAAUS,IAAAA,CAAI,GAAI8B,mBAAAA,CAAoBR,YAAAA,EAAY,CAAA;KACpD,MAAO;AACL,MAAA,MAAM,IAAA,CAAKzG,KAAAA,CAAMe,MAAAA,CAAOoF,OAAAA,CAAQtD,SAASK,QAAQ,CAAA;AACnD;AAGA,IAAA,KAAA,MAAWwE,eAAehD,SAAAA,EAAW;AACnC,MAAA,IAAA,CAAKjE,oBAAAA,CAAqBkH,SAASD,WAAAA,CAAAA;AACnC,MAAA,IAAI,IAAA,CAAKjH,qBAAqBmH,UAAAA,EAAY;AACxC,QAAA,IAAA,CAAKnH,qBAAqBoH,KAAAA,EAAK;AAC/B,QAAA,MAAMV,cAAAA,GAAiB,MAAM,IAAA,CAAK5C,aAAAA,CAAc;AAC9C,UAAA;AACE6C,YAAAA,MAAAA,EAAQM,WAAAA,CAAYvD,QAAAA;YACpBkD,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,CAAgBjC,OAAAA,CAAQa,cAAcqB,SAAAA,CAAAA;AACrE,IAAA,MAAM,IAAA,CAAK1E,KAAAA,CAAMe,MAAAA,CAAOoF,OAAAA,CAAQxB,WAAAA,CAAAA;AAGhC,IAAA,MAAMmD,YAAAA,GAAe,IAAA,CAAK9H,KAAAA,CAAMe,MAAAA,CAAOmC,QAAAA,CAAS6E,MAAAA,CAAO,CAAC/C,GAAAA,KAAQA,GAAAA,CAAIgD,IAAAA,CAAKlI,gBAAAA,CAAiB,CAAA;AAC1F,IAAA,MAAM,IAAA,CAAKE,KAAAA,CAAMe,MAAAA,CAAOkH,UAAAA,CAAWH,YAAAA,CAAAA;AAEnC,IAAA,OAAOjF,QAAAA;AACT;AAEQH,EAAAA,uBAAAA,CAAwBhB,WAAAA,EAA8B;AAC5D,IAAA,MAAMqB,UAAAA,GAAa,IAAImF,wBAAAA,CAAyB;MAC9Cd,MAAAA,EAAQ1F,WAAAA;MACRyG,cAAAA,EAAgB,KAAA;MAChBC,GAAAA,EAAK,UAAA;MACLC,WAAAA,EAAa;KACf,CAAA;AACAtF,IAAAA,UAAAA,CAAWuD,OAAAA,CAAQgC,EAAAA,CAAG,QAAA,EAAU,OAAOC,MAAMC,CAAAA,KAAAA;AAC3C,MAAA,MAAM,IAAA,CAAK9H,GAAAA,CAAI4F,OAAAA,CAAQC,IAAAA,CAAK,aAAA,EAAe;AACzCvG,QAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AACZqF,QAAAA,MAAAA,EAAQkD,IAAAA,CAAKlD,MAAAA;AACboD,QAAAA,KAAAA,EAAOF,IAAAA,CAAKE,KAAAA;QACZC,gBAAAA,EAAkBjH;OACpB,CAAA;KACF,CAAA;AACA,IAAA,OAAOsB,UAAAA;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, ChatModelInput, 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\";\nimport { StreamToolCallMiddleware } from \"@/middleware/streamToolCall.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 streamMiddleware = this.createFinalAnswerStream(request.finalAnswer);\n    try {\n      const input = await this.prepareLLMRequest(request);\n      const response = await this.llm.create(input).middleware(streamMiddleware);\n\n      if (response.usage) {\n        mergeTokenUsage(this.state.usage, response.usage);\n      }\n\n      return response;\n    } finally {\n      streamMiddleware.unbind();\n    }\n  }\n\n  protected async prepareLLMRequest(request: RequirementAgentRequest): Promise<ChatModelInput> {\n    const messages: Message[] = [\n      await createSystemMessage(this.templates.system, request),\n      ...this.state.memory.messages,\n    ];\n\n    return {\n      messages,\n      tools: request.allowedTools,\n      toolChoice: request.toolChoice,\n      streamPartialToolCalls: true,\n    };\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      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  private createFinalAnswerStream(finalAnswer: FinalAnswerTool) {\n    const middleware = new StreamToolCallMiddleware({\n      target: finalAnswer,\n      forceStreaming: false,\n      key: \"response\",\n      matchNested: false,\n    });\n    middleware.emitter.on(\"update\", async (data, _) => {\n      await this.ctx.emitter.emit(\"finalAnswer\", {\n        state: this.state,\n        output: data.output,\n        delta: data.delta,\n        outputStructured: undefined,\n      });\n    });\n    return middleware;\n  }\n}\n"]}
+{"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","streamMiddleware","createFinalAnswerStream","input","prepareLLMRequest","response","create","middleware","mergeTokenUsage","unbind","messages","createSystemMessage","system","allowedTools","toolChoice","streamPartialToolCalls","createFinalAnswerToolCall","fullText","parsed","parseBrokenJson","pair","customSchema","AssistantMessage","type","toolCallId","uuidv4","substring","toolName","name","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","finalAnswerToolCall","canStop","stream","add","ChatModelOutput","err","updatedRequest","target","allowed","hidden","forced","preventStop","toolCallMsg","register","cycleFound","reset","tempMessages","filter","meta","deleteMany","StreamToolCallMiddleware","forceStreaming","key","matchNested","on","data","_","delta","outputStructured"],"mappings":";;;;;;;;;;;;;;AA8BA,MAAMA,gBAAAA,GAAmB,aAAA;AAKlB,MAAMC,sBAAAA,CAAAA;EAnCb;;;;;;;;;;AAoCqBC,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,MAAMC,gBAAAA,GAAmB,IAAA,CAAKC,uBAAAA,CAAwBF,OAAAA,CAAQd,WAAW,CAAA;AACzE,IAAA,IAAI;AACF,MAAA,MAAMiB,KAAAA,GAAQ,MAAM,IAAA,CAAKC,iBAAAA,CAAkBJ,OAAAA,CAAAA;AAC3C,MAAA,MAAMK,QAAAA,GAAW,MAAM,IAAA,CAAKzC,GAAAA,CAAI0C,OAAOH,KAAAA,CAAAA,CAAOI,WAAWN,gBAAAA,CAAAA;AAEzD,MAAA,IAAII,SAAS1B,KAAAA,EAAO;AAClB6B,QAAAA,eAAAA,CAAgB,IAAA,CAAKhD,KAAAA,CAAMmB,KAAAA,EAAO0B,QAAAA,CAAS1B,KAAK,CAAA;AAClD;AAEA,MAAA,OAAO0B,QAAAA;KACT,SAAA;AACEJ,MAAAA,gBAAAA,CAAiBQ,MAAAA,EAAM;AACzB;AACF;AAEA,EAAA,MAAgBL,kBAAkBJ,OAAAA,EAA2D;AAC3F,IAAA,MAAMU,QAAAA,GAAsB;AAC1B,MAAA,MAAMC,mBAAAA,CAAoB,IAAA,CAAKvC,SAAAA,CAAUwC,MAAAA,EAAQZ,OAAAA,CAAAA;AAC9C,MAAA,GAAA,IAAA,CAAKxC,MAAMe,MAAAA,CAAOmC;;AAGvB,IAAA,OAAO;AACLA,MAAAA,QAAAA;AACA5C,MAAAA,KAAAA,EAAOkC,OAAAA,CAAQa,YAAAA;AACfC,MAAAA,UAAAA,EAAYd,OAAAA,CAAQc,UAAAA;MACpBC,sBAAAA,EAAwB,IAAA;MACxBtB,UAAAA,EAAY,IAAA,CAAK5B,UAAUyB,iBAAAA,IAAqB;AAClD,KAAA;AACF;AAEA,EAAA,MAAgB0B,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,CAAKzD,QAAAA,CAASyB,WAAAA,CAAYmC,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,CAAKlE,SAASyB,WAAAA,CAAY0C,IAAAA;AACpCzB,YAAAA,KAAAA,EAAO,IAAA,CAAK1C,QAAAA,CAASyB,WAAAA,CAAY2C,WAAAA,GAAcC,KAAAA,CAAM;cAAEzB,QAAAA,EAAUY;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,CAAKlE,SAASyB,WAAAA,CAAY0C,IAAAA;QACpCzB,KAAAA,EAAOe;AACT;AACD,KAAA,CAAA;AACH;EAEA,MAAgBa,aAAAA,CAAcC,UAAAA,GAAqB,EAAA,EAAsC;AACvF,IAAA,OAAO,MAAM,KAAKvE,QAAAA,CAASsE,aAAAA,CAAc,KAAKvE,KAAAA,EAAO,IAAA,CAAKW,wBAAwB6D,UAAAA,CAAAA;AACpF;EAEA,MAAgBC,eAAAA,CACdnE,OACAoE,SAAAA,EACwB;AACxB,IAAA,MAAMC,cAA6B,EAAA;AAEnC,IAAA,MAAMC,OAAAA,GAAU,MAAMC,OAAAA,CAAQC,GAAAA,CAC5BJ,SAAAA,CAAUK,IAAI,CAACC,GAAAA,KAAQC,OAAAA,CAAQ3E,KAAAA,EAAO0E,GAAAA,EAAK;AAAEhF,MAAAA,KAAAA,EAAO,IAAA,CAAKA;AAAM,KAAA,CAAA,CAAA,CAAA;AAGjE,IAAA,KAAA,MAAWkF,YAAYN,OAAAA,EAAS;AAC9B,MAAA,IAAA,CAAK5E,KAAAA,CAAMiB,MAAMkE,IAAAA,CAAK;AACpBC,QAAAA,EAAAA,EAAInB,EAAAA,EAAAA;AACJ/C,QAAAA,SAAAA,EAAW,KAAKlB,KAAAA,CAAMkB,SAAAA;AACtByB,QAAAA,KAAAA,EAAOuC,QAAAA,CAASvC,KAAAA;AAChB0C,QAAAA,MAAAA,EAAQH,QAAAA,CAASG,MAAAA;AACjBC,QAAAA,IAAAA,EAAMJ,QAAAA,CAASI,IAAAA;AACfC,QAAAA,KAAAA,EAAOL,QAAAA,CAASK;OAClB,CAAA;AAEA,MAAA,IAAIzE,MAAAA;AACJ,MAAA,IAAIoE,SAASK,KAAAA,EAAO;AAClBzE,QAAAA,MAAAA,GAAS,IAAA,CAAKF,SAAAA,CAAU4E,SAAAA,CAAUC,MAAAA,CAAO;UACvCC,MAAAA,EAAQR,QAAAA,CAASK,MAAMI,OAAAA;SACzB,CAAA;OACF,MAAO;AACL7E,QAAAA,MAAAA,GAAS,CAACoE,QAAAA,CAASG,MAAAA,CAAOO,OAAAA,EAAO,GAC7BV,QAAAA,CAASG,MAAAA,CAAOQ,cAAAA,EAAc,GAC9B,IAAA,CAAKjF,SAAAA,CAAUkF,YAAAA,CAAaL,MAAAA,CAAO;UACjCP,QAAAA,EAAU;AAAEI,YAAAA,IAAAA,EAAMJ,QAAAA,CAASI,IAAAA;AAAM3C,YAAAA,KAAAA,EAAOuC,QAAAA,CAASvC;AAAM;SACzD,CAAA;AACN;AACAgC,MAAAA,WAAAA,CAAYQ,IAAAA,CACV,IAAIY,WAAAA,CAAY;QACdhC,IAAAA,EAAM,aAAA;AACNI,QAAAA,QAAAA,EAAUe,QAAAA,CAASI,IAAAA,EAAMlB,IAAAA,IAAQc,QAAAA,CAASF,GAAAA,CAAIb,QAAAA;AAC9CH,QAAAA,UAAAA,EAAYkB,SAASF,GAAAA,CAAIhB,UAAAA;QACzBqB,MAAAA,EAAQ;UAAEtB,IAAAA,EAAM,MAAA;UAAQiC,KAAAA,EAAOlF;AAAO;AACxC,OAAA,CAAA,CAAA;AAGF,MAAA,IAAIoE,SAASK,KAAAA,EAAO;AAClB,QAAA,IAAA,CAAKrF,qBAAAA,CAAsB+F,GAAAA,CAAIf,QAAAA,CAASK,KAAK,CAAA;AAC7C,QAAA,IAAA,CAAKpF,kBAAAA,CAAmB8F,GAAAA,CAAIf,QAAAA,CAASK,KAAK,CAAA;AAC5C;AACF;AAEA,IAAA,OAAOZ,WAAAA;AACT;AAEA,EAAA,MAAMuB,YAAYhD,QAAAA,EAAoC;AACpD,IAAA,MAAM,IAAA,CAAKlD,KAAAA,CAAMe,MAAAA,CAAOoF,OAAAA,CAAQjD,QAAAA,CAAAA;AAClC;AAEA,EAAA,MAAMkD,GAAAA,GAAyC;AAC7C,IAAA,IAAI,IAAA,CAAKpG,MAAMa,MAAAA,EAAQ;AACrB,MAAA,OAAO,IAAA,CAAKb,KAAAA;AACd;AAGA,IAAA,MAAM,IAAA,CAAKC,QAAAA,CAASoG,MAAAA,CAAO,IAAA,CAAK9F,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,CAAK+B,aAAAA,EAAa;AACxC,MAAA,MAAM,IAAA,CAAK7D,GAAAA,CAAI4F,OAAAA,CAAQC,IAAAA,CAAK,OAAA,EAAS;AAAEvG,QAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AAAOwC,QAAAA;OAAQ,CAAA;AAClE,MAAA,MAAMK,QAAAA,GAAW,MAAM,IAAA,CAAK2D,YAAAA,CAAahE,OAAAA,CAAAA;AACzC,MAAA,MAAM,IAAA,CAAK9B,GAAAA,CAAI4F,OAAAA,CAAQC,IAAAA,CAAK,SAAA,EAAW;AAAEvG,QAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AAAO6C,QAAAA;OAAS,CAAA;AACvE;AAEA,IAAA,OAAO,IAAA,CAAK7C,KAAAA;AACd;AAEA,EAAA,MAAgBwG,aAAahE,OAAAA,EAA4D;AACvF,IAAA,MAAMK,QAAAA,GAAW,MAAM,IAAA,CAAKN,MAAAA,CAAOC,OAAAA,CAAAA;AAGnC,IAAA,IAAIK,QAAAA,CAAS4D,YAAAA,EAAY,CAAGC,MAAAA,KAAW,CAAA,EAAG;AACxC,MAAA,MAAMC,YAAAA,GAAe9D,SAAS+D,eAAAA,EAAe;AAC7C,MAAA,MAAMC,IAAAA,GAAOF,aAAa5B,GAAAA,CAAI,CAAC+B,MAAMA,CAAAA,CAAED,IAAI,CAAA,CAAEE,IAAAA,CAAK,IAAA,CAAA;AAElD,MAAA,MAAMC,mBAAAA,GACJH,QAAQrE,OAAAA,CAAQyE,OAAAA,GAAU,MAAM,IAAA,CAAKzD,yBAAAA,CAA0BqD,IAAAA,CAAAA,GAAQ,IAAA;AACzE,MAAA,IAAIG,mBAAAA,EAAqB;AACvB,QAAA,MAAME,MAAAA,GAAS,IAAA,CAAKxE,uBAAAA,CAAwBF,OAAAA,CAAQd,WAAW,CAAA;AAC/D,QAAA,MAAMwF,MAAAA,CAAOC,GAAAA,CAAI,IAAIC,eAAAA,CAAgB;AAACJ,UAAAA;SAAoB,CAAA,CAAA;OAC5D,MAAO;AACL,QAAA,MAAMK,GAAAA,GAAM,IAAIrF,UAAAA,CAAW,mDAAA,CAAA;AAC3B,QAAA,IAAA,CAAK9B,qBAAAA,CAAsB+F,IAAIoB,GAAAA,CAAAA;AAC/B,QAAA,IAAA,CAAKlH,kBAAAA,CAAmB8F,IAAIoB,GAAAA,CAAAA;AAE5B,QAAA,IAAI,CAAC7E,QAAQyE,OAAAA,EAAS;AACpB,UAAA,OAAO,MAAM,IAAA,CAAKT,YAAAA,CAAahE,OAAAA,CAAAA;AACjC;AAEA,QAAA,MAAM,IAAA,CAAKvC,QAAAA,CAASoG,MAAAA,CAAO,EAAE,CAAA;AAC7B,QAAA,MAAMiB,cAAAA,GAAiB,MAAM,IAAA,CAAK/C,aAAAA,CAAc;AAC9C,UAAA;YACEgD,MAAAA,EAAQ,IAAA,CAAKtH,SAASyB,WAAAA,CAAY0C,IAAAA;YAClCoD,OAAAA,EAAS,IAAA;YACTC,MAAAA,EAAQ,KAAA;YACRC,MAAAA,EAAQ,KAAA;YACRC,WAAAA,EAAa;AACf;AACD,SAAA,CAAA;AACD,QAAA,IAAA,CAAKhH,sBAAAA,GAAyB,IAAA;AAC9B,QAAA,OAAO,MAAM,IAAA,CAAK6F,YAAAA,CAAac,cAAAA,CAAAA;AACjC;AAEAzE,MAAAA,QAAAA,CAASK,SAASwD,MAAAA,GAAS,CAAA;AAC3B7D,MAAAA,QAAAA,CAASK,QAAAA,CAASiC,KAAK6B,mBAAAA,CAAAA;AACzB;AAGA,IAAA,MAAMtC,SAAAA,GAAY7B,SAAS4D,YAAAA,EAAY;AACvC,IAAA,KAAA,MAAWmB,eAAelD,SAAAA,EAAW;AACnC,MAAA,IAAA,CAAKjE,oBAAAA,CAAqBoH,SAASD,WAAAA,CAAAA;AACnC,MAAA,IAAI,IAAA,CAAKnH,qBAAqBqH,UAAAA,EAAY;AACxC,QAAA,IAAA,CAAKrH,qBAAqBsH,KAAAA,EAAK;AAC/B,QAAA,MAAMT,cAAAA,GAAiB,MAAM,IAAA,CAAK/C,aAAAA,CAAc;AAC9C,UAAA;AACEgD,YAAAA,MAAAA,EAAQK,WAAAA,CAAYzD,QAAAA;YACpBqD,OAAAA,EAAS,KAAA;YACTC,MAAAA,EAAQ,KAAA;YACRC,MAAAA,EAAQ,IAAA;YACRC,WAAAA,EAAa;AACf;AACD,SAAA,CAAA;AACD,QAAA,OAAO,MAAM,IAAA,CAAKnB,YAAAA,CAAac,cAAAA,CAAAA;AACjC;AACF;AAEA,IAAA,MAAM3C,cAAc,MAAM,IAAA,CAAKF,eAAAA,CAAgBjC,OAAAA,CAAQa,cAAcqB,SAAAA,CAAAA;AACrE,IAAA,MAAM,IAAA,CAAK1E,KAAAA,CAAMe,MAAAA,CAAOoF,OAAAA,CAAQ;SAAItD,QAAAA,CAASK,QAAAA;AAAayB,MAAAA,GAAAA;AAAY,KAAA,CAAA;AAGtE,IAAA,MAAMqD,YAAAA,GAAe,IAAA,CAAKhI,KAAAA,CAAMe,MAAAA,CAAOmC,QAAAA,CAAS+E,MAAAA,CAAO,CAACjD,GAAAA,KAAQA,GAAAA,CAAIkD,IAAAA,CAAKpI,gBAAAA,CAAiB,CAAA;AAC1F,IAAA,MAAM,IAAA,CAAKE,KAAAA,CAAMe,MAAAA,CAAOoH,UAAAA,CAAWH,YAAAA,CAAAA;AAEnC,IAAA,OAAOnF,QAAAA;AACT;AAEQH,EAAAA,uBAAAA,CAAwBhB,WAAAA,EAA8B;AAC5D,IAAA,MAAMqB,UAAAA,GAAa,IAAIqF,wBAAAA,CAAyB;MAC9Cb,MAAAA,EAAQ7F,WAAAA;MACR2G,cAAAA,EAAgB,KAAA;MAChBC,GAAAA,EAAK,UAAA;MACLC,WAAAA,EAAa;KACf,CAAA;AACAxF,IAAAA,UAAAA,CAAWuD,OAAAA,CAAQkC,EAAAA,CAAG,QAAA,EAAU,OAAOC,MAAMC,CAAAA,KAAAA;AAC3C,MAAA,MAAM,IAAA,CAAKhI,GAAAA,CAAI4F,OAAAA,CAAQC,IAAAA,CAAK,aAAA,EAAe;AACzCvG,QAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AACZqF,QAAAA,MAAAA,EAAQoD,IAAAA,CAAKpD,MAAAA;AACbsD,QAAAA,KAAAA,EAAOF,IAAAA,CAAKE,KAAAA;QACZC,gBAAAA,EAAkBnH;OACpB,CAAA;KACF,CAAA;AACA,IAAA,OAAOsB,UAAAA;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, ChatModelInput, 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\";\nimport { StreamToolCallMiddleware } from \"@/middleware/streamToolCall.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 streamMiddleware = this.createFinalAnswerStream(request.finalAnswer);\n    try {\n      const input = await this.prepareLLMRequest(request);\n      const response = await this.llm.create(input).middleware(streamMiddleware);\n\n      if (response.usage) {\n        mergeTokenUsage(this.state.usage, response.usage);\n      }\n\n      return response;\n    } finally {\n      streamMiddleware.unbind();\n    }\n  }\n\n  protected async prepareLLMRequest(request: RequirementAgentRequest): Promise<ChatModelInput> {\n    const messages: Message[] = [\n      await createSystemMessage(this.templates.system, request),\n      ...this.state.memory.messages,\n    ];\n\n    return {\n      messages,\n      tools: request.allowedTools,\n      toolChoice: request.toolChoice,\n      streamPartialToolCalls: true,\n      maxRetries: this.runConfig.maxRetriesPerStep ?? 0,\n    };\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      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    if (response.getToolCalls().length === 0) {\n      const textMessages = response.getTextMessages();\n      const text = textMessages.map((m) => m.text).join(\"\\n\");\n\n      const finalAnswerToolCall =\n        text && request.canStop ? await this.createFinalAnswerToolCall(text) : null;\n      if (finalAnswerToolCall) {\n        const stream = this.createFinalAnswerStream(request.finalAnswer);\n        await stream.add(new ChatModelOutput([finalAnswerToolCall]));\n      } else {\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        if (!request.canStop) {\n          return await this.runIteration(request);\n        }\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      response.messages.length = 0;\n      response.messages.push(finalAnswerToolCall);\n    }\n\n    // Check for cycles\n    const toolCalls = response.getToolCalls();\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([...response.messages, ...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  private createFinalAnswerStream(finalAnswer: FinalAnswerTool) {\n    const middleware = new StreamToolCallMiddleware({\n      target: finalAnswer,\n      forceStreaming: false,\n      key: \"response\",\n      matchNested: false,\n    });\n    middleware.emitter.on(\"update\", async (data, _) => {\n      await this.ctx.emitter.emit(\"finalAnswer\", {\n        state: this.state,\n        output: data.output,\n        delta: data.delta,\n        outputStructured: undefined,\n      });\n    });\n    return middleware;\n  }\n}\n"]}

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

@@ -3,17 +3,17 @@ import '../../backend/message.cjs';
 import '../../tools/base.cjs';
 import '../../errors.cjs';
 import '../../template.cjs';
-import '../../chat-CC0s8z23.cjs';
-import '../../base-XfWeXQ-S.cjs';
+import '../../chat-DaqV3woF.cjs';
+import '../../base-BqzN2QwB.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-D9pn62yF.cjs';
+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-YoNQSxQH.cjs';
 import '../../internals/serializable.cjs';
 import '../../internals/types.cjs';
 import '../../internals/helpers/guards.cjs';
 import 'ai';
 import 'ajv';
 import '../../context.cjs';
-import '../../emitter-jN55XZZq.cjs';
+import '../../emitter-UkIkNrek.cjs';
 import '../../internals/helpers/promise.cjs';
 import 'promise-based-task';
 import '../../cache/base.cjs';