54 lines
2.0 KiB
Python
54 lines
2.0 KiB
Python
import os
|
||
from typing import Any
|
||
|
||
import requests
|
||
from langchain.tools import BaseTool
|
||
from urllib.parse import quote
|
||
|
||
class Weather(BaseTool):
|
||
name = "weather"
|
||
description = "此工具用于获取天气预报信息，需传入英文的城市名，参数格式：Guangzhou"
|
||
|
||
def __init__(self):
|
||
super().__init__()
|
||
|
||
async def _arun(self, *args: Any, **kwargs: Any) -> Any:
|
||
# 用例中没有用到 arun 不予具体实现
|
||
pass
|
||
|
||
|
||
def _run(self, para: str) -> str:
|
||
try:
|
||
if not para:
|
||
return "参数不能为空"
|
||
encoded_city = quote(para)
|
||
|
||
api_url = f"http://api.openweathermap.org/data/2.5/weather?q={encoded_city}&appid=272fcb70d2c4e6f5134c2dce7d091df6"
|
||
response = requests.get(api_url)
|
||
if response.status_code == 200:
|
||
weather_data = response.json()
|
||
# 提取天气信息
|
||
temperature_kelvin = weather_data['main']['temp']
|
||
temperature_celsius = temperature_kelvin - 273.15
|
||
min_temperature_kelvin = weather_data['main']['temp_min']
|
||
max_temperature_kelvin = weather_data['main']['temp_max']
|
||
min_temperature_celsius = min_temperature_kelvin - 273.15
|
||
max_temperature_celsius = max_temperature_kelvin - 273.15
|
||
description = weather_data['weather'][0]['description']
|
||
wind_speed = weather_data['wind']['speed']
|
||
|
||
# 构建天气描述
|
||
weather_description = f"今天天气：{description}，气温：{temperature_celsius:.2f}摄氏度，风速：{wind_speed} m/s。"
|
||
|
||
return f"天气预报信息：{weather_description}"
|
||
else:
|
||
return f"无法获取天气预报信息，状态码：{response.status_code}"
|
||
except Exception as e:
|
||
return f"发生错误：{str(e)}"
|
||
|
||
|
||
if __name__ == "__main__":
|
||
weather_tool = Weather()
|
||
weather_info = weather_tool.run("Guangzhou")
|
||
print(weather_info)
|