Added Weather Tool

agent.json

@@ -2,7 +2,8 @@
     "tools": [
         "web_search",
         "visit_webpage",
-        "final_answer"
+        "final_answer",
+        "get_weather"
     ],
     "model": {
         "class": "HfApiModel",
@@ -48,6 +49,7 @@
         "queue",
         "time",
         "collections",
-        "re"
+        "re",
+        "requests"
     ]
 }

app.py

@@ -3,15 +3,15 @@ import datetime
 import requests
 import pytz
 import yaml
-from geopy.geocoders import Nominatim
-from timezonefinder import TimezoneFinder
+
 from tools.final_answer import FinalAnswerTool
+from tools.get_weather import GetWeather
 
 from Gradio_UI import GradioUI
 
 # 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
+def my_custom_tool(arg1:str, arg2:int)-> str: #it's important 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:
@@ -34,69 +34,10 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"The current local time in {timezone} is: {local_time}"
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
-
-def get_lat_long(address: str) -> tuple | str:
-    """
-    Get the latitude and longitude of a given address
-    Args:
-        address: A string representing a valid address (e.g., '1600 Amphitheatre Parkway, Mountain View, CA').
-    """
-    if not address:
-        return "No address provided", "No address provided"
-    
-    geolocator = Nominatim(user_agent="myGeocoder", timeout=10)
-    try:
-        location = geolocator.geocode(address)
-        return location.latitude, location.longitude
-    except:
-        return "Location not found", "Location not found"
-    
-
-def get_timezone(lat: float, long: float) -> str:
-    """
-    Get the timezone of a given latitude and longitude
-    Args:
-        lat: A float representing the latitude of a location.
-        long: A float representing the longitude of a location.
-    """
-    if not lat or not long:
-        return "No latitude or longitude provided"
-    
-    tf = TimezoneFinder()
-    try:
-        timezone_str = tf.timezone_at(lng=long, lat=lat)
-        return timezone_str
-    except:
-        return "Timezone not found"
-
-@tool 
-def get_weather(location: str) -> str:
-    """
-    Get the weather of a given latitude and longitude
-    Args:
-        location: A string representing a valid location (e.g., '1600 Amphitheatre Parkway, Mountain View, CA').
-    """
-    if not location:
-        return "No location provided"
-    
-    lat, long = get_lat_long(location)
-    if lat == "Location not found" or long == "Location not found":
-        return "Location not found"
-    
-    timezone = get_timezone(lat, long)
-    if timezone == "Timezone not found":
-        return "Timezone not found"
-    
-    api_call = f"https://api.open-meteo.com/v1/forecast?latitude={lat}&longitude={long}&current=temperature_2m,apparent_temperature,precipitation,wind_speed_10m&timezone={timezone}&past_days=1&forecast_days=1"
-    try:
-        response = requests.get(api_call)
-        data = response.json()
-        return f"Temperature: {data['current']['temperature_2m']}{data['current_units']['temperature_2m']}\n" + f"Feels like: {data['current']['apparent_temperature']}{data['current_units']['apparent_temperature']}\n" + f"Precipitation: {data['current']['precipitation']}{data['current_units']['precipitation']}\n" + f"Wind speed: {data['current']['wind_speed_10m']}{data['current_units']['wind_speed_10m']}"
-    except Exception as e:
-        return "Error fetching weather data"
     
 
 final_answer = FinalAnswerTool()
+get_weather = GetWeather()
 
 # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
 # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 
@@ -117,7 +58,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[get_weather, final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,

tools/get_weather.py

@@ -0,0 +1,82 @@
+from smolagents import Tool
+
+class GetWeather(Tool):
+    name: str = "get_weather"
+    description: str = "Get the weather of a given location."
+    inputs: dict = {'location': {'type': 'string', 'description': 'The location to get the weather of.'}}
+    output_type: str = "string"
+
+    def forward(self, location: str) -> str:
+        import requests
+        if not location:
+            return "No location provided"
+        
+        lat, long = self.get_lat_long(location)
+        if lat == "Location not found" or long == "Location not found":
+            return "Location not found"
+        
+        timezone = self.get_timezone(lat, long)
+        if timezone == "Timezone not found":
+            return "Timezone not found"
+        
+        api_call = f"https://api.open-meteo.com/v1/forecast?latitude={lat}&longitude={long}&current=temperature_2m,apparent_temperature,precipitation,wind_speed_10m&timezone={timezone}&past_days=1&forecast_days=1"
+        try:
+            response = requests.get(api_call)
+            data = response.json()
+            return f"Temperature: {data['current']['temperature_2m']}{data['current_units']['temperature_2m']}\n" + f"Feels like: {data['current']['apparent_temperature']}{data['current_units']['apparent_temperature']}\n" + f"Precipitation: {data['current']['precipitation']}{data['current_units']['precipitation']}\n" + f"Wind speed: {data['current']['wind_speed_10m']}{data['current_units']['wind_speed_10m']}"
+        except Exception as e:
+            return "Error fetching weather data"
+
+
+    def get_lat_long(self, address: str) -> tuple | str:
+        """
+        Get the latitude and longitude of a given address
+        Args:
+            address: A string representing a valid address (e.g., '1600 Amphitheatre Parkway, Mountain View, CA').
+        """
+        try:
+            from geopy.geocoders import Nominatim
+        except ImportError as e:
+            raise ImportError(
+                "You must install the `geopy` package to run this tool: for instance run `pip install geopy`."
+            )
+        
+        if not address:
+            return "No address provided", "No address provided"
+        
+        geolocator = Nominatim(user_agent="myGeocoder", timeout=10)
+        try:
+            location = geolocator.geocode(address)
+            return location.latitude, location.longitude
+        except:
+            return "Location not found", "Location not found"
+        
+
+    def get_timezone(self, lat: float, long: float) -> str:
+        """
+        Get the timezone of a given latitude and longitude
+        Args:
+            lat: A float representing the latitude of a location.
+            long: A float representing the longitude of a location.
+        """
+        try:
+            from timezonefinder import TimezoneFinder
+        except ImportError as e:
+            raise ImportError(
+                "You must install the `timezonefinder` package to run this tool: for instance run `pip install timezonefinder`."
+            )
+        
+
+        if not lat or not long:
+            return "No latitude or longitude provided"
+        
+        tf = TimezoneFinder()
+        try:
+            timezone_str = tf.timezone_at(lng=long, lat=lat)
+            return timezone_str
+        except:
+            return "Timezone not found"
+    
+    def __init__(self, *args, **kwargs):
+        self.is_initialized = False
+