beeai-framework 0.1.24 → 0.1.25

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

@@ -3,17 +3,17 @@ import '../../backend/message.js';
 import '../../tools/base.js';
 import '../../errors.js';
 import '../../template.js';
-import '../../chat-D8SVtw5Q.js';
-import '../../base-CyFx7BRL.js';
+import '../../chat-DTxmaKP0.js';
+import '../../base-BfhNzUYf.js';
 import 'zod';
-export { h as RequirementAgentCallbacks, c as RequirementAgentExecutionConfig, f as RequirementAgentOutput, a as RequirementAgentRequest, e as RequirementAgentRunInput, g as RequirementAgentRunOptions, R as RequirementAgentRunState, l as RequirementAgentRunStateStep, b as RequirementAgentSystemPromptInputSchema, i as RequirementAgentTaskPromptInputSchema, d as RequirementAgentTemplates, j as RequirementAgentToolErrorPromptInputSchema, k as RequirementAgentToolNoResultPromptInputSchema } from '../../types-Dyd4e1G6.js';
+export { h as RequirementAgentCallbacks, c as RequirementAgentExecutionConfig, f as RequirementAgentOutput, a as RequirementAgentRequest, e as RequirementAgentRunInput, g as RequirementAgentRunOptions, R as RequirementAgentRunState, l as RequirementAgentRunStateStep, b as RequirementAgentSystemPromptInputSchema, i as RequirementAgentTaskPromptInputSchema, d as RequirementAgentTemplates, j as RequirementAgentToolErrorPromptInputSchema, k as RequirementAgentToolNoResultPromptInputSchema } from '../../types-YT-WdK3m.js';
 import '../../internals/serializable.js';
 import '../../internals/types.js';
 import '../../internals/helpers/guards.js';
 import 'ai';
 import 'ajv';
 import '../../context.js';
-import '../../emitter-36-9MnvA.js';
+import '../../emitter-CL56qab4.js';
 import '../../internals/helpers/promise.js';
 import 'promise-based-task';
 import '../../cache/base.js';

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

@@ -2,7 +2,7 @@ import { AnyTool } from '../../../tools/base.cjs';
 import { RunContext } from '../../../context.cjs';
 import { SystemMessage } from '../../../backend/message.cjs';
 import { PromptTemplate } from '../../../template.cjs';
-import { F as FinalAnswerTool, R as RequirementAgentRunState, a as RequirementAgentRequest, b as RequirementAgentSystemPromptInputSchema } from '../../../types-D9pn62yF.cjs';
+import { F as FinalAnswerTool, R as RequirementAgentRunState, a as RequirementAgentRequest, b as RequirementAgentSystemPromptInputSchema } from '../../../types-YoNQSxQH.cjs';
 import { Requirement, Rule } from '../requirements/requirement.cjs';
 import 'ajv';
 import '../../../errors.cjs';
@@ -14,15 +14,15 @@ import '../../../cache/base.cjs';
 import '../../../internals/helpers/schema.cjs';
 import 'zod';
 import 'zod-to-json-schema';
-import '../../../emitter-jN55XZZq.cjs';
+import '../../../emitter-UkIkNrek.cjs';
 import '../../../internals/helpers/promise.cjs';
 import 'ai';
 import '../../../memory/base.cjs';
-import '../../../chat-CC0s8z23.cjs';
+import '../../../chat-DaqV3woF.cjs';
 import '../../../backend/constants.cjs';
 import '../../../logger/logger.cjs';
 import 'pino';
