from app.tools.base import BaseTool, ToolResult


_ADD_DESCRIPTION = """计算两个整数的和。"""


class Add(BaseTool):
    """加法计算工具"""

    name: str = "add"
    description: str = _ADD_DESCRIPTION
    parameters: dict = {
        "type": "object",
        "properties": {
            "a": {
                "type": "integer",
                "description": "第一个加数",
            },
            "b": {
                "type": "integer",
                "description": "第二个加数",
            },
        },
        "required": ["a", "b"],
    }

    async def execute(self, a: int, b: int) -> ToolResult:
        """执行加法计算"""
        result = a + b
        return self.success_response({"result": result})