Update app.py

app.py

@@ -4,6 +4,9 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+import requests
+import json
+from datetime import datetime
 
 from Gradio_UI import GradioUI
 
@@ -18,6 +21,73 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     """
     return "What magic will you build ?"
 
+
+
+
+@tool
+def query_weather(location: str, days: int = 1) -> str:
+    """查询指定地点的天气情况
+    
+    参数：
+        location：要查询天气的地点（城市名称）
+        days：要查询的天数（默认为1天，最大3天）
+    """
+    if days < 1 or days > 3:
+        return "查询天数必须在1-3之间"
+    
+    # 使用wttr.in API，完全免费，无需API密钥
+    base_url = "https://wttr.in/"
+    
+    # 构建API请求URL
+    params = {
+        "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}"
+        
+        data = response.json()
+        
+        # 处理天气数据
+        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"
+            
+            result += "\n"
+        
+        return result
+        
+    except Exception as e:
+        return f"查询天气时发生错误: {str(e)}"
+        
+        
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -55,7 +125,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,query_weather], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,