Upload tool

app.py

@@ -0,0 +1,4 @@
+from transformers import launch_gradio_demo
+from calculator import SimpleCalculatorTool
+
+launch_gradio_demo(SimpleCalculatorTool)

calculator.py

@@ -0,0 +1,20 @@
+import re
+from transformers import Tool
+from huggingface_hub import list_models
+
+
+class SimpleCalculatorTool(Tool):
+    name = "simple_calculator"
+    description = (
+        "This is a tool that returns the result of a simple mathematical expression written in a string."
+    )
+
+    inputs = ["text"]
+    outputs = ["text"]
+    regex = r"(\d+(?:\.\d+)?)(\+|\-|\*|\/)(\d+(?:\.\d+)?)"
+
+    def __call__(self, input_str: str):
+        match = re.match(self.regex, input_str)
+        if match is None:
+            raise ValueError("Input string is not a valid mathematical expression.")
+        return eval(f"{match.group(1)} {match.group(2)} {match.group(3)}")

requirements.txt

@@ -0,0 +1,3 @@
+transformers
+re
+huggingface_hub

tool_config.json

@@ -0,0 +1,5 @@
+{
+  "description": "This is a tool that returns the result of a simple mathematical expression written in a string.",
+  "name": "simple_calculator",
+  "tool_class": "calculator.SimpleCalculatorTool"
+}