-import '../../../base-XfWeXQ-S.cjs';
+import '../../../base-BqzN2QwB.cjs';
 
 /**
  * Copyright 2025 © BeeAI a Series of LF Projects, LLC

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

@@ -2,7 +2,7 @@ import { AnyTool } from '../../../tools/base.js';
 import { RunContext } from '../../../context.js';
 import { SystemMessage } from '../../../backend/message.js';
 import { PromptTemplate } from '../../../template.js';
-import { F as FinalAnswerTool, R as RequirementAgentRunState, a as RequirementAgentRequest, b as RequirementAgentSystemPromptInputSchema } from '../../../types-Dyd4e1G6.js';
+import { F as FinalAnswerTool, R as RequirementAgentRunState, a as RequirementAgentRequest, b as RequirementAgentSystemPromptInputSchema } from '../../../types-YT-WdK3m.js';
 import { Requirement, Rule } from '../requirements/requirement.js';
 import 'ajv';
 import '../../../errors.js';
@@ -14,15 +14,15 @@ import '../../../cache/base.js';
 import '../../../internals/helpers/schema.js';
 import 'zod';
 import 'zod-to-json-schema';
-import '../../../emitter-36-9MnvA.js';
+import '../../../emitter-CL56qab4.js';
 import '../../../internals/helpers/promise.js';
 import 'ai';
 import '../../../memory/base.js';
-import '../../../chat-D8SVtw5Q.js';
+import '../../../chat-DTxmaKP0.js';
 import '../../../backend/constants.js';
 import '../../../logger/logger.js';
 import 'pino';
-import '../../../base-CyFx7BRL.js';
+import '../../../base-BfhNzUYf.js';
 
 /**
  * Copyright 2025 © BeeAI a Series of LF Projects, LLC

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

@@ -1,14 +1,25 @@
 'use strict';
 
 var base_cjs = require('../../../tools/base.cjs');
-var errors_cjs = require('../../../errors.cjs');
+var errors_cjs$1 = require('../../../errors.cjs');
 var emitter_cjs = require('../../../emitter/emitter.cjs');
 var zod = require('zod');
 var message_cjs = require('../../../backend/message.cjs');
+var utils_cjs = require('../../../adapters/vercel/backend/utils.cjs');
+var errors_cjs = require('../../../backend/errors.cjs');
 
 var __defProp = Object.defineProperty;
 var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 async function runTool(tools, msg, context) {
+  if (!utils_cjs.isToolCallValid(msg)) {
+    throw new errors_cjs.ChatModelToolCallError(`Tool '${msg.toolName}' does not exist!`, [], {
+      generatedContent: JSON.stringify({
+        name: msg.toolName,
+        parameters: msg.input
+      }),
+      generatedError: "The generated tool call is invalid. Cannot parse the args."
+    });
+  }
   const result = {
     msg,
     tool: null,
@@ -28,7 +39,7 @@ async function runTool(tools, msg, context) {
     });
   } catch (e) {
     if (e instanceof base_cjs.ToolError) {
-      result.error = errors_cjs.FrameworkError.ensure(e);
+      result.error = errors_cjs$1.FrameworkError.ensure(e);
     } else {
       throw e;
     }

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

@@ -1 +1 @@
-{"version":3,"sources":["../../../../src/agents/requirement/utils/tool.ts"],"names":["runTool","tools","msg","context","result","tool","input","output","StringToolOutput","error","find","name","toolName","ToolError","run","toolCallMsg","Tool","contextKeys","Memory","state","memory","e","FrameworkError","ensure","FinalAnswerToolSchema","z","object","response","string","describe","FinalAnswerTool","description","instructions","customSchema","emitter","expectedOutput","undefined","ZodSchema","Emitter","root","child","namespace","creator","inputSchema","_run","_options","_ctx","answer","AssistantMessage","JSON","stringify"],"mappings":";;;;;;;;;;AAiCA,eAAsBA,OAAAA,CACpBC,KAAAA,EACAC,GAAAA,EACAC,OAAAA,EAA4B;AAE5B,EAAA,MAAMC,MAAAA,GAA+B;AACnCF,IAAAA,GAAAA;IACAG,IAAAA,EAAM,IAAA;AACNC,IAAAA,KAAAA,EAAOJ,GAAAA,CAAII,KAAAA;IACXC,MAAAA,EAAQ,IAAIC,0BAAiB,EAAA,CAAA;IAC7BC,KAAAA,EAAO;AACT,GAAA;AAEA,EAAA,IAAI;AACFL,IAAAA,MAAAA,CAAOC,IAAAA,GAAOJ,MAAMS,IAAAA,CAAK,CAACL,SAASA,IAAAA,CAAKM,IAAAA,KAAST,GAAAA,CAAIU,QAAQ,CAAA,IAAK,IAAA;AAClE,IAAA,IAAI,CAACR,OAAOC,IAAAA,EAAM;AAChB,MAAA,MAAM,IAAIQ,kBAAAA,CAAU,CAAA,MAAA,EAASX,GAAAA,CAAIU,QAAQ,CAAA,iBAAA,CAAmB,CAAA;AAC9D;AAEAR,IAAAA,MAAAA,CAAOG,MAAAA,GAAS,MAAMH,MAAAA,CAAOC,IAAAA,CAAKS,IAAIV,MAAAA,CAAOE,KAAK,EAAEH,OAAAA,CAAQ;MAC1D,GAAGA,OAAAA;MACHY,WAAAA,EAAab,GAAAA;AACb,MAAA,CAACc,aAAAA,CAAKC,WAAAA,CAAYC,MAAM,GAAGf,QAAQgB,KAAAA,CAAMC;KAC3C,CAAA;AACF,GAAA,CAAA,OAASC,CAAAA,EAAG;AACV,IAAA,IAAIA,aAAaR,kBAAAA,EAAW;AAC1BT,MAAAA,MAAAA,CAAOK,KAAAA,GAAQa,yBAAAA,CAAeC,MAAAA,CAAOF,CAAAA,CAAAA;KACvC,MAAO;AACL,MAAA,MAAMA,CAAAA;AACR;AACF;AAEA,EAAA,OAAOjB,MAAAA;AACT;AAjCsBJ,MAAAA,CAAAA,OAAAA,EAAAA,SAAAA,CAAAA;AAoCtB,MAAMwB,qBAAAA,GAAwBC,MAAEC,MAAAA,CAAO;AACrCC,EAAAA,QAAAA,EAAUF,KAAAA,CAAEG,MAAAA,EAAM,CAAGC,QAAAA,CAAS,8BAAA;AAChC,CAAA,CAAA;AAKO,MAAMC,wBAAwBd,aAAAA,CAAAA;EA5ErC;;;EA6EkBL,IAAAA,GAAO,cAAA;EACPoB,WAAAA,GAAc,oCAAA;AACvBC,EAAAA,YAAAA;AACAC,EAAAA,YAAAA;AACSC,EAAAA,OAAAA;AAENC,EAAAA,cAAAA;AACAhB,EAAAA,KAAAA;AAEV,EAAA,WAAA,CAAYgB,gBAA2ChB,KAAAA,EAAiC;AACtF,IAAA,KAAA,EAAK;AACL,IAAA,IAAA,CAAKgB,cAAAA,GAAiBA,cAAAA;AACtB,IAAA,IAAA,CAAKhB,KAAAA,GAAQA,KAAAA;AACb,IAAA,IAAA,CAAKa,YAAAA,GAAe,OAAOG,cAAAA,KAAmB,QAAA,GAAWA,cAAAA,GAAiBC,MAAAA;AAC1E,IAAA,IAAA,CAAKH,eAAeE,cAAAA,YAA0BE,aAAAA;AAC9C,IAAA,IAAA,CAAKH,OAAAA,GAAUI,mBAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAAM;MAChCC,SAAAA,EAAW;AAAC,QAAA,MAAA;AAAQ,QAAA;;MACpBC,OAAAA,EAAS;KACX,CAAA;AACF;EAEAC,WAAAA,GAAyB;AACvB,IAAA,MAAMR,iBAAiB,IAAA,CAAKA,cAAAA;AAE5B,IAAA,IAAI,CAACA,cAAAA,EAAgB;AACnB,MAAA,OAAOX,qBAAAA;AACT,KAAA,MAAA,IAAWW,0BAA0BE,aAAAA,EAAW;AAC9C,MAAA,OAAOF,cAAAA;KACT,MAAA,IAAW,OAAOA,mBAAmB,QAAA,EAAU;AAC7C,MAAA,OAAOV,MAAEC,MAAAA,CAAO;AACdC,QAAAA,QAAAA,EAAUF,KAAAA,CAAEG,MAAAA,EAAM,CAAGC,QAAAA,CAASM,cAAAA;OAChC,CAAA;KACF,MAAO;AACL,MAAA,OAAOX,qBAAAA;AACT;AACF;EAEA,MAAMoB,IAAAA,CACJtC,KAAAA,EACAuC,QAAAA,EACAC,IAAAA,EAC2B;AAC3B,IAAA,IAAA,CAAK3B,MAAMf,MAAAA,GAASE,KAAAA;AAEpB,IAAA,IAAI,IAAA,CAAK6B,0BAA0BE,aAAAA,EAAW;AAE5C,MAAA,IAAA,CAAKlB,MAAM4B,MAAAA,GAAS,IAAIC,6BAAiBC,IAAAA,CAAKC,SAAAA,CAAU5C,KAAAA,CAAAA,CAAAA;KAC1D,MAAO;AAEL,MAAA,IAAA,CAAKa,KAAAA,CAAM4B,MAAAA,GAAS,IAAIC,4BAAAA,CAAiB1C,MAAMqB,QAAQ,CAAA;AACzD;AAEA,IAAA,OAAO,IAAInB,0BAAiB,uBAAA,CAAA;AAC9B;AACF","file":"tool.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport {\n  AnyTool,\n  BaseToolRunOptions,\n  StringToolOutput,\n  Tool,\n  ToolError,\n  ToolOutput,\n} from \"@/tools/base.js\";\nimport { ToolCallPart } from \"ai\";\nimport { FrameworkError } from \"@/errors.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { z, ZodSchema } from \"zod\";\nimport { RequirementAgentRunState } from \"@/agents/requirement/types.js\";\nimport { AssistantMessage } from \"@/backend/message.js\";\nimport { RunContext } from \"@/context.js\";\n\n// Tool invocation result\nexport interface ToolInvocationResult {\n  msg: ToolCallPart;\n  tool: AnyTool | null;\n  input: unknown;\n  output: ToolOutput;\n  error: FrameworkError | null;\n}\n\n/**\n * Run a single tool with error handling\n */\nexport async function runTool(\n  tools: AnyTool[],\n  msg: ToolCallPart,\n  context: Record<string, any>,\n): Promise<ToolInvocationResult> {\n  const result: ToolInvocationResult = {\n    msg,\n    tool: null,\n    input: msg.input,\n    output: new StringToolOutput(\"\"),\n    error: null,\n  };\n\n  try {\n    result.tool = tools.find((tool) => tool.name === msg.toolName) || null;\n    if (!result.tool) {\n      throw new ToolError(`Tool '${msg.toolName}' does not exist!`);\n    }\n\n    result.output = await result.tool.run(result.input).context({\n      ...context,\n      toolCallMsg: msg,\n      [Tool.contextKeys.Memory]: context.state.memory,\n    });\n  } catch (e) {\n    if (e instanceof ToolError) {\n      result.error = FrameworkError.ensure(e);\n    } else {\n      throw e;\n    }\n  }\n\n  return result;\n}\n\n// Final answer tool schema\nconst FinalAnswerToolSchema = z.object({\n  response: z.string().describe(\"The final answer to the user\"),\n});\n\n/**\n * Special tool for capturing final answers\n */\nexport class FinalAnswerTool extends Tool {\n  public readonly name = \"final_answer\";\n  public readonly description = \"Sends the final answer to the user\";\n  public instructions?: string;\n  public customSchema: boolean;\n  public readonly emitter: Emitter<any>;\n\n  protected expectedOutput: string | ZodSchema | null;\n  protected state: RequirementAgentRunState;\n\n  constructor(expectedOutput: string | ZodSchema | null, state: RequirementAgentRunState) {\n    super();\n    this.expectedOutput = expectedOutput;\n    this.state = state;\n    this.instructions = typeof expectedOutput === \"string\" ? expectedOutput : undefined;\n    this.customSchema = expectedOutput instanceof ZodSchema;\n    this.emitter = Emitter.root.child({\n      namespace: [\"tool\", \"final_answer\"],\n      creator: this,\n    });\n  }\n\n  inputSchema(): ZodSchema {\n    const expectedOutput = this.expectedOutput;\n\n    if (!expectedOutput) {\n      return FinalAnswerToolSchema;\n    } else if (expectedOutput instanceof ZodSchema) {\n      return expectedOutput;\n    } else if (typeof expectedOutput === \"string\") {\n      return z.object({\n        response: z.string().describe(expectedOutput),\n      });\n    } else {\n      return FinalAnswerToolSchema;\n    }\n  }\n\n  async _run(\n    input: any,\n    _options: BaseToolRunOptions,\n    _ctx: RunContext<typeof this>,\n  ): Promise<StringToolOutput> {\n    this.state.result = input;\n\n    if (this.expectedOutput instanceof ZodSchema) {\n      // For custom schemas, serialize the entire input\n      this.state.answer = new AssistantMessage(JSON.stringify(input));\n    } else {\n      // For string schemas, use the response field\n      this.state.answer = new AssistantMessage(input.response);\n    }\n\n    return new StringToolOutput(\"Message has been sent\");\n  }\n}\n"]}
+{"version":3,"sources":["../../../../src/agents/requirement/utils/tool.ts"],"names":["runTool","tools","msg","context","isToolCallValid","ChatModelToolCallError","toolName","generatedContent","JSON","stringify","name","parameters","input","generatedError","result","tool","output","StringToolOutput","error","find","ToolError","run","toolCallMsg","Tool","contextKeys","Memory","state","memory","e","FrameworkError","ensure","FinalAnswerToolSchema","z","object","response","string","describe","FinalAnswerTool","description","instructions","customSchema","emitter","expectedOutput","undefined","ZodSchema","Emitter","root","child","namespace","creator","inputSchema","_run","_options","_ctx","answer","AssistantMessage"],"mappings":";;;;;;;;;;;;AAmCA,eAAsBA,OAAAA,CACpBC,KAAAA,EACAC,GAAAA,EACAC,OAAAA,EAA4B;AAE5B,EAAA,IAAI,CAACC,yBAAAA,CAAgBF,GAAAA,CAAAA,EAAM;AACzB,IAAA,MAAM,IAAIG,iCAAAA,CAAuB,CAAA,MAAA,EAASH,IAAII,QAAQ,CAAA,iBAAA,CAAA,EAAqB,EAAA,EAAI;AAC7EC,MAAAA,gBAAAA,EAAkBC,KAAKC,SAAAA,CAAU;AAAEC,QAAAA,IAAAA,EAAMR,GAAAA,CAAII,QAAAA;AAAUK,QAAAA,UAAAA,EAAYT,GAAAA,CAAIU;OAAM,CAAA;MAC7EC,cAAAA,EAAgB;KAClB,CAAA;AACF;AAEA,EAAA,MAAMC,MAAAA,GAA+B;AACnCZ,IAAAA,GAAAA;IACAa,IAAAA,EAAM,IAAA;AACNH,IAAAA,KAAAA,EAAOV,GAAAA,CAAIU,KAAAA;IACXI,MAAAA,EAAQ,IAAIC,0BAAiB,EAAA,CAAA;IAC7BC,KAAAA,EAAO;AACT,GAAA;AAEA,EAAA,IAAI;AACFJ,IAAAA,MAAAA,CAAOC,IAAAA,GAAOd,MAAMkB,IAAAA,CAAK,CAACJ,SAASA,IAAAA,CAAKL,IAAAA,KAASR,GAAAA,CAAII,QAAQ,CAAA,IAAK,IAAA;AAClE,IAAA,IAAI,CAACQ,OAAOC,IAAAA,EAAM;AAChB,MAAA,MAAM,IAAIK,kBAAAA,CAAU,CAAA,MAAA,EAASlB,GAAAA,CAAII,QAAQ,CAAA,iBAAA,CAAmB,CAAA;AAC9D;AAEAQ,IAAAA,MAAAA,CAAOE,MAAAA,GAAS,MAAMF,MAAAA,CAAOC,IAAAA,CAAKM,IAAIP,MAAAA,CAAOF,KAAK,EAAET,OAAAA,CAAQ;MAC1D,GAAGA,OAAAA;MACHmB,WAAAA,EAAapB,GAAAA;AACb,MAAA,CAACqB,aAAAA,CAAKC,WAAAA,CAAYC,MAAM,GAAGtB,QAAQuB,KAAAA,CAAMC;KAC3C,CAAA;AACF,GAAA,CAAA,OAASC,CAAAA,EAAG;AACV,IAAA,IAAIA,aAAaR,kBAAAA,EAAW;AAC1BN,MAAAA,MAAAA,CAAOI,KAAAA,GAAQW,2BAAAA,CAAeC,MAAAA,CAAOF,CAAAA,CAAAA;KACvC,MAAO;AACL,MAAA,MAAMA,CAAAA;AACR;AACF;AAEA,EAAA,OAAOd,MAAAA;AACT;AAxCsBd,MAAAA,CAAAA,OAAAA,EAAAA,SAAAA,CAAAA;AA2CtB,MAAM+B,qBAAAA,GAAwBC,MAAEC,MAAAA,CAAO;AACrCC,EAAAA,QAAAA,EAAUF,KAAAA,CAAEG,MAAAA,EAAM,CAAGC,QAAAA,CAAS,8BAAA;AAChC,CAAA,CAAA;AAKO,MAAMC,wBAAwBd,aAAAA,CAAAA;EArFrC;;;EAsFkBb,IAAAA,GAAO,cAAA;EACP4B,WAAAA,GAAc,oCAAA;AACvBC,EAAAA,YAAAA;AACAC,EAAAA,YAAAA;AACSC,EAAAA,OAAAA;AAENC,EAAAA,cAAAA;AACAhB,EAAAA,KAAAA;AAEV,EAAA,WAAA,CAAYgB,gBAA2ChB,KAAAA,EAAiC;AACtF,IAAA,KAAA,EAAK;AACL,IAAA,IAAA,CAAKgB,cAAAA,GAAiBA,cAAAA;AACtB,IAAA,IAAA,CAAKhB,KAAAA,GAAQA,KAAAA;AACb,IAAA,IAAA,CAAKa,YAAAA,GAAe,OAAOG,cAAAA,KAAmB,QAAA,GAAWA,cAAAA,GAAiBC,MAAAA;AAC1E,IAAA,IAAA,CAAKH,eAAeE,cAAAA,YAA0BE,aAAAA;AAC9C,IAAA,IAAA,CAAKH,OAAAA,GAAUI,mBAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAAM;MAChCC,SAAAA,EAAW;AAAC,QAAA,MAAA;AAAQ,QAAA;;MACpBC,OAAAA,EAAS;KACX,CAAA;AACF;EAEAC,WAAAA,GAAyB;AACvB,IAAA,MAAMR,iBAAiB,IAAA,CAAKA,cAAAA;AAE5B,IAAA,IAAI,CAACA,cAAAA,EAAgB;AACnB,MAAA,OAAOX,qBAAAA;AACT,KAAA,MAAA,IAAWW,0BAA0BE,aAAAA,EAAW;AAC9C,MAAA,OAAOF,cAAAA;KACT,MAAA,IAAW,OAAOA,mBAAmB,QAAA,EAAU;AAC7C,MAAA,OAAOV,MAAEC,MAAAA,CAAO;AACdC,QAAAA,QAAAA,EAAUF,KAAAA,CAAEG,MAAAA,EAAM,CAAGC,QAAAA,CAASM,cAAAA;OAChC,CAAA;KACF,MAAO;AACL,MAAA,OAAOX,qBAAAA;AACT;AACF;EAEA,MAAMoB,IAAAA,CACJvC,KAAAA,EACAwC,QAAAA,EACAC,IAAAA,EAC2B;AAC3B,IAAA,IAAA,CAAK3B,MAAMZ,MAAAA,GAASF,KAAAA;AAEpB,IAAA,IAAI,IAAA,CAAK8B,0BAA0BE,aAAAA,EAAW;AAE5C,MAAA,IAAA,CAAKlB,MAAM4B,MAAAA,GAAS,IAAIC,6BAAiB/C,IAAAA,CAAKC,SAAAA,CAAUG,KAAAA,CAAAA,CAAAA;KAC1D,MAAO;AAEL,MAAA,IAAA,CAAKc,KAAAA,CAAM4B,MAAAA,GAAS,IAAIC,4BAAAA,CAAiB3C,MAAMsB,QAAQ,CAAA;AACzD;AAEA,IAAA,OAAO,IAAIjB,0BAAiB,uBAAA,CAAA;AAC9B;AACF","file":"tool.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport {\n  AnyTool,\n  BaseToolRunOptions,\n  StringToolOutput,\n  Tool,\n  ToolError,\n  ToolOutput,\n} from \"@/tools/base.js\";\nimport { ToolCallPart } from \"ai\";\nimport { FrameworkError } from \"@/errors.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { z, ZodSchema } from \"zod\";\nimport { RequirementAgentRunState } from \"@/agents/requirement/types.js\";\nimport { AssistantMessage } from \"@/backend/message.js\";\nimport { RunContext } from \"@/context.js\";\nimport { isToolCallValid } from \"@/adapters/vercel/backend/utils.js\";\nimport { ChatModelToolCallError } from \"@/backend/errors.js\";\n\n// Tool invocation result\nexport interface ToolInvocationResult {\n  msg: ToolCallPart;\n  tool: AnyTool | null;\n  input: unknown;\n  output: ToolOutput;\n  error: FrameworkError | null;\n}\n\n/**\n * Run a single tool with error handling\n */\nexport async function runTool(\n  tools: AnyTool[],\n  msg: ToolCallPart,\n  context: Record<string, any>,\n): Promise<ToolInvocationResult> {\n  if (!isToolCallValid(msg)) {\n    throw new ChatModelToolCallError(`Tool '${msg.toolName}' does not exist!`, [], {\n      generatedContent: JSON.stringify({ name: msg.toolName, parameters: msg.input }),\n      generatedError: \"The generated tool call is invalid. Cannot parse the args.\",\n    });\n  }\n\n  const result: ToolInvocationResult = {\n    msg,\n    tool: null,\n    input: msg.input,\n    output: new StringToolOutput(\"\"),\n    error: null,\n  };\n\n  try {\n    result.tool = tools.find((tool) => tool.name === msg.toolName) || null;\n    if (!result.tool) {\n      throw new ToolError(`Tool '${msg.toolName}' does not exist!`);\n    }\n\n    result.output = await result.tool.run(result.input).context({\n      ...context,\n      toolCallMsg: msg,\n      [Tool.contextKeys.Memory]: context.state.memory,\n    });\n  } catch (e) {\n    if (e instanceof ToolError) {\n      result.error = FrameworkError.ensure(e);\n    } else {\n      throw e;\n    }\n  }\n\n  return result;\n}\n\n// Final answer tool schema\nconst FinalAnswerToolSchema = z.object({\n  response: z.string().describe(\"The final answer to the user\"),\n});\n\n/**\n * Special tool for capturing final answers\n */\nexport class FinalAnswerTool extends Tool {\n  public readonly name = \"final_answer\";\n  public readonly description = \"Sends the final answer to the user\";\n  public instructions?: string;\n  public customSchema: boolean;\n  public readonly emitter: Emitter<any>;\n\n  protected expectedOutput: string | ZodSchema | null;\n  protected state: RequirementAgentRunState;\n\n  constructor(expectedOutput: string | ZodSchema | null, state: RequirementAgentRunState) {\n    super();\n    this.expectedOutput = expectedOutput;\n    this.state = state;\n    this.instructions = typeof expectedOutput === \"string\" ? expectedOutput : undefined;\n    this.customSchema = expectedOutput instanceof ZodSchema;\n    this.emitter = Emitter.root.child({\n      namespace: [\"tool\", \"final_answer\"],\n      creator: this,\n    });\n  }\n\n  inputSchema(): ZodSchema {\n    const expectedOutput = this.expectedOutput;\n\n    if (!expectedOutput) {\n      return FinalAnswerToolSchema;\n    } else if (expectedOutput instanceof ZodSchema) {\n      return expectedOutput;\n    } else if (typeof expectedOutput === \"string\") {\n      return z.object({\n        response: z.string().describe(expectedOutput),\n      });\n    } else {\n      return FinalAnswerToolSchema;\n    }\n  }\n\n  async _run(\n    input: any,\n    _options: BaseToolRunOptions,\n    _ctx: RunContext<typeof this>,\n  ): Promise<StringToolOutput> {\n    this.state.result = input;\n\n    if (this.expectedOutput instanceof ZodSchema) {\n      // For custom schemas, serialize the entire input\n      this.state.answer = new AssistantMessage(JSON.stringify(input));\n    } else {\n      // For string schemas, use the response field\n      this.state.answer = new AssistantMessage(input.response);\n    }\n\n    return new StringToolOutput(\"Message has been sent\");\n  }\n}\n"]}

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

