From 04b5c459b8795f3ab9d199e0305e832147d1aede Mon Sep 17 00:00:00 2001
From: Tomas Dvorak <toomas2d@gmail.com>
Date: Tue, 3 Dec 2024 20:10:03 +0100
Subject: [PATCH] feat(agents)!: set auto retry for Bee as a default behaviour

Signed-off-by: Tomas Dvorak <toomas2d@gmail.com>
---
 src/agents/bee/agent.ts | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/src/agents/bee/agent.ts b/src/agents/bee/agent.ts
index 62c40901..5f7d673d 100644
--- a/src/agents/bee/agent.ts
+++ b/src/agents/bee/agent.ts
@@ -102,7 +102,18 @@ export class BeeAgent extends BaseAgent<BeeRunInput, BeeRunOutput, BeeRunOptions
     options: BeeRunOptions = {},
     run: GetRunContext<typeof this>,
   ): Promise<BeeRunOutput> {
-    const runner = new this.runner(this.input, options, run);
+    const runner = new this.runner(
+      this.input,
+      {
+        ...options,
+        execution: options?.execution ?? {
+          maxRetriesPerStep: 3,
+          totalMaxRetries: 20,
+          maxIterations: 10,
+        },
+      },
+      run,
+    );
     await runner.init(input);
 
     let finalMessage: BaseMessage | undefined;