from smolagents import Tool from typing import Any, Optional class SimpleTool(Tool): name = "add_numbers" description = "Adds two numbers." inputs = {'a': {'type': 'integer', 'description': 'The first number.'}, 'b': {'type': 'integer', 'description': 'The second number.'}} output_type = "integer" def forward(self, a: int, b: int) -> int: """Adds two numbers. Args: a: The first number. b: The second number. """ return a + b