Fix tool compatibility with smolagents 1.9.2 by creating proper Tool classes

app.py

@@ -1,38 +1,13 @@
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
-import datetime
-import requests
-import pytz
+from smolagents import CodeAgent, HfApiModel
 import yaml
-from tools.final_answer import FinalAnswerTool
-
+from tools import FinalAnswerTool, WeatherTool, DuckDuckGoSearchTool
 from Gradio_UI import GradioUI
 
-# Weather tool to get current weather information
-def get_weather(location: str) -> str:
-    """A tool that fetches the current weather for a specified location.
-    Args:
-        location: A string representing a city or location (e.g., 'New York', 'Paris, France').
-    """
-    try:
-        # Using OpenWeatherMap API with a free tier (limited requests)
-        api_key = "1ae035abc608d0e1095a5472dc989299"  # OpenWeatherMap API key
-        url = f"http://api.openweathermap.org/data/2.5/weather?q={location}&appid={api_key}&units=metric"
-        
-        response = requests.get(url)
-        data = response.json()
-        
-        if response.status_code == 200:
-            temp = data["main"]["temp"]
-            weather = data["weather"][0]["description"]
-            humidity = data["main"]["humidity"]
-            wind_speed = data["wind"]["speed"]
-            return f"The current weather in {location} is {temp}°C with {weather}. Humidity: {humidity}%, Wind speed: {wind_speed} m/s."
-        else:
-            return f"Error fetching weather for {location}: {data.get('message', 'Unknown error')}"
-    except Exception as e:
-        return f"Error fetching weather for {location}: {str(e)}"
-
+# Initialize our tools
 final_answer = FinalAnswerTool()
+weather_tool = WeatherTool()
+search_tool = DuckDuckGoSearchTool()
+
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
@@ -48,8 +23,8 @@ agent = CodeAgent(
     model=model,
     tools=[
         final_answer,
-        DuckDuckGoSearchTool(),
-        get_weather
+        search_tool,
+        weather_tool
     ],
     max_steps=6,
     verbosity_level=1,

tools/__init__.py

@@ -1,5 +1,7 @@
 """Tools for the agent."""
 
 from .final_answer import FinalAnswerTool
+from .weather import WeatherTool
+from .search import DuckDuckGoSearchTool
 
-__all__ = ["FinalAnswerTool"] 
+__all__ = ["FinalAnswerTool", "WeatherTool", "DuckDuckGoSearchTool"] 

tools/search.py

@@ -0,0 +1,52 @@
+from smolagents.tools import Tool
+import requests
+from typing import Any, List, Dict, Optional
+import json
+
+class DuckDuckGoSearchTool(Tool):
+    """Tool for searching the web using DuckDuckGo."""
+    
+    name = "duck_duck_go_search"
+    description = "Search the web for information using DuckDuckGo."
+    
+    def __call__(self, query: str) -> str:
+        """
+        Search the web for information using DuckDuckGo.
+        
+        Args:
+            query: A string representing the search query.
+            
+        Returns:
+            A string with the search results.
+        """
+        try:
+            # Using the DuckDuckGo API
+            url = "https://api.duckduckgo.com/"
+            params = {
+                'q': query,
+                'format': 'json',
+                'no_html': 1,
+                'skip_disambig': 1
+            }
+            
+            response = requests.get(url, params=params)
+            data = response.json()
+            
+            results = []
+            
+            # Add the abstract if available
+            if data.get('Abstract'):
+                results.append(f"Abstract: {data['Abstract']}")
+            
+            # Add related topics
+            if data.get('RelatedTopics'):
+                for i, topic in enumerate(data['RelatedTopics'][:5]):  # Limit to 5 topics
+                    if 'Text' in topic:
+                        results.append(f"Result {i+1}: {topic['Text']}")
+            
+            if not results:
+                return f"No results found for query: {query}"
+            
+            return "\n\n".join(results)
+        except Exception as e:
+            return f"Error searching for {query}: {str(e)}" 

tools/weather.py

@@ -0,0 +1,38 @@
+from smolagents.tools import Tool
+import requests
+from typing import Any, Optional
+
+class WeatherTool(Tool):
+    """Tool for getting weather information for a location."""
+    
+    name = "get_weather"
+    description = "Get the current weather for a specified location."
+    
+    def __call__(self, location: str) -> str:
+        """
+        Get the current weather for a specified location.
+        
+        Args:
+            location: A string representing a city or location (e.g., 'New York', 'Paris, France').
+            
+        Returns:
+            A string with the current weather information.
+        """
+        try:
+            # Using OpenWeatherMap API with a free tier (limited requests)
+            api_key = "1ae035abc608d0e1095a5472dc989299"  # OpenWeatherMap API key
+            url = f"http://api.openweathermap.org/data/2.5/weather?q={location}&appid={api_key}&units=metric"
+            
+            response = requests.get(url)
+            data = response.json()
+            
+            if response.status_code == 200:
+                temp = data["main"]["temp"]
+                weather = data["weather"][0]["description"]
+                humidity = data["main"]["humidity"]
+                wind_speed = data["wind"]["speed"]
+                return f"The current weather in {location} is {temp}°C with {weather}. Humidity: {humidity}%, Wind speed: {wind_speed} m/s."
+            else:
+                return f"Error fetching weather for {location}: {data.get('message', 'Unknown error')}"
+        except Exception as e:
+            return f"Error fetching weather for {location}: {str(e)}"