beeai-framework 0.1.10 → 0.1.12

package/dist/agents/toolCalling/agent.cjs

@@ -41,6 +41,7 @@ class ToolCallingAgent extends base_cjs.BaseAgent {
       ],
       creator: this
     });
+    this.input.saveIntermediateSteps = this.input.saveIntermediateSteps ?? true;
   }
   static {
     this.register();
@@ -61,6 +62,7 @@ class ToolCallingAgent extends base_cjs.BaseAgent {
       role: void 0,
       instructions: void 0
     })));
+    await state.memory.addMany(this.memory.messages);
     if (input.prompt) {
       const userMessage = new message_cjs.UserMessage(this.templates.task.render({
         prompt: input.prompt,
@@ -70,6 +72,25 @@ class ToolCallingAgent extends base_cjs.BaseAgent {
       await state.memory.add(userMessage);
     }
     const globalRetriesCounter = new counter_cjs.RetryCounter(execution.totalMaxRetries || 1, base_cjs.AgentError);
+    const usePlainResponse = !input.expectedOutput || !(input.expectedOutput instanceof zod.ZodSchema);
+    const finalAnswerToolSchema = usePlainResponse ? zod.z.object({
+      response: zod.z.string().describe(String(input.expectedOutput ?? ""))
+    }) : input.expectedOutput;
+    const finalAnswerTool = new base_cjs$1.DynamicTool({
+      name: "final_answer",
+      description: "Sends the final answer to the user",
+      inputSchema: finalAnswerToolSchema,
+      handler: /* @__PURE__ */ __name(async (input2) => {
+        const result = usePlainResponse ? input2.response : JSON.stringify(input2.response);
+        state.result = new message_cjs.AssistantMessage(result);
+        return new base_cjs$1.StringToolOutput("Message has been sent");
+      }, "handler")
+    });
+    const tools = [
+      ...this.input.tools,
+      finalAnswerTool
+    ];
+    let forceFinalAnswer = false;
     while (!state.result) {
       state.iteration++;
       if (state.iteration > (execution.totalMaxRetries ?? Infinity)) {
@@ -80,14 +101,15 @@ class ToolCallingAgent extends base_cjs.BaseAgent {
       });
       const response = await this.input.llm.create({
         messages: state.memory.messages.slice(),
-        tools: this.input.tools,
+        tools,
+        toolChoice: forceFinalAnswer ? finalAnswerTool : tools.length > 1 ? "required" : tools[0],
         stream: false
       });
       await state.memory.addMany(response.messages);
       const toolCallMessages = response.getToolCalls();
       for (const toolCall of toolCallMessages) {
         try {
-          const tool = this.input.tools.find((tool2) => tool2.name === toolCall.toolName);
+          const tool = tools.find((tool2) => tool2.name === toolCall.toolName);
           if (!tool) {
             throw new base_cjs.AgentError(`Tool ${toolCall.toolName} does not exist!`);
           }
@@ -127,21 +149,20 @@ class ToolCallingAgent extends base_cjs.BaseAgent {
         await state.memory.deleteMany(state.memory.messages.filter((msg) => msg.meta[tempMessageKey]));
       }
       if (!remeda.isEmpty(textMessages) && remeda.isEmpty(toolCallMessages)) {
-        if (input.expectedOutput && input.expectedOutput instanceof zod.ZodSchema) {
-          const structured = await this.input.llm.createStructure({
-            schema: input.expectedOutput,
-            messages: state.memory.messages.slice()
-          });
-          state.result = new message_cjs.AssistantMessage(JSON.stringify(structured, null, 4));
-        } else {
-          state.result = message_cjs.AssistantMessage.fromChunks(textMessages);
-        }
+        forceFinalAnswer = true;
+        tools.length = 0;
+        tools.push(finalAnswerTool);
       }
       await run.emitter.emit("success", {
         state
       });
     }
-    await this.memory.addMany(state.memory.messages.slice(1));
+    if (this.input.saveIntermediateSteps) {
+      this.memory.reset();
+      await this.memory.addMany(state.memory.messages.slice(1));
+    } else {
+      await this.memory.addMany(state.memory.messages.slice(-2));
+    }
     return {
       memory: state.memory,
       result: state.result

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

@@ -1 +1 @@
-{"version":3,"sources":["agent.ts"],"names":["ToolCallingAgent","BaseAgent","emitter","constructor","input","Emitter","root","child","namespace","creator","register","_run","options","run","tempMessageKey","execution","totalMaxRetries","omitUndefined","state","memory","UnconstrainedMemory","result","undefined","iteration","add","SystemMessage","templates","system","render","role","instructions","prompt","userMessage","UserMessage","task","context","expectedOutput","isString","globalRetriesCounter","RetryCounter","AgentError","Infinity","emit","response","llm","create","messages","slice","tools","stream","addMany","toolCallMessages","getToolCalls","toolCall","tool","find","name","toolName","toolInput","args","toolResponse","toolCallMsg","ToolMessage","type","toolCallId","getTextContent","isError","e","ToolError","use","explain","textMessages","getTextMessages","isEmpty","AssistantMessage","deleteMany","filter","msg","meta","ZodSchema","structured","createStructure","schema","JSON","stringify","fromChunks","description","length","extraDescription","map","join","overrides","defaultTemplates","ToolCallingAgentSystemPrompt","ToolCallingAgentTaskPrompt","mapObj","key","defaultTemplate","override","PromptTemplate","createSnapshot","shallowCopy","enumerable"],"mappings":";;;;;;;;;;;;;;;;;;AAcC,SAAA,YAAA,CAAA,UAAA,EAAA,MAAA,EAAA,GAAA,EAAA,IAAA,EAAA;;;;;;AAAA,MAAA,CAAA,YAAA,EAAA,cAAA,CAAA;;;;;AAiDM,MAAMA,yBAAyBC,kBAAAA,CAAAA;EA/DtC;;;;AAoEkBC,EAAAA,OAAAA;AAKhBC,EAAAA,WAAAA,CAA4BC,KAA8B,EAAA;AACxD,IAAK,KAAA,EAAA,EAAA,KADqBA,KAAAA,GAAAA,KAAAA,EAAAA,KALZF,OAAUG,GAAAA,mBAAAA,CAAQC,KAAKC,KAAiC,CAAA;MACtEC,SAAW,EAAA;AAAC,QAAA,OAAA;AAAS,QAAA;;MACrBC,OAAS,EAAA;KACX,CAAA;AAIA;EAEA;AACE,IAAA,IAAA,CAAKC,QAAQ,EAAA;AACf;AAEA,EAAA,MAAgBC,IACdP,CAAAA,KAAAA,EACAQ,OAAsC,GAAA,IACtCC,GACoC,EAAA;AACpC,IAAA,MAAMC,cAAiB,GAAA,aAAA;AACvB,IAAA,MAAMC,SAAY,GAAA;MAEhBC,eAAiB,EAAA,EAAA;MAEjB,GAAGC,wBAAc,CAAA,IAAA,CAAKb,KAAMW,CAAAA,SAAAA,IAAa,EAAC,CAAA;AAC1C,MAAA,GAAGE,wBAAcL,CAAAA,OAAAA,CAAQG,SAAa,IAAA,EAAC;AACzC,KAAA;AAEA,IAAA,MAAMG,KAAkC,GAAA;AACtCC,MAAAA,MAAAA,EAAQ,IAAIC,2CAAAA,EAAAA;MACZC,MAAQC,EAAAA,MAAAA;MACRC,SAAW,EAAA;AACb,KAAA;AACA,IAAML,MAAAA,KAAAA,CAAMC,OAAOK,GACjB,CAAA,IAAIC,0BACF,IAAKC,CAAAA,SAAAA,CAAUC,OAAOC,MAAO,CAAA;MAC3BC,IAAMP,EAAAA,MAAAA;MACNQ,YAAcR,EAAAA;AAChB,KAAA,CAAA,CAAA,CAAA;AAIJ,IAAA,IAAIlB,MAAM2B,MAAQ,EAAA;AAChB,MAAA,MAAMC,cAAc,IAAIC,uBAAAA,CACtB,IAAKP,CAAAA,SAAAA,CAAUQ,KAAKN,MAAO,CAAA;AACzBG,QAAAA,MAAAA,EAAQ3B,KAAM2B,CAAAA,MAAAA;AACdI,QAAAA,OAAAA,EAAS/B,KAAM+B,CAAAA,OAAAA;AACfC,QAAAA,cAAAA,EAAgBC,eAASjC,CAAAA,KAAAA,CAAMgC,cAAc,CAAA,GAAIhC,MAAMgC,cAAiBd,GAAAA;AAC1E,OAAA,CAAA,CAAA;AAEF,MAAMJ,MAAAA,KAAAA,CAAMC,MAAOK,CAAAA,GAAAA,CAAIQ,WAAAA,CAAAA;AACzB;AAEA,IAAA,MAAMM,uBAAuB,IAAIC,wBAAAA,CAAaxB,SAAUC,CAAAA,eAAAA,IAAmB,GAAGwB,mBAAAA,CAAAA;AAE9E,IAAO,OAAA,CAACtB,MAAMG,MAAQ,EAAA;AACpBH,MAAMK,KAAAA,CAAAA,SAAAA,EAAAA;AACN,MAAA,IAAIL,KAAMK,CAAAA,SAAAA,IAAaR,SAAUC,CAAAA,eAAAA,IAAmByB,QAAW,CAAA,EAAA;AAC7D,QAAA,MAAM,IAAID,mBAAAA,CACR,CAA6CtB,0CAAAA,EAAAA,KAAAA,CAAMK,SAAS,CAAc,YAAA,CAAA,CAAA;AAE9E;AAEA,MAAMV,MAAAA,GAAAA,CAAIX,OAAQwC,CAAAA,IAAAA,CAAK,OAAS,EAAA;AAAExB,QAAAA;OAAM,CAAA;AACxC,MAAA,MAAMyB,QAAW,GAAA,MAAM,IAAKvC,CAAAA,KAAAA,CAAMwC,IAAIC,MAAO,CAAA;QAC3CC,QAAU5B,EAAAA,KAAAA,CAAMC,MAAO2B,CAAAA,QAAAA,CAASC,KAAK,EAAA;AACrCC,QAAAA,KAAAA,EAAO,KAAK5C,KAAM4C,CAAAA,KAAAA;QAClBC,MAAQ,EAAA;OACV,CAAA;AACA,MAAA,MAAM/B,KAAMC,CAAAA,MAAAA,CAAO+B,OAAQP,CAAAA,QAAAA,CAASG,QAAQ,CAAA;AAE5C,MAAMK,MAAAA,gBAAAA,GAAmBR,SAASS,YAAY,EAAA;AAC9C,MAAA,KAAA,MAAWC,YAAYF,gBAAkB,EAAA;AACvC,QAAI,IAAA;AACF,UAAMG,MAAAA,IAAAA,GAAO,IAAKlD,CAAAA,KAAAA,CAAM4C,KAAMO,CAAAA,IAAAA,CAAK,CAACD,KAASA,KAAAA,KAAAA,CAAKE,IAASH,KAAAA,QAAAA,CAASI,QAAQ,CAAA;AAC5E,UAAA,IAAI,CAACH,IAAM,EAAA;AACT,YAAA,MAAM,IAAId,mBAAAA,CAAW,CAAQa,KAAAA,EAAAA,QAAAA,CAASI,QAAQ,CAAkB,gBAAA,CAAA,CAAA;AAClE;AAEA,UAAA,MAAMC,YAAiBL,QAASM,CAAAA,IAAAA;AAChC,UAAA,MAAMC,eAA2B,MAAMN,IAAAA,CAAKzC,GAAI6C,CAAAA,SAAAA,EAAWvB,OAAQ,CAAA;AACjEjB,YAAAA,KAAAA;YACA2C,WAAaR,EAAAA;WACf,CAAA;AACA,UAAA,MAAMnC,KAAMC,CAAAA,MAAAA,CAAOK,GACjB,CAAA,IAAIsC,uBAAY,CAAA;YACdC,IAAM,EAAA,aAAA;AACNC,YAAAA,UAAAA,EAAYX,QAASW,CAAAA,UAAAA;AACrBP,YAAAA,QAAAA,EAAUJ,QAASI,CAAAA,QAAAA;AACnBpC,YAAAA,MAAAA,EAAQuC,aAAaK,cAAc,EAAA;YACnCC,OAAS,EAAA;AACX,WAAA,CAAA,CAAA;AAEJ,SAAA,CAAA,OAASC,CAAG,EAAA;AACV,UAAA,IAAIA,aAAaC,oBAAW,EAAA;AAC1B9B,YAAAA,oBAAAA,CAAqB+B,IAAIF,CAAAA,CAAAA;AACzB,YAAA,MAAMjD,KAAMC,CAAAA,MAAAA,CAAOK,GACjB,CAAA,IAAIsC,uBAAY,CAAA;cACdC,IAAM,EAAA,aAAA;AACNC,cAAAA,UAAAA,EAAYX,QAASW,CAAAA,UAAAA;AACrBP,cAAAA,QAAAA,EAAUJ,QAASI,CAAAA,QAAAA;AACnBpC,cAAAA,MAAAA,EAAQ8C,EAAEG,OAAO,EAAA;cACjBJ,OAAS,EAAA;AACX,aAAA,CAAA,CAAA;WAEG,MAAA;AACL,YAAMC,MAAAA,CAAAA;AACR;AACF;AACF;AAGA,MAAMI,MAAAA,YAAAA,GAAe5B,SAAS6B,eAAe,EAAA;AAC7C,MAAA,IAAIC,cAAQtB,CAAAA,gBAAAA,CAAqBsB,IAAAA,cAAAA,CAAQF,YAAAA,CAAe,EAAA;AACtD,QAAA,MAAMrD,KAAMC,CAAAA,MAAAA,CAAOK,GAAI,CAAA,IAAIkD,6BAAiB,IAAM,EAAA;AAAE,UAAA,CAAC5D,cAAAA,GAAiB;AAAK,SAAA,CAAA,CAAA;OACtE,MAAA;AACL,QAAA,MAAMI,KAAMC,CAAAA,MAAAA,CAAOwD,UACjBzD,CAAAA,KAAAA,CAAMC,MAAO2B,CAAAA,QAAAA,CAAS8B,MAAO,CAAA,CAACC,GAAQA,KAAAA,GAAAA,CAAIC,IAAKhE,CAAAA,cAAAA,CAAe,CAAA,CAAA;AAElE;AAEA,MAAA,IAAI,CAAC2D,cAAQF,CAAAA,YAAAA,CAAiBE,IAAAA,cAAAA,CAAQtB,gBAAAA,CAAmB,EAAA;AACvD,QAAA,IAAI/C,KAAMgC,CAAAA,cAAAA,IAAkBhC,KAAMgC,CAAAA,cAAAA,YAA0B2C,aAAW,EAAA;AACrE,UAAA,MAAMC,UAAa,GAAA,MAAM,IAAK5E,CAAAA,KAAAA,CAAMwC,IAAIqC,eAAgB,CAAA;AACtDC,YAAAA,MAAAA,EAAQ9E,KAAMgC,CAAAA,cAAAA;YACdU,QAAU5B,EAAAA,KAAAA,CAAMC,MAAO2B,CAAAA,QAAAA,CAASC,KAAK;WACvC,CAAA;AACA7B,UAAMG,KAAAA,CAAAA,MAAAA,GAAS,IAAIqD,4BAAiBS,CAAAA,IAAAA,CAAKC,UAAUJ,UAAY,EAAA,IAAA,EAAM,CAAA,CAAA,CAAA;SAChE,MAAA;AACL9D,UAAMG,KAAAA,CAAAA,MAAAA,GAASqD,4BAAiBW,CAAAA,UAAAA,CAAWd,YAAAA,CAAAA;AAC7C;AACF;AAEA,MAAM1D,MAAAA,GAAAA,CAAIX,OAAQwC,CAAAA,IAAAA,CAAK,SAAW,EAAA;AAAExB,QAAAA;OAAM,CAAA;AAC5C;AAEA,IAAM,MAAA,IAAA,CAAKC,OAAO+B,OAAQhC,CAAAA,KAAAA,CAAMC,OAAO2B,QAASC,CAAAA,KAAAA,CAAM,CAAA,CAAA,CAAA;AACtD,IAAO,OAAA;AAAE5B,MAAAA,MAAAA,EAAQD,KAAMC,CAAAA,MAAAA;AAAQE,MAAAA,MAAAA,EAAQH,KAAMG,CAAAA;AAAO,KAAA;AACtD;AAEA,EAAA,IAAIyD,IAAkB,GAAA;AACpB,IAAA,MAAM9B,KAAQ,GAAA,IAAA,CAAK5C,KAAM4C,CAAAA,KAAAA,CAAMD,KAAK,EAAA;AAEpC,IAAI,IAAA,IAAA,CAAK3C,MAAM0E,IAAM,EAAA;AACnB,MAAO,OAAA;AAAE,QAAA,GAAG,KAAK1E,KAAM0E,CAAAA,IAAAA;AAAM9B,QAAAA;AAAM,OAAA;AACrC;AAEA,IAAO,OAAA;MACLQ,IAAM,EAAA,aAAA;AACNR,MAAAA,KAAAA;MACAsC,WAAa,EAAA,0DAAA;MACb,GAAItC,KAAAA,CAAMuC,SAAS,CAAK,IAAA;QACtBC,gBAAkB,EAAA;AAChB,UAAA,CAAA,8CAAA,CAAA;aACGxC,KAAMyC,CAAAA,GAAAA,CAAI,CAACnC,IAAS,KAAA,CAAA,MAAA,EAASA,KAAKE,IAAI,CAAA,GAAA,EAAMF,IAAKgC,CAAAA,WAAW,CAAG,CAAA,CAAA;AAClEI,SAAAA,CAAAA,IAAAA,CAAK,IAAA;AACT;AACF,KAAA;AACF;AAEA,EAAA,IACchE,SAAuC,GAAA;AACnD,IAAA,MAAMiE,SAAY,GAAA,IAAA,CAAKvF,KAAMsB,CAAAA,SAAAA,IAAa,EAAC;AAC3C,IAAA,MAAMkE,gBAA8C,GAAA;MAClDjE,MAAQkE,EAAAA,wCAAAA;MACR3D,IAAM4D,EAAAA;AACR,KAAA;AAEA,IAAA,OAAOC,iBAAOH,CAAAA,gBAAAA,CACZ,CAAA,CAACI,KAAKC,eAAAA,KAAAA;AACJ,MAAMC,MAAAA,QAAAA,GAAWP,SAAUK,CAAAA,GAAAA,CAAQC,IAAAA,eAAAA;AACnC,MAAA,IAAIC,oBAAoBC,2BAAgB,EAAA;AACtC,QAAOD,OAAAA,QAAAA;AACT;AACA,MAAOA,OAAAA,QAAAA,CAASD,eAAAA,CAAoBA,IAAAA,eAAAA;KACtC,CAAA;AAEJ;EAEAG,cAAiB,GAAA;AACf,IAAO,OAAA;AACL,MAAA,GAAG,MAAMA,cAAAA,EAAAA;MACThG,KAAOiG,EAAAA,qBAAAA,CAAY,KAAKjG,KAAK,CAAA;AAC7BF,MAAAA,OAAAA,EAAS,IAAKA,CAAAA;AAChB,KAAA;AACF;AAEA,EAAA,IAAIiB,OAAOA,MAAoB,EAAA;AAC7B,IAAA,IAAA,CAAKf,MAAMe,MAASA,GAAAA,MAAAA;AACtB;AAEA,EAAA,IAAIA,MAAS,GAAA;AACX,IAAA,OAAO,KAAKf,KAAMe,CAAAA,MAAAA;AACpB;AACF;;;IAlCWmF,UAAY,EAAA","file":"agent.cjs","sourcesContent":["/**\n * Copyright 2025 IBM Corp.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { AgentError, BaseAgent } from \"@/agents/base.js\";\nimport { AnyTool, ToolError, ToolOutput } from \"@/tools/base.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\nimport { AgentMeta } from \"@/agents/types.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport type {\n  ToolCallingAgentExecutionConfig,\n  ToolCallingAgentTemplates,\n  ToolCallingAgentCallbacks,\n  ToolCallingAgentRunInput,\n  ToolCallingAgentRunOptions,\n  ToolCallingAgentRunOutput,\n  ToolCallingAgentRunState,\n} from \"@/agents/toolCalling/types.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { ChatModel } from \"@/backend/chat.js\";\nimport { shallowCopy } from \"@/serializer/utils.js\";\nimport { UnconstrainedMemory } from \"@/memory/unconstrainedMemory.js\";\nimport { AssistantMessage, SystemMessage, ToolMessage, UserMessage } from \"@/backend/message.js\";\nimport { isEmpty, isString } from \"remeda\";\nimport { RetryCounter } from \"@/internals/helpers/counter.js\";\nimport { mapObj, omitUndefined } from \"@/internals/helpers/object.js\";\nimport { Cache } from \"@/cache/decoratorCache.js\";\nimport { PromptTemplate } from \"@/template.js\";\nimport {\n  ToolCallingAgentSystemPrompt,\n  ToolCallingAgentTaskPrompt,\n} from \"@/agents/toolCalling/prompts.js\";\nimport { ZodSchema } from \"zod\";\n\nexport type ToolCallingAgentTemplateFactory<K extends keyof ToolCallingAgentTemplates> = (\n  template: ToolCallingAgentTemplates[K],\n) => ToolCallingAgentTemplates[K];\n\nexport interface ToolCallingAgentInput {\n  llm: ChatModel;\n  memory: BaseMemory;\n  tools: AnyTool[];\n  meta?: Omit<AgentMeta, \"tools\">;\n  templates?: Partial<{\n    [K in keyof ToolCallingAgentTemplates]:\n      | ToolCallingAgentTemplates[K]\n      | ToolCallingAgentTemplateFactory<K>;\n  }>;\n  execution?: ToolCallingAgentExecutionConfig;\n}\n\nexport class ToolCallingAgent extends BaseAgent<\n  ToolCallingAgentRunInput,\n  ToolCallingAgentRunOutput,\n  ToolCallingAgentRunOptions\n> {\n  public readonly emitter = Emitter.root.child<ToolCallingAgentCallbacks>({\n    namespace: [\"agent\", \"toolCalling\"],\n    creator: this,\n  });\n\n  constructor(public readonly input: ToolCallingAgentInput) {\n    super();\n  }\n\n  static {\n    this.register();\n  }\n\n  protected async _run(\n    input: ToolCallingAgentRunInput,\n    options: ToolCallingAgentRunOptions = {},\n    run: GetRunContext<typeof this>,\n  ): Promise<ToolCallingAgentRunOutput> {\n    const tempMessageKey = \"tempMessage\" as const;\n    const execution = {\n      maxRetriesPerStep: 3,\n      totalMaxRetries: 20,\n      maxIterations: 10,\n      ...omitUndefined(this.input.execution ?? {}),\n      ...omitUndefined(options.execution ?? {}),\n    };\n\n    const state: ToolCallingAgentRunState = {\n      memory: new UnconstrainedMemory(),\n      result: undefined,\n      iteration: 0,\n    };\n    await state.memory.add(\n      new SystemMessage(\n        this.templates.system.render({\n          role: undefined,\n          instructions: undefined,\n        }),\n      ),\n    );\n\n    if (input.prompt) {\n      const userMessage = new UserMessage(\n        this.templates.task.render({\n          prompt: input.prompt,\n          context: input.context,\n          expectedOutput: isString(input.expectedOutput) ? input.expectedOutput : undefined,\n        }),\n      );\n      await state.memory.add(userMessage);\n    }\n\n    const globalRetriesCounter = new RetryCounter(execution.totalMaxRetries || 1, AgentError);\n\n    while (!state.result) {\n      state.iteration++;\n      if (state.iteration > (execution.totalMaxRetries ?? Infinity)) {\n        throw new AgentError(\n          `Agent was not able to resolve the task in ${state.iteration} iterations.`,\n        );\n      }\n\n      await run.emitter.emit(\"start\", { state });\n      const response = await this.input.llm.create({\n        messages: state.memory.messages.slice(),\n        tools: this.input.tools,\n        stream: false,\n      });\n      await state.memory.addMany(response.messages);\n\n      const toolCallMessages = response.getToolCalls();\n      for (const toolCall of toolCallMessages) {\n        try {\n          const tool = this.input.tools.find((tool) => tool.name === toolCall.toolName);\n          if (!tool) {\n            throw new AgentError(`Tool ${toolCall.toolName} does not exist!`);\n          }\n\n          const toolInput: any = toolCall.args;\n          const toolResponse: ToolOutput = await tool.run(toolInput).context({\n            state,\n            toolCallMsg: toolCall,\n          });\n          await state.memory.add(\n            new ToolMessage({\n              type: \"tool-result\",\n              toolCallId: toolCall.toolCallId,\n              toolName: toolCall.toolName,\n              result: toolResponse.getTextContent(),\n              isError: false,\n            }),\n          );\n        } catch (e) {\n          if (e instanceof ToolError) {\n            globalRetriesCounter.use(e);\n            await state.memory.add(\n              new ToolMessage({\n                type: \"tool-result\",\n                toolCallId: toolCall.toolCallId,\n                toolName: toolCall.toolName,\n                result: e.explain(),\n                isError: true,\n              }),\n            );\n          } else {\n            throw e;\n          }\n        }\n      }\n\n      // handle empty messages for some models\n      const textMessages = response.getTextMessages();\n      if (isEmpty(toolCallMessages) && isEmpty(textMessages)) {\n        await state.memory.add(new AssistantMessage(\"\\n\", { [tempMessageKey]: true }));\n      } else {\n        await state.memory.deleteMany(\n          state.memory.messages.filter((msg) => msg.meta[tempMessageKey]),\n        );\n      }\n\n      if (!isEmpty(textMessages) && isEmpty(toolCallMessages)) {\n        if (input.expectedOutput && input.expectedOutput instanceof ZodSchema) {\n          const structured = await this.input.llm.createStructure({\n            schema: input.expectedOutput,\n            messages: state.memory.messages.slice(),\n          });\n          state.result = new AssistantMessage(JSON.stringify(structured, null, 4));\n        } else {\n          state.result = AssistantMessage.fromChunks(textMessages);\n        }\n      }\n\n      await run.emitter.emit(\"success\", { state });\n    }\n\n    await this.memory.addMany(state.memory.messages.slice(1));\n    return { memory: state.memory, result: state.result };\n  }\n\n  get meta(): AgentMeta {\n    const tools = this.input.tools.slice();\n\n    if (this.input.meta) {\n      return { ...this.input.meta, tools };\n    }\n\n    return {\n      name: \"ToolCalling\",\n      tools,\n      description: \"ToolCallingAgent that uses tools to accomplish the task.\",\n      ...(tools.length > 0 && {\n        extraDescription: [\n          `Tools that I can use to accomplish given task.`,\n          ...tools.map((tool) => `Tool '${tool.name}': ${tool.description}.`),\n        ].join(\"\\n\"),\n      }),\n    };\n  }\n\n  @Cache({ enumerable: false })\n  protected get templates(): ToolCallingAgentTemplates {\n    const overrides = this.input.templates ?? {};\n    const defaultTemplates: ToolCallingAgentTemplates = {\n      system: ToolCallingAgentSystemPrompt,\n      task: ToolCallingAgentTaskPrompt,\n    } as const;\n\n    return mapObj(defaultTemplates)(\n      (key, defaultTemplate: ToolCallingAgentTemplates[typeof key]) => {\n        const override = overrides[key] ?? defaultTemplate;\n        if (override instanceof PromptTemplate) {\n          return override;\n        }\n        return override(defaultTemplate) ?? defaultTemplate;\n      },\n    );\n  }\n\n  createSnapshot() {\n    return {\n      ...super.createSnapshot(),\n      input: shallowCopy(this.input),\n      emitter: this.emitter,\n    };\n  }\n\n  set memory(memory: BaseMemory) {\n    this.input.memory = memory;\n  }\n\n  get memory() {\n    return this.input.memory;\n  }\n}\n"]}
+{"version":3,"sources":["agent.ts"],"names":["ToolCallingAgent","BaseAgent","emitter","constructor","input","Emitter","root","child","namespace","creator","saveIntermediateSteps","register","_run","options","run","tempMessageKey","execution","totalMaxRetries","omitUndefined","state","memory","UnconstrainedMemory","result","undefined","iteration","add","SystemMessage","templates","system","render","role","instructions","addMany","messages","prompt","userMessage","UserMessage","task","context","expectedOutput","isString","globalRetriesCounter","RetryCounter","AgentError","usePlainResponse","ZodSchema","finalAnswerToolSchema","z","object","response","string","describe","String","finalAnswerTool","DynamicTool","name","description","inputSchema","handler","JSON","stringify","AssistantMessage","StringToolOutput","tools","forceFinalAnswer","Infinity","emit","llm","create","slice","toolChoice","length","stream","toolCallMessages","getToolCalls","toolCall","tool","find","toolName","toolInput","args","toolResponse","toolCallMsg","ToolMessage","type","toolCallId","getTextContent","isError","e","ToolError","use","explain","textMessages","getTextMessages","isEmpty","deleteMany","filter","msg","meta","push","reset","extraDescription","map","join","overrides","defaultTemplates","ToolCallingAgentSystemPrompt","ToolCallingAgentTaskPrompt","mapObj","key","defaultTemplate","override","PromptTemplate","createSnapshot","shallowCopy","enumerable"],"mappings":";;;;;;;;;;;;;;;;;;AAcC,SAAA,YAAA,CAAA,UAAA,EAAA,MAAA,EAAA,GAAA,EAAA,IAAA,EAAA;;;;;;AAAA,MAAA,CAAA,YAAA,EAAA,cAAA,CAAA;;;;;AAkDM,MAAMA,yBAAyBC,kBAAAA,CAAAA;EAhEtC;;;;AAqEkBC,EAAAA,OAAAA;AAKhBC,EAAAA,WAAAA,CAA4BC,KAA8B,EAAA;AACxD,IAAK,KAAA,EAAA,EAAA,KADqBA,KAAAA,GAAAA,KAAAA,EAAAA,KALZF,OAAUG,GAAAA,mBAAAA,CAAQC,KAAKC,KAAiC,CAAA;MACtEC,SAAW,EAAA;AAAC,QAAA,OAAA;AAAS,QAAA;;MACrBC,OAAS,EAAA;KACX,CAAA;AAIE,IAAA,IAAA,CAAKL,KAAMM,CAAAA,qBAAAA,GAAwB,IAAKN,CAAAA,KAAAA,CAAMM,qBAAyB,IAAA,IAAA;AACzE;EAEA;AACE,IAAA,IAAA,CAAKC,QAAQ,EAAA;AACf;AAEA,EAAA,MAAgBC,IACdR,CAAAA,KAAAA,EACAS,OAAsC,GAAA,IACtCC,GACoC,EAAA;AACpC,IAAA,MAAMC,cAAiB,GAAA,aAAA;AACvB,IAAA,MAAMC,SAAY,GAAA;MAEhBC,eAAiB,EAAA,EAAA;MAEjB,GAAGC,wBAAc,CAAA,IAAA,CAAKd,KAAMY,CAAAA,SAAAA,IAAa,EAAC,CAAA;AAC1C,MAAA,GAAGE,wBAAcL,CAAAA,OAAAA,CAAQG,SAAa,IAAA,EAAC;AACzC,KAAA;AAEA,IAAA,MAAMG,KAAkC,GAAA;AACtCC,MAAAA,MAAAA,EAAQ,IAAIC,2CAAAA,EAAAA;MACZC,MAAQC,EAAAA,MAAAA;MACRC,SAAW,EAAA;AACb,KAAA;AACA,IAAML,MAAAA,KAAAA,CAAMC,OAAOK,GACjB,CAAA,IAAIC,0BACF,IAAKC,CAAAA,SAAAA,CAAUC,OAAOC,MAAO,CAAA;MAC3BC,IAAMP,EAAAA,MAAAA;MACNQ,YAAcR,EAAAA;AAChB,KAAA,CAAA,CAAA,CAAA;AAGJ,IAAA,MAAMJ,KAAMC,CAAAA,MAAAA,CAAOY,OAAQ,CAAA,IAAA,CAAKZ,OAAOa,QAAQ,CAAA;AAE/C,IAAA,IAAI7B,MAAM8B,MAAQ,EAAA;AAChB,MAAA,MAAMC,cAAc,IAAIC,uBAAAA,CACtB,IAAKT,CAAAA,SAAAA,CAAUU,KAAKR,MAAO,CAAA;AACzBK,QAAAA,MAAAA,EAAQ9B,KAAM8B,CAAAA,MAAAA;AACdI,QAAAA,OAAAA,EAASlC,KAAMkC,CAAAA,OAAAA;AACfC,QAAAA,cAAAA,EAAgBC,eAASpC,CAAAA,KAAAA,CAAMmC,cAAc,CAAA,GAAInC,MAAMmC,cAAiBhB,GAAAA;AAC1E,OAAA,CAAA,CAAA;AAEF,MAAMJ,MAAAA,KAAAA,CAAMC,MAAOK,CAAAA,GAAAA,CAAIU,WAAAA,CAAAA;AACzB;AAEA,IAAA,MAAMM,uBAAuB,IAAIC,wBAAAA,CAAa1B,SAAUC,CAAAA,eAAAA,IAAmB,GAAG0B,mBAAAA,CAAAA;AAE9E,IAAA,MAAMC,mBAAmB,CAACxC,KAAAA,CAAMmC,cAAkB,IAAA,EAAEnC,MAAMmC,cAA0BM,YAAAA,aAAAA,CAAAA;AACpF,IAAMC,MAAAA,qBAAAA,GAAwBF,gBAC1BG,GAAAA,KAAAA,CAAEC,MAAO,CAAA;MACPC,QAAUF,EAAAA,KAAAA,CAAEG,QAASC,CAAAA,QAAAA,CAASC,OAAOhD,KAAMmC,CAAAA,cAAAA,IAAkB,EAAA,CAAA;AAC/D,KAAA,IACCnC,KAAMmC,CAAAA,cAAAA;AAEX,IAAMc,MAAAA,eAAAA,GAAkB,IAAIC,sBAAY,CAAA;MACtCC,IAAM,EAAA,cAAA;MACNC,WAAa,EAAA,oCAAA;MACbC,WAAaX,EAAAA,qBAAAA;AACbY,MAAAA,OAAAA,gCAAgBtD,MAAAA,KAAAA;AACd,QAAA,MAAMkB,SAASsB,gBAAmBxC,GAAAA,MAAAA,CAAM6C,WAAWU,IAAKC,CAAAA,SAAAA,CAAUxD,OAAM6C,QAAQ,CAAA;AAChF9B,QAAMG,KAAAA,CAAAA,MAAAA,GAAS,IAAIuC,4BAAAA,CAAiBvC,MAAAA,CAAAA;AACpC,QAAO,OAAA,IAAIwC,4BAAiB,uBAAA,CAAA;OAHrB,EAAA,SAAA;KAKX,CAAA;AAEA,IAAA,MAAMC,KAAQ,GAAA;AAAI,MAAA,GAAA,IAAA,CAAK3D,KAAM2D,CAAAA,KAAAA;AAAOV,MAAAA;;AACpC,IAAA,IAAIW,gBAAmB,GAAA,KAAA;AAEvB,IAAO,OAAA,CAAC7C,MAAMG,MAAQ,EAAA;AACpBH,MAAMK,KAAAA,CAAAA,SAAAA,EAAAA;AACN,MAAA,IAAIL,KAAMK,CAAAA,SAAAA,IAAaR,SAAUC,CAAAA,eAAAA,IAAmBgD,QAAW,CAAA,EAAA;AAC7D,QAAA,MAAM,IAAItB,mBAAAA,CACR,CAA6CxB,0CAAAA,EAAAA,KAAAA,CAAMK,SAAS,CAAc,YAAA,CAAA,CAAA;AAE9E;AAEA,MAAMV,MAAAA,GAAAA,CAAIZ,OAAQgE,CAAAA,IAAAA,CAAK,OAAS,EAAA;AAAE/C,QAAAA;OAAM,CAAA;AACxC,MAAA,MAAM8B,QAAW,GAAA,MAAM,IAAK7C,CAAAA,KAAAA,CAAM+D,IAAIC,MAAO,CAAA;QAC3CnC,QAAUd,EAAAA,KAAAA,CAAMC,MAAOa,CAAAA,QAAAA,CAASoC,KAAK,EAAA;AACrCN,QAAAA,KAAAA;AACAO,QAAAA,UAAAA,EAAYN,mBAAmBX,eAAkBU,GAAAA,KAAAA,CAAMQ,SAAS,CAAI,GAAA,UAAA,GAAaR,MAAM,CAAA,CAAA;QACvFS,MAAQ,EAAA;OACV,CAAA;AACA,MAAA,MAAMrD,KAAMC,CAAAA,MAAAA,CAAOY,OAAQiB,CAAAA,QAAAA,CAAShB,QAAQ,CAAA;AAE5C,MAAMwC,MAAAA,gBAAAA,GAAmBxB,SAASyB,YAAY,EAAA;AAC9C,MAAA,KAAA,MAAWC,YAAYF,gBAAkB,EAAA;AACvC,QAAI,IAAA;AACF,UAAMG,MAAAA,IAAAA,GAAOb,MAAMc,IAAK,CAAA,CAACD,UAASA,KAAKrB,CAAAA,IAAAA,KAASoB,SAASG,QAAQ,CAAA;AACjE,UAAA,IAAI,CAACF,IAAM,EAAA;AACT,YAAA,MAAM,IAAIjC,mBAAAA,CAAW,CAAQgC,KAAAA,EAAAA,QAAAA,CAASG,QAAQ,CAAkB,gBAAA,CAAA,CAAA;AAClE;AAEA,UAAA,MAAMC,YAAiBJ,QAASK,CAAAA,IAAAA;AAChC,UAAA,MAAMC,eAA2B,MAAML,IAAAA,CAAK9D,GAAIiE,CAAAA,SAAAA,EAAWzC,OAAQ,CAAA;AACjEnB,YAAAA,KAAAA;YACA+D,WAAaP,EAAAA;WACf,CAAA;AACA,UAAA,MAAMxD,KAAMC,CAAAA,MAAAA,CAAOK,GACjB,CAAA,IAAI0D,uBAAY,CAAA;YACdC,IAAM,EAAA,aAAA;AACNC,YAAAA,UAAAA,EAAYV,QAASU,CAAAA,UAAAA;AACrBP,YAAAA,QAAAA,EAAUH,QAASG,CAAAA,QAAAA;AACnBxD,YAAAA,MAAAA,EAAQ2D,aAAaK,cAAc,EAAA;YACnCC,OAAS,EAAA;AACX,WAAA,CAAA,CAAA;AAEJ,SAAA,CAAA,OAASC,CAAG,EAAA;AACV,UAAA,IAAIA,aAAaC,oBAAW,EAAA;AAC1BhD,YAAAA,oBAAAA,CAAqBiD,IAAIF,CAAAA,CAAAA;AACzB,YAAA,MAAMrE,KAAMC,CAAAA,MAAAA,CAAOK,GACjB,CAAA,IAAI0D,uBAAY,CAAA;cACdC,IAAM,EAAA,aAAA;AACNC,cAAAA,UAAAA,EAAYV,QAASU,CAAAA,UAAAA;AACrBP,cAAAA,QAAAA,EAAUH,QAASG,CAAAA,QAAAA;AACnBxD,cAAAA,MAAAA,EAAQkE,EAAEG,OAAO,EAAA;cACjBJ,OAAS,EAAA;AACX,aAAA,CAAA,CAAA;WAEG,MAAA;AACL,YAAMC,MAAAA,CAAAA;AACR;AACF;AACF;AAGA,MAAMI,MAAAA,YAAAA,GAAe3C,SAAS4C,eAAe,EAAA;AAC7C,MAAA,IAAIC,cAAQrB,CAAAA,gBAAAA,CAAqBqB,IAAAA,cAAAA,CAAQF,YAAAA,CAAe,EAAA;AACtD,QAAA,MAAMzE,KAAMC,CAAAA,MAAAA,CAAOK,GAAI,CAAA,IAAIoC,6BAAiB,IAAM,EAAA;AAAE,UAAA,CAAC9C,cAAAA,GAAiB;AAAK,SAAA,CAAA,CAAA;OACtE,MAAA;AACL,QAAA,MAAMI,KAAMC,CAAAA,MAAAA,CAAO2E,UACjB5E,CAAAA,KAAAA,CAAMC,MAAOa,CAAAA,QAAAA,CAAS+D,MAAO,CAAA,CAACC,GAAQA,KAAAA,GAAAA,CAAIC,IAAKnF,CAAAA,cAAAA,CAAe,CAAA,CAAA;AAElE;AAGA,MAAA,IAAI,CAAC+E,cAAQF,CAAAA,YAAAA,CAAiBE,IAAAA,cAAAA,CAAQrB,gBAAAA,CAAmB,EAAA;AACvDT,QAAmB,gBAAA,GAAA,IAAA;AACnBD,QAAAA,KAAAA,CAAMQ,MAAS,GAAA,CAAA;AACfR,QAAAA,KAAAA,CAAMoC,KAAK9C,eAAAA,CAAAA;AACb;AAEA,MAAMvC,MAAAA,GAAAA,CAAIZ,OAAQgE,CAAAA,IAAAA,CAAK,SAAW,EAAA;AAAE/C,QAAAA;OAAM,CAAA;AAC5C;AAEA,IAAI,IAAA,IAAA,CAAKf,MAAMM,qBAAuB,EAAA;AACpC,MAAA,IAAA,CAAKU,OAAOgF,KAAK,EAAA;AACjB,MAAM,MAAA,IAAA,CAAKhF,OAAOY,OAAQb,CAAAA,KAAAA,CAAMC,OAAOa,QAASoC,CAAAA,KAAAA,CAAM,CAAA,CAAA,CAAA;KACjD,MAAA;AACL,MAAM,MAAA,IAAA,CAAKjD,OAAOY,OAAQb,CAAAA,KAAAA,CAAMC,OAAOa,QAASoC,CAAAA,KAAAA,CAAM,EAAC,CAAA,CAAA;AACzD;AACA,IAAO,OAAA;AAAEjD,MAAAA,MAAAA,EAAQD,KAAMC,CAAAA,MAAAA;AAAQE,MAAAA,MAAAA,EAAQH,KAAMG,CAAAA;AAAO,KAAA;AACtD;AAEA,EAAA,IAAI4E,IAAkB,GAAA;AACpB,IAAA,MAAMnC,KAAQ,GAAA,IAAA,CAAK3D,KAAM2D,CAAAA,KAAAA,CAAMM,KAAK,EAAA;AAEpC,IAAI,IAAA,IAAA,CAAKjE,MAAM8F,IAAM,EAAA;AACnB,MAAO,OAAA;AAAE,QAAA,GAAG,KAAK9F,KAAM8F,CAAAA,IAAAA;AAAMnC,QAAAA;AAAM,OAAA;AACrC;AAEA,IAAO,OAAA;MACLR,IAAM,EAAA,aAAA;AACNQ,MAAAA,KAAAA;MACAP,WAAa,EAAA,0DAAA;MACb,GAAIO,KAAAA,CAAMQ,SAAS,CAAK,IAAA;QACtB8B,gBAAkB,EAAA;AAChB,UAAA,CAAA,8CAAA,CAAA;aACGtC,KAAMuC,CAAAA,GAAAA,CAAI,CAAC1B,IAAS,KAAA,CAAA,MAAA,EAASA,KAAKrB,IAAI,CAAA,GAAA,EAAMqB,IAAKpB,CAAAA,WAAW,CAAG,CAAA,CAAA;AAClE+C,SAAAA,CAAAA,IAAAA,CAAK,IAAA;AACT;AACF,KAAA;AACF;AAEA,EAAA,IACc5E,SAAuC,GAAA;AACnD,IAAA,MAAM6E,SAAY,GAAA,IAAA,CAAKpG,KAAMuB,CAAAA,SAAAA,IAAa,EAAC;AAC3C,IAAA,MAAM8E,gBAA8C,GAAA;MAClD7E,MAAQ8E,EAAAA,wCAAAA;MACRrE,IAAMsE,EAAAA;AACR,KAAA;AAEA,IAAA,OAAOC,iBAAOH,CAAAA,gBAAAA,CACZ,CAAA,CAACI,KAAKC,eAAAA,KAAAA;AACJ,MAAMC,MAAAA,QAAAA,GAAWP,SAAUK,CAAAA,GAAAA,CAAQC,IAAAA,eAAAA;AACnC,MAAA,IAAIC,oBAAoBC,2BAAgB,EAAA;AACtC,QAAOD,OAAAA,QAAAA;AACT;AACA,MAAOA,OAAAA,QAAAA,CAASD,eAAAA,CAAoBA,IAAAA,eAAAA;KACtC,CAAA;AAEJ;EAEAG,cAAiB,GAAA;AACf,IAAO,OAAA;AACL,MAAA,GAAG,MAAMA,cAAAA,EAAAA;MACT7G,KAAO8G,EAAAA,qBAAAA,CAAY,KAAK9G,KAAK,CAAA;AAC7BF,MAAAA,OAAAA,EAAS,IAAKA,CAAAA;AAChB,KAAA;AACF;AAEA,EAAA,IAAIkB,OAAOA,MAAoB,EAAA;AAC7B,IAAA,IAAA,CAAKhB,MAAMgB,MAASA,GAAAA,MAAAA;AACtB;AAEA,EAAA,IAAIA,MAAS,GAAA;AACX,IAAA,OAAO,KAAKhB,KAAMgB,CAAAA,MAAAA;AACpB;AACF;;;IAlCW+F,UAAY,EAAA","file":"agent.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { AgentError, BaseAgent } from \"@/agents/base.js\";\nimport { AnyTool, DynamicTool, StringToolOutput, ToolError, ToolOutput } from \"@/tools/base.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\nimport { AgentMeta } from \"@/agents/types.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport type {\n  ToolCallingAgentExecutionConfig,\n  ToolCallingAgentTemplates,\n  ToolCallingAgentCallbacks,\n  ToolCallingAgentRunInput,\n  ToolCallingAgentRunOptions,\n  ToolCallingAgentRunOutput,\n  ToolCallingAgentRunState,\n} from \"@/agents/toolCalling/types.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { ChatModel } from \"@/backend/chat.js\";\nimport { shallowCopy } from \"@/serializer/utils.js\";\nimport { UnconstrainedMemory } from \"@/memory/unconstrainedMemory.js\";\nimport { AssistantMessage, SystemMessage, ToolMessage, UserMessage } from \"@/backend/message.js\";\nimport { isEmpty, isString } from \"remeda\";\nimport { RetryCounter } from \"@/internals/helpers/counter.js\";\nimport { mapObj, omitUndefined } from \"@/internals/helpers/object.js\";\nimport { Cache } from \"@/cache/decoratorCache.js\";\nimport { PromptTemplate } from \"@/template.js\";\nimport {\n  ToolCallingAgentSystemPrompt,\n  ToolCallingAgentTaskPrompt,\n} from \"@/agents/toolCalling/prompts.js\";\nimport { z, ZodSchema } from \"zod\";\n\nexport type ToolCallingAgentTemplateFactory<K extends keyof ToolCallingAgentTemplates> = (\n  template: ToolCallingAgentTemplates[K],\n) => ToolCallingAgentTemplates[K];\n\nexport interface ToolCallingAgentInput {\n  llm: ChatModel;\n  memory: BaseMemory;\n  tools: AnyTool[];\n  meta?: Omit<AgentMeta, \"tools\">;\n  templates?: Partial<{\n    [K in keyof ToolCallingAgentTemplates]:\n      | ToolCallingAgentTemplates[K]\n      | ToolCallingAgentTemplateFactory<K>;\n  }>;\n  execution?: ToolCallingAgentExecutionConfig;\n  saveIntermediateSteps?: boolean;\n}\n\nexport class ToolCallingAgent extends BaseAgent<\n  ToolCallingAgentRunInput,\n  ToolCallingAgentRunOutput,\n  ToolCallingAgentRunOptions\n> {\n  public readonly emitter = Emitter.root.child<ToolCallingAgentCallbacks>({\n    namespace: [\"agent\", \"toolCalling\"],\n    creator: this,\n  });\n\n  constructor(public readonly input: ToolCallingAgentInput) {\n    super();\n    this.input.saveIntermediateSteps = this.input.saveIntermediateSteps ?? true;\n  }\n\n  static {\n    this.register();\n  }\n\n  protected async _run(\n    input: ToolCallingAgentRunInput,\n    options: ToolCallingAgentRunOptions = {},\n    run: GetRunContext<typeof this>,\n  ): Promise<ToolCallingAgentRunOutput> {\n    const tempMessageKey = \"tempMessage\" as const;\n    const execution = {\n      maxRetriesPerStep: 3,\n      totalMaxRetries: 20,\n      maxIterations: 10,\n      ...omitUndefined(this.input.execution ?? {}),\n      ...omitUndefined(options.execution ?? {}),\n    };\n\n    const state: ToolCallingAgentRunState = {\n      memory: new UnconstrainedMemory(),\n      result: undefined,\n      iteration: 0,\n    };\n    await state.memory.add(\n      new SystemMessage(\n        this.templates.system.render({\n          role: undefined,\n          instructions: undefined,\n        }),\n      ),\n    );\n    await state.memory.addMany(this.memory.messages);\n\n    if (input.prompt) {\n      const userMessage = new UserMessage(\n        this.templates.task.render({\n          prompt: input.prompt,\n          context: input.context,\n          expectedOutput: isString(input.expectedOutput) ? input.expectedOutput : undefined,\n        }),\n      );\n      await state.memory.add(userMessage);\n    }\n\n    const globalRetriesCounter = new RetryCounter(execution.totalMaxRetries || 1, AgentError);\n\n    const usePlainResponse = !input.expectedOutput || !(input.expectedOutput instanceof ZodSchema);\n    const finalAnswerToolSchema = usePlainResponse\n      ? z.object({\n          response: z.string().describe(String(input.expectedOutput ?? \"\")),\n        })\n      : (input.expectedOutput as ZodSchema);\n\n    const finalAnswerTool = new DynamicTool({\n      name: \"final_answer\",\n      description: \"Sends the final answer to the user\",\n      inputSchema: finalAnswerToolSchema,\n      handler: async (input) => {\n        const result = usePlainResponse ? input.response : JSON.stringify(input.response);\n        state.result = new AssistantMessage(result);\n        return new StringToolOutput(\"Message has been sent\");\n      },\n    });\n\n    const tools = [...this.input.tools, finalAnswerTool];\n    let forceFinalAnswer = false;\n\n    while (!state.result) {\n      state.iteration++;\n      if (state.iteration > (execution.totalMaxRetries ?? Infinity)) {\n        throw new AgentError(\n          `Agent was not able to resolve the task in ${state.iteration} iterations.`,\n        );\n      }\n\n      await run.emitter.emit(\"start\", { state });\n      const response = await this.input.llm.create({\n        messages: state.memory.messages.slice(),\n        tools,\n        toolChoice: forceFinalAnswer ? finalAnswerTool : tools.length > 1 ? \"required\" : tools[0],\n        stream: false,\n      });\n      await state.memory.addMany(response.messages);\n\n      const toolCallMessages = response.getToolCalls();\n      for (const toolCall of toolCallMessages) {\n        try {\n          const tool = tools.find((tool) => tool.name === toolCall.toolName);\n          if (!tool) {\n            throw new AgentError(`Tool ${toolCall.toolName} does not exist!`);\n          }\n\n          const toolInput: any = toolCall.args;\n          const toolResponse: ToolOutput = await tool.run(toolInput).context({\n            state,\n            toolCallMsg: toolCall,\n          });\n          await state.memory.add(\n            new ToolMessage({\n              type: \"tool-result\",\n              toolCallId: toolCall.toolCallId,\n              toolName: toolCall.toolName,\n              result: toolResponse.getTextContent(),\n              isError: false,\n            }),\n          );\n        } catch (e) {\n          if (e instanceof ToolError) {\n            globalRetriesCounter.use(e);\n            await state.memory.add(\n              new ToolMessage({\n                type: \"tool-result\",\n                toolCallId: toolCall.toolCallId,\n                toolName: toolCall.toolName,\n                result: e.explain(),\n                isError: true,\n              }),\n            );\n          } else {\n            throw e;\n          }\n        }\n      }\n\n      // handle empty messages for some models\n      const textMessages = response.getTextMessages();\n      if (isEmpty(toolCallMessages) && isEmpty(textMessages)) {\n        await state.memory.add(new AssistantMessage(\"\\n\", { [tempMessageKey]: true }));\n      } else {\n        await state.memory.deleteMany(\n          state.memory.messages.filter((msg) => msg.meta[tempMessageKey]),\n        );\n      }\n\n      // Fallback for providers that do not support structured outputs\n      if (!isEmpty(textMessages) && isEmpty(toolCallMessages)) {\n        forceFinalAnswer = true;\n        tools.length = 0;\n        tools.push(finalAnswerTool);\n      }\n\n      await run.emitter.emit(\"success\", { state });\n    }\n\n    if (this.input.saveIntermediateSteps) {\n      this.memory.reset();\n      await this.memory.addMany(state.memory.messages.slice(1));\n    } else {\n      await this.memory.addMany(state.memory.messages.slice(-2));\n    }\n    return { memory: state.memory, result: state.result };\n  }\n\n  get meta(): AgentMeta {\n    const tools = this.input.tools.slice();\n\n    if (this.input.meta) {\n      return { ...this.input.meta, tools };\n    }\n\n    return {\n      name: \"ToolCalling\",\n      tools,\n      description: \"ToolCallingAgent that uses tools to accomplish the task.\",\n      ...(tools.length > 0 && {\n        extraDescription: [\n          `Tools that I can use to accomplish given task.`,\n          ...tools.map((tool) => `Tool '${tool.name}': ${tool.description}.`),\n        ].join(\"\\n\"),\n      }),\n    };\n  }\n\n  @Cache({ enumerable: false })\n  protected get templates(): ToolCallingAgentTemplates {\n    const overrides = this.input.templates ?? {};\n    const defaultTemplates: ToolCallingAgentTemplates = {\n      system: ToolCallingAgentSystemPrompt,\n      task: ToolCallingAgentTaskPrompt,\n    } as const;\n\n    return mapObj(defaultTemplates)(\n      (key, defaultTemplate: ToolCallingAgentTemplates[typeof key]) => {\n        const override = overrides[key] ?? defaultTemplate;\n        if (override instanceof PromptTemplate) {\n          return override;\n        }\n        return override(defaultTemplate) ?? defaultTemplate;\n      },\n    );\n  }\n\n  createSnapshot() {\n    return {\n      ...super.createSnapshot(),\n      input: shallowCopy(this.input),\n      emitter: this.emitter,\n    };\n  }\n\n  set memory(memory: BaseMemory) {\n    this.input.memory = memory;\n  }\n\n  get memory() {\n    return this.input.memory;\n  }\n}\n"]}

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

@@ -1,10 +1,10 @@
-import { b as AgentMeta, a as BaseAgent } from '../../base-Csr9guq2.cjs';
+import { b as AgentMeta, a as BaseAgent } from '../../base-BPsHNV2O.cjs';
 import { AnyTool } from '../../tools/base.cjs';
 import { BaseMemory } from '../../memory/base.cjs';
-import { E as Emitter } from '../../emitter-BWtGHYn0.cjs';
+import { E as Emitter } from '../../emitter-BTr99H7I.cjs';
 import { ToolCallingAgentTemplates, ToolCallingAgentExecutionConfig, ToolCallingAgentRunInput, ToolCallingAgentRunOutput, ToolCallingAgentRunOptions, ToolCallingAgentCallbacks } from './types.cjs';
 import { GetRunContext } from '../../context.cjs';
-import { ChatModel } from '../../backend/chat.cjs';
+import { m as ChatModel } from '../../chat-BvyMdWFo.cjs';
 import '../../errors.cjs';
 import '../../internals/types.cjs';
 import '../../internals/helpers/guards.cjs';
@@ -20,12 +20,12 @@ import '../../backend/message.cjs';
 import 'ai';
 import './prompts.cjs';
 import '../../template.cjs';
-import '../../backend/utils.cjs';
 import '../../backend/constants.cjs';
-import '@ai-sdk/provider';
+import '../../logger/logger.cjs';
+import 'pino';
 
 /**
- * Copyright 2025 IBM Corp.
+ * Copyright 2025 © BeeAI a Series of LF Projects, LLC
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -50,6 +50,7 @@ interface ToolCallingAgentInput {
         [K in keyof ToolCallingAgentTemplates]: ToolCallingAgentTemplates[K] | ToolCallingAgentTemplateFactory<K>;
     }>;
     execution?: ToolCallingAgentExecutionConfig;
+    saveIntermediateSteps?: boolean;
 }
 declare class ToolCallingAgent extends BaseAgent<ToolCallingAgentRunInput, ToolCallingAgentRunOutput, ToolCallingAgentRunOptions> {
     readonly input: ToolCallingAgentInput;

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

@@ -1,10 +1,10 @@
-import { b as AgentMeta, a as BaseAgent } from '../../base-DHhpWHQm.js';
+import { b as AgentMeta, a as BaseAgent } from '../../base-ZnwDK6ZB.js';
 import { AnyTool } from '../../tools/base.js';
 import { BaseMemory } from '../../memory/base.js';
-import { E as Emitter } from '../../emitter-l0W9gC1A.js';
+import { E as Emitter } from '../../emitter-DiGmS_5O.js';
 import { ToolCallingAgentTemplates, ToolCallingAgentExecutionConfig, ToolCallingAgentRunInput, ToolCallingAgentRunOutput, ToolCallingAgentRunOptions, ToolCallingAgentCallbacks } from './types.js';
 import { GetRunContext } from '../../context.js';
-import { ChatModel } from '../../backend/chat.js';
+import { m as ChatModel } from '../../chat-diSK3Sui.js';
 import '../../errors.js';
 import '../../internals/types.js';
 import '../../internals/helpers/guards.js';
@@ -20,12 +20,12 @@ import '../../backend/message.js';
 import 'ai';
 import './prompts.js';
 import '../../template.js';
-import '../../backend/utils.js';
 import '../../backend/constants.js';
-import '@ai-sdk/provider';
+import '../../logger/logger.js';
+import 'pino';
 
 /**
- * Copyright 2025 IBM Corp.
+ * Copyright 2025 © BeeAI a Series of LF Projects, LLC
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -50,6 +50,7 @@ interface ToolCallingAgentInput {
         [K in keyof ToolCallingAgentTemplates]: ToolCallingAgentTemplates[K] | ToolCallingAgentTemplateFactory<K>;
     }>;
     execution?: ToolCallingAgentExecutionConfig;
+    saveIntermediateSteps?: boolean;
 }
 declare class ToolCallingAgent extends BaseAgent<ToolCallingAgentRunInput, ToolCallingAgentRunOutput, ToolCallingAgentRunOptions> {
     readonly input: ToolCallingAgentInput;

package/dist/agents/toolCalling/agent.js

@@ -1,5 +1,5 @@
 import { BaseAgent, AgentError } from '../base.js';
-import { ToolError } from '../../tools/base.js';
+import { DynamicTool, ToolError, StringToolOutput } from '../../tools/base.js';
 import { Emitter } from '../../emitter/emitter.js';
 import { shallowCopy } from '../../serializer/utils.js';
 import { UnconstrainedMemory } from '../../memory/unconstrainedMemory.js';
@@ -10,7 +10,7 @@ import { omitUndefined, mapObj } from '../../internals/helpers/object.js';
 import { Cache } from '../../cache/decoratorCache.js';
 import { PromptTemplate } from '../../template.js';
 import { ToolCallingAgentTaskPrompt, ToolCallingAgentSystemPrompt } from './prompts.js';
-import { ZodSchema } from 'zod';
+import { ZodSchema, z } from 'zod';
 
 var __defProp = Object.defineProperty;
 var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
@@ -39,6 +39,7 @@ class ToolCallingAgent extends BaseAgent {
       ],
       creator: this
     });
+    this.input.saveIntermediateSteps = this.input.saveIntermediateSteps ?? true;
   }
   static {
     this.register();
@@ -59,6 +60,7 @@ class ToolCallingAgent extends BaseAgent {
       role: void 0,
       instructions: void 0
     })));
+    await state.memory.addMany(this.memory.messages);
     if (input.prompt) {
       const userMessage = new UserMessage(this.templates.task.render({
         prompt: input.prompt,
@@ -68,6 +70,25 @@ class ToolCallingAgent extends BaseAgent {
       await state.memory.add(userMessage);
     }
     const globalRetriesCounter = new RetryCounter(execution.totalMaxRetries || 1, AgentError);
+    const usePlainResponse = !input.expectedOutput || !(input.expectedOutput instanceof ZodSchema);
+    const finalAnswerToolSchema = usePlainResponse ? z.object({
+      response: z.string().describe(String(input.expectedOutput ?? ""))
+    }) : input.expectedOutput;
+    const finalAnswerTool = new DynamicTool({
+      name: "final_answer",
+      description: "Sends the final answer to the user",
+      inputSchema: finalAnswerToolSchema,
+      handler: /* @__PURE__ */ __name(async (input2) => {
+        const result = usePlainResponse ? input2.response : JSON.stringify(input2.response);
+        state.result = new AssistantMessage(result);
+        return new StringToolOutput("Message has been sent");
+      }, "handler")
+    });
+    const tools = [
+      ...this.input.tools,
+      finalAnswerTool
+    ];
+    let forceFinalAnswer = false;
     while (!state.result) {
       state.iteration++;
       if (state.iteration > (execution.totalMaxRetries ?? Infinity)) {
@@ -78,14 +99,15 @@ class ToolCallingAgent extends BaseAgent {
       });
       const response = await this.input.llm.create({
         messages: state.memory.messages.slice(),
-        tools: this.input.tools,
+        tools,
+        toolChoice: forceFinalAnswer ? finalAnswerTool : tools.length > 1 ? "required" : tools[0],
         stream: false
       });
       await state.memory.addMany(response.messages);
       const toolCallMessages = response.getToolCalls();
       for (const toolCall of toolCallMessages) {
         try {
-          const tool = this.input.tools.find((tool2) => tool2.name === toolCall.toolName);
+          const tool = tools.find((tool2) => tool2.name === toolCall.toolName);
           if (!tool) {
             throw new AgentError(`Tool ${toolCall.toolName} does not exist!`);
           }
@@ -125,21 +147,20 @@ class ToolCallingAgent extends BaseAgent {
         await state.memory.deleteMany(state.memory.messages.filter((msg) => msg.meta[tempMessageKey]));
       }
       if (!isEmpty(textMessages) && isEmpty(toolCallMessages)) {
-        if (input.expectedOutput && input.expectedOutput instanceof ZodSchema) {
-          const structured = await this.input.llm.createStructure({
-            schema: input.expectedOutput,
-            messages: state.memory.messages.slice()
-          });
-          state.result = new AssistantMessage(JSON.stringify(structured, null, 4));
-        } else {
-          state.result = AssistantMessage.fromChunks(textMessages);
-        }
+        forceFinalAnswer = true;
+        tools.length = 0;
+        tools.push(finalAnswerTool);
       }
       await run.emitter.emit("success", {
         state
       });
     }
-    await this.memory.addMany(state.memory.messages.slice(1));
+    if (this.input.saveIntermediateSteps) {
+      this.memory.reset();
+      await this.memory.addMany(state.memory.messages.slice(1));
+    } else {
+      await this.memory.addMany(state.memory.messages.slice(-2));
+    }
     return {
       memory: state.memory,
       result: state.result

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

@@ -1 +1 @@
-{"version":3,"sources":["agent.ts"],"names":["ToolCallingAgent","BaseAgent","emitter","constructor","input","Emitter","root","child","namespace","creator","register","_run","options","run","tempMessageKey","execution","totalMaxRetries","omitUndefined","state","memory","UnconstrainedMemory","result","undefined","iteration","add","SystemMessage","templates","system","render","role","instructions","prompt","userMessage","UserMessage","task","context","expectedOutput","isString","globalRetriesCounter","RetryCounter","AgentError","Infinity","emit","response","llm","create","messages","slice","tools","stream","addMany","toolCallMessages","getToolCalls","toolCall","tool","find","name","toolName","toolInput","args","toolResponse","toolCallMsg","ToolMessage","type","toolCallId","getTextContent","isError","e","ToolError","use","explain","textMessages","getTextMessages","isEmpty","AssistantMessage","deleteMany","filter","msg","meta","ZodSchema","structured","createStructure","schema","JSON","stringify","fromChunks","description","length","extraDescription","map","join","overrides","defaultTemplates","ToolCallingAgentSystemPrompt","ToolCallingAgentTaskPrompt","mapObj","key","defaultTemplate","override","PromptTemplate","createSnapshot","shallowCopy","enumerable"],"mappings":";;;;;;;;;;;;;;;;AAcC,SAAA,YAAA,CAAA,UAAA,EAAA,MAAA,EAAA,GAAA,EAAA,IAAA,EAAA;;;;;;AAAA,MAAA,CAAA,YAAA,EAAA,cAAA,CAAA;;;;;AAiDM,MAAMA,yBAAyBC,SAAAA,CAAAA;EA/DtC;;;;AAoEkBC,EAAAA,OAAAA;AAKhBC,EAAAA,WAAAA,CAA4BC,KAA8B,EAAA;AACxD,IAAK,KAAA,EAAA,EAAA,KADqBA,KAAAA,GAAAA,KAAAA,EAAAA,KALZF,OAAUG,GAAAA,OAAAA,CAAQC,KAAKC,KAAiC,CAAA;MACtEC,SAAW,EAAA;AAAC,QAAA,OAAA;AAAS,QAAA;;MACrBC,OAAS,EAAA;KACX,CAAA;AAIA;EAEA;AACE,IAAA,IAAA,CAAKC,QAAQ,EAAA;AACf;AAEA,EAAA,MAAgBC,IACdP,CAAAA,KAAAA,EACAQ,OAAsC,GAAA,IACtCC,GACoC,EAAA;AACpC,IAAA,MAAMC,cAAiB,GAAA,aAAA;AACvB,IAAA,MAAMC,SAAY,GAAA;MAEhBC,eAAiB,EAAA,EAAA;MAEjB,GAAGC,aAAc,CAAA,IAAA,CAAKb,KAAMW,CAAAA,SAAAA,IAAa,EAAC,CAAA;AAC1C,MAAA,GAAGE,aAAcL,CAAAA,OAAAA,CAAQG,SAAa,IAAA,EAAC;AACzC,KAAA;AAEA,IAAA,MAAMG,KAAkC,GAAA;AACtCC,MAAAA,MAAAA,EAAQ,IAAIC,mBAAAA,EAAAA;MACZC,MAAQC,EAAAA,MAAAA;MACRC,SAAW,EAAA;AACb,KAAA;AACA,IAAML,MAAAA,KAAAA,CAAMC,OAAOK,GACjB,CAAA,IAAIC,cACF,IAAKC,CAAAA,SAAAA,CAAUC,OAAOC,MAAO,CAAA;MAC3BC,IAAMP,EAAAA,MAAAA;MACNQ,YAAcR,EAAAA;AAChB,KAAA,CAAA,CAAA,CAAA;AAIJ,IAAA,IAAIlB,MAAM2B,MAAQ,EAAA;AAChB,MAAA,MAAMC,cAAc,IAAIC,WAAAA,CACtB,IAAKP,CAAAA,SAAAA,CAAUQ,KAAKN,MAAO,CAAA;AACzBG,QAAAA,MAAAA,EAAQ3B,KAAM2B,CAAAA,MAAAA;AACdI,QAAAA,OAAAA,EAAS/B,KAAM+B,CAAAA,OAAAA;AACfC,QAAAA,cAAAA,EAAgBC,QAASjC,CAAAA,KAAAA,CAAMgC,cAAc,CAAA,GAAIhC,MAAMgC,cAAiBd,GAAAA;AAC1E,OAAA,CAAA,CAAA;AAEF,MAAMJ,MAAAA,KAAAA,CAAMC,MAAOK,CAAAA,GAAAA,CAAIQ,WAAAA,CAAAA;AACzB;AAEA,IAAA,MAAMM,uBAAuB,IAAIC,YAAAA,CAAaxB,SAAUC,CAAAA,eAAAA,IAAmB,GAAGwB,UAAAA,CAAAA;AAE9E,IAAO,OAAA,CAACtB,MAAMG,MAAQ,EAAA;AACpBH,MAAMK,KAAAA,CAAAA,SAAAA,EAAAA;AACN,MAAA,IAAIL,KAAMK,CAAAA,SAAAA,IAAaR,SAAUC,CAAAA,eAAAA,IAAmByB,QAAW,CAAA,EAAA;AAC7D,QAAA,MAAM,IAAID,UAAAA,CACR,CAA6CtB,0CAAAA,EAAAA,KAAAA,CAAMK,SAAS,CAAc,YAAA,CAAA,CAAA;AAE9E;AAEA,MAAMV,MAAAA,GAAAA,CAAIX,OAAQwC,CAAAA,IAAAA,CAAK,OAAS,EAAA;AAAExB,QAAAA;OAAM,CAAA;AACxC,MAAA,MAAMyB,QAAW,GAAA,MAAM,IAAKvC,CAAAA,KAAAA,CAAMwC,IAAIC,MAAO,CAAA;QAC3CC,QAAU5B,EAAAA,KAAAA,CAAMC,MAAO2B,CAAAA,QAAAA,CAASC,KAAK,EAAA;AACrCC,QAAAA,KAAAA,EAAO,KAAK5C,KAAM4C,CAAAA,KAAAA;QAClBC,MAAQ,EAAA;OACV,CAAA;AACA,MAAA,MAAM/B,KAAMC,CAAAA,MAAAA,CAAO+B,OAAQP,CAAAA,QAAAA,CAASG,QAAQ,CAAA;AAE5C,MAAMK,MAAAA,gBAAAA,GAAmBR,SAASS,YAAY,EAAA;AAC9C,MAAA,KAAA,MAAWC,YAAYF,gBAAkB,EAAA;AACvC,QAAI,IAAA;AACF,UAAMG,MAAAA,IAAAA,GAAO,IAAKlD,CAAAA,KAAAA,CAAM4C,KAAMO,CAAAA,IAAAA,CAAK,CAACD,KAASA,KAAAA,KAAAA,CAAKE,IAASH,KAAAA,QAAAA,CAASI,QAAQ,CAAA;AAC5E,UAAA,IAAI,CAACH,IAAM,EAAA;AACT,YAAA,MAAM,IAAId,UAAAA,CAAW,CAAQa,KAAAA,EAAAA,QAAAA,CAASI,QAAQ,CAAkB,gBAAA,CAAA,CAAA;AAClE;AAEA,UAAA,MAAMC,YAAiBL,QAASM,CAAAA,IAAAA;AAChC,UAAA,MAAMC,eAA2B,MAAMN,IAAAA,CAAKzC,GAAI6C,CAAAA,SAAAA,EAAWvB,OAAQ,CAAA;AACjEjB,YAAAA,KAAAA;YACA2C,WAAaR,EAAAA;WACf,CAAA;AACA,UAAA,MAAMnC,KAAMC,CAAAA,MAAAA,CAAOK,GACjB,CAAA,IAAIsC,WAAY,CAAA;YACdC,IAAM,EAAA,aAAA;AACNC,YAAAA,UAAAA,EAAYX,QAASW,CAAAA,UAAAA;AACrBP,YAAAA,QAAAA,EAAUJ,QAASI,CAAAA,QAAAA;AACnBpC,YAAAA,MAAAA,EAAQuC,aAAaK,cAAc,EAAA;YACnCC,OAAS,EAAA;AACX,WAAA,CAAA,CAAA;AAEJ,SAAA,CAAA,OAASC,CAAG,EAAA;AACV,UAAA,IAAIA,aAAaC,SAAW,EAAA;AAC1B9B,YAAAA,oBAAAA,CAAqB+B,IAAIF,CAAAA,CAAAA;AACzB,YAAA,MAAMjD,KAAMC,CAAAA,MAAAA,CAAOK,GACjB,CAAA,IAAIsC,WAAY,CAAA;cACdC,IAAM,EAAA,aAAA;AACNC,cAAAA,UAAAA,EAAYX,QAASW,CAAAA,UAAAA;AACrBP,cAAAA,QAAAA,EAAUJ,QAASI,CAAAA,QAAAA;AACnBpC,cAAAA,MAAAA,EAAQ8C,EAAEG,OAAO,EAAA;cACjBJ,OAAS,EAAA;AACX,aAAA,CAAA,CAAA;WAEG,MAAA;AACL,YAAMC,MAAAA,CAAAA;AACR;AACF;AACF;AAGA,MAAMI,MAAAA,YAAAA,GAAe5B,SAAS6B,eAAe,EAAA;AAC7C,MAAA,IAAIC,OAAQtB,CAAAA,gBAAAA,CAAqBsB,IAAAA,OAAAA,CAAQF,YAAAA,CAAe,EAAA;AACtD,QAAA,MAAMrD,KAAMC,CAAAA,MAAAA,CAAOK,GAAI,CAAA,IAAIkD,iBAAiB,IAAM,EAAA;AAAE,UAAA,CAAC5D,cAAAA,GAAiB;AAAK,SAAA,CAAA,CAAA;OACtE,MAAA;AACL,QAAA,MAAMI,KAAMC,CAAAA,MAAAA,CAAOwD,UACjBzD,CAAAA,KAAAA,CAAMC,MAAO2B,CAAAA,QAAAA,CAAS8B,MAAO,CAAA,CAACC,GAAQA,KAAAA,GAAAA,CAAIC,IAAKhE,CAAAA,cAAAA,CAAe,CAAA,CAAA;AAElE;AAEA,MAAA,IAAI,CAAC2D,OAAQF,CAAAA,YAAAA,CAAiBE,IAAAA,OAAAA,CAAQtB,gBAAAA,CAAmB,EAAA;AACvD,QAAA,IAAI/C,KAAMgC,CAAAA,cAAAA,IAAkBhC,KAAMgC,CAAAA,cAAAA,YAA0B2C,SAAW,EAAA;AACrE,UAAA,MAAMC,UAAa,GAAA,MAAM,IAAK5E,CAAAA,KAAAA,CAAMwC,IAAIqC,eAAgB,CAAA;AACtDC,YAAAA,MAAAA,EAAQ9E,KAAMgC,CAAAA,cAAAA;YACdU,QAAU5B,EAAAA,KAAAA,CAAMC,MAAO2B,CAAAA,QAAAA,CAASC,KAAK;WACvC,CAAA;AACA7B,UAAMG,KAAAA,CAAAA,MAAAA,GAAS,IAAIqD,gBAAiBS,CAAAA,IAAAA,CAAKC,UAAUJ,UAAY,EAAA,IAAA,EAAM,CAAA,CAAA,CAAA;SAChE,MAAA;AACL9D,UAAMG,KAAAA,CAAAA,MAAAA,GAASqD,gBAAiBW,CAAAA,UAAAA,CAAWd,YAAAA,CAAAA;AAC7C;AACF;AAEA,MAAM1D,MAAAA,GAAAA,CAAIX,OAAQwC,CAAAA,IAAAA,CAAK,SAAW,EAAA;AAAExB,QAAAA;OAAM,CAAA;AAC5C;AAEA,IAAM,MAAA,IAAA,CAAKC,OAAO+B,OAAQhC,CAAAA,KAAAA,CAAMC,OAAO2B,QAASC,CAAAA,KAAAA,CAAM,CAAA,CAAA,CAAA;AACtD,IAAO,OAAA;AAAE5B,MAAAA,MAAAA,EAAQD,KAAMC,CAAAA,MAAAA;AAAQE,MAAAA,MAAAA,EAAQH,KAAMG,CAAAA;AAAO,KAAA;AACtD;AAEA,EAAA,IAAIyD,IAAkB,GAAA;AACpB,IAAA,MAAM9B,KAAQ,GAAA,IAAA,CAAK5C,KAAM4C,CAAAA,KAAAA,CAAMD,KAAK,EAAA;AAEpC,IAAI,IAAA,IAAA,CAAK3C,MAAM0E,IAAM,EAAA;AACnB,MAAO,OAAA;AAAE,QAAA,GAAG,KAAK1E,KAAM0E,CAAAA,IAAAA;AAAM9B,QAAAA;AAAM,OAAA;AACrC;AAEA,IAAO,OAAA;MACLQ,IAAM,EAAA,aAAA;AACNR,MAAAA,KAAAA;MACAsC,WAAa,EAAA,0DAAA;MACb,GAAItC,KAAAA,CAAMuC,SAAS,CAAK,IAAA;QACtBC,gBAAkB,EAAA;AAChB,UAAA,CAAA,8CAAA,CAAA;aACGxC,KAAMyC,CAAAA,GAAAA,CAAI,CAACnC,IAAS,KAAA,CAAA,MAAA,EAASA,KAAKE,IAAI,CAAA,GAAA,EAAMF,IAAKgC,CAAAA,WAAW,CAAG,CAAA,CAAA;AAClEI,SAAAA,CAAAA,IAAAA,CAAK,IAAA;AACT;AACF,KAAA;AACF;AAEA,EAAA,IACchE,SAAuC,GAAA;AACnD,IAAA,MAAMiE,SAAY,GAAA,IAAA,CAAKvF,KAAMsB,CAAAA,SAAAA,IAAa,EAAC;AAC3C,IAAA,MAAMkE,gBAA8C,GAAA;MAClDjE,MAAQkE,EAAAA,4BAAAA;MACR3D,IAAM4D,EAAAA;AACR,KAAA;AAEA,IAAA,OAAOC,MAAOH,CAAAA,gBAAAA,CACZ,CAAA,CAACI,KAAKC,eAAAA,KAAAA;AACJ,MAAMC,MAAAA,QAAAA,GAAWP,SAAUK,CAAAA,GAAAA,CAAQC,IAAAA,eAAAA;AACnC,MAAA,IAAIC,oBAAoBC,cAAgB,EAAA;AACtC,QAAOD,OAAAA,QAAAA;AACT;AACA,MAAOA,OAAAA,QAAAA,CAASD,eAAAA,CAAoBA,IAAAA,eAAAA;KACtC,CAAA;AAEJ;EAEAG,cAAiB,GAAA;AACf,IAAO,OAAA;AACL,MAAA,GAAG,MAAMA,cAAAA,EAAAA;MACThG,KAAOiG,EAAAA,WAAAA,CAAY,KAAKjG,KAAK,CAAA;AAC7BF,MAAAA,OAAAA,EAAS,IAAKA,CAAAA;AAChB,KAAA;AACF;AAEA,EAAA,IAAIiB,OAAOA,MAAoB,EAAA;AAC7B,IAAA,IAAA,CAAKf,MAAMe,MAASA,GAAAA,MAAAA;AACtB;AAEA,EAAA,IAAIA,MAAS,GAAA;AACX,IAAA,OAAO,KAAKf,KAAMe,CAAAA,MAAAA;AACpB;AACF;;;IAlCWmF,UAAY,EAAA","file":"agent.js","sourcesContent":["/**\n * Copyright 2025 IBM Corp.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { AgentError, BaseAgent } from \"@/agents/base.js\";\nimport { AnyTool, ToolError, ToolOutput } from \"@/tools/base.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\nimport { AgentMeta } from \"@/agents/types.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport type {\n  ToolCallingAgentExecutionConfig,\n  ToolCallingAgentTemplates,\n  ToolCallingAgentCallbacks,\n  ToolCallingAgentRunInput,\n  ToolCallingAgentRunOptions,\n  ToolCallingAgentRunOutput,\n  ToolCallingAgentRunState,\n} from \"@/agents/toolCalling/types.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { ChatModel } from \"@/backend/chat.js\";\nimport { shallowCopy } from \"@/serializer/utils.js\";\nimport { UnconstrainedMemory } from \"@/memory/unconstrainedMemory.js\";\nimport { AssistantMessage, SystemMessage, ToolMessage, UserMessage } from \"@/backend/message.js\";\nimport { isEmpty, isString } from \"remeda\";\nimport { RetryCounter } from \"@/internals/helpers/counter.js\";\nimport { mapObj, omitUndefined } from \"@/internals/helpers/object.js\";\nimport { Cache } from \"@/cache/decoratorCache.js\";\nimport { PromptTemplate } from \"@/template.js\";\nimport {\n  ToolCallingAgentSystemPrompt,\n  ToolCallingAgentTaskPrompt,\n} from \"@/agents/toolCalling/prompts.js\";\nimport { ZodSchema } from \"zod\";\n\nexport type ToolCallingAgentTemplateFactory<K extends keyof ToolCallingAgentTemplates> = (\n  template: ToolCallingAgentTemplates[K],\n) => ToolCallingAgentTemplates[K];\n\nexport interface ToolCallingAgentInput {\n  llm: ChatModel;\n  memory: BaseMemory;\n  tools: AnyTool[];\n  meta?: Omit<AgentMeta, \"tools\">;\n  templates?: Partial<{\n    [K in keyof ToolCallingAgentTemplates]:\n      | ToolCallingAgentTemplates[K]\n      | ToolCallingAgentTemplateFactory<K>;\n  }>;\n  execution?: ToolCallingAgentExecutionConfig;\n}\n\nexport class ToolCallingAgent extends BaseAgent<\n  ToolCallingAgentRunInput,\n  ToolCallingAgentRunOutput,\n  ToolCallingAgentRunOptions\n> {\n  public readonly emitter = Emitter.root.child<ToolCallingAgentCallbacks>({\n    namespace: [\"agent\", \"toolCalling\"],\n    creator: this,\n  });\n\n  constructor(public readonly input: ToolCallingAgentInput) {\n    super();\n  }\n\n  static {\n    this.register();\n  }\n\n  protected async _run(\n    input: ToolCallingAgentRunInput,\n    options: ToolCallingAgentRunOptions = {},\n    run: GetRunContext<typeof this>,\n  ): Promise<ToolCallingAgentRunOutput> {\n    const tempMessageKey = \"tempMessage\" as const;\n    const execution = {\n      maxRetriesPerStep: 3,\n      totalMaxRetries: 20,\n      maxIterations: 10,\n      ...omitUndefined(this.input.execution ?? {}),\n      ...omitUndefined(options.execution ?? {}),\n    };\n\n    const state: ToolCallingAgentRunState = {\n      memory: new UnconstrainedMemory(),\n      result: undefined,\n      iteration: 0,\n    };\n    await state.memory.add(\n      new SystemMessage(\n        this.templates.system.render({\n          role: undefined,\n          instructions: undefined,\n        }),\n      ),\n    );\n\n    if (input.prompt) {\n      const userMessage = new UserMessage(\n        this.templates.task.render({\n          prompt: input.prompt,\n          context: input.context,\n          expectedOutput: isString(input.expectedOutput) ? input.expectedOutput : undefined,\n        }),\n      );\n      await state.memory.add(userMessage);\n    }\n\n    const globalRetriesCounter = new RetryCounter(execution.totalMaxRetries || 1, AgentError);\n\n    while (!state.result) {\n      state.iteration++;\n      if (state.iteration > (execution.totalMaxRetries ?? Infinity)) {\n        throw new AgentError(\n          `Agent was not able to resolve the task in ${state.iteration} iterations.`,\n        );\n      }\n\n      await run.emitter.emit(\"start\", { state });\n      const response = await this.input.llm.create({\n        messages: state.memory.messages.slice(),\n        tools: this.input.tools,\n        stream: false,\n      });\n      await state.memory.addMany(response.messages);\n\n      const toolCallMessages = response.getToolCalls();\n      for (const toolCall of toolCallMessages) {\n        try {\n          const tool = this.input.tools.find((tool) => tool.name === toolCall.toolName);\n          if (!tool) {\n            throw new AgentError(`Tool ${toolCall.toolName} does not exist!`);\n          }\n\n          const toolInput: any = toolCall.args;\n          const toolResponse: ToolOutput = await tool.run(toolInput).context({\n            state,\n            toolCallMsg: toolCall,\n          });\n          await state.memory.add(\n            new ToolMessage({\n              type: \"tool-result\",\n              toolCallId: toolCall.toolCallId,\n              toolName: toolCall.toolName,\n              result: toolResponse.getTextContent(),\n              isError: false,\n            }),\n          );\n        } catch (e) {\n          if (e instanceof ToolError) {\n            globalRetriesCounter.use(e);\n            await state.memory.add(\n              new ToolMessage({\n                type: \"tool-result\",\n                toolCallId: toolCall.toolCallId,\n                toolName: toolCall.toolName,\n                result: e.explain(),\n                isError: true,\n              }),\n            );\n          } else {\n            throw e;\n          }\n        }\n      }\n\n      // handle empty messages for some models\n      const textMessages = response.getTextMessages();\n      if (isEmpty(toolCallMessages) && isEmpty(textMessages)) {\n        await state.memory.add(new AssistantMessage(\"\\n\", { [tempMessageKey]: true }));\n      } else {\n        await state.memory.deleteMany(\n          state.memory.messages.filter((msg) => msg.meta[tempMessageKey]),\n        );\n      }\n\n      if (!isEmpty(textMessages) && isEmpty(toolCallMessages)) {\n        if (input.expectedOutput && input.expectedOutput instanceof ZodSchema) {\n          const structured = await this.input.llm.createStructure({\n            schema: input.expectedOutput,\n            messages: state.memory.messages.slice(),\n          });\n          state.result = new AssistantMessage(JSON.stringify(structured, null, 4));\n        } else {\n          state.result = AssistantMessage.fromChunks(textMessages);\n        }\n      }\n\n      await run.emitter.emit(\"success\", { state });\n    }\n\n    await this.memory.addMany(state.memory.messages.slice(1));\n    return { memory: state.memory, result: state.result };\n  }\n\n  get meta(): AgentMeta {\n    const tools = this.input.tools.slice();\n\n    if (this.input.meta) {\n      return { ...this.input.meta, tools };\n    }\n\n    return {\n      name: \"ToolCalling\",\n      tools,\n      description: \"ToolCallingAgent that uses tools to accomplish the task.\",\n      ...(tools.length > 0 && {\n        extraDescription: [\n          `Tools that I can use to accomplish given task.`,\n          ...tools.map((tool) => `Tool '${tool.name}': ${tool.description}.`),\n        ].join(\"\\n\"),\n      }),\n    };\n  }\n\n  @Cache({ enumerable: false })\n  protected get templates(): ToolCallingAgentTemplates {\n    const overrides = this.input.templates ?? {};\n    const defaultTemplates: ToolCallingAgentTemplates = {\n      system: ToolCallingAgentSystemPrompt,\n      task: ToolCallingAgentTaskPrompt,\n    } as const;\n\n    return mapObj(defaultTemplates)(\n      (key, defaultTemplate: ToolCallingAgentTemplates[typeof key]) => {\n        const override = overrides[key] ?? defaultTemplate;\n        if (override instanceof PromptTemplate) {\n          return override;\n        }\n        return override(defaultTemplate) ?? defaultTemplate;\n      },\n    );\n  }\n\n  createSnapshot() {\n    return {\n      ...super.createSnapshot(),\n      input: shallowCopy(this.input),\n      emitter: this.emitter,\n    };\n  }\n\n  set memory(memory: BaseMemory) {\n    this.input.memory = memory;\n  }\n\n  get memory() {\n    return this.input.memory;\n  }\n}\n"]}
+{"version":3,"sources":["agent.ts"],"names":["ToolCallingAgent","BaseAgent","emitter","constructor","input","Emitter","root","child","namespace","creator","saveIntermediateSteps","register","_run","options","run","tempMessageKey","execution","totalMaxRetries","omitUndefined","state","memory","UnconstrainedMemory","result","undefined","iteration","add","SystemMessage","templates","system","render","role","instructions","addMany","messages","prompt","userMessage","UserMessage","task","context","expectedOutput","isString","globalRetriesCounter","RetryCounter","AgentError","usePlainResponse","ZodSchema","finalAnswerToolSchema","z","object","response","string","describe","String","finalAnswerTool","DynamicTool","name","description","inputSchema","handler","JSON","stringify","AssistantMessage","StringToolOutput","tools","forceFinalAnswer","Infinity","emit","llm","create","slice","toolChoice","length","stream","toolCallMessages","getToolCalls","toolCall","tool","find","toolName","toolInput","args","toolResponse","toolCallMsg","ToolMessage","type","toolCallId","getTextContent","isError","e","ToolError","use","explain","textMessages","getTextMessages","isEmpty","deleteMany","filter","msg","meta","push","reset","extraDescription","map","join","overrides","defaultTemplates","ToolCallingAgentSystemPrompt","ToolCallingAgentTaskPrompt","mapObj","key","defaultTemplate","override","PromptTemplate","createSnapshot","shallowCopy","enumerable"],"mappings":";;;;;;;;;;;;;;;;AAcC,SAAA,YAAA,CAAA,UAAA,EAAA,MAAA,EAAA,GAAA,EAAA,IAAA,EAAA;;;;;;AAAA,MAAA,CAAA,YAAA,EAAA,cAAA,CAAA;;;;;AAkDM,MAAMA,yBAAyBC,SAAAA,CAAAA;EAhEtC;;;;AAqEkBC,EAAAA,OAAAA;AAKhBC,EAAAA,WAAAA,CAA4BC,KAA8B,EAAA;AACxD,IAAK,KAAA,EAAA,EAAA,KADqBA,KAAAA,GAAAA,KAAAA,EAAAA,KALZF,OAAUG,GAAAA,OAAAA,CAAQC,KAAKC,KAAiC,CAAA;MACtEC,SAAW,EAAA;AAAC,QAAA,OAAA;AAAS,QAAA;;MACrBC,OAAS,EAAA;KACX,CAAA;AAIE,IAAA,IAAA,CAAKL,KAAMM,CAAAA,qBAAAA,GAAwB,IAAKN,CAAAA,KAAAA,CAAMM,qBAAyB,IAAA,IAAA;AACzE;EAEA;AACE,IAAA,IAAA,CAAKC,QAAQ,EAAA;AACf;AAEA,EAAA,MAAgBC,IACdR,CAAAA,KAAAA,EACAS,OAAsC,GAAA,IACtCC,GACoC,EAAA;AACpC,IAAA,MAAMC,cAAiB,GAAA,aAAA;AACvB,IAAA,MAAMC,SAAY,GAAA;MAEhBC,eAAiB,EAAA,EAAA;MAEjB,GAAGC,aAAc,CAAA,IAAA,CAAKd,KAAMY,CAAAA,SAAAA,IAAa,EAAC,CAAA;AAC1C,MAAA,GAAGE,aAAcL,CAAAA,OAAAA,CAAQG,SAAa,IAAA,EAAC;AACzC,KAAA;AAEA,IAAA,MAAMG,KAAkC,GAAA;AACtCC,MAAAA,MAAAA,EAAQ,IAAIC,mBAAAA,EAAAA;MACZC,MAAQC,EAAAA,MAAAA;MACRC,SAAW,EAAA;AACb,KAAA;AACA,IAAML,MAAAA,KAAAA,CAAMC,OAAOK,GACjB,CAAA,IAAIC,cACF,IAAKC,CAAAA,SAAAA,CAAUC,OAAOC,MAAO,CAAA;MAC3BC,IAAMP,EAAAA,MAAAA;MACNQ,YAAcR,EAAAA;AAChB,KAAA,CAAA,CAAA,CAAA;AAGJ,IAAA,MAAMJ,KAAMC,CAAAA,MAAAA,CAAOY,OAAQ,CAAA,IAAA,CAAKZ,OAAOa,QAAQ,CAAA;AAE/C,IAAA,IAAI7B,MAAM8B,MAAQ,EAAA;AAChB,MAAA,MAAMC,cAAc,IAAIC,WAAAA,CACtB,IAAKT,CAAAA,SAAAA,CAAUU,KAAKR,MAAO,CAAA;AACzBK,QAAAA,MAAAA,EAAQ9B,KAAM8B,CAAAA,MAAAA;AACdI,QAAAA,OAAAA,EAASlC,KAAMkC,CAAAA,OAAAA;AACfC,QAAAA,cAAAA,EAAgBC,QAASpC,CAAAA,KAAAA,CAAMmC,cAAc,CAAA,GAAInC,MAAMmC,cAAiBhB,GAAAA;AAC1E,OAAA,CAAA,CAAA;AAEF,MAAMJ,MAAAA,KAAAA,CAAMC,MAAOK,CAAAA,GAAAA,CAAIU,WAAAA,CAAAA;AACzB;AAEA,IAAA,MAAMM,uBAAuB,IAAIC,YAAAA,CAAa1B,SAAUC,CAAAA,eAAAA,IAAmB,GAAG0B,UAAAA,CAAAA;AAE9E,IAAA,MAAMC,mBAAmB,CAACxC,KAAAA,CAAMmC,cAAkB,IAAA,EAAEnC,MAAMmC,cAA0BM,YAAAA,SAAAA,CAAAA;AACpF,IAAMC,MAAAA,qBAAAA,GAAwBF,gBAC1BG,GAAAA,CAAAA,CAAEC,MAAO,CAAA;MACPC,QAAUF,EAAAA,CAAAA,CAAEG,QAASC,CAAAA,QAAAA,CAASC,OAAOhD,KAAMmC,CAAAA,cAAAA,IAAkB,EAAA,CAAA;AAC/D,KAAA,IACCnC,KAAMmC,CAAAA,cAAAA;AAEX,IAAMc,MAAAA,eAAAA,GAAkB,IAAIC,WAAY,CAAA;MACtCC,IAAM,EAAA,cAAA;MACNC,WAAa,EAAA,oCAAA;MACbC,WAAaX,EAAAA,qBAAAA;AACbY,MAAAA,OAAAA,gCAAgBtD,MAAAA,KAAAA;AACd,QAAA,MAAMkB,SAASsB,gBAAmBxC,GAAAA,MAAAA,CAAM6C,WAAWU,IAAKC,CAAAA,SAAAA,CAAUxD,OAAM6C,QAAQ,CAAA;AAChF9B,QAAMG,KAAAA,CAAAA,MAAAA,GAAS,IAAIuC,gBAAAA,CAAiBvC,MAAAA,CAAAA;AACpC,QAAO,OAAA,IAAIwC,iBAAiB,uBAAA,CAAA;OAHrB,EAAA,SAAA;KAKX,CAAA;AAEA,IAAA,MAAMC,KAAQ,GAAA;AAAI,MAAA,GAAA,IAAA,CAAK3D,KAAM2D,CAAAA,KAAAA;AAAOV,MAAAA;;AACpC,IAAA,IAAIW,gBAAmB,GAAA,KAAA;AAEvB,IAAO,OAAA,CAAC7C,MAAMG,MAAQ,EAAA;AACpBH,MAAMK,KAAAA,CAAAA,SAAAA,EAAAA;AACN,MAAA,IAAIL,KAAMK,CAAAA,SAAAA,IAAaR,SAAUC,CAAAA,eAAAA,IAAmBgD,QAAW,CAAA,EAAA;AAC7D,QAAA,MAAM,IAAItB,UAAAA,CACR,CAA6CxB,0CAAAA,EAAAA,KAAAA,CAAMK,SAAS,CAAc,YAAA,CAAA,CAAA;AAE9E;AAEA,MAAMV,MAAAA,GAAAA,CAAIZ,OAAQgE,CAAAA,IAAAA,CAAK,OAAS,EAAA;AAAE/C,QAAAA;OAAM,CAAA;AACxC,MAAA,MAAM8B,QAAW,GAAA,MAAM,IAAK7C,CAAAA,KAAAA,CAAM+D,IAAIC,MAAO,CAAA;QAC3CnC,QAAUd,EAAAA,KAAAA,CAAMC,MAAOa,CAAAA,QAAAA,CAASoC,KAAK,EAAA;AACrCN,QAAAA,KAAAA;AACAO,QAAAA,UAAAA,EAAYN,mBAAmBX,eAAkBU,GAAAA,KAAAA,CAAMQ,SAAS,CAAI,GAAA,UAAA,GAAaR,MAAM,CAAA,CAAA;QACvFS,MAAQ,EAAA;OACV,CAAA;AACA,MAAA,MAAMrD,KAAMC,CAAAA,MAAAA,CAAOY,OAAQiB,CAAAA,QAAAA,CAAShB,QAAQ,CAAA;AAE5C,MAAMwC,MAAAA,gBAAAA,GAAmBxB,SAASyB,YAAY,EAAA;AAC9C,MAAA,KAAA,MAAWC,YAAYF,gBAAkB,EAAA;AACvC,QAAI,IAAA;AACF,UAAMG,MAAAA,IAAAA,GAAOb,MAAMc,IAAK,CAAA,CAACD,UAASA,KAAKrB,CAAAA,IAAAA,KAASoB,SAASG,QAAQ,CAAA;AACjE,UAAA,IAAI,CAACF,IAAM,EAAA;AACT,YAAA,MAAM,IAAIjC,UAAAA,CAAW,CAAQgC,KAAAA,EAAAA,QAAAA,CAASG,QAAQ,CAAkB,gBAAA,CAAA,CAAA;AAClE;AAEA,UAAA,MAAMC,YAAiBJ,QAASK,CAAAA,IAAAA;AAChC,UAAA,MAAMC,eAA2B,MAAML,IAAAA,CAAK9D,GAAIiE,CAAAA,SAAAA,EAAWzC,OAAQ,CAAA;AACjEnB,YAAAA,KAAAA;YACA+D,WAAaP,EAAAA;WACf,CAAA;AACA,UAAA,MAAMxD,KAAMC,CAAAA,MAAAA,CAAOK,GACjB,CAAA,IAAI0D,WAAY,CAAA;YACdC,IAAM,EAAA,aAAA;AACNC,YAAAA,UAAAA,EAAYV,QAASU,CAAAA,UAAAA;AACrBP,YAAAA,QAAAA,EAAUH,QAASG,CAAAA,QAAAA;AACnBxD,YAAAA,MAAAA,EAAQ2D,aAAaK,cAAc,EAAA;YACnCC,OAAS,EAAA;AACX,WAAA,CAAA,CAAA;AAEJ,SAAA,CAAA,OAASC,CAAG,EAAA;AACV,UAAA,IAAIA,aAAaC,SAAW,EAAA;AAC1BhD,YAAAA,oBAAAA,CAAqBiD,IAAIF,CAAAA,CAAAA;AACzB,YAAA,MAAMrE,KAAMC,CAAAA,MAAAA,CAAOK,GACjB,CAAA,IAAI0D,WAAY,CAAA;cACdC,IAAM,EAAA,aAAA;AACNC,cAAAA,UAAAA,EAAYV,QAASU,CAAAA,UAAAA;AACrBP,cAAAA,QAAAA,EAAUH,QAASG,CAAAA,QAAAA;AACnBxD,cAAAA,MAAAA,EAAQkE,EAAEG,OAAO,EAAA;cACjBJ,OAAS,EAAA;AACX,aAAA,CAAA,CAAA;WAEG,MAAA;AACL,YAAMC,MAAAA,CAAAA;AACR;AACF;AACF;AAGA,MAAMI,MAAAA,YAAAA,GAAe3C,SAAS4C,eAAe,EAAA;AAC7C,MAAA,IAAIC,OAAQrB,CAAAA,gBAAAA,CAAqBqB,IAAAA,OAAAA,CAAQF,YAAAA,CAAe,EAAA;AACtD,QAAA,MAAMzE,KAAMC,CAAAA,MAAAA,CAAOK,GAAI,CAAA,IAAIoC,iBAAiB,IAAM,EAAA;AAAE,UAAA,CAAC9C,cAAAA,GAAiB;AAAK,SAAA,CAAA,CAAA;OACtE,MAAA;AACL,QAAA,MAAMI,KAAMC,CAAAA,MAAAA,CAAO2E,UACjB5E,CAAAA,KAAAA,CAAMC,MAAOa,CAAAA,QAAAA,CAAS+D,MAAO,CAAA,CAACC,GAAQA,KAAAA,GAAAA,CAAIC,IAAKnF,CAAAA,cAAAA,CAAe,CAAA,CAAA;AAElE;AAGA,MAAA,IAAI,CAAC+E,OAAQF,CAAAA,YAAAA,CAAiBE,IAAAA,OAAAA,CAAQrB,gBAAAA,CAAmB,EAAA;AACvDT,QAAmB,gBAAA,GAAA,IAAA;AACnBD,QAAAA,KAAAA,CAAMQ,MAAS,GAAA,CAAA;AACfR,QAAAA,KAAAA,CAAMoC,KAAK9C,eAAAA,CAAAA;AACb;AAEA,MAAMvC,MAAAA,GAAAA,CAAIZ,OAAQgE,CAAAA,IAAAA,CAAK,SAAW,EAAA;AAAE/C,QAAAA;OAAM,CAAA;AAC5C;AAEA,IAAI,IAAA,IAAA,CAAKf,MAAMM,qBAAuB,EAAA;AACpC,MAAA,IAAA,CAAKU,OAAOgF,KAAK,EAAA;AACjB,MAAM,MAAA,IAAA,CAAKhF,OAAOY,OAAQb,CAAAA,KAAAA,CAAMC,OAAOa,QAASoC,CAAAA,KAAAA,CAAM,CAAA,CAAA,CAAA;KACjD,MAAA;AACL,MAAM,MAAA,IAAA,CAAKjD,OAAOY,OAAQb,CAAAA,KAAAA,CAAMC,OAAOa,QAASoC,CAAAA,KAAAA,CAAM,EAAC,CAAA,CAAA;AACzD;AACA,IAAO,OAAA;AAAEjD,MAAAA,MAAAA,EAAQD,KAAMC,CAAAA,MAAAA;AAAQE,MAAAA,MAAAA,EAAQH,KAAMG,CAAAA;AAAO,KAAA;AACtD;AAEA,EAAA,IAAI4E,IAAkB,GAAA;AACpB,IAAA,MAAMnC,KAAQ,GAAA,IAAA,CAAK3D,KAAM2D,CAAAA,KAAAA,CAAMM,KAAK,EAAA;AAEpC,IAAI,IAAA,IAAA,CAAKjE,MAAM8F,IAAM,EAAA;AACnB,MAAO,OAAA;AAAE,QAAA,GAAG,KAAK9F,KAAM8F,CAAAA,IAAAA;AAAMnC,QAAAA;AAAM,OAAA;AACrC;AAEA,IAAO,OAAA;MACLR,IAAM,EAAA,aAAA;AACNQ,MAAAA,KAAAA;MACAP,WAAa,EAAA,0DAAA;MACb,GAAIO,KAAAA,CAAMQ,SAAS,CAAK,IAAA;QACtB8B,gBAAkB,EAAA;AAChB,UAAA,CAAA,8CAAA,CAAA;aACGtC,KAAMuC,CAAAA,GAAAA,CAAI,CAAC1B,IAAS,KAAA,CAAA,MAAA,EAASA,KAAKrB,IAAI,CAAA,GAAA,EAAMqB,IAAKpB,CAAAA,WAAW,CAAG,CAAA,CAAA;AAClE+C,SAAAA,CAAAA,IAAAA,CAAK,IAAA;AACT;AACF,KAAA;AACF;AAEA,EAAA,IACc5E,SAAuC,GAAA;AACnD,IAAA,MAAM6E,SAAY,GAAA,IAAA,CAAKpG,KAAMuB,CAAAA,SAAAA,IAAa,EAAC;AAC3C,IAAA,MAAM8E,gBAA8C,GAAA;MAClD7E,MAAQ8E,EAAAA,4BAAAA;MACRrE,IAAMsE,EAAAA;AACR,KAAA;AAEA,IAAA,OAAOC,MAAOH,CAAAA,gBAAAA,CACZ,CAAA,CAACI,KAAKC,eAAAA,KAAAA;AACJ,MAAMC,MAAAA,QAAAA,GAAWP,SAAUK,CAAAA,GAAAA,CAAQC,IAAAA,eAAAA;AACnC,MAAA,IAAIC,oBAAoBC,cAAgB,EAAA;AACtC,QAAOD,OAAAA,QAAAA;AACT;AACA,MAAOA,OAAAA,QAAAA,CAASD,eAAAA,CAAoBA,IAAAA,eAAAA;KACtC,CAAA;AAEJ;EAEAG,cAAiB,GAAA;AACf,IAAO,OAAA;AACL,MAAA,GAAG,MAAMA,cAAAA,EAAAA;MACT7G,KAAO8G,EAAAA,WAAAA,CAAY,KAAK9G,KAAK,CAAA;AAC7BF,MAAAA,OAAAA,EAAS,IAAKA,CAAAA;AAChB,KAAA;AACF;AAEA,EAAA,IAAIkB,OAAOA,MAAoB,EAAA;AAC7B,IAAA,IAAA,CAAKhB,MAAMgB,MAASA,GAAAA,MAAAA;AACtB;AAEA,EAAA,IAAIA,MAAS,GAAA;AACX,IAAA,OAAO,KAAKhB,KAAMgB,CAAAA,MAAAA;AACpB;AACF;;;IAlCW+F,UAAY,EAAA","file":"agent.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { AgentError, BaseAgent } from \"@/agents/base.js\";\nimport { AnyTool, DynamicTool, StringToolOutput, ToolError, ToolOutput } from \"@/tools/base.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\nimport { AgentMeta } from \"@/agents/types.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport type {\n  ToolCallingAgentExecutionConfig,\n  ToolCallingAgentTemplates,\n  ToolCallingAgentCallbacks,\n  ToolCallingAgentRunInput,\n  ToolCallingAgentRunOptions,\n  ToolCallingAgentRunOutput,\n  ToolCallingAgentRunState,\n} from \"@/agents/toolCalling/types.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { ChatModel } from \"@/backend/chat.js\";\nimport { shallowCopy } from \"@/serializer/utils.js\";\nimport { UnconstrainedMemory } from \"@/memory/unconstrainedMemory.js\";\nimport { AssistantMessage, SystemMessage, ToolMessage, UserMessage } from \"@/backend/message.js\";\nimport { isEmpty, isString } from \"remeda\";\nimport { RetryCounter } from \"@/internals/helpers/counter.js\";\nimport { mapObj, omitUndefined } from \"@/internals/helpers/object.js\";\nimport { Cache } from \"@/cache/decoratorCache.js\";\nimport { PromptTemplate } from \"@/template.js\";\nimport {\n  ToolCallingAgentSystemPrompt,\n  ToolCallingAgentTaskPrompt,\n} from \"@/agents/toolCalling/prompts.js\";\nimport { z, ZodSchema } from \"zod\";\n\nexport type ToolCallingAgentTemplateFactory<K extends keyof ToolCallingAgentTemplates> = (\n  template: ToolCallingAgentTemplates[K],\n) => ToolCallingAgentTemplates[K];\n\nexport interface ToolCallingAgentInput {\n  llm: ChatModel;\n  memory: BaseMemory;\n  tools: AnyTool[];\n  meta?: Omit<AgentMeta, \"tools\">;\n  templates?: Partial<{\n    [K in keyof ToolCallingAgentTemplates]:\n      | ToolCallingAgentTemplates[K]\n      | ToolCallingAgentTemplateFactory<K>;\n  }>;\n  execution?: ToolCallingAgentExecutionConfig;\n  saveIntermediateSteps?: boolean;\n}\n\nexport class ToolCallingAgent extends BaseAgent<\n  ToolCallingAgentRunInput,\n  ToolCallingAgentRunOutput,\n  ToolCallingAgentRunOptions\n> {\n  public readonly emitter = Emitter.root.child<ToolCallingAgentCallbacks>({\n    namespace: [\"agent\", \"toolCalling\"],\n    creator: this,\n  });\n\n  constructor(public readonly input: ToolCallingAgentInput) {\n    super();\n    this.input.saveIntermediateSteps = this.input.saveIntermediateSteps ?? true;\n  }\n\n  static {\n    this.register();\n  }\n\n  protected async _run(\n    input: ToolCallingAgentRunInput,\n    options: ToolCallingAgentRunOptions = {},\n    run: GetRunContext<typeof this>,\n  ): Promise<ToolCallingAgentRunOutput> {\n    const tempMessageKey = \"tempMessage\" as const;\n    const execution = {\n      maxRetriesPerStep: 3,\n      totalMaxRetries: 20,\n      maxIterations: 10,\n      ...omitUndefined(this.input.execution ?? {}),\n      ...omitUndefined(options.execution ?? {}),\n    };\n\n    const state: ToolCallingAgentRunState = {\n      memory: new UnconstrainedMemory(),\n      result: undefined,\n      iteration: 0,\n    };\n    await state.memory.add(\n      new SystemMessage(\n        this.templates.system.render({\n          role: undefined,\n          instructions: undefined,\n        }),\n      ),\n    );\n    await state.memory.addMany(this.memory.messages);\n\n    if (input.prompt) {\n      const userMessage = new UserMessage(\n        this.templates.task.render({\n          prompt: input.prompt,\n          context: input.context,\n          expectedOutput: isString(input.expectedOutput) ? input.expectedOutput : undefined,\n        }),\n      );\n      await state.memory.add(userMessage);\n    }\n\n    const globalRetriesCounter = new RetryCounter(execution.totalMaxRetries || 1, AgentError);\n\n    const usePlainResponse = !input.expectedOutput || !(input.expectedOutput instanceof ZodSchema);\n    const finalAnswerToolSchema = usePlainResponse\n      ? z.object({\n          response: z.string().describe(String(input.expectedOutput ?? \"\")),\n        })\n      : (input.expectedOutput as ZodSchema);\n\n    const finalAnswerTool = new DynamicTool({\n      name: \"final_answer\",\n      description: \"Sends the final answer to the user\",\n      inputSchema: finalAnswerToolSchema,\n      handler: async (input) => {\n        const result = usePlainResponse ? input.response : JSON.stringify(input.response);\n        state.result = new AssistantMessage(result);\n        return new StringToolOutput(\"Message has been sent\");\n      },\n    });\n\n    const tools = [...this.input.tools, finalAnswerTool];\n    let forceFinalAnswer = false;\n\n    while (!state.result) {\n      state.iteration++;\n      if (state.iteration > (execution.totalMaxRetries ?? Infinity)) {\n        throw new AgentError(\n          `Agent was not able to resolve the task in ${state.iteration} iterations.`,\n        );\n      }\n\n      await run.emitter.emit(\"start\", { state });\n      const response = await this.input.llm.create({\n        messages: state.memory.messages.slice(),\n        tools,\n        toolChoice: forceFinalAnswer ? finalAnswerTool : tools.length > 1 ? \"required\" : tools[0],\n        stream: false,\n      });\n      await state.memory.addMany(response.messages);\n\n      const toolCallMessages = response.getToolCalls();\n      for (const toolCall of toolCallMessages) {\n        try {\n          const tool = tools.find((tool) => tool.name === toolCall.toolName);\n          if (!tool) {\n            throw new AgentError(`Tool ${toolCall.toolName} does not exist!`);\n          }\n\n          const toolInput: any = toolCall.args;\n          const toolResponse: ToolOutput = await tool.run(toolInput).context({\n            state,\n            toolCallMsg: toolCall,\n          });\n          await state.memory.add(\n            new ToolMessage({\n              type: \"tool-result\",\n              toolCallId: toolCall.toolCallId,\n              toolName: toolCall.toolName,\n              result: toolResponse.getTextContent(),\n              isError: false,\n            }),\n          );\n        } catch (e) {\n          if (e instanceof ToolError) {\n            globalRetriesCounter.use(e);\n            await state.memory.add(\n              new ToolMessage({\n                type: \"tool-result\",\n                toolCallId: toolCall.toolCallId,\n                toolName: toolCall.toolName,\n                result: e.explain(),\n                isError: true,\n              }),\n            );\n          } else {\n            throw e;\n          }\n        }\n      }\n\n      // handle empty messages for some models\n      const textMessages = response.getTextMessages();\n      if (isEmpty(toolCallMessages) && isEmpty(textMessages)) {\n        await state.memory.add(new AssistantMessage(\"\\n\", { [tempMessageKey]: true }));\n      } else {\n        await state.memory.deleteMany(\n          state.memory.messages.filter((msg) => msg.meta[tempMessageKey]),\n        );\n      }\n\n      // Fallback for providers that do not support structured outputs\n      if (!isEmpty(textMessages) && isEmpty(toolCallMessages)) {\n        forceFinalAnswer = true;\n        tools.length = 0;\n        tools.push(finalAnswerTool);\n      }\n\n      await run.emitter.emit(\"success\", { state });\n    }\n\n    if (this.input.saveIntermediateSteps) {\n      this.memory.reset();\n      await this.memory.addMany(state.memory.messages.slice(1));\n    } else {\n      await this.memory.addMany(state.memory.messages.slice(-2));\n    }\n    return { memory: state.memory, result: state.result };\n  }\n\n  get meta(): AgentMeta {\n    const tools = this.input.tools.slice();\n\n    if (this.input.meta) {\n      return { ...this.input.meta, tools };\n    }\n\n    return {\n      name: \"ToolCalling\",\n      tools,\n      description: \"ToolCallingAgent that uses tools to accomplish the task.\",\n      ...(tools.length > 0 && {\n        extraDescription: [\n          `Tools that I can use to accomplish given task.`,\n          ...tools.map((tool) => `Tool '${tool.name}': ${tool.description}.`),\n        ].join(\"\\n\"),\n      }),\n    };\n  }\n\n  @Cache({ enumerable: false })\n  protected get templates(): ToolCallingAgentTemplates {\n    const overrides = this.input.templates ?? {};\n    const defaultTemplates: ToolCallingAgentTemplates = {\n      system: ToolCallingAgentSystemPrompt,\n      task: ToolCallingAgentTaskPrompt,\n    } as const;\n\n    return mapObj(defaultTemplates)(\n      (key, defaultTemplate: ToolCallingAgentTemplates[typeof key]) => {\n        const override = overrides[key] ?? defaultTemplate;\n        if (override instanceof PromptTemplate) {\n          return override;\n        }\n        return override(defaultTemplate) ?? defaultTemplate;\n      },\n    );\n  }\n\n  createSnapshot() {\n    return {\n      ...super.createSnapshot(),\n      input: shallowCopy(this.input),\n      emitter: this.emitter,\n    };\n  }\n\n  set memory(memory: BaseMemory) {\n    this.input.memory = memory;\n  }\n\n  get memory() {\n    return this.input.memory;\n  }\n}\n"]}

package/dist/agents/toolCalling/prompts.cjs

@@ -26,13 +26,14 @@ Your instructions are:
 {{.}}
 {{/instructions}}
 
-When the user sends a message, figure out a solution and provide a final answer.
-You can use tools to improve your answers if available.
+When the user sends a message, figure out a solution and provide a final answer to the user by calling the 'final_answer' tool.
+Before you call the 'final_answer' tool, ensure that you have gathered sufficient evidence to support the final answer.
 
 # Best practices
 - Use markdown syntax to format code snippets, links, JSON, tables, images, and files.
 - If the provided task is unclear, ask the user for clarification.
 - Do not refer to tools or tool outputs by name when responding.
+- Do not call the same tool twice with the similar inputs.
 
 # Date and Time
 The current date and time is: {{formatDate}}

package/dist/agents/toolCalling/prompts.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["prompts.ts"],"names":["ToolCallingAgentSystemPrompt","PromptTemplate","schema","z","object","role","string","instructions","optional","functions","formatDate","Date","toISOString","defaults","template","ToolCallingAgentTaskPrompt","prompt","context","expectedOutput"],"mappings":";;;;;;;AAmBaA,MAAAA,4BAAAA,GAA+B,IAAIC,2BAAe,CAAA;AAC7DC,EAAAA,MAAAA,EAAQC,MAAEC,MAAO,CAAA;AACfC,IAAAA,IAAAA,EAAMF,MAAEG,MAAM,EAAA;IACdC,YAAcJ,EAAAA,KAAAA,CAAEG,MAAM,EAAA,CAAGE,QAAQ;GACnC,CAAA;EACAC,SAAW,EAAA;AACTC,IAAAA,UAAAA,kBAAY,MAAA,CAAA,WAAA;AACV,MAAO,OAAA,iBAAA,IAAIC,IAAAA,EAAAA,EAAOC,WAAW,EAAA;KADnB,EAAA,YAAA;AAGd,GAAA;EACAC,QAAU,EAAA;IAAER,IAAM,EAAA,wBAAA;IAA0BE,YAAc,EAAA;AAAG,GAAA;EAC7DO,QAAU,EAAA,CAAA;;;;;;;;;;;;;;;;;;;AAmBZ,CAAA;AAEaC,MAAAA,0BAAAA,GAA6B,IAAId,2BAAe,CAAA;AAC3DC,EAAAA,MAAAA,EAAQC,MAAEC,MAAO,CAAA;AACfY,IAAAA,MAAAA,EAAQb,MAAEG,MAAM,EAAA;IAChBW,OAASd,EAAAA,KAAAA,CAAEG,MAAM,EAAA,CAAGE,QAAQ,EAAA;IAC5BU,cAAgBf,EAAAA,KAAAA,CAAEG,MAAM,EAAA,CAAGE,QAAQ;GACrC,CAAA;EACAM,QAAU,EAAA,CAAA;;;;;;;;;;;AAWZ,CAAA","file":"prompts.cjs","sourcesContent":["/**\n * Copyright 2025 IBM Corp.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { PromptTemplate } from \"@/template.js\";\nimport { z } from \"zod\";\n\nexport const ToolCallingAgentSystemPrompt = new PromptTemplate({\n  schema: z.object({\n    role: z.string(),\n    instructions: z.string().optional(),\n  }),\n  functions: {\n    formatDate: function () {\n      return new Date().toISOString();\n    },\n  },\n  defaults: { role: \"A helpful AI assistant\", instructions: \"\" },\n  template: `Assume the role of {{role}}.\n{{#instructions}}\n\nYour instructions are:\n{{.}}\n{{/instructions}}\n\nWhen the user sends a message, figure out a solution and provide a final answer.\nYou can use tools to improve your answers if available.\n\n# Best practices\n- Use markdown syntax to format code snippets, links, JSON, tables, images, and files.\n- If the provided task is unclear, ask the user for clarification.\n- Do not refer to tools or tool outputs by name when responding.\n\n# Date and Time\nThe current date and time is: {{formatDate}}\nYou do not need a tool to get the current Date and Time. Use the information available here.\n`,\n});\n\nexport const ToolCallingAgentTaskPrompt = new PromptTemplate({\n  schema: z.object({\n    prompt: z.string(),\n    context: z.string().optional(),\n    expectedOutput: z.string().optional(),\n  }),\n  template: `{{#context}}This is the context that you are working with:\n{{.}}\n\n{{/context}}\n{{#expectedOutput}}\nThis is the expected criteria for your output:\n{{.}}\n\n{{/expectedOutput}}\nYour task: {{prompt}}\n`,\n});\n"]}
+{"version":3,"sources":["prompts.ts"],"names":["ToolCallingAgentSystemPrompt","PromptTemplate","schema","z","object","role","string","instructions","optional","functions","formatDate","Date","toISOString","defaults","template","ToolCallingAgentTaskPrompt","prompt","context","expectedOutput"],"mappings":";;;;;;;AAmBaA,MAAAA,4BAAAA,GAA+B,IAAIC,2BAAe,CAAA;AAC7DC,EAAAA,MAAAA,EAAQC,MAAEC,MAAO,CAAA;AACfC,IAAAA,IAAAA,EAAMF,MAAEG,MAAM,EAAA;IACdC,YAAcJ,EAAAA,KAAAA,CAAEG,MAAM,EAAA,CAAGE,QAAQ;GACnC,CAAA;EACAC,SAAW,EAAA;AACTC,IAAAA,UAAAA,kBAAY,MAAA,CAAA,WAAA;AACV,MAAO,OAAA,iBAAA,IAAIC,IAAAA,EAAAA,EAAOC,WAAW,EAAA;KADnB,EAAA,YAAA;AAGd,GAAA;EACAC,QAAU,EAAA;IAAER,IAAM,EAAA,wBAAA;IAA0BE,YAAc,EAAA;AAAG,GAAA;EAC7DO,QAAU,EAAA,CAAA;;;;;;;;;;;;;;;;;;;;AAoBZ,CAAA;AAEaC,MAAAA,0BAAAA,GAA6B,IAAId,2BAAe,CAAA;AAC3DC,EAAAA,MAAAA,EAAQC,MAAEC,MAAO,CAAA;AACfY,IAAAA,MAAAA,EAAQb,MAAEG,MAAM,EAAA;IAChBW,OAASd,EAAAA,KAAAA,CAAEG,MAAM,EAAA,CAAGE,QAAQ,EAAA;IAC5BU,cAAgBf,EAAAA,KAAAA,CAAEG,MAAM,EAAA,CAAGE,QAAQ;GACrC,CAAA;EACAM,QAAU,EAAA,CAAA;;;;;;;;;;;AAWZ,CAAA","file":"prompts.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { PromptTemplate } from \"@/template.js\";\nimport { z } from \"zod\";\n\nexport const ToolCallingAgentSystemPrompt = new PromptTemplate({\n  schema: z.object({\n    role: z.string(),\n    instructions: z.string().optional(),\n  }),\n  functions: {\n    formatDate: function () {\n      return new Date().toISOString();\n    },\n  },\n  defaults: { role: \"A helpful AI assistant\", instructions: \"\" },\n  template: `Assume the role of {{role}}.\n{{#instructions}}\n\nYour instructions are:\n{{.}}\n{{/instructions}}\n\nWhen the user sends a message, figure out a solution and provide a final answer to the user by calling the 'final_answer' tool.\nBefore you call the 'final_answer' tool, ensure that you have gathered sufficient evidence to support the final answer.\n\n# Best practices\n- Use markdown syntax to format code snippets, links, JSON, tables, images, and files.\n- If the provided task is unclear, ask the user for clarification.\n- Do not refer to tools or tool outputs by name when responding.\n- Do not call the same tool twice with the similar inputs.\n\n# Date and Time\nThe current date and time is: {{formatDate}}\nYou do not need a tool to get the current Date and Time. Use the information available here.\n`,\n});\n\nexport const ToolCallingAgentTaskPrompt = new PromptTemplate({\n  schema: z.object({\n    prompt: z.string(),\n    context: z.string().optional(),\n    expectedOutput: z.string().optional(),\n  }),\n  template: `{{#context}}This is the context that you are working with:\n{{.}}\n\n{{/context}}\n{{#expectedOutput}}\nThis is the expected criteria for your output:\n{{.}}\n\n{{/expectedOutput}}\nYour task: {{prompt}}\n`,\n});\n"]}

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

@@ -7,7 +7,7 @@ import '../../internals/serializable.cjs';
 import 'ajv';
 
 /**
- * Copyright 2025 IBM Corp.
+ * Copyright 2025 © BeeAI a Series of LF Projects, LLC
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.

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

@@ -7,7 +7,7 @@ import '../../internals/serializable.js';
 import 'ajv';
 
 /**
- * Copyright 2025 IBM Corp.
+ * Copyright 2025 © BeeAI a Series of LF Projects, LLC
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.

package/dist/agents/toolCalling/prompts.js

@@ -24,13 +24,14 @@ Your instructions are:
 {{.}}
 {{/instructions}}
 
-When the user sends a message, figure out a solution and provide a final answer.
-You can use tools to improve your answers if available.
+When the user sends a message, figure out a solution and provide a final answer to the user by calling the 'final_answer' tool.
+Before you call the 'final_answer' tool, ensure that you have gathered sufficient evidence to support the final answer.
 
 # Best practices
 - Use markdown syntax to format code snippets, links, JSON, tables, images, and files.
 - If the provided task is unclear, ask the user for clarification.
 - Do not refer to tools or tool outputs by name when responding.
+- Do not call the same tool twice with the similar inputs.
 
 # Date and Time
 The current date and time is: {{formatDate}}

package/dist/agents/toolCalling/prompts.js.map

@@ -1 +1 @@
-{"version":3,"sources":["prompts.ts"],"names":["ToolCallingAgentSystemPrompt","PromptTemplate","schema","z","object","role","string","instructions","optional","functions","formatDate","Date","toISOString","defaults","template","ToolCallingAgentTaskPrompt","prompt","context","expectedOutput"],"mappings":";;;;;AAmBaA,MAAAA,4BAAAA,GAA+B,IAAIC,cAAe,CAAA;AAC7DC,EAAAA,MAAAA,EAAQC,EAAEC,MAAO,CAAA;AACfC,IAAAA,IAAAA,EAAMF,EAAEG,MAAM,EAAA;IACdC,YAAcJ,EAAAA,CAAAA,CAAEG,MAAM,EAAA,CAAGE,QAAQ;GACnC,CAAA;EACAC,SAAW,EAAA;AACTC,IAAAA,UAAAA,kBAAY,MAAA,CAAA,WAAA;AACV,MAAO,OAAA,iBAAA,IAAIC,IAAAA,EAAAA,EAAOC,WAAW,EAAA;KADnB,EAAA,YAAA;AAGd,GAAA;EACAC,QAAU,EAAA;IAAER,IAAM,EAAA,wBAAA;IAA0BE,YAAc,EAAA;AAAG,GAAA;EAC7DO,QAAU,EAAA,CAAA;;;;;;;;;;;;;;;;;;;AAmBZ,CAAA;AAEaC,MAAAA,0BAAAA,GAA6B,IAAId,cAAe,CAAA;AAC3DC,EAAAA,MAAAA,EAAQC,EAAEC,MAAO,CAAA;AACfY,IAAAA,MAAAA,EAAQb,EAAEG,MAAM,EAAA;IAChBW,OAASd,EAAAA,CAAAA,CAAEG,MAAM,EAAA,CAAGE,QAAQ,EAAA;IAC5BU,cAAgBf,EAAAA,CAAAA,CAAEG,MAAM,EAAA,CAAGE,QAAQ;GACrC,CAAA;EACAM,QAAU,EAAA,CAAA;;;;;;;;;;;AAWZ,CAAA","file":"prompts.js","sourcesContent":["/**\n * Copyright 2025 IBM Corp.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { PromptTemplate } from \"@/template.js\";\nimport { z } from \"zod\";\n\nexport const ToolCallingAgentSystemPrompt = new PromptTemplate({\n  schema: z.object({\n    role: z.string(),\n    instructions: z.string().optional(),\n  }),\n  functions: {\n    formatDate: function () {\n      return new Date().toISOString();\n    },\n  },\n  defaults: { role: \"A helpful AI assistant\", instructions: \"\" },\n  template: `Assume the role of {{role}}.\n{{#instructions}}\n\nYour instructions are:\n{{.}}\n{{/instructions}}\n\nWhen the user sends a message, figure out a solution and provide a final answer.\nYou can use tools to improve your answers if available.\n\n# Best practices\n- Use markdown syntax to format code snippets, links, JSON, tables, images, and files.\n- If the provided task is unclear, ask the user for clarification.\n- Do not refer to tools or tool outputs by name when responding.\n\n# Date and Time\nThe current date and time is: {{formatDate}}\nYou do not need a tool to get the current Date and Time. Use the information available here.\n`,\n});\n\nexport const ToolCallingAgentTaskPrompt = new PromptTemplate({\n  schema: z.object({\n    prompt: z.string(),\n    context: z.string().optional(),\n    expectedOutput: z.string().optional(),\n  }),\n  template: `{{#context}}This is the context that you are working with:\n{{.}}\n\n{{/context}}\n{{#expectedOutput}}\nThis is the expected criteria for your output:\n{{.}}\n\n{{/expectedOutput}}\nYour task: {{prompt}}\n`,\n});\n"]}
+{"version":3,"sources":["prompts.ts"],"names":["ToolCallingAgentSystemPrompt","PromptTemplate","schema","z","object","role","string","instructions","optional","functions","formatDate","Date","toISOString","defaults","template","ToolCallingAgentTaskPrompt","prompt","context","expectedOutput"],"mappings":";;;;;AAmBaA,MAAAA,4BAAAA,GAA+B,IAAIC,cAAe,CAAA;AAC7DC,EAAAA,MAAAA,EAAQC,EAAEC,MAAO,CAAA;AACfC,IAAAA,IAAAA,EAAMF,EAAEG,MAAM,EAAA;IACdC,YAAcJ,EAAAA,CAAAA,CAAEG,MAAM,EAAA,CAAGE,QAAQ;GACnC,CAAA;EACAC,SAAW,EAAA;AACTC,IAAAA,UAAAA,kBAAY,MAAA,CAAA,WAAA;AACV,MAAO,OAAA,iBAAA,IAAIC,IAAAA,EAAAA,EAAOC,WAAW,EAAA;KADnB,EAAA,YAAA;AAGd,GAAA;EACAC,QAAU,EAAA;IAAER,IAAM,EAAA,wBAAA;IAA0BE,YAAc,EAAA;AAAG,GAAA;EAC7DO,QAAU,EAAA,CAAA;;;;;;;;;;;;;;;;;;;;AAoBZ,CAAA;AAEaC,MAAAA,0BAAAA,GAA6B,IAAId,cAAe,CAAA;AAC3DC,EAAAA,MAAAA,EAAQC,EAAEC,MAAO,CAAA;AACfY,IAAAA,MAAAA,EAAQb,EAAEG,MAAM,EAAA;IAChBW,OAASd,EAAAA,CAAAA,CAAEG,MAAM,EAAA,CAAGE,QAAQ,EAAA;IAC5BU,cAAgBf,EAAAA,CAAAA,CAAEG,MAAM,EAAA,CAAGE,QAAQ;GACrC,CAAA;EACAM,QAAU,EAAA,CAAA;;;;;;;;;;;AAWZ,CAAA","file":"prompts.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { PromptTemplate } from \"@/template.js\";\nimport { z } from \"zod\";\n\nexport const ToolCallingAgentSystemPrompt = new PromptTemplate({\n  schema: z.object({\n    role: z.string(),\n    instructions: z.string().optional(),\n  }),\n  functions: {\n    formatDate: function () {\n      return new Date().toISOString();\n    },\n  },\n  defaults: { role: \"A helpful AI assistant\", instructions: \"\" },\n  template: `Assume the role of {{role}}.\n{{#instructions}}\n\nYour instructions are:\n{{.}}\n{{/instructions}}\n\nWhen the user sends a message, figure out a solution and provide a final answer to the user by calling the 'final_answer' tool.\nBefore you call the 'final_answer' tool, ensure that you have gathered sufficient evidence to support the final answer.\n\n# Best practices\n- Use markdown syntax to format code snippets, links, JSON, tables, images, and files.\n- If the provided task is unclear, ask the user for clarification.\n- Do not refer to tools or tool outputs by name when responding.\n- Do not call the same tool twice with the similar inputs.\n\n# Date and Time\nThe current date and time is: {{formatDate}}\nYou do not need a tool to get the current Date and Time. Use the information available here.\n`,\n});\n\nexport const ToolCallingAgentTaskPrompt = new PromptTemplate({\n  schema: z.object({\n    prompt: z.string(),\n    context: z.string().optional(),\n    expectedOutput: z.string().optional(),\n  }),\n  template: `{{#context}}This is the context that you are working with:\n{{.}}\n\n{{/context}}\n{{#expectedOutput}}\nThis is the expected criteria for your output:\n{{.}}\n\n{{/expectedOutput}}\nYour task: {{prompt}}\n`,\n});\n"]}

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

@@ -1,6 +1,6 @@
 import { BaseMemory } from '../../memory/base.cjs';
 import { AssistantMessage } from '../../backend/message.cjs';
-import { C as Callback } from '../../emitter-BWtGHYn0.cjs';
+import { C as Callback } from '../../emitter-BTr99H7I.cjs';
 import { ToolCallingAgentSystemPrompt, ToolCallingAgentTaskPrompt } from './prompts.cjs';
 import { ZodSchema } from 'zod';
 import '../../errors.cjs';
@@ -12,7 +12,7 @@ import '../../template.cjs';
 import 'ajv';
 
 /**
- * Copyright 2025 IBM Corp.
+ * Copyright 2025 © BeeAI a Series of LF Projects, LLC
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.

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

@@ -1,6 +1,6 @@
 import { BaseMemory } from '../../memory/base.js';
 import { AssistantMessage } from '../../backend/message.js';
-import { C as Callback } from '../../emitter-l0W9gC1A.js';
+import { C as Callback } from '../../emitter-DiGmS_5O.js';
 import { ToolCallingAgentSystemPrompt, ToolCallingAgentTaskPrompt } from './prompts.js';
 import { ZodSchema } from 'zod';
 import '../../errors.js';
@@ -12,7 +12,7 @@ import '../../template.js';
 import 'ajv';
 
 /**
- * Copyright 2025 IBM Corp.
+ * Copyright 2025 © BeeAI a Series of LF Projects, LLC
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.

package/dist/agents/types.d.cts

@@ -1,8 +1,8 @@
-export { d as AgentCallback, c as AgentCallbackValue, b as AgentMeta, f as AnyAgent, G as GetAgentInput, e as GetAgentOutput, I as InternalAgentCallbackValue, P as PublicAgentCallbackValue } from '../base-Csr9guq2.cjs';
+export { d as AgentCallback, c as AgentCallbackValue, b as AgentMeta, f as AnyAgent, G as GetAgentInput, e as GetAgentOutput, I as InternalAgentCallbackValue, P as PublicAgentCallbackValue } from '../base-BPsHNV2O.cjs';
 import '../internals/types.cjs';
 import '../tools/base.cjs';
 import '../context.cjs';
-import '../emitter-BWtGHYn0.cjs';
+import '../emitter-BTr99H7I.cjs';
 import '../internals/serializable.cjs';
 import '../internals/helpers/guards.cjs';
 import '../internals/helpers/promise.cjs';

package/dist/agents/types.d.ts

@@ -1,8 +1,8 @@
-export { d as AgentCallback, c as AgentCallbackValue, b as AgentMeta, f as AnyAgent, G as GetAgentInput, e as GetAgentOutput, I as InternalAgentCallbackValue, P as PublicAgentCallbackValue } from '../base-DHhpWHQm.js';
+export { d as AgentCallback, c as AgentCallbackValue, b as AgentMeta, f as AnyAgent, G as GetAgentInput, e as GetAgentOutput, I as InternalAgentCallbackValue, P as PublicAgentCallbackValue } from '../base-ZnwDK6ZB.js';
 import '../internals/types.js';
 import '../tools/base.js';
 import '../context.js';
-import '../emitter-l0W9gC1A.js';
+import '../emitter-DiGmS_5O.js';
 import '../internals/serializable.js';
 import '../internals/helpers/guards.js';
 import '../internals/helpers/promise.js';