首先，在你的项目目录中创建my_calculator_tool.py
代码实现
# my_calculator_tool.py
import ast
import operator
import math
from hello_agents import ToolRegistrydef my_calculate(expression: str) -> str:"""简单的数学计算函数"""if not expression.strip():return "计算表达式不能为空"# 支持的基本运算operators = {ast.Add: operator.add, # +ast.Sub: operator.sub, # -ast.Mult: operator.mul, # *ast.Div: operator.truediv, # /}# 支持的基本函数functions = {'sqrt': math.sqrt,'pi': math.pi,}try:node = ast.parse(expression, mode='eval')result = _eval_node(node.body, operators, functions)return str(result)except:return "计算失败，请检查表达式格式"def _eval_node(node, operators, functions):"""简化的表达式求值"""if isinstance(node, ast.Constant):return node.valueelif isinstance(node, ast.BinOp):left = _eval_node(node.left, operators, functions)right = _eval_node(node.right, operators, functions)op = operators.get(type(node.op))return op(left, right)elif isinstance(node, ast.Call):func_name = node.func.idif func_name in functions:args = [_eval_node(arg, operators, functions) for arg in node.args]return functions[func_name](*args)elif isinstance(node, ast.Name):if node.id in functions:return functions[node.id]def create_calculator_registry():"""创建包含计算器的工具注册表"""registry = ToolRegistry()# 注册计算器函数registry.register_function(name="my_calculator",description="简单的数学计算工具，支持基本运算(+,-,*,/)和sqrt函数",func=my_calculate)return registry
代码实现
# test_my_calculator.py
from dotenv import load_dotenv
from my_calculator_tool import create_calculator_registry# 加载环境变量
load_dotenv()def test_calculator_tool():"""测试自定义计算器工具"""# 创建包含计算器的注册表registry = create_calculator_registry()print("🧪 测试自定义计算器工具\n")# 简单测试用例test_cases = ["2 + 3", # 基本加法"10 - 4", # 基本减法"5 * 6", # 基本乘法"15 / 3", # 基本除法"sqrt(16)", # 平方根]for i, expression in enumerate(test_cases, 1):print(f"测试 {i}: {expression}")result = registry.execute_tool("my_calculator", expression)print(f"结果: {result}\n")def test_with_simple_agent():"""测试与SimpleAgent的集成"""from hello_agents import HelloAgentsLLM# 创建LLM客户端llm = HelloAgentsLLM()# 创建包含计算器的注册表registry = create_calculator_registry()print("🤖 与SimpleAgent集成测试:")# 模拟SimpleAgent使用工具的场景user_question = "请帮我计算 sqrt(16) + 2 * 3"print(f"用户问题: {user_question}")# 使用工具计算calc_result = registry.execute_tool("my_calculator", "sqrt(16) + 2 * 3")print(f"计算结果: {calc_result}")# 构建最终回答final_messages = [{"role": "user", "content": f"计算结果是 {calc_result}，请用自然语言回答用户的问题：{user_question}"}]print("\n🎯 SimpleAgent的回答:")response = llm.think(final_messages)for chunk in response:passprint("\n")if __name__ == "__main__":test_calculator_tool()test_with_simple_agent()