fix(agent): ignore newlines for a tool name

Signed-off-by: Tomas Dvorak <[email protected]>
i-am-bee · Oct 15, 2024 · a2c6202 · a2c6202
1 parent ed6405d
commit a2c6202
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/src/agents/bee/runner.ts b/src/agents/bee/runner.ts
@@ -180,7 +180,10 @@ export class BeeAgentRunner {
           prefix: "Function Name:",
           next: ["tool_input"],
           field: new ZodParserField(
-            z.enum(tools.map((tool) => tool.name) as [string, ...string[]]),
+            z.pipeline(
+              z.string().trim(),
+              z.enum(tools.map((tool) => tool.name) as [string, ...string[]]),
+            ),
           ),
         },
         tool_input: {

diff --git a/src/agents/parsers/field.test.ts b/src/agents/parsers/field.test.ts
@@ -94,6 +94,16 @@ describe("Parser Fields", () => {
   });
 
   describe("Zod", () => {
+    it("Enum", async () => {
+      const values = {
+        apple: "apple",
+      };
+      const field = new ZodParserField(z.pipeline(z.string().trim(), z.nativeEnum(values)));
+      field.write(" apple\n");
+      await field.end();
+      expect(field.get()).toBe(values.apple);
+    });
+
     it("Number", async () => {
       const field = new ZodParserField(z.coerce.number().int());
       expect(field.getPartial()).toBe("");