Support load_tool

requirements.txt

@@ -0,0 +1 @@
+transformers

tool_config.json

@@ -0,0 +1,5 @@
+{
+  "description": "This is a tool that uses WolframAlpha to compute any mathematical question. It takes one input query, and returns a verbose result in xml format, which includes the solution.",
+  "name": "wolfram_alpha",
+  "tool_class": "wolfram_alpha_tool.WolframAlpha"
+}

wolfram_alpha_tool.py

@@ -0,0 +1,37 @@
+from transformers import Tool
+
+class WolframAlpha(Tool):
+    name = "wolfram_alpha"
+    description = ("This is a tool that uses WolframAlpha to compute any mathematical question. It takes one input query, and returns a verbose result in xml format, which includes the solution.")
+
+    inputs = ["query"]
+    outputs = ["result"]
+
+    def __call__(self, query, output='xml', pod_format='plaintext'):
+        base_url = 'http://api.wolframalpha.com/v2/query'
+
+        output_opts = ['xml','json']
+        format_opts = ['plaintext', 'image', 'image,imagemap', 'mathml', 'sound', 'wav']
+
+        if output not in output_opts:
+            return f"{output} is not a correct output parameter! It must be one of these: {output_opts}"
+
+        if pod_format not in format_opts:
+            return f"{pod_format} is not a correct pod_format parameter! It must be one of these: {format_opts}"
+
+        params = {
+            'input': query,
+            'output': output,
+            'appid': CFG_appid,
+        }
+        print("About to response")
+        response = requests.get(base_url, params=params)
+        print("response:",response)
+        if response.status_code == 200:
+            if output == 'xml':
+                return response.text
+            elif output == 'json':
+                # Remove unnecessary empty spaces
+                return str(response.json())
+        else:
+            return f"There was an error with the request, with response: {response}"