From 9b96054c3cd10fa8d2b62c2fffe5579ca61d1ae5 Mon Sep 17 00:00:00 2001 From: SnowindMe <1945743455@qq.com> Date: Sat, 19 Apr 2025 18:47:38 +0800 Subject: [PATCH] =?UTF-8?q?ai=E5=BB=BA=E8=AE=AE=E4=BF=AE=E6=94=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/do_tool/tool_can_use/compare_numbers_tool.py | 2 ++ src/do_tool/tool_can_use/letter_count_tool.py | 4 ++-- 2 files changed, 4 insertions(+), 2 deletions(-) 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 ab96eb18..4e4ce96f 100644 --- a/src/do_tool/tool_can_use/compare_numbers_tool.py +++ b/src/do_tool/tool_can_use/compare_numbers_tool.py @@ -32,6 +32,8 @@ class CompareNumbersTool(BaseTool): try: num1 = function_args.get("num1") num2 = function_args.get("num2") + if not (isinstance(num1, (int, float)) and isinstance(num2, (int, float))): + raise ValueError("参数'num1'和'num2'必须为数字") lua_code = """ function CompareNumbers(a, b) return a .. (a > b and " 大于 " or a < b and " 小于 " or " 等于 ") .. b diff --git a/src/do_tool/tool_can_use/letter_count_tool.py b/src/do_tool/tool_can_use/letter_count_tool.py index 999ac53c..9731606a 100644 --- a/src/do_tool/tool_can_use/letter_count_tool.py +++ b/src/do_tool/tool_can_use/letter_count_tool.py @@ -36,8 +36,8 @@ class LetterCountTool(BaseTool): try: word = function_args.get("word") letter = function_args.get("letter") - if re.match(r"^[a-zA-Z]+$", letter) is None: - raise ValueError("请输入英文字母") + if re.match(r"^[a-zA-Z]$", letter) is None: + raise ValueError("请输入单个英文字母") lua_code = """ function LetterCount(inputStr, targetLetter) local lower = (inputStr:gsub("[^"..targetLetter:lower().."]", "")):len()