Support boolean tool args (#34208)

aymeric-roucher · web-flow · commit 1fb575fcf0ed · 2024-10-23T16:48:21.000+02:00
Support boolean tool arguments
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)

Original file line number	Diff line number	Diff line change
`@@ -138,7 +138,7 @@ def validate_arguments(self):`
`138`	`138`	`"inputs": Dict,`
`139`	`139`	`"output_type": str,`
`140`	`140`	`}`
`141`		`- authorized_types = ["string", "integer", "number", "image", "audio", "any"]`
	`141`	`+ authorized_types = ["string", "integer", "number", "image", "audio", "any", "boolean"]`
`142`	`142`
`143`	`143`	`for attr, expected_type in required_attributes.items():`
`144`	`144`	`attr_value = getattr(self, attr, None)`