feat(react): improve max retries error message

i-am-bee · Aug 28, 2024 · 54895f9 · 54895f9
1 parent dba7ad7
commit 54895f9
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 4 deletions.
diff --git a/src/agents/bee/runner.ts b/src/agents/bee/runner.ts
@@ -36,6 +36,7 @@ import {
   BeeUserPrompt,
 } from "@/agents/bee/prompts.js";
 import { BeeIterationToolResult, BeeOutputParser } from "@/agents/bee/parser.js";
+import { AgentError } from "@/agents/base.js";
 
 export class BeeAgentRunnerFatalError extends BeeAgentError {
   isFatal = true;
@@ -50,7 +51,7 @@ export class BeeAgentRunner {
     protected readonly options: BeeRunOptions,
     public readonly memory: TokenMemory,
   ) {
-    this.failedAttemptsCounter = new RetryCounter(options?.execution?.totalMaxRetries);
+    this.failedAttemptsCounter = new RetryCounter(options?.execution?.totalMaxRetries, AgentError);
   }
 
   static async create(

diff --git a/src/internals/helpers/counter.ts b/src/internals/helpers/counter.ts
@@ -22,7 +22,10 @@ export class RetryCounter extends Serializable {
   protected readonly maxRetries: number;
   protected lastError?: Error;
 
-  constructor(maxRetries = 0) {
+  constructor(
+    maxRetries = 0,
+    protected ErrorClass: typeof FrameworkError,
+  ) {
     super();
     this.maxRetries = maxRetries;
     this.remaining = maxRetries;
@@ -32,8 +35,8 @@ export class RetryCounter extends Serializable {
     this.lastError = error ?? this.lastError;
 
     if (this.remaining <= 0) {
-      throw new FrameworkError(
-        `Maximal amount of global retries '${this.maxRetries}' has been reached!`,
+      throw new this.ErrorClass(
+        `Maximal amount of global retries (${this.maxRetries}) has been reached.`,
         this.lastError ? [this.lastError] : undefined,
         { isFatal: true },
       );
@@ -46,6 +49,7 @@ export class RetryCounter extends Serializable {
       remaining: this.remaining,
       maxRetries: this.maxRetries,
       lastError: this.lastError,
+      ErrorClass: this.ErrorClass,
     };
   }
 

diff --git a/src/tools/base.ts b/src/tools/base.ts
@@ -189,6 +189,7 @@ export abstract class Tool<
 
   protected toError(e: Error, context: any) {
     if (e instanceof ToolError) {
+      Object.assign(e.context, context);
       return e;
     } else {
       return new ToolError(`Tool "${this.name}" has occurred an error!`, [e], {