Update app.py

app.py

@@ -37,49 +37,52 @@ def get_current_time_in_timezone(timezone: str) -> str:
 @tool
 def get_weather_by_city(city_name: str) -> str:
     """
-    A tool that returns the current weather for the specified city.
-    Args:
-        city_name: city name 
+    Возвращает текущую погоду для указанного города в формате: "Температура в [город]: X°C".
+
+    Аргументы:
+        city_name (str): Название города на английском или русском языке. Например: "Paris" или "Париж".
+
+    Возвращает:
+        str: Строка с температурой или сообщение об ошибке.
     """
-    # Получаем координаты города
-    # Формируем URL для запроса к API
-    city_name = city_name
-    url = f"https://geocoding-api.open-meteo.com/v1/search?name={city_name}&count=1&language=en&format=json"
-    try:
-        # Отправляем GET-запрос к API
-        response = requests.get(url)
-        response.raise_for_status()  # Проверяем на ошибки HTTP
-        
-        # Парсим JSON-ответ
-        data = response.json()
-        
-        # Проверяем, есть ли результаты
-        if "results" in data and len(data["results"]) > 0:
-            city_data = data["results"][0]
-        else:
-            return {"error": "Город не найден"}
+    # Шаг 1: Получаем координаты города
+    geocoding_url = f"https://geocoding-api.open-meteo.com/v1/search?name={city_name}&count=1&language=en&format=json"
     
+    try:
+        # Запрос координат
+        geo_response = requests.get(geocoding_url)
+        geo_response.raise_for_status()
+        geo_data = geo_response.json()
+
+        # Проверка наличия результатов
+        if not geo_data.get("results"):
+            return f"Город '{city_name}' не найден"
+            
+        city_data = geo_data["results"][0]
+        latitude = city_data["latitude"]
+        longitude = city_data["longitude"]
+
     except requests.RequestException as e:
-        return {"error": f"Ошибка при запросе к API: {str(e)}"}
-    except Exception as e:
-        return {"error": f"Неожиданная ошибка: {str(e)}"}    
-    
-    coordinates = {
-                "latitude": city_data["latitude"],
-                "longitude": city_data["longitude"]
-            }
-    if "error" in coordinates:
-        return coordinates["error"]
+        return f"Ошибка геокодинга: {str(e)}"
+    except KeyError:
+        return "Некорректный ответ от API геокодинга"
+
+    # Шаг 2: Получаем погоду по координатам
+    weather_url = f"https://api.open-meteo.com/v1/forecast?latitude={latitude}&longitude={longitude}&current_weather=true"
     
-    # Используем координаты для запроса погоды
-    weather_url = f"https://api.open-meteo.com/v1/forecast?latitude={coordinates['latitude']}&longitude={coordinates['longitude']}&current_weather=true"
     try:
-        response = requests.get(weather_url)
-        response.raise_for_status()
-        weather_data = response.json()
-        return f"Текущая температура в {city_name}: {weather_data['current_weather']['temperature']}°C"
+        weather_response = requests.get(weather_url)
+        weather_response.raise_for_status()
+        weather_data = weather_response.json()
+        
+        temperature = weather_data["current_weather"]["temperature"]
+        return f"Текущая температура в {city_name}: {temperature}°C"
+
     except requests.RequestException as e:
-        return f"Ошибка при запросе погоды: {str(e)}"
+        return f"Ошибка запроса погоды: {str(e)}"
+    except KeyError:
+        return "Некорректный ответ от API погоды"
+        
 
 final_answer = FinalAnswerTool()
 
@@ -102,7 +105,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_weather_by_city()], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_weather_by_city], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,