Update app.py

app.py

@@ -7,6 +7,8 @@ from tools.final_answer import FinalAnswerTool
 import requests
 import json
 from datetime import datetime
+from typing import Optional
+
 
 from Gradio_UI import GradioUI
 import os
@@ -29,69 +31,62 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
 
 @tool
 def query_weather(location: str, days: int = 1) -> str:
-    """查询指定地点的天气预报。
+    """
+    查询指定地点的天气预报。
     
     Args:
-        location: 要查询天气的地点名称，如"北京"、"上海"等。
-        days: 要查询的天数，默认为1天。
+        location: 要查询天气的地点名称，如 "北京"、"上海"。
+        days: 要查询的天数，默认为 1 天（最多 5 天）。
     
     Returns:
         包含天气预报信息的字符串。
     """
-    
-    if days < 1 or days > 3:
-        return "查询天数必须在1-3之间"
-    
-    # 使用wttr
+    if days < 1 or days > 5:
+        return "查询天数必须在 1-5 之间。"
+
     base_url = "https://wttr.in/"
-    
-    # 构建API请求URL
     params = {
-        "format": "j1",  
-        "lang": "zh-cn" 
+        "format": "j1",  # 返回 JSON 格式
+        "lang": "zh-cn"  # 使用中文
     }
-    
+
     try:
         # 发送请求
         response = requests.get(f"{base_url}{location}", params=params)
-        
-        # 检查响应状态
-        if response.status_code != 200:
-            return f"查询失败: HTTP状态码 {response.status_code}"
-        
+        response.raise_for_status()  # 检查 HTTP 状态码
+
         data = response.json()
-        
+
         # 处理天气数据
-        result = f"{location}天气预报:\n\n"
-        
-        # 处理预报数据
+        result = f"{location} 天气预报:\n\n"
         for i, day in enumerate(data["weather"]):
             if i >= days:
                 break
-                
+
             date = day["date"]
-            
             result += f"【{date}】\n"
-            
+
             # 添加当天的天气信息
             for period in day["hourly"]:
-                # 每隔6小时显示一次预报
-                if int(period["time"]) % 600 == 0:
-                    time_str = f"{int(period['time'])//100:02d}:00"
-                    temp = period["tempC"]
-                    weather_desc = period["lang_zh"][0]["value"]
-                    humidity = period["humidity"]
-                    wind_speed = period["windspeedKmph"]
-                    wind_dir = period["winddir16Point"]
-                    
-                    result += f"- {time_str}: {weather_desc}, 温度{temp}°C, 湿度{humidity}%, 风速{wind_speed}km/h, 风向{wind_dir}\n"
-            
+                time_str = f"{int(period['time'])//100:02d}:00"
+                temp = period["tempC"]
+                weather_desc = period["weatherDesc"][0]["value"]
+                humidity = period["humidity"]
+                wind_speed = period["windspeedKmph"]
+                wind_dir = period["winddir16Point"]
+
+                result += (
+                    f"- {time_str}: {weather_desc}, "
+                    f"温度 {temp}°C, 湿度 {humidity}%, "
+                    f"风速 {wind_speed}km/h, 风向 {wind_dir}\n"
+                )
+
             result += "\n"
-        
+
         return result
-        
-    except Exception as e:
-        return f"查询天气时发生错误: {str(e)}"
+
+    except requests.exceptions.RequestException as e:
+        return f"查询天气失败: {str(e)}"