| """Tool: calculadora aritmética segura. |
| |
| Avalia expressões matemáticas básicas (+, -, *, /, **, parênteses). |
| Usada principalmente pelo EducIAção em exemplos de matemática. |
| """ |
| import ast |
| import operator |
|
|
| |
| _OPS: dict = { |
| ast.Add: operator.add, |
| ast.Sub: operator.sub, |
| ast.Mult: operator.mul, |
| ast.Div: operator.truediv, |
| ast.Pow: operator.pow, |
| ast.Mod: operator.mod, |
| ast.FloorDiv: operator.floordiv, |
| ast.USub: operator.neg, |
| ast.UAdd: operator.pos, |
| } |
|
|
|
|
| def _safe_eval(node: ast.AST) -> float: |
| """Avalia AST de forma segura, sem chamar eval() bruto.""" |
| if isinstance(node, ast.Constant) and isinstance(node.value, (int, float)): |
| return node.value |
| if isinstance(node, ast.BinOp): |
| op_type = type(node.op) |
| if op_type not in _OPS: |
| raise ValueError(f"Operador {op_type.__name__} não permitido") |
| return _OPS[op_type](_safe_eval(node.left), _safe_eval(node.right)) |
| if isinstance(node, ast.UnaryOp): |
| op_type = type(node.op) |
| if op_type not in _OPS: |
| raise ValueError(f"Operador unário {op_type.__name__} não permitido") |
| return _OPS[op_type](_safe_eval(node.operand)) |
| raise ValueError(f"Nó AST não permitido: {type(node).__name__}") |
|
|
|
|
| def run_calculator(expression: str) -> str: |
| """Avalia uma expressão matemática e retorna string formatada.""" |
| try: |
| tree = ast.parse(expression, mode="eval") |
| result = _safe_eval(tree.body) |
| |
| if isinstance(result, float) and result.is_integer(): |
| result = int(result) |
| elif isinstance(result, float): |
| result = round(result, 6) |
| return f"{expression} = {result}" |
| except (SyntaxError, ValueError, ZeroDivisionError) as exc: |
| return f"ERRO ao calcular '{expression}': {exc}" |
|
|
|
|
| calculator_tool: dict = { |
| "type": "function", |
| "function": { |
| "name": "calculator", |
| "description": ( |
| "Avalia uma expressão matemática (números, +, -, *, /, **, parênteses). " |
| "Use SEMPRE que precisar fazer conta exata. Não invente resultados." |
| ), |
| "parameters": { |
| "type": "object", |
| "properties": { |
| "expression": { |
| "type": "string", |
| "description": ( |
| "Expressão matemática em formato Python. " |
| "Exemplos: '3/4 * 5/2', '2 ** 10', '(15 + 7) * 3'" |
| ), |
| } |
| }, |
| "required": ["expression"], |
| "additionalProperties": False, |
| }, |
| }, |
| } |
|
|