@@ -1,9 +1,9 @@
 import '../../../tools/base.cjs';
 import 'ai';
 import '../../../errors.cjs';
-import '../../../emitter-jN55XZZq.cjs';
+import '../../../emitter-UkIkNrek.cjs';
 import 'zod';
-export { F as FinalAnswerTool, T as ToolInvocationResult, r as runTool } from '../../../types-D9pn62yF.cjs';
+export { F as FinalAnswerTool, T as ToolInvocationResult, r as runTool } from '../../../types-YoNQSxQH.cjs';
 import '../../../context.cjs';
 import 'ajv';
 import '../../../internals/serializable.cjs';
@@ -17,8 +17,8 @@ import '../../../internals/helpers/promise.cjs';
 import '../../../memory/base.cjs';
 import '../../../backend/message.cjs';
 import '../../../template.cjs';
-import '../../../chat-CC0s8z23.cjs';
+import '../../../chat-DaqV3woF.cjs';
 import '../../../backend/constants.cjs';
 import '../../../logger/logger.cjs';
 import 'pino';
-import '../../../base-XfWeXQ-S.cjs';
+import '../../../base-BqzN2QwB.cjs';

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

@@ -1,9 +1,9 @@
 import '../../../tools/base.js';
 import 'ai';
 import '../../../errors.js';
