mirror of
https://github.com/RYDE-WORK/Langchain-Chatchat.git
synced 2026-01-25 16:23:22 +08:00
26 lines
890 B
Python
26 lines
890 B
Python
"""
|
|
简单的单参数输入工具实现，用于查询现在天气的情况
|
|
"""
|
|
from pydantic.v1 import BaseModel, Field
|
|
import requests
|
|
|
|
def weather(location: str, api_key: str):
|
|
url = f"https://api.seniverse.com/v3/weather/now.json?key={api_key}&location={location}&language=zh-Hans&unit=c"
|
|
response = requests.get(url)
|
|
if response.status_code == 200:
|
|
data = response.json()
|
|
weather = {
|
|
"temperature": data["results"][0]["now"]["temperature"],
|
|
"description": data["results"][0]["now"]["text"],
|
|
}
|
|
return weather
|
|
else:
|
|
raise Exception(
|
|
f"Failed to retrieve weather: {response.status_code}")
|
|
|
|
|
|
def weather_check(location: str):
|
|
return weather(location, "S8vrB4U_-c5mvAMiK")
|
|
class WeatherInput(BaseModel):
|
|
location: str = Field(description="City name,include city and county,like '厦门'")
|