Update tools.py

tools.py

@@ -2,7 +2,7 @@ from smolagents import DuckDuckGoSearchTool
 from smolagents import Tool
 import random
 from huggingface_hub import list_models
-
+import requests
 
 # Initialize the DuckDuckGo search tool
 #search_tool = DuckDuckGoSearchTool()
@@ -54,3 +54,23 @@ class HubStatsTool(Tool):
         except Exception as e:
             return f"Error fetching models for {author}: {str(e)}"
 
+class NewsTool(Tool):
+    name = "get_latest_news"
+    description = "Fetches the latest news headlines about a given topic."
+
+    inputs = {
+        "topic": {
+            "type": "string",
+            "description": "The topic to search news for."
+        }
+    }
+
+    def forward(self, topic: str):
+        url = f"https://newsapi.org/v2/everything?q={topic}&apiKey=15768396aec7461d927ed37dfffa2f84"
+        response = requests.get(url)
+        data = response.json()
+        if "articles" in data and len(data["articles"]) > 0:
+            headlines = [a["title"] for a in data["articles"][:5]]
+            return "\n".join(headlines)
+        return "No recent news found."
+