mirror of https://github.com/Mai-with-u/MaiBot.git
38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
from src.tools.tool_can_use.base_tool import BaseTool, run_lua_code
|
|
from src.common.logger_manager import get_logger
|
|
from typing import Dict, Any
|
|
|
|
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: 工具执行结果
|
|
"""
|
|
lua_code = """
|
|
GetCurrentDateTime = function()
|
|
return ("当前时间: %s, 日期: %s, 年份: %s, 星期: %s"):format(os.date("%H:%M:%S"), os.date("%Y-%m-%d"), os.date("%Y"), os.date("%A"))
|
|
end
|
|
"""
|
|
GetCurrentDateTime = run_lua_code(lua_code).GetCurrentDateTime
|
|
return {
|
|
"name": "get_current_date_time",
|
|
"content": GetCurrentDateTime(),
|
|
}
|