add real tool get_current_weather_by_city

app.py

@@ -1,6 +1,7 @@
 from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel, load_tool, tool, TransformersModel
 import datetime
 import requests
+import json
 import pytz
 import yaml
 import os
@@ -37,6 +38,58 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+@tool
+def get_current_weather_by_city(city: str) -> str:
+    '''Retrieves the current and forecasted weather for a specified city.
+    Args:
+        city: The name of the city (e.g., "New York")
+    Return:
+        A string with weather information
+    '''
+    # API Geocoding от Open-Meteo
+    geocoding_url = f"https://geocoding-api.open-meteo.com/v1/search?name={city}&count=1&language=en&format=json"
+    response_geocoding = requests.get(geocoding_url)
+    
+    if response_geocoding.status_code == 200:
+        data_geocoding = response_geocoding.json()
+        
+        if 'results' in data_geocoding and len(data_geocoding['results']) > 0:
+            result = data_geocoding['results'][0]
+            lat = result.get('latitude', None)
+            lon = result.get('longitude', None)
+            
+            if lat is not None and lon is not None:
+                # Получаем погоду по координатам
+                weather_url = f"https://api.open-meteo.com/v1/forecast?latitude={lat}&longitude={lon}&current_weather=true&hourly=temperature_2m"
+                response_weather = requests.get(weather_url)
+                
+                if response_weather.status_code == 200:
+                    data_weather = response_weather.json()
+                    
+                    current_temp = data_weather['current_weather']['temperature'] if 'current_weather' in data_weather else None
+                    hourly_temps = data_weather['hourly']['temperature_2m'] if 'hourly' in data_weather and 'temperature_2m' in data_weather['hourly'] else []
+                    
+                    weather_info = f"Current temperature: {current_temp}°C\n"
+                    weather_info += "Hourly temperatures:\n"
+                    
+                    for temp in hourly_temps:
+                        weather_info += f"{temp}°C\n"
+                    
+                    return weather_info
+                else:
+                    return f"Failed to retrieve weather data. Status code: {response_weather.status_code}"
+            else:
+                return "Latitude and Longitude not found."
+        else:
+            return "No results found for the specified city."
+    else:
+        return f"Geocoding failed. Status code: {response_geocoding.status_code}"
+
+# Execute the function and prepare the final answer
+#result = get_current_weather_by_city("New York")
+#final_answer = f"The current weather in New York is:\n{result}"
+#print(final_answer)
+
 final_answer = FinalAnswerTool()
 
 # Local: mcedit  ~/.bashrc
@@ -59,7 +112,11 @@ with open("prompts.yaml", 'r') as stream:
 
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_current_time_in_timezone], ## add your tools here (don't remove final answer)
+    tools=[
+        final_answer,
+        get_current_time_in_timezone,
+        get_current_weather_by_city,
+    ], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,