58 lines
1.4 KiB
Python
58 lines
1.4 KiB
Python
"""Weather related tools"""
|
|
from backend.tools.factory import tool
|
|
|
|
|
|
@tool(
|
|
name="get_weather",
|
|
description="Get weather information for a specified city. Use when user asks about weather.",
|
|
parameters={
|
|
"type": "object",
|
|
"properties": {
|
|
"city": {
|
|
"type": "string",
|
|
"description": "City name, e.g.: 北京, 上海, 广州"
|
|
}
|
|
},
|
|
"required": ["city"]
|
|
},
|
|
category="weather"
|
|
)
|
|
def get_weather(arguments: dict) -> dict:
|
|
"""
|
|
Weather query tool (simulated)
|
|
|
|
Args:
|
|
arguments: {
|
|
"city": "北京"
|
|
}
|
|
|
|
Returns:
|
|
{
|
|
"city": "北京",
|
|
"temperature": 25,
|
|
"humidity": 60,
|
|
"description": "晴天"
|
|
}
|
|
"""
|
|
city = arguments["city"]
|
|
|
|
# 模拟天气数据
|
|
weather_data = {
|
|
"北京": {"temperature": 25, "humidity": 60, "description": "晴天"},
|
|
"上海": {"temperature": 28, "humidity": 75, "description": "多云"},
|
|
"广州": {"temperature": 32, "humidity": 85, "description": "雷阵雨"},
|
|
"深圳": {"temperature": 30, "humidity": 80, "description": "阴天"},
|
|
}
|
|
|
|
data = weather_data.get(city, {
|
|
"temperature": 22,
|
|
"humidity": 65,
|
|
"description": "晴天"
|
|
})
|
|
|
|
return {
|
|
"city": city,
|
|
**data,
|
|
"query_time": "2026-03-24 12:00:00"
|
|
}
|