added news tool

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import random
-from smolagents import GradioUI, CodeAgent, HfApiModel, DuckDuckGoSearchTool
+from smolagents import GradioUI, CodeAgent, HfApiModel, DuckDuckGoSearchTool, LatestNewsTool
 # Import our custom tools from their modules
 from tools import WeatherInfoTool, HubStatsTool
 from retriever import load_guest_dataset
@@ -10,7 +10,8 @@ model = HfApiModel()
 
 # Initialize the DuckDuckGo search tool
 search_tool = DuckDuckGoSearchTool()
-
+# Initialize the News tool
+latest_news_tool = LatestNewsTool()
 # Initialize the weather tool
 weather_info_tool = WeatherInfoTool()
 
@@ -23,12 +24,11 @@ hub_stats_tool = HubStatsTool()
 
 # Create Alfred with all the tools
 agent = CodeAgent(
-    tools=[guest_info_tool, search_tool, weather_info_tool, hub_stats_tool], 
+    tools=[guest_info_tool, latest_news_tool, weather_info_tool, hub_stats_tool, search_tool], 
     model=model,
     add_base_tools=True,  # Add any additional base tools
-    planning_interval=3,   # Enable planning every 3 steps
+    # planning_interval=3,   # Enable planning every 3 steps
 )
 #
 if __name__ == "__main__":
-   if __name__ == "__main__":
     GradioUI(agent).launch()

tools.py

@@ -2,7 +2,9 @@ from smolagents import DuckDuckGoSearchTool
 from smolagents import Tool
 import requests
 from huggingface_hub import list_models
+from dotenv import find_dotenv, load_dotenv
 import os 
+from datetime import datetime, timedelta
 
 class WeatherInfoTool(Tool):
     name = "weather_info"
@@ -23,6 +25,7 @@ class WeatherInfoTool(Tool):
 
     def forward(self, location:str, units:str="imperial")->str:
         # Weather from openweather api
+        #load_dotenv(find_dotenv())
         weather_api_key = os.getenv("WEATHER_API")
         url = f"http://api.openweathermap.org/geo/1.0/direct?limit=5"
         params = {
@@ -49,7 +52,36 @@ class WeatherInfoTool(Tool):
         result = f"""The weather in {location} is {combined_dictionary['description']}. The temperature is {combined_dictionary['temp']}, feels like {combined_dictionary['feels_like']}"""
 
         return result
-    
+
+class LatestNewsTool(Tool):
+    name = "latest_news"
+    description = "Fetches latest news on a specified topic."
+    inputs = {
+        "topic": {
+            "type": "string",
+            "description": "Topic for which to get the latest news."
+        }
+        }
+    output_type = "string"
+
+    def forward(self, topic:str)->str:
+        # news from newsapi
+        load_dotenv(find_dotenv())
+        news_api_key = os.getenv("NEWS_API")
+        today = datetime.now()
+        yesterday = today - timedelta(days=1)
+        yesterday = str(yesterday.date())     
+        url = ('https://newsapi.org/v2/everything?'
+                f'q={topic}&'
+                f'from={yesterday}&'
+                'sortBy=popularity&'
+                f'apiKey={news_api_key}')
+        response = requests.get(url)
+        resp = response.json()
+        results = [f'Title: {el["title"]}. Description: {el["description"]}' for el in resp["articles"]][:10]
+        results_str = ' '.join(el for el in results)
+        return results_str
+
 class HubStatsTool(Tool):
     name = "hub_stats"
     description = "Fetches the most downloaded model from a specific author on the Hugging Face Hub."