-import '../../../emitter-36-9MnvA.js';
+import '../../../emitter-CL56qab4.js';
 import 'zod';
-export { F as FinalAnswerTool, T as ToolInvocationResult, r as runTool } from '../../../types-Dyd4e1G6.js';
+export { F as FinalAnswerTool, T as ToolInvocationResult, r as runTool } from '../../../types-YT-WdK3m.js';
 import '../../../context.js';
 import 'ajv';
 import '../../../internals/serializable.js';
@@ -17,8 +17,8 @@ import '../../../internals/helpers/promise.js';
 import '../../../memory/base.js';
 import '../../../backend/message.js';
 import '../../../template.js';
-import '../../../chat-D8SVtw5Q.js';
+import '../../../chat-DTxmaKP0.js';
 import '../../../backend/constants.js';
 import '../../../logger/logger.js';
 import 'pino';
-import '../../../base-CyFx7BRL.js';
+import '../../../base-BfhNzUYf.js';

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

@@ -3,10 +3,21 @@ import { FrameworkError } from '../../../errors.js';
 import { Emitter } from '../../../emitter/emitter.js';
 import { z, ZodSchema } from 'zod';
 import { AssistantMessage } from '../../../backend/message.js';
+import { isToolCallValid } from '../../../adapters/vercel/backend/utils.js';
+import { ChatModelToolCallError } from '../../../backend/errors.js';
 
 var __defProp = Object.defineProperty;
 var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 async function runTool(tools, msg, context) {
+  if (!isToolCallValid(msg)) {
+    throw new ChatModelToolCallError(`Tool '${msg.toolName}' does not exist!`, [], {
+      generatedContent: JSON.stringify({
+        name: msg.toolName,
+        parameters: msg.input
+      }),
+      generatedError: "The generated tool call is invalid. Cannot parse the args."
+    });
+  }
   const result = {
     msg,
     tool: null,

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

@@ -1 +1 @@
-{"version":3,"sources":["../../../../src/agents/requirement/utils/tool.ts"],"names":["runTool","tools","msg","context","result","tool","input","output","StringToolOutput","error","find","name","toolName","ToolError","run","toolCallMsg","Tool","contextKeys","Memory","state","memory","e","FrameworkError","ensure","FinalAnswerToolSchema","z","object","response","string","describe","FinalAnswerTool","description","instructions","customSchema","emitter","expectedOutput","undefined","ZodSchema","Emitter","root","child","namespace","creator","inputSchema","_run","_options","_ctx","answer","AssistantMessage","JSON","stringify"],"mappings":";;;;;;;;AAiCA,eAAsBA,OAAAA,CACpBC,KAAAA,EACAC,GAAAA,EACAC,OAAAA,EAA4B;AAE5B,EAAA,MAAMC,MAAAA,GAA+B;AACnCF,IAAAA,GAAAA;IACAG,IAAAA,EAAM,IAAA;AACNC,IAAAA,KAAAA,EAAOJ,GAAAA,CAAII,KAAAA;IACXC,MAAAA,EAAQ,IAAIC,iBAAiB,EAAA,CAAA;IAC7BC,KAAAA,EAAO;AACT,GAAA;AAEA,EAAA,IAAI;AACFL,IAAAA,MAAAA,CAAOC,IAAAA,GAAOJ,MAAMS,IAAAA,CAAK,CAACL,SAASA,IAAAA,CAAKM,IAAAA,KAAST,GAAAA,CAAIU,QAAQ,CAAA,IAAK,IAAA;AAClE,IAAA,IAAI,CAACR,OAAOC,IAAAA,EAAM;AAChB,MAAA,MAAM,IAAIQ,SAAAA,CAAU,CAAA,MAAA,EAASX,GAAAA,CAAIU,QAAQ,CAAA,iBAAA,CAAmB,CAAA;AAC9D;AAEAR,IAAAA,MAAAA,CAAOG,MAAAA,GAAS,MAAMH,MAAAA,CAAOC,IAAAA,CAAKS,IAAIV,MAAAA,CAAOE,KAAK,EAAEH,OAAAA,CAAQ;MAC1D,GAAGA,OAAAA;MACHY,WAAAA,EAAab,GAAAA;AACb,MAAA,CAACc,IAAAA,CAAKC,WAAAA,CAAYC,MAAM,GAAGf,QAAQgB,KAAAA,CAAMC;KAC3C,CAAA;AACF,GAAA,CAAA,OAASC,CAAAA,EAAG;AACV,IAAA,IAAIA,aAAaR,SAAAA,EAAW;AAC1BT,MAAAA,MAAAA,CAAOK,KAAAA,GAAQa,cAAAA,CAAeC,MAAAA,CAAOF,CAAAA,CAAAA;KACvC,MAAO;AACL,MAAA,MAAMA,CAAAA;AACR;AACF;AAEA,EAAA,OAAOjB,MAAAA;AACT;AAjCsBJ,MAAAA,CAAAA,OAAAA,EAAAA,SAAAA,CAAAA;AAoCtB,MAAMwB,qBAAAA,GAAwBC,EAAEC,MAAAA,CAAO;AACrCC,EAAAA,QAAAA,EAAUF,CAAAA,CAAEG,MAAAA,EAAM,CAAGC,QAAAA,CAAS,8BAAA;AAChC,CAAA,CAAA;AAKO,MAAMC,wBAAwBd,IAAAA,CAAAA;EA5ErC;;;EA6EkBL,IAAAA,GAAO,cAAA;EACPoB,WAAAA,GAAc,oCAAA;AACvBC,EAAAA,YAAAA;AACAC,EAAAA,YAAAA;AACSC,EAAAA,OAAAA;AAENC,EAAAA,cAAAA;AACAhB,EAAAA,KAAAA;AAEV,EAAA,WAAA,CAAYgB,gBAA2ChB,KAAAA,EAAiC;AACtF,IAAA,KAAA,EAAK;AACL,IAAA,IAAA,CAAKgB,cAAAA,GAAiBA,cAAAA;AACtB,IAAA,IAAA,CAAKhB,KAAAA,GAAQA,KAAAA;AACb,IAAA,IAAA,CAAKa,YAAAA,GAAe,OAAOG,cAAAA,KAAmB,QAAA,GAAWA,cAAAA,GAAiBC,MAAAA;AAC1E,IAAA,IAAA,CAAKH,eAAeE,cAAAA,YAA0BE,SAAAA;AAC9C,IAAA,IAAA,CAAKH,OAAAA,GAAUI,OAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAAM;MAChCC,SAAAA,EAAW;AAAC,QAAA,MAAA;AAAQ,QAAA;;MACpBC,OAAAA,EAAS;KACX,CAAA;AACF;EAEAC,WAAAA,GAAyB;AACvB,IAAA,MAAMR,iBAAiB,IAAA,CAAKA,cAAAA;AAE5B,IAAA,IAAI,CAACA,cAAAA,EAAgB;AACnB,MAAA,OAAOX,qBAAAA;AACT,KAAA,MAAA,IAAWW,0BAA0BE,SAAAA,EAAW;AAC9C,MAAA,OAAOF,cAAAA;KACT,MAAA,IAAW,OAAOA,mBAAmB,QAAA,EAAU;AAC7C,MAAA,OAAOV,EAAEC,MAAAA,CAAO;AACdC,QAAAA,QAAAA,EAAUF,CAAAA,CAAEG,MAAAA,EAAM,CAAGC,QAAAA,CAASM,cAAAA;OAChC,CAAA;KACF,MAAO;AACL,MAAA,OAAOX,qBAAAA;AACT;AACF;EAEA,MAAMoB,IAAAA,CACJtC,KAAAA,EACAuC,QAAAA,EACAC,IAAAA,EAC2B;AAC3B,IAAA,IAAA,CAAK3B,MAAMf,MAAAA,GAASE,KAAAA;AAEpB,IAAA,IAAI,IAAA,CAAK6B,0BAA0BE,SAAAA,EAAW;AAE5C,MAAA,IAAA,CAAKlB,MAAM4B,MAAAA,GAAS,IAAIC,iBAAiBC,IAAAA,CAAKC,SAAAA,CAAU5C,KAAAA,CAAAA,CAAAA;KAC1D,MAAO;AAEL,MAAA,IAAA,CAAKa,KAAAA,CAAM4B,MAAAA,GAAS,IAAIC,gBAAAA,CAAiB1C,MAAMqB,QAAQ,CAAA;AACzD;AAEA,IAAA,OAAO,IAAInB,iBAAiB,uBAAA,CAAA;AAC9B;AACF","file":"tool.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport {\n  AnyTool,\n  BaseToolRunOptions,\n  StringToolOutput,\n  Tool,\n  ToolError,\n  ToolOutput,\n} from \"@/tools/base.js\";\nimport { ToolCallPart } from \"ai\";\nimport { FrameworkError } from \"@/errors.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { z, ZodSchema } from \"zod\";\nimport { RequirementAgentRunState } from \"@/agents/requirement/types.js\";\nimport { AssistantMessage } from \"@/backend/message.js\";\nimport { RunContext } from \"@/context.js\";\n\n// Tool invocation result\nexport interface ToolInvocationResult {\n  msg: ToolCallPart;\n  tool: AnyTool | null;\n  input: unknown;\n  output: ToolOutput;\n  error: FrameworkError | null;\n}\n\n/**\n * Run a single tool with error handling\n */\nexport async function runTool(\n  tools: AnyTool[],\n  msg: ToolCallPart,\n  context: Record<string, any>,\n): Promise<ToolInvocationResult> {\n  const result: ToolInvocationResult = {\n    msg,\n    tool: null,\n    input: msg.input,\n    output: new StringToolOutput(\"\"),\n    error: null,\n  };\n\n  try {\n    result.tool = tools.find((tool) => tool.name === msg.toolName) || null;\n    if (!result.tool) {\n      throw new ToolError(`Tool '${msg.toolName}' does not exist!`);\n    }\n\n    result.output = await result.tool.run(result.input).context({\n      ...context,\n      toolCallMsg: msg,\n      [Tool.contextKeys.Memory]: context.state.memory,\n    });\n  } catch (e) {\n    if (e instanceof ToolError) {\n      result.error = FrameworkError.ensure(e);\n    } else {\n      throw e;\n    }\n  }\n\n  return result;\n}\n\n// Final answer tool schema\nconst FinalAnswerToolSchema = z.object({\n  response: z.string().describe(\"The final answer to the user\"),\n});\n\n/**\n * Special tool for capturing final answers\n */\nexport class FinalAnswerTool extends Tool {\n  public readonly name = \"final_answer\";\n  public readonly description = \"Sends the final answer to the user\";\n  public instructions?: string;\n  public customSchema: boolean;\n  public readonly emitter: Emitter<any>;\n\n  protected expectedOutput: string | ZodSchema | null;\n  protected state: RequirementAgentRunState;\n\n  constructor(expectedOutput: string | ZodSchema | null, state: RequirementAgentRunState) {\n    super();\n    this.expectedOutput = expectedOutput;\n    this.state = state;\n    this.instructions = typeof expectedOutput === \"string\" ? expectedOutput : undefined;\n    this.customSchema = expectedOutput instanceof ZodSchema;\n    this.emitter = Emitter.root.child({\n      namespace: [\"tool\", \"final_answer\"],\n      creator: this,\n    });\n  }\n\n  inputSchema(): ZodSchema {\n    const expectedOutput = this.expectedOutput;\n\n    if (!expectedOutput) {\n      return FinalAnswerToolSchema;\n    } else if (expectedOutput instanceof ZodSchema) {\n      return expectedOutput;\n    } else if (typeof expectedOutput === \"string\") {\n      return z.object({\n        response: z.string().describe(expectedOutput),\n      });\n    } else {\n      return FinalAnswerToolSchema;\n    }\n  }\n\n  async _run(\n    input: any,\n    _options: BaseToolRunOptions,\n    _ctx: RunContext<typeof this>,\n  ): Promise<StringToolOutput> {\n    this.state.result = input;\n\n    if (this.expectedOutput instanceof ZodSchema) {\n      // For custom schemas, serialize the entire input\n      this.state.answer = new AssistantMessage(JSON.stringify(input));\n    } else {\n      // For string schemas, use the response field\n      this.state.answer = new AssistantMessage(input.response);\n    }\n\n    return new StringToolOutput(\"Message has been sent\");\n  }\n}\n"]}
+{"version":3,"sources":["../../../../src/agents/requirement/utils/tool.ts"],"names":["runTool","tools","msg","context","isToolCallValid","ChatModelToolCallError","toolName","generatedContent","JSON","stringify","name","parameters","input","generatedError","result","tool","output","StringToolOutput","error","find","ToolError","run","toolCallMsg","Tool","contextKeys","Memory","state","memory","e","FrameworkError","ensure","FinalAnswerToolSchema","z","object","response","string","describe","FinalAnswerTool","description","instructions","customSchema","emitter","expectedOutput","undefined","ZodSchema","Emitter","root","child","namespace","creator","inputSchema","_run","_options","_ctx","answer","AssistantMessage"],"mappings":";;;;;;;;;;AAmCA,eAAsBA,OAAAA,CACpBC,KAAAA,EACAC,GAAAA,EACAC,OAAAA,EAA4B;AAE5B,EAAA,IAAI,CAACC,eAAAA,CAAgBF,GAAAA,CAAAA,EAAM;AACzB,IAAA,MAAM,IAAIG,sBAAAA,CAAuB,CAAA,MAAA,EAASH,IAAII,QAAQ,CAAA,iBAAA,CAAA,EAAqB,EAAA,EAAI;AAC7EC,MAAAA,gBAAAA,EAAkBC,KAAKC,SAAAA,CAAU;AAAEC,QAAAA,IAAAA,EAAMR,GAAAA,CAAII,QAAAA;AAAUK,QAAAA,UAAAA,EAAYT,GAAAA,CAAIU;OAAM,CAAA;MAC7EC,cAAAA,EAAgB;KAClB,CAAA;AACF;AAEA,EAAA,MAAMC,MAAAA,GAA+B;AACnCZ,IAAAA,GAAAA;IACAa,IAAAA,EAAM,IAAA;AACNH,IAAAA,KAAAA,EAAOV,GAAAA,CAAIU,KAAAA;IACXI,MAAAA,EAAQ,IAAIC,iBAAiB,EAAA,CAAA;IAC7BC,KAAAA,EAAO;AACT,GAAA;AAEA,EAAA,IAAI;AACFJ,IAAAA,MAAAA,CAAOC,IAAAA,GAAOd,MAAMkB,IAAAA,CAAK,CAACJ,SAASA,IAAAA,CAAKL,IAAAA,KAASR,GAAAA,CAAII,QAAQ,CAAA,IAAK,IAAA;AAClE,IAAA,IAAI,CAACQ,OAAOC,IAAAA,EAAM;AAChB,MAAA,MAAM,IAAIK,SAAAA,CAAU,CAAA,MAAA,EAASlB,GAAAA,CAAII,QAAQ,CAAA,iBAAA,CAAmB,CAAA;AAC9D;AAEAQ,IAAAA,MAAAA,CAAOE,MAAAA,GAAS,MAAMF,MAAAA,CAAOC,IAAAA,CAAKM,IAAIP,MAAAA,CAAOF,KAAK,EAAET,OAAAA,CAAQ;MAC1D,GAAGA,OAAAA;MACHmB,WAAAA,EAAapB,GAAAA;AACb,MAAA,CAACqB,IAAAA,CAAKC,WAAAA,CAAYC,MAAM,GAAGtB,QAAQuB,KAAAA,CAAMC;KAC3C,CAAA;AACF,GAAA,CAAA,OAASC,CAAAA,EAAG;AACV,IAAA,IAAIA,aAAaR,SAAAA,EAAW;AAC1BN,MAAAA,MAAAA,CAAOI,KAAAA,GAAQW,cAAAA,CAAeC,MAAAA,CAAOF,CAAAA,CAAAA;KACvC,MAAO;AACL,MAAA,MAAMA,CAAAA;AACR;AACF;AAEA,EAAA,OAAOd,MAAAA;AACT;AAxCsBd,MAAAA,CAAAA,OAAAA,EAAAA,SAAAA,CAAAA;AA2CtB,MAAM+B,qBAAAA,GAAwBC,EAAEC,MAAAA,CAAO;AACrCC,EAAAA,QAAAA,EAAUF,CAAAA,CAAEG,MAAAA,EAAM,CAAGC,QAAAA,CAAS,8BAAA;AAChC,CAAA,CAAA;AAKO,MAAMC,wBAAwBd,IAAAA,CAAAA;EArFrC;;;EAsFkBb,IAAAA,GAAO,cAAA;EACP4B,WAAAA,GAAc,oCAAA;AACvBC,EAAAA,YAAAA;AACAC,EAAAA,YAAAA;AACSC,EAAAA,OAAAA;AAENC,EAAAA,cAAAA;AACAhB,EAAAA,KAAAA;AAEV,EAAA,WAAA,CAAYgB,gBAA2ChB,KAAAA,EAAiC;AACtF,IAAA,KAAA,EAAK;AACL,IAAA,IAAA,CAAKgB,cAAAA,GAAiBA,cAAAA;AACtB,IAAA,IAAA,CAAKhB,KAAAA,GAAQA,KAAAA;AACb,IAAA,IAAA,CAAKa,YAAAA,GAAe,OAAOG,cAAAA,KAAmB,QAAA,GAAWA,cAAAA,GAAiBC,MAAAA;AAC1E,IAAA,IAAA,CAAKH,eAAeE,cAAAA,YAA0BE,SAAAA;AAC9C,IAAA,IAAA,CAAKH,OAAAA,GAAUI,OAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAAM;MAChCC,SAAAA,EAAW;AAAC,QAAA,MAAA;AAAQ,QAAA;;MACpBC,OAAAA,EAAS;KACX,CAAA;AACF;EAEAC,WAAAA,GAAyB;AACvB,IAAA,MAAMR,iBAAiB,IAAA,CAAKA,cAAAA;AAE5B,IAAA,IAAI,CAACA,cAAAA,EAAgB;AACnB,MAAA,OAAOX,qBAAAA;AACT,KAAA,MAAA,IAAWW,0BAA0BE,SAAAA,EAAW;AAC9C,MAAA,OAAOF,cAAAA;KACT,MAAA,IAAW,OAAOA,mBAAmB,QAAA,EAAU;AAC7C,MAAA,OAAOV,EAAEC,MAAAA,CAAO;AACdC,QAAAA,QAAAA,EAAUF,CAAAA,CAAEG,MAAAA,EAAM,CAAGC,QAAAA,CAASM,cAAAA;OAChC,CAAA;KACF,MAAO;AACL,MAAA,OAAOX,qBAAAA;AACT;AACF;EAEA,MAAMoB,IAAAA,CACJvC,KAAAA,EACAwC,QAAAA,EACAC,IAAAA,EAC2B;AAC3B,IAAA,IAAA,CAAK3B,MAAMZ,MAAAA,GAASF,KAAAA;AAEpB,IAAA,IAAI,IAAA,CAAK8B,0BAA0BE,SAAAA,EAAW;AAE5C,MAAA,IAAA,CAAKlB,MAAM4B,MAAAA,GAAS,IAAIC,iBAAiB/C,IAAAA,CAAKC,SAAAA,CAAUG,KAAAA,CAAAA,CAAAA;KAC1D,MAAO;AAEL,MAAA,IAAA,CAAKc,KAAAA,CAAM4B,MAAAA,GAAS,IAAIC,gBAAAA,CAAiB3C,MAAMsB,QAAQ,CAAA;AACzD;AAEA,IAAA,OAAO,IAAIjB,iBAAiB,uBAAA,CAAA;AAC9B;AACF","file":"tool.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport {\n  AnyTool,\n  BaseToolRunOptions,\n  StringToolOutput,\n  Tool,\n  ToolError,\n  ToolOutput,\n} from \"@/tools/base.js\";\nimport { ToolCallPart } from \"ai\";\nimport { FrameworkError } from \"@/errors.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { z, ZodSchema } from \"zod\";\nimport { RequirementAgentRunState } from \"@/agents/requirement/types.js\";\nimport { AssistantMessage } from \"@/backend/message.js\";\nimport { RunContext } from \"@/context.js\";\nimport { isToolCallValid } from \"@/adapters/vercel/backend/utils.js\";\nimport { ChatModelToolCallError } from \"@/backend/errors.js\";\n\n// Tool invocation result\nexport interface ToolInvocationResult {\n  msg: ToolCallPart;\n  tool: AnyTool | null;\n  input: unknown;\n  output: ToolOutput;\n  error: FrameworkError | null;\n}\n\n/**\n * Run a single tool with error handling\n */\nexport async function runTool(\n  tools: AnyTool[],\n  msg: ToolCallPart,\n  context: Record<string, any>,\n): Promise<ToolInvocationResult> {\n  if (!isToolCallValid(msg)) {\n    throw new ChatModelToolCallError(`Tool '${msg.toolName}' does not exist!`, [], {\n      generatedContent: JSON.stringify({ name: msg.toolName, parameters: msg.input }),\n      generatedError: \"The generated tool call is invalid. Cannot parse the args.\",\n    });\n  }\n\n  const result: ToolInvocationResult = {\n    msg,\n    tool: null,\n    input: msg.input,\n    output: new StringToolOutput(\"\"),\n    error: null,\n  };\n\n  try {\n    result.tool = tools.find((tool) => tool.name === msg.toolName) || null;\n    if (!result.tool) {\n      throw new ToolError(`Tool '${msg.toolName}' does not exist!`);\n    }\n\n    result.output = await result.tool.run(result.input).context({\n      ...context,\n      toolCallMsg: msg,\n      [Tool.contextKeys.Memory]: context.state.memory,\n    });\n  } catch (e) {\n    if (e instanceof ToolError) {\n      result.error = FrameworkError.ensure(e);\n    } else {\n      throw e;\n    }\n  }\n\n  return result;\n}\n\n// Final answer tool schema\nconst FinalAnswerToolSchema = z.object({\n  response: z.string().describe(\"The final answer to the user\"),\n});\n\n/**\n * Special tool for capturing final answers\n */\nexport class FinalAnswerTool extends Tool {\n  public readonly name = \"final_answer\";\n  public readonly description = \"Sends the final answer to the user\";\n  public instructions?: string;\n  public customSchema: boolean;\n  public readonly emitter: Emitter<any>;\n\n  protected expectedOutput: string | ZodSchema | null;\n  protected state: RequirementAgentRunState;\n\n  constructor(expectedOutput: string | ZodSchema | null, state: RequirementAgentRunState) {\n    super();\n    this.expectedOutput = expectedOutput;\n    this.state = state;\n    this.instructions = typeof expectedOutput === \"string\" ? expectedOutput : undefined;\n    this.customSchema = expectedOutput instanceof ZodSchema;\n    this.emitter = Emitter.root.child({\n      namespace: [\"tool\", \"final_answer\"],\n      creator: this,\n    });\n  }\n\n  inputSchema(): ZodSchema {\n    const expectedOutput = this.expectedOutput;\n\n    if (!expectedOutput) {\n      return FinalAnswerToolSchema;\n    } else if (expectedOutput instanceof ZodSchema) {\n      return expectedOutput;\n    } else if (typeof expectedOutput === \"string\") {\n      return z.object({\n        response: z.string().describe(expectedOutput),\n      });\n    } else {\n      return FinalAnswerToolSchema;\n    }\n  }\n\n  async _run(\n    input: any,\n    _options: BaseToolRunOptions,\n    _ctx: RunContext<typeof this>,\n  ): Promise<StringToolOutput> {\n    this.state.result = input;\n\n    if (this.expectedOutput instanceof ZodSchema) {\n      // For custom schemas, serialize the entire input\n      this.state.answer = new AssistantMessage(JSON.stringify(input));\n    } else {\n      // For string schemas, use the response field\n      this.state.answer = new AssistantMessage(input.response);\n    }\n\n    return new StringToolOutput(\"Message has been sent\");\n  }\n}\n"]}

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

