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 db43df1c0..8d616f85c 100644 --- a/src/do_tool/tool_can_use/compare_numbers_tool.py +++ b/src/do_tool/tool_can_use/compare_numbers_tool.py @@ -4,56 +4,47 @@ from typing import Dict, Any logger = get_module_logger("compare_numbers_tool") + class CompareNumbersTool(BaseTool): """比较两个数大小的工具""" + name = "compare_numbers" description = "比较两个数的大小,返回较大的数" parameters = { "type": "object", "properties": { - "num1": { - "type": "number", - "description": "第一个数字" - }, - "num2": { - "type": "number", - "description": "第二个数字" - } + "num1": {"type": "number", "description": "第一个数字"}, + "num2": {"type": "number", "description": "第二个数字"}, }, - "required": ["num1", "num2"] + "required": ["num1", "num2"], } - + async def execute(self, function_args: Dict[str, Any], message_txt: str = "") -> Dict[str, Any]: """执行比较两个数的大小 - + Args: function_args: 工具参数 message_txt: 原始消息文本 - + Returns: Dict: 工具执行结果 """ try: num1 = function_args.get("num1") num2 = function_args.get("num2") - + if num1 > num2: result = f"{num1} 大于 {num2}" elif num1 < num2: result = f"{num1} 小于 {num2}" else: result = f"{num1} 等于 {num2}" - - return { - "name": self.name, - "content": result - } + + return {"name": self.name, "content": result} except Exception as e: logger.error(f"比较数字失败: {str(e)}") - return { - "name": self.name, - "content": f"比较数字失败: {str(e)}" - } + return {"name": self.name, "content": f"比较数字失败: {str(e)}"} + # 注册工具 -register_tool(CompareNumbersTool) \ No newline at end of file +register_tool(CompareNumbersTool) diff --git a/src/do_tool/tool_use.py b/src/do_tool/tool_use.py index 237dc1409..dd1ca7044 100644 --- a/src/do_tool/tool_use.py +++ b/src/do_tool/tool_use.py @@ -104,7 +104,7 @@ class ToolUser: # 定义可用工具 tools = self._define_tools() - + # 使用llm_model_tool发送带工具定义的请求 payload = { "model": self.llm_model_tool.model_name,