32 lines
1.3 KiB
Python
32 lines
1.3 KiB
Python
import datetime
|
|
|
|
from langchain_core.tools import tool
|
|
|
|
|
|
@tool(name_or_callable="tool_get_datetime", description="一个获取当前时间信息的一个工具")
|
|
def tool_get_datetime():
|
|
now = datetime.datetime.now()
|
|
current_time = now.strftime("%Y-%m-%d %H:%M:%S")
|
|
|
|
yesterday = (now - datetime.timedelta(days=1)).strftime("%Y-%m-%d %H:%M:%S")
|
|
tomorrow = (now + datetime.timedelta(days=1)).strftime("%Y-%m-%d %H:%M:%S")
|
|
day_after_tomorrow = (now + datetime.timedelta(days=2)).strftime("%Y-%m-%d %H:%M:%S")
|
|
two_days_after_tomorrow = (now + datetime.timedelta(days=3)).strftime("%Y-%m-%d %H:%M:%S")
|
|
|
|
weekdays = ['星期一', '星期二', '星期三', '星期四', '星期五', '星期六', '星期日']
|
|
current_weekday = weekdays[now.weekday()]
|
|
yesterday_weekday = weekdays[(now.weekday() - 1) % 7]
|
|
tomorrow_weekday = weekdays[(now.weekday() + 1) % 7]
|
|
day_after_tomorrow_weekday = weekdays[(now.weekday() + 2) % 7]
|
|
two_days_after_tomorrow_weekday = weekdays[(now.weekday() + 3) % 7]
|
|
|
|
result = {
|
|
"当前时间": f"{current_time} {current_weekday}",
|
|
"昨天": f"{yesterday} {yesterday_weekday}",
|
|
"明天": f"{tomorrow} {tomorrow_weekday}",
|
|
"后天": f"{day_after_tomorrow} {day_after_tomorrow_weekday}",
|
|
"大后天": f"{two_days_after_tomorrow} {two_days_after_tomorrow_weekday}"
|
|
}
|
|
|
|
return result
|