@@ -1,10 +1,10 @@
 import { GetRunContext, MiddlewareType } from '../../context.cjs';
-import { c as AgentMeta, a as BaseAgent } from '../../base-XfWeXQ-S.cjs';
+import { c as AgentMeta, a as BaseAgent } from '../../base-BqzN2QwB.cjs';
 import { AnyTool } from '../../tools/base.cjs';
 import { BaseMemory } from '../../memory/base.cjs';
-import { E as Emitter } from '../../emitter-jN55XZZq.cjs';
+import { E as Emitter } from '../../emitter-UkIkNrek.cjs';
 import { ToolCallingAgentTemplates, ToolCallingAgentExecutionConfig, ToolCallingAgentRunInput, ToolCallingAgentRunOutput, ToolCallingAgentRunOptions, ToolCallingAgentCallbacks } from './types.cjs';
-import { m as ChatModel } from '../../chat-CC0s8z23.cjs';
+import { m as ChatModel } from '../../chat-DaqV3woF.cjs';
 import '../../internals/serializable.cjs';
 import '../../internals/types.cjs';
 import '../../internals/helpers/guards.cjs';

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

@@ -1,10 +1,10 @@
 import { GetRunContext, MiddlewareType } from '../../context.js';
-import { c as AgentMeta, a as BaseAgent } from '../../base-CyFx7BRL.js';
+import { c as AgentMeta, a as BaseAgent } from '../../base-BfhNzUYf.js';
 import { AnyTool } from '../../tools/base.js';
 import { BaseMemory } from '../../memory/base.js';
