update get_weather tool

app.py

@@ -36,10 +36,16 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 @tool
-def get_weather(location: str, api_key: str) -> str:
-     """A tool that fetches the current weather data in a specified location.
+def get_weather(location: str, api_key: str) -> dict:
+    """
+    A tool that fetches the current weather data for a specified location.
+
     Args:
-        location: A string representing the location to get weather data (e.g., 'New_York').
+        location (str): The city name to get weather data (e.g., 'New York').
+        api_key (str): Your OpenWeatherMap API key.
+
+    Returns:
+        dict: A dictionary containing weather details or an error message.
     """
     base_url = "https://api.openweathermap.org/data/2.5/weather"
     params = {
@@ -50,15 +56,15 @@ def get_weather(location: str, api_key: str) -> str:
     
     try:
         response = requests.get(base_url, params=params)
-        response.raise_for_status()
+        response.raise_for_status()  # Raise an exception for HTTP errors
         data = response.json()
         
         weather_info = {
-            "city": data["name"],
-            "temperature": data["main"]["temp"],
-            "description": data["weather"][0]["description"],
-            "humidity": data["main"]["humidity"],
-            "wind_speed": data["wind"]["speed"]
+            "city": data.get("name"),
+            "temperature": data["main"].get("temp"),
+            "description": data["weather"][0].get("description"),
+            "humidity": data["main"].get("humidity"),
+            "wind_speed": data["wind"].get("speed")
         }
         return weather_info
     except requests.exceptions.RequestException as e: