Update my_tools.py

my_tools.py

@@ -5,6 +5,46 @@ import pytz
 import os
 import datetime
 
+WOLFRAM_RESPONSE_KEYS = [
+    "Result",
+    "Solution",
+    "RealSolution",
+]
+
+
+@tool
+def wolfram(query: str)-> str:
+    """
+        A wrapper around Wolfram Alpha. Useful for when you need to answer questions about Math,
+        Science, Technology, Culture, Society and Everyday Life. Input should be a search query."    
+    Args:
+        query: search query.
+    Returns:
+        A string containing the answer for the query.
+    """
+    api_key = os.environ["WOLFRAM_API_KEY"]
+    formatted_query = query
+    url = f"http://api.wolframalpha.com/v2/query?appid={api_key}&input={formatted_query}&output=json&format=plaintext"
+    for key in WOLFRAM_RESPONSE_KEYS:
+        url += f"&includepodid={key}"
+    print(f"\nURL:\n{url}")
+
+    try:
+        response = requests.get(url)
+        response.raise_for_status()  # Raise an exception for HTTP errors
+
+        data = response.json()
+
+        if data.get("error"):  # Check if there's an error in the response
+            return f"Error: {data['error'].get('info', 'Unable to fetch Wolfram response.')}"
+
+        response = data.get("queryresult").get("pods")[0].get("subpods")[0].get("plaintext")
+        return response
+        
+    except requests.exceptions.RequestException as e:
+        return f"Error fetching Wolfram response: {str(e)}"
+
+
 @tool
 def get_weather(location: str, celsius: Optional[bool] = False) -> str:
     """
@@ -211,18 +251,6 @@ def search_wikipedia(query: str) -> str:
         return f"Error fetching Wikipedia data: {str(e)}"
 
 
-
-# Below is an example of a tool that does nothing. Amaze us with your creativity !
-@tool
-def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
-    #Keep this format for the description / args / args description but feel free to modify the tool
-    """A tool that does nothing yet 
-    Args:
-        arg1: the first argument
-        arg2: the second argument
-    """
-    return "What magic will you build ?"
-
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.