-import { E as Emitter } from '../../emitter-36-9MnvA.js';
+import { E as Emitter } from '../../emitter-CL56qab4.js';
 import { ToolCallingAgentTemplates, ToolCallingAgentExecutionConfig, ToolCallingAgentRunInput, ToolCallingAgentRunOutput, ToolCallingAgentRunOptions, ToolCallingAgentCallbacks } from './types.js';
-import { m as ChatModel } from '../../chat-D8SVtw5Q.js';
+import { m as ChatModel } from '../../chat-DTxmaKP0.js';
 import '../../internals/serializable.js';
 import '../../internals/types.js';
 import '../../internals/helpers/guards.js';

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-jN55XZZq.cjs';
+import { C as Callback } from '../../emitter-UkIkNrek.cjs';
 import { ToolCallingAgentSystemPrompt, ToolCallingAgentTaskPrompt } from './prompts.cjs';
 import { ZodSchema } from 'zod';
 import '../../errors.cjs';

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-36-9MnvA.js';
+import { C as Callback } from '../../emitter-CL56qab4.js';
 import { ToolCallingAgentSystemPrompt, ToolCallingAgentTaskPrompt } from './prompts.js';
 import { ZodSchema } from 'zod';
 import '../../errors.js';

package/dist/agents/types.d.cts

