Update app.py

app.py

@@ -489,6 +489,7 @@ def get_time_tool2(query: str) -> datetime:
         return now  # Default return the current datetime object
 
 def weather_agent_tool(query: str) -> str:
+    """Weather Agent: Return current, hourly, or historical weather info using WeatherAPI."""
     try:
         weather_api_key = os.environ.get("WEATHER_API_KEY")
         if not weather_api_key:
@@ -513,9 +514,13 @@ def weather_agent_tool(query: str) -> str:
         # Step 2: Get timezone and time
         target_dt = get_time_tool(query)  # Use your get_time_tool here to get the target time
 
+        # 确保 target_dt 是 datetime 对象，如果是字符串，进行转换
+        if isinstance(target_dt, str):
+            target_dt = datetime.strptime(target_dt, "%Y-%m-%d %H:%M:%S")  # 转换字符串为 datetime 对象
+
         # Ensure we calculate target time correctly
         tz_str = location_to_timezone(location)  # We can also pass the correct location here for timezone
-        target_dt = target_dt.replace(tzinfo=ZoneInfo(tz_str))
+        target_dt = target_dt.replace(tzinfo=ZoneInfo(tz_str))  # 添加时区信息
 
         # Step 3: Call WeatherAPI
         if target_dt.date() < datetime.now().date() - timedelta(days=7):
@@ -607,9 +612,13 @@ def weather_tool(query: str) -> str:
         # Step 2: Get timezone and time
         target_dt = get_time_tool(query)  # Use your get_time_tool here to get the target time
 
+        # 确保 target_dt 是 datetime 对象，如果是字符串，进行转换
+        if isinstance(target_dt, str):
+            target_dt = datetime.strptime(target_dt, "%Y-%m-%d %H:%M:%S")  # 转换字符串为 datetime 对象
+
         # Ensure we calculate target time correctly
         tz_str = location_to_timezone(location)  # We can also pass the correct location here for timezone
-        target_dt = target_dt.replace(tzinfo=ZoneInfo(tz_str))
+        target_dt = target_dt.replace(tzinfo=ZoneInfo(tz_str))  # 添加时区信息
 
         # Step 3: Call WeatherAPI
         if target_dt.date() < datetime.now().date() - timedelta(days=7):