feat(agent): simplify tools' schema that is passing to prompt

Signed-off-by: Tomas Dvorak <[email protected]>
i-am-bee · Oct 4, 2024 · c927201 · c927201
1 parent 699e45f
commit c927201
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/agents/bee/runner.ts b/src/agents/bee/runner.ts
@@ -119,6 +119,7 @@ export class BeeAgentRunner {
         },
       },
     });
+
     await memory.addMany([
       BaseMessage.of({
         role: Role.SYSTEM,
@@ -127,7 +128,13 @@ export class BeeAgentRunner {
             input.tools.map(async (tool) => ({
               name: tool.name,
               description: tool.description.replaceAll("\n", ".").replace(/\.$/, "").concat("."),
-              schema: JSON.stringify(await tool.getInputJsonSchema()),
+              schema: JSON.stringify(
+                await tool.getInputJsonSchema(),
+                (() => {
+                  const ignoredKeys = new Set(["minLength", "maxLength", "$schema"]);
+                  return (key, value) => (ignoredKeys.has(key) ? undefined : value);
+                })(),
+              ),
             })),
           ),
           instructions: undefined,