@@ -1,8 +1,8 @@
-export { e as AgentCallback, d as AgentCallbackValue, c as AgentMeta, b as AnyAgent, G as GetAgentInput, f as GetAgentOutput, I as InternalAgentCallbackValue, P as PublicAgentCallbackValue } from '../base-XfWeXQ-S.cjs';
+export { e as AgentCallback, d as AgentCallbackValue, c as AgentMeta, b as AnyAgent, G as GetAgentInput, f as GetAgentOutput, I as InternalAgentCallbackValue, P as PublicAgentCallbackValue } from '../base-BqzN2QwB.cjs';
 import '../internals/types.cjs';
 import '../tools/base.cjs';
 import '../context.cjs';
-import '../emitter-jN55XZZq.cjs';
+import '../emitter-UkIkNrek.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 { e as AgentCallback, d as AgentCallbackValue, c as AgentMeta, b as AnyAgent, G as GetAgentInput, f as GetAgentOutput, I as InternalAgentCallbackValue, P as PublicAgentCallbackValue } from '../base-CyFx7BRL.js';
+export { e as AgentCallback, d as AgentCallbackValue, c as AgentMeta, b as AnyAgent, G as GetAgentInput, f as GetAgentOutput, I as InternalAgentCallbackValue, P as PublicAgentCallbackValue } from '../base-BfhNzUYf.js';
 import '../internals/types.js';
 import '../tools/base.js';
 import '../context.js';
-import '../emitter-36-9MnvA.js';
+import '../emitter-CL56qab4.js';
 import '../internals/serializable.js';
 import '../internals/helpers/guards.js';
 import '../internals/helpers/promise.js';

package/dist/backend/backend.d.cts

@@ -1,12 +1,12 @@
 import { Serializable } from '../internals/serializable.cjs';
-import { m as ChatModel, F as FullModelName } from '../chat-CC0s8z23.cjs';
+import { m as ChatModel, F as FullModelName } from '../chat-DaqV3woF.cjs';
 import { EmbeddingModel } from './embedding.cjs';
 import { ProviderName } from './constants.cjs';
 import { OptionalExcept } from '../internals/types.cjs';
 import './message.cjs';
 import 'ai';
 import '../context.cjs';
-import '../emitter-jN55XZZq.cjs';
+import '../emitter-UkIkNrek.cjs';
 import '../internals/helpers/promise.cjs';
 import '../errors.cjs';
 import 'promise-based-task';

package/dist/backend/backend.d.ts

@@ -1,12 +1,12 @@
 import { Serializable } from '../internals/serializable.js';
-import { m as ChatModel, F as FullModelName } from '../chat-D8SVtw5Q.js';
+import { m as ChatModel, F as FullModelName } from '../chat-DTxmaKP0.js';
 import { EmbeddingModel } from './embedding.js';
 import { ProviderName } from './constants.js';
 import { OptionalExcept } from '../internals/types.js';
 import './message.js';
 import 'ai';
 import '../context.js';
-import '../emitter-36-9MnvA.js';
+import '../emitter-CL56qab4.js';
 import '../internals/helpers/promise.js';
 import '../errors.js';
 import 'promise-based-task';

package/dist/backend/chat.cjs

@@ -20,6 +20,7 @@ var template_cjs = require('../template.cjs');
 var promise_cjs = require('../internals/helpers/promise.cjs');
 var serializer_cjs = require('../serializer/serializer.cjs');
 var logger_cjs = require('../logger/logger.cjs');
