Upload tool

requirements.txt

@@ -1,2 +1,3 @@
-requests
+os
 transformers
+requests

wolfram_alpha_tool.py

@@ -1,5 +1,6 @@
-from transformers import Tool
+import os
 import requests
+from transformers import Tool
 
 class WolframAlpha(Tool):
     name = "wolfram_alpha"
@@ -8,9 +9,27 @@ class WolframAlpha(Tool):
     inputs = ["query"]
     outputs = ["result"]
 
+    def __init__(self):
+        self.base_url = 'http://api.wolframalpha.com/v2/query'
+
+        self.app_id = os.environ.get('WOLFRAM_APP_ID')
+        if self.app_id is None:
+            raise ValueError("Please set the `WOLFRAM_APP_ID` as an environment variable in order to instantiate the Wolfram tool.\nTo do so, before instantiating the class, run:\nos.environ['WOLFRAM_APP_ID'] = 'YOUR_WOLFRAM_APP_ID'")
+
+        print("Making sure APP_ID is valid... ", end="")
+        dummy_params = {
+            'input': '1+1',
+            'output': 'xml',
+            'appid': self.app_id,
+        }
+        response = self.make_request(params=dummy_params)
+        if "Invalid appid" in response:
+            appid_url = 'https://developer.wolframalpha.com/portal/myapps/index.html'
+            raise ValueError(f"Please set a valid `WOLFRAM_APP_ID` as an environment variable.\nWolframAlpha is not validating APP_ID: {self.app_id}\nTo get an APP_ID, go to:\n{appid_url}\nand click on [Get an AppID]")
+        print("APP_ID validated! Tool ready to use.")
+
     def __call__(self, query, output='xml', pod_format='plaintext'):
-        CFG_appid="YOUR_WOLFRAM_API_KEY_HERE"
-        base_url = 'http://api.wolframalpha.com/v2/query'
+        
 
         output_opts = ['xml','json']
         format_opts = ['plaintext', 'image', 'image,imagemap', 'mathml', 'sound', 'wav']
@@ -24,13 +43,19 @@ class WolframAlpha(Tool):
         params = {
             'input': query,
             'output': output,
-            'appid': CFG_appid,
+            'appid': self.app_id,
         }
-        response = requests.get(base_url, params=params)
+        
+        response = self.make_request(params)
+
+        return response
+    
+    def make_request(self, params):
+        response = requests.get(self.base_url, params=params)
         if response.status_code == 200:
-            if output == 'xml':
+            if params['output'] == 'xml':
                 return response.text
-            elif output == 'json':
+            elif params['output'] == 'json':
                 # Remove unnecessary empty spaces
                 return str(response.json())
         else: