updated tool arsenal

app.py

@@ -18,6 +18,63 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     """
     return "What magic will you build ?"
 
+# code from: https://github.com/huggingface/smolagents/blob/main/examples/multiple_tools.py
+@tool
+def search_wikipedia(query: str) -> str:
+    """
+    Fetches a summary of a Wikipedia page for a given query.
+    Args:
+        query: The search term to look up on Wikipedia.
+    Returns:
+        str: A summary of the Wikipedia page if successful, or an error message if the request fails.
+    Raises:
+        requests.exceptions.RequestException: If there is an issue with the HTTP request.
+    """
+    url = f"https://en.wikipedia.org/api/rest_v1/page/summary/{query}"
+
+    try:
+        response = requests.get(url)
+        response.raise_for_status()
+
+        data = response.json()
+        title = data["title"]
+        extract = data["extract"]
+
+        return f"Summary for {title}: {extract}"
+
+    except requests.exceptions.RequestException as e:
+        return f"Error fetching Wikipedia data: {str(e)}"
+
+@tool
+def get_news_headlines() -> str:
+    """
+    Fetches the top news headlines from the News API for the United States.
+    This function makes a GET request to the News API to retrieve the top news headlines
+    for the United States. It returns the titles and sources of the top 5 articles as a
+    formatted string. If no articles are available, it returns a message indicating that
+    no news is available. In case of a request error, it returns an error message.
+    Returns:
+        str: A string containing the top 5 news headlines and their sources, or an error message.
+    """
+    api_key = "995553d64cc54505859b26e81d7065a2"  # Replace with your actual API key from https://newsapi.org/
+    url = f"https://newsapi.org/v2/top-headlines?country=us&apiKey={api_key}"
+
+    try:
+        response = requests.get(url)
+        response.raise_for_status()
+
+        data = response.json()
+        articles = data["articles"]
+
+        if not articles:
+            return "No news available at the moment."
+
+        headlines = [f"{article['title']} - {article['source']['name']}" for article in articles[:5]]
+        return "\n".join(headlines)
+
+    except requests.exceptions.RequestException as e:
+        return f"Error fetching news data: {str(e)}"
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -55,8 +112,14 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
-    max_steps=6,
+    tools=[
+        final_answer,
+        get_current_time_in_timezone,
+        get_news_headlines,
+        image_generation_tool,
+        search_wikipedia,
+    ], ## add your tools here (don't remove final answer)
+    max_steps=10,
     verbosity_level=1,
     grammar=None,
     planning_interval=None,