revert(agents): update constrained decoding for Bee (#220)

Signed-off-by: Matous Havlena <[email protected]>
i-am-bee · Dec 4, 2024 · 6e616ed · 6e616ed
1 parent 643cd8e
commit 6e616ed
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/agents/bee/runners/default/runner.ts b/src/agents/bee/runners/default/runner.ts
@@ -363,7 +363,7 @@ export class DefaultRunner extends BaseRunner {
     const parserRegex = isEmpty(tools)
       ? new RegExp(`Thought: .+\\nFinal Answer: [\\s\\S]+`)
       : new RegExp(
-          `Thought: (?!.*Function Name:).+\\n(?:Final Answer: [\\s\\S]+|Function Name: (?:${tools.map((tool) => tool.name).join("|")})\\nFunction Input: \\{.*\\}\\nFunction Output:)`,
+          `Thought: .+\\n(?:Final Answer: [\\s\\S]+|Function Name: (${tools.map((tool) => tool.name).join("|")})\\nFunction Input: \\{.*\\}\\nFunction Output:)`,
         );
 
     const parser = new LinePrefixParser<BeeParserInput>(

diff --git a/src/agents/bee/runners/granite/runner.ts b/src/agents/bee/runners/granite/runner.ts
@@ -90,7 +90,7 @@ export class GraniteRunner extends DefaultRunner {
       parserRegex: isEmpty(tools)
         ? new RegExp(`Thought: .+\\nFinal Answer: [\\s\\S]+`)
         : new RegExp(
-            `Thought: (?!.*Tool Name:).+\\n(?:Final Answer: [\\s\\S]+|Tool Name: (?:${tools.map((tool) => tool.name).join("|")})\\nTool Input: \\{.*\\})`,
+            `Thought: .+\\n(?:Final Answer: [\\s\\S]+|Tool Name: (${tools.map((tool) => tool.name).join("|")})\\nTool Input: \\{.*\\})`,
           ),
       parser: parser.fork<BeeParserInput>((nodes, options) => ({
         options,