Langchain-Chatchat

1
"""
2
更简单的单参数输入工具实现，用于查询现在天气的情况
3
"""
4
from pydantic import BaseModel, Field
5
import requests
6
from configs.kb_config import SENIVERSE_API_KEY
7

8

9
def weather(location: str, api_key: str):
10
    url = f"https://api.seniverse.com/v3/weather/now.json?key={api_key}&location={location}&language=zh-Hans&unit=c"
11
    response = requests.get(url)
12
    if response.status_code == 200:
13
        data = response.json()
14
        weather = {
15
            "temperature": data["results"][0]["now"]["temperature"],
16
            "description": data["results"][0]["now"]["text"],
17
        }
18
        return weather
19
    else:
20
        raise Exception(
21
            f"Failed to retrieve weather: {response.status_code}")
22

23

24
def weathercheck(location: str):
25
    return weather(location, SENIVERSE_API_KEY)
26

27

28
class WeatherInput(BaseModel):
29
    location: str = Field(description="City name,include city and county")
30