feat(agent): add template for tool not found error

i-am-bee · Sep 3, 2024 · c4ea1c8 · c4ea1c8
1 parent 6cda741
commit c4ea1c8
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 9 deletions.
diff --git a/src/agents/bee/prompts.ts b/src/agents/bee/prompts.ts
@@ -114,3 +114,11 @@ export const BeeToolNoResultsPrompt = new PromptTemplate({
   variables: [],
   template: `No results were found!`,
 });
+
+export const BeeToolNotFoundPrompt = new PromptTemplate({
+  variables: ["tools"],
+  template: `Tool does not exist!
+{{#tools.length}}
+Use one of the following tools: {{#trim}}{{#tools}}{{name}},{{/tools}}{{/trim}}
+{{/tools.length}}`,
+});
diff --git a/src/agents/bee/runner.ts b/src/agents/bee/runner.ts
@@ -31,6 +31,7 @@ import {
   BeeToolErrorPrompt,
   BeeToolInputErrorPrompt,
   BeeToolNoResultsPrompt,
+  BeeToolNotFoundPrompt,
   BeeUserPrompt,
 } from "@/agents/bee/prompts.js";
 import { BeeIterationToolResult, BeeOutputParser } from "@/agents/bee/parser.js";
@@ -190,17 +191,18 @@ export class BeeAgentRunner {
       (tool) => tool.name.trim().toUpperCase() == iteration.tool_name?.toUpperCase(),
     );
     if (!tool) {
-      this.failedAttemptsCounter.use();
-      const availableTools = this.input.tools.map((tool) => tool.name);
+      this.failedAttemptsCounter.use(
+        new AgentError(`Agent was trying to use non-existing tool "${iteration.tool_name}"`, [], {
+          context: { iteration, meta },
+        }),
+      );
+
+      const template = this.input.templates?.toolNotFoundError ?? BeeToolNotFoundPrompt;
       return {
         success: false,
-        output: [
-          `Tool does not exist!`,
-          availableTools.length > 0 &&
-            `Use one of the following tools: ${availableTools.join(",")}`,
-        ]
-          .filter(Boolean)
-          .join("\n"),
+        output: template.render({
+          tools: this.input.tools,
+        }),
       };
     }
     const options = await (async () => {

diff --git a/src/agents/bee/types.ts b/src/agents/bee/types.ts
@@ -25,6 +25,7 @@ import {
   BeeToolErrorPrompt,
   BeeToolInputErrorPrompt,
   BeeToolNoResultsPrompt,
+  BeeToolNotFoundPrompt,
   BeeUserPrompt,
 } from "@/agents/bee/prompts.js";
 
@@ -126,4 +127,5 @@ export interface BeeAgentTemplates {
   toolError: typeof BeeToolErrorPrompt;
   toolInputError: typeof BeeToolInputErrorPrompt;
   toolNoResultError: typeof BeeToolNoResultsPrompt;
+  toolNotFoundError: typeof BeeToolNotFoundPrompt;
 }