+var utils_cjs$2 = require('../adapters/vercel/backend/utils.cjs');
 
 var __defProp = Object.defineProperty;
 var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
@@ -41,6 +42,8 @@ class ChatModel extends serializable_cjs.Serializable {
   ];
   toolCallFallbackViaResponseFormat = true;
   modelSupportsToolCalling = true;
+  fixInvalidToolCalls = true;
+  retryOnEmptyResponse = true;
   create(input) {
     input = utils_cjs.shallowCopy(input);
     if (input.stream === void 0) {
@@ -64,56 +67,124 @@ class ChatModel extends serializable_cjs.Serializable {
         this.logger.warn(`The following tool choice value '${input.toolChoice}' is not supported. Ignoring.`);
         input.toolChoice = void 0;
       }
+      const modelInputMessagesBackup = input.messages.slice();
       const cacheEntry = await this.createCacheAccessor(input);
       try {
         await run.emitter.emit("start", {
           input
         });
-        const chunks = [];
-        const generator = cacheEntry.value ?? (input.stream ? this._createStream(input, run) : promise_cjs.toAsyncGenerator(this._create(input, run)));
-        const controller = new AbortController();
-        for await (const value of generator) {
-          chunks.push(value);
-          await run.emitter.emit("newToken", {
-            value,
-            callbacks: {
-              abort: /* @__PURE__ */ __name(() => controller.abort(), "abort")
+        const result = await new retryable_cjs.Retryable({
+          executor: /* @__PURE__ */ __name(async () => {
+            const chunks = [];
+            const generator = cacheEntry.value ?? (input.stream ? this._createStream(input, run) : promise_cjs.toAsyncGenerator(this._create(input, run)));
+            const controller = new AbortController();
+            for await (const value of generator) {
+              chunks.push(value);
+              await run.emitter.emit("newToken", {
+                value,
+                callbacks: {
+                  abort: /* @__PURE__ */ __name(() => controller.abort(), "abort")
+                }
+              });
+              if (controller.signal.aborted) {
+                break;
+              }
             }
-          });
-          if (controller.signal.aborted) {
-            break;
-          }
-        }
-        cacheEntry.resolve(chunks);
-        const result = ChatModelOutput.fromChunks(chunks);
-        for (const toolCall of result.getToolCalls()) {
-          if (remeda.isString(toolCall.input)) {
-            toolCall.input = JSON.parse(toolCall.input);
-          }
-        }
-        if (forceToolCallViaResponseFormat && remeda.isEmpty(result.getToolCalls())) {
-          const lastMsg = result.messages.at(-1);
-          const toolCall = schema_cjs.parseBrokenJson(lastMsg.text, {
-            pair: [
-              "{",
-              "}"
-            ]
-          });
-          if (!toolCall || !toolCall.name || !toolCall.parameters) {
-            throw new errors_cjs.ChatModelError(`Failed to produce a valid tool call. Generate output: ${lastMsg.text}`, [], {
-              isFatal: true,
-              isRetryable: false
-            });
-          }
-          lastMsg.content.length = 0;
-          lastMsg.content.push({
-            type: "tool-call",
-            toolCallId: `call_${remeda.randomString(8).toLowerCase()}`,
-            toolName: toolCall.name,
-            // todo: add types
-            input: toolCall.parameters
-          });
-        }
+            const result2 = ChatModelOutput.fromChunks(chunks);
+            if (result2.isEmpty()) {
+              throw new errors_cjs.EmptyChatModelResponseError();
+            }
+            if (remeda.isEmpty(result2.getToolCalls()) && (forceToolCallViaResponseFormat || input.toolChoice === "required" || input.toolChoice instanceof base_cjs.Tool)) {
+              const lastMsg = result2.messages.at(-1);
+              let toolCall = schema_cjs.parseBrokenJson(lastMsg.text, {
+                pair: [
+                  "{",
+                  "}"
+                ]
+              });
+              if (toolCall && !toolCall.name && !toolCall.parameters && input.toolChoice instanceof base_cjs.Tool) {
+                toolCall = {
+                  name: input.toolChoice.name,
+                  parameters: toolCall
+                };
+              }
+              if (!toolCall || !toolCall.name || !toolCall.parameters) {
+                throw new errors_cjs.ChatModelToolCallError(`Failed to produce a valid tool call. Generate output: ${lastMsg.text}`, [], {
+                  generatedContent: lastMsg.text,
+                  generatedError: "Tool call was not produced.",
+                  response: result2
+                });
+              }
+              lastMsg.content.length = 0;
+              lastMsg.content.push({
+                type: "tool-call",
+                toolCallId: `call_${remeda.randomString(8).toLowerCase()}`,
+                toolName: toolCall.name,
+                // todo: add types
+                input: toolCall.parameters
+              });
+            }
+            for (const toolCall of result2.getToolCalls()) {
+              const tool = input.tools?.find((t) => t.name === toolCall.toolName);
+              if (!tool) {
+                const availableTools = input.tools?.map((t) => t.name).join(",") || "None";
+                throw new errors_cjs.ChatModelToolCallError("Non existing tool call.", [], {
+                  generatedError: `Error: Unknown tool '${toolCall.toolName}'.
+Use on of the available tools: ${availableTools}`,
+                  generatedContent: JSON.stringify({
+                    name: toolCall.toolName,
+                    input: remeda.isString(toolCall.input) ? toolCall.input : JSON.stringify(toolCall.input)
+                  }),
+                  response: result2
+                });
+              }
+              if (!utils_cjs$2.isToolCallValid(toolCall)) {
+                throw new errors_cjs.ChatModelToolCallError("Malformed tool call.", [], {
+                  generatedContent: remeda.isString(toolCall.input) ? toolCall.input : JSON.stringify(toolCall.input),
+                  generatedError: `The tool call for the '${toolCall.toolName}' tool has malformed parameters. It must be a valid JSON.`,
+                  response: result2
+                });
+              }
+              if (remeda.isString(toolCall.input)) {
+                toolCall.input = JSON.parse(toolCall.input);
+              }
+            }
+            cacheEntry.resolve(chunks);
+            return result2;
+          }, "executor"),
+          config: {
+            maxRetries: input.maxRetries ?? 0
+          },
+          onRetry: /* @__PURE__ */ __name(async (_, lastError) => {
+            if (this.fixInvalidToolCalls && lastError instanceof errors_cjs.ChatModelToolCallError) {
+              input.messages = input.messages.slice();
+              if (lastError.data.generatedContent) {
+                input.messages.push(new message_cjs.AssistantMessage(lastError.data.generatedContent, {
+                  tempMessage: true
+                }));
+              }
+              const toolNames = input.tools?.map((t) => t.name).join(", ") || "None";
+              input.messages.push(new message_cjs.UserMessage(`${lastError.data.generatedError}
+
+Available Tools: ${toolNames}`, {
+                tempMessage: true
+              }));
+            } else if (this.retryOnEmptyResponse && lastError instanceof errors_cjs.EmptyChatModelResponseError) {
+              input.messages = input.messages.slice();
+              const lastMessage = input.messages.at(-1);
+              if (lastMessage && lastMessage instanceof message_cjs.AssistantMessage && lastMessage.meta["tempMessage"] && lastMessage.text === "") {
+                input.messages.push(new message_cjs.UserMessage("No output received. Please regenerate your previous response.", {
+                  tempMessage: true
+                }));
+              } else {
+                input.messages.push(new message_cjs.AssistantMessage("", {
+                  tempMessage: true
+                }));
+              }
+            }
+          }, "onRetry")
+        }).get();
+        input.messages = modelInputMessagesBackup;
         await run.emitter.emit("success", {
           value: result
         });
@@ -243,7 +314,9 @@ Validation Errors: {{errors}}`
       logger: this.logger,
       toolChoiceSupport: this.toolChoiceSupport.slice(),
       toolCallFallbackViaResponseFormat: this.toolCallFallbackViaResponseFormat,
-      modelSupportsToolCalling: this.modelSupportsToolCalling
+      modelSupportsToolCalling: this.modelSupportsToolCalling,
+      retryOnEmptyResponse: this.retryOnEmptyResponse,
+      fixInvalidToolCalls: this.fixInvalidToolCalls
     };
   }
   destroy() {
@@ -303,6 +376,12 @@ class ChatModelOutput extends serializable_cjs.Serializable {
     chunks.forEach((cur) => final.merge(cur));
     return final;
   }
+  isEmpty() {
+    if (this.messages.length === 0) {
+      return true;
+    }
+    return this.getTextContent() === "" && this.getToolCalls().length === 0;
+  }
   merge(other) {
     if (other.messages.length > 0) {
       const clones = other.messages.map(cloneSync);