Support boolean tool args (#34208)

Support boolean tool arguments
huggingface · Oct 23, 2024 · 1fb575f · 1fb575f
1 parent 343c8cb
commit 1fb575f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/agents/tools.py b/src/transformers/agents/tools.py
@@ -138,7 +138,7 @@ def validate_arguments(self):
             "inputs": Dict,
             "output_type": str,
         }
-        authorized_types = ["string", "integer", "number", "image", "audio", "any"]
+        authorized_types = ["string", "integer", "number", "image", "audio", "any", "boolean"]
 
         for attr, expected_type in required_attributes.items():
             attr_value = getattr(self, attr, None)