40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
from src.tools.tool_can_use.base_tool import BaseTool
|
|
from src.common.logger_manager import get_logger
|
|
from typing import Dict, Any
|
|
from datetime import datetime
|
|
import time
|
|
|
|
logger = get_logger("get_time_date")
|
|
|
|
|
|
class GetCurrentDateTimeTool(BaseTool):
|
|
"""获取当前时间、日期、年份和星期的工具"""
|
|
|
|
name = "get_current_date_time"
|
|
description = "当有人询问或者涉及到具体时间或者日期的时候，必须使用这个工具"
|
|
parameters = {
|
|
"type": "object",
|
|
"properties": {},
|
|
"required": [],
|
|
}
|
|
|
|
async def execute(self, function_args: Dict[str, Any]) -> Dict[str, Any]:
|
|
"""执行获取当前时间、日期、年份和星期
|
|
|
|
Args:
|
|
function_args: 工具参数（此工具不使用）
|
|
|
|
Returns:
|
|
Dict: 工具执行结果
|
|
"""
|
|
current_time = datetime.now().strftime("%H:%M:%S")
|
|
current_date = datetime.now().strftime("%Y-%m-%d")
|
|
current_year = datetime.now().strftime("%Y")
|
|
current_weekday = datetime.now().strftime("%A")
|
|
|
|
return {
|
|
"type": "time_info",
|
|
"id": f"time_info_{time.time()}",
|
|
"content": f"当前时间: {current_time}, 日期: {current_date}, 年份: {current_year}, 星期: {current_weekday}",
|
|
}
|