weather_yool

app.py

@@ -4,6 +4,7 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+import os
 
 from Gradio_UI import GradioUI
 
@@ -17,6 +18,48 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
         arg2: the second argument
     """
     return "What magic will you build ?"
+    
+@tool
+def check_weather(city: str, country: str) -> str:
+    """A tool that checks the weather at a specified city.
+    Args:
+        city: The city for which to check the weather (e.g., 'Brezno').
+        country: The country code (e.g., 'SK' for Slovakia).
+    """
+    # Replace with your actual API key
+    api_key = os.getenv("weather")
+    
+    # Geocoding API URL
+    geo_url = f"http://api.openweathermap.org/geo/1.0/direct?q={city},{country}&limit=1&appid={api_key}"
+    
+    try:
+        # Fetch the geocoding data
+        geo_response = requests.get(geo_url)
+        geo_data = geo_response.json()
+        
+        if not geo_data:
+            return f"Error: Could not find location '{city}, {country}'."
+        
+        # Extract latitude and longitude
+        lat = geo_data[0]['lat']
+        lon = geo_data[0]['lon']
+        
+        # Weather API URL
+        weather_url = f"https://api.openweathermap.org/data/3.0/onecall?lat={lat}&lon={lon}&exclude=hourly,daily&appid={api_key}"
+        
+        # Fetch the weather data
+        weather_response = requests.get(weather_url)
+        weather_data = weather_response.json()
+        
+        # Extract relevant weather information
+        current_weather = weather_data['current']
+        temperature = current_weather['temp']
+        weather_description = current_weather['weather'][0]['description']
+        
+        return f"The current weather in {city}, {country} is {weather_description} with a temperature of {temperature}°C."
+    
+    except Exception as e:
+        return f"Error fetching weather data: {str(e)}"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -55,7 +98,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=[final_answer,check_weather,get_current_time_in_timezone], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,