feat(agent): make repetition checker disabled by default

i-am-bee · Sep 10, 2024 · 4bc3a60 · 4bc3a60
1 parent e414674
commit 4bc3a60
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/agents/bee/parser.ts b/src/agents/bee/parser.ts
@@ -61,13 +61,18 @@ interface Callbacks {
 
 class RepetitionChecker {
   protected stash: string[] = [];
+  public enabled = true;
 
   constructor(
     protected readonly size: number,
     protected threshold: number,
   ) {}
 
   add(chunk: string) {
+    if (!this.enabled) {
+      return false;
+    }
+
     if (this.stash.length > this.size) {
       this.stash.shift();
     }
@@ -96,6 +101,7 @@ export class BeeOutputParser {
     repetition: {
       size: 20,
       threshold: 10,
+      enabled: true,
     },
   };
 
@@ -122,6 +128,7 @@ export class BeeOutputParser {
       BeeOutputParser.Config.repetition.size,
       BeeOutputParser.Config.repetition.threshold,
     );
+    this.repetitionChecker.enabled = BeeOutputParser.Config.repetition.enabled;
   }
 
   async add(chunk: string) {