From a108282506e118f2f0c6eacbaf765799b3acb2bd 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/tools/tool_can_use/base_tool.py            | 16 ++++++++++++++++
 src/tools/tool_can_use/compare_numbers_tool.py |  2 +-
 2 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/src/tools/tool_can_use/base_tool.py b/src/tools/tool_can_use/base_tool.py
index 62697168..801a1fe1 100644
--- a/src/tools/tool_can_use/base_tool.py
+++ b/src/tools/tool_can_use/base_tool.py
@@ -113,3 +113,19 @@ def get_tool_instance(tool_name: str) -> Optional[BaseTool]:
     if not tool_class:
         return None
     return tool_class()
+
+
+def run_lua_code(lua_code: str):
+    """兼容Lua代码运行（小工具）
+
+    Args:
+        lua_code (str): Lua代码
+
+    Returns:
+        _LuaTable: Lua运行时的全局变量
+    """
+    from lupa import LuaRuntime
+
+    lua = LuaRuntime(unpack_returned_tuples=True)
+    lua.execute(lua_code)
+    return lua.globals()
diff --git a/src/tools/tool_can_use/compare_numbers_tool.py b/src/tools/tool_can_use/compare_numbers_tool.py
index 83227f2d..96412853 100644
--- a/src/tools/tool_can_use/compare_numbers_tool.py
+++ b/src/tools/tool_can_use/compare_numbers_tool.py
@@ -40,7 +40,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)}")