From c0b511273e4096e8614d2937b8eda932d2165519 Mon Sep 17 00:00:00 2001 From: SnowindMe <1945743455@qq.com> Date: Sat, 19 Apr 2025 18:51:04 +0800 Subject: [PATCH] ruff --- src/do_tool/tool_can_use/base_tool.py | 4 +++- src/do_tool/tool_can_use/compare_numbers_tool.py | 4 ++-- src/do_tool/tool_can_use/get_time_date.py | 2 +- 3 files changed, 6 insertions(+), 4 deletions(-) diff --git a/src/do_tool/tool_can_use/base_tool.py b/src/do_tool/tool_can_use/base_tool.py index d1cb6a8c..10581c1a 100644 --- a/src/do_tool/tool_can_use/base_tool.py +++ b/src/do_tool/tool_can_use/base_tool.py @@ -114,6 +114,7 @@ def get_tool_instance(tool_name: str) -> Optional[BaseTool]: return None return tool_class() + def run_lua_code(lua_code: str): """兼容Lua代码运行(小工具) @@ -124,6 +125,7 @@ def run_lua_code(lua_code: str): _LuaTable: Lua运行时的全局变量 """ from lupa import LuaRuntime + lua = LuaRuntime(unpack_returned_tuples=True) lua.execute(lua_code) - return lua.globals() \ No newline at end of file + return lua.globals() diff --git a/src/do_tool/tool_can_use/compare_numbers_tool.py b/src/do_tool/tool_can_use/compare_numbers_tool.py index 4e4ce96f..b2f5dd18 100644 --- a/src/do_tool/tool_can_use/compare_numbers_tool.py +++ b/src/do_tool/tool_can_use/compare_numbers_tool.py @@ -1,4 +1,4 @@ -from src.do_tool.tool_can_use.base_tool import BaseTool,run_lua_code +from src.do_tool.tool_can_use.base_tool import BaseTool, run_lua_code from src.common.logger import get_module_logger from typing import Dict, Any @@ -41,7 +41,7 @@ class CompareNumbersTool(BaseTool): """ CompareNumbers = run_lua_code(lua_code).CompareNumbers result = CompareNumbers(num1, num2) - + return {"name": self.name, "content": result} except Exception as e: logger.error(f"比较数字失败: {str(e)}") diff --git a/src/do_tool/tool_can_use/get_time_date.py b/src/do_tool/tool_can_use/get_time_date.py index b3153e08..b54210fc 100644 --- a/src/do_tool/tool_can_use/get_time_date.py +++ b/src/do_tool/tool_can_use/get_time_date.py @@ -1,4 +1,4 @@ -from src.do_tool.tool_can_use.base_tool import BaseTool,run_lua_code +from src.do_tool.tool_can_use.base_tool import BaseTool, run_lua_code from src.common.logger import get_module_logger from typing import